-
Notifications
You must be signed in to change notification settings - Fork 4
/
ossh_util.go
52 lines (45 loc) · 1.04 KB
/
ossh_util.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
package main
import (
"encoding/pem"
"errors"
"fmt"
"io/ioutil"
"os"
"syscall"
"golang.org/x/crypto/ssh"
"golang.org/x/crypto/ssh/terminal"
)
func readBytePasswordFromTerminal(prompt string) []byte {
fmt.Printf("%s ", prompt)
bytePassword, _ := terminal.ReadPassword(int(syscall.Stdin))
fmt.Printf("\n")
return bytePassword
}
func publicKeyFile(file string) (ssh.AuthMethod, error) {
var key ssh.Signer
var err error
buffer, err := ioutil.ReadFile(file)
if err != nil {
return nil, err
}
_, rest := pem.Decode(buffer)
if len(rest) > 0 {
return nil, errors.New("Failed to decode the key \"" + file + "\"")
}
if key, err = ssh.ParsePrivateKey(buffer); err != nil {
if err == err.(*ssh.PassphraseMissingError) {
if key, err = ssh.ParsePrivateKeyWithPassphrase(buffer, readBytePasswordFromTerminal("SSH Passphrase:")); err != nil {
return nil, err
}
return ssh.PublicKeys(key), nil
}
return nil, err
}
return ssh.PublicKeys(key), nil
}
func abortOnError(err error) {
if err != nil {
fmt.Println(err)
os.Exit(1)
}
}