golang ssh host 连接华为NE5k路由器并支持mo
2017-03-09 本文已影响0人
飞天神猫
- 利用golang/x/crypto/ssh实现并参考作者: CodyGuo代码
- ssh登录华为路由器执行命令,可用作单台设备执行命令打印输出之用
- 对终端屏宽产生的分页符做处理
- github地址
use case
go run ./testConfig.go --username="aaa" --passwd='aaa' --ip_port="192.168.6.87" --cmd='display version'
-bash-4.3$ go run ./testConfig.go -h
Usage of /tmp/go-build918643740/command-line-arguments/_obj/exe/testConfig:
-cmd string
cmdstring (default "display arp statistics all")
-ip_port string
ip and port (default "1.1.1.1:22")
-passwd string
password (default "aaa")
-username string
username (default "aaa")
exit status 2
重点code说明
...
in <- "display version" //执行输入goroutine
in <- "display arp statistics all"
...
go func() {
for cmd := range in {
wg.Add(1)
w.Write([]byte(cmd + "\n"))
wg.Wait() //控制每次goroutine执行一条命令
}
}()
go func() {
var (
buf [1024 * 1024]byte
t int
)
for {
n, err := r.Read(buf[t:])
if err != nil {
fmt.Println(err.Error())
close(in)
close(out)
return
}
t += n
result := string(buf[:t])
//循环处理设备分页符
if strings.Contains(string(buf[t-n:t]), "More") {
w.Write([]byte("\n"))
}
//匹配等待符完成一条操作清空goroutine
if strings.Contains(result, "username:") ||
strings.Contains(result, "password:") ||
strings.Contains(result, ">") {
out <- string(buf[:t])
t = 0
wg.Done()
}
}
}()
...
TestUnit
display version
Huawei Versatile Routing Platform Software
VRP (R) software, Version 5.90 (NE40E&80E V600R003C00SPCa00)
Copyright (C) 2000-2012 Huawei Technologies Co., Ltd.
HUAWEI NE80E uptime is 1695 days, 20 hours, 57 minutes
NE80E version information:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
BKP 1 version information:
PCB Version : CR52BKPA REV B
MPU Slot Quantity : 2
SRU Slot Quantity : 0
SFU Slot Quantity : 4
LPU Slot Quantity : 16
...
...
...
<HK-HK-CW-1>
display arp statistics all
Dynamic: 345 Static: 0
<HK-HK-CW-1>