mirror of
https://github.com/gorhill/uMatrix.git
synced 2024-06-03 02:44:57 +12:00
378 lines
12 KiB
JavaScript
378 lines
12 KiB
JavaScript
/*******************************************************************************
|
|
|
|
µMatrix - a Chromium browser extension to black/white list requests.
|
|
Copyright (C) 2014 Raymond Hill
|
|
|
|
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/uMatrix
|
|
*/
|
|
|
|
/* global chrome, µMatrix, punycode, publicSuffixList */
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.getBytesInUse = function() {
|
|
var getBytesInUseHandler = function(bytesInUse) {
|
|
µMatrix.storageUsed = bytesInUse;
|
|
};
|
|
chrome.storage.local.getBytesInUse(null, getBytesInUseHandler);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.saveUserSettings = function() {
|
|
chrome.storage.local.set(this.userSettings, function() {
|
|
µMatrix.getBytesInUse();
|
|
});
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.loadUserSettings = function() {
|
|
var settingsLoaded = function(store) {
|
|
// console.log('storage.js > loaded user settings');
|
|
|
|
// Ensure backward-compatibility
|
|
// https://github.com/gorhill/httpswitchboard/issues/229
|
|
if ( store.smartAutoReload === true ) {
|
|
store.smartAutoReload = 'all';
|
|
} else if ( store.smartAutoReload === false ) {
|
|
store.smartAutoReload = 'none';
|
|
}
|
|
// https://github.com/gorhill/httpswitchboard/issues/250
|
|
if ( typeof store.autoCreateSiteScope === 'boolean' ) {
|
|
store.autoCreateScope = store.autoCreateSiteScope ? 'site' : '';
|
|
delete store.autoCreateSiteScope;
|
|
}
|
|
// https://github.com/gorhill/httpswitchboard/issues/299
|
|
// No longer needed.
|
|
delete store.subframeFgColor;
|
|
|
|
µMatrix.userSettings = store;
|
|
|
|
// https://github.com/gorhill/httpswitchboard/issues/344
|
|
µMatrix.userAgentSpoofer.shuffle();
|
|
};
|
|
|
|
chrome.storage.local.get(this.userSettings, settingsLoaded);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
// save white/blacklist
|
|
µMatrix.saveMatrix = function() {
|
|
µMatrix.XAL.keyvalSetOne('userMatrix', this.pMatrix.toString());
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.loadMatrix = function() {
|
|
var µm = this;
|
|
var onLoaded = function(bin) {
|
|
if ( bin.hasOwnProperty('userMatrix') ) {
|
|
µm.pMatrix.fromString(bin.userMatrix);
|
|
µm.tMatrix.assign(µm.pMatrix);
|
|
}
|
|
};
|
|
this.XAL.keyvalGetOne('userMatrix', onLoaded);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.getAvailableHostsFiles = function(callback) {
|
|
var availableHostsFiles = {};
|
|
var redirections = {};
|
|
var µm = this;
|
|
|
|
// selected lists
|
|
var onSelectedHostsFilesLoaded = function(store) {
|
|
var lists = store.liveHostsFiles;
|
|
var locations = Object.keys(lists);
|
|
var oldLocation, newLocation;
|
|
var availableEntry, storedEntry;
|
|
|
|
while ( oldLocation = locations.pop() ) {
|
|
newLocation = redirections[oldLocation] || oldLocation;
|
|
availableEntry = availableHostsFiles[newLocation];
|
|
if ( availableEntry === undefined ) {
|
|
continue;
|
|
}
|
|
storedEntry = lists[oldLocation];
|
|
availableEntry.off = storedEntry.off || false;
|
|
µm.assets.setHomeURL(newLocation, availableEntry.homeURL);
|
|
if ( storedEntry.entryCount !== undefined ) {
|
|
availableEntry.entryCount = storedEntry.entryCount;
|
|
}
|
|
if ( storedEntry.entryUsedCount !== undefined ) {
|
|
availableEntry.entryUsedCount = storedEntry.entryUsedCount;
|
|
}
|
|
// This may happen if the list name was pulled from the list content
|
|
if ( availableEntry.title === '' && storedEntry.title !== '' ) {
|
|
availableEntry.title = storedEntry.title;
|
|
}
|
|
}
|
|
callback(availableHostsFiles);
|
|
};
|
|
|
|
// built-in lists
|
|
var onBuiltinHostsFilesLoaded = function(details) {
|
|
var location, locations;
|
|
try {
|
|
locations = JSON.parse(details.content);
|
|
} catch (e) {
|
|
locations = {};
|
|
}
|
|
var hostsFileEntry;
|
|
for ( location in locations ) {
|
|
if ( locations.hasOwnProperty(location) === false ) {
|
|
continue;
|
|
}
|
|
hostsFileEntry = locations[location];
|
|
availableHostsFiles['assets/thirdparties/' + location] = hostsFileEntry;
|
|
if ( hostsFileEntry.old !== undefined ) {
|
|
redirections[hostsFileEntry.old] = location;
|
|
delete hostsFileEntry.old;
|
|
}
|
|
}
|
|
|
|
// Now get user's selection of lists
|
|
chrome.storage.local.get(
|
|
{ 'liveHostsFiles': availableHostsFiles },
|
|
onSelectedHostsFilesLoaded
|
|
);
|
|
};
|
|
|
|
// permanent hosts files
|
|
var location;
|
|
var lists = this.permanentHostsFiles;
|
|
for ( location in lists ) {
|
|
if ( lists.hasOwnProperty(location) === false ) {
|
|
continue;
|
|
}
|
|
availableHostsFiles[location] = lists[location];
|
|
}
|
|
|
|
// custom lists
|
|
var c;
|
|
var locations = this.userSettings.externalHostsFiles.split('\n');
|
|
for ( var i = 0; i < locations.length; i++ ) {
|
|
location = locations[i].trim();
|
|
c = location.charAt(0);
|
|
if ( location === '' || c === '!' || c === '#' ) {
|
|
continue;
|
|
}
|
|
// Coarse validation
|
|
if ( /[^0-9A-Za-z!*'();:@&=+$,\/?%#\[\]_.~-]/.test(location) ) {
|
|
continue;
|
|
}
|
|
availableHostsFiles[location] = {
|
|
title: '',
|
|
external: true
|
|
};
|
|
}
|
|
|
|
// get built-in block lists.
|
|
this.assets.get('assets/umatrix/hosts-files.json', onBuiltinHostsFilesLoaded);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.loadHostsFiles = function(callback) {
|
|
var µm = µMatrix;
|
|
var hostsFileLoadCount;
|
|
|
|
if ( typeof callback !== 'function' ) {
|
|
callback = this.noopFunc;
|
|
}
|
|
|
|
var loadHostsFilesEnd = function() {
|
|
µm.ubiquitousBlacklist.freeze();
|
|
chrome.storage.local.set({ 'liveHostsFiles': µm.liveHostsFiles });
|
|
µm.messaging.announce({ what: 'loadHostsFilesCompleted' });
|
|
callback();
|
|
};
|
|
|
|
var mergeHostsFile = function(details) {
|
|
µm.mergeHostsFile(details);
|
|
hostsFileLoadCount -= 1;
|
|
if ( hostsFileLoadCount === 0 ) {
|
|
loadHostsFilesEnd();
|
|
}
|
|
};
|
|
|
|
var loadHostsFilesStart = function(hostsFiles) {
|
|
µm.liveHostsFiles = hostsFiles;
|
|
µm.ubiquitousBlacklist.reset();
|
|
var locations = Object.keys(hostsFiles);
|
|
hostsFileLoadCount = locations.length;
|
|
if ( hostsFileLoadCount === 0 ) {
|
|
loadHostsFilesEnd();
|
|
return;
|
|
}
|
|
|
|
// Load all hosts file which are not disabled.
|
|
var location;
|
|
while ( location = locations.pop() ) {
|
|
if ( hostsFiles[location].off ) {
|
|
hostsFileLoadCount -= 1;
|
|
continue;
|
|
}
|
|
µm.assets.get(location, mergeHostsFile);
|
|
}
|
|
};
|
|
|
|
this.getAvailableHostsFiles(loadHostsFilesStart);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.mergeHostsFile = function(details) {
|
|
// console.log('storage.js > mergeHostsFile from "%s": "%s..."', details.path, details.content.slice(0, 40));
|
|
|
|
var usedCount = this.ubiquitousBlacklist.count;
|
|
var duplicateCount = this.ubiquitousBlacklist.duplicateCount;
|
|
|
|
this.mergeHostsFileContent(details.content);
|
|
|
|
usedCount = this.ubiquitousBlacklist.count - usedCount;
|
|
duplicateCount = this.ubiquitousBlacklist.duplicateCount - duplicateCount;
|
|
|
|
var hostsFilesMeta = this.liveHostsFiles[details.path];
|
|
hostsFilesMeta.entryCount = usedCount + duplicateCount;
|
|
hostsFilesMeta.entryUsedCount = usedCount;
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.mergeHostsFileContent = function(rawText) {
|
|
// console.log('storage.js > mergeHostsFileContent from "%s": "%s..."', details.path, details.content.slice(0, 40));
|
|
|
|
var rawEnd = rawText.length;
|
|
var ubiquitousBlacklist = this.ubiquitousBlacklist;
|
|
var reLocalhost = /(^|\s)(localhost\.localdomain|localhost|local|broadcasthost|0\.0\.0\.0|127\.0\.0\.1|::1|fe80::1%lo0)(?=\s|$)/g;
|
|
var reAsciiSegment = /^[\x21-\x7e]+$/;
|
|
var matches;
|
|
var lineBeg = 0, lineEnd;
|
|
var line;
|
|
|
|
while ( lineBeg < rawEnd ) {
|
|
lineEnd = rawText.indexOf('\n', lineBeg);
|
|
if ( lineEnd < 0 ) {
|
|
lineEnd = rawText.indexOf('\r', lineBeg);
|
|
if ( lineEnd < 0 ) {
|
|
lineEnd = rawEnd;
|
|
}
|
|
}
|
|
|
|
// rhill 2014-04-18: The trim is important here, as without it there
|
|
// could be a lingering `\r` which would cause problems in the
|
|
// following parsing code.
|
|
line = rawText.slice(lineBeg, lineEnd).trim();
|
|
lineBeg = lineEnd + 1;
|
|
|
|
// https://github.com/gorhill/httpswitchboard/issues/15
|
|
// Ensure localhost et al. don't end up in the ubiquitous blacklist.
|
|
line = line
|
|
.replace(/#.*$/, '')
|
|
.toLowerCase()
|
|
.replace(reLocalhost, '')
|
|
.trim();
|
|
|
|
// The filter is whatever sequence of printable ascii character without
|
|
// whitespaces
|
|
matches = reAsciiSegment.exec(line);
|
|
if ( !matches || matches.length === 0 ) {
|
|
continue;
|
|
}
|
|
|
|
// Bypass anomalies
|
|
// For example, when a filter contains whitespace characters, or
|
|
// whatever else outside the range of printable ascii characters.
|
|
if ( matches[0] !== line ) {
|
|
// console.error('"%s": "%s" !== "%s"', details.path, matches[0], line);
|
|
continue;
|
|
}
|
|
|
|
line = matches[0];
|
|
if ( line === '' ) {
|
|
continue;
|
|
}
|
|
|
|
ubiquitousBlacklist.add(line);
|
|
}
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
// `switches` contains the preset blacklists for which the switch must be
|
|
// revisited.
|
|
|
|
µMatrix.reloadHostsFiles = function(switches) {
|
|
var liveHostsFiles = this.liveHostsFiles;
|
|
|
|
// Toggle switches
|
|
var i = switches.length;
|
|
while ( i-- ) {
|
|
if ( !liveHostsFiles[switches[i].location] ) {
|
|
continue;
|
|
}
|
|
liveHostsFiles[switches[i].location].off = !!switches[i].off;
|
|
}
|
|
|
|
// Save switch states
|
|
chrome.storage.local.set(
|
|
{ 'liveHostsFiles': liveHostsFiles },
|
|
this.loadHostsFiles.bind(this)
|
|
);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
µMatrix.loadPublicSuffixList = function() {
|
|
var applyPublicSuffixList = function(details) {
|
|
// TODO: Not getting proper suffix list is a bit serious, I think
|
|
// the extension should be force-restarted if it occurs..
|
|
if ( !details.error ) {
|
|
publicSuffixList.parse(details.content, punycode.toASCII);
|
|
}
|
|
};
|
|
this.assets.get(
|
|
'assets/thirdparties/publicsuffix.org/list/effective_tld_names.dat',
|
|
applyPublicSuffixList
|
|
);
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
// Load updatable assets
|
|
|
|
µMatrix.loadUpdatableAssets = function() {
|
|
this.loadHostsFiles();
|
|
this.loadPublicSuffixList();
|
|
};
|
|
|
|
/******************************************************************************/
|
|
|
|
// Load all
|
|
|
|
µMatrix.load = function() {
|
|
this.loadUserSettings();
|
|
this.loadMatrix();
|
|
this.loadUpdatableAssets();
|
|
|
|
this.getBytesInUse();
|
|
};
|
|
|