소스 검색

log err on tcp test server

Darien Raymond 8 년 전
부모
커밋
6515cf12e5
1개의 변경된 파일9개의 추가작업 그리고 2개의 파일을 삭제
  1. 9 2
      testing/servers/tcp/tcp.go

+ 9 - 2
testing/servers/tcp/tcp.go

@@ -2,6 +2,7 @@ package tcp
 
 import (
 	"fmt"
+	"io"
 	"net"
 
 	v2net "v2ray.com/core/common/net"
@@ -54,13 +55,19 @@ func (server *Server) handleConnection(conn net.Conn) {
 		conn.Write(server.SendFirst)
 	}
 	request := make([]byte, 4096)
-	for true {
+	for {
 		nBytes, err := conn.Read(request)
 		if err != nil {
+			if err != io.EOF {
+				fmt.Println("Failed to read request:", err)
+			}
 			break
 		}
 		response := server.MsgProcessor(request[:nBytes])
-		conn.Write(response)
+		if _, err := conn.Write(response); err != nil {
+			fmt.Println("Failed to write response:", err)
+			break
+		}
 	}
 	conn.Close()
 }