Hurricanezwf 9 vuotta sitten
vanhempi
commit
60c9804776
7 muutettua tiedostoa jossa 19 lisäystä ja 18 poistoa
  1. 1 0
      .gitignore
  2. 1 1
      cmd/frpc/main.go
  3. 1 1
      cmd/frps/main.go
  4. 8 8
      pkg/models/client.go
  5. 6 7
      pkg/models/msg.go
  6. 1 0
      pkg/models/server.go
  7. 1 1
      pkg/utils/log/log.go

+ 1 - 0
.gitignore

@@ -28,3 +28,4 @@ bin/
 
 # Cache
 *.swp
+*.swo

+ 1 - 1
cmd/frpc/main.go

@@ -3,7 +3,7 @@ package main
 import (
 	"os"
 	"sync"
-	
+
 	"frp/pkg/utils/log"
 )
 

+ 1 - 1
cmd/frps/main.go

@@ -3,8 +3,8 @@ package main
 import (
 	"os"
 
-	"frp/pkg/utils/log"
 	"frp/pkg/utils/conn"
+	"frp/pkg/utils/log"
 )
 
 func main() {

+ 8 - 8
pkg/models/client.go

@@ -8,9 +8,9 @@ import (
 )
 
 type ProxyClient struct {
-	Name		string
-	Passwd		string
-	LocalPort	int64
+	Name      string
+	Passwd    string
+	LocalPort int64
 }
 
 func (p *ProxyClient) GetLocalConn() (c *conn.Conn, err error) {
@@ -24,7 +24,7 @@ func (p *ProxyClient) GetLocalConn() (c *conn.Conn, err error) {
 
 func (p *ProxyClient) GetRemoteConn(addr string, port int64) (c *conn.Conn, err error) {
 	c = &conn.Conn{}
-	defer func(){
+	defer func() {
 		if err != nil {
 			c.Close()
 		}
@@ -37,9 +37,9 @@ func (p *ProxyClient) GetRemoteConn(addr string, port int64) (c *conn.Conn, err
 	}
 
 	req := &ClientCtlReq{
-		Type:		WorkConn,
-		ProxyName:	p.Name,
-		Passwd:		p.Passwd,
+		Type:      WorkConn,
+		ProxyName: p.Name,
+		Passwd:    p.Passwd,
 	}
 
 	buf, _ := json.Marshal(req)
@@ -64,7 +64,7 @@ func (p *ProxyClient) StartTunnel(serverAddr string, serverPort int64) (err erro
 	}
 
 	log.Debug("Join two conns, (l[%s] r[%s]) (l[%s] r[%s])", localConn.GetLocalAddr(), localConn.GetRemoteAddr(),
-			remoteConn.GetLocalAddr(), remoteConn.GetRemoteAddr())
+		remoteConn.GetLocalAddr(), remoteConn.GetRemoteAddr())
 	go conn.Join(localConn, remoteConn)
 	return nil
 }

+ 6 - 7
pkg/models/msg.go

@@ -1,8 +1,8 @@
 package models
 
 type GeneralRes struct {
-	Code			int64	`json:"code"`
-	Msg				string	`json:"msg"`
+	Code int64  `json:"code"`
+	Msg  string `json:"msg"`
 }
 
 // type
@@ -12,16 +12,15 @@ const (
 )
 
 type ClientCtlReq struct {
-	Type			int64	`json:"type"`
-	ProxyName		string	`json:"proxy_name"`
-	Passwd			string	`json:"passwd"`
+	Type      int64  `json:"type"`
+	ProxyName string `json:"proxy_name"`
+	Passwd    string `json:"passwd"`
 }
 
 type ClientCtlRes struct {
 	GeneralRes
 }
 
-
 type ServerCtlReq struct {
-	Type			int64	`json:"type"`
+	Type int64 `json:"type"`
 }

+ 1 - 0
pkg/models/server.go

@@ -89,6 +89,7 @@ func (p *ProxyServer) Start() (err error) {
 				p.UserConnList.Remove(element)
 			} else {
 				cliConn.Close()
+				p.Unlock()
 				continue
 			}
 			p.Unlock()

+ 1 - 1
pkg/utils/log/log.go

@@ -22,7 +22,7 @@ func SetLogFile(logWay string, logFile string) {
 	if logWay == "console" {
 		Log.SetLogger("console", "")
 	} else {
-		Log.SetLogger("file", `{"filename": "` + logFile + `"}`)
+		Log.SetLogger("file", `{"filename": "`+logFile+`"}`)
 	}
 }