Initial support for partitioning cookies for isolated apps.

chrome/browser/extensions/isolated_app_apitest.cc

Index: chrome/browser/extensions/isolated_app_apitest.cc
diff --git a/chrome/browser/extensions/isolated_app_apitest.cc b/chrome/browser/extensions/isolated_app_apitest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..4bc2252b8d307acbb815823edce144d774c694b9
--- /dev/null
+++ b/chrome/browser/extensions/isolated_app_apitest.cc
@@ -0,0 +1,162 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.

[Matt Perry, 2011/01/26 20:09:23]

One other thing that might be worth testing is an app1 page with an iframe
pointing to non_app. Make sure the iframe's cookies are registered with app1
rather than non_app. Maybe the same with a window.open. (to verify that cookies
are isolated per-renderer rather than per URL).

[Charlie Reis, 2011/03/01 21:33:11]

Good idea-- I've added that.  We plan to keep iframes in their parent's process
for the time being, so it's good to be explicit about it.
Hopefully that won't be true soon-- I'm working on a CL to fix the window.open
case, so that opening a non-app URL in a new tab will end up in a different
process.

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "base/utf_string_conversions.h"
+#include "chrome/browser/browser_thread.h"
+#include "chrome/browser/extensions/extension_apitest.h"
+#include "chrome/browser/extensions/extension_host.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/renderer_host/render_view_host.h"
+#include "chrome/browser/tab_contents/tab_contents.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/net/url_request_context_getter.h"
+#include "chrome/test/ui_test_utils.h"
+#include "net/base/cookie_store.h"
+#include "net/base/mock_host_resolver.h"
+
+namespace {
+
+void GetCookiesOnIOThread(
+    const GURL& url,
+    const scoped_refptr<URLRequestContextGetter>& context_getter,
+    base::WaitableEvent* event,
+    std::string* cookies) {
+  *cookies = context_getter->GetCookieStore()->GetCookies(url);
+  event->Signal();
+}
+
+class IsolatedAppApiTest : public ExtensionApiTest {
+ public:
+  // Gets the cookie string and size for the given URL, using the context
+  // associated with the currently visible page or app in the given
+  // TabContents.
+  void GetCookies(const GURL& url, TabContents* contents, int* value_size,
+                  std::string* value) {
+    *value_size = -1;
+    if (url.is_valid() && contents) {
+      // Since we are on the UI thread don't call GetURLRequestContext().
+      // Be sure to get the request context specific to the current app.
+      const Extension* installed_app =
+          contents->render_view_host()->installed_app();
+      scoped_refptr<URLRequestContextGetter> context_getter =
+          contents->profile()->GetRequestContext(installed_app);
+
+      base::WaitableEvent event(true /* manual reset */,
+                                false /* not initially signaled */);
+      CHECK(BrowserThread::PostTask(
+                BrowserThread::IO, FROM_HERE,
+                NewRunnableFunction(&GetCookiesOnIOThread,
+                                    url, context_getter, &event, value)));
+      event.Wait();
+
+      *value_size = static_cast<int>(value->size());

[Matt Perry, 2011/01/26 20:09:23]

seems like this param is unused

[Charlie Reis, 2011/03/01 21:33:11]

True.  I was keeping it around from the TestingAutomationProvider approach to be
general, but I don't think it's necessary here.

+    }
+  }
+
+  // Returns whether the given tab's current URL has the given cookie.
+  bool WARN_UNUSED_RESULT HasCookie(TabContents* contents, std::string cookie) {
+    int cookie_size;
+    std::string actual_cookie;
+    GetCookies(contents->GetURL(), contents, &cookie_size, &actual_cookie);
+    return actual_cookie.find(cookie) != std::string::npos;
+  }
+};
+
+}  // namespace
+
+// Tests that cookies set within an isolated app are not visible to normal
+// pages or other apps.
+IN_PROC_BROWSER_TEST_F(IsolatedAppApiTest, CookieIsolation) {
+  CommandLine::ForCurrentProcess()->AppendSwitch(
+      switches::kDisablePopupBlocking);
+  CommandLine::ForCurrentProcess()->AppendSwitch(
+      switches::kEnableExperimentalAppManifests);
+
+  host_resolver()->AddRule("*", "127.0.0.1");
+  ASSERT_TRUE(test_server()->Start());
+
+  ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("isolated_apps/app1")));
+  ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("isolated_apps/app2")));
+
+  // The app under test acts on URLs whose host is "localhost",
+  // so the URLs we navigate to must have host "localhost".
+  GURL base_url = test_server()->GetURL(
+      "files/extensions/api_test/isolated_apps/");
+  GURL::Replacements replace_host;
+  std::string host_str("localhost");  // must stay in scope with replace_host
+  replace_host.SetHostStr(host_str);
+  base_url = base_url.ReplaceComponents(replace_host);
+
+  browser()->NewTab();
+  ui_test_utils::NavigateToURL(browser(), base_url.Resolve("app1/main.html"));
+  browser()->NewTab();
+  ui_test_utils::NavigateToURL(browser(), base_url.Resolve("app2/main.html"));
+  browser()->NewTab();
+  ui_test_utils::NavigateToURL(browser(),
+                               base_url.Resolve("non_app/main.html"));
+
+  // Ensure first two tabs have installed apps

[Matt Perry, 2011/01/26 20:09:23]

nit: end with .

[Charlie Reis, 2011/03/01 21:33:11]

Done.

+  TabContents* tab1 = browser()->GetTabContentsAt(1);
+  TabContents* tab2 = browser()->GetTabContentsAt(2);
+  TabContents* tab3 = browser()->GetTabContentsAt(3);
+  ASSERT_TRUE(tab1->render_view_host()->installed_app() != NULL);
+  ASSERT_TRUE(tab2->render_view_host()->installed_app() != NULL);
+  ASSERT_TRUE(tab3->render_view_host()->installed_app() == NULL);
+
+  // Check that each tab sees its own cookie.
+  ASSERT_TRUE(HasCookie(tab1, "app1=3"));
+  ASSERT_TRUE(HasCookie(tab2, "app2=4"));
+  ASSERT_TRUE(HasCookie(tab3, "normalPage=5"));
+
+  // Check that app1 tab cannot see the other cookies.
+  ASSERT_FALSE(HasCookie(tab1, "app2"));
+  ASSERT_FALSE(HasCookie(tab1, "normalPage"));
+
+  // Check that app2 tab cannot see the other cookies.
+  ASSERT_FALSE(HasCookie(tab2, "app1"));
+  ASSERT_FALSE(HasCookie(tab2, "normalPage"));
+
+  // Check that normal tab cannot see the other cookies.
+  ASSERT_FALSE(HasCookie(tab3, "app1"));
+  ASSERT_FALSE(HasCookie(tab3, "app2"));
+}
+
+// Without the --enable-experimental-app-manifests flag, all the tabs
+// should see each others' cookies.
+IN_PROC_BROWSER_TEST_F(IsolatedAppApiTest, CookieIsolationRequiresFlag) {
+  CommandLine::ForCurrentProcess()->AppendSwitch(
+      switches::kDisablePopupBlocking);
+
+  host_resolver()->AddRule("*", "127.0.0.1");
+  ASSERT_TRUE(test_server()->Start());
+
+  ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("isolated_apps/app1")));
+  ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("isolated_apps/app2")));
+
+  // The app under test acts on URLs whose host is "localhost",
+  // so the URLs we navigate to must have host "localhost".
+  GURL base_url = test_server()->GetURL(
+      "files/extensions/api_test/isolated_apps/");
+  GURL::Replacements replace_host;
+  std::string host_str("localhost");  // must stay in scope with replace_host
+  replace_host.SetHostStr(host_str);
+  base_url = base_url.ReplaceComponents(replace_host);
+
+  browser()->NewTab();
+  ui_test_utils::NavigateToURL(browser(), base_url.Resolve("app1/main.html"));
+  browser()->NewTab();
+  ui_test_utils::NavigateToURL(browser(), base_url.Resolve("app2/main.html"));
+  browser()->NewTab();
+  ui_test_utils::NavigateToURL(browser(),
+                               base_url.Resolve("non_app/main.html"));
+
+  // Check that tabs see each others' cookies.
+  ASSERT_TRUE(HasCookie(browser()->GetTabContentsAt(1), "app2=4"));
+  ASSERT_TRUE(HasCookie(browser()->GetTabContentsAt(1), "normalPage=5"));
+  ASSERT_TRUE(HasCookie(browser()->GetTabContentsAt(2), "app1=3"));
+  ASSERT_TRUE(HasCookie(browser()->GetTabContentsAt(2), "normalPage=5"));
+  ASSERT_TRUE(HasCookie(browser()->GetTabContentsAt(3), "app1=3"));
+  ASSERT_TRUE(HasCookie(browser()->GetTabContentsAt(3), "app2=4"));
+}