酷站(www.ku0.com)-致力于为互联网从业者提供动力!

热门关键词:  企业  as  baidu  c4rp3nt3r  美女
【ECS精选特惠】新用户上云低至1折起
百度资源

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

来源:互联网搜集 作者:秩名 人气: 发布时间:2020-06-11
本篇文章主要介绍了Golang通过SSH执行交换机操作的实现,对大家的学习或者工作具有一定的参考学习价值,感兴趣的小伙伴们可以参考一下,也感谢大家对酷站(ku0.com)的支持。

简单实现通过输入指令,两步执行交换机命令。

  • 输入执行换机的账号和密码。可以一次输入多个账号和密码,为了方便操作,规定了输入格式。如 用户名;主机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
}

版权声明:本文内容来源于互联网或用户自行发布贡献,该文观点仅代表原作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 959677720#qq.cn(#换@) 举报,一经查实,本站将立刻删除。
原文链接:https://blog.51cto.com/9406836/2503197

相关文章

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

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

    简单实现通过输入指令,两步执行交换机命令。 输入执行换机的账号和密码。可以一次输入多个账号和密码,为了方便操作,规定了输入格式。如 用户名;主机IP;密码|用户名;主机IP;密码。举例admin;192.168.56.10;h3csw1|admin;192.168.56.1......
    06-11
  • golang使用json格式实现增删查改的方法

    golang使用json格式实现增删查改的方法

    需求和思路 在一般的小项目或者一个小软件,例如客户端之类的小程序中,可能会需要数据的持久化.但是使用一般的数据库(Mysql)之类的不合适.使用sqlite3这种嵌入式的是个较好的方法,但是Go语言中sqlite3的库是C语言的,Cgo不支持跨平台编译.......
    05-21
  • Golang Cron定时任务的实现方法

    Golang Cron定时任务的实现方法

    开门见山写一个 package mainimport ( fmt github.com/robfig/cron log strings time)func CronTask() { log.Println(******** ******* *******)}func CronTest() { log.Println(Starting Cron...) c := cron.New() c.AddFunc(* * * * * ......
    05-03
  • Windows下安装VScode并使用及中文配置方法

    Windows下安装VScode并使用及中文配置方法

    首先明确一点,VScode是开发Go应用的基础编辑器,是Microsoft(微软的产品),可以运行在Windows、Linux、Mac Os X上使用,默认提供Go语言语法高亮,安装Go语言插件后,就可以智能提示,编译等运行功能, 当然市场上还有很多其他GO语言的......
    03-29
  • go中的引用类型的介绍

    go中的引用类型的介绍

    值类型和引用类型 值类型:int、float、bool和string这些类型都属于值类型,使用这些类型的变量直接指向存在内存中的值,值类型的变量的值存储在栈中。当使用等号=将一个变量的值赋给另一个变量时,如 j = i ,实际上是在内存中将 i 的值......
    03-15
  • 详解用golang运行第一个项目

    详解用golang运行第一个项目

    安装一些必要的环境 1.下载go sdk (本人装的是1.9) 2.下载golang 3.下载git 因为有些依赖 要用 go get 去github上获取 4.配置gopath bin 生成的执行文件 pkg 编译的中间文件 src 程序的代码文件 里面有3个文件夹,然后把项目放在src里。......
    03-10
  • Go语言中确保Cookie数据的安全传输

    Go语言中确保Cookie数据的安全传输

    什么是Cookie Cookie(也叫Web Cookie或浏览器Cookie)是服务器发送到用户浏览器并保存在本地的一小块数据,它会在浏览器下次向同一服务器再发起请求时被携带并发送到服务器上。通常,它用于告知服务端两个请求是否来自同一浏览器,如保......
    03-08
  • 详解Go语言defer语句的三种机制

    详解Go语言defer语句的三种机制

    Golang 的 1.13 版本 与 1.14 版本对 defer 进行了两次优化,使得 defer 的性能开销在大部分场景下都得到大幅降低,其中到底经历了什么原理? 这是因为这两个版本对 defer 各加入了一项新的机制,使得 defer 语句在编译时,编译器会根据......
    03-01
  • Golang使用lua脚本实现redis原子

    Golang使用lua脚本实现redis原子

    目录 [redis 调用Lua脚本](#redis 调用Lua脚本) [redis+lua 实现评分排行榜实时更新](#redis+lua 实现评分排行榜实时更新) [lua 脚本](#lua 脚本) Golang调用redis+lua示例 byte切片与string的转换优化 redis 调用Lua脚本 EVAL命令 redis......
    03-01
  • 使用go module导入本地包的方法

    使用go module导入本地包的方法

    go module 是Go1.11版本之后官方推出的版本管理工具,并且从Go1.13 版本开始,go module 将是Go语言默认的依赖管理工具。到今天Go1.14 版本推出之后Go modules 功能已经被正式推荐在生产环境下使用了。 这几天已经有很多教程讲解如何使用......
    03-01

最新更新