Repository navigation
Expand file tree
/
Copy pathsandbox.js
More file actions
108 lines (100 loc) · 2.6 KB
/
Copy pathsandbox.js
File metadata and controls
108 lines (100 loc) · 2.6 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
// https://github.com/terser/terser/blob/v5.37.0/test/sandbox.js
import vm from 'vm';
// import { Console } from "console";
// global.console = new Console({
// stdout: process.stdout,
// stderr: process.stderr,
// colorMode: false
// });
function safe_log(arg, level) {
if (arg) {
switch (typeof arg) {
case 'function':
return arg.toString();
case 'object':
if (/Error$/.test(arg.name)) return arg.toString();
arg.constructor.toString();
if (level--) {
for (var key in arg) {
if (!Object.getOwnPropertyDescriptor(arg, key).get) {
arg[key] = safe_log(arg[key], level);
}
}
}
}
}
return arg;
}
function strip_func_ids(text) {
return text.toString().replace(/F[0-9]{6}N/g, '<F<>N>');
}
var FUNC_TOSTRING = `
[ Array, Boolean, Error, Function, Number, Object, RegExp, String].forEach(function(f) {
f.toString = Function.prototype.toString;
});
Function.prototype.toString = function() {
var id = 100000;
return function() {
var n = this.name;
if (!/^F[0-9]{6}N$/.test(n)) {
n = "F" + ++id + "N";
${
Object.getOwnPropertyDescriptor(Function.prototype, 'name').configurable
? 'Object.defineProperty(this, "name", { get: () => n });'
: ''
}
}
return "function " + n + "() {...}";
}
}();
`;
export function run_code(code, prepend_code = '') {
var stdout = '';
var original_write = process.stdout.write;
process.stdout.write = function(chunk) {
stdout += chunk;
};
try {
const global = {
console: {
log: function(msg) {
if (arguments.length == 1 && typeof msg == 'string') {
return console.log(msg);
}
return console.log.apply(
console,
[].map.call(arguments, function(arg) {
return safe_log(arg, 3);
}),
);
},
},
id: x => x,
leak: () => {},
pass: () => {
global.console.log('PASS');
},
fail: () => {
global.console.log('FAIL');
},
};
global.global = global;
vm.runInNewContext(
[
FUNC_TOSTRING,
'!function() {',
prepend_code + code,
'}();',
].join('\n'),
global,
);
return stdout;
} catch (ex) {
return ex;
} finally {
process.stdout.write = original_write;
}
}
export function same_stdout(expected, actual) {
return typeof expected == typeof actual && strip_func_ids(expected) == strip_func_ids(actual);
}