当前位置： > > > > 在 go 中通过 SSH 发送文件

在 go 中通过 SSH 发送文件

来源：stackoverflow
2024-04-27 14:45:36
0浏览
收藏

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天米云就整理分享《在 go 中通过 SSH 发送文件》，聊聊，希望可以帮助到正在努力赚钱的你。

问题内容

我在发布这个问题之前找到了这个答案，但我不清楚答案。

这是答案的代码：

conn, err := ssh.Dial("tcp", hostname+":22", config)
if err != nil {
    return err
}

session, err := conn.NewSession()
if err != nil {
    return err
}
defer session.Close()

r, err := session.StdoutPipe()
if err != nil {
    return err
}

name := fmt.Sprintf("%s/backup_folder_%v.tar.gz", path, time.Now().Unix())
file, err := os.OpenFile(name, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0644)
if err != nil {
    return err
}
defer file.Close()

if err := session.Start(cmd); err != nil {
    return err
}

n, err := io.Copy(file, r)
if err != nil {
    return err
}

if err := session.Wait(); err != nil {
    return err
}

return nil

我不明白 cmd 变量和 io.copy 之间的关系，它在哪里以及如何知道要复制哪个文件。 我喜欢使用 io.copy 的想法，但我不知道如何通过 ssh 创建文件并开始使用 io.copy 向其发送内容。

解决方案

以下是有关如何使用 go 作为 scp 客户端的最小示例：

config := &ssh.clientconfig{
    user: "user",
    auth: []ssh.authmethod{
        ssh.password("pass"),
    },
    hostkeycallback: ssh.insecureignorehostkey(),
}

client, _ := ssh.dial("tcp", "remotehost:22", config)
defer client.close()

session, _ := client.newsession()
defer session.close()

file, _ := os.open("filetocopy")
defer file.close()
stat, _ := file.stat()

wg := sync.waitgroup{}
wg.add(1)

go func() {
    hostin, _ := session.stdinpipe()
    defer hostin.close()
    fmt.fprintf(hostin, "c0664 %d %s\n", stat.size(), "filecopyname")
    io.copy(hostin, file)
    fmt.fprint(hostin, "\x00")
    wg.done()
}()

session.run("/usr/bin/scp -t /remotedirectory/")
wg.wait()

请注意，为了简洁起见，我忽略了所有错误。

1. session.stdinpipe() 将为远程主机创建一个可写管道。
2. fmt.fprintf(... "c0664 ...") 将发出带有 0664 权限、stat.size() 大小和远程文件名 filecopyname 的文件开始信号。
3. io.copy(hostin, file) 会将 file 的内容写入 hostin。
4. fmt.fprint(hostin, "\x00") 将发出文件结束信号。
5. session.run("/usr/bin/scp -qt /remotedirectory/") 将运行 scp 命令。

编辑：根据 op 的请求添加等待组

您也可以使用 sftp 包来完成此操作 – “github.com/pkg/sftp”

func sshcopyfile(srcpath, dstpath string) error {
    config := &ssh.clientconfig{
        user: "user",
        auth: []ssh.authmethod{
            ssh.password("pass"),
        },
        hostkeycallback: ssh.insecureignorehostkey(),
    }

    client, _ := ssh.dial("tcp", "remotehost:22", config)
    defer client.close()


    // open an sftp session over an existing ssh connection.
    sftp, err := sftp.newclient(client)
    if err != nil {
        return err
    }
    defer sftp.close()

    // open the source file
    srcfile, err := os.open(srcpath)
    if err != nil {
        return err
    }
    defer srcfile.close()

    // create the destination file
    dstfile, err := sftp.create(dstpath)
    if err != nil {
        return err
    }
    defer dstfile.close()

    // write to file
    if  _, err := dstfile.readfrom(srcfile); err!= nil {
        return err
    }
    return nil
}

然后这样称呼它

sshcopyfile("/path/to/local/file.txt", "/path/on/remote/file.txt")

这些是必需的包

"github.com/pkg/sftp"
"golang.org/x/crypto/ssh"

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注米云公众号，一起学习编程~