bughz/glider
1
0
Fork 0
mirror of https://github.com/nadoo/glider.git synced 2025-02-23 17:35:40 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

ssh: init conn in Dial

Browse Source
This commit is contained in:
nadoo 2021-12-06 23:53:10 +08:00
parent 638adc63d6
commit ff09c45fb6
3 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
go.mod
View File

@ -12,7 +12,7 @@ require (
github.com/nadoo/ipset v0.3.0 github.com/nadoo/ipset v0.3.0
github.com/xtaci/kcp-go/v5 v5.6.1 github.com/xtaci/kcp-go/v5 v5.6.1
golang.org/x/crypto v0.0.0-20211202192323-5770296d904e golang.org/x/crypto v0.0.0-20211202192323-5770296d904e
golang.org/x/sys v0.0.0-20211204120058-94396e421777 golang.org/x/sys v0.0.0-20211205182925-97ca703d548d
) )
require ( require (

4
go.sum
View File

@ -157,8 +157,8 @@ golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7w
golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210525143221-35b2ab0089ea/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.0.0-20210525143221-35b2ab0089ea/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20211204120058-94396e421777 h1:QAkhGVjOxMa+n4mlsAWeAU+BMZmimQAaNiMu+iUi94E= golang.org/x/sys v0.0.0-20211205182925-97ca703d548d h1:FjkYO/PPp4Wi0EAUOVLxePm7qVW4r4ctbWpURyuOD0E=
golang.org/x/sys v0.0.0-20211204120058-94396e421777/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.0.0-20211205182925-97ca703d548d/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1 h1:v+OssWQX+hTHEmOBgwxdZxK4zHq3yOs8F9J7mk0PY8E= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1 h1:v+OssWQX+hTHEmOBgwxdZxK4zHq3yOs8F9J7mk0PY8E=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=

18
proxy/ssh/ssh.go
View File

@ -21,9 +21,7 @@ type SSH struct {
mu sync.Mutex mu sync.Mutex
sshCfg *ssh.ClientConfig sshCfg *ssh.ClientConfig
sshConn ssh.Conn sshClient *ssh.Client
sshChan <-chan ssh.NewChannel
sshReq <-chan *ssh.Request
} }
func init() { func init() {
@ -75,7 +73,7 @@ func NewSSH(s string, d proxy.Dialer, p proxy.Proxy) (*SSH, error) {
t.addr = net.JoinHostPort(t.addr, "22") t.addr = net.JoinHostPort(t.addr, "22")
} }
return t, t.initConn() return t, nil
} }
// NewSSHDialer returns a ssh proxy dialer. // NewSSHDialer returns a ssh proxy dialer.
@ -98,12 +96,12 @@ func (s *SSH) initConn() error {
return err return err
} }
s.sshConn, s.sshChan, s.sshReq, err = ssh.NewClientConn(c, s.addr, s.sshCfg) sshConn, sshChan, sshReq, err := ssh.NewClientConn(c, s.addr, s.sshCfg)
if err != nil { if err != nil {
log.F("[ssh]: initial connection to %s error: %s", s.addr, err) log.F("[ssh]: initial connection to %s error: %s", s.addr, err)
return err return err
} }
s.sshClient = ssh.NewClient(sshConn, sshChan, sshReq)
return nil return nil
} }
@ -112,14 +110,16 @@ func (s *SSH) Dial(network, addr string) (net.Conn, error) {
s.mu.Lock() s.mu.Lock()
defer s.mu.Unlock() defer s.mu.Unlock()
if c, err := ssh.NewClient(s.sshConn, s.sshChan, s.sshReq).Dial(network, addr); err == nil { if s.sshClient != nil {
if c, err := s.sshClient.Dial(network, addr); err == nil {
return c, nil return c, nil
} }
s.sshConn.Close() s.sshClient.Conn.Close()
}
if err := s.initConn(); err != nil { if err := s.initConn(); err != nil {
return nil, err return nil, err
} }
return ssh.NewClient(s.sshConn, s.sshChan, s.sshReq).Dial(network, addr) return s.sshClient.Dial(network, addr)
} }
// DialUDP connects to the given address via the proxy. // DialUDP connects to the given address via the proxy.