package scenarios import ( "os" "path/filepath" _ "github.com/v2ray/v2ray-core/app/router/rules" "github.com/v2ray/v2ray-core/common/log" "github.com/v2ray/v2ray-core/shell/point" // The following are necessary as they register handlers in their init functions. _ "github.com/v2ray/v2ray-core/proxy/blackhole" _ "github.com/v2ray/v2ray-core/proxy/dokodemo" _ "github.com/v2ray/v2ray-core/proxy/freedom" _ "github.com/v2ray/v2ray-core/proxy/http" _ "github.com/v2ray/v2ray-core/proxy/shadowsocks" _ "github.com/v2ray/v2ray-core/proxy/socks" _ "github.com/v2ray/v2ray-core/proxy/vmess/inbound" _ "github.com/v2ray/v2ray-core/proxy/vmess/outbound" ) var ( runningServers = make([]*point.Point, 0, 10) ) func TestFile(filename string) string { return filepath.Join(os.Getenv("GOPATH"), "src", "github.com", "v2ray", "v2ray-core", "testing", "scenarios", "data", filename) } func InitializeServerSetOnce(testcase string) error { if err := InitializeServerServer(testcase); err != nil { return err } if err := InitializeServerClient(testcase); err != nil { return err } return nil } func InitializeServerServer(testcase string) error { return InitializeServer(TestFile(testcase + "_server.json")) } func InitializeServerClient(testcase string) error { return InitializeServer(TestFile(testcase + "_client.json")) } func InitializeServer(configFile string) error { config, err := point.LoadConfig(configFile) if err != nil { log.Error("Failed to read config file (", configFile, "): ", configFile, err) return err } vPoint, err := point.NewPoint(config) if err != nil { log.Error("Failed to create Point server: ", err) return err } err = vPoint.Start() if err != nil { log.Error("Error starting Point server: ", err) return err } runningServers = append(runningServers, vPoint) return nil } func CloseAllServers() { for _, server := range runningServers { server.Close() } runningServers = make([]*point.Point, 0, 10) }