1// Copyright (c) 2011 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/command_line.h"
6#include "chrome/browser/extensions/extension_apitest.h"
7#include "chrome/browser/extensions/extension_service.h"
8#include "chrome/browser/profiles/profile.h"
9#include "chrome/browser/ui/browser.h"
10#include "chrome/test/base/test_switches.h"
11#include "chrome/test/base/ui_test_utils.h"
12#include "components/crx_file/id_util.h"
13#include "extensions/browser/extension_system.h"
14#include "extensions/common/extension.h"
15#include "extensions/common/extensions_client.h"
16#include "extensions/test/extension_test_message_listener.h"
17
18const std::string kAllUrlsTarget =
19    "files/extensions/api_test/all_urls/index.html";
20
21typedef ExtensionApiTest AllUrlsApiTest;
22
23#if defined(OS_WIN) && !defined(NDEBUG)
24// http://crbug.com/174341
25#define MAYBE_WhitelistedExtension DISABLED_WhitelistedExtension
26#else
27#define MAYBE_WhitelistedExtension WhitelistedExtension
28#endif
29IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, MAYBE_WhitelistedExtension) {
30#if defined(OS_WIN) && defined(USE_ASH)
31  // Disable this test in Metro+Ash for now (http://crbug.com/262796).
32  if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
33    return;
34#endif
35
36  // First setup the two extensions.
37  base::FilePath extension_dir1 = test_data_dir_.AppendASCII("all_urls")
38                                          .AppendASCII("content_script");
39  base::FilePath extension_dir2 = test_data_dir_.AppendASCII("all_urls")
40                                          .AppendASCII("execute_script");
41
42  // Then add the two extensions to the whitelist.
43  extensions::ExtensionsClient::ScriptingWhitelist whitelist;
44  whitelist.push_back(crx_file::id_util::GenerateIdForPath(extension_dir1));
45  whitelist.push_back(crx_file::id_util::GenerateIdForPath(extension_dir2));
46  extensions::ExtensionsClient::Get()->SetScriptingWhitelist(whitelist);
47
48  // Then load extensions.
49  ExtensionService* service = extensions::ExtensionSystem::Get(
50      browser()->profile())->extension_service();
51  const size_t size_before = service->extensions()->size();
52  ASSERT_TRUE(LoadExtension(extension_dir1));
53  ASSERT_TRUE(LoadExtension(extension_dir2));
54  EXPECT_EQ(size_before + 2, service->extensions()->size());
55
56  std::string url;
57
58  // Now verify we run content scripts on chrome://newtab/.
59  url = "chrome://newtab/";
60  ExtensionTestMessageListener listener1a("content script: " + url, false);
61  ExtensionTestMessageListener listener1b("execute: " + url, false);
62  ui_test_utils::NavigateToURL(browser(), GURL(url));
63  ASSERT_TRUE(listener1a.WaitUntilSatisfied());
64  ASSERT_TRUE(listener1b.WaitUntilSatisfied());
65
66  // Now verify data: urls.
67  url = "data:text/html;charset=utf-8,<html>asdf</html>";
68  ExtensionTestMessageListener listener2a("content script: " + url, false);
69  ExtensionTestMessageListener listener2b("execute: " + url, false);
70  ui_test_utils::NavigateToURL(browser(), GURL(url));
71  ASSERT_TRUE(listener2a.WaitUntilSatisfied());
72  ASSERT_TRUE(listener2b.WaitUntilSatisfied());
73
74  // Now verify chrome://version/.
75  url = "chrome://version/";
76  ExtensionTestMessageListener listener3a("content script: " + url, false);
77  ExtensionTestMessageListener listener3b("execute: " + url, false);
78  ui_test_utils::NavigateToURL(browser(), GURL(url));
79  ASSERT_TRUE(listener3a.WaitUntilSatisfied());
80  ASSERT_TRUE(listener3b.WaitUntilSatisfied());
81
82  // Now verify about:blank.
83  url = "about:blank";
84  ExtensionTestMessageListener listener4a("content script: " + url, false);
85  ExtensionTestMessageListener listener4b("execute: " + url, false);
86  ui_test_utils::NavigateToURL(browser(), GURL(url));
87  ASSERT_TRUE(listener4a.WaitUntilSatisfied());
88  ASSERT_TRUE(listener4b.WaitUntilSatisfied());
89
90  // Now verify we can script a regular http page.
91  ASSERT_TRUE(test_server()->Start());
92  GURL page_url = test_server()->GetURL(kAllUrlsTarget);
93  ExtensionTestMessageListener listener5a("content script: " + page_url.spec(),
94                                          false);
95  ExtensionTestMessageListener listener5b("execute: " + page_url.spec(), false);
96  ui_test_utils::NavigateToURL(browser(), page_url);
97  ASSERT_TRUE(listener5a.WaitUntilSatisfied());
98  ASSERT_TRUE(listener5b.WaitUntilSatisfied());
99}
100
101// Test that an extension NOT whitelisted for scripting can ask for <all_urls>
102// and run scripts on non-restricted all pages.
103IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, RegularExtensions) {
104  // First load the two extensions.
105  base::FilePath extension_dir1 = test_data_dir_.AppendASCII("all_urls")
106                                          .AppendASCII("content_script");
107  base::FilePath extension_dir2 = test_data_dir_.AppendASCII("all_urls")
108                                          .AppendASCII("execute_script");
109
110  ExtensionService* service = extensions::ExtensionSystem::Get(
111      browser()->profile())->extension_service();
112  const size_t size_before = service->extensions()->size();
113  ASSERT_TRUE(LoadExtension(extension_dir1));
114  ASSERT_TRUE(LoadExtension(extension_dir2));
115  EXPECT_EQ(size_before + 2, service->extensions()->size());
116
117  // Now verify we can script a regular http page.
118  ASSERT_TRUE(test_server()->Start());
119  GURL page_url = test_server()->GetURL(kAllUrlsTarget);
120  ExtensionTestMessageListener listener1a("content script: " + page_url.spec(),
121                                          false);
122  ExtensionTestMessageListener listener1b("execute: " + page_url.spec(), false);
123  ui_test_utils::NavigateToURL(browser(), page_url);
124  ASSERT_TRUE(listener1a.WaitUntilSatisfied());
125  ASSERT_TRUE(listener1b.WaitUntilSatisfied());
126}
127