blob: 475222c776f8acaeb72065207e55f22407b68f3c (
plain)
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
|
// Copyright 2023 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
package subsystem
import (
"maps"
"regexp"
"slices"
"strings"
)
type PathMatcher struct {
matches []*match
}
type match struct {
include *regexp.Regexp
exclude *regexp.Regexp
object *Subsystem
}
func MakePathMatcher(list []*Subsystem) *PathMatcher {
m := &PathMatcher{}
for _, item := range list {
m.register(item)
}
return m
}
func (p *PathMatcher) register(item *Subsystem) {
onlyInclude := []string{}
list := []PathRule{}
for _, r := range item.PathRules {
if r.ExcludeRegexp == "" {
// It's expected that almost everything will go to this branch.
onlyInclude = append(onlyInclude, r.IncludeRegexp)
} else {
list = append(list, r)
}
}
if len(onlyInclude) > 0 {
list = append(list, PathRule{
IncludeRegexp: strings.Join(onlyInclude, "|"),
})
}
for _, rule := range list {
p.matches = append(p.matches, buildMatch(rule, item))
}
}
func (p *PathMatcher) Match(path string) []*Subsystem {
ret := map[*Subsystem]struct{}{}
for _, m := range p.matches {
if m.exclude != nil && m.exclude.MatchString(path) {
continue
}
if m.include != nil && !m.include.MatchString(path) {
continue
}
ret[m.object] = struct{}{}
}
return slices.Collect(maps.Keys(ret))
}
func buildMatch(rule PathRule, item *Subsystem) *match {
m := &match{object: item}
if rule.IncludeRegexp != "" {
m.include = regexp.MustCompile(rule.IncludeRegexp)
}
if rule.ExcludeRegexp != "" {
m.exclude = regexp.MustCompile(rule.ExcludeRegexp)
}
return m
}
|
