1ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Use of this source code is governed by a BSD-style license that can be 3c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// found in the LICENSE file. 4c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 5c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/task.h" 63f50c38dc070f4bb515c1b64450dae14f316474eKristian Monsen#include "base/synchronization/waitable_event.h" 721d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen#include "chrome/browser/content_settings/host_content_settings_map.h" 821d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen#include "chrome/browser/profiles/profile.h" 94a5e2dc747d50c653511c68ccb2cfbfb740bd5a7Ben Murdoch#include "chrome/browser/ui/browser.h" 10c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/test/in_process_browser_test.h" 11c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/test/ui_test_utils.h" 123345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#include "net/base/cookie_store.h" 13c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "net/base/mock_host_resolver.h" 143345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#include "net/test/test_server.h" 15ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen#include "net/url_request/url_request_context.cc" 16ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen#include "net/url_request/url_request_context_getter.h" 17c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 18c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochnamespace { 19c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 20c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochclass GetCookiesTask : public Task { 21c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch public: 22c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch GetCookiesTask(const GURL& url, 23ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen net::URLRequestContextGetter* context_getter, 24c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch base::WaitableEvent* event, 25c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string* cookies) 26c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch : url_(url), 27c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch context_getter_(context_getter), 28c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch event_(event), 29c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch cookies_(cookies) {} 30c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 31c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch virtual void Run() { 32ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen *cookies_ = 33ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen context_getter_->GetURLRequestContext()->cookie_store()-> 34ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen GetCookies(url_); 35c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch event_->Signal(); 36c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 37c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 38c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch private: 39c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const GURL& url_; 40ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen net::URLRequestContextGetter* const context_getter_; 41c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch base::WaitableEvent* const event_; 42c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string* const cookies_; 43c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 44c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DISALLOW_COPY_AND_ASSIGN(GetCookiesTask); 45c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch}; 46c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 47c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochclass CookiePolicyBrowserTest : public InProcessBrowserTest { 48c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch protected: 49c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch CookiePolicyBrowserTest() {} 50c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 51c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string GetCookies(const GURL& url) { 52c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string cookies; 53c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch base::WaitableEvent event(true /* manual reset */, 54c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch false /* not initially signaled */); 55ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen net::URLRequestContextGetter* context_getter = 56c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch browser()->profile()->GetRequestContext(); 57c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch EXPECT_TRUE( 58731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick BrowserThread::PostTask( 59731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick BrowserThread::IO, FROM_HERE, 60c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch new GetCookiesTask(url, context_getter, &event, &cookies))); 61c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch EXPECT_TRUE(event.Wait()); 62c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return cookies; 63c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 64c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 65c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch private: 66c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch DISALLOW_COPY_AND_ASSIGN(CookiePolicyBrowserTest); 67c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch}; 68c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 69c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Visits a page that sets a first-party cookie. 70c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochIN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, AllowFirstPartyCookies) { 713345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick ASSERT_TRUE(test_server()->Start()); 72c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 73c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch browser()->profile()->GetHostContentSettingsMap()-> 74c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch SetBlockThirdPartyCookies(true); 75c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 763345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick GURL url(test_server()->GetURL("set-cookie?cookie1")); 77c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 78c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string cookie = GetCookies(url); 79c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ASSERT_EQ("", cookie); 80c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 81c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ui_test_utils::NavigateToURL(browser(), url); 82c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 83c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch cookie = GetCookies(url); 84c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch EXPECT_EQ("cookie1", cookie); 85c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 86c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 87c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Visits a page that is a redirect across domain boundary to a page that sets 88c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// a first-party cookie. 89c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochIN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, 90c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch AllowFirstPartyCookiesRedirect) { 913345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick ASSERT_TRUE(test_server()->Start()); 92c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 93c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch browser()->profile()->GetHostContentSettingsMap()-> 94c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch SetBlockThirdPartyCookies(true); 95c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 963345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick GURL url(test_server()->GetURL("server-redirect?")); 973345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick GURL redirected_url(test_server()->GetURL("set-cookie?cookie2")); 98c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 993345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick // Change the host name from 127.0.0.1 to www.example.com so it triggers 100c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // third-party cookie blocking if the first party for cookies URL is not 101c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // changed when we follow a redirect. 1023345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick ASSERT_EQ("127.0.0.1", redirected_url.host()); 103c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch GURL::Replacements replacements; 104c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string new_host("www.example.com"); 105c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch replacements.SetHostStr(new_host); 106c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch redirected_url = redirected_url.ReplaceComponents(replacements); 107c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 108c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string cookie = GetCookies(redirected_url); 109c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ASSERT_EQ("", cookie); 110c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 111c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch host_resolver()->AddRule("www.example.com", "127.0.0.1"); 112c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 113c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ui_test_utils::NavigateToURL(browser(), 114c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch GURL(url.spec() + redirected_url.spec())); 115c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 116c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch cookie = GetCookies(redirected_url); 117c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch EXPECT_EQ("cookie2", cookie); 118c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 119c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 120c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} // namespace 121