-
Notifications
You must be signed in to change notification settings - Fork 0
/
client_scp.go
82 lines (67 loc) · 1.34 KB
/
client_scp.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
// Author: yangzq80@gmail.com
// Date: 2021-10-08
//
package ssh
import (
"bytes"
"fmt"
"io/ioutil"
"log"
"os"
"strings"
)
const (
chunkSize = 65536 // chunk size in bytes for scp
)
func (s *SSHClient) UploadFile(localSourceFile string, remoteTargetFile string) (stdout, stderr string, err error) {
fp, err := os.Open(localSourceFile)
if err != nil {
log.Println(err.Error())
return
}
defer fp.Close()
contents, err := ioutil.ReadAll(fp)
if err != nil {
log.Println(err.Error())
return
}
session, err := s.Client.NewSession()
if err != nil {
return
}
defer session.Close()
cmd := "cat >'" + strings.Replace(remoteTargetFile, "'", "'\\''", -1) + "'"
stdinPipe, err := session.StdinPipe()
if err != nil {
return
}
var stdoutBuf bytes.Buffer
var stderrBuf bytes.Buffer
session.Stdout = &stdoutBuf
session.Stderr = &stderrBuf
err = session.Start(cmd)
if err != nil {
return
}
for start, maxEnd := 0, len(contents); start < maxEnd; start += chunkSize {
//todo limit scp
//<-maxThroughputChan
end := start + chunkSize
if end > maxEnd {
end = maxEnd
}
_, err = stdinPipe.Write(contents[start:end])
if err != nil {
return
}
}
err = stdinPipe.Close()
if err != nil {
fmt.Printf(err.Error())
return
}
err = session.Wait()
stdout = stdoutBuf.String()
stderr = stderrBuf.String()
return
}