OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/command_line.h" | 5 #include "base/command_line.h" |
6 #include "base/files/file_path.h" | 6 #include "base/files/file_path.h" |
7 #include "base/message_loop/message_loop.h" | 7 #include "base/message_loop/message_loop.h" |
8 #include "base/path_service.h" | 8 #include "base/path_service.h" |
9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
10 #include "base/strings/utf_string_conversions.h" | 10 #include "base/strings/utf_string_conversions.h" |
(...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
106 int GetBlockedContentsCount() { | 106 int GetBlockedContentsCount() { |
107 // Do a round trip to the renderer first to flush any in-flight IPCs to | 107 // Do a round trip to the renderer first to flush any in-flight IPCs to |
108 // create a to-be-blocked window. | 108 // create a to-be-blocked window. |
109 WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents(); | 109 WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents(); |
110 CHECK(content::ExecuteScript(tab, std::string())); | 110 CHECK(content::ExecuteScript(tab, std::string())); |
111 PopupBlockerTabHelper* popup_blocker_helper = | 111 PopupBlockerTabHelper* popup_blocker_helper = |
112 PopupBlockerTabHelper::FromWebContents(tab); | 112 PopupBlockerTabHelper::FromWebContents(tab); |
113 return popup_blocker_helper->GetBlockedPopupsCount(); | 113 return popup_blocker_helper->GetBlockedPopupsCount(); |
114 } | 114 } |
115 | 115 |
116 void NavigateAndCheckPopupShown(const GURL& url) { | |
117 content::WindowedNotificationObserver observer( | |
118 chrome::NOTIFICATION_TAB_ADDED, | |
119 content::NotificationService::AllSources()); | |
120 ui_test_utils::NavigateToURL(browser(), url); | |
121 observer.Wait(); | |
122 | |
123 ASSERT_EQ(2u, chrome::GetBrowserCount(browser()->profile(), | |
124 browser()->host_desktop_type())); | |
125 | |
126 ASSERT_EQ(0, GetBlockedContentsCount()); | |
127 } | |
128 | |
129 enum WhatToExpect { | 116 enum WhatToExpect { |
130 ExpectPopup, | 117 ExpectPopup, |
131 ExpectTab | 118 ExpectTab |
132 }; | 119 }; |
133 | 120 |
134 enum ShouldCheckTitle { | 121 enum ShouldCheckTitle { |
135 CheckTitle, | 122 CheckTitle, |
136 DontCheckTitle | 123 DontCheckTitle |
137 }; | 124 }; |
138 | 125 |
| 126 void NavigateAndCheckPopupShown(const GURL& url, |
| 127 WhatToExpect what_to_expect) { |
| 128 content::WindowedNotificationObserver observer( |
| 129 chrome::NOTIFICATION_TAB_ADDED, |
| 130 content::NotificationService::AllSources()); |
| 131 ui_test_utils::NavigateToURL(browser(), url); |
| 132 observer.Wait(); |
| 133 |
| 134 if (what_to_expect == ExpectPopup) { |
| 135 ASSERT_EQ(2u, |
| 136 chrome::GetBrowserCount(browser()->profile(), |
| 137 browser()->host_desktop_type())); |
| 138 } else { |
| 139 ASSERT_EQ(1u, |
| 140 chrome::GetBrowserCount(browser()->profile(), |
| 141 browser()->host_desktop_type())); |
| 142 ASSERT_EQ(2, browser()->tab_strip_model()->count()); |
| 143 } |
| 144 |
| 145 ASSERT_EQ(0, GetBlockedContentsCount()); |
| 146 } |
| 147 |
139 // Navigates to the test indicated by |test_name| using |browser| which is | 148 // Navigates to the test indicated by |test_name| using |browser| which is |
140 // expected to try to open a popup. Verifies that the popup was blocked and | 149 // expected to try to open a popup. Verifies that the popup was blocked and |
141 // then opens the blocked popup. Once the popup stopped loading, verifies | 150 // then opens the blocked popup. Once the popup stopped loading, verifies |
142 // that the title of the page is "PASS" if |check_title| is set. | 151 // that the title of the page is "PASS" if |check_title| is set. |
143 // | 152 // |
144 // If |what_to_expect| is ExpectPopup, the popup is expected to open a new | 153 // If |what_to_expect| is ExpectPopup, the popup is expected to open a new |
145 // window, or a background tab if it is false. | 154 // window, or a background tab if it is false. |
146 // | 155 // |
147 // Returns the WebContents of the launched popup. | 156 // Returns the WebContents of the launched popup. |
148 WebContents* RunCheckTest(Browser* browser, | 157 WebContents* RunCheckTest(Browser* browser, |
(...skipping 134 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
283 AllowPopupThroughContentSetting) { | 292 AllowPopupThroughContentSetting) { |
284 GURL url(embedded_test_server()->GetURL( | 293 GURL url(embedded_test_server()->GetURL( |
285 "/popup_blocker/popup-blocked-to-post-blank.html")); | 294 "/popup_blocker/popup-blocked-to-post-blank.html")); |
286 browser()->profile()->GetHostContentSettingsMap() | 295 browser()->profile()->GetHostContentSettingsMap() |
287 ->SetContentSetting(ContentSettingsPattern::FromURL(url), | 296 ->SetContentSetting(ContentSettingsPattern::FromURL(url), |
288 ContentSettingsPattern::Wildcard(), | 297 ContentSettingsPattern::Wildcard(), |
289 CONTENT_SETTINGS_TYPE_POPUPS, | 298 CONTENT_SETTINGS_TYPE_POPUPS, |
290 std::string(), | 299 std::string(), |
291 CONTENT_SETTING_ALLOW); | 300 CONTENT_SETTING_ALLOW); |
292 | 301 |
293 NavigateAndCheckPopupShown(url); | 302 NavigateAndCheckPopupShown(url, ExpectTab); |
294 } | 303 } |
295 | 304 |
296 // Verify that content settings are applied based on the top-level frame URL. | 305 // Verify that content settings are applied based on the top-level frame URL. |
297 IN_PROC_BROWSER_TEST_F(PopupBlockerBrowserTest, | 306 IN_PROC_BROWSER_TEST_F(PopupBlockerBrowserTest, |
298 AllowPopupThroughContentSettingIFrame) { | 307 AllowPopupThroughContentSettingIFrame) { |
299 GURL url(embedded_test_server()->GetURL("/popup_blocker/popup-frames.html")); | 308 GURL url(embedded_test_server()->GetURL("/popup_blocker/popup-frames.html")); |
300 browser()->profile()->GetHostContentSettingsMap() | 309 browser()->profile()->GetHostContentSettingsMap() |
301 ->SetContentSetting(ContentSettingsPattern::FromURL(url), | 310 ->SetContentSetting(ContentSettingsPattern::FromURL(url), |
302 ContentSettingsPattern::Wildcard(), | 311 ContentSettingsPattern::Wildcard(), |
303 CONTENT_SETTINGS_TYPE_POPUPS, | 312 CONTENT_SETTINGS_TYPE_POPUPS, |
304 std::string(), | 313 std::string(), |
305 CONTENT_SETTING_ALLOW); | 314 CONTENT_SETTING_ALLOW); |
306 | 315 |
307 // Popup from the iframe should be allowed since the top-level URL is | 316 // Popup from the iframe should be allowed since the top-level URL is |
308 // whitelisted. | 317 // whitelisted. |
309 NavigateAndCheckPopupShown(url); | 318 NavigateAndCheckPopupShown(url, ExpectTab); |
310 | 319 |
311 // Whitelist iframe URL instead. | 320 // Whitelist iframe URL instead. |
312 GURL::Replacements replace_host; | 321 GURL::Replacements replace_host; |
313 std::string host_str("www.a.com"); // Must stay in scope with replace_host | 322 std::string host_str("www.a.com"); // Must stay in scope with replace_host |
314 replace_host.SetHostStr(host_str); | 323 replace_host.SetHostStr(host_str); |
315 GURL frame_url(embedded_test_server() | 324 GURL frame_url(embedded_test_server() |
316 ->GetURL("/popup_blocker/popup-frames-iframe.html") | 325 ->GetURL("/popup_blocker/popup-frames-iframe.html") |
317 .ReplaceComponents(replace_host)); | 326 .ReplaceComponents(replace_host)); |
318 browser()->profile()->GetHostContentSettingsMap()->ClearSettingsForOneType( | 327 browser()->profile()->GetHostContentSettingsMap()->ClearSettingsForOneType( |
319 CONTENT_SETTINGS_TYPE_POPUPS); | 328 CONTENT_SETTINGS_TYPE_POPUPS); |
(...skipping 10 matching lines...) Expand all Loading... |
330 } | 339 } |
331 | 340 |
332 IN_PROC_BROWSER_TEST_F(PopupBlockerBrowserTest, | 341 IN_PROC_BROWSER_TEST_F(PopupBlockerBrowserTest, |
333 PopupsLaunchWhenTabIsClosed) { | 342 PopupsLaunchWhenTabIsClosed) { |
334 CommandLine::ForCurrentProcess()->AppendSwitch( | 343 CommandLine::ForCurrentProcess()->AppendSwitch( |
335 switches::kDisablePopupBlocking); | 344 switches::kDisablePopupBlocking); |
336 GURL url( | 345 GURL url( |
337 embedded_test_server()->GetURL("/popup_blocker/popup-on-unload.html")); | 346 embedded_test_server()->GetURL("/popup_blocker/popup-on-unload.html")); |
338 ui_test_utils::NavigateToURL(browser(), url); | 347 ui_test_utils::NavigateToURL(browser(), url); |
339 | 348 |
340 NavigateAndCheckPopupShown(embedded_test_server()->GetURL("/popup_blocker/")); | 349 NavigateAndCheckPopupShown(embedded_test_server()->GetURL("/popup_blocker/"), |
| 350 ExpectPopup); |
341 } | 351 } |
342 | 352 |
343 // Verify that when you unblock popup, the popup shows in history and omnibox. | 353 // Verify that when you unblock popup, the popup shows in history and omnibox. |
344 IN_PROC_BROWSER_TEST_F(PopupBlockerBrowserTest, | 354 IN_PROC_BROWSER_TEST_F(PopupBlockerBrowserTest, |
345 UnblockedPopupShowsInHistoryAndOmnibox) { | 355 UnblockedPopupShowsInHistoryAndOmnibox) { |
346 CommandLine::ForCurrentProcess()->AppendSwitch( | 356 CommandLine::ForCurrentProcess()->AppendSwitch( |
347 switches::kDisablePopupBlocking); | 357 switches::kDisablePopupBlocking); |
348 GURL url(embedded_test_server()->GetURL( | 358 GURL url(embedded_test_server()->GetURL( |
349 "/popup_blocker/popup-blocked-to-post-blank.html")); | 359 "/popup_blocker/popup-blocked-to-post-blank.html")); |
350 NavigateAndCheckPopupShown(url); | 360 NavigateAndCheckPopupShown(url, ExpectTab); |
351 | 361 |
352 std::string search_string = | 362 std::string search_string = |
353 "data:text/html,<title>Popup Success!</title>you should not see this " | 363 "data:text/html,<title>Popup Success!</title>you should not see this " |
354 "message if popup blocker is enabled"; | 364 "message if popup blocker is enabled"; |
355 | 365 |
356 ui_test_utils::HistoryEnumerator history(browser()->profile()); | 366 ui_test_utils::HistoryEnumerator history(browser()->profile()); |
357 std::vector<GURL>& history_urls = history.urls(); | 367 std::vector<GURL>& history_urls = history.urls(); |
358 ASSERT_EQ(2u, history_urls.size()); | 368 ASSERT_EQ(2u, history_urls.size()); |
359 ASSERT_EQ(GURL(search_string), history_urls[0]); | 369 ASSERT_EQ(GURL(search_string), history_urls[0]); |
360 ASSERT_EQ(url, history_urls[1]); | 370 ASSERT_EQ(url, history_urls[1]); |
(...skipping 110 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
471 | 481 |
472 // Verify that the renderer can't DOS the browser by creating arbitrarily many | 482 // Verify that the renderer can't DOS the browser by creating arbitrarily many |
473 // popups. | 483 // popups. |
474 IN_PROC_BROWSER_TEST_F(PopupBlockerBrowserTest, DenialOfService) { | 484 IN_PROC_BROWSER_TEST_F(PopupBlockerBrowserTest, DenialOfService) { |
475 GURL url(embedded_test_server()->GetURL("/popup_blocker/popup-dos.html")); | 485 GURL url(embedded_test_server()->GetURL("/popup_blocker/popup-dos.html")); |
476 ui_test_utils::NavigateToURL(browser(), url); | 486 ui_test_utils::NavigateToURL(browser(), url); |
477 ASSERT_EQ(25, GetBlockedContentsCount()); | 487 ASSERT_EQ(25, GetBlockedContentsCount()); |
478 } | 488 } |
479 | 489 |
480 } // namespace | 490 } // namespace |
OLD | NEW |