84 lines
2.6 KiB
JavaScript
84 lines
2.6 KiB
JavaScript
'use strict'
|
|
|
|
const { PuppeteerExtraPlugin } = require('puppeteer-extra-plugin')
|
|
|
|
/**
|
|
* Strip sourceURL from scripts injected by puppeteer.
|
|
* It can be used to identify the presence of pptr via stacktraces.
|
|
*/
|
|
class Plugin extends PuppeteerExtraPlugin {
|
|
constructor(opts = {}) {
|
|
super(opts)
|
|
}
|
|
|
|
get name() {
|
|
return 'stealth/evasions/sourceurl'
|
|
}
|
|
|
|
async onPageCreated(page) {
|
|
const client =
|
|
page && typeof page._client === 'function' ? page._client() : page._client
|
|
if (!client) {
|
|
this.debug('Warning, missing properties to intercept CDP.', { page })
|
|
return
|
|
}
|
|
|
|
// Intercept CDP commands and strip identifying and unnecessary sourceURL
|
|
// https://github.com/puppeteer/puppeteer/blob/9b3005c105995cd267fdc7fb95b78aceab82cf0e/new-docs/puppeteer.cdpsession.md
|
|
const debug = this.debug
|
|
client.send = (function(originalMethod, context) {
|
|
return async function() {
|
|
const [method, paramArgs] = arguments || []
|
|
const next = async () => {
|
|
try {
|
|
return await originalMethod.apply(context, [method, paramArgs])
|
|
} catch (error) {
|
|
// This seems to happen sometimes when redirects cause other outstanding requests to be cut short
|
|
if (
|
|
error instanceof Error &&
|
|
error.message.includes(
|
|
`Protocol error (Network.getResponseBody): No resource with given identifier found`
|
|
)
|
|
) {
|
|
debug(
|
|
`Caught and ignored an error about a missing network resource.`,
|
|
{ error }
|
|
)
|
|
} else {
|
|
throw error
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!method || !paramArgs) {
|
|
return next()
|
|
}
|
|
|
|
// To find the methods/props in question check `_evaluateInternal` at:
|
|
// https://github.com/puppeteer/puppeteer/blob/main/src/common/ExecutionContext.ts#L186
|
|
const methodsToPatch = {
|
|
'Runtime.evaluate': 'expression',
|
|
'Runtime.callFunctionOn': 'functionDeclaration'
|
|
}
|
|
const SOURCE_URL_SUFFIX =
|
|
'//# sourceURL=__puppeteer_evaluation_script__'
|
|
|
|
if (!methodsToPatch[method] || !paramArgs[methodsToPatch[method]]) {
|
|
return next()
|
|
}
|
|
|
|
debug('Stripping sourceURL', { method })
|
|
paramArgs[methodsToPatch[method]] = paramArgs[
|
|
methodsToPatch[method]
|
|
].replace(SOURCE_URL_SUFFIX, '')
|
|
|
|
return next()
|
|
}
|
|
})(client.send, client)
|
|
}
|
|
}
|
|
|
|
module.exports = function(pluginConfig) {
|
|
return new Plugin(pluginConfig)
|
|
}
|