mirror of
https://github.com/gorhill/uBlock.git
synced 2024-11-09 12:22:33 +01:00
448 lines
14 KiB
JavaScript
448 lines
14 KiB
JavaScript
/*******************************************************************************
|
|
|
|
uBlock Origin - a browser extension to block requests.
|
|
Copyright (C) 2014-2017 The uBlock Origin authors
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see {http://www.gnu.org/licenses/}.
|
|
|
|
Home: https://github.com/gorhill/uBlock
|
|
*/
|
|
|
|
// For non-background page
|
|
|
|
'use strict';
|
|
|
|
/******************************************************************************/
|
|
|
|
// https://github.com/chrisaljoudi/uBlock/issues/456
|
|
// Skip if already injected.
|
|
|
|
if ( typeof vAPI === 'object' && !vAPI.clientScript ) { // >>>>>>>> start of HUGE-IF-BLOCK
|
|
|
|
/******************************************************************************/
|
|
/******************************************************************************/
|
|
|
|
vAPI.clientScript = true;
|
|
|
|
vAPI.randomToken = function() {
|
|
return String.fromCharCode(Date.now() % 26 + 97) +
|
|
Math.floor(Math.random() * 982451653 + 982451653).toString(36);
|
|
};
|
|
|
|
vAPI.sessionId = vAPI.randomToken();
|
|
vAPI.chrome = true;
|
|
vAPI.setTimeout = vAPI.setTimeout || self.setTimeout.bind(self);
|
|
|
|
/******************************************************************************/
|
|
|
|
vAPI.shutdown = {
|
|
jobs: [],
|
|
add: function(job) {
|
|
this.jobs.push(job);
|
|
},
|
|
exec: function() {
|
|
var job;
|
|
while ( (job = this.jobs.pop()) ) {
|
|
job();
|
|
}
|
|
},
|
|
remove: function(job) {
|
|
var pos;
|
|
while ( (pos = this.jobs.indexOf(job)) !== -1 ) {
|
|
this.jobs.splice(pos, 1);
|
|
}
|
|
}
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
vAPI.messaging = {
|
|
port: null,
|
|
portTimer: null,
|
|
portTimerDelay: 10000,
|
|
channels: new Map(),
|
|
connections: new Map(),
|
|
pending: new Map(),
|
|
auxProcessId: 1,
|
|
shuttingDown: false,
|
|
|
|
Connection: function(handler, details) {
|
|
var messaging = vAPI.messaging;
|
|
this.messaging = messaging;
|
|
this.handler = handler;
|
|
this.id = details.id;
|
|
this.to = details.to;
|
|
this.toToken = details.toToken;
|
|
this.from = details.from;
|
|
this.fromToken = details.fromToken;
|
|
this.checkBound = this.check.bind(this);
|
|
this.checkTimer = undefined;
|
|
// On Firefox it appears ports are not automatically disconnected when
|
|
// navigating to another page.
|
|
if ( messaging.Connection.pagehide !== undefined ) { return; }
|
|
messaging.Connection.pagehide = function() {
|
|
for ( var connection of this.connections.values() ) {
|
|
connection.disconnect();
|
|
connection.handler(connection.toDetails('connectionBroken'));
|
|
}
|
|
}.bind(messaging);
|
|
window.addEventListener('pagehide', messaging.Connection.pagehide);
|
|
},
|
|
|
|
shutdown: function() {
|
|
this.shuttingDown = true;
|
|
this.destroyPort();
|
|
},
|
|
|
|
disconnectListener: function() {
|
|
this.port = null;
|
|
vAPI.shutdown.exec();
|
|
},
|
|
disconnectListenerBound: null,
|
|
|
|
messageListener: function(details) {
|
|
if ( !details ) { return; }
|
|
|
|
// Sent to all channels
|
|
if ( details.broadcast ) {
|
|
for ( var channelName of this.channels.keys() ) {
|
|
this.sendToChannelListeners(channelName, details.msg);
|
|
}
|
|
return;
|
|
}
|
|
|
|
// Response to specific message previously sent
|
|
var listener;
|
|
if ( details.auxProcessId ) {
|
|
listener = this.pending.get(details.auxProcessId);
|
|
if ( listener !== undefined ) {
|
|
this.pending.delete(details.auxProcessId);
|
|
listener(details.msg);
|
|
return;
|
|
}
|
|
}
|
|
|
|
if ( details.channelName !== 'vapi' ) { return; }
|
|
|
|
// Internal handler
|
|
var connection;
|
|
|
|
switch ( details.msg.what ) {
|
|
case 'connectionAccepted':
|
|
case 'connectionBroken':
|
|
case 'connectionCheck':
|
|
case 'connectionMessage':
|
|
case 'connectionRefused':
|
|
connection = this.connections.get(details.msg.id);
|
|
if ( connection === undefined ) { return; }
|
|
connection.receive(details.msg);
|
|
break;
|
|
case 'connectionRequested':
|
|
var listeners = this.channels.get(details.msg.to);
|
|
if ( listeners === undefined ) { return; }
|
|
var port = this.getPort();
|
|
if ( port === null ) { return; }
|
|
for ( listener of listeners ) {
|
|
if ( listener(details.msg) !== true ) { continue; }
|
|
details.msg.what = 'connectionAccepted';
|
|
details.msg.toToken = port.name;
|
|
connection = new this.Connection(listener, details.msg);
|
|
this.connections.set(connection.id, connection);
|
|
break;
|
|
}
|
|
if ( details.msg.what !== 'connectionAccepted' ) {
|
|
details.msg.what = 'connectionRefused';
|
|
}
|
|
port.postMessage(details);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
},
|
|
messageListenerCallback: null,
|
|
|
|
portPoller: function() {
|
|
this.portTimer = null;
|
|
if (
|
|
this.port !== null &&
|
|
this.channels.size === 0 &&
|
|
this.connections.size === 0 &&
|
|
this.pending.size === 0
|
|
) {
|
|
return this.destroyPort();
|
|
}
|
|
this.portTimer = vAPI.setTimeout(this.portPollerBound, this.portTimerDelay);
|
|
this.portTimerDelay = Math.min(this.portTimerDelay * 2, 60 * 60 * 1000);
|
|
},
|
|
portPollerBound: null,
|
|
|
|
destroyPort: function() {
|
|
if ( this.portTimer !== null ) {
|
|
clearTimeout(this.portTimer);
|
|
this.portTimer = null;
|
|
}
|
|
var port = this.port;
|
|
if ( port !== null ) {
|
|
port.disconnect();
|
|
port.onMessage.removeListener(this.messageListenerCallback);
|
|
port.onDisconnect.removeListener(this.disconnectListenerBound);
|
|
this.port = null;
|
|
}
|
|
this.channels.clear();
|
|
if ( this.connections.size !== 0 ) {
|
|
for ( var connection of this.connections.values() ) {
|
|
connection.receive({ what: 'connectionBroken' });
|
|
}
|
|
this.connections.clear();
|
|
}
|
|
// service pending callbacks
|
|
if ( this.pending.size !== 0 ) {
|
|
var pending = this.pending;
|
|
this.pending = new Map();
|
|
for ( var callback of pending.values() ) {
|
|
if ( typeof callback === 'function' ) {
|
|
callback(null);
|
|
}
|
|
}
|
|
}
|
|
},
|
|
|
|
createPort: function() {
|
|
if ( this.shuttingDown ) { return null; }
|
|
if ( this.messageListenerCallback === null ) {
|
|
this.messageListenerCallback = this.messageListener.bind(this);
|
|
this.disconnectListenerBound = this.disconnectListener.bind(this);
|
|
this.portPollerBound = this.portPoller.bind(this);
|
|
}
|
|
try {
|
|
this.port = chrome.runtime.connect({name: vAPI.sessionId}) || null;
|
|
} catch (ex) {
|
|
this.port = null;
|
|
}
|
|
if ( this.port !== null ) {
|
|
this.port.onMessage.addListener(this.messageListenerCallback);
|
|
this.port.onDisconnect.addListener(this.disconnectListenerBound);
|
|
this.portTimerDelay = 10000;
|
|
if ( this.portTimer === null ) {
|
|
this.portTimer = vAPI.setTimeout(
|
|
this.portPollerBound,
|
|
this.portTimerDelay
|
|
);
|
|
}
|
|
}
|
|
return this.port;
|
|
},
|
|
|
|
getPort: function() {
|
|
return this.port !== null ? this.port : this.createPort();
|
|
},
|
|
|
|
send: function(channelName, message, callback) {
|
|
// Too large a gap between the last request and the last response means
|
|
// the main process is no longer reachable: memory leaks and bad
|
|
// performance become a risk -- especially for long-lived, dynamic
|
|
// pages. Guard against this.
|
|
if ( this.pending.size > 25 ) {
|
|
vAPI.shutdown.exec();
|
|
}
|
|
var port = this.getPort();
|
|
if ( port === null ) {
|
|
if ( typeof callback === 'function' ) { callback(); }
|
|
return;
|
|
}
|
|
var auxProcessId;
|
|
if ( callback ) {
|
|
auxProcessId = this.auxProcessId++;
|
|
this.pending.set(auxProcessId, callback);
|
|
}
|
|
port.postMessage({
|
|
channelName: channelName,
|
|
auxProcessId: auxProcessId,
|
|
msg: message
|
|
});
|
|
},
|
|
|
|
connectTo: function(from, to, handler) {
|
|
var port = this.getPort();
|
|
if ( port === null ) { return; }
|
|
var connection = new this.Connection(handler, {
|
|
id: from + '-' + to + '-' + vAPI.sessionId,
|
|
to: to,
|
|
from: from,
|
|
fromToken: port.name
|
|
});
|
|
this.connections.set(connection.id, connection);
|
|
port.postMessage({
|
|
channelName: 'vapi',
|
|
msg: {
|
|
what: 'connectionRequested',
|
|
id: connection.id,
|
|
from: from,
|
|
fromToken: port.name,
|
|
to: to
|
|
}
|
|
});
|
|
return connection.id;
|
|
},
|
|
|
|
disconnectFrom: function(connectionId) {
|
|
var connection = this.connections.get(connectionId);
|
|
if ( connection === undefined ) { return; }
|
|
connection.disconnect();
|
|
},
|
|
|
|
sendTo: function(connectionId, payload) {
|
|
var connection = this.connections.get(connectionId);
|
|
if ( connection === undefined ) { return; }
|
|
connection.send(payload);
|
|
},
|
|
|
|
addChannelListener: function(channelName, listener) {
|
|
var listeners = this.channels.get(channelName);
|
|
if ( listeners === undefined ) {
|
|
this.channels.set(channelName, [ listener ]);
|
|
} else if ( listeners.indexOf(listener) === -1 ) {
|
|
listeners.push(listener);
|
|
}
|
|
this.getPort();
|
|
},
|
|
|
|
removeChannelListener: function(channelName, listener) {
|
|
var listeners = this.channels.get(channelName);
|
|
if ( listeners === undefined ) { return; }
|
|
var pos = listeners.indexOf(listener);
|
|
if ( pos === -1 ) { return; }
|
|
listeners.splice(pos, 1);
|
|
if ( listeners.length === 0 ) {
|
|
this.channels.delete(channelName);
|
|
}
|
|
},
|
|
|
|
removeAllChannelListeners: function(channelName) {
|
|
this.channels.delete(channelName);
|
|
},
|
|
|
|
sendToChannelListeners: function(channelName, msg) {
|
|
var listeners = this.channels.get(channelName);
|
|
if ( listeners === undefined ) { return; }
|
|
listeners = listeners.slice(0);
|
|
var response;
|
|
for ( var listener of listeners ) {
|
|
response = listener(msg);
|
|
if ( response !== undefined ) { break; }
|
|
}
|
|
return response;
|
|
}
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
vAPI.messaging.Connection.prototype = {
|
|
toDetails: function(what, payload) {
|
|
return {
|
|
what: what,
|
|
id: this.id,
|
|
from: this.from,
|
|
fromToken: this.fromToken,
|
|
to: this.to,
|
|
toToken: this.toToken,
|
|
payload: payload
|
|
};
|
|
},
|
|
disconnect: function() {
|
|
if ( this.checkTimer !== undefined ) {
|
|
clearTimeout(this.checkTimer);
|
|
this.checkTimer = undefined;
|
|
}
|
|
this.messaging.connections.delete(this.id);
|
|
var port = this.messaging.getPort();
|
|
if ( port === null ) { return; }
|
|
port.postMessage({
|
|
channelName: 'vapi',
|
|
msg: this.toDetails('connectionBroken')
|
|
});
|
|
},
|
|
checkAsync: function() {
|
|
if ( this.checkTimer !== undefined ) {
|
|
clearTimeout(this.checkTimer);
|
|
}
|
|
this.checkTimer = vAPI.setTimeout(this.checkBound, 499);
|
|
},
|
|
check: function() {
|
|
this.checkTimer = undefined;
|
|
if ( this.messaging.connections.has(this.id) === false ) { return; }
|
|
var port = this.messaging.getPort();
|
|
if ( port === null ) { return; }
|
|
port.postMessage({
|
|
channelName: 'vapi',
|
|
msg: this.toDetails('connectionCheck')
|
|
});
|
|
this.checkAsync();
|
|
},
|
|
receive: function(details) {
|
|
switch ( details.what ) {
|
|
case 'connectionAccepted':
|
|
this.toToken = details.toToken;
|
|
this.handler(details);
|
|
this.checkAsync();
|
|
break;
|
|
case 'connectionBroken':
|
|
this.messaging.connections.delete(this.id);
|
|
this.handler(details);
|
|
break;
|
|
case 'connectionMessage':
|
|
this.handler(details);
|
|
this.checkAsync();
|
|
break;
|
|
case 'connectionCheck':
|
|
var port = this.messaging.getPort();
|
|
if ( port === null ) { return; }
|
|
if ( this.messaging.connections.has(this.id) ) {
|
|
this.checkAsync();
|
|
} else {
|
|
details.what = 'connectionBroken';
|
|
port.postMessage({ channelName: 'vapi', msg: details });
|
|
}
|
|
break;
|
|
case 'connectionRefused':
|
|
this.messaging.connections.delete(this.id);
|
|
this.handler(details);
|
|
break;
|
|
}
|
|
},
|
|
send: function(payload) {
|
|
var port = this.messaging.getPort();
|
|
if ( port === null ) { return; }
|
|
port.postMessage({
|
|
channelName: 'vapi',
|
|
msg: this.toDetails('connectionMessage', payload)
|
|
});
|
|
}
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
vAPI.shutdown.add(function() {
|
|
vAPI.messaging.shutdown();
|
|
window.vAPI = undefined;
|
|
});
|
|
|
|
// https://www.youtube.com/watch?v=rT5zCHn0tsg
|
|
// https://www.youtube.com/watch?v=E-jS4e3zacI
|
|
|
|
/******************************************************************************/
|
|
/******************************************************************************/
|
|
|
|
} // <<<<<<<< end of HUGE-IF-BLOCK
|