From 6d0efaeb0757ee47dadedfdc6aa5efa78d78981e Mon Sep 17 00:00:00 2001
From: "rui.zheng" <rui.zheng@tcl.com>
Date: Wed, 25 Mar 2015 16:13:01 +0800
Subject: [PATCH] add print

---
 gost.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/gost.go b/gost.go
index 707c4f1..b9d1230 100644
--- a/gost.go
+++ b/gost.go
@@ -233,7 +233,9 @@ func (g *Gost) srv(conn net.Conn) {
 			}
 			lg.Logln("|<<<", cmd)
 
-			tunnelUdp(conn, uconn, true)
+			if err := tunnelUdp(conn, uconn, true); err != nil {
+				lg.Logln(err)
+			}
 			/*
 				up, err := ReadUdpPayload(uconn)
 				if err != nil {
@@ -348,7 +350,6 @@ func tunnelUdp(conn net.Conn, uconn *net.UDPConn, rawUdp bool) (err error) {
 				rChan <- err
 				return
 			}
-			log.Println("r", up)
 
 			addr, err := net.ResolveUDPAddr("udp", net.JoinHostPort(up.Addr, strconv.Itoa(int(up.Port))))
 			if err != nil {
@@ -360,6 +361,7 @@ func tunnelUdp(conn net.Conn, uconn *net.UDPConn, rawUdp bool) (err error) {
 					rChan <- err
 					return
 				}
+				log.Println("r", up)
 			} else {
 				up.Rsv = 0
 				buf := &bytes.Buffer{}
@@ -368,6 +370,7 @@ func tunnelUdp(conn net.Conn, uconn *net.UDPConn, rawUdp bool) (err error) {
 					rChan <- err
 					return
 				}
+				log.Println("r", up)
 			}
 		}
 	}()
@@ -394,7 +397,7 @@ func tunnelUdp(conn net.Conn, uconn *net.UDPConn, rawUdp bool) (err error) {
 				continue
 			}
 
-			rbuf := bytes.NewReader(b)
+			rbuf := bytes.NewReader(b[:n])
 			up, err := ReadUdpPayload(rbuf)
 			if err != nil {
 				wChan <- err