ShellPkg/ShellLib: Constructor doesn't depend on ShellParameters

[mirror_edk2.git] / ShellPkg / Library / UefiShellTftpCommandLib / Tftp.c

diff --git a/ShellPkg/Library/UefiShellTftpCommandLib/Tftp.c b/ShellPkg/Library/UefiShellTftpCommandLib/Tftp.c
index 5c50797631f2886c0024d1fb30297dd4a275e59f..fbde3bfe6048f50beb2e4c7be8d188737ccd5916 100755 (executable)
--- a/ShellPkg/Library/UefiShellTftpCommandLib/Tftp.c
+++ b/ShellPkg/Library/UefiShellTftpCommandLib/Tftp.c
@@ -284,6 +284,7 @@ ShellCommandRunTftp (
   EFI_HANDLE              Mtftp4ChildHandle;\r
   EFI_MTFTP4_PROTOCOL     *Mtftp4;\r
   UINTN                   FileSize;\r
+  UINTN                   DataSize;\r
   VOID                    *Data;\r
   SHELL_FILE_HANDLE       FileHandle;\r
   UINT16                  BlockSize;\r
@@ -294,6 +295,7 @@ ShellCommandRunTftp (
   AsciiRemoteFilePath = NULL;\r
   Handles             = NULL;\r
   FileSize            = 0;\r
+  DataSize            = 0;\r
   BlockSize           = MTFTP_DEFAULT_BLKSIZE;\r
 \r
   //\r
@@ -537,6 +539,7 @@ ShellCommandRunTftp (
       goto NextHandle;\r
     }\r
 \r
+    DataSize = FileSize;\r
     Status = ShellWriteFile (FileHandle, &FileSize, Data);\r
     if (!EFI_ERROR (Status)) {\r
       ShellStatus = SHELL_SUCCESS;\r
@@ -551,7 +554,7 @@ ShellCommandRunTftp (
     NextHandle:\r
 \r
     if (Data != NULL) {\r
-      gBS->FreePages ((EFI_PHYSICAL_ADDRESS)(UINTN)Data, EFI_SIZE_TO_PAGES (FileSize));\r
+      gBS->FreePages ((EFI_PHYSICAL_ADDRESS)(UINTN)Data, EFI_SIZE_TO_PAGES (DataSize));\r
     }\r
 \r
     CloseProtocolAndDestroyServiceChild (\r