forked from gen0cide/laforge
/
ssh.go
108 lines (89 loc) · 2.1 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
103
104
105
106
107
108
package command
import (
"io/ioutil"
"log"
"os"
"github.com/codegangsta/cli"
"github.com/gen0cide/laforge/competition"
"github.com/shiena/ansicolor"
"golang.org/x/crypto/ssh"
"golang.org/x/crypto/ssh/terminal"
)
func CmdSsh(c *cli.Context) {
hostname := c.Args().Get(0)
if len(hostname) < 1 {
competition.LogFatal("You did not provide a hostname to use.")
}
comp, env := InitConfig()
sshHosts := env.NewSSHConfig()
var (
ip string
socket string
port = "22"
user = "root"
)
if val, ok := sshHosts.Hosts[hostname]; ok {
ip = val
socket = ip + ":" + port
} else {
competition.LogFatal("Unknown host: " + hostname)
}
publicKey, err := PublicKeyFile(comp.SSHPrivateKeyPath())
if err != nil {
panic(err)
}
config := &ssh.ClientConfig{
User: user,
Auth: []ssh.AuthMethod{
publicKey,
},
HostKeyCallback: ssh.InsecureIgnoreHostKey(),
}
conn, err := ssh.Dial("tcp", socket, config)
if err != nil {
panic("Failed to dial: " + err.Error())
}
defer conn.Close()
fd := int(os.Stdin.Fd())
oldState, err := terminal.MakeRaw(fd)
if err != nil {
panic(err)
}
termWidth, termHeight, err := terminal.GetSize(fd)
if err != nil {
panic(err)
}
session, err := conn.NewSession()
if err != nil {
log.Fatal("unable to create session: ", err)
}
defer session.Close()
session.Stdout = ansicolor.NewAnsiColorWriter(os.Stdout)
session.Stderr = ansicolor.NewAnsiColorWriter(os.Stderr)
session.Stdin = os.Stdin
modes := ssh.TerminalModes{
ssh.ECHO: 1,
ssh.ECHOCTL: 1,
ssh.TTY_OP_ISPEED: 14400,
ssh.TTY_OP_OSPEED: 14400,
}
if err := session.RequestPty("xterm", termHeight, termWidth, modes); err != nil {
log.Fatal("request for pseudo terminal failed: ", err)
}
if err := session.Shell(); err != nil {
log.Fatal("failed to start shell: ", err)
}
session.Wait()
terminal.Restore(fd, oldState)
}
func PublicKeyFile(file string) (ssh.AuthMethod, error) {
buffer, err := ioutil.ReadFile(file)
if err != nil {
return nil, err
}
key, err := ssh.ParsePrivateKey(buffer)
if err != nil {
return nil, err
}
return ssh.PublicKeys(key), nil
}