1
0
mirror of https://github.com/gorhill/uBlock.git synced 2024-11-25 11:52:51 +01:00

Redesign cache storage

In uBO, the "cache storage" is used to save resources which can
be safely discarded, though at the cost of having to fetch or
recompute them again.

Extension storage (browser.storage.local) is now always used as
cache storage backend. This has always been the default for
Chromium-based browsers.

For Firefox-based browsers, IndexedDB was used as backend for
cache storage, with fallback to extension storage when using
Firefox in private mode by default.

Extension storage is reliable since it works in all contexts,
though it may not be the most performant one.

To speed-up loading of resources from extension storage, uBO will
now make use of Cache API storage, which will mirror content of
key assets saved to extension storage. Typically loading resources
from Cache API is faster than loading the same resources from
the extension storage.

Only resources which must be loaded in memory as fast as possible
will make use of the Cache API storage layered on top of the
extension storage.

Compiled filter lists and memory snapshot of filtering engines
(aka "selfies") will be mirrored to the Cache API storage, since
these must be loaded into memory as fast as possible, and reloading
filter lists from their compiled counterpart is a common
operation.

This new design makes it now seamless to work in permanent private
mode for Firefox-based browsers, since extension storage now
always contains cache-related assets.

Support for IndexedDB is removed for the time being, except to
support migration of cached assets the first time uBO runs with
the new cache storage design.

In order to easily support all choices of storage, a new serializer
has been introduced, which is capable of serializing/deserializing
structure-cloneable data to/from a JS string.

Because of this new serializer, JS data structures can be stored
directly from their native representation, and deserialized
directly to their native representation from uBO's point of view,
since the serialization occurs (if needed) only at the storage
interface level.

This new serializer simplifies many code paths where data
structures such as Set, Map, TypedArray, RegExp, etc. had to be
converted in a disparate manner to be able to persist them to
extension storage.

The new serializer supports workers and LZ4 compression. These
can be configured through advanced settings.

With this new layered design, it's possible to introduce more
storage layers if measured as beneficial (i.e. maybe
browser.storage.session)

References:
- https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/storage/local
- https://developer.mozilla.org/en-US/docs/Web/API/Cache
- https://developer.mozilla.org/en-US/docs/Web/API/Web_Workers_API/Structured_clone_algorithm
This commit is contained in:
Raymond Hill 2024-02-26 16:50:11 -05:00
parent 2262a129ec
commit 086766a924
No known key found for this signature in database
GPG Key ID: 25E1490B761470C2
19 changed files with 1920 additions and 928 deletions

View File

@ -1671,10 +1671,7 @@ vAPI.cloud = (( ) => {
const push = async function(details) { const push = async function(details) {
const { datakey, data, encode } = details; const { datakey, data, encode } = details;
if ( if ( data === undefined || typeof data === 'string' && data === '' ) {
data === undefined ||
typeof data === 'string' && data === ''
) {
return deleteChunks(datakey, 0); return deleteChunks(datakey, 0);
} }
const item = { const item = {
@ -1682,10 +1679,9 @@ vAPI.cloud = (( ) => {
tstamp: Date.now(), tstamp: Date.now(),
data, data,
}; };
const json = JSON.stringify(item);
const encoded = encode instanceof Function const encoded = encode instanceof Function
? await encode(json) ? await encode(item)
: json; : JSON.stringify(item);
// Chunkify taking into account QUOTA_BYTES_PER_ITEM: // Chunkify taking into account QUOTA_BYTES_PER_ITEM:
// https://developer.chrome.com/extensions/storage#property-sync // https://developer.chrome.com/extensions/storage#property-sync
@ -1750,13 +1746,16 @@ vAPI.cloud = (( ) => {
i += 1; i += 1;
} }
encoded = encoded.join(''); encoded = encoded.join('');
const json = decode instanceof Function
? await decode(encoded)
: encoded;
let entry = null; let entry = null;
try { try {
entry = JSON.parse(json); if ( decode instanceof Function ) {
} catch(ex) { entry = await decode(encoded) || null;
}
if ( typeof entry === 'string' ) {
entry = JSON.parse(entry);
}
} catch(_) {
} }
return entry; return entry;
}; };

View File

@ -528,12 +528,12 @@ function getAssetSourceRegistry() {
assetSourceRegistryPromise = cacheStorage.get( assetSourceRegistryPromise = cacheStorage.get(
'assetSourceRegistry' 'assetSourceRegistry'
).then(bin => { ).then(bin => {
if ( if ( bin instanceof Object ) {
bin instanceof Object && if ( bin.assetSourceRegistry instanceof Object ) {
bin.assetSourceRegistry instanceof Object assetSourceRegistry = bin.assetSourceRegistry;
) { ubolog('Loaded assetSourceRegistry');
assetSourceRegistry = bin.assetSourceRegistry; return assetSourceRegistry;
return assetSourceRegistry; }
} }
return assets.fetchText( return assets.fetchText(
µb.assetsBootstrapLocation || µb.assetsJsonPath µb.assetsBootstrapLocation || µb.assetsJsonPath
@ -543,6 +543,7 @@ function getAssetSourceRegistry() {
: assets.fetchText(µb.assetsJsonPath); : assets.fetchText(µb.assetsJsonPath);
}).then(details => { }).then(details => {
updateAssetSourceRegistry(details.content, true); updateAssetSourceRegistry(details.content, true);
ubolog('Loaded assetSourceRegistry');
return assetSourceRegistry; return assetSourceRegistry;
}); });
}); });
@ -673,39 +674,27 @@ let assetCacheRegistryPromise;
let assetCacheRegistry = {}; let assetCacheRegistry = {};
function getAssetCacheRegistry() { function getAssetCacheRegistry() {
if ( assetCacheRegistryPromise === undefined ) { if ( assetCacheRegistryPromise !== undefined ) {
assetCacheRegistryPromise = cacheStorage.get( return assetCacheRegistryPromise;
'assetCacheRegistry'
).then(bin => {
if (
bin instanceof Object &&
bin.assetCacheRegistry instanceof Object
) {
if ( Object.keys(assetCacheRegistry).length === 0 ) {
assetCacheRegistry = bin.assetCacheRegistry;
} else {
console.error(
'getAssetCacheRegistry(): assetCacheRegistry reassigned!'
);
if (
Object.keys(bin.assetCacheRegistry).sort().join() !==
Object.keys(assetCacheRegistry).sort().join()
) {
console.error(
'getAssetCacheRegistry(): assetCacheRegistry changes overwritten!'
);
}
}
}
return assetCacheRegistry;
});
} }
assetCacheRegistryPromise = cacheStorage.get(
'assetCacheRegistry'
).then(bin => {
if ( bin instanceof Object === false ) { return; }
if ( bin.assetCacheRegistry instanceof Object === false ) { return; }
if ( Object.keys(assetCacheRegistry).length !== 0 ) {
return console.error('getAssetCacheRegistry(): assetCacheRegistry reassigned!');
}
ubolog('Loaded assetCacheRegistry');
assetCacheRegistry = bin.assetCacheRegistry;
}).then(( ) =>
assetCacheRegistry
);
return assetCacheRegistryPromise; return assetCacheRegistryPromise;
} }
const saveAssetCacheRegistry = (( ) => { const saveAssetCacheRegistry = (( ) => {
const save = function() { const save = ( ) => {
timer.off(); timer.off();
cacheStorage.set({ assetCacheRegistry }); cacheStorage.set({ assetCacheRegistry });
}; };
@ -726,7 +715,9 @@ async function assetCacheRead(assetKey, updateReadTime = false) {
const reportBack = function(content) { const reportBack = function(content) {
if ( content instanceof Blob ) { content = ''; } if ( content instanceof Blob ) { content = ''; }
const details = { assetKey, content }; const details = { assetKey, content };
if ( content === '' ) { details.error = 'ENOTFOUND'; } if ( content === '' || content === undefined ) {
details.error = 'ENOTFOUND';
}
return details; return details;
}; };
@ -742,17 +733,11 @@ async function assetCacheRead(assetKey, updateReadTime = false) {
) + ' ms'; ) + ' ms';
} }
if ( if ( bin instanceof Object === false ) { return reportBack(''); }
bin instanceof Object === false || if ( bin.hasOwnProperty(internalKey) === false ) { return reportBack(''); }
bin.hasOwnProperty(internalKey) === false
) {
return reportBack('');
}
const entry = assetCacheRegistry[assetKey]; const entry = assetCacheRegistry[assetKey];
if ( entry === undefined ) { if ( entry === undefined ) { return reportBack(''); }
return reportBack('');
}
entry.readTime = Date.now(); entry.readTime = Date.now();
if ( updateReadTime ) { if ( updateReadTime ) {
@ -762,34 +747,22 @@ async function assetCacheRead(assetKey, updateReadTime = false) {
return reportBack(bin[internalKey]); return reportBack(bin[internalKey]);
} }
async function assetCacheWrite(assetKey, details) { async function assetCacheWrite(assetKey, content, options = {}) {
let content = ''; if ( content === '' || content === undefined ) {
let options = {};
if ( typeof details === 'string' ) {
content = details;
} else if ( details instanceof Object ) {
content = details.content || '';
options = details;
}
if ( content === '' ) {
return assetCacheRemove(assetKey); return assetCacheRemove(assetKey);
} }
const cacheDict = await getAssetCacheRegistry(); const { resourceTime, url } = options;
let entry = cacheDict[assetKey]; getAssetCacheRegistry().then(cacheDict => {
if ( entry === undefined ) { const entry = cacheDict[assetKey] || {};
entry = cacheDict[assetKey] = {}; cacheDict[assetKey] = entry;
} entry.writeTime = entry.readTime = Date.now();
entry.writeTime = entry.readTime = Date.now(); entry.resourceTime = resourceTime || 0;
entry.resourceTime = options.resourceTime || 0; if ( typeof url === 'string' ) {
if ( typeof options.url === 'string' ) { entry.remoteURL = url;
entry.remoteURL = options.url; }
} cacheStorage.set({ assetCacheRegistry, [`cache/${assetKey}`]: content });
cacheStorage.set({
assetCacheRegistry,
[`cache/${assetKey}`]: content
}); });
const result = { assetKey, content }; const result = { assetKey, content };
@ -800,21 +773,31 @@ async function assetCacheWrite(assetKey, details) {
return result; return result;
} }
async function assetCacheRemove(pattern) { async function assetCacheRemove(pattern, options = {}) {
const cacheDict = await getAssetCacheRegistry(); const cacheDict = await getAssetCacheRegistry();
const removedEntries = []; const removedEntries = [];
const removedContent = []; const removedContent = [];
for ( const assetKey in cacheDict ) { for ( const assetKey in cacheDict ) {
if ( pattern instanceof RegExp && !pattern.test(assetKey) ) { if ( pattern instanceof RegExp ) {
continue; if ( pattern.test(assetKey) === false ) { continue; }
} } else if ( typeof pattern === 'string' ) {
if ( typeof pattern === 'string' && assetKey !== pattern ) { if ( assetKey !== pattern ) { continue; }
continue;
} }
removedEntries.push(assetKey); removedEntries.push(assetKey);
removedContent.push('cache/' + assetKey); removedContent.push(`cache/${assetKey}`);
delete cacheDict[assetKey]; delete cacheDict[assetKey];
} }
if ( options.janitor && pattern instanceof RegExp ) {
const re = new RegExp(
pattern.source.replace(/^\^/, 'cache\/'),
pattern.flags
);
const keys = await cacheStorage.keys(re);
for ( const key of keys ) {
removedContent.push(key);
ubolog(`Removing stray ${key}`);
}
}
if ( removedContent.length !== 0 ) { if ( removedContent.length !== 0 ) {
await Promise.all([ await Promise.all([
cacheStorage.remove(removedContent), cacheStorage.remove(removedContent),
@ -980,8 +963,7 @@ assets.get = async function(assetKey, options = {}) {
} }
if ( details.content === '' ) { continue; } if ( details.content === '' ) { continue; }
if ( reIsExternalPath.test(contentURL) && options.dontCache !== true ) { if ( reIsExternalPath.test(contentURL) && options.dontCache !== true ) {
assetCacheWrite(assetKey, { assetCacheWrite(assetKey, details.content, {
content: details.content,
url: contentURL, url: contentURL,
silent: options.silent === true, silent: options.silent === true,
}); });
@ -1057,8 +1039,7 @@ async function getRemote(assetKey, options = {}) {
} }
// Success // Success
assetCacheWrite(assetKey, { assetCacheWrite(assetKey, result.content, {
content: result.content,
url: contentURL, url: contentURL,
resourceTime: result.resourceTime || 0, resourceTime: result.resourceTime || 0,
}); });
@ -1101,6 +1082,17 @@ assets.put = async function(assetKey, content) {
/******************************************************************************/ /******************************************************************************/
assets.toCache = async function(assetKey, content) {
return assetCacheWrite(assetKey, content);
};
assets.fromCache = async function(assetKey) {
const details = await assetCacheRead(assetKey);
return details && details.content;
};
/******************************************************************************/
assets.metadata = async function() { assets.metadata = async function() {
await Promise.all([ await Promise.all([
getAssetSourceRegistry(), getAssetSourceRegistry(),
@ -1147,8 +1139,8 @@ assets.metadata = async function() {
assets.purge = assetCacheMarkAsDirty; assets.purge = assetCacheMarkAsDirty;
assets.remove = function(pattern) { assets.remove = function(...args) {
return assetCacheRemove(pattern); return assetCacheRemove(...args);
}; };
assets.rmrf = function() { assets.rmrf = function() {
@ -1300,8 +1292,7 @@ async function diffUpdater() {
'Diff-Path', 'Diff-Path',
'Diff-Expires', 'Diff-Expires',
]); ]);
assetCacheWrite(data.assetKey, { assetCacheWrite(data.assetKey, data.text, {
content: data.text,
resourceTime: metadata.lastModified || 0, resourceTime: metadata.lastModified || 0,
}); });
metadata.diffUpdated = true; metadata.diffUpdated = true;

View File

@ -56,6 +56,7 @@ const hiddenSettingsDefault = {
blockingProfiles: '11111/#F00 11010/#C0F 11001/#00F 00001', blockingProfiles: '11111/#F00 11010/#C0F 11001/#00F 00001',
cacheStorageAPI: 'unset', cacheStorageAPI: 'unset',
cacheStorageCompression: true, cacheStorageCompression: true,
cacheStorageMultithread: 2,
cacheControlForFirefox1376932: 'no-cache, no-store, must-revalidate', cacheControlForFirefox1376932: 'no-cache, no-store, must-revalidate',
cloudStorageCompression: true, cloudStorageCompression: true,
cnameIgnoreList: 'unset', cnameIgnoreList: 'unset',
@ -181,7 +182,7 @@ const µBlock = { // jshint ignore:line
// Read-only // Read-only
systemSettings: { systemSettings: {
compiledMagic: 57, // Increase when compiled format changes compiledMagic: 57, // Increase when compiled format changes
selfieMagic: 57, // Increase when selfie format changes selfieMagic: 58, // 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

View File

@ -46,105 +46,6 @@ const digitToVal = new Uint8Array(128);
} }
} }
// The sparse base64 codec is best for buffers which contains a lot of
// small u32 integer values. Those small u32 integer values are better
// represented with stringified integers, because small values can be
// represented with fewer bits than the usual base64 codec. For example,
// 0 become '0 ', i.e. 16 bits instead of 48 bits with official base64
// codec.
const sparseBase64 = {
magic: 'Base64_1',
encode: function(arrbuf, arrlen) {
const inputLength = (arrlen + 3) >>> 2;
const inbuf = new Uint32Array(arrbuf, 0, inputLength);
const outputLength = this.magic.length + 7 + inputLength * 7;
const outbuf = new Uint8Array(outputLength);
// magic bytes
let j = 0;
for ( let i = 0; i < this.magic.length; i++ ) {
outbuf[j++] = this.magic.charCodeAt(i);
}
// array size
let v = inputLength;
do {
outbuf[j++] = valToDigit[v & 0b111111];
v >>>= 6;
} while ( v !== 0 );
outbuf[j++] = 0x20 /* ' ' */;
// array content
for ( let i = 0; i < inputLength; i++ ) {
v = inbuf[i];
do {
outbuf[j++] = valToDigit[v & 0b111111];
v >>>= 6;
} while ( v !== 0 );
outbuf[j++] = 0x20 /* ' ' */;
}
if ( typeof TextDecoder === 'undefined' ) {
return JSON.stringify(
Array.from(new Uint32Array(outbuf.buffer, 0, j >>> 2))
);
}
const textDecoder = new TextDecoder();
return textDecoder.decode(new Uint8Array(outbuf.buffer, 0, j));
},
decode: function(instr, arrbuf) {
if ( instr.charCodeAt(0) === 0x5B /* '[' */ ) {
const inbuf = JSON.parse(instr);
if ( arrbuf instanceof ArrayBuffer === false ) {
return new Uint32Array(inbuf);
}
const outbuf = new Uint32Array(arrbuf);
outbuf.set(inbuf);
return outbuf;
}
if ( instr.startsWith(this.magic) === false ) {
throw new Error('Invalid µBlock.base64 encoding');
}
const inputLength = instr.length;
const outputLength = this.decodeSize(instr) >> 2;
const outbuf = arrbuf instanceof ArrayBuffer === false
? new Uint32Array(outputLength)
: new Uint32Array(arrbuf);
let i = instr.indexOf(' ', this.magic.length) + 1;
if ( i === -1 ) {
throw new Error('Invalid µBlock.base64 encoding');
}
// array content
let j = 0;
for (;;) {
if ( j === outputLength || i >= inputLength ) { break; }
let v = 0, l = 0;
for (;;) {
const c = instr.charCodeAt(i++);
if ( c === 0x20 /* ' ' */ ) { break; }
v += digitToVal[c] << l;
l += 6;
}
outbuf[j++] = v;
}
if ( i < inputLength || j < outputLength ) {
throw new Error('Invalid µBlock.base64 encoding');
}
return outbuf;
},
decodeSize: function(instr) {
if ( instr.startsWith(this.magic) === false ) { return 0; }
let v = 0, l = 0, i = this.magic.length;
for (;;) {
const c = instr.charCodeAt(i++);
if ( c === 0x20 /* ' ' */ ) { break; }
v += digitToVal[c] << l;
l += 6;
}
return v << 2;
},
};
// The dense base64 codec is best for typed buffers which values are // The dense base64 codec is best for typed buffers which values are
// more random. For example, buffer contents as a result of compression // more random. For example, buffer contents as a result of compression
// contain less repetitive values and thus the content is more // contain less repetitive values and thus the content is more
@ -154,7 +55,7 @@ const sparseBase64 = {
// ArrayBuffer fails, the content of the resulting Uint8Array is // ArrayBuffer fails, the content of the resulting Uint8Array is
// non-sensical. WASM-related? // non-sensical. WASM-related?
const denseBase64 = { export const denseBase64 = {
magic: 'DenseBase64_1', magic: 'DenseBase64_1',
encode: function(input) { encode: function(input) {
@ -242,5 +143,3 @@ const denseBase64 = {
}; };
/******************************************************************************/ /******************************************************************************/
export { denseBase64, sparseBase64 };

View File

@ -576,34 +576,19 @@ class BidiTrieContainer {
}; };
} }
serialize(encoder) { toSelfie() {
if ( encoder instanceof Object ) { return this.buf32.subarray(
return encoder.encode( 0,
this.buf32.buffer, this.buf32[CHAR1_SLOT] + 3 >>> 2
this.buf32[CHAR1_SLOT]
);
}
return Array.from(
new Uint32Array(
this.buf32.buffer,
0,
this.buf32[CHAR1_SLOT] + 3 >>> 2
)
); );
} }
unserialize(selfie, decoder) { fromSelfie(selfie) {
const shouldDecode = typeof selfie === 'string'; if ( selfie instanceof Uint32Array === false ) { return false; }
let byteLength = shouldDecode let byteLength = selfie.length << 2;
? decoder.decodeSize(selfie)
: selfie.length << 2;
if ( byteLength === 0 ) { return false; } if ( byteLength === 0 ) { return false; }
this.reallocateBuf(byteLength); this.reallocateBuf(byteLength);
if ( shouldDecode ) { this.buf32.set(selfie);
decoder.decode(selfie, this.buf8.buffer);
} else {
this.buf32.set(selfie);
}
return true; return true;
} }

View File

@ -19,179 +19,362 @@
Home: https://github.com/gorhill/uBlock Home: https://github.com/gorhill/uBlock
*/ */
/* global browser, IDBDatabase, indexedDB */ /* global browser, indexedDB */
'use strict'; 'use strict';
/******************************************************************************/ /******************************************************************************/
import lz4Codec from './lz4.js'; import lz4Codec from './lz4.js';
import µb from './background.js';
import webext from './webext.js'; import webext from './webext.js';
import µb from './background.js';
import { ubolog } from './console.js';
import * as scuo from './scuo-serializer.js';
/******************************************************************************/ /******************************************************************************/
// The code below has been originally manually imported from:
// Commit: https://github.com/nikrolls/uBlock-Edge/commit/d1538ea9bea89d507219d3219592382eee306134
// Commit date: 29 October 2016
// Commit author: https://github.com/nikrolls
// Commit message: "Implement cacheStorage using IndexedDB"
// The original imported code has been subsequently modified as it was not
// compatible with Firefox.
// (a Promise thing, see https://github.com/dfahlander/Dexie.js/issues/317)
// Furthermore, code to migrate from browser.storage.local to vAPI.storage
// has been added, for seamless migration of cache-related entries into
// indexedDB.
// https://bugzilla.mozilla.org/show_bug.cgi?id=1371255
// Firefox-specific: we use indexedDB because browser.storage.local() has
// poor performance in Firefox.
// https://github.com/uBlockOrigin/uBlock-issues/issues/328
// Use IndexedDB for Chromium as well, to take advantage of LZ4
// compression.
// https://github.com/uBlockOrigin/uBlock-issues/issues/399
// Revert Chromium support of IndexedDB, use advanced setting to force
// IndexedDB.
// https://github.com/uBlockOrigin/uBlock-issues/issues/409
// Allow forcing the use of webext storage on Firefox.
const STORAGE_NAME = 'uBlock0CacheStorage'; const STORAGE_NAME = 'uBlock0CacheStorage';
const extensionStorage = webext.storage.local;
// Default to webext storage. const keysFromGetArg = arg => {
const storageLocal = webext.storage.local; if ( arg === null || arg === undefined ) { return []; }
const type = typeof arg;
let storageReadyResolve; if ( type === 'string' ) { return [ arg ]; }
const storageReadyPromise = new Promise(resolve => { if ( Array.isArray(arg) ) { return arg; }
storageReadyResolve = resolve; if ( type !== 'object' ) { return; }
}); return Object.keys(arg);
const cacheStorage = {
name: 'browser.storage.local',
get(...args) {
return storageReadyPromise.then(( ) =>
storageLocal.get(...args).catch(reason => {
console.log(reason);
})
);
},
set(...args) {
return storageReadyPromise.then(( ) =>
storageLocal.set(...args).catch(reason => {
console.log(reason);
})
);
},
remove(...args) {
return storageReadyPromise.then(( ) =>
storageLocal.remove(...args).catch(reason => {
console.log(reason);
})
);
},
clear(...args) {
return storageReadyPromise.then(( ) =>
storageLocal.clear(...args).catch(reason => {
console.log(reason);
})
);
},
select: function(selectedBackend) {
let actualBackend = selectedBackend;
if ( actualBackend === undefined || actualBackend === 'unset' ) {
actualBackend = vAPI.webextFlavor.soup.has('firefox')
? 'indexedDB'
: 'browser.storage.local';
}
if ( actualBackend === 'indexedDB' ) {
return selectIDB().then(success => {
if ( success || selectedBackend === 'indexedDB' ) {
clearWebext();
storageReadyResolve();
return 'indexedDB';
}
clearIDB();
storageReadyResolve();
return 'browser.storage.local';
});
}
if ( actualBackend === 'browser.storage.local' ) {
clearIDB();
}
storageReadyResolve();
return Promise.resolve('browser.storage.local');
},
error: undefined
}; };
// Cache API is subject to quota so we will use it only for what is key
// performance-wise
const shouldCache = bin => {
const out = {};
for ( const key of Object.keys(bin) ) {
if ( key.startsWith('cache/') ) {
if ( /^cache\/(compiled|selfie)\//.test(key) === false ) { continue; }
}
out[key] = bin[key];
}
return out;
};
/*******************************************************************************
*
* Extension storage
*
* Always available.
*
* */
const cacheStorage = (( ) => {
const LARGE = 65536;
const compress = async (key, data) => {
const isLarge = typeof data === 'string' && data.length >= LARGE;
const µbhs = µb.hiddenSettings;
const after = await scuo.serializeAsync(data, {
compress: isLarge && µbhs.cacheStorageCompression,
multithreaded: isLarge && µbhs.cacheStorageMultithread || 0,
});
return { key, data: after };
};
const decompress = async (key, data) => {
if ( scuo.canDeserialize(data) === false ) {
return { key, data };
}
const isLarge = data.length >= LARGE;
const after = await scuo.deserializeAsync(data, {
multithreaded: isLarge && µb.hiddenSettings.cacheStorageMultithread || 0,
});
return { key, data: after };
};
return {
name: 'browser.storage.local',
get(arg) {
const keys = arg;
return cacheAPI.get(keysFromGetArg(arg)).then(bin => {
if ( bin !== undefined ) { return bin; }
return extensionStorage.get(keys).catch(reason => {
ubolog(reason);
});
}).then(bin => {
if ( bin instanceof Object === false ) { return bin; }
const promises = [];
for ( const key of Object.keys(bin) ) {
promises.push(decompress(key, bin[key]));
}
return Promise.all(promises);
}).then(results => {
const bin = {};
for ( const { key, data } of results ) {
bin[key] = data;
}
return bin;
}).catch(reason => {
ubolog(reason);
});
},
async keys(regex) {
const results = await Promise.all([
cacheAPI.keys(regex),
extensionStorage.get(null).catch(( ) => {}),
]);
const keys = new Set(results[0]);
const bin = results[1] || {};
for ( const key of Object.keys(bin) ) {
if ( regex && regex.test(key) === false ) { continue; }
keys.add(key);
}
return keys;
},
async set(keyvalStore) {
const keys = Object.keys(keyvalStore);
if ( keys.length === 0 ) { return; }
const promises = [];
for ( const key of keys ) {
promises.push(compress(key, keyvalStore[key]));
}
const results = await Promise.all(promises);
const serializedStore = {};
for ( const { key, data } of results ) {
serializedStore[key] = data;
}
cacheAPI.set(shouldCache(serializedStore));
return extensionStorage.set(serializedStore).catch(reason => {
ubolog(reason);
});
},
remove(...args) {
cacheAPI.remove(...args);
return extensionStorage.remove(...args).catch(reason => {
ubolog(reason);
});
},
clear(...args) {
cacheAPI.clear(...args);
return extensionStorage.clear(...args).catch(reason => {
ubolog(reason);
});
},
async migrate(cacheAPI) {
if ( cacheAPI === 'browser.storage.local' ) { return; }
if ( cacheAPI !== 'indexedDB' ) {
if ( vAPI.webextFlavor.soup.has('firefox') === false ) { return; }
}
if ( browser.extension.inIncognitoContext ) { return; }
// Copy all items to new cache storage
const bin = await idbStorage.get(null);
if ( typeof bin !== 'object' || bin === null ) { return; }
const toMigrate = [];
for ( const key of Object.keys(bin) ) {
if ( key.startsWith('cache/selfie/') ) { continue; }
ubolog(`Migrating ${key}=${JSON.stringify(bin[key]).slice(0,32)}`);
toMigrate.push(cacheStorage.set({ [key]: bin[key] }));
}
idbStorage.clear();
return Promise.all(toMigrate);
},
error: undefined
};
})();
// Not all platforms support getBytesInUse // Not all platforms support getBytesInUse
if ( storageLocal.getBytesInUse instanceof Function ) { if ( extensionStorage.getBytesInUse instanceof Function ) {
cacheStorage.getBytesInUse = function(...args) { cacheStorage.getBytesInUse = function(...args) {
return storageLocal.getBytesInUse(...args).catch(reason => { return extensionStorage.getBytesInUse(...args).catch(reason => {
console.log(reason); ubolog(reason);
}); });
}; };
} }
// Reassign API entries to that of indexedDB-based ones /*******************************************************************************
const selectIDB = async function() { *
let db; * Cache API
let dbPromise; *
* Purpose is to mirror cache-related items from extension storage, as its
* read/write operations are faster. May not be available/populated in
* private/incognito mode.
*
* */
const noopfn = function () { const cacheAPI = (( ) => {
}; const caches = globalThis.caches;
const cacheStoragePromise = new Promise(resolve => {
const disconnect = function() { if ( typeof caches !== 'object' || caches === null ) {
dbTimer.off(); ubolog('CacheStorage API not available');
if ( db instanceof IDBDatabase ) { resolve(null);
db.close(); return;
db = undefined;
} }
}; resolve(caches.open(STORAGE_NAME).catch(reason => {
ubolog(reason);
const dbTimer = vAPI.defer.create(( ) => { }));
disconnect();
}); });
const keepAlive = function() { const urlPrefix = 'https://ublock0.invalid/';
dbTimer.offon(Math.max(
µb.hiddenSettings.autoUpdateAssetFetchPeriod * 2 * 1000, const keyToURL = key =>
180000 `${urlPrefix}${encodeURIComponent(key)}`;
));
const urlToKey = url =>
decodeURIComponent(url.slice(urlPrefix.length));
const getOne = async key => {
const cache = await cacheStoragePromise;
if ( cache === null ) { return; }
return cache.match(keyToURL(key)).then(response => {
if ( response instanceof Response === false ) { return; }
return response.text();
}).then(text => {
if ( text === undefined ) { return; }
return { key, text };
}).catch(reason => {
ubolog(reason);
});
}; };
// https://github.com/gorhill/uBlock/issues/3156 const getAll = async ( ) => {
// I have observed that no event was fired in Tor Browser 7.0.7 + const cache = await cacheStoragePromise;
// medium security level after the request to open the database was if ( cache === null ) { return; }
// created. When this occurs, I have also observed that the `error` return cache.keys().then(requests => {
// property was already set, so this means uBO can detect here whether const promises = [];
// the database can be opened successfully. A try-catch block is for ( const request of requests ) {
// necessary when reading the `error` property because we are not promises.push(getOne(urlToKey(request.url)));
// allowed to read this property outside of event handlers in newer }
// implementation of IDBRequest (my understanding). return Promise.all(promises);
}).then(responses => {
const bin = {};
for ( const response of responses ) {
if ( response === undefined ) { continue; }
bin[response.key] = response.text;
}
return bin;
}).catch(reason => {
ubolog(reason);
});
};
const setOne = async (key, text) => {
if ( text === undefined ) { return removeOne(key); }
const blob = new Blob([ text ], { type: 'text/plain;charset=utf-8'});
const cache = await cacheStoragePromise;
if ( cache === null ) { return; }
return cache
.put(keyToURL(key), new Response(blob))
.catch(reason => {
ubolog(reason);
});
};
const removeOne = async key => {
const cache = await cacheStoragePromise;
if ( cache === null ) { return; }
return cache.delete(keyToURL(key)).catch(reason => {
ubolog(reason);
});
};
return {
async get(arg) {
const keys = keysFromGetArg(arg);
if ( keys === undefined ) { return; }
if ( keys.length === 0 ) {
return getAll();
}
const bin = {};
const toFetch = keys.slice();
const hasDefault = typeof arg === 'object' && Array.isArray(arg) === false;
for ( let i = 0; i < toFetch.length; i++ ) {
const key = toFetch[i];
if ( hasDefault && arg[key] !== undefined ) {
bin[key] = arg[key];
}
toFetch[i] = getOne(key);
}
const responses = await Promise.all(toFetch);
for ( const response of responses ) {
if ( response instanceof Object === false ) { continue; }
const { key, text } = response;
if ( typeof key !== 'string' ) { continue; }
if ( typeof text !== 'string' ) { continue; }
bin[key] = text;
}
if ( Object.keys(bin).length === 0 ) { return; }
return bin;
},
async keys(regex) {
const cache = await cacheStoragePromise;
if ( cache === null ) { return []; }
return cache.keys().then(requests =>
requests.map(r => urlToKey(r.url))
.filter(k => regex === undefined || regex.test(k))
).catch(( ) => []);
},
async set(keyvalStore) {
const keys = Object.keys(keyvalStore);
if ( keys.length === 0 ) { return; }
const promises = [];
for ( const key of keys ) {
promises.push(setOne(key, keyvalStore[key]));
}
return Promise.all(promises);
},
async remove(keys) {
const toRemove = [];
if ( typeof keys === 'string' ) {
toRemove.push(removeOne(keys));
} else if ( Array.isArray(keys) ) {
for ( const key of keys ) {
toRemove.push(removeOne(key));
}
}
return Promise.all(toRemove);
},
async clear() {
return globalThis.caches.delete(STORAGE_NAME).catch(reason => {
ubolog(reason);
});
},
};
})();
/*******************************************************************************
*
* IndexedDB
*
* Deprecated, exists only for the purpose of migrating from older versions.
*
* */
const idbStorage = (( ) => {
let dbPromise;
const getDb = function() { const getDb = function() {
keepAlive(); if ( dbPromise !== undefined ) { return dbPromise; }
if ( db !== undefined ) {
return Promise.resolve(db);
}
if ( dbPromise !== undefined ) {
return dbPromise;
}
dbPromise = new Promise(resolve => { dbPromise = new Promise(resolve => {
let req; let req;
try { try {
req = indexedDB.open(STORAGE_NAME, 1); req = indexedDB.open(STORAGE_NAME, 1);
if ( req.error ) { if ( req.error ) {
console.log(req.error); ubolog(req.error);
req = undefined; req = undefined;
} }
} catch(ex) { } catch(ex) {
} }
if ( req === undefined ) { if ( req === undefined ) {
db = null;
dbPromise = undefined;
return resolve(null); return resolve(null);
} }
req.onupgradeneeded = function(ev) { req.onupgradeneeded = function(ev) {
@ -215,24 +398,16 @@ const selectIDB = async function() {
req.onsuccess = function(ev) { req.onsuccess = function(ev) {
if ( resolve === undefined ) { return; } if ( resolve === undefined ) { return; }
req = undefined; req = undefined;
db = ev.target.result; resolve(ev.target.result);
dbPromise = undefined;
resolve(db);
resolve = undefined; resolve = undefined;
}; };
req.onerror = req.onblocked = function() { req.onerror = req.onblocked = function() {
if ( resolve === undefined ) { return; } if ( resolve === undefined ) { return; }
req = undefined;
console.log(this.error);
db = null;
dbPromise = undefined;
resolve(null); resolve(null);
resolve = undefined; resolve = undefined;
}; };
vAPI.defer.once(5000).then(( ) => { vAPI.defer.once(5000).then(( ) => {
if ( resolve === undefined ) { return; } if ( resolve === undefined ) { return; }
db = null;
dbPromise = undefined;
resolve(null); resolve(null);
resolve = undefined; resolve = undefined;
}); });
@ -253,60 +428,12 @@ const selectIDB = async function() {
}); });
}; };
const toBlob = function(data) {
const value = data instanceof Uint8Array
? new Blob([ data ])
: data;
return Promise.resolve(value);
};
const compress = function(store, key, data) {
return lz4Codec.encode(data, toBlob).then(value => {
store.push({ key, value });
});
};
const decompress = function(store, key, data) { const decompress = function(store, key, data) {
return lz4Codec.decode(data, fromBlob).then(data => { return lz4Codec.decode(data, fromBlob).then(data => {
store[key] = data; store[key] = data;
}); });
}; };
const getFromDb = async function(keys, keyvalStore, callback) {
if ( typeof callback !== 'function' ) { return; }
if ( keys.length === 0 ) { return callback(keyvalStore); }
const promises = [];
const gotOne = function() {
if ( typeof this.result !== 'object' ) { return; }
const { key, value } = this.result;
keyvalStore[key] = value;
if ( value instanceof Blob === false ) { return; }
promises.push(decompress(keyvalStore, key, value));
};
try {
const db = await getDb();
if ( !db ) { return callback(); }
const transaction = db.transaction(STORAGE_NAME, 'readonly');
transaction.oncomplete =
transaction.onerror =
transaction.onabort = ( ) => {
Promise.all(promises).then(( ) => {
callback(keyvalStore);
});
};
const table = transaction.objectStore(STORAGE_NAME);
for ( const key of keys ) {
const req = table.get(key);
req.onsuccess = gotOne;
req.onerror = noopfn;
}
}
catch(reason) {
console.info(`cacheStorage.getFromDb() failed: ${reason}`);
callback();
}
};
const visitAllFromDb = async function(visitFn) { const visitAllFromDb = async function(visitFn) {
const db = await getDb(); const db = await getDb();
if ( !db ) { return visitFn(); } if ( !db ) { return visitFn(); }
@ -341,190 +468,40 @@ const selectIDB = async function() {
if ( entry.value instanceof Blob === false ) { return; } if ( entry.value instanceof Blob === false ) { return; }
promises.push(decompress(keyvalStore, key, value)); promises.push(decompress(keyvalStore, key, value));
}).catch(reason => { }).catch(reason => {
console.info(`cacheStorage.getAllFromDb() failed: ${reason}`); ubolog(`cacheStorage.getAllFromDb() failed: ${reason}`);
callback(); callback();
}); });
}; };
// https://github.com/uBlockOrigin/uBlock-issues/issues/141
// Mind that IDBDatabase.transaction() and IDBObjectStore.put()
// can throw:
// https://developer.mozilla.org/en-US/docs/Web/API/IDBDatabase/transaction
// https://developer.mozilla.org/en-US/docs/Web/API/IDBObjectStore/put
const putToDb = async function(keyvalStore, callback) {
if ( typeof callback !== 'function' ) {
callback = noopfn;
}
const keys = Object.keys(keyvalStore);
if ( keys.length === 0 ) { return callback(); }
const promises = [ getDb() ];
const entries = [];
const dontCompress =
µb.hiddenSettings.cacheStorageCompression !== true;
for ( const key of keys ) {
const value = keyvalStore[key];
const isString = typeof value === 'string';
if ( isString === false || dontCompress ) {
entries.push({ key, value });
continue;
}
promises.push(compress(entries, key, value));
}
const finish = ( ) => {
if ( callback === undefined ) { return; }
let cb = callback;
callback = undefined;
cb();
};
try {
const results = await Promise.all(promises);
const db = results[0];
if ( !db ) { return callback(); }
const transaction = db.transaction(
STORAGE_NAME,
'readwrite'
);
transaction.oncomplete =
transaction.onerror =
transaction.onabort = finish;
const table = transaction.objectStore(STORAGE_NAME);
for ( const entry of entries ) {
table.put(entry);
}
} catch (ex) {
finish();
}
};
const deleteFromDb = async function(input, callback) {
if ( typeof callback !== 'function' ) {
callback = noopfn;
}
const keys = Array.isArray(input) ? input.slice() : [ input ];
if ( keys.length === 0 ) { return callback(); }
const finish = ( ) => {
if ( callback === undefined ) { return; }
let cb = callback;
callback = undefined;
cb();
};
try {
const db = await getDb();
if ( !db ) { return callback(); }
const transaction = db.transaction(STORAGE_NAME, 'readwrite');
transaction.oncomplete =
transaction.onerror =
transaction.onabort = finish;
const table = transaction.objectStore(STORAGE_NAME);
for ( const key of keys ) {
table.delete(key);
}
} catch (ex) {
finish();
}
};
const clearDb = async function(callback) { const clearDb = async function(callback) {
if ( typeof callback !== 'function' ) { if ( typeof callback !== 'function' ) {
callback = noopfn; callback = ()=>{};
} }
try { try {
const db = await getDb(); const db = await getDb();
if ( !db ) { return callback(); } if ( !db ) { return callback(); }
const transaction = db.transaction(STORAGE_NAME, 'readwrite'); db.close();
transaction.oncomplete = indexedDB.deleteDatabase(STORAGE_NAME);
transaction.onerror = callback();
transaction.onabort = ( ) => {
callback();
};
transaction.objectStore(STORAGE_NAME).clear();
} }
catch(reason) { catch(reason) {
console.info(`cacheStorage.clearDb() failed: ${reason}`);
callback(); callback();
} }
}; };
await getDb(); return {
if ( !db ) { return false; } get: function get() {
return new Promise(resolve => {
cacheStorage.name = 'indexedDB'; return getAllFromDb(bin => resolve(bin));
cacheStorage.get = function get(keys) { });
return storageReadyPromise.then(( ) => },
new Promise(resolve => { clear: function clear() {
if ( keys === null ) { return new Promise(resolve => {
return getAllFromDb(bin => resolve(bin));
}
let toRead, output = {};
if ( typeof keys === 'string' ) {
toRead = [ keys ];
} else if ( Array.isArray(keys) ) {
toRead = keys;
} else /* if ( typeof keys === 'object' ) */ {
toRead = Object.keys(keys);
output = keys;
}
getFromDb(toRead, output, bin => resolve(bin));
})
);
};
cacheStorage.set = function set(keys) {
return storageReadyPromise.then(( ) =>
new Promise(resolve => {
putToDb(keys, details => resolve(details));
})
);
};
cacheStorage.remove = function remove(keys) {
return storageReadyPromise.then(( ) =>
new Promise(resolve => {
deleteFromDb(keys, ( ) => resolve());
})
);
};
cacheStorage.clear = function clear() {
return storageReadyPromise.then(( ) =>
new Promise(resolve => {
clearDb(( ) => resolve()); clearDb(( ) => resolve());
}) });
); },
}; };
cacheStorage.getBytesInUse = function getBytesInUse() { })();
return Promise.resolve(0);
};
return true;
};
// https://github.com/uBlockOrigin/uBlock-issues/issues/328
// Delete cache-related entries from webext storage.
const clearWebext = async function() {
let bin;
try {
bin = await webext.storage.local.get('assetCacheRegistry');
} catch(ex) {
console.error(ex);
}
if ( bin instanceof Object === false ) { return; }
if ( bin.assetCacheRegistry instanceof Object === false ) { return; }
const toRemove = [
'assetCacheRegistry',
'assetSourceRegistry',
];
for ( const key in bin.assetCacheRegistry ) {
if ( bin.assetCacheRegistry.hasOwnProperty(key) ) {
toRemove.push('cache/' + key);
}
}
webext.storage.local.remove(toRemove);
};
const clearIDB = function() {
try {
indexedDB.deleteDatabase(STORAGE_NAME);
} catch(ex) {
}
};
/******************************************************************************/ /******************************************************************************/

View File

@ -292,7 +292,7 @@ FilterContainer.prototype.reset = function() {
this.highlyGeneric.complex.str = ''; this.highlyGeneric.complex.str = '';
this.highlyGeneric.complex.mru.reset(); this.highlyGeneric.complex.mru.reset();
this.selfieVersion = 1; this.selfieVersion = 2;
}; };
/******************************************************************************/ /******************************************************************************/
@ -576,9 +576,11 @@ FilterContainer.prototype.toSelfie = function() {
acceptedCount: this.acceptedCount, acceptedCount: this.acceptedCount,
discardedCount: this.discardedCount, discardedCount: this.discardedCount,
specificFilters: this.specificFilters.toSelfie(), specificFilters: this.specificFilters.toSelfie(),
lowlyGeneric: Array.from(this.lowlyGeneric), lowlyGeneric: this.lowlyGeneric,
highSimpleGenericHideArray: Array.from(this.highlyGeneric.simple.dict), highSimpleGenericHideDict: this.highlyGeneric.simple.dict,
highComplexGenericHideArray: Array.from(this.highlyGeneric.complex.dict), highSimpleGenericHideStr: this.highlyGeneric.simple.str,
highComplexGenericHideDict: this.highlyGeneric.complex.dict,
highComplexGenericHideStr: this.highlyGeneric.complex.str,
}; };
}; };
@ -593,11 +595,11 @@ FilterContainer.prototype.fromSelfie = function(selfie) {
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 = new Map(selfie.lowlyGeneric); this.lowlyGeneric = selfie.lowlyGeneric;
this.highlyGeneric.simple.dict = new Set(selfie.highSimpleGenericHideArray); this.highlyGeneric.simple.dict = selfie.highSimpleGenericHideDict;
this.highlyGeneric.simple.str = selfie.highSimpleGenericHideArray.join(',\n'); this.highlyGeneric.simple.str = selfie.highSimpleGenericHideStr;
this.highlyGeneric.complex.dict = new Set(selfie.highComplexGenericHideArray); this.highlyGeneric.complex.dict = selfie.highComplexGenericHideDict;
this.highlyGeneric.complex.str = selfie.highComplexGenericHideArray.join(',\n'); this.highlyGeneric.complex.str = selfie.highComplexGenericHideStr;
this.frozen = true; this.frozen = true;
}; };

View File

@ -445,28 +445,17 @@ class HNTrieContainer {
}; };
} }
serialize(encoder) { toSelfie() {
if ( encoder instanceof Object ) { return this.buf32.subarray(
return encoder.encode( 0,
this.buf32.buffer, this.buf32[CHAR1_SLOT] + 3 >>> 2
this.buf32[CHAR1_SLOT]
);
}
return Array.from(
new Uint32Array(
this.buf32.buffer,
0,
this.buf32[CHAR1_SLOT] + 3 >>> 2
)
); );
} }
unserialize(selfie, decoder) { fromSelfie(selfie) {
if ( selfie instanceof Uint32Array === false ) { return false; }
this.needle = ''; this.needle = '';
const shouldDecode = typeof selfie === 'string'; let byteLength = selfie.length << 2;
let byteLength = shouldDecode
? decoder.decodeSize(selfie)
: selfie.length << 2;
if ( byteLength === 0 ) { return false; } if ( byteLength === 0 ) { return false; }
byteLength = roundToPageSize(byteLength); byteLength = roundToPageSize(byteLength);
if ( this.wasmMemory !== null ) { if ( this.wasmMemory !== null ) {
@ -477,14 +466,10 @@ class HNTrieContainer {
this.buf = new Uint8Array(this.wasmMemory.buffer); this.buf = new Uint8Array(this.wasmMemory.buffer);
this.buf32 = new Uint32Array(this.buf.buffer); this.buf32 = new Uint32Array(this.buf.buffer);
} }
} else if ( byteLength > this.buf.length ) {
this.buf = new Uint8Array(byteLength);
this.buf32 = new Uint32Array(this.buf.buffer);
}
if ( shouldDecode ) {
decoder.decode(selfie, this.buf.buffer);
} else {
this.buf32.set(selfie); this.buf32.set(selfie);
} else {
this.buf32 = selfie;
this.buf = new Uint8Array(this.buf32.buffer);
} }
// https://github.com/uBlockOrigin/uBlock-issues/issues/2925 // https://github.com/uBlockOrigin/uBlock-issues/issues/2925
this.buf[255] = 0; this.buf[255] = 0;

View File

@ -45,6 +45,7 @@ import { dnrRulesetFromRawLists } from './static-dnr-filtering.js';
import { i18n$ } from './i18n.js'; import { i18n$ } from './i18n.js';
import { redirectEngine } from './redirect-engine.js'; import { redirectEngine } from './redirect-engine.js';
import * as sfp from './static-filtering-parser.js'; import * as sfp from './static-filtering-parser.js';
import * as scuo from './scuo-serializer.js';
import { import {
permanentFirewall, permanentFirewall,
@ -925,21 +926,6 @@ const fromBase64 = function(encoded) {
return Promise.resolve(u8array !== undefined ? u8array : encoded); return Promise.resolve(u8array !== undefined ? u8array : encoded);
}; };
const toBase64 = function(data) {
const value = data instanceof Uint8Array
? denseBase64.encode(data)
: data;
return Promise.resolve(value);
};
const compress = function(json) {
return lz4Codec.encode(json, toBase64);
};
const decompress = function(encoded) {
return lz4Codec.decode(encoded, fromBase64);
};
const onMessage = function(request, sender, callback) { const onMessage = function(request, sender, callback) {
// Cloud storage support is optional. // Cloud storage support is optional.
if ( µb.cloudStorageSupported !== true ) { if ( µb.cloudStorageSupported !== true ) {
@ -961,15 +947,25 @@ const onMessage = function(request, sender, callback) {
return; return;
case 'cloudPull': case 'cloudPull':
request.decode = decompress; request.decode = encoded => {
if ( scuo.canDeserialize(encoded) ) {
return scuo.deserializeAsync(encoded, { thread: true });
}
// Legacy decoding: needs to be kept around for the foreseeable future.
return lz4Codec.decode(encoded, fromBase64);
};
return vAPI.cloud.pull(request).then(result => { return vAPI.cloud.pull(request).then(result => {
callback(result); callback(result);
}); });
case 'cloudPush': case 'cloudPush':
if ( µb.hiddenSettings.cloudStorageCompression ) { request.encode = data => {
request.encode = compress; const options = {
} compress: µb.hiddenSettings.cloudStorageCompression,
thread: true,
};
return scuo.serializeAsync(data, options);
};
return vAPI.cloud.push(request).then(result => { return vAPI.cloud.push(request).then(result => {
callback(result); callback(result);
}); });

View File

@ -24,11 +24,7 @@
/******************************************************************************/ /******************************************************************************/
import redirectableResources from './redirect-resources.js'; import redirectableResources from './redirect-resources.js';
import { LineIterator, orphanizeString } from './text-utils.js';
import {
LineIterator,
orphanizeString,
} from './text-utils.js';
/******************************************************************************/ /******************************************************************************/
@ -448,33 +444,22 @@ class RedirectEngine {
} }
selfieFromResources(storage) { selfieFromResources(storage) {
storage.put( return storage.toCache(RESOURCES_SELFIE_NAME, {
RESOURCES_SELFIE_NAME, version: RESOURCES_SELFIE_VERSION,
JSON.stringify({ aliases: this.aliases,
version: RESOURCES_SELFIE_VERSION, resources: this.resources,
aliases: Array.from(this.aliases), });
resources: Array.from(this.resources),
})
);
} }
async resourcesFromSelfie(storage) { async resourcesFromSelfie(storage) {
const result = await storage.get(RESOURCES_SELFIE_NAME); const selfie = await storage.fromCache(RESOURCES_SELFIE_NAME);
let selfie; if ( selfie instanceof Object === false ) { return false; }
try { if ( selfie.version !== RESOURCES_SELFIE_VERSION ) { return false; }
selfie = JSON.parse(result.content); if ( selfie.aliases instanceof Map === false ) { return false; }
} catch(ex) { if ( selfie.resources instanceof Map === false ) { return false; }
} this.aliases = selfie.aliases;
if ( this.resources = selfie.resources;
selfie instanceof Object === false || for ( const [ token, entry ] of this.resources ) {
selfie.version !== RESOURCES_SELFIE_VERSION ||
Array.isArray(selfie.resources) === false
) {
return false;
}
this.aliases = new Map(selfie.aliases);
this.resources = new Map();
for ( const [ token, entry ] of selfie.resources ) {
this.resources.set(token, RedirectEntry.fromDetails(entry)); this.resources.set(token, RedirectEntry.fromDetails(entry));
} }
return true; return true;

View File

@ -62,7 +62,7 @@ const stopWorker = function() {
}; };
const workerTTLTimer = vAPI.defer.create(stopWorker); const workerTTLTimer = vAPI.defer.create(stopWorker);
const workerTTL = { min: 5 }; const workerTTL = { min: 1.5 };
const initWorker = function() { const initWorker = function() {
if ( worker === null ) { if ( worker === null ) {

View File

@ -200,7 +200,7 @@ export class ScriptletFilteringEngine {
} }
fromSelfie(selfie) { fromSelfie(selfie) {
if ( selfie instanceof Object === false ) { return false; } if ( typeof selfie !== 'object' || selfie === null ) { return false; }
if ( selfie.version !== VERSION ) { return false; } if ( selfie.version !== VERSION ) { return false; }
this.scriptletDB.fromSelfie(selfie); this.scriptletDB.fromSelfie(selfie);
return true; return true;

1307
src/js/scuo-serializer.js Normal file

File diff suppressed because it is too large Load Diff

View File

@ -139,7 +139,7 @@ const initializeTabs = async ( ) => {
// https://www.reddit.com/r/uBlockOrigin/comments/s7c9go/ // https://www.reddit.com/r/uBlockOrigin/comments/s7c9go/
// Abort suspending network requests when uBO is merely being installed. // Abort suspending network requests when uBO is merely being installed.
const onVersionReady = lastVersion => { const onVersionReady = async lastVersion => {
if ( lastVersion === vAPI.app.version ) { return; } if ( lastVersion === vAPI.app.version ) { return; }
vAPI.storage.set({ vAPI.storage.set({
@ -155,6 +155,11 @@ const onVersionReady = lastVersion => {
return; return;
} }
// Migrate cache storage
if ( lastVersionInt < vAPI.app.intFromVersion('1.56.1b1') ) {
await cacheStorage.migrate(µb.hiddenSettings.cacheStorageAPI);
}
// Since built-in resources may have changed since last version, we // Since built-in resources may have changed since last version, we
// force a reload of all resources. // force a reload of all resources.
redirectEngine.invalidateResourcesSelfie(io); redirectEngine.invalidateResourcesSelfie(io);
@ -252,19 +257,19 @@ const onUserSettingsReady = fetched => {
// Wait for removal of invalid cached data to be completed. // Wait for removal of invalid cached data to be completed.
const onCacheSettingsReady = async (fetched = {}) => { const onCacheSettingsReady = async (fetched = {}) => {
let selfieIsInvalid = false;
if ( fetched.compiledMagic !== µb.systemSettings.compiledMagic ) { if ( fetched.compiledMagic !== µb.systemSettings.compiledMagic ) {
µb.compiledFormatChanged = true; µb.compiledFormatChanged = true;
µb.selfieIsInvalid = true; selfieIsInvalid = true;
ubolog(`Serialized format of static filter lists changed`); ubolog(`Serialized format of static filter lists changed`);
} }
if ( fetched.selfieMagic !== µb.systemSettings.selfieMagic ) { if ( fetched.selfieMagic !== µb.systemSettings.selfieMagic ) {
µb.selfieIsInvalid = true; selfieIsInvalid = true;
ubolog(`Serialized format of selfie changed`); ubolog(`Serialized format of selfie changed`);
} }
if ( µb.selfieIsInvalid ) { if ( selfieIsInvalid === false ) { return; }
µb.selfieManager.destroy(); µb.selfieManager.destroy({ janitor: true });
cacheStorage.set(µb.systemSettings); cacheStorage.set(µb.systemSettings);
}
}; };
/******************************************************************************/ /******************************************************************************/
@ -305,10 +310,7 @@ const onHiddenSettingsReady = async ( ) => {
} }
// Maybe override default cache storage // Maybe override default cache storage
µb.supportStats.cacheBackend = await cacheStorage.select( µb.supportStats.cacheBackend = 'browser.storage.local';
µb.hiddenSettings.cacheStorageAPI
);
ubolog(`Backend storage for cache will be ${µb.supportStats.cacheBackend}`);
}; };
/******************************************************************************/ /******************************************************************************/
@ -333,7 +335,6 @@ const onFirstFetchReady = (fetched, adminExtra) => {
sessionSwitches.assign(permanentSwitches); sessionSwitches.assign(permanentSwitches);
onNetWhitelistReady(fetched.netWhitelist, adminExtra); onNetWhitelistReady(fetched.netWhitelist, adminExtra);
onVersionReady(fetched.version);
}; };
/******************************************************************************/ /******************************************************************************/
@ -389,23 +390,20 @@ try {
const adminExtra = await vAPI.adminStorage.get('toAdd'); const adminExtra = await vAPI.adminStorage.get('toAdd');
ubolog(`Extra admin settings ready ${Date.now()-vAPI.T0} ms after launch`); ubolog(`Extra admin settings ready ${Date.now()-vAPI.T0} ms after launch`);
// https://github.com/uBlockOrigin/uBlock-issues/issues/1365 const lastVersion = await vAPI.storage.get(createDefaultProps()).then(async fetched => {
// Wait for onCacheSettingsReady() to be fully ready. ubolog(`Version ready ${Date.now()-vAPI.T0} ms after launch`);
const [ , , lastVersion ] = await Promise.all([ await onVersionReady(fetched.version);
return fetched;
}).then(fetched => {
ubolog(`First fetch ready ${Date.now()-vAPI.T0} ms after launch`);
onFirstFetchReady(fetched, adminExtra);
return fetched.version;
});
await Promise.all([
µb.loadSelectedFilterLists().then(( ) => { µb.loadSelectedFilterLists().then(( ) => {
ubolog(`List selection ready ${Date.now()-vAPI.T0} ms after launch`); ubolog(`List selection ready ${Date.now()-vAPI.T0} ms after launch`);
}), }),
cacheStorage.get(
{ compiledMagic: 0, selfieMagic: 0 }
).then(fetched => {
ubolog(`Cache magic numbers ready ${Date.now()-vAPI.T0} ms after launch`);
onCacheSettingsReady(fetched);
}),
vAPI.storage.get(createDefaultProps()).then(fetched => {
ubolog(`First fetch ready ${Date.now()-vAPI.T0} ms after launch`);
onFirstFetchReady(fetched, adminExtra);
return fetched.version;
}),
µb.loadUserSettings().then(fetched => { µb.loadUserSettings().then(fetched => {
ubolog(`User settings ready ${Date.now()-vAPI.T0} ms after launch`); ubolog(`User settings ready ${Date.now()-vAPI.T0} ms after launch`);
onUserSettingsReady(fetched); onUserSettingsReady(fetched);
@ -413,6 +411,10 @@ try {
µb.loadPublicSuffixList().then(( ) => { µb.loadPublicSuffixList().then(( ) => {
ubolog(`PSL ready ${Date.now()-vAPI.T0} ms after launch`); ubolog(`PSL ready ${Date.now()-vAPI.T0} ms after launch`);
}), }),
cacheStorage.get({ compiledMagic: 0, selfieMagic: 0 }).then(bin => {
ubolog(`Cache magic numbers ready ${Date.now()-vAPI.T0} ms after launch`);
onCacheSettingsReady(bin);
}),
]); ]);
// https://github.com/uBlockOrigin/uBlock-issues/issues/1547 // https://github.com/uBlockOrigin/uBlock-issues/issues/1547

View File

@ -141,8 +141,8 @@ const StaticExtFilteringHostnameDB = class {
toSelfie() { toSelfie() {
return { return {
version: this.version, version: this.version,
hostnameToSlotIdMap: Array.from(this.hostnameToSlotIdMap), hostnameToSlotIdMap: this.hostnameToSlotIdMap,
regexToSlotIdMap: Array.from(this.regexToSlotIdMap), regexToSlotIdMap: this.regexToSlotIdMap,
hostnameSlots: this.hostnameSlots, hostnameSlots: this.hostnameSlots,
strSlots: this.strSlots, strSlots: this.strSlots,
size: this.size size: this.size
@ -150,11 +150,11 @@ const StaticExtFilteringHostnameDB = class {
} }
fromSelfie(selfie) { fromSelfie(selfie) {
if ( selfie === undefined ) { return; } if ( typeof selfie !== 'object' || selfie === null ) { return; }
this.hostnameToSlotIdMap = new Map(selfie.hostnameToSlotIdMap); this.hostnameToSlotIdMap = selfie.hostnameToSlotIdMap;
// Regex-based lookup available in uBO 1.47.0 and above // Regex-based lookup available in uBO 1.47.0 and above
if ( Array.isArray(selfie.regexToSlotIdMap) ) { if ( selfie.regexToSlotIdMap ) {
this.regexToSlotIdMap = new Map(selfie.regexToSlotIdMap); this.regexToSlotIdMap = selfie.regexToSlotIdMap;
} }
this.hostnameSlots = selfie.hostnameSlots; this.hostnameSlots = selfie.hostnameSlots;
this.strSlots = selfie.strSlots; this.strSlots = selfie.strSlots;

View File

@ -26,9 +26,8 @@
import cosmeticFilteringEngine from './cosmetic-filtering.js'; import cosmeticFilteringEngine from './cosmetic-filtering.js';
import htmlFilteringEngine from './html-filtering.js'; import htmlFilteringEngine from './html-filtering.js';
import httpheaderFilteringEngine from './httpheader-filtering.js'; import httpheaderFilteringEngine from './httpheader-filtering.js';
import io from './assets.js';
import logger from './logger.js';
import scriptletFilteringEngine from './scriptlet-filtering.js'; import scriptletFilteringEngine from './scriptlet-filtering.js';
import logger from './logger.js';
/******************************************************************************* /*******************************************************************************
@ -147,34 +146,24 @@ staticExtFilteringEngine.fromCompiledContent = function(reader, options) {
htmlFilteringEngine.fromCompiledContent(reader, options); htmlFilteringEngine.fromCompiledContent(reader, options);
}; };
staticExtFilteringEngine.toSelfie = function(path) { staticExtFilteringEngine.toSelfie = function() {
return io.put( return {
`${path}/main`, cosmetic: cosmeticFilteringEngine.toSelfie(),
JSON.stringify({ scriptlets: scriptletFilteringEngine.toSelfie(),
cosmetic: cosmeticFilteringEngine.toSelfie(), httpHeaders: httpheaderFilteringEngine.toSelfie(),
scriptlets: scriptletFilteringEngine.toSelfie(), html: htmlFilteringEngine.toSelfie(),
httpHeaders: httpheaderFilteringEngine.toSelfie(), };
html: htmlFilteringEngine.toSelfie(),
})
);
}; };
staticExtFilteringEngine.fromSelfie = function(path) { staticExtFilteringEngine.fromSelfie = async function(selfie) {
return io.get(`${path}/main`).then(details => { if ( typeof selfie !== 'object' || selfie === null ) { return false; }
let selfie; cosmeticFilteringEngine.fromSelfie(selfie.cosmetic);
try { httpheaderFilteringEngine.fromSelfie(selfie.httpHeaders);
selfie = JSON.parse(details.content); htmlFilteringEngine.fromSelfie(selfie.html);
} catch (ex) { if ( scriptletFilteringEngine.fromSelfie(selfie.scriptlets) === false ) {
} return false;
if ( selfie instanceof Object === false ) { return false; } }
cosmeticFilteringEngine.fromSelfie(selfie.cosmetic); return true;
httpheaderFilteringEngine.fromSelfie(selfie.httpHeaders);
htmlFilteringEngine.fromSelfie(selfie.html);
if ( scriptletFilteringEngine.fromSelfie(selfie.scriptlets) === false ) {
return false;
}
return true;
});
}; };
/******************************************************************************/ /******************************************************************************/

View File

@ -28,7 +28,6 @@
import { queueTask, dropTask } from './tasks.js'; import { queueTask, dropTask } from './tasks.js';
import BidiTrieContainer from './biditrie.js'; import BidiTrieContainer from './biditrie.js';
import HNTrieContainer from './hntrie.js'; import HNTrieContainer from './hntrie.js';
import { sparseBase64 } from './base64-custom.js';
import { CompiledListReader } from './static-filtering-io.js'; import { CompiledListReader } from './static-filtering-io.js';
import * as sfp from './static-filtering-parser.js'; import * as sfp from './static-filtering-parser.js';
@ -493,17 +492,13 @@ const filterDataReset = ( ) => {
filterData.fill(0); filterData.fill(0);
filterDataWritePtr = 2; filterDataWritePtr = 2;
}; };
const filterDataToSelfie = ( ) => { const filterDataToSelfie = ( ) =>
return JSON.stringify(Array.from(filterData.subarray(0, filterDataWritePtr))); filterData.subarray(0, filterDataWritePtr);
};
const filterDataFromSelfie = selfie => { const filterDataFromSelfie = selfie => {
if ( typeof selfie !== 'string' || selfie === '' ) { return false; } if ( selfie instanceof Int32Array === false ) { return false; }
const data = JSON.parse(selfie); filterData = selfie;
if ( Array.isArray(data) === false ) { return false; } filterDataWritePtr = selfie.length;
filterDataGrow(data.length);
filterDataWritePtr = data.length;
filterData.set(data);
filterDataShrink();
return true; return true;
}; };
@ -519,53 +514,15 @@ const filterRefsReset = ( ) => {
filterRefs.fill(null); filterRefs.fill(null);
filterRefsWritePtr = 1; filterRefsWritePtr = 1;
}; };
const filterRefsToSelfie = ( ) => { const filterRefsToSelfie = ( ) =>
const refs = []; filterRefs.slice(0, filterRefsWritePtr);
for ( let i = 0; i < filterRefsWritePtr; i++ ) {
const v = filterRefs[i];
if ( v instanceof RegExp ) {
refs.push({ t: 1, s: v.source, f: v.flags });
continue;
}
if ( Array.isArray(v) ) {
refs.push({ t: 2, v });
continue;
}
if ( typeof v !== 'object' || v === null ) {
refs.push({ t: 0, v });
continue;
}
const out = Object.create(null);
for ( const prop of Object.keys(v) ) {
const value = v[prop];
out[prop] = prop.startsWith('$')
? (typeof value === 'string' ? '' : null)
: value;
}
refs.push({ t: 3, v: out });
}
return JSON.stringify(refs);
};
const filterRefsFromSelfie = selfie => { const filterRefsFromSelfie = selfie => {
if ( typeof selfie !== 'string' || selfie === '' ) { return false; } if ( Array.isArray(selfie) === false ) { return false; }
const refs = JSON.parse(selfie); for ( let i = 0, n = selfie.length; i < n; i++ ) {
if ( Array.isArray(refs) === false ) { return false; } filterRefs[i] = selfie[i];
for ( let i = 0; i < refs.length; i++ ) {
const v = refs[i];
switch ( v.t ) {
case 0:
case 2:
case 3:
filterRefs[i] = v.v;
break;
case 1:
filterRefs[i] = new RegExp(v.s, v.f);
break;
default:
throw new Error('Unknown filter reference!');
}
} }
filterRefsWritePtr = refs.length; filterRefsWritePtr = selfie.length;
return true; return true;
}; };
@ -3121,14 +3078,11 @@ const urlTokenizer = new (class {
} }
toSelfie() { toSelfie() {
return sparseBase64.encode( return this.knownTokens;
this.knownTokens.buffer,
this.knownTokens.byteLength
);
} }
fromSelfie(selfie) { fromSelfie(selfie) {
return sparseBase64.decode(selfie, this.knownTokens.buffer); this.knownTokens = selfie;
} }
// https://github.com/chrisaljoudi/uBlock/issues/1118 // https://github.com/chrisaljoudi/uBlock/issues/1118
@ -4674,52 +4628,24 @@ FilterContainer.prototype.optimize = function(throttle = 0) {
/******************************************************************************/ /******************************************************************************/
FilterContainer.prototype.toSelfie = async function(storage, path) { FilterContainer.prototype.toSelfie = function() {
if ( typeof storage !== 'object' || storage === null ) { return; }
if ( typeof storage.put !== 'function' ) { return; }
bidiTrieOptimize(true); bidiTrieOptimize(true);
keyvalStore.setItem( keyvalStore.setItem('SNFE.origHNTrieContainer.trieDetails',
'SNFE.origHNTrieContainer.trieDetails',
origHNTrieContainer.optimize() origHNTrieContainer.optimize()
); );
return {
return Promise.all([ version: this.selfieVersion,
storage.put( processedFilterCount: this.processedFilterCount,
`${path}/destHNTrieContainer`, acceptedCount: this.acceptedCount,
destHNTrieContainer.serialize(sparseBase64) discardedCount: this.discardedCount,
), bitsToBucket: this.bitsToBucket,
storage.put( urlTokenizer: urlTokenizer.toSelfie(),
`${path}/origHNTrieContainer`, destHNTrieContainer: destHNTrieContainer.toSelfie(),
origHNTrieContainer.serialize(sparseBase64) origHNTrieContainer: origHNTrieContainer.toSelfie(),
), bidiTrie: bidiTrie.toSelfie(),
storage.put( filterData: filterDataToSelfie(),
`${path}/bidiTrie`, filterRefs: filterRefsToSelfie(),
bidiTrie.serialize(sparseBase64) };
),
storage.put(
`${path}/filterData`,
filterDataToSelfie()
),
storage.put(
`${path}/filterRefs`,
filterRefsToSelfie()
),
storage.put(
`${path}/main`,
JSON.stringify({
version: this.selfieVersion,
processedFilterCount: this.processedFilterCount,
acceptedCount: this.acceptedCount,
discardedCount: this.discardedCount,
bitsToBucket: Array.from(this.bitsToBucket).map(kv => {
kv[1] = Array.from(kv[1]);
return kv;
}),
urlTokenizer: urlTokenizer.toSelfie(),
})
)
]);
}; };
FilterContainer.prototype.serialize = async function() { FilterContainer.prototype.serialize = async function() {
@ -4735,53 +4661,27 @@ FilterContainer.prototype.serialize = async function() {
/******************************************************************************/ /******************************************************************************/
FilterContainer.prototype.fromSelfie = async function(storage, path) { FilterContainer.prototype.fromSelfie = async function(selfie) {
if ( typeof storage !== 'object' || storage === null ) { return; } if ( typeof selfie !== 'object' || selfie === null ) { return; }
if ( typeof storage.get !== 'function' ) { return; }
this.reset(); this.reset();
this.notReady = true; this.notReady = true;
const results = await Promise.all([ const results = [
storage.get(`${path}/main`), destHNTrieContainer.fromSelfie(selfie.destHNTrieContainer),
storage.get(`${path}/destHNTrieContainer`).then(details => origHNTrieContainer.fromSelfie(selfie.origHNTrieContainer),
destHNTrieContainer.unserialize(details.content, sparseBase64) bidiTrie.fromSelfie(selfie.bidiTrie),
), filterDataFromSelfie(selfie.filterData),
storage.get(`${path}/origHNTrieContainer`).then(details => filterRefsFromSelfie(selfie.filterRefs),
origHNTrieContainer.unserialize(details.content, sparseBase64) ];
),
storage.get(`${path}/bidiTrie`).then(details =>
bidiTrie.unserialize(details.content, sparseBase64)
),
storage.get(`${path}/filterData`).then(details =>
filterDataFromSelfie(details.content)
),
storage.get(`${path}/filterRefs`).then(details =>
filterRefsFromSelfie(details.content)
),
]);
if ( results.slice(1).every(v => v === true) === false ) { return false; } if ( results.slice(1).every(v => v === true) === false ) { return false; }
const details = results[0];
if ( typeof details !== 'object' || details === null ) { return false; }
if ( typeof details.content !== 'string' ) { return false; }
if ( details.content === '' ) { return false; }
let selfie;
try {
selfie = JSON.parse(details.content);
} catch (ex) {
}
if ( typeof selfie !== 'object' || selfie === null ) { return false; }
if ( selfie.version !== this.selfieVersion ) { return false; } if ( selfie.version !== this.selfieVersion ) { return false; }
this.processedFilterCount = selfie.processedFilterCount; this.processedFilterCount = selfie.processedFilterCount;
this.acceptedCount = selfie.acceptedCount; this.acceptedCount = selfie.acceptedCount;
this.discardedCount = selfie.discardedCount; this.discardedCount = selfie.discardedCount;
this.bitsToBucket = new Map(selfie.bitsToBucket.map(kv => { this.bitsToBucket = selfie.bitsToBucket;
kv[1] = new Map(kv[1]);
return kv;
}));
urlTokenizer.fromSelfie(selfie.urlTokenizer); urlTokenizer.fromSelfie(selfie.urlTokenizer);
// If this point is never reached, it means the internal state is // If this point is never reached, it means the internal state is

View File

@ -38,7 +38,6 @@ import µb from './background.js';
import { hostnameFromURI } from './uri-utils.js'; import { hostnameFromURI } from './uri-utils.js';
import { i18n, i18n$ } from './i18n.js'; import { i18n, i18n$ } from './i18n.js';
import { redirectEngine } from './redirect-engine.js'; import { redirectEngine } from './redirect-engine.js';
import { sparseBase64 } from './base64-custom.js';
import { ubolog, ubologSet } from './console.js'; import { ubolog, ubologSet } from './console.js';
import * as sfp from './static-filtering-parser.js'; import * as sfp from './static-filtering-parser.js';
@ -974,7 +973,7 @@ onBroadcast(msg => {
/******************************************************************************/ /******************************************************************************/
µb.getCompiledFilterList = async function(assetKey) { µb.getCompiledFilterList = async function(assetKey) {
const compiledPath = 'compiled/' + assetKey; const compiledPath = `compiled/${assetKey}`;
// https://github.com/uBlockOrigin/uBlock-issues/issues/1365 // https://github.com/uBlockOrigin/uBlock-issues/issues/1365
// Verify that the list version matches that of the current compiled // Verify that the list version matches that of the current compiled
@ -983,11 +982,10 @@ onBroadcast(msg => {
this.compiledFormatChanged === false && this.compiledFormatChanged === false &&
this.badLists.has(assetKey) === false this.badLists.has(assetKey) === false
) { ) {
const compiledDetails = await io.get(compiledPath); const content = await io.fromCache(compiledPath);
const compilerVersion = `${this.systemSettings.compiledMagic}\n`; const compilerVersion = `${this.systemSettings.compiledMagic}\n`;
if ( compiledDetails.content.startsWith(compilerVersion) ) { if ( content.startsWith(compilerVersion) ) {
compiledDetails.assetKey = assetKey; return { assetKey, content };
return compiledDetails;
} }
} }
@ -1017,7 +1015,7 @@ onBroadcast(msg => {
assetKey, assetKey,
trustedSource: this.isTrustedList(assetKey), trustedSource: this.isTrustedList(assetKey),
}); });
io.put(compiledPath, compiledContent); io.toCache(compiledPath, compiledContent);
return { assetKey, content: compiledContent }; return { assetKey, content: compiledContent };
}; };
@ -1046,7 +1044,7 @@ onBroadcast(msg => {
/******************************************************************************/ /******************************************************************************/
µb.removeCompiledFilterList = function(assetKey) { µb.removeCompiledFilterList = function(assetKey) {
io.remove('compiled/' + assetKey); io.remove(`compiled/${assetKey}`);
}; };
µb.removeFilterList = function(assetKey) { µb.removeFilterList = function(assetKey) {
@ -1173,20 +1171,17 @@ onBroadcast(msg => {
const results = await Promise.all(fetchPromises); const results = await Promise.all(fetchPromises);
if ( Array.isArray(results) === false ) { return results; } if ( Array.isArray(results) === false ) { return results; }
let content = ''; const content = [];
for ( let i = 1; i < results.length; i++ ) { for ( let i = 1; i < results.length; i++ ) {
const result = results[i]; const result = results[i];
if ( if ( result instanceof Object === false ) { continue; }
result instanceof Object === false || if ( typeof result.content !== 'string' ) { continue; }
typeof result.content !== 'string' || if ( result.content === '' ) { continue; }
result.content === '' content.push(result.content);
) { }
continue; if ( content.length !== 0 ) {
} redirectEngine.resourcesFromString(content.join('\n\n'));
content += '\n\n' + result.content;
} }
redirectEngine.resourcesFromString(content);
redirectEngine.selfieFromResources(io); redirectEngine.selfieFromResources(io);
} catch(ex) { } catch(ex) {
ubolog(ex); ubolog(ex);
@ -1225,8 +1220,8 @@ onBroadcast(msg => {
} }
try { try {
const result = await io.get(`compiled/${this.pslAssetKey}`); const selfie = await io.fromCache(`compiled/${this.pslAssetKey}`);
if ( psl.fromSelfie(result.content, sparseBase64) ) { return; } if ( psl.fromSelfie(selfie) ) { return; }
} catch (reason) { } catch (reason) {
ubolog(reason); ubolog(reason);
} }
@ -1240,7 +1235,7 @@ onBroadcast(msg => {
µb.compilePublicSuffixList = function(content) { µb.compilePublicSuffixList = function(content) {
const psl = publicSuffixList; const psl = publicSuffixList;
psl.parse(content, punycode.toASCII); psl.parse(content, punycode.toASCII);
io.put(`compiled/${this.pslAssetKey}`, psl.toSelfie(sparseBase64)); return io.toCache(`compiled/${this.pslAssetKey}`, psl.toSelfie());
}; };
/******************************************************************************/ /******************************************************************************/
@ -1260,39 +1255,24 @@ onBroadcast(msg => {
if ( µb.inMemoryFilters.length !== 0 ) { return; } if ( µb.inMemoryFilters.length !== 0 ) { return; }
if ( Object.keys(µb.availableFilterLists).length === 0 ) { return; } if ( Object.keys(µb.availableFilterLists).length === 0 ) { return; }
await Promise.all([ await Promise.all([
io.put( io.toCache('selfie/main', {
'selfie/main', magic: µb.systemSettings.selfieMagic,
JSON.stringify({ availableFilterLists: µb.availableFilterLists,
magic: µb.systemSettings.selfieMagic, }),
availableFilterLists: µb.availableFilterLists, io.toCache('selfie/staticExtFilteringEngine',
}) staticExtFilteringEngine.toSelfie()
), ),
redirectEngine.toSelfie('selfie/redirectEngine'), io.toCache('selfie/staticNetFilteringEngine',
staticExtFilteringEngine.toSelfie( staticNetFilteringEngine.toSelfie()
'selfie/staticExtFilteringEngine'
),
staticNetFilteringEngine.toSelfie(io,
'selfie/staticNetFilteringEngine'
), ),
]); ]);
lz4Codec.relinquish(); lz4Codec.relinquish();
µb.selfieIsInvalid = false; µb.selfieIsInvalid = false;
ubolog(`Selfie was created`);
}; };
const loadMain = async function() { const loadMain = async function() {
const details = await io.get('selfie/main'); const selfie = await io.fromCache('selfie/main');
if (
details instanceof Object === false ||
typeof details.content !== 'string' ||
details.content === ''
) {
return false;
}
let selfie;
try {
selfie = JSON.parse(details.content);
} catch(ex) {
}
if ( selfie instanceof Object === false ) { return false; } if ( selfie instanceof Object === false ) { return false; }
if ( selfie.magic !== µb.systemSettings.selfieMagic ) { return false; } if ( selfie.magic !== µb.systemSettings.selfieMagic ) { return false; }
if ( selfie.availableFilterLists instanceof Object === false ) { return false; } if ( selfie.availableFilterLists instanceof Object === false ) { return false; }
@ -1306,12 +1286,11 @@ onBroadcast(msg => {
try { try {
const results = await Promise.all([ const results = await Promise.all([
loadMain(), loadMain(),
redirectEngine.fromSelfie('selfie/redirectEngine'), io.fromCache('selfie/staticExtFilteringEngine').then(selfie =>
staticExtFilteringEngine.fromSelfie( staticExtFilteringEngine.fromSelfie(selfie)
'selfie/staticExtFilteringEngine'
), ),
staticNetFilteringEngine.fromSelfie(io, io.fromCache('selfie/staticNetFilteringEngine').then(selfie =>
'selfie/staticNetFilteringEngine' staticNetFilteringEngine.fromSelfie(selfie)
), ),
]); ]);
if ( results.every(v => v) ) { if ( results.every(v => v) ) {
@ -1325,10 +1304,11 @@ onBroadcast(msg => {
return false; return false;
}; };
const destroy = function() { const destroy = function(options = {}) {
if ( µb.selfieIsInvalid === false ) { if ( µb.selfieIsInvalid === false ) {
io.remove(/^selfie\//); io.remove(/^selfie\//, options);
µb.selfieIsInvalid = true; µb.selfieIsInvalid = true;
ubolog(`Selfie was removed`);
} }
if ( µb.wakeupReason === 'createSelfie' ) { if ( µb.wakeupReason === 'createSelfie' ) {
µb.wakeupReason = ''; µb.wakeupReason = '';
@ -1594,8 +1574,7 @@ onBroadcast(msg => {
if ( topic === 'after-asset-updated' ) { if ( topic === 'after-asset-updated' ) {
// Skip selfie-related content. // Skip selfie-related content.
if ( details.assetKey.startsWith('selfie/') ) { return; } if ( details.assetKey.startsWith('selfie/') ) { return; }
const cached = typeof details.content === 'string' && const cached = typeof details.content === 'string' && details.content !== '';
details.content !== '';
if ( this.availableFilterLists.hasOwnProperty(details.assetKey) ) { if ( this.availableFilterLists.hasOwnProperty(details.assetKey) ) {
if ( cached ) { if ( cached ) {
if ( this.selectedFilterLists.indexOf(details.assetKey) !== -1 ) { if ( this.selectedFilterLists.indexOf(details.assetKey) !== -1 ) {
@ -1604,8 +1583,7 @@ onBroadcast(msg => {
details.content details.content
); );
if ( this.badLists.has(details.assetKey) === false ) { if ( this.badLists.has(details.assetKey) === false ) {
io.put( io.toCache(`compiled/${details.assetKey}`,
'compiled/' + details.assetKey,
this.compileFilters(details.content, { this.compileFilters(details.content, {
assetKey: details.assetKey, assetKey: details.assetKey,
trustedSource: this.isTrustedList(details.assetKey), trustedSource: this.isTrustedList(details.assetKey),

View File

@ -13,8 +13,6 @@
/*! Home: https://github.com/gorhill/publicsuffixlist.js -- GPLv3 APLv2 */ /*! Home: https://github.com/gorhill/publicsuffixlist.js -- GPLv3 APLv2 */
/* globals WebAssembly, exports:true, module */
'use strict'; 'use strict';
/******************************************************************************* /*******************************************************************************
@ -70,7 +68,7 @@ const RULES_PTR_SLOT = 100; // 100 / 400 (400-256=144 => 144>128)
const SUFFIX_NOT_FOUND_SLOT = 399; // -- / 399 (safe, see above) const SUFFIX_NOT_FOUND_SLOT = 399; // -- / 399 (safe, see above)
const CHARDATA_PTR_SLOT = 101; // 101 / 404 const CHARDATA_PTR_SLOT = 101; // 101 / 404
const EMPTY_STRING = ''; const EMPTY_STRING = '';
const SELFIE_MAGIC = 2; const SELFIE_MAGIC = 3;
let wasmMemory; let wasmMemory;
let pslBuffer32; let pslBuffer32;
@ -499,9 +497,7 @@ const toSelfie = function(encoder) {
} }
return { return {
magic: SELFIE_MAGIC, magic: SELFIE_MAGIC,
buf32: Array.from( buf32: pslBuffer32.subarray(0, pslByteLength >> 2),
new Uint32Array(pslBuffer8.buffer, 0, pslByteLength >>> 2)
),
}; };
}; };
@ -524,7 +520,7 @@ const fromSelfie = function(selfie, decoder) {
} else if ( } else if (
selfie instanceof Object && selfie instanceof Object &&
selfie.magic === SELFIE_MAGIC && selfie.magic === SELFIE_MAGIC &&
Array.isArray(selfie.buf32) selfie.buf32 instanceof Uint32Array
) { ) {
byteLength = selfie.buf32.length << 2; byteLength = selfie.buf32.length << 2;
allocateBuffers(byteLength); allocateBuffers(byteLength);