1 files changed, 13 insertions, 0 deletions
diff --git a/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.c b/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.c
index 2433c76a8c..8befbae1de 100644
--- a/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.c
+++ b/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleLib.c
@@ -11,6 +11,7 @@
   performs basic validation.
 
   Copyright (c) 2016 - 2019, Intel Corporation. All rights reserved.<BR>
+  Copyright (c) 2024, Ampere Computing LLC. All rights reserved.<BR>
   SPDX-License-Identifier: BSD-2-Clause-Patent
 
 **/
@@ -48,6 +49,8 @@ EFI_EVENT  mDxeCapsuleLibEndOfDxeEvent = NULL;
 
 EDKII_FIRMWARE_MANAGEMENT_PROGRESS_PROTOCOL  *mFmpProgress = NULL;
 
+BOOLEAN  mDxeCapsuleLibReadyToBootEvent = FALSE;
+
 /**
   Initialize capsule related variables.
 **/
@@ -1402,6 +1405,16 @@ IsNestedFmpCapsule (
       }
     }
   } else {
+    if (mDxeCapsuleLibReadyToBootEvent) {
+      //
+      // The ESRT table (mEsrtTable) in the Configuration Table would be located
+      // at the ReadyToBoot event if it exists. Hence, it should return here to
+      // avoid a crash due to calling gBS->LocateProtocol () at runtime in case
+      // there is no ERST table installed.
+      //
+      return FALSE;
+    }
+
     //
     // Check ESRT protocol
     //