ShellPkg: Add checks for NULL pointers.

[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / Edit / Edit.c

diff --git a/ShellPkg/Library/UefiShellDebug1CommandsLib/Edit/Edit.c b/ShellPkg/Library/UefiShellDebug1CommandsLib/Edit/Edit.c
index c4dbf80c96f01a9a154493eed736ea8288006ce8..c28f9bc59135315f21d057b2a90ed94cefc242d1 100644 (file)
--- a/ShellPkg/Library/UefiShellDebug1CommandsLib/Edit/Edit.c
+++ b/ShellPkg/Library/UefiShellDebug1CommandsLib/Edit/Edit.c
@@ -36,6 +36,7 @@ ShellCommandRunEdit (
   CONST CHAR16        *Cwd;\r
   CHAR16              *Nfs;\r
   CHAR16              *Spot;\r
+  CONST CHAR16        *TempParam;\r
 //  SHELL_FILE_HANDLE   TempHandle;\r
 \r
   Buffer      = NULL;\r
@@ -101,7 +102,9 @@ ShellCommandRunEdit (
         // if editor launched with file named\r
         //\r
         if (ShellCommandLineGetCount(Package) == 2) {\r
-          FileBufferSetFileName (ShellCommandLineGetRawValue(Package, 1));\r
+          TempParam = ShellCommandLineGetRawValue(Package, 1);\r
+          ASSERT(TempParam != NULL);\r
+          FileBufferSetFileName (TempParam);\r
 //          if (EFI_ERROR(ShellFileExists(MainEditor.FileBuffer->FileName))) {\r
 //            Status = ShellOpenFileByName(MainEditor.FileBuffer->FileName, &TempHandle, EFI_FILE_MODE_CREATE|EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE, 0);\r
 //            if (!EFI_ERROR(Status)) {\r