Golang通过SSH执行交换机操作实现

 更新时间:2020年06月11日 11:36:24   作者:moakia  
这篇文章主要介绍了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", "aes128-gcm@openssh.com", "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
}

到此这篇关于Golang通过SSH执行交换机操作实现的文章就介绍到这了,更多相关Golang SSH执行交换机内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • golang API开发过程的中的自动重启方式(基于gin框架)

    golang API开发过程的中的自动重启方式(基于gin框架)

    这篇文章主要介绍了golang API开发过程的中的自动重启方式(基于gin框架),本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-12-12
  • golang如何实现mapreduce单进程版本详解

    golang如何实现mapreduce单进程版本详解

    这篇文章主要给大家介绍了关于golang如何实现mapreduce单进程版本的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧。
    2018-01-01
  • GO 语言学习指南

    GO 语言学习指南

    这篇文章主要介绍了GO 语言学习指南,介绍的相当全面,有需要的小伙伴参考下吧。
    2015-01-01
  • golang中使用proto3协议导致的空值字段不显示的问题处理方案

    golang中使用proto3协议导致的空值字段不显示的问题处理方案

    这篇文章主要介绍了golang中使用proto3协议导致的空值字段不显示的问题处理方案,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-10-10
  • Golang 实现分片读取http超大文件流和并发控制

    Golang 实现分片读取http超大文件流和并发控制

    这篇文章主要介绍了Golang 实现分片读取http超大文件流和并发控制,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-12-12
  • golang 中的 nil的场景分析

    golang 中的 nil的场景分析

    这篇文章主要介绍了golang 中的 nil,本文通过多种场景分析给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-03-03
  • Go语言基础学习教程

    Go语言基础学习教程

    这篇文章主要介绍了Go语言基础知识,包括基本语法、语句、数组等的定义与用法,需要的朋友可以参考下
    2016-07-07
  • Go语言实现字符串切片赋值的方法小结

    Go语言实现字符串切片赋值的方法小结

    这篇文章主要给大家介绍了Go语言实现字符串切片赋值的两种方法,分别是在for循环的range中以及在函数的参数传递中实现,有需要的朋友们可以根据自己的需要选择使用。下面来一起看看吧。
    2016-10-10
  • Golang科学计数法转换string数字输出的实现

    Golang科学计数法转换string数字输出的实现

    最近接手一个商城运单号模块,接手后发现有部分运单号返回给前端是按照科学计数法的方式返回,本文就介绍一下Golang科学计数法转换string数字输出,感兴趣的可以了解一下
    2021-07-07
  • golang中strconv.ParseInt函数用法示例

    golang中strconv.ParseInt函数用法示例

    这篇文章主要介绍了golang中strconv.ParseInt函数用法,实例分析了strconv.ParseInt函数将字符串转换为数字的简单使用方法,需要的朋友可以参考下
    2016-07-07

最新评论