v_jstools/tools/window_hooker.js
2021-10-30 00:54:17 +08:00

110 lines
5.4 KiB
JavaScript

// 暂时还在考虑的一种 window hook 方式。需要配合全局 ast 代码修改的方式
function fetch_hook(code){
var newn = t.ConditionalExpression(
t.BinaryExpression('===', t.ThisExpression(), t.Identifier('v_window')),
t.Identifier('v_win'),
t.ThisExpression()
)
function protect_this(path){
var node = path.node
path.replaceWith(newn)
path.stop()
}
var ast = parser.parse(code, {allowReturnOutsideFunction: true});
traverse(ast, {ThisExpression: protect_this});
var { code } = generator(ast, { jsescOption: { minimal: true, } });
return `
// 暂时还在考虑的一种 window hook 方式。需要配合全局 ast 代码修改的方式
;(function(){
var cache = {}
function make_cache_hooker(obj, name){
if (name in cache){ return cache[name] }
return cache[name] = new Proxy(obj, {
set: function(a,b,c){ return filter_log('set', b, c), obj[b]=c },
get: function(a,b){
var r = obj[b]
if (!(b == Symbol.unscopables || b == Symbol.toStringTag || b == Symbol.toPrimitive)){ filter_log(name, 'get', b, r) }
if (typeof r == 'function'){ return r.bind(obj) }
return r
},
})
}
var filter_log = console.log
!function make_fake_window(){
if (window.v_win){ return }
var _win = {}
function mainobj(b, r){
switch(b){
case 'window':
case 'self':
case 'top':
case 'frames': // 这两个可能存在问题
case 'parent': // 这两个可能存在问题
case 'globalThis': r = win; break
case 'clientInformation': r = make_cache_hooker(clientInformation, 'window.clientInformation'); break
case 'crypto': r = make_cache_hooker(crypto, 'window.crypto'); break
case 'customElements': r = make_cache_hooker(customElements, 'window.customElements'); break
case 'document': r = make_cache_hooker(document, 'window.document'); break
case 'external': r = make_cache_hooker(external, 'window.external'); break
case 'history': r = make_cache_hooker(history, 'window.history'); break
case 'indexedDB': r = make_cache_hooker(indexedDB, 'window.indexedDB'); break
case 'localStorage': r = make_cache_hooker(localStorage, 'window.localStorage'); break
case 'locationbar': r = make_cache_hooker(locationbar, 'window.locationbar'); break
case 'menubar': r = make_cache_hooker(menubar, 'window.menubar'); break
case 'navigator': r = make_cache_hooker(navigator, 'window.navigator'); break
case 'performance': r = make_cache_hooker(performance, 'window.performance'); break
case 'personalbar': r = make_cache_hooker(personalbar, 'window.personalbar'); break
case 'screen': r = make_cache_hooker(screen, 'window.screen'); break
case 'scrollbars': r = make_cache_hooker(scrollbars, 'window.scrollbars'); break
case 'sessionStorage': r = make_cache_hooker(sessionStorage, 'window.sessionStorage'); break
case 'statusbar': r = make_cache_hooker(statusbar, 'window.statusbar'); break
case 'toolbar': r = make_cache_hooker(toolbar, 'window.toolbar'); break
case 'trustedTypes': r = make_cache_hooker(trustedTypes, 'window.trustedTypes'); break
case 'visualViewport': r = make_cache_hooker(visualViewport, 'window.visualViewport'); break
case 'location': r = make_cache_hooker(location, 'window.location'); break
default: r = window[b]; break
}
return r
}
var unlogs = [
'undefined',
'v_window',
]
var localeval = eval
var win = new Proxy(_win, {
has: function(a,b){ return true },
set: function(a,b,c){ return filter_log('window set', b, c), window[b]=c },
get: function(a,b){
var r = mainobj(b)
if (!(b == Symbol.unscopables || b == Symbol.toStringTag || b == Symbol.toPrimitive || unlogs.indexOf(b) != -1)){ filter_log('window', 'get', b, r) }
if (typeof r == 'function' && !r.prototype){ return r.bind(window) }
return r
},
})
var interceptor = new Proxy(_win, {
has: function(a,b){ return b in window }, // win 和 interceptor 的区别在这里
set: function(a,b,c){ return filter_log('window set', b, c), window[b]=c },
get: function(a,b){
if (!(b in window) && typeof b != 'symbol'){ throw ReferenceError(b + ' is not defined') } // win 和 interceptor 的区别在这里
var r = mainobj(b)
if (!(b == Symbol.unscopables || b == Symbol.toStringTag || b == Symbol.toPrimitive || unlogs.indexOf(b) != -1)){ filter_log('window', 'get', b, r) }
if (typeof r == 'function' && !r.prototype){ return r.bind(window) }
return r
},
})
window.v_win = win
window.v_interceptor = interceptor
window.v_window = window
// window.v_eval = eval
}()
})()
with (window.v_interceptor){
${code}
}
`
}