1 files changed, 153 insertions, 0 deletions
diff --git a/EDK/Foundation/Library/Pei/PeiLib/ia32/PeiServicePointer.c b/EDK/Foundation/Library/Pei/PeiLib/ia32/PeiServicePointer.c
new file mode 100644
index 0000000..162415c
--- /dev/null
+++ b/EDK/Foundation/Library/Pei/PeiLib/ia32/PeiServicePointer.c
@@ -0,0 +1,153 @@
+/*++
+
+Copyright (c) 2004 - 2007, Intel Corporation                                                         
+All rights reserved. This program and the accompanying materials                          
+are licensed and made available under the terms and conditions of the BSD License         
+which accompanies this distribution.  The full text of the license may be found at        
+http://opensource.org/licenses/bsd-license.php                                            
+                                                                                          
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,                     
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.             
+
+Module Name:
+
+    PeiServicePointer.c
+
+Abstract:
+
+--*/
+
+#include "Tiano.h"
+#include "PeiApi.h"
+#include "PeiLib.h"
+
+//;;## ...AMI_ADD FILE... Support PI1.x
+#if (PI_SPECIFICATION_VERSION >= 0x00010000)
+
+#ifdef EFI_NT_EMULATOR
+EFI_PEI_SERVICES  **gPeiServices;
+#endif
+
+
+VOID
+SetPeiServicesTablePointer (
+  IN EFI_PEI_SERVICES  **PeiServices
+  )
+/*++
+
+Routine Description:
+
+  Save PeiService pointer so that it can be retrieved anywhere.
+
+Arguments:
+
+  PeiServices     - The direct pointer to PeiServiceTable.
+  PhyscialAddress - The physcial address of variable PeiServices.
+  
+Returns:
+  NONE
+  
+--*/          
+{
+
+#ifdef EFI_NT_EMULATOR
+
+  //
+  // For NT32, set EFI_PEI_SERVICES** to global variable.
+  //
+  gPeiServices = PeiServices;
+#else
+
+  //
+  // For X86 processor,the EFI_PEI_SERVICES** is stored in the 
+  // 4 bytes immediately preceding the Interrupt Descriptor Table.
+  //
+  UINTN IdtBaseAddress;
+  IdtBaseAddress = (UINTN)ReadIdtBase();
+  *(UINTN*)(IdtBaseAddress - 4) = (UINTN)PeiServices;
+
+#endif  
+}
+
+
+EFI_PEI_SERVICES **
+GetPeiServicesTablePointer ( 
+  VOID
+  )
+/*++
+
+Routine Description:
+
+  Get PeiService pointer.
+
+Arguments:
+
+  NONE.
+  
+Returns:
+  The direct pointer to PeiServiceTable.
+  
+--*/          
+{
+  EFI_PEI_SERVICES  **PeiServices;
+
+#ifdef EFI_NT_EMULATOR
+
+  //
+  // For NT32, set EFI_PEI_SERVICES** to global variable.
+  //
+  PeiServices = gPeiServices;
+#else
+
+  //
+  // For X86 processor,the EFI_PEI_SERVICES** is stored in the 
+  // 4 bytes immediately preceding the Interrupt Descriptor Table.
+  //
+  UINTN IdtBaseAddress;
+  IdtBaseAddress = (UINTN)ReadIdtBase();
+  PeiServices = (EFI_PEI_SERVICES **)(UINTN)(*(UINTN*)(IdtBaseAddress - 4));
+#endif
+  return PeiServices;
+}
+
+
+VOID
+MigrateIdtTable (
+  IN EFI_PEI_SERVICES  **PeiServices
+  )
+/*++
+
+Routine Description:
+
+  Migrate IDT from CAR to real memory where preceded with 4 bytes for
+  storing PeiService pointer.
+
+Arguments:
+
+  PeiServices   - The direct pointer to PeiServiceTable.
+  
+Returns:
+
+  NONE.
+  
+--*/            
+{
+#ifndef EFI_NT_EMULATOR
+  UINT16          IdtEntrySize;
+  UINTN           OldIdtBase;
+  UINTN           Size;
+  VOID            *NewIdtBase;
+  EFI_STATUS      Status;
+  
+  IdtEntrySize    = ReadIdtLimit();
+  OldIdtBase      = ReadIdtBase();
+  Size = sizeof(PEI_IDT_TABLE) + (IdtEntrySize + 1);  
+  Status = (*PeiServices)->AllocatePool (PeiServices, Size, &NewIdtBase);
+  ASSERT_PEI_ERROR (PeiServices, Status);
+  (*PeiServices)->CopyMem ((VOID*)((UINTN)NewIdtBase + sizeof(PEI_IDT_TABLE)), (VOID*)OldIdtBase, (IdtEntrySize + 1));
+  SetIdtBase(((UINTN)NewIdtBase + sizeof(PEI_IDT_TABLE)), IdtEntrySize);
+  SetPeiServicesTablePointer(PeiServices);
+#endif
+}
+
+#endif