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

feat: add config_parser

Browse Source
This commit is contained in:
mzz2017
2023-01-27 02:10:27 +08:00
parent 916a55d480
commit edbce81e88
53 changed files with 6696 additions and 733 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
component/control/routing_matcher_builder.go
View File

@ -20,6 +20,7 @@ type DomainSet struct {
RuleIndex int
Domains []string
}
type RoutingMatcherBuilder struct {
*routing.DefaultMatcherBuilder
outboundName2Id map[string]uint8
@ -74,7 +75,7 @@ func (b *RoutingMatcherBuilder) AddDomain(key string, values []string, outbound
})
}
func (b *RoutingMatcherBuilder) AddMac(macAddrs [][6]byte, outbound string) {
func (b *RoutingMatcherBuilder) AddSourceMac(macAddrs [][6]byte, outbound string) {
if b.err != nil {
return
}
@ -108,7 +109,7 @@ func (b *RoutingMatcherBuilder) AddIp(values []netip.Prefix, outbound string) {
})
}
func (b *RoutingMatcherBuilder) AddSource(values []netip.Prefix, outbound string) {
func (b *RoutingMatcherBuilder) AddSourceIp(values []netip.Prefix, outbound string) {
if b.err != nil {
return
}
@ -166,9 +167,9 @@ func (b *RoutingMatcherBuilder) Build() (err error) {
keys = append(keys, cidrToBpfLpmKey(cidr))
values = append(values, 1)
}
m, err := b.bpf.NewLpmMap(keys, values)
m, err := b.bpf.newLpmMap(keys, values)
if err != nil {
return fmt.Errorf("NewLpmMap: %w", err)
return fmt.Errorf("newLpmMap: %w", err)
}
// ebpf.Map cannot be BatchUpdate
if err = b.bpf.LpmArrayMap.Update(uint32(i), m, ebpf.UpdateAny); err != nil {
@ -177,7 +178,12 @@ func (b *RoutingMatcherBuilder) Build() (err error) {
}
m.Close()
}
// Update routings.
// Write routings.
// Final rule MUST be the last.
if b.rules[len(b.rules)-1].Type != uint8(consts.RoutingType_Final) {
b.err = fmt.Errorf("final rule MUST be the last")
return b.err
}
routingsLen := uint32(len(b.rules))
routingsKeys := common.ARangeU32(routingsLen)
if _, err = b.bpf.RoutingMap.BatchUpdate(routingsKeys, b.rules, &ebpf.BatchOptions{
@ -185,8 +191,5 @@ func (b *RoutingMatcherBuilder) Build() (err error) {
}); err != nil {
return fmt.Errorf("BatchUpdate: %w", err)
}
if err = b.bpf.ParamMap.Update(consts.RoutingsLenKey, routingsLen, ebpf.UpdateAny); err != nil {
return fmt.Errorf("Update: %w", err)
}
return nil
}