'首次提交'
This commit is contained in:
44
main.go
Normal file
44
main.go
Normal file
@@ -0,0 +1,44 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"prot/sshtunnel"
|
||||
"syscall"
|
||||
)
|
||||
|
||||
func main() {
|
||||
var sts []*sshtunnel.Config
|
||||
p := "config.json"
|
||||
if len(os.Args) == 2 {
|
||||
p = os.Args[1]
|
||||
}
|
||||
f, err := ioutil.ReadFile(p)
|
||||
if err != nil {
|
||||
log.Printf("载入配置文件出错, 错误: %v", err)
|
||||
os.Exit(-1)
|
||||
}
|
||||
err = json.Unmarshal(f, &sts)
|
||||
if nil != err {
|
||||
log.Printf("解析配置文件内容出错, 错误: %v", err)
|
||||
os.Exit(-1)
|
||||
}
|
||||
|
||||
var tunnels []*sshtunnel.SSHTunnel
|
||||
for _, st := range sts {
|
||||
tunnel := sshtunnel.NewSSHTunnel(st)
|
||||
tunnel.Start()
|
||||
tunnels = append(tunnels, tunnel)
|
||||
}
|
||||
|
||||
signalChan := make(chan os.Signal)
|
||||
signal.Notify(signalChan, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
|
||||
<-signalChan
|
||||
for _, t := range tunnels {
|
||||
t.Close()
|
||||
}
|
||||
os.Exit(0)
|
||||
}
|
||||
Reference in New Issue
Block a user