cleanup

sibalma · May 3, 2017 · 976395b · 976395b
1 parent 5b15d9a
commit 976395b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/testing/scenarios/vmess_test.go b/testing/scenarios/vmess_test.go
@@ -1209,15 +1209,15 @@ func TestVMessGCMMuxUDP(t *testing.T) {
 
 				xorpayload := xor(payload)
 
-				for j := 0; j < 5; j++ {
+				for j := 0; j < 2; j++ {
 					nBytes, _, err := conn.WriteMsgUDP(payload, nil, nil)
 					assert.Error(err).IsNil()
 					assert.Int(nBytes).Equals(len(payload))
 				}
 
 				response := make([]byte, 1024)
 				oob := make([]byte, 16)
-				for j := 0; j < 5; j++ {
+				for j := 0; j < 2; j++ {
 					nBytes, _, _, _, err := conn.ReadMsgUDP(response, oob)
 					assert.Error(err).IsNil()
 					assert.Int(nBytes).Equals(1024)

diff --git a/testing/servers/udp/udp.go b/testing/servers/udp/udp.go
@@ -41,7 +41,9 @@ func (server *Server) handleConnection(conn *net.UDPConn) {
 		}
 
 		response := server.MsgProcessor(buffer[:nBytes])
-		conn.WriteToUDP(response, addr)
+		if _, err := conn.WriteToUDP(response, addr); err != nil {
+			fmt.Println("Failed to write to UDP: ", err.Error())
+		}
 	}
 }