Browsertests for third-party cookie blocking in double-nested frames.

This patch tests [1], which I'd like to keep small for ease of merging.

[1]: https://chromium-review.googlesource.com/c/chromium/src/+/1212846

Bug: 881715
Change-Id: Ib046afb94be0995cd0dc3ac2bda2df90ddbc16fa
Reviewed-on: https://chromium-review.googlesource.com/1213086
Commit-Queue: Mike West <[email protected]>
Reviewed-by: Josh Karlin <[email protected]>
Reviewed-by: Christian Dullweber <[email protected]>
Reviewed-by: Martin Šrámek <[email protected]>
Cr-Commit-Position: refs/heads/master@{#589573}
diff --git a/chrome/browser/net/cookie_policy_browsertest.cc b/chrome/browser/net/cookie_policy_browsertest.cc
index 61f257a..1d2cf31 100644
--- a/chrome/browser/net/cookie_policy_browsertest.cc
+++ b/chrome/browser/net/cookie_policy_browsertest.cc
@@ -15,6 +15,7 @@
 #include "components/content_settings/core/common/pref_names.h"
 #include "components/prefs/pref_service.h"
 #include "content/public/test/browser_test_utils.h"
+#include "content/public/test/test_navigation_observer.h"
 #include "net/dns/mock_host_resolver.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 
@@ -28,6 +29,72 @@
 
   void SetUpOnMainThread() override {
     host_resolver()->AddRule("*", "127.0.0.1");
+    ASSERT_TRUE(embedded_test_server()->Start());
+  }
+
+  void SetBlockThirdPartyCookies(bool value) {
+    browser()->profile()->GetPrefs()->SetBoolean(prefs::kBlockThirdPartyCookies,
+                                                 value);
+  }
+
+  void NavigateToPageWithFrame(const std::string& host) {
+    GURL main_url(embedded_test_server()->GetURL(host, "/iframe.html"));
+    ui_test_utils::NavigateToURL(browser(), main_url);
+  }
+
+  void NavigateFrameTo(const std::string& host, const std::string& path) {
+    GURL page = embedded_test_server()->GetURL(host, path);
+    content::WebContents* web_contents =
+        browser()->tab_strip_model()->GetActiveWebContents();
+    EXPECT_TRUE(NavigateIframeToURL(web_contents, "test", page));
+  }
+
+  void ExpectFrameContent(const std::string& expected) {
+    content::WebContents* web_contents =
+        browser()->tab_strip_model()->GetActiveWebContents();
+    content::RenderFrameHost* nested =
+        ChildFrameAt(web_contents->GetMainFrame(), 0);
+    std::string content;
+    ASSERT_TRUE(ExecuteScriptAndExtractString(
+        nested,
+        "window.domAutomationController.send(document.body.textContent)",
+        &content));
+    EXPECT_EQ(expected, content);
+  }
+
+  void NavigateNestedFrameTo(const std::string& host, const std::string& path) {
+    GURL url(embedded_test_server()->GetURL(host, path));
+    content::WebContents* web_contents =
+        browser()->tab_strip_model()->GetActiveWebContents();
+    content::RenderFrameHost* nested =
+        ChildFrameAt(web_contents->GetMainFrame(), 0);
+    content::TestNavigationObserver load_observer(web_contents);
+    ASSERT_TRUE(ExecuteScript(
+        nested,
+        base::StringPrintf("document.body.querySelector('iframe').src = '%s';",
+                           url.spec().c_str())));
+    load_observer.Wait();
+  }
+
+  void ExpectNestedFrameContent(const std::string& expected) {
+    content::WebContents* web_contents =
+        browser()->tab_strip_model()->GetActiveWebContents();
+    content::RenderFrameHost* nested =
+        ChildFrameAt(web_contents->GetMainFrame(), 0);
+    content::RenderFrameHost* double_nested = ChildFrameAt(nested, 0);
+    std::string content;
+    ASSERT_TRUE(ExecuteScriptAndExtractString(
+        double_nested,
+        "window.domAutomationController.send(document.body.textContent)",
+        &content));
+    EXPECT_EQ(expected, content);
+  }
+
+  void ExpectCookiesOnHost(const std::string& host,
+                           const std::string& expected) {
+    EXPECT_EQ(expected,
+              content::GetCookies(browser()->profile(),
+                                  embedded_test_server()->GetURL(host, "/")));
   }
 
  private:
@@ -36,10 +103,7 @@
 
 // Visits a page that sets a first-party cookie.
 IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest, AllowFirstPartyCookies) {
-  ASSERT_TRUE(embedded_test_server()->Start());
-
-  browser()->profile()->GetPrefs()->SetBoolean(prefs::kBlockThirdPartyCookies,
-                                               true);
+  SetBlockThirdPartyCookies(false);
 
   GURL url(embedded_test_server()->GetURL("/set-cookie?cookie1"));
 
@@ -56,10 +120,7 @@
 // a first-party cookie.
 IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest,
                        AllowFirstPartyCookiesRedirect) {
-  ASSERT_TRUE(embedded_test_server()->Start());
-
-  browser()->profile()->GetPrefs()->SetBoolean(prefs::kBlockThirdPartyCookies,
-                                               true);
+  SetBlockThirdPartyCookies(true);
 
   GURL url(embedded_test_server()->GetURL("/server-redirect?"));
   GURL redirected_url(embedded_test_server()->GetURL("/set-cookie?cookie2"));
@@ -83,4 +144,123 @@
   EXPECT_EQ("cookie2", cookie);
 }
 
+// Third-Party Frame Tests
+IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest,
+                       ThirdPartyCookiesIFrameAllowSetting) {
+  SetBlockThirdPartyCookies(false);
+
+  NavigateToPageWithFrame("a.com");
+
+  ExpectCookiesOnHost("b.com", "");
+
+  // Navigate iframe to a cross-site, cookie-setting endpoint, and verify that
+  // the cookie is set:
+  NavigateFrameTo("b.com", "/set-cookie?thirdparty");
+  ExpectCookiesOnHost("b.com", "thirdparty");
+
+  // Navigate iframe to a cross-site frame with a frame, and navigate _that_
+  // frame to a cross-site, cookie-setting endpoint, and verify that the cookie
+  // is set:
+  NavigateFrameTo("b.com", "/iframe.html");
+  NavigateNestedFrameTo("b.com", "/set-cookie?thirdparty");
+  ExpectCookiesOnHost("b.com", "thirdparty");
+
+  // Navigate iframe to a cross-site frame with a frame, and navigate _that_
+  // frame to a cross-site, cookie-setting endpoint, and verify that the cookie
+  // is set:
+  NavigateFrameTo("c.com", "/iframe.html");
+  NavigateNestedFrameTo("b.com", "/set-cookie?thirdparty");
+  ExpectCookiesOnHost("b.com", "thirdparty");
+}
+
+IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest,
+                       ThirdPartyCookiesIFrameBlockSetting) {
+  SetBlockThirdPartyCookies(true);
+
+  NavigateToPageWithFrame("a.com");
+
+  // Navigate iframe to a cross-site, cookie-setting endpoint, and verify that
+  // the cookie is not set:
+  NavigateFrameTo("b.com", "/set-cookie?thirdparty");
+  ExpectCookiesOnHost("b.com", "");
+
+  // Navigate iframe to a cross-site frame with a frame, and navigate _that_
+  // frame to a cross-site, cookie-setting endpoint, and verify that the cookie
+  // is not set:
+  NavigateFrameTo("b.com", "/iframe.html");
+  NavigateNestedFrameTo("b.com", "/set-cookie?thirdparty");
+  ExpectCookiesOnHost("b.com", "");
+
+  // Navigate iframe to a cross-site frame with a frame, and navigate _that_
+  // frame to a cross-site, cookie-setting endpoint, and verify that the cookie
+  // is not set:
+  NavigateFrameTo("c.com", "/iframe.html");
+  NavigateNestedFrameTo("b.com", "/set-cookie?thirdparty");
+  ExpectCookiesOnHost("b.com", "");
+}
+
+IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest,
+                       ThirdPartyCookiesIFrameAllowReading) {
+  SetBlockThirdPartyCookies(false);
+
+  // Set a cookie on `b.com`.
+  content::SetCookie(browser()->profile(),
+                     embedded_test_server()->GetURL("b.com", "/"),
+                     "thirdparty");
+  ExpectCookiesOnHost("b.com", "thirdparty");
+
+  NavigateToPageWithFrame("a.com");
+
+  // Navigate iframe to a cross-site, cookie-reading endpoint, and verify that
+  // the cookie is sent:
+  NavigateFrameTo("b.com", "/echoheader?cookie");
+  ExpectFrameContent("thirdparty");
+
+  // Navigate iframe to a cross-site frame with a frame, and navigate _that_
+  // frame to a cross-site page that echos the cookie header, and verify that
+  // the cookie is sent:
+  NavigateFrameTo("b.com", "/iframe.html");
+  NavigateNestedFrameTo("b.com", "/echoheader?cookie");
+  ExpectNestedFrameContent("thirdparty");
+
+  // Navigate iframe to a cross-site frame with a frame, and navigate _that_
+  // frame to a distinct cross-site page that echos the cookie header, and
+  // verify that the cookie is not sent:
+  NavigateFrameTo("c.com", "/iframe.html");
+  NavigateNestedFrameTo("b.com", "/echoheader?cookie");
+  ExpectNestedFrameContent("thirdparty");
+}
+
+IN_PROC_BROWSER_TEST_F(CookiePolicyBrowserTest,
+                       ThirdPartyCookiesIFrameBlockReading) {
+  SetBlockThirdPartyCookies(true);
+
+  // Set a cookie on `b.com`.
+  content::SetCookie(browser()->profile(),
+                     embedded_test_server()->GetURL("b.com", "/"),
+                     "thirdparty");
+  ExpectCookiesOnHost("b.com", "thirdparty");
+
+  NavigateToPageWithFrame("a.com");
+
+  // Navigate iframe to a cross-site, cookie-reading endpoint, and verify that
+  // the cookie is not sent:
+  NavigateFrameTo("b.com", "/echoheader?cookie");
+  ExpectFrameContent("None");
+
+  // Navigate iframe to a cross-site frame with a frame, and navigate _that_
+  // frame to a cross-site page that echos the cookie header, and verify that
+  // the cookie is not sent:
+  NavigateFrameTo("b.com", "/iframe.html");
+  NavigateNestedFrameTo("b.com", "/echoheader?cookie");
+  ExpectNestedFrameContent("None");
+
+  // Navigate iframe to a cross-site frame with a frame, and navigate _that_
+  // frame to a distinct cross-site page that echos the cookie header, and
+  // verify that the cookie is not sent:
+  NavigateFrameTo("c.com", "/iframe.html");
+  NavigateNestedFrameTo("b.com", "/echoheader?cookie");
+  ExpectNestedFrameContent("None");
+}
+
 }  // namespace