Drivers: SSH allow skipping known host check

driver/ssh.go

@@ -2,8 +2,10 @@ package driver
 
 import (
 	"fmt"
+
 	"github.com/melbahja/goph"
 	log "github.com/sirupsen/logrus"
+	"golang.org/x/crypto/ssh"
 )
 
 // SSH : Driver for handling ssh executions
@@ -19,6 +21,8 @@ type SSH struct {
 	PubKeyFile string
 	// Pass key for public key file
 	PubKeyPass string
+	// Check known hosts (only disable for tests
+	CheckKnownHosts bool
 }
 
 func (d *SSH) String() string {
@@ -27,6 +31,9 @@ func (d *SSH) String() string {
 
 // set the goph Client
 func (d *SSH) Client() (*goph.Client, error) {
+	var err error
+	var client *goph.Client
+	var callback ssh.HostKeyCallback
 	port := 22
 	if d.Port != 0 {
 		port = d.Port
@@ -35,21 +42,22 @@ func (d *SSH) Client() (*goph.Client, error) {
 	if err != nil {
 		return nil, err
 	}
-	callback, err := goph.DefaultKnownHosts()
-	if err != nil {
-		return nil, err
+	if d.CheckKnownHosts {
+		callback, err = goph.DefaultKnownHosts()
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		callback = ssh.InsecureIgnoreHostKey()
 	}
-	client, err := goph.NewConn(&goph.Config{
+	client, err = goph.NewConn(&goph.Config{
 		User:     d.User,
 		Addr:     d.Host,
 		Port:     uint(port),
 		Auth:     auth,
 		Timeout:  goph.DefaultTimeout,
 		Callback: callback,
 	})
-	if err != nil {
-		return nil, err
-	}
 	return client, err
 }
 

driver/ssh_test.go

@@ -1 +1,18 @@
 package driver
+
+import "testing"
+
+func TestSSHRunCommand(t *testing.T) {
+	d := SSH{
+		User:            "root",
+		Host:            "172.17.0.2",
+		Port:            2222,
+		PubKeyFile:      "/home/deven/.ssh/id_rsa",
+		PubKeyPass:      "",
+		CheckKnownHosts: false,
+	}
+	_, err := d.runCommand(`ps -A`)
+	if err != nil {
+		t.Error(err)
+	}
+}

go.mod

@@ -7,4 +7,5 @@ require (
 	github.com/sirupsen/logrus v1.8.1
 	github.com/spf13/cobra v1.2.1
 	github.com/spf13/viper v1.9.0
+	golang.org/x/crypto v0.0.0-20210817164053-32db794688a5
 )