NetworkPkg/UefiPxeBcDxe: Use the specified MTFTP windowsize.

[mirror_edk2.git] / NetworkPkg / UefiPxeBcDxe / PxeBcImpl.c

diff --git a/NetworkPkg/UefiPxeBcDxe/PxeBcImpl.c b/NetworkPkg/UefiPxeBcDxe/PxeBcImpl.c
index 13396903f5853176d55667488ff4095641f4842f..468b38d887d85005f70752b4bbd6923ce7743cae 100644 (file)
--- a/NetworkPkg/UefiPxeBcDxe/PxeBcImpl.c
+++ b/NetworkPkg/UefiPxeBcDxe/PxeBcImpl.c
@@ -849,7 +849,7 @@ EfiPxeBcMtftp (
   VOID                            *Config;\r
   EFI_STATUS                      Status;\r
   EFI_PXE_BASE_CODE_IP_FILTER     IpFilter;\r
-\r
+  UINTN                           WindowSize;\r
 \r
   if ((This == NULL) ||\r
       (Filename == NULL) ||\r
@@ -873,6 +873,11 @@ EfiPxeBcMtftp (
   Private   = PXEBC_PRIVATE_DATA_FROM_PXEBC (This);\r
   Mode      = Private->PxeBc.Mode;\r
 \r
+  //\r
+  // Get PcdPxeTftpWindowSize.\r
+  //\r
+  WindowSize = (UINTN) PcdGet64 (PcdPxeTftpWindowSize);\r
+\r
   if (Mode->UsingIpv6) {\r
     if (!NetIp6IsValidUnicast (&ServerIp->v6)) {\r
       return EFI_INVALID_PARAMETER;\r
@@ -930,6 +935,7 @@ EfiPxeBcMtftp (
                Config,\r
                Filename,\r
                BlockSize,\r
+               (WindowSize > 1) ? &WindowSize : NULL,\r
                BufferSize\r
                );\r
 \r
@@ -944,6 +950,7 @@ EfiPxeBcMtftp (
                Config,\r
                Filename,\r
                BlockSize,\r
+               (WindowSize > 1) ? &WindowSize : NULL,\r
                BufferPtr,\r
                BufferSize,\r
                DontUseBuffer\r
@@ -976,6 +983,7 @@ EfiPxeBcMtftp (
                Config,\r
                Filename,\r
                BlockSize,\r
+               (WindowSize > 1) ? &WindowSize : NULL,\r
                BufferPtr,\r
                BufferSize,\r
                DontUseBuffer\r