diff options
-rw-r--r-- | MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c b/MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c index 5903b0f641..ee10655244 100644 --- a/MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c +++ b/MdeModulePkg/Library/PiDxeS3BootScriptLib/BootScriptSave.c @@ -1989,13 +1989,10 @@ S3BootScriptCompare ( if (RelativePosition == NULL) {
return EFI_INVALID_PARAMETER;
}
- TableLength = ((EFI_BOOT_SCRIPT_TABLE_HEADER*)Script)->TableLength;
//
- // If in boot time, TableLength does not include the termination node. so add it up
+ // mS3BootScriptTablePtr->TableLength does not include the termination node, so add it up
//
- if (!mS3BootScriptTablePtr->AtRuntime) {
- TableLength += sizeof(EFI_BOOT_SCRIPT_TERMINATE);
- }
+ TableLength = mS3BootScriptTablePtr->TableLength + sizeof (EFI_BOOT_SCRIPT_TERMINATE);
if (Position1 < Script || Position1 > Script+TableLength) {
return EFI_INVALID_PARAMETER;
}
|