Expand/harden some scriptlets

Related issue:
- https://github.com/uBlockOrigin/uBlock-issues/issues/2615

Expand `set-constant`: 3rd parameter and beyond are tokens which
modify the behavior of `set-contant`. Valid tokens:

- `interactive`, `end`, `2`: set the constant when the event
  `DOMContentInteractive` is fired.
- `complete`, `idle`, `3`: set the constant when the event
  `load` is fired.
- `asFunction`: the constant will be a function returning the
  specified value.
- `asCallback`: the constant will be a function returning a
  function returning the specified value.
- `asResolved`: the constant will be a promise resolving to
  the specified value.
- `asRejected`: the constant will be a promise failing with
  the specified value.

Harden `no-setimeout-if` and `no-setinterval-if` as per feedback
from filter list maintainers.
This commit is contained in:
Raymond Hill 2023-05-19 12:55:01 -04:00
parent 0442718efd
commit 2ef2888805
No known key found for this signature in database
GPG Key ID: 25E1490B761470C2
1 changed files with 48 additions and 14 deletions

View File

@ -130,13 +130,18 @@ builtinScriptlets.push({
}); });
function runAt(fn, when) { function runAt(fn, when) {
const intFromReadyState = state => { const intFromReadyState = state => {
return ({ const targets = {
loading: 1, 'loading': 1,
interactive: 2, 'interactive': 2, 'end': 2, '2': 2,
end: 2, 'complete': 3, 'idle': 3, '3': 3,
complete: 3, };
idle: 3, const tokens = Array.isArray(state) ? state : [ state ];
})[`${state}`] || 0; for ( const token of tokens ) {
const prop = `${token}`;
if ( targets.hasOwnProperty(prop) === false ) { continue; }
return targets[prop];
}
return 0;
}; };
const runAt = intFromReadyState(when); const runAt = intFromReadyState(when);
if ( intFromReadyState(document.readyState) >= runAt ) { if ( intFromReadyState(document.readyState) >= runAt ) {
@ -1030,14 +1035,22 @@ builtinScriptlets.push({
function setConstant( function setConstant(
arg1 = '', arg1 = '',
arg2 = '', arg2 = '',
arg3 = 0 arg3 = ''
) { ) {
const details = typeof arg1 !== 'object' const details = typeof arg1 !== 'object'
? { prop: arg1, value: arg2, runAt: parseInt(arg3, 10) || 0 } ? { prop: arg1, value: arg2 }
: arg1; : arg1;
if ( arg3 !== '' ) {
if ( /^\d$/.test(arg3) ) {
details.options = [ arg3 ];
} else {
details.options = Array.from(arguments).slice(2);
}
}
const { prop: chain = '', value: cValue = '' } = details; const { prop: chain = '', value: cValue = '' } = details;
if ( typeof chain !== 'string' ) { return; } if ( typeof chain !== 'string' ) { return; }
if ( chain === '' ) { return; } if ( chain === '' ) { return; }
const options = details.options || [];
function setConstant(chain, cValue) { function setConstant(chain, cValue) {
const trappedProp = (( ) => { const trappedProp = (( ) => {
const pos = chain.lastIndexOf('.'); const pos = chain.lastIndexOf('.');
@ -1093,13 +1106,22 @@ function setConstant(
cValue = cloakFunc(function(){ return true; }); cValue = cloakFunc(function(){ return true; });
} else if ( cValue === 'falseFunc' ) { } else if ( cValue === 'falseFunc' ) {
cValue = cloakFunc(function(){ return false; }); cValue = cloakFunc(function(){ return false; });
} else if ( /^\d+$/.test(cValue) ) { } else if ( /^-?\d+$/.test(cValue) ) {
cValue = parseFloat(cValue); cValue = parseInt(cValue);
if ( isNaN(cValue) ) { return; } if ( isNaN(cValue) ) { return; }
if ( Math.abs(cValue) > 0x7FFF ) { return; } if ( Math.abs(cValue) > 0x7FFF ) { return; }
} else { } else {
return; return;
} }
if ( options.includes('asFunction') ) {
cValue = ( ) => cValue;
} else if ( options.includes('asCallback') ) {
cValue = ( ) => (( ) => cValue);
} else if ( options.includes('asResolved') ) {
cValue = Promise.resolve(cValue);
} else if ( options.includes('asRejected') ) {
cValue = Promise.reject(cValue);
}
let aborted = false; let aborted = false;
const mustAbort = function(v) { const mustAbort = function(v) {
if ( aborted ) { return true; } if ( aborted ) { return true; }
@ -1193,7 +1215,7 @@ function setConstant(
} }
runAt(( ) => { runAt(( ) => {
setConstant(chain, cValue); setConstant(chain, cValue);
}, details.runAt); }, options);
} }
/******************************************************************************/ /******************************************************************************/
@ -1243,7 +1265,13 @@ function noSetIntervalIf(
} }
} }
return target.apply(thisArg, args); return target.apply(thisArg, args);
},
get(target, prop, receiver) {
if ( prop === 'toString' ) {
return target.toString.bind(target);
} }
return Reflect.get(target, prop, receiver);
},
}); });
} }
@ -1294,7 +1322,13 @@ function noSetTimeoutIf(
} }
} }
return target.apply(thisArg, args); return target.apply(thisArg, args);
},
get(target, prop, receiver) {
if ( prop === 'toString' ) {
return target.toString.bind(target);
} }
return Reflect.get(target, prop, receiver);
},
}); });
} }