Method sshRunCommand

main.go:83–140

Run command via SSH

(command string, env *env.Env)

// Run command via SSH
func (ssh *SSH) sshRunCommand(command string, env *env.Env) ([]byte, error) {
	// Get signer from private key
	signer, _ := sshClient.ParsePrivateKey(ssh.SSH_PRIVATE_KEY)

	// Convert timeout param
	timeoutSeconds, _ := strconv.Atoi(env.SSH_CONNECT_TIMEOUT)
	timeoutDuration := time.Duration(timeoutSeconds) * time.Second

	// SSH client config
	config := &sshClient.ClientConfig{
		User:            ssh.SSH_USER,
		HostKeyCallback: sshClient.InsecureIgnoreHostKey(), // StrictHostKeyChecking=no
		Timeout:         time.Duration(timeoutDuration),    // ConnectTimeout
		Auth: []sshClient.AuthMethod{
			sshClient.Password(env.SSH_PASSWORD),
		},
	}

	// SSH auth config
	if signer != nil {
		config.Auth = []sshClient.AuthMethod{
			sshClient.PublicKeys(signer),
			sshClient.Password(env.SSH_PASSWORD),
		}
	} else {
		log.Println("[WARN] Private key not found")
		if len(env.SSH_PASSWORD) == 0 {
			log.Println("[WARN] Password not set")
		}
	}

	// Establishing TCP connection
	client, err := sshClient.Dial("tcp", fmt.Sprintf("%s:%s", ssh.SSH_HOST, ssh.SSH_PORT), config)
	if err != nil {
		log.Println("[WARN] Failed establishing tcp connection")
		return nil, fmt.Errorf("Failed establishing tcp connection: %w", err)
	}
	defer client.Close()

	// Creating SSH session
	session, err := client.NewSession()
	if err != nil {
		log.Println("[WARN] Failed creating ssh session")
		return nil, fmt.Errorf("Failed creating ssh session: %w", err)
	}
	defer session.Close()

	// Run command
	var stdoutBuf bytes.Buffer
	var stderrBuf bytes.Buffer
	session.Stdout = &stdoutBuf
	session.Stderr = &stderrBuf
	err = session.Run(command)
	if len(stderrBuf.Bytes()) > 0 {
		return stderrBuf.Bytes(), fmt.Errorf("Execution error (error output is not empty)")
	}
	return stdoutBuf.Bytes(), err
}