mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-11-22 08:31:30 +00:00
Add source_port_range/port_range rule item
This commit is contained in:
parent
af19ba6119
commit
7d340e7ef9
|
@ -102,7 +102,9 @@ type DefaultDNSRule struct {
|
||||||
SourceGeoIP Listable[string] `json:"source_geoip,omitempty"`
|
SourceGeoIP Listable[string] `json:"source_geoip,omitempty"`
|
||||||
SourceIPCIDR Listable[string] `json:"source_ip_cidr,omitempty"`
|
SourceIPCIDR Listable[string] `json:"source_ip_cidr,omitempty"`
|
||||||
SourcePort Listable[uint16] `json:"source_port,omitempty"`
|
SourcePort Listable[uint16] `json:"source_port,omitempty"`
|
||||||
|
SourcePortRange Listable[string] `json:"source_port_range,omitempty"`
|
||||||
Port Listable[uint16] `json:"port,omitempty"`
|
Port Listable[uint16] `json:"port,omitempty"`
|
||||||
|
PortRange Listable[string] `json:"port_range,omitempty"`
|
||||||
ProcessName Listable[string] `json:"process_name,omitempty"`
|
ProcessName Listable[string] `json:"process_name,omitempty"`
|
||||||
PackageName Listable[string] `json:"package_name,omitempty"`
|
PackageName Listable[string] `json:"package_name,omitempty"`
|
||||||
User Listable[string] `json:"user,omitempty"`
|
User Listable[string] `json:"user,omitempty"`
|
||||||
|
@ -132,7 +134,9 @@ func (r DefaultDNSRule) Equals(other DefaultDNSRule) bool {
|
||||||
common.ComparableSliceEquals(r.SourceGeoIP, other.SourceGeoIP) &&
|
common.ComparableSliceEquals(r.SourceGeoIP, other.SourceGeoIP) &&
|
||||||
common.ComparableSliceEquals(r.SourceIPCIDR, other.SourceIPCIDR) &&
|
common.ComparableSliceEquals(r.SourceIPCIDR, other.SourceIPCIDR) &&
|
||||||
common.ComparableSliceEquals(r.SourcePort, other.SourcePort) &&
|
common.ComparableSliceEquals(r.SourcePort, other.SourcePort) &&
|
||||||
|
common.ComparableSliceEquals(r.SourcePortRange, other.SourcePortRange) &&
|
||||||
common.ComparableSliceEquals(r.Port, other.Port) &&
|
common.ComparableSliceEquals(r.Port, other.Port) &&
|
||||||
|
common.ComparableSliceEquals(r.PortRange, other.PortRange) &&
|
||||||
common.ComparableSliceEquals(r.ProcessName, other.ProcessName) &&
|
common.ComparableSliceEquals(r.ProcessName, other.ProcessName) &&
|
||||||
common.ComparableSliceEquals(r.UserID, other.UserID) &&
|
common.ComparableSliceEquals(r.UserID, other.UserID) &&
|
||||||
common.ComparableSliceEquals(r.PackageName, other.PackageName) &&
|
common.ComparableSliceEquals(r.PackageName, other.PackageName) &&
|
||||||
|
|
|
@ -102,7 +102,9 @@ type DefaultRule struct {
|
||||||
SourceIPCIDR Listable[string] `json:"source_ip_cidr,omitempty"`
|
SourceIPCIDR Listable[string] `json:"source_ip_cidr,omitempty"`
|
||||||
IPCIDR Listable[string] `json:"ip_cidr,omitempty"`
|
IPCIDR Listable[string] `json:"ip_cidr,omitempty"`
|
||||||
SourcePort Listable[uint16] `json:"source_port,omitempty"`
|
SourcePort Listable[uint16] `json:"source_port,omitempty"`
|
||||||
|
SourcePortRange Listable[string] `json:"source_port_range,omitempty"`
|
||||||
Port Listable[uint16] `json:"port,omitempty"`
|
Port Listable[uint16] `json:"port,omitempty"`
|
||||||
|
PortRange Listable[string] `json:"port_range,omitempty"`
|
||||||
ProcessName Listable[string] `json:"process_name,omitempty"`
|
ProcessName Listable[string] `json:"process_name,omitempty"`
|
||||||
PackageName Listable[string] `json:"package_name,omitempty"`
|
PackageName Listable[string] `json:"package_name,omitempty"`
|
||||||
User Listable[string] `json:"user,omitempty"`
|
User Listable[string] `json:"user,omitempty"`
|
||||||
|
@ -133,7 +135,9 @@ func (r DefaultRule) Equals(other DefaultRule) bool {
|
||||||
common.ComparableSliceEquals(r.SourceIPCIDR, other.SourceIPCIDR) &&
|
common.ComparableSliceEquals(r.SourceIPCIDR, other.SourceIPCIDR) &&
|
||||||
common.ComparableSliceEquals(r.IPCIDR, other.IPCIDR) &&
|
common.ComparableSliceEquals(r.IPCIDR, other.IPCIDR) &&
|
||||||
common.ComparableSliceEquals(r.SourcePort, other.SourcePort) &&
|
common.ComparableSliceEquals(r.SourcePort, other.SourcePort) &&
|
||||||
|
common.ComparableSliceEquals(r.SourcePortRange, other.SourcePortRange) &&
|
||||||
common.ComparableSliceEquals(r.Port, other.Port) &&
|
common.ComparableSliceEquals(r.Port, other.Port) &&
|
||||||
|
common.ComparableSliceEquals(r.PortRange, other.PortRange) &&
|
||||||
common.ComparableSliceEquals(r.ProcessName, other.ProcessName) &&
|
common.ComparableSliceEquals(r.ProcessName, other.ProcessName) &&
|
||||||
common.ComparableSliceEquals(r.PackageName, other.PackageName) &&
|
common.ComparableSliceEquals(r.PackageName, other.PackageName) &&
|
||||||
common.ComparableSliceEquals(r.User, other.User) &&
|
common.ComparableSliceEquals(r.User, other.User) &&
|
||||||
|
|
|
@ -148,11 +148,27 @@ func NewDefaultRule(router adapter.Router, logger log.ContextLogger, options opt
|
||||||
rule.items = append(rule.items, item)
|
rule.items = append(rule.items, item)
|
||||||
rule.allItems = append(rule.allItems, item)
|
rule.allItems = append(rule.allItems, item)
|
||||||
}
|
}
|
||||||
|
if len(options.SourcePortRange) > 0 {
|
||||||
|
item, err := NewPortRangeItem(true, options.SourcePortRange)
|
||||||
|
if err != nil {
|
||||||
|
return nil, E.Cause(err, "source_port_range")
|
||||||
|
}
|
||||||
|
rule.items = append(rule.items, item)
|
||||||
|
rule.allItems = append(rule.allItems, item)
|
||||||
|
}
|
||||||
if len(options.Port) > 0 {
|
if len(options.Port) > 0 {
|
||||||
item := NewPortItem(false, options.Port)
|
item := NewPortItem(false, options.Port)
|
||||||
rule.items = append(rule.items, item)
|
rule.items = append(rule.items, item)
|
||||||
rule.allItems = append(rule.allItems, item)
|
rule.allItems = append(rule.allItems, item)
|
||||||
}
|
}
|
||||||
|
if len(options.PortRange) > 0 {
|
||||||
|
item, err := NewPortRangeItem(false, options.PortRange)
|
||||||
|
if err != nil {
|
||||||
|
return nil, E.Cause(err, "port_range")
|
||||||
|
}
|
||||||
|
rule.items = append(rule.items, item)
|
||||||
|
rule.allItems = append(rule.allItems, item)
|
||||||
|
}
|
||||||
if len(options.ProcessName) > 0 {
|
if len(options.ProcessName) > 0 {
|
||||||
item := NewProcessItem(options.ProcessName)
|
item := NewProcessItem(options.ProcessName)
|
||||||
rule.items = append(rule.items, item)
|
rule.items = append(rule.items, item)
|
||||||
|
|
|
@ -121,11 +121,27 @@ func NewDefaultDNSRule(router adapter.Router, logger log.ContextLogger, options
|
||||||
rule.items = append(rule.items, item)
|
rule.items = append(rule.items, item)
|
||||||
rule.allItems = append(rule.allItems, item)
|
rule.allItems = append(rule.allItems, item)
|
||||||
}
|
}
|
||||||
|
if len(options.SourcePortRange) > 0 {
|
||||||
|
item, err := NewPortRangeItem(true, options.SourcePortRange)
|
||||||
|
if err != nil {
|
||||||
|
return nil, E.Cause(err, "source_port_range")
|
||||||
|
}
|
||||||
|
rule.items = append(rule.items, item)
|
||||||
|
rule.allItems = append(rule.allItems, item)
|
||||||
|
}
|
||||||
if len(options.Port) > 0 {
|
if len(options.Port) > 0 {
|
||||||
item := NewPortItem(false, options.Port)
|
item := NewPortItem(false, options.Port)
|
||||||
rule.items = append(rule.items, item)
|
rule.items = append(rule.items, item)
|
||||||
rule.allItems = append(rule.allItems, item)
|
rule.allItems = append(rule.allItems, item)
|
||||||
}
|
}
|
||||||
|
if len(options.PortRange) > 0 {
|
||||||
|
item, err := NewPortRangeItem(false, options.PortRange)
|
||||||
|
if err != nil {
|
||||||
|
return nil, E.Cause(err, "port_range")
|
||||||
|
}
|
||||||
|
rule.items = append(rule.items, item)
|
||||||
|
rule.allItems = append(rule.allItems, item)
|
||||||
|
}
|
||||||
if len(options.ProcessName) > 0 {
|
if len(options.ProcessName) > 0 {
|
||||||
item := NewProcessItem(options.ProcessName)
|
item := NewProcessItem(options.ProcessName)
|
||||||
rule.items = append(rule.items, item)
|
rule.items = append(rule.items, item)
|
||||||
|
|
87
route/rule_port_range.go
Normal file
87
route/rule_port_range.go
Normal file
|
@ -0,0 +1,87 @@
|
||||||
|
package route
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/sagernet/sing-box/adapter"
|
||||||
|
E "github.com/sagernet/sing/common/exceptions"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ErrBadPortRange = E.New("bad port range")
|
||||||
|
|
||||||
|
var _ RuleItem = (*PortRangeItem)(nil)
|
||||||
|
|
||||||
|
type PortRangeItem struct {
|
||||||
|
isSource bool
|
||||||
|
portRanges []string
|
||||||
|
portRangeList []rangeItem
|
||||||
|
}
|
||||||
|
|
||||||
|
type rangeItem struct {
|
||||||
|
start uint16
|
||||||
|
end uint16
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewPortRangeItem(isSource bool, rangeList []string) (*PortRangeItem, error) {
|
||||||
|
portRangeList := make([]rangeItem, 0, len(rangeList))
|
||||||
|
for _, portRange := range rangeList {
|
||||||
|
if !strings.Contains(portRange, ":") {
|
||||||
|
return nil, E.Extend(ErrBadPortRange, portRange)
|
||||||
|
}
|
||||||
|
subIndex := strings.Index(portRange, ":")
|
||||||
|
var start, end uint64
|
||||||
|
var err error
|
||||||
|
if subIndex > 0 {
|
||||||
|
start, err = strconv.ParseUint(portRange[:subIndex], 10, 16)
|
||||||
|
if err != nil {
|
||||||
|
return nil, E.Cause(err, E.Extend(ErrBadPortRange, portRange))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if subIndex == len(portRange)-1 {
|
||||||
|
end = 0xFF
|
||||||
|
} else {
|
||||||
|
end, err = strconv.ParseUint(portRange[subIndex+1:], 10, 16)
|
||||||
|
if err != nil {
|
||||||
|
return nil, E.Cause(err, E.Extend(ErrBadPortRange, portRange))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
portRangeList = append(portRangeList, rangeItem{uint16(start), uint16(end)})
|
||||||
|
}
|
||||||
|
return &PortRangeItem{
|
||||||
|
isSource: isSource,
|
||||||
|
portRanges: rangeList,
|
||||||
|
portRangeList: portRangeList,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *PortRangeItem) Match(metadata *adapter.InboundContext) bool {
|
||||||
|
var port uint16
|
||||||
|
if r.isSource {
|
||||||
|
port = metadata.Source.Port
|
||||||
|
} else {
|
||||||
|
port = metadata.Destination.Port
|
||||||
|
}
|
||||||
|
for _, portRange := range r.portRangeList {
|
||||||
|
if port >= portRange.start && port <= portRange.end {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *PortRangeItem) String() string {
|
||||||
|
var description string
|
||||||
|
if r.isSource {
|
||||||
|
description = "source_port_range="
|
||||||
|
} else {
|
||||||
|
description = "port_range="
|
||||||
|
}
|
||||||
|
pLen := len(r.portRanges)
|
||||||
|
if pLen == 1 {
|
||||||
|
description += r.portRanges[0]
|
||||||
|
} else {
|
||||||
|
description += "[" + strings.Join(r.portRanges, " ") + "]"
|
||||||
|
}
|
||||||
|
return description
|
||||||
|
}
|
Loading…
Reference in a new issue