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 <memory> | 5 #include <memory> |
6 #include <sstream> | 6 #include <sstream> |
7 #include <string> | 7 #include <string> |
8 #include <utility> | 8 #include <utility> |
9 | 9 |
10 #include "base/base64.h" | 10 #include "base/base64.h" |
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
61 | 61 |
62 HeadlessShell::~HeadlessShell() {} | 62 HeadlessShell::~HeadlessShell() {} |
63 | 63 |
64 void HeadlessShell::OnStart(HeadlessBrowser* browser) { | 64 void HeadlessShell::OnStart(HeadlessBrowser* browser) { |
65 browser_ = browser; | 65 browser_ = browser; |
66 | 66 |
67 HeadlessBrowserContext::Builder context_builder = | 67 HeadlessBrowserContext::Builder context_builder = |
68 browser_->CreateBrowserContextBuilder(); | 68 browser_->CreateBrowserContextBuilder(); |
69 // TODO(eseckler): These switches should also affect BrowserContexts that | 69 // TODO(eseckler): These switches should also affect BrowserContexts that |
70 // are created via DevTools later. | 70 // are created via DevTools later. |
| 71 DeterministicHttpProtocolHandler* http_handler; |
| 72 DeterministicHttpProtocolHandler* https_handler; |
71 if (base::CommandLine::ForCurrentProcess()->HasSwitch( | 73 if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
72 switches::kDeterministicFetch)) { | 74 switches::kDeterministicFetch)) { |
73 deterministic_dispatcher_.reset( | 75 deterministic_dispatcher_.reset( |
74 new DeterministicDispatcher(browser_->BrowserIOThread())); | 76 new DeterministicDispatcher(browser_->BrowserIOThread())); |
75 | 77 |
76 ProtocolHandlerMap protocol_handlers; | 78 ProtocolHandlerMap protocol_handlers; |
77 protocol_handlers[url::kHttpScheme] = | 79 protocol_handlers[url::kHttpScheme] = |
78 base::MakeUnique<DeterministicHttpProtocolHandler>( | 80 base::MakeUnique<DeterministicHttpProtocolHandler>( |
79 deterministic_dispatcher_.get(), browser->BrowserIOThread()); | 81 deterministic_dispatcher_.get(), browser->BrowserIOThread()); |
| 82 http_handler = static_cast<DeterministicHttpProtocolHandler*>( |
| 83 protocol_handlers[url::kHttpScheme].get()); |
80 protocol_handlers[url::kHttpsScheme] = | 84 protocol_handlers[url::kHttpsScheme] = |
81 base::MakeUnique<DeterministicHttpProtocolHandler>( | 85 base::MakeUnique<DeterministicHttpProtocolHandler>( |
82 deterministic_dispatcher_.get(), browser->BrowserIOThread()); | 86 deterministic_dispatcher_.get(), browser->BrowserIOThread()); |
| 87 https_handler = static_cast<DeterministicHttpProtocolHandler*>( |
| 88 protocol_handlers[url::kHttpsScheme].get()); |
83 | 89 |
84 context_builder.SetProtocolHandlers(std::move(protocol_handlers)); | 90 context_builder.SetProtocolHandlers(std::move(protocol_handlers)); |
85 } | 91 } |
86 browser_context_ = context_builder.Build(); | 92 browser_context_ = context_builder.Build(); |
| 93 if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
| 94 switches::kDeterministicFetch)) { |
| 95 http_handler->SetHeadlessBrowserContext(browser_context_); |
| 96 https_handler->SetHeadlessBrowserContext(browser_context_); |
| 97 } |
87 browser_->SetDefaultBrowserContext(browser_context_); | 98 browser_->SetDefaultBrowserContext(browser_context_); |
88 | 99 |
89 HeadlessWebContents::Builder builder( | 100 HeadlessWebContents::Builder builder( |
90 browser_context_->CreateWebContentsBuilder()); | 101 browser_context_->CreateWebContentsBuilder()); |
91 base::CommandLine::StringVector args = | 102 base::CommandLine::StringVector args = |
92 base::CommandLine::ForCurrentProcess()->GetArgs(); | 103 base::CommandLine::ForCurrentProcess()->GetArgs(); |
93 | 104 |
94 // TODO(alexclarke): Should we navigate to about:blank first if using | 105 // TODO(alexclarke): Should we navigate to about:blank first if using |
95 // virtual time? | 106 // virtual time? |
96 if (args.empty()) | 107 if (args.empty()) |
(...skipping 525 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
622 if (net::HttpUtil::IsValidHeaderValue(ua)) | 633 if (net::HttpUtil::IsValidHeaderValue(ua)) |
623 builder.SetUserAgent(ua); | 634 builder.SetUserAgent(ua); |
624 } | 635 } |
625 | 636 |
626 return HeadlessBrowserMain( | 637 return HeadlessBrowserMain( |
627 builder.Build(), | 638 builder.Build(), |
628 base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell))); | 639 base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell))); |
629 } | 640 } |
630 | 641 |
631 } // namespace headless | 642 } // namespace headless |
OLD | NEW |