Declarative Web Request: firstPartyForCookiesUrl

chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_registry.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/declarative_webrequest/webrequest_rules_registry.h"
 
+#include <algorithm>
 #include <limits>
 
 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_condition.h"
 #include "chrome/browser/extensions/api/web_request/web_request_api_helpers.h"
 #include "chrome/browser/extensions/api/web_request/web_request_permissions.h"
 #include "chrome/browser/extensions/extension_system.h"
 #include "net/url_request/url_request.h"
 
 namespace extensions {
 
 WebRequestRulesRegistry::WebRequestRulesRegistry(Profile* profile,
                                                  Delegate* delegate)
     : RulesRegistryWithCache(delegate) {
   if (profile)
     extension_info_map_ = ExtensionSystem::Get(profile)->info_map();
 }
 
 std::set<WebRequestRule::GlobalRuleId>
 WebRequestRulesRegistry::GetMatches(
     const WebRequestRule::RequestData& request_data) {
   std::set<WebRequestRule::GlobalRuleId> result;
 
   // Figure out for which rules the URL match conditions were fulfilled.
   typedef std::set<URLMatcherConditionSet::ID> URLMatches;
   URLMatches url_matches = url_matcher_.MatchURL(request_data.request->url());
+  std::set<const WebRequestRule*> url_matched_rules;
+  std::transform(url_matches.begin(), url_matches.end(),
+                 std::inserter(url_matched_rules, url_matched_rules.begin()),
+                 RuleTriggersOperator(rule_triggers_));
+
+  URLMatches first_party_url_matches = first_party_url_matcher_.MatchURL(
+      request_data.request->first_party_for_cookies());
 
   // Then we need to check for each of these, whether the other
   // WebRequestConditionAttributes are also fulfilled.
-  for (URLMatches::iterator url_match = url_matches.begin();
-       url_match != url_matches.end(); ++url_match) {
+  for (URLMatches::iterator url_match = first_party_url_matches.begin();
+       url_match != first_party_url_matches.end(); ++url_match) {
     RuleTriggers::iterator rule_trigger = rule_triggers_.find(*url_match);
     CHECK(rule_trigger != rule_triggers_.end());
 
     WebRequestRule* rule = rule_trigger->second;
-    if (rule->conditions().IsFulfilled(*url_match, request_data))
+    // We already iterate over rules which have the first-party URL condition
+    // satisfied. Now check that the request URL is also satisfied, and finally
+    // check the rest of the attributes.
+    if (url_matched_rules.find(rule) != url_matched_rules.end() &&
+        rule->conditions().IsFulfilled(*url_match, request_data))
       result.insert(rule->id());
   }
   return result;
 }
 
 std::list<LinkedPtrEventResponseDelta> WebRequestRulesRegistry::CreateDeltas(
     const ExtensionInfoMap* extension_info_map,
     const WebRequestRule::RequestData& request_data,
     bool crosses_incognito) {
   if (webrequest_rules_.empty())
@@ skipping 65 matching lines @@
 
   std::string error;
   RulesMap new_webrequest_rules;
 
   for (std::vector<linked_ptr<RulesRegistry::Rule> >::const_iterator rule =
        rules.begin(); rule != rules.end(); ++rule) {
     WebRequestRule::GlobalRuleId rule_id(extension_id, *(*rule)->id);
     DCHECK(webrequest_rules_.find(rule_id) == webrequest_rules_.end());
 
     scoped_ptr<WebRequestRule> webrequest_rule(
-        WebRequestRule::Create(url_matcher_.condition_factory(), extension_id,
+        WebRequestRule::Create(url_matcher_.condition_factory(),
+                               first_party_url_matcher_.condition_factory(),
+                               extension_id,
                                extension_installation_time, *rule, &error));
     if (!error.empty()) {
       // We don't return here, because we want to clear temporary
-      // condition sets in the url_matcher_.
+      // condition sets in |url_matcher_| and |first_party_url_matcher_|.
       break;
     }
 
     new_webrequest_rules[rule_id] = make_linked_ptr(webrequest_rule.release());
   }
 
   if (!error.empty()) {
     // Clean up temporary condition sets created during rule creation.
     url_matcher_.ClearUnusedConditionSets();
+    first_party_url_matcher_.ClearUnusedConditionSets();
     return error;
   }
 
   // Wohoo, everything worked fine.
   webrequest_rules_.insert(new_webrequest_rules.begin(),
                            new_webrequest_rules.end());
 
   // Create the triggers.
   for (RulesMap::iterator i = new_webrequest_rules.begin();
        i != new_webrequest_rules.end(); ++i) {
     URLMatcherConditionSet::Vector url_condition_sets;
     const WebRequestConditionSet& conditions = i->second->conditions();
     conditions.GetURLMatcherConditionSets(&url_condition_sets);
+    conditions.GetFirstPartyURLMatcherConditionSets(&url_condition_sets);
     for (URLMatcherConditionSet::Vector::iterator j =
          url_condition_sets.begin(); j != url_condition_sets.end(); ++j) {
       rule_triggers_[(*j)->id()] = i->second.get();
     }
   }
 
-  // Register url patterns in url_matcher_.
+  // Register url patterns in |url_matcher_| and |first_party_url_matcher_|.
   URLMatcherConditionSet::Vector all_new_condition_sets;
+  URLMatcherConditionSet::Vector all_new_condition_sets_first_party;
   for (RulesMap::iterator i = new_webrequest_rules.begin();
        i != new_webrequest_rules.end(); ++i) {
     i->second->conditions().GetURLMatcherConditionSets(&all_new_condition_sets);
+    i->second->conditions().GetFirstPartyURLMatcherConditionSets(
+        &all_new_condition_sets_first_party);
   }
   url_matcher_.AddConditionSets(all_new_condition_sets);
+  first_party_url_matcher_.AddConditionSets(all_new_condition_sets_first_party);
 
   ClearCacheOnNavigation();
 
   return "";
 }
 
 std::string WebRequestRulesRegistry::RemoveRulesImpl(
     const std::string& extension_id,
     const std::vector<std::string>& rule_identifiers) {
-  // URLMatcherConditionSet IDs that can be removed from URLMatcher.
+  // URLMatcherConditionSet IDs that can be removed from URL matchers.
   std::vector<URLMatcherConditionSet::ID> remove_from_url_matcher;
+  std::vector<URLMatcherConditionSet::ID> remove_from_first_party_url_matcher;
 
   for (std::vector<std::string>::const_iterator i = rule_identifiers.begin();
        i != rule_identifiers.end(); ++i) {
     WebRequestRule::GlobalRuleId rule_id(extension_id, *i);
 
     // Skip unknown rules.
     RulesMap::iterator webrequest_rules_entry = webrequest_rules_.find(rule_id);
     if (webrequest_rules_entry == webrequest_rules_.end())
       continue;
 
     // Remove all triggers but collect their IDs.
     URLMatcherConditionSet::Vector condition_sets;
+    URLMatcherConditionSet::Vector condition_sets_first_party;
     WebRequestRule* rule = webrequest_rules_entry->second.get();
     rule->conditions().GetURLMatcherConditionSets(&condition_sets);
+    rule->conditions().GetFirstPartyURLMatcherConditionSets(
+        &condition_sets_first_party);
     for (URLMatcherConditionSet::Vector::iterator j = condition_sets.begin();
          j != condition_sets.end(); ++j) {
       remove_from_url_matcher.push_back((*j)->id());
       rule_triggers_.erase((*j)->id());
     }
+    for (URLMatcherConditionSet::Vector::iterator j =
+             condition_sets_first_party.begin();
+         j != condition_sets_first_party.end(); ++j) {
+      remove_from_first_party_url_matcher.push_back((*j)->id());
+      rule_triggers_.erase((*j)->id());
+    }
 
     // Remove reference to actual rule.
     webrequest_rules_.erase(webrequest_rules_entry);
   }
 
-  // Clear URLMatcher based on condition_set_ids that are not needed any more.
+  // Clear URL matchers based on condition_set_ids that are not needed any more.
   url_matcher_.RemoveConditionSets(remove_from_url_matcher);
+  first_party_url_matcher_.RemoveConditionSets(
+      remove_from_first_party_url_matcher);
 
   ClearCacheOnNavigation();
 
   return "";
 }
 
 std::string WebRequestRulesRegistry::RemoveAllRulesImpl(
     const std::string& extension_id) {
   // Search all identifiers of rules that belong to extension |extension_id|.
   std::vector<std::string> rule_identifiers;
   for (RulesMap::iterator i = webrequest_rules_.begin();
        i != webrequest_rules_.end(); ++i) {
     const WebRequestRule::GlobalRuleId& global_rule_id = i->first;
     if (global_rule_id.first == extension_id)
       rule_identifiers.push_back(global_rule_id.second);
   }
 
   // No need to call ClearCacheOnNavigation() here because RemoveRulesImpl
   // takes care of that.
   return RemoveRulesImpl(extension_id, rule_identifiers);
 }
 
 content::BrowserThread::ID WebRequestRulesRegistry::GetOwnerThread() const {
   return content::BrowserThread::IO;
 }
 
 bool WebRequestRulesRegistry::IsEmpty() const {
   return rule_triggers_.empty() && webrequest_rules_.empty() &&
-      url_matcher_.IsEmpty();
+      url_matcher_.IsEmpty() && first_party_url_matcher_.IsEmpty();
 }
 
 WebRequestRulesRegistry::~WebRequestRulesRegistry() {}
 
 base::Time WebRequestRulesRegistry::GetExtensionInstallationTime(
     const std::string& extension_id) const {
   if (!extension_info_map_.get())  // May be NULL during testing.
     return base::Time();
 
   return extension_info_map_->GetInstallTime(extension_id);
 }
 
 void WebRequestRulesRegistry::ClearCacheOnNavigation() {
   extension_web_request_api_helpers::ClearCacheOnNavigation();
 }
 
+WebRequestRulesRegistry::RuleTriggersOperator::RuleTriggersOperator(
+    const RuleTriggers& triggers)
+    : triggers_(triggers) {
+}
+
+WebRequestRulesRegistry::RuleTriggersOperator::~RuleTriggersOperator() {
+}
+
 }  // namespace extensions