fix: support to bind dae to wg lan (and other tun) (#221)

daeuniverse · Jul 16, 2023 · b34b5ac · b34b5ac
1 parent ef6dc5d
commit b34b5ac
Show file tree

Hide file tree

Showing 4 changed files with 216 additions and 124 deletions.
diff --git a/common/consts/ebpf.go b/common/consts/ebpf.go
@@ -168,3 +168,8 @@ const (
 	LanWanFlag_IsLan
 	LanWanFlag_NotApplicable
 )
+
+const (
+	LinkType_None uint32 = iota
+	LinkType_Ethernet
+)
diff --git a/control/control_plane.go b/control/control_plane.go
@@ -192,6 +192,7 @@ func NewControlPlane(
 	// Add clsact qdisc
 	for _, ifname := range common.Deduplicate(append(append([]string{}, global.LanInterface...), global.WanInterface...)) {
 		_ = core.addQdisc(ifname)
+		_ = core.mapLinkType(ifname)
 	}
 	// Bind to LAN
 	if len(global.LanInterface) > 0 {

diff --git a/control/control_plane_core.go b/control/control_plane_core.go
@@ -135,6 +135,22 @@ func getIfParamsFromLink(link netlink.Link) (ifParams bpfIfParams, err error) {
 	return ifParams, nil
 }
 
+func (c *controlPlaneCore) mapLinkType(ifname string) error {
+	link, err := netlink.LinkByName(ifname)
+	if err != nil {
+		return err
+	}
+	linkType := uint32(0xffff)
+	switch link.Attrs().EncapType {
+	case "none":
+		linkType = consts.LinkType_None
+	case "ether":
+		linkType = consts.LinkType_Ethernet
+	default:
+	}
+	return c.bpf.bpfMaps.LinktypeMap.Update(uint32(link.Attrs().Index), linkType, ebpf.UpdateAny)
+}
+
 func (c *controlPlaneCore) addQdisc(ifname string) error {
 	link, err := netlink.LinkByName(ifname)
 	if err != nil {