minor

shadowsocks · Jan 12, 2015 · 9b92445 · 9b92445
1 parent c9ad1e8
commit 9b92445
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/local.js b/local.js
diff --git a/server.js b/server.js
diff --git a/src/local.coffee b/src/local.coffee
@@ -184,7 +184,7 @@ server = net.createServer (connection) ->
       return
 
   connection.on "drain", ->
-    ws.resume()
+    ws._socket.resume()
 
   connection.setTimeout timeout, ->
     console.log "local timeout"

diff --git a/src/server.coffee b/src/server.coffee
@@ -102,10 +102,10 @@ wss.on "connection", (ws) ->
           console.log "remote disconnected"
 
         remote.on "drain", ->
-          ws.resume()
+          ws._socket.resume()
 
         remote.on "error", (e)->
-          ws.close()
+          ws.terminate()
           console.log "remote: #{e}"
 
         remote.setTimeout timeout, ->
@@ -134,7 +134,7 @@ wss.on "connection", (ws) ->
     ws.pong '', null, true
 
   ws._socket.on "drain", ->
-    remote.resume() if remote
+    remote.resume() if stage is 5
 
   ws.on "close", ->
     console.log "server disconnected"