check error for the first payload

AllenTom · Jan 6, 2017 · f2fcc90 · f2fcc90
1 parent 06a73ad
commit f2fcc90
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/proxy/shadowsocks/server.go b/proxy/shadowsocks/server.go
@@ -187,8 +187,14 @@ func (v *Server) handleConnection(conn internet.Connection) {
 		if err != nil {
 			return err
 		}
-		responseWriter.Write(payload)
-		bufferedWriter.SetBuffered(false)
+		if err := responseWriter.Write(payload); err != nil {
+			return err
+		}
+		payload.Release()
+
+		if err := bufferedWriter.SetBuffered(false); err != nil {
+			return err
+		}
 
 		if err := buf.PipeUntilEOF(ray.InboundOutput(), responseWriter); err != nil {
 			log.Info("Shadowsocks|Server: Failed to transport all TCP response: ", err)