Golang通過SSH執行交換機操作

Golang通過SSH執行交換機操作

​ 簡單實現通過輸入指令，兩步執行交換機命令。

• 輸入執行換機的賬號和密碼。可以一次輸入多個賬號和密碼，爲了方便操作，規定了輸入格式。如 用戶名；主機IP;密碼|用戶名；主機IP;密碼。舉例admin;192.168.56.10;h3csw1|admin;192.168.56.11;h3csw2
• 輸入要執行的命令，以;分割。例如system-view;dis cu;

​ 存在問題：

• 不夠靈活。輸入方式限制太死，輸入特別字符也可能存在錯誤。
• 過於簡陋。
• 功能簡單。

​ 不過我的目的已經達到，我主要是瞭解ssh的使用。

package main

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

//獲取賬號和密碼的對應關係
type HostPassword struct {
    Host string
    Username string
    Password string
}
var (
    a,b string  //臨時存儲變量
    commands = []string{}  //執行命令組
    hp []HostPassword  //保存賬號和密碼
    wg sync.WaitGroup    //執行goroutine

)
func main()  {
    //1. 選擇交換機
    //2. 輸入要執行命令
    //3. 建立會話連接
    //4. 新建session，並執行命令

    //1. 選擇操作交換機
    // 1.1 輸入要執行交換機
    fmt.Println("請輸入計劃執行命令的交換機賬號和密碼,賬號密碼直接使用|分割，多個賬號密碼之間使用;分割,例如admin;192.168.56.10;h3csw1|admin;192.168.56.11;h3csw2")
    _, err := fmt.Scanln(&a)
    if err != nil {
        log.Fatal("輸入錯誤:",err)
    }
    fmt.Println("請輸入要執行的命令行，以；號間隔")
    //1.1.1切割交換機命令
    switchgroups := strings.Split(a, "|")
    length := len(switchgroups)
    hp = make([]HostPassword,length)
    for i,singleswitch := range switchgroups{
        hp[i]=HostPassword{}
        switchsplit := strings.Split(singleswitch, ";")
        hp[i].Username=switchsplit[0]
        hp[i].Host=switchsplit[1]
        hp[i].Password=switchsplit[2]
    }

    // 1.2 輸入要執行命令
    input := bufio.NewReader(os.Stdin)
    b, err := input.ReadString('\n')
    if err != nil {
        log.Fatal("輸入錯誤",err)
    }
    commands = strings.Split(b, ";")

    //2. 執行交換機操作
    err = SshSwitch(hp)
    if err != nil {
        log.Fatalln(err)
    }

    // 同步等待
    wg.Wait()
}

//建立ssh連接
func SshSwitch(hostpasswords []HostPassword) (error){
    //循環獲取hostpasswords的賬號和密碼
    for i,_ := range hp{
        //添加同步組，下面會執行goroutin
        wg.Add(1)
        config := &ssh.ClientConfig{
            Config:            ssh.Config{
                Ciphers:        []string{"aes128-ctr", "aes192-ctr", "aes256-ctr", "[email protected]", "arcfour256", "arcfour128", "aes128-cbc", "3des-cbc", "aes192-cbc", "aes256-cbc"},
            }, //添加了很多加密方式，爲了應對不同的密碼規則
            User:              hp[i].Username,
            Auth:              []ssh.AuthMethod{
                ssh.Password(hp[i].Password),
            },
            HostKeyCallback:   ssh.InsecureIgnoreHostKey(), //此處相當於執行nil，但是並不安全
        }
        client, err := ssh.Dial("tcp",hp[i].Host+":22", config)
        if err != nil {
            log.Fatalln("建立ssh連接錯誤:",err)
            return err
        }
        //執行goroutine，但是沒有返回錯誤。
        go HandleSession(client, commands,&wg)

    }
    return  nil
}

//建立session，執行命令。
func HandleSession(client *ssh.Client,commands []string,wg *sync.WaitGroup) error {
    //建立session
    session, err := client.NewSession()
    if err != nil {
        log.Fatalln("創建session出錯",err)
        return  err
    }
    //延遲關閉session
    defer  session.Close()

    //設置terminalmodes的方式
    modes := ssh.TerminalModes{
        ssh.ECHO:          0,     // disable echoing
        ssh.TTY_OP_ISPEED: 14400, // input speed = 14.4kbaud
        ssh.TTY_OP_OSPEED: 14400, // output speed = 14.4kbaud
    }
    //建立僞終端
    err = session.RequestPty("xterm",80,40,modes)
    if err != nil {
        log.Fatal("創建requestpty出錯",err)
        return  err
    }
    //設置session的標準輸入是stdin
    stdin, err := session.StdinPipe()
    if err != nil {
        log.Fatal("輸入錯誤",err)
        return  err
    }
    //設置session的標準輸出和錯誤輸出分別是os.stdout,os,stderr.就是輸出到後臺
    session.Stdout = os.Stdout
    session.Stderr = os.Stderr
    err = session.Shell()
    if err != nil {
        log.Fatal("創建shell出錯",err)
        return  err
    }
    //將命令依次執行
    for _, cmd := range commands {
        fmt.Println(cmd)
        _, err = fmt.Fprintf(stdin, "%s\n", cmd)
        if err != nil {
            log.Fatal("寫入stdin出錯",err)
            return  err
        }
    }

    //執行等待
    err = session.Wait()
    if err != nil {
        log.Fatal("等待session出錯",err)
        return  err
    }
    //減少同步組的次數
    wg.Done()
    return  nil
}