2015-09-12 05:51:42 -04:00
|
|
|
package main
|
|
|
|
|
2015-09-12 14:36:21 -04:00
|
|
|
import (
|
|
|
|
"flag"
|
2015-09-18 05:59:36 -04:00
|
|
|
"fmt"
|
2016-10-03 05:18:24 -04:00
|
|
|
"io"
|
2015-10-18 07:14:34 -04:00
|
|
|
"os"
|
2016-05-10 18:51:38 -04:00
|
|
|
"os/signal"
|
2015-10-18 07:14:34 -04:00
|
|
|
"path/filepath"
|
2016-10-16 08:22:21 -04:00
|
|
|
"strings"
|
2016-08-09 16:10:44 -04:00
|
|
|
"syscall"
|
2015-09-12 14:36:21 -04:00
|
|
|
|
2016-08-20 14:55:45 -04:00
|
|
|
"v2ray.com/core"
|
2017-02-01 15:47:02 -05:00
|
|
|
"v2ray.com/core/app/log"
|
2016-12-16 17:38:58 -05:00
|
|
|
|
|
|
|
_ "v2ray.com/core/main/distro/all"
|
2015-09-12 14:36:21 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
2015-10-18 07:14:34 -04:00
|
|
|
configFile string
|
2015-09-18 05:59:36 -04:00
|
|
|
version = flag.Bool("version", false, "Show current version of V2Ray.")
|
2016-01-24 06:58:04 -05:00
|
|
|
test = flag.Bool("test", false, "Test config file only, without launching V2Ray server.")
|
2016-10-03 05:18:24 -04:00
|
|
|
format = flag.String("format", "json", "Format of input file.")
|
2015-09-12 14:36:21 -04:00
|
|
|
)
|
|
|
|
|
2015-10-18 07:14:34 -04:00
|
|
|
func init() {
|
|
|
|
defaultConfigFile := ""
|
|
|
|
workingDir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
|
|
|
if err == nil {
|
|
|
|
defaultConfigFile = filepath.Join(workingDir, "config.json")
|
|
|
|
}
|
|
|
|
flag.StringVar(&configFile, "config", defaultConfigFile, "Config file for this Point server.")
|
|
|
|
}
|
|
|
|
|
2016-10-16 08:22:21 -04:00
|
|
|
func GetConfigFormat() core.ConfigFormat {
|
|
|
|
switch strings.ToLower(*format) {
|
|
|
|
case "json":
|
|
|
|
return core.ConfigFormat_JSON
|
|
|
|
case "pb", "protobuf":
|
|
|
|
return core.ConfigFormat_Protobuf
|
|
|
|
default:
|
|
|
|
return core.ConfigFormat_JSON
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-10-12 10:46:02 -04:00
|
|
|
func startV2Ray() *core.Point {
|
2015-10-18 07:14:34 -04:00
|
|
|
if len(configFile) == 0 {
|
2015-10-07 04:05:11 -04:00
|
|
|
log.Error("Config file is not set.")
|
2016-07-10 17:11:42 -04:00
|
|
|
return nil
|
2015-09-12 14:36:21 -04:00
|
|
|
}
|
2016-10-03 05:18:24 -04:00
|
|
|
var configInput io.Reader
|
|
|
|
if configFile == "stdin:" {
|
|
|
|
configInput = os.Stdin
|
|
|
|
} else {
|
|
|
|
fixedFile := os.ExpandEnv(configFile)
|
|
|
|
file, err := os.Open(fixedFile)
|
|
|
|
if err != nil {
|
|
|
|
log.Error("Config file not readable: ", err)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
defer file.Close()
|
|
|
|
configInput = file
|
|
|
|
}
|
2016-10-16 08:22:21 -04:00
|
|
|
config, err := core.LoadConfig(GetConfigFormat(), configInput)
|
2015-09-12 14:36:21 -04:00
|
|
|
if err != nil {
|
2016-01-18 06:24:33 -05:00
|
|
|
log.Error("Failed to read config file (", configFile, "): ", configFile, err)
|
2016-07-10 17:11:42 -04:00
|
|
|
return nil
|
2015-09-12 14:36:21 -04:00
|
|
|
}
|
|
|
|
|
2016-10-12 10:46:02 -04:00
|
|
|
vPoint, err := core.NewPoint(config)
|
2015-09-12 14:36:21 -04:00
|
|
|
if err != nil {
|
2016-01-18 06:24:33 -05:00
|
|
|
log.Error("Failed to create Point server: ", err)
|
2016-07-10 17:11:42 -04:00
|
|
|
return nil
|
2015-09-12 14:36:21 -04:00
|
|
|
}
|
|
|
|
|
2016-01-24 06:58:04 -05:00
|
|
|
if *test {
|
|
|
|
fmt.Println("Configuration OK.")
|
2016-07-10 17:11:42 -04:00
|
|
|
return nil
|
2016-01-24 06:58:04 -05:00
|
|
|
}
|
|
|
|
|
2015-09-12 14:36:21 -04:00
|
|
|
err = vPoint.Start()
|
|
|
|
if err != nil {
|
2016-01-18 06:24:33 -05:00
|
|
|
log.Error("Error starting Point server: ", err)
|
2016-07-10 17:11:42 -04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
return vPoint
|
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
core.PrintVersion()
|
|
|
|
|
|
|
|
if *version {
|
2015-10-07 04:05:11 -04:00
|
|
|
return
|
2015-09-12 14:36:21 -04:00
|
|
|
}
|
2015-09-12 05:51:42 -04:00
|
|
|
|
2016-07-10 17:11:42 -04:00
|
|
|
if point := startV2Ray(); point != nil {
|
|
|
|
osSignals := make(chan os.Signal, 1)
|
2016-08-09 16:10:44 -04:00
|
|
|
signal.Notify(osSignals, os.Interrupt, os.Kill, syscall.SIGTERM)
|
2016-05-10 18:51:38 -04:00
|
|
|
|
2016-07-10 17:11:42 -04:00
|
|
|
<-osSignals
|
|
|
|
point.Close()
|
|
|
|
}
|
2015-09-12 05:51:42 -04:00
|
|
|
}
|