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 connections, (l[%s] r[%s]) (l[%s] r[%s])", localConn.GetLocalAddr(), localConn.GetRemoteAddr(), remoteConn.GetLocalAddr(), remoteConn.GetRemoteAddr()) go msg.JoinMore(localConn, remoteConn, p.BaseConf) return nil }
// start listening for user conns func (p *ProxyServer) Start(c *conn.Conn) (err error) { p.CtlConn = c p.Init() if p.Type == "tcp" { l, err := conn.Listen(p.BindAddr, p.ListenPort) if err != nil { return err } p.listeners = append(p.listeners, l) } else if p.Type == "http" { for _, domain := range p.CustomDomains { l, err := VhostHttpMuxer.Listen(domain) if err != nil { return err } p.listeners = append(p.listeners, l) } } else if p.Type == "https" { for _, domain := range p.CustomDomains { l, err := VhostHttpsMuxer.Listen(domain) if err != nil { return err } p.listeners = append(p.listeners, l) } } p.Lock() p.Status = consts.Working p.Unlock() // start a goroutine for every listener to accept user connection for _, listener := range p.listeners { go func(l Listener) { for { // block // if listener is closed, err returned c, err := l.Accept() if err != nil { log.Info("ProxyName [%s], listener is closed", p.Name) return } log.Debug("ProxyName [%s], get one new user conn [%s]", p.Name, c.GetRemoteAddr()) if p.Status != consts.Working { log.Debug("ProxyName [%s] is not working, new user conn close", p.Name) c.Close() return } // start another goroutine for join two conns from frpc and user go func() { workConn, err := p.getWorkConn() if err != nil { return } userConn := c // msg will transfer to another without modifying // l means local, r means remote log.Debug("Join two connections, (l[%s] r[%s]) (l[%s] r[%s])", workConn.GetLocalAddr(), workConn.GetRemoteAddr(), userConn.GetLocalAddr(), userConn.GetRemoteAddr()) go msg.JoinMore(userConn, workConn, p.BaseConf) }() } }(listener) } return nil }