80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
package ssh
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"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
|
|
clientsBlacklist map[string]time.Time
|
|
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),
|
|
clientsBlacklist: make(map[string]time.Time),
|
|
}
|
|
}
|
|
|
|
func (m *Manager) ConnectTo(addr net.TCPAddr) *ssh.Client {
|
|
m.clientsMUX.Lock()
|
|
defer m.clientsMUX.Unlock()
|
|
if t, ok := m.clientsBlacklist[addr.IP.String()]; ok {
|
|
if time.Now().Add(-time.Hour * 24).After(t) {
|
|
return nil
|
|
} else {
|
|
delete(m.clientsBlacklist, addr.IP.String())
|
|
}
|
|
}
|
|
|
|
if client, ok := m.clients[addr.IP.String()]; ok {
|
|
return client
|
|
}
|
|
|
|
client, err := ssh.Dial("tcp", addr.String(), m.config)
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "no supported methods remain") {
|
|
m.clientsBlacklist[addr.IP.String()] = time.Now()
|
|
log.Log.Warnf("node was set on the blacklist: %s", err)
|
|
} else {
|
|
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)
|
|
}
|
|
}
|