InOsEmuPkg: Make XIP work properly

[mirror_edk2.git] / InOsEmuPkg / Library / PeiEmuPeCoffExtraActionLib / PeiEmuPeCoffExtraActionLib.c

diff --git a/InOsEmuPkg/Library/PeiEmuPeCoffExtraActionLib/PeiEmuPeCoffExtraActionLib.c b/InOsEmuPkg/Library/PeiEmuPeCoffExtraActionLib/PeiEmuPeCoffExtraActionLib.c
index b88bd009fe63acf62a65df1b7a6debe7f35a051c..b398b53e512f8cdb4929dd98916a4339402930ba 100644 (file)
--- a/InOsEmuPkg/Library/PeiEmuPeCoffExtraActionLib/PeiEmuPeCoffExtraActionLib.c
+++ b/InOsEmuPkg/Library/PeiEmuPeCoffExtraActionLib/PeiEmuPeCoffExtraActionLib.c
@@ -23,6 +23,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Library/DebugLib.h>\r
 #include <Library/BaseLib.h>\r
 #include <Library/PeCoffExtraActionLib.h>\r
+#include <Library/EmuMagicPageLib.h>\r
 \r
 //\r
 // Cache of UnixThunk protocol \r
@@ -56,7 +57,7 @@ EmuPeCoffGetThunkStucture (
               );\r
   ASSERT_EFI_ERROR (Status);\r
 \r
-  mThunk  = (EMU_THUNK_PROTOCOL *) ThunkPpi->Thunk ();\r
+  EMU_MAGIC_PAGE()->Thunk = (EMU_THUNK_PROTOCOL *) ThunkPpi->Thunk ();\r
 \r
   return EFI_SUCCESS;\r
 }\r
@@ -76,10 +77,10 @@ PeCoffLoaderRelocateImageExtraAction (
   IN OUT PE_COFF_LOADER_IMAGE_CONTEXT  *ImageContext\r
   )\r
 {\r
-  if (mThunk == NULL) {\r
+  if (EMU_MAGIC_PAGE()->Thunk == NULL) {\r
     EmuPeCoffGetThunkStucture ();\r
   }\r
-    mThunk->PeCoffRelocateImageExtraAction (ImageContext);\r
+    EMU_MAGIC_PAGE()->Thunk->PeCoffRelocateImageExtraAction (ImageContext);\r
   }\r
 \r
 \r
@@ -99,8 +100,8 @@ PeCoffLoaderUnloadImageExtraAction (
   IN OUT PE_COFF_LOADER_IMAGE_CONTEXT  *ImageContext\r
   )\r
 {\r
-  if (mThunk == NULL) {\r
+  if (EMU_MAGIC_PAGE()->Thunk == NULL) {\r
     EmuPeCoffGetThunkStucture ();\r
   }\r
-  mThunk->PeCoffUnloadImageExtraAction (ImageContext);\r
+  EMU_MAGIC_PAGE()->Thunk->PeCoffUnloadImageExtraAction (ImageContext);\r
 }\r