check memory allocations for success.

check pointer before access. git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11503 6f19259b-4bc3-4df7-8a09-765794883524
2011-04-04 21:41:24 +00:00
parent 4aa4377fda
commit cbdd109b43
5 changed files with 48 additions and 33 deletions
--- a/ShellPkg/Library/UefiShellLevel1CommandsLib/For.c
+++ b/ShellPkg/Library/UefiShellLevel1CommandsLib/For.c
@ -565,7 +565,9 @@ ShellCommandRunFor (
          Info->RemoveSubstAlias  = TRUE;
        }
      }
-      CurrentScriptFile->CurrentCommand->Data = Info;
+      if (CurrentScriptFile->CurrentCommand != NULL) {
+        CurrentScriptFile->CurrentCommand->Data = Info;
+      }
    } else {
      ShellPrintHiiEx(
        -1,