ShellPkg: fix 'ls' handling of empty drives where there is not even an "." or "....

[mirror_edk2.git] / ShellPkg / Library / UefiFileHandleLib / UefiFileHandleLib.c

diff --git a/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c b/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
index 0bee8e38992d5b601a163f2641be02c923648e54..fb9d4ec0be50978b806952b47dec0d4b8d06d710 100644 (file)
--- a/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
+++ b/ShellPkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
@@ -509,9 +509,12 @@ FileHandleFindFirstFile (
   //\r
   Status = FileHandleRead (DirHandle, &BufferSize, *Buffer);\r
   ASSERT(Status != EFI_BUFFER_TOO_SMALL);\r
-  if (EFI_ERROR(Status)) {\r
+  if (EFI_ERROR(Status) || BufferSize == 0) {\r
     FreePool(*Buffer);\r
     *Buffer = NULL;\r
+    if (BufferSize == 0) {\r
+      return (EFI_NOT_FOUND);\r
+    }\r
     return (Status);\r
   }\r
   return (EFI_SUCCESS);\r