NetworkPkg: Fix incorrect SizeofHeaders returned from HttpTcpReceiveHeader().

[mirror_edk2.git] / NetworkPkg / HttpDxe / HttpProto.c

diff --git a/NetworkPkg/HttpDxe/HttpProto.c b/NetworkPkg/HttpDxe/HttpProto.c
index ab00f3d6f25f050bd5177c7c21b020e5e67690bd..1aa1816c527ac454adcff1bd7ea4d4925d131546 100644 (file)
--- a/NetworkPkg/HttpDxe/HttpProto.c
+++ b/NetworkPkg/HttpDxe/HttpProto.c
@@ -1876,7 +1876,10 @@ HttpTcpReceiveHeader (
       //\r
       // Check whether we received end of HTTP headers.\r
       //\r
-      *EndofHeader = AsciiStrStr (*HttpHeaders, HTTP_END_OF_HDR_STR); \r
+      *EndofHeader = AsciiStrStr (*HttpHeaders, HTTP_END_OF_HDR_STR);\r
+      if (*EndofHeader != NULL) {\r
+        *SizeofHeaders = *EndofHeader - *HttpHeaders;\r
+      }\r
     };\r
     \r
     //\r
@@ -1976,6 +1979,9 @@ HttpTcpReceiveHeader (
       // Check whether we received end of HTTP headers.\r
       //\r
       *EndofHeader = AsciiStrStr (*HttpHeaders, HTTP_END_OF_HDR_STR); \r
+      if (*EndofHeader != NULL) {\r
+        *SizeofHeaders = *EndofHeader - *HttpHeaders;\r
+      }\r
     };\r
 \r
     //\r