Golang 執行遠程命令

概述

遠程執行命令有什麼用？為什麼要遠程執行命令？ 如果你只有2，3台服務器需要管理的時候，遠程執行命令確實沒有沒多大作用，你可以登錄到每台服務器上去完成各種操作。 當你的服務器大於3台的時候，遠程執行的命令的方式就可以大大提高你的生產力了。

如果你有一個可以遠程執行命令的工具，那麼就可以像操作單台機器那樣操作多台機器，機器越多，效率提高的越多。 遠程執行命令最常用的方法就是利用 SSH 協議，將命令發送到遠程機器上執行，並獲取返回結果。

本文介紹如何使用 golang 實現遠程執行命令。

一般命令

所謂一般命令，就是在一定時間內會執行完的命令。比如 grep, cat 等等。 執行命令的步驟是：連接，執行，獲取結果

連接

連接包含了認證，可以使用 password 或者 sshkey 2種方式來認證。下面的示例為了簡單，使用了密碼認證的方式來完成連接。

import (  
  "fmt"
  "time"

  "golang.org/x/crypto/ssh"
)

func connect(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))

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

  // 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
}

連接的方法很簡單，只要提供登錄主機的 用戶*， *密碼*， *主機名或者IP*， *SSH端口

執行，命令獲取結果

連接成功後，執行命令很簡單

import (  
  "fmt"
  "log"
  "os"
  "time"

  "golang.org/x/crypto/ssh"
)

func main() {  
  session, err := connect("root", "xxxxx", "127.0.0.1", 22)
  if err != nil {
    log.Fatal(err)
  }
  defer session.Close()

  session.Run("ls /; ls /abc")
}

上面代碼運行之後，雖然命令正常執行了，但是沒有正常輸出的結果，也沒有異常輸出的結果。 要想顯示結果，需要將 session 的 Stdout 和 Stderr 重定向 修改 func main 為如下：

func main() {  
  session, err := connect("root", "xxxxx", "127.0.0.1", 22)
  if err != nil {
    log.Fatal(err)
  }
  defer session.Close()

  session.Stdout = os.Stdout
  session.Stderr = os.Stderr
  session.Run("ls /; ls /abc")
}

這樣就能在屏幕上顯示正常，異常的信息了。

交互式命令

上面的方式無法遠程執行交互式命令，比如 top ， 遠程編輯一個文件，比如 vi /etc/nginx/nginx.conf 如果要支持交互式的命令，需要當前的terminal來接管遠程的 PTY。

func main() {  
  session, err := connect("root", "olordjesus", "dockers.iotalabs.io", 2210)
  if err != nil {
    log.Fatal(err)
  }
  defer session.Close()

  fd := int(os.Stdin.Fd())
  oldState, err := terminal.MakeRaw(fd)
  if err != nil {
    panic(err)
  }
  defer terminal.Restore(fd, oldState)

  // excute command
  session.Stdout = os.Stdout
  session.Stderr = os.Stderr
  session.Stdin = os.Stdin

  termWidth, termHeight, err := terminal.GetSize(fd)
  if err != nil {
    panic(err)
  }

  // Set up terminal modes
  modes := ssh.TerminalModes{
    ssh.ECHO:          1,     // enable echoing
    ssh.TTY_OP_ISPEED: 14400, // input speed = 14.4kbaud
    ssh.TTY_OP_OSPEED: 14400, // output speed = 14.4kbaud
  }

  // Request pseudo terminal
  if err := session.RequestPty("xterm-256color", termHeight, termWidth, modes); err != nil {
    log.Fatal(err)
  }

  session.Run("top")
}

這樣就可以執行交互式命令了，比如上面的 top 也可以通過 vi /etc/nginx/nignx.conf 之類的命令來遠程編輯文件。