ip.WithPort

This commit is contained in:
Josh Bleecher Snyder 2021-05-12 16:23:38 -07:00
parent ebcd7ab890
commit eec74d42ef
13 changed files with 42 additions and 52 deletions

View File

@ -164,10 +164,8 @@ func parseCallMeMaybe(ver uint8, p []byte) (m *CallMeMaybe, err error) {
for len(p) > 0 { for len(p) > 0 {
var a [16]byte var a [16]byte
copy(a[:], p) copy(a[:], p)
m.MyNumber = append(m.MyNumber, netaddr.IPPort{ port := binary.BigEndian.Uint16(p[16:18])
IP: netaddr.IPFrom16(a), m.MyNumber = append(m.MyNumber, netaddr.IPFrom16(a).WithPort(port))
Port: binary.BigEndian.Uint16(p[16:18]),
})
p = p[epLength:] p = p[epLength:]
} }
return m, nil return m, nil

View File

@ -1796,10 +1796,7 @@ func parseResolver(cfg tailcfg.DNSResolver) (netaddr.IPPort, error) {
if err != nil { if err != nil {
return netaddr.IPPort{}, fmt.Errorf("[unexpected] non-IP resolver %q", cfg.Addr) return netaddr.IPPort{}, fmt.Errorf("[unexpected] non-IP resolver %q", cfg.Addr)
} }
return netaddr.IPPort{ return ip.WithPort(53), nil
IP: ip,
Port: 53,
}, nil
} }
// tailscaleVarRoot returns the root directory of Tailscale's writable // tailscaleVarRoot returns the root directory of Tailscale's writable
@ -2580,9 +2577,9 @@ func peerAPIBase(nm *netmap.NetworkMap, peer *tailcfg.Node) string {
var ipp netaddr.IPPort var ipp netaddr.IPPort
switch { switch {
case have4 && p4 != 0: case have4 && p4 != 0:
ipp = netaddr.IPPort{IP: nodeIP(peer, netaddr.IP.Is4), Port: p4} ipp = nodeIP(peer, netaddr.IP.Is4).WithPort(p4)
case have6 && p6 != 0: case have6 && p6 != 0:
ipp = netaddr.IPPort{IP: nodeIP(peer, netaddr.IP.Is6), Port: p6} ipp = nodeIP(peer, netaddr.IP.Is6).WithPort(p6)
} }
if ipp.IP.IsZero() { if ipp.IP.IsZero() {
return "" return ""

View File

@ -219,7 +219,7 @@ func toIPsOnly(ipps []netaddr.IPPort) (ret []netaddr.IP) {
func toIPPorts(ips []netaddr.IP) (ret []netaddr.IPPort) { func toIPPorts(ips []netaddr.IP) (ret []netaddr.IPPort) {
ret = make([]netaddr.IPPort, 0, len(ips)) ret = make([]netaddr.IPPort, 0, len(ips))
for _, ip := range ips { for _, ip := range ips {
ret = append(ret, netaddr.IPPort{IP: ip, Port: 53}) ret = append(ret, ip.WithPort(53))
} }
return ret return ret
} }

View File

@ -433,8 +433,8 @@ func TestDelegateCollision(t *testing.T) {
qtype dns.Type qtype dns.Type
addr netaddr.IPPort addr netaddr.IPPort
}{ }{
{"test.site.", dns.TypeA, netaddr.IPPort{IP: netaddr.IPv4(1, 1, 1, 1), Port: 1001}}, {"test.site.", dns.TypeA, netaddr.IPv4(1, 1, 1, 1).WithPort(1001)},
{"test.site.", dns.TypeAAAA, netaddr.IPPort{IP: netaddr.IPv4(1, 1, 1, 1), Port: 1002}}, {"test.site.", dns.TypeAAAA, netaddr.IPv4(1, 1, 1, 1).WithPort(1002)},
} }
// packets will have the same dns txid. // packets will have the same dns txid.

View File

@ -1172,7 +1172,7 @@ func (c *Client) nodeAddr(ctx context.Context, n *tailcfg.DERPNode, proto probeP
if proto == probeIPv6 && ip.Is4() { if proto == probeIPv6 && ip.Is4() {
return nil return nil
} }
return netaddr.IPPort{IP: ip, Port: uint16(port)}.UDPAddr() return ip.WithPort(uint16(port)).UDPAddr()
} }
switch proto { switch proto {
@ -1182,7 +1182,7 @@ func (c *Client) nodeAddr(ctx context.Context, n *tailcfg.DERPNode, proto probeP
if !ip.Is4() { if !ip.Is4() {
return nil return nil
} }
return netaddr.IPPort{IP: ip, Port: uint16(port)}.UDPAddr() return ip.WithPort(uint16(port)).UDPAddr()
} }
case probeIPv6: case probeIPv6:
if n.IPv6 != "" { if n.IPv6 != "" {
@ -1190,7 +1190,7 @@ func (c *Client) nodeAddr(ctx context.Context, n *tailcfg.DERPNode, proto probeP
if !ip.Is6() { if !ip.Is6() {
return nil return nil
} }
return netaddr.IPPort{IP: ip, Port: uint16(port)}.UDPAddr() return ip.WithPort(uint16(port)).UDPAddr()
} }
default: default:
return nil return nil

View File

@ -192,8 +192,8 @@ func (pp *Parsed) AsTailscaleRejectedHeader() (h TailscaleRejectedHeader, ok boo
Reason: TailscaleRejectReason(p[2]), Reason: TailscaleRejectReason(p[2]),
IPSrc: pp.Src.IP, IPSrc: pp.Src.IP,
IPDst: pp.Dst.IP, IPDst: pp.Dst.IP,
Src: netaddr.IPPort{IP: pp.Dst.IP, Port: binary.BigEndian.Uint16(p[3:5])}, Src: pp.Dst.IP.WithPort(binary.BigEndian.Uint16(p[3:5])),
Dst: netaddr.IPPort{IP: pp.Src.IP, Port: binary.BigEndian.Uint16(p[5:7])}, Dst: pp.Src.IP.WithPort(binary.BigEndian.Uint16(p[5:7])),
} }
if len(p) > 7 { if len(p) > 7 {
flags := p[7] flags := p[7]

View File

@ -95,7 +95,7 @@ func (m *pmpMapping) release() {
} }
defer uc.Close() defer uc.Close()
pkt := buildPMPRequestMappingPacket(m.internal.Port, m.external.Port, pmpMapLifetimeDelete) pkt := buildPMPRequestMappingPacket(m.internal.Port, m.external.Port, pmpMapLifetimeDelete)
uc.WriteTo(pkt, netaddr.IPPort{IP: m.gw, Port: pmpPort}.UDPAddr()) uc.WriteTo(pkt, m.gw.WithPort(pmpPort).UDPAddr())
} }
// NewClient returns a new portmapping client. // NewClient returns a new portmapping client.
@ -256,7 +256,7 @@ func (c *Client) CreateOrGetMapping(ctx context.Context) (external netaddr.IPPor
localPort := c.localPort localPort := c.localPort
m := &pmpMapping{ m := &pmpMapping{
gw: gw, gw: gw,
internal: netaddr.IPPort{IP: myIP, Port: localPort}, internal: myIP.WithPort(localPort),
} }
// prevPort is the port we had most previously, if any. We try // prevPort is the port we had most previously, if any. We try
@ -297,7 +297,7 @@ func (c *Client) CreateOrGetMapping(ctx context.Context) (external netaddr.IPPor
uc.SetReadDeadline(time.Now().Add(portMapServiceTimeout)) uc.SetReadDeadline(time.Now().Add(portMapServiceTimeout))
defer closeCloserOnContextDone(ctx, uc)() defer closeCloserOnContextDone(ctx, uc)()
pmpAddr := netaddr.IPPort{IP: gw, Port: pmpPort} pmpAddr := gw.WithPort(pmpPort)
pmpAddru := pmpAddr.UDPAddr() pmpAddru := pmpAddr.UDPAddr()
// Ask for our external address if needed. // Ask for our external address if needed.
@ -468,9 +468,9 @@ func (c *Client) Probe(ctx context.Context) (res ProbeResult, err error) {
defer cancel() defer cancel()
defer closeCloserOnContextDone(ctx, uc)() defer closeCloserOnContextDone(ctx, uc)()
pcpAddr := netaddr.IPPort{IP: gw, Port: pcpPort}.UDPAddr() pcpAddr := gw.WithPort(pcpPort).UDPAddr()
pmpAddr := netaddr.IPPort{IP: gw, Port: pmpPort}.UDPAddr() pmpAddr := gw.WithPort(pmpPort).UDPAddr()
upnpAddr := netaddr.IPPort{IP: gw, Port: upnpPort}.UDPAddr() upnpAddr := gw.WithPort(upnpPort).UDPAddr()
// Don't send probes to services that we recently learned (for // Don't send probes to services that we recently learned (for
// the same gw/myIP) are available. See // the same gw/myIP) are available. See

View File

@ -237,10 +237,8 @@ func (n *SNAT44) allocateMappedPort() (net.PacketConn, netaddr.IPPort) {
if err != nil { if err != nil {
panic(fmt.Sprintf("ran out of NAT ports: %v", err)) panic(fmt.Sprintf("ran out of NAT ports: %v", err))
} }
addr := netaddr.IPPort{ port := uint16(pc.LocalAddr().(*net.UDPAddr).Port)
IP: ip, addr := ip.WithPort(port)
Port: uint16(pc.LocalAddr().(*net.UDPAddr).Port),
}
return pc, addr return pc, addr
} }

View File

@ -396,8 +396,8 @@ func (m *Machine) deliverLocalPacket(p *Packet, iface *Interface) {
} }
possibleDsts := []netaddr.IPPort{ possibleDsts := []netaddr.IPPort{
p.Dst, p.Dst,
netaddr.IPPort{IP: v6unspec, Port: p.Dst.Port}, v6unspec.WithPort(p.Dst.Port),
netaddr.IPPort{IP: v4unspec, Port: p.Dst.Port}, v4unspec.WithPort(p.Dst.Port),
} }
for _, dest := range possibleDsts { for _, dest := range possibleDsts {
c, ok := conns[dest] c, ok := conns[dest]
@ -707,7 +707,7 @@ func (m *Machine) ListenPacket(ctx context.Context, network, address string) (ne
return nil, nil return nil, nil
} }
} }
ipp := netaddr.IPPort{IP: ip, Port: port} ipp := ip.WithPort(port)
c := &conn{ c := &conn{
m: m, m: m,

View File

@ -49,8 +49,8 @@ func TestSendPacket(t *testing.T) {
ifFoo := foo.Attach("eth0", internet) ifFoo := foo.Attach("eth0", internet)
ifBar := bar.Attach("enp0s1", internet) ifBar := bar.Attach("enp0s1", internet)
fooAddr := netaddr.IPPort{IP: ifFoo.V4(), Port: 123} fooAddr := ifFoo.V4().WithPort(123)
barAddr := netaddr.IPPort{IP: ifBar.V4(), Port: 456} barAddr := ifBar.V4().WithPort(456)
ctx := context.Background() ctx := context.Background()
fooPC, err := foo.ListenPacket(ctx, "udp4", fooAddr.String()) fooPC, err := foo.ListenPacket(ctx, "udp4", fooAddr.String())
@ -111,10 +111,10 @@ func TestMultiNetwork(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
clientAddr := netaddr.IPPort{IP: ifClient.V4(), Port: 123} clientAddr := ifClient.V4().WithPort(123)
natLANAddr := netaddr.IPPort{IP: ifNATLAN.V4(), Port: 456} natLANAddr := ifNATLAN.V4().WithPort(456)
natWANAddr := netaddr.IPPort{IP: ifNATWAN.V4(), Port: 456} natWANAddr := ifNATWAN.V4().WithPort(456)
serverAddr := netaddr.IPPort{IP: ifServer.V4(), Port: 789} serverAddr := ifServer.V4().WithPort(789)
const msg1, msg2 = "hello", "world" const msg1, msg2 = "hello", "world"
if _, err := natPC.WriteTo([]byte(msg1), clientAddr.UDPAddr()); err != nil { if _, err := natPC.WriteTo([]byte(msg1), clientAddr.UDPAddr()); err != nil {
@ -216,7 +216,7 @@ func TestPacketHandler(t *testing.T) {
} }
const msg = "some message" const msg = "some message"
serverAddr := netaddr.IPPort{IP: ifServer.V4(), Port: 456} serverAddr := ifServer.V4().WithPort(456)
if _, err := clientPC.WriteTo([]byte(msg), serverAddr.UDPAddr()); err != nil { if _, err := clientPC.WriteTo([]byte(msg), serverAddr.UDPAddr()); err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -230,7 +230,7 @@ func TestPacketHandler(t *testing.T) {
if string(buf) != msg { if string(buf) != msg {
t.Errorf("read %q; want %q", buf, msg) t.Errorf("read %q; want %q", buf, msg)
} }
mappedAddr := netaddr.IPPort{IP: ifNATWAN.V4(), Port: 123} mappedAddr := ifNATWAN.V4().WithPort(123)
if addr.String() != mappedAddr.String() { if addr.String() != mappedAddr.String() {
t.Errorf("addr = %q; want %q", addr, mappedAddr) t.Errorf("addr = %q; want %q", addr, mappedAddr)
} }

View File

@ -425,10 +425,10 @@ func TestLoggingPrivacy(t *testing.T) {
f.logIPs = logB.IPSet() f.logIPs = logB.IPSet()
var ( var (
ts4 = netaddr.IPPort{IP: tsaddr.CGNATRange().IP.Next(), Port: 1234} ts4 = tsaddr.CGNATRange().IP.Next().WithPort(1234)
internet4 = netaddr.IPPort{IP: netaddr.MustParseIP("8.8.8.8"), Port: 1234} internet4 = netaddr.MustParseIP("8.8.8.8").WithPort(1234)
ts6 = netaddr.IPPort{IP: tsaddr.TailscaleULARange().IP.Next(), Port: 1234} ts6 = tsaddr.TailscaleULARange().IP.Next().WithPort(1234)
internet6 = netaddr.IPPort{IP: netaddr.MustParseIP("2001::1"), Port: 1234} internet6 = netaddr.MustParseIP("2001::1").WithPort(1234)
) )
tests := []struct { tests := []struct {

View File

@ -965,7 +965,7 @@ func (c *Conn) goDerpConnect(node int) {
if node == 0 { if node == 0 {
return return
} }
go c.derpWriteChanOfAddr(netaddr.IPPort{IP: derpMagicIPAddr, Port: uint16(node)}, key.Public{}) go c.derpWriteChanOfAddr(derpMagicIPAddr.WithPort(uint16(node)), key.Public{})
} }
// determineEndpoints returns the machine's endpoint addresses. It // determineEndpoints returns the machine's endpoint addresses. It
@ -1037,7 +1037,7 @@ func (c *Conn) determineEndpoints(ctx context.Context) ([]tailcfg.Endpoint, erro
ips = loopback ips = loopback
} }
for _, ip := range ips { for _, ip := range ips {
addAddr(netaddr.IPPort{IP: ip, Port: uint16(localAddr.Port)}, tailcfg.EndpointLocal) addAddr(ip.WithPort(uint16(localAddr.Port)), tailcfg.EndpointLocal)
} }
} else { } else {
// Our local endpoint is bound to a particular address. // Our local endpoint is bound to a particular address.
@ -1676,7 +1676,7 @@ func (c *Conn) processDERPReadResult(dm derpReadResult, b []byte) (n int, ep con
return 0, nil return 0, nil
} }
ipp := netaddr.IPPort{IP: derpMagicIPAddr, Port: uint16(regionID)} ipp := derpMagicIPAddr.WithPort(uint16(regionID))
if c.handleDiscoMessage(b[:n], ipp) { if c.handleDiscoMessage(b[:n], ipp) {
return 0, nil return 0, nil
} }
@ -3254,10 +3254,7 @@ func (de *discoEndpoint) initFakeUDPAddr() {
addr[0] = 0xfd addr[0] = 0xfd
addr[1] = 0x00 addr[1] = 0x00
binary.BigEndian.PutUint64(addr[2:], uint64(reflect.ValueOf(de).Pointer())) binary.BigEndian.PutUint64(addr[2:], uint64(reflect.ValueOf(de).Pointer()))
de.fakeWGAddr = netaddr.IPPort{ de.fakeWGAddr = netaddr.IPFrom16(addr).WithPort(12345)
IP: netaddr.IPFrom16(addr),
Port: 12345,
}
} }
// isFirstRecvActivityInAwhile notes that receive activity has occured for this // isFirstRecvActivityInAwhile notes that receive activity has occured for this

View File

@ -316,7 +316,7 @@ func (m DNSMap) Resolve(ctx context.Context, addr string) (netaddr.IPPort, error
// Try MagicDNS first, else otherwise a real DNS lookup. // Try MagicDNS first, else otherwise a real DNS lookup.
ip := m[host] ip := m[host]
if !ip.IsZero() { if !ip.IsZero() {
return netaddr.IPPort{IP: ip, Port: uint16(port16)}, nil return ip.WithPort(uint16(port16)), nil
} }
// No MagicDNS name so try real DNS. // No MagicDNS name so try real DNS.
@ -329,7 +329,7 @@ func (m DNSMap) Resolve(ctx context.Context, addr string) (netaddr.IPPort, error
return netaddr.IPPort{}, fmt.Errorf("DNS lookup returned no results for %q", host) return netaddr.IPPort{}, fmt.Errorf("DNS lookup returned no results for %q", host)
} }
ip, _ = netaddr.FromStdIP(ips[0]) ip, _ = netaddr.FromStdIP(ips[0])
return netaddr.IPPort{IP: ip, Port: uint16(port16)}, nil return ip.WithPort(uint16(port16)), nil
} }
func (ns *Impl) DialContextTCP(ctx context.Context, addr string) (*gonet.TCPConn, error) { func (ns *Impl) DialContextTCP(ctx context.Context, addr string) (*gonet.TCPConn, error) {