Golang实现SSH、SFTP操作小结

更新时间：2024年04月02日 11:23:31 作者：畱䶒

在日常的一些开发场景中,我们需要去和远程服务器进行一些通信,本文主要介绍了Golang实现SSH、SFTP操作小结,具有一定的参考价值,感兴趣的可以了解一下

1 前言

在日常的一些开发场景中，我们需要去和远程服务器进行一些通信，执行一些相关命令操作，这个时候我们就可以使用SSH协议实现目标。SSH协议是建立在应用层上的安全协议，全称为Secure Shell，采用的是面向连接的TCP协议进行传输，也就意味着它是安全可靠的。需要注意的是文件传输并不能在SSH协议上完成，需要在下面提到的SFTP协议完成。

2 Go实现

Go官方为我们提供了用于实现SSH连接的package，位于golang.org/x/crypto下，通过在程序中调用包中提供的相关方法，便可以实现与其他机器进行通信。使用前我们需要使用go get导入相关的依赖包。

go get golang.org/x/crypto/ssh

2.1 配置相关参数

在进行通信之前，我们还需要配置一些用于配置一些用于建立连接的相关参数。ssh包下的ClientConfig结构体中，定义了建立SSH连接需要用到的一些配置项，部分项提供了默认参数，我们使用时可以不进行声明。

下面的代码段中，我们首先是声明了用户名和密码，连接超时时间设置为10秒钟，addr变量定义了目标机器的IP地址以及端口。

HostKeyCallback项，我们设置了忽略，这是因为SSH协议为客户端提供了两种安全验证方式，一种是基于口令的安全验证，也就是我们常常使用的账号密码形式，另外一种则是基于密钥的安全验证，相较于第一种，这种形式的校验方法极大的提升了安全等级，缺点则是时间损耗相对较长。

如果需要使用这种方式进行校验，首先我们需要在服务器上为自己创建一对密钥，作为客户端进行访问时，首先会向服务端发送安全验证请求，服务端收到请求后，首先会将机器上保存的公钥与客户端发送的公钥进行比较，如果一致，服务端则会向客户端响应加密质询，客户端接受到质询之后，使用私钥进行解密，然后再将解密结果发送给服务端，服务端进行校验后再返回响应结果，到这里就算是完成了一段密钥校验。

        //添加配置
        config := &ssh.ClientConfig{
                User: "root",
                Auth: []ssh.AuthMethod{ssh.Password("Password")},
                HostKeyCallback: ssh.InsecureIgnoreHostKey(),
                Timeout: 10 * time.Second,
            }
        }
        addr := fmt.Sprintf("%v:%v", IP, Port)

2.2 建立连接

在完成了所有的参数初始化之后，我们便可以调用Dial方法建立SSH连接。Dial方法一共有三个参数和两个返回值，第一个参数network为网络类型，这里我们使用面向连接的TCP协议，第二个参数addr则为目标机器的IP地址和端口号，第三个参数config则为前面我们生命的配置项。Dial会返回一个SSH连接和错误类型。

func Dial(network, addr string, config *ClientConfig) (*Client, error)

        //建立SSH连接
        sshClient, err := ssh.Dial("tcp", addr, config)
        if err != nil {
            log.Fatal("unable to create ssh conn")
        }

2.3 创建会话

在建立了与目标机器的SSH连接之后，我们就可以通过创建SSH会话来与目标机器进行通信。通过NewSession()方法便可以实现这一操作。

        //建立SSH会话
        sshSession, err := sshClient.NewSession()
        if err != nil {
           log.Fatal("unable to create ssh session")
        }

2.4 执行操作

与目标机器建立会话后，我们就可以通过执行命令等来操作远程服务器。Go目前为我们提供了五个用于操作远程机器的方法，分别是Run(), Start(), Output(), CombineOutpt(), Shell()。

🦫其中 Output(), **CombineOutpt()**这两个方法是对Run()方法进行不同程度上的封装，校验了输出流，错误流等相关内容。

        // Output runs cmd on the remote host and returns its standard output.
        func (s *Session) Output(cmd string) ([]byte, error) {
           if s.Stdout != nil {
              return nil, errors.New("ssh: Stdout already set")
           }
           var b bytes.Buffer
           s.Stdout = &b
           err := s.Run(cmd)
           return b.Bytes(), err
        }
        
        
        // CombinedOutput runs cmd on the remote host and returns its combined
        // standard output and standard error.
        func (s *Session) CombinedOutput(cmd string) ([]byte, error) {
           if s.Stdout != nil {
              return nil, errors.New("ssh: Stdout already set")
           }
           if s.Stderr != nil {
              return nil, errors.New("ssh: Stderr already set")
           }
           var b singleWriter
           s.Stdout = &b
           s.Stderr = &b
           err := s.Run(cmd)
           return b.b.Bytes(), err
        }

Run()方法则是对Start()方法进行了封装，添加了Wait方法，用于校验远程服务器的退出指令。Wait()方法中有一个管道类型的变量exitStatus，它是用来保存每次执行命令后，机器返回的退出状态的。有兴趣的朋友可以去看看这块的代码，这里就不贴代码了。

这里面有一个坑，如果我们在远程机器上去运行一个永远不会停止的程序，这个时候我们的程序一直等待不到远程机器发送的退出指令，就会造成程序一直阻塞而无法正常返回。解决的办法是用一个协程去单独执行这一块的任务，或者是使用定时器来定时结束session会话，来正常返回。

Start()方法与Shell方法一致，都是返回一个error类型，在底层都是调用了start()方法和SendRequest方法，关于这两个方法的内容这里就不做详细介绍了，有兴趣的朋友可以自行去阅读。唯一的区别是Start()方法有一个string类型的参数，用于接收用户输入的参数，而Shell()方法是无参数的。

使用Shell()方法配合RequestPty()等方法可以在本地建立一个伪终端，可以直接通过输入命令的形式操作目标机器。下面都会做一个示例。

        //Run
        func (s *Session) Run(cmd string) error {
           err := s.Start(cmd)
           if err != nil {
              fmt.Println(err)
              return err
           }
           return s.Wait()
                }
                
                
        // Start runs cmd on the remote host. Typically, the remote
        // server passes cmd to the shell for interpretation.
        // A Session only accepts one call to Run, Start or Shell.
        func (s *Session) Start(cmd string) error {
           if s.started {
              return errors.New("ssh: session already started")
           }
           req := execMsg{
              Command: cmd,
           }
        
           ok, err := s.ch.SendRequest("exec", true, Marshal(&req))
           if err == nil && !ok {
              err = fmt.Errorf("ssh: command %v failed", cmd)
           }
           if err != nil {
              return err
           }
           return s.start()
        }

2.5 示例代码(执行命令)

这里我们使用Run()方法来演示一下如果去执行命令，其他方法类型就不做演示了。这里我们使用一个标准输出流、错误流来保存执行结果。

这里演示了一个简单的执行过程，使用了cd命令到/home/min目录下，在给helloworld程序添加可执行权限，最后运行程序。

        var stdoutBuf, stderrBuf bytes.Buffer
        session.Stdout = &stdoutBuf
        session.Stderr = &stderrBuf
    
        // cd /home/min
        // chmod +x helloworld
        // ./helloworld
        cmd := fmt.Sprintf("cd %v ; chmod +x %v ; %v &", "/home/min", "helloworld", ./helloworld)
        err := session.Run(cmd)
        if err != nil {
            log.Fatal("[ERROR]: ", session.Stderr, err)
        }

2.6(创建伪终端)

        // 设置Terminal Mode
	modes := ssh.TerminalModes{
		ssh.ECHO:          0,     // 关闭回显
		ssh.TTY_OP_ISPEED: 14400, // 设置传输速率
		ssh.TTY_OP_OSPEED: 14400,
	}
    
        // 请求伪终端
	err = session.RequestPty("linux", 32, 160, modes)
	if err != nil {
		log.Println(err)
		return
	}
    
        // 设置输入输出
	session.Stdout = os.Stdout
	session.Stdin = os.Stdin
	session.Stderr = os.Stderr
 
	session.Shell() // 启动shell
	session.Wait()  // 等待退出

2.7 完整代码

//机器平台信息
type Machine struct {
   IP       string
   Port     string
   Username string
   Password string
}

//建立SSH连接
func CreateSSHConn(m *model.Machine) error {
   //初始化连接信息
   config := &ssh.ClientConfig{
      User:            m.Username,
      Auth:            []ssh.AuthMethod{ssh.Password(m.Password)},
      HostKeyCallback: ssh.InsecureIgnoreHostKey(),
      Timeout:         10 * time.Second,
   }
   addr := fmt.Sprintf("%v:%v", m.IP, m.Port)

   //建立ssh连接
   sshClient, err := ssh.Dial("tcp", addr, config)
   if err != nil {
      fmt.Println("unable create ssh conn", err)
      return err
   }
   defer sshClient.Close()
   
   //建立ssh会话
   session, err := sshClient.NewSession()
   if err != nil {
      fmt.Println("unable create ssh conn", err)
      return err
   }
   defer session.Close()
   
   
   //执行命令
   var stdoutBuf, stderrBuf bytes.Buffer
   session.Stdout = &stdoutBuf
   session.Stderr = &stderrBuf
   
   cmd := fmt.Sprintf("cd %v ; chmod +x %v ; %v &", "/home/min", "helloworld", ./helloworld)
   if err := session.Run(cmd); err != nil {
       log.Fatal("[ERROR]: ", session.Stderr, err)
   }
  
   //创建伪终端
   // 设置Terminal Mode
   modes := ssh.TerminalModes{
           ssh.ECHO:          0,     // 关闭回显
           ssh.TTY_OP_ISPEED: 14400, // 设置传输速率
           ssh.TTY_OP_OSPEED: 14400,
   }
      
   // 请求伪终端
   err = session.RequestPty("linux", 32, 160, modes)
   if err != nil {
           log.Fatal(err)
   }
      
   // 设置输入输出
   session.Stdout = os.Stdout
   session.Stdin = os.Stdin
   session.Stderr = os.Stderr
   
   session.Shell() // 启动shell
   session.Wait()  // 等待退出
   
   return err
}

到此这篇关于Golang实现SSH、SFTP操作小结的文章就介绍到这了,更多相关Golang实现SSH和SFT内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

golang 解析word文档操作
这篇文章主要介绍了golang 解析word文档操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-12-12
Golang环境变量设置和查看工具go env详解
go env 是 Go 工具链中的一个命令，用于设置和查看当前 Golang 环境的相关信息，对于理解、编译和运行 Golang 程序非常有用，本文就给大家简单的介绍一下Golang环境变量设置和查看工具go env,需要的朋友可以参考下
2023-07-07
一文详解Golang中new和make的区别
在Go语言中，new和make是两个用于创建对象的内建函数。本文将详细介绍new和make的区别，并通过多个方面的分析和代码示例，帮助大家理解它们的使用场景
2023-05-05
浅谈golang类型断言,失败类型断言返回值问题
这篇文章主要介绍了浅谈golang类型断言,失败类型断言返回值问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-12-12
GoLang BoltDB数据库详解
这篇文章主要介绍了GoLang BoltDB数据库，boltdb是使用Go语言编写的开源的键值对数据库，boltdb存储数据时 key和value都要求是字节数据，此处需要使用到序列化和反序列化
2023-02-02
浅析Golang开发中goroutine的正确使用姿势
很多初级的Gopher在学习了goroutine之后,在项目中其实使用率不高,所以这篇文章小编主要来带大家深入了解一下goroutine的常见使用方法,希望对大家有所帮助
2024-03-03
Go map发生内存泄漏解决方法
这篇文章主要介绍了Go map发生内存泄漏解决方法详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-11-11
Golang pprof性能测试与分析讲解
刚开始接触go就遇到了一个内存问题,在进行内存分析的时候发现了一下比较好的工具,在此留下记录,下面这篇文章主要给大家介绍了关于go性能分析工具pprof的性能测试,需要的朋友可以参考下
2023-04-04
Golang实现四种负载均衡的算法(随机,轮询等)
本文介绍了示例介绍了Golang 负载均衡的四种实现，主要包括了随机，轮询，加权轮询负载，一致性hash，感兴趣的小伙伴们可以参考一下
2021-06-06
详解Golang中NewTimer计时器的底层实现原理
本文将主要介绍一下Go语言中的NewTimer，首先展示基于NewTimer创建的定时器来实现超时控制。接着通过一系列问题的跟进，展示了NewTimer的底层实现原理，需要的可以参考一下
2023-05-05