From 5d8ab112e0cbc9b6ef355566a7fb022668c67445 Mon Sep 17 00:00:00 2001
From: Robert Nagy <ronagy@icloud.com>
Date: Tue, 17 Apr 2018 11:37:50 +0200
Subject: [PATCH] http: added aborted property to request

PR-URL: https://github.com/nodejs/node/pull/20094
Reviewed-By: James M Snell <jasnell@gmail.com>
Reviewed-By: Luigi Pinca <luigipinca@gmail.com>
---
 doc/api/http.md                            | 12 +++++++++-
 doc/api/http2.md                           | 10 ++++++++
 lib/_http_client.js                        |  2 ++
 lib/_http_incoming.js                      |  2 ++
 lib/_http_server.js                        |  1 +
 lib/internal/http2/compat.js               |  7 ++++++
 test/parallel/test-http-aborted.js         | 26 +++++++++++++++++++++
 test/parallel/test-http2-compat-aborted.js | 27 ++++++++++++++++++++++
 8 files changed, 86 insertions(+), 1 deletion(-)
 create mode 100644 test/parallel/test-http-aborted.js
 create mode 100644 test/parallel/test-http2-compat-aborted.js

diff --git a/doc/api/http.md b/doc/api/http.md
index 79ed99a388065d..4609cfe9adef54 100644
--- a/doc/api/http.md
+++ b/doc/api/http.md
@@ -1403,7 +1403,7 @@ following additional events, methods, and properties.
 added: v0.3.8
 -->
 
-Emitted when the request has been aborted and the network socket has closed.
+Emitted when the request has been aborted.
 
 ### Event: 'close'
 <!-- YAML
@@ -1413,6 +1413,16 @@ added: v0.4.2
 Indicates that the underlying connection was closed.
 Just like `'end'`, this event occurs only once per response.
 
+### message.aborted
+<!-- YAML
+added: REPLACEME
+-->
+
+* {boolean}
+
+The `message.aborted` property will be `true` if the request has
+been aborted.
+
 ### message.destroy([error])
 <!-- YAML
 added: v0.3.0
diff --git a/doc/api/http2.md b/doc/api/http2.md
index b8db1c982bf96c..3433716f5ee7e6 100644
--- a/doc/api/http2.md
+++ b/doc/api/http2.md
@@ -2421,6 +2421,16 @@ added: v8.4.0
 Indicates that the underlying [`Http2Stream`][] was closed.
 Just like `'end'`, this event occurs only once per response.
 
+#### request.aborted
+<!-- YAML
+added: REPLACEME
+-->
+
+* {boolean}
+
+The `request.aborted` property will be `true` if the request has
+been aborted.
+
 #### request.destroy([error])
 <!-- YAML
 added: v8.4.0
diff --git a/lib/_http_client.js b/lib/_http_client.js
index 135d31acaaec2d..1b7a01709ba1d1 100644
--- a/lib/_http_client.js
+++ b/lib/_http_client.js
@@ -299,6 +299,7 @@ ClientRequest.prototype.abort = function abort() {
   if (!this.aborted) {
     process.nextTick(emitAbortNT.bind(this));
   }
+
   // Mark as aborting so we can avoid sending queued request data
   // This is used as a truthy flag elsewhere. The use of Date.now is for
   // debugging purposes only.
@@ -350,6 +351,7 @@ function socketCloseListener() {
   req.emit('close');
   if (req.res && req.res.readable) {
     // Socket closed before we emitted 'end' below.
+    req.res.aborted = true;
     req.res.emit('aborted');
     var res = req.res;
     res.on('end', function() {
diff --git a/lib/_http_incoming.js b/lib/_http_incoming.js
index 696fcc3b4ce53d..d51bc3ae6bfd41 100644
--- a/lib/_http_incoming.js
+++ b/lib/_http_incoming.js
@@ -59,6 +59,8 @@ function IncomingMessage(socket) {
 
   this.readable = true;
 
+  this.aborted = false;
+
   this.upgrade = null;
 
   // request (server) only
diff --git a/lib/_http_server.js b/lib/_http_server.js
index 2de2112dfca2ed..b281db03510681 100644
--- a/lib/_http_server.js
+++ b/lib/_http_server.js
@@ -436,6 +436,7 @@ function socketOnClose(socket, state) {
 function abortIncoming(incoming) {
   while (incoming.length) {
     var req = incoming.shift();
+    req.aborted = true;
     req.emit('aborted');
     req.emit('close');
   }
diff --git a/lib/internal/http2/compat.js b/lib/internal/http2/compat.js
index 5913c83b724b43..02b595b6a023f8 100644
--- a/lib/internal/http2/compat.js
+++ b/lib/internal/http2/compat.js
@@ -19,6 +19,7 @@ const kTrailers = Symbol('trailers');
 const kRawTrailers = Symbol('rawTrailers');
 const kProxySocket = Symbol('proxySocket');
 const kSetHeader = Symbol('setHeader');
+const kAborted = Symbol('aborted');
 
 const {
   HTTP2_HEADER_AUTHORITY,
@@ -125,6 +126,7 @@ function onStreamDrain() {
 function onStreamAbortedRequest() {
   const request = this[kRequest];
   if (request !== undefined && request[kState].closed === false) {
+    request[kAborted] = true;
     request.emit('aborted');
   }
 }
@@ -221,6 +223,7 @@ class Http2ServerRequest extends Readable {
     this[kTrailers] = {};
     this[kRawTrailers] = [];
     this[kStream] = stream;
+    this[kAborted] = false;
     stream[kProxySocket] = null;
     stream[kRequest] = this;
 
@@ -236,6 +239,10 @@ class Http2ServerRequest extends Readable {
     this.on('resume', onRequestResume);
   }
 
+  get aborted() {
+    return this[kAborted];
+  }
+
   get complete() {
     return this._readableState.ended ||
            this[kState].closed ||
diff --git a/test/parallel/test-http-aborted.js b/test/parallel/test-http-aborted.js
new file mode 100644
index 00000000000000..c3d7e4641f4501
--- /dev/null
+++ b/test/parallel/test-http-aborted.js
@@ -0,0 +1,26 @@
+'use strict';
+
+const common = require('../common');
+const http = require('http');
+const assert = require('assert');
+
+const server = http.createServer(common.mustCall(function(req, res) {
+  req.on('aborted', common.mustCall(function() {
+    assert.strictEqual(this.aborted, true);
+    server.close();
+  }));
+  assert.strictEqual(req.aborted, false);
+  res.write('hello');
+}));
+
+server.listen(0, common.mustCall(() => {
+  const req = http.get({
+    port: server.address().port,
+    headers: { connection: 'keep-alive' }
+  }, common.mustCall((res) => {
+    res.on('aborted', common.mustCall(() => {
+      assert.strictEqual(res.aborted, true);
+    }));
+    req.abort();
+  }));
+}));
diff --git a/test/parallel/test-http2-compat-aborted.js b/test/parallel/test-http2-compat-aborted.js
new file mode 100644
index 00000000000000..01caf95f98688a
--- /dev/null
+++ b/test/parallel/test-http2-compat-aborted.js
@@ -0,0 +1,27 @@
+'use strict';
+
+const common = require('../common');
+if (!common.hasCrypto)
+  common.skip('missing crypto');
+const h2 = require('http2');
+const assert = require('assert');
+
+
+const server = h2.createServer(common.mustCall(function(req, res) {
+  req.on('aborted', common.mustCall(function() {
+    assert.strictEqual(this.aborted, true);
+  }));
+  assert.strictEqual(req.aborted, false);
+  res.write('hello');
+  server.close();
+}));
+
+server.listen(0, common.mustCall(function() {
+  const url = `http://localhost:${server.address().port}`;
+  const client = h2.connect(url, common.mustCall(() => {
+    const request = client.request();
+    request.on('data', common.mustCall((chunk) => {
+      client.destroy();
+    }));
+  }));
+}));