test/extended/util/compat_otp/ssh_client.go - main - githubmirror/origin

test/extended/util/compat_otp/ssh_client.go
101 строка3 KB
Justin Pierce
Add compat_otp package with openshift-tests-private (OTP) util implementations
11 авг 2025, 18:00
11 авг 2025, 18:0080ac5c1
package compat_otp

import (
	"fmt"
	"io/ioutil"
	"net"
	"os"
	"path/filepath"

	"golang.org/x/crypto/ssh"

	e2e "k8s.io/kubernetes/test/e2e/framework"
)

type SshClient struct {
	User       string
	Host       string
	Port       int
	PrivateKey string
}

func (sshClient *SshClient) getConfig() (*ssh.ClientConfig, error) {
	pemBytes, err := ioutil.ReadFile(sshClient.PrivateKey)
	if err != nil {
		e2e.Logf("Pem byte failed:%v", err)
	}
	signer, err := ssh.ParsePrivateKey(pemBytes)
	if err != nil {
		e2e.Logf("Parse key failed:%v", err)
	}
	config := &ssh.ClientConfig{
		User: sshClient.User,
		Auth: []ssh.AuthMethod{ssh.PublicKeys(signer)},
		HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
			return nil
		},
	}
	return config, err
}

// Run runs cmd on the remote host.
func (sshClient *SshClient) Run(cmd string) error {
	combinedOutput, err := sshClient.RunOutput(cmd)
	if err != nil {
		return err
	}
	e2e.Logf("Successfully executed cmd '%s' with output:\n%s", cmd, combinedOutput)
	return nil
}

// RunOutput runs cmd on the remote host and returns its combined standard output and standard error.
func (sshClient *SshClient) RunOutput(cmd string) (string, error) {
	config, err := sshClient.getConfig()
	if err != nil {
		return "", fmt.Errorf("failed to get SSH config: %v", err)
	}

	connection, err := ssh.Dial("tcp", fmt.Sprintf("%v:%v", sshClient.Host, sshClient.Port), config)
	if err != nil {
		return "", fmt.Errorf("failed to dial %s:%d: %v", sshClient.Host, sshClient.Port, err)
	}
	defer connection.Close()

	session, err := connection.NewSession()
	if err != nil {
		return "", fmt.Errorf("failed to create session: %v", err)
	}
	defer session.Close()

	combinedOutputBuffer := NewSynchronizedBuffer()
	session.Stdout = combinedOutputBuffer
	session.Stderr = combinedOutputBuffer

	err = session.Run(cmd)
	if err != nil {
		return "", fmt.Errorf("failed to run cmd '%s': %v\n%s", cmd, err, combinedOutputBuffer.String())
	}
	return combinedOutputBuffer.String(), nil
}

func GetPrivateKey() (string, error) {
	privateKey := os.Getenv("SSH_CLOUD_PRIV_KEY")
	if privateKey == "" {
		privateKey = filepath.Join("../internal/config/keys/", "openshift-qe.pem")
	}
	if _, err := os.Stat(privateKey); os.IsNotExist(err) {
		return "", fmt.Errorf("private key file not found: %s", privateKey)
	}
	return privateKey, nil
}

func GetPublicKey() (string, error) {
	publicKey := os.Getenv("SSH_CLOUD_PUB_KEY")
	if publicKey == "" {
		publicKey = filepath.Join("../internal/config/keys/", "openshift-qe.pub")
	}
	if _, err := os.Stat(publicKey); os.IsNotExist(err) {
		return "", fmt.Errorf("public key file not found: %s", publicKey)
	}
	return publicKey, nil
}