From 079d3dedad9088cbd0190238bf8008be7dd3704a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E6=B2=B3?= <shixi_liuhe1@staff.sina.com.cn>
Date: Tue, 6 Nov 2018 16:43:06 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=A2=E6=88=B7=E7=AB=AF=E9=94=99=E8=AF=AF?=
 =?UTF-8?q?=E5=A4=84=E7=90=86=E6=9C=BA=E5=88=B6=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/client.go b/client.go
index 0c60dc8..483d17f 100755
--- a/client.go
+++ b/client.go
@@ -93,6 +93,7 @@ func (c *TRPClient) deal(conn net.Conn) error {
 	if nlen <= 0 {
 		log.Println("数据长度错误。")
 		c.werror(conn)
+		return errors.New("数据长度错误")
 	}
 	raw := make([]byte, nlen)
 	n, err := conn.Read(raw)
@@ -102,11 +103,13 @@ func (c *TRPClient) deal(conn net.Conn) error {
 	if n != int(nlen) {
 		log.Printf("读取服务端数据长度错误，已经读取%dbyte，总长度%d字节\n", n, nlen)
 		c.werror(conn)
+		return errors.New("读取服务端数据长度错误")
 	}
 	req, err := DecodeRequest(raw)
 	if err != nil {
 		log.Println("DecodeRequest错误：", err)
 		c.werror(conn)
+		return err
 	}
 	rawQuery := ""
 	if req.URL.RawQuery != "" {
@@ -122,6 +125,7 @@ func (c *TRPClient) deal(conn net.Conn) error {
 	if err != nil && !disRedirect {
 		log.Println("请求本地客户端错误：", err)
 		c.werror(conn)
+		return err
 	}
 	if !disRedirect {
 		defer resp.Body.Close()
@@ -133,10 +137,12 @@ func (c *TRPClient) deal(conn net.Conn) error {
 	if err != nil {
 		log.Println("EncodeResponse错误：", err)
 		c.werror(conn)
+		return err
 	}
 	n, err = conn.Write(respBytes)
 	if err != nil {
 		log.Println("发送数据错误，错误：", err)
+		return err
 	}
 	if n != len(respBytes) {
 		log.Printf("发送数据长度错误，已经发送：%dbyte，总字节长：%dbyte\n", n, len(respBytes))