[Golang]ssh登录及执行命令

1
package main
2

3
import (
4
  "fmt"
5
  "github.com/mitchellh/go-homedir"
6
  "golang.org/x/crypto/ssh"
7
  "io/ioutil"
8
  "log"
9
  "time"
10
)
11

12
func main() {
13
  re,err := sshL("192.168.199.1","root","rehttPassword","password","","uname -a",22)
14
  if err != nil{
15
  fmt.println(err)
16
  }
17
  fmt.println(string(re))
18
}
19

20
func sshL(sshHost,sshUser,sshPassword,sshType,sshKeyPath,command string,sshPort int) ([]byte,error) {
21

22
  //创建sshp登陆配置
23
  config := &ssh.ClientConfig{
24
    Timeout:         time.Second,//ssh 连接time out 时间一秒钟, 如果ssh验证错误 会在一秒内返回
25
    User:            sshUser,
26
    HostKeyCallback: ssh.InsecureIgnoreHostKey(), //这个可以， 但是不够安全
27
    //HostKeyCallback: hostKeyCallBackFunc(h.Host),
28
  }
29
  if sshType == "password" {
30
    config.Auth = []ssh.AuthMethod{ssh.Password(sshPassword)}
31
  } else {
32
    config.Auth = []ssh.AuthMethod{publicKeyAuthFunc(sshKeyPath)}
33
  }
34

35

36

37
  //dial 获取ssh client
38
  addr := fmt.Sprintf("%s:%d", sshHost, sshPort)
39
  sshClient, err := ssh.Dial("tcp", addr, config)
40
  if err != nil {
41
    log.Fatal("创建ssh client 失败",err)
42
    return nil,err
43
  }
44
  defer sshClient.Close()
45

46

47
  //创建ssh-session
48
  session, err := sshClient.NewSession()
49
  if err != nil {
50
    log.Fatal("创建ssh session 失败",err)
51
    return nil,err
52
  }
53
  defer session.Close()
54
  //执行远程命令
55
  combo,err := session.CombinedOutput(command)
56
  if err != nil {
57
    log.Fatal("远程执行cmd 失败",err)
58
    return nil,err
59
  }
60
  return combo,nil
61
}
62

63
func publicKeyAuthFunc(kPath string) ssh.AuthMethod {
64
  keyPath, err := homedir.Expand(kPath)
65
  if err != nil {
66
    log.Fatal("find key's home dir failed", err)
67
  }
68
  key, err := ioutil.ReadFile(keyPath)
69
  if err != nil {
70
    log.Fatal("ssh key file read failed", err)
71
  }
72
  // Create the Signer for this private key.
73
  signer, err := ssh.ParsePrivateKey(key)
74
  if err != nil {
75
    log.Fatal("ssh key signer failed", err)
76
  }
77
  return ssh.PublicKeys(signer)
78
}