[Script Injection] Allow whitelisted extensions to inject scripts everywhere

chrome/browser/extensions/all_urls_apitest.cc

 // 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/command_line.h"
 #include "chrome/browser/extensions/extension_apitest.h"
+#include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/test/base/test_switches.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/crx_file/id_util.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/common/extension.h"
 #include "extensions/common/extensions_client.h"
 #include "extensions/test/extension_test_message_listener.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 
+namespace extensions {
+
+namespace {
 const std::string kAllUrlsTarget = "/extensions/api_test/all_urls/index.html";
+}
 
-typedef ExtensionApiTest AllUrlsApiTest;
+class AllUrlsApiTest : public ExtensionApiTest {
+ protected:
+  AllUrlsApiTest() {}
+  ~AllUrlsApiTest() override {}
+
+  const Extension* content_script() const { return content_script_.get(); }
+  const Extension* execute_script() const { return execute_script_.get(); }
+
+  void WhitelistExtensions() {
+    ExtensionsClient::ScriptingWhitelist whitelist;
+    whitelist.push_back(content_script_->id());
+    whitelist.push_back(execute_script_->id());
+    ExtensionsClient::Get()->SetScriptingWhitelist(whitelist);
+    // Extensions will have certain permissions withheld at initialization if
+    // they aren't whitelisted, so we need to reload them.
+    extension_service()->ReloadExtension(content_script_->id());
+    extension_service()->ReloadExtension(execute_script_->id());
+  }
+
+ private:
+  void SetUpOnMainThread() override {
+    ExtensionApiTest::SetUpOnMainThread();
+    base::FilePath data_dir = test_data_dir_.AppendASCII("all_urls");
+    content_script_ = LoadExtension(data_dir.AppendASCII("content_script"));
+    ASSERT_TRUE(content_script_);
+    execute_script_ = LoadExtension(data_dir.AppendASCII("execute_script"));
+    ASSERT_TRUE(execute_script_);
+  }
+
+  scoped_refptr<const Extension> content_script_;
+  scoped_refptr<const Extension> execute_script_;
+
+  DISALLOW_COPY_AND_ASSIGN(AllUrlsApiTest);
+};
 
 #if (defined(OS_WIN) && !defined(NDEBUG)) || defined(OS_CHROMEOS)
 // http://crbug.com/174341
 #define MAYBE_WhitelistedExtension DISABLED_WhitelistedExtension
 #else
 #define MAYBE_WhitelistedExtension WhitelistedExtension
 #endif
 IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, MAYBE_WhitelistedExtension) {
 #if defined(OS_WIN) && defined(USE_ASH)
   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
           switches::kAshBrowserTests))
     return;
 #endif
 
-  // First setup the two extensions.
-  base::FilePath extension_dir1 = test_data_dir_.AppendASCII("all_urls")
-                                          .AppendASCII("content_script");
-  base::FilePath extension_dir2 = test_data_dir_.AppendASCII("all_urls")
-                                          .AppendASCII("execute_script");
-
-  // Then add the two extensions to the whitelist.
-  extensions::ExtensionsClient::ScriptingWhitelist whitelist;
-  whitelist.push_back(crx_file::id_util::GenerateIdForPath(extension_dir1));
-  whitelist.push_back(crx_file::id_util::GenerateIdForPath(extension_dir2));
-  extensions::ExtensionsClient::Get()->SetScriptingWhitelist(whitelist);
-
-  // Then load extensions.
-  extensions::ExtensionRegistry* registry =
-      extensions::ExtensionRegistry::Get(browser()->profile());
-  const size_t size_before = registry->enabled_extensions().size();
-  ASSERT_TRUE(LoadExtension(extension_dir1));
-  ASSERT_TRUE(LoadExtension(extension_dir2));
-  EXPECT_EQ(size_before + 2, registry->enabled_extensions().size());
+  WhitelistExtensions();
 
   std::string url;
 
   // Now verify we run content scripts on chrome://newtab/.
   url = "chrome://newtab/";
   ExtensionTestMessageListener listener1a("content script: " + url, false);
   ExtensionTestMessageListener listener1b("execute: " + url, false);
   ui_test_utils::NavigateToURL(browser(), GURL(url));
   ASSERT_TRUE(listener1a.WaitUntilSatisfied());
   ASSERT_TRUE(listener1b.WaitUntilSatisfied());
@@ skipping 29 matching lines @@
                                           false);
   ExtensionTestMessageListener listener5b("execute: " + page_url.spec(), false);
   ui_test_utils::NavigateToURL(browser(), page_url);
   ASSERT_TRUE(listener5a.WaitUntilSatisfied());
   ASSERT_TRUE(listener5b.WaitUntilSatisfied());
 }
 
 // Test that an extension NOT whitelisted for scripting can ask for <all_urls>
 // and run scripts on non-restricted all pages.
 IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, RegularExtensions) {
-  // First load the two extensions.
-  base::FilePath extension_dir1 = test_data_dir_.AppendASCII("all_urls")
-                                          .AppendASCII("content_script");
-  base::FilePath extension_dir2 = test_data_dir_.AppendASCII("all_urls")
-                                          .AppendASCII("execute_script");
-
-  extensions::ExtensionRegistry* registry =
-      extensions::ExtensionRegistry::Get(browser()->profile());
-  const size_t size_before = registry->enabled_extensions().size();
-  ASSERT_TRUE(LoadExtension(extension_dir1));
-  ASSERT_TRUE(LoadExtension(extension_dir2));
-  EXPECT_EQ(size_before + 2, registry->enabled_extensions().size());
-
   // Now verify we can script a regular http page.
   ASSERT_TRUE(StartEmbeddedTestServer());
   GURL page_url = embedded_test_server()->GetURL(kAllUrlsTarget);
   ExtensionTestMessageListener listener1a("content script: " + page_url.spec(),
                                           false);
   ExtensionTestMessageListener listener1b("execute: " + page_url.spec(), false);
   ui_test_utils::NavigateToURL(browser(), page_url);
   ASSERT_TRUE(listener1a.WaitUntilSatisfied());
   ASSERT_TRUE(listener1b.WaitUntilSatisfied());
 }
+
+IN_PROC_BROWSER_TEST_F(AllUrlsApiTest,
+                       WhitelistedExtensionRunsOnExtensionPages) {
+  WhitelistExtensions();
+  const Extension* bystander =
+      LoadExtension(test_data_dir_.AppendASCII("all_urls")
+                                  .AppendASCII("bystander"));
+  ASSERT_TRUE(bystander);
+
+  // TODO(devlin): This test should probably go in the WhitelistedExtension test
+  // above, but that one has a bunch of disableds, so it wouldn't be very
+  // useful.
+  GURL url(bystander->GetResourceURL("page.html"));
+  ExtensionTestMessageListener listenerA(
+      "content script: " + url.spec(), false);
+  ExtensionTestMessageListener listenerB("execute: " + url.spec(), false);
+  ui_test_utils::NavigateToURL(browser(), GURL(url));
+  ASSERT_TRUE(listenerA.WaitUntilSatisfied());
+  ASSERT_TRUE(listenerB.WaitUntilSatisfied());
+}
+
+}  // namespace extensions