123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282 |
- // Copyright 2013 The Go Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style
- // license that can be found in the LICENSE file.
- package net
- import (
- "reflect"
- "testing"
- )
- var testInetaddr = func(ip IPAddr) Addr { return &TCPAddr{IP: ip.IP, Port: 5682, Zone: ip.Zone} }
- var addrListTests = []struct {
- filter func(IPAddr) bool
- ips []IPAddr
- inetaddr func(IPAddr) Addr
- first Addr
- primaries addrList
- fallbacks addrList
- err error
- }{
- {
- nil,
- []IPAddr{
- {IP: IPv4(127, 0, 0, 1)},
- {IP: IPv6loopback},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
- addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
- nil,
- },
- {
- nil,
- []IPAddr{
- {IP: IPv6loopback},
- {IP: IPv4(127, 0, 0, 1)},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
- addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
- nil,
- },
- {
- nil,
- []IPAddr{
- {IP: IPv4(127, 0, 0, 1)},
- {IP: IPv4(192, 168, 0, 1)},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- &TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
- },
- nil,
- nil,
- },
- {
- nil,
- []IPAddr{
- {IP: IPv6loopback},
- {IP: ParseIP("fe80::1"), Zone: "eth0"},
- },
- testInetaddr,
- &TCPAddr{IP: IPv6loopback, Port: 5682},
- addrList{
- &TCPAddr{IP: IPv6loopback, Port: 5682},
- &TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
- },
- nil,
- nil,
- },
- {
- nil,
- []IPAddr{
- {IP: IPv4(127, 0, 0, 1)},
- {IP: IPv4(192, 168, 0, 1)},
- {IP: IPv6loopback},
- {IP: ParseIP("fe80::1"), Zone: "eth0"},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- &TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
- },
- addrList{
- &TCPAddr{IP: IPv6loopback, Port: 5682},
- &TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
- },
- nil,
- },
- {
- nil,
- []IPAddr{
- {IP: IPv6loopback},
- {IP: ParseIP("fe80::1"), Zone: "eth0"},
- {IP: IPv4(127, 0, 0, 1)},
- {IP: IPv4(192, 168, 0, 1)},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{
- &TCPAddr{IP: IPv6loopback, Port: 5682},
- &TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
- },
- addrList{
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- &TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
- },
- nil,
- },
- {
- nil,
- []IPAddr{
- {IP: IPv4(127, 0, 0, 1)},
- {IP: IPv6loopback},
- {IP: IPv4(192, 168, 0, 1)},
- {IP: ParseIP("fe80::1"), Zone: "eth0"},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- &TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
- },
- addrList{
- &TCPAddr{IP: IPv6loopback, Port: 5682},
- &TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
- },
- nil,
- },
- {
- nil,
- []IPAddr{
- {IP: IPv6loopback},
- {IP: IPv4(127, 0, 0, 1)},
- {IP: ParseIP("fe80::1"), Zone: "eth0"},
- {IP: IPv4(192, 168, 0, 1)},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{
- &TCPAddr{IP: IPv6loopback, Port: 5682},
- &TCPAddr{IP: ParseIP("fe80::1"), Port: 5682, Zone: "eth0"},
- },
- addrList{
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- &TCPAddr{IP: IPv4(192, 168, 0, 1), Port: 5682},
- },
- nil,
- },
- {
- ipv4only,
- []IPAddr{
- {IP: IPv4(127, 0, 0, 1)},
- {IP: IPv6loopback},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
- nil,
- nil,
- },
- {
- ipv4only,
- []IPAddr{
- {IP: IPv6loopback},
- {IP: IPv4(127, 0, 0, 1)},
- },
- testInetaddr,
- &TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682},
- addrList{&TCPAddr{IP: IPv4(127, 0, 0, 1), Port: 5682}},
- nil,
- nil,
- },
- {
- ipv6only,
- []IPAddr{
- {IP: IPv4(127, 0, 0, 1)},
- {IP: IPv6loopback},
- },
- testInetaddr,
- &TCPAddr{IP: IPv6loopback, Port: 5682},
- addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
- nil,
- nil,
- },
- {
- ipv6only,
- []IPAddr{
- {IP: IPv6loopback},
- {IP: IPv4(127, 0, 0, 1)},
- },
- testInetaddr,
- &TCPAddr{IP: IPv6loopback, Port: 5682},
- addrList{&TCPAddr{IP: IPv6loopback, Port: 5682}},
- nil,
- nil,
- },
- {nil, nil, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
- {ipv4only, nil, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
- {ipv4only, []IPAddr{{IP: IPv6loopback}}, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
- {ipv6only, nil, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
- {ipv6only, []IPAddr{{IP: IPv4(127, 0, 0, 1)}}, testInetaddr, nil, nil, nil, &AddrError{errNoSuitableAddress.Error(), "ADDR"}},
- }
- func TestAddrList(t *testing.T) {
- if !supportsIPv4() || !supportsIPv6() {
- t.Skip("both IPv4 and IPv6 are required")
- }
- for i, tt := range addrListTests {
- addrs, err := filterAddrList(tt.filter, tt.ips, tt.inetaddr, "ADDR")
- if !reflect.DeepEqual(err, tt.err) {
- t.Errorf("#%v: got %v; want %v", i, err, tt.err)
- }
- if tt.err != nil {
- if len(addrs) != 0 {
- t.Errorf("#%v: got %v; want 0", i, len(addrs))
- }
- continue
- }
- first := addrs.first(isIPv4)
- if !reflect.DeepEqual(first, tt.first) {
- t.Errorf("#%v: got %v; want %v", i, first, tt.first)
- }
- primaries, fallbacks := addrs.partition(isIPv4)
- if !reflect.DeepEqual(primaries, tt.primaries) {
- t.Errorf("#%v: got %v; want %v", i, primaries, tt.primaries)
- }
- if !reflect.DeepEqual(fallbacks, tt.fallbacks) {
- t.Errorf("#%v: got %v; want %v", i, fallbacks, tt.fallbacks)
- }
- expectedLen := len(primaries) + len(fallbacks)
- if len(addrs) != expectedLen {
- t.Errorf("#%v: got %v; want %v", i, len(addrs), expectedLen)
- }
- }
- }
- func TestAddrListPartition(t *testing.T) {
- addrs := addrList{
- &IPAddr{IP: ParseIP("fe80::"), Zone: "eth0"},
- &IPAddr{IP: ParseIP("fe80::1"), Zone: "eth0"},
- &IPAddr{IP: ParseIP("fe80::2"), Zone: "eth0"},
- }
- cases := []struct {
- lastByte byte
- primaries addrList
- fallbacks addrList
- }{
- {0, addrList{addrs[0]}, addrList{addrs[1], addrs[2]}},
- {1, addrList{addrs[0], addrs[2]}, addrList{addrs[1]}},
- {2, addrList{addrs[0], addrs[1]}, addrList{addrs[2]}},
- {3, addrList{addrs[0], addrs[1], addrs[2]}, nil},
- }
- for i, tt := range cases {
- // Inverting the function's output should not affect the outcome.
- for _, invert := range []bool{false, true} {
- primaries, fallbacks := addrs.partition(func(a Addr) bool {
- ip := a.(*IPAddr).IP
- return (ip[len(ip)-1] == tt.lastByte) != invert
- })
- if !reflect.DeepEqual(primaries, tt.primaries) {
- t.Errorf("#%v: got %v; want %v", i, primaries, tt.primaries)
- }
- if !reflect.DeepEqual(fallbacks, tt.fallbacks) {
- t.Errorf("#%v: got %v; want %v", i, fallbacks, tt.fallbacks)
- }
- }
- }
- }
|