
本文详细阐述了使用go语言ssh库连接cisco设备时,发送长命令可能出现的截断问题及其解决方案。通过分析go ssh `requestpty` 函数的参数配置,揭示了终端宽度设置不当是导致命令被错误分割的根本原因,并提供了正确的pty尺寸配置方法,确保长命令能够完整无误地发送和执行。
使用Go SSH库连接Cisco设备
在使用Go语言进行网络设备自动化管理时,SSH是与Cisco等设备交互的常用协议。通常,我们需要通过SSH建立连接,然后模拟终端会话,发送show命令来获取设备状态或配置信息。Go的golang.org/x/crypto/ssh库提供了强大的功能来实现这一点。
一个典型的SSH连接及会话建立流程如下:
配置SSH客户端: 设置用户名、认证方式(密码或密钥)、主机密钥回调等。建立连接: 使用ssh.Dial函数连接到目标设备。创建会话: 从客户端创建新的SSH会话。获取I/O管道: 获取会话的标准输入(StdinPipe)和标准输出(StdoutPipe),用于发送命令和接收响应。请求伪终端(PTY): 这是模拟交互式终端的关键步骤,它定义了终端类型、尺寸等。启动Shell: 在会话中启动一个远程Shell。
以下是Go语言实现这些步骤的基本代码结构:
package mainimport ( "fmt" "io" "log" "strings" "time" "golang.org/x/crypto/ssh")// readUntilPrompt 从SSH输出中读取数据直到遇到指定提示符func readUntilPrompt(reader io.Reader, prompt string) (string, error) { var output strings.Builder buf := make([]byte, 1024) for { n, err := reader.Read(buf) if err != nil { return "", fmt.Errorf("error reading from sshOut: %v", err) } output.Write(buf[:n]) if strings.Contains(output.String(), prompt) { break } } return output.String(), nil}func main() { // 1. SSH配置 config := &ssh.ClientConfig{ User: "your_username", Auth: []ssh.AuthMethod{ ssh.Password("your_password"), }, HostKeyCallback: ssh.InsecureHostKeyCallback, // 生产环境请使用更安全的HostKeyCallback Timeout: 5 * time.Second, } // 2. 建立SSH连接 client, err := ssh.Dial("tcp", "172.16.32.95:22", config) if err != nil { log.Fatalf("Failed to dial: %v", err) } defer client.Close() // 3. 创建新的会话 session, err := client.NewSession() if err != nil { log.Fatalf("Failed to create session: %v", err) } defer session.Close() // 4. 获取标准输入输出管道 sshOut, err := session.StdoutPipe() if err != nil { log.Fatalf("Failed to get stdout pipe: %v", err) } sshIn, err := session.StdinPipe() if err != nil { log.Fatalf("Failed to get stdin pipe: %v", err) } // 5. 请求伪终端 (PTY) - 初始配置(可能导致问题) modes := ssh.TerminalModes{ ssh.ECHO: 0, // 禁用回显 ssh.TTY_OP_ISPEED: 14400, // input speed = 14.4kbaud ssh.TTY_OP_OSPEED: 14400, // output speed = 14.4kbaud } // 注意:这里的 width 参数可能过小 if err := session.RequestPty("xterm", 80, 40, modes); err != nil { // height=80, width=40 log.Fatalf("Request for pseudo terminal failed: %v", err) } // 6. 启动Shell if err := session.Shell(); err != nil { log.Fatalf("Failed to start shell: %v", err) } // 读取初始欢迎信息和提示符 initialOutput, err := readUntilPrompt(sshOut, "[local]ewag#") if err != nil { log.Fatalf("Failed to read initial prompt: %v", err) } fmt.Println("--- Initial Output ---") fmt.Println(initialOutput) // 发送短命令示例 commandShort := "show clockr" fmt.Printf("--- Sending command: %s ---n", strings.TrimSpace(commandShort)) if _, err := sshIn.Write([]byte(commandShort)); err != nil { log.Fatalf("Failed to write command: %v", err) } responseShort, err := readUntilPrompt(sshOut, "[local]ewag#") if err != nil { log.Fatalf("Failed to read short command response: %v", err) } fmt.Println(responseShort) // ... 更多操作 ...}
在上述代码中,session.RequestPty(“xterm”, 80, 40, modes)这行代码是关键,它请求一个伪终端,并指定了终端类型为xterm,高度为80行,宽度为40列。
长命令截断问题
当使用上述配置发送较短的命令(例如show clock)时,通常不会出现问题,命令能够被完整执行并返回结果。然而,当尝试发送一个较长的命令时,例如show session progress ipsg-service ipsg-gprs-svc,可能会观察到命令在远程设备上被意外截断,导致设备识别为未知命令或语法错误。
以下是截断问题的典型输出示例:
Last login: Tue Jan 14 18:09:19 -0500 2014 on pts/2 from 10.7.7.14.Cisco Systems SSI[local]ewag# show session progress ipsg-service ipsg-gprs-svcUnknown command - "ipsg-service", unrecognized keyword[local]ewag#
从输出中可以看出,发送的命令show session progress ipsg-service ipsg-gprs-svc在ipsg之后被分割成了两部分。这使得Cisco设备无法正确识别完整的命令,从而报错。
问题根源:PTY宽度配置不当
这个问题的根本原因在于session.RequestPty函数中伪终端的宽度(width)参数设置过小。session.RequestPty函数的签名是RequestPty(term string, h, w int, termmodes TerminalModes),其中h代表终端高度,w代表终端宽度。
当我们将终端宽度设置为一个较小的值(例如40列)时,Go SSH库在模拟终端行为时,可能会在客户端侧对发送的命令进行“换行”处理,以适应这个较小的终端宽度。这意味着,即使我们通过sshIn.Write发送了一个完整的长字符串,SSH客户端在将其传输到远程服务器之前,可能已经根据PTY的宽度限制将其分割成了多行。远程Cisco设备接收到的就是这些被分割的命令片段,而不是一个完整的命令。
解决方案:优化PTY宽度配置
解决这个问题的关键是确保为伪终端配置一个足够大的宽度,或者让远程服务器决定其宽度。
根据实际测试,将PTY宽度设置为一个较大的值(例如200),同时可以将高度设置为0(让服务器决定或使用默认值),可以有效解决命令截断问题。
将原有的Pty请求:
if err := session.RequestPty("xterm", 80, 40, modes); err != nil { // height=80, width=40 log.Fatalf("Request for pseudo terminal failed: %v", err)}
修改为:
// 正确配置:增加宽度,将高度设置为0if err := session.RequestPty("vt100", 0, 200, modes); err != nil { // height=0, width=200 log.Fatalf("Request for pseudo terminal failed: %v", err)}
修改后的代码片段:
// ... (之前的连接和会话建立代码) ...// 重新配置并启动会话以应用新的PTY配置// 注意:PTY只能请求一次,如果要在同一会话中更改,需要重新创建会话或连接// 假设这里是为了演示,我们重新创建了一个会话sessionFixed, err := client.NewSession() // 使用同一个clientif err != nil { log.Fatalf("Failed to create new session for fix: %v", err)}defer sessionFixed.Close()sshOutFixed, err := sessionFixed.StdoutPipe()if err != nil { log.Fatalf("Failed to get stdout pipe for fix: %v", err)}sshInFixed, err := sessionFixed.StdinPipe()if err != nil { log.Fatalf("Failed to get stdin pipe for fix: %v", err)}// 优化
以上就是Go SSH与Cisco设备通信:Pty配置优化解决长命令发送问题的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1425309.html
微信扫一扫
支付宝扫一扫