@dev-swarup/http-mitm-proxy 0.9.6 → 0.9.7

package/lib/proxy.js

@@ -37,6 +37,7 @@ var Proxy = function () {
   this.onResponseHeadersHandlers = [];
   this.onResponseDataHandlers = [];
   this.onResponseEndHandlers = [];
+  this.socketMapping = new Map();
   this.responseContentPotentiallyModified = false;
 };
 
@@ -424,16 +425,44 @@ Proxy.prototype._onHttpServerConnectData = function (req, socket, head) {
         port: port,
       },
       function () {
+        if (conn.localPort) {
+            self.socketMapping.set(conn.localPort, socket);
+        }
+
         // create a tunnel between the two hosts
         conn.write(head);
         conn.pipe(socket);
         return socket.pipe(conn);
       }
     );
-    conn.on('error', self._onSocketError.bind(self, 'PROXY_TO_PROXY_SOCKET'));
-    conn.once('end', () => connClosing(socket));
-    conn.once('finish', () => connClosing(socket));
-    conn.once('close', () => connClosing(socket));
+    conn.on('error', ()=> {
+      if (conn.localPort) {
+        self.socketMapping.delete(conn.localPort);
+      };
+
+      self._onSocketError.bind(self, 'PROXY_TO_PROXY_SOCKET');
+    });
+    conn.once('end', () => {
+      if (conn.localPort) {
+        self.socketMapping.delete(conn.localPort);
+      };
+
+      connClosing(socket);
+    });
+    conn.once('finish', () => {
+      if (conn.localPort) {
+        self.socketMapping.delete(conn.localPort);
+      };
+
+      connClosing(socket);
+    });
+    conn.once('close', () => {
+      if (conn.localPort) {
+        self.socketMapping.delete(conn.localPort);
+      };
+
+      connClosing(socket);
+    });
     conn.setNoDelay();
     socket.once('end', () => socketClosing(conn));
     socket.once('finish', () => socketClosing(conn));
@@ -664,6 +693,14 @@ Proxy.prototype._onWebSocketServerConnect = function (isSSL, ws, upgradeReq) {
 
 Proxy.prototype._onHttpServerRequest = function (isSSL, clientToProxyRequest, proxyToClientResponse) {
   var self = this;
+
+  const remotePort = clientToProxyRequest.socket.remotePort;
+  if (remotePort && self.socketMapping.has(remotePort)) {
+    clientToProxyRequest.originalSocket = self.socketMapping.get(remotePort);
+  }
+
+  self.socketMapping.delete(remotePort);
+  
   var ctx = {
     isSSL: isSSL,
     clientToProxyRequest: clientToProxyRequest,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dev-swarup/http-mitm-proxy",
-  "version": "0.9.6",
+  "version": "0.9.7",
   "description": "HTTP Man In The Middle (MITM) Proxy. This is a fork of Joe Ferners' library node-http-mitm-proxy. Its first release was identical to the master version of the original library, commit 66ac0f5d3298f66b731f90ebf1e9b430fa5d76eb. I decided to publish a scoped version of this library to npm, since I needed the codebase in npm. It is only intended for my own library cypress-ntlm-auth. Use at you own risk!",
   "main": "index.js",
   "types": "index.d.ts",