
本文探讨了使用 go 语言 ssh 库与 cisco 设备交互时,发送长命令可能遇到的截断问题。通过分析问题根源,发现是由于 pty (pseudo-terminal) 尺寸配置不当导致。教程将详细指导如何正确设置 pty 宽度,以确保长命令完整传输,从而实现稳定可靠的网络设备自动化管理。
在自动化管理网络设备,特别是像 Cisco ASR 这样的路由器时,通过 SSH 建立连接并执行 show 命令是常见的操作。由于大多数网络设备不支持 SSH exec 命令直接执行,我们通常需要模拟一个交互式终端会话(shell),通过向其标准输入写入命令,并从标准输出读取响应。Go 语言的 x/crypto/ssh 库提供了强大的功能来实现这一目标。
一个典型的 Go SSH 会话建立流程如下:
package main
import (
"fmt"
"io"
"log"
"strings"
"time"
"golang.org/x/crypto/ssh"
)
// SSH配置,实际应用中应从安全方式获取
var config = &ssh.ClientConfig{
User: "your_username",
Auth: []ssh.AuthMethod{
ssh.Password("your_password"),
},
HostKeyCallback: ssh.InsecureIgnoreHostKey(), // 生产环境请使用ssh.FixedHostKey或ssh.KnownHosts
Timeout: 10 * time.Second,
}
func main() {
// 1. 建立SSH连接
client, err := ssh.Dial("tcp", "172.16.32.95:22", config)
if err != nil {
log.Fatalf("无法连接SSH服务器: %v", err)
}
defer client.Close()
// 2. 创建新的会话
session, err := client.NewSession()
if err != nil {
log.Fatalf("无法创建SSH会话: %v", err)
}
defer session.Close()
// 3. 获取标准输入和标准输出管道
sshOut, err := session.StdoutPipe()
if err != nil {
log.Fatalf("无法获取StdoutPipe: %v", err)
}
sshIn, err := session.StdinPipe()
if err != nil {
log.Fatalf("无法获取StdinPipe: %v", err)
}
// 4. 请求一个伪终端 (PTY)
// 注意:这里的Pty尺寸是解决问题的关键
modes := ssh.TerminalModes{
ssh.ECHO: 0, // 禁用回显
ssh.TTY_OP_ISPEED: 14400, // 输入速度
ssh.TTY_OP_OSPEED: 14400, // 输出速度
}
// 初始的错误配置可能类似这样:session.RequestPty("vt100", 80, 40, modes)
// 正确的配置将在后续章节详细说明
err = session.RequestPty("vt100", 0, 200, modes) // 示例中正确的配置
if err != nil {
log.Fatalf("无法请求PTY: %v", err)
}
// 5. 启动Shell
if err = session.Shell(); err != nil {
log.Fatalf("无法启动Shell: %v", err)
}
// 6. 读取初始欢迎信息和等待命令提示符
readUntilPrompt(sshOut, "[local]ewag# ") // 假设提示符是 "[local]ewag#"
// 7. 发送命令并读取响应 (示例:发送短命令)
fmt.Println("发送短命令: show clock")
if _, err := sshIn.Write([]byte("show clock\r")); err != nil {
log.Fatalf("无法写入命令: %v", err)
}
readUntilPrompt(sshOut, "[local]ewag# ")
// 8. 尝试发送长命令 (这里会复现问题,如果PTY宽度设置不当)
fmt.Println("\n发送长命令: show session progress ipsg-service ipsg-gprs-svc")
longCommand := "show session progress ipsg-service ipsg-gprs-svc\r"
if _, err := sshIn.Write([]byte(longCommand)); err != nil {
log.Fatalf("无法写入长命令: %v", err)
}
readUntilPrompt(sshOut, "[local]ewag# ")
fmt.Println("\n教程执行完毕。")
}
// readUntilPrompt 辅助函数,用于读取SSH输出直到遇到指定的命令提示符
func readUntilPrompt(r io.Reader, prompt string) string {
buf := make([]byte, 1024)
var sb strings.Builder
for {
n, err := r.Read(buf)
if err != nil {
if err == io.EOF {
break
}
log.Printf("读取SSH输出错误: %v", err)
break
}
chunk := string(buf[:n])
sb.WriteString(chunk)
fmt.Print(chunk) // 实时打印输出,便于观察
if strings.Contains(sb.String(), prompt) {
break
}
// 避免无限循环,如果长时间未收到提示符
if sb.Len() > 4096 { // 限制缓冲大小
log.Println("输出过长,可能未找到提示符或陷入循环。")
break
}
}
return sb.String()
}在上述基础流程中,当尝试发送一个较短的命令(例如 show clock)时,通常不会出现问题,命令能够完整地发送并获得正确的响应。然而,当发送一个较长的命令,如 show session progress ipsg-service ipsg-gprs-svc 时,可能会观察到命令在传输过程中被意外截断。
例如,输出可能显示为:
[local]ewag# show session progress ipsg -service ipsg-gprs-svc Unknown command - "ipsg-service", unrecognized keyword [local]ewag#
这表明路由器只接收到了 show session progress ipsg,而命令的其余部分 -service ipsg-gprs-svc 则被视为新的一行或错误输入。这导致命令执行失败,并返回“未知命令”或“无法识别的关键字”错误。
问题的根源在于伪终端(PTY)的宽度配置。当我们通过 session.RequestPty 请求一个终端时,需要指定终端的类型、高度和宽度。如果指定的宽度过小,当发送的命令字符串长度超过这个宽度时,SSH 客户端或服务器可能会将其视为多行输入,从而导致命令被分割。对于网络设备而言,这通常意味着命令无法被正确识别和执行。
解决长命令截断问题的关键在于为 session.RequestPty 函数提供一个足够大的宽度参数。RequestPty 函数的签名通常为 func (s *Session) RequestPty(term string, h, w int, modes TerminalModes) error,其中 h 代表终端高度,w 代表终端宽度。
原始问题中,可能存在如下错误的 PTY 配置:
// 错误的Pty配置示例:宽度过小
session.RequestPty("vt100", 80, 40, modes) // 这里将宽度(w)设为40当宽度被设置为 40 时,任何超过 40 个字符的命令都可能被截断。
正确的做法是确保 PTY 宽度足够大,以容纳最长的命令。如果对具体的宽度没有严格要求,或者希望尽可能避免截断,可以设置一个较大的宽度值。例如,将宽度设置为 200 或更大,同时将高度设置为 0(表示不关心具体高度,或由终端自动调整)。
// 正确的Pty配置示例:提供足够大的宽度
err = session.RequestPty("vt100", 0, 200, modes) // 高度为0,宽度为200
if err != nil {
log.Fatalf("无法请求PTY: %v", err)
}通过将宽度设置为 200,我们为命令提供了充足的空间,确保其在传输过程中不会被意外分割。vt100 是一种常见的、兼容性良好的终端类型,适合大多数网络设备。
通过 Go 语言的 x/crypto/ssh 库与 Cisco 等网络设备进行交互时,正确配置伪终端(PTY)的尺寸至关重要。特别是 PTY 的宽度,它直接影响长命令能否完整传输。当遇到长命令被截断并导致执行失败的问题时,检查并增大 session.RequestPty 函数中的宽度参数,通常能有效解决此问题。遵循上述最佳实践,可以构建出更健壮、更安全的网络设备自动化管理工具。
以上就是使用 Go SSH 库与 Cisco 设备交互:解决长命令截断问题的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号