frp/server/proxy/xtcp.go

86 lines
2.0 KiB
Go
Raw Normal View History

2019-01-14 17:11:08 +01:00
// Copyright 2019 fatedier, fatedier@gmail.com
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package proxy
import (
"fmt"
2022-08-28 19:02:53 +02:00
"github.com/fatedier/golib/errors"
"golang.org/x/time/rate"
2022-08-28 19:02:53 +02:00
2020-09-23 07:49:14 +02:00
"github.com/fatedier/frp/pkg/config"
"github.com/fatedier/frp/pkg/msg"
2019-01-14 17:11:08 +01:00
)
2020-05-24 11:48:37 +02:00
type XTCPProxy struct {
2019-01-31 09:49:23 +01:00
*BaseProxy
2020-05-24 11:48:37 +02:00
cfg *config.XTCPProxyConf
2019-01-14 17:11:08 +01:00
closeCh chan struct{}
}
2020-05-24 11:48:37 +02:00
func (pxy *XTCPProxy) Run() (remoteAddr string, err error) {
2019-10-12 14:13:12 +02:00
xl := pxy.xl
2019-01-14 17:11:08 +01:00
if pxy.rc.NatHoleController == nil {
err = fmt.Errorf("xtcp is not supported in frps")
return
}
allowUsers := pxy.cfg.AllowUsers
// if allowUsers is empty, only allow same user from proxy
if len(allowUsers) == 0 {
allowUsers = []string{pxy.GetUserInfo().User}
}
sidCh := pxy.rc.NatHoleController.ListenClient(pxy.GetName(), pxy.cfg.Sk, allowUsers)
2019-01-14 17:11:08 +01:00
go func() {
for {
select {
case <-pxy.closeCh:
return
case sid := <-sidCh:
2019-03-29 12:01:18 +01:00
workConn, errRet := pxy.GetWorkConnFromPool(nil, nil)
2019-01-14 17:11:08 +01:00
if errRet != nil {
continue
}
m := &msg.NatHoleSid{
Sid: sid,
2019-01-14 17:11:08 +01:00
}
errRet = msg.WriteMsg(workConn, m)
if errRet != nil {
2019-10-12 14:13:12 +02:00
xl.Warn("write nat hole sid package error, %v", errRet)
2019-01-14 17:11:08 +01:00
}
workConn.Close()
2019-01-14 17:11:08 +01:00
}
}
}()
return
}
2020-05-24 11:48:37 +02:00
func (pxy *XTCPProxy) GetConf() config.ProxyConf {
2019-01-14 17:11:08 +01:00
return pxy.cfg
}
func (pxy *XTCPProxy) GetLimiter() *rate.Limiter {
return pxy.limiter
}
2020-05-24 11:48:37 +02:00
func (pxy *XTCPProxy) Close() {
2019-01-14 17:11:08 +01:00
pxy.BaseProxy.Close()
pxy.rc.NatHoleController.CloseClient(pxy.GetName())
2022-08-28 19:02:53 +02:00
_ = errors.PanicToError(func() {
2019-01-14 17:11:08 +01:00
close(pxy.closeCh)
})
}