]> git.proxmox.com Git - mirror_edk2.git/blobdiff - NetworkPkg/HttpDxe/HttpImpl.c
NetworkPkg/HttpDxe: Fix the bug when parsing HTTP(S) message body.
[mirror_edk2.git] / NetworkPkg / HttpDxe / HttpImpl.c
index 1f7a4fa52a8ee36f52124942c2e958e156698a52..17deceb395473e7457e2c3616c7059eea423cc56 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
   Implementation of EFI_HTTP_PROTOCOL protocol interfaces.\r
 \r
-  Copyright (c) 2015 - 2017, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2015 - 2018, Intel Corporation. All rights reserved.<BR>\r
   (C) Copyright 2015-2016 Hewlett Packard Enterprise Development LP<BR>\r
 \r
   This program and the accompanying materials\r
@@ -33,16 +33,18 @@ EFI_HTTP_PROTOCOL  mEfiHttpTemplate = {
 \r
   @param[in]  This                Pointer to EFI_HTTP_PROTOCOL instance.\r
   @param[out] HttpConfigData      Point to buffer for operational parameters of this\r
-                                  HTTP instance.\r
+                                  HTTP instance. It is the responsibility of the caller\r
+                                  to allocate the memory for HttpConfigData and\r
+                                  HttpConfigData->AccessPoint.IPv6Node/IPv4Node. In fact,\r
+                                  it is recommended to allocate sufficient memory to record\r
+                                  IPv6Node since it is big enough for all possibilities.\r
 \r
   @retval EFI_SUCCESS             Operation succeeded.\r
   @retval EFI_INVALID_PARAMETER   One or more of the following conditions is TRUE:\r
                                   This is NULL.\r
                                   HttpConfigData is NULL.\r
-                                  HttpInstance->LocalAddressIsIPv6 is FALSE and\r
-                                  HttpConfigData->IPv4Node is NULL.\r
-                                  HttpInstance->LocalAddressIsIPv6 is TRUE and\r
-                                  HttpConfigData->IPv6Node is NULL.\r
+                                  HttpConfigData->AccessPoint.IPv4Node or\r
+                                  HttpConfigData->AccessPoint.IPv6Node is NULL.\r
   @retval EFI_NOT_STARTED         This EFI HTTP Protocol instance has not been started.\r
 \r
 **/\r
@@ -63,10 +65,9 @@ EfiHttpGetModeData (
   }\r
 \r
   HttpInstance = HTTP_INSTANCE_FROM_PROTOCOL (This);\r
-  ASSERT (HttpInstance != NULL);\r
 \r
-  if ((HttpInstance->LocalAddressIsIPv6 && HttpConfigData->AccessPoint.IPv6Node == NULL) ||\r
-      (!HttpInstance->LocalAddressIsIPv6 && HttpConfigData->AccessPoint.IPv4Node == NULL)) {\r
+  if ((HttpConfigData->AccessPoint.IPv6Node == NULL) ||\r
+      (HttpConfigData->AccessPoint.IPv4Node == NULL)) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
@@ -115,9 +116,9 @@ EfiHttpGetModeData (
   @retval EFI_INVALID_PARAMETER   One or more of the following conditions is TRUE:\r
                                   This is NULL.\r
                                   HttpConfigData->LocalAddressIsIPv6 is FALSE and\r
-                                  HttpConfigData->IPv4Node is NULL.\r
+                                  HttpConfigData->AccessPoint.IPv4Node is NULL.\r
                                   HttpConfigData->LocalAddressIsIPv6 is TRUE and\r
-                                  HttpConfigData->IPv6Node is NULL.\r
+                                  HttpConfigData->AccessPoint.IPv6Node is NULL.\r
   @retval EFI_ALREADY_STARTED     Reinitialize this HTTP instance without calling\r
                                   Configure() with NULL to reset it.\r
   @retval EFI_DEVICE_ERROR        An unexpected system or network error occurred.\r
@@ -130,27 +131,31 @@ EFI_STATUS
 EFIAPI\r
 EfiHttpConfigure (\r
   IN  EFI_HTTP_PROTOCOL         *This,\r
-  IN  EFI_HTTP_CONFIG_DATA      *HttpConfigData\r
-  ) \r
+  IN  EFI_HTTP_CONFIG_DATA      *HttpConfigData OPTIONAL\r
+  )\r
 {\r
   HTTP_PROTOCOL                 *HttpInstance;\r
   EFI_STATUS                    Status;\r
-  \r
+\r
   //\r
   // Check input parameters.\r
   //\r
   if (This == NULL ||\r
-      (HttpConfigData != NULL && \r
+      (HttpConfigData != NULL &&\r
        ((HttpConfigData->LocalAddressIsIPv6 && HttpConfigData->AccessPoint.IPv6Node == NULL) ||\r
         (!HttpConfigData->LocalAddressIsIPv6 && HttpConfigData->AccessPoint.IPv4Node == NULL)))) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
   HttpInstance = HTTP_INSTANCE_FROM_PROTOCOL (This);\r
-  ASSERT (HttpInstance != NULL && HttpInstance->Service != NULL);\r
+  ASSERT (HttpInstance->Service != NULL);\r
 \r
   if (HttpConfigData != NULL) {\r
 \r
+    if (HttpConfigData->HttpVersion >= HttpVersionUnsupported) {\r
+      return EFI_UNSUPPORTED;\r
+    }\r
+\r
     //\r
     // Now configure this HTTP instance.\r
     //\r
@@ -161,8 +166,8 @@ EfiHttpConfigure (
     HttpInstance->HttpVersion        = HttpConfigData->HttpVersion;\r
     HttpInstance->TimeOutMillisec    = HttpConfigData->TimeOutMillisec;\r
     HttpInstance->LocalAddressIsIPv6 = HttpConfigData->LocalAddressIsIPv6;\r
-    \r
-    if (HttpConfigData->LocalAddressIsIPv6) { \r
+\r
+    if (HttpConfigData->LocalAddressIsIPv6) {\r
       CopyMem (\r
         &HttpInstance->Ipv6Node,\r
         HttpConfigData->AccessPoint.IPv6Node,\r
@@ -175,7 +180,7 @@ EfiHttpConfigure (
         sizeof (HttpInstance->IPv4Node)\r
         );\r
     }\r
-    \r
+\r
     //\r
     // Creat Tcp child\r
     //\r
@@ -183,7 +188,7 @@ EfiHttpConfigure (
     if (EFI_ERROR (Status)) {\r
       return Status;\r
     }\r
-    \r
+\r
     HttpInstance->State = HTTP_STATE_HTTP_CONFIGED;\r
     return EFI_SUCCESS;\r
 \r
@@ -196,7 +201,7 @@ EfiHttpConfigure (
     return EFI_SUCCESS;\r
   }\r
 }\r
\r
+\r
 \r
 /**\r
   The Request() function queues an HTTP request to this HTTP instance.\r
@@ -276,15 +281,16 @@ EfiHttpRequest (
   Request = HttpMsg->Data.Request;\r
 \r
   //\r
-  // Only support GET, HEAD, PUT and POST method in current implementation.\r
+  // Only support GET, HEAD, DELETE, PATCH, PUT and POST method in current implementation.\r
   //\r
   if ((Request != NULL) && (Request->Method != HttpMethodGet) &&\r
-      (Request->Method != HttpMethodHead) && (Request->Method != HttpMethodPut) && (Request->Method != HttpMethodPost)) {\r
+      (Request->Method != HttpMethodHead) && (Request->Method != HttpMethodDelete) &&\r
+      (Request->Method != HttpMethodPut) && (Request->Method != HttpMethodPost) &&\r
+      (Request->Method != HttpMethodPatch)) {\r
     return EFI_UNSUPPORTED;\r
   }\r
 \r
   HttpInstance = HTTP_INSTANCE_FROM_PROTOCOL (This);\r
-  ASSERT (HttpInstance != NULL);\r
 \r
   //\r
   // Capture the method into HttpInstance.\r
@@ -299,14 +305,16 @@ EfiHttpRequest (
 \r
   if (Request == NULL) {\r
     //\r
-    // Request would be NULL only for PUT/POST operation (in the current implementation)\r
+    // Request would be NULL only for PUT/POST/PATCH operation (in the current implementation)\r
     //\r
-    if ((HttpInstance->Method != HttpMethodPut) && (HttpInstance->Method != HttpMethodPost)) {\r
+    if ((HttpInstance->Method != HttpMethodPut) &&\r
+        (HttpInstance->Method != HttpMethodPost) &&\r
+        (HttpInstance->Method != HttpMethodPatch)) {\r
       return EFI_INVALID_PARAMETER;\r
     }\r
 \r
     //\r
-    // For PUT/POST, we need to have the TCP already configured. Bail out if it is not!\r
+    // For PUT/POST/PATCH, we need to have the TCP already configured. Bail out if it is not!\r
     //\r
     if (HttpInstance->State < HTTP_STATE_TCP_CONFIGED) {\r
       return EFI_INVALID_PARAMETER;\r
@@ -350,7 +358,7 @@ EfiHttpRequest (
     UnicodeStrToAsciiStrS (Request->Url, Url, UrlLen);\r
 \r
     //\r
-    // From the information in Url, the HTTP instance will \r
+    // From the information in Url, the HTTP instance will\r
     // be able to determine whether to use http or https.\r
     //\r
     HttpInstance->UseHttps = IsHttpsUrl (Url);\r
@@ -359,7 +367,7 @@ EfiHttpRequest (
     // HTTP is disabled, return directly if the URI is not HTTPS.\r
     //\r
     if (!PcdGetBool (PcdAllowHttpConnections) && !(HttpInstance->UseHttps)) {\r
-      \r
+\r
       DEBUG ((EFI_D_ERROR, "EfiHttpRequest: HTTP is disabled.\n"));\r
 \r
       return EFI_ACCESS_DENIED;\r
@@ -380,6 +388,7 @@ EfiHttpRequest (
 \r
       HttpInstance->TlsChildHandle = TlsCreateChild (\r
                                        ImageHandle,\r
+                                       &(HttpInstance->TlsSb),\r
                                        &(HttpInstance->Tls),\r
                                        &(HttpInstance->TlsConfiguration)\r
                                        );\r
@@ -425,9 +434,9 @@ EfiHttpRequest (
       ReConfigure = FALSE;\r
     } else {\r
       if ((HttpInstance->RemotePort == RemotePort) &&\r
-          (AsciiStrCmp (HttpInstance->RemoteHost, HostName) == 0) && \r
-          (!HttpInstance->UseHttps || (HttpInstance->UseHttps && \r
-                                       !TlsConfigure && \r
+          (AsciiStrCmp (HttpInstance->RemoteHost, HostName) == 0) &&\r
+          (!HttpInstance->UseHttps || (HttpInstance->UseHttps &&\r
+                                       !TlsConfigure &&\r
                                        HttpInstance->TlsSessionState == EfiTlsSessionDataTransferring))) {\r
         //\r
         // Host Name and port number of the request URL are the same with previous call to Request().\r
@@ -462,6 +471,8 @@ EfiHttpRequest (
 \r
           FreePool (HostName);\r
 \r
+          HttpUrlFreeParser (UrlParser);\r
+\r
           //\r
           // Queue the HTTP token and return.\r
           //\r
@@ -484,7 +495,7 @@ EfiHttpRequest (
         }\r
       }\r
     }\r
-  } \r
+  }\r
 \r
   if (Configure) {\r
     //\r
@@ -503,16 +514,17 @@ EfiHttpRequest (
         Status = EFI_OUT_OF_RESOURCES;\r
         goto Error1;\r
       }\r
-      \r
+\r
       AsciiStrToUnicodeStrS (HostName, HostNameStr, HostNameSize);\r
       if (!HttpInstance->LocalAddressIsIPv6) {\r
         Status = HttpDns4 (HttpInstance, HostNameStr, &HttpInstance->RemoteAddr);\r
       } else {\r
         Status = HttpDns6 (HttpInstance, HostNameStr, &HttpInstance->RemoteIpv6Addr);\r
       }\r
-      \r
+\r
       FreePool (HostNameStr);\r
       if (EFI_ERROR (Status)) {\r
+        DEBUG ((EFI_D_ERROR, "Error: Could not retrieve the host address from DNS server.\n"));\r
         goto Error1;\r
       }\r
     }\r
@@ -541,10 +553,10 @@ EfiHttpRequest (
       if (EFI_ERROR (Status)) {\r
         goto Error1;\r
       }\r
-      \r
+\r
       TlsCloseTxRxEvent (HttpInstance);\r
     }\r
-    \r
+\r
     HttpCloseConnection (HttpInstance);\r
     EfiHttpCancel (This, NULL);\r
   }\r
@@ -563,11 +575,11 @@ EfiHttpRequest (
   if (Request != NULL) {\r
     Wrap->TcpWrap.Method = Request->Method;\r
   }\r
-  \r
+\r
   Status = HttpInitSession (\r
-             HttpInstance, \r
-             Wrap, \r
-             Configure || ReConfigure, \r
+             HttpInstance,\r
+             Wrap,\r
+             Configure || ReConfigure,\r
              TlsConfigure\r
              );\r
   if (EFI_ERROR (Status)) {\r
@@ -576,14 +588,14 @@ EfiHttpRequest (
 \r
   if (!Configure && !ReConfigure && !TlsConfigure) {\r
     //\r
-    // For the new HTTP token, create TX TCP token events.    \r
+    // For the new HTTP token, create TX TCP token events.\r
     //\r
     Status = HttpCreateTcpTxEvent (Wrap);\r
     if (EFI_ERROR (Status)) {\r
       goto Error1;\r
     }\r
   }\r
-  \r
+\r
   //\r
   // Create request message.\r
   //\r
@@ -612,11 +624,9 @@ EfiHttpRequest (
     goto Error3;\r
   }\r
 \r
-  ASSERT (RequestMsg != NULL);\r
-\r
   //\r
   // Every request we insert a TxToken and a response call would remove the TxToken.\r
-  // In cases of PUT/POST, after an initial request-response pair, we would do a\r
+  // In cases of PUT/POST/PATCH, after an initial request-response pair, we would do a\r
   // continuous request without a response call. So, in such cases, where Request\r
   // structure is NULL, we would not insert a TxToken.\r
   //\r
@@ -637,15 +647,19 @@ EfiHttpRequest (
              RequestMsgSize\r
              );\r
   if (EFI_ERROR (Status)) {\r
-    goto Error5;    \r
+    goto Error5;\r
   }\r
 \r
   DispatchDpc ();\r
-  \r
+\r
   if (HostName != NULL) {\r
     FreePool (HostName);\r
   }\r
-  \r
+\r
+  if (UrlParser != NULL) {\r
+    HttpUrlFreeParser (UrlParser);\r
+  }\r
+\r
   return EFI_SUCCESS;\r
 \r
 Error5:\r
@@ -660,7 +674,7 @@ Error5:
 Error4:\r
   if (RequestMsg != NULL) {\r
     FreePool (RequestMsg);\r
-  }  \r
+  }\r
 \r
 Error3:\r
   if (HttpInstance->UseHttps) {\r
@@ -670,7 +684,7 @@ Error3:
 \r
 Error2:\r
   HttpCloseConnection (HttpInstance);\r
-  \r
+\r
   HttpCloseTcpConnCloseEvent (HttpInstance);\r
   if (NULL != Wrap->TcpWrap.Tx4Token.CompletionToken.Event) {\r
     gBS->CloseEvent (Wrap->TcpWrap.Tx4Token.CompletionToken.Event);\r
@@ -688,17 +702,17 @@ Error1:
   if (Wrap != NULL) {\r
     FreePool (Wrap);\r
   }\r
-  if (UrlParser!= NULL) {\r
+  if (UrlParser != NULL) {\r
     HttpUrlFreeParser (UrlParser);\r
   }\r
 \r
   return Status;\r
-  \r
+\r
 }\r
 \r
 /**\r
-  Cancel a user's Token. \r
\r
+  Cancel a user's Token.\r
+\r
   @param[in]  Map                The HTTP instance's token queue.\r
   @param[in]  Item               Object container for one HTTP token and token's wrap.\r
   @param[in]  Context            The user's token to cancel.\r
@@ -732,7 +746,7 @@ HttpCancelTokens (
   Wrap = (HTTP_TOKEN_WRAP *) Item->Value;\r
   ASSERT (Wrap != NULL);\r
   HttpInstance = Wrap->HttpInstance;\r
-  \r
+\r
   if (!HttpInstance->LocalAddressIsIPv6) {\r
     if (Wrap->TcpWrap.Rx4Token.CompletionToken.Event != NULL) {\r
       //\r
@@ -767,7 +781,7 @@ HttpCancelTokens (
     return EFI_ABORTED;\r
   }\r
 \r
-  return EFI_SUCCESS; \r
+  return EFI_SUCCESS;\r
 }\r
 \r
 /**\r
@@ -780,7 +794,7 @@ HttpCancelTokens (
                                  cancelled.\r
 \r
   @retval EFI_SUCCESS            The token is cancelled.\r
-  @retval EFI_NOT_FOUND          The asynchronous request or response token is not found.                                 \r
+  @retval EFI_NOT_FOUND          The asynchronous request or response token is not found.\r
   @retval Others                 Other error as indicated.\r
 \r
 **/\r
@@ -800,7 +814,7 @@ HttpCancel (
     if (Token != NULL) {\r
       if (Status == EFI_ABORTED) {\r
         return EFI_SUCCESS;\r
-      } \r
+      }\r
     } else {\r
       return Status;\r
     }\r
@@ -829,7 +843,7 @@ HttpCancel (
       HttpInstance->Tcp6->Cancel (HttpInstance->Tcp6, &HttpInstance->Tcp6TlsRxToken.CompletionToken);\r
     }\r
   }\r
-  \r
+\r
   return EFI_SUCCESS;\r
 }\r
 \r
@@ -870,7 +884,6 @@ EfiHttpCancel (
   }\r
 \r
   HttpInstance = HTTP_INSTANCE_FROM_PROTOCOL (This);\r
-  ASSERT (HttpInstance != NULL);\r
 \r
   if (HttpInstance->State != HTTP_STATE_TCP_CONNECTED) {\r
     return EFI_NOT_STARTED;\r
@@ -903,6 +916,7 @@ HttpBodyParserCallback (
   IN VOID                       *Context\r
   )\r
 {\r
+  HTTP_CALLBACK_DATA            *CallbackData;\r
   HTTP_TOKEN_WRAP               *Wrap;\r
   UINTN                         BodyLength;\r
   CHAR8                         *Body;\r
@@ -915,20 +929,17 @@ HttpBodyParserCallback (
     return EFI_SUCCESS;\r
   }\r
 \r
-  Wrap = (HTTP_TOKEN_WRAP *) Context;\r
-  Body = Wrap->HttpToken->Message->Body;\r
-  BodyLength = Wrap->HttpToken->Message->BodyLength;\r
+  CallbackData = (HTTP_CALLBACK_DATA *) Context;\r
+\r
+  Wrap       = (HTTP_TOKEN_WRAP *) (CallbackData->Wrap);\r
+  Body       = CallbackData->ParseData;\r
+  BodyLength = CallbackData->ParseDataLength;\r
+\r
   if (Data < Body + BodyLength) {\r
     Wrap->HttpInstance->NextMsg = Data;\r
   } else {\r
     Wrap->HttpInstance->NextMsg = NULL;\r
   }\r
-  \r
-\r
-  //\r
-  // Free Tx4Token or Tx6Token since already received corrsponding HTTP response.\r
-  //\r
-  FreePool (Wrap);\r
 \r
   return EFI_SUCCESS;\r
 }\r
@@ -940,7 +951,7 @@ HttpBodyParserCallback (
 \r
   @retval EFI_SUCCESS             Allocation succeeded.\r
   @retval EFI_OUT_OF_RESOURCES    Failed to complete the opration due to lack of resources.\r
-  @retval EFI_NOT_READY           Can't find a corresponding Tx4Token/Tx6Token or \r
+  @retval EFI_NOT_READY           Can't find a corresponding Tx4Token/Tx6Token or\r
                                   the EFI_HTTP_UTILITIES_PROTOCOL is not available.\r
 \r
 **/\r
@@ -970,7 +981,7 @@ HttpResponseWorker (
   if (Wrap == NULL || Wrap->HttpInstance == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
-  \r
+\r
   HttpInstance = Wrap->HttpInstance;\r
   Token = Wrap->HttpToken;\r
   HttpMsg = Token->Message;\r
@@ -985,7 +996,7 @@ HttpResponseWorker (
   ValueInItem               = NULL;\r
   Fragment.Len              = 0;\r
   Fragment.Bulk             = NULL;\r
\r
+\r
   if (HttpMsg->Data.Response != NULL) {\r
     //\r
     // Check whether we have cached header from previous call.\r
@@ -1012,8 +1023,8 @@ HttpResponseWorker (
       //\r
       // Check whether we cached the whole HTTP headers.\r
       //\r
-      EndofHeader = AsciiStrStr (HttpHeaders, HTTP_END_OF_HDR_STR); \r
-    }   \r
+      EndofHeader = AsciiStrStr (HttpHeaders, HTTP_END_OF_HDR_STR);\r
+    }\r
 \r
     HttpInstance->EndofHeader = &EndofHeader;\r
     HttpInstance->HttpHeaders = &HttpHeaders;\r
@@ -1112,7 +1123,7 @@ HttpResponseWorker (
     ValueInItem = NULL;\r
 \r
     //\r
-    // In cases of PUT/POST, after an initial request-response pair, we would do a\r
+    // In cases of PUT/POST/PATCH, after an initial request-response pair, we would do a\r
     // continuous request without a response call. So, we would not do an insert of\r
     // TxToken. After we have sent the complete file, we will call a response to get\r
     // a final response from server. In such a case, we would not have any TxTokens.\r
@@ -1178,7 +1189,7 @@ HttpResponseWorker (
                  HttpMsg->HeaderCount,\r
                  HttpMsg->Headers,\r
                  HttpBodyParserCallback,\r
-                 (VOID *) ValueInItem,\r
+                 (VOID *) (&HttpInstance->CallbackData),\r
                  &HttpInstance->MsgParser\r
                  );\r
       if (EFI_ERROR (Status)) {\r
@@ -1189,18 +1200,28 @@ HttpResponseWorker (
       // Check whether we received a complete HTTP message.\r
       //\r
       if (HttpInstance->CacheBody != NULL) {\r
+        //\r
+        // Record the CallbackData data.\r
+        //\r
+        HttpInstance->CallbackData.Wrap = (VOID *) Wrap;\r
+        HttpInstance->CallbackData.ParseData = (VOID *) HttpInstance->CacheBody;\r
+        HttpInstance->CallbackData.ParseDataLength = HttpInstance->CacheLen;\r
+\r
+        //\r
+        // Parse message with CallbackData data.\r
+        //\r
         Status = HttpParseMessageBody (HttpInstance->MsgParser, HttpInstance->CacheLen, HttpInstance->CacheBody);\r
         if (EFI_ERROR (Status)) {\r
           goto Error2;\r
         }\r
+      }\r
 \r
-        if (HttpIsMessageComplete (HttpInstance->MsgParser)) {\r
-          //\r
-          // Free the MsgParse since we already have a full HTTP message.\r
-          //\r
-          HttpFreeMsgParser (HttpInstance->MsgParser);\r
-          HttpInstance->MsgParser = NULL;\r
-        }\r
+      if (HttpIsMessageComplete (HttpInstance->MsgParser)) {\r
+        //\r
+        // Free the MsgParse since we already have a full HTTP message.\r
+        //\r
+        HttpFreeMsgParser (HttpInstance->MsgParser);\r
+        HttpInstance->MsgParser = NULL;\r
       }\r
     }\r
 \r
@@ -1226,7 +1247,7 @@ HttpResponseWorker (
       //\r
       // We have a cached HTTP message which includes a part of HTTP header of next message.\r
       //\r
-      BodyLen = HttpInstance->NextMsg - (HttpInstance->CacheBody + HttpInstance->CacheOffset);      \r
+      BodyLen = HttpInstance->NextMsg - (HttpInstance->CacheBody + HttpInstance->CacheOffset);\r
     } else {\r
       BodyLen = HttpInstance->CacheLen - HttpInstance->CacheOffset;\r
     }\r
@@ -1261,7 +1282,7 @@ HttpResponseWorker (
       //\r
       Status = EFI_SUCCESS;\r
       goto Exit;\r
-    } \r
+    }\r
 \r
     if (BodyLen == 0 && HttpInstance->MsgParser == NULL) {\r
       //\r
@@ -1269,8 +1290,8 @@ HttpResponseWorker (
       //\r
       HttpMsg->BodyLength = 0;\r
       Status = EFI_SUCCESS;\r
-      goto Exit;      \r
-    }    \r
+      goto Exit;\r
+    }\r
   }\r
 \r
   ASSERT (HttpInstance->MsgParser != NULL);\r
@@ -1284,7 +1305,7 @@ HttpResponseWorker (
     if (EFI_ERROR (Status)) {\r
       goto Error2;\r
     }\r
-    \r
+\r
   } else {\r
     if (HttpInstance->TimeoutEvent == NULL) {\r
       //\r
@@ -1309,22 +1330,36 @@ HttpResponseWorker (
     if (EFI_ERROR (Status)) {\r
       goto Error2;\r
     }\r
-  \r
+\r
     Status = HttpsReceive (HttpInstance, &Fragment, HttpInstance->TimeoutEvent);\r
 \r
     gBS->SetTimer (HttpInstance->TimeoutEvent, TimerCancel, 0);\r
-    \r
+\r
     if (EFI_ERROR (Status)) {\r
       goto Error2;\r
     }\r
 \r
     //\r
-    // Check whether we receive a complete HTTP message.\r
+    // Process the received the body packet.\r
+    //\r
+    HttpMsg->BodyLength = MIN (Fragment.Len, (UINT32) HttpMsg->BodyLength);\r
+\r
+    CopyMem (HttpMsg->Body, Fragment.Bulk, HttpMsg->BodyLength);\r
+\r
+    //\r
+    // Record the CallbackData data.\r
+    //\r
+    HttpInstance->CallbackData.Wrap = (VOID *) Wrap;\r
+    HttpInstance->CallbackData.ParseData = HttpMsg->Body;\r
+    HttpInstance->CallbackData.ParseDataLength = HttpMsg->BodyLength;\r
+\r
+    //\r
+    // Parse Body with CallbackData data.\r
     //\r
     Status = HttpParseMessageBody (\r
                HttpInstance->MsgParser,\r
-               (UINTN) Fragment.Len,\r
-               (CHAR8 *) Fragment.Bulk\r
+               HttpMsg->BodyLength,\r
+               HttpMsg->Body\r
                );\r
     if (EFI_ERROR (Status)) {\r
       goto Error2;\r
@@ -1339,46 +1374,28 @@ HttpResponseWorker (
     }\r
 \r
     //\r
-    // We receive part of header of next HTTP msg.\r
+    // Check whether there is the next message header in the HttpMsg->Body.\r
     //\r
     if (HttpInstance->NextMsg != NULL) {\r
-      HttpMsg->BodyLength = MIN ((UINTN) HttpInstance->NextMsg - (UINTN) Fragment.Bulk, HttpMsg->BodyLength);\r
-      CopyMem (HttpMsg->Body, Fragment.Bulk, HttpMsg->BodyLength);\r
-      \r
-      HttpInstance->CacheLen = Fragment.Len - HttpMsg->BodyLength;\r
-      if (HttpInstance->CacheLen != 0) {\r
-        if (HttpInstance->CacheBody != NULL) {\r
-          FreePool (HttpInstance->CacheBody);\r
-        }\r
-        \r
-        HttpInstance->CacheBody = AllocateZeroPool (HttpInstance->CacheLen);\r
-        if (HttpInstance->CacheBody == NULL) {\r
-          Status = EFI_OUT_OF_RESOURCES;\r
-          goto Error2;\r
-        }\r
-        \r
-        CopyMem (HttpInstance->CacheBody, Fragment.Bulk + HttpMsg->BodyLength, HttpInstance->CacheLen);\r
-        HttpInstance->CacheOffset = 0;\r
+      HttpMsg->BodyLength = HttpInstance->NextMsg - (CHAR8 *) HttpMsg->Body;\r
+    }\r
 \r
-        HttpInstance->NextMsg = HttpInstance->CacheBody + ((UINTN) HttpInstance->NextMsg - (UINTN) (Fragment.Bulk + HttpMsg->BodyLength));\r
+    HttpInstance->CacheLen = Fragment.Len - HttpMsg->BodyLength;\r
+    if (HttpInstance->CacheLen != 0) {\r
+      if (HttpInstance->CacheBody != NULL) {\r
+        FreePool (HttpInstance->CacheBody);\r
       }\r
-    } else {\r
-      HttpMsg->BodyLength = MIN (Fragment.Len, (UINT32) HttpMsg->BodyLength);\r
-      CopyMem (HttpMsg->Body, Fragment.Bulk, HttpMsg->BodyLength);\r
-      HttpInstance->CacheLen = Fragment.Len - HttpMsg->BodyLength;\r
-      if (HttpInstance->CacheLen != 0) {\r
-        if (HttpInstance->CacheBody != NULL) {\r
-          FreePool (HttpInstance->CacheBody);\r
-        }\r
-        \r
-        HttpInstance->CacheBody = AllocateZeroPool (HttpInstance->CacheLen);\r
-        if (HttpInstance->CacheBody == NULL) {\r
-          Status = EFI_OUT_OF_RESOURCES;\r
-          goto Error2;\r
-        }\r
 \r
-        CopyMem (HttpInstance->CacheBody, Fragment.Bulk + HttpMsg->BodyLength, HttpInstance->CacheLen);\r
-        HttpInstance->CacheOffset = 0;\r
+      HttpInstance->CacheBody = AllocateZeroPool (HttpInstance->CacheLen);\r
+      if (HttpInstance->CacheBody == NULL) {\r
+        Status = EFI_OUT_OF_RESOURCES;\r
+        goto Error2;\r
+      }\r
+\r
+      CopyMem (HttpInstance->CacheBody, Fragment.Bulk + HttpMsg->BodyLength, HttpInstance->CacheLen);\r
+      HttpInstance->CacheOffset = 0;\r
+      if (HttpInstance->NextMsg != NULL) {\r
+        HttpInstance->NextMsg = HttpInstance->CacheBody;\r
       }\r
     }\r
 \r
@@ -1425,7 +1442,7 @@ Error:
   } else {\r
     FreePool (Wrap);\r
   }\r
-  \r
+\r
   if (HttpHeaders != NULL) {\r
     FreePool (HttpHeaders);\r
     HttpHeaders = NULL;\r
@@ -1454,7 +1471,7 @@ Error:
 \r
   gBS->SignalEvent (Token->Event);\r
 \r
-  return Status;  \r
+  return Status;\r
 \r
 }\r
 \r
@@ -1526,9 +1543,8 @@ EfiHttpResponse (
   if (HttpMsg == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
-  \r
+\r
   HttpInstance = HTTP_INSTANCE_FROM_PROTOCOL (This);\r
-  ASSERT (HttpInstance != NULL);\r
 \r
   if (HttpInstance->State != HTTP_STATE_TCP_CONNECTED) {\r
     return EFI_NOT_STARTED;\r
@@ -1538,7 +1554,7 @@ EfiHttpResponse (
   // Check whether the token already existed.\r
   //\r
   if (EFI_ERROR (NetMapIterate (&HttpInstance->RxTokens, HttpTokenExist, Token))) {\r
-    return EFI_ACCESS_DENIED;   \r
+    return EFI_ACCESS_DENIED;\r
   }\r
 \r
   Wrap = AllocateZeroPool (sizeof (HTTP_TOKEN_WRAP));\r
@@ -1550,8 +1566,8 @@ EfiHttpResponse (
   Wrap->HttpToken    = Token;\r
 \r
   //\r
-  // Notes: For Https, receive token wrapped in HTTP_TOKEN_WRAP is not used to \r
-  // receive the https response. A special TlsRxToken is used for receiving TLS \r
+  // Notes: For Https, receive token wrapped in HTTP_TOKEN_WRAP is not used to\r
+  // receive the https response. A special TlsRxToken is used for receiving TLS\r
   // related messages. It should be a blocking response.\r
   //\r
   if (!HttpInstance->UseHttps) {\r
@@ -1585,9 +1601,9 @@ Error:
       gBS->CloseEvent (Wrap->TcpWrap.Rx6Token.CompletionToken.Event);\r
     }\r
     FreePool (Wrap);\r
-  }  \r
+  }\r
 \r
-  return Status;  \r
+  return Status;\r
 }\r
 \r
 /**\r
@@ -1624,12 +1640,11 @@ EfiHttpPoll (
   }\r
 \r
   HttpInstance = HTTP_INSTANCE_FROM_PROTOCOL (This);\r
-  ASSERT (HttpInstance != NULL);\r
 \r
   if (HttpInstance->State != HTTP_STATE_TCP_CONNECTED) {\r
     return EFI_NOT_STARTED;\r
   }\r
-  \r
+\r
   if (HttpInstance->LocalAddressIsIPv6) {\r
     if (HttpInstance->Tcp6 == NULL) {\r
       return EFI_NOT_STARTED;\r
@@ -1641,8 +1656,8 @@ EfiHttpPoll (
     }\r
     Status = HttpInstance->Tcp4->Poll (HttpInstance->Tcp4);\r
   }\r
-  \r
+\r
   DispatchDpc ();\r
\r
+\r
   return Status;\r
 }\r