mirror of https://github.com/acidanthera/audk.git
Check the pointer before use it to avoid potential access violation.
Signed-off-by: ydong10 Reviewed-by: jcarsey git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12533 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
parent
90f6df07f8
commit
0c1950ba5e
|
@ -2236,14 +2236,9 @@ ShellCommandLineGetFlag (
|
||||||
CHAR16 *TempString;
|
CHAR16 *TempString;
|
||||||
|
|
||||||
//
|
//
|
||||||
// ASSERT that both CheckPackage and KeyString aren't NULL
|
// return FALSE for no package or KeyString is NULL
|
||||||
//
|
//
|
||||||
ASSERT(KeyString != NULL);
|
if (CheckPackage == NULL || KeyString == NULL) {
|
||||||
|
|
||||||
//
|
|
||||||
// return FALSE for no package
|
|
||||||
//
|
|
||||||
if (CheckPackage == NULL) {
|
|
||||||
return (FALSE);
|
return (FALSE);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -2305,9 +2300,9 @@ ShellCommandLineGetValue (
|
||||||
CHAR16 *TempString;
|
CHAR16 *TempString;
|
||||||
|
|
||||||
//
|
//
|
||||||
// check for CheckPackage == NULL
|
// return NULL for no package or KeyString is NULL
|
||||||
//
|
//
|
||||||
if (CheckPackage == NULL) {
|
if (CheckPackage == NULL || KeyString == NULL) {
|
||||||
return (NULL);
|
return (NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue