mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 18:17:52 -05:00
165 lines
3.7 KiB
Go
165 lines
3.7 KiB
Go
package main
|
|
|
|
//go:generate go run github.com/v2fly/v2ray-core/v4/common/errors/errorgen
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"path"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"syscall"
|
|
|
|
core "github.com/v2fly/v2ray-core/v4"
|
|
"github.com/v2fly/v2ray-core/v4/common/cmdarg"
|
|
"github.com/v2fly/v2ray-core/v4/common/platform"
|
|
_ "github.com/v2fly/v2ray-core/v4/main/distro/all"
|
|
)
|
|
|
|
var (
|
|
configFiles cmdarg.Arg // "Config file for V2Ray.", the option is customed type, parse in main
|
|
configDir string
|
|
version = flag.Bool("version", false, "Show current version of V2Ray.")
|
|
test = flag.Bool("test", false, "Test config file only, without launching V2Ray server.")
|
|
format = flag.String("format", "json", "Format of input file.")
|
|
|
|
/* 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() error { // nolint: unparam
|
|
flag.Var(&configFiles, "config", "Config file for V2Ray. Multiple assign is accepted (only json). Latter ones overrides the former ones.")
|
|
flag.Var(&configFiles, "c", "Short alias of -config")
|
|
flag.StringVar(&configDir, "confdir", "", "A dir with multiple json config")
|
|
|
|
return nil
|
|
}()
|
|
)
|
|
|
|
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()
|
|
}
|
|
|
|
func readConfDir(dirPath string) {
|
|
confs, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
for _, f := range confs {
|
|
if strings.HasSuffix(f.Name(), ".json") {
|
|
configFiles.Set(path.Join(dirPath, f.Name()))
|
|
}
|
|
}
|
|
}
|
|
|
|
func getConfigFilePath() cmdarg.Arg {
|
|
if dirExists(configDir) {
|
|
log.Println("Using confdir from arg:", configDir)
|
|
readConfDir(configDir)
|
|
} else if envConfDir := platform.GetConfDirPath(); dirExists(envConfDir) {
|
|
log.Println("Using confdir from env:", envConfDir)
|
|
readConfDir(envConfDir)
|
|
}
|
|
|
|
if len(configFiles) > 0 {
|
|
return configFiles
|
|
}
|
|
|
|
if workingDir, err := os.Getwd(); err == nil {
|
|
configFile := filepath.Join(workingDir, "config.json")
|
|
if fileExists(configFile) {
|
|
log.Println("Using default config: ", configFile)
|
|
return cmdarg.Arg{configFile}
|
|
}
|
|
}
|
|
|
|
if configFile := platform.GetConfigurationPath(); fileExists(configFile) {
|
|
log.Println("Using config from env: ", configFile)
|
|
return cmdarg.Arg{configFile}
|
|
}
|
|
|
|
log.Println("Using config from STDIN")
|
|
return cmdarg.Arg{"stdin:"}
|
|
}
|
|
|
|
func GetConfigFormat() string {
|
|
switch strings.ToLower(*format) {
|
|
case "pb", "protobuf":
|
|
return "protobuf"
|
|
default:
|
|
return "json"
|
|
}
|
|
}
|
|
|
|
func startV2Ray() (core.Server, error) {
|
|
configFiles := getConfigFilePath()
|
|
|
|
config, err := core.LoadConfig(GetConfigFormat(), configFiles[0], configFiles)
|
|
if err != nil {
|
|
return nil, newError("failed to read config files: [", configFiles.String(), "]").Base(err)
|
|
}
|
|
|
|
server, err := core.New(config)
|
|
if err != nil {
|
|
return nil, newError("failed to create server").Base(err)
|
|
}
|
|
|
|
return server, nil
|
|
}
|
|
|
|
func printVersion() {
|
|
version := core.VersionStatement()
|
|
for _, s := range version {
|
|
fmt.Println(s)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
printVersion()
|
|
|
|
if *version {
|
|
return
|
|
}
|
|
|
|
server, err := startV2Ray()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
// Configuration error. Exit with a special value to prevent systemd from restarting.
|
|
os.Exit(23)
|
|
}
|
|
|
|
if *test {
|
|
fmt.Println("Configuration OK.")
|
|
os.Exit(0)
|
|
}
|
|
|
|
if err := server.Start(); err != nil {
|
|
fmt.Println("Failed to start", err)
|
|
os.Exit(-1)
|
|
}
|
|
defer server.Close()
|
|
|
|
// Explicitly triggering GC to remove garbage from config loading.
|
|
runtime.GC()
|
|
|
|
{
|
|
osSignals := make(chan os.Signal, 1)
|
|
signal.Notify(osSignals, os.Interrupt, syscall.SIGTERM)
|
|
<-osSignals
|
|
}
|
|
}
|