Protect Chrome WebStore based on process IDs

chrome/browser/extensions/api/web_request/web_request_permissions_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/extensions/api/web_request/web_request_permissions.h"
 
 #include "base/message_loop.h"
+#include "chrome/browser/extensions/extension_info_map.h"
+#include "chrome/common/extensions/extension_constants.h"
 #include "chrome/test/base/testing_profile.h"
+#include "content/public/browser/resource_request_info.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
+using content::ResourceRequestInfo;
+
 TEST(ExtensionWebRequestHelpersTest, TestHideRequestForURL) {
   MessageLoopForIO message_loop;
   TestURLRequestContext context;
+  scoped_refptr<ExtensionInfoMap> extension_info_map(new ExtensionInfoMap);
   const char* sensitive_urls[] = {
-      "http://www.google.com/chrome",
-      "https://www.google.com/chrome",
-      "http://www.google.com/chrome/foobar",
-      "https://www.google.com/chrome/foobar",
-      "http://chrome.google.com",
-      "https://chrome.google.com",
       "http://client2.google.com",
       "https://client2.google.com",
-      // No http version of webstore.
-      "https://chrome.google.com/webstore",
       "http://clients2.google.com/service/update2/crx",
       "https://clients2.google.com/service/update2/crx",
       "http://www.gstatic.com/chrome/extensions/blacklist",
       "https://www.gstatic.com/chrome/extensions/blacklist",
       "notregisteredscheme://www.foobar.com"
   };
   const char* non_sensitive_urls[] = {
       "http://www.google.com/"
   };
   // Check that requests are rejected based on the destination
   for (size_t i = 0; i < arraysize(sensitive_urls); ++i) {
     GURL sensitive_url(sensitive_urls[i]);
     TestURLRequest request(sensitive_url, NULL, &context);
-    EXPECT_TRUE(WebRequestPermissions::HideRequest(&request))
+    EXPECT_TRUE(
+        WebRequestPermissions::HideRequest(extension_info_map.get(), &request))
         << sensitive_urls[i];
   }
   // Check that requests are accepted if they don't touch sensitive urls.
   for (size_t i = 0; i < arraysize(non_sensitive_urls); ++i) {
     GURL non_sensitive_url(non_sensitive_urls[i]);
     TestURLRequest request(non_sensitive_url, NULL, &context);
-    EXPECT_FALSE(WebRequestPermissions::HideRequest(&request))
+    EXPECT_FALSE(
+        WebRequestPermissions::HideRequest(extension_info_map.get(), &request))
         << non_sensitive_urls[i];
   }
-  // Check that requests are rejected if their first party url is sensitive.
-  ASSERT_GE(arraysize(non_sensitive_urls), 1u);
-  GURL non_sensitive_url(non_sensitive_urls[0]);
-  for (size_t i = 0; i < arraysize(sensitive_urls); ++i) {
-    TestURLRequest request(non_sensitive_url, NULL, &context);
-    GURL sensitive_url(sensitive_urls[i]);
-    request.set_first_party_for_cookies(sensitive_url);
-    EXPECT_TRUE(WebRequestPermissions::HideRequest(&request))
-        << sensitive_urls[i];
-  }
+
+  // Check protection of requests originating from the frame showing the Chrome
+  // WebStore.
+  // Normally this request is not protected:
+  GURL non_sensitive_url("http://www.google.com/test.js");
+  TestURLRequest non_sensitive_request(non_sensitive_url, NULL, &context);
+  EXPECT_FALSE(WebRequestPermissions::HideRequest(extension_info_map.get(),
+                                                  &non_sensitive_request));
+  // If the origin is labeled by the WebStoreAppId, it becomes protected.
+  int process_id = 42;
+  int site_instance_id = 23;
+  int frame_id = 17;
+  TestURLRequest sensitive_request(non_sensitive_url, NULL, &context);
+  ResourceRequestInfo::AllocateForTesting(&sensitive_request,
+      ResourceType::SCRIPT, NULL, process_id, frame_id);
+  extension_info_map->RegisterExtensionProcess(extension_misc::kWebStoreAppId,
+      process_id, site_instance_id);
+  EXPECT_TRUE(WebRequestPermissions::HideRequest(extension_info_map.get(),
+                                                 &sensitive_request));
 }