Add more check to make sure code run safely.
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10574 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
/*++
|
||||
|
||||
Copyright (c) 2007, Intel Corporation. All rights reserved.<BR>
|
||||
Copyright (c) 2007 - 2010, Intel Corporation. All rights reserved.<BR>
|
||||
This program and the accompanying materials
|
||||
are licensed and made available under the terms and conditions of the BSD License
|
||||
which accompanies this distribution. The full text of the license may be found at
|
||||
@@ -554,7 +554,11 @@ Returns:
|
||||
|
||||
Status = HiiDatabase->ExportPackageLists (HiiDatabase, Handle, &BufferSize, HiiPackageList);
|
||||
}
|
||||
if (HiiPackageList == NULL) {
|
||||
return EFI_OUT_OF_RESOURCES;
|
||||
}
|
||||
if (EFI_ERROR (Status)) {
|
||||
gBS->FreePool (HiiPackageList);
|
||||
return Status;
|
||||
}
|
||||
|
||||
|
@@ -1,6 +1,6 @@
|
||||
/*++
|
||||
|
||||
Copyright (c) 2007, Intel Corporation. All rights reserved.<BR>
|
||||
Copyright (c) 2007 - 2010, Intel Corporation. All rights reserved.<BR>
|
||||
This program and the accompanying materials
|
||||
are licensed and made available under the terms and conditions of the BSD License
|
||||
which accompanies this distribution. The full text of the license may be found at
|
||||
@@ -551,6 +551,9 @@ IfrLibNewString (
|
||||
LocateHiiProtocols ();
|
||||
|
||||
Languages = GetSupportedLanguages (PackageList);
|
||||
if (Languages == NULL) {
|
||||
return EFI_NOT_FOUND;
|
||||
}
|
||||
|
||||
LangStrings = Languages;
|
||||
while (*LangStrings != 0) {
|
||||
@@ -677,6 +680,9 @@ IfrLibSetString (
|
||||
LocateHiiProtocols ();
|
||||
|
||||
Languages = GetSupportedLanguages (PackageList);
|
||||
if (Languages == NULL) {
|
||||
return EFI_NOT_FOUND;
|
||||
}
|
||||
|
||||
LangStrings = Languages;
|
||||
while (*LangStrings != 0) {
|
||||
|
Reference in New Issue
Block a user