Go语言:crypto/ssh执行远程命令
2015 年 5 月 28 日
前言
远程执行命令有什么用？为什么要远程执行命令？ 如果你只有2，3台服务器需要管理的时候，远程执行命令确实没有没多大作用，你可以登录到每台服务器上去完成各种操作。 当你的服务器大于3台的时候，远程执行的命令的方式就可以大大提高你的生产力了。
如果你有一个可以远程执行命令的工具，那么就可以像操作单台机器那样操作多台机器，机器越多，效率提高的越多。 远程执行命令最常用的方法就是利用 SSH 协议，将命令发送到远程机器上执行，并获取返回结果。
代码
连接包含了认证，可以使用 password 或者 sshkey 2种方式来认证。下面的示例为了简单，使用了密码认证的方式来完成连接。
package main
import (
"fmt"
"github.com/mitchellh/go-homedir"
"golang.org/x/crypto/ssh"
"io/ioutil"
"log"
"time"
)
func main(){
sshHost := "home.xxx.cn"
sshUser := "x"
sshPassword := "xxxxxx"
sshType := "password"//password 或者 key
sshKeyPath := ""//ssh id_rsa.id 路径"
sshPort := 22
//创建sshp登陆配置
config := &ssh.ClientConfig{
Timeout: time.Second,//ssh 连接time out 时间一秒钟, 如果ssh验证错误 会在一秒内返回
User: sshUser,
HostKeyCallback: ssh.InsecureIgnoreHostKey(), //这个可以， 但是不够安全
//HostKeyCallback: hostKeyCallBackFunc(h.Host),
}
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;echo https://github.com/dejavuzhou/felix")
if err != nil {
log.Fatal("远程执行cmd 失败",err)
}
log.Println("命令输出:",string(combo))
}
func publicKeyAuthFunc(kPath string) ssh.AuthMethod {
keyPath, err := homedir.Expand(kPath)
if err != nil {
log.Fatal("find key's home dir failed", err)
}
key, err := ioutil.ReadFile(keyPath)
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)
}
代码详解
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
3 执行命令 CombinedOutput run …
打印结果
2019/05/21 18:39:22 命令输出: pi 总用量 91 drwxr-xr-x 23 root root 4096 5月 20 11:13 . drwxr-xr-x 23 root root 4096 5月 20 11:13 .. drwxr-xr-x 2 root root 4096 4月 8 17:51 bin drwxr-xr-x 6 root root 2560 1月 1 1970 boot drwxr-xr-x 14 root root 3280 5月 21 12:17 dev drwxr-xr-x 87 root root 4096 5月 17 09:57 etc drwxr-xr-x 4 root root 4096 5月 17 09:56 home drwxr-xr-x 16 root root 4096 4月 8 17:58 lib drwx------ 2 root root 16384 4月 8 18:24 lost+found drwxr-xr-x 2 root root 4096 4月 8 17:37 media drwxr-xr-x 2 root root 4096 4月 18 22:18 miwifi drwxr-xr-x 2 root root 4096 4月 8 17:37 mnt -rw-r--r-- 1 root root 2787 4月 19 10:42 nginx_default_site.conf drwxr-xr-x 3 root root 4096 4月 8 17:48 opt dr-xr-xr-x 139 root root 0 1月 1 1970 proc drwx------ 6 root root 4096 5月 20 11:12 root drwxr-xr-x 24 root root 760 5月 21 18:39 run drwxr-xr-x 2 root root 4096 4月 19 13:48 sbin drwxr-xr-x 2 root root 4096 4月 8 17:37 srv dr-xr-xr-x 12 root root 0 5月 21 18:25 sys drwxrwxrwt 8 root root 4096 5月 21 18:35 tmp drwxr-xr-x 10 root root 4096 4月 8 17:37 usr drwxr-xr-x 12 root root 4096 4月 19 10:10 var drwxrwxrwx 3 root root 4096 4月 19 10:35 www https://github.com/dejavuzhou/felix
高级用法
- 使用命令行快速远程执行cmd github.com/dejavuzhou/felix/ssh
- golang ssh 转websocket ssh2ws
- golang ssh 自动输入sudo密码 line 50
About The Author
maynard
懒散，不想无聊