From 68443c61b4e51bdef20a60692bf1c417c5e06fb8 Mon Sep 17 00:00:00 2001
From: qhuang8 <qhuang8@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Wed, 11 Feb 2009 07:42:24 +0000
Subject: Fix entry point prototype error.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7505 6f19259b-4bc3-4df7-8a09-765794883524
---
 Nt32Pkg/WinNtAutoScanPei/WinNtAutoScan.c      | 35 +++++++++++++--------------
 Nt32Pkg/WinNtAutoScanPei/WinNtAutoScanPei.inf |  2 +-
 2 files changed, 18 insertions(+), 19 deletions(-)

(limited to 'Nt32Pkg/WinNtAutoScanPei')

diff --git a/Nt32Pkg/WinNtAutoScanPei/WinNtAutoScan.c b/Nt32Pkg/WinNtAutoScanPei/WinNtAutoScan.c
index 2c2b3025b0..9591c39fcf 100644
--- a/Nt32Pkg/WinNtAutoScanPei/WinNtAutoScan.c
+++ b/Nt32Pkg/WinNtAutoScanPei/WinNtAutoScan.c
@@ -35,12 +35,13 @@ Revision History
 #include <Library/DebugLib.h>
 #include <Library/PeimEntryPoint.h>
 #include <Library/HobLib.h>
+#include <Library/PeiServicesLib.h>
 
 EFI_STATUS
 EFIAPI
 PeimInitializeWinNtAutoScan (
-  IN EFI_FFS_FILE_HEADER       *FfsHeader,
-  IN EFI_PEI_SERVICES          **PeiServices
+  IN       EFI_PEI_FILE_HANDLE       FileHandle,
+  IN CONST EFI_PEI_SERVICES          **PeiServices
   )
 /*++
 
@@ -72,25 +73,23 @@ Returns:
   //
   // Get the PEI NT Autoscan PPI
   //
-  Status = (**PeiServices).LocatePpi (
-                            (const EFI_PEI_SERVICES **)PeiServices,
-                            &gPeiNtAutoScanPpiGuid, // GUID
-                            0,                      // INSTANCE
-                            &PpiDescriptor,         // EFI_PEI_PPI_DESCRIPTOR
-                            (VOID**)&PeiNtService           // PPI
-                            );
+  Status = PeiServicesLocatePpi (
+             &gPeiNtAutoScanPpiGuid, // GUID
+             0,                      // INSTANCE
+             &PpiDescriptor,         // EFI_PEI_PPI_DESCRIPTOR
+             (VOID**)&PeiNtService           // PPI
+             );
   ASSERT_EFI_ERROR (Status);
 
   //
   // Get the Memory Test PPI
   //
-  Status = (**PeiServices).LocatePpi (
-                            (const EFI_PEI_SERVICES **)PeiServices,
-                            &gPeiBaseMemoryTestPpiGuid,
-                            0,
-                            NULL,
-                            (VOID**)&MemoryTestPpi
-                            );
+  Status = PeiServicesLocatePpi (
+             &gPeiBaseMemoryTestPpiGuid,
+             0,
+             NULL,
+            (VOID**)&MemoryTestPpi
+            );
   ASSERT_EFI_ERROR (Status);
 
   Index = 0;
@@ -112,7 +111,7 @@ Returns:
         // For the first area register it as PEI tested memory
         //
         Status = MemoryTestPpi->BaseMemoryTest (
-                                  PeiServices,
+                                  (EFI_PEI_SERVICES **) PeiServices,
                                   MemoryTestPpi,
                                   MemoryBase,
                                   MemorySize,
@@ -124,7 +123,7 @@ Returns:
         //
         // Register the "tested" memory with the PEI Core
         //
-        Status = (**PeiServices).InstallPeiMemory ((const EFI_PEI_SERVICES **)PeiServices, MemoryBase, MemorySize);
+        Status = PeiServicesInstallPeiMemory (MemoryBase, MemorySize);
         ASSERT_EFI_ERROR (Status);
 
         Attributes |= EFI_RESOURCE_ATTRIBUTE_TESTED;
diff --git a/Nt32Pkg/WinNtAutoScanPei/WinNtAutoScanPei.inf b/Nt32Pkg/WinNtAutoScanPei/WinNtAutoScanPei.inf
index 078293142d..c04d062f72 100644
--- a/Nt32Pkg/WinNtAutoScanPei/WinNtAutoScanPei.inf
+++ b/Nt32Pkg/WinNtAutoScanPei/WinNtAutoScanPei.inf
@@ -44,7 +44,7 @@
   HobLib
   PeimEntryPoint
   DebugLib
-
+  PeiServicesLib
 
 [Ppis]
   gPeiBaseMemoryTestPpiGuid                     # PPI ALWAYS_CONSUMED
-- 
cgit v1.2.3