diff options
Diffstat (limited to 'ShellPkg/Application/Shell/FileHandleWrappers.c')
| -rw-r--r-- | ShellPkg/Application/Shell/FileHandleWrappers.c | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ShellPkg/Application/Shell/FileHandleWrappers.c b/ShellPkg/Application/Shell/FileHandleWrappers.c index 2ba9b34..f98d86a 100644 --- a/ShellPkg/Application/Shell/FileHandleWrappers.c +++ b/ShellPkg/Application/Shell/FileHandleWrappers.c @@ -2144,6 +2144,10 @@ FileInterfaceFileWrite ( // Ascii
//
AsciiBuffer = AllocateZeroPool (*BufferSize);
+ if (AsciiBuffer == NULL) {
+ return EFI_OUT_OF_RESOURCES;
+ }
+
AsciiSPrint (AsciiBuffer, *BufferSize, "%S", Buffer);
Size = AsciiStrSize (AsciiBuffer) - 1; // (we dont need the null terminator)
Status = (((EFI_FILE_PROTOCOL_FILE *)This)->Orig->Write (((EFI_FILE_PROTOCOL_FILE *)This)->Orig, &Size, AsciiBuffer));
|