mirror of
https://github.com/nadoo/glider.git
synced 2025-02-23 09:25:41 +08:00
93 lines
3.0 KiB
Go
93 lines
3.0 KiB
Go
package rule
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/nadoo/conflag"
|
|
)
|
|
|
|
// Config is config of rule.
|
|
type Config struct {
|
|
Name string
|
|
|
|
Forward []string
|
|
StrategyConfig StrategyConfig
|
|
|
|
DNSServers []string
|
|
IPSet string
|
|
|
|
Domain []string
|
|
IP []string
|
|
CIDR []string
|
|
}
|
|
|
|
// StrategyConfig is config of strategy.
|
|
type StrategyConfig struct {
|
|
Strategy string
|
|
CheckType string
|
|
CheckAddr string
|
|
CheckInterval int
|
|
CheckTimeout int
|
|
CheckTolerance int
|
|
CheckDisabledOnly bool
|
|
MaxFailures int
|
|
DialTimeout int
|
|
RelayTimeout int
|
|
IntFace string
|
|
}
|
|
|
|
// NewConfFromFile returns a new config from file.
|
|
func NewConfFromFile(ruleFile string) (*Config, error) {
|
|
p := &Config{Name: ruleFile}
|
|
|
|
f := conflag.NewFromFile("rule", ruleFile)
|
|
f.StringSliceUniqVar(&p.Forward, "forward", nil, "forward url, format: SCHEME://[USER|METHOD:PASSWORD@][HOST]:PORT?PARAMS[,SCHEME://[USER|METHOD:PASSWORD@][HOST]:PORT?PARAMS]")
|
|
f.StringVar(&p.StrategyConfig.Strategy, "strategy", "rr", "forward strategy, default: rr")
|
|
f.StringVar(&p.StrategyConfig.CheckType, "checktype", "http", "fowarder check type, http/tcp, default: http")
|
|
f.StringVar(&p.StrategyConfig.CheckAddr, "checkaddr", "www.apple.com:80", "fowarder check addr, format: HOST[:PORT], default port: 80,")
|
|
f.IntVar(&p.StrategyConfig.CheckInterval, "checkinterval", 30, "fowarder check interval(seconds)")
|
|
f.IntVar(&p.StrategyConfig.CheckTimeout, "checktimeout", 10, "fowarder check timeout(seconds)")
|
|
f.IntVar(&p.StrategyConfig.CheckTolerance, "checktolerance", 0, "fowarder check tolerance(ms), switch only when new_latency < old_latency - tolerance, only used in lha mode")
|
|
f.BoolVar(&p.StrategyConfig.CheckDisabledOnly, "checkdisabledonly", false, "check disabled fowarders only")
|
|
f.IntVar(&p.StrategyConfig.MaxFailures, "maxfailures", 3, "max failures to change forwarder status to disabled")
|
|
f.IntVar(&p.StrategyConfig.DialTimeout, "dialtimeout", 3, "dial timeout(seconds)")
|
|
f.IntVar(&p.StrategyConfig.RelayTimeout, "relaytimeout", 0, "relay timeout(seconds)")
|
|
f.StringVar(&p.StrategyConfig.IntFace, "interface", "", "source ip or source interface")
|
|
|
|
f.StringSliceUniqVar(&p.DNSServers, "dnsserver", nil, "remote dns server")
|
|
f.StringVar(&p.IPSet, "ipset", "", "ipset name")
|
|
|
|
f.StringSliceUniqVar(&p.Domain, "domain", nil, "domain")
|
|
f.StringSliceUniqVar(&p.IP, "ip", nil, "ip")
|
|
f.StringSliceUniqVar(&p.CIDR, "cidr", nil, "cidr")
|
|
|
|
err := f.Parse()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return p, err
|
|
}
|
|
|
|
// ListDir returns file list named with suffix in dirPth.
|
|
func ListDir(dirPth string, suffix string) (files []string, err error) {
|
|
files = make([]string, 0, 10)
|
|
dir, err := ioutil.ReadDir(dirPth)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
PthSep := string(os.PathSeparator)
|
|
suffix = strings.ToLower(suffix)
|
|
for _, fi := range dir {
|
|
if fi.IsDir() {
|
|
continue
|
|
}
|
|
if strings.HasSuffix(strings.ToLower(fi.Name()), suffix) {
|
|
files = append(files, dirPth+PthSep+fi.Name())
|
|
}
|
|
}
|
|
return files, nil
|
|
}
|