// Copyright (c) 2011 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/stl_util-inl.h" #include "base/string16.h" #include "chrome/common/chrome_constants.h" #include "chrome/common/render_messages.h" #include "chrome/common/url_constants.h" #include "chrome/test/testing_profile.h" #include "content/browser/browser_thread.h" #include "content/browser/browsing_instance.h" #include "content/browser/child_process_security_policy.h" #include "content/browser/content_browser_client.h" #include "content/browser/renderer_host/browser_render_process_host.h" #include "content/browser/renderer_host/render_view_host.h" #include "content/browser/renderer_host/test_render_view_host.h" #include "content/browser/site_instance.h" #include "content/browser/tab_contents/navigation_entry.h" #include "content/browser/tab_contents/tab_contents.h" #include "content/browser/webui/empty_web_ui_factory.h" #include "content/common/content_client.h" #include "testing/gtest/include/gtest/gtest.h" namespace { // TODO(estade): this shouldn't need to be chrome:, but it does (or else GURL // doesn't think that the webui URLs have a host). Figure out where this is // coming from and fix it. const char kWebUIScheme[] = "chrome"; class SiteInstanceTestWebUIFactory : public content::EmptyWebUIFactory { public: virtual bool UseWebUIForURL(Profile* profile, const GURL& url) const { return HasWebUIScheme(url); } virtual bool HasWebUIScheme(const GURL& url) const { return url.SchemeIs(kWebUIScheme); } }; class SiteInstanceTestBrowserClient : public content::ContentBrowserClient { public: virtual content::WebUIFactory* GetWebUIFactory() { return &factory_; } private: SiteInstanceTestWebUIFactory factory_; }; class SiteInstanceTest : public testing::Test { public: SiteInstanceTest() : ui_thread_(BrowserThread::UI, &message_loop_), old_browser_client_(NULL) { } virtual void SetUp() { old_browser_client_ = content::GetContentClient()->browser(); content::GetContentClient()->set_browser(&browser_client_); } virtual void TearDown() { content::GetContentClient()->set_browser(old_browser_client_); } private: MessageLoopForUI message_loop_; BrowserThread ui_thread_; SiteInstanceTestBrowserClient browser_client_; content::ContentBrowserClient* old_browser_client_; }; class TestBrowsingInstance : public BrowsingInstance { public: TestBrowsingInstance(Profile* profile, int* deleteCounter) : BrowsingInstance(profile), use_process_per_site(false), deleteCounter_(deleteCounter) { } // Overrides BrowsingInstance::ShouldUseProcessPerSite so that we can test // both alternatives without using command-line switches. bool ShouldUseProcessPerSite(const GURL& url) { return use_process_per_site; } // Set by individual tests. bool use_process_per_site; private: ~TestBrowsingInstance() { (*deleteCounter_)++; } int* deleteCounter_; }; class TestSiteInstance : public SiteInstance { public: static TestSiteInstance* CreateTestSiteInstance(Profile* profile, int* siteDeleteCounter, int* browsingDeleteCounter) { TestBrowsingInstance* browsing_instance = new TestBrowsingInstance(profile, browsingDeleteCounter); return new TestSiteInstance(browsing_instance, siteDeleteCounter); } private: TestSiteInstance(BrowsingInstance* browsing_instance, int* deleteCounter) : SiteInstance(browsing_instance), deleteCounter_(deleteCounter) {} ~TestSiteInstance() { (*deleteCounter_)++; } int* deleteCounter_; }; } // namespace // Test to ensure no memory leaks for SiteInstance objects. TEST_F(SiteInstanceTest, SiteInstanceDestructor) { // The existence of these factories will cause TabContents to create our test // one instead of the real one. MockRenderProcessHostFactory rph_factory; TestRenderViewHostFactory rvh_factory(&rph_factory); int siteDeleteCounter = 0; int browsingDeleteCounter = 0; const GURL url("test:foo"); // Ensure that instances are deleted when their NavigationEntries are gone. TestSiteInstance* instance = TestSiteInstance::CreateTestSiteInstance(NULL, &siteDeleteCounter, &browsingDeleteCounter); EXPECT_EQ(0, siteDeleteCounter); NavigationEntry* e1 = new NavigationEntry(instance, 0, url, GURL(), string16(), PageTransition::LINK); // Redundantly setting e1's SiteInstance shouldn't affect the ref count. e1->set_site_instance(instance); EXPECT_EQ(0, siteDeleteCounter); // Add a second reference NavigationEntry* e2 = new NavigationEntry(instance, 0, url, GURL(), string16(), PageTransition::LINK); // Now delete both entries and be sure the SiteInstance goes away. delete e1; EXPECT_EQ(0, siteDeleteCounter); EXPECT_EQ(0, browsingDeleteCounter); delete e2; EXPECT_EQ(1, siteDeleteCounter); // instance is now deleted EXPECT_EQ(1, browsingDeleteCounter); // browsing_instance is now deleted // Ensure that instances are deleted when their RenderViewHosts are gone. scoped_ptr<TestingProfile> profile(new TestingProfile()); instance = TestSiteInstance::CreateTestSiteInstance(profile.get(), &siteDeleteCounter, &browsingDeleteCounter); { TabContents contents(profile.get(), instance, MSG_ROUTING_NONE, NULL, NULL); EXPECT_EQ(1, siteDeleteCounter); EXPECT_EQ(1, browsingDeleteCounter); } // Make sure that we flush any messages related to the above TabContents // destruction. MessageLoop::current()->RunAllPending(); EXPECT_EQ(2, siteDeleteCounter); EXPECT_EQ(2, browsingDeleteCounter); // contents is now deleted, along with instance and browsing_instance } // Test that NavigationEntries with SiteInstances can be cloned, but that their // SiteInstances can be changed afterwards. Also tests that the ref counts are // updated properly after the change. TEST_F(SiteInstanceTest, CloneNavigationEntry) { int siteDeleteCounter1 = 0; int siteDeleteCounter2 = 0; int browsingDeleteCounter = 0; const GURL url("test:foo"); SiteInstance* instance1 = TestSiteInstance::CreateTestSiteInstance(NULL, &siteDeleteCounter1, &browsingDeleteCounter); SiteInstance* instance2 = TestSiteInstance::CreateTestSiteInstance(NULL, &siteDeleteCounter2, &browsingDeleteCounter); NavigationEntry* e1 = new NavigationEntry(instance1, 0, url, GURL(), string16(), PageTransition::LINK); // Clone the entry NavigationEntry* e2 = new NavigationEntry(*e1); // Should be able to change the SiteInstance of the cloned entry. e2->set_site_instance(instance2); // The first SiteInstance should go away after deleting e1, since e2 should // no longer be referencing it. delete e1; EXPECT_EQ(1, siteDeleteCounter1); EXPECT_EQ(0, siteDeleteCounter2); // The second SiteInstance should go away after deleting e2. delete e2; EXPECT_EQ(1, siteDeleteCounter1); EXPECT_EQ(1, siteDeleteCounter2); // Both BrowsingInstances are also now deleted EXPECT_EQ(2, browsingDeleteCounter); } // Test to ensure UpdateMaxPageID is working properly. TEST_F(SiteInstanceTest, UpdateMaxPageID) { scoped_refptr<SiteInstance> instance(SiteInstance::CreateSiteInstance(NULL)); EXPECT_EQ(-1, instance->max_page_id()); // Make sure max_page_id_ is monotonically increasing. instance->UpdateMaxPageID(3); instance->UpdateMaxPageID(1); EXPECT_EQ(3, instance->max_page_id()); } // Test to ensure GetProcess returns and creates processes correctly. TEST_F(SiteInstanceTest, GetProcess) { // Ensure that GetProcess returns a process. scoped_ptr<TestingProfile> profile(new TestingProfile()); scoped_ptr<RenderProcessHost> host1; scoped_refptr<SiteInstance> instance( SiteInstance::CreateSiteInstance(profile.get())); host1.reset(instance->GetProcess()); EXPECT_TRUE(host1.get() != NULL); // Ensure that GetProcess creates a new process. scoped_refptr<SiteInstance> instance2( SiteInstance::CreateSiteInstance(profile.get())); scoped_ptr<RenderProcessHost> host2(instance2->GetProcess()); EXPECT_TRUE(host2.get() != NULL); EXPECT_NE(host1.get(), host2.get()); } // Test to ensure SetSite and site() work properly. TEST_F(SiteInstanceTest, SetSite) { scoped_refptr<SiteInstance> instance(SiteInstance::CreateSiteInstance(NULL)); EXPECT_FALSE(instance->has_site()); EXPECT_TRUE(instance->site().is_empty()); instance->SetSite(GURL("http://www.google.com/index.html")); EXPECT_EQ(GURL("http://google.com"), instance->site()); EXPECT_TRUE(instance->has_site()); } // Test to ensure GetSiteForURL properly returns sites for URLs. TEST_F(SiteInstanceTest, GetSiteForURL) { // Pages are irrelevant. GURL test_url = GURL("http://www.google.com/index.html"); EXPECT_EQ(GURL("http://google.com"), SiteInstance::GetSiteForURL(NULL, test_url)); // Ports are irrlevant. test_url = GURL("https://www.google.com:8080"); EXPECT_EQ(GURL("https://google.com"), SiteInstance::GetSiteForURL(NULL, test_url)); // Javascript URLs have no site. test_url = GURL("javascript:foo();"); EXPECT_EQ(GURL(), SiteInstance::GetSiteForURL(NULL, test_url)); test_url = GURL("http://foo/a.html"); EXPECT_EQ(GURL("http://foo"), SiteInstance::GetSiteForURL(NULL, test_url)); test_url = GURL("file:///C:/Downloads/"); EXPECT_EQ(GURL(), SiteInstance::GetSiteForURL(NULL, test_url)); // TODO(creis): Do we want to special case file URLs to ensure they have // either no site or a special "file://" site? We currently return // "file://home/" as the site, which seems broken. // test_url = GURL("file://home/"); // EXPECT_EQ(GURL(), SiteInstance::GetSiteForURL(NULL, test_url)); } // Test of distinguishing URLs from different sites. Most of this logic is // tested in RegistryControlledDomainTest. This test focuses on URLs with // different schemes or ports. TEST_F(SiteInstanceTest, IsSameWebSite) { GURL url_foo = GURL("http://foo/a.html"); GURL url_foo2 = GURL("http://foo/b.html"); GURL url_foo_https = GURL("https://foo/a.html"); GURL url_foo_port = GURL("http://foo:8080/a.html"); GURL url_javascript = GURL("javascript:alert(1);"); GURL url_crash = GURL(chrome::kAboutCrashURL); GURL url_hang = GURL(chrome::kAboutHangURL); GURL url_shorthang = GURL(chrome::kAboutShorthangURL); // Same scheme and port -> same site. EXPECT_TRUE(SiteInstance::IsSameWebSite(NULL, url_foo, url_foo2)); // Different scheme -> different site. EXPECT_FALSE(SiteInstance::IsSameWebSite(NULL, url_foo, url_foo_https)); // Different port -> same site. // (Changes to document.domain make renderer ignore the port.) EXPECT_TRUE(SiteInstance::IsSameWebSite(NULL, url_foo, url_foo_port)); // JavaScript links should be considered same site for anything. EXPECT_TRUE(SiteInstance::IsSameWebSite(NULL, url_javascript, url_foo)); EXPECT_TRUE(SiteInstance::IsSameWebSite(NULL, url_javascript, url_foo_https)); EXPECT_TRUE(SiteInstance::IsSameWebSite(NULL, url_javascript, url_foo_port)); // The crash/hang URLs should also be treated as same site. (Bug 1143809.) EXPECT_TRUE(SiteInstance::IsSameWebSite(NULL, url_crash, url_foo)); EXPECT_TRUE(SiteInstance::IsSameWebSite(NULL, url_hang, url_foo)); EXPECT_TRUE(SiteInstance::IsSameWebSite(NULL, url_shorthang, url_foo)); } // Test to ensure that there is only one SiteInstance per site in a given // BrowsingInstance, when process-per-site is not in use. TEST_F(SiteInstanceTest, OneSiteInstancePerSite) { int deleteCounter = 0; TestBrowsingInstance* browsing_instance = new TestBrowsingInstance(NULL, &deleteCounter); browsing_instance->use_process_per_site = false; const GURL url_a1("http://www.google.com/1.html"); scoped_refptr<SiteInstance> site_instance_a1( browsing_instance->GetSiteInstanceForURL(url_a1)); EXPECT_TRUE(site_instance_a1.get() != NULL); // A separate site should create a separate SiteInstance. const GURL url_b1("http://www.yahoo.com/"); scoped_refptr<SiteInstance> site_instance_b1( browsing_instance->GetSiteInstanceForURL(url_b1)); EXPECT_NE(site_instance_a1.get(), site_instance_b1.get()); // Getting the new SiteInstance from the BrowsingInstance and from another // SiteInstance in the BrowsingInstance should give the same result. EXPECT_EQ(site_instance_b1.get(), site_instance_a1->GetRelatedSiteInstance(url_b1)); // A second visit to the original site should return the same SiteInstance. const GURL url_a2("http://www.google.com/2.html"); EXPECT_EQ(site_instance_a1.get(), browsing_instance->GetSiteInstanceForURL(url_a2)); EXPECT_EQ(site_instance_a1.get(), site_instance_a1->GetRelatedSiteInstance(url_a2)); // A visit to the original site in a new BrowsingInstance (same or different // profile) should return a different SiteInstance. TestBrowsingInstance* browsing_instance2 = new TestBrowsingInstance(NULL, &deleteCounter); browsing_instance2->use_process_per_site = false; // Ensure the new SiteInstance is ref counted so that it gets deleted. scoped_refptr<SiteInstance> site_instance_a2_2( browsing_instance2->GetSiteInstanceForURL(url_a2)); EXPECT_NE(site_instance_a1.get(), site_instance_a2_2.get()); // Should be able to see that we do have SiteInstances. EXPECT_TRUE(browsing_instance->HasSiteInstance( GURL("http://mail.google.com"))); EXPECT_TRUE(browsing_instance2->HasSiteInstance( GURL("http://mail.google.com"))); EXPECT_TRUE(browsing_instance->HasSiteInstance( GURL("http://mail.yahoo.com"))); // Should be able to see that we don't have SiteInstances. EXPECT_FALSE(browsing_instance->HasSiteInstance( GURL("https://www.google.com"))); EXPECT_FALSE(browsing_instance2->HasSiteInstance( GURL("http://www.yahoo.com"))); // browsing_instances will be deleted when their SiteInstances are deleted } // Test to ensure that there is only one SiteInstance per site for an entire // Profile, if process-per-site is in use. TEST_F(SiteInstanceTest, OneSiteInstancePerSiteInProfile) { int deleteCounter = 0; TestBrowsingInstance* browsing_instance = new TestBrowsingInstance(NULL, &deleteCounter); browsing_instance->use_process_per_site = true; const GURL url_a1("http://www.google.com/1.html"); scoped_refptr<SiteInstance> site_instance_a1( browsing_instance->GetSiteInstanceForURL(url_a1)); EXPECT_TRUE(site_instance_a1.get() != NULL); // A separate site should create a separate SiteInstance. const GURL url_b1("http://www.yahoo.com/"); scoped_refptr<SiteInstance> site_instance_b1( browsing_instance->GetSiteInstanceForURL(url_b1)); EXPECT_NE(site_instance_a1.get(), site_instance_b1.get()); // Getting the new SiteInstance from the BrowsingInstance and from another // SiteInstance in the BrowsingInstance should give the same result. EXPECT_EQ(site_instance_b1.get(), site_instance_a1->GetRelatedSiteInstance(url_b1)); // A second visit to the original site should return the same SiteInstance. const GURL url_a2("http://www.google.com/2.html"); EXPECT_EQ(site_instance_a1.get(), browsing_instance->GetSiteInstanceForURL(url_a2)); EXPECT_EQ(site_instance_a1.get(), site_instance_a1->GetRelatedSiteInstance(url_a2)); // A visit to the original site in a new BrowsingInstance (same profile) // should also return the same SiteInstance. // This BrowsingInstance doesn't get its own SiteInstance within the test, so // it won't be deleted by its children. Thus, we'll keep a ref count to it // to make sure it gets deleted. scoped_refptr<TestBrowsingInstance> browsing_instance2( new TestBrowsingInstance(NULL, &deleteCounter)); browsing_instance2->use_process_per_site = true; EXPECT_EQ(site_instance_a1.get(), browsing_instance2->GetSiteInstanceForURL(url_a2)); // A visit to the original site in a new BrowsingInstance (different profile) // should return a different SiteInstance. scoped_ptr<TestingProfile> profile(new TestingProfile()); TestBrowsingInstance* browsing_instance3 = new TestBrowsingInstance(profile.get(), &deleteCounter); browsing_instance3->use_process_per_site = true; // Ensure the new SiteInstance is ref counted so that it gets deleted. scoped_refptr<SiteInstance> site_instance_a2_3( browsing_instance3->GetSiteInstanceForURL(url_a2)); EXPECT_NE(site_instance_a1.get(), site_instance_a2_3.get()); // Should be able to see that we do have SiteInstances. EXPECT_TRUE(browsing_instance->HasSiteInstance( GURL("http://mail.google.com"))); // visited before EXPECT_TRUE(browsing_instance2->HasSiteInstance( GURL("http://mail.google.com"))); // visited before EXPECT_TRUE(browsing_instance->HasSiteInstance( GURL("http://mail.yahoo.com"))); // visited before EXPECT_TRUE(browsing_instance2->HasSiteInstance( GURL("http://www.yahoo.com"))); // different BI, but same profile // Should be able to see that we don't have SiteInstances. EXPECT_FALSE(browsing_instance->HasSiteInstance( GURL("https://www.google.com"))); // not visited before EXPECT_FALSE(browsing_instance3->HasSiteInstance( GURL("http://www.yahoo.com"))); // different BI, different profile // browsing_instances will be deleted when their SiteInstances are deleted } static SiteInstance* CreateSiteInstance(RenderProcessHostFactory* factory, const GURL& url) { SiteInstance* instance = SiteInstance::CreateSiteInstanceForURL(NULL, url); instance->set_render_process_host_factory(factory); return instance; } // Test to ensure that pages that require certain privileges are grouped // in processes with similar pages. TEST_F(SiteInstanceTest, ProcessSharingByType) { MockRenderProcessHostFactory rph_factory; ChildProcessSecurityPolicy* policy = ChildProcessSecurityPolicy::GetInstance(); // Make a bunch of mock renderers so that we hit the limit. std::vector<MockRenderProcessHost*> hosts; for (size_t i = 0; i < chrome::kMaxRendererProcessCount; ++i) hosts.push_back(new MockRenderProcessHost(NULL)); // Create some extension instances and make sure they share a process. scoped_refptr<SiteInstance> extension1_instance( CreateSiteInstance(&rph_factory, GURL("chrome-extension://foo/bar"))); policy->GrantExtensionBindings(extension1_instance->GetProcess()->id()); scoped_refptr<SiteInstance> extension2_instance( CreateSiteInstance(&rph_factory, GURL("chrome-extension://baz/bar"))); scoped_ptr<RenderProcessHost> extension_host( extension1_instance->GetProcess()); EXPECT_EQ(extension1_instance->GetProcess(), extension2_instance->GetProcess()); // Create some WebUI instances and make sure they share a process. scoped_refptr<SiteInstance> webui1_instance( CreateSiteInstance(&rph_factory, GURL(kWebUIScheme + std::string("://newtab")))); policy->GrantWebUIBindings(webui1_instance->GetProcess()->id()); scoped_refptr<SiteInstance> webui2_instance( CreateSiteInstance(&rph_factory, GURL(kWebUIScheme + std::string("://history")))); scoped_ptr<RenderProcessHost> dom_host(webui1_instance->GetProcess()); EXPECT_EQ(webui1_instance->GetProcess(), webui2_instance->GetProcess()); // Make sure none of differing privilege processes are mixed. EXPECT_NE(extension1_instance->GetProcess(), webui1_instance->GetProcess()); for (size_t i = 0; i < chrome::kMaxRendererProcessCount; ++i) { EXPECT_NE(extension1_instance->GetProcess(), hosts[i]); EXPECT_NE(webui1_instance->GetProcess(), hosts[i]); } STLDeleteContainerPointers(hosts.begin(), hosts.end()); } // Test to ensure that profiles that derive from each other share site // information. TEST_F(SiteInstanceTest, GetSiteInstanceMap) { int deleteCounter = 0; scoped_ptr<Profile> p1(new TestingProfile()); scoped_ptr<Profile> p2(new TestingProfile()); scoped_ptr<Profile> p3(new DerivedTestingProfile(p1.get())); // In this test, instances 1 and 2 will be deleted automatically when the // SiteInstance objects they return are deleted. However, instance 3 never // returns any SitesIntance objects in this test, so will not be automatically // deleted. It must be deleted manually. TestBrowsingInstance* instance1(new TestBrowsingInstance(p1.get(), &deleteCounter)); TestBrowsingInstance* instance2(new TestBrowsingInstance(p2.get(), &deleteCounter)); scoped_refptr<TestBrowsingInstance> instance3( new TestBrowsingInstance(p3.get(), &deleteCounter)); instance1->use_process_per_site = true; instance2->use_process_per_site = true; instance3->use_process_per_site = true; // The same profile with the same site. scoped_refptr<SiteInstance> s1a(instance1->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); scoped_refptr<SiteInstance> s1b(instance1->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); EXPECT_EQ(s1a, s1b); // The same profile with different sites. scoped_refptr<SiteInstance> s2a(instance1->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); scoped_refptr<SiteInstance> s2b(instance1->GetSiteInstanceForURL( GURL("chrome-extension://foo/boo"))); EXPECT_NE(s2a, s2b); // The different profiles with the same site. scoped_refptr<SiteInstance> s3a(instance1->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); scoped_refptr<SiteInstance> s3b(instance2->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); EXPECT_NE(s3a, s3b); // The different profiles with different sites. scoped_refptr<SiteInstance> s4a(instance1->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); scoped_refptr<SiteInstance> s4b(instance2->GetSiteInstanceForURL( GURL("chrome-extension://foo/boo"))); EXPECT_NE(s4a, s4b); // The derived profiles with the same site. scoped_refptr<SiteInstance> s5a(instance1->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); scoped_refptr<SiteInstance> s5b(instance3->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); EXPECT_EQ(s5a, s5b); // The derived profiles with the different sites. scoped_refptr<SiteInstance> s6a(instance1->GetSiteInstanceForURL( GURL("chrome-extension://baz/bar"))); scoped_refptr<SiteInstance> s6b(instance3->GetSiteInstanceForURL( GURL("chrome-extension://foo/boo"))); EXPECT_NE(s6a, s6b); }