Handle socket errors

tjgq · Apr 8, 2014 · 044611d · 044611d
1 parent afa0851
commit 044611d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/bin/throttleproxy.js b/bin/throttleproxy.js
@@ -29,7 +29,16 @@ function runProxy(localAddr, remoteAddr, downRate, upRate) {
         var remoteThrottle = new Throttle({rate: downRate});
 
         local.pipe(localThrottle).pipe(remote);
+        local.on('error', function() {
+            remote.destroy();
+            local.destroy();
+        });
+
         remote.pipe(remoteThrottle).pipe(local);
+        remote.on('error', function() {
+            local.destroy();
+            remote.destroy();
+        });
     });
 
     server.listen(localAddr.port, localAddr.host);