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