package/tun2socks
1
0
Fork 0
mirror of https://github.com/xjasonlyu/tun2socks.git synced 2025-11-02 21:04:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: main.go

Browse Source
This commit is contained in:
xjasonlyu
2022-03-26 18:18:43 +08:00
parent ae07fbdb68
commit c6ca52326a
2 changed files with 47 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
main.go
View File

@@ -2,24 +2,32 @@ package main
import (
"flag"
"fmt"
"os"
"os/signal"
"syscall"
"github.com/xjasonlyu/tun2socks/v2/engine"
"github.com/xjasonlyu/tun2socks/v2/internal/version"
"github.com/xjasonlyu/tun2socks/v2/log"
"go.uber.org/automaxprocs/maxprocs"
"gopkg.in/yaml.v3"
)
var key = new(engine.Key)
var (
key = new(engine.Key)
configFile string
versionFlag bool
)
func init() {
flag.BoolVar(&versionFlag, "version", false, "Show version and then quit")
flag.IntVar(&key.Mark, "fwmark", 0, "Set firewall MARK (Linux only)")
flag.IntVar(&key.MTU, "mtu", 0, "Set device maximum transmission unit (MTU)")
flag.IntVar(&key.UDPTimeout, "udp-timeout", 0, "Set timeout for each UDP session")
flag.BoolVar(&key.Version, "version", false, "Show version information and quit")
flag.StringVar(&key.Config, "config", "", "YAML format configuration file")
flag.StringVar(&configFile, "config", "", "YAML format configuration file")
flag.StringVar(&key.Device, "device", "", "Use this device [driver://]name")
flag.StringVar(&key.Interface, "interface", "", "Use network INTERFACE (Linux/MacOS only)")
flag.StringVar(&key.LogLevel, "loglevel", "info", "Log level [debug|info|warning|error|silent]")
@@ -32,16 +40,32 @@ func init() {
func main() {
maxprocs.Set(maxprocs.Logger(func(string, ...any) {}))
if versionFlag {
fmt.Println(version.String())
fmt.Println(version.BuildString())
os.Exit(0)
}
if configFile != "" {
data, err := os.ReadFile(configFile)
if err != nil {
log.Fatalf("Failed to read config %s: %v", configFile, err)
}
if err = yaml.Unmarshal(data, key); err != nil {
log.Fatalf("Failed to unmarshal config %s: %v", configFile, err)
}
}
engine.Insert(key)
checkErr := func(msg string, f func() error) {
if err := f(); err != nil {
assert := func(msg string, err error) {
if err != nil {
log.Fatalf("Failed to %s: %v", msg, err)
}
}
checkErr("start engine", engine.Start)
defer checkErr("stop engine", engine.Stop)
assert("start engine", engine.Start())
defer assert("stop engine", engine.Stop())
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)