溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

Golang如何通過(guò)SSH執(zhí)行交換機(jī)操作?

發(fā)布時(shí)間:2020-06-23 17:53:27 來(lái)源:億速云 閱讀:369 作者:清晨 欄目:編程語(yǔ)言

這篇文章將為大家詳細(xì)講解有關(guān)Golang如何通過(guò)SSH執(zhí)行交換機(jī)操作?,小編覺(jué)得挺實(shí)用的,因此分享給大家做個(gè)參考,希望大家閱讀完這篇文章后可以有所收獲。

簡(jiǎn)單實(shí)現(xiàn)通過(guò)輸入指令,兩步執(zhí)行交換機(jī)命令。

  • 輸入執(zhí)行換機(jī)的賬號(hào)和密碼??梢砸淮屋斎攵鄠€(gè)賬號(hào)和密碼,為了方便操作,規(guī)定了輸入格式。如 用戶名;主機(jī)IP;密碼|用戶名;主機(jī)IP;密碼。舉例admin;192.168.56.10;h4csw1|admin;192.168.56.11;h4csw2
  • 輸入要執(zhí)行的命令,以;分割。例如system-view;dis cu;

​ 存在問(wèn)題:

  • 不夠靈活。輸入方式限制太死,輸入特別字符也可能存在錯(cuò)誤。
  • 過(guò)于簡(jiǎn)陋。
  • 功能簡(jiǎn)單。

​ 不過(guò)我的目的已經(jīng)達(dá)到,我主要是了解ssh的使用。

package main

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

//獲取賬號(hào)和密碼的對(duì)應(yīng)關(guān)系
type HostPassword struct {
  Host string
  Username string
  Password string
}
var (
  a,b string //臨時(shí)存儲(chǔ)變量
  commands = []string{} //執(zhí)行命令組
  hp []HostPassword //保存賬號(hào)和密碼
  wg sync.WaitGroup  //執(zhí)行g(shù)oroutine

)
func main() {
  //1. 選擇交換機(jī)
  //2. 輸入要執(zhí)行命令
  //3. 建立會(huì)話連接
  //4. 新建session,并執(zhí)行命令

  //1. 選擇操作交換機(jī)
  // 1.1 輸入要執(zhí)行交換機(jī)
  fmt.Println("請(qǐng)輸入計(jì)劃執(zhí)行命令的交換機(jī)賬號(hào)和密碼,賬號(hào)密碼直接使用|分割,多個(gè)賬號(hào)密碼之間使用;分割,例如admin;192.168.56.10;h4csw1|admin;192.168.56.11;h4csw2")
  _, err := fmt.Scanln(&a)
  if err != nil {
    log.Fatal("輸入錯(cuò)誤:",err)
  }
  fmt.Println("請(qǐng)輸入要執(zhí)行的命令行,以;號(hào)間隔")
  //1.1.1切割交換機(jī)命令
  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 輸入要執(zhí)行命令
  input := bufio.NewReader(os.Stdin)
  b, err := input.ReadString('\n')
  if err != nil {
    log.Fatal("輸入錯(cuò)誤",err)
  }
  commands = strings.Split(b, ";")

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

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

//建立ssh連接
func SshSwitch(hostpasswords []HostPassword) (error){
  //循環(huán)獲取hostpasswords的賬號(hào)和密碼
  for i,_ := range hp{
    //添加同步組,下面會(huì)執(zhí)行g(shù)oroutin
    wg.Add(1)
    config := &ssh.ClientConfig{
      Config:      ssh.Config{
        Ciphers:    []string{"aes128-ctr", "aes192-ctr", "aes256-ctr", "aes128-gcm@openssh.com", "arcfour256", "arcfour128", "aes128-cbc", "3des-cbc", "aes192-cbc", "aes256-cbc"},
      }, //添加了很多加密方式,為了應(yīng)對(duì)不同的密碼規(guī)則
      User:       hp[i].Username,
      Auth:       []ssh.AuthMethod{
        ssh.Password(hp[i].Password),
      },
      HostKeyCallback:  ssh.InsecureIgnoreHostKey(), //此處相當(dāng)于執(zhí)行nil,但是并不安全
    }
    client, err := ssh.Dial("tcp",hp[i].Host+":22", config)
    if err != nil {
      log.Fatalln("建立ssh連接錯(cuò)誤:",err)
      return err
    }
    //執(zhí)行g(shù)oroutine,但是沒(méi)有返回錯(cuò)誤。
    go HandleSession(client, commands,&wg)

  }
  return nil
}

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

  //設(shè)置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("創(chuàng)建requestpty出錯(cuò)",err)
    return err
  }
  //設(shè)置session的標(biāo)準(zhǔn)輸入是stdin
  stdin, err := session.StdinPipe()
  if err != nil {
    log.Fatal("輸入錯(cuò)誤",err)
    return err
  }
  //設(shè)置session的標(biāo)準(zhǔn)輸出和錯(cuò)誤輸出分別是os.stdout,os,stderr.就是輸出到后臺(tái)
  session.Stdout = os.Stdout
  session.Stderr = os.Stderr
  err = session.Shell()
  if err != nil {
    log.Fatal("創(chuàng)建shell出錯(cuò)",err)
    return err
  }
  //將命令依次執(zhí)行
  for _, cmd := range commands {
    fmt.Println(cmd)
    _, err = fmt.Fprintf(stdin, "%s\n", cmd)
    if err != nil {
      log.Fatal("寫入stdin出錯(cuò)",err)
      return err
    }
  }

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

關(guān)于Golang如何通過(guò)SSH執(zhí)行交換機(jī)操作?就分享到這里了,希望以上內(nèi)容可以對(duì)大家有一定的幫助,可以學(xué)到更多知識(shí)。如果覺(jué)得文章不錯(cuò),可以把它分享出去讓更多的人看到。

向AI問(wèn)一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI