优化一些内容

core/mylog/log.go

@@ -30,8 +30,6 @@ func init() {
 }
 
 func InitLog(file string, level string) {
-	fmt.Printf("### Log file is %s, env is %s ###\n", file, env)
-
 	if env == "develop" {
 		logrus.SetFormatter(&logrus.TextFormatter{
 			ForceColors: false,
@@ -68,4 +66,6 @@ func InitLog(file string, level string) {
 		}
 		logrus.SetOutput(logFile)
 	}
+
+	fmt.Printf("### Log file is [%s], log level is [%s], env is [%s] ###\n", file, logrus.GetLevel().String(), env)
 }

core/network/connection.go

@@ -3,7 +3,6 @@ package network
 import (
 	"encoding/binary"
 	"errors"
-	"fmt"
 	"github.com/sirupsen/logrus"
 	"net"
 )
@@ -38,12 +37,6 @@ func (c Connection) Read(length uint32) ([]byte, error) {
 
 	var buf = make([]byte, length)
 	var bufSize, err = c.conn.Read(buf)
-	// 连接被远程主机主动关闭
-	if bufSize == 0 {
-		e := fmt.Sprintf("conn closed by remote, %s -> %s", c.LocalAddress(), c.RemoteAddress())
-		logrus.Debug(e)
-		return nil, errors.New(e)
-	}
 	if err != nil {
 		c.Close()
 		return nil, err

local/main.go

@@ -69,11 +69,12 @@ func handlerSocks5(conn network.Connection) {
 		logrus.Warn(err)
 		return
 	}
+	logrus.Debugf("Socks5 auth success: %s", conn.RemoteAddress())
 	remoteConn, err := socks.ConnectRemote(conn, remoteServer, password)
 	if err != nil {
 		logrus.Warn(err)
 		return
 	}
-
+	logrus.Debugf("Socks5 remote connect success: %s", conn.RemoteAddress())
 	socks.HandlerSocks5Data(conn, remoteConn)
 }