123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596 |
- package proxy
- import (
- "fmt"
- "net"
- "reflect"
- "strconv"
- v1 "github.com/fatedier/frp/pkg/config/v1"
- )
- func init() {
- RegisterProxyFactory(reflect.TypeOf(&v1.TCPProxyConfig{}), NewTCPProxy)
- }
- type TCPProxy struct {
- *BaseProxy
- cfg *v1.TCPProxyConfig
- realBindPort int
- }
- func NewTCPProxy(baseProxy *BaseProxy) Proxy {
- unwrapped, ok := baseProxy.GetConfigurer().(*v1.TCPProxyConfig)
- if !ok {
- return nil
- }
- baseProxy.usedPortsNum = 1
- return &TCPProxy{
- BaseProxy: baseProxy,
- cfg: unwrapped,
- }
- }
- func (pxy *TCPProxy) Run() (remoteAddr string, err error) {
- xl := pxy.xl
- if pxy.cfg.LoadBalancer.Group != "" {
- l, realBindPort, errRet := pxy.rc.TCPGroupCtl.Listen(pxy.name, pxy.cfg.LoadBalancer.Group, pxy.cfg.LoadBalancer.GroupKey,
- pxy.serverCfg.ProxyBindAddr, pxy.cfg.RemotePort)
- if errRet != nil {
- err = errRet
- return
- }
- defer func() {
- if err != nil {
- l.Close()
- }
- }()
- pxy.realBindPort = realBindPort
- pxy.listeners = append(pxy.listeners, l)
- xl.Infof("tcp proxy listen port [%d] in group [%s]", pxy.cfg.RemotePort, pxy.cfg.LoadBalancer.Group)
- } else {
- pxy.realBindPort, err = pxy.rc.TCPPortManager.Acquire(pxy.name, pxy.cfg.RemotePort)
- if err != nil {
- return
- }
- defer func() {
- if err != nil {
- pxy.rc.TCPPortManager.Release(pxy.realBindPort)
- }
- }()
- listener, errRet := net.Listen("tcp", net.JoinHostPort(pxy.serverCfg.ProxyBindAddr, strconv.Itoa(pxy.realBindPort)))
- if errRet != nil {
- err = errRet
- return
- }
- pxy.listeners = append(pxy.listeners, listener)
- xl.Infof("tcp proxy listen port [%d]", pxy.cfg.RemotePort)
- }
- pxy.cfg.RemotePort = pxy.realBindPort
- remoteAddr = fmt.Sprintf(":%d", pxy.realBindPort)
- pxy.startCommonTCPListenersHandler()
- return
- }
- func (pxy *TCPProxy) Close() {
- pxy.BaseProxy.Close()
- if pxy.cfg.LoadBalancer.Group == "" {
- pxy.rc.TCPPortManager.Release(pxy.realBindPort)
- }
- }
|