[TASK] add temporary blacklist

ssh/manager.go

@@ -2,7 +2,9 @@ package ssh
 
 import (
 	"net"
+	"strings"
 	"sync"
+	"time"
 
 	"golang.org/x/crypto/ssh"
 
@@ -13,6 +15,7 @@ import (
 type Manager struct {
 	config           *ssh.ClientConfig
 	clients          map[string]*ssh.Client
+	clientsBlacklist map[string]time.Time
 	clientsMUX       sync.Mutex
 }
 
@@ -34,12 +37,20 @@ func NewManager(file string) *Manager {
 	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
@@ -47,7 +58,12 @@ func (m *Manager) ConnectTo(addr net.TCPAddr) *ssh.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
 	}