func main() {
flag.Parse()
c, logfile, err := loadConfig()
if err != nil {
log.ErrorExit(err)
}
if len(logfile) > 0 {
f, err := os.OpenFile(logfile, os.O_WRONLY|os.O_APPEND, 0644)
if err != nil {
log.ErrorExit(err)
}
defer f.Close()
log.DefaultLogger().SetOutput(f)
}
srv, err := transocks.NewServer(c)
if err != nil {
log.ErrorExit(err)
}
log.Info("server starts", nil)
srv.Serve()
log.Info("server ends", nil)
}
func loadConfig() (*transocks.Config, string, error) {
tc := new(tomlConfig)
md, err := toml.DecodeFile(*configFile, tc)
if err != nil {
return nil, "", err
}
if len(md.Undecoded()) > 0 {
return nil, "", fmt.Errorf("undecoded key in TOML: %v", md.Undecoded())
}
c := transocks.NewConfig()
c.Listen = tc.Listen
u, err := url.Parse(tc.ProxyURL)
if err != nil {
return nil, "", err
}
c.ProxyURL = u
if err = log.DefaultLogger().SetThresholdByName(tc.LogLevel); err != nil {
return nil, "", err
}
return c, tc.LogFile, nil
}