Xray-core/main/run.go

231 lines
5.1 KiB
Go
Raw Normal View History

2020-11-25 13:01:53 +02:00
package main
import (
"fmt"
"log"
"os"
"os/signal"
"path"
"path/filepath"
"regexp"
2020-11-25 13:01:53 +02:00
"runtime"
"runtime/debug"
2021-03-21 11:13:51 +02:00
"strings"
2020-11-25 13:01:53 +02:00
"syscall"
"time"
2020-11-25 13:01:53 +02:00
2020-12-04 03:36:16 +02:00
"github.com/xtls/xray-core/common/cmdarg"
"github.com/xtls/xray-core/common/errors"
clog "github.com/xtls/xray-core/common/log"
2020-12-04 03:36:16 +02:00
"github.com/xtls/xray-core/common/platform"
"github.com/xtls/xray-core/core"
"github.com/xtls/xray-core/main/commands/base"
2020-11-25 13:01:53 +02:00
)
var cmdRun = &base.Command{
UsageLine: "{{.Exec}} run [-c config.json] [-confdir dir]",
Short: "Run Xray with config, the default command",
Long: `
Run Xray with config, the default command.
The -config=file, -c=file flags set the config files for
Xray. Multiple assign is accepted.
The -confdir=dir flag sets a dir with multiple json config
The -format=json flag sets the format of config files.
2021-03-21 11:13:51 +02:00
Default "auto".
2020-11-25 13:01:53 +02:00
The -test flag tells Xray to test config files only,
without launching the server.
The -dump flag tells Xray to print the merged config.
2020-11-25 13:01:53 +02:00
`,
}
func init() {
2021-10-19 19:57:14 +03:00
cmdRun.Run = executeRun // break init loop
2020-11-25 13:01:53 +02:00
}
var (
configFiles cmdarg.Arg // "Config file for Xray.", the option is customed type, parse in main
configDir string
dump = cmdRun.Flag.Bool("dump", false, "Dump merged config only, without launching Xray server.")
2020-11-25 13:01:53 +02:00
test = cmdRun.Flag.Bool("test", false, "Test config file only, without launching Xray server.")
2021-03-21 11:13:51 +02:00
format = cmdRun.Flag.String("format", "auto", "Format of input file.")
2020-11-25 13:01:53 +02:00
/* We have to do this here because Golang's Test will also need to parse flag, before
* main func in this file is run.
*/
_ = func() bool {
cmdRun.Flag.Var(&configFiles, "config", "Config path for Xray.")
cmdRun.Flag.Var(&configFiles, "c", "Short alias of -config")
cmdRun.Flag.StringVar(&configDir, "confdir", "", "A dir with multiple json config")
return true
}()
)
func executeRun(cmd *base.Command, args []string) {
if *dump {
clog.ReplaceWithSeverityLogger(clog.Severity_Warning)
errCode := dumpConfig()
os.Exit(errCode)
}
2020-11-25 13:01:53 +02:00
printVersion()
server, err := startXray()
if err != nil {
2020-12-25 12:53:17 +02:00
fmt.Println("Failed to start:", err)
// Configuration error. Exit with a special value to prevent systemd from restarting.
os.Exit(23)
2020-11-25 13:01:53 +02:00
}
if *test {
fmt.Println("Configuration OK.")
2020-12-25 12:53:17 +02:00
os.Exit(0)
2020-11-25 13:01:53 +02:00
}
if err := server.Start(); err != nil {
2020-12-25 12:53:17 +02:00
fmt.Println("Failed to start:", err)
os.Exit(-1)
2020-11-25 13:01:53 +02:00
}
defer server.Close()
/*
conf.FileCache = nil
conf.IPCache = nil
conf.SiteCache = nil
*/
2020-11-25 13:01:53 +02:00
// Explicitly triggering GC to remove garbage from config loading.
runtime.GC()
debug.FreeOSMemory()
2020-11-25 13:01:53 +02:00
{
osSignals := make(chan os.Signal, 1)
signal.Notify(osSignals, os.Interrupt, syscall.SIGTERM)
<-osSignals
}
}
func dumpConfig() int {
files := getConfigFilePath(false)
if config, err := core.GetMergedConfig(files); err != nil {
fmt.Println(err)
time.Sleep(1 * time.Second)
return 23
} else {
fmt.Print(config)
}
return 0
}
2020-11-25 13:01:53 +02:00
func fileExists(file string) bool {
info, err := os.Stat(file)
return err == nil && !info.IsDir()
}
func dirExists(file string) bool {
if file == "" {
return false
}
info, err := os.Stat(file)
return err == nil && info.IsDir()
}
2021-03-21 11:13:51 +02:00
func getRegepxByFormat() string {
switch strings.ToLower(*format) {
case "json":
return `^.+\.(json|jsonc)$`
2021-03-21 11:13:51 +02:00
case "toml":
return `^.+\.toml$`
case "yaml", "yml":
return `^.+\.(yaml|yml)$`
default:
return `^.+\.(json|jsonc|toml|yaml|yml)$`
2021-03-21 11:13:51 +02:00
}
}
2020-11-25 13:01:53 +02:00
func readConfDir(dirPath string) {
confs, err := os.ReadDir(dirPath)
2020-11-25 13:01:53 +02:00
if err != nil {
log.Fatalln(err)
}
for _, f := range confs {
2021-03-21 11:13:51 +02:00
matched, err := regexp.MatchString(getRegepxByFormat(), f.Name())
if err != nil {
log.Fatalln(err)
}
if matched {
2020-11-25 13:01:53 +02:00
configFiles.Set(path.Join(dirPath, f.Name()))
}
}
}
func getConfigFilePath(verbose bool) cmdarg.Arg {
2020-11-25 13:01:53 +02:00
if dirExists(configDir) {
if verbose {
log.Println("Using confdir from arg:", configDir)
}
2020-11-25 13:01:53 +02:00
readConfDir(configDir)
} else if envConfDir := platform.GetConfDirPath(); dirExists(envConfDir) {
if verbose {
log.Println("Using confdir from env:", envConfDir)
}
2020-11-25 13:01:53 +02:00
readConfDir(envConfDir)
}
if len(configFiles) > 0 {
return configFiles
}
if workingDir, err := os.Getwd(); err == nil {
configFile := filepath.Join(workingDir, "config.json")
if fileExists(configFile) {
if verbose {
log.Println("Using default config: ", configFile)
}
2020-11-25 13:01:53 +02:00
return cmdarg.Arg{configFile}
}
}
if configFile := platform.GetConfigurationPath(); fileExists(configFile) {
if verbose {
log.Println("Using config from env: ", configFile)
}
2020-11-25 13:01:53 +02:00
return cmdarg.Arg{configFile}
}
if verbose {
log.Println("Using config from STDIN")
}
2020-11-25 13:01:53 +02:00
return cmdarg.Arg{"stdin:"}
}
func getConfigFormat() string {
f := core.GetFormatByExtension(*format)
if f == "" {
2021-03-21 11:13:51 +02:00
f = "auto"
2020-11-25 13:01:53 +02:00
}
return f
2020-11-25 13:01:53 +02:00
}
func startXray() (core.Server, error) {
configFiles := getConfigFilePath(true)
2020-11-25 13:01:53 +02:00
2021-10-19 19:57:14 +03:00
// config, err := core.LoadConfig(getConfigFormat(), configFiles[0], configFiles)
2020-12-24 21:30:26 +02:00
c, err := core.LoadConfig(getConfigFormat(), configFiles)
2020-11-25 13:01:53 +02:00
if err != nil {
return nil, errors.New("failed to load config files: [", configFiles.String(), "]").Base(err)
2020-11-25 13:01:53 +02:00
}
server, err := core.New(c)
2020-11-25 13:01:53 +02:00
if err != nil {
return nil, errors.New("failed to create server").Base(err)
2020-11-25 13:01:53 +02:00
}
return server, nil
}