-
Notifications
You must be signed in to change notification settings - Fork 19
/
main.go
83 lines (72 loc) · 1.47 KB
/
main.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
package main
import (
"context"
"flag"
"fmt"
"io"
"log"
"os"
tun "github.com/glycerine/sshego"
ssh "github.com/glycerine/sshego/xendor/github.com/glycerine/xcryptossh"
)
const ProgramName = "gosshtun"
func main() {
myflags := flag.NewFlagSet(ProgramName, flag.ExitOnError)
cfg := tun.NewSshegoConfig()
cfg.DefineFlags(myflags)
err := myflags.Parse(os.Args[1:])
if cfg.ShowVersion {
fmt.Printf("\n%v\n", tun.SourceVersion())
os.Exit(0)
}
err = cfg.ValidateConfig()
if err != nil {
log.Fatalf("%s command line flag error: '%s'", ProgramName, err)
}
//p("cfg = %#v", cfg)
h, err := tun.NewKnownHosts(cfg.ClientKnownHostsPath, tun.KHJson)
panicOn(err)
cfg.KnownHosts = h
if cfg.WriteConfigOut != "" {
var o io.WriteCloser
if cfg.WriteConfigOut == "-" {
o = os.Stdout
} else {
o, err = os.Create(cfg.WriteConfigOut)
if err != nil {
panic(err)
}
}
err = cfg.SaveConfig(o)
if err != nil {
panic(err)
}
}
if cfg.AddUser != "" {
tun.AddUserAndExit(cfg)
}
if cfg.DelUser != "" {
tun.DelUserAndExit(cfg)
}
passphrase := ""
totpUrl := ""
ctx := context.Background()
halt := ssh.NewHalter()
_, _, err = cfg.SSHConnect(ctx, h, cfg.Username, cfg.PrivateKeyPath,
cfg.SSHdServer.Host, cfg.SSHdServer.Port, passphrase, totpUrl, halt)
if err != nil {
fmt.Println(err.Error())
os.Exit(0)
}
if err != nil {
panic(err)
}
if !cfg.WriteConfigOnly {
select {}
}
}
func panicOn(err error) {
if err != nil {
panic(err)
}
}