blob: 5dc15085888f35f0edbc2c8e459b7db9340827a2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
// 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.
#ifndef CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_WEBREQUEST_WEBREQUEST_CONDITION_H_
#define CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_WEBREQUEST_WEBREQUEST_CONDITION_H_
#include <map>
#include <set>
#include <string>
#include <vector>
#include "base/basictypes.h"
#include "base/memory/linked_ptr.h"
#include "chrome/browser/extensions/api/declarative/declarative_rule.h"
#include "chrome/browser/extensions/api/declarative_webrequest/webrequest_condition_attribute.h"
#include "components/url_matcher/url_matcher.h"
#include "net/http/http_response_headers.h"
namespace extensions {
// Container for information about a URLRequest to determine which
// rules apply to the request.
struct WebRequestData {
WebRequestData(net::URLRequest* request, RequestStage stage);
WebRequestData(
net::URLRequest* request,
RequestStage stage,
const net::HttpResponseHeaders* original_response_headers);
~WebRequestData();
// The network request that is currently being processed.
net::URLRequest* request;
// The stage (progress) of the network request.
RequestStage stage;
// Additional information about requests that is not
// available in all request stages.
const net::HttpResponseHeaders* original_response_headers;
};
// Adds information about URL matches to WebRequestData.
struct WebRequestDataWithMatchIds {
explicit WebRequestDataWithMatchIds(const WebRequestData* request_data);
~WebRequestDataWithMatchIds();
const WebRequestData* data;
std::set<url_matcher::URLMatcherConditionSet::ID> url_match_ids;
std::set<url_matcher::URLMatcherConditionSet::ID> first_party_url_match_ids;
};
// Representation of a condition in the Declarative WebRequest API. A condition
// consists of several attributes. Each of these attributes needs to be
// fulfilled in order for the condition to be fulfilled.
//
// We distinguish between two types of conditions:
// - URL Matcher conditions are conditions that test the URL of a request.
// These are treated separately because we use a URLMatcher to efficiently
// test many of these conditions in parallel by using some advanced
// data structures. The URLMatcher tells us if all URL Matcher conditions
// are fulfilled for a WebRequestCondition.
// - All other conditions are represented as WebRequestConditionAttributes.
// These conditions are probed linearly (only if the URL Matcher found a hit).
//
// TODO(battre) Consider making the URLMatcher an owner of the
// URLMatcherConditionSet and only pass a pointer to URLMatcherConditionSet
// in url_matcher_condition_set(). This saves some copying in
// WebRequestConditionSet::GetURLMatcherConditionSets.
class WebRequestCondition {
public:
typedef WebRequestDataWithMatchIds MatchData;
WebRequestCondition(
scoped_refptr<url_matcher::URLMatcherConditionSet> url_matcher_conditions,
scoped_refptr<url_matcher::URLMatcherConditionSet>
first_party_url_matcher_conditions,
const WebRequestConditionAttributes& condition_attributes);
~WebRequestCondition();
// Factory method that instantiates a WebRequestCondition according to
// the description |condition| passed by the extension API.
static scoped_ptr<WebRequestCondition> Create(
const Extension* extension,
url_matcher::URLMatcherConditionFactory* url_matcher_condition_factory,
const base::Value& condition,
std::string* error);
// Returns whether the request matches this condition.
bool IsFulfilled(const MatchData& request_data) const;
// If this condition has url attributes, appends them to |condition_sets|.
void GetURLMatcherConditionSets(
url_matcher::URLMatcherConditionSet::Vector* condition_sets) const;
// Returns a bit vector representing extensions::RequestStage. The bit vector
// contains a 1 for each request stage during which the condition can be
// tested.
int stages() const { return applicable_request_stages_; }
private:
// URL attributes of this condition.
scoped_refptr<url_matcher::URLMatcherConditionSet> url_matcher_conditions_;
scoped_refptr<url_matcher::URLMatcherConditionSet>
first_party_url_matcher_conditions_;
// All non-UrlFilter attributes of this condition.
WebRequestConditionAttributes condition_attributes_;
// Bit vector indicating all RequestStage during which all
// |condition_attributes_| can be evaluated.
int applicable_request_stages_;
DISALLOW_COPY_AND_ASSIGN(WebRequestCondition);
};
typedef DeclarativeConditionSet<WebRequestCondition> WebRequestConditionSet;
} // namespace extensions
#endif // CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_WEBREQUEST_WEBREQUEST_CONDITION_H_
|