destroy socket after a timeout

TheKhanj · Dec 22, 2023 · 05ffd6d · 05ffd6d
1 parent 6077bd8
commit 05ffd6d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "moxyjs",
-  "version": "0.0.7",
+  "version": "0.0.8",
   "description": "Distributed transparent proxy with traffic control facilities",
   "author": "Pooyan Khanjankhani <[email protected]>",
   "license": "MIT",

diff --git a/src/proxy.ts b/src/proxy.ts
@@ -95,6 +95,10 @@ class TcpProxy implements Proxy {
       clientSocket.on("close", () => {
         this.logger.log("Client disconnected");
         forwardSocket.end();
+        setTimeout(() => {
+          if (forwardSocket.destroyed) return;
+          forwardSocket.destroy();
+        }, 10_000);
       });
 
       clientSocket.on("error", (err) => {
@@ -104,6 +108,10 @@ class TcpProxy implements Proxy {
         } else {
           this.logger.error("Local socket error:", err);
           clientSocket.end();
+          setTimeout(() => {
+            if (clientSocket.destroyed) return;
+            clientSocket.destroy();
+          }, 10_000);
         }
       });
 
@@ -114,6 +122,10 @@ class TcpProxy implements Proxy {
       forwardSocket.on("error", (err) => {
         this.logger.error(`Forward socket error: ${err}`);
         clientSocket.end();
+        setTimeout(() => {
+          if (clientSocket.destroyed) return;
+          clientSocket.destroy();
+        }, 10_000);
       });
     }