SecurityPkg: Fix the loop in DoDeviceMeasurement

Signed-off-by: Wenxing Hou <wenxing.hou@intel.com>
tianocore · Aug 25, 2023 · 4c39ec9 · 4c39ec9
1 parent 3d47429
commit 4c39ec9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/SecurityPkg/DeviceSecurity/SpdmSecurityLib/SpdmMeasurement.c b/SecurityPkg/DeviceSecurity/SpdmSecurityLib/SpdmMeasurement.c
@@ -679,7 +679,6 @@ DoDeviceMeasurement (
         }
       }
 
-      ReceivedNumberOfBlock += 1;
       DEBUG ((DEBUG_INFO, "ExtendMeasurement...\n"));
       AuthState                       = TCG_DEVICE_SECURITY_EVENT_DATA_DEVICE_AUTH_STATE_SUCCESS;
       SecurityState->MeasurementState = EDKII_DEVICE_SECURITY_STATE_SUCCESS;
@@ -692,6 +691,8 @@ DoDeviceMeasurement (
       if (Status != EFI_SUCCESS) {
         return Status;
       }
+
+      ReceivedNumberOfBlock += 1;
     }
 
     if (ReceivedNumberOfBlock != NumberOfBlocks) {