X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=Nt32Pkg%2FLibrary%2FNt32PeCoffLoaderLib%2FNt32PeCoffLoader.c;h=3d5353b4238082fc981734edaf5cc97dd62f29ad;hp=8eaa2aba432bbd929e07f62830c832c19f0842c8;hb=b0d803fe3e5c55a9a0c75bc90ccf40ebbbc9ffa3;hpb=f25695723bfe21bf664589ae086a0c472d7fec80

diff --git a/Nt32Pkg/Library/Nt32PeCoffLoaderLib/Nt32PeCoffLoader.c b/Nt32Pkg/Library/Nt32PeCoffLoaderLib/Nt32PeCoffLoader.c
index 8eaa2aba43..3d5353b423 100644
--- a/Nt32Pkg/Library/Nt32PeCoffLoaderLib/Nt32PeCoffLoader.c
+++ b/Nt32Pkg/Library/Nt32PeCoffLoaderLib/Nt32PeCoffLoader.c
@@ -21,35 +21,30 @@ Abstract:
 --*/
 
 #include <PiPei.h>
+#include <Library/DebugLib.h>
 #include <Guid/PeiPeCoffLoader.h>
-#include <Library/PeCoffLoaderLib.h>
+#include <Library/PeCoffLoaderLib.h>  
+#include <Library/PeiServicesLib.h>
 
 
-EFI_PEI_PE_COFF_LOADER_PROTOCOL  *mPeiEfiPeiPeCoffLoader;
+EFI_PEI_PE_COFF_LOADER_PROTOCOL  *mPeiEfiPeiPeCoffLoader = NULL;
 
-EFI_STATUS
-EFIAPI
-PeCoffLoaderConstructor (
-  IN EFI_FFS_FILE_HEADER      *FfsHeader,
-  IN EFI_PEI_SERVICES          **PeiServices
-  )
-{
-  EFI_STATUS  Status;
-
-  Status = (*PeiServices)->LocatePpi (
-                            PeiServices,
-                            &gEfiPeiPeCoffLoaderGuid,
-                            0,
-                            NULL,
-                            &mPeiEfiPeiPeCoffLoader
-                            );
-  return Status;
-}
 
 EFI_PEI_PE_COFF_LOADER_PROTOCOL *
 EFIAPI
 GetPeCoffLoaderProtocol (
   )
 {
+  EFI_STATUS Status;
+  
+  if (mPeiEfiPeiPeCoffLoader == NULL) {
+    Status = PeiServicesLocatePpi(
+                              &gEfiPeiPeCoffLoaderGuid,
+                              0,
+                              NULL,
+                              (VOID **) &mPeiEfiPeiPeCoffLoader
+                              );
+    ASSERT_EFI_ERROR (Status);
+   }
   return mPeiEfiPeiPeCoffLoader;
 }