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

Fix always allocate memory issue. 

Signed-off-by: Eric Dong <eric.dong@intel.com>
Reviewed-by: Jaben Carsey <Jaben.carsey@intel.com>

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@13706 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
ydong10 2012-09-07 01:55:33 +00:00
parent 81634bfb53
commit f480fdc0f9
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3142,11 +3142,13 @@ StrnCatGrow (
// //
if (CurrentSize != NULL) { if (CurrentSize != NULL) {
NewSize = *CurrentSize; NewSize = *CurrentSize;
if (NewSize < DestinationStartSize + (Count * sizeof(CHAR16))) {
while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) { while (NewSize < (DestinationStartSize + (Count*sizeof(CHAR16)))) {
NewSize += 2 * Count * sizeof(CHAR16); NewSize += 2 * Count * sizeof(CHAR16);
} }
*Destination = ReallocatePool(*CurrentSize, NewSize, *Destination); *Destination = ReallocatePool(*CurrentSize, NewSize, *Destination);
*CurrentSize = NewSize; *CurrentSize = NewSize;
}
} else { } else {
*Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16)); *Destination = AllocateZeroPool((Count+1)*sizeof(CHAR16));
} }