1 files changed, 7 insertions, 0 deletions
diff --git a/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c b/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c
index 0858d8f4d7..db36a0ad3c 100644
--- a/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c
+++ b/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c
@@ -97,6 +97,11 @@ UINTN mNumberOfCpus = 1;
 BOOLEAN mSmmReadyToLock = FALSE;
 
 //
+// S3 boot flag
+//
+BOOLEAN mSmmS3Flag = FALSE;
+
+//
 // Global used to cache PCD for SMM Code Access Check enable
 //
 BOOLEAN                  mSmmCodeAccessCheckEnable = FALSE;
@@ -485,6 +490,8 @@ SmmRestoreCpu (
 
   DEBUG ((EFI_D_INFO, "SmmRestoreCpu()\n"));
 
+  mSmmS3Flag = TRUE;
+
   InitializeSpinLock (mMemoryMappedLock);
 
   //