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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
//This is the background script. It is responsible for actually redirecting requests,
//as well as monitoring changes in the redirects and the disabled status and reacting to them.
//TODO: Better browser detection...
var isFirefox = !!navigator.userAgent.match(/Firefox\//);
var storage = chrome.storage.local; //TODO: Change to sync when Firefox supports it...
//Hopefully Firefox will fix this at some point and we can just use onBeforeRequest everywhere...
var redirectEvent = isFirefox ? chrome.webRequest.onBeforeSendHeaders : chrome.webRequest.onBeforeRequest;
//Redirects partitioned by request type, so we have to run through
//the minimum number of redirects for each request.
var partitionedRedirects = {};
//Keep track of tabids where the main_frame url has been redirected.
//Mark it as green until a new url is loaded.
var tabIdToIcon = {
};
function log(msg) {
if (log.enabled) {
console.log(msg);
}
}
log.enabled = true;
function setIcon(image19, image38, tabId) {
var data = {
path: {
19: image19,
38: image38
}
};
if (typeof tabId !== 'undefined') {
data.tabId = tabId;
}
chrome.browserAction.setIcon(data, function(tab) {
var err = chrome.runtime.lastError;
if (err) {
//If not checked we will get unchecked errors in the background page console...
log('Error in SetIcon: ' + err.message);
}
});
}
//This is the actual function that gets called for each request and must
//decide whether or not we want to redirect.
function checkRedirects(details) {
log('Checking: ' + details.url);
//We only allow GET request to be redirected, don't want to accidentally redirect
//sensitive POST parameters
if (details.method != 'GET') {
return null;
}
var list = partitionedRedirects[details.type];
if (!list) {
log('No list for type: ' + details.type);
return;
}
for (var i = 0; i < list.length; i++) {
var r = list[i];
var result = r.getMatch(details.url);
if (result.isMatch) {
//Have to check if the result also matches, which would cause a loop...
//Based on tests in chrome it actually looks like we don't get passed the redirect url back into our listener so
//this should be unneccessary. But lets verify on Firefox and Opera first, before removing this code (and the recursive warning in the
//edit box)
var recursiveResult = r.getMatch(result.redirectTo);
if (recursiveResult.isMatch) {
log('Ignoring pattern ' + r.includePattern + ' for url ' + details.url + ', because it would also match the result: ' + result.redirectTo);
} else {
log('Redirecting ' + details.url + ' ===> ' + result.redirectTo + ', type: ' + details.type + ', pattern: ' + r.includePattern);
/* Unfortunately the setBrowserIcon for a specific tab function is way too unreliable, fails all the time with tab not found,
even though the tab is there. So, for now I'm cancelling this feature, which would have been pretty great ... :/
if (details.type == 'main_frame') {
log('Setting icon on tab ' + details.tabId + ' to green');
setIcon("images/icon19redirected.png", "images/icon38redirected.png", details.tabId);
tabIdToIcon[details.tabId] = true;
}*/
return { redirectUrl: result.redirectTo };
}
}
}
/* Cancelled for now because of setBrowserIcon being really unreliable...
if (details.type == 'main_frame' && tabIdToIcon[details.tabId]) {
log('Setting icon on tab ' + details.tabId + ' back to active');
setIcon("images/icon19active.png", "images/icon38active.png", details.tabId);
delete tabIdToIcon[details.tabId];
}*/
return null;
}
//Monitor changes in data, and setup everything again.
//This could probably be optimized to not do everything on every change
//but why bother?
chrome.storage.onChanged.addListener(function(changes, namespace) {
if (changes.disabled) {
updateIcon();
if (changes.disabled.newValue == true) {
log('Disabling Redirector, removing listener');
redirectEvent.removeListener(checkRedirects);
} else {
log('Enabling Redirector, setting up listener');
setUpRedirectListener();
}
}
if (changes.redirects) {
log('Redirects have changed, setting up listener again');
setUpRedirectListener();
}
});
//Creates a filter to pass to the listener so we don't have to run through
//all the redirects for all the request types we don't have any redirects for anyway.
function createFilter(redirects) {
var types = [];
for (var i = 0; i < redirects.length; i++) {
redirects[i].appliesTo.forEach(function(type) {
if (types.indexOf(type) == -1) {
types.push(type);
}
});
}
types.sort();
return {
urls: ["http://*/*", "https://*/*"],
types : types
};
}
function createPartitionedRedirects(redirects) {
var partitioned = {};
for (var i = 0; i < redirects.length; i++) {
var redirect = new Redirect(redirects[i]);
redirect.compile();
for (var j=0; j<redirect.appliesTo.length;j++) {
var requestType = redirect.appliesTo[j];
if (partitioned[requestType]) {
partitioned[requestType].push(redirect);
} else {
partitioned[requestType] = [redirect];
}
}
}
return partitioned;
}
//Sets up the listener, partitions the redirects, creates the appropriate filters etc.
function setUpRedirectListener() {
redirectEvent.removeListener(checkRedirects); //Unsubscribe first, in case there are changes...
storage.get('redirects', function(obj) {
if (!obj.redirects) {
log('No redirects to set up');
return;
}
partitionedRedirects = createPartitionedRedirects(obj.redirects);
var filter = createFilter(obj.redirects);
log('Setting filter for listener: ' + JSON.stringify(filter));
redirectEvent.addListener(checkRedirects, filter, ["blocking"]);
});
}
function updateIcon() {
storage.get({disabled:false}, function(obj) {
if (obj.disabled) {
setIcon("images/icon19disabled.png", "images/icon38disabled.png");
} else {
setIcon("images/icon19active.png", "images/icon38active.png");
}
});
}
//First time setup
updateIcon();
storage.get({disabled:false}, function(obj) {
if (!obj.disabled) {
setUpRedirectListener();
}
});
|
