//go:build linux /* SPDX-License-Identifier: MIT * * Copyright (C) 2017-2023 WireGuard LLC. All Rights Reserved. */ package conn import ( "context" "net" "net/netip" "runtime" "testing" "unsafe" "golang.org/x/sys/unix" ) func Test_setSrcControl(t *testing.T) { t.Run("IPv4", func(t *testing.T) { ep := &StdNetEndpoint{ AddrPort: netip.MustParseAddrPort("127.0.0.1:1234"), } ep.src.Addr = netip.MustParseAddr("127.0.0.1") ep.src.ifidx = 5 control := make([]byte, srcControlSize) setSrcControl(&control, ep) hdr := (*unix.Cmsghdr)(unsafe.Pointer(&control[0])) if hdr.Level != unix.IPPROTO_IP { t.Errorf("unexpected level: %d", hdr.Level) } if hdr.Type != unix.IP_PKTINFO { t.Errorf("unexpected type: %d", hdr.Type) } if uint(hdr.Len) != uint(unix.CmsgLen(int(unsafe.Sizeof(unix.Inet4Pktinfo{})))) { t.Errorf("unexpected length: %d", hdr.Len) } info := (*unix.Inet4Pktinfo)(unsafe.Pointer(&control[unix.CmsgLen(0)])) if info.Spec_dst[0] != 127 || info.Spec_dst[1] != 0 || info.Spec_dst[2] != 0 || info.Spec_dst[3] != 1 { t.Errorf("unexpected address: %v", info.Spec_dst) } if info.Ifindex != 5 { t.Errorf("unexpected ifindex: %d", info.Ifindex) } }) t.Run("IPv6", func(t *testing.T) { ep := &StdNetEndpoint{ AddrPort: netip.MustParseAddrPort("[::1]:1234"), } ep.src.Addr = netip.MustParseAddr("::1") ep.src.ifidx = 5 control := make([]byte, srcControlSize) setSrcControl(&control, ep) hdr := (*unix.Cmsghdr)(unsafe.Pointer(&control[0])) if hdr.Level != unix.IPPROTO_IPV6 { t.Errorf("unexpected level: %d", hdr.Level) } if hdr.Type != unix.IPV6_PKTINFO { t.Errorf("unexpected type: %d", hdr.Type) } if uint(hdr.Len) != uint(unix.CmsgLen(int(unsafe.Sizeof(unix.Inet6Pktinfo{})))) { t.Errorf("unexpected length: %d", hdr.Len) } info := (*unix.Inet6Pktinfo)(unsafe.Pointer(&control[unix.CmsgLen(0)])) if info.Addr != ep.SrcIP().As16() { t.Errorf("unexpected address: %v", info.Addr) } if info.Ifindex != 5 { t.Errorf("unexpected ifindex: %d", info.Ifindex) } }) t.Run("ClearOnNoSrc", func(t *testing.T) { control := make([]byte, srcControlSize) hdr := (*unix.Cmsghdr)(unsafe.Pointer(&control[0])) hdr.Level = 1 hdr.Type = 2 hdr.Len = 3 setSrcControl(&control, &StdNetEndpoint{}) if len(control) != 0 { t.Errorf("unexpected control: %v", control) } }) } func Test_getSrcFromControl(t *testing.T) { t.Run("IPv4", func(t *testing.T) { control := make([]byte, srcControlSize) hdr := (*unix.Cmsghdr)(unsafe.Pointer(&control[0])) hdr.Level = unix.IPPROTO_IP hdr.Type = unix.IP_PKTINFO hdr.SetLen(unix.CmsgLen(int(unsafe.Sizeof(unix.Inet4Pktinfo{})))) info := (*unix.Inet4Pktinfo)(unsafe.Pointer(&control[unix.CmsgLen(0)])) info.Spec_dst = [4]byte{127, 0, 0, 1} info.Ifindex = 5 ep := &StdNetEndpoint{} getSrcFromControl(control, ep) if ep.src.Addr != netip.MustParseAddr("127.0.0.1") { t.Errorf("unexpected address: %v", ep.src.Addr) } if ep.src.ifidx != 5 { t.Errorf("unexpected ifindex: %d", ep.src.ifidx) } }) t.Run("IPv6", func(t *testing.T) { control := make([]byte, srcControlSize) hdr := (*unix.Cmsghdr)(unsafe.Pointer(&control[0])) hdr.Level = unix.IPPROTO_IPV6 hdr.Type = unix.IPV6_PKTINFO hdr.SetLen(unix.CmsgLen(int(unsafe.Sizeof(unix.Inet6Pktinfo{})))) info := (*unix.Inet6Pktinfo)(unsafe.Pointer(&control[unix.CmsgLen(0)])) info.Addr = [16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1} info.Ifindex = 5 ep := &StdNetEndpoint{} getSrcFromControl(control, ep) if ep.SrcIP() != netip.MustParseAddr("::1") { t.Errorf("unexpected address: %v", ep.SrcIP()) } if ep.src.ifidx != 5 { t.Errorf("unexpected ifindex: %d", ep.src.ifidx) } }) t.Run("ClearOnEmpty", func(t *testing.T) { control := make([]byte, srcControlSize) ep := &StdNetEndpoint{} ep.src.Addr = netip.MustParseAddr("::1") ep.src.ifidx = 5 getSrcFromControl(control, ep) if ep.SrcIP().IsValid() { t.Errorf("unexpected address: %v", ep.src.Addr) } if ep.src.ifidx != 0 { t.Errorf("unexpected ifindex: %d", ep.src.ifidx) } }) t.Run("Multiple", func(t *testing.T) { zeroControl := make([]byte, unix.CmsgSpace(0)) zeroHdr := (*unix.Cmsghdr)(unsafe.Pointer(&zeroControl[0])) zeroHdr.SetLen(unix.CmsgLen(0)) control := make([]byte, srcControlSize) hdr := (*unix.Cmsghdr)(unsafe.Pointer(&control[0])) hdr.Level = unix.IPPROTO_IP hdr.Type = unix.IP_PKTINFO hdr.SetLen(unix.CmsgLen(int(unsafe.Sizeof(unix.Inet4Pktinfo{})))) info := (*unix.Inet4Pktinfo)(unsafe.Pointer(&control[unix.CmsgLen(0)])) info.Spec_dst = [4]byte{127, 0, 0, 1} info.Ifindex = 5 combined := make([]byte, 0) combined = append(combined, zeroControl...) combined = append(combined, control...) ep := &StdNetEndpoint{} getSrcFromControl(combined, ep) if ep.src.Addr != netip.MustParseAddr("127.0.0.1") { t.Errorf("unexpected address: %v", ep.src.Addr) } if ep.src.ifidx != 5 { t.Errorf("unexpected ifindex: %d", ep.src.ifidx) } }) } func Test_listenConfig(t *testing.T) { t.Run("IPv4", func(t *testing.T) { conn, err := listenConfig().ListenPacket(context.Background(), "udp4", ":0") if err != nil { t.Fatal(err) } defer conn.Close() sc, err := conn.(*net.UDPConn).SyscallConn() if err != nil { t.Fatal(err) } if runtime.GOOS == "linux" { var i int sc.Control(func(fd uintptr) { i, err = unix.GetsockoptInt(int(fd), unix.IPPROTO_IP, unix.IP_PKTINFO) }) if err != nil { t.Fatal(err) } if i != 1 { t.Error("IP_PKTINFO not set!") } } else { t.Logf("listenConfig() does not set IPV6_RECVPKTINFO on %s", runtime.GOOS) } }) t.Run("IPv6", func(t *testing.T) { conn, err := listenConfig().ListenPacket(context.Background(), "udp6", ":0") if err != nil { t.Fatal(err) } sc, err := conn.(*net.UDPConn).SyscallConn() if err != nil { t.Fatal(err) } if runtime.GOOS == "linux" { var i int sc.Control(func(fd uintptr) { i, err = unix.GetsockoptInt(int(fd), unix.IPPROTO_IPV6, unix.IPV6_RECVPKTINFO) }) if err != nil { t.Fatal(err) } if i != 1 { t.Error("IPV6_PKTINFO not set!") } } else { t.Logf("listenConfig() does not set IPV6_RECVPKTINFO on %s", runtime.GOOS) } }) }