Skip to content

Commit

Permalink
fix(core): SafePromiseAll to be unaffected by Array#@@iterator (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
petamoriken committed Mar 13, 2023
1 parent bcb6ee9 commit e5673f5
Show file tree
Hide file tree
Showing 6 changed files with 82 additions and 20 deletions.
6 changes: 4 additions & 2 deletions cli/tests/unit/command_test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -851,17 +851,19 @@ Deno.test(

Deno.test(
{ permissions: { read: true, run: true } },
async function commandWithPromisePrototypeThenOverride() {
async function commandWithPrototypePollution() {
const originalThen = Promise.prototype.then;
const originalSymbolIterator = Array.prototype[Symbol.iterator];
try {
Promise.prototype.then = () => {
Promise.prototype.then = Array.prototype[Symbol.iterator] = () => {
throw new Error();
};
await new Deno.Command(Deno.execPath(), {
args: ["eval", "console.log('hello world')"],
}).output();
} finally {
Promise.prototype.then = originalThen;
Array.prototype[Symbol.iterator] = originalSymbolIterator;
}
},
);
6 changes: 4 additions & 2 deletions cli/tests/unit/flash_test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2272,10 +2272,11 @@ Deno.test(

Deno.test(
{ permissions: { net: true } },
async function serveWithPromisePrototypeThenOverride() {
async function serveWithPrototypePollution() {
const originalThen = Promise.prototype.then;
const originalSymbolIterator = Array.prototype[Symbol.iterator];
try {
Promise.prototype.then = () => {
Promise.prototype.then = Array.prototype[Symbol.iterator] = () => {
throw new Error();
};
const ac = new AbortController();
Expand All @@ -2292,6 +2293,7 @@ Deno.test(
await server;
} finally {
Promise.prototype.then = originalThen;
Array.prototype[Symbol.iterator] = originalSymbolIterator;
}
},
);
Expand Down
80 changes: 66 additions & 14 deletions core/00_primordials.js
Original file line number Diff line number Diff line change
Expand Up @@ -329,10 +329,11 @@
return SafeIterator;
};

primordials.SafeArrayIterator = createSafeIterator(
const SafeArrayIterator = createSafeIterator(
primordials.ArrayPrototypeSymbolIterator,
primordials.ArrayIteratorPrototypeNext,
);
primordials.SafeArrayIterator = SafeArrayIterator;
primordials.SafeSetIterator = createSafeIterator(
primordials.SetPrototypeSymbolIterator,
primordials.SetIteratorPrototypeNext,
Expand Down Expand Up @@ -479,29 +480,80 @@
primordials.PromisePrototypeCatch = (thisPromise, onRejected) =>
PromisePrototypeThen(thisPromise, undefined, onRejected);

const arrayToSafePromiseIterable = (array) =>
new SafeArrayIterator(
ArrayPrototypeMap(
array,
(p) => {
if (ObjectPrototypeIsPrototypeOf(PromisePrototype, p)) {
return new SafePromise((c, d) => PromisePrototypeThen(p, c, d));
}
return p;
},
),
);

/**
* Creates a Promise that is resolved with an array of results when all of the
* provided Promises resolve, or rejected when any Promise is rejected.
* @param {unknown[]} values An array of Promises.
* @returns A new Promise.
* @template T
* @param {Array<T | PromiseLike<T>>} values
* @returns {Promise<Awaited<T>[]>}
*/
primordials.SafePromiseAll = (values) =>
// Wrapping on a new Promise is necessary to not expose the SafePromise
// prototype to user-land.
new Promise((a, b) =>
SafePromise.all(
ArrayPrototypeMap(
values,
(p) => {
if (ObjectPrototypeIsPrototypeOf(PromisePrototype, p)) {
return new SafePromise((c, d) => PromisePrototypeThen(p, c, d));
}
return p;
},
),
).then(a, b)
SafePromise.all(arrayToSafePromiseIterable(values)).then(a, b)
);

// NOTE: Uncomment the following functions when you need to use them

// /**
// * Creates a Promise that is resolved with an array of results when all
// * of the provided Promises resolve or reject.
// * @template T
// * @param {Array<T | PromiseLike<T>>} values
// * @returns {Promise<PromiseSettledResult<T>[]>}
// */
// primordials.SafePromiseAllSettled = (values) =>
// // Wrapping on a new Promise is necessary to not expose the SafePromise
// // prototype to user-land.
// new Promise((a, b) =>
// SafePromise.allSettled(arrayToSafePromiseIterable(values)).then(a, b)
// );

// /**
// * The any function returns a promise that is fulfilled by the first given
// * promise to be fulfilled, or rejected with an AggregateError containing
// * an array of rejection reasons if all of the given promises are rejected.
// * It resolves all elements of the passed iterable to promises as it runs
// * this algorithm.
// * @template T
// * @param {T} values
// * @returns {Promise<Awaited<T[number]>>}
// */
// primordials.SafePromiseAny = (values) =>
// // Wrapping on a new Promise is necessary to not expose the SafePromise
// // prototype to user-land.
// new Promise((a, b) =>
// SafePromise.any(arrayToSafePromiseIterable(values)).then(a, b)
// );

// /**
// * Creates a Promise that is resolved or rejected when any of the provided
// * Promises are resolved or rejected.
// * @template T
// * @param {T} values
// * @returns {Promise<Awaited<T[number]>>}
// */
// primordials.SafePromiseRace = (values) =>
// // Wrapping on a new Promise is necessary to not expose the SafePromise
// // prototype to user-land.
// new Promise((a, b) =>
// SafePromise.race(arrayToSafePromiseIterable(values)).then(a, b)
// );

/**
* Attaches a callback that is invoked when the Promise is settled (fulfilled or
* rejected). The resolved value cannot be modified from the callback.
Expand Down
3 changes: 2 additions & 1 deletion core/01_core.js
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
PromisePrototypeThen,
RangeError,
ReferenceError,
SafeArrayIterator,
SafePromisePrototypeFinally,
setQueueMicrotask,
StringPrototypeSlice,
Expand Down Expand Up @@ -198,7 +199,7 @@
const id = rollPromiseId();
let promise = PromisePrototypeThen(setPromise(id), unwrapOpResult);
try {
ops[name](id, ...args);
ops[name](id, ...new SafeArrayIterator(args));
} catch (err) {
// Cleanup the just-created promise
getPromise(id);
Expand Down
4 changes: 4 additions & 0 deletions core/internal.d.ts
Original file line number Diff line number Diff line change
Expand Up @@ -75,6 +75,10 @@ declare namespace __bootstrap {
typeof globalThis.FinalizationRegistry;
export const SafeWeakRef: typeof globalThis.WeakRef;
export const SafePromiseAll: typeof Promise.all;
// NOTE: Uncomment the following functions when you need to use them
// export const SafePromiseAllSettled: typeof Promise.allSettled;
// export const SafePromiseAny: typeof Promise.any;
// export const SafePromiseRace: typeof Promise.race;
export const SafePromisePrototypeFinally: UncurryThis<
Promise.prototype.finally
>;
Expand Down
3 changes: 2 additions & 1 deletion ext/http/01_http.js
Original file line number Diff line number Diff line change
Expand Up @@ -510,7 +510,8 @@ function buildCaseInsensitiveCommaValueFinder(checkText) {

/** @param value {string} */
function hasWord(value) {
for (const [cLower, cUpper] of charCodes) {
for (let j = 0; j < charCodes.length; ++j) {
const { 0: cLower, 1: cUpper } = charCodes[j];
if (cLower === char || cUpper === char) {
char = StringPrototypeCharCodeAt(value, ++i);
} else {
Expand Down

0 comments on commit e5673f5

Please sign in to comment.