http: fix socketOnWrap edge cases

Properly handle prependListener wrapping on http server
socket, in addition to on and addListener.

PR-URL: https://github.com/nodejs/node/pull/27968
Reviewed-By: Yongsheng Zhang <zyszys98@gmail.com>
Reviewed-By: Luigi Pinca <luigipinca@gmail.com>
Reviewed-By: Rich Trott <rtrott@gmail.com>

lib/_http_server.js

@@ -404,9 +404,10 @@ function connectionListenerInternal(server, socket) {
   socket.on('resume', onSocketResume);
   socket.on('pause', onSocketPause);
 
-  // Override on to unconsume on `data`, `readable` listeners
+  // Overrides to unconsume on `data`, `readable` listeners
-  socket.on = socketOnWrap;
+  socket.on = generateSocketListenerWrapper('on');
-  socket.addListener = socket.on;
+  socket.addListener = generateSocketListenerWrapper('addListener');
+  socket.prependListener = generateSocketListenerWrapper('prependListener');
 
   // We only consume the socket if it has never been consumed before.
   if (socket._handle && socket._handle.isStreamBase &&
@@ -754,12 +755,13 @@ function unconsume(parser, socket) {
   }
 }
 
-function socketOnWrap(ev, fn) {
+function generateSocketListenerWrapper(originalFnName) {
-  const res = net.Socket.prototype.on.call(this, ev, fn);
+  return function socketListenerWrap(ev, fn) {
+    const res = net.Socket.prototype[originalFnName].call(this, ev, fn);
     if (!this.parser) {
-    this.prependListener = net.Socket.prototype.prependListener;
       this.on = net.Socket.prototype.on;
-    this.addListener = this.on;
+      this.addListener = net.Socket.prototype.addListener;
+      this.prependListener = net.Socket.prototype.prependListener;
       return res;
     }
 
@@ -767,6 +769,7 @@ function socketOnWrap(ev, fn) {
       unconsume(this.parser, this);
 
     return res;
+  };
 }
 
 function resetHeadersTimeoutOnReqEnd() {

test/parallel/test-http-server-unconsume.js

@@ -1,34 +1,33 @@
 'use strict';
-require('../common');
+const common = require('../common');
 const assert = require('assert');
 const http = require('http');
 const net = require('net');
 
-let received = '';
+['on', 'addListener', 'prependListener'].forEach((testFn) => {
+  let received = '';
 
-const server = http.createServer(function(req, res) {
+  const server = http.createServer(function(req, res) {
     res.writeHead(200);
     res.end();
 
-  req.socket.on('data', function(data) {
+    req.socket[testFn]('data', function(data) {
       received += data;
     });
 
-  assert.strictEqual(req.socket.on, req.socket.addListener);
-  assert.strictEqual(req.socket.prependListener,
-                     net.Socket.prototype.prependListener);
-
     server.close();
-}).listen(0, function() {
+  }).listen(0, function() {
     const socket = net.connect(this.address().port, function() {
       socket.write('PUT / HTTP/1.1\r\n\r\n');
 
       socket.once('data', function() {
         socket.end('hello world');
       });
+
+      socket.on('end', common.mustCall(() => {
+        assert.strictEqual(received, 'hello world',
+                           `failed for socket.${testFn}`);
+      }));
+    });
   });
 });
-
-process.on('exit', function() {
-  assert.strictEqual(received, 'hello world');
-});