OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 <iostream> | 5 #include <iostream> |
6 #include <memory> | 6 #include <memory> |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/base64.h" | 9 #include "base/base64.h" |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
11 #include "base/callback.h" | 11 #include "base/callback.h" |
12 #include "base/command_line.h" | 12 #include "base/command_line.h" |
13 #include "base/files/file_path.h" | 13 #include "base/files/file_path.h" |
14 #include "base/json/json_writer.h" | 14 #include "base/json/json_writer.h" |
15 #include "base/location.h" | 15 #include "base/location.h" |
16 #include "base/memory/ref_counted.h" | 16 #include "base/memory/ref_counted.h" |
17 #include "base/numerics/safe_conversions.h" | 17 #include "base/numerics/safe_conversions.h" |
18 #include "base/strings/string_number_conversions.h" | 18 #include "base/strings/string_number_conversions.h" |
19 #include "content/public/common/content_switches.h" | 19 #include "content/public/common/content_switches.h" |
20 #include "headless/app/headless_shell_switches.h" | 20 #include "headless/app/headless_shell_switches.h" |
21 #include "headless/public/domains/emulation.h" | 21 #include "headless/public/domains/emulation.h" |
22 #include "headless/public/domains/page.h" | 22 #include "headless/public/domains/page.h" |
23 #include "headless/public/domains/runtime.h" | 23 #include "headless/public/domains/runtime.h" |
24 #include "headless/public/headless_browser.h" | 24 #include "headless/public/headless_browser.h" |
25 #include "headless/public/headless_devtools_client.h" | 25 #include "headless/public/headless_devtools_client.h" |
26 #include "headless/public/headless_devtools_target.h" | 26 #include "headless/public/headless_devtools_target.h" |
27 #include "headless/public/headless_web_contents.h" | 27 #include "headless/public/headless_web_contents.h" |
| 28 #include "headless/public/util/deterministic_dispatcher.h" |
| 29 #include "headless/public/util/deterministic_http_protocol_handler.h" |
28 #include "net/base/file_stream.h" | 30 #include "net/base/file_stream.h" |
29 #include "net/base/io_buffer.h" | 31 #include "net/base/io_buffer.h" |
30 #include "net/base/ip_address.h" | 32 #include "net/base/ip_address.h" |
31 #include "net/base/net_errors.h" | 33 #include "net/base/net_errors.h" |
32 #include "ui/gfx/geometry/size.h" | 34 #include "ui/gfx/geometry/size.h" |
33 | 35 |
34 using headless::HeadlessBrowser; | 36 using headless::HeadlessBrowser; |
35 using headless::HeadlessBrowserContext; | 37 using headless::HeadlessBrowserContext; |
36 using headless::HeadlessDevToolsClient; | 38 using headless::HeadlessDevToolsClient; |
37 using headless::HeadlessWebContents; | 39 using headless::HeadlessWebContents; |
(...skipping 28 matching lines...) Expand all Loading... |
66 : browser_(nullptr), | 68 : browser_(nullptr), |
67 devtools_client_(HeadlessDevToolsClient::Create()), | 69 devtools_client_(HeadlessDevToolsClient::Create()), |
68 web_contents_(nullptr), | 70 web_contents_(nullptr), |
69 processed_page_ready_(false), | 71 processed_page_ready_(false), |
70 browser_context_(nullptr) {} | 72 browser_context_(nullptr) {} |
71 ~HeadlessShell() override {} | 73 ~HeadlessShell() override {} |
72 | 74 |
73 void OnStart(HeadlessBrowser* browser) { | 75 void OnStart(HeadlessBrowser* browser) { |
74 browser_ = browser; | 76 browser_ = browser; |
75 | 77 |
76 browser_context_ = browser_->CreateBrowserContextBuilder().Build(); | 78 if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
| 79 headless::switches::kDeterministicFetch)) { |
| 80 deterministic_dispatcher_.reset( |
| 81 new headless::DeterministicDispatcher(browser_->BrowserIOThread())); |
| 82 |
| 83 headless::ProtocolHandlerMap protocol_handlers; |
| 84 protocol_handlers[url::kHttpScheme] = |
| 85 base::MakeUnique<headless::DeterministicHttpProtocolHandler>( |
| 86 deterministic_dispatcher_.get()); |
| 87 protocol_handlers[url::kHttpsScheme] = |
| 88 base::MakeUnique<headless::DeterministicHttpProtocolHandler>( |
| 89 deterministic_dispatcher_.get()); |
| 90 |
| 91 headless::DeterministicHttpProtocolHandler* http_handler = |
| 92 static_cast<headless::DeterministicHttpProtocolHandler*>( |
| 93 protocol_handlers[url::kHttpScheme].get()); |
| 94 headless::DeterministicHttpProtocolHandler* https_handler = |
| 95 static_cast<headless::DeterministicHttpProtocolHandler*>( |
| 96 protocol_handlers[url::kHttpsScheme].get()); |
| 97 |
| 98 browser_context_ = browser_->CreateBrowserContextBuilder() |
| 99 .SetProtocolHandlers(std::move(protocol_handlers)) |
| 100 .Build(); |
| 101 |
| 102 http_handler->set_browser_context(browser_context_); |
| 103 https_handler->set_browser_context(browser_context_); |
| 104 } else { |
| 105 browser_context_ = browser_->CreateBrowserContextBuilder().Build(); |
| 106 } |
77 | 107 |
78 HeadlessWebContents::Builder builder( | 108 HeadlessWebContents::Builder builder( |
79 browser_context_->CreateWebContentsBuilder()); | 109 browser_context_->CreateWebContentsBuilder()); |
80 base::CommandLine::StringVector args = | 110 base::CommandLine::StringVector args = |
81 base::CommandLine::ForCurrentProcess()->GetArgs(); | 111 base::CommandLine::ForCurrentProcess()->GetArgs(); |
82 | 112 |
83 // TODO(alexclarke): Should we navigate to about:blank first if using | 113 // TODO(alexclarke): Should we navigate to about:blank first if using |
84 // virtual time? | 114 // virtual time? |
85 if (!args.empty() && !args[0].empty()) | 115 if (!args.empty() && !args[0].empty()) |
86 builder.SetInitialURL(GURL(args[0])); | 116 builder.SetInitialURL(GURL(args[0])); |
(...skipping 241 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
328 } | 358 } |
329 | 359 |
330 private: | 360 private: |
331 GURL url_; | 361 GURL url_; |
332 HeadlessBrowser* browser_; // Not owned. | 362 HeadlessBrowser* browser_; // Not owned. |
333 std::unique_ptr<HeadlessDevToolsClient> devtools_client_; | 363 std::unique_ptr<HeadlessDevToolsClient> devtools_client_; |
334 HeadlessWebContents* web_contents_; | 364 HeadlessWebContents* web_contents_; |
335 bool processed_page_ready_; | 365 bool processed_page_ready_; |
336 std::unique_ptr<net::FileStream> screenshot_file_stream_; | 366 std::unique_ptr<net::FileStream> screenshot_file_stream_; |
337 HeadlessBrowserContext* browser_context_; | 367 HeadlessBrowserContext* browser_context_; |
| 368 std::unique_ptr<headless::DeterministicDispatcher> deterministic_dispatcher_; |
338 | 369 |
339 DISALLOW_COPY_AND_ASSIGN(HeadlessShell); | 370 DISALLOW_COPY_AND_ASSIGN(HeadlessShell); |
340 }; | 371 }; |
341 | 372 |
342 int main(int argc, const char** argv) { | 373 int main(int argc, const char** argv) { |
343 headless::RunChildProcessIfNeeded(argc, argv); | 374 headless::RunChildProcessIfNeeded(argc, argv); |
344 HeadlessShell shell; | 375 HeadlessShell shell; |
345 HeadlessBrowser::Options::Builder builder(argc, argv); | 376 HeadlessBrowser::Options::Builder builder(argc, argv); |
346 | 377 |
347 // Enable devtools if requested. | 378 // Enable devtools if requested. |
(...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
408 LOG(ERROR) << "Malformed window size"; | 439 LOG(ERROR) << "Malformed window size"; |
409 return EXIT_FAILURE; | 440 return EXIT_FAILURE; |
410 } | 441 } |
411 builder.SetWindowSize(parsed_window_size); | 442 builder.SetWindowSize(parsed_window_size); |
412 } | 443 } |
413 | 444 |
414 return HeadlessBrowserMain( | 445 return HeadlessBrowserMain( |
415 builder.Build(), | 446 builder.Build(), |
416 base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell))); | 447 base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell))); |
417 } | 448 } |
OLD | NEW |