Add test to check if ssh is working

The test checks if ssh is working on the guest by attempting to use the
credentials generated by images verify to login into the guest and to
run a simple command (e.g. 'echo hello').

Signed-off-by: Felix Matouschek <[email protected]>

Author: 0xFelix

artifacts/centos/centos.go

@@ -117,6 +117,7 @@ func (c *centos) UserData(data *docs.UserData) string {
 func (c *centos) Tests() []api.ArtifactTest {
 	return []api.ArtifactTest{
 		tests.GuestOsInfo,
+		tests.SSH,
 	}
 }
 

artifacts/centosstream/centos-stream.go

@@ -101,6 +101,7 @@ func (c *centos) UserData(data *docs.UserData) string {
 func (c *centos) Tests() []api.ArtifactTest {
 	return []api.ArtifactTest{
 		tests.GuestOsInfo,
+		tests.SSH,
 	}
 }
 

artifacts/fedora/fedora.go

@@ -87,6 +87,7 @@ func (f *fedora) UserData(data *docs.UserData) string {
 func (f *fedora) Tests() []api.ArtifactTest {
 	return []api.ArtifactTest{
 		tests.GuestOsInfo,
+		tests.SSH,
 	}
 }
 

artifacts/rhcos/rhcos.go

@@ -73,6 +73,7 @@ func (r *rhcos) UserData(data *docs.UserData) string {
 func (r *rhcos) Tests() []api.ArtifactTest {
 	return []api.ArtifactTest{
 		tests.GuestOsInfo,
+		tests.SSH,
 	}
 }
 

artifacts/rhcosprerelease/rhcos.go

@@ -92,6 +92,7 @@ func (r *rhcos) UserData(data *docs.UserData) string {
 func (r *rhcos) Tests() []api.ArtifactTest {
 	return []api.ArtifactTest{
 		tests.GuestOsInfo,
+		tests.SSH,
 	}
 }
 

pkg/tests/ssh.go

@@ -0,0 +1,61 @@
+package tests
+
+import (
+	"context"
+	"fmt"
+
+	"golang.org/x/crypto/ssh"
+	v1 "kubevirt.io/api/core/v1"
+	kvirtcli "kubevirt.io/client-go/kubecli"
+	"kubevirt.io/containerdisks/pkg/api"
+)
+
+func SSH(ctx context.Context, vmi *v1.VirtualMachineInstance, params *api.ArtifactTestParams) error {
+	kvirtClient, err := kvirtcli.GetKubevirtClient()
+	if err != nil {
+		return err
+	}
+
+	signer, err := ssh.NewSignerFromKey(params.PrivateKey)
+	if err != nil {
+		return err
+	}
+
+	config := &ssh.ClientConfig{
+		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
+		User:            params.Username,
+		Auth: []ssh.AuthMethod{
+			ssh.PublicKeys(signer),
+		},
+	}
+
+	return retryTest(ctx, func() error {
+		return testSSH(vmi, kvirtClient, config)
+	})
+}
+
+func testSSH(vmi *v1.VirtualMachineInstance, kvirtClient kvirtcli.KubevirtClient, config *ssh.ClientConfig) error {
+	tunnel, err := kvirtClient.VirtualMachineInstance(vmi.Namespace).PortForward(vmi.Name, 22, "tcp")
+	if err != nil {
+		return fmt.Errorf("failed to forward ssh port: %w", err)
+	}
+
+	conn := tunnel.AsConn()
+	addr := fmt.Sprintf("vmi/%s.%s:22", vmi.Name, vmi.Namespace)
+	sshConn, chans, reqs, err := ssh.NewClientConn(conn, addr, config)
+	if err != nil {
+		return err
+	}
+
+	session, err := ssh.NewClient(sshConn, chans, reqs).NewSession()
+	if err != nil {
+		return err
+	}
+
+	err = session.Run("echo hello")
+	if err != nil {
+		return err
+	}
+
+	return nil
+}