CodeMirror/budibase
1
0
Fork 0
mirror of synced 2024-07-06 15:00:49 +12:00
Code Issues Projects Releases Wiki Activity

Replace ivm.module for script

Browse source
This commit is contained in:
Adria Navarro 2024-02-13 14:13:09 +01:00
parent 7e43fff018
commit 53d4fc2fa0
3 changed files with 3592 additions and 68 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3554
packages/server/src/jsRunner/bundles/index-helpers.ivm.bundle.js
View file

File diff suppressed because one or more lines are too long

5
packages/server/src/jsRunner/bundles/index-helpers.ts
View file

@ -2,9 +2,8 @@ const {
getJsHelperList,
} = require("../../../../string-templates/src/helpers/list.js")
const helpers = getJsHelperList()
export default {
...helpers,
export const helpers = {
...getJsHelperList(),
// pointing stripProtocol to a unexisting function to be able to declare it on isolated-vm
// @ts-ignore
// eslint-disable-next-line no-undef

101
packages/server/src/jsRunner/vm/index.ts
View file

@ -16,31 +16,14 @@ class ExecutionTimeoutError extends Error {
}
class ModuleHandler {
private modules: {
import: string
moduleKey: string
module: ivm.Module
}[] = []
private modules: string[] = []
private generateRandomKey = () => `i${crypto.randomUUID().replace(/-/g, "")}`
registerModule(module: ivm.Module, imports: string) {
this.modules.push({
moduleKey: this.generateRandomKey(),
import: imports,
module: module,
})
registerModule(code: string) {
this.modules.push(code)
}
generateImports() {
return this.modules
.map(m => `import ${m.import} from "${m.moduleKey}"`)
.join(";")
}
getModule(key: string) {
const module = this.modules.find(m => m.moduleKey === key)
return module?.module
return this.modules.join(";")
}
}
@ -98,34 +81,39 @@ export class IsolatedVM implements VM {
}),
})
const cryptoModule = this.registerCallbacks({
randomUUID: crypto.randomUUID,
})
const injectedRequire = `const require=function req(val) {
switch (val) {
case "url": return ${urlModule};
case "querystring": return ${querystringModule};
case "crypto": return ${cryptoModule};
}
}`
const helpersSource = loadBundle(BundleType.HELPERS)
const helpersModule = this.isolate.compileModuleSync(
`${injectedRequire};${helpersSource}`
)
// const helpersModule = this.isolate.compileModuleSync(
// `${injectedRequire};${helpersSource}`
// )
helpersModule.instantiateSync(this.vm, specifier => {
if (specifier === "crypto") {
const cryptoModule = this.registerCallbacks({
randomUUID: crypto.randomUUID,
})
const module = this.isolate.compileModuleSync(
`export default ${cryptoModule}`
)
module.instantiateSync(this.vm, specifier => {
throw new Error(`No imports allowed. Required: ${specifier}`)
})
return module
}
throw new Error(`No imports allowed. Required: ${specifier}`)
})
// helpersModule.instantiateSync(this.vm, specifier => {
// if (specifier === "crypto") {
// const cryptoModule = this.registerCallbacks({
// randomUUID: crypto.randomUUID,
// })
// const module = this.isolate.compileModuleSync(
// `export default ${cryptoModule}`
// )
// module.instantiateSync(this.vm, specifier => {
// throw new Error(`No imports allowed. Required: ${specifier}`)
// })
// return module
// }
// throw new Error(`No imports allowed. Required: ${specifier}`)
// })
this.moduleHandler.registerModule(helpersModule, "helpers")
this.moduleHandler.registerModule(`${injectedRequire};${helpersSource}`)
return this
}
@ -191,7 +179,11 @@ export class IsolatedVM implements VM {
throw new Error(`No imports allowed. Required: ${specifier}`)
})
this.moduleHandler.registerModule(bsonModule, "{deserialize, toJson}")
this.moduleHandler.registerModule(
bsonModule,
"{deserialize, toJson}",
"bson"
)
return this
}
@ -206,25 +198,13 @@ export class IsolatedVM implements VM {
}
}
code = `${this.moduleHandler.generateImports()};results.out=${this.codeWrapper(
code
)};`
code = `${this.moduleHandler.generateImports()};${this.codeWrapper(code)};`
const script = this.isolate.compileModuleSync(code)
const script = this.isolate.compileScriptSync(code)
script.instantiateSync(this.vm, specifier => {
const module = this.moduleHandler.getModule(specifier)
if (module) {
return module
}
const result = script.runSync(this.vm, { timeout: this.invocationTimeout })
throw new Error(`"${specifier}" import not allowed`)
})
script.evaluateSync({ timeout: this.invocationTimeout })
const result = this.getFromContext(this.resultKey)
return result.out
return result
}
private registerCallbacks(functions: Record<string, any>) {
@ -260,11 +240,4 @@ export class IsolatedVM implements VM {
)
}
}
private getFromContext(key: string) {
const ref = this.vm.global.getSync(key, { reference: true })
const result = ref.copySync()
ref.release()
return result
}
}