\r
**/\r
\r
-#include "InternalPeiServicesTablePointer.h"\r
+#include <PiPei.h>\r
+\r
+#include <Library/BaseLib.h>\r
+#include <Library/PeiServicesTablePointerLib.h>\r
+#include <Library/DebugLib.h>\r
\r
/**\r
\r
IA32_DESCRIPTOR Idtr;\r
\r
AsmReadIdtr (&Idtr);\r
- PeiServices = (EFI_PEI_SERVICES **) (*(UINTN*)(Idtr.Base - 4));\r
+ PeiServices = (EFI_PEI_SERVICES **) (*(UINTN*)(Idtr.Base - sizeof (UINTN)));\r
ASSERT (PeiServices != NULL);\r
return PeiServices;\r
}\r
IA32_DESCRIPTOR Idtr;\r
\r
AsmReadIdtr (&Idtr);\r
- (*(UINTN*)(Idtr.Base - 4)) = (UINTN)PeiServicesTablePointer;\r
+ (*(UINTN*)(Idtr.Base - sizeof (UINTN))) = (UINTN)PeiServicesTablePointer;\r
}\r
\r
-/**\r
- After memory initialization in PEI phase, the IDT table in temporary memory should \r
- be migrated to memory, and the address of PeiServicesPointer also need to be updated \r
- immediately preceding the new IDT table.\r
- \r
- @param PeiServices The address of PeiServices pointer.\r
-**/\r
-VOID\r
-EFIAPI\r
-MigrateIdtTable (\r
- IN EFI_PEI_SERVICES **PeiServices\r
- )\r
-{\r
- UINTN Size;\r
- VOID *NewBase;\r
- EFI_STATUS Status;\r
- IA32_DESCRIPTOR Idtr;\r
- \r
- AsmReadIdtr (&Idtr);\r
- \r
- Size = sizeof(UINTN) + (Idtr.Limit + 1); \r
- \r
- Status = PeiServicesAllocatePool (Size, &NewBase);\r
- ASSERT_EFI_ERROR (Status);\r
- \r
- CopyMem ((VOID*)((UINTN)NewBase + sizeof(UINTN)), (VOID*)Idtr.Base, (Idtr.Limit + 1));\r
- \r
- Idtr.Base = (UINTN)NewBase + sizeof(UINTN);\r
- AsmWriteIdtr (&Idtr);\r
- SetPeiServicesTablePointer(PeiServices); \r
-} \r
\r