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 <vector>
6
7 #include "base/command_line.h"
8 #include "base/file_util.h"
9 #include "base/files/file_path.h"
10 #include "base/path_service.h"
11 #include "base/strings/string_util.h"
12 #include "base/strings/stringprintf.h"
13 #include "chrome/browser/chrome_notification_types.h"
14 #include "chrome/browser/extensions/extension_service.h"
15 #include "chrome/browser/extensions/extension_util.h"
16 #include "chrome/browser/extensions/user_script_master.h"
17 #include "chrome/browser/prefs/chrome_pref_service_factory.h"
18 #include "chrome/browser/profiles/profile.h"
19 #include "chrome/browser/ui/browser.h"
20 #include "chrome/browser/ui/tabs/tab_strip_model.h"
21 #include "chrome/common/chrome_constants.h"
22 #include "chrome/common/chrome_paths.h"
23 #include "chrome/common/chrome_switches.h"
24 #include "chrome/test/base/in_process_browser_test.h"
25 #include "chrome/test/base/testing_profile.h"
26 #include "chrome/test/base/ui_test_utils.h"
27 #include "content/public/browser/notification_details.h"
28 #include "content/public/browser/notification_service.h"
29 #include "content/public/browser/web_contents.h"
30 #include "content/public/common/content_switches.h"
31 #include "content/public/test/browser_test_utils.h"
32 #include "extensions/browser/extension_registry.h"
33 #include "extensions/browser/extension_system.h"
34 #include "extensions/common/extension.h"
35 #include "extensions/common/extension_set.h"
36 #include "extensions/common/feature_switch.h"
37 #include "net/base/filename_util.h"
38
39 using extensions::FeatureSwitch;
40
41 // This file contains high-level startup tests for the extensions system. We've
42 // had many silly bugs where command line flags did not get propagated correctly
43 // into the services, so we didn't start correctly.
44
45 class ExtensionStartupTestBase : public InProcessBrowserTest {
46 public:
ExtensionStartupTestBase()47 ExtensionStartupTestBase() : unauthenticated_load_allowed_(true) {
48 num_expected_extensions_ = 3;
49 }
50
51 protected:
52 // InProcessBrowserTest
SetUpCommandLine(CommandLine * command_line)53 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
54 if (load_extensions_.empty()) {
55 // If no |load_extensions_| were specified, allow unauthenticated
56 // extension settings to be loaded from Preferences as if they had been
57 // authenticated correctly before they were handed to the ExtensionSystem.
58 command_line->AppendSwitchASCII(
59 switches::kForceFieldTrials,
60 base::StringPrintf(
61 "%s/%s/",
62 chrome_prefs::internals::kSettingsEnforcementTrialName,
63 chrome_prefs::internals::kSettingsEnforcementGroupNoEnforcement));
64 #if defined(OFFICIAL_BUILD) && defined(OS_WIN)
65 // In Windows official builds, it is not possible to disable settings
66 // authentication.
67 unauthenticated_load_allowed_ = false;
68 #endif
69 } else {
70 base::FilePath::StringType paths = JoinString(load_extensions_, ',');
71 command_line->AppendSwitchNative(switches::kLoadExtension,
72 paths);
73 command_line->AppendSwitch(switches::kDisableExtensionsFileAccessCheck);
74 }
75 }
76
SetUpUserDataDirectory()77 virtual bool SetUpUserDataDirectory() OVERRIDE {
78 base::FilePath profile_dir;
79 PathService::Get(chrome::DIR_USER_DATA, &profile_dir);
80 profile_dir = profile_dir.AppendASCII(TestingProfile::kTestUserProfileDir);
81 base::CreateDirectory(profile_dir);
82
83 preferences_file_ = profile_dir.Append(chrome::kPreferencesFilename);
84 user_scripts_dir_ = profile_dir.AppendASCII("User Scripts");
85 extensions_dir_ = profile_dir.AppendASCII("Extensions");
86
87 if (load_extensions_.empty()) {
88 base::FilePath src_dir;
89 PathService::Get(chrome::DIR_TEST_DATA, &src_dir);
90 src_dir = src_dir.AppendASCII("extensions").AppendASCII("good");
91
92 base::CopyFile(src_dir.Append(chrome::kPreferencesFilename),
93 preferences_file_);
94 base::CopyDirectory(src_dir.AppendASCII("Extensions"),
95 profile_dir, true); // recursive
96 }
97 return true;
98 }
99
TearDown()100 virtual void TearDown() {
101 EXPECT_TRUE(base::DeleteFile(preferences_file_, false));
102
103 // TODO(phajdan.jr): Check return values of the functions below, carefully.
104 base::DeleteFile(user_scripts_dir_, true);
105 base::DeleteFile(extensions_dir_, true);
106
107 InProcessBrowserTest::TearDown();
108 }
109
WaitForServicesToStart(int num_expected_extensions,bool expect_extensions_enabled)110 void WaitForServicesToStart(int num_expected_extensions,
111 bool expect_extensions_enabled) {
112 ExtensionService* service = extensions::ExtensionSystem::Get(
113 browser()->profile())->extension_service();
114
115 // Count the number of non-component extensions.
116 int found_extensions = 0;
117 for (extensions::ExtensionSet::const_iterator it =
118 service->extensions()->begin();
119 it != service->extensions()->end(); ++it) {
120 if ((*it)->location() != extensions::Manifest::COMPONENT)
121 found_extensions++;
122 }
123
124 if (!unauthenticated_load_allowed_)
125 num_expected_extensions = 0;
126
127 ASSERT_EQ(static_cast<uint32>(num_expected_extensions),
128 static_cast<uint32>(found_extensions));
129 ASSERT_EQ(expect_extensions_enabled, service->extensions_enabled());
130
131 content::WindowedNotificationObserver user_scripts_observer(
132 chrome::NOTIFICATION_USER_SCRIPTS_UPDATED,
133 content::NotificationService::AllSources());
134 extensions::UserScriptMaster* master =
135 extensions::ExtensionSystem::Get(browser()->profile())->
136 user_script_master();
137 if (!master->ScriptsReady())
138 user_scripts_observer.Wait();
139 ASSERT_TRUE(master->ScriptsReady());
140 }
141
TestInjection(bool expect_css,bool expect_script)142 void TestInjection(bool expect_css, bool expect_script) {
143 if (!unauthenticated_load_allowed_) {
144 expect_css = false;
145 expect_script = false;
146 }
147
148 // Load a page affected by the content script and test to see the effect.
149 base::FilePath test_file;
150 PathService::Get(chrome::DIR_TEST_DATA, &test_file);
151 test_file = test_file.AppendASCII("extensions")
152 .AppendASCII("test_file.html");
153
154 ui_test_utils::NavigateToURL(browser(), net::FilePathToFileURL(test_file));
155
156 bool result = false;
157 ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
158 browser()->tab_strip_model()->GetActiveWebContents(),
159 "window.domAutomationController.send("
160 " document.defaultView.getComputedStyle(document.body, null)."
161 " getPropertyValue('background-color') == 'rgb(245, 245, 220)')",
162 &result));
163 EXPECT_EQ(expect_css, result);
164
165 result = false;
166 ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
167 browser()->tab_strip_model()->GetActiveWebContents(),
168 "window.domAutomationController.send(document.title == 'Modified')",
169 &result));
170 EXPECT_EQ(expect_script, result);
171 }
172
173 base::FilePath preferences_file_;
174 base::FilePath extensions_dir_;
175 base::FilePath user_scripts_dir_;
176 // True unless unauthenticated extension settings are not allowed to be
177 // loaded in this configuration.
178 bool unauthenticated_load_allowed_;
179 // Extensions to load from the command line.
180 std::vector<base::FilePath::StringType> load_extensions_;
181
182 int num_expected_extensions_;
183 };
184
185
186 // ExtensionsStartupTest
187 // Ensures that we can startup the browser with --enable-extensions and some
188 // extensions installed and see them run and do basic things.
189 typedef ExtensionStartupTestBase ExtensionsStartupTest;
190
IN_PROC_BROWSER_TEST_F(ExtensionsStartupTest,Test)191 IN_PROC_BROWSER_TEST_F(ExtensionsStartupTest, Test) {
192 WaitForServicesToStart(num_expected_extensions_, true);
193 TestInjection(true, true);
194 }
195
196 // Sometimes times out on Mac. http://crbug.com/48151
197 #if defined(OS_MACOSX)
198 #define MAYBE_NoFileAccess DISABLED_NoFileAccess
199 #else
200 #define MAYBE_NoFileAccess NoFileAccess
201 #endif
202 // Tests that disallowing file access on an extension prevents it from injecting
203 // script into a page with a file URL.
IN_PROC_BROWSER_TEST_F(ExtensionsStartupTest,MAYBE_NoFileAccess)204 IN_PROC_BROWSER_TEST_F(ExtensionsStartupTest, MAYBE_NoFileAccess) {
205 WaitForServicesToStart(num_expected_extensions_, true);
206
207 // Keep a separate list of extensions for which to disable file access, since
208 // doing so reloads them.
209 std::vector<const extensions::Extension*> extension_list;
210
211 extensions::ExtensionRegistry* registry =
212 extensions::ExtensionRegistry::Get(browser()->profile());
213 for (extensions::ExtensionSet::const_iterator it =
214 registry->enabled_extensions().begin();
215 it != registry->enabled_extensions().end(); ++it) {
216 if ((*it)->location() == extensions::Manifest::COMPONENT)
217 continue;
218 if (extensions::util::AllowFileAccess((*it)->id(), browser()->profile()))
219 extension_list.push_back(it->get());
220 }
221
222 for (size_t i = 0; i < extension_list.size(); ++i) {
223 content::WindowedNotificationObserver user_scripts_observer(
224 chrome::NOTIFICATION_USER_SCRIPTS_UPDATED,
225 content::NotificationService::AllSources());
226 extensions::util::SetAllowFileAccess(
227 extension_list[i]->id(), browser()->profile(), false);
228 user_scripts_observer.Wait();
229 }
230
231 TestInjection(false, false);
232 }
233
234 // ExtensionsLoadTest
235 // Ensures that we can startup the browser with --load-extension and see them
236 // run.
237 class ExtensionsLoadTest : public ExtensionStartupTestBase {
238 public:
ExtensionsLoadTest()239 ExtensionsLoadTest() {
240 base::FilePath one_extension_path;
241 PathService::Get(chrome::DIR_TEST_DATA, &one_extension_path);
242 one_extension_path = one_extension_path
243 .AppendASCII("extensions")
244 .AppendASCII("good")
245 .AppendASCII("Extensions")
246 .AppendASCII("behllobkkfkfnphdnhnkndlbkcpglgmj")
247 .AppendASCII("1.0.0.0");
248 load_extensions_.push_back(one_extension_path.value());
249 }
250 };
251
252 // Fails inconsistently on Linux x64. http://crbug.com/80961
253 // TODO(dpapad): Has not failed since October 2011, let's reenable, monitor
254 // and act accordingly.
IN_PROC_BROWSER_TEST_F(ExtensionsLoadTest,Test)255 IN_PROC_BROWSER_TEST_F(ExtensionsLoadTest, Test) {
256 WaitForServicesToStart(1, true);
257 TestInjection(true, true);
258 }
259
260 // ExtensionsLoadMultipleTest
261 // Ensures that we can startup the browser with multiple extensions
262 // via --load-extension=X1,X2,X3.
263 class ExtensionsLoadMultipleTest : public ExtensionStartupTestBase {
264 public:
ExtensionsLoadMultipleTest()265 ExtensionsLoadMultipleTest() {
266 base::FilePath one_extension_path;
267 PathService::Get(chrome::DIR_TEST_DATA, &one_extension_path);
268 one_extension_path = one_extension_path
269 .AppendASCII("extensions")
270 .AppendASCII("good")
271 .AppendASCII("Extensions")
272 .AppendASCII("behllobkkfkfnphdnhnkndlbkcpglgmj")
273 .AppendASCII("1.0.0.0");
274 load_extensions_.push_back(one_extension_path.value());
275
276 base::FilePath second_extension_path;
277 PathService::Get(chrome::DIR_TEST_DATA, &second_extension_path);
278 second_extension_path = second_extension_path
279 .AppendASCII("extensions")
280 .AppendASCII("app");
281 load_extensions_.push_back(second_extension_path.value());
282
283 base::FilePath third_extension_path;
284 PathService::Get(chrome::DIR_TEST_DATA, &third_extension_path);
285 third_extension_path = third_extension_path
286 .AppendASCII("extensions")
287 .AppendASCII("app1");
288 load_extensions_.push_back(third_extension_path.value());
289
290 base::FilePath fourth_extension_path;
291 PathService::Get(chrome::DIR_TEST_DATA, &fourth_extension_path);
292 fourth_extension_path = fourth_extension_path
293 .AppendASCII("extensions")
294 .AppendASCII("app2");
295 load_extensions_.push_back(fourth_extension_path.value());
296 }
297 };
298
IN_PROC_BROWSER_TEST_F(ExtensionsLoadMultipleTest,Test)299 IN_PROC_BROWSER_TEST_F(ExtensionsLoadMultipleTest, Test) {
300 WaitForServicesToStart(4, true);
301 TestInjection(true, true);
302 }
303