OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "mojo/fetcher/network_fetcher.h" | 5 #include "mojo/shell/fetcher/network_fetcher.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 | 8 |
9 #include <utility> | 9 #include <utility> |
10 | 10 |
11 #include "base/bind.h" | 11 #include "base/bind.h" |
12 #include "base/command_line.h" | 12 #include "base/command_line.h" |
13 #include "base/files/file.h" | 13 #include "base/files/file.h" |
14 #include "base/files/file_path.h" | 14 #include "base/files/file_path.h" |
15 #include "base/files/file_util.h" | 15 #include "base/files/file_util.h" |
16 #include "base/message_loop/message_loop.h" | 16 #include "base/message_loop/message_loop.h" |
17 #include "base/process/process.h" | 17 #include "base/process/process.h" |
18 #include "base/stl_util.h" | 18 #include "base/stl_util.h" |
19 #include "base/strings/string_number_conversions.h" | 19 #include "base/strings/string_number_conversions.h" |
20 #include "base/strings/string_util.h" | 20 #include "base/strings/string_util.h" |
21 #include "base/strings/stringprintf.h" | 21 #include "base/strings/stringprintf.h" |
22 #include "base/strings/utf_string_conversions.h" | 22 #include "base/strings/utf_string_conversions.h" |
23 #include "base/trace_event/trace_event.h" | 23 #include "base/trace_event/trace_event.h" |
24 #include "crypto/secure_hash.h" | 24 #include "crypto/secure_hash.h" |
25 #include "crypto/sha2.h" | 25 #include "crypto/sha2.h" |
26 #include "mojo/common/common_type_converters.h" | 26 #include "mojo/common/common_type_converters.h" |
27 #include "mojo/common/data_pipe_utils.h" | 27 #include "mojo/common/data_pipe_utils.h" |
28 #include "mojo/common/url_type_converters.h" | 28 #include "mojo/common/url_type_converters.h" |
29 #include "mojo/services/network/public/interfaces/url_loader_factory.mojom.h" | 29 #include "mojo/services/network/public/interfaces/url_loader_factory.mojom.h" |
30 #include "mojo/shell/data_pipe_peek.h" | 30 #include "mojo/shell/data_pipe_peek.h" |
31 #include "mojo/shell/switches.h" | 31 #include "mojo/shell/switches.h" |
32 | 32 |
33 namespace mojo { | 33 namespace mojo { |
34 namespace fetcher { | 34 namespace shell { |
35 | 35 |
36 NetworkFetcher::NetworkFetcher(bool disable_cache, | 36 NetworkFetcher::NetworkFetcher(bool disable_cache, |
37 mojo::URLRequestPtr request, | 37 mojo::URLRequestPtr request, |
38 URLLoaderFactory* url_loader_factory, | 38 URLLoaderFactory* url_loader_factory, |
39 const FetchCallback& loader_callback) | 39 const FetchCallback& loader_callback) |
40 : Fetcher(loader_callback), | 40 : Fetcher(loader_callback), |
41 disable_cache_(false), | 41 disable_cache_(false), |
42 url_(request->url.To<GURL>()), | 42 url_(request->url.To<GURL>()), |
43 weak_ptr_factory_(this) { | 43 weak_ptr_factory_(this) { |
44 StartNetworkRequest(std::move(request), url_loader_factory); | 44 StartNetworkRequest(std::move(request), url_loader_factory); |
(...skipping 160 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
205 base::Bind(&NetworkFetcher::CopyCompleted, | 205 base::Bind(&NetworkFetcher::CopyCompleted, |
206 weak_ptr_factory_.GetWeakPtr(), callback)); | 206 weak_ptr_factory_.GetWeakPtr(), callback)); |
207 } | 207 } |
208 | 208 |
209 std::string NetworkFetcher::MimeType() { | 209 std::string NetworkFetcher::MimeType() { |
210 return response_->mime_type; | 210 return response_->mime_type; |
211 } | 211 } |
212 | 212 |
213 bool NetworkFetcher::HasMojoMagic() { | 213 bool NetworkFetcher::HasMojoMagic() { |
214 std::string magic; | 214 std::string magic; |
215 return shell::BlockingPeekNBytes(response_->body.get(), &magic, | 215 return BlockingPeekNBytes(response_->body.get(), &magic, strlen(kMojoMagic), |
216 strlen(kMojoMagic), kPeekTimeout) && | 216 kPeekTimeout) && magic == kMojoMagic; |
217 magic == kMojoMagic; | |
218 } | 217 } |
219 | 218 |
220 bool NetworkFetcher::PeekFirstLine(std::string* line) { | 219 bool NetworkFetcher::PeekFirstLine(std::string* line) { |
221 return shell::BlockingPeekLine(response_->body.get(), line, kMaxShebangLength, | 220 return BlockingPeekLine(response_->body.get(), line, kMaxShebangLength, |
222 kPeekTimeout); | 221 kPeekTimeout); |
223 } | 222 } |
224 | 223 |
225 void NetworkFetcher::StartNetworkRequest(mojo::URLRequestPtr request, | 224 void NetworkFetcher::StartNetworkRequest(mojo::URLRequestPtr request, |
226 URLLoaderFactory* url_loader_factory) { | 225 URLLoaderFactory* url_loader_factory) { |
227 TRACE_EVENT_ASYNC_BEGIN1("mojo_shell", "NetworkFetcher::NetworkRequest", this, | 226 TRACE_EVENT_ASYNC_BEGIN1("mojo_shell", "NetworkFetcher::NetworkRequest", this, |
228 "url", request->url.To<std::string>()); | 227 "url", request->url.To<std::string>()); |
229 request->auto_follow_redirects = false; | 228 request->auto_follow_redirects = false; |
230 request->bypass_cache = disable_cache_; | 229 request->bypass_cache = disable_cache_; |
231 | 230 |
232 url_loader_factory->CreateURLLoader(GetProxy(&url_loader_)); | 231 url_loader_factory->CreateURLLoader(GetProxy(&url_loader_)); |
(...skipping 10 matching lines...) Expand all Loading... |
243 << response->error->description << ") while fetching " | 242 << response->error->description << ") while fetching " |
244 << response->url; | 243 << response->url; |
245 loader_callback_.Run(nullptr); | 244 loader_callback_.Run(nullptr); |
246 return; | 245 return; |
247 } | 246 } |
248 | 247 |
249 response_ = std::move(response); | 248 response_ = std::move(response); |
250 loader_callback_.Run(std::move(owner)); | 249 loader_callback_.Run(std::move(owner)); |
251 } | 250 } |
252 | 251 |
253 } // namespace fetcher | 252 } // namespace shell |
254 } // namespace mojo | 253 } // namespace mojo |
OLD | NEW |