Minor code update to make TcpIo unchanged when return EFI_DEVICE_ERROR.

[mirror_edk2.git] / MdeModulePkg / Library / DxeTcpIoLib / DxeTcpIoLib.c

diff --git a/MdeModulePkg/Library/DxeTcpIoLib/DxeTcpIoLib.c b/MdeModulePkg/Library/DxeTcpIoLib/DxeTcpIoLib.c
index 48120b435308564e522b019125a5c78f81fd6313..110e4890c88f9c244c7febe002a76e33cfbd0dce 100644 (file)
--- a/MdeModulePkg/Library/DxeTcpIoLib/DxeTcpIoLib.c
+++ b/MdeModulePkg/Library/DxeTcpIoLib/DxeTcpIoLib.c
@@ -882,7 +882,6 @@ TcpIoReceive (
   Tcp6 = NULL;\r
 \r
   if (TcpIo->TcpVersion == TCP_VERSION_4) {\r
-    TcpIo->RxToken.Tcp4Token.Packet.RxData = &RxData;\r
     Tcp4 = TcpIo->Tcp.Tcp4;\r
 \r
     if (TcpIo->IsListenDone) {\r
@@ -893,8 +892,9 @@ TcpIoReceive (
       return EFI_DEVICE_ERROR;\r
     }\r
 \r
+    TcpIo->RxToken.Tcp4Token.Packet.RxData = &RxData;\r
+\r
   } else if (TcpIo->TcpVersion == TCP_VERSION_6) {\r
-    TcpIo->RxToken.Tcp6Token.Packet.RxData = (EFI_TCP6_RECEIVE_DATA *) &RxData;\r
     Tcp6 = TcpIo->Tcp.Tcp6;\r
 \r
     if (TcpIo->IsListenDone) {\r
@@ -905,6 +905,8 @@ TcpIoReceive (
       return EFI_DEVICE_ERROR; \r
     }\r
 \r
+    TcpIo->RxToken.Tcp6Token.Packet.RxData = (EFI_TCP6_RECEIVE_DATA *) &RxData;\r
+\r
   } else {\r
     return EFI_UNSUPPORTED;\r
   }\r