Golang 實現 SSH 執行遠端命令

本篇文章會介紹使用 Golang 執行遠端命令，並獲取香瓜你執行結果的兩種方法，SSH 客戶端連接，和執行系統命令，最大的區別是，前者會建立 SSH 連接，之後執行相應的命令；後者是執行系統 ssh 命令， 把想要執行的操作當做 ssh 的參數。

在執行遠端命令相關的文章我在我的另一篇文章已經介紹了 Python, Shell, Perl 這三種方法，詳見：http://blog.csdn.net/u010649766/article/details/78465503

執行系統命令

這個方法需要有一個環境的準備：與目標服務器建立免密碼登陸，並且執行程序的用戶與執行用戶一致。

import (
    "net"
    "log"
    "fmt"
    "bytes"
    "os/exec"
    "strconv"
    str "strings"
    "golang.org/x/crypto/ssh"
)

func runCmd(){

    var stdOut, stdErr bytes.Buffer

    cmd := exec.Command( "ssh", "[email protected]", "if [ -d liujx/project ];then echo 0;else echo 1;fi" )
    cmd.Stdout = &stdOut
    cmd.Stderr = &stdErr
    if err := cmd.Run(); err != nil {
        fmt.Printf( "cmd exec failed: %s : %s", fmt.Sprint( err ), stdErr.String() )
    }

    fmt.Print( stdOut.String() )
    ret, err := strconv.Atoi( str.Replace( stdOut.String(), "\n", "", -1 )  )
    if err != nil {
        panic(err)
    }

    fmt.Printf("%d, %s\n", ret, stdErr.String() )
}

SSH 客戶端連接

這種方法可以不用搭建免密碼登陸環境，連接時可指定用戶和密碼的。

func SSHConnect( user, password, host string, port int ) ( *ssh.Session, error ) {
    var (
        auth         []ssh.AuthMethod
        addr         string
        clientConfig *ssh.ClientConfig
        client       *ssh.Client
        session      *ssh.Session
        err          error
    )
    // get auth method
    auth = make([]ssh.AuthMethod, 0)
    auth = append(auth, ssh.Password(password))

    hostKeyCallbk := func(hostname string, remote net.Addr, key ssh.PublicKey) error {
            return nil
    }

    clientConfig = &ssh.ClientConfig{
        User:               user,
        Auth:               auth,
        // Timeout:             30 * time.Second,
        HostKeyCallback:    hostKeyCallbk, 
    }

    // connet to ssh
    addr = fmt.Sprintf( "%s:%d", host, port )

    if client, err = ssh.Dial( "tcp", addr, clientConfig ); err != nil {
        return nil, err
    }

    // create session
    if session, err = client.NewSession(); err != nil {
        return nil, err
    }

    return session, nil
}

func runSsh(){

    var stdOut, stdErr bytes.Buffer

    session, err := SSHConnect( "username", "passworld", "192.168.1.4", 22 )
    if err != nil {
        log.Fatal(err)
    }
    defer session.Close()

    session.Stdout = &stdOut
    session.Stderr = &stdErr

    session.Run("if [ -d liujx/project ]; then echo 0; else echo 1; fi")
    ret, err := strconv.Atoi( str.Replace( stdOut.String(), "\n", "", -1 )  )
    if err != nil {
        panic(err)
    }

    fmt.Printf("%d, %s\n", ret, stdErr.String() )

}

ps: 由於 golang.org 被牆，所以直接用 go get golang.org/x/crypto/ssh 會報超時，所以如果我們可以從 GitHub 上獲取資源拷貝到這個目錄下， 具體操作如下：

mkdir -p $GOPATH/src/golang.org/x/
cd $GOPATH/src/golang.org/x/
git clone https://github.com/golang/crypto.git