From 4f60364201e90348bbb03638eff51d02826c153c Mon Sep 17 00:00:00 2001
From: Ouyang Yadong <oyydoibh@gmail.com>
Date: Wed, 17 Oct 2018 00:44:49 +0800
Subject: [PATCH] http2: add Http2Stream.bufferSize

This commit adds `bufferSize` for `Http2Stream`.

Refs: https://github.com/nodejs/node/issues/21631

PR-URL: https://github.com/nodejs/node/pull/23711
Reviewed-By: James M Snell <jasnell@gmail.com>
---
 doc/api/http2.md                              | 10 +++
 lib/internal/http2/core.js                    |  6 ++
 test/parallel/test-http2-buffersize.js        | 51 +++++++++++++
 .../test-tls-streamwrap-buffersize.js         | 72 +++++++++++++++++++
 4 files changed, 139 insertions(+)
 create mode 100644 test/parallel/test-http2-buffersize.js
 create mode 100644 test/parallel/test-tls-streamwrap-buffersize.js

diff --git a/doc/api/http2.md b/doc/api/http2.md
index 5606f78e051e81..214107cb88171b 100644
--- a/doc/api/http2.md
+++ b/doc/api/http2.md
@@ -1012,6 +1012,15 @@ added: v8.4.0
 Set to `true` if the `Http2Stream` instance was aborted abnormally. When set,
 the `'aborted'` event will have been emitted.
 
+#### http2stream.bufferSize
+<!-- YAML
+added: REPLACEME
+-->
+* {number}
+
+This property shows the number of characters currently buffered to be written.
+See [`net.Socket.bufferSize`][] for details.
+
 #### http2stream.close(code[, callback])
 <!-- YAML
 added: v8.4.0
@@ -3435,6 +3444,7 @@ following additional properties:
 [`http2session.close()`]: #http2_http2session_close_callback
 [`http2stream.pushStream()`]: #http2_http2stream_pushstream_headers_options_callback
 [`net.Server.close()`]: net.html#net_server_close_callback
+[`net.Socket.bufferSize`]: net.html#net_socket_buffersize
 [`net.Socket.prototype.ref()`]: net.html#net_socket_ref
 [`net.Socket.prototype.unref()`]: net.html#net_socket_unref
 [`net.Socket`]: net.html#net_class_net_socket
diff --git a/lib/internal/http2/core.js b/lib/internal/http2/core.js
index de89197b4f834f..e7720b35be884e 100644
--- a/lib/internal/http2/core.js
+++ b/lib/internal/http2/core.js
@@ -1685,6 +1685,12 @@ class Http2Stream extends Duplex {
     return `Http2Stream ${util.format(obj)}`;
   }
 
+  get bufferSize() {
+    // `bufferSize` properties of `net.Socket` are `undefined` when
+    // their `_handle` are falsy. Here we avoid the behavior.
+    return this[kState].writeQueueSize + this.writableLength;
+  }
+
   get endAfterHeaders() {
     return this[kState].endAfterHeaders;
   }
diff --git a/test/parallel/test-http2-buffersize.js b/test/parallel/test-http2-buffersize.js
new file mode 100644
index 00000000000000..be56d8ce3d9344
--- /dev/null
+++ b/test/parallel/test-http2-buffersize.js
@@ -0,0 +1,51 @@
+'use strict';
+
+const { mustCall, hasCrypto, skip } = require('../common');
+if (!hasCrypto)
+  skip('missing crypto');
+const assert = require('assert');
+const { createServer, connect } = require('http2');
+const Countdown = require('../common/countdown');
+
+// This test ensures that `bufferSize` of Http2Session and Http2Stream work
+// as expected.
+{
+  const SOCKETS = 2;
+  const TIMES = 10;
+  const BUFFER_SIZE = 30;
+  const server = createServer();
+
+  // Other `bufferSize` tests for net module and tls module
+  // don't assert `bufferSize` of server-side sockets.
+  server.on('stream', mustCall((stream) => {
+    stream.on('data', mustCall());
+    stream.on('end', mustCall());
+  }, SOCKETS));
+
+  server.listen(0, mustCall(() => {
+    const authority = `http://localhost:${server.address().port}`;
+    const client = connect(authority);
+    const countdown = new Countdown(SOCKETS, () => {
+      client.close();
+      server.close();
+    });
+
+    client.once('connect', mustCall());
+
+    for (let j = 0; j < SOCKETS; j += 1) {
+      const stream = client.request({ ':method': 'POST' });
+      stream.on('data', () => {});
+      stream.on('close', mustCall(() => {
+        countdown.dec();
+      }));
+
+      for (let i = 0; i < TIMES; i += 1) {
+        stream.write(Buffer.allocUnsafe(BUFFER_SIZE), mustCall());
+        const expectedSocketBufferSize = BUFFER_SIZE * (i + 1);
+        assert.strictEqual(stream.bufferSize, expectedSocketBufferSize);
+      }
+      stream.end();
+      stream.close();
+    }
+  }));
+}
diff --git a/test/parallel/test-tls-streamwrap-buffersize.js b/test/parallel/test-tls-streamwrap-buffersize.js
new file mode 100644
index 00000000000000..f65a3d6e76b9d8
--- /dev/null
+++ b/test/parallel/test-tls-streamwrap-buffersize.js
@@ -0,0 +1,72 @@
+'use strict';
+const common = require('../common');
+if (!common.hasCrypto)
+  common.skip('missing crypto');
+const assert = require('assert');
+const fixtures = require('../common/fixtures');
+const makeDuplexPair = require('../common/duplexpair');
+const tls = require('tls');
+const net = require('net');
+
+// This test ensures that `bufferSize` also works for those tlsSockets
+// created from `socket` of `Duplex`, with which, TLSSocket will wrap
+// sockets in `StreamWrap`.
+{
+  const iter = 10;
+
+  function createDuplex(port) {
+    const { clientSide, serverSide } = makeDuplexPair();
+
+    return new Promise((resolve, reject) => {
+      const socket = net.connect({
+        port,
+      }, common.mustCall(() => {
+        clientSide.pipe(socket);
+        socket.pipe(clientSide);
+        clientSide.on('close', common.mustCall(() => socket.destroy()));
+        socket.on('close', common.mustCall(() => clientSide.destroy()));
+
+        resolve(serverSide);
+      }));
+    });
+  }
+
+  const server = tls.createServer({
+    key: fixtures.readKey('agent2-key.pem'),
+    cert: fixtures.readKey('agent2-cert.pem')
+  }, common.mustCall((socket) => {
+    let str = '';
+    socket.setEncoding('utf-8');
+    socket.on('data', (chunk) => { str += chunk; });
+
+    socket.on('end', common.mustCall(() => {
+      assert.strictEqual(str, 'a'.repeat(iter - 1));
+      server.close();
+    }));
+  }));
+
+  server.listen(0, common.mustCall(() => {
+    const { port } = server.address();
+    createDuplex(port).then((socket) => {
+      const client = tls.connect({
+        socket,
+        rejectUnauthorized: false,
+      }, common.mustCall(() => {
+        assert.strictEqual(client.bufferSize, 0);
+
+        for (let i = 1; i < iter; i++) {
+          client.write('a');
+          assert.strictEqual(client.bufferSize, i + 1);
+        }
+
+        // It seems that tlsSockets created from sockets of `Duplex` emit no
+        // "finish" events. We use "end" event instead.
+        client.on('end', common.mustCall(() => {
+          assert.strictEqual(client.bufferSize, undefined);
+        }));
+
+        client.end();
+      }));
+    });
+  }));
+}