Bug fix for "Unplug U Disk during exploer using File Explorer cause tiano halt"

[mirror_edk2.git] / EdkNt32Pkg / Dxe / PlatformBds / Generic / BootMaint / FileExplorer.c

diff --git a/EdkNt32Pkg/Dxe/PlatformBds/Generic/BootMaint/FileExplorer.c b/EdkNt32Pkg/Dxe/PlatformBds/Generic/BootMaint/FileExplorer.c
index 5f304e0b8f32e62c11886c4ce71a0b9cb7e61156..db835cf534ca9d7829d6ffcdf3c8138ff9353eeb 100644 (file)
--- a/EdkNt32Pkg/Dxe/PlatformBds/Generic/BootMaint/FileExplorer.c
+++ b/EdkNt32Pkg/Dxe/PlatformBds/Generic/BootMaint/FileExplorer.c
@@ -148,7 +148,8 @@ Returns:
   BM_FILE_CONTEXT *NewFileContext;\r
   FORM_ID         FormId;\r
   BOOLEAN         ExitFileExplorer;\r
-\r
+  EFI_STATUS      Status;\r
+  \r
   NewMenuEntry      = NULL;\r
   NewFileContext    = NULL;\r
   ExitFileExplorer  = FALSE;\r
@@ -180,7 +181,11 @@ Returns:
     if (NewFileContext->IsDir ) {\r
       RemoveEntryList (&NewMenuEntry->Link);\r
       BOpt_FreeMenu (&DirectoryMenu);\r
-      BOpt_FindFiles (CallbackData, NewMenuEntry);\r
+      Status = BOpt_FindFiles (CallbackData, NewMenuEntry);\r
+       if (EFI_ERROR (Status)) {\r
+         ExitFileExplorer = TRUE;\r
+         goto exit;\r
+       }\r
       CreateMenuStringToken (CallbackData, CallbackData->FeHiiHandle, &DirectoryMenu);\r
       BOpt_DestroyMenuEntry (NewMenuEntry);\r
 \r
@@ -247,7 +252,7 @@ Returns:
       }\r
     }\r
   }\r
-\r
+exit:\r
   return ExitFileExplorer;\r
 }\r
 \r