net: implement capture rejections for 'connection' event

PR-URL: https://github.com/nodejs/node/pull/27867
Reviewed-By: Benjamin Gruenbaum <benjamingr@gmail.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
Reviewed-By: Jeremiah Senkpiel <fishrock123@rocketmail.com>
Reviewed-By: Anna Henningsen <anna@addaleax.net>
Reviewed-By: Michaël Zasso <targos@protonmail.com>

2 files changed, 17 insertions, 2 deletions

diff --git a/lib/_http_server.js b/lib/_http_server.js
index 3444c44a6c..b967cac3ae 100644
--- a/lib/_http_server.js
+++ b/lib/_http_server.js
@@ -359,10 +359,11 @@ Server.prototype.setTimeout = function setTimeout(msecs, callback) {
 };
 
 Server.prototype[EE.captureRejectionSymbol] = function(
-  err, event, req, res) {
+  err, event, ...args) {
 
   switch (event) {
     case 'request':
+      const [ , res] = args;
       if (!res.headersSent && !res.writableEnded) {
         // Don't leak headers.
         for (const name of res.getHeaderNames()) {
@@ -375,7 +376,8 @@ Server.prototype[EE.captureRejectionSymbol] = function(
       }
       break;
     default:
-      this.emit('error', err);
+      net.Server.prototype[Symbol.for('nodejs.rejection')]
+        .call(this, err, event, ...args);
   }
 };
 
diff --git a/lib/net.js b/lib/net.js
index 02fd187480..7e5d0ea621 100644
--- a/lib/net.js
+++ b/lib/net.js
@@ -1653,6 +1653,19 @@ function emitCloseNT(self) {
 }
 
 
+Server.prototype[EventEmitter.captureRejectionSymbol] = function(
+  err, event, sock) {
+
+  switch (event) {
+    case 'connection':
+      sock.destroy(err);
+      break;
+    default:
+      this.emit('error', err);
+  }
+};
+
+
 // Legacy alias on the C++ wrapper object. This is not public API, so we may
 // want to runtime-deprecate it at some point. There's no hurry, though.
 ObjectDefineProperty(TCP.prototype, 'owner', {