better handling of requests that die

proxy.js

@@ -139,8 +139,14 @@ Proxy.prototype.next_socket = function(cb) {
         return self.waiting.push(cb);
     }
 
+    var done_called = false;
     // put the socket back
     function done() {
+        if (done_called) {
+            throw new Error('done called multiple times');
+        }
+
+        done_called = true;
         if (!sock.destroyed) {
             debug('retuning socket');
             self.sockets.push(sock);

server.js

@@ -49,6 +49,8 @@ function maybe_bounce(req, res, bounce) {
         return true;
     }
 
+    // flag if we already finished before we get a socket
+    // we can't respond to these requests
     var finished = false;
     on_finished(res, function(err) {
         finished = true;
@@ -63,6 +65,7 @@ function maybe_bounce(req, res, bounce) {
         if (finished) {
             return done();
         }
+
         // happens when client upstream is disconnected
         // we gracefully inform the user and kill their conn
         // without this, the browser will leave some connections open
@@ -79,12 +82,22 @@ function maybe_bounce(req, res, bounce) {
 
         stream.on('error', function(err) {
             socket.destroy();
+            req.connection.destroy();
+            done();
         });
 
         // return the socket to the client pool
         stream.once('end', function() {
             done();
         });
+
+        on_finished(res, function(err) {
+            if (err) {
+                req.connection.destroy();
+                socket.destroy();
+                done();
+            }
+        });
     });
 
     return true;