路漫漫其修远兮
吾将上下而求索

go学习:ssh远程执行命令

用到的库:golang.org/x/crypto/ssh(隔墙有代理https://goproxy.cn)

1. 发送指令执行 session.Run()

package main

import (
    "bytes"
    "fmt"
    "golang.org/x/crypto/ssh"
    "log"
)

func main() {
    // 建立SSH客户端连接
    client, err := ssh.Dial("tcp", "127.0.0.1:2222", &ssh.ClientConfig{
        User:            "root",
        Auth:            []ssh.AuthMethod{ssh.Password("123456")},
        HostKeyCallback: ssh.InsecureIgnoreHostKey(),
    })
    if err != nil {
        log.Fatalf("SSH dial error: %s", err.Error())
    }

    // 建立新会话
    session, err := client.NewSession()
    if err != nil {
        log.Fatalf("new session error: %s", err.Error())
    }

    defer session.Close()


    var b bytes.Buffer
    session.Stdout = &b
    if err := session.Run("ls"); err != nil {
        panic("Failed to run: " + err.Error())
    }
    fmt.Println(b.String())
}

2. 发送指令执行 session.Output()

 session.run(command)是直接在host执行命令,不关心执行结果。session.Output是将执行命令之后的Stdout返回

package main

import (
    "fmt"
    "golang.org/x/crypto/ssh"
    "log"
    "os"
)

func test() {

    // 建立SSH客户端连接
    client, err := ssh.Dial("tcp", "127.0.0.1:2222", &ssh.ClientConfig{
        User:            "root",
        Auth:            []ssh.AuthMethod{ssh.Password("123456")},
        HostKeyCallback: ssh.InsecureIgnoreHostKey(),
    })
    if err != nil {
        log.Fatalf("SSH dial error: %s", err.Error())
    }

    // 建立新会话
    session, err := client.NewSession()
    defer session.Close()
    if err != nil {
        log.Fatalf("new session error: %s", err.Error())
    }

    result, err := session.Output("/bin/bash /root/a.sh")
    if err != nil {
        fmt.Fprintf(os.Stdout, "Failed to run command, Err:%s", err.Error())
        os.Exit(0)
    }
    fmt.Println(string(result))
}

另外一台机器有脚本,测试可以执行,100s不断线,还行

cat a.sh 
while [[ $i -le 100 ]]; do
    sleep 1
    echo $i
    let i=i+1
done

3. 模拟交互terminal

package main

import (
    "golang.org/x/crypto/ssh"
    "log"
    "os"
)

func main() {
    // 建立SSH客户端连接
    client, err := ssh.Dial("tcp", "127.0.0.1:2222", &ssh.ClientConfig{
        User:            "root",
        Auth:            []ssh.AuthMethod{ssh.Password("123456")},
        HostKeyCallback: ssh.InsecureIgnoreHostKey(),
    })
    if err != nil {
        log.Fatalf("SSH dial error: %s", err.Error())
    }

    // 建立新会话
    session, err := client.NewSession()
    defer session.Close()
    if err != nil {
        log.Fatalf("new session error: %s", err.Error())
    }

    session.Stdout = os.Stdout // 会话输出关联到系统标准输出设备
    session.Stderr = os.Stderr // 会话错误输出关联到系统标准错误输出设备
    session.Stdin = os.Stdin   // 会话输入关联到系统标准输入设备
    modes := ssh.TerminalModes{
        ssh.ECHO:          0,  // 禁用回显(0禁用,1启动)
        ssh.TTY_OP_ISPEED: 14400, // input speed = 14.4kbaud
        ssh.TTY_OP_OSPEED: 14400, //output speed = 14.4kbaud
    }
    if err = session.RequestPty("linux", 32, 160, modes); err != nil {
        log.Fatalf("request pty error: %s", err.Error())
    }
    if err = session.Shell(); err != nil {
        log.Fatalf("start shell error: %s", err.Error())
    }
    if err = session.Wait(); err != nil {
        log.Fatalf("return error: %s", err.Error())
    }
}

禁用回显:

//如果不禁用回显
[root@65a9c031a770 ~]# ls
ls
anaconda-ks.cfg


//禁用回显
[root@65a9c031a770 ~]# ls
anaconda-ks.cfg

注意:

这里的ssh.InsecureIgnoreHostKey是不检查host key,需要检查的话得参考client源码重写函数

使用GO语言灵活批量ssh登录服务器执行操作: https://www.cnblogs.com/findumars/p/5930584.html

github一个非常好的web ssh项目: https://github.com/libragen/felix

未经允许不得转载:江哥架构师笔记 » go学习:ssh远程执行命令

分享到:更多 ()

评论 抢沙发

评论前必须登录!