package ssh import ( "net" "sync" "golang.org/x/crypto/ssh" "github.com/FreifunkBremen/freifunkmanager/lib/log" ) // the SSH Connection Manager for multiple connections type Manager struct { config *ssh.ClientConfig clients map[string]*ssh.Client clientsMUX sync.Mutex } // create a new SSH Connection Manager by ssh file func NewManager(file string) *Manager { var auths []ssh.AuthMethod if auth := SSHAgent(); auth != nil { auths = append(auths, auth) } if auth := PublicKeyFile(file); auth != nil { auths = append(auths, auth) } sshConfig := &ssh.ClientConfig{ User: "root", Auth: auths, HostKeyCallback: ssh.InsecureIgnoreHostKey(), } return &Manager{ config: sshConfig, clients: make(map[string]*ssh.Client), } } func (m *Manager) ConnectTo(addr net.TCPAddr) *ssh.Client { m.clientsMUX.Lock() defer m.clientsMUX.Unlock() if client, ok := m.clients[addr.IP.String()]; ok { return client } client, err := ssh.Dial("tcp", addr.String(), m.config) if err != nil { log.Log.Error(err) return nil } m.clients[addr.IP.String()] = client return client } func (m *Manager) Close() { for host, client := range m.clients { client.Close() delete(m.clients, host) } }