From aa4d897a21e2628f4c3d1050998a49ce2363af90 Mon Sep 17 00:00:00 2001 From: V2Ray Date: Tue, 6 Oct 2015 09:33:46 +0200 Subject: [PATCH] Leverage v2net functions --- proxy/vmess/vmessout.go | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/proxy/vmess/vmessout.go b/proxy/vmess/vmessout.go index 8d18e178..d9492c2d 100644 --- a/proxy/vmess/vmessout.go +++ b/proxy/vmess/vmessout.go @@ -171,12 +171,10 @@ func handleResponse(conn net.Conn, request *protocol.VMessRequest, output chan<- log.Error("VMessOut: Failed to create decrypt reader: %v", err) return } - - buffer := make([]byte, 2*1024) - - nBytes, err := decryptResponseReader.Read(buffer) + + buffer, err := v2net.ReadFrom(decryptResponseReader) if err != nil { - //log.Error("VMessOut: Failed to read VMess response (%d bytes): %v", nBytes, err) + log.Error("VMessOut: Failed to read VMess response (%d bytes): %v", nBytes, err) return } if !bytes.Equal(buffer[:4], request.ResponseHeader[:]) {