content/browser/devtools/site_per_process_devtools_browsertest.cc - chromium/src - Git at Google

 // Copyright (c) 2014 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 "base/run_loop.h"
 #include "build/build_config.h"
 #include "content/browser/devtools/render_frame_devtools_agent_host.h"
 #include "content/browser/frame_host/frame_tree.h"
 #include "content/browser/site_per_process_browsertest.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/download_manager.h"
 #include "content/public/test/content_browser_test_utils.h"
 #include "content/public/test/test_utils.h"
 #include "content/shell/browser/shell.h"
 #include "content/shell/browser/shell_download_manager_delegate.h"
 #include "content/test/content_browser_test_utils_internal.h"
 #include "net/dns/mock_host_resolver.h"

 namespace content {

 class SitePerProcessDevToolsBrowserTest : public SitePerProcessBrowserTest {
  public:
   SitePerProcessDevToolsBrowserTest() {}
   void SetUpOnMainThread() override {
     host_resolver()->AddRule("*", "127.0.0.1");
     SitePerProcessBrowserTest::SetUpOnMainThread();
   }
 };

 class TestClient: public DevToolsAgentHostClient {
  public:
   TestClient() : closed_(false), waiting_for_reply_(false) {}
   ~TestClient() override {}

   bool closed() { return closed_; }

   void DispatchProtocolMessage(
       DevToolsAgentHost* agent_host,
       const std::string& message) override {
     if (waiting_for_reply_) {
       waiting_for_reply_ = false;
       base::RunLoop::QuitCurrentDeprecated();
     }
   }

   void AgentHostClosed(DevToolsAgentHost* agent_host) override {
     closed_ = true;
   }

   void WaitForReply() {
     waiting_for_reply_ = true;
     base::RunLoop().Run();
   }

  private:
   bool closed_;
   bool waiting_for_reply_;
 };

 // Fails on Android, http://crbug.com/464993.
 #if defined(OS_ANDROID)
 #define MAYBE_CrossSiteIframeAgentHost DISABLED_CrossSiteIframeAgentHost
 #else
 #define MAYBE_CrossSiteIframeAgentHost CrossSiteIframeAgentHost
 #endif
 IN_PROC_BROWSER_TEST_F(SitePerProcessDevToolsBrowserTest,
                        MAYBE_CrossSiteIframeAgentHost) {
   DevToolsAgentHost::List list;
   GURL main_url(embedded_test_server()->GetURL("/site_per_process_main.html"));
   NavigateToURL(shell(), main_url);

   // It is safe to obtain the root frame tree node here, as it doesn't change.
   FrameTreeNode* root =
       static_cast<WebContentsImpl*>(shell()->web_contents())->
           GetFrameTree()->root();

   list = DevToolsAgentHost::GetOrCreateAll();
   EXPECT_EQ(1U, list.size());
   EXPECT_EQ(DevToolsAgentHost::kTypePage, list[0]->GetType());
   EXPECT_EQ(main_url.spec(), list[0]->GetURL().spec());

   // Load same-site page into iframe.
   FrameTreeNode* child = root->child_at(0);
   GURL http_url(embedded_test_server()->GetURL("/title1.html"));
   NavigateFrameToURL(child, http_url);

   list = DevToolsAgentHost::GetOrCreateAll();
   EXPECT_EQ(1U, list.size());
   EXPECT_EQ(DevToolsAgentHost::kTypePage, list[0]->GetType());
   EXPECT_EQ(main_url.spec(), list[0]->GetURL().spec());

   // Load cross-site page into iframe.
   GURL::Replacements replace_host;
   GURL cross_site_url(embedded_test_server()->GetURL("/title2.html"));
   replace_host.SetHostStr("foo.com");
   cross_site_url = cross_site_url.ReplaceComponents(replace_host);
   NavigateFrameToURL(root->child_at(0), cross_site_url);

   list = DevToolsAgentHost::GetOrCreateAll();
   EXPECT_EQ(2U, list.size());
   EXPECT_EQ(DevToolsAgentHost::kTypePage, list[0]->GetType());
   EXPECT_EQ(main_url.spec(), list[0]->GetURL().spec());
   EXPECT_EQ(DevToolsAgentHost::kTypeFrame, list[1]->GetType());
   EXPECT_EQ(cross_site_url.spec(), list[1]->GetURL().spec());
   EXPECT_EQ(std::string(), list[0]->GetParentId());
   EXPECT_EQ(list[0]->GetId(), list[1]->GetParentId());
   EXPECT_NE(list[1]->GetId(), list[0]->GetId());

   // Attaching to both agent hosts.
   scoped_refptr<DevToolsAgentHost> child_host = list[1];
   TestClient child_client;
   child_host->AttachClient(&child_client);
   scoped_refptr<DevToolsAgentHost> parent_host = list[0];
   TestClient parent_client;
   parent_host->AttachClient(&parent_client);

   // Send message to parent and child frames and get result back.
   char message[] = "{\"id\": 0, \"method\": \"incorrect.method\"}";
   child_host->DispatchProtocolMessage(&child_client, message);
   child_client.WaitForReply();
   parent_host->DispatchProtocolMessage(&parent_client, message);
   parent_client.WaitForReply();

   // Load back same-site page into iframe.
   NavigateFrameToURL(root->child_at(0), http_url);

   list = DevToolsAgentHost::GetOrCreateAll();
   EXPECT_EQ(1U, list.size());
   EXPECT_EQ(DevToolsAgentHost::kTypePage, list[0]->GetType());
   EXPECT_EQ(main_url.spec(), list[0]->GetURL().spec());
   EXPECT_TRUE(child_client.closed());
   child_host->DetachClient(&child_client);
   child_host = nullptr;
   EXPECT_FALSE(parent_client.closed());
   parent_host->DetachClient(&parent_client);
   parent_host = nullptr;
 }

 IN_PROC_BROWSER_TEST_F(SitePerProcessDevToolsBrowserTest, AgentHostForFrames) {
   GURL main_url(embedded_test_server()->GetURL("/site_per_process_main.html"));
   NavigateToURL(shell(), main_url);

   scoped_refptr<DevToolsAgentHost> page_agent =
       DevToolsAgentHost::GetOrCreateFor(shell()->web_contents());

   // It is safe to obtain the root frame tree node here, as it doesn't change.
   FrameTreeNode* root =
       static_cast<WebContentsImpl*>(shell()->web_contents())->
           GetFrameTree()->root();

   scoped_refptr<DevToolsAgentHost> main_frame_agent =
       RenderFrameDevToolsAgentHost::GetOrCreateFor(root);
   EXPECT_EQ(page_agent.get(), main_frame_agent.get());

   // Load same-site page into iframe.
   FrameTreeNode* child = root->child_at(0);
   GURL http_url(embedded_test_server()->GetURL("/title1.html"));
   NavigateFrameToURL(child, http_url);

   scoped_refptr<DevToolsAgentHost> child_frame_agent =
       RenderFrameDevToolsAgentHost::GetOrCreateFor(child);
   EXPECT_EQ(page_agent.get(), child_frame_agent.get());

   // Load cross-site page into iframe.
   GURL::Replacements replace_host;
   GURL cross_site_url(embedded_test_server()->GetURL("/title2.html"));
   replace_host.SetHostStr("foo.com");
   cross_site_url = cross_site_url.ReplaceComponents(replace_host);
   NavigateFrameToURL(root->child_at(0), cross_site_url);

   child_frame_agent = RenderFrameDevToolsAgentHost::GetOrCreateFor(child);
   EXPECT_NE(page_agent.get(), child_frame_agent.get());
   EXPECT_EQ(child_frame_agent->GetParentId(), page_agent->GetId());
   EXPECT_NE(child_frame_agent->GetId(), page_agent->GetId());
 }

 IN_PROC_BROWSER_TEST_F(SitePerProcessDevToolsBrowserTest,
     AgentHostForPageEqualsOneForMainFrame) {
   GURL main_url(embedded_test_server()->GetURL("/site_per_process_main.html"));
   NavigateToURL(shell(), main_url);

   // It is safe to obtain the root frame tree node here, as it doesn't change.
   FrameTreeNode* root =
       static_cast<WebContentsImpl*>(shell()->web_contents())->
           GetFrameTree()->root();
   FrameTreeNode* child = root->child_at(0);

   // Load cross-site page into iframe.
   GURL::Replacements replace_host;
   GURL cross_site_url(embedded_test_server()->GetURL("/title2.html"));
   replace_host.SetHostStr("foo.com");
   cross_site_url = cross_site_url.ReplaceComponents(replace_host);
   NavigateFrameToURL(child, cross_site_url);

   // First ask for child frame, then for main frame.
   scoped_refptr<DevToolsAgentHost> child_frame_agent =
       RenderFrameDevToolsAgentHost::GetOrCreateFor(child);
   scoped_refptr<DevToolsAgentHost> main_frame_agent =
       RenderFrameDevToolsAgentHost::GetOrCreateFor(root);
   EXPECT_NE(main_frame_agent.get(), child_frame_agent.get());
   EXPECT_EQ(child_frame_agent->GetParentId(), main_frame_agent->GetId());
   EXPECT_NE(child_frame_agent->GetId(), main_frame_agent->GetId());

   // Agent for web contents should be the the main frame's one.
   scoped_refptr<DevToolsAgentHost> page_agent =
       DevToolsAgentHost::GetOrCreateFor(shell()->web_contents());
   EXPECT_EQ(page_agent.get(), main_frame_agent.get());
 }

 class SitePerProcessDownloadDevToolsBrowserTest
     : public SitePerProcessBrowserTest {
  public:
   SitePerProcessDownloadDevToolsBrowserTest() {}

   void SetUpOnMainThread() override {
     SitePerProcessBrowserTest::SetUpOnMainThread();
     ASSERT_TRUE(downloads_directory_.CreateUniqueTempDir());
     DownloadManager* download_manager = BrowserContext::GetDownloadManager(
         shell()->web_contents()->GetBrowserContext());
     ShellDownloadManagerDelegate* download_delegate =
         static_cast<ShellDownloadManagerDelegate*>(
             download_manager->GetDelegate());
     download_delegate->SetDownloadBehaviorForTesting(
         downloads_directory_.GetPath());
   }

   base::ScopedTempDir downloads_directory_;
 };

 IN_PROC_BROWSER_TEST_F(SitePerProcessDownloadDevToolsBrowserTest,
                        NotCommittedNavigationDoesNotBlockAgent) {
   ASSERT_TRUE(
       NavigateToURL(shell(), embedded_test_server()->GetURL("/title1.html")));
   scoped_refptr<DevToolsAgentHost> agent =
       DevToolsAgentHost::GetOrCreateFor(shell()->web_contents());
   TestClient client;
   agent->AttachClient(&client);
   char message[] = "{\"id\": 0, \"method\": \"incorrect.method\"}";
   // Check that client is responsive.
   agent->DispatchProtocolMessage(&client, message);
   client.WaitForReply();

   // Do cross process navigation that ends up being download, which will be
   // not committed navigation in that web contents/render frame.
   GURL::Replacements replace_host;
   GURL cross_site_url(embedded_test_server()->GetURL("/download/empty.bin"));
   replace_host.SetHostStr("foo.com");
   cross_site_url = cross_site_url.ReplaceComponents(replace_host);
   ASSERT_TRUE(NavigateToURLAndExpectNoCommit(shell(), cross_site_url));

   // Check that client is still responding after not committed navigation
   // is finished.
   agent->DispatchProtocolMessage(&client, message);
   client.WaitForReply();
   ASSERT_TRUE(agent->DetachClient(&client));
 }

 }  // namespace content
	// Copyright (c) 2014 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 "base/run_loop.h"
	#include "build/build_config.h"
	#include "content/browser/devtools/render_frame_devtools_agent_host.h"
	#include "content/browser/frame_host/frame_tree.h"
	#include "content/browser/site_per_process_browsertest.h"
	#include "content/browser/web_contents/web_contents_impl.h"
	#include "content/public/browser/browser_context.h"
	#include "content/public/browser/devtools_agent_host.h"
	#include "content/public/browser/download_manager.h"
	#include "content/public/test/content_browser_test_utils.h"
	#include "content/public/test/test_utils.h"
	#include "content/shell/browser/shell.h"
	#include "content/shell/browser/shell_download_manager_delegate.h"
	#include "content/test/content_browser_test_utils_internal.h"
	#include "net/dns/mock_host_resolver.h"

	namespace content {

	class SitePerProcessDevToolsBrowserTest : public SitePerProcessBrowserTest {
	public:
	SitePerProcessDevToolsBrowserTest() {}
	void SetUpOnMainThread() override {
	host_resolver()->AddRule("*", "127.0.0.1");
	SitePerProcessBrowserTest::SetUpOnMainThread();
	}
	};

	class TestClient: public DevToolsAgentHostClient {
	public:
	TestClient() : closed_(false), waiting_for_reply_(false) {}
	~TestClient() override {}

	bool closed() { return closed_; }

	void DispatchProtocolMessage(
	DevToolsAgentHost* agent_host,
	const std::string& message) override {
	if (waiting_for_reply_) {
	waiting_for_reply_ = false;
	base::RunLoop::QuitCurrentDeprecated();
	}
	}

	void AgentHostClosed(DevToolsAgentHost* agent_host) override {
	closed_ = true;
	}

	void WaitForReply() {
	waiting_for_reply_ = true;
	base::RunLoop().Run();
	}

	private:
	bool closed_;
	bool waiting_for_reply_;
	};

	// Fails on Android, http://crbug.com/464993.
	#if defined(OS_ANDROID)
	#define MAYBE_CrossSiteIframeAgentHost DISABLED_CrossSiteIframeAgentHost
	#else
	#define MAYBE_CrossSiteIframeAgentHost CrossSiteIframeAgentHost
	#endif
	IN_PROC_BROWSER_TEST_F(SitePerProcessDevToolsBrowserTest,
	MAYBE_CrossSiteIframeAgentHost) {
	DevToolsAgentHost::List list;
	GURL main_url(embedded_test_server()->GetURL("/site_per_process_main.html"));
	NavigateToURL(shell(), main_url);

	// It is safe to obtain the root frame tree node here, as it doesn't change.
	FrameTreeNode* root =
	static_cast<WebContentsImpl*>(shell()->web_contents())->
	GetFrameTree()->root();

	list = DevToolsAgentHost::GetOrCreateAll();
	EXPECT_EQ(1U, list.size());
	EXPECT_EQ(DevToolsAgentHost::kTypePage, list[0]->GetType());
	EXPECT_EQ(main_url.spec(), list[0]->GetURL().spec());

	// Load same-site page into iframe.
	FrameTreeNode* child = root->child_at(0);
	GURL http_url(embedded_test_server()->GetURL("/title1.html"));
	NavigateFrameToURL(child, http_url);

	list = DevToolsAgentHost::GetOrCreateAll();
	EXPECT_EQ(1U, list.size());
	EXPECT_EQ(DevToolsAgentHost::kTypePage, list[0]->GetType());
	EXPECT_EQ(main_url.spec(), list[0]->GetURL().spec());

	// Load cross-site page into iframe.
	GURL::Replacements replace_host;
	GURL cross_site_url(embedded_test_server()->GetURL("/title2.html"));
	replace_host.SetHostStr("foo.com");
	cross_site_url = cross_site_url.ReplaceComponents(replace_host);
	NavigateFrameToURL(root->child_at(0), cross_site_url);

	list = DevToolsAgentHost::GetOrCreateAll();
	EXPECT_EQ(2U, list.size());
	EXPECT_EQ(DevToolsAgentHost::kTypePage, list[0]->GetType());
	EXPECT_EQ(main_url.spec(), list[0]->GetURL().spec());
	EXPECT_EQ(DevToolsAgentHost::kTypeFrame, list[1]->GetType());
	EXPECT_EQ(cross_site_url.spec(), list[1]->GetURL().spec());
	EXPECT_EQ(std::string(), list[0]->GetParentId());
	EXPECT_EQ(list[0]->GetId(), list[1]->GetParentId());
	EXPECT_NE(list[1]->GetId(), list[0]->GetId());

	// Attaching to both agent hosts.
	scoped_refptr<DevToolsAgentHost> child_host = list[1];
	TestClient child_client;
	child_host->AttachClient(&child_client);
	scoped_refptr<DevToolsAgentHost> parent_host = list[0];
	TestClient parent_client;
	parent_host->AttachClient(&parent_client);

	// Send message to parent and child frames and get result back.
	char message[] = "{\"id\": 0, \"method\": \"incorrect.method\"}";
	child_host->DispatchProtocolMessage(&child_client, message);
	child_client.WaitForReply();
	parent_host->DispatchProtocolMessage(&parent_client, message);
	parent_client.WaitForReply();

	// Load back same-site page into iframe.
	NavigateFrameToURL(root->child_at(0), http_url);

	list = DevToolsAgentHost::GetOrCreateAll();
	EXPECT_EQ(1U, list.size());
	EXPECT_EQ(DevToolsAgentHost::kTypePage, list[0]->GetType());
	EXPECT_EQ(main_url.spec(), list[0]->GetURL().spec());
	EXPECT_TRUE(child_client.closed());
	child_host->DetachClient(&child_client);
	child_host = nullptr;
	EXPECT_FALSE(parent_client.closed());
	parent_host->DetachClient(&parent_client);
	parent_host = nullptr;
	}

	IN_PROC_BROWSER_TEST_F(SitePerProcessDevToolsBrowserTest, AgentHostForFrames) {
	GURL main_url(embedded_test_server()->GetURL("/site_per_process_main.html"));
	NavigateToURL(shell(), main_url);

	scoped_refptr<DevToolsAgentHost> page_agent =
	DevToolsAgentHost::GetOrCreateFor(shell()->web_contents());

	// It is safe to obtain the root frame tree node here, as it doesn't change.
	FrameTreeNode* root =
	static_cast<WebContentsImpl*>(shell()->web_contents())->
	GetFrameTree()->root();

	scoped_refptr<DevToolsAgentHost> main_frame_agent =
	RenderFrameDevToolsAgentHost::GetOrCreateFor(root);
	EXPECT_EQ(page_agent.get(), main_frame_agent.get());

	// Load same-site page into iframe.
	FrameTreeNode* child = root->child_at(0);
	GURL http_url(embedded_test_server()->GetURL("/title1.html"));
	NavigateFrameToURL(child, http_url);

	scoped_refptr<DevToolsAgentHost> child_frame_agent =
	RenderFrameDevToolsAgentHost::GetOrCreateFor(child);
	EXPECT_EQ(page_agent.get(), child_frame_agent.get());

	// Load cross-site page into iframe.
	GURL::Replacements replace_host;
	GURL cross_site_url(embedded_test_server()->GetURL("/title2.html"));
	replace_host.SetHostStr("foo.com");
	cross_site_url = cross_site_url.ReplaceComponents(replace_host);
	NavigateFrameToURL(root->child_at(0), cross_site_url);

	child_frame_agent = RenderFrameDevToolsAgentHost::GetOrCreateFor(child);
	EXPECT_NE(page_agent.get(), child_frame_agent.get());
	EXPECT_EQ(child_frame_agent->GetParentId(), page_agent->GetId());
	EXPECT_NE(child_frame_agent->GetId(), page_agent->GetId());
	}

	IN_PROC_BROWSER_TEST_F(SitePerProcessDevToolsBrowserTest,
	AgentHostForPageEqualsOneForMainFrame) {
	GURL main_url(embedded_test_server()->GetURL("/site_per_process_main.html"));
	NavigateToURL(shell(), main_url);

	// It is safe to obtain the root frame tree node here, as it doesn't change.
	FrameTreeNode* root =
	static_cast<WebContentsImpl*>(shell()->web_contents())->
	GetFrameTree()->root();
	FrameTreeNode* child = root->child_at(0);

	// Load cross-site page into iframe.
	GURL::Replacements replace_host;
	GURL cross_site_url(embedded_test_server()->GetURL("/title2.html"));
	replace_host.SetHostStr("foo.com");
	cross_site_url = cross_site_url.ReplaceComponents(replace_host);
	NavigateFrameToURL(child, cross_site_url);

	// First ask for child frame, then for main frame.
	scoped_refptr<DevToolsAgentHost> child_frame_agent =
	RenderFrameDevToolsAgentHost::GetOrCreateFor(child);
	scoped_refptr<DevToolsAgentHost> main_frame_agent =
	RenderFrameDevToolsAgentHost::GetOrCreateFor(root);
	EXPECT_NE(main_frame_agent.get(), child_frame_agent.get());
	EXPECT_EQ(child_frame_agent->GetParentId(), main_frame_agent->GetId());
	EXPECT_NE(child_frame_agent->GetId(), main_frame_agent->GetId());

	// Agent for web contents should be the the main frame's one.
	scoped_refptr<DevToolsAgentHost> page_agent =
	DevToolsAgentHost::GetOrCreateFor(shell()->web_contents());
	EXPECT_EQ(page_agent.get(), main_frame_agent.get());
	}

	class SitePerProcessDownloadDevToolsBrowserTest
	: public SitePerProcessBrowserTest {
	public:
	SitePerProcessDownloadDevToolsBrowserTest() {}

	void SetUpOnMainThread() override {
	SitePerProcessBrowserTest::SetUpOnMainThread();
	ASSERT_TRUE(downloads_directory_.CreateUniqueTempDir());
	DownloadManager* download_manager = BrowserContext::GetDownloadManager(
	shell()->web_contents()->GetBrowserContext());
	ShellDownloadManagerDelegate* download_delegate =
	static_cast<ShellDownloadManagerDelegate*>(
	download_manager->GetDelegate());
	download_delegate->SetDownloadBehaviorForTesting(
	downloads_directory_.GetPath());
	}

	base::ScopedTempDir downloads_directory_;
	};

	IN_PROC_BROWSER_TEST_F(SitePerProcessDownloadDevToolsBrowserTest,
	NotCommittedNavigationDoesNotBlockAgent) {
	ASSERT_TRUE(
	NavigateToURL(shell(), embedded_test_server()->GetURL("/title1.html")));
	scoped_refptr<DevToolsAgentHost> agent =
	DevToolsAgentHost::GetOrCreateFor(shell()->web_contents());
	TestClient client;
	agent->AttachClient(&client);
	char message[] = "{\"id\": 0, \"method\": \"incorrect.method\"}";
	// Check that client is responsive.
	agent->DispatchProtocolMessage(&client, message);
	client.WaitForReply();

	// Do cross process navigation that ends up being download, which will be
	// not committed navigation in that web contents/render frame.
	GURL::Replacements replace_host;
	GURL cross_site_url(embedded_test_server()->GetURL("/download/empty.bin"));
	replace_host.SetHostStr("foo.com");
	cross_site_url = cross_site_url.ReplaceComponents(replace_host);
	ASSERT_TRUE(NavigateToURLAndExpectNoCommit(shell(), cross_site_url));

	// Check that client is still responding after not committed navigation
	// is finished.
	agent->DispatchProtocolMessage(&client, message);
	client.WaitForReply();
	ASSERT_TRUE(agent->DetachClient(&client));
	}

	} // namespace content