Mirrors
/
uBlock
mirror of https://github.com/gorhill/uBlock.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Rework generic cosmetic filtering code 

Related issue:
- https://github.com/uBlockOrigin/uBlock-issues/issues/2248
This commit is contained in:
Raymond Hill 2022-12-07 10:30:09 -05:00
parent 76d70102f0
commit 26594fb902
No known key found for this signature in database
GPG Key ID: 25E1490B761470C2
4 changed files with 351 additions and 550 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/js/background.js
View File

@ -176,8 +176,8 @@ const µBlock = { // jshint ignore:line
// Read-only // Read-only
systemSettings: { systemSettings: {
compiledMagic: 47, // Increase when compiled format changes compiledMagic: 48, // Increase when compiled format changes
selfieMagic: 47, // Increase when selfie format changes selfieMagic: 48, // Increase when selfie format changes
}, },
// https://github.com/uBlockOrigin/uBlock-issues/issues/759#issuecomment-546654501 // https://github.com/uBlockOrigin/uBlock-issues/issues/759#issuecomment-546654501

358
src/js/contentscript.js
View File

@ -945,72 +945,65 @@ vAPI.DOMFilterer = class {
// vAPI.domSurveyor // vAPI.domSurveyor
{ {
const messaging = vAPI.messaging; // https://werxltd.com/wp/2010/05/13/javascript-implementation-of-javas-string-hashcode-method/
const queriedIds = new Set(); // Must mirror cosmetic filtering compiler's version
const queriedClasses = new Set(); const hashFromStr = (type, s) => {
const len = s.length;
const step = len + 7 >>> 3;
let hash = (type << 5) - type + (len & 0xFF) | 0;
for ( let i = 0; i < len; i += step ) {
hash = (hash << 5) - hash + s.charCodeAt(i) | 0;
}
return hash & 0xFFFFFF;
};
const addHashes = hashes => {
for ( const hash of hashes ) {
queriedHashes.add(hash);
}
};
const queriedHashes = new Set();
const maxSurveyNodes = 65536; const maxSurveyNodes = 65536;
const maxSurveyTimeSlice = 4; const pendingLists = [];
const maxSurveyBuffer = 64; const pendingNodes = [];
const processedSet = new Set();
let domFilterer;
let hostname = '';
let domChanged = false;
let scannedCount = 0;
let stopped = false;
let domFilterer, const addPendingList = list => {
hostname = '', if ( list.length === 0 ) { return; }
surveyCost = 0; pendingLists.push(Array.from(list));
};
const pendingNodes = { const nextPendingNodes = ( ) => {
nodeLists: [], if ( pendingLists.length === 0 ) { return 0; }
buffer: [ const bufferSize = 256;
null, null, null, null, null, null, null, null, let j = 0;
null, null, null, null, null, null, null, null, do {
null, null, null, null, null, null, null, null, const nodeList = pendingLists[0];
null, null, null, null, null, null, null, null, let n = bufferSize - j;
null, null, null, null, null, null, null, null, if ( n > nodeList.length ) {
null, null, null, null, null, null, null, null, n = nodeList.length;
null, null, null, null, null, null, null, null,
null, null, null, null, null, null, null, null,
],
j: 0,
accepted: 0,
iterated: 0,
stopped: false,
add(nodes) {
if ( nodes.length === 0 || this.accepted >= maxSurveyNodes ) {
return;
} }
this.nodeLists.push(nodes); for ( let i = 0; i < n; i++ ) {
this.accepted += nodes.length; pendingNodes[j+i] = nodeList[i];
},
next() {
if ( this.nodeLists.length === 0 || this.stopped ) { return 0; }
const nodeLists = this.nodeLists;
let ib = 0;
do {
const nodeList = nodeLists[0];
let j = this.j;
let n = j + maxSurveyBuffer - ib;
if ( n > nodeList.length ) {
n = nodeList.length;
}
for ( let i = j; i < n; i++ ) {
this.buffer[ib++] = nodeList[j++];
}
if ( j !== nodeList.length ) {
this.j = j;
break;
}
this.j = 0;
this.nodeLists.shift();
} while ( ib < maxSurveyBuffer && nodeLists.length !== 0 );
this.iterated += ib;
if ( this.iterated >= maxSurveyNodes ) {
this.nodeLists = [];
this.stopped = true;
//console.info(`domSurveyor> Surveyed a total of ${this.iterated} nodes. Enough.`);
} }
return ib; j += n;
}, if ( n !== nodeList.length ) {
hasNodes() { pendingLists[0] = nodeList.slice(n);
return this.nodeLists.length !== 0; break;
}, }
pendingLists.shift();
} while ( j < bufferSize && pendingLists.length !== 0 );
return j;
};
const hasPendingNodes = ( ) => {
return pendingLists.length !== 0;
}; };
// Extract all classes/ids: these will be passed to the cosmetic // Extract all classes/ids: these will be passed to the cosmetic
@ -1024,10 +1017,10 @@ vAPI.DOMFilterer = class {
const idFromNode = (node, out) => { const idFromNode = (node, out) => {
const raw = node.id; const raw = node.id;
if ( typeof raw !== 'string' || raw.length === 0 ) { return; } if ( typeof raw !== 'string' || raw.length === 0 ) { return; }
const s = raw.trim(); const hash = hashFromStr(0x23 /* '#' */, raw.trim());
if ( queriedIds.has(s) || s.length === 0 ) { return; } if ( queriedHashes.has(hash) ) { return; }
out.push(s); queriedHashes.add(hash);
queriedIds.add(s); out.push(hash);
}; };
// https://github.com/uBlockOrigin/uBlock-issues/discussions/2076 // https://github.com/uBlockOrigin/uBlock-issues/discussions/2076
@ -1036,73 +1029,83 @@ vAPI.DOMFilterer = class {
const s = node.getAttribute('class'); const s = node.getAttribute('class');
if ( typeof s !== 'string' ) { return; } if ( typeof s !== 'string' ) { return; }
const len = s.length; const len = s.length;
for ( let beg = 0, end = 0, token = ''; beg < len; beg += 1 ) { for ( let beg = 0, end = 0; beg < len; beg += 1 ) {
end = s.indexOf(' ', beg); end = s.indexOf(' ', beg);
if ( end === beg ) { continue; } if ( end === beg ) { continue; }
if ( end === -1 ) { end = len; } if ( end === -1 ) { end = len; }
token = s.slice(beg, end); const hash = hashFromStr(0x2E /* '.' */, s.slice(beg, end));
beg = end; beg = end;
if ( queriedClasses.has(token) ) { continue; } if ( queriedHashes.has(hash) ) { continue; }
out.push(token); queriedHashes.add(hash);
queriedClasses.add(token); out.push(hash);
} }
}; };
const surveyPhase1 = function() { const getSurveyResults = hashes => {
//console.time('dom surveyor/surveying'); if ( self.vAPI.messaging instanceof Object === false ) {
stop(); return;
}
const promise = hashes.length === 0
? Promise.resolve(null)
: self.vAPI.messaging.send('contentscript', {
what: 'retrieveGenericCosmeticSelectors',
hostname,
hashes,
exceptions: domFilterer.exceptions,
});
promise.then(response => {
processSurveyResults(response);
});
};
const doSurvey = ( ) => {
const t0 = performance.now(); const t0 = performance.now();
const ids = []; const hashes = [];
const classes = []; const nodes = pendingNodes;
const nodes = pendingNodes.buffer; const deadline = t0 + 4;
const deadline = t0 + maxSurveyTimeSlice;
let processed = 0; let processed = 0;
let scanned = 0;
for (;;) { for (;;) {
const n = pendingNodes.next(); const n = nextPendingNodes();
if ( n === 0 ) { break; } if ( n === 0 ) { break; }
for ( let i = 0; i < n; i++ ) { for ( let i = 0; i < n; i++ ) {
const node = nodes[i]; nodes[i] = null; const node = nodes[i]; nodes[i] = null;
idFromNode(node, ids); if ( domChanged ) {
classesFromNode(node, classes); if ( processedSet.has(node) ) { continue; }
processedSet.add(node);
}
idFromNode(node, hashes);
classesFromNode(node, hashes);
scanned += 1;
} }
processed += n; processed += n;
if ( performance.now() >= deadline ) { break; } if ( performance.now() >= deadline ) { break; }
} }
const t1 = performance.now(); //console.info(`[domSurveyor][${hostname}] Surveyed ${scanned}/${processed} nodes in ${(performance.now()-t0).toFixed(2)} ms: ${hashes.length} hashes`);
surveyCost += t1 - t0; scannedCount += scanned;
//console.info(`domSurveyor> Surveyed ${processed} nodes in ${(t1-t0).toFixed(2)} ms`); if ( scannedCount >= maxSurveyNodes ) {
// Phase 2: Ask main process to lookup relevant cosmetic filters. stop();
if ( ids.length !== 0 || classes.length !== 0 ) {
messaging.send('contentscript', {
what: 'retrieveGenericCosmeticSelectors',
hostname,
ids, classes,
exceptions: domFilterer.exceptions,
cost: surveyCost,
}).then(response => {
surveyPhase3(response);
});
} else {
surveyPhase3(null);
} }
//console.timeEnd('dom surveyor/surveying'); processedSet.clear();
getSurveyResults(hashes);
}; };
const surveyTimer = new vAPI.SafeAnimationFrame(surveyPhase1); const surveyTimer = new vAPI.SafeAnimationFrame(doSurvey);
// This is to shutdown the surveyor if result of surveying keeps being // This is to shutdown the surveyor if result of surveying keeps being
// fruitless. This is useful on long-lived web page. I arbitrarily // fruitless. This is useful on long-lived web page. I arbitrarily
// picked 5 minutes before the surveyor is allowed to shutdown. I also // picked 5 minutes before the surveyor is allowed to shutdown. I also
// arbitrarily picked 256 misses before the surveyor is allowed to // arbitrarily picked 256 misses before the surveyor is allowed to
// shutdown. // shutdown.
let canShutdownAfter = Date.now() + 300000, let canShutdownAfter = Date.now() + 300000;
surveyingMissCount = 0; let surveyResultMissCount = 0;
// Handle main process' response. // Handle main process' response.
const surveyPhase3 = function(response) { const processSurveyResults = response => {
if ( stopped ) { return; }
const result = response && response.result; const result = response && response.result;
let mustCommit = false; let mustCommit = false;
if ( result ) { if ( result ) {
const css = result.injectedCSS; const css = result.injectedCSS;
if ( typeof css === 'string' && css.length !== 0 ) { if ( typeof css === 'string' && css.length !== 0 ) {
@ -1114,99 +1117,86 @@ vAPI.DOMFilterer = class {
domFilterer.exceptCSSRules(selectors); domFilterer.exceptCSSRules(selectors);
} }
} }
if ( hasPendingNodes() ) {
if ( pendingNodes.stopped === false ) { surveyTimer.start(1);
if ( pendingNodes.hasNodes() ) {
surveyTimer.start(1);
}
if ( mustCommit ) {
surveyingMissCount = 0;
canShutdownAfter = Date.now() + 300000;
return;
}
surveyingMissCount += 1;
if ( surveyingMissCount < 256 || Date.now() < canShutdownAfter ) {
return;
}
} }
if ( mustCommit ) {
//console.info('dom surveyor shutting down: too many misses'); surveyResultMissCount = 0;
canShutdownAfter = Date.now() + 300000;
surveyTimer.clear(); return;
vAPI.domWatcher.removeListener(domWatcherInterface); }
vAPI.domSurveyor = null; surveyResultMissCount += 1;
if ( surveyResultMissCount < 256 || Date.now() < canShutdownAfter ) {
return;
}
//console.info(`[domSurveyor][${hostname}] Shutting down, too many misses`);
stop();
self.vAPI.messaging.send('contentscript', {
what: 'disableGenericCosmeticFilteringSurveyor',
hostname,
});
}; };
const domWatcherInterface = { const domWatcherInterface = {
onDOMCreated: function() { onDOMCreated: function() {
if (
self.vAPI instanceof Object === false ||
vAPI.domSurveyor instanceof Object === false ||
vAPI.domFilterer instanceof Object === false
) {
if ( self.vAPI instanceof Object ) {
if ( vAPI.domWatcher instanceof Object ) {
vAPI.domWatcher.removeListener(domWatcherInterface);
}
vAPI.domSurveyor = null;
}
return;
}
//console.time('dom surveyor/dom layout created');
domFilterer = vAPI.domFilterer; domFilterer = vAPI.domFilterer;
pendingNodes.add(document.querySelectorAll(
'[id]:not(html):not(body),[class]:not(html):not(body)'
));
surveyTimer.start();
// https://github.com/uBlockOrigin/uBlock-issues/issues/1692 // https://github.com/uBlockOrigin/uBlock-issues/issues/1692
// Look-up safe-only selectors to mitigate probability of // Look-up safe-only selectors to mitigate probability of
// html/body elements of erroneously being targeted. // html/body elements of erroneously being targeted.
const ids = [], classes = []; const hashes = [];
if ( document.documentElement !== null ) { if ( document.documentElement !== null ) {
idFromNode(document.documentElement, ids); idFromNode(document.documentElement, hashes);
classesFromNode(document.documentElement, classes); classesFromNode(document.documentElement, hashes);
} }
if ( document.body !== null ) { if ( document.body !== null ) {
idFromNode(document.body, ids); idFromNode(document.body, hashes);
classesFromNode(document.body, classes); classesFromNode(document.body, hashes);
} }
if ( ids.length !== 0 || classes.length !== 0 ) { addPendingList(document.querySelectorAll(
messaging.send('contentscript', { '[id]:not(html):not(body),[class]:not(html):not(body)'
what: 'retrieveGenericCosmeticSelectors', ));
hostname, if ( hasPendingNodes() ) {
ids, classes, surveyTimer.start();
exceptions: domFilterer.exceptions,
safeOnly: true,
}).then(response => {
surveyPhase3(response);
});
} }
//console.timeEnd('dom surveyor/dom layout created');
}, },
onDOMChanged: function(addedNodes) { onDOMChanged: function(addedNodes) {
if ( addedNodes.length === 0 ) { return; } if ( addedNodes.length === 0 ) { return; }
//console.time('dom surveyor/dom layout changed'); domChanged = true;
for ( const node of addedNodes ) { for ( const node of addedNodes ) {
pendingNodes.add([ node ]); addPendingList([ node ]);
if ( node.firstElementChild === null ) { continue; } if ( node.firstElementChild === null ) { continue; }
pendingNodes.add(node.querySelectorAll( addPendingList(
'[id]:not(html):not(body),[class]:not(html):not(body)' node.querySelectorAll(
)); '[id]:not(html):not(body),[class]:not(html):not(body)'
)
);
} }
if ( pendingNodes.hasNodes() ) { if ( hasPendingNodes() ) {
surveyTimer.start(1); surveyTimer.start(1);
} }
//console.timeEnd('dom surveyor/dom layout changed');
} }
}; };
const start = function(details) { const start = details => {
if ( vAPI.domWatcher instanceof Object === false ) { return; } if ( self.vAPI instanceof Object === false ) { return; }
if ( self.vAPI.domFilterer instanceof Object === false ) { return; }
if ( self.vAPI.domWatcher instanceof Object === false ) { return; }
hostname = details.hostname; hostname = details.hostname;
vAPI.domWatcher.addListener(domWatcherInterface); self.vAPI.domWatcher.addListener(domWatcherInterface);
}; };
vAPI.domSurveyor = { start }; const stop = ( ) => {
stopped = true;
pendingLists.length = 0;
surveyTimer.clear();
if ( self.vAPI instanceof Object === false ) { return; }
if ( self.vAPI.domWatcher instanceof Object ) {
self.vAPI.domWatcher.removeListener(domWatcherInterface);
}
self.vAPI.domSurveyor = null;
};
self.vAPI.domSurveyor = { start, addHashes };
} }
/******************************************************************************/ /******************************************************************************/
@ -1218,7 +1208,7 @@ vAPI.DOMFilterer = class {
// to be launched if/when needed. // to be launched if/when needed.
{ {
const bootstrapPhase2 = function() { const onDomReady = ( ) => {
// This can happen on Firefox. For instance: // This can happen on Firefox. For instance:
// https://github.com/gorhill/uBlock/issues/1893 // https://github.com/gorhill/uBlock/issues/1893
if ( window.location === null ) { return; } if ( window.location === null ) { return; }
@ -1279,9 +1269,8 @@ vAPI.DOMFilterer = class {
// an object -- let's stay around, we may be given the opportunity // an object -- let's stay around, we may be given the opportunity
// to try bootstrapping again later. // to try bootstrapping again later.
const bootstrapPhase1 = function(response) { const onResponseReady = response => {
if ( response instanceof Object === false ) { return; } if ( response instanceof Object === false ) { return; }
vAPI.bootstrap = undefined; vAPI.bootstrap = undefined;
// cosmetic filtering engine aka 'cfe' // cosmetic filtering engine aka 'cfe'
@ -1308,7 +1297,7 @@ vAPI.DOMFilterer = class {
vAPI.domSurveyor = null; vAPI.domSurveyor = null;
} else { } else {
const domFilterer = vAPI.domFilterer = new vAPI.DOMFilterer(); const domFilterer = vAPI.domFilterer = new vAPI.DOMFilterer();
if ( noGenericCosmeticFiltering || cfeDetails.noDOMSurveying ) { if ( noGenericCosmeticFiltering || cfeDetails.disableSurveyor ) {
vAPI.domSurveyor = null; vAPI.domSurveyor = null;
} }
domFilterer.exceptions = cfeDetails.exceptionFilters; domFilterer.exceptions = cfeDetails.exceptionFilters;
@ -1316,10 +1305,9 @@ vAPI.DOMFilterer = class {
domFilterer.addProceduralSelectors(cfeDetails.proceduralFilters); domFilterer.addProceduralSelectors(cfeDetails.proceduralFilters);
domFilterer.exceptCSSRules(cfeDetails.exceptedFilters); domFilterer.exceptCSSRules(cfeDetails.exceptedFilters);
domFilterer.convertedProceduralFilters = cfeDetails.convertedProceduralFilters; domFilterer.convertedProceduralFilters = cfeDetails.convertedProceduralFilters;
vAPI.userStylesheet.apply();
} }
vAPI.userStylesheet.apply();
// Library of resources is located at: // Library of resources is located at:
// https://github.com/gorhill/uBlock/blob/master/assets/ublock/resources.txt // https://github.com/gorhill/uBlock/blob/master/assets/ublock/resources.txt
if ( scriptlets && typeof self.uBO_scriptletsInjected !== 'boolean' ) { if ( scriptlets && typeof self.uBO_scriptletsInjected !== 'boolean' ) {
@ -1328,26 +1316,18 @@ vAPI.DOMFilterer = class {
vAPI.injectedScripts = scriptlets; vAPI.injectedScripts = scriptlets;
} }
if ( vAPI.domSurveyor instanceof Object ) { if ( vAPI.domSurveyor ) {
if ( Array.isArray(cfeDetails.genericCosmeticHashes) ) {
vAPI.domSurveyor.addHashes(cfeDetails.genericCosmeticHashes);
}
vAPI.domSurveyor.start(cfeDetails); vAPI.domSurveyor.start(cfeDetails);
} }
// https://github.com/chrisaljoudi/uBlock/issues/587 const readyState = document.readyState;
// If no filters were found, maybe the script was injected before if ( readyState === 'interactive' || readyState === 'complete' ) {
// uBlock's process was fully initialized. When this happens, pages return onDomReady();
// won't be cleaned right after browser launch.
if (
typeof document.readyState === 'string' &&
document.readyState !== 'loading'
) {
bootstrapPhase2();
} else {
document.addEventListener(
'DOMContentLoaded',
bootstrapPhase2,
{ once: true }
);
} }
document.addEventListener('DOMContentLoaded', onDomReady, { once: true });
}; };
vAPI.bootstrap = function() { vAPI.bootstrap = function() {
@ -1356,7 +1336,7 @@ vAPI.DOMFilterer = class {
url: vAPI.effectiveSelf.location.href, url: vAPI.effectiveSelf.location.href,
needScriptlets: typeof self.uBO_scriptletsInjected !== 'boolean', needScriptlets: typeof self.uBO_scriptletsInjected !== 'boolean',
}).then(response => { }).then(response => {
bootstrapPhase1(response); onResponseReady(response);
}); });
}; };
} }

535
src/js/cosmetic-filtering.js
View File

@ -32,12 +32,6 @@ import {
StaticExtFilteringSessionDB, StaticExtFilteringSessionDB,
} from './static-ext-filtering-db.js'; } from './static-ext-filtering-db.js';
/******************************************************************************/
const cosmeticSurveyingMissCountMax =
parseInt(vAPI.localStorage.getItem('cosmeticSurveyingMissCountMax'), 10) ||
15;
/******************************************************************************/ /******************************************************************************/
/******************************************************************************/ /******************************************************************************/
@ -48,71 +42,55 @@ const SelectorCacheEntry = class {
reset() { reset() {
this.cosmetic = new Set(); this.cosmetic = new Set();
this.cosmeticSurveyingMissCount = 0; this.cosmeticHashes = new Set();
this.disableSurveyor = false;
this.net = new Map(); this.net = new Map();
this.lastAccessTime = Date.now(); this.accessId = SelectorCacheEntry.accessId++;
return this; return this;
} }
dispose() { dispose() {
this.cosmetic = this.net = null; this.cosmetic = this.cosmeticHashes = this.net = null;
if ( SelectorCacheEntry.junkyard.length < 25 ) { if ( SelectorCacheEntry.junkyard.length < 25 ) {
SelectorCacheEntry.junkyard.push(this); SelectorCacheEntry.junkyard.push(this);
} }
} }
addCosmetic(details) { addCosmetic(details) {
const selectors = details.selectors; const selectors = details.selectors.join(',\n');
let i = selectors.length || 0; if ( selectors.length !== 0 ) {
// https://github.com/gorhill/uBlock/issues/2011 this.cosmetic.add(selectors);
// Avoiding seemingly pointless surveys only if they appear costly.
if ( details.first && i === 0 ) {
if ( (details.cost || 0) >= 80 ) {
this.cosmeticSurveyingMissCount += 1;
}
return;
} }
this.cosmeticSurveyingMissCount = 0; for ( const hash of details.hashes ) {
while ( i-- ) { this.cosmeticHashes.add(hash);
this.cosmetic.add(selectors[i]);
} }
} }
addNet(selectors) { addNet(selectors) {
if ( typeof selectors === 'string' ) { if ( typeof selectors === 'string' ) {
this.addNetOne(selectors, Date.now()); this.net.set(selectors, this.accessId);
} else { } else {
this.addNetMany(selectors, Date.now()); this.net.set(selectors.join(',\n'), this.accessId);
} }
// Net request-derived selectors: I limit the number of cached // Net request-derived selectors: I limit the number of cached
// selectors, as I expect cases where the blocked net-requests // selectors, as I expect cases where the blocked network requests
// are never the exact same URL. // are never the exact same URL.
if ( this.net.size < SelectorCacheEntry.netHighWaterMark ) { if ( this.net.size < SelectorCacheEntry.netHighWaterMark ) { return; }
return; const keys = Array.from(this.net)
} .sort((a, b) => b[1] - a[1])
const dict = this.net; .slice(SelectorCacheEntry.netLowWaterMark)
const keys = Array.from(dict.keys()).sort(function(a, b) { .map(a => a[0]);
return dict.get(b) - dict.get(a); for ( const key of keys ) {
}).slice(SelectorCacheEntry.netLowWaterMark); this.net.delete(key);
let i = keys.length;
while ( i-- ) {
dict.delete(keys[i]);
} }
} }
addNetOne(selector, now) { addNetOne(selector, token) {
this.net.set(selector, now); this.net.set(selector, token);
}
addNetMany(selectors, now) {
let i = selectors.length || 0;
while ( i-- ) {
this.net.set(selectors[i], now);
}
} }
add(details) { add(details) {
this.lastAccessTime = Date.now(); this.accessId = SelectorCacheEntry.accessId++;
if ( details.type === 'cosmetic' ) { if ( details.type === 'cosmetic' ) {
this.addCosmetic(details); this.addCosmetic(details);
} else { } else {
@ -122,10 +100,9 @@ const SelectorCacheEntry = class {
// https://github.com/chrisaljoudi/uBlock/issues/420 // https://github.com/chrisaljoudi/uBlock/issues/420
remove(type) { remove(type) {
this.lastAccessTime = Date.now(); this.accessId = SelectorCacheEntry.accessId++;
if ( type === undefined || type === 'cosmetic' ) { if ( type === undefined || type === 'cosmetic' ) {
this.cosmetic.clear(); this.cosmetic.clear();
this.cosmeticSurveyingMissCount = 0;
} }
if ( type === undefined || type === 'net' ) { if ( type === undefined || type === 'net' ) {
this.net.clear(); this.net.clear();
@ -133,36 +110,41 @@ const SelectorCacheEntry = class {
} }
retrieveToArray(iterator, out) { retrieveToArray(iterator, out) {
for ( let selector of iterator ) { for ( const selector of iterator ) {
out.push(selector); out.push(selector);
} }
} }
retrieveToSet(iterator, out) { retrieveToSet(iterator, out) {
for ( let selector of iterator ) { for ( const selector of iterator ) {
out.add(selector); out.add(selector);
} }
} }
retrieve(type, out) { retrieveNet(out) {
this.lastAccessTime = Date.now(); this.accessId = SelectorCacheEntry.accessId++;
const iterator = type === 'cosmetic' ? this.cosmetic : this.net.keys(); if ( this.net.size === 0 ) { return false; }
if ( Array.isArray(out) ) { this.retrieveToArray(this.net.keys(), out);
this.retrieveToArray(iterator, out); return true;
} else { }
this.retrieveToSet(iterator, out);
} retrieveCosmetic(selectors, hashes) {
this.accessId = SelectorCacheEntry.accessId++;
if ( this.cosmetic.size === 0 ) { return false; }
this.retrieveToSet(this.cosmetic, selectors);
this.retrieveToArray(this.cosmeticHashes, hashes);
return true;
} }
static factory() { static factory() {
const entry = SelectorCacheEntry.junkyard.pop(); const entry = SelectorCacheEntry.junkyard.pop();
if ( entry ) { return entry
return entry.reset(); ? entry.reset()
} : new SelectorCacheEntry();
return new SelectorCacheEntry();
} }
}; };
SelectorCacheEntry.accessId = 1;
SelectorCacheEntry.netLowWaterMark = 20; SelectorCacheEntry.netLowWaterMark = 20;
SelectorCacheEntry.netHighWaterMark = 30; SelectorCacheEntry.netHighWaterMark = 30;
SelectorCacheEntry.junkyard = []; SelectorCacheEntry.junkyard = [];
@ -170,6 +152,61 @@ SelectorCacheEntry.junkyard = [];
/******************************************************************************/ /******************************************************************************/
/******************************************************************************/ /******************************************************************************/
// https://werxltd.com/wp/2010/05/13/javascript-implementation-of-javas-string-hashcode-method/
// Must mirror content script surveyor's version
const hashFromStr = (type, s) => {
const len = s.length;
const step = len + 7 >>> 3;
let hash = (type << 5) - type + (len & 0xFF) | 0;
for ( let i = 0; i < len; i += step ) {
hash = (hash << 5) - hash + s.charCodeAt(i) | 0;
}
return hash & 0xFFFFFF;
};
// https://github.com/gorhill/uBlock/issues/1668
// The key must be literal: unescape escaped CSS before extracting key.
// It's an uncommon case, so it's best to unescape only when needed.
const keyFromSelector = selector => {
let matches = rePlainSelector.exec(selector);
if ( matches === null ) {
matches = rePlainSelectorEx.exec(selector);
if ( matches !== null ) { return matches[1] || matches[2]; }
return;
}
let key = matches[0];
if ( key.includes('\\') === false ) { return key; }
matches = rePlainSelectorEscaped.exec(selector);
if ( matches === null ) { return; }
key = '';
const escaped = matches[0];
let beg = 0;
reEscapeSequence.lastIndex = 0;
for (;;) {
matches = reEscapeSequence.exec(escaped);
if ( matches === null ) {
return key + escaped.slice(beg);
}
key += escaped.slice(beg, matches.index);
beg = reEscapeSequence.lastIndex;
if ( matches[1].length === 1 ) {
key += matches[1];
} else {
key += String.fromCharCode(parseInt(matches[1], 16));
}
}
};
const rePlainSelector = /^[#.][\w\\-]+/;
const rePlainSelectorEx = /^[^#.\[(]+([#.][\w-]+)|([#.][\w-]+)$/;
const rePlainSelectorEscaped = /^[#.](?:\\[0-9A-Fa-f]+ |\\.|\w|-)+/;
const reEscapeSequence = /\\([0-9A-Fa-f]+ |.)/g;
/******************************************************************************/
/******************************************************************************/
// Cosmetic filter family tree: // Cosmetic filter family tree:
// //
// Generic // Generic
@ -186,18 +223,12 @@ SelectorCacheEntry.junkyard = [];
// Specific filers can be enforced before the main document is loaded. // Specific filers can be enforced before the main document is loaded.
const FilterContainer = function() { const FilterContainer = function() {
this.rePlainSelector = /^[#.][\w\\-]+/;
this.rePlainSelectorEscaped = /^[#.](?:\\[0-9A-Fa-f]+ |\\.|\w|-)+/;
this.rePlainSelectorEx = /^[^#.\[(]+([#.][\w-]+)|([#.][\w-]+)$/;
this.reEscapeSequence = /\\([0-9A-Fa-f]+ |.)/g;
this.reSimpleHighGeneric = /^(?:[a-z]*\[[^\]]+\]|\S+)$/; this.reSimpleHighGeneric = /^(?:[a-z]*\[[^\]]+\]|\S+)$/;
this.reHighMedium = /^\[href\^="https?:\/\/([^"]{8})[^"]*"\]$/;
this.selectorCache = new Map(); this.selectorCache = new Map();
this.selectorCachePruneDelay = 10 * 60 * 1000; // 10 minutes this.selectorCachePruneDelay = 10 * 60 * 1000; // 10 minutes
this.selectorCacheAgeMax = 120 * 60 * 1000; // 120 minutes this.selectorCacheCountMin = 40;
this.selectorCacheCountMin = 25; this.selectorCacheCountMax = 50;
this.netSelectorCacheCountMax = SelectorCacheEntry.netHighWaterMark;
this.selectorCacheTimer = null; this.selectorCacheTimer = null;
// specific filters // specific filters
@ -206,20 +237,8 @@ const FilterContainer = function() {
// temporary filters // temporary filters
this.sessionFilterDB = new StaticExtFilteringSessionDB(); this.sessionFilterDB = new StaticExtFilteringSessionDB();
// low generic cosmetic filters, organized by id/class then simple/complex. // low generic cosmetic filters: map of hash => array of selectors
this.lowlyGeneric = Object.create(null); this.lowlyGeneric = new Map();
this.lowlyGeneric.id = {
canonical: 'ids',
prefix: '#',
simple: new Set(),
complex: new Map()
};
this.lowlyGeneric.cl = {
canonical: 'classes',
prefix: '.',
simple: new Set(),
complex: new Map()
};
// highly generic selectors sets // highly generic selectors sets
this.highlyGeneric = Object.create(null); this.highlyGeneric = Object.create(null);
@ -240,8 +259,6 @@ const FilterContainer = function() {
// is to prevent repeated allocation/deallocation overheads -- the // is to prevent repeated allocation/deallocation overheads -- the
// constructors/destructors of javascript Set/Map is assumed to be costlier // constructors/destructors of javascript Set/Map is assumed to be costlier
// than just calling clear() on these. // than just calling clear() on these.
this.$simpleSet = new Set();
this.$complexSet = new Set();
this.$specificSet = new Set(); this.$specificSet = new Set();
this.$exceptionSet = new Set(); this.$exceptionSet = new Set();
this.$proceduralSet = new Set(); this.$proceduralSet = new Set();
@ -266,17 +283,11 @@ FilterContainer.prototype.reset = function() {
this.selectorCacheTimer = null; this.selectorCacheTimer = null;
} }
// whether there is at least one surveyor-based filter
this.needDOMSurveyor = false;
// hostname, entity-based filters // hostname, entity-based filters
this.specificFilters.clear(); this.specificFilters.clear();
// low generic cosmetic filters, organized by id/class then simple/complex. // low generic cosmetic filters
this.lowlyGeneric.id.simple.clear(); this.lowlyGeneric.clear();
this.lowlyGeneric.id.complex.clear();
this.lowlyGeneric.cl.simple.clear();
this.lowlyGeneric.cl.complex.clear();
// highly generic selectors sets // highly generic selectors sets
this.highlyGeneric.simple.dict.clear(); this.highlyGeneric.simple.dict.clear();
@ -285,6 +296,8 @@ FilterContainer.prototype.reset = function() {
this.highlyGeneric.complex.dict.clear(); this.highlyGeneric.complex.dict.clear();
this.highlyGeneric.complex.str = ''; this.highlyGeneric.complex.str = '';
this.highlyGeneric.complex.mru.reset(); this.highlyGeneric.complex.mru.reset();
this.selfieVersion = 1;
}; };
/******************************************************************************/ /******************************************************************************/
@ -293,12 +306,6 @@ FilterContainer.prototype.freeze = function() {
this.duplicateBuster.clear(); this.duplicateBuster.clear();
this.specificFilters.collectGarbage(); this.specificFilters.collectGarbage();
this.needDOMSurveyor =
this.lowlyGeneric.id.simple.size !== 0 ||
this.lowlyGeneric.id.complex.size !== 0 ||
this.lowlyGeneric.cl.simple.size !== 0 ||
this.lowlyGeneric.cl.complex.size !== 0;
this.highlyGeneric.simple.str = Array.from(this.highlyGeneric.simple.dict).join(',\n'); this.highlyGeneric.simple.str = Array.from(this.highlyGeneric.simple.dict).join(',\n');
this.highlyGeneric.simple.mru.reset(); this.highlyGeneric.simple.mru.reset();
this.highlyGeneric.complex.str = Array.from(this.highlyGeneric.complex.dict).join(',\n'); this.highlyGeneric.complex.str = Array.from(this.highlyGeneric.complex.dict).join(',\n');
@ -309,40 +316,6 @@ FilterContainer.prototype.freeze = function() {
/******************************************************************************/ /******************************************************************************/
// https://github.com/gorhill/uBlock/issues/1668
// The key must be literal: unescape escaped CSS before extracting key.
// It's an uncommon case, so it's best to unescape only when needed.
FilterContainer.prototype.keyFromSelector = function(selector) {
let matches = this.rePlainSelector.exec(selector);
if ( matches === null ) { return; }
let key = matches[0];
if ( key.indexOf('\\') === -1 ) {
return key;
}
matches = this.rePlainSelectorEscaped.exec(selector);
if ( matches === null ) { return; }
key = '';
const escaped = matches[0];
let beg = 0;
this.reEscapeSequence.lastIndex = 0;
for (;;) {
matches = this.reEscapeSequence.exec(escaped);
if ( matches === null ) {
return key + escaped.slice(beg);
}
key += escaped.slice(beg, matches.index);
beg = this.reEscapeSequence.lastIndex;
if ( matches[1].length === 1 ) {
key += matches[1];
} else {
key += String.fromCharCode(parseInt(matches[1], 16));
}
}
};
/******************************************************************************/
FilterContainer.prototype.compile = function(parser, writer) { FilterContainer.prototype.compile = function(parser, writer) {
if ( parser.hasOptions() === false ) { if ( parser.hasOptions() === false ) {
this.compileGenericSelector(parser, writer); this.compileGenericSelector(parser, writer);
@ -396,38 +369,8 @@ FilterContainer.prototype.compileGenericHideSelector = function(
writer.select('COSMETIC_FILTERS:GENERIC'); writer.select('COSMETIC_FILTERS:GENERIC');
const type = compiled.charCodeAt(0);
let key;
// Simple selector-based CSS rule: no need to test for whether the
// selector is valid, the regex took care of this. Most generic selector
// falls into that category:
// - ###ad-bigbox
// - ##.ads-bigbox
if ( type === 0x23 /* '#' */ ) {
key = this.keyFromSelector(compiled);
if ( key === compiled ) {
writer.push([ 0, key.slice(1) ]);
return;
}
} else if ( type === 0x2E /* '.' */ ) {
key = this.keyFromSelector(compiled);
if ( key === compiled ) {
writer.push([ 2, key.slice(1) ]);
return;
}
}
// Invalid cosmetic filter, possible reasons:
// - Bad syntax
// - Procedural filters (can't be generic): the compiled version of
// a procedural selector is NEVER equal to its raw version.
// https://github.com/uBlockOrigin/uBlock-issues/issues/464
// Pseudoclass-based selectors can be compiled, but are also valid
// plain selectors.
// https://github.com/uBlockOrigin/uBlock-issues/issues/131 // https://github.com/uBlockOrigin/uBlock-issues/issues/131
// Support generic procedural filters as per advanced settings. // Support generic procedural filters as per advanced settings.
// TODO: prevent double compilation.
if ( compiled.charCodeAt(0) === 0x7B /* '{' */ ) { if ( compiled.charCodeAt(0) === 0x7B /* '{' */ ) {
if ( µb.hiddenSettings.allowGenericProceduralFilters === true ) { if ( µb.hiddenSettings.allowGenericProceduralFilters === true ) {
return this.compileSpecificSelector(parser, '', false, writer); return this.compileSpecificSelector(parser, '', false, writer);
@ -441,28 +384,12 @@ FilterContainer.prototype.compileGenericHideSelector = function(
return; return;
} }
// Complex selector-based CSS rule: const key = keyFromSelector(compiled);
// - ###tads + div + .c
// - ##.rscontainer > .ellip
if ( key !== undefined ) { if ( key !== undefined ) {
writer.push([ writer.push([
type === 0x23 /* '#' */ ? 1 : 3, 0,
key.slice(1), hashFromStr(key.charCodeAt(0), key.slice(1)),
compiled compiled,
]);
return;
}
// https://github.com/gorhill/uBlock/issues/909
// Anything which contains a plain id/class selector can be classified
// as a low generic cosmetic filter.
const matches = this.rePlainSelectorEx.exec(compiled);
if ( matches !== null ) {
const key = matches[1] || matches[2];
writer.push([
key.charCodeAt(0) === 0x23 /* '#' */ ? 1 : 3,
key.slice(1),
compiled
]); ]);
return; return;
} }
@ -618,36 +545,13 @@ FilterContainer.prototype.fromCompiledContent = function(reader, options) {
this.duplicateBuster.add(fingerprint); this.duplicateBuster.add(fingerprint);
const args = reader.args(); const args = reader.args();
switch ( args[0] ) { switch ( args[0] ) {
// low generic, simple // low generic
case 0: // #AdBanner case 0: {
case 2: { // .largeAd if ( this.lowlyGeneric.has(args[1]) ) {
const db = args[0] === 0 ? this.lowlyGeneric.id : this.lowlyGeneric.cl; const selector = this.lowlyGeneric.get(args[1]);
const bucket = db.complex.get(args[1]); this.lowlyGeneric.set(args[1], `${selector},\n${args[2]}`);
if ( bucket === undefined ) {
db.simple.add(args[1]);
} else if ( Array.isArray(bucket) ) {
bucket.push(db.prefix + args[1]);
} else { } else {
db.complex.set(args[1], [ bucket, db.prefix + args[1] ]); this.lowlyGeneric.set(args[1], args[2]);
}
break;
}
// low generic, complex
case 1: // #tads + div + .c
case 3: { // .Mpopup + #Mad > #MadZone
const db = args[0] === 1 ? this.lowlyGeneric.id : this.lowlyGeneric.cl;
const bucket = db.complex.get(args[1]);
if ( bucket === undefined ) {
if ( db.simple.has(args[1]) ) {
db.complex.set(args[1], [ db.prefix + args[1], args[2] ]);
} else {
db.complex.set(args[1], args[2]);
db.simple.add(args[1]);
}
} else if ( Array.isArray(bucket) ) {
bucket.push(args[2]);
} else {
db.complex.set(args[1], [ bucket, args[2] ]);
} }
break; break;
} }
@ -682,13 +586,11 @@ FilterContainer.prototype.skipCompiledContent = function(reader, sectionId) {
FilterContainer.prototype.toSelfie = function() { FilterContainer.prototype.toSelfie = function() {
return { return {
version: this.selfieVersion,
acceptedCount: this.acceptedCount, acceptedCount: this.acceptedCount,
discardedCount: this.discardedCount, discardedCount: this.discardedCount,
specificFilters: this.specificFilters.toSelfie(), specificFilters: this.specificFilters.toSelfie(),
lowlyGenericSID: Array.from(this.lowlyGeneric.id.simple), lowlyGeneric: Array.from(this.lowlyGeneric),
lowlyGenericCID: Array.from(this.lowlyGeneric.id.complex),
lowlyGenericSCL: Array.from(this.lowlyGeneric.cl.simple),
lowlyGenericCCL: Array.from(this.lowlyGeneric.cl.complex),
highSimpleGenericHideArray: Array.from(this.highlyGeneric.simple.dict), highSimpleGenericHideArray: Array.from(this.highlyGeneric.simple.dict),
highComplexGenericHideArray: Array.from(this.highlyGeneric.complex.dict), highComplexGenericHideArray: Array.from(this.highlyGeneric.complex.dict),
}; };
@ -697,22 +599,19 @@ FilterContainer.prototype.toSelfie = function() {
/******************************************************************************/ /******************************************************************************/
FilterContainer.prototype.fromSelfie = function(selfie) { FilterContainer.prototype.fromSelfie = function(selfie) {
if ( selfie.version !== this.selfieVersion ) {
throw new Error(
`cosmeticFilteringEngine: mismatched selfie version, ${selfie.version}, expected ${this.selfieVersion}`
);
}
this.acceptedCount = selfie.acceptedCount; this.acceptedCount = selfie.acceptedCount;
this.discardedCount = selfie.discardedCount; this.discardedCount = selfie.discardedCount;
this.specificFilters.fromSelfie(selfie.specificFilters); this.specificFilters.fromSelfie(selfie.specificFilters);
this.lowlyGeneric.id.simple = new Set(selfie.lowlyGenericSID); this.lowlyGeneric = new Map(selfie.lowlyGeneric);
this.lowlyGeneric.id.complex = new Map(selfie.lowlyGenericCID);
this.lowlyGeneric.cl.simple = new Set(selfie.lowlyGenericSCL);
this.lowlyGeneric.cl.complex = new Map(selfie.lowlyGenericCCL);
this.highlyGeneric.simple.dict = new Set(selfie.highSimpleGenericHideArray); this.highlyGeneric.simple.dict = new Set(selfie.highSimpleGenericHideArray);
this.highlyGeneric.simple.str = selfie.highSimpleGenericHideArray.join(',\n'); this.highlyGeneric.simple.str = selfie.highSimpleGenericHideArray.join(',\n');
this.highlyGeneric.complex.dict = new Set(selfie.highComplexGenericHideArray); this.highlyGeneric.complex.dict = new Set(selfie.highComplexGenericHideArray);
this.highlyGeneric.complex.str = selfie.highComplexGenericHideArray.join(',\n'); this.highlyGeneric.complex.str = selfie.highComplexGenericHideArray.join(',\n');
this.needDOMSurveyor =
selfie.lowlyGenericSID.length !== 0 ||
selfie.lowlyGenericCID.length !== 0 ||
selfie.lowlyGenericSCL.length !== 0 ||
selfie.lowlyGenericCCL.length !== 0;
this.frozen = true; this.frozen = true;
}; };
@ -721,12 +620,11 @@ FilterContainer.prototype.fromSelfie = function(selfie) {
FilterContainer.prototype.triggerSelectorCachePruner = function() { FilterContainer.prototype.triggerSelectorCachePruner = function() {
// Of interest: http://fitzgeraldnick.com/weblog/40/ // Of interest: http://fitzgeraldnick.com/weblog/40/
// http://googlecode.blogspot.ca/2009/07/gmail-for-mobile-html5-series-using.html // http://googlecode.blogspot.ca/2009/07/gmail-for-mobile-html5-series-using.html
if ( this.selectorCacheTimer === null ) { if ( this.selectorCacheTimer !== null ) { return; }
this.selectorCacheTimer = vAPI.setTimeout( this.selectorCacheTimer = vAPI.setTimeout(
this.pruneSelectorCacheAsync.bind(this), ( ) => { this.pruneSelectorCacheAsync(); },
this.selectorCachePruneDelay this.selectorCachePruneDelay
); );
}
}; };
/******************************************************************************/ /******************************************************************************/
@ -740,7 +638,7 @@ FilterContainer.prototype.addToSelectorCache = function(details) {
if ( entry === undefined ) { if ( entry === undefined ) {
entry = SelectorCacheEntry.factory(); entry = SelectorCacheEntry.factory();
this.selectorCache.set(hostname, entry); this.selectorCache.set(hostname, entry);
if ( this.selectorCache.size > this.selectorCacheCountMin ) { if ( this.selectorCache.size > this.selectorCacheCountMax ) {
this.triggerSelectorCachePruner(); this.triggerSelectorCachePruner();
} }
} }
@ -753,7 +651,7 @@ FilterContainer.prototype.removeFromSelectorCache = function(
targetHostname = '*', targetHostname = '*',
type = undefined type = undefined
) { ) {
let targetHostnameLength = targetHostname.length; const targetHostnameLength = targetHostname.length;
for ( let entry of this.selectorCache ) { for ( let entry of this.selectorCache ) {
let hostname = entry[0]; let hostname = entry[0];
let item = entry[1]; let item = entry[1];
@ -772,46 +670,27 @@ FilterContainer.prototype.removeFromSelectorCache = function(
/******************************************************************************/ /******************************************************************************/
FilterContainer.prototype.retrieveFromSelectorCache = function( FilterContainer.prototype.pruneSelectorCacheAsync = function() {
hostname, this.selectorCacheTimer = null;
type, if ( this.selectorCache.size <= this.selectorCacheCountMax ) { return; }
out const cache = this.selectorCache;
) { const hostnames = Array.from(cache.keys())
let entry = this.selectorCache.get(hostname); .sort((a, b) => cache.get(b).accessId - cache.get(a).accessId)
if ( entry !== undefined ) { .slice(this.selectorCacheCountMin);
entry.retrieve(type, out); for ( const hn of hostnames ) {
cache.get(hn).dispose();
cache.delete(hn);
} }
}; };
/******************************************************************************/ /******************************************************************************/
FilterContainer.prototype.pruneSelectorCacheAsync = function() { FilterContainer.prototype.disableSurveyor = function(details) {
this.selectorCacheTimer = null; const hostname = details.hostname;
if ( this.selectorCache.size <= this.selectorCacheCountMin ) { return; } if ( typeof hostname !== 'string' || hostname === '' ) { return; }
let cache = this.selectorCache; const cacheEntry = this.selectorCache.get(hostname);
// Sorted from most-recently-used to least-recently-used, because if ( cacheEntry === undefined ) { return; }
// we loop beginning at the end below. cacheEntry.disableSurveyor = true;
// We can't avoid sorting because we have to keep a minimum number of
// entries, and these entries should always be the most-recently-used.
let hostnames = Array.from(cache.keys())
.sort(function(a, b) {
return cache.get(b).lastAccessTime -
cache.get(a).lastAccessTime;
})
.slice(this.selectorCacheCountMin);
let obsolete = Date.now() - this.selectorCacheAgeMax,
i = hostnames.length;
while ( i-- ) {
let hostname = hostnames[i];
let entry = cache.get(hostname);
if ( entry.lastAccessTime > obsolete ) { break; }
// console.debug('pruneSelectorCacheAsync: flushing "%s"', hostname);
entry.dispose();
cache.delete(hostname);
}
if ( cache.size > this.selectorCacheCountMin ) {
this.triggerSelectorCachePruner();
}
}; };
/******************************************************************************/ /******************************************************************************/
@ -850,43 +729,19 @@ FilterContainer.prototype.cssRuleFromProcedural = function(json) {
/******************************************************************************/ /******************************************************************************/
FilterContainer.prototype.retrieveGenericSelectors = function(request) { FilterContainer.prototype.retrieveGenericSelectors = function(request) {
if ( this.acceptedCount === 0 ) { return; } if ( this.lowlyGeneric.size === 0 ) { return; }
if ( !request.ids && !request.classes ) { return; } if ( Array.isArray(request.hashes) === false ) { return; }
if ( request.hashes.length === 0 ) { return; }
const { safeOnly = false } = request; const selectorsSet = new Set();
//console.time('cosmeticFilteringEngine.retrieveGenericSelectors'); const hashes = [];
for ( const hash of request.hashes ) {
const simpleSelectors = this.$simpleSet; const bucket = this.lowlyGeneric.get(hash);
const complexSelectors = this.$complexSet; if ( bucket === undefined ) { continue; }
for ( const selector of bucket.split(',\n') ) {
const cacheEntry = this.selectorCache.get(request.hostname); selectorsSet.add(selector);
const previousHits = cacheEntry && cacheEntry.cosmetic || this.$dummySet;
for ( const type in this.lowlyGeneric ) {
const entry = this.lowlyGeneric[type];
const selectors = request[entry.canonical];
if ( Array.isArray(selectors) === false ) { continue; }
for ( const identifier of selectors ) {
if ( entry.simple.has(identifier) === false ) { continue; }
const bucket = entry.complex.get(identifier);
if ( typeof bucket === 'string' ) {
if ( previousHits.has(bucket) ) { continue; }
complexSelectors.add(bucket);
continue;
}
const simpleSelector = entry.prefix + identifier;
if ( Array.isArray(bucket) ) {
for ( const complexSelector of bucket ) {
if ( previousHits.has(complexSelector) ) { continue; }
if ( safeOnly && complexSelector === simpleSelector ) { continue; }
complexSelectors.add(complexSelector);
}
continue;
}
if ( previousHits.has(simpleSelector) ) { continue; }
if ( safeOnly ) { continue; }
simpleSelectors.add(simpleSelector);
} }
hashes.push(hash);
} }
// Apply exceptions: it is the responsibility of the caller to provide // Apply exceptions: it is the responsibility of the caller to provide
@ -894,48 +749,29 @@ FilterContainer.prototype.retrieveGenericSelectors = function(request) {
const excepted = []; const excepted = [];
if ( Array.isArray(request.exceptions) ) { if ( Array.isArray(request.exceptions) ) {
for ( const exception of request.exceptions ) { for ( const exception of request.exceptions ) {
if ( if ( selectorsSet.delete(exception) ) {
simpleSelectors.delete(exception) ||
complexSelectors.delete(exception)
) {
excepted.push(exception); excepted.push(exception);
} }
} }
} }
if ( if ( selectorsSet.size === 0 && excepted.length === 0 ) { return; }
simpleSelectors.size === 0 &&
complexSelectors.size === 0 &&
excepted.length === 0
) {
return;
}
const out = { injectedCSS: '', excepted, }; const out = { injectedCSS: '', excepted, };
const selectors = Array.from(selectorsSet);
const injected = [];
if ( simpleSelectors.size !== 0 ) {
injected.push(...simpleSelectors);
simpleSelectors.clear();
}
if ( complexSelectors.size !== 0 ) {
injected.push(...complexSelectors);
complexSelectors.clear();
}
// Cache and inject looked-up low generic cosmetic filters.
if ( injected.length === 0 ) { return out; }
if ( typeof request.hostname === 'string' && request.hostname !== '' ) { if ( typeof request.hostname === 'string' && request.hostname !== '' ) {
this.addToSelectorCache({ this.addToSelectorCache({
cost: request.surveyCost || 0,
hostname: request.hostname, hostname: request.hostname,
selectors: injected, selectors,
hashes,
type: 'cosmetic', type: 'cosmetic',
}); });
} }
out.injectedCSS = `${injected.join(',\n')}\n{display:none!important;}`; if ( selectors.length === 0 ) { return out; }
out.injectedCSS = `${selectors.join(',\n')}\n{display:none!important;}`;
vAPI.tabs.insertCSS(request.tabId, { vAPI.tabs.insertCSS(request.tabId, {
code: out.injectedCSS, code: out.injectedCSS,
frameId: request.frameId, frameId: request.frameId,
@ -943,8 +779,6 @@ FilterContainer.prototype.retrieveGenericSelectors = function(request) {
runAt: 'document_start', runAt: 'document_start',
}); });
//console.timeEnd('cosmeticFilteringEngine.retrieveGenericSelectors');
return out; return out;
}; };
@ -972,7 +806,7 @@ FilterContainer.prototype.retrieveSpecificSelectors = function(
exceptedFilters: [], exceptedFilters: [],
proceduralFilters: [], proceduralFilters: [],
convertedProceduralFilters: [], convertedProceduralFilters: [],
noDOMSurveying: this.needDOMSurveyor === false, disableSurveyor: this.lowlyGeneric.size === 0,
}; };
const injectedCSS = []; const injectedCSS = [];
@ -987,10 +821,9 @@ FilterContainer.prototype.retrieveSpecificSelectors = function(
// Cached cosmetic filters: these are always declarative. // Cached cosmetic filters: these are always declarative.
if ( cacheEntry !== undefined ) { if ( cacheEntry !== undefined ) {
cacheEntry.retrieve('cosmetic', specificSet); cacheEntry.retrieveCosmetic(specificSet, out.genericCosmeticHashes = []);
if ( out.noDOMSurveying === false ) { if ( cacheEntry.disableSurveyor ) {
out.noDOMSurveying = cacheEntry.cosmeticSurveyingMissCount > out.disableSurveyor = true;
cosmeticSurveyingMissCountMax;
} }
} }
@ -1123,8 +956,7 @@ FilterContainer.prototype.retrieveSpecificSelectors = function(
// CSS selectors for collapsible blocked elements // CSS selectors for collapsible blocked elements
if ( cacheEntry ) { if ( cacheEntry ) {
const networkFilters = []; const networkFilters = [];
cacheEntry.retrieve('net', networkFilters); if ( cacheEntry.retrieveNet(networkFilters) ) {
if ( networkFilters.length !== 0 ) {
details.code = `${networkFilters.join('\n')}\n{display:none!important;}`; details.code = `${networkFilters.join('\n')}\n{display:none!important;}`;
if ( request.tabId !== undefined ) { if ( request.tabId !== undefined ) {
vAPI.tabs.insertCSS(request.tabId, details); vAPI.tabs.insertCSS(request.tabId, details);
@ -1144,31 +976,16 @@ FilterContainer.prototype.getFilterCount = function() {
/******************************************************************************/ /******************************************************************************/
FilterContainer.prototype.dump = function() { FilterContainer.prototype.dump = function() {
let genericCount = 0; const generics = [];
for ( const i of [ 'simple', 'complex' ] ) { for ( const selectors of this.lowlyGeneric.values() ) {
for ( const j of [ 'id', 'cl' ] ) { generics.push(...selectors.split(',\n'));
genericCount += this.lowlyGeneric[j][i].size;
}
} }
return [ return [
'Cosmetic Filtering Engine internals:', 'Cosmetic Filtering Engine internals:',
`specific: ${this.specificFilters.size}`, `specific: ${this.specificFilters.size}`,
`generic: ${genericCount}`, `generic: ${generics.length}`,
`+ lowly.id: ${this.lowlyGeneric.id.simple.size + this.lowlyGeneric.id.complex.size}`, `+ selectors: ${this.lowlyGeneric.size}`,
` + simple: ${this.lowlyGeneric.id.simple.size}`, ...generics.map(a => ` ${a}`),
...Array.from(this.lowlyGeneric.id.simple).map(a => ` ###${a}`),
` + complex: ${this.lowlyGeneric.id.complex.size}`,
...Array.from(this.lowlyGeneric.id.complex.values()).map(a => ` ##${a}`),
`+ lowly.class: ${this.lowlyGeneric.cl.simple.size + this.lowlyGeneric.cl.complex.size}`,
` + simple: ${this.lowlyGeneric.cl.simple.size}`,
...Array.from(this.lowlyGeneric.cl.simple).map(a => ` ##.${a}`),
` + complex: ${this.lowlyGeneric.cl.complex.size}`,
...Array.from(this.lowlyGeneric.cl.complex.values()).map(a => ` ##${a}`),
`+ highly: ${this.highlyGeneric.simple.dict.size + this.highlyGeneric.complex.dict.size}`,
` + highly.simple: ${this.highlyGeneric.simple.dict.size}`,
...Array.from(this.highlyGeneric.simple.dict).map(a => ` ##${a}`),
` + highly.complex: ${this.highlyGeneric.complex.dict.size}`,
...Array.from(this.highlyGeneric.complex.dict).map(a => ` ##${a}`),
].join('\n'); ].join('\n');
}; };

4
src/js/messaging.js
View File

@ -823,6 +823,10 @@ const onMessage = function(request, sender, callback) {
cosmeticFilteringEngine.addToSelectorCache(request); cosmeticFilteringEngine.addToSelectorCache(request);
break; break;
case 'disableGenericCosmeticFilteringSurveyor':
cosmeticFilteringEngine.disableSurveyor(request);
break;
case 'getCollapsibleBlockedRequests': case 'getCollapsibleBlockedRequests':
response = { response = {
id: request.id, id: request.id,