Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

http2: handle existing socket data when creating HTTP/2 server sessions #41185

Merged
merged 1 commit into from Dec 23, 2021
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
30 changes: 15 additions & 15 deletions lib/internal/http2/core.js
Expand Up @@ -1262,6 +1262,21 @@ class Http2Session extends EventEmitter {
this.on('newListener', sessionListenerAdded);
this.on('removeListener', sessionListenerRemoved);

// Process data on the next tick - a remoteSettings handler may be attached.
// https://github.com/nodejs/node/issues/35981
process.nextTick(() => {
// Socket already has some buffered data - emulate receiving it
// https://github.com/nodejs/node/issues/35475
// https://github.com/nodejs/node/issues/34532
if (socket.readableLength) {
let buf;
while ((buf = socket.read()) !== null) {
debugSession(type, `${buf.length} bytes already in buffer`);
this[kHandle].receive(buf);
}
}
});

debugSession(type, 'created');
}

Expand Down Expand Up @@ -3276,21 +3291,6 @@ function connect(authority, options, listener) {
if (typeof listener === 'function')
session.once('connect', listener);

// Process data on the next tick - a remoteSettings handler may be attached.
// https://github.com/nodejs/node/issues/35981
process.nextTick(() => {
debug('Http2Session connect', options.createConnection);
// Socket already has some buffered data - emulate receiving it
// https://github.com/nodejs/node/issues/35475
if (socket && socket.readableLength) {
let buf;
while ((buf = socket.read()) !== null) {
debug(`Http2Session connect: ${buf.length} bytes already in buffer`);
session[kHandle].receive(buf);
}
}
});

return session;
}

Expand Down
72 changes: 72 additions & 0 deletions test/parallel/test-http2-autoselect-protocol.js
@@ -0,0 +1,72 @@
'use strict';
const common = require('../common');
if (!common.hasCrypto)
common.skip('missing crypto');

const assert = require('assert');
const net = require('net');
const http = require('http');
const http2 = require('http2');

// Example test for HTTP/1 vs HTTP/2 protocol autoselection.
// Refs: https://github.com/nodejs/node/issues/34532

const h1Server = http.createServer(common.mustCall((req, res) => {
res.end('HTTP/1 Response');
}));

const h2Server = http2.createServer(common.mustCall((req, res) => {
res.end('HTTP/2 Response');
}));

const rawServer = net.createServer(common.mustCall(function listener(socket) {
const data = socket.read(3);

if (!data) { // Repeat until data is available
socket.once('readable', () => listener(socket));
return;
}

// Put the data back, so the real server can handle it:
socket.unshift(data);

if (data.toString('ascii') === 'PRI') { // Very dumb preface check
h2Server.emit('connection', socket);
} else {
h1Server.emit('connection', socket);
}
}, 2));

rawServer.listen(common.mustCall(() => {
const { port } = rawServer.address();

let done = 0;
{
// HTTP/2 Request
const client = http2.connect(`http://localhost:${port}`);
const req = client.request({ ':path': '/' });
req.end();

let content = '';
req.setEncoding('utf8');
req.on('data', (chunk) => content += chunk);
req.on('end', common.mustCall(() => {
assert.strictEqual(content, 'HTTP/2 Response');
if (++done === 2) rawServer.close();
client.close();
}));
}

{
// HTTP/1 Request
http.get(`http://localhost:${port}`, common.mustCall((res) => {
let content = '';
res.setEncoding('utf8');
res.on('data', (chunk) => content += chunk);
res.on('end', common.mustCall(() => {
assert.strictEqual(content, 'HTTP/1 Response');
if (++done === 2) rawServer.close();
}));
}));
}
}));