Fix style and add test

patriksimek · Jul 5, 2022 · c01280c · c01280c
1 parent 3a98764
commit c01280c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/lib/setup-sandbox.js b/lib/setup-sandbox.js
@@ -54,7 +54,7 @@ const {
 const {
 	get: localWeakMapGet,
 	set: localWeakMapSet
- } = LocalWeakMap.prototype;
+} = LocalWeakMap.prototype;
 
 function localUnexpected() {
 	return new VMError('Should not happen');

diff --git a/test/vm.js b/test/vm.js
@@ -1058,6 +1058,18 @@ describe('VM', () => {
 		const sst = vm2.run('Error.prepareStackTrace = (e,sst)=>sst;const sst = new Error().stack;Error.prepareStackTrace = undefined;sst');
 		assert.strictEqual(vm2.run('sst=>Object.getPrototypeOf(sst)')(sst), vm2.run('Array.prototype'));
 		assert.throws(()=>vm2.run('sst=>sst[0].getThis().constructor.constructor')(sst), /TypeError: Cannot read propert.*constructor/);
+		assert.throws(()=>vm2.run(`
+			const { set } = WeakMap.prototype;
+			WeakMap.prototype.set = function(v) {
+				return set.call(this, v, v);
+			};
+			Error.prepareStackTrace =
+			Error.prepareStackTrace =
+			(_, c) => c.map(c => c.getThis()).find(a => a);
+			const { stack } = new Error();
+			Error.prepareStackTrace = undefined;
+			stack.process
+		`));
 	});
 
 	it('Node internal prepareStackTrace attack', () => {