Skip to content

Commit ee2e7fc

Browse files
committed
http2: remember sent headers
Add sentHeaders, sentTrailers, and sentInfoHeaders properties on `Http2Stream`. PR-URL: #18045 Fixes: #16619 Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
1 parent 20fe04f commit ee2e7fc

File tree

3 files changed

+99
-0
lines changed

3 files changed

+99
-0
lines changed

doc/api/http2.md

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -989,6 +989,34 @@ destroyed after either receiving an `RST_STREAM` frame from the connected peer,
989989
calling `http2stream.close()`, or `http2stream.destroy()`. Will be
990990
`undefined` if the `Http2Stream` has not been closed.
991991

992+
#### http2stream.sentHeaders
993+
<!-- YAML
994+
added: REPLACEME
995+
-->
996+
997+
* Value: {[Headers Object][]}
998+
999+
An object containing the outbound headers sent for this `Http2Stream`.
1000+
1001+
#### http2stream.sentInfoHeaders
1002+
<!-- YAML
1003+
added: REPLACEME
1004+
-->
1005+
1006+
* Value: {[Headers Object][]\[\]}
1007+
1008+
An array of objects containing the outbound informational (additional) headers
1009+
sent for this `Http2Stream`.
1010+
1011+
#### http2stream.sentTrailers
1012+
<!-- YAML
1013+
added: REPLACEME
1014+
-->
1015+
1016+
* Value: {[Headers Object][]}
1017+
1018+
An object containing the outbound trailers sent for this this `HttpStream`.
1019+
9921020
#### http2stream.session
9931021
<!-- YAML
9941022
added: v8.4.0

lib/internal/http2/core.js

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ const kEncrypted = Symbol('encrypted');
7676
const kHandle = Symbol('handle');
7777
const kID = Symbol('id');
7878
const kInit = Symbol('init');
79+
const kInfoHeaders = Symbol('sent-info-headers');
7980
const kMaybeDestroy = Symbol('maybe-destroy');
8081
const kLocalSettings = Symbol('local-settings');
8182
const kOptions = Symbol('options');
@@ -84,6 +85,8 @@ const kProceed = Symbol('proceed');
8485
const kProtocol = Symbol('protocol');
8586
const kProxySocket = Symbol('proxy-socket');
8687
const kRemoteSettings = Symbol('remote-settings');
88+
const kSentHeaders = Symbol('sent-headers');
89+
const kSentTrailers = Symbol('sent-trailers');
8790
const kServer = Symbol('server');
8891
const kSession = Symbol('session');
8992
const kState = Symbol('state');
@@ -258,6 +261,7 @@ function onStreamTrailers() {
258261
stream.destroy(headersList);
259262
return [];
260263
}
264+
stream[kSentTrailers] = trailers;
261265
return headersList;
262266
}
263267

@@ -1348,6 +1352,7 @@ class ClientHttp2Session extends Http2Session {
13481352
throw headersList;
13491353

13501354
const stream = new ClientHttp2Stream(this, undefined, undefined, {});
1355+
stream[kSentHeaders] = headers;
13511356

13521357
// Close the writable side of the stream if options.endStream is set.
13531358
if (options.endStream)
@@ -1514,6 +1519,18 @@ class Http2Stream extends Duplex {
15141519
return `Http2Stream ${util.format(obj)}`;
15151520
}
15161521

1522+
get sentHeaders() {
1523+
return this[kSentHeaders];
1524+
}
1525+
1526+
get sentTrailers() {
1527+
return this[kSentTrailers];
1528+
}
1529+
1530+
get sentInfoHeaders() {
1531+
return this[kInfoHeaders];
1532+
}
1533+
15171534
get pending() {
15181535
return this[kID] === undefined;
15191536
}
@@ -1855,6 +1872,7 @@ function processRespondWithFD(self, fd, headers, offset = 0, length = -1,
18551872
state.flags |= STREAM_FLAGS_HEADERS_SENT;
18561873

18571874
const headersList = mapToHeaders(headers, assertValidPseudoHeaderResponse);
1875+
self[kSentHeaders] = headers;
18581876
if (!Array.isArray(headersList)) {
18591877
self.destroy(headersList);
18601878
return;
@@ -2085,6 +2103,7 @@ class ServerHttp2Stream extends Http2Stream {
20852103

20862104
const id = ret.id();
20872105
const stream = new ServerHttp2Stream(session, ret, id, options, headers);
2106+
stream[kSentHeaders] = headers;
20882107

20892108
if (options.endStream)
20902109
stream.end();
@@ -2144,6 +2163,7 @@ class ServerHttp2Stream extends Http2Stream {
21442163
const headersList = mapToHeaders(headers, assertValidPseudoHeaderResponse);
21452164
if (!Array.isArray(headersList))
21462165
throw headersList;
2166+
this[kSentHeaders] = headers;
21472167

21482168
state.flags |= STREAM_FLAGS_HEADERS_SENT;
21492169

@@ -2329,6 +2349,10 @@ class ServerHttp2Stream extends Http2Stream {
23292349
const headersList = mapToHeaders(headers, assertValidPseudoHeaderResponse);
23302350
if (!Array.isArray(headersList))
23312351
throw headersList;
2352+
if (!this[kInfoHeaders])
2353+
this[kInfoHeaders] = [headers];
2354+
else
2355+
this[kInfoHeaders].push(headers);
23322356

23332357
const ret = this[kHandle].info(headersList);
23342358
if (ret < 0)
Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,47 @@
1+
'use strict';
2+
3+
const common = require('../common');
4+
if (!common.hasCrypto)
5+
common.skip('missing crypto');
6+
const assert = require('assert');
7+
const h2 = require('http2');
8+
9+
const server = h2.createServer();
10+
11+
server.on('stream', common.mustCall((stream) => {
12+
stream.additionalHeaders({ ':status': 102 });
13+
assert.strictEqual(stream.sentInfoHeaders[0][':status'], 102);
14+
15+
stream.respond({ abc: 'xyz' }, {
16+
getTrailers(headers) {
17+
headers.xyz = 'abc';
18+
}
19+
});
20+
assert.strictEqual(stream.sentHeaders.abc, 'xyz');
21+
assert.strictEqual(stream.sentHeaders[':status'], 200);
22+
assert.notStrictEqual(stream.sentHeaders.date, undefined);
23+
stream.end();
24+
stream.on('close', () => {
25+
assert.strictEqual(stream.sentTrailers.xyz, 'abc');
26+
});
27+
}));
28+
29+
server.listen(0, common.mustCall(() => {
30+
const client = h2.connect(`http://localhost:${server.address().port}`);
31+
const req = client.request();
32+
33+
req.on('headers', common.mustCall((headers) => {
34+
assert.strictEqual(headers[':status'], 102);
35+
}));
36+
37+
assert.strictEqual(req.sentHeaders[':method'], 'GET');
38+
assert.strictEqual(req.sentHeaders[':authority'],
39+
`localhost:${server.address().port}`);
40+
assert.strictEqual(req.sentHeaders[':scheme'], 'http');
41+
assert.strictEqual(req.sentHeaders[':path'], '/');
42+
req.resume();
43+
req.on('close', () => {
44+
server.close();
45+
client.close();
46+
});
47+
}));

0 commit comments

Comments
 (0)