Update copyright for files modified in this year

[mirror_edk2.git] / MdePkg / Library / PeiServicesTablePointerLibIdt / PeiServicesTablePointer.c

diff --git a/MdePkg/Library/PeiServicesTablePointerLibIdt/PeiServicesTablePointer.c b/MdePkg/Library/PeiServicesTablePointerLibIdt/PeiServicesTablePointer.c
index a75ddb140055f10943cebb781ec8acf7d23d1c11..a5304e8e3432930a72872c4f56f019c51b41a7e6 100644 (file)
--- a/MdePkg/Library/PeiServicesTablePointerLibIdt/PeiServicesTablePointer.c
+++ b/MdePkg/Library/PeiServicesTablePointerLibIdt/PeiServicesTablePointer.c
@@ -1,7 +1,10 @@
 /** @file\r
-  PEI Services Table Pointer Library for IA-32 and X64.\r
+  PEI Services Table Pointer Library for IA-32 and x64.\r
 \r
-  Copyright (c) 2006 - 2007, Intel Corporation.<BR>\r
+  According to PI specification, the peiservice pointer is stored prior at IDT\r
+  table in IA32 and x64 architecture.\r
+  \r
+  Copyright (c) 2006 - 2008, Intel Corporation.<BR>\r
   All rights reserved. This program and the accompanying materials                          \r
   are licensed and made available under the terms and conditions of the BSD License         \r
   which accompanies this distribution.  The full text of the license may be found at        \r
@@ -12,31 +15,63 @@
 \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
+  Retrieves the cached value of the PEI Services Table pointer.\r
+\r
+  Returns the cached value of the PEI Services Table pointer in a CPU specific manner \r
+  as specified in the CPU binding section of the Platform Initialization Pre-EFI \r
+  Initialization Core Interface Specification.\r
   \r
-  The function returns the pointer to PeiServicee following\r
-  PI1.0.\r
-  \r
-  For IA32, the four-bytes field immediately prior to new IDT\r
-  base addres is used to save the EFI_PEI_SERVICES**.\r
-  For x64, the eight-bytes field immediately prior to new IDT\r
-  base addres is used to save the EFI_PEI_SERVICES**\r
-  @retval  The pointer to PeiServices.\r
+  If the cached PEI Services Table pointer is NULL, then ASSERT().\r
+\r
+  @return  The pointer to PeiServices.\r
 \r
 **/\r
-EFI_PEI_SERVICES **\r
+CONST EFI_PEI_SERVICES **\r
 EFIAPI\r
 GetPeiServicesTablePointer (\r
   VOID\r
   )\r
 {\r
-  EFI_PEI_SERVICES  **PeiServices;\r
-\r
-  PeiServices = (EFI_PEI_SERVICES **) AsmPeiSevicesTablePointer ();\r
+  CONST EFI_PEI_SERVICES  **PeiServices;\r
+  IA32_DESCRIPTOR   Idtr;\r
+  \r
+  AsmReadIdtr (&Idtr);\r
+  PeiServices = (CONST EFI_PEI_SERVICES **) (*(UINTN*)(Idtr.Base - sizeof (UINTN)));\r
   ASSERT (PeiServices != NULL);\r
   return PeiServices;\r
 }\r
 \r
+/**\r
+  Caches a pointer PEI Services Table. \r
+ \r
+  Caches the pointer to the PEI Services Table specified by PeiServicesTablePointer \r
+  in a CPU specific manner as specified in the CPU binding section of the Platform Initialization \r
+  Pre-EFI Initialization Core Interface Specification. \r
+  The function set the pointer of PEI services immediately preceding the IDT table\r
+  according to PI specification.\r
+  \r
+  If PeiServicesTablePointer is NULL, then ASSERT().\r
+  \r
+  @param    PeiServicesTablePointer   The address of PeiServices pointer.\r
+**/\r
+VOID\r
+EFIAPI\r
+SetPeiServicesTablePointer (\r
+  IN CONST EFI_PEI_SERVICES ** PeiServicesTablePointer\r
+  )\r
+{\r
+  IA32_DESCRIPTOR   Idtr;\r
+  \r
+  ASSERT (PeiServicesTablePointer != NULL);\r
+  AsmReadIdtr (&Idtr);\r
+  (*(UINTN*)(Idtr.Base - sizeof (UINTN))) = (UINTN)PeiServicesTablePointer;\r
+}\r
+\r
+\r