当前位置：首页 > news >正文

go客户端ssh交换机

news 2025/7/20 18:56:51

连接交换机代码

使用库golang.org/x/crypto/ssh

func (s *SwitchConf) switchNewSession() error {config := &ssh.ClientConfig{User: s.User,Auth: []ssh.AuthMethod{ssh.Password(s.Password),},HostKeyCallback: ssh.InsecureIgnoreHostKey(),Timeout:         30 * time.Second, Config: ssh.Config{			Ciphers: []string{"aes128-ctr", "aes192-ctr", "aes256-ctr", "aes128-gcm@openssh.com", "aes256-gcm@openssh.com", "chacha20-poly1305@openssh.com"},}, }address := fmt.Sprintf("%s:22", s.Ip)client, err := ssh.Dial("tcp", address, config)if err != nil {log.Fatalf("Failed to dial: %v", err)return nil}go s.waitConnectClosed(client)session, err := client.NewSession()if err != nil {fmt.Println("Failed to create session:", err)return nil}s.Session = sessionmodes := ssh.TerminalModes{ssh.ECHO:          0,     // disable echoingssh.TTY_OP_ISPEED: 14400, // input speed = 14.4kbaudssh.TTY_OP_OSPEED: 14400, // output speed = 14.4kbaud}err := s.Session.RequestPty("xterm", 80, 80, modes)if err != nil {fmt.Println("创建requestpty出错", err)return err}// 使用 io.MultiWriter 同时写入 os.Stdout 和缓冲区//multiWriter := io.MultiWriter(os.Stdout, &buf)s.input, err = s.Session.StdinPipe()if err != nil {fmt.Println("StdinPipe错误: ", err)return err}s.output, err = s.Session.StdoutPipe()if err != nil {fmt.Println("StdoutPipe: ", err)return err}s.errput, err = s.Session.StderrPipe()if err != nil {fmt.Println("StderrPipe: ", err)return err}err = s.Session.Shell()if err != nil {fmt.Println("创建shell出错: ", err)return err}go s.getErrorRealTime()return nil
}

执行命令

// 如果不判断什么时候结束，结束后就会一直卡在output.Read

func (s *DeviceConf) getOutputRealTime(expect string, unExpect string) {lenBuf := 128 * 1024var n ints.outBuf = make([]byte, lenBuf)s.bufLen = 0for {if s.output == nil {return}lenr, err := s.output.Read(s.outBuf[n:])if err != nil {log.Errorf("read err: %s", err)s.bufLen = n - 1return}if lenr == 0 {log.Errorf("read len 0")s.bufLen = n - 1return}if strings.Contains(string(s.outBuf[n:n+lenr-1]), expect) || strings.Contains(string(s.outBuf[n:n+lenr-1]), unExpect) {			s.bufLen = n + lenr - 1return}n += lenr//留1024给下次读取if n >= lenBuf+1024 {log.Errorf("BUFF is too short")s.bufLen = n - 1return}}
}
func (s *DeviceConf) runCmd(cmd string, expectSub string, unExpect string) string {if s.Session == nil {err := s.switchNewSession()if err != nil {log.Errorf("switchNewSession error: %s", err)return ""}err = s.getShell()if err != nil {log.Errorf("getShell error: %s", err)return ""}}go s.getOutputRealTime(expectSub, unExpect)go s.getErrorRealTime()_, err := s.input.Write(nputil.Str2bytes("\n " + cmd + "\n"))if err != nil {log.Errorf("写入stdin出错111: %s", err)return ""}timeoutChan := time.After(defaultTimeout)for {select {case <-timeoutChan:log.Errorf("超时了，退出循环: %s", time.Now())return ""default:if s.bufLen != 0 {out := s.outBuf[:s.bufLen]s.bufLen = 0log.Errorf("default return: %s", time.Now())return string(out)}}}}