|
|
@ -377,29 +377,33 @@ func (sv *SUDPVisitor) Run() (err error) {
|
|
|
|
func (sv *SUDPVisitor) dispatcher() {
|
|
|
|
func (sv *SUDPVisitor) dispatcher() {
|
|
|
|
xl := xlog.FromContextSafe(sv.ctx)
|
|
|
|
xl := xlog.FromContextSafe(sv.ctx)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var (
|
|
|
|
|
|
|
|
visitorConn net.Conn
|
|
|
|
|
|
|
|
err error
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
firstPacket *msg.UDPPacket
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
for {
|
|
|
|
for {
|
|
|
|
// loop for get frpc to frps tcp conn
|
|
|
|
select {
|
|
|
|
// setup worker
|
|
|
|
case firstPacket = <-sv.sendCh:
|
|
|
|
// wait worker to finished
|
|
|
|
if firstPacket == nil {
|
|
|
|
// retry or exit
|
|
|
|
|
|
|
|
visitorConn, err := sv.getNewVisitorConn()
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
// check if proxy is closed
|
|
|
|
|
|
|
|
// if checkCloseCh is close, we will return, other case we will continue to reconnect
|
|
|
|
|
|
|
|
select {
|
|
|
|
|
|
|
|
case <-sv.checkCloseCh:
|
|
|
|
|
|
|
|
xl.Info("frpc sudp visitor proxy is closed")
|
|
|
|
xl.Info("frpc sudp visitor proxy is closed")
|
|
|
|
return
|
|
|
|
return
|
|
|
|
default:
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
case <-sv.checkCloseCh:
|
|
|
|
|
|
|
|
xl.Info("frpc sudp visitor proxy is closed")
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
time.Sleep(3 * time.Second)
|
|
|
|
visitorConn, err = sv.getNewVisitorConn()
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
xl.Warn("newVisitorConn to frps error: %v, try to reconnect", err)
|
|
|
|
xl.Warn("newVisitorConn to frps error: %v, try to reconnect", err)
|
|
|
|
continue
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
sv.worker(visitorConn)
|
|
|
|
// visitorConn always be closed when worker done.
|
|
|
|
|
|
|
|
sv.worker(visitorConn, firstPacket)
|
|
|
|
|
|
|
|
|
|
|
|
select {
|
|
|
|
select {
|
|
|
|
case <-sv.checkCloseCh:
|
|
|
|
case <-sv.checkCloseCh:
|
|
|
@ -407,9 +411,10 @@ func (sv *SUDPVisitor) dispatcher() {
|
|
|
|
default:
|
|
|
|
default:
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func (sv *SUDPVisitor) worker(workConn net.Conn) {
|
|
|
|
func (sv *SUDPVisitor) worker(workConn net.Conn, firstPacket *msg.UDPPacket) {
|
|
|
|
xl := xlog.FromContextSafe(sv.ctx)
|
|
|
|
xl := xlog.FromContextSafe(sv.ctx)
|
|
|
|
xl.Debug("starting sudp proxy worker")
|
|
|
|
xl.Debug("starting sudp proxy worker")
|
|
|
|
|
|
|
|
|
|
|
@ -463,6 +468,14 @@ func (sv *SUDPVisitor) worker(workConn net.Conn) {
|
|
|
|
}()
|
|
|
|
}()
|
|
|
|
|
|
|
|
|
|
|
|
var errRet error
|
|
|
|
var errRet error
|
|
|
|
|
|
|
|
if firstPacket != nil {
|
|
|
|
|
|
|
|
if errRet = msg.WriteMsg(conn, firstPacket); errRet != nil {
|
|
|
|
|
|
|
|
xl.Warn("sender goroutine for udp work connection closed: %v", errRet)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
xl.Trace("send udp package to workConn: %s", firstPacket.Content)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
for {
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
select {
|
|
|
|
case udpMsg, ok := <-sv.sendCh:
|
|
|
|
case udpMsg, ok := <-sv.sendCh:
|
|
|
|