fix for windows/macos

net/connUDP.go

@@ -24,8 +24,10 @@ type UDPConn struct {
 }
 
 type ControlMessage struct {
-	Dst     net.IP // destination address, receiving only
-	Src     net.IP // source address, specifying only
+	// For connection oriented packetConn the ControlMessage fields are ignored, only linux supports set control message.
+
+	Dst     net.IP // destination address of the packet
+	Src     net.IP // source address of the packet
 	IfIndex int    // interface index, 0 means any interface
 }
 
@@ -72,18 +74,11 @@ type packetConnIPv4 struct {
 	supportsControlMessage bool
 }
 
-func isNotImplemented(err error) bool {
-	return strings.Contains(err.Error(), "not implemented on")
-}
-
-func newPacketConnIPv4(p *ipv4.PacketConn) (*packetConnIPv4, error) {
+func newPacketConnIPv4(p *ipv4.PacketConn) *packetConnIPv4 {
 	if err := p.SetControlMessage(ipv4.FlagDst|ipv4.FlagInterface|ipv4.FlagSrc, true); err != nil {
-		if isNotImplemented(err) {
-			return &packetConnIPv4{packetConn: p, supportsControlMessage: false}, nil
-		}
-		return nil, err
+		return &packetConnIPv4{packetConn: p, supportsControlMessage: false}
 	}
-	return &packetConnIPv4{packetConn: p, supportsControlMessage: true}, nil
+	return &packetConnIPv4{packetConn: p, supportsControlMessage: true}
 }
 
 func (p *packetConnIPv4) SupportsControlMessage() bool {
@@ -150,14 +145,11 @@ type packetConnIPv6 struct {
 	supportsControlMessage bool
 }
 
-func newPacketConnIPv6(p *ipv6.PacketConn) (*packetConnIPv6, error) {
+func newPacketConnIPv6(p *ipv6.PacketConn) *packetConnIPv6 {
 	if err := p.SetControlMessage(ipv6.FlagDst|ipv6.FlagInterface|ipv6.FlagSrc, true); err != nil {
-		if isNotImplemented(err) {
-			return &packetConnIPv6{packetConn: p, supportsControlMessage: false}, nil
-		}
-		return nil, err
+		return &packetConnIPv6{packetConn: p, supportsControlMessage: false}
 	}
-	return &packetConnIPv6{packetConn: p, supportsControlMessage: true}, nil
+	return &packetConnIPv6{packetConn: p, supportsControlMessage: true}
 }
 
 func (p *packetConnIPv6) SupportsControlMessage() bool {
@@ -265,12 +257,9 @@ func newPacketConnWithAddr(addr *net.UDPAddr, c *net.UDPConn) (packetConn, error
 	var pc packetConn
 	var err error
 	if IsIPv6(addr.IP) {
-		pc, err = newPacketConnIPv6(ipv6.NewPacketConn(c))
-		if err != nil {
-			return nil, fmt.Errorf("invalid UDPv6 connection: %w", err)
-		}
+		pc = newPacketConnIPv6(ipv6.NewPacketConn(c))
 	} else {
-		pc, err = newPacketConnIPv4(ipv4.NewPacketConn(c))
+		pc = newPacketConnIPv4(ipv4.NewPacketConn(c))
 		if err != nil {
 			return nil, fmt.Errorf("invalid UDPv4 connection: %w", err)
 		}
@@ -524,6 +513,16 @@ func (c *UDPConn) writeMulticast(ctx context.Context, raddr *net.UDPAddr, buffer
 	return nil
 }
 
+func (c *UDPConn) writeTo(raddr *net.UDPAddr, cm *ControlMessage, buffer []byte) (int, error) {
+	if !supportsOverrideRemoteAddr(c.connection) {
+		// If the remote address is set, we can use it as the destination address
+		// because the connection is already established.
+		// Note: Overwriting the destination address is only supported on Linux.
+		return c.connection.Write(buffer)
+	}
+	return c.packetConn.WriteTo(buffer, cm, raddr)
+}
+
 // WriteWithContext writes data with context.
 func (c *UDPConn) WriteWithContext(ctx context.Context, raddr *net.UDPAddr, cm *ControlMessage, buffer []byte) error {
 	if raddr == nil {
@@ -538,7 +537,7 @@ func (c *UDPConn) WriteWithContext(ctx context.Context, raddr *net.UDPAddr, cm *
 	if c.closed.Load() {
 		return ErrConnectionIsClosed
 	}
-	n, err := c.packetConn.WriteTo(buffer, cm, raddr)
+	n, err := c.writeTo(raddr, cm, buffer)
 	if err != nil {
 		return err
 	}

net/connUDP_internal_test.go

@@ -369,20 +369,25 @@ func TestUDPConnWriteToAddr(t *testing.T) {
 	}
 }
 
-func TestPacketConnIPv4ReadFrom(t *testing.T) {
+func TestPacketConnReadFrom(t *testing.T) {
 	readUDP4Conn, err := net.ListenUDP("udp4", &net.UDPAddr{Port: 1234})
 	require.NoError(t, err)
 	defer func() {
 		errC := readUDP4Conn.Close()
 		require.NoError(t, errC)
 	}()
+
+	require.Nil(t, readUDP4Conn.RemoteAddr())
+
 	writeUDP4Conn, err := net.DialUDP("udp4", nil, readUDP4Conn.LocalAddr().(*net.UDPAddr))
 	require.NoError(t, err)
 	defer func() {
 		errC := writeUDP4Conn.Close()
 		require.NoError(t, errC)
 	}()
 
+	require.NotNil(t, writeUDP4Conn.RemoteAddr())
+
 	readUDP6Conn, err := net.ListenUDP("udp6", &net.UDPAddr{Port: 1235})
 	require.NoError(t, err)
 	defer func() {

net/supportsOverrideRemoteAddr.go

@@ -0,0 +1,9 @@
+//go:build !linux
+
+package net
+
+import "net"
+
+func supportsOverrideRemoteAddr(c *net.UDPConn) bool {
+	return c.RemoteAddr() == nil
+}

net/supportsOverrideRemoteAddr_linux.go

@@ -0,0 +1,7 @@
+package net
+
+import "net"
+
+func supportsOverrideRemoteAddr(*net.UDPConn) bool {
+	return true
+}