From 9e926b693de28a1b77f64adda952463dcc481d12 Mon Sep 17 00:00:00 2001 From: jcarsey Date: Thu, 14 Jan 2010 20:26:39 +0000 Subject: [PATCH] git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@9758 6f19259b-4bc3-4df7-8a09-765794883524 --- ShellPkg/Library/UefiShellLib/UefiShellLib.c | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ShellPkg/Library/UefiShellLib/UefiShellLib.c b/ShellPkg/Library/UefiShellLib/UefiShellLib.c index 369bb81a58..db46b80f9a 100644 --- a/ShellPkg/Library/UefiShellLib/UefiShellLib.c +++ b/ShellPkg/Library/UefiShellLib/UefiShellLib.c @@ -1564,6 +1564,7 @@ ShellFindFilePathEx ( CHAR16 *TestPath; CHAR16 *RetVal; CONST CHAR16 *ExtensionWalker; + UINTN Size; ASSERT(FileName != NULL); if (FileExtension == NULL) { return (ShellFindFilePath(FileName)); @@ -1572,7 +1573,9 @@ ShellFindFilePathEx ( if (RetVal != NULL) { return (RetVal); } - TestPath = AllocateZeroPool(StrSize(FileName) + StrSize(FileExtension)); + Size = StrSize(FileName); + Size += StrSize(FileExtension); + TestPath = AllocateZeroPool(Size); for (ExtensionWalker = FileExtension ; ; ExtensionWalker = StrStr(ExtensionWalker, L";") + 1 ){ StrCpy(TestPath, FileName); StrCat(TestPath, ExtensionWalker); -- 2.39.2