tyler.durden/audk
1
0
Fork 0
mirror of https://github.com/acidanthera/audk.git synced 2025-07-29 00:24:07 +02:00
Code Issues Packages Projects Releases Wiki Activity

ShellPkg: complete previous incomplete patch

Browse Source 
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Jim Dailey <Jim_Dailey@Dell.com>
Reviewed-by: Jaben Carsey <jaben.carsey@intel.com>
This commit is contained in:
Jim Dailey 2016-02-10 13:17:56 -08:00 committed by Jaben Carsey
parent 62989e0bd2
commit 1095b4323d
1 changed files with 13 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -4176,43 +4176,27 @@ ShellFileHandleReadLine(
// //
// if we have space save it... // if we have space save it...
// //
if ((CountSoFar + 1) * CharSize < *Size){ if ((CountSoFar+1)*sizeof(CHAR16) < *Size){
ASSERT(Buffer != NULL); ASSERT(Buffer != NULL);
if (*Ascii) { ((CHAR16*)Buffer)[CountSoFar] = CharBuffer;
((CHAR8*)Buffer)[CountSoFar] = (CHAR8) CharBuffer; ((CHAR16*)Buffer)[CountSoFar+1] = CHAR_NULL;
((CHAR8*)Buffer)[CountSoFar+1] = '\0';
}
else {
((CHAR16*)Buffer)[CountSoFar] = CharBuffer;
((CHAR16*)Buffer)[CountSoFar+1] = CHAR_NULL;
}
} }
} }
// //
// if we ran out of space tell when... // if we ran out of space tell when...
// //
if (Status != EFI_END_OF_FILE){ if ((CountSoFar+1)*sizeof(CHAR16) > *Size){
if ((CountSoFar + 1) * CharSize > *Size){ *Size = (CountSoFar+1)*sizeof(CHAR16);
*Size = (CountSoFar + 1) * CharSize; if (!Truncate) {
if (!Truncate) { gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition);
gEfiShellProtocol->SetFilePosition(Handle, OriginalFilePosition); } else {
} else { DEBUG((DEBUG_WARN, "The line was truncated in ShellFileHandleReadLine"));
DEBUG((DEBUG_WARN, "The line was truncated in ShellFileHandleReadLine"));
}
return (EFI_BUFFER_TOO_SMALL);
}
if (*Ascii) {
if (CountSoFar && ((CHAR8*)Buffer)[CountSoFar - 1] == '\r') {
((CHAR8*)Buffer)[CountSoFar - 1] = '\0';
}
}
else {
if (CountSoFar && Buffer[CountSoFar - 1] == L'\r') {
Buffer[CountSoFar - 1] = CHAR_NULL;
}
} }
return (EFI_BUFFER_TOO_SMALL);
}
while(Buffer[StrLen(Buffer)-1] == L'\r') {
Buffer[StrLen(Buffer)-1] = CHAR_NULL;
} }
return (Status); return (Status);