Refine code to make code run more safely.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11143 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
ydong10 2010-12-09 08:25:59 +00:00
parent 96756716c0
commit 2349cf6f63
2 changed files with 5 additions and 1 deletions

View File

@ -1395,6 +1395,10 @@ Returns:
return EFI_BUFFER_TOO_SMALL; return EFI_BUFFER_TOO_SMALL;
} }
if (ConfigHdr == NULL) {
return EFI_INVALID_PARAMETER;
}
*StrBufferLen = BufferSize; *StrBufferLen = BufferSize;
StrPtr = ConfigHdr; StrPtr = ConfigHdr;

View File

@ -930,7 +930,7 @@ Returns:
// Try to get the PEI version of ReportStatusCode. // Try to get the PEI version of ReportStatusCode.
// //
Status = GetPeiProtocol (&gEfiStatusCodeRuntimeProtocolGuid, (VOID **) &gReportStatusCode); Status = GetPeiProtocol (&gEfiStatusCodeRuntimeProtocolGuid, (VOID **) &gReportStatusCode);
if (EFI_ERROR (Status)) { if (EFI_ERROR (Status) || (gReportStatusCode == NULL)) {
return EFI_UNSUPPORTED; return EFI_UNSUPPORTED;
} }
} }