OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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/task.h" | |
6 #include "base/waitable_event.h" | |
7 #include "chrome/browser/browser.h" | 5 #include "chrome/browser/browser.h" |
8 #include "chrome/browser/host_content_settings_map.h" | 6 #include "chrome/browser/host_content_settings_map.h" |
9 #include "chrome/browser/profile.h" | 7 #include "chrome/browser/profile.h" |
10 #include "chrome/common/net/url_request_context_getter.h" | 8 #include "chrome/common/net/url_request_context_getter.h" |
11 #include "chrome/test/in_process_browser_test.h" | 9 #include "chrome/test/in_process_browser_test.h" |
12 #include "chrome/test/ui_test_utils.h" | 10 #include "chrome/test/ui_test_utils.h" |
13 #include "net/base/mock_host_resolver.h" | 11 #include "net/base/mock_host_resolver.h" |
14 | 12 |
15 namespace { | 13 class CookiePolicyBrowserTest : public InProcessBrowserTest { |
16 | |
17 class GetCookiesTask : public Task { | |
18 public: | 14 public: |
19 GetCookiesTask(const GURL& url, | |
20 URLRequestContextGetter* context_getter, | |
21 base::WaitableEvent* event, | |
22 std::string* cookies) | |
23 : url_(url), | |
24 context_getter_(context_getter), | |
25 event_(event), | |
26 cookies_(cookies) {} | |
27 | |
28 virtual void Run() { | |
29 *cookies_ = context_getter_->GetCookieStore()->GetCookies(url_); | |
30 event_->Signal(); | |
31 } | |
32 | |
33 private: | |
34 const GURL& url_; | |
35 URLRequestContextGetter* const context_getter_; | |
36 base::WaitableEvent* const event_; | |
37 std::string* const cookies_; | |
38 | |
39 DISALLOW_COPY_AND_ASSIGN(GetCookiesTask); | |
40 }; | |
41 | |
42 class CookiePolicyBrowserTest : public InProcessBrowserTest { | |
43 protected: | |
44 CookiePolicyBrowserTest() {} | 15 CookiePolicyBrowserTest() {} |
45 | 16 |
46 std::string GetCookies(const GURL& url) { | |
47 std::string cookies; | |
48 base::WaitableEvent event(true /* manual reset */, | |
49 false /* not initially signaled */); | |
50 URLRequestContextGetter* context_getter = | |
51 browser()->profile()->GetRequestContext(); | |
52 EXPECT_TRUE( | |
53 ChromeThread::PostTask( | |
54 ChromeThread::IO, FROM_HERE, | |
55 new GetCookiesTask(url, context_getter, &event, &cookies))); | |
56 EXPECT_TRUE(event.Wait()); | |
57 return cookies; | |
58 } | |
59 | |
60 private: | 17 private: |
61 DISALLOW_COPY_AND_ASSIGN(CookiePolicyBrowserTest); | 18 DISALLOW_COPY_AND_ASSIGN(CookiePolicyBrowserTest); |
62 }; | 19 }; |
63 | 20 |
64 // Visits a page that sets a first-party cookie. | 21 // Visits a page that sets a first-party cookie. |
65 IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, AllowFirstPartyCookies) { | 22 IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, AllowFirstPartyCookies) { |
66 HTTPTestServer* server = StartHTTPServer(); | 23 HTTPTestServer* server = StartHTTPServer(); |
67 ASSERT_TRUE(server != NULL); | 24 ASSERT_TRUE(server != NULL); |
68 | 25 |
69 browser()->profile()->GetHostContentSettingsMap()-> | 26 browser()->profile()->GetHostContentSettingsMap()-> |
70 SetBlockThirdPartyCookies(true); | 27 SetBlockThirdPartyCookies(true); |
71 | 28 |
| 29 net::CookieStore* cookie_store = |
| 30 browser()->profile()->GetRequestContext()->GetCookieStore(); |
| 31 |
72 GURL url = server->TestServerPage("set-cookie?cookie1"); | 32 GURL url = server->TestServerPage("set-cookie?cookie1"); |
73 | 33 |
74 std::string cookie = GetCookies(url); | 34 std::string cookie = cookie_store->GetCookies(url); |
75 ASSERT_EQ("", cookie); | 35 ASSERT_EQ("", cookie); |
76 | 36 |
77 ui_test_utils::NavigateToURL(browser(), url); | 37 ui_test_utils::NavigateToURL(browser(), url); |
78 | 38 |
79 cookie = GetCookies(url); | 39 cookie = cookie_store->GetCookies(url); |
80 EXPECT_EQ("cookie1", cookie); | 40 EXPECT_EQ("cookie1", cookie); |
81 } | 41 } |
82 | 42 |
| 43 |
83 // Visits a page that is a redirect across domain boundary to a page that sets | 44 // Visits a page that is a redirect across domain boundary to a page that sets |
84 // a first-party cookie. | 45 // a first-party cookie. |
85 IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, | 46 IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, |
86 AllowFirstPartyCookiesRedirect) { | 47 AllowFirstPartyCookiesRedirect) { |
87 HTTPTestServer* server = StartHTTPServer(); | 48 HTTPTestServer* server = StartHTTPServer(); |
88 ASSERT_TRUE(server != NULL); | 49 ASSERT_TRUE(server != NULL); |
89 | 50 |
90 browser()->profile()->GetHostContentSettingsMap()-> | 51 browser()->profile()->GetHostContentSettingsMap()-> |
91 SetBlockThirdPartyCookies(true); | 52 SetBlockThirdPartyCookies(true); |
92 | 53 |
| 54 net::CookieStore* cookie_store = |
| 55 browser()->profile()->GetRequestContext()->GetCookieStore(); |
| 56 |
93 GURL url = server->TestServerPage("server-redirect?"); | 57 GURL url = server->TestServerPage("server-redirect?"); |
94 | 58 |
95 GURL redirected_url = server->TestServerPage("set-cookie?cookie2"); | 59 GURL redirected_url = server->TestServerPage("set-cookie?cookie2"); |
96 // Change the host name from localhost to www.example.com so it triggers | 60 // Change the host name from localhost to www.example.com so it triggers |
97 // third-party cookie blocking if the first party for cookies URL is not | 61 // third-party cookie blocking if the first party for cookies URL is not |
98 // changed when we follow a redirect. | 62 // changed when we follow a redirect. |
99 ASSERT_EQ("localhost", redirected_url.host()); | 63 ASSERT_EQ("localhost", redirected_url.host()); |
100 GURL::Replacements replacements; | 64 GURL::Replacements replacements; |
101 std::string new_host("www.example.com"); | 65 std::string new_host("www.example.com"); |
102 replacements.SetHostStr(new_host); | 66 replacements.SetHostStr(new_host); |
103 redirected_url = redirected_url.ReplaceComponents(replacements); | 67 redirected_url = redirected_url.ReplaceComponents(replacements); |
104 | 68 |
105 std::string cookie = GetCookies(redirected_url); | 69 std::string cookie = cookie_store->GetCookies(redirected_url); |
106 ASSERT_EQ("", cookie); | 70 ASSERT_EQ("", cookie); |
107 | 71 |
108 host_resolver()->AddRule("www.example.com", "127.0.0.1"); | 72 host_resolver()->AddRule("www.example.com", "127.0.0.1"); |
109 | 73 |
110 ui_test_utils::NavigateToURL(browser(), | 74 ui_test_utils::NavigateToURL(browser(), |
111 GURL(url.spec() + redirected_url.spec())); | 75 GURL(url.spec() + redirected_url.spec())); |
112 | 76 |
113 cookie = GetCookies(redirected_url); | 77 cookie = cookie_store->GetCookies(redirected_url); |
114 EXPECT_EQ("cookie2", cookie); | 78 EXPECT_EQ("cookie2", cookie); |
115 } | 79 } |
116 | |
117 } // namespace | |
OLD | NEW |