| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_
registry.h" | 5 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_
registry.h" |
| 6 | 6 |
| 7 #include <limits> | 7 #include <limits> |
| 8 | 8 |
| 9 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_condit
ion.h" | 9 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_condit
ion.h" |
| 10 #include "chrome/browser/extensions/api/web_request/web_request_api_helpers.h" | 10 #include "chrome/browser/extensions/api/web_request/web_request_api_helpers.h" |
| 11 #include "chrome/browser/extensions/api/web_request/web_request_permissions.h" |
| 11 #include "chrome/browser/extensions/extension_system.h" | 12 #include "chrome/browser/extensions/extension_system.h" |
| 12 #include "net/url_request/url_request.h" | 13 #include "net/url_request/url_request.h" |
| 13 | 14 |
| 14 namespace extensions { | 15 namespace extensions { |
| 15 | 16 |
| 16 WebRequestRulesRegistry::WebRequestRulesRegistry(Profile* profile, | 17 WebRequestRulesRegistry::WebRequestRulesRegistry(Profile* profile, |
| 17 Delegate* delegate) | 18 Delegate* delegate) |
| 18 : RulesRegistryWithCache(delegate) { | 19 : RulesRegistryWithCache(delegate) { |
| 19 if (profile) | 20 if (profile) |
| 20 extension_info_map_ = ExtensionSystem::Get(profile)->info_map(); | 21 extension_info_map_ = ExtensionSystem::Get(profile)->info_map(); |
| (...skipping 18 matching lines...) Expand all Loading... |
| 39 WebRequestRule* rule = rule_trigger->second; | 40 WebRequestRule* rule = rule_trigger->second; |
| 40 if (rule->conditions().IsFulfilled(*url_match, request, request_stage)) | 41 if (rule->conditions().IsFulfilled(*url_match, request, request_stage)) |
| 41 result.insert(rule->id()); | 42 result.insert(rule->id()); |
| 42 } | 43 } |
| 43 return result; | 44 return result; |
| 44 } | 45 } |
| 45 | 46 |
| 46 std::list<LinkedPtrEventResponseDelta> WebRequestRulesRegistry::CreateDeltas( | 47 std::list<LinkedPtrEventResponseDelta> WebRequestRulesRegistry::CreateDeltas( |
| 47 const ExtensionInfoMap* extension_info_map, | 48 const ExtensionInfoMap* extension_info_map, |
| 48 net::URLRequest* request, | 49 net::URLRequest* request, |
| 50 bool crosses_incognito, |
| 49 RequestStages request_stage, | 51 RequestStages request_stage, |
| 50 const WebRequestRule::OptionalRequestData& optional_request_data) { | 52 const WebRequestRule::OptionalRequestData& optional_request_data) { |
| 51 if (webrequest_rules_.empty()) | 53 if (webrequest_rules_.empty()) |
| 52 return std::list<LinkedPtrEventResponseDelta>(); | 54 return std::list<LinkedPtrEventResponseDelta>(); |
| 53 | 55 |
| 54 std::set<WebRequestRule::GlobalRuleId> matches = | 56 std::set<WebRequestRule::GlobalRuleId> matches = |
| 55 GetMatches(request, request_stage); | 57 GetMatches(request, request_stage); |
| 56 | 58 |
| 57 // Sort all matching rules by their priority so that they can be processed | 59 // Sort all matching rules by their priority so that they can be processed |
| 58 // in decreasing order. | 60 // in decreasing order. |
| (...skipping 26 matching lines...) Expand all Loading... |
| 85 | 87 |
| 86 // Create deltas until we have passed the minimum priority. | 88 // Create deltas until we have passed the minimum priority. |
| 87 std::list<LinkedPtrEventResponseDelta> result; | 89 std::list<LinkedPtrEventResponseDelta> result; |
| 88 for (std::vector<PriorityRuleIdPair>::iterator i = ordered_matches.begin(); | 90 for (std::vector<PriorityRuleIdPair>::iterator i = ordered_matches.begin(); |
| 89 i != ordered_matches.end(); ++i) { | 91 i != ordered_matches.end(); ++i) { |
| 90 const WebRequestRule::Priority priority_of_rule = i->first; | 92 const WebRequestRule::Priority priority_of_rule = i->first; |
| 91 const WebRequestRule::GlobalRuleId& rule_id = i->second; | 93 const WebRequestRule::GlobalRuleId& rule_id = i->second; |
| 92 const ExtensionId& extension_id = rule_id.first; | 94 const ExtensionId& extension_id = rule_id.first; |
| 93 const WebRequestRule* rule = webrequest_rules_[rule_id].get(); | 95 const WebRequestRule* rule = webrequest_rules_[rule_id].get(); |
| 94 CHECK(rule); | 96 CHECK(rule); |
| 95 const extensions::Extension* extension = NULL; | |
| 96 if (extension_info_map) | |
| 97 extension = extension_info_map->extensions().GetByID(extension_id); | |
| 98 | 97 |
| 99 // Skip rule if a previous rule of this extension instructed to ignore | 98 // Skip rule if a previous rule of this extension instructed to ignore |
| 100 // all rules with a lower priority than min_priorities[extension_id]. | 99 // all rules with a lower priority than min_priorities[extension_id]. |
| 101 int current_min_priority = min_priorities[extension_id]; | 100 int current_min_priority = min_priorities[extension_id]; |
| 102 if (priority_of_rule < current_min_priority) | 101 if (priority_of_rule < current_min_priority) |
| 103 continue; | 102 continue; |
| 104 | 103 |
| 105 std::list<LinkedPtrEventResponseDelta> rule_result = | 104 std::list<LinkedPtrEventResponseDelta> rule_result = |
| 106 rule->CreateDeltas(extension, request, request_stage, | 105 rule->CreateDeltas(extension_info_map, request, crosses_incognito, |
| 107 optional_request_data); | 106 request_stage, optional_request_data); |
| 108 result.splice(result.begin(), rule_result); | 107 result.splice(result.begin(), rule_result); |
| 109 | 108 |
| 110 min_priorities[extension_id] = std::max(current_min_priority, | 109 min_priorities[extension_id] = std::max(current_min_priority, |
| 111 rule->GetMinimumPriority()); | 110 rule->GetMinimumPriority()); |
| 112 } | 111 } |
| 113 return result; | 112 return result; |
| 114 } | 113 } |
| 115 | 114 |
| 116 std::string WebRequestRulesRegistry::AddRulesImpl( | 115 std::string WebRequestRulesRegistry::AddRulesImpl( |
| 117 const std::string& extension_id, | 116 const std::string& extension_id, |
| (...skipping 116 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 234 | 233 |
| 235 base::Time WebRequestRulesRegistry::GetExtensionInstallationTime( | 234 base::Time WebRequestRulesRegistry::GetExtensionInstallationTime( |
| 236 const std::string& extension_id) const { | 235 const std::string& extension_id) const { |
| 237 if (!extension_info_map_.get()) // May be NULL during testing. | 236 if (!extension_info_map_.get()) // May be NULL during testing. |
| 238 return base::Time(); | 237 return base::Time(); |
| 239 | 238 |
| 240 return extension_info_map_->GetInstallTime(extension_id); | 239 return extension_info_map_->GetInstallTime(extension_id); |
| 241 } | 240 } |
| 242 | 241 |
| 243 } // namespace extensions | 242 } // namespace extensions |
| OLD | NEW |