From 70a7493d65ddd302cc791013a34f264217a6e475 Mon Sep 17 00:00:00 2001
From: Jeff Fan <jeff.fan@intel.com>
Date: Sat, 2 Jul 2016 11:08:04 +0800
Subject: [PATCH] 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>
---
 UefiCpuPkg/PiSmmCpuDxeSmm/PiSmmCpuDxeSmm.c | 7 +++++++
 1 file changed, 7 insertions(+)

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
@@ -96,6 +96,11 @@ UINTN mNumberOfCpus = 1;
 //
 BOOLEAN mSmmReadyToLock = FALSE;
 
+//
+// S3 boot flag
+//
+BOOLEAN mSmmS3Flag = FALSE;
+
 //
 // Global used to cache PCD for SMM Code Access Check enable
 //
@@ -485,6 +490,8 @@ SmmRestoreCpu (
 
   DEBUG ((EFI_D_INFO, "SmmRestoreCpu()\n"));
 
+  mSmmS3Flag = TRUE;
+
   InitializeSpinLock (mMemoryMappedLock);
 
   //
-- 
2.39.2