Deduping conversions between ResourceRequestBody/WebHTTPBody/ExplodedHttpBody.

content/child/web_url_request_util.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/child/web_url_request_util.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <limits>
 
 #include "base/logging.h"
 #include "base/strings/string_util.h"
 #include "net/base/load_flags.h"
 #include "net/base/net_errors.h"
 #include "third_party/WebKit/public/platform/FilePathConversion.h"
 #include "third_party/WebKit/public/platform/WebCachePolicy.h"
+#include "third_party/WebKit/public/platform/WebData.h"
 #include "third_party/WebKit/public/platform/WebHTTPHeaderVisitor.h"
 #include "third_party/WebKit/public/platform/WebString.h"
 #include "third_party/WebKit/public/platform/WebURL.h"
 #include "third_party/WebKit/public/platform/WebURLError.h"
 #include "third_party/WebKit/public/platform/WebURLRequest.h"
 
 using blink::WebCachePolicy;
+using blink::WebData;
 using blink::WebHTTPBody;
 using blink::WebString;
 using blink::WebURLRequest;
 
 namespace content {
 
 namespace {
 
 const char kThrottledErrorDescription[] =
     "Request throttled. Visit http://dev.chromium.org/throttling for more "
@@ skipping 175 matching lines @@
     load_flags |= net::LOAD_DO_NOT_SAVE_COOKIES;
     load_flags |= net::LOAD_DO_NOT_SEND_COOKIES;
   }
 
   if (!request.allowStoredCredentials())
     load_flags |= net::LOAD_DO_NOT_SEND_AUTH_DATA;
 
   return load_flags;
 }
 
+WebHTTPBody GetWebHTTPBodyForRequestBody(
+    const scoped_refptr<ResourceRequestBody>& input) {
+  WebHTTPBody http_body;
+  http_body.initialize();
+  http_body.setIdentifier(input->identifier());
+  for (const auto& element : *input->elements()) {
+    switch (element.type()) {
+      case ResourceRequestBody::Element::TYPE_BYTES:
+        http_body.appendData(WebData(element.bytes(), element.length()));
+        break;
+      case ResourceRequestBody::Element::TYPE_FILE:
+        http_body.appendFileRange(
+            element.path().AsUTF16Unsafe(), element.offset(),
+            (element.length() != std::numeric_limits<uint64_t>::max())
+                ? element.length()
+                : -1,
+            element.expected_modification_time().ToDoubleT());
+        break;
+      case ResourceRequestBody::Element::TYPE_FILE_FILESYSTEM:
+        http_body.appendFileSystemURLRange(
+            element.filesystem_url(), element.offset(),
+            (element.length() != std::numeric_limits<uint64_t>::max())
+                ? element.length()
+                : -1,
+            element.expected_modification_time().ToDoubleT());
+        break;
+      case ResourceRequestBody::Element::TYPE_BLOB:
+        http_body.appendBlob(WebString::fromUTF8(element.blob_uuid()));
+        break;
+      case ResourceRequestBody::Element::TYPE_BYTES_DESCRIPTION:
+      case ResourceRequestBody::Element::TYPE_DISK_CACHE_ENTRY:
+      default:
+        NOTREACHED();
+        break;
+    }
+  }
+  return http_body;
+}
+
 scoped_refptr<ResourceRequestBody> GetRequestBodyForWebURLRequest(
     const blink::WebURLRequest& request) {
   scoped_refptr<ResourceRequestBody> request_body;
 
   if (request.httpBody().isNull()) {
     return request_body;
   }
 
   const std::string& method = request.httpMethod().latin1();
   // GET and HEAD requests shouldn't have http bodies.
   DCHECK(method != "GET" && method != "HEAD");
 
-  const WebHTTPBody& httpBody = request.httpBody();
-  request_body = new ResourceRequestBody();
+  return GetRequestBodyForWebHTTPBody(request.httpBody());
+}
+
+scoped_refptr<ResourceRequestBody> GetRequestBodyForWebHTTPBody(
+    const blink::WebHTTPBody& httpBody) {
+  scoped_refptr<ResourceRequestBody> request_body = new ResourceRequestBody();
   size_t i = 0;
   WebHTTPBody::Element element;
   while (httpBody.elementAt(i++, element)) {
     switch (element.type) {
       case WebHTTPBody::Element::TypeData:
         if (!element.data.isEmpty()) {
           // Blink sometimes gives empty data to append. These aren't
           // necessary so they are just optimized out here.
           request_body->AppendBytes(
               element.data.data(), static_cast<int>(element.data.size()));
@@ skipping 21 matching lines @@
             base::Time::FromDoubleT(element.modificationTime));
         break;
       }
       case WebHTTPBody::Element::TypeBlob:
         request_body->AppendBlob(element.blobUUID.utf8());
         break;
       default:
         NOTREACHED();
     }
   }
-  request_body->set_identifier(request.httpBody().identifier());
+  request_body->set_identifier(httpBody.identifier());
   return request_body;
 }
 
 #define STATIC_ASSERT_ENUM(a, b)                            \
   static_assert(static_cast<int>(a) == static_cast<int>(b), \
                 "mismatching enums: " #a)
 
 STATIC_ASSERT_ENUM(FETCH_REQUEST_MODE_SAME_ORIGIN,
                    WebURLRequest::FetchRequestModeSameOrigin);
 STATIC_ASSERT_ENUM(FETCH_REQUEST_MODE_NO_CORS,
@@ skipping 148 matching lines @@
                                      bool stale_copy_in_cache,
                                      int reason,
                                      bool was_ignored_by_handler) {
   blink::WebURLError error =
       CreateWebURLError(unreachable_url, stale_copy_in_cache, reason);
   error.wasIgnoredByHandler = was_ignored_by_handler;
   return error;
 }
 
 }  // namespace content