summaryrefslogtreecommitdiff
path: root/ext
diff options
context:
space:
mode:
authorMatt Mastracci <matthew@mastracci.com>2023-08-22 08:45:10 -0600
committerGitHub <noreply@github.com>2023-08-22 14:45:10 +0000
commitc37b9655b6a6ccff1cedc6e43d245fd55415d76f (patch)
tree8850a4248ec05c5273f34ecf49c58ff7d6901e9e /ext
parent792dc754712ecf913a76c6bcbf3074ef7fb51cd7 (diff)
fix(ext/node): simultaneous reads can leak into each other (#20223)
Reported in #20188 This was caused by re-use of a global buffer `BUF` during simultaneous async reads.
Diffstat (limited to 'ext')
-rw-r--r--ext/node/polyfills/internal_binding/stream_wrap.ts88
1 files changed, 48 insertions, 40 deletions
diff --git a/ext/node/polyfills/internal_binding/stream_wrap.ts b/ext/node/polyfills/internal_binding/stream_wrap.ts
index 66ebbe682..8e976da2c 100644
--- a/ext/node/polyfills/internal_binding/stream_wrap.ts
+++ b/ext/node/polyfills/internal_binding/stream_wrap.ts
@@ -311,56 +311,61 @@ export class LibuvStreamWrap extends HandleWrap {
/** Internal method for reading from the attached stream. */
async #read() {
- let buf = BUF;
-
- let nread: number | null;
- const ridBefore = this[kStreamBaseField]!.rid;
+ const isOwnedBuf = bufLocked;
+ let buf = bufLocked ? new Uint8Array(SUGGESTED_SIZE) : BUF;
+ bufLocked = true;
try {
- nread = await this[kStreamBaseField]!.read(buf);
- } catch (e) {
- // Try to read again if the underlying stream resource
- // changed. This can happen during TLS upgrades (eg. STARTTLS)
- if (ridBefore != this[kStreamBaseField]!.rid) {
- return this.#read();
- }
+ let nread: number | null;
+ const ridBefore = this[kStreamBaseField]!.rid;
+ try {
+ nread = await this[kStreamBaseField]!.read(buf);
+ } catch (e) {
+ // Try to read again if the underlying stream resource
+ // changed. This can happen during TLS upgrades (eg. STARTTLS)
+ if (ridBefore != this[kStreamBaseField]!.rid) {
+ return this.#read();
+ }
- if (
- e instanceof Deno.errors.Interrupted ||
- e instanceof Deno.errors.BadResource
- ) {
- nread = codeMap.get("EOF")!;
- } else if (
- e instanceof Deno.errors.ConnectionReset ||
- e instanceof Deno.errors.ConnectionAborted
- ) {
- nread = codeMap.get("ECONNRESET")!;
- } else {
- nread = codeMap.get("UNKNOWN")!;
- }
+ if (
+ e instanceof Deno.errors.Interrupted ||
+ e instanceof Deno.errors.BadResource
+ ) {
+ nread = codeMap.get("EOF")!;
+ } else if (
+ e instanceof Deno.errors.ConnectionReset ||
+ e instanceof Deno.errors.ConnectionAborted
+ ) {
+ nread = codeMap.get("ECONNRESET")!;
+ } else {
+ nread = codeMap.get("UNKNOWN")!;
+ }
- buf = new Uint8Array(0);
- }
+ buf = new Uint8Array(0);
+ }
- nread ??= codeMap.get("EOF")!;
+ nread ??= codeMap.get("EOF")!;
- streamBaseState[kReadBytesOrError] = nread;
+ streamBaseState[kReadBytesOrError] = nread;
- if (nread > 0) {
- this.bytesRead += nread;
- }
+ if (nread > 0) {
+ this.bytesRead += nread;
+ }
- buf = buf.slice(0, nread);
+ buf = isOwnedBuf ? buf.subarray(0, nread) : buf.slice(0, nread);
- streamBaseState[kArrayBufferOffset] = 0;
+ streamBaseState[kArrayBufferOffset] = 0;
- try {
- this.onread!(buf, nread);
- } catch {
- // swallow callback errors.
- }
+ try {
+ this.onread!(buf, nread);
+ } catch {
+ // swallow callback errors.
+ }
- if (nread >= 0 && this.#reading) {
- this.#read();
+ if (nread >= 0 && this.#reading) {
+ this.#read();
+ }
+ } finally {
+ bufLocked = false;
}
}
@@ -423,4 +428,7 @@ export class LibuvStreamWrap extends HandleWrap {
}
}
+// Used in #read above
const BUF = new Uint8Array(SUGGESTED_SIZE);
+// We need to ensure that only one inflight read request uses the cached buffer above
+let bufLocked = false;