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 "chrome/service/cloud_print/cloud_print_url_fetcher.h" | 5 #include "chrome/service/cloud_print/cloud_print_url_fetcher.h" |
6 | 6 |
7 #include "base/stringprintf.h" | 7 #include "base/stringprintf.h" |
8 #include "base/values.h" | 8 #include "base/values.h" |
9 #include "chrome/common/cloud_print/cloud_print_helpers.h" | 9 #include "chrome/common/cloud_print/cloud_print_helpers.h" |
10 #include "chrome/service/cloud_print/cloud_print_consts.h" | 10 #include "chrome/service/cloud_print/cloud_print_consts.h" |
11 #include "chrome/service/cloud_print/cloud_print_helpers.h" | 11 #include "chrome/service/cloud_print/cloud_print_helpers.h" |
12 #include "chrome/service/cloud_print/cloud_print_token_store.h" | 12 #include "chrome/service/cloud_print/cloud_print_token_store.h" |
13 #include "chrome/service/net/service_url_request_context.h" | 13 #include "chrome/service/net/service_url_request_context.h" |
14 #include "chrome/service/service_process.h" | 14 #include "chrome/service/service_process.h" |
| 15 #include "content/public/common/url_fetcher.h" |
15 #include "googleurl/src/gurl.h" | 16 #include "googleurl/src/gurl.h" |
16 #include "net/http/http_status_code.h" | 17 #include "net/http/http_status_code.h" |
17 #include "net/url_request/url_request_status.h" | 18 #include "net/url_request/url_request_status.h" |
18 | 19 |
19 CloudPrintURLFetcher::CloudPrintURLFetcher() | 20 CloudPrintURLFetcher::CloudPrintURLFetcher() |
20 : delegate_(NULL), | 21 : delegate_(NULL), |
21 num_retries_(0) { | 22 num_retries_(0) { |
22 } | 23 } |
23 | 24 |
24 bool CloudPrintURLFetcher::IsSameRequest(const net::URLFetcher* source) { | 25 bool CloudPrintURLFetcher::IsSameRequest(const net::URLFetcher* source) { |
25 return (request_.get() == source); | 26 return (request_.get() == source); |
26 } | 27 } |
27 | 28 |
28 void CloudPrintURLFetcher::StartGetRequest( | 29 void CloudPrintURLFetcher::StartGetRequest( |
29 const GURL& url, | 30 const GURL& url, |
30 Delegate* delegate, | 31 Delegate* delegate, |
31 int max_retries, | 32 int max_retries, |
32 const std::string& additional_headers) { | 33 const std::string& additional_headers) { |
33 StartRequestHelper(url, | 34 StartRequestHelper(url, |
34 content::URLFetcher::GET, | 35 net::URLFetcher::GET, |
35 delegate, | 36 delegate, |
36 max_retries, | 37 max_retries, |
37 std::string(), | 38 std::string(), |
38 std::string(), | 39 std::string(), |
39 additional_headers); | 40 additional_headers); |
40 } | 41 } |
41 | 42 |
42 void CloudPrintURLFetcher::StartPostRequest( | 43 void CloudPrintURLFetcher::StartPostRequest( |
43 const GURL& url, | 44 const GURL& url, |
44 Delegate* delegate, | 45 Delegate* delegate, |
45 int max_retries, | 46 int max_retries, |
46 const std::string& post_data_mime_type, | 47 const std::string& post_data_mime_type, |
47 const std::string& post_data, | 48 const std::string& post_data, |
48 const std::string& additional_headers) { | 49 const std::string& additional_headers) { |
49 StartRequestHelper(url, | 50 StartRequestHelper(url, |
50 content::URLFetcher::POST, | 51 net::URLFetcher::POST, |
51 delegate, | 52 delegate, |
52 max_retries, | 53 max_retries, |
53 post_data_mime_type, | 54 post_data_mime_type, |
54 post_data, | 55 post_data, |
55 additional_headers); | 56 additional_headers); |
56 } | 57 } |
57 | 58 |
58 void CloudPrintURLFetcher::OnURLFetchComplete( | 59 void CloudPrintURLFetcher::OnURLFetchComplete( |
59 const net::URLFetcher* source) { | 60 const net::URLFetcher* source) { |
60 VLOG(1) << "CP_PROXY: OnURLFetchComplete, url: " << source->GetURL() | 61 VLOG(1) << "CP_PROXY: OnURLFetchComplete, url: " << source->GetURL() |
(...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
126 // may have changed. | 127 // may have changed. |
127 SetupRequestHeaders(); | 128 SetupRequestHeaders(); |
128 request_->SetRequestContext(GetRequestContextGetter()); | 129 request_->SetRequestContext(GetRequestContextGetter()); |
129 request_->Start(); | 130 request_->Start(); |
130 } | 131 } |
131 } | 132 } |
132 } | 133 } |
133 | 134 |
134 void CloudPrintURLFetcher::StartRequestHelper( | 135 void CloudPrintURLFetcher::StartRequestHelper( |
135 const GURL& url, | 136 const GURL& url, |
136 content::URLFetcher::RequestType request_type, | 137 net::URLFetcher::RequestType request_type, |
137 Delegate* delegate, | 138 Delegate* delegate, |
138 int max_retries, | 139 int max_retries, |
139 const std::string& post_data_mime_type, | 140 const std::string& post_data_mime_type, |
140 const std::string& post_data, | 141 const std::string& post_data, |
141 const std::string& additional_headers) { | 142 const std::string& additional_headers) { |
142 DCHECK(delegate); | 143 DCHECK(delegate); |
143 // Persist the additional headers in case we need to retry the request. | 144 // Persist the additional headers in case we need to retry the request. |
144 additional_headers_ = additional_headers; | 145 additional_headers_ = additional_headers; |
145 request_.reset(content::URLFetcher::Create(url, request_type, this)); | 146 request_.reset(content::URLFetcher::Create(url, request_type, this)); |
146 request_->SetRequestContext(GetRequestContextGetter()); | 147 request_->SetRequestContext(GetRequestContextGetter()); |
147 // Since we implement our own retry logic, disable the retry in URLFetcher. | 148 // Since we implement our own retry logic, disable the retry in URLFetcher. |
148 request_->SetAutomaticallyRetryOn5xx(false); | 149 request_->SetAutomaticallyRetryOn5xx(false); |
149 request_->SetMaxRetries(max_retries); | 150 request_->SetMaxRetries(max_retries); |
150 delegate_ = delegate; | 151 delegate_ = delegate; |
151 SetupRequestHeaders(); | 152 SetupRequestHeaders(); |
152 if (request_type == content::URLFetcher::POST) { | 153 if (request_type == net::URLFetcher::POST) { |
153 request_->SetUploadData(post_data_mime_type, post_data); | 154 request_->SetUploadData(post_data_mime_type, post_data); |
154 } | 155 } |
155 | 156 |
156 request_->Start(); | 157 request_->Start(); |
157 } | 158 } |
158 | 159 |
159 void CloudPrintURLFetcher::SetupRequestHeaders() { | 160 void CloudPrintURLFetcher::SetupRequestHeaders() { |
160 std::string headers = delegate_->GetAuthHeader(); | 161 std::string headers = delegate_->GetAuthHeader(); |
161 if (!headers.empty()) | 162 if (!headers.empty()) |
162 headers += "\r\n"; | 163 headers += "\r\n"; |
163 headers += cloud_print::kChromeCloudPrintProxyHeader; | 164 headers += cloud_print::kChromeCloudPrintProxyHeader; |
164 if (!additional_headers_.empty()) { | 165 if (!additional_headers_.empty()) { |
165 headers += "\r\n"; | 166 headers += "\r\n"; |
166 headers += additional_headers_; | 167 headers += additional_headers_; |
167 } | 168 } |
168 request_->SetExtraRequestHeaders(headers); | 169 request_->SetExtraRequestHeaders(headers); |
169 } | 170 } |
170 | 171 |
171 CloudPrintURLFetcher::~CloudPrintURLFetcher() {} | 172 CloudPrintURLFetcher::~CloudPrintURLFetcher() {} |
172 | 173 |
173 net::URLRequestContextGetter* CloudPrintURLFetcher::GetRequestContextGetter() { | 174 net::URLRequestContextGetter* CloudPrintURLFetcher::GetRequestContextGetter() { |
174 ServiceURLRequestContextGetter* getter = | 175 ServiceURLRequestContextGetter* getter = |
175 g_service_process->GetServiceURLRequestContextGetter(); | 176 g_service_process->GetServiceURLRequestContextGetter(); |
176 // Now set up the user agent for cloudprint. | 177 // Now set up the user agent for cloudprint. |
177 std::string user_agent = getter->user_agent(); | 178 std::string user_agent = getter->user_agent(); |
178 base::StringAppendF(&user_agent, " %s", kCloudPrintUserAgent); | 179 base::StringAppendF(&user_agent, " %s", kCloudPrintUserAgent); |
179 getter->set_user_agent(user_agent); | 180 getter->set_user_agent(user_agent); |
180 return getter; | 181 return getter; |
181 } | 182 } |
OLD | NEW |