mirror of
https://github.com/netbirdio/netbird.git
synced 2025-07-23 17:05:29 +02:00
f89332fcd2
Ensure we use WG address instead of loopback addresses for eBPF. - First try to use 53 port - Try to use 5053 port on WG interface for eBPF - Try to use 5053 on WG interface or loopback interface
52 lines
1015 B
Go
52 lines
1015 B
Go
package ebpf
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"net"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
mapKeyDNSIP uint32 = 0
|
|
mapKeyDNSPort uint32 = 1
|
|
)
|
|
|
|
func (tf *GeneralManager) LoadDNSFwd(ip string, dnsPort int) error {
|
|
log.Debugf("load eBPF DNS forwarder, watching addr: %s:53, redirect to port: %d", ip, dnsPort)
|
|
tf.lock.Lock()
|
|
defer tf.lock.Unlock()
|
|
|
|
err := tf.loadXdp()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = tf.bpfObjs.NbMapDnsIp.Put(mapKeyDNSIP, ip2int(ip))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = tf.bpfObjs.NbMapDnsPort.Put(mapKeyDNSPort, uint16(dnsPort))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
tf.setFeatureFlag(featureFlagDnsForwarder)
|
|
err = tf.bpfObjs.NbFeatures.Put(mapKeyFeatures, tf.featureFlags)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (tf *GeneralManager) FreeDNSFwd() error {
|
|
log.Debugf("free ebpf DNS forwarder")
|
|
return tf.unsetFeatureFlag(featureFlagDnsForwarder)
|
|
}
|
|
|
|
func ip2int(ipString string) uint32 {
|
|
ip := net.ParseIP(ipString)
|
|
return binary.BigEndian.Uint32(ip.To4())
|
|
}
|