ssh.go

/*
 * @Description:
 * @Author: gphper
 * @Date: 2021-11-07 10:43:45
 */
package common

import (
	"log"
	"net"
	"time"

	"github.com/pkg/errors"
	"golang.org/x/crypto/ssh"
)

func GetSSHClient(user, pass, addr string) (*ssh.Client, error) {
	config := &ssh.ClientConfig{
		User: user,
		Auth: []ssh.AuthMethod{
			ssh.Password(pass),
		},
		//需要验证服务端，不做验证返回nil，没有该参数会报错
		HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
			return nil
		},
		Timeout: 10 * time.Second,
	}

	sshConn, err := net.Dial("tcp", addr)
	if nil != err {
		log.Printf("%+v", errors.WithStack(err))
		return nil, err
	}

	clientConn, chans, reqs, err := ssh.NewClientConn(sshConn, addr, config)
	if nil != err {
		sshConn.Close()
		log.Printf("%+v", errors.WithStack(err))
		return nil, err
	}

	client := ssh.NewClient(clientConn, chans, reqs)

	return client, nil
}