forked from kubernetes/kops
/
ssh.go
102 lines (90 loc) · 2.23 KB
/
ssh.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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package kutil
import (
"fmt"
"golang.org/x/crypto/ssh"
"io/ioutil"
"k8s.io/kops/util/pkg/vfs"
)
type NodeSSH struct {
Hostname string
SSHConfig ssh.ClientConfig
sshClient *ssh.Client
}
func (m *NodeSSH) Root() (*vfs.SSHPath, error) {
client, err := m.GetSSHClient()
if err != nil {
return nil, err
}
sudo := true
return vfs.NewSSHPath(client, m.Hostname, "/", sudo), nil
}
func AddSSHIdentity(sshConfig *ssh.ClientConfig, p string) error {
a, err := parsePrivateKeyFile(p)
if err != nil {
return err
}
sshConfig.Auth = append(sshConfig.Auth, a)
return nil
}
func (m *NodeSSH) dial() (*ssh.Client, error) {
users := []string{"admin", "ubuntu"}
if m.SSHConfig.User != "" {
users = []string{m.SSHConfig.User}
}
var lastError error
for _, user := range users {
m.SSHConfig.User = user
sshClient, err := ssh.Dial("tcp", m.Hostname+":22", &m.SSHConfig)
if err == nil {
return sshClient, err
}
lastError = err
}
return nil, fmt.Errorf("error connecting to SSH on server %q: %v", m.Hostname, lastError)
}
func (m *NodeSSH) GetSSHClient() (*ssh.Client, error) {
if m.sshClient == nil {
sshClient, err := m.dial()
if err != nil {
return nil, err
}
m.sshClient = sshClient
}
return m.sshClient, nil
}
//func (m *NodeSSH) ReadFile(remotePath string) ([]byte, error) {
// b, err := m.exec("sudo cat " + remotePath)
// if err != nil {
// return nil, fmt.Errorf("error reading remote file %q: %v", remotePath, err)
// }
// return b, nil
//}
//func (m *NodeSSH) exec(cmd string) ([]byte, error) {
// client, err := m.GetSSHClient()
// if err != nil {
// return nil, err
// }
//
// session, err := client.NewSession()
// if err != nil {
// return nil, fmt.Errorf("error creating SSH session: %v", err)
// }
// defer session.Close()
//
// b, err := session.Output(cmd)
// if err != nil {
// return nil, fmt.Errorf("error executing command %q: %v", cmd, err)
// }
// return b, nil
//}
func parsePrivateKeyFile(p string) (ssh.AuthMethod, error) {
buffer, err := ioutil.ReadFile(p)
if err != nil {
return nil, fmt.Errorf("error reading SSH key file %q: %v", p, err)
}
key, err := ssh.ParsePrivateKey(buffer)
if err != nil {
return nil, fmt.Errorf("error parsing key file %q: %v", p, err)
}
return ssh.PublicKeys(key), nil
}