SecurityPkg:Tcg2Smm: Fix compile issue

Update Tcg2Smm _PRS patching logic to fix compile issue

Cc: Liming Gao <liming.gao@intel.com>
Cc: Dandan Bi <dandan.bi@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.1
Signed-off-by: Chao Zhang <chao.b.zhang@intel.com>
Reviewed-by: Liming Gao <liming.gao@intel.com>

diff --git a/SecurityPkg/Tcg/Tcg2Smm/Tcg2Smm.c b/SecurityPkg/Tcg/Tcg2Smm/Tcg2Smm.c
index e3938cb8ece02b9f2196e196da45f28c506e4980..6eb62ae64ca3c7ec49ce158b045885c2aea81b1d 100644 (file)
--- a/SecurityPkg/Tcg/Tcg2Smm/Tcg2Smm.c
+++ b/SecurityPkg/Tcg/Tcg2Smm/Tcg2Smm.c
@@ -539,10 +539,11 @@ UpdatePossibleResource (
   *(DataPtr + 1) = 0;\r
 \r
   //\r
   *(DataPtr + 1) = 0;\r
 \r
   //\r

-  // 5. Jump over whole ResourceTemplate. Stuff rest bytes to NOOP\r
+  // 5. Jump over new ResourceTemplate. Stuff rest bytes to NOOP\r

   //\r
   //\r

-  for (DataPtr += 2; DataPtr < DataEndPtr; DataPtr++) {\r
-    *DataPtr = AML_NOOP_OP;\r
+  DataPtr += 2;\r
+  if (DataPtr < DataEndPtr) {\r
+    SetMem(DataPtr, (UINTN)(DataEndPtr - DataPtr), AML_NOOP_OP);\r

   }\r
 \r
   return EFI_SUCCESS;\r
   }\r
 \r
   return EFI_SUCCESS;\r