X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=EdkModulePkg%2FCore%2FDxeIplX64Peim%2FDxeLoadX64.c;h=d01daafbe2deace2d88a7e154f2b06f6f6d81b1a;hp=4015222036eee951fa1f98689c90a58a6495d85f;hb=e29d70965dbc1e5ab3ed744e05413304ae0f059a;hpb=f47629a4822eb80ed8f5e0ce79f12e86b54bf5d5

diff --git a/EdkModulePkg/Core/DxeIplX64Peim/DxeLoadX64.c b/EdkModulePkg/Core/DxeIplX64Peim/DxeLoadX64.c
index 4015222036..d01daafbe2 100644
--- a/EdkModulePkg/Core/DxeIplX64Peim/DxeLoadX64.c
+++ b/EdkModulePkg/Core/DxeIplX64Peim/DxeLoadX64.c
@@ -62,17 +62,17 @@ static EFI_PEI_PPI_DESCRIPTOR     mPpiSignal = {
   NULL
 };
 
-DECOMPRESS_LIBRARY  gEfiDecompress = {
+GLOBAL_REMOVE_IF_UNREFERENCED DECOMPRESS_LIBRARY  gEfiDecompress = {
   UefiDecompressGetInfo,
   UefiDecompress
 };
 
-DECOMPRESS_LIBRARY  gTianoDecompress = {
+GLOBAL_REMOVE_IF_UNREFERENCED DECOMPRESS_LIBRARY  gTianoDecompress = {
   TianoDecompressGetInfo,
   TianoDecompress
 };
 
-DECOMPRESS_LIBRARY  gCustomDecompress = {
+GLOBAL_REMOVE_IF_UNREFERENCED DECOMPRESS_LIBRARY  gCustomDecompress = {
   CustomDecompressGetInfo,
   CustomDecompress
 };
@@ -842,14 +842,24 @@ Returns:
 
         switch (CompressionSection->CompressionType) {
         case EFI_STANDARD_COMPRESSION:
-          DecompressLibrary = &gTianoDecompress;
+          if (FeaturePcdGet (PcdDxeIplSupportTianoDecompress)) {
+            DecompressLibrary = &gTianoDecompress;
+          } else {
+            ASSERT (FALSE);
+            return EFI_NOT_FOUND;
+          }
           break;
 
         case EFI_CUSTOMIZED_COMPRESSION:
           //
           // Load user customized compression protocol.
           //
-          DecompressLibrary = &gCustomDecompress;
+          if (FeaturePcdGet (PcdDxeIplSupportCustomDecompress)) {
+            DecompressLibrary = &gCustomDecompress;
+          } else {
+            ASSERT (FALSE);
+            return EFI_NOT_FOUND;
+          }
           break;
 
         case EFI_NOT_COMPRESSED: