http2: fix end without read
Adjust http2 behaviour to allow ending a stream even after some data comes in (when the user has no intention of reading that data). Also correctly end a stream when trailers are present. PR-URL: https://github.com/nodejs/node/pull/20621 Fixes: https://github.com/nodejs/node/issues/20060 Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Matteo Collina <matteo.collina@gmail.com> Reviewed-By: Trivikram Kamat <trivikr.dev@gmail.com>
This commit is contained in:
parent
f9de6f5804
commit
8d38288a80
@ -260,8 +260,6 @@ class Http2ServerRequest extends Readable {
|
|||||||
stream[kRequest] = this;
|
stream[kRequest] = this;
|
||||||
|
|
||||||
// Pause the stream..
|
// Pause the stream..
|
||||||
stream.pause();
|
|
||||||
stream.on('data', onStreamData);
|
|
||||||
stream.on('trailers', onStreamTrailers);
|
stream.on('trailers', onStreamTrailers);
|
||||||
stream.on('end', onStreamEnd);
|
stream.on('end', onStreamEnd);
|
||||||
stream.on('error', onStreamError);
|
stream.on('error', onStreamError);
|
||||||
@ -328,8 +326,12 @@ class Http2ServerRequest extends Readable {
|
|||||||
_read(nread) {
|
_read(nread) {
|
||||||
const state = this[kState];
|
const state = this[kState];
|
||||||
if (!state.closed) {
|
if (!state.closed) {
|
||||||
state.didRead = true;
|
if (!state.didRead) {
|
||||||
process.nextTick(resumeStream, this[kStream]);
|
state.didRead = true;
|
||||||
|
this[kStream].on('data', onStreamData);
|
||||||
|
} else {
|
||||||
|
process.nextTick(resumeStream, this[kStream]);
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
this.emit('error', new ERR_HTTP2_INVALID_STREAM());
|
this.emit('error', new ERR_HTTP2_INVALID_STREAM());
|
||||||
}
|
}
|
||||||
|
@ -349,11 +349,11 @@ function onStreamClose(code) {
|
|||||||
// Push a null so the stream can end whenever the client consumes
|
// Push a null so the stream can end whenever the client consumes
|
||||||
// it completely.
|
// it completely.
|
||||||
stream.push(null);
|
stream.push(null);
|
||||||
|
// If the client hasn't tried to consume the stream and there is no
|
||||||
// Same as net.
|
// resume scheduled (which would indicate they would consume in the future),
|
||||||
if (stream.readableLength === 0) {
|
// then just dump the incoming data so that the stream can be destroyed.
|
||||||
stream.read(0);
|
if (!stream[kState].didRead && !stream._readableState.resumeScheduled)
|
||||||
}
|
stream.resume();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1795,6 +1795,8 @@ class Http2Stream extends Duplex {
|
|||||||
const ret = this[kHandle].trailers(headersList);
|
const ret = this[kHandle].trailers(headersList);
|
||||||
if (ret < 0)
|
if (ret < 0)
|
||||||
this.destroy(new NghttpError(ret));
|
this.destroy(new NghttpError(ret));
|
||||||
|
else
|
||||||
|
this[kMaybeDestroy]();
|
||||||
}
|
}
|
||||||
|
|
||||||
get closed() {
|
get closed() {
|
||||||
|
@ -20,12 +20,15 @@ fs.readFile(loc, common.mustCall((err, data) => {
|
|||||||
const server = http2.createServer();
|
const server = http2.createServer();
|
||||||
|
|
||||||
server.on('stream', common.mustCall((stream) => {
|
server.on('stream', common.mustCall((stream) => {
|
||||||
stream.on('close', common.mustCall(() => {
|
// Wait for some data to come through.
|
||||||
assert.strictEqual(stream.rstCode, 0);
|
setImmediate(() => {
|
||||||
}));
|
stream.on('close', common.mustCall(() => {
|
||||||
|
assert.strictEqual(stream.rstCode, 0);
|
||||||
|
}));
|
||||||
|
|
||||||
stream.respond({ ':status': 400 });
|
stream.respond({ ':status': 400 });
|
||||||
stream.end();
|
stream.end();
|
||||||
|
});
|
||||||
}));
|
}));
|
||||||
|
|
||||||
server.listen(0, common.mustCall(() => {
|
server.listen(0, common.mustCall(() => {
|
||||||
|
44
test/parallel/test-http2-compat-client-upload-reject.js
Normal file
44
test/parallel/test-http2-compat-client-upload-reject.js
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
'use strict';
|
||||||
|
|
||||||
|
// Verifies that uploading data from a client works
|
||||||
|
|
||||||
|
const common = require('../common');
|
||||||
|
if (!common.hasCrypto)
|
||||||
|
common.skip('missing crypto');
|
||||||
|
const assert = require('assert');
|
||||||
|
const http2 = require('http2');
|
||||||
|
const fs = require('fs');
|
||||||
|
const fixtures = require('../common/fixtures');
|
||||||
|
|
||||||
|
const loc = fixtures.path('person-large.jpg');
|
||||||
|
|
||||||
|
assert(fs.existsSync(loc));
|
||||||
|
|
||||||
|
fs.readFile(loc, common.mustCall((err, data) => {
|
||||||
|
assert.ifError(err);
|
||||||
|
|
||||||
|
const server = http2.createServer(common.mustCall((req, res) => {
|
||||||
|
setImmediate(() => {
|
||||||
|
res.writeHead(400);
|
||||||
|
res.end();
|
||||||
|
});
|
||||||
|
}));
|
||||||
|
|
||||||
|
server.listen(0, common.mustCall(() => {
|
||||||
|
const client = http2.connect(`http://localhost:${server.address().port}`);
|
||||||
|
|
||||||
|
const req = client.request({ ':method': 'POST' });
|
||||||
|
req.on('response', common.mustCall((headers) => {
|
||||||
|
assert.strictEqual(headers[':status'], 400);
|
||||||
|
}));
|
||||||
|
|
||||||
|
req.resume();
|
||||||
|
req.on('end', common.mustCall(() => {
|
||||||
|
server.close();
|
||||||
|
client.close();
|
||||||
|
}));
|
||||||
|
|
||||||
|
const str = fs.createReadStream(loc);
|
||||||
|
str.pipe(req);
|
||||||
|
}));
|
||||||
|
}));
|
Loading…
x
Reference in New Issue
Block a user