Golang ssh並ping包

工做須要寫了個Go小程序,須要ssh到host上,並執行ping包,檢查host的網絡是否通。簡單版的以下:git

package main

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

func main() {
	// get ssh client
	addrPort := "10.42.10.11:22"
	client, err := ssh.Dial("tcp", addrPort, &ssh.ClientConfig{
		User:            "root",
		Auth:            []ssh.AuthMethod{ssh.Password("123456")},
		HostKeyCallback: ssh.InsecureIgnoreHostKey(),
	})
	if err != nil {
		fmt.Fprintf(os.Stdout, "Failed to get ssh client %v\n", err)
		os.Exit(0)
	}
	defer client.Close()

	// connect host
	session, err := client.NewSession()
	if err != nil {
		fmt.Fprintf(os.Stdout, "Failed to connect session \n")
		os.Exit(0)
	}
	defer session.Close()

	pingAccess := "ping -c 2 10.42.10.12 | grep '100% packet loss' | wc -l"
	result, err := session.Output(pingAccess)
	if err != nil || 49 == result[0] {
		fmt.Fprintf(os.Stdout, "Failed to ping result:%d, Err:%v \n", result[0], err)
		os.Exit(0)
	}
	//result is ASCII
	fmt.Fprintf(os.Stdout, "finish %d \n", result[0])
}

簡單的說明:github

1. 下載crypto源碼git clone https://github.com/golang/crypto.git,把crypto文件夾拷貝到vendor/golang.org/x/目錄下golang

2. 這裏的ssh.InsecureIgnoreHostKey是不檢查host key,須要檢查的話得參考client源碼重寫函數。小程序

3. session.run(command)是直接在host執行命令,不關心執行結果。session.Output是將執行命令以後的Stdout返回網絡

相關文章
相關標籤/搜索