From fa6d3ee4913eaf3b0fd9ddb8a4e3b618584bd965 Mon Sep 17 00:00:00 2001
From: ljin6 <ljin6@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Mon, 1 Feb 2010 03:32:28 +0000
Subject: Get MaxPacketSize from IP4 mode data and reduce the head size of UDP
 and MTFTP, and take it as BlockSize to download.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@9894 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcImpl.c | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

(limited to 'MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcImpl.c')

diff --git a/MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcImpl.c b/MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcImpl.c
index e5ff4addf6..df9d2df5be 100644
--- a/MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcImpl.c
+++ b/MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcImpl.c
@@ -347,6 +347,11 @@ EfiPxeBcStart (
     return Status;
   }
 
+  //
+  // Configure block size for TFTP as a default value to handle all link layers.
+  // 
+  Private->BlockSize   = (UINTN) (MIN (Private->Ip4MaxPacketSize, PXEBC_DEFAULT_PACKET_SIZE) - 
+                           PXEBC_DEFAULT_UDP_OVERHEAD_SIZE - PXEBC_DEFAULT_TFTP_OVERHEAD_SIZE);
   Private->AddressIsOk = FALSE;
 
   ZeroMem (Mode, sizeof (EFI_PXE_BASE_CODE_MODE));
@@ -2421,7 +2426,6 @@ DiscoverBootFile (
   UINT16                      Type;
   UINT16                      Layer;
   BOOLEAN                     UseBis;
-  UINTN                       BlockSize;
   PXEBC_CACHED_DHCP4_PACKET   *Packet;
   UINT16                      Value;
 
@@ -2466,7 +2470,6 @@ DiscoverBootFile (
   }
 
   *BufferSize = 0;
-  BlockSize   = 0x8000;
 
   //
   // Get bootfile name and (m)tftp server ip addresss
@@ -2526,7 +2529,7 @@ DiscoverBootFile (
                       Buffer,
                       FALSE,
                       BufferSize,
-                      &BlockSize,
+                      &Private->BlockSize,
                       &Private->ServerIp,
                       (UINT8 *) Private->BootFileName,
                       NULL,
@@ -2580,14 +2583,12 @@ EfiPxeLoadFile (
   PXEBC_PRIVATE_DATA          *Private;
   EFI_PXE_BASE_CODE_PROTOCOL  *PxeBc;
   BOOLEAN                     NewMakeCallback;
-  UINTN                       BlockSize;
   EFI_STATUS                  Status;
   UINT64                      TmpBufSize;
 
   Private         = PXEBC_PRIVATE_DATA_FROM_LOADFILE (This);
   PxeBc           = &Private->PxeBc;
   NewMakeCallback = FALSE;
-  BlockSize       = 0x8000;
   Status          = EFI_DEVICE_ERROR;
 
   if (This == NULL || BufferSize == NULL) {
@@ -2646,7 +2647,7 @@ EfiPxeLoadFile (
                         Buffer,
                         FALSE,
                         &TmpBufSize,
-                        &BlockSize,
+                        &Private->BlockSize,
                         &Private->ServerIp,
                         (UINT8 *) Private->BootFileName,
                         NULL,
@@ -2670,7 +2671,7 @@ EfiPxeLoadFile (
                       Buffer,
                       FALSE,
                       &TmpBufSize,
-                      &BlockSize,
+                      &Private->BlockSize,
                       &Private->ServerIp,
                       (UINT8 *) Private->BootFileName,
                       NULL,
-- 
cgit v1.2.3