Add check code to avoid access violation.

[mirror_edk2.git] / ShellPkg / Library / UefiShellDriver1CommandsLib / DrvCfg.c

diff --git a/ShellPkg/Library/UefiShellDriver1CommandsLib/DrvCfg.c b/ShellPkg/Library/UefiShellDriver1CommandsLib/DrvCfg.c
index 5203f11f9bd88ccf7d4558d62d7576ecf11979d6..1b6b189f25beb7fd04907aa3e2f3befcaefe7e92 100644 (file)
--- a/ShellPkg/Library/UefiShellDriver1CommandsLib/DrvCfg.c
+++ b/ShellPkg/Library/UefiShellDriver1CommandsLib/DrvCfg.c
@@ -64,6 +64,10 @@ FindHiiHandleViaDevPath(
     return (Status);\r
   }\r
 \r
+  if (HandleBuffer == NULL) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
   for (LoopVariable = 0 ; LoopVariable < (HandleBufferSize/sizeof(HandleBuffer[0])) && *HiiHandle == NULL ; LoopVariable++) {\r
     MainBufferSize    = 0;\r
     MainBuffer        = NULL;\r