1 // Copyright (c) 2011 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.
4
5 #include "storage/browser/blob/blob_url_request_job_factory.h"
6
7 #include "base/basictypes.h"
8 #include "base/compiler_specific.h"
9 #include "base/message_loop/message_loop_proxy.h"
10 #include "base/strings/string_util.h"
11 #include "net/base/request_priority.h"
12 #include "net/url_request/url_request_context.h"
13 #include "net/url_request/url_request_job_factory.h"
14 #include "storage/browser/blob/blob_data_handle.h"
15 #include "storage/browser/blob/blob_storage_context.h"
16 #include "storage/browser/blob/blob_url_request_job.h"
17 #include "storage/browser/fileapi/file_system_context.h"
18
19 namespace storage {
20
21 namespace {
22
23 int kUserDataKey; // The value is not important, the addr is a key.
24
GetRequestedBlobDataHandle(net::URLRequest * request)25 BlobDataHandle* GetRequestedBlobDataHandle(net::URLRequest* request) {
26 return static_cast<BlobDataHandle*>(request->GetUserData(&kUserDataKey));
27 }
28
29 } // namespace
30
31 // static
CreateBlobRequest(scoped_ptr<BlobDataHandle> blob_data_handle,const net::URLRequestContext * request_context,net::URLRequest::Delegate * request_delegate)32 scoped_ptr<net::URLRequest> BlobProtocolHandler::CreateBlobRequest(
33 scoped_ptr<BlobDataHandle> blob_data_handle,
34 const net::URLRequestContext* request_context,
35 net::URLRequest::Delegate* request_delegate) {
36 const GURL kBlobUrl("blob://see_user_data/");
37 scoped_ptr<net::URLRequest> request = request_context->CreateRequest(
38 kBlobUrl, net::DEFAULT_PRIORITY, request_delegate, NULL);
39 SetRequestedBlobDataHandle(request.get(), blob_data_handle.Pass());
40 return request.Pass();
41 }
42
43 // static
SetRequestedBlobDataHandle(net::URLRequest * request,scoped_ptr<BlobDataHandle> blob_data_handle)44 void BlobProtocolHandler::SetRequestedBlobDataHandle(
45 net::URLRequest* request,
46 scoped_ptr<BlobDataHandle> blob_data_handle) {
47 request->SetUserData(&kUserDataKey, blob_data_handle.release());
48 }
49
BlobProtocolHandler(BlobStorageContext * context,storage::FileSystemContext * file_system_context,const scoped_refptr<base::MessageLoopProxy> & loop_proxy)50 BlobProtocolHandler::BlobProtocolHandler(
51 BlobStorageContext* context,
52 storage::FileSystemContext* file_system_context,
53 const scoped_refptr<base::MessageLoopProxy>& loop_proxy)
54 : file_system_context_(file_system_context), file_loop_proxy_(loop_proxy) {
55 if (context)
56 context_ = context->AsWeakPtr();
57 }
58
~BlobProtocolHandler()59 BlobProtocolHandler::~BlobProtocolHandler() {
60 }
61
MaybeCreateJob(net::URLRequest * request,net::NetworkDelegate * network_delegate) const62 net::URLRequestJob* BlobProtocolHandler::MaybeCreateJob(
63 net::URLRequest* request, net::NetworkDelegate* network_delegate) const {
64 return new storage::BlobURLRequestJob(request,
65 network_delegate,
66 LookupBlobData(request),
67 file_system_context_.get(),
68 file_loop_proxy_.get());
69 }
70
LookupBlobData(net::URLRequest * request) const71 scoped_refptr<storage::BlobData> BlobProtocolHandler::LookupBlobData(
72 net::URLRequest* request) const {
73 BlobDataHandle* blob_data_handle = GetRequestedBlobDataHandle(request);
74 if (blob_data_handle)
75 return blob_data_handle->data();
76 if (!context_.get())
77 return NULL;
78
79 // Support looking up based on uuid, the FeedbackExtensionAPI relies on this.
80 // TODO(michaeln): Replace this use case and others like it with a BlobReader
81 // impl that does not depend on urlfetching to perform this function.
82 const std::string kPrefix("blob:uuid/");
83 if (!StartsWithASCII(request->url().spec(), kPrefix, true))
84 return NULL;
85 std::string uuid = request->url().spec().substr(kPrefix.length());
86 scoped_ptr<BlobDataHandle> handle = context_->GetBlobDataFromUUID(uuid);
87 return handle.get() ? handle->data() : NULL;
88 }
89
90 } // namespace storage
91