mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
912 B
48 lines
912 B
package rules
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/v2ray/v2ray-core/app/router"
|
|
"github.com/v2ray/v2ray-core/app/router/rules/config"
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
)
|
|
|
|
var (
|
|
InvalidRule = errors.New("Invalid Rule")
|
|
NoRuleApplicable = errors.New("No rule applicable")
|
|
)
|
|
|
|
type Router struct {
|
|
rules []config.Rule
|
|
}
|
|
|
|
func (this *Router) TakeDetour(dest v2net.Destination) (string, error) {
|
|
for _, rule := range this.rules {
|
|
if rule.Apply(dest) {
|
|
return rule.Tag(), nil
|
|
}
|
|
}
|
|
return "", NoRuleApplicable
|
|
}
|
|
|
|
type RouterFactory struct {
|
|
}
|
|
|
|
func (this *RouterFactory) Create(rawConfig interface{}) (router.Router, error) {
|
|
config := rawConfig.(config.RouterRuleConfig)
|
|
rules := config.Rules()
|
|
for _, rule := range rules {
|
|
if rule == nil {
|
|
return nil, InvalidRule
|
|
}
|
|
}
|
|
return &Router{
|
|
rules: rules,
|
|
}, nil
|
|
}
|
|
|
|
func init() {
|
|
router.RegisterRouter("rules", &RouterFactory{})
|
|
}
|