mirror of https://github.com/fatedier/frp
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
1.5 KiB
71 lines
1.5 KiB
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
|
|
}
|
|
|
|
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
|
|
}
|