golang第三方库-远程ssh

2021-05-15 约 1832 字预计阅读 4 分钟次阅读

安装

设置goproxy环境变量为 set GOPROXY=https://goproxy.io

1
2


go mod init webssh
go get -u golang.org/x/crypto/ssh

示例

连接包含了认证，可以使用password或者sshkey 两种方式来认证

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67


package main

import (
   "fmt"
   "golang.org/x/crypto/ssh"
   "io/ioutil"
   "log"
   "time"
)

func publicKeyAuthFunc(kPath string)ssh.AuthMethod{
   key,err := ioutil.ReadFile(kPath)
   if err != nil {
      log.Fatal("ssh key file read failed", err)
   }
   // Create the Signer for this private key.
   signer, err := ssh.ParsePrivateKey(key)
   if err != nil {
      log.Fatal("ssh key signer failed", err)
   }
   return ssh.PublicKeys(signer)
}

func main()  {
   //可以使用 password 或者 sshkey 2种方式来认证。
   sshHost := "192.168.1.58" // 主机名
   sshUser := "root"     //用户名
   sshPassword := "120110" //密码
   sshType := "password"   //ssh认证类型
   sshKeyPath := ""        //ssh id_rsa.id路径
   sshPort := 22

   //创建ssh登陆配置
   config := &ssh.ClientConfig{
      Timeout: time.Second, //ssh 连接timeout时间一秒钟，如果ssh验证错误 会在1秒内返回
      User: sshUser, //指定ssh连接用户
      HostKeyCallback: ssh.InsecureIgnoreHostKey(), //这个可以，但是不够安全
   }

   if sshType == "password"{
      config.Auth = []ssh.AuthMethod{ssh.Password(sshPassword)}
   }else {
      config.Auth = []ssh.AuthMethod{publicKeyAuthFunc(sshKeyPath)}
   }

   //dial获取ssh Client
   addr := fmt.Sprintf("%s:%d",sshHost,sshPort)
   sshClient,err := ssh.Dial("tcp",addr,config)
   if err != nil {
      log.Fatal("创建ssh client 失败",err)
   }
   defer sshClient.Close()

   //创建ssh-session
   session,err := sshClient.NewSession()
   if err != nil{
      log.Fatal("创建ssh session 失败",err)
   }
   defer  session.Close()

   //执行远程命令
   combo,err := session.CombinedOutput("whoami; cd /; ls -al")
   if err != nil{
      log.Fatal("远程执行cmd 失败",err)
   }
   log.Println("命令输出:",string(combo))
}

代码详解

1 配置ssh.ClientConfig

建议Timeout自定义一个比较短的时间
自定义HostKeyCallback 如果想简便使用就使用 ssh.InsecureIgnoreHostKey回调，这种方式不是很安全
publicKeyAuthFunc 如果使用key登陆，就需要用这个函数来读取id_rsa私钥，当然你可以自定义这个访问让他支持字符串

2 ssh.Dial创建ssh客户端

拼接字符串得到ssh连接地址,同时不要忘记 defer client.Close()

3 sshClient.NewSession 创建session会话

可以自定义stdin,stdout
可以创建pty
可以SetEnv

执行命令的方式

CombinedOutput

1

func (s *Session) CombinedOutput(cmd string) ([]byte, error)

在远程主机上运行CMD命令，标准输出和标准错误都通过[]byte返回，否则就返回 nil, errors.New("ssh: Stdout already set")

每个session下只能调用一次

Run

1

func (s *Session) Run(cmd string) error

在远程主机上运行CMD命令，不关心命令执行结果