1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "base/strings/utf_string_conversions.h" 6 #include "chrome/browser/extensions/extension_apitest.h" 7 #include "chrome/browser/extensions/extension_host.h" 8 #include "chrome/browser/extensions/extension_service.h" 9 #include "chrome/browser/extensions/extension_system.h" 10 #include "chrome/browser/profiles/profile.h" 11 #include "chrome/browser/profiles/profile_manager.h" 12 #include "chrome/browser/ui/browser.h" 13 #include "chrome/browser/ui/tabs/tab_strip_model.h" 14 #include "chrome/common/url_constants.h" 15 #include "chrome/test/base/ui_test_utils.h" 16 #include "content/public/browser/render_process_host.h" 17 #include "content/public/browser/render_view_host.h" 18 #include "content/public/browser/site_instance.h" 19 #include "content/public/browser/web_contents.h" 20 #include "extensions/browser/process_manager.h" 21 #include "extensions/common/switches.h" 22 #include "net/dns/mock_host_resolver.h" 23 #include "net/test/embedded_test_server/embedded_test_server.h" 24 25 using content::NavigationController; 26 using content::WebContents; 27 28 namespace { 29 30 class ProcessManagementTest : public ExtensionBrowserTest { 31 private: 32 // This is needed for testing isolated apps, which are still experimental. 33 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { 34 ExtensionBrowserTest::SetUpCommandLine(command_line); 35 command_line->AppendSwitch( 36 extensions::switches::kEnableExperimentalExtensionApis); 37 } 38 }; 39 40 } // namespace 41 42 43 // TODO(nasko): crbug.com/173137 44 #if defined(OS_WIN) 45 #define MAYBE_ProcessOverflow DISABLED_ProcessOverflow 46 #else 47 #define MAYBE_ProcessOverflow ProcessOverflow 48 #endif 49 50 // Ensure that an isolated app never shares a process with WebUIs, non-isolated 51 // extensions, and normal webpages. None of these should ever comingle 52 // RenderProcessHosts even if we hit the process limit. 53 IN_PROC_BROWSER_TEST_F(ProcessManagementTest, MAYBE_ProcessOverflow) { 54 // Set max renderers to 1 to force running out of processes. 55 content::RenderProcessHost::SetMaxRendererProcessCount(1); 56 57 host_resolver()->AddRule("*", "127.0.0.1"); 58 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); 59 60 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("isolated_apps/app1"))); 61 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("isolated_apps/app2"))); 62 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("hosted_app"))); 63 ASSERT_TRUE( 64 LoadExtension(test_data_dir_.AppendASCII("api_test/app_process"))); 65 66 // The app under test acts on URLs whose host is "localhost", 67 // so the URLs we navigate to must have host "localhost". 68 GURL base_url = embedded_test_server()->GetURL( 69 "/extensions/"); 70 GURL::Replacements replace_host; 71 std::string host_str("localhost"); // Must stay in scope with replace_host. 72 replace_host.SetHostStr(host_str); 73 base_url = base_url.ReplaceComponents(replace_host); 74 75 // Load an extension before adding tabs. 76 const extensions::Extension* extension1 = LoadExtension( 77 test_data_dir_.AppendASCII("api_test/browser_action/basics")); 78 ASSERT_TRUE(extension1); 79 GURL extension1_url = extension1->url(); 80 81 // Create multiple tabs for each type of renderer that might exist. 82 ui_test_utils::NavigateToURLWithDisposition( 83 browser(), base_url.Resolve("isolated_apps/app1/main.html"), 84 CURRENT_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 85 ui_test_utils::NavigateToURLWithDisposition( 86 browser(), GURL(chrome::kChromeUINewTabURL), 87 NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 88 ui_test_utils::NavigateToURLWithDisposition( 89 browser(), base_url.Resolve("hosted_app/main.html"), 90 NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 91 ui_test_utils::NavigateToURLWithDisposition( 92 browser(), base_url.Resolve("test_file.html"), 93 NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 94 95 ui_test_utils::NavigateToURLWithDisposition( 96 browser(), base_url.Resolve("isolated_apps/app2/main.html"), 97 NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 98 ui_test_utils::NavigateToURLWithDisposition( 99 browser(), GURL(chrome::kChromeUINewTabURL), 100 NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 101 ui_test_utils::NavigateToURLWithDisposition( 102 browser(), base_url.Resolve("api_test/app_process/path1/empty.html"), 103 NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 104 ui_test_utils::NavigateToURLWithDisposition( 105 browser(), base_url.Resolve("test_file_with_body.html"), 106 NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 107 108 // Load another copy of isolated app 1. 109 ui_test_utils::NavigateToURLWithDisposition( 110 browser(), base_url.Resolve("isolated_apps/app1/main.html"), 111 NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 112 113 // Load another extension. 114 const extensions::Extension* extension2 = LoadExtension( 115 test_data_dir_.AppendASCII("api_test/browser_action/close_background")); 116 ASSERT_TRUE(extension2); 117 GURL extension2_url = extension2->url(); 118 119 // Get tab processes. 120 ASSERT_EQ(9, browser()->tab_strip_model()->count()); 121 content::RenderProcessHost* isolated1_host = 122 browser()->tab_strip_model()->GetWebContentsAt(0)->GetRenderProcessHost(); 123 content::RenderProcessHost* ntp1_host = 124 browser()->tab_strip_model()->GetWebContentsAt(1)->GetRenderProcessHost(); 125 content::RenderProcessHost* hosted1_host = 126 browser()->tab_strip_model()->GetWebContentsAt(2)->GetRenderProcessHost(); 127 content::RenderProcessHost* web1_host = 128 browser()->tab_strip_model()->GetWebContentsAt(3)->GetRenderProcessHost(); 129 130 content::RenderProcessHost* isolated2_host = 131 browser()->tab_strip_model()->GetWebContentsAt(4)->GetRenderProcessHost(); 132 content::RenderProcessHost* ntp2_host = 133 browser()->tab_strip_model()->GetWebContentsAt(5)->GetRenderProcessHost(); 134 content::RenderProcessHost* hosted2_host = 135 browser()->tab_strip_model()->GetWebContentsAt(6)->GetRenderProcessHost(); 136 content::RenderProcessHost* web2_host = 137 browser()->tab_strip_model()->GetWebContentsAt(7)->GetRenderProcessHost(); 138 139 content::RenderProcessHost* second_isolated1_host = 140 browser()->tab_strip_model()->GetWebContentsAt(8)->GetRenderProcessHost(); 141 142 // Get extension processes. 143 extensions::ProcessManager* process_manager = 144 extensions::ExtensionSystem::Get(browser()->profile())-> 145 process_manager(); 146 content::RenderProcessHost* extension1_host = 147 process_manager->GetSiteInstanceForURL(extension1_url)->GetProcess(); 148 content::RenderProcessHost* extension2_host = 149 process_manager->GetSiteInstanceForURL(extension2_url)->GetProcess(); 150 151 // An isolated app only shares with other instances of itself, not other 152 // isolated apps or anything else. 153 EXPECT_EQ(isolated1_host, second_isolated1_host); 154 EXPECT_NE(isolated1_host, isolated2_host); 155 EXPECT_NE(isolated1_host, ntp1_host); 156 EXPECT_NE(isolated1_host, hosted1_host); 157 EXPECT_NE(isolated1_host, web1_host); 158 EXPECT_NE(isolated1_host, extension1_host); 159 EXPECT_NE(isolated2_host, ntp1_host); 160 EXPECT_NE(isolated2_host, hosted1_host); 161 EXPECT_NE(isolated2_host, web1_host); 162 EXPECT_NE(isolated2_host, extension1_host); 163 164 // Everything else is clannish. WebUI only shares with other WebUI. 165 EXPECT_EQ(ntp1_host, ntp2_host); 166 EXPECT_NE(ntp1_host, hosted1_host); 167 EXPECT_NE(ntp1_host, web1_host); 168 EXPECT_NE(ntp1_host, extension1_host); 169 170 // Hosted apps only share with each other. 171 // Note that hosted2_host's app has the background permission and will use 172 // process-per-site mode, but it should still share with hosted1_host's app. 173 EXPECT_EQ(hosted1_host, hosted2_host); 174 EXPECT_NE(hosted1_host, web1_host); 175 EXPECT_NE(hosted1_host, extension1_host); 176 177 // Web pages only share with each other. 178 EXPECT_EQ(web1_host, web2_host); 179 EXPECT_NE(web1_host, extension1_host); 180 181 // Extensions only share with each other. 182 EXPECT_EQ(extension1_host, extension2_host); 183 } 184 185 // See 186 #if defined(OS_WIN) 187 #define MAYBE_ExtensionProcessBalancing DISABLED_ExtensionProcessBalancing 188 #else 189 #define MAYBE_ExtensionProcessBalancing ExtensionProcessBalancing 190 #endif 191 // Test to verify that the policy of maximum share of extension processes is 192 // properly enforced. 193 IN_PROC_BROWSER_TEST_F(ProcessManagementTest, MAYBE_ExtensionProcessBalancing) { 194 // Set max renderers to 6 so we can expect 2 extension processes to be 195 // allocated. 196 content::RenderProcessHost::SetMaxRendererProcessCount(6); 197 198 host_resolver()->AddRule("*", "127.0.0.1"); 199 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); 200 201 // The app under test acts on URLs whose host is "localhost", 202 // so the URLs we navigate to must have host "localhost". 203 GURL base_url = embedded_test_server()->GetURL( 204 "/extensions/"); 205 GURL::Replacements replace_host; 206 std::string host_str("localhost"); // Must stay in scope with replace_host. 207 replace_host.SetHostStr(host_str); 208 base_url = base_url.ReplaceComponents(replace_host); 209 210 ASSERT_TRUE(LoadExtension( 211 test_data_dir_.AppendASCII("api_test/browser_action/none"))); 212 ASSERT_TRUE(LoadExtension( 213 test_data_dir_.AppendASCII("api_test/browser_action/basics"))); 214 ASSERT_TRUE(LoadExtension( 215 test_data_dir_.AppendASCII("api_test/browser_action/remove_popup"))); 216 ASSERT_TRUE(LoadExtension( 217 test_data_dir_.AppendASCII("api_test/browser_action/add_popup"))); 218 ASSERT_TRUE(LoadExtension( 219 test_data_dir_.AppendASCII("api_test/browser_action/no_icon"))); 220 ASSERT_TRUE(LoadExtension( 221 test_data_dir_.AppendASCII("isolated_apps/app1"))); 222 ASSERT_TRUE(LoadExtension( 223 test_data_dir_.AppendASCII("api_test/management/test"))); 224 225 ui_test_utils::NavigateToURLWithDisposition( 226 browser(), base_url.Resolve("isolated_apps/app1/main.html"), 227 CURRENT_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 228 229 ui_test_utils::NavigateToURLWithDisposition( 230 browser(), base_url.Resolve("api_test/management/test/basics.html"), 231 CURRENT_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 232 233 std::set<int> process_ids; 234 Profile* profile = browser()->profile(); 235 extensions::ProcessManager* epm = extensions::ExtensionSystem::Get(profile)-> 236 process_manager(); 237 for (extensions::ProcessManager::const_iterator iter = 238 epm->background_hosts().begin(); 239 iter != epm->background_hosts().end(); ++iter) { 240 process_ids.insert((*iter)->render_process_host()->GetID()); 241 } 242 243 // We've loaded 5 extensions with background pages, 1 extension without 244 // background page, and one isolated app. We expect only 2 unique processes 245 // hosting those extensions. 246 ExtensionService* service = 247 extensions::ExtensionSystem::Get(profile)->extension_service(); 248 249 EXPECT_GE((size_t) 6, service->process_map()->size()); 250 EXPECT_EQ((size_t) 2, process_ids.size()); 251 } 252
