tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@9758 6f19259b-4bc3-4df7-8a09-765794883524

This commit is contained in:
jcarsey 2010-01-14 20:26:39 +00:00
parent bc230a23ed
commit 9e926b693d
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ShellPkg/Library/UefiShellLib/UefiShellLib.c
View File

@ -1564,6 +1564,7 @@ ShellFindFilePathEx (
CHAR16 *TestPath; CHAR16 *TestPath;
CHAR16 *RetVal; CHAR16 *RetVal;
CONST CHAR16 *ExtensionWalker; CONST CHAR16 *ExtensionWalker;
UINTN Size;
ASSERT(FileName != NULL); ASSERT(FileName != NULL);
if (FileExtension == NULL) { if (FileExtension == NULL) {
return (ShellFindFilePath(FileName)); return (ShellFindFilePath(FileName));
@ -1572,7 +1573,9 @@ ShellFindFilePathEx (
if (RetVal != NULL) { if (RetVal != NULL) {
return (RetVal); 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 ){ for (ExtensionWalker = FileExtension ; ; ExtensionWalker = StrStr(ExtensionWalker, L";") + 1 ){
StrCpy(TestPath, FileName); StrCpy(TestPath, FileName);
StrCat(TestPath, ExtensionWalker); StrCat(TestPath, ExtensionWalker);