ArmPkg/BdsLib: Prevent memory leak whith TFTP

[mirror_edk2.git] / ArmPkg / Library / BdsLib / BdsFilePath.c

diff --git a/ArmPkg/Library/BdsLib/BdsFilePath.c b/ArmPkg/Library/BdsLib/BdsFilePath.c
index f3fa4456555b76c8982390e4463b959826d632c0..411fcc1fcc8f810a983b76fba1e6f340a7a1a536 100644 (file)
--- a/ArmPkg/Library/BdsLib/BdsFilePath.c
+++ b/ArmPkg/Library/BdsLib/BdsFilePath.c
@@ -756,6 +756,7 @@ BdsTftpLoadImage (
   EFI_STATUS                  Status;\r
   EFI_PXE_BASE_CODE_PROTOCOL  *Pxe;\r
   UINT64                      TftpBufferSize;\r
+  UINT64                      TftpTransferSize;\r
   EFI_IP_ADDRESS              ServerIp;\r
   IPv4_DEVICE_PATH*           IPv4DevicePathNode;\r
   FILEPATH_DEVICE_PATH*       FilePathDevicePath;\r
@@ -857,12 +858,13 @@ BdsTftpLoadImage (
         goto EXIT;\r
       }\r
 \r
+      TftpTransferSize = TftpBufferSize;\r
       Status = Pxe->Mtftp (\r
                       Pxe,\r
                       EFI_PXE_BASE_CODE_TFTP_READ_FILE,\r
                       (VOID *)(UINTN)*Image,\r
                       FALSE,\r
-                      &TftpBufferSize,\r
+                      &TftpTransferSize,\r
                       NULL,\r
                       &ServerIp,\r
                       (UINT8*)AsciiPathName,\r