mirrors/dae
1
0
Fork 0
mirror of https://github.com/daeuniverse/dae.git synced 2025-07-18 20:00:28 +07:00
Code Issues Packages Projects Releases Wiki Activity

feat: support ToS routing rule (#292)

Browse Source
This commit is contained in:
mzz
2023-08-20 01:55:42 +08:00
committed by GitHub
parent c784ebaf53
commit 7273be6a06
14 changed files with 110 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
component/routing/function_parser.go
View File

@ -6,13 +6,17 @@
package routing
import (
"encoding/binary"
"fmt"
"net/netip"
"strconv"
"strings"
"github.com/daeuniverse/dae/common"
"github.com/daeuniverse/dae/common/consts"
"github.com/daeuniverse/dae/pkg/config_parser"
"github.com/sirupsen/logrus"
"net/netip"
"strings"
"golang.org/x/exp/constraints"
)
type FunctionParser func(log *logrus.Logger, f *config_parser.Function, key string, paramValueGroup []string, overrideOutbound *Outbound) (err error)
@ -137,3 +141,18 @@ func toProcessName(processName string) (procName [consts.TaskCommLen]byte) {
copy(procName[:], n)
return procName
}
func UintParserFactory[T constraints.Unsigned](callback func(f *config_parser.Function, values []T, overrideOutbound *Outbound) (err error)) FunctionParser {
size := binary.Size(new(T))
return func(log *logrus.Logger, f *config_parser.Function, key string, paramValueGroup []string, overrideOutbound *Outbound) (err error) {
var values []T
for _, v := range paramValueGroup {
val, err := strconv.ParseUint(v, 10, 8*size)
if err != nil {
return fmt.Errorf("cannot parse %v: %w", v, err)
}
values = append(values, T(val))
}
return callback(f, values, overrideOutbound)
}
}