2021-07-25 16:55:35 +02:00
|
|
|
/*******************************************************************************
|
|
|
|
|
|
|
|
uBlock Origin - a browser extension to block requests.
|
|
|
|
Copyright (C) 2014-present 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/uBlock
|
|
|
|
*/
|
|
|
|
|
2021-08-21 22:02:32 +02:00
|
|
|
/* globals WebAssembly */
|
|
|
|
|
2021-07-25 16:55:35 +02:00
|
|
|
'use strict';
|
|
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
|
2021-07-28 16:38:24 +02:00
|
|
|
import { createRequire } from 'module';
|
|
|
|
|
2021-08-02 18:32:47 +02:00
|
|
|
import { readFileSync } from 'fs';
|
|
|
|
import { dirname, resolve } from 'path';
|
|
|
|
import { fileURLToPath } from 'url';
|
|
|
|
|
|
|
|
const __dirname = dirname(fileURLToPath(import.meta.url));
|
|
|
|
|
2021-08-22 18:03:59 +02:00
|
|
|
import punycode from './lib/punycode.js';
|
2021-07-25 16:55:35 +02:00
|
|
|
import './lib/publicsuffixlist/publicsuffixlist.js';
|
|
|
|
|
|
|
|
import globals from './js/globals.js';
|
2021-07-29 01:48:38 +02:00
|
|
|
import snfe from './js/static-net-filtering.js';
|
2021-07-25 16:55:35 +02:00
|
|
|
import { FilteringContext } from './js/filtering-context.js';
|
2021-07-31 14:38:33 +02:00
|
|
|
import { LineIterator } from './js/text-utils.js';
|
2021-07-25 16:55:35 +02:00
|
|
|
import { StaticFilteringParser } from './js/static-filtering-parser.js';
|
|
|
|
|
|
|
|
import {
|
|
|
|
CompiledListReader,
|
|
|
|
CompiledListWriter,
|
|
|
|
} from './js/static-filtering-io.js';
|
|
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
|
2021-08-02 18:32:47 +02:00
|
|
|
function loadJSON(path) {
|
|
|
|
return JSON.parse(readFileSync(resolve(__dirname, path), 'utf8'));
|
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
/******************************************************************************/
|
2021-07-25 16:55:35 +02:00
|
|
|
|
2021-07-29 22:54:51 +02:00
|
|
|
async function enableWASM() {
|
2021-07-29 23:04:32 +02:00
|
|
|
const wasmModuleFetcher = function(path) {
|
2021-07-31 13:06:11 +02:00
|
|
|
const require = createRequire(import.meta.url); // jshint ignore:line
|
|
|
|
const wasm = new Uint8Array(require(`${path}.wasm.json`));
|
2021-08-21 22:02:32 +02:00
|
|
|
return WebAssembly.compile(wasm);
|
2021-07-29 22:54:51 +02:00
|
|
|
};
|
|
|
|
try {
|
|
|
|
const results = await Promise.all([
|
2021-07-31 13:06:11 +02:00
|
|
|
globals.publicSuffixList.enableWASM(wasmModuleFetcher, './lib/publicsuffixlist/wasm/'),
|
|
|
|
snfe.enableWASM(wasmModuleFetcher, './js/wasm/'),
|
2021-07-29 22:54:51 +02:00
|
|
|
]);
|
|
|
|
return results.every(a => a === true);
|
|
|
|
} catch(reason) {
|
2021-07-31 13:06:11 +02:00
|
|
|
console.log(reason);
|
2021-07-29 22:54:51 +02:00
|
|
|
}
|
|
|
|
return false;
|
2021-07-25 16:55:35 +02:00
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
/******************************************************************************/
|
2021-08-01 22:42:05 +02:00
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
function pslInit(raw) {
|
|
|
|
if ( typeof raw === 'string' && raw.trim() !== '' ) {
|
2021-08-22 18:03:59 +02:00
|
|
|
globals.publicSuffixList.parse(raw, punycode.toASCII);
|
2021-08-08 15:17:14 +02:00
|
|
|
return globals.publicSuffixList;
|
|
|
|
}
|
2021-08-01 22:42:05 +02:00
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
// Use serialized version if available
|
|
|
|
let serialized = null;
|
|
|
|
try {
|
|
|
|
// Use loadJSON() because require() would keep the string in memory.
|
|
|
|
serialized = loadJSON('build/publicsuffixlist.json');
|
|
|
|
} catch (error) {
|
2021-08-10 02:02:58 +02:00
|
|
|
if ( process.env.npm_lifecycle_event !== 'build' ) {
|
|
|
|
// This should never happen except during package building.
|
2021-08-08 15:17:14 +02:00
|
|
|
console.error(error);
|
2021-08-01 22:42:05 +02:00
|
|
|
}
|
2021-08-08 15:17:14 +02:00
|
|
|
}
|
|
|
|
if ( serialized !== null ) {
|
|
|
|
globals.publicSuffixList.fromSelfie(serialized);
|
|
|
|
return globals.publicSuffixList;
|
|
|
|
}
|
2021-08-01 22:42:05 +02:00
|
|
|
|
2021-08-15 17:18:01 +02:00
|
|
|
raw = readFileSync(
|
|
|
|
resolve(__dirname, './assets/thirdparties/publicsuffix.org/list/effective_tld_names.dat'),
|
|
|
|
'utf8'
|
|
|
|
);
|
2021-08-08 15:17:14 +02:00
|
|
|
if ( typeof raw !== 'string' || raw.trim() === '' ) {
|
|
|
|
console.error('Unable to populate public suffix list');
|
|
|
|
return;
|
2021-08-01 21:42:46 +02:00
|
|
|
}
|
2021-08-22 18:03:59 +02:00
|
|
|
globals.publicSuffixList.parse(raw, punycode.toASCII);
|
2021-07-31 13:09:01 +02:00
|
|
|
return globals.publicSuffixList;
|
2021-07-25 16:55:35 +02:00
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
/******************************************************************************/
|
2021-08-05 19:37:41 +02:00
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
function compileList({ name, raw }, compiler, writer, options = {}) {
|
|
|
|
const lineIter = new LineIterator(raw);
|
|
|
|
const events = Array.isArray(options.events) ? options.events : undefined;
|
2021-07-25 16:55:35 +02:00
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
if ( name ) {
|
|
|
|
writer.properties.set('name', name);
|
2021-08-05 19:30:20 +02:00
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
const { parser } = compiler;
|
2021-08-05 19:30:20 +02:00
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
while ( lineIter.eot() === false ) {
|
|
|
|
let line = lineIter.next();
|
|
|
|
while ( line.endsWith(' \\') ) {
|
|
|
|
if ( lineIter.peek(4) !== ' ' ) { break; }
|
|
|
|
line = line.slice(0, -2).trim() + lineIter.next().trim();
|
|
|
|
}
|
|
|
|
parser.analyze(line);
|
|
|
|
if ( parser.shouldIgnore() ) { continue; }
|
|
|
|
if ( parser.category !== parser.CATStaticNetFilter ) { continue; }
|
|
|
|
if ( parser.patternHasUnicode() && parser.toASCII() === false ) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if ( compiler.compile(writer) ) { continue; }
|
|
|
|
if ( compiler.error !== undefined && events !== undefined ) {
|
|
|
|
options.events.push({
|
|
|
|
type: 'error',
|
|
|
|
text: compiler.error
|
|
|
|
});
|
|
|
|
}
|
2021-07-25 16:55:35 +02:00
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
return writer.toString();
|
2021-08-05 19:30:20 +02:00
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
/******************************************************************************/
|
|
|
|
|
|
|
|
async function useLists(lists, options = {}) {
|
2021-08-17 22:57:39 +02:00
|
|
|
if ( useLists.promise !== null ) {
|
|
|
|
throw new Error('Pending useLists() operation');
|
|
|
|
}
|
|
|
|
|
2021-08-05 19:30:20 +02:00
|
|
|
// Remove all filters
|
2021-08-08 15:17:14 +02:00
|
|
|
snfe.reset();
|
2021-08-05 19:30:20 +02:00
|
|
|
|
|
|
|
if ( Array.isArray(lists) === false || lists.length === 0 ) {
|
2021-08-15 17:18:01 +02:00
|
|
|
return;
|
2021-08-05 19:30:20 +02:00
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
let compiler = null;
|
2021-08-05 19:30:20 +02:00
|
|
|
|
|
|
|
const consumeList = list => {
|
2021-08-08 15:17:14 +02:00
|
|
|
let { compiled } = list;
|
|
|
|
if ( typeof compiled !== 'string' || compiled === '' ) {
|
|
|
|
const writer = new CompiledListWriter();
|
|
|
|
if ( compiler === null ) {
|
|
|
|
compiler = snfe.createCompiler(new StaticFilteringParser());
|
|
|
|
}
|
|
|
|
compiled = compileList(list, compiler, writer, options);
|
|
|
|
}
|
|
|
|
snfe.fromCompiled(new CompiledListReader(compiled));
|
2021-08-05 19:30:20 +02:00
|
|
|
};
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
// Populate filtering engine with resolved filter lists
|
2021-08-05 19:30:20 +02:00
|
|
|
const promises = [];
|
|
|
|
for ( const list of lists ) {
|
|
|
|
const promise = list instanceof Promise ? list : Promise.resolve(list);
|
|
|
|
promises.push(promise.then(list => consumeList(list)));
|
|
|
|
}
|
|
|
|
|
2021-08-17 22:57:39 +02:00
|
|
|
useLists.promise = Promise.all(promises);
|
|
|
|
await useLists.promise;
|
|
|
|
useLists.promise = null;
|
2021-08-05 19:30:20 +02:00
|
|
|
|
|
|
|
// Commit changes
|
|
|
|
snfe.freeze();
|
|
|
|
snfe.optimize();
|
2021-07-25 16:55:35 +02:00
|
|
|
}
|
|
|
|
|
2021-08-17 22:57:39 +02:00
|
|
|
useLists.promise = null;
|
|
|
|
|
2021-08-08 13:22:02 +02:00
|
|
|
/******************************************************************************/
|
|
|
|
|
2021-08-09 22:52:41 +02:00
|
|
|
class MockStorage {
|
|
|
|
constructor(serialized) {
|
|
|
|
this.map = new Map(serialized);
|
|
|
|
}
|
|
|
|
|
|
|
|
async put(assetKey, content) {
|
|
|
|
this.map.set(assetKey, content);
|
|
|
|
return ({ assetKey, content });
|
|
|
|
}
|
|
|
|
|
|
|
|
async get(assetKey) {
|
|
|
|
return ({ assetKey, content: this.map.get(assetKey) });
|
|
|
|
}
|
|
|
|
|
|
|
|
*[Symbol.iterator]() {
|
|
|
|
yield* this.map;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
const fctx = new FilteringContext();
|
2021-08-15 16:43:36 +02:00
|
|
|
let snfeProxyInstance = null;
|
2021-08-08 13:22:02 +02:00
|
|
|
|
|
|
|
class StaticNetFilteringEngine {
|
|
|
|
constructor() {
|
2021-08-15 16:43:36 +02:00
|
|
|
if ( snfeProxyInstance !== null ) {
|
2021-08-08 13:22:02 +02:00
|
|
|
throw new Error('Only a single instance is supported.');
|
|
|
|
}
|
2021-08-15 16:43:36 +02:00
|
|
|
snfeProxyInstance = this;
|
2021-08-08 13:22:02 +02:00
|
|
|
}
|
|
|
|
|
2021-08-19 13:34:02 +02:00
|
|
|
useLists(lists) {
|
|
|
|
return useLists(lists);
|
2021-08-08 13:22:02 +02:00
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
matchRequest(details) {
|
|
|
|
return snfe.matchRequest(fctx.fromDetails(details));
|
2021-08-08 13:22:02 +02:00
|
|
|
}
|
|
|
|
|
2021-08-17 18:48:39 +02:00
|
|
|
matchAndFetchModifiers(details, modifier) {
|
|
|
|
return snfe.matchAndFetchModifiers(fctx.fromDetails(details), modifier);
|
|
|
|
}
|
|
|
|
|
2021-08-17 22:49:43 +02:00
|
|
|
hasQuery(details) {
|
|
|
|
return snfe.hasQuery(details);
|
|
|
|
}
|
|
|
|
|
2021-08-08 13:22:02 +02:00
|
|
|
toLogData() {
|
|
|
|
return snfe.toLogData();
|
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
createCompiler(parser) {
|
|
|
|
return snfe.createCompiler(parser);
|
|
|
|
}
|
|
|
|
|
|
|
|
compileList(...args) {
|
|
|
|
return compileList(...args);
|
|
|
|
}
|
|
|
|
|
2021-08-09 22:52:41 +02:00
|
|
|
async serialize() {
|
|
|
|
const storage = new MockStorage();
|
|
|
|
await snfe.toSelfie(storage, 'path');
|
|
|
|
return JSON.stringify([...storage]);
|
|
|
|
}
|
|
|
|
|
|
|
|
async deserialize(serialized) {
|
|
|
|
const storage = new MockStorage(JSON.parse(serialized));
|
|
|
|
await snfe.fromSelfie(storage, 'path');
|
|
|
|
}
|
|
|
|
|
|
|
|
static async create({ noPSL = false } = {}) {
|
2021-08-08 15:17:14 +02:00
|
|
|
const instance = new StaticNetFilteringEngine();
|
|
|
|
|
|
|
|
if ( noPSL !== true && !pslInit() ) {
|
2021-08-08 13:22:02 +02:00
|
|
|
throw new Error('Failed to initialize public suffix list.');
|
|
|
|
}
|
|
|
|
|
2021-08-08 15:17:14 +02:00
|
|
|
return instance;
|
2021-08-08 13:22:02 +02:00
|
|
|
}
|
2021-08-15 16:43:36 +02:00
|
|
|
|
2021-08-15 17:18:01 +02:00
|
|
|
static async release() {
|
|
|
|
if ( snfeProxyInstance === null ) { return; }
|
2021-08-15 16:43:36 +02:00
|
|
|
snfeProxyInstance = null;
|
2021-08-15 17:18:01 +02:00
|
|
|
await useLists([]);
|
2021-08-15 16:43:36 +02:00
|
|
|
}
|
2021-08-08 15:17:14 +02:00
|
|
|
}
|
2021-08-08 13:22:02 +02:00
|
|
|
|
|
|
|
/******************************************************************************/
|
|
|
|
|
2021-08-02 00:33:13 +02:00
|
|
|
// rollup.js needs module.exports to be set back to the local exports object.
|
|
|
|
// This is because some of the code (e.g. publicsuffixlist.js) sets
|
|
|
|
// module.exports. Once all included files are written like ES modules, using
|
|
|
|
// export statements, this should no longer be necessary.
|
2021-08-03 18:03:11 +02:00
|
|
|
if ( typeof module !== 'undefined' && typeof exports !== 'undefined' ) {
|
2021-08-03 16:14:40 +02:00
|
|
|
module.exports = exports;
|
2021-08-02 00:33:13 +02:00
|
|
|
}
|
|
|
|
|
2021-07-25 16:55:35 +02:00
|
|
|
export {
|
|
|
|
enableWASM,
|
|
|
|
pslInit,
|
2021-08-08 15:17:14 +02:00
|
|
|
StaticNetFilteringEngine,
|
2021-07-25 16:55:35 +02:00
|
|
|
};
|