engine: mac: more ipv6 gm
Some checks failed
Quality check / Static analysis (push) Has been cancelled
Quality check / Tests (push) Has been cancelled

This commit is contained in:
2026-02-11 13:18:34 +05:30
parent 9c2786212b
commit 198f72814c

View File

@@ -4,6 +4,7 @@ import (
"bufio"
"net"
"os"
"os/exec"
"strings"
"sync"
"time"
@@ -91,6 +92,17 @@ func (r *sourceMACResolver) Resolve(ip net.IP) net.HardwareAddr {
if mac := r.ndpByIP[ipKey]; len(mac) != 0 {
return append(net.HardwareAddr(nil), mac...)
}
// On-demand IPv6 neighbor lookup via route-netlink as a last fast path.
if ip.To4() == nil {
if mac, ok := lookupNeighborMACNetlink(ip); ok {
out := append(net.HardwareAddr(nil), mac...)
r.mu.Lock()
r.ndpByIP[ipKey] = append(net.HardwareAddr(nil), mac...)
r.mu.Unlock()
return out
}
}
return nil
}
@@ -170,6 +182,10 @@ func (r *sourceMACResolver) refreshNDPCache(now time.Time) {
// Fallback for environments without /proc/net/ndisc_cache.
m = readIPv6NeighNetlink()
}
if len(m) == 0 {
// Last-resort fallback for environments where route-netlink dumps are restricted.
m = readIPv6NeighCommand()
}
r.mu.Lock()
r.ndpByIP = m
r.lastNDPRefresh = now
@@ -258,6 +274,100 @@ func readIPv6NeighNetlink() map[string]net.HardwareAddr {
return m
}
func lookupNeighborMACNetlink(target net.IP) (net.HardwareAddr, bool) {
const (
ndMsgLen = 12
ndaDst = 1
ndaLLAddr = 2
)
if target == nil || target.To4() != nil {
return nil, false
}
target16 := target.To16()
if target16 == nil {
return nil, false
}
conn, err := netlink.Dial(unix.NETLINK_ROUTE, nil)
if err != nil {
return nil, false
}
defer conn.Close()
req := make([]byte, ndMsgLen)
req[0] = unix.AF_INET6
msgs, err := conn.Execute(netlink.Message{
Header: netlink.Header{
Type: unix.RTM_GETNEIGH,
Flags: netlink.Request | netlink.Dump,
},
Data: req,
})
if err != nil {
return nil, false
}
for _, msg := range msgs {
if msg.Header.Type != unix.RTM_NEWNEIGH || len(msg.Data) < ndMsgLen || msg.Data[0] != unix.AF_INET6 {
continue
}
attrs, err := netlink.UnmarshalAttributes(msg.Data[ndMsgLen:])
if err != nil {
continue
}
var dstIP net.IP
var mac net.HardwareAddr
for _, a := range attrs {
switch a.Type {
case ndaDst:
if len(a.Data) == net.IPv6len {
dstIP = net.IP(append([]byte(nil), a.Data...))
}
case ndaLLAddr:
if len(a.Data) >= 6 {
candidate := append(net.HardwareAddr(nil), a.Data...)
if candidate.String() != "00:00:00:00:00:00" {
mac = candidate
}
}
}
}
if dstIP != nil && mac != nil && dstIP.Equal(target16) {
return mac, true
}
}
return nil, false
}
func readIPv6NeighCommand() map[string]net.HardwareAddr {
commands := [][]string{
{"ip", "-6", "neigh", "show"},
{"/sbin/ip", "-6", "neigh", "show"},
{"/usr/sbin/ip", "-6", "neigh", "show"},
{"busybox", "ip", "-6", "neigh", "show"},
{"/bin/busybox", "ip", "-6", "neigh", "show"},
}
for _, cmd := range commands {
out, err := exec.Command(cmd[0], cmd[1:]...).Output()
if err != nil || len(out) == 0 {
continue
}
m := make(map[string]net.HardwareAddr)
for _, line := range strings.Split(string(out), "\n") {
ip, mac, ok := parseNeighborLine(line)
if !ok {
continue
}
m[ip] = mac
}
if len(m) != 0 {
return m
}
}
return map[string]net.HardwareAddr{}
}
func parseNeighborLine(line string) (string, net.HardwareAddr, bool) {
fields := strings.Fields(line)
if len(fields) < 2 {