Split net::UploadData into two: for IPC and for upload handling

webkit/glue/resource_request_body.cc

+// Copyright (c) 2012 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 "webkit/glue/resource_request_body.h"
+
+#include "base/logging.h"
+#include "net/base/upload_data.h"
+#include "webkit/blob/blob_storage_controller.h"
+
+using webkit_blob::BlobData;
+using webkit_blob::BlobStorageController;
+
+namespace webkit_glue {
+
+ResourceRequestBody::Element::Element()
+    : type_(TYPE_BYTES),
+      bytes_start_(NULL),
+      bytes_length_(0),
+      file_range_offset_(0),
+      file_range_length_(kuint64max) {
+}
+
+ResourceRequestBody::Element::~Element() {}
+
+ResourceRequestBody::ResourceRequestBody() : identifier_(0) {}
+
+void ResourceRequestBody::AppendBytes(const char* bytes, int bytes_len) {
+  if (bytes_len > 0) {
+    elements_.push_back(Element());
+    elements_.back().SetToBytes(bytes, bytes_len);
+  }
+}
+
+void ResourceRequestBody::AppendFileRange(
+    const FilePath& file_path,
+    uint64 offset, uint64 length,
+    const base::Time& expected_modification_time) {
+  elements_.push_back(Element());
+  elements_.back().SetToFilePathRange(file_path, offset, length,
+                                      expected_modification_time);
+}
+
+void ResourceRequestBody::AppendBlob(const GURL& blob_url) {
+  elements_.push_back(Element());
+  elements_.back().SetToBlobUrl(blob_url);
+}
+
+net::UploadData* ResourceRequestBody::ResolveElementsAndCreateUploadData(
+    BlobStorageController* blob_controller) {
+  net::UploadData* upload_data = new net::UploadData;
+  // We attach 'this' to UploadData so that we do not need to copy
+  // bytes for TYPE_BYTES.
+  upload_data->SetUserData(
+      this, new base::UserDataAdapter<ResourceRequestBody>(this));
+  std::vector<net::UploadElement>* elements =
+      upload_data->elements_mutable();
+  for (size_t i = 0; i < elements_.size(); ++i) {
+    const Element& element = elements_[i];
+    switch (element.type()) {
+      case TYPE_BYTES:
+        elements->push_back(net::UploadElement());
+        elements->back().SetToSharedBytes(element.bytes(),
+                                        element.bytes_length());
+        break;
+      case TYPE_FILE:
+        elements->push_back(net::UploadElement());
+        elements->back().SetToFilePathRange(
+            element.file_path(),
+            element.file_range_offset(),
+            element.file_range_length(),
+            element.expected_file_modification_time());
+        break;
+      case TYPE_BLOB:
+        ResolveBlobReference(blob_controller, element.blob_url(), elements);
+        break;
+    }
+  }
+  upload_data->set_identifier(identifier_);
+  return upload_data;
+}
+
+ResourceRequestBody::~ResourceRequestBody() {}
+
+void ResourceRequestBody::ResolveBlobReference(
+    webkit_blob::BlobStorageController* blob_controller,
+    const GURL& blob_url, std::vector<net::UploadElement>* elements) {
+  DCHECK(blob_controller);
+  BlobData* blob_data = blob_controller->GetBlobDataFromUrl(blob_url);
+  DCHECK(blob_data);
+  if (!blob_data)
+    return;
+
+  // If there is no element in the referred blob data, just return true.
+  if (blob_data->items().empty())
+    return;
+
+  // Ensure the blob and any attached shareable files survive until
+  // upload completion.
+  SetUserData(blob_data, new base::UserDataAdapter<BlobData>(blob_data));
+
+  // Append the elements in the referred blob data.
+  for (size_t i = 0; i < blob_data->items().size(); ++i) {
+    elements->push_back(net::UploadElement());
+    net::UploadElement& element = elements->back();
+    const BlobData::Item& item = blob_data->items().at(i);
+    switch (item.type) {
+      case BlobData::TYPE_DATA:
+        element.SetToSharedBytes(
+            &item.data.at(0) + static_cast<int>(item.offset),
+            static_cast<int>(item.length));
+        break;
+      case BlobData::TYPE_FILE:
+        element.SetToFilePathRange(
+            item.file_path,
+            item.offset,
+            item.length,
+            item.expected_modification_time);
+        break;
+      default:
+        NOTREACHED();
+        break;
+    }
+  }
+}
+
+}  // namespace webkit_glue