Fix a bug that mtftp4 driver reply 2 ACK in answer to incoming No.65535 block data. That issue incurs incorrect network transfer.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10619 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 3 insertions, 1 deletions

diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c
index 4dcdf827ea..bb4ce0ae65 100644
--- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c
+++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c
@@ -158,6 +158,7 @@ Mtftp4SetLastBlockNum (
 

   @param  Head                  The block range list to remove from

   @param  Num                   The block number to remove

+  @param  Completed             Whether Num is the last block number

   @param  TotalBlock            The continuous block number in all 

 

   @retval EFI_NOT_FOUND         The block number isn't in the block range list

@@ -169,6 +170,7 @@ EFI_STATUS
 Mtftp4RemoveBlockNum (

   IN LIST_ENTRY             *Head,

   IN UINT16                 Num,

+  IN BOOLEAN                Completed,

   OUT UINT64                *TotalBlock

   )

 {

@@ -229,7 +231,7 @@ Mtftp4RemoveBlockNum (
 		  Range->Round ++;

       }

 

-      if (Range->Start > Range->End) {

+      if ((Range->Start > Range->End) || Completed) {

         RemoveEntryList (&Range->Link);

         FreePool (Range);

       }