원격 호스트에서 명령을 실행하고 파일을 업로드 또는 다운로드하기 위해 golang을 사용하여 ssh를 캡슐화하는 방법

                                                        O 다음은 lGolang

튜토리얼 칼럼에서 소개한 것으로 원격 호스트에서 명령을 실행하고 파일을 업로드하거나 다운로드하기 위해 SSH를 Golang으로 묶는 방법을 소개합니다. 도움이 필요한 친구들에게 도움이 되길 바랍니다!

Python에서는 paramiko를 사용하여 원격 호스트에서 명령을 실행하고 파일을 업로드 및 다운로드할 수 있습니다. Go에서는 ssh 및 sftp 패키지를 사용하여 다음 기능을 수행할 수도 있습니다.

원격 호스트는 명령을 실행하여 결과, 반환 값을 반환합니다.

원격 호스트에 파일 업로드 및 다운로드, 전송되는 바이트 수

• 인증 방법

비밀번호가 지정된 경우 인증은 사용자 + 비밀번호 방식을 사용하고, 그렇지 않으면 사용자 + 비밀키 방식을 사용합니다사용자를 지정하지 않으면 기본적으로 현재 사용자가 사용됩니다

• 비밀번호를 지정하지 않으면 사용자 + 비밀키 방식이 사용됩니다. ~/.ssh/id_rsa 파일의 개인 키 파일이 기본적으로 획득됩니다. 비밀 키
  • 는 paramik
코드와 유사합니다

• package mainimport (
      "errors"
      "fmt"
      "github.com/pkg/sftp"
      "golang.org/x/crypto/ssh"
      "io"
      "io/ioutil"
      "log"
      "os"
      "os/user"
      "time")var (
      DefaultSShTcpTimeout = 15 * time.Second   // 与ssh建立连接的默认时间，自己设置一个就行)// 错误定义var (
      InvalidHostName = errors.New("invalid parameters: hostname is empty")
      InvalidPort     = errors.New("invalid parameters: port must be range 0 ~ 65535"))// 返回当前用户名func getCurrentUser() string {
      user, _ := user.Current()
      return user.Username}// 存放上传或下载的信息type TransferInfo struct {
      Kind         string   // upload或download
      Local        string   // 本地路径
      Dst          string   // 目标路径
      TransferByte int64    // 传输的字节数(byte)}func (t *TransferInfo) String()  string {
      return fmt.Sprintf(`TransforInfo(Kind:"%s", Local: "%s", Dst: "%s", TransferByte: %d)`,
          t.Kind, t.Local, t.Dst, t.TransferByte)}// 存放执行结果的结构体信息type ExecInfo struct {
      Cmd         string
      Output     []byte
      ExitCode int}func (e *ExecInfo) OutputString() string {
      return string(e.Output)}func (e *ExecInfo) String() string {
      return fmt.Sprintf(`ExecInfo(cmd: "%s", exitcode: %d)`,
          e.Cmd, e.ExitCode)}type AuthConfig struct {
      *ssh.ClientConfig
      User     string
      Password string
      KeyFile  string
      Timeout  time.Duration}func (a *AuthConfig) setDefault()  {
      if a.User == "" {
          a.User = getCurrentUser()
      }
  
      if a.KeyFile == "" {
          userHome, _ := os.UserHomeDir()
          a.KeyFile = fmt.Sprintf("%s/.ssh/id_rsa", userHome)
      }
  
      if a.Timeout == 0 {
          a.Timeout = DefaultSShTcpTimeout    }}func (a *AuthConfig) SetAuthMethod() (ssh.AuthMethod, error) {
      a.setDefault()
      if a.Password != "" {
          return ssh.Password(a.Password), nil    }
      data, err := ioutil.ReadFile(a.KeyFile)
      if err != nil {
          return nil, err    }
      singer, err := ssh.ParsePrivateKey(data)
      if err != nil {
          return nil, err    }
      return ssh.PublicKeys(singer), nil}func (a *AuthConfig) ApplyConfig() error {
      authMethod, err := a.SetAuthMethod()
      if err != nil {
          return err    }
      a.ClientConfig = &ssh.ClientConfig{
          User: a.User,
          Auth: []ssh.AuthMethod{authMethod},
          HostKeyCallback: ssh.InsecureIgnoreHostKey(),
          Timeout: a.Timeout,
      }
      return nil}// 存放连接的结构体type conn struct {
      client     *ssh.Client
      sftpClient *sftp.Client}func (c *conn) Close()  {
      if c.sftpClient != nil {
          c.sftpClient.Close()
          c.sftpClient = nil    }
      if c.client != nil {
          c.client.Close()
          c.client = nil    }}// SSHClient结构体type SSHClient struct {
      conn
      HostName   string
      Port        int
      AuthConfig AuthConfig}// 设置默认端口信息func (s *SSHClient) setDefaultValue()  {
      if s.Port == 0 {
          s.Port = 22
      }}// 与远程主机连接func (s *SSHClient) Connect() error {
      if s.client != nil {
          log.Println("Already Login")
          return nil    }
      if err := s.AuthConfig.ApplyConfig(); err != nil {
          return err    }
      s.setDefaultValue()
      addr := fmt.Sprintf("%s:%d", s.HostName, s.Port)
      var err error
      s.client, err = ssh.Dial("tcp", addr, s.AuthConfig.ClientConfig)
      if err != nil {
          return err    }
      return nil}// 一个session只能执行一次命令，也就是说不能在同一个session执行多次s.session.CombinedOutput// 如果想执行多次，需要每条为每个命令创建一个session(这里是这样做)func (s *SSHClient) Exec(cmd string) (*ExecInfo, error) {
      session, err := s.client.NewSession()
      if err != nil {
          return nil, err    }
      defer session.Close()
      output, err := session.CombinedOutput(cmd)
      var exitcode int    if err != nil {
          // 断言转成具体实现类型，获取返回值
          exitcode = err.(*ssh.ExitError).ExitStatus()
      }
      return &ExecInfo{
          Cmd: cmd,
          Output: output,
          ExitCode: exitcode,
      }, nil}// 将本地文件上传到远程主机上func (s *SSHClient) Upload(localPath string, dstPath string) (*TransferInfo, error) {
      transferInfo := &TransferInfo{Kind: "upload", Local: localPath, Dst: dstPath, TransferByte: 0}
      var err error    // 如果sftp客户端没有打开，就打开，为了复用
      if s.sftpClient == nil {
          if s.sftpClient, err = sftp.NewClient(s.client); err != nil {
              return transferInfo, err        }
      }
      localFileObj, err := os.Open(localPath)
      if err != nil {
          return transferInfo, err    }
      defer localFileObj.Close()
  
      dstFileObj, err := s.sftpClient.Create(dstPath)
      if err != nil {
          return transferInfo, err    }
      defer dstFileObj.Close()
  
      written, err := io.Copy(dstFileObj, localFileObj)
      if err != nil {
          return transferInfo, err    }
      transferInfo.TransferByte = written    return transferInfo, nil}// 从远程主机上下载文件到本地func (s *SSHClient) Download(dstPath string, localPath string)  (*TransferInfo, error) {
      transferInfo := &TransferInfo{Kind: "download", Local: localPath, Dst: dstPath, TransferByte: 0}
      var err error    if s.sftpClient == nil {
          if s.sftpClient, err = sftp.NewClient(s.client); err != nil {
              return transferInfo, err        }
      }
      //defer s.sftpClient.Close()
      localFileObj, err := os.Create(localPath)
      if err != nil {
          return transferInfo, err    }
      defer localFileObj.Close()
  
      dstFileObj, err := s.sftpClient.Open(dstPath)
      if err != nil {
          return transferInfo, err    }
      defer dstFileObj.Close()
  
      written, err := io.Copy(localFileObj, dstFileObj)
      if err != nil {
          return transferInfo, err    }
      transferInfo.TransferByte = written    return transferInfo, nil}// SSHclient的构造方法func NewSSHClient(hostname string, port int, authConfig AuthConfig) (*SSHClient, error) {
      switch {
      case hostname == "":
          return nil, InvalidHostName    case port > 65535 || port < 0:
          return nil, InvalidPort    }
      sshClient := &SSHClient{HostName: hostname, Port: port, AuthConfig: authConfig}
      err := sshClient.Connect()
      if err != nil {
          return nil, err    }
      return sshClient, nil}func main()  {// 测试
      sshClient, err := NewSSHClient("172.16.0.178", 22, AuthConfig{User: "root"})
      if err != nil {
          fmt.Println(err)
          return
      }
      defer sshClient.Close()
      //第一次 执行命令
      execinfo, err := sshClient.Exec("ls -l")
      fmt.Println(execinfo.OutputString(), err)
      //第二次执行命令
      out1, exitcode2 := sshClient.Exec("ifconfig -a")
       fmt.Println(string(out1), exitcode2)
       // 上传文件
      transInfoUpload, err := sshClient.Upload("/tmp/passwd", "/tmp/password_upload")
      fmt.Println(transInfoUpload, err)
      // 下载文件
      transInfoDownload, err := sshClient.Download("/etc/passwd", "/tmp/passwd_download")
      fmt.Println(transInfoDownload, err)}

위 내용은 원격 호스트에서 명령을 실행하고 파일을 업로드 또는 다운로드하기 위해 golang을 사용하여 ssh를 캡슐화하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!