Add check code to avoid access violation.

Signed-off-by: ydong10
Reviewed-by: jcarsey
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11873 6f19259b-4bc3-4df7-8a09-765794883524

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