UefiCpuPkg/PiSmmCpuDxeSmm: Add SMM S3 boot flag

It will be set to TRUE during S3 resume.

Cc: Michael Kinney <michael.d.kinney@intel.com>
Cc: Feng Tian <feng.tian@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Jeff Fan <jeff.fan@intel.com>
Reviewed-by: Feng Tian <feng.tian@intel.com>
Reviewed-by: Michael Kinney <michael.d.kinney@intel.com>
Regression-tested-by: Laszlo Ersek <lersek@redhat.com>
diff --git a/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c b/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c
index 0858d8f..db36a0a 100644
--- a/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c
+++ b/UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c
@@ -97,6 +97,11 @@
 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 @@
 

   DEBUG ((EFI_D_INFO, "SmmRestoreCpu()\n"));

 

+  mSmmS3Flag = TRUE;

+

   InitializeSpinLock (mMemoryMappedLock);

 

   //