diff options
Diffstat (limited to 'EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c')
| -rw-r--r-- | EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c b/EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c index 8bbfd8049b..51240bbdc7 100644 --- a/EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c +++ b/EdkCompatibilityPkg/Compatibility/SmmBaseOnSmmBase2Thunk/SmmBaseOnSmmBase2Thunk.c @@ -75,6 +75,7 @@ SmmBaseHelperService ( UINTN DataSize;
mCommunicationData.FunctionData.Status = EFI_UNSUPPORTED;
+ mCommunicationData.FunctionData.SmmBaseImageHandle = mSmmBaseHandle;
if ((mCommunicationData.FunctionData.Function != SmmBaseFunctionCommunicate) && IsInSmm()) {
///
@@ -421,6 +422,8 @@ SmmBaseThunkMain ( EFI_STATUS Status;
EFI_EVENT Event;
+ mSmmBaseHandle = ImageHandle;
+
//
// Assume only one instance of SMM Base2 Protocol in the system
// Locate SMM Base2 Protocol
|