summaryrefslogtreecommitdiff
path: root/ShellPkg
diff options
context:
space:
mode:
authorJim_Dailey@Dell.com <Jim_Dailey@Dell.com>2016-02-18 22:40:46 +0800
committerLiming Gao <liming.gao@intel.com>2016-03-03 12:45:18 +0800
commit7bcd3ff611821e9f3f05207ebf157e30d030795b (patch)
treed6534eb64128728fe0dddde941169c23411ba293 /ShellPkg
parentb9da8fe0e3a1353af0f5d698f449210908bab491 (diff)
downloadedk2-platforms-7bcd3ff611821e9f3f05207ebf157e30d030795b.tar.xz
ShellPkg: Add FileSize member to shell memory file structure.
The shell uses the memory file structure to manage temporary files in memory that support piping of output from one command into the the input of another command. The BufferSize member is the size of the internal buffer, not the size of the data that was written to the file. So, it was possible to read beyond the EOF of these files as reads used BufferSize. Now FileSize tracks the actual size of these files (the number of bytes written, not the number of bytes available in the buffer), and the reads use this member. Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Jim Dailey <jim_dailey@dell.com> Reviewed-by: Jaben Carsey <jaben.carsey@intel.com>
Diffstat (limited to 'ShellPkg')
-rw-r--r--ShellPkg/Application/Shell/FileHandleWrappers.c9
1 files changed, 6 insertions, 3 deletions
diff --git a/ShellPkg/Application/Shell/FileHandleWrappers.c b/ShellPkg/Application/Shell/FileHandleWrappers.c
index f8306e2147..893e5ffc04 100644
--- a/ShellPkg/Application/Shell/FileHandleWrappers.c
+++ b/ShellPkg/Application/Shell/FileHandleWrappers.c
@@ -1323,6 +1323,7 @@ typedef struct {
UINT64 Position;
UINT64 BufferSize;
BOOLEAN Unicode;
+ UINT64 FileSize;
} EFI_FILE_PROTOCOL_MEM;
/**
@@ -1341,7 +1342,7 @@ FileInterfaceMemSetPosition(
OUT UINT64 Position
)
{
- if (Position <= ((EFI_FILE_PROTOCOL_MEM*)This)->BufferSize) {
+ if (Position <= ((EFI_FILE_PROTOCOL_MEM*)This)->FileSize) {
((EFI_FILE_PROTOCOL_MEM*)This)->Position = Position;
return (EFI_SUCCESS);
} else {
@@ -1400,6 +1401,7 @@ FileInterfaceMemWrite(
}
CopyMem(((UINT8*)MemFile->Buffer) + MemFile->Position, Buffer, *BufferSize);
MemFile->Position += (*BufferSize);
+ MemFile->FileSize = MemFile->Position;
return (EFI_SUCCESS);
} else {
//
@@ -1416,6 +1418,7 @@ FileInterfaceMemWrite(
}
CopyMem(((UINT8*)MemFile->Buffer) + MemFile->Position, AsciiBuffer, AsciiStrSize(AsciiBuffer));
MemFile->Position += (*BufferSize / sizeof(CHAR16));
+ MemFile->FileSize = MemFile->Position;
FreePool(AsciiBuffer);
return (EFI_SUCCESS);
}
@@ -1441,8 +1444,8 @@ FileInterfaceMemRead(
EFI_FILE_PROTOCOL_MEM *MemFile;
MemFile = (EFI_FILE_PROTOCOL_MEM *) This;
- if (*BufferSize > (UINTN)((MemFile->BufferSize) - (UINTN)(MemFile->Position))) {
- (*BufferSize) = (UINTN)((MemFile->BufferSize) - (UINTN)(MemFile->Position));
+ if (*BufferSize > (UINTN)((MemFile->FileSize) - (UINTN)(MemFile->Position))) {
+ (*BufferSize) = (UINTN)((MemFile->FileSize) - (UINTN)(MemFile->Position));
}
CopyMem(Buffer, ((UINT8*)MemFile->Buffer) + MemFile->Position, (*BufferSize));
MemFile->Position = MemFile->Position + (*BufferSize);