mirror of https://github.com/v2ray/v2ray-core
89 lines
2.0 KiB
Go
Executable File
89 lines
2.0 KiB
Go
Executable File
package core
|
|
|
|
import (
|
|
"io"
|
|
"strings"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
"v2ray.com/core/common"
|
|
"v2ray.com/core/common/buf"
|
|
)
|
|
|
|
// ConfigFormat is a configurable format of V2Ray config file.
|
|
type ConfigFormat struct {
|
|
Name string
|
|
Extension []string
|
|
Loader ConfigLoader
|
|
}
|
|
|
|
// ConfigLoader is a utility to load V2Ray config from external source.
|
|
type ConfigLoader func(input io.Reader) (*Config, error)
|
|
|
|
var (
|
|
configLoaderByName = make(map[string]*ConfigFormat)
|
|
configLoaderByExt = make(map[string]*ConfigFormat)
|
|
)
|
|
|
|
// RegisterConfigLoader add a new ConfigLoader.
|
|
func RegisterConfigLoader(format *ConfigFormat) error {
|
|
name := strings.ToLower(format.Name)
|
|
if _, found := configLoaderByName[name]; found {
|
|
return newError(format.Name, " already registered.")
|
|
}
|
|
configLoaderByName[name] = format
|
|
|
|
for _, ext := range format.Extension {
|
|
lext := strings.ToLower(ext)
|
|
if f, found := configLoaderByExt[lext]; found {
|
|
return newError(ext, " already registered to ", f.Name)
|
|
}
|
|
configLoaderByExt[lext] = format
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func getExtension(filename string) string {
|
|
idx := strings.LastIndexByte(filename, '.')
|
|
if idx == -1 {
|
|
return ""
|
|
}
|
|
return filename[idx+1:]
|
|
}
|
|
|
|
// LoadConfig loads config with given format from given source.
|
|
func LoadConfig(formatName string, filename string, input io.Reader) (*Config, error) {
|
|
ext := getExtension(filename)
|
|
if len(ext) > 0 {
|
|
if f, found := configLoaderByExt[ext]; found {
|
|
return f.Loader(input)
|
|
}
|
|
}
|
|
|
|
if f, found := configLoaderByName[formatName]; found {
|
|
return f.Loader(input)
|
|
}
|
|
|
|
return nil, newError("Unable to load config in ", formatName).AtWarning()
|
|
}
|
|
|
|
func loadProtobufConfig(input io.Reader) (*Config, error) {
|
|
config := new(Config)
|
|
data, err := buf.ReadAllToBytes(input)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := proto.Unmarshal(data, config); err != nil {
|
|
return nil, err
|
|
}
|
|
return config, nil
|
|
}
|
|
|
|
func init() {
|
|
common.Must(RegisterConfigLoader(&ConfigFormat{
|
|
Name: "Protobuf",
|
|
Extension: []string{"pb"},
|
|
Loader: loadProtobufConfig,
|
|
}))
|
|
}
|