Blink roll 25b6bd3a7a131ffe68d809546ad1a20707915cdc:3a503f41ae42e5b79cfcd2ff10e65afde...
[chromium-blink-merge.git] / content / browser / loader / upload_data_stream_builder.cc
blob376e221d2a6e2ebd8b7f26247ecb6a9b57697a12
1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "content/browser/loader/upload_data_stream_builder.h"
7 #include "base/logging.h"
8 #include "content/browser/fileapi/upload_file_system_file_element_reader.h"
9 #include "content/common/resource_request_body.h"
10 #include "net/base/elements_upload_data_stream.h"
11 #include "net/base/upload_bytes_element_reader.h"
12 #include "net/base/upload_file_element_reader.h"
13 #include "storage/browser/blob/blob_data_handle.h"
14 #include "storage/browser/blob/blob_storage_context.h"
16 using storage::BlobData;
17 using storage::BlobDataHandle;
18 using storage::BlobStorageContext;
20 namespace content {
21 namespace {
23 // A subclass of net::UploadBytesElementReader which owns ResourceRequestBody.
24 class BytesElementReader : public net::UploadBytesElementReader {
25 public:
26 BytesElementReader(ResourceRequestBody* resource_request_body,
27 const ResourceRequestBody::Element& element)
28 : net::UploadBytesElementReader(element.bytes(), element.length()),
29 resource_request_body_(resource_request_body) {
30 DCHECK_EQ(ResourceRequestBody::Element::TYPE_BYTES, element.type());
33 ~BytesElementReader() override {}
35 private:
36 scoped_refptr<ResourceRequestBody> resource_request_body_;
38 DISALLOW_COPY_AND_ASSIGN(BytesElementReader);
41 // A subclass of net::UploadFileElementReader which owns ResourceRequestBody.
42 // This class is necessary to ensure the BlobData and any attached shareable
43 // files survive until upload completion.
44 class FileElementReader : public net::UploadFileElementReader {
45 public:
46 FileElementReader(ResourceRequestBody* resource_request_body,
47 base::TaskRunner* task_runner,
48 const ResourceRequestBody::Element& element)
49 : net::UploadFileElementReader(task_runner,
50 element.path(),
51 element.offset(),
52 element.length(),
53 element.expected_modification_time()),
54 resource_request_body_(resource_request_body) {
55 DCHECK_EQ(ResourceRequestBody::Element::TYPE_FILE, element.type());
58 ~FileElementReader() override {}
60 private:
61 scoped_refptr<ResourceRequestBody> resource_request_body_;
63 DISALLOW_COPY_AND_ASSIGN(FileElementReader);
66 void ResolveBlobReference(
67 storage::BlobStorageContext* blob_context,
68 const ResourceRequestBody::Element& element,
69 std::vector<const ResourceRequestBody::Element*>* resolved_elements) {
70 DCHECK(blob_context);
71 scoped_ptr<storage::BlobDataHandle> handle =
72 blob_context->GetBlobDataFromUUID(element.blob_uuid());
73 DCHECK(handle);
74 if (!handle)
75 return;
77 // If there is no element in the referred blob data, just return.
78 if (handle->data()->items().empty())
79 return;
81 // Append the elements in the referenced blob data.
82 for (size_t i = 0; i < handle->data()->items().size(); ++i) {
83 const BlobData::Item& item = handle->data()->items().at(i);
84 DCHECK_NE(BlobData::Item::TYPE_BLOB, item.type());
85 resolved_elements->push_back(&item);
89 } // namespace
91 scoped_ptr<net::UploadDataStream> UploadDataStreamBuilder::Build(
92 ResourceRequestBody* body,
93 BlobStorageContext* blob_context,
94 storage::FileSystemContext* file_system_context,
95 base::TaskRunner* file_task_runner) {
96 // Resolve all blob elements.
97 std::vector<const ResourceRequestBody::Element*> resolved_elements;
98 for (size_t i = 0; i < body->elements()->size(); ++i) {
99 const ResourceRequestBody::Element& element = (*body->elements())[i];
100 if (element.type() == ResourceRequestBody::Element::TYPE_BLOB)
101 ResolveBlobReference(blob_context, element, &resolved_elements);
102 else
103 resolved_elements.push_back(&element);
106 ScopedVector<net::UploadElementReader> element_readers;
107 for (size_t i = 0; i < resolved_elements.size(); ++i) {
108 const ResourceRequestBody::Element& element = *resolved_elements[i];
109 switch (element.type()) {
110 case ResourceRequestBody::Element::TYPE_BYTES:
111 element_readers.push_back(new BytesElementReader(body, element));
112 break;
113 case ResourceRequestBody::Element::TYPE_FILE:
114 element_readers.push_back(
115 new FileElementReader(body, file_task_runner, element));
116 break;
117 case ResourceRequestBody::Element::TYPE_FILE_FILESYSTEM:
118 // If |body| contains any filesystem URLs, the caller should have
119 // supplied a FileSystemContext.
120 DCHECK(file_system_context);
121 element_readers.push_back(
122 new content::UploadFileSystemFileElementReader(
123 file_system_context,
124 element.filesystem_url(),
125 element.offset(),
126 element.length(),
127 element.expected_modification_time()));
128 break;
129 case ResourceRequestBody::Element::TYPE_BLOB:
130 // Blob elements should be resolved beforehand.
131 NOTREACHED();
132 break;
133 case ResourceRequestBody::Element::TYPE_UNKNOWN:
134 NOTREACHED();
135 break;
139 return make_scoped_ptr(
140 new net::ElementsUploadDataStream(element_readers.Pass(),
141 body->identifier()));
144 } // namespace content