ShellPkg/Ls: Handle the case when SearchString is NULL

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Ruiyu Ni <ruiyu.ni@intel.com>
Reviewed-by: Hao Wu <hao.a.wu@intel.com>

1 files changed, 5 insertions, 0 deletions

diff --git a/ShellPkg/Library/UefiShellLevel2CommandsLib/Ls.c b/ShellPkg/Library/UefiShellLevel2CommandsLib/Ls.c
index 0b8019516f..9b4c452ca3 100644
--- a/ShellPkg/Library/UefiShellLevel2CommandsLib/Ls.c
+++ b/ShellPkg/Library/UefiShellLevel2CommandsLib/Ls.c
@@ -709,6 +709,11 @@ ShellCommandRunLs (
               // must split off the search part that applies to files from the end of the directory part

               //

               StrnCatGrow(&SearchString, NULL, FullPath, 0);

+              if (SearchString == NULL) {

+                FreePool (FullPath);

+                ShellCommandLineFreeVarList (Package);

+                return SHELL_OUT_OF_RESOURCES;

+              }

               PathRemoveLastItem (FullPath);

               CopyMem (SearchString, SearchString + StrLen (FullPath), StrSize (SearchString + StrLen (FullPath)));

             }