OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "webkit/blob/blob_storage_controller.h" | 5 #include "webkit/blob/blob_storage_controller.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "googleurl/src/gurl.h" | 8 #include "googleurl/src/gurl.h" |
9 #include "net/base/upload_data.h" | |
10 #include "webkit/blob/blob_data.h" | 9 #include "webkit/blob/blob_data.h" |
11 | 10 |
12 namespace webkit_blob { | 11 namespace webkit_blob { |
13 | 12 |
14 namespace { | 13 namespace { |
15 | 14 |
16 // We can't use GURL directly for these hash fragment manipulations | 15 // We can't use GURL directly for these hash fragment manipulations |
17 // since it doesn't have specific knowlege of the BlobURL format. GURL | 16 // since it doesn't have specific knowlege of the BlobURL format. GURL |
18 // treats BlobURLs as if they were PathURLs which don't support hash | 17 // treats BlobURLs as if they were PathURLs which don't support hash |
19 // fragments. | 18 // fragments. |
(...skipping 142 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
162 return true; | 161 return true; |
163 } | 162 } |
164 | 163 |
165 | 164 |
166 BlobData* BlobStorageController::GetBlobDataFromUrl(const GURL& url) { | 165 BlobData* BlobStorageController::GetBlobDataFromUrl(const GURL& url) { |
167 BlobMap::iterator found = blob_map_.find( | 166 BlobMap::iterator found = blob_map_.find( |
168 BlobUrlHasRef(url) ? ClearBlobUrlRef(url).spec() : url.spec()); | 167 BlobUrlHasRef(url) ? ClearBlobUrlRef(url).spec() : url.spec()); |
169 return (found != blob_map_.end()) ? found->second : NULL; | 168 return (found != blob_map_.end()) ? found->second : NULL; |
170 } | 169 } |
171 | 170 |
172 void BlobStorageController::ResolveBlobReferencesInUploadData( | |
173 net::UploadData* upload_data) { | |
174 DCHECK(upload_data); | |
175 | |
176 std::vector<net::UploadElement>* uploads = | |
177 upload_data->elements_mutable(); | |
178 std::vector<net::UploadElement>::iterator iter; | |
179 for (iter = uploads->begin(); iter != uploads->end();) { | |
180 if (iter->type() != net::UploadElement::TYPE_BLOB) { | |
181 iter++; | |
182 continue; | |
183 } | |
184 | |
185 // Find the referred blob data. | |
186 BlobData* blob_data = GetBlobDataFromUrl(iter->blob_url()); | |
187 DCHECK(blob_data); | |
188 if (!blob_data) { | |
189 // TODO(jianli): We should probably fail uploading the data | |
190 iter++; | |
191 continue; | |
192 } | |
193 | |
194 // Remove this element. | |
195 iter = uploads->erase(iter); | |
196 | |
197 // If there is no element in the referred blob data, continue the loop. | |
198 // Note that we should not increase iter since it already points to the one | |
199 // after the removed element. | |
200 if (blob_data->items().empty()) | |
201 continue; | |
202 | |
203 // Ensure the blob and any attached shareable files survive until | |
204 // upload completion. | |
205 upload_data->SetUserData(blob_data, | |
206 new base::UserDataAdapter<BlobData>(blob_data)); | |
207 | |
208 // Insert the elements in the referred blob data. | |
209 // Note that we traverse from the bottom so that the elements can be | |
210 // inserted in the original order. | |
211 for (size_t i = blob_data->items().size(); i > 0; --i) { | |
212 iter = uploads->insert(iter, net::UploadElement()); | |
213 | |
214 const BlobData::Item& item = blob_data->items().at(i - 1); | |
215 switch (item.type) { | |
216 case BlobData::TYPE_DATA: | |
217 // TODO(jianli): Figure out how to avoid copying the data. | |
218 // TODO(michaeln): Now that blob_data surives for the duration, | |
219 // maybe UploadData could take a raw ptr without having to copy. | |
220 iter->SetToBytes( | |
221 &item.data.at(0) + static_cast<int>(item.offset), | |
222 static_cast<int>(item.length)); | |
223 break; | |
224 case BlobData::TYPE_FILE: | |
225 iter->SetToFilePathRange( | |
226 item.file_path, | |
227 item.offset, | |
228 item.length, | |
229 item.expected_modification_time); | |
230 break; | |
231 default: | |
232 NOTREACHED(); | |
233 break; | |
234 } | |
235 } | |
236 } | |
237 } | |
238 | |
239 void BlobStorageController::AppendStorageItems( | 171 void BlobStorageController::AppendStorageItems( |
240 BlobData* target_blob_data, BlobData* src_blob_data, | 172 BlobData* target_blob_data, BlobData* src_blob_data, |
241 uint64 offset, uint64 length) { | 173 uint64 offset, uint64 length) { |
242 DCHECK(target_blob_data && src_blob_data && | 174 DCHECK(target_blob_data && src_blob_data && |
243 length != static_cast<uint64>(-1)); | 175 length != static_cast<uint64>(-1)); |
244 | 176 |
245 std::vector<BlobData::Item>::const_iterator iter = | 177 std::vector<BlobData::Item>::const_iterator iter = |
246 src_blob_data->items().begin(); | 178 src_blob_data->items().begin(); |
247 if (offset) { | 179 if (offset) { |
248 for (; iter != src_blob_data->items().end(); ++iter) { | 180 for (; iter != src_blob_data->items().end(); ++iter) { |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
294 bool BlobStorageController::DecrementBlobDataUsage(BlobData* blob_data) { | 226 bool BlobStorageController::DecrementBlobDataUsage(BlobData* blob_data) { |
295 BlobDataUsageMap::iterator found = blob_data_usage_count_.find(blob_data); | 227 BlobDataUsageMap::iterator found = blob_data_usage_count_.find(blob_data); |
296 DCHECK(found != blob_data_usage_count_.end()); | 228 DCHECK(found != blob_data_usage_count_.end()); |
297 if (--(found->second)) | 229 if (--(found->second)) |
298 return false; // Still in use | 230 return false; // Still in use |
299 blob_data_usage_count_.erase(found); | 231 blob_data_usage_count_.erase(found); |
300 return true; | 232 return true; |
301 } | 233 } |
302 | 234 |
303 } // namespace webkit_blob | 235 } // namespace webkit_blob |
OLD | NEW |