Skip to content

Commit 658b5e6

Browse files
committed
net: return nil UDPAddr from ReadFromUDP
In cases where the socket operation has no underlying address, golang.org/cl/291509 unintentionally changed ReadFromUDP from return a nil *UDPAddr to a non-nil (but zero value) *UDPAddr. This may break callers that assume "no address" is always addr == nil, so change it back to remain nil. Fixes #46238 Change-Id: I8531e8fa16b853ed7560088eabda0b9e3e53f5be Reviewed-on: https://go-review.googlesource.com/c/go/+/320909 Trust: Michael Pratt <[email protected]> Trust: Josh Bleecher Snyder <[email protected]> Reviewed-by: Filippo Valsorda <[email protected]> Reviewed-by: Josh Bleecher Snyder <[email protected]>
1 parent 15a374d commit 658b5e6

File tree

2 files changed

+32
-0
lines changed

2 files changed

+32
-0
lines changed

src/net/udpsock_posix.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,9 @@ func (c *UDPConn) readFrom(b []byte, addr *UDPAddr) (int, *UDPAddr, error) {
5050
*addr = UDPAddr{IP: sa.Addr[0:], Port: sa.Port}
5151
case *syscall.SockaddrInet6:
5252
*addr = UDPAddr{IP: sa.Addr[0:], Port: sa.Port, Zone: zoneCache.name(int(sa.ZoneId))}
53+
default:
54+
// No sockaddr, so don't return UDPAddr.
55+
addr = nil
5356
}
5457
return n, addr, err
5558
}

src/net/udpsock_test.go

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,9 @@
88
package net
99

1010
import (
11+
"errors"
1112
"internal/testenv"
13+
"os"
1214
"reflect"
1315
"runtime"
1416
"testing"
@@ -446,6 +448,33 @@ func TestUDPReadSizeError(t *testing.T) {
446448
}
447449
}
448450

451+
// TestUDPReadTimeout verifies that ReadFromUDP with timeout returns an error
452+
// without data or an address.
453+
func TestUDPReadTimeout(t *testing.T) {
454+
la, err := ResolveUDPAddr("udp4", "127.0.0.1:0")
455+
if err != nil {
456+
t.Fatal(err)
457+
}
458+
c, err := ListenUDP("udp4", la)
459+
if err != nil {
460+
t.Fatal(err)
461+
}
462+
defer c.Close()
463+
464+
c.SetDeadline(time.Now())
465+
b := make([]byte, 1)
466+
n, addr, err := c.ReadFromUDP(b)
467+
if !errors.Is(err, os.ErrDeadlineExceeded) {
468+
t.Errorf("ReadFromUDP got err %v want os.ErrDeadlineExceeded", err)
469+
}
470+
if n != 0 {
471+
t.Errorf("ReadFromUDP got n %d want 0", n)
472+
}
473+
if addr != nil {
474+
t.Errorf("ReadFromUDP got addr %+#v want nil", addr)
475+
}
476+
}
477+
449478
func BenchmarkWriteToReadFromUDP(b *testing.B) {
450479
conn, err := ListenUDP("udp4", &UDPAddr{IP: IPv4(127, 0, 0, 1)})
451480
if err != nil {

0 commit comments

Comments
 (0)