-
Notifications
You must be signed in to change notification settings - Fork 305
/
domain_matcher.go
103 lines (90 loc) · 2.94 KB
/
domain_matcher.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*
* Copyright (C) 2020-2022, IrineSistiana
*
* This file is part of mosdns.
*
* mosdns is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* mosdns is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package base_domain
import (
"context"
"fmt"
"github.com/IrineSistiana/mosdns/v5/pkg/matcher/domain"
"github.com/IrineSistiana/mosdns/v5/pkg/query_context"
"github.com/IrineSistiana/mosdns/v5/plugin/data_provider"
"github.com/IrineSistiana/mosdns/v5/plugin/data_provider/domain_set"
"github.com/IrineSistiana/mosdns/v5/plugin/executable/sequence"
"strings"
)
var _ sequence.Matcher = (*Matcher)(nil)
type Args struct {
Exps []string `yaml:"exps"`
DomainSets []string `yaml:"domain_sets"`
Files []string `yaml:"files"`
}
type MatchFunc func(qCtx *query_context.Context, m domain.Matcher[struct{}]) (bool, error)
type Matcher struct {
match MatchFunc
mg []domain.Matcher[struct{}]
}
func (m *Matcher) Match(_ context.Context, qCtx *query_context.Context) (bool, error) {
return m.match(qCtx, domain_set.MatcherGroup(m.mg))
}
func NewMatcher(bq sequence.BQ, args *Args, f MatchFunc) (m *Matcher, err error) {
m = &Matcher{
match: f,
}
// Acquire matchers from other plugins.
for _, tag := range args.DomainSets {
p := bq.M().GetPlugin(tag)
dsProvider, _ := p.(data_provider.DomainMatcherProvider)
if dsProvider == nil {
return nil, fmt.Errorf("cannot find domain set %s", tag)
}
dm := dsProvider.GetDomainMatcher()
m.mg = append(m.mg, dm)
}
// Anonymous set from plugin's args and files.
if len(args.Exps)+len(args.Files) > 0 {
anonymousSet := domain.NewDomainMixMatcher()
if err := domain_set.LoadExpsAndFiles(args.Exps, args.Files, anonymousSet); err != nil {
return nil, err
}
if anonymousSet.Len() > 0 {
m.mg = append(m.mg, anonymousSet)
}
}
return m, nil
}
// ParseQuickSetupArgs parses expressions and domain set to args.
// Format: "([exp] | [$domain_set_tag] | [&domain_list_file])..."
func ParseQuickSetupArgs(s string) *Args {
cutPrefix := func(s string, p string) (string, bool) {
if strings.HasPrefix(s, p) {
return strings.TrimPrefix(s, p), true
}
return s, false
}
args := new(Args)
for _, exp := range strings.Fields(s) {
if tag, ok := cutPrefix(exp, "$"); ok {
args.DomainSets = append(args.DomainSets, tag)
} else if path, ok := cutPrefix(exp, "&"); ok {
args.Files = append(args.Files, path)
} else {
args.Exps = append(args.Exps, exp)
}
}
return args
}