mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-09-06 09:02:32 +00:00
98 lines
2 KiB
Go
98 lines
2 KiB
Go
package common
|
|
|
|
import (
|
|
"github.com/Dreamacro/clash/common/utils"
|
|
"github.com/Dreamacro/clash/component/process"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
"github.com/Dreamacro/clash/log"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type Uid struct {
|
|
*Base
|
|
uids []utils.Range[int32]
|
|
oUid string
|
|
adapter string
|
|
}
|
|
|
|
func NewUid(oUid, adapter string) (*Uid, error) {
|
|
//if len(_uids) > 28 {
|
|
// return nil, fmt.Errorf("%s, too many uid to use, maximum support 28 uid", errPayload.Error())
|
|
//}
|
|
|
|
var uidRange []utils.Range[int32]
|
|
for _, u := range strings.Split(oUid, "/") {
|
|
if u == "" {
|
|
continue
|
|
}
|
|
|
|
subUids := strings.Split(u, "-")
|
|
subUidsLen := len(subUids)
|
|
if subUidsLen > 2 {
|
|
return nil, errPayload
|
|
}
|
|
|
|
uidStart, err := strconv.ParseUint(strings.Trim(subUids[0], "[ ]"), 10, 32)
|
|
if err != nil {
|
|
return nil, errPayload
|
|
}
|
|
|
|
switch subUidsLen {
|
|
case 1:
|
|
uidRange = append(uidRange, *utils.NewRange(int32(uidStart), int32(uidStart)))
|
|
case 2:
|
|
uidEnd, err := strconv.ParseUint(strings.Trim(subUids[1], "[ ]"), 10, 32)
|
|
if err != nil {
|
|
return nil, errPayload
|
|
}
|
|
|
|
uidRange = append(uidRange, *utils.NewRange(int32(uidStart), int32(uidEnd)))
|
|
}
|
|
}
|
|
|
|
if len(uidRange) == 0 {
|
|
return nil, errPayload
|
|
}
|
|
return &Uid{
|
|
Base: &Base{},
|
|
adapter: adapter,
|
|
oUid: oUid,
|
|
uids: uidRange,
|
|
}, nil
|
|
}
|
|
|
|
func (u *Uid) RuleType() C.RuleType {
|
|
return C.Uid
|
|
}
|
|
|
|
func (u *Uid) Match(metadata *C.Metadata) bool {
|
|
srcPort, err := strconv.Atoi(metadata.SrcPort)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
var uid int32
|
|
if metadata.Uid != 0 {
|
|
uid = metadata.Uid
|
|
} else if uid, err = process.FindUid(metadata.NetWork.String(), metadata.SrcIP, srcPort); err == nil {
|
|
metadata.Uid = uid
|
|
} else {
|
|
log.Warnln("[UID] could not get uid from %s", metadata.String())
|
|
return false
|
|
}
|
|
|
|
for _, _uid := range u.uids {
|
|
if _uid.Contains(uid) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (u *Uid) Adapter() string {
|
|
return u.adapter
|
|
}
|
|
|
|
func (u *Uid) Payload() string {
|
|
return u.oUid
|
|
}
|