XiaoMo/frp
1
0
Fork 0
mirror of https://github.com/fatedier/frp.git synced 2025-07-27 07:35:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use encryption in frp protocol.

Browse Source
This commit is contained in:
fatedier
2017-03-10 01:42:06 +08:00
parent f83a2a73ab
commit f90028cf96
5 changed files with 38 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
client/control.go
View File

@@ -23,6 +23,7 @@ import (
"github.com/fatedier/frp/models/config"
"github.com/fatedier/frp/models/msg"
"github.com/fatedier/frp/utils/crypto"
"github.com/fatedier/frp/utils/log"
"github.com/fatedier/frp/utils/net"
"github.com/fatedier/frp/utils/util"
@@ -135,7 +136,7 @@ func (ctl *Control) NewWorkConn() {
var startMsg msg.StartWorkConn
if err = msg.ReadMsgInto(workConn, &startMsg); err != nil {
ctl.Error("work connection closed and no response from server, %v", err)
ctl.Error("work connection closed, %v", err)
workConn.Close()
return
}
@@ -205,12 +206,17 @@ func (ctl *Control) reader() {
ctl.Error("panic error: %v", err)
}
}()
defer close(ctl.closedCh)
encReader, err := crypto.NewReader(ctl.conn, []byte(config.ClientCommonCfg.PrivilegeToken))
if err != nil {
ctl.conn.Error("crypto new reader error: %v", err)
return
}
for {
if m, err := msg.ReadMsg(ctl.conn); err != nil {
if m, err := msg.ReadMsg(encReader); err != nil {
if err == io.EOF {
ctl.Debug("read from control connection EOF")
close(ctl.closedCh)
return
} else {
ctl.Warn("read error: %v", err)
@@ -223,12 +229,18 @@ func (ctl *Control) reader() {
}
func (ctl *Control) writer() {
encWriter, err := crypto.NewWriter(ctl.conn, []byte(config.ClientCommonCfg.PrivilegeToken))
if err != nil {
ctl.conn.Error("crypto new writer error: %v", err)
ctl.conn.Close()
return
}
for {
if m, ok := <-ctl.sendCh; !ok {
ctl.Info("control writer is closing")
return
} else {
if err := msg.WriteMsg(ctl.conn, m); err != nil {
if err := msg.WriteMsg(encWriter, m); err != nil {
ctl.Warn("write message to control connection error: %v", err)
return
}

3
client/proxy.go
View File

@@ -23,8 +23,11 @@ import (
"github.com/fatedier/frp/utils/net"
)
// Proxy defines how to work for different proxy type.
type Proxy interface {
Run()
// InWorkConn accept work connections registered to server.
InWorkConn(conn net.Conn)
Close()
}