Commit 575dc7d

Browse files
jasnelladuh95
authored andcommitted
quic: fixup quic stream variable chunk len
Signed-off-by: James M Snell <jasnell@gmail.com> PR-URL: #63230 Backport-PR-URL: #64675Fixes: #63216 Reviewed-By: Tim Perry <pimterry@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 2551f9e commit 575dc7d

3 files changed

Lines changed: 110 additions & 16 deletions

File tree

β€Žlib/internal/quic/quic.jsβ€Ž

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,19 +1280,14 @@ function waitForDrain(stream) {
12801280

12811281
// Writes a batch to the handle, awaiting drain if backpressured.
12821282
// Returns true if the stream was destroyed during the wait.
1283-
// Checks writeDesiredSize before writing to enforce backpressure
1284-
// against the outbound DataQueue's uncommitted bytes.
1283+
// Only waits when writeDesiredSize is 0 (no capacity at all).
1284+
// When there is any capacity, the write proceeds even if the batch
1285+
// is larger -- the C++ side buffers the data and writeDesiredSize
1286+
// drops toward 0, letting the normal drain mechanism take over.
12851287
asyncfunctionwriteBatchWithDrain(handle,stream,batch){
12861288
conststate=getQuicStreamState(stream);
12871289

1288-
// Calculate total batch size for the capacity check.
1289-
letlen=0;
1290-
for(constchunkofbatch)len+=TypedArrayPrototypeGetByteLength(chunk);
1291-
1292-
// If insufficient capacity, wait for the C++ drain signal which
1293-
// fires when writeDesiredSize transitions from 0 to > 0 (i.e.,
1294-
// ngtcp2 has consumed data from the outbound DataQueue).
1295-
if(len>state.writeDesiredSize){
1290+
if(state.writeDesiredSize===0){
12961291
awaitwaitForDrain(stream);
12971292
if(stream.destroyed)returntrue;
12981293
}
@@ -2030,9 +2025,15 @@ class QuicStream {
20302025
chunk=toUint8Array(chunk);
20312026
constlen=TypedArrayPrototypeGetByteLength(chunk);
20322027
if(len===0)returntrue;
2033-
// Refuse the write if the chunk doesn't fit in the available
2034-
// buffer capacity. The caller should wait for drain and retry.
2035-
if(len>stream.#state.writeDesiredSize)returnfalse;
2028+
// Refuse the write only when there is no available capacity at
2029+
// all. When writeDesiredSize > 0 we allow the write even if the
2030+
// chunk is larger than the remaining capacity -- the C++ side
2031+
// will accept the data into the DataQueue and
2032+
// UpdateWriteDesiredSize() will drop writeDesiredSize toward 0,
2033+
// at which point the standard drain mechanism takes over.
2034+
// This follows the Web Streams model where writes beyond the HWM
2035+
// succeed and backpressure applies to *subsequent* writes.
2036+
if(stream.#state.writeDesiredSize===0)returnfalse;
20362037
constresult=handle.write([chunk]);
20372038
if(result===undefined)returnfalse;
20382039
totalBytesWritten+=len;
@@ -2071,7 +2072,7 @@ class QuicStream {
20712072
letlen=0;
20722073
for(constcofchunks)len+=TypedArrayPrototypeGetByteLength(c);
20732074
if(len===0)returntrue;
2074-
if(len>stream.#state.writeDesiredSize)returnfalse;
2075+
if(stream.#state.writeDesiredSize===0)returnfalse;
20752076
constresult=handle.write(chunks);
20762077
if(result===undefined)returnfalse;
20772078
totalBytesWritten+=len;

β€Žsrc/quic/streams.ccβ€Ž

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,8 +1592,10 @@ void Stream::UpdateWriteDesiredSize() {
15921592
uint32_t old_size = state_->write_desired_size;
15931593
state_->write_desired_size = clamped;
15941594

1595-
// Fire drain when transitioning from 0 to non-zero
1596-
if (old_size == 0 && desired > 0) {
1595+
// Fire drain when transitioning from 0 to non-zero.
1596+
// writeDesiredSize == 0 means the buffer is full or flow control is
1597+
// exhausted, so the JS side may be waiting for capacity.
1598+
if (old_size == 0 && clamped > 0) {
15971599
EmitDrain();
15981600
}
15991601
}
Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
// Flags: --experimental-quic --experimental-stream-iter --no-warnings
2+
3+
// Test: bidirectional data transfer with varying chunk sizes.
4+
// This is a regression test for a stall caused by a mismatch between
5+
// writeSync (which rejects when chunk > writeDesiredSize) and
6+
// drainableProtocol (which returned null when writeDesiredSize > 0).
7+
// When chunks don't evenly fill the high water mark, writeDesiredSize
8+
// can be positive but smaller than the next chunk, causing the
9+
// while(!writeSync) { dp(); await } loop to spin without yielding.
10+
// See: https://github.com/nodejs/node/issues/63216
11+
12+
import{hasQuic,skip,mustCall}from'../common/index.mjs';
13+
importassertfrom'node:assert';
14+
15+
const{ strictEqual }=assert;
16+
17+
if(!hasQuic){
18+
skip('QUIC is not enabled');
19+
}
20+
21+
const{ listen, connect }=awaitimport('../common/quic.mjs');
22+
const{ bytes,drainableProtocol: dp}=awaitimport('stream/iter');
23+
24+
// Varying chunk sizes β€” the pattern of alternating large and small
25+
// chunks is effective at triggering the writeDesiredSize gap.
26+
constchunkSizes=[60000,12,50000,1600,20000,30000,0,100];
27+
constnumChunks=chunkSizes.length;
28+
constbyteLength=chunkSizes.reduce((a,b)=>a+b,0);
29+
30+
// Build a deterministic payload so we can verify integrity.
31+
functionbuildChunk(index){
32+
constchunk=newUint8Array(chunkSizes[index]);
33+
constval=index&0xff;
34+
for(leti=0;i<chunkSizes[index];i++){
35+
chunk[i]=(val+i)&0xff;
36+
}
37+
returnchunk;
38+
}
39+
40+
functionchecksum(data){
41+
letsum=0;
42+
for(leti=0;i<data.byteLength;i++){
43+
sum=(sum+data[i])|0;
44+
}
45+
returnsum;
46+
}
47+
48+
// Compute expected checksum.
49+
letexpectedChecksum=0;
50+
for(leti=0;i<numChunks;i++){
51+
constchunk=buildChunk(i);
52+
expectedChecksum=(expectedChecksum+checksum(chunk))|0;
53+
}
54+
55+
constdone=Promise.withResolvers();
56+
57+
constserverEndpoint=awaitlisten(mustCall((serverSession)=>{
58+
serverSession.onstream=mustCall(async(stream)=>{
59+
constreceived=awaitbytes(stream);
60+
strictEqual(received.byteLength,byteLength);
61+
strictEqual(checksum(received),expectedChecksum);
62+
63+
stream.writer.endSync();
64+
awaitstream.closed;
65+
serverSession.close();
66+
done.resolve();
67+
});
68+
}));
69+
70+
constclientSession=awaitconnect(serverEndpoint.address);
71+
awaitclientSession.opened;
72+
73+
conststream=awaitclientSession.createBidirectionalStream();
74+
constw=stream.writer;
75+
76+
// Write chunks, respecting backpressure via drainableProtocol.
77+
for(leti=0;i<numChunks;i++){
78+
constchunk=buildChunk(i);
79+
while(!w.writeSync(chunk)){
80+
// Flow controlled β€” wait for drain before retrying.
81+
constdrainable=w[dp]();
82+
if(drainable)awaitdrainable;
83+
}
84+
}
85+
86+
consttotalWritten=w.endSync();
87+
strictEqual(totalWritten,byteLength);
88+
89+
awaitPromise.all([stream.closed,done.promise]);
90+
awaitclientSession.close();
91+
awaitserverEndpoint.close();

0 commit comments

Comments
Β (0)
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '
Skip to content

Commit 575dc7d

Browse files
jasnelladuh95
authored andcommitted
quic: fixup quic stream variable chunk len
Signed-off-by: James M Snell <jasnell@gmail.com> PR-URL: #63230 Backport-PR-URL: #64675Fixes: #63216 Reviewed-By: Tim Perry <pimterry@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 2551f9e commit 575dc7d

3 files changed

Lines changed: 110 additions & 16 deletions

File tree

β€Žlib/internal/quic/quic.jsβ€Ž

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,19 +1280,14 @@ function waitForDrain(stream) {
12801280

12811281
// Writes a batch to the handle, awaiting drain if backpressured.
12821282
// Returns true if the stream was destroyed during the wait.
1283-
// Checks writeDesiredSize before writing to enforce backpressure
1284-
// against the outbound DataQueue's uncommitted bytes.
1283+
// Only waits when writeDesiredSize is 0 (no capacity at all).
1284+
// When there is any capacity, the write proceeds even if the batch
1285+
// is larger -- the C++ side buffers the data and writeDesiredSize
1286+
// drops toward 0, letting the normal drain mechanism take over.
12851287
asyncfunctionwriteBatchWithDrain(handle,stream,batch){
12861288
conststate=getQuicStreamState(stream);
12871289

1288-
// Calculate total batch size for the capacity check.
1289-
letlen=0;
1290-
for(constchunkofbatch)len+=TypedArrayPrototypeGetByteLength(chunk);
1291-
1292-
// If insufficient capacity, wait for the C++ drain signal which
1293-
// fires when writeDesiredSize transitions from 0 to > 0 (i.e.,
1294-
// ngtcp2 has consumed data from the outbound DataQueue).
1295-
if(len>state.writeDesiredSize){
1290+
if(state.writeDesiredSize===0){
12961291
awaitwaitForDrain(stream);
12971292
if(stream.destroyed)returntrue;
12981293
}
@@ -2030,9 +2025,15 @@ class QuicStream {
20302025
chunk=toUint8Array(chunk);
20312026
constlen=TypedArrayPrototypeGetByteLength(chunk);
20322027
if(len===0)returntrue;
2033-
// Refuse the write if the chunk doesn't fit in the available
2034-
// buffer capacity. The caller should wait for drain and retry.
2035-
if(len>stream.#state.writeDesiredSize)returnfalse;
2028+
// Refuse the write only when there is no available capacity at
2029+
// all. When writeDesiredSize > 0 we allow the write even if the
2030+
// chunk is larger than the remaining capacity -- the C++ side
2031+
// will accept the data into the DataQueue and
2032+
// UpdateWriteDesiredSize() will drop writeDesiredSize toward 0,
2033+
// at which point the standard drain mechanism takes over.
2034+
// This follows the Web Streams model where writes beyond the HWM
2035+
// succeed and backpressure applies to *subsequent* writes.
2036+
if(stream.#state.writeDesiredSize===0)returnfalse;
20362037
constresult=handle.write([chunk]);
20372038
if(result===undefined)returnfalse;
20382039
totalBytesWritten+=len;
@@ -2071,7 +2072,7 @@ class QuicStream {
20712072
letlen=0;
20722073
for(constcofchunks)len+=TypedArrayPrototypeGetByteLength(c);
20732074
if(len===0)returntrue;
2074-
if(len>stream.#state.writeDesiredSize)returnfalse;
2075+
if(stream.#state.writeDesiredSize===0)returnfalse;
20752076
constresult=handle.write(chunks);
20762077
if(result===undefined)returnfalse;
20772078
totalBytesWritten+=len;

β€Žsrc/quic/streams.ccβ€Ž

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,8 +1592,10 @@ void Stream::UpdateWriteDesiredSize() {
15921592
uint32_t old_size = state_->write_desired_size;
15931593
state_->write_desired_size = clamped;
15941594

1595-
// Fire drain when transitioning from 0 to non-zero
1596-
if (old_size == 0 && desired > 0) {
1595+
// Fire drain when transitioning from 0 to non-zero.
1596+
// writeDesiredSize == 0 means the buffer is full or flow control is
1597+
// exhausted, so the JS side may be waiting for capacity.
1598+
if (old_size == 0 && clamped > 0) {
15971599
EmitDrain();
15981600
}
15991601
}
Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
// Flags: --experimental-quic --experimental-stream-iter --no-warnings
2+
3+
// Test: bidirectional data transfer with varying chunk sizes.
4+
// This is a regression test for a stall caused by a mismatch between
5+
// writeSync (which rejects when chunk > writeDesiredSize) and
6+
// drainableProtocol (which returned null when writeDesiredSize > 0).
7+
// When chunks don't evenly fill the high water mark, writeDesiredSize
8+
// can be positive but smaller than the next chunk, causing the
9+
// while(!writeSync) { dp(); await } loop to spin without yielding.
10+
// See: https://github.com/nodejs/node/issues/63216
11+
12+
import{hasQuic,skip,mustCall}from'../common/index.mjs';
13+
importassertfrom'node:assert';
14+
15+
const{ strictEqual }=assert;
16+
17+
if(!hasQuic){
18+
skip('QUIC is not enabled');
19+
}
20+
21+
const{ listen, connect }=awaitimport('../common/quic.mjs');
22+
const{ bytes,drainableProtocol: dp}=awaitimport('stream/iter');
23+
24+
// Varying chunk sizes β€” the pattern of alternating large and small
25+
// chunks is effective at triggering the writeDesiredSize gap.
26+
constchunkSizes=[60000,12,50000,1600,20000,30000,0,100];
27+
constnumChunks=chunkSizes.length;
28+
constbyteLength=chunkSizes.reduce((a,b)=>a+b,0);
29+
30+
// Build a deterministic payload so we can verify integrity.
31+
functionbuildChunk(index){
32+
constchunk=newUint8Array(chunkSizes[index]);
33+
constval=index&0xff;
34+
for(leti=0;i<chunkSizes[index];i++){
35+
chunk[i]=(val+i)&0xff;
36+
}
37+
returnchunk;
38+
}
39+
40+
functionchecksum(data){
41+
letsum=0;
42+
for(leti=0;i<data.byteLength;i++){
43+
sum=(sum+data[i])|0;
44+
}
45+
returnsum;
46+
}
47+
48+
// Compute expected checksum.
49+
letexpectedChecksum=0;
50+
for(leti=0;i<numChunks;i++){
51+
constchunk=buildChunk(i);
52+
expectedChecksum=(expectedChecksum+checksum(chunk))|0;
53+
}
54+
55+
constdone=Promise.withResolvers();
56+
57+
constserverEndpoint=awaitlisten(mustCall((serverSession)=>{
58+
serverSession.onstream=mustCall(async(stream)=>{
59+
constreceived=awaitbytes(stream);
60+
strictEqual(received.byteLength,byteLength);
61+
strictEqual(checksum(received),expectedChecksum);
62+
63+
stream.writer.endSync();
64+
awaitstream.closed;
65+
serverSession.close();
66+
done.resolve();
67+
});
68+
}));
69+
70+
constclientSession=awaitconnect(serverEndpoint.address);
71+
awaitclientSession.opened;
72+
73+
conststream=awaitclientSession.createBidirectionalStream();
74+
constw=stream.writer;
75+
76+
// Write chunks, respecting backpressure via drainableProtocol.
77+
for(leti=0;i<numChunks;i++){
78+
constchunk=buildChunk(i);
79+
while(!w.writeSync(chunk)){
80+
// Flow controlled β€” wait for drain before retrying.
81+
constdrainable=w[dp]();
82+
if(drainable)awaitdrainable;
83+
}
84+
}
85+
86+
consttotalWritten=w.endSync();
87+
strictEqual(totalWritten,byteLength);
88+
89+
awaitPromise.all([stream.closed,done.promise]);
90+
awaitclientSession.close();
91+
awaitserverEndpoint.close();

0 commit comments

Comments
Β (0)
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix"); } } catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Commit 575dc7d

Browse files
jasnelladuh95
authored andcommitted
quic: fixup quic stream variable chunk len
Signed-off-by: James M Snell <jasnell@gmail.com> PR-URL: #63230 Backport-PR-URL: #64675Fixes: #63216 Reviewed-By: Tim Perry <pimterry@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 2551f9e commit 575dc7d

3 files changed

Lines changed: 110 additions & 16 deletions

File tree

β€Žlib/internal/quic/quic.jsβ€Ž

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,19 +1280,14 @@ function waitForDrain(stream) {
12801280

12811281
// Writes a batch to the handle, awaiting drain if backpressured.
12821282
// Returns true if the stream was destroyed during the wait.
1283-
// Checks writeDesiredSize before writing to enforce backpressure
1284-
// against the outbound DataQueue's uncommitted bytes.
1283+
// Only waits when writeDesiredSize is 0 (no capacity at all).
1284+
// When there is any capacity, the write proceeds even if the batch
1285+
// is larger -- the C++ side buffers the data and writeDesiredSize
1286+
// drops toward 0, letting the normal drain mechanism take over.
12851287
asyncfunctionwriteBatchWithDrain(handle,stream,batch){
12861288
conststate=getQuicStreamState(stream);
12871289

1288-
// Calculate total batch size for the capacity check.
1289-
letlen=0;
1290-
for(constchunkofbatch)len+=TypedArrayPrototypeGetByteLength(chunk);
1291-
1292-
// If insufficient capacity, wait for the C++ drain signal which
1293-
// fires when writeDesiredSize transitions from 0 to > 0 (i.e.,
1294-
// ngtcp2 has consumed data from the outbound DataQueue).
1295-
if(len>state.writeDesiredSize){
1290+
if(state.writeDesiredSize===0){
12961291
awaitwaitForDrain(stream);
12971292
if(stream.destroyed)returntrue;
12981293
}
@@ -2030,9 +2025,15 @@ class QuicStream {
20302025
chunk=toUint8Array(chunk);
20312026
constlen=TypedArrayPrototypeGetByteLength(chunk);
20322027
if(len===0)returntrue;
2033-
// Refuse the write if the chunk doesn't fit in the available
2034-
// buffer capacity. The caller should wait for drain and retry.
2035-
if(len>stream.#state.writeDesiredSize)returnfalse;
2028+
// Refuse the write only when there is no available capacity at
2029+
// all. When writeDesiredSize > 0 we allow the write even if the
2030+
// chunk is larger than the remaining capacity -- the C++ side
2031+
// will accept the data into the DataQueue and
2032+
// UpdateWriteDesiredSize() will drop writeDesiredSize toward 0,
2033+
// at which point the standard drain mechanism takes over.
2034+
// This follows the Web Streams model where writes beyond the HWM
2035+
// succeed and backpressure applies to *subsequent* writes.
2036+
if(stream.#state.writeDesiredSize===0)returnfalse;
20362037
constresult=handle.write([chunk]);
20372038
if(result===undefined)returnfalse;
20382039
totalBytesWritten+=len;
@@ -2071,7 +2072,7 @@ class QuicStream {
20712072
letlen=0;
20722073
for(constcofchunks)len+=TypedArrayPrototypeGetByteLength(c);
20732074
if(len===0)returntrue;
2074-
if(len>stream.#state.writeDesiredSize)returnfalse;
2075+
if(stream.#state.writeDesiredSize===0)returnfalse;
20752076
constresult=handle.write(chunks);
20762077
if(result===undefined)returnfalse;
20772078
totalBytesWritten+=len;

β€Žsrc/quic/streams.ccβ€Ž

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,8 +1592,10 @@ void Stream::UpdateWriteDesiredSize() {
15921592
uint32_t old_size = state_->write_desired_size;
15931593
state_->write_desired_size = clamped;
15941594

1595-
// Fire drain when transitioning from 0 to non-zero
1596-
if (old_size == 0 && desired > 0) {
1595+
// Fire drain when transitioning from 0 to non-zero.
1596+
// writeDesiredSize == 0 means the buffer is full or flow control is
1597+
// exhausted, so the JS side may be waiting for capacity.
1598+
if (old_size == 0 && clamped > 0) {
15971599
EmitDrain();
15981600
}
15991601
}
Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
// Flags: --experimental-quic --experimental-stream-iter --no-warnings
2+
3+
// Test: bidirectional data transfer with varying chunk sizes.
4+
// This is a regression test for a stall caused by a mismatch between
5+
// writeSync (which rejects when chunk > writeDesiredSize) and
6+
// drainableProtocol (which returned null when writeDesiredSize > 0).
7+
// When chunks don't evenly fill the high water mark, writeDesiredSize
8+
// can be positive but smaller than the next chunk, causing the
9+
// while(!writeSync) { dp(); await } loop to spin without yielding.
10+
// See: https://github.com/nodejs/node/issues/63216
11+
12+
import{hasQuic,skip,mustCall}from'../common/index.mjs';
13+
importassertfrom'node:assert';
14+
15+
const{ strictEqual }=assert;
16+
17+
if(!hasQuic){
18+
skip('QUIC is not enabled');
19+
}
20+
21+
const{ listen, connect }=awaitimport('../common/quic.mjs');
22+
const{ bytes,drainableProtocol: dp}=awaitimport('stream/iter');
23+
24+
// Varying chunk sizes β€” the pattern of alternating large and small
25+
// chunks is effective at triggering the writeDesiredSize gap.
26+
constchunkSizes=[60000,12,50000,1600,20000,30000,0,100];
27+
constnumChunks=chunkSizes.length;
28+
constbyteLength=chunkSizes.reduce((a,b)=>a+b,0);
29+
30+
// Build a deterministic payload so we can verify integrity.
31+
functionbuildChunk(index){
32+
constchunk=newUint8Array(chunkSizes[index]);
33+
constval=index&0xff;
34+
for(leti=0;i<chunkSizes[index];i++){
35+
chunk[i]=(val+i)&0xff;
36+
}
37+
returnchunk;
38+
}
39+
40+
functionchecksum(data){
41+
letsum=0;
42+
for(leti=0;i<data.byteLength;i++){
43+
sum=(sum+data[i])|0;
44+
}
45+
returnsum;
46+
}
47+
48+
// Compute expected checksum.
49+
letexpectedChecksum=0;
50+
for(leti=0;i<numChunks;i++){
51+
constchunk=buildChunk(i);
52+
expectedChecksum=(expectedChecksum+checksum(chunk))|0;
53+
}
54+
55+
constdone=Promise.withResolvers();
56+
57+
constserverEndpoint=awaitlisten(mustCall((serverSession)=>{
58+
serverSession.onstream=mustCall(async(stream)=>{
59+
constreceived=awaitbytes(stream);
60+
strictEqual(received.byteLength,byteLength);
61+
strictEqual(checksum(received),expectedChecksum);
62+
63+
stream.writer.endSync();
64+
awaitstream.closed;
65+
serverSession.close();
66+
done.resolve();
67+
});
68+
}));
69+
70+
constclientSession=awaitconnect(serverEndpoint.address);
71+
awaitclientSession.opened;
72+
73+
conststream=awaitclientSession.createBidirectionalStream();
74+
constw=stream.writer;
75+
76+
// Write chunks, respecting backpressure via drainableProtocol.
77+
for(leti=0;i<numChunks;i++){
78+
constchunk=buildChunk(i);
79+
while(!w.writeSync(chunk)){
80+
// Flow controlled β€” wait for drain before retrying.
81+
constdrainable=w[dp]();
82+
if(drainable)awaitdrainable;
83+
}
84+
}
85+
86+
consttotalWritten=w.endSync();
87+
strictEqual(totalWritten,byteLength);
88+
89+
awaitPromise.all([stream.closed,done.promise]);
90+
awaitclientSession.close();
91+
awaitserverEndpoint.close();

0 commit comments

Comments
Β (0)
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Highlight search terms from Google/DuckDuckGo/Bing referrer\n(function() {\n var ref = document.referrer;\n var terms = [];\n \n if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) {\n var url = new URL(ref);\n var q = url.searchParams.get('q') || url.searchParams.get('p');\n if (q) {\n terms = q.split(/\\s+/).filter(function(t) { return t.length > 2; });\n }\n }\n \n if (terms.length === 0) return;\n \n var style = document.createElement('style');\n style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }';\n document.head.appendChild(style);\n \n function highlight(node) {\n if (node.nodeType === 3) { // text node\n var text = node.textContent;\n var found = false;\n terms.forEach(function(term) {\n var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\') + ')', 'gi');\n if (regex.test(text)) {\n found = true;\n var frag = document.createDocumentFragment();\n var parts = text.split(regex);\n parts.forEach(function(part, i) {\n if (i % 2 === 0) {\n frag.appendChild(document.createTextNode(part));\n } else {\n var span = document.createElement('span');\n span.className = 'userscript-highlight';\n span.textContent = part;\n frag.appendChild(span);\n }\n });\n node.parentNode.replaceChild(frag, node);\n }\n });\n } else if (node.nodeType === 1 && node.childNodes) { // element\n var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT'];\n if (!skipTags.includes(node.tagName)) {\n Array.from(node.childNodes).forEach(highlight);\n }\n }\n }\n \n highlight(document.body);\n \n // Re-highlight on dynamic content\n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1 || node.nodeType === 3) highlight(node);\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Highlight Search Terms"); } } catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Commit 575dc7d

Browse files
jasnelladuh95
authored andcommitted
quic: fixup quic stream variable chunk len
Signed-off-by: James M Snell <jasnell@gmail.com> PR-URL: #63230 Backport-PR-URL: #64675Fixes: #63216 Reviewed-By: Tim Perry <pimterry@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 2551f9e commit 575dc7d

3 files changed

Lines changed: 110 additions & 16 deletions

File tree

β€Žlib/internal/quic/quic.jsβ€Ž

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,19 +1280,14 @@ function waitForDrain(stream) {
12801280

12811281
// Writes a batch to the handle, awaiting drain if backpressured.
12821282
// Returns true if the stream was destroyed during the wait.
1283-
// Checks writeDesiredSize before writing to enforce backpressure
1284-
// against the outbound DataQueue's uncommitted bytes.
1283+
// Only waits when writeDesiredSize is 0 (no capacity at all).
1284+
// When there is any capacity, the write proceeds even if the batch
1285+
// is larger -- the C++ side buffers the data and writeDesiredSize
1286+
// drops toward 0, letting the normal drain mechanism take over.
12851287
asyncfunctionwriteBatchWithDrain(handle,stream,batch){
12861288
conststate=getQuicStreamState(stream);
12871289

1288-
// Calculate total batch size for the capacity check.
1289-
letlen=0;
1290-
for(constchunkofbatch)len+=TypedArrayPrototypeGetByteLength(chunk);
1291-
1292-
// If insufficient capacity, wait for the C++ drain signal which
1293-
// fires when writeDesiredSize transitions from 0 to > 0 (i.e.,
1294-
// ngtcp2 has consumed data from the outbound DataQueue).
1295-
if(len>state.writeDesiredSize){
1290+
if(state.writeDesiredSize===0){
12961291
awaitwaitForDrain(stream);
12971292
if(stream.destroyed)returntrue;
12981293
}
@@ -2030,9 +2025,15 @@ class QuicStream {
20302025
chunk=toUint8Array(chunk);
20312026
constlen=TypedArrayPrototypeGetByteLength(chunk);
20322027
if(len===0)returntrue;
2033-
// Refuse the write if the chunk doesn't fit in the available
2034-
// buffer capacity. The caller should wait for drain and retry.
2035-
if(len>stream.#state.writeDesiredSize)returnfalse;
2028+
// Refuse the write only when there is no available capacity at
2029+
// all. When writeDesiredSize > 0 we allow the write even if the
2030+
// chunk is larger than the remaining capacity -- the C++ side
2031+
// will accept the data into the DataQueue and
2032+
// UpdateWriteDesiredSize() will drop writeDesiredSize toward 0,
2033+
// at which point the standard drain mechanism takes over.
2034+
// This follows the Web Streams model where writes beyond the HWM
2035+
// succeed and backpressure applies to *subsequent* writes.
2036+
if(stream.#state.writeDesiredSize===0)returnfalse;
20362037
constresult=handle.write([chunk]);
20372038
if(result===undefined)returnfalse;
20382039
totalBytesWritten+=len;
@@ -2071,7 +2072,7 @@ class QuicStream {
20712072
letlen=0;
20722073
for(constcofchunks)len+=TypedArrayPrototypeGetByteLength(c);
20732074
if(len===0)returntrue;
2074-
if(len>stream.#state.writeDesiredSize)returnfalse;
2075+
if(stream.#state.writeDesiredSize===0)returnfalse;
20752076
constresult=handle.write(chunks);
20762077
if(result===undefined)returnfalse;
20772078
totalBytesWritten+=len;

β€Žsrc/quic/streams.ccβ€Ž

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,8 +1592,10 @@ void Stream::UpdateWriteDesiredSize() {
15921592
uint32_t old_size = state_->write_desired_size;
15931593
state_->write_desired_size = clamped;
15941594

1595-
// Fire drain when transitioning from 0 to non-zero
1596-
if (old_size == 0 && desired > 0) {
1595+
// Fire drain when transitioning from 0 to non-zero.
1596+
// writeDesiredSize == 0 means the buffer is full or flow control is
1597+
// exhausted, so the JS side may be waiting for capacity.
1598+
if (old_size == 0 && clamped > 0) {
15971599
EmitDrain();
15981600
}
15991601
}
Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
// Flags: --experimental-quic --experimental-stream-iter --no-warnings
2+
3+
// Test: bidirectional data transfer with varying chunk sizes.
4+
// This is a regression test for a stall caused by a mismatch between
5+
// writeSync (which rejects when chunk > writeDesiredSize) and
6+
// drainableProtocol (which returned null when writeDesiredSize > 0).
7+
// When chunks don't evenly fill the high water mark, writeDesiredSize
8+
// can be positive but smaller than the next chunk, causing the
9+
// while(!writeSync) { dp(); await } loop to spin without yielding.
10+
// See: https://github.com/nodejs/node/issues/63216
11+
12+
import{hasQuic,skip,mustCall}from'../common/index.mjs';
13+
importassertfrom'node:assert';
14+
15+
const{ strictEqual }=assert;
16+
17+
if(!hasQuic){
18+
skip('QUIC is not enabled');
19+
}
20+
21+
const{ listen, connect }=awaitimport('../common/quic.mjs');
22+
const{ bytes,drainableProtocol: dp}=awaitimport('stream/iter');
23+
24+
// Varying chunk sizes β€” the pattern of alternating large and small
25+
// chunks is effective at triggering the writeDesiredSize gap.
26+
constchunkSizes=[60000,12,50000,1600,20000,30000,0,100];
27+
constnumChunks=chunkSizes.length;
28+
constbyteLength=chunkSizes.reduce((a,b)=>a+b,0);
29+
30+
// Build a deterministic payload so we can verify integrity.
31+
functionbuildChunk(index){
32+
constchunk=newUint8Array(chunkSizes[index]);
33+
constval=index&0xff;
34+
for(leti=0;i<chunkSizes[index];i++){
35+
chunk[i]=(val+i)&0xff;
36+
}
37+
returnchunk;
38+
}
39+
40+
functionchecksum(data){
41+
letsum=0;
42+
for(leti=0;i<data.byteLength;i++){
43+
sum=(sum+data[i])|0;
44+
}
45+
returnsum;
46+
}
47+
48+
// Compute expected checksum.
49+
letexpectedChecksum=0;
50+
for(leti=0;i<numChunks;i++){
51+
constchunk=buildChunk(i);
52+
expectedChecksum=(expectedChecksum+checksum(chunk))|0;
53+
}
54+
55+
constdone=Promise.withResolvers();
56+
57+
constserverEndpoint=awaitlisten(mustCall((serverSession)=>{
58+
serverSession.onstream=mustCall(async(stream)=>{
59+
constreceived=awaitbytes(stream);
60+
strictEqual(received.byteLength,byteLength);
61+
strictEqual(checksum(received),expectedChecksum);
62+
63+
stream.writer.endSync();
64+
awaitstream.closed;
65+
serverSession.close();
66+
done.resolve();
67+
});
68+
}));
69+
70+
constclientSession=awaitconnect(serverEndpoint.address);
71+
awaitclientSession.opened;
72+
73+
conststream=awaitclientSession.createBidirectionalStream();
74+
constw=stream.writer;
75+
76+
// Write chunks, respecting backpressure via drainableProtocol.
77+
for(leti=0;i<numChunks;i++){
78+
constchunk=buildChunk(i);
79+
while(!w.writeSync(chunk)){
80+
// Flow controlled β€” wait for drain before retrying.
81+
constdrainable=w[dp]();
82+
if(drainable)awaitdrainable;
83+
}
84+
}
85+
86+
consttotalWritten=w.endSync();
87+
strictEqual(totalWritten,byteLength);
88+
89+
awaitPromise.all([stream.closed,done.promise]);
90+
awaitclientSession.close();
91+
awaitserverEndpoint.close();

0 commit comments

Comments
Β (0)
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Strip utm_, fbclid, gclid, etc. from all links on page\n(function() {\n var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content',\n 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid',\n 'ref', 'ref_src', 'source', 'medium', 'campaign'];\n \n function cleanUrl(url) {\n try {\n var u = new URL(url, window.location.origin);\n var changed = false;\n trackingParams.forEach(function(p) {\n if (u.searchParams.has(p)) {\n u.searchParams.delete(p);\n changed = true;\n }\n });\n return changed ? u.toString() : url;\n } catch (e) {\n return url;\n }\n }\n \n function cleanLinks() {\n document.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n \n cleanLinks();\n \n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1) {\n if (node.tagName === 'A') cleanLinks();\n node.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Remove Tracking Parameters from Links"); } } catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); } })(); (function(){ try { var __m = "youtube.com"; var __re = new RegExp('^' + "youtube\\.com" + '
Skip to content

Commit 575dc7d

Browse files
jasnelladuh95
authored andcommitted
quic: fixup quic stream variable chunk len
Signed-off-by: James M Snell <jasnell@gmail.com> PR-URL: #63230 Backport-PR-URL: #64675Fixes: #63216 Reviewed-By: Tim Perry <pimterry@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 2551f9e commit 575dc7d

3 files changed

Lines changed: 110 additions & 16 deletions

File tree

β€Žlib/internal/quic/quic.jsβ€Ž

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,19 +1280,14 @@ function waitForDrain(stream) {
12801280

12811281
// Writes a batch to the handle, awaiting drain if backpressured.
12821282
// Returns true if the stream was destroyed during the wait.
1283-
// Checks writeDesiredSize before writing to enforce backpressure
1284-
// against the outbound DataQueue's uncommitted bytes.
1283+
// Only waits when writeDesiredSize is 0 (no capacity at all).
1284+
// When there is any capacity, the write proceeds even if the batch
1285+
// is larger -- the C++ side buffers the data and writeDesiredSize
1286+
// drops toward 0, letting the normal drain mechanism take over.
12851287
asyncfunctionwriteBatchWithDrain(handle,stream,batch){
12861288
conststate=getQuicStreamState(stream);
12871289

1288-
// Calculate total batch size for the capacity check.
1289-
letlen=0;
1290-
for(constchunkofbatch)len+=TypedArrayPrototypeGetByteLength(chunk);
1291-
1292-
// If insufficient capacity, wait for the C++ drain signal which
1293-
// fires when writeDesiredSize transitions from 0 to > 0 (i.e.,
1294-
// ngtcp2 has consumed data from the outbound DataQueue).
1295-
if(len>state.writeDesiredSize){
1290+
if(state.writeDesiredSize===0){
12961291
awaitwaitForDrain(stream);
12971292
if(stream.destroyed)returntrue;
12981293
}
@@ -2030,9 +2025,15 @@ class QuicStream {
20302025
chunk=toUint8Array(chunk);
20312026
constlen=TypedArrayPrototypeGetByteLength(chunk);
20322027
if(len===0)returntrue;
2033-
// Refuse the write if the chunk doesn't fit in the available
2034-
// buffer capacity. The caller should wait for drain and retry.
2035-
if(len>stream.#state.writeDesiredSize)returnfalse;
2028+
// Refuse the write only when there is no available capacity at
2029+
// all. When writeDesiredSize > 0 we allow the write even if the
2030+
// chunk is larger than the remaining capacity -- the C++ side
2031+
// will accept the data into the DataQueue and
2032+
// UpdateWriteDesiredSize() will drop writeDesiredSize toward 0,
2033+
// at which point the standard drain mechanism takes over.
2034+
// This follows the Web Streams model where writes beyond the HWM
2035+
// succeed and backpressure applies to *subsequent* writes.
2036+
if(stream.#state.writeDesiredSize===0)returnfalse;
20362037
constresult=handle.write([chunk]);
20372038
if(result===undefined)returnfalse;
20382039
totalBytesWritten+=len;
@@ -2071,7 +2072,7 @@ class QuicStream {
20712072
letlen=0;
20722073
for(constcofchunks)len+=TypedArrayPrototypeGetByteLength(c);
20732074
if(len===0)returntrue;
2074-
if(len>stream.#state.writeDesiredSize)returnfalse;
2075+
if(stream.#state.writeDesiredSize===0)returnfalse;
20752076
constresult=handle.write(chunks);
20762077
if(result===undefined)returnfalse;
20772078
totalBytesWritten+=len;

β€Žsrc/quic/streams.ccβ€Ž

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,8 +1592,10 @@ void Stream::UpdateWriteDesiredSize() {
15921592
uint32_t old_size = state_->write_desired_size;
15931593
state_->write_desired_size = clamped;
15941594

1595-
// Fire drain when transitioning from 0 to non-zero
1596-
if (old_size == 0 && desired > 0) {
1595+
// Fire drain when transitioning from 0 to non-zero.
1596+
// writeDesiredSize == 0 means the buffer is full or flow control is
1597+
// exhausted, so the JS side may be waiting for capacity.
1598+
if (old_size == 0 && clamped > 0) {
15971599
EmitDrain();
15981600
}
15991601
}
Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
// Flags: --experimental-quic --experimental-stream-iter --no-warnings
2+
3+
// Test: bidirectional data transfer with varying chunk sizes.
4+
// This is a regression test for a stall caused by a mismatch between
5+
// writeSync (which rejects when chunk > writeDesiredSize) and
6+
// drainableProtocol (which returned null when writeDesiredSize > 0).
7+
// When chunks don't evenly fill the high water mark, writeDesiredSize
8+
// can be positive but smaller than the next chunk, causing the
9+
// while(!writeSync) { dp(); await } loop to spin without yielding.
10+
// See: https://github.com/nodejs/node/issues/63216
11+
12+
import{hasQuic,skip,mustCall}from'../common/index.mjs';
13+
importassertfrom'node:assert';
14+
15+
const{ strictEqual }=assert;
16+
17+
if(!hasQuic){
18+
skip('QUIC is not enabled');
19+
}
20+
21+
const{ listen, connect }=awaitimport('../common/quic.mjs');
22+
const{ bytes,drainableProtocol: dp}=awaitimport('stream/iter');
23+
24+
// Varying chunk sizes β€” the pattern of alternating large and small
25+
// chunks is effective at triggering the writeDesiredSize gap.
26+
constchunkSizes=[60000,12,50000,1600,20000,30000,0,100];
27+
constnumChunks=chunkSizes.length;
28+
constbyteLength=chunkSizes.reduce((a,b)=>a+b,0);
29+
30+
// Build a deterministic payload so we can verify integrity.
31+
functionbuildChunk(index){
32+
constchunk=newUint8Array(chunkSizes[index]);
33+
constval=index&0xff;
34+
for(leti=0;i<chunkSizes[index];i++){
35+
chunk[i]=(val+i)&0xff;
36+
}
37+
returnchunk;
38+
}
39+
40+
functionchecksum(data){
41+
letsum=0;
42+
for(leti=0;i<data.byteLength;i++){
43+
sum=(sum+data[i])|0;
44+
}
45+
returnsum;
46+
}
47+
48+
// Compute expected checksum.
49+
letexpectedChecksum=0;
50+
for(leti=0;i<numChunks;i++){
51+
constchunk=buildChunk(i);
52+
expectedChecksum=(expectedChecksum+checksum(chunk))|0;
53+
}
54+
55+
constdone=Promise.withResolvers();
56+
57+
constserverEndpoint=awaitlisten(mustCall((serverSession)=>{
58+
serverSession.onstream=mustCall(async(stream)=>{
59+
constreceived=awaitbytes(stream);
60+
strictEqual(received.byteLength,byteLength);
61+
strictEqual(checksum(received),expectedChecksum);
62+
63+
stream.writer.endSync();
64+
awaitstream.closed;
65+
serverSession.close();
66+
done.resolve();
67+
});
68+
}));
69+
70+
constclientSession=awaitconnect(serverEndpoint.address);
71+
awaitclientSession.opened;
72+
73+
conststream=awaitclientSession.createBidirectionalStream();
74+
constw=stream.writer;
75+
76+
// Write chunks, respecting backpressure via drainableProtocol.
77+
for(leti=0;i<numChunks;i++){
78+
constchunk=buildChunk(i);
79+
while(!w.writeSync(chunk)){
80+
// Flow controlled β€” wait for drain before retrying.
81+
constdrainable=w[dp]();
82+
if(drainable)awaitdrainable;
83+
}
84+
}
85+
86+
consttotalWritten=w.endSync();
87+
strictEqual(totalWritten,byteLength);
88+
89+
awaitPromise.all([stream.closed,done.promise]);
90+
awaitclientSession.close();
91+
awaitserverEndpoint.close();

0 commit comments

Comments
Β (0)
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default"); } } catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Commit 575dc7d

Browse files
jasnelladuh95
authored andcommitted
quic: fixup quic stream variable chunk len
Signed-off-by: James M Snell <jasnell@gmail.com> PR-URL: #63230 Backport-PR-URL: #64675Fixes: #63216 Reviewed-By: Tim Perry <pimterry@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 2551f9e commit 575dc7d

3 files changed

Lines changed: 110 additions & 16 deletions

File tree

β€Žlib/internal/quic/quic.jsβ€Ž

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,19 +1280,14 @@ function waitForDrain(stream) {
12801280

12811281
// Writes a batch to the handle, awaiting drain if backpressured.
12821282
// Returns true if the stream was destroyed during the wait.
1283-
// Checks writeDesiredSize before writing to enforce backpressure
1284-
// against the outbound DataQueue's uncommitted bytes.
1283+
// Only waits when writeDesiredSize is 0 (no capacity at all).
1284+
// When there is any capacity, the write proceeds even if the batch
1285+
// is larger -- the C++ side buffers the data and writeDesiredSize
1286+
// drops toward 0, letting the normal drain mechanism take over.
12851287
asyncfunctionwriteBatchWithDrain(handle,stream,batch){
12861288
conststate=getQuicStreamState(stream);
12871289

1288-
// Calculate total batch size for the capacity check.
1289-
letlen=0;
1290-
for(constchunkofbatch)len+=TypedArrayPrototypeGetByteLength(chunk);
1291-
1292-
// If insufficient capacity, wait for the C++ drain signal which
1293-
// fires when writeDesiredSize transitions from 0 to > 0 (i.e.,
1294-
// ngtcp2 has consumed data from the outbound DataQueue).
1295-
if(len>state.writeDesiredSize){
1290+
if(state.writeDesiredSize===0){
12961291
awaitwaitForDrain(stream);
12971292
if(stream.destroyed)returntrue;
12981293
}
@@ -2030,9 +2025,15 @@ class QuicStream {
20302025
chunk=toUint8Array(chunk);
20312026
constlen=TypedArrayPrototypeGetByteLength(chunk);
20322027
if(len===0)returntrue;
2033-
// Refuse the write if the chunk doesn't fit in the available
2034-
// buffer capacity. The caller should wait for drain and retry.
2035-
if(len>stream.#state.writeDesiredSize)returnfalse;
2028+
// Refuse the write only when there is no available capacity at
2029+
// all. When writeDesiredSize > 0 we allow the write even if the
2030+
// chunk is larger than the remaining capacity -- the C++ side
2031+
// will accept the data into the DataQueue and
2032+
// UpdateWriteDesiredSize() will drop writeDesiredSize toward 0,
2033+
// at which point the standard drain mechanism takes over.
2034+
// This follows the Web Streams model where writes beyond the HWM
2035+
// succeed and backpressure applies to *subsequent* writes.
2036+
if(stream.#state.writeDesiredSize===0)returnfalse;
20362037
constresult=handle.write([chunk]);
20372038
if(result===undefined)returnfalse;
20382039
totalBytesWritten+=len;
@@ -2071,7 +2072,7 @@ class QuicStream {
20712072
letlen=0;
20722073
for(constcofchunks)len+=TypedArrayPrototypeGetByteLength(c);
20732074
if(len===0)returntrue;
2074-
if(len>stream.#state.writeDesiredSize)returnfalse;
2075+
if(stream.#state.writeDesiredSize===0)returnfalse;
20752076
constresult=handle.write(chunks);
20762077
if(result===undefined)returnfalse;
20772078
totalBytesWritten+=len;

β€Žsrc/quic/streams.ccβ€Ž

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,8 +1592,10 @@ void Stream::UpdateWriteDesiredSize() {
15921592
uint32_t old_size = state_->write_desired_size;
15931593
state_->write_desired_size = clamped;
15941594

1595-
// Fire drain when transitioning from 0 to non-zero
1596-
if (old_size == 0 && desired > 0) {
1595+
// Fire drain when transitioning from 0 to non-zero.
1596+
// writeDesiredSize == 0 means the buffer is full or flow control is
1597+
// exhausted, so the JS side may be waiting for capacity.
1598+
if (old_size == 0 && clamped > 0) {
15971599
EmitDrain();
15981600
}
15991601
}
Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
// Flags: --experimental-quic --experimental-stream-iter --no-warnings
2+
3+
// Test: bidirectional data transfer with varying chunk sizes.
4+
// This is a regression test for a stall caused by a mismatch between
5+
// writeSync (which rejects when chunk > writeDesiredSize) and
6+
// drainableProtocol (which returned null when writeDesiredSize > 0).
7+
// When chunks don't evenly fill the high water mark, writeDesiredSize
8+
// can be positive but smaller than the next chunk, causing the
9+
// while(!writeSync) { dp(); await } loop to spin without yielding.
10+
// See: https://github.com/nodejs/node/issues/63216
11+
12+
import{hasQuic,skip,mustCall}from'../common/index.mjs';
13+
importassertfrom'node:assert';
14+
15+
const{ strictEqual }=assert;
16+
17+
if(!hasQuic){
18+
skip('QUIC is not enabled');
19+
}
20+
21+
const{ listen, connect }=awaitimport('../common/quic.mjs');
22+
const{ bytes,drainableProtocol: dp}=awaitimport('stream/iter');
23+
24+
// Varying chunk sizes β€” the pattern of alternating large and small
25+
// chunks is effective at triggering the writeDesiredSize gap.
26+
constchunkSizes=[60000,12,50000,1600,20000,30000,0,100];
27+
constnumChunks=chunkSizes.length;
28+
constbyteLength=chunkSizes.reduce((a,b)=>a+b,0);
29+
30+
// Build a deterministic payload so we can verify integrity.
31+
functionbuildChunk(index){
32+
constchunk=newUint8Array(chunkSizes[index]);
33+
constval=index&0xff;
34+
for(leti=0;i<chunkSizes[index];i++){
35+
chunk[i]=(val+i)&0xff;
36+
}
37+
returnchunk;
38+
}
39+
40+
functionchecksum(data){
41+
letsum=0;
42+
for(leti=0;i<data.byteLength;i++){
43+
sum=(sum+data[i])|0;
44+
}
45+
returnsum;
46+
}
47+
48+
// Compute expected checksum.
49+
letexpectedChecksum=0;
50+
for(leti=0;i<numChunks;i++){
51+
constchunk=buildChunk(i);
52+
expectedChecksum=(expectedChecksum+checksum(chunk))|0;
53+
}
54+
55+
constdone=Promise.withResolvers();
56+
57+
constserverEndpoint=awaitlisten(mustCall((serverSession)=>{
58+
serverSession.onstream=mustCall(async(stream)=>{
59+
constreceived=awaitbytes(stream);
60+
strictEqual(received.byteLength,byteLength);
61+
strictEqual(checksum(received),expectedChecksum);
62+
63+
stream.writer.endSync();
64+
awaitstream.closed;
65+
serverSession.close();
66+
done.resolve();
67+
});
68+
}));
69+
70+
constclientSession=awaitconnect(serverEndpoint.address);
71+
awaitclientSession.opened;
72+
73+
conststream=awaitclientSession.createBidirectionalStream();
74+
constw=stream.writer;
75+
76+
// Write chunks, respecting backpressure via drainableProtocol.
77+
for(leti=0;i<numChunks;i++){
78+
constchunk=buildChunk(i);
79+
while(!w.writeSync(chunk)){
80+
// Flow controlled β€” wait for drain before retrying.
81+
constdrainable=w[dp]();
82+
if(drainable)awaitdrainable;
83+
}
84+
}
85+
86+
consttotalWritten=w.endSync();
87+
strictEqual(totalWritten,byteLength);
88+
89+
awaitPromise.all([stream.closed,done.promise]);
90+
awaitclientSession.close();
91+
awaitserverEndpoint.close();

0 commit comments

Comments
Β (0)
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers"); } } catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); } })(); (function(){ try { var __m = "*"; var __re = new RegExp('^' + ".*" + '
Skip to content

Commit 575dc7d

Browse files
jasnelladuh95
authored andcommitted
quic: fixup quic stream variable chunk len
Signed-off-by: James M Snell <jasnell@gmail.com> PR-URL: #63230 Backport-PR-URL: #64675Fixes: #63216 Reviewed-By: Tim Perry <pimterry@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 2551f9e commit 575dc7d

3 files changed

Lines changed: 110 additions & 16 deletions

File tree

β€Žlib/internal/quic/quic.jsβ€Ž

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,19 +1280,14 @@ function waitForDrain(stream) {
12801280

12811281
// Writes a batch to the handle, awaiting drain if backpressured.
12821282
// Returns true if the stream was destroyed during the wait.
1283-
// Checks writeDesiredSize before writing to enforce backpressure
1284-
// against the outbound DataQueue's uncommitted bytes.
1283+
// Only waits when writeDesiredSize is 0 (no capacity at all).
1284+
// When there is any capacity, the write proceeds even if the batch
1285+
// is larger -- the C++ side buffers the data and writeDesiredSize
1286+
// drops toward 0, letting the normal drain mechanism take over.
12851287
asyncfunctionwriteBatchWithDrain(handle,stream,batch){
12861288
conststate=getQuicStreamState(stream);
12871289

1288-
// Calculate total batch size for the capacity check.
1289-
letlen=0;
1290-
for(constchunkofbatch)len+=TypedArrayPrototypeGetByteLength(chunk);
1291-
1292-
// If insufficient capacity, wait for the C++ drain signal which
1293-
// fires when writeDesiredSize transitions from 0 to > 0 (i.e.,
1294-
// ngtcp2 has consumed data from the outbound DataQueue).
1295-
if(len>state.writeDesiredSize){
1290+
if(state.writeDesiredSize===0){
12961291
awaitwaitForDrain(stream);
12971292
if(stream.destroyed)returntrue;
12981293
}
@@ -2030,9 +2025,15 @@ class QuicStream {
20302025
chunk=toUint8Array(chunk);
20312026
constlen=TypedArrayPrototypeGetByteLength(chunk);
20322027
if(len===0)returntrue;
2033-
// Refuse the write if the chunk doesn't fit in the available
2034-
// buffer capacity. The caller should wait for drain and retry.
2035-
if(len>stream.#state.writeDesiredSize)returnfalse;
2028+
// Refuse the write only when there is no available capacity at
2029+
// all. When writeDesiredSize > 0 we allow the write even if the
2030+
// chunk is larger than the remaining capacity -- the C++ side
2031+
// will accept the data into the DataQueue and
2032+
// UpdateWriteDesiredSize() will drop writeDesiredSize toward 0,
2033+
// at which point the standard drain mechanism takes over.
2034+
// This follows the Web Streams model where writes beyond the HWM
2035+
// succeed and backpressure applies to *subsequent* writes.
2036+
if(stream.#state.writeDesiredSize===0)returnfalse;
20362037
constresult=handle.write([chunk]);
20372038
if(result===undefined)returnfalse;
20382039
totalBytesWritten+=len;
@@ -2071,7 +2072,7 @@ class QuicStream {
20712072
letlen=0;
20722073
for(constcofchunks)len+=TypedArrayPrototypeGetByteLength(c);
20732074
if(len===0)returntrue;
2074-
if(len>stream.#state.writeDesiredSize)returnfalse;
2075+
if(stream.#state.writeDesiredSize===0)returnfalse;
20752076
constresult=handle.write(chunks);
20762077
if(result===undefined)returnfalse;
20772078
totalBytesWritten+=len;

β€Žsrc/quic/streams.ccβ€Ž

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,8 +1592,10 @@ void Stream::UpdateWriteDesiredSize() {
15921592
uint32_t old_size = state_->write_desired_size;
15931593
state_->write_desired_size = clamped;
15941594

1595-
// Fire drain when transitioning from 0 to non-zero
1596-
if (old_size == 0 && desired > 0) {
1595+
// Fire drain when transitioning from 0 to non-zero.
1596+
// writeDesiredSize == 0 means the buffer is full or flow control is
1597+
// exhausted, so the JS side may be waiting for capacity.
1598+
if (old_size == 0 && clamped > 0) {
15971599
EmitDrain();
15981600
}
15991601
}
Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
// Flags: --experimental-quic --experimental-stream-iter --no-warnings
2+
3+
// Test: bidirectional data transfer with varying chunk sizes.
4+
// This is a regression test for a stall caused by a mismatch between
5+
// writeSync (which rejects when chunk > writeDesiredSize) and
6+
// drainableProtocol (which returned null when writeDesiredSize > 0).
7+
// When chunks don't evenly fill the high water mark, writeDesiredSize
8+
// can be positive but smaller than the next chunk, causing the
9+
// while(!writeSync) { dp(); await } loop to spin without yielding.
10+
// See: https://github.com/nodejs/node/issues/63216
11+
12+
import{hasQuic,skip,mustCall}from'../common/index.mjs';
13+
importassertfrom'node:assert';
14+
15+
const{ strictEqual }=assert;
16+
17+
if(!hasQuic){
18+
skip('QUIC is not enabled');
19+
}
20+
21+
const{ listen, connect }=awaitimport('../common/quic.mjs');
22+
const{ bytes,drainableProtocol: dp}=awaitimport('stream/iter');
23+
24+
// Varying chunk sizes β€” the pattern of alternating large and small
25+
// chunks is effective at triggering the writeDesiredSize gap.
26+
constchunkSizes=[60000,12,50000,1600,20000,30000,0,100];
27+
constnumChunks=chunkSizes.length;
28+
constbyteLength=chunkSizes.reduce((a,b)=>a+b,0);
29+
30+
// Build a deterministic payload so we can verify integrity.
31+
functionbuildChunk(index){
32+
constchunk=newUint8Array(chunkSizes[index]);
33+
constval=index&0xff;
34+
for(leti=0;i<chunkSizes[index];i++){
35+
chunk[i]=(val+i)&0xff;
36+
}
37+
returnchunk;
38+
}
39+
40+
functionchecksum(data){
41+
letsum=0;
42+
for(leti=0;i<data.byteLength;i++){
43+
sum=(sum+data[i])|0;
44+
}
45+
returnsum;
46+
}
47+
48+
// Compute expected checksum.
49+
letexpectedChecksum=0;
50+
for(leti=0;i<numChunks;i++){
51+
constchunk=buildChunk(i);
52+
expectedChecksum=(expectedChecksum+checksum(chunk))|0;
53+
}
54+
55+
constdone=Promise.withResolvers();
56+
57+
constserverEndpoint=awaitlisten(mustCall((serverSession)=>{
58+
serverSession.onstream=mustCall(async(stream)=>{
59+
constreceived=awaitbytes(stream);
60+
strictEqual(received.byteLength,byteLength);
61+
strictEqual(checksum(received),expectedChecksum);
62+
63+
stream.writer.endSync();
64+
awaitstream.closed;
65+
serverSession.close();
66+
done.resolve();
67+
});
68+
}));
69+
70+
constclientSession=awaitconnect(serverEndpoint.address);
71+
awaitclientSession.opened;
72+
73+
conststream=awaitclientSession.createBidirectionalStream();
74+
constw=stream.writer;
75+
76+
// Write chunks, respecting backpressure via drainableProtocol.
77+
for(leti=0;i<numChunks;i++){
78+
constchunk=buildChunk(i);
79+
while(!w.writeSync(chunk)){
80+
// Flow controlled β€” wait for drain before retrying.
81+
constdrainable=w[dp]();
82+
if(drainable)awaitdrainable;
83+
}
84+
}
85+
86+
consttotalWritten=w.endSync();
87+
strictEqual(totalWritten,byteLength);
88+
89+
awaitPromise.all([stream.closed,done.promise]);
90+
awaitclientSession.close();
91+
awaitserverEndpoint.close();

0 commit comments

Comments
Β (0)
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Universal Dark Mode - works on any site\n(function() {\n var enabled = true;\n \n function applyDarkMode() {\n if (!enabled) return;\n \n // Create style element if it doesn't exist\n var style = document.getElementById('universal-dark-mode-style');\n if (!style) {\n style = document.createElement('style');\n style.id = 'universal-dark-mode-style';\n document.head.appendChild(style);\n }\n \n // Dark mode CSS - inverts colors but preserves images/video\n style.textContent = '\n /* Invert everything except media */\n html {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #1a1a2e !important;\n }\n \n /* Restore images, videos, iframes, canvas */\n img, video, iframe, canvas, svg, picture, [style*=\"background-image\"] {\n filter: invert(1) hue-rotate(180deg) !important;\n }\n \n /* Preserve specific elements that should not be inverted */\n .no-dark-mode, .no-dark-mode *,\n [data-theme=\"light\"], [data-theme=\"light\"],\n .ace_editor, .ace_editor *,\n .CodeMirror, .CodeMirror *,\n .monaco-editor, .monaco-editor *,\n .markdown-body pre, .markdown-body pre *,\n .highlight, .highlight *,\n pre code, pre code * {\n filter: none !important;\n }\n \n /* Fix common UI elements */\n .modal, .popup, .dropdown-menu, .tooltip, .popover {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #2d2d44 !important;\n border-color: #444 !important;\n }\n \n /* Scrollbars */\n ::-webkit-scrollbar { background: #1a1a2e !important; }\n ::-webkit-scrollbar-thumb { background: #444 !important; }\n ::-webkit-scrollbar-thumb:hover { background: #555 !important; }\n \n /* Selection */\n ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ';\n }\n \n function removeDarkMode() {\n var style = document.getElementById('universal-dark-mode-style');\n if (style) style.remove();\n }\n \n // Toggle with Alt+Shift+D\n document.addEventListener('keydown', function(e) {\n if (e.altKey && e.shiftKey && e.key === 'D') {\n e.preventDefault();\n enabled = !enabled;\n if (enabled) {\n applyDarkMode();\n console.log('[Universal Dark Mode] Enabled');\n } else {\n removeDarkMode();\n console.log('[Universal Dark Mode] Disabled');\n }\n }\n });\n \n // Apply on load\n applyDarkMode();\n \n // Re-apply on dynamic content\n var observer = new MutationObserver(function(mutations) {\n if (enabled && !document.getElementById('universal-dark-mode-style')) {\n applyDarkMode();\n }\n });\n observer.observe(document.head, { childList: true });\n \n console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle');\n})();", "Universal Dark Mode"); } } catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); } })(); })();
Skip to content

Commit 575dc7d

Browse files
jasnelladuh95
authored andcommitted
quic: fixup quic stream variable chunk len
Signed-off-by: James M Snell <jasnell@gmail.com> PR-URL: #63230 Backport-PR-URL: #64675Fixes: #63216 Reviewed-By: Tim Perry <pimterry@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 2551f9e commit 575dc7d

3 files changed

Lines changed: 110 additions & 16 deletions

File tree

β€Žlib/internal/quic/quic.jsβ€Ž

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1280,19 +1280,14 @@ function waitForDrain(stream) {
12801280

12811281
// Writes a batch to the handle, awaiting drain if backpressured.
12821282
// Returns true if the stream was destroyed during the wait.
1283-
// Checks writeDesiredSize before writing to enforce backpressure
1284-
// against the outbound DataQueue's uncommitted bytes.
1283+
// Only waits when writeDesiredSize is 0 (no capacity at all).
1284+
// When there is any capacity, the write proceeds even if the batch
1285+
// is larger -- the C++ side buffers the data and writeDesiredSize
1286+
// drops toward 0, letting the normal drain mechanism take over.
12851287
asyncfunctionwriteBatchWithDrain(handle,stream,batch){
12861288
conststate=getQuicStreamState(stream);
12871289

1288-
// Calculate total batch size for the capacity check.
1289-
letlen=0;
1290-
for(constchunkofbatch)len+=TypedArrayPrototypeGetByteLength(chunk);
1291-
1292-
// If insufficient capacity, wait for the C++ drain signal which
1293-
// fires when writeDesiredSize transitions from 0 to > 0 (i.e.,
1294-
// ngtcp2 has consumed data from the outbound DataQueue).
1295-
if(len>state.writeDesiredSize){
1290+
if(state.writeDesiredSize===0){
12961291
awaitwaitForDrain(stream);
12971292
if(stream.destroyed)returntrue;
12981293
}
@@ -2030,9 +2025,15 @@ class QuicStream {
20302025
chunk=toUint8Array(chunk);
20312026
constlen=TypedArrayPrototypeGetByteLength(chunk);
20322027
if(len===0)returntrue;
2033-
// Refuse the write if the chunk doesn't fit in the available
2034-
// buffer capacity. The caller should wait for drain and retry.
2035-
if(len>stream.#state.writeDesiredSize)returnfalse;
2028+
// Refuse the write only when there is no available capacity at
2029+
// all. When writeDesiredSize > 0 we allow the write even if the
2030+
// chunk is larger than the remaining capacity -- the C++ side
2031+
// will accept the data into the DataQueue and
2032+
// UpdateWriteDesiredSize() will drop writeDesiredSize toward 0,
2033+
// at which point the standard drain mechanism takes over.
2034+
// This follows the Web Streams model where writes beyond the HWM
2035+
// succeed and backpressure applies to *subsequent* writes.
2036+
if(stream.#state.writeDesiredSize===0)returnfalse;
20362037
constresult=handle.write([chunk]);
20372038
if(result===undefined)returnfalse;
20382039
totalBytesWritten+=len;
@@ -2071,7 +2072,7 @@ class QuicStream {
20712072
letlen=0;
20722073
for(constcofchunks)len+=TypedArrayPrototypeGetByteLength(c);
20732074
if(len===0)returntrue;
2074-
if(len>stream.#state.writeDesiredSize)returnfalse;
2075+
if(stream.#state.writeDesiredSize===0)returnfalse;
20752076
constresult=handle.write(chunks);
20762077
if(result===undefined)returnfalse;
20772078
totalBytesWritten+=len;

β€Žsrc/quic/streams.ccβ€Ž

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1592,8 +1592,10 @@ void Stream::UpdateWriteDesiredSize() {
15921592
uint32_t old_size = state_->write_desired_size;
15931593
state_->write_desired_size = clamped;
15941594

1595-
// Fire drain when transitioning from 0 to non-zero
1596-
if (old_size == 0 && desired > 0) {
1595+
// Fire drain when transitioning from 0 to non-zero.
1596+
// writeDesiredSize == 0 means the buffer is full or flow control is
1597+
// exhausted, so the JS side may be waiting for capacity.
1598+
if (old_size == 0 && clamped > 0) {
15971599
EmitDrain();
15981600
}
15991601
}
Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,91 @@
1+
// Flags: --experimental-quic --experimental-stream-iter --no-warnings
2+
3+
// Test: bidirectional data transfer with varying chunk sizes.
4+
// This is a regression test for a stall caused by a mismatch between
5+
// writeSync (which rejects when chunk > writeDesiredSize) and
6+
// drainableProtocol (which returned null when writeDesiredSize > 0).
7+
// When chunks don't evenly fill the high water mark, writeDesiredSize
8+
// can be positive but smaller than the next chunk, causing the
9+
// while(!writeSync) { dp(); await } loop to spin without yielding.
10+
// See: https://github.com/nodejs/node/issues/63216
11+
12+
import{hasQuic,skip,mustCall}from'../common/index.mjs';
13+
importassertfrom'node:assert';
14+
15+
const{ strictEqual }=assert;
16+
17+
if(!hasQuic){
18+
skip('QUIC is not enabled');
19+
}
20+
21+
const{ listen, connect }=awaitimport('../common/quic.mjs');
22+
const{ bytes,drainableProtocol: dp}=awaitimport('stream/iter');
23+
24+
// Varying chunk sizes β€” the pattern of alternating large and small
25+
// chunks is effective at triggering the writeDesiredSize gap.
26+
constchunkSizes=[60000,12,50000,1600,20000,30000,0,100];
27+
constnumChunks=chunkSizes.length;
28+
constbyteLength=chunkSizes.reduce((a,b)=>a+b,0);
29+
30+
// Build a deterministic payload so we can verify integrity.
31+
functionbuildChunk(index){
32+
constchunk=newUint8Array(chunkSizes[index]);
33+
constval=index&0xff;
34+
for(leti=0;i<chunkSizes[index];i++){
35+
chunk[i]=(val+i)&0xff;
36+
}
37+
returnchunk;
38+
}
39+
40+
functionchecksum(data){
41+
letsum=0;
42+
for(leti=0;i<data.byteLength;i++){
43+
sum=(sum+data[i])|0;
44+
}
45+
returnsum;
46+
}
47+
48+
// Compute expected checksum.
49+
letexpectedChecksum=0;
50+
for(leti=0;i<numChunks;i++){
51+
constchunk=buildChunk(i);
52+
expectedChecksum=(expectedChecksum+checksum(chunk))|0;
53+
}
54+
55+
constdone=Promise.withResolvers();
56+
57+
constserverEndpoint=awaitlisten(mustCall((serverSession)=>{
58+
serverSession.onstream=mustCall(async(stream)=>{
59+
constreceived=awaitbytes(stream);
60+
strictEqual(received.byteLength,byteLength);
61+
strictEqual(checksum(received),expectedChecksum);
62+
63+
stream.writer.endSync();
64+
awaitstream.closed;
65+
serverSession.close();
66+
done.resolve();
67+
});
68+
}));
69+
70+
constclientSession=awaitconnect(serverEndpoint.address);
71+
awaitclientSession.opened;
72+
73+
conststream=awaitclientSession.createBidirectionalStream();
74+
constw=stream.writer;
75+
76+
// Write chunks, respecting backpressure via drainableProtocol.
77+
for(leti=0;i<numChunks;i++){
78+
constchunk=buildChunk(i);
79+
while(!w.writeSync(chunk)){
80+
// Flow controlled β€” wait for drain before retrying.
81+
constdrainable=w[dp]();
82+
if(drainable)awaitdrainable;
83+
}
84+
}
85+
86+
consttotalWritten=w.endSync();
87+
strictEqual(totalWritten,byteLength);
88+
89+
awaitPromise.all([stream.closed,done.promise]);
90+
awaitclientSession.close();
91+
awaitserverEndpoint.close();

0 commit comments

Comments
Β (0)