1
0

rule_item_process_path_regex.go 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. package rule
  2. import (
  3. "regexp"
  4. "strings"
  5. "github.com/sagernet/sing-box/adapter"
  6. E "github.com/sagernet/sing/common/exceptions"
  7. F "github.com/sagernet/sing/common/format"
  8. )
  9. var _ RuleItem = (*ProcessPathRegexItem)(nil)
  10. type ProcessPathRegexItem struct {
  11. matchers []*regexp.Regexp
  12. description string
  13. }
  14. func NewProcessPathRegexItem(expressions []string) (*ProcessPathRegexItem, error) {
  15. matchers := make([]*regexp.Regexp, 0, len(expressions))
  16. for i, regex := range expressions {
  17. matcher, err := regexp.Compile(regex)
  18. if err != nil {
  19. return nil, E.Cause(err, "parse expression ", i)
  20. }
  21. matchers = append(matchers, matcher)
  22. }
  23. description := "process_path_regex="
  24. eLen := len(expressions)
  25. if eLen == 1 {
  26. description += expressions[0]
  27. } else if eLen > 3 {
  28. description += F.ToString("[", strings.Join(expressions[:3], " "), "]")
  29. } else {
  30. description += F.ToString("[", strings.Join(expressions, " "), "]")
  31. }
  32. return &ProcessPathRegexItem{matchers, description}, nil
  33. }
  34. func (r *ProcessPathRegexItem) Match(metadata *adapter.InboundContext) bool {
  35. if metadata.ProcessInfo == nil || metadata.ProcessInfo.ProcessPath == "" {
  36. return false
  37. }
  38. for _, matcher := range r.matchers {
  39. if matcher.MatchString(metadata.ProcessInfo.ProcessPath) {
  40. return true
  41. }
  42. }
  43. return false
  44. }
  45. func (r *ProcessPathRegexItem) String() string {
  46. return r.description
  47. }