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 #ifndef CONTENT_BROWSER_WEBUI_URL_DATA_MANAGER_BACKEND_H_ | 5 #ifndef CONTENT_BROWSER_WEBUI_URL_DATA_MANAGER_BACKEND_H_ |
6 #define CONTENT_BROWSER_WEBUI_URL_DATA_MANAGER_BACKEND_H_ | 6 #define CONTENT_BROWSER_WEBUI_URL_DATA_MANAGER_BACKEND_H_ |
7 | 7 |
8 #include <map> | 8 #include <map> |
9 #include <memory> | 9 #include <memory> |
10 #include <string> | 10 #include <string> |
11 #include <vector> | 11 #include <vector> |
12 | 12 |
13 #include "base/compiler_specific.h" | 13 #include "base/compiler_specific.h" |
14 #include "base/macros.h" | 14 #include "base/macros.h" |
15 #include "base/supports_user_data.h" | 15 #include "base/supports_user_data.h" |
16 #include "base/values.h" | 16 #include "base/values.h" |
17 #include "content/browser/webui/url_data_manager.h" | 17 #include "content/browser/webui/url_data_manager.h" |
18 #include "content/public/browser/url_data_source.h" | 18 #include "content/public/browser/url_data_source.h" |
| 19 #include "net/http/http_response_headers.h" |
19 #include "net/url_request/url_request_job_factory.h" | 20 #include "net/url_request/url_request_job_factory.h" |
20 | 21 |
21 class GURL; | 22 class GURL; |
22 | 23 |
23 namespace base { | 24 namespace base { |
24 class RefCountedMemory; | 25 class RefCountedMemory; |
25 } | 26 } |
26 | 27 |
27 namespace content { | 28 namespace content { |
28 | 29 |
29 class ChromeBlobStorageContext; | 30 class ChromeBlobStorageContext; |
30 class ResourceContext; | 31 class ResourceContext; |
31 class URLDataManagerBackend; | 32 class URLDataManagerBackend; |
32 class URLDataSourceImpl; | 33 class URLDataSourceImpl; |
33 class URLRequestChromeJob; | 34 class URLRequestChromeJob; |
34 | 35 |
35 // URLDataManagerBackend is used internally by ChromeURLDataManager on the IO | 36 // URLDataManagerBackend is used internally by ChromeURLDataManager on the IO |
36 // thread. In most cases you can use the API in ChromeURLDataManager and ignore | 37 // thread. In most cases you can use the API in ChromeURLDataManager and ignore |
37 // this class. URLDataManagerBackend is owned by ResourceContext. | 38 // this class. URLDataManagerBackend is owned by ResourceContext. |
38 class URLDataManagerBackend : public base::SupportsUserData::Data { | 39 class URLDataManagerBackend : public base::SupportsUserData::Data { |
39 public: | 40 public: |
40 typedef int RequestID; | 41 typedef int RequestID; |
41 | 42 |
42 URLDataManagerBackend(); | 43 URLDataManagerBackend(); |
43 ~URLDataManagerBackend() override; | 44 ~URLDataManagerBackend() override; |
44 | 45 |
45 // Invoked to create the protocol handler for chrome://. |is_incognito| should | 46 // Invoked to create the protocol handler for chrome://. Called on the UI |
46 // be set for incognito profiles. Called on the UI thread. | 47 // thread. |
47 CONTENT_EXPORT static std::unique_ptr< | 48 CONTENT_EXPORT static std::unique_ptr< |
48 net::URLRequestJobFactory::ProtocolHandler> | 49 net::URLRequestJobFactory::ProtocolHandler> |
49 CreateProtocolHandler(content::ResourceContext* resource_context, | 50 CreateProtocolHandler(ResourceContext* resource_context, |
50 bool is_incognito, | |
51 ChromeBlobStorageContext* blob_storage_context); | 51 ChromeBlobStorageContext* blob_storage_context); |
52 | 52 |
53 // Adds a DataSource to the collection of data sources. | 53 // Adds a DataSource to the collection of data sources. |
54 void AddDataSource(URLDataSourceImpl* source); | 54 void AddDataSource(URLDataSourceImpl* source); |
55 | 55 |
56 void UpdateWebUIDataSource(const std::string& source_name, | 56 void UpdateWebUIDataSource(const std::string& source_name, |
57 const base::DictionaryValue& update); | 57 const base::DictionaryValue& update); |
58 | 58 |
59 // DataSource invokes this. Sends the data to the URLRequest. |bytes| may be | 59 // DataSource invokes this. Sends the data to the URLRequest. |bytes| may be |
60 // null, which signals an error handling the request. | 60 // null, which signals an error handling the request. |
61 void DataAvailable(RequestID request_id, base::RefCountedMemory* bytes); | 61 void DataAvailable(RequestID request_id, base::RefCountedMemory* bytes); |
62 | 62 |
63 static net::URLRequestJob* Factory(net::URLRequest* request, | 63 // Look up the data source for the request. Returns the source if it is found, |
64 const std::string& scheme); | 64 // else NULL. |
| 65 URLDataSourceImpl* GetDataSourceFromURL(const GURL& url); |
| 66 |
| 67 // Creates and sets the response headers for the given request. |
| 68 static scoped_refptr<net::HttpResponseHeaders> GetHeaders( |
| 69 URLDataSourceImpl* source, |
| 70 const std::string& path, |
| 71 const std::string& origin); |
| 72 |
| 73 // Returns whether |url| passes some sanity checks and is a valid GURL. |
| 74 static bool CheckURLIsValid(const GURL& url); |
| 75 |
| 76 // Parse |url| to get the path which will be used to resolve the request. The |
| 77 // path is the remaining portion after the scheme and hostname. |
| 78 static void URLToRequestPath(const GURL& url, std::string* path); |
65 | 79 |
66 private: | 80 private: |
67 friend class URLRequestChromeJob; | 81 friend class URLRequestChromeJob; |
68 | 82 |
69 typedef std::map<std::string, | 83 typedef std::map<std::string, |
70 scoped_refptr<URLDataSourceImpl> > DataSourceMap; | 84 scoped_refptr<URLDataSourceImpl> > DataSourceMap; |
71 typedef std::map<RequestID, URLRequestChromeJob*> PendingRequestMap; | 85 typedef std::map<RequestID, URLRequestChromeJob*> PendingRequestMap; |
72 | 86 |
73 // Called by the job when it's starting up. | 87 // Called by the job when it's starting up. |
74 // Returns false if |url| is not a URL managed by this object. | 88 // Returns false if |url| is not a URL managed by this object. |
75 bool StartRequest(const net::URLRequest* request, URLRequestChromeJob* job); | 89 bool StartRequest(const net::URLRequest* request, URLRequestChromeJob* job); |
76 | 90 |
77 // Helper function to call StartDataRequest on |source|'s delegate. This is | 91 // Helper function to call StartDataRequest on |source|'s delegate. This is |
78 // needed because while we want to call URLDataSourceDelegate's method, we | 92 // needed because while we want to call URLDataSourceDelegate's method, we |
79 // need to add a refcount on the source. | 93 // need to add a refcount on the source. |
80 static void CallStartRequest( | 94 static void CallStartRequest( |
81 scoped_refptr<URLDataSourceImpl> source, | 95 scoped_refptr<URLDataSourceImpl> source, |
82 const std::string& path, | 96 const std::string& path, |
83 const ResourceRequestInfo::WebContentsGetter& wc_getter, | 97 const ResourceRequestInfo::WebContentsGetter& wc_getter, |
84 int request_id); | 98 int request_id); |
85 | 99 |
86 // Remove a request from the list of pending requests. | 100 // Remove a request from the list of pending requests. |
87 void RemoveRequest(URLRequestChromeJob* job); | 101 void RemoveRequest(URLRequestChromeJob* job); |
88 | 102 |
89 // Returns true if the job exists in |pending_requests_|. False otherwise. | 103 // Returns true if the job exists in |pending_requests_|. False otherwise. |
90 // Called by ~URLRequestChromeJob to verify that |pending_requests_| is kept | 104 // Called by ~URLRequestChromeJob to verify that |pending_requests_| is kept |
91 // up to date. | 105 // up to date. |
92 bool HasPendingJob(URLRequestChromeJob* job) const; | 106 bool HasPendingJob(URLRequestChromeJob* job) const; |
93 | 107 |
94 // Look up the data source for the request. Returns the source if it is found, | |
95 // else NULL. | |
96 URLDataSourceImpl* GetDataSourceFromURL(const GURL& url); | |
97 | |
98 // Custom sources of data, keyed by source path (e.g. "favicon"). | 108 // Custom sources of data, keyed by source path (e.g. "favicon"). |
99 DataSourceMap data_sources_; | 109 DataSourceMap data_sources_; |
100 | 110 |
101 // All pending URLRequestChromeJobs, keyed by ID of the request. | 111 // All pending URLRequestChromeJobs, keyed by ID of the request. |
102 // URLRequestChromeJob calls into this object when it's constructed and | 112 // URLRequestChromeJob calls into this object when it's constructed and |
103 // destructed to ensure that the pointers in this map remain valid. | 113 // destructed to ensure that the pointers in this map remain valid. |
104 PendingRequestMap pending_requests_; | 114 PendingRequestMap pending_requests_; |
105 | 115 |
106 // The ID we'll use for the next request we receive. | 116 // The ID we'll use for the next request we receive. |
107 RequestID next_request_id_; | 117 RequestID next_request_id_; |
108 | 118 |
109 DISALLOW_COPY_AND_ASSIGN(URLDataManagerBackend); | 119 DISALLOW_COPY_AND_ASSIGN(URLDataManagerBackend); |
110 }; | 120 }; |
111 | 121 |
112 // Creates protocol handler for chrome-devtools://. |is_incognito| should be | 122 // Creates protocol handler for chrome-devtools://. |
113 // set for incognito profiles. | 123 net::URLRequestJobFactory::ProtocolHandler* CreateDevToolsProtocolHandler( |
114 net::URLRequestJobFactory::ProtocolHandler* | 124 ResourceContext* resource_context); |
115 CreateDevToolsProtocolHandler(content::ResourceContext* resource_context, | |
116 bool is_incognito); | |
117 | 125 |
118 } // namespace content | 126 } // namespace content |
119 | 127 |
120 #endif // CONTENT_BROWSER_WEBUI_URL_DATA_MANAGER_BACKEND_H_ | 128 #endif // CONTENT_BROWSER_WEBUI_URL_DATA_MANAGER_BACKEND_H_ |
OLD | NEW |