mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-10-07 11:33:59 -04:00
212 lines
5.0 KiB
Go
212 lines
5.0 KiB
Go
package commands
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"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/commands/base"
|
|
)
|
|
|
|
// CmdRun runs V2Ray with config
|
|
var CmdRun = &base.Command{
|
|
CustomFlags: true,
|
|
UsageLine: "{{.Exec}} run [-c config.json] [-d dir]",
|
|
Short: "run V2Ray with config",
|
|
Long: `
|
|
Run V2Ray with config.
|
|
|
|
{{.Exec}} will also use the config directory specified by environment
|
|
variable "v2ray.location.confdir". If no config found, it tries
|
|
to load config from one of below:
|
|
|
|
1. The default "config.json" in the current directory
|
|
2. The config file from ENV "v2ray.location.config"
|
|
3. The stdin if all failed above
|
|
|
|
Arguments:
|
|
|
|
-c, -config <file>
|
|
Config file for V2Ray. Multiple assign is accepted.
|
|
|
|
-d, -confdir <dir>
|
|
A directory with config files. Multiple assign is accepted.
|
|
|
|
-r
|
|
Load confdir recursively.
|
|
|
|
-format <format>
|
|
Format of config input. (default "auto")
|
|
|
|
Examples:
|
|
|
|
{{.Exec}} {{.LongName}} -c config.json
|
|
{{.Exec}} {{.LongName}} -d path/to/dir
|
|
|
|
Use "{{.Exec}} help format-loader" for more information about format.
|
|
`,
|
|
Run: executeRun,
|
|
}
|
|
|
|
var (
|
|
configFiles cmdarg.Arg
|
|
configDirs cmdarg.Arg
|
|
configFormat *string
|
|
configDirRecursively *bool
|
|
)
|
|
|
|
func setConfigFlags(cmd *base.Command) {
|
|
configFormat = cmd.Flag.String("format", core.FormatAuto, "")
|
|
configDirRecursively = cmd.Flag.Bool("r", false, "")
|
|
|
|
cmd.Flag.Var(&configFiles, "config", "")
|
|
cmd.Flag.Var(&configFiles, "c", "")
|
|
cmd.Flag.Var(&configDirs, "confdir", "")
|
|
cmd.Flag.Var(&configDirs, "d", "")
|
|
}
|
|
|
|
func executeRun(cmd *base.Command, args []string) {
|
|
setConfigFlags(cmd)
|
|
cmd.Flag.Parse(args)
|
|
printVersion()
|
|
configFiles = getConfigFilePath()
|
|
server, err := startV2Ray()
|
|
if err != nil {
|
|
base.Fatalf("Failed to start: %s", err)
|
|
}
|
|
|
|
if err := server.Start(); err != nil {
|
|
base.Fatalf("Failed to start: %s", err)
|
|
}
|
|
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
|
|
}
|
|
}
|
|
|
|
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, extension []string) cmdarg.Arg {
|
|
confs, err := ioutil.ReadDir(dirPath)
|
|
if err != nil {
|
|
base.Fatalf("failed to read dir %s: %s", dirPath, err)
|
|
}
|
|
files := make(cmdarg.Arg, 0)
|
|
for _, f := range confs {
|
|
ext := filepath.Ext(f.Name())
|
|
for _, e := range extension {
|
|
if strings.EqualFold(e, ext) {
|
|
files.Set(filepath.Join(dirPath, f.Name()))
|
|
break
|
|
}
|
|
}
|
|
}
|
|
return files
|
|
}
|
|
|
|
// getFolderFiles get files in the folder and it's children
|
|
func readConfDirRecursively(dirPath string, extension []string) cmdarg.Arg {
|
|
files := make(cmdarg.Arg, 0)
|
|
err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
|
|
ext := filepath.Ext(path)
|
|
for _, e := range extension {
|
|
if strings.EqualFold(e, ext) {
|
|
files.Set(path)
|
|
break
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
base.Fatalf("failed to read dir %s: %s", dirPath, err)
|
|
}
|
|
return files
|
|
}
|
|
|
|
func getConfigFilePath() cmdarg.Arg {
|
|
extension, err := core.GetLoaderExtensions(*configFormat)
|
|
if err != nil {
|
|
base.Fatalf(err.Error())
|
|
}
|
|
dirReader := readConfDir
|
|
if *configDirRecursively {
|
|
dirReader = readConfDirRecursively
|
|
}
|
|
if len(configDirs) > 0 {
|
|
for _, d := range configDirs {
|
|
log.Println("Using confdir from arg:", d)
|
|
configFiles = append(configFiles, dirReader(d, extension)...)
|
|
}
|
|
} else if envConfDir := platform.GetConfDirPath(); dirExists(envConfDir) {
|
|
log.Println("Using confdir from env:", envConfDir)
|
|
configFiles = append(configFiles, dirReader(envConfDir, extension)...)
|
|
}
|
|
if len(configFiles) > 0 {
|
|
return configFiles
|
|
}
|
|
|
|
if len(configFiles) == 0 && len(configDirs) > 0 {
|
|
base.Fatalf("no config file found with extension: %s", extension)
|
|
}
|
|
|
|
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}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func startV2Ray() (core.Server, error) {
|
|
config, err := core.LoadConfig(*configFormat, configFiles)
|
|
if err != nil {
|
|
if len(configFiles) == 0 {
|
|
err = newError("failed to load config").Base(err)
|
|
} else {
|
|
err = newError(fmt.Sprintf("failed to load config: %s", configFiles)).Base(err)
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
server, err := core.New(config)
|
|
if err != nil {
|
|
return nil, newError("failed to create server").Base(err)
|
|
}
|
|
|
|
return server, nil
|
|
}
|