sshtargate/main.go

121 lines
2.2 KiB
Go
Raw Normal View History

2020-01-04 21:22:45 +00:00
package main
import (
"flag"
"fmt"
"log"
"os"
"os/signal"
"strings"
"sync"
"syscall"
2021-07-09 00:23:10 +00:00
"code.rocketnine.space/tslocum/ez"
2021-04-20 01:35:57 +00:00
"code.rocketnine.space/tslocum/sshtargate/portal"
2020-01-04 21:22:45 +00:00
"github.com/anmitsu/go-shlex"
)
const (
version = "0.0.0"
versionInfo = `sshtargate - Host SSH portals to applications - v` + version + `
2021-04-20 01:35:57 +00:00
https://code.rocketnine.space/tslocum/sshtargate
2020-01-04 21:22:45 +00:00
The MIT License (MIT)
Copyright (c) 2020 Trevor Slocum <trevor@rocketnine.space>
`
)
var (
printVersionInfo bool
configPath string
2020-02-07 23:03:07 +00:00
portals []*portal.Portal
2020-01-04 21:22:45 +00:00
portalsLock = new(sync.Mutex)
done = make(chan bool)
)
func main() {
flag.BoolVar(&printVersionInfo, "version", false, "print version information and exit")
flag.StringVar(&configPath, "config", "", "path to configuration file")
flag.Parse()
if printVersionInfo {
fmt.Print(versionInfo)
return
}
2021-07-09 00:23:10 +00:00
if configPath == "" {
var err error
configPath, err = ez.DefaultConfigPath("sshtargate")
if err != nil {
log.Fatal(err)
}
}
2020-01-04 21:22:45 +00:00
// TODO: Allow portals to be specified via arguments
// TODO: Catch SIGHUP
sigc := make(chan os.Signal, 1)
signal.Notify(sigc,
syscall.SIGINT,
syscall.SIGTERM)
go func() {
<-sigc
done <- true
}()
log.Println("Initializing sshtargate...")
2021-07-09 00:23:10 +00:00
err := ez.Deserialize(config, configPath)
if err != nil {
log.Fatalf("failed to read configuration file: %s", err)
2020-01-04 21:22:45 +00:00
}
2021-07-09 00:23:10 +00:00
if len(config.Portals) == 0 {
log.Println("Warning: No portals are defined")
2020-01-04 21:22:45 +00:00
}
for pname, pcfg := range config.Portals {
cs, err := shlex.Split(pcfg.Command, true)
if err != nil {
log.Fatalf("failed to split command %s", pcfg.Command)
}
pname, pcfg := pname, pcfg // Capture
go func() {
wg := new(sync.WaitGroup)
for _, address := range pcfg.Host {
wg.Add(1)
address := address // Capture
go func() {
2020-02-07 23:03:07 +00:00
p, err := portal.New(pname, address, cs)
2020-01-04 21:22:45 +00:00
if err != nil {
log.Fatalf("failed to start portal %s on %s: %s", pname, address, err)
}
portalsLock.Lock()
portals = append(portals, p)
portalsLock.Unlock()
wg.Done()
}()
}
wg.Wait()
log.Printf("Opened portal %s on %s to %s", pname, strings.Join(pcfg.Host, ","), pcfg.Command)
}()
}
<-done
portalsLock.Lock()
for _, p := range portals {
p.Shutdown()
}
portalsLock.Unlock()
}