mirror of
https://gitee.com/IrisVega/frp.git
synced 2024-11-01 22:31:29 +08:00
e805acd1e3
Conflicts: cmd/frpc/config.go cmd/frpc/control.go cmd/frpc/main.go cmd/frps/config.go cmd/frps/control.go cmd/frps/main.go pkg/models/server.go
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/fatedier/frp/pkg/utils/conn"
|
|
"github.com/fatedier/frp/pkg/utils/log"
|
|
)
|
|
|
|
type ProxyClient struct {
|
|
Name string
|
|
Passwd string
|
|
LocalPort int64
|
|
}
|
|
|
|
func (p *ProxyClient) GetLocalConn() (c *conn.Conn, err error) {
|
|
c = &conn.Conn{}
|
|
err = c.ConnectServer("127.0.0.1", p.LocalPort)
|
|
if err != nil {
|
|
log.Error("ProxyName [%s], connect to local port error, %v", p.Name, err)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (p *ProxyClient) GetRemoteConn(addr string, port int64) (c *conn.Conn, err error) {
|
|
c = &conn.Conn{}
|
|
defer func() {
|
|
if err != nil {
|
|
c.Close()
|
|
}
|
|
}()
|
|
|
|
err = c.ConnectServer(addr, port)
|
|
if err != nil {
|
|
log.Error("ProxyName [%s], connect to server [%s:%d] error, %v", p.Name, addr, port, err)
|
|
return
|
|
}
|
|
|
|
req := &ClientCtlReq{
|
|
Type: WorkConn,
|
|
ProxyName: p.Name,
|
|
Passwd: p.Passwd,
|
|
}
|
|
|
|
buf, _ := json.Marshal(req)
|
|
err = c.Write(string(buf) + "\n")
|
|
if err != nil {
|
|
log.Error("ProxyName [%s], write to server error, %v", p.Name, err)
|
|
return
|
|
}
|
|
|
|
err = nil
|
|
return
|
|
}
|
|
|
|
func (p *ProxyClient) StartTunnel(serverAddr string, serverPort int64) (err error) {
|
|
localConn, err := p.GetLocalConn()
|
|
if err != nil {
|
|
return
|
|
}
|
|
remoteConn, err := p.GetRemoteConn(serverAddr, serverPort)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// l means local, r means remote
|
|
log.Debug("Join two conns, (l[%s] r[%s]) (l[%s] r[%s])", localConn.GetLocalAddr(), localConn.GetRemoteAddr(),
|
|
remoteConn.GetLocalAddr(), remoteConn.GetRemoteAddr())
|
|
go conn.Join(localConn, remoteConn)
|
|
return nil
|
|
}
|