在go中通过SSH发送文件

Posted 2021-03-26

I found this answer before posting this question but the answer is not clear to me.

这是答案的代码：

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
", 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的请求添加了waitgroup