Index: content/shell/browser/layout_test/devtools_protocol_test_bindings.cc |
diff --git a/content/shell/browser/layout_test/devtools_protocol_test_bindings.cc b/content/shell/browser/layout_test/devtools_protocol_test_bindings.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..cfc153310f83e6629f737cedb68c7638b09c6307 |
--- /dev/null |
+++ b/content/shell/browser/layout_test/devtools_protocol_test_bindings.cc |
@@ -0,0 +1,131 @@ |
+// Copyright 2017 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "content/shell/browser/layout_test/devtools_protocol_test_bindings.h" |
+ |
+#include "base/json/json_reader.h" |
+#include "base/json/string_escape.h" |
+#include "base/strings/string_number_conversions.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "base/values.h" |
+#include "content/public/browser/devtools_agent_host.h" |
+#include "content/public/browser/navigation_handle.h" |
+#include "content/public/browser/render_frame_host.h" |
+#include "content/public/browser/web_contents.h" |
+ |
+#if !defined(OS_ANDROID) |
+#include "content/public/browser/devtools_frontend_host.h" |
+#endif |
+ |
+namespace content { |
+ |
+namespace { |
+// This constant should be in sync with |
+// the constant at devtools_ui_bindings.cc. |
+const size_t kMaxMessageChunkSize = IPC::Channel::kMaximumMessageSize / 4; |
+} // namespace |
+ |
+DevToolsProtocolTestBindings::DevToolsProtocolTestBindings( |
+ WebContents* devtools) |
+ : WebContentsObserver(devtools), |
+ agent_host_(DevToolsAgentHost::CreateForDiscovery()) { |
+ agent_host_->AttachClient(this); |
+} |
+ |
+DevToolsProtocolTestBindings::~DevToolsProtocolTestBindings() { |
+ if (agent_host_) { |
+ agent_host_->DetachClient(this); |
+ agent_host_ = nullptr; |
+ } |
+} |
+ |
+// static |
+GURL DevToolsProtocolTestBindings::GetProtocolTestURL(const GURL& test_url) { |
+ std::string spec = test_url.spec(); |
+ std::string dir = "/inspector-protocol/"; |
+ size_t pos = spec.find(dir); |
+ if (pos == std::string::npos) |
+ return GURL(); |
+ if (spec.rfind(".js") != spec.length() - 3) |
+ return GURL(); |
+ spec = spec.substr(0, pos + dir.length()) + |
+ "resources/inspector-protocol-test.html?" + spec; |
+ return GURL(spec); |
+} |
+ |
+void DevToolsProtocolTestBindings::ReadyToCommitNavigation( |
+ NavigationHandle* navigation_handle) { |
+#if !defined(OS_ANDROID) |
+ content::RenderFrameHost* frame = navigation_handle->GetRenderFrameHost(); |
+ if (frame->GetParent()) |
+ return; |
+ frontend_host_.reset(DevToolsFrontendHost::Create( |
+ frame, base::Bind(&DevToolsProtocolTestBindings::HandleMessageFromTest, |
+ base::Unretained(this)))); |
+#endif |
+} |
+ |
+void DevToolsProtocolTestBindings::WebContentsDestroyed() { |
+ if (agent_host_) { |
+ agent_host_->DetachClient(this); |
+ agent_host_ = nullptr; |
+ } |
+} |
+ |
+void DevToolsProtocolTestBindings::HandleMessageFromTest( |
+ const std::string& message) { |
+ std::string method; |
+ base::ListValue* params = nullptr; |
+ base::DictionaryValue* dict = nullptr; |
+ std::unique_ptr<base::Value> parsed_message = base::JSONReader::Read(message); |
+ if (!parsed_message || !parsed_message->GetAsDictionary(&dict) || |
+ !dict->GetString("method", &method)) { |
+ return; |
+ } |
+ |
+ int request_id = 0; |
+ dict->GetInteger("id", &request_id); |
+ dict->GetList("params", ¶ms); |
+ |
+ if (method == "dispatchProtocolMessage" && params && params->GetSize() == 1) { |
+ std::string protocol_message; |
+ if (!params->GetString(0, &protocol_message)) |
+ return; |
+ if (agent_host_) |
+ agent_host_->DispatchProtocolMessage(this, protocol_message); |
+ return; |
+ } |
+} |
+ |
+void DevToolsProtocolTestBindings::DispatchProtocolMessage( |
+ DevToolsAgentHost* agent_host, |
+ const std::string& message) { |
+ if (message.length() < kMaxMessageChunkSize) { |
+ std::string param; |
+ base::EscapeJSONString(message, true, ¶m); |
+ std::string code = "DevToolsAPI.dispatchMessage(" + param + ");"; |
+ base::string16 javascript = base::UTF8ToUTF16(code); |
+ web_contents()->GetMainFrame()->ExecuteJavaScriptForTests(javascript); |
+ return; |
+ } |
+ |
+ size_t total_size = message.length(); |
+ for (size_t pos = 0; pos < message.length(); pos += kMaxMessageChunkSize) { |
+ std::string param; |
+ base::EscapeJSONString(message.substr(pos, kMaxMessageChunkSize), true, |
+ ¶m); |
+ std::string code = "DevToolsAPI.dispatchMessageChunk(" + param + "," + |
+ std::to_string(pos ? 0 : total_size) + ");"; |
+ base::string16 javascript = base::UTF8ToUTF16(code); |
+ web_contents()->GetMainFrame()->ExecuteJavaScriptForTests(javascript); |
+ } |
+} |
+ |
+void DevToolsProtocolTestBindings::AgentHostClosed( |
+ DevToolsAgentHost* agent_host, |
+ bool replaced) { |
+ agent_host_ = nullptr; |
+} |
+ |
+} // namespace content |