uBlock/platform/chromium/vapi-client.js

423 lines
13 KiB
JavaScript
Raw Normal View History

/*******************************************************************************
2016-03-06 08:51:06 -07:00
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
*/
/* global HTMLDocument, XMLDocument */
'use strict';
// For non background pages
/******************************************************************************/
2017-04-01 10:40:02 -06:00
(function(self) {
2016-03-06 08:51:06 -07:00
/******************************************************************************/
/******************************************************************************/
// https://github.com/chrisaljoudi/uBlock/issues/464
if ( document instanceof HTMLDocument === false ) {
// https://github.com/chrisaljoudi/uBlock/issues/1528
// A XMLDocument can be a valid HTML document.
if (
document instanceof XMLDocument === false ||
document.createElement('div') instanceof HTMLDivElement === false
) {
return;
}
}
2016-01-02 17:33:32 -07:00
// https://github.com/gorhill/uBlock/issues/1124
// Looks like `contentType` is on track to be standardized:
// https://dom.spec.whatwg.org/#concept-document-content-type
// https://forums.lanik.us/viewtopic.php?f=64&t=31522
// Skip text/plain documents.
var contentType = document.contentType || '';
if ( /^image\/|^text\/plain/.test(contentType) ) {
2016-01-02 17:33:32 -07:00
return;
}
/******************************************************************************/
// https://bugs.chromium.org/p/project-zero/issues/detail?id=1225&desc=6#c10
2017-04-03 08:10:27 -06:00
if ( !self.vAPI || self.vAPI.uBO !== true ) {
self.vAPI = { uBO: true };
}
var vAPI = self.vAPI;
2015-01-01 19:14:53 -07:00
var chrome = self.chrome;
2015-04-06 19:26:05 -06:00
// https://github.com/chrisaljoudi/uBlock/issues/456
// Already injected?
if ( vAPI.sessionId ) {
return;
}
2014-11-24 15:20:21 -07:00
/******************************************************************************/
var referenceCounter = 0;
vAPI.lock = function() {
referenceCounter += 1;
};
vAPI.unlock = function() {
referenceCounter -= 1;
if ( referenceCounter === 0 ) {
// Eventually there will be code here to flush the javascript code
// from this file out of memory when it ends up unused.
}
};
/******************************************************************************/
vAPI.executionCost = {
start: function(){},
stop: function(){}
};
/*
vAPI.executionCost = {
tcost: 0,
tstart: 0,
nstart: 0,
level: 1,
start: function() {
if ( this.nstart === 0 ) {
this.tstart = window.performance.now();
}
this.nstart += 1;
},
stop: function(mark) {
this.nstart -= 1;
if ( this.nstart !== 0 ) {
return;
}
var tcost = window.performance.now() - this.tstart;
this.tcost += tcost;
if ( mark === undefined ) {
return;
}
var top = window === window.top;
if ( !top && this.level < 2 ) {
return;
}
var context = window === window.top ? ' top' : 'frame';
var percent = this.tcost / window.performance.now() * 100;
console.log(
'uBO cost (%s): %sms/%s%% (%s: %sms)',
context,
this.tcost.toFixed(1),
percent.toFixed(1),
mark,
tcost.toFixed(2)
);
}
};
*/
vAPI.executionCost.start();
/******************************************************************************/
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() {
2015-04-07 17:34:22 -06:00
var job;
while ( (job = this.jobs.pop()) ) {
2015-04-07 17:34:22 -06:00
job();
}
},
remove: function(job) {
var pos;
while ( (pos = this.jobs.indexOf(job)) !== -1 ) {
this.jobs.splice(pos, 1);
}
}
};
2015-04-07 17:34:22 -06:00
2016-03-06 08:51:06 -07:00
/******************************************************************************/
2015-04-07 17:34:22 -06:00
/******************************************************************************/
vAPI.messaging = {
port: null,
2016-03-06 08:51:06 -07:00
portTimer: null,
portTimerDelay: 10000,
channels: Object.create(null),
channelCount: 0,
pending: Object.create(null),
2015-07-07 09:03:26 -06:00
pendingCount: 0,
auxProcessId: 1,
2016-03-06 08:51:06 -07:00
shuttingDown: false,
shutdown: function() {
this.shuttingDown = true;
this.destroyPort();
},
2015-06-25 22:08:41 -06:00
2016-03-06 08:51:06 -07:00
disconnectListener: function() {
this.port = null;
vAPI.shutdown.exec();
},
2016-03-06 08:51:06 -07:00
disconnectListenerCallback: null,
2016-03-06 08:51:06 -07:00
messageListener: function(details) {
if ( !details ) {
return;
2015-07-07 09:03:26 -06:00
}
2016-03-06 08:51:06 -07:00
// Sent to all channels
if ( details.broadcast === true && !details.channelName ) {
for ( var channelName in this.channels ) {
this.sendToChannelListeners(channelName, details.msg);
}
return;
}
// Response to specific message previously sent
if ( details.auxProcessId ) {
var listener = this.pending[details.auxProcessId];
delete this.pending[details.auxProcessId];
delete details.auxProcessId; // TODO: why?
if ( listener ) {
this.pendingCount -= 1;
listener(details.msg);
return;
}
}
// Sent to a specific channel
var response = this.sendToChannelListeners(details.channelName, details.msg);
// Respond back if required
if ( details.mainProcessId === undefined ) {
return;
}
var port = this.connect();
if ( port !== null ) {
port.postMessage({
mainProcessId: details.mainProcessId,
msg: response
});
2015-07-07 09:03:26 -06:00
}
},
2016-03-06 08:51:06 -07:00
messageListenerCallback: null,
portPoller: function() {
this.portTimer = null;
if ( this.port !== null ) {
if ( this.channelCount !== 0 || this.pendingCount !== 0 ) {
this.portTimer = vAPI.setTimeout(this.portPollerCallback, this.portTimerDelay);
2016-03-09 09:24:21 -07:00
this.portTimerDelay = Math.min(this.portTimerDelay * 2, 60 * 60 * 1000);
2016-03-06 08:51:06 -07:00
return;
}
}
this.destroyPort();
},
portPollerCallback: null,
2015-06-25 22:08:41 -06:00
2016-03-06 08:51:06 -07:00
destroyPort: function() {
if ( this.portTimer !== null ) {
clearTimeout(this.portTimer);
this.portTimer = null;
}
2015-07-07 09:03:26 -06:00
var port = this.port;
if ( port !== null ) {
port.disconnect();
2016-03-06 08:51:06 -07:00
port.onMessage.removeListener(this.messageListenerCallback);
port.onDisconnect.removeListener(this.disconnectListenerCallback);
this.port = null;
}
2016-03-06 08:51:06 -07:00
if ( this.channelCount !== 0 ) {
this.channels = Object.create(null);
this.channelCount = 0;
}
2015-07-07 09:03:26 -06:00
// service pending callbacks
2016-03-06 08:51:06 -07:00
if ( this.pendingCount !== 0 ) {
var pending = this.pending, callback;
this.pending = Object.create(null);
this.pendingCount = 0;
for ( var auxId in pending ) {
callback = pending[auxId];
if ( typeof callback === 'function' ) {
callback(null);
2015-07-07 09:03:26 -06:00
}
}
}
},
2015-06-28 15:42:08 -06:00
2016-03-06 08:51:06 -07:00
createPort: function() {
if ( this.shuttingDown ) {
return null;
}
2016-03-06 08:51:06 -07:00
if ( this.messageListenerCallback === null ) {
this.messageListenerCallback = this.messageListener.bind(this);
this.disconnectListenerCallback = this.disconnectListener.bind(this);
this.portPollerCallback = this.portPoller.bind(this);
}
2016-03-06 08:51:06 -07:00
try {
2016-03-15 09:17:56 -06:00
this.port = chrome.runtime.connect({name: vAPI.sessionId}) || null;
2016-03-06 08:51:06 -07:00
} catch (ex) {
2016-03-15 09:17:56 -06:00
this.port = null;
}
if ( this.port !== null ) {
this.port.onMessage.addListener(this.messageListenerCallback);
this.port.onDisconnect.addListener(this.disconnectListenerCallback);
}
2016-03-09 09:24:21 -07:00
this.portTimerDelay = 10000;
2016-03-06 08:51:06 -07:00
if ( this.portTimer === null ) {
this.portTimer = vAPI.setTimeout(this.portPollerCallback, this.portTimerDelay);
2015-06-25 22:08:41 -06:00
}
2016-03-15 09:17:56 -06:00
return this.port;
2016-03-06 08:51:06 -07:00
},
2016-03-06 08:51:06 -07:00
connect: function() {
return this.port !== null ? this.port : this.createPort();
},
2015-06-25 22:08:41 -06:00
2016-03-06 08:51:06 -07:00
send: function(channelName, message, callback) {
this.sendTo(channelName, message, undefined, undefined, callback);
},
2015-06-25 22:08:41 -06:00
2016-03-06 08:51:06 -07:00
sendTo: function(channelName, message, toTabId, toChannel, 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.pendingCount > 25 ) {
vAPI.shutdown.exec();
2015-06-25 22:08:41 -06:00
}
2016-03-06 08:51:06 -07:00
var port = this.connect();
if ( port === null ) {
2015-07-07 09:03:26 -06:00
if ( typeof callback === 'function' ) {
callback();
}
return;
}
2016-03-06 08:51:06 -07:00
var auxProcessId;
if ( callback ) {
auxProcessId = this.auxProcessId++;
this.pending[auxProcessId] = callback;
this.pendingCount += 1;
}
port.postMessage({
channelName: channelName,
auxProcessId: auxProcessId,
toTabId: toTabId,
toChannel: toChannel,
msg: message
});
},
2016-03-06 08:51:06 -07:00
addChannelListener: function(channelName, callback) {
if ( typeof callback !== 'function' ) {
return;
}
var listeners = this.channels[channelName];
if ( listeners !== undefined && listeners.indexOf(callback) !== -1 ) {
console.error('Duplicate listener on channel "%s"', channelName);
return;
}
if ( listeners === undefined ) {
this.channels[channelName] = [callback];
this.channelCount += 1;
} else {
listeners.push(callback);
}
this.connect();
},
2016-03-06 08:51:06 -07:00
removeChannelListener: function(channelName, callback) {
if ( typeof callback !== 'function' ) {
return;
}
var listeners = this.channels[channelName];
if ( listeners === undefined ) {
return;
}
var pos = listeners.indexOf(callback);
2016-03-06 08:51:06 -07:00
if ( pos === -1 ) {
console.error('Listener not found on channel "%s"', channelName);
return;
}
listeners.splice(pos, 1);
if ( listeners.length === 0 ) {
delete this.channels[channelName];
this.channelCount -= 1;
}
},
2016-03-06 08:51:06 -07:00
removeAllChannelListeners: function(channelName) {
var listeners = this.channels[channelName];
if ( listeners === undefined ) {
return;
}
delete this.channels[channelName];
this.channelCount -= 1;
},
2015-06-28 15:42:08 -06:00
2016-03-06 08:51:06 -07:00
sendToChannelListeners: function(channelName, msg) {
var listeners = this.channels[channelName];
if ( listeners === undefined ) {
return;
}
var response;
for ( var i = 0, n = listeners.length; i < n; i++ ) {
response = listeners[i](msg);
if ( response !== undefined ) {
break;
}
2015-06-25 22:08:41 -06:00
}
2016-03-06 08:51:06 -07:00
return response;
}
};
2016-03-06 08:51:06 -07:00
/******************************************************************************/
vAPI.shutdown.add(function() {
vAPI.messaging.shutdown();
2016-03-15 09:17:56 -06:00
delete window.vAPI;
2016-03-06 08:51:06 -07:00
});
// https://www.youtube.com/watch?v=rT5zCHn0tsg
// https://www.youtube.com/watch?v=E-jS4e3zacI
vAPI.executionCost.stop('vapi-client.js');
/******************************************************************************/
/******************************************************************************/
2017-04-01 10:40:02 -06:00
})(this);
/******************************************************************************/