Source file
src/net/ip_test.go
1
2
3
4
5
6
7 package net
8
9 import (
10 "bytes"
11 "math/rand"
12 "reflect"
13 "runtime"
14 "testing"
15 )
16
17 var parseIPTests = []struct {
18 in string
19 out IP
20 }{
21 {"127.0.1.2", IPv4(127, 0, 1, 2)},
22 {"127.0.0.1", IPv4(127, 0, 0, 1)},
23 {"::ffff:127.1.2.3", IPv4(127, 1, 2, 3)},
24 {"::ffff:7f01:0203", IPv4(127, 1, 2, 3)},
25 {"0:0:0:0:0000:ffff:127.1.2.3", IPv4(127, 1, 2, 3)},
26 {"0:0:0:0:000000:ffff:127.1.2.3", IPv4(127, 1, 2, 3)},
27 {"0:0:0:0::ffff:127.1.2.3", IPv4(127, 1, 2, 3)},
28
29 {"2001:4860:0:2001::68", IP{0x20, 0x01, 0x48, 0x60, 0, 0, 0x20, 0x01, 0, 0, 0, 0, 0, 0, 0x00, 0x68}},
30 {"2001:4860:0000:2001:0000:0000:0000:0068", IP{0x20, 0x01, 0x48, 0x60, 0, 0, 0x20, 0x01, 0, 0, 0, 0, 0, 0, 0x00, 0x68}},
31
32 {"-0.0.0.0", nil},
33 {"0.-1.0.0", nil},
34 {"0.0.-2.0", nil},
35 {"0.0.0.-3", nil},
36 {"127.0.0.256", nil},
37 {"abc", nil},
38 {"123:", nil},
39 {"fe80::1%lo0", nil},
40 {"fe80::1%911", nil},
41 {"", nil},
42 {"a1:a2:a3:a4::b1:b2:b3:b4", nil},
43 {"127.001.002.003", nil},
44 {"::ffff:127.001.002.003", nil},
45 {"123.000.000.000", nil},
46 {"1.2..4", nil},
47 {"0123.0.0.1", nil},
48 }
49
50 func TestParseIP(t *testing.T) {
51 for _, tt := range parseIPTests {
52 if out := ParseIP(tt.in); !reflect.DeepEqual(out, tt.out) {
53 t.Errorf("ParseIP(%q) = %v, want %v", tt.in, out, tt.out)
54 }
55 if tt.in == "" {
56
57 continue
58 }
59 var out IP
60 if err := out.UnmarshalText([]byte(tt.in)); !reflect.DeepEqual(out, tt.out) || (tt.out == nil) != (err != nil) {
61 t.Errorf("IP.UnmarshalText(%q) = %v, %v, want %v", tt.in, out, err, tt.out)
62 }
63 }
64 }
65
66 func TestLookupWithIP(t *testing.T) {
67 _, err := LookupIP("")
68 if err == nil {
69 t.Errorf(`LookupIP("") succeeded, should fail`)
70 }
71 _, err = LookupHost("")
72 if err == nil {
73 t.Errorf(`LookupIP("") succeeded, should fail`)
74 }
75
76
77
78 for _, tt := range parseIPTests {
79 if tt.out != nil {
80 addrs, err := LookupHost(tt.in)
81 if len(addrs) != 1 || addrs[0] != tt.in || err != nil {
82 t.Errorf("LookupHost(%q) = %v, %v, want %v, nil", tt.in, addrs, err, []string{tt.in})
83 }
84 } else if !testing.Short() {
85
86
87
88 addrs, err := LookupHost(tt.in)
89 if err == nil {
90 t.Logf("warning: LookupHost(%q) = %v, want error", tt.in, addrs)
91 }
92 }
93
94 if tt.out != nil {
95 ips, err := LookupIP(tt.in)
96 if len(ips) != 1 || !reflect.DeepEqual(ips[0], tt.out) || err != nil {
97 t.Errorf("LookupIP(%q) = %v, %v, want %v, nil", tt.in, ips, err, []IP{tt.out})
98 }
99 } else if !testing.Short() {
100 ips, err := LookupIP(tt.in)
101
102 if err == nil {
103 t.Logf("warning: LookupIP(%q) = %v, want error", tt.in, ips)
104 }
105 }
106 }
107 }
108
109 func BenchmarkParseIP(b *testing.B) {
110 testHookUninstaller.Do(uninstallTestHooks)
111
112 for i := 0; i < b.N; i++ {
113 for _, tt := range parseIPTests {
114 ParseIP(tt.in)
115 }
116 }
117 }
118
119
120 func TestMarshalEmptyIP(t *testing.T) {
121 for _, in := range [][]byte{nil, []byte("")} {
122 var out = IP{1, 2, 3, 4}
123 if err := out.UnmarshalText(in); err != nil || out != nil {
124 t.Errorf("UnmarshalText(%v) = %v, %v; want nil, nil", in, out, err)
125 }
126 }
127 var ip IP
128 got, err := ip.MarshalText()
129 if err != nil {
130 t.Fatal(err)
131 }
132 if !reflect.DeepEqual(got, []byte("")) {
133 t.Errorf(`got %#v, want []byte("")`, got)
134 }
135 }
136
137 var ipStringTests = []*struct {
138 in IP
139 str string
140 byt []byte
141 error
142 }{
143
144 {
145 IP{192, 0, 2, 1},
146 "192.0.2.1",
147 []byte("192.0.2.1"),
148 nil,
149 },
150 {
151 IP{0, 0, 0, 0},
152 "0.0.0.0",
153 []byte("0.0.0.0"),
154 nil,
155 },
156
157
158 {
159 IP{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xff, 0xff, 192, 0, 2, 1},
160 "192.0.2.1",
161 []byte("192.0.2.1"),
162 nil,
163 },
164 {
165 IP{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xff, 0xff, 0, 0, 0, 0},
166 "0.0.0.0",
167 []byte("0.0.0.0"),
168 nil,
169 },
170
171
172 {
173 IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0, 0, 0, 0x1, 0x23, 0, 0x12, 0, 0x1},
174 "2001:db8::123:12:1",
175 []byte("2001:db8::123:12:1"),
176 nil,
177 },
178 {
179 IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0x1},
180 "2001:db8::1",
181 []byte("2001:db8::1"),
182 nil,
183 },
184 {
185 IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0x1, 0, 0, 0, 0x1, 0, 0, 0, 0x1},
186 "2001:db8:0:1:0:1:0:1",
187 []byte("2001:db8:0:1:0:1:0:1"),
188 nil,
189 },
190 {
191 IP{0x20, 0x1, 0xd, 0xb8, 0, 0x1, 0, 0, 0, 0x1, 0, 0, 0, 0x1, 0, 0},
192 "2001:db8:1:0:1:0:1:0",
193 []byte("2001:db8:1:0:1:0:1:0"),
194 nil,
195 },
196 {
197 IP{0x20, 0x1, 0, 0, 0, 0, 0, 0, 0, 0x1, 0, 0, 0, 0, 0, 0x1},
198 "2001::1:0:0:1",
199 []byte("2001::1:0:0:1"),
200 nil,
201 },
202 {
203 IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0, 0, 0x1, 0, 0, 0, 0, 0, 0},
204 "2001:db8:0:0:1::",
205 []byte("2001:db8:0:0:1::"),
206 nil,
207 },
208 {
209 IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0, 0, 0x1, 0, 0, 0, 0, 0, 0x1},
210 "2001:db8::1:0:0:1",
211 []byte("2001:db8::1:0:0:1"),
212 nil,
213 },
214 {
215 IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0, 0, 0xa, 0, 0xb, 0, 0xc, 0, 0xd},
216 "2001:db8::a:b:c:d",
217 []byte("2001:db8::a:b:c:d"),
218 nil,
219 },
220 {
221 IPv6unspecified,
222 "::",
223 []byte("::"),
224 nil,
225 },
226
227
228 {
229 nil,
230 "<nil>",
231 nil,
232 nil,
233 },
234
235
236 {
237 IP{0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef},
238 "?0123456789abcdef",
239 nil,
240 &AddrError{Err: "invalid IP address", Addr: "0123456789abcdef"},
241 },
242 }
243
244 func TestIPString(t *testing.T) {
245 for _, tt := range ipStringTests {
246 if out := tt.in.String(); out != tt.str {
247 t.Errorf("IP.String(%v) = %q, want %q", tt.in, out, tt.str)
248 }
249 if out, err := tt.in.MarshalText(); !bytes.Equal(out, tt.byt) || !reflect.DeepEqual(err, tt.error) {
250 t.Errorf("IP.MarshalText(%v) = %v, %v, want %v, %v", tt.in, out, err, tt.byt, tt.error)
251 }
252 }
253 }
254
255 var sink string
256
257 func BenchmarkIPString(b *testing.B) {
258 testHookUninstaller.Do(uninstallTestHooks)
259
260 b.Run("IPv4", func(b *testing.B) {
261 benchmarkIPString(b, IPv4len)
262 })
263
264 b.Run("IPv6", func(b *testing.B) {
265 benchmarkIPString(b, IPv6len)
266 })
267 }
268
269 func benchmarkIPString(b *testing.B, size int) {
270 b.ReportAllocs()
271 b.ResetTimer()
272 for i := 0; i < b.N; i++ {
273 for _, tt := range ipStringTests {
274 if tt.in != nil && len(tt.in) == size {
275 sink = tt.in.String()
276 }
277 }
278 }
279 }
280
281 var ipMaskTests = []struct {
282 in IP
283 mask IPMask
284 out IP
285 }{
286 {IPv4(192, 168, 1, 127), IPv4Mask(255, 255, 255, 128), IPv4(192, 168, 1, 0)},
287 {IPv4(192, 168, 1, 127), IPMask(ParseIP("255.255.255.192")), IPv4(192, 168, 1, 64)},
288 {IPv4(192, 168, 1, 127), IPMask(ParseIP("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffe0")), IPv4(192, 168, 1, 96)},
289 {IPv4(192, 168, 1, 127), IPv4Mask(255, 0, 255, 0), IPv4(192, 0, 1, 0)},
290 {ParseIP("2001:db8::1"), IPMask(ParseIP("ffff:ff80::")), ParseIP("2001:d80::")},
291 {ParseIP("2001:db8::1"), IPMask(ParseIP("f0f0:0f0f::")), ParseIP("2000:d08::")},
292 }
293
294 func TestIPMask(t *testing.T) {
295 for _, tt := range ipMaskTests {
296 if out := tt.in.Mask(tt.mask); out == nil || !tt.out.Equal(out) {
297 t.Errorf("IP(%v).Mask(%v) = %v, want %v", tt.in, tt.mask, out, tt.out)
298 }
299 }
300 }
301
302 var ipMaskStringTests = []struct {
303 in IPMask
304 out string
305 }{
306 {IPv4Mask(255, 255, 255, 240), "fffffff0"},
307 {IPv4Mask(255, 0, 128, 0), "ff008000"},
308 {IPMask(ParseIP("ffff:ff80::")), "ffffff80000000000000000000000000"},
309 {IPMask(ParseIP("ef00:ff80::cafe:0")), "ef00ff800000000000000000cafe0000"},
310 {nil, "<nil>"},
311 }
312
313 func TestIPMaskString(t *testing.T) {
314 for _, tt := range ipMaskStringTests {
315 if out := tt.in.String(); out != tt.out {
316 t.Errorf("IPMask.String(%v) = %q, want %q", tt.in, out, tt.out)
317 }
318 }
319 }
320
321 func BenchmarkIPMaskString(b *testing.B) {
322 testHookUninstaller.Do(uninstallTestHooks)
323
324 for i := 0; i < b.N; i++ {
325 for _, tt := range ipMaskStringTests {
326 sink = tt.in.String()
327 }
328 }
329 }
330
331 var parseCIDRTests = []struct {
332 in string
333 ip IP
334 net *IPNet
335 err error
336 }{
337 {"135.104.0.0/32", IPv4(135, 104, 0, 0), &IPNet{IP: IPv4(135, 104, 0, 0), Mask: IPv4Mask(255, 255, 255, 255)}, nil},
338 {"0.0.0.0/24", IPv4(0, 0, 0, 0), &IPNet{IP: IPv4(0, 0, 0, 0), Mask: IPv4Mask(255, 255, 255, 0)}, nil},
339 {"135.104.0.0/24", IPv4(135, 104, 0, 0), &IPNet{IP: IPv4(135, 104, 0, 0), Mask: IPv4Mask(255, 255, 255, 0)}, nil},
340 {"135.104.0.1/32", IPv4(135, 104, 0, 1), &IPNet{IP: IPv4(135, 104, 0, 1), Mask: IPv4Mask(255, 255, 255, 255)}, nil},
341 {"135.104.0.1/24", IPv4(135, 104, 0, 1), &IPNet{IP: IPv4(135, 104, 0, 0), Mask: IPv4Mask(255, 255, 255, 0)}, nil},
342 {"::1/128", ParseIP("::1"), &IPNet{IP: ParseIP("::1"), Mask: IPMask(ParseIP("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff"))}, nil},
343 {"abcd:2345::/127", ParseIP("abcd:2345::"), &IPNet{IP: ParseIP("abcd:2345::"), Mask: IPMask(ParseIP("ffff:ffff:ffff:ffff:ffff:ffff:ffff:fffe"))}, nil},
344 {"abcd:2345::/65", ParseIP("abcd:2345::"), &IPNet{IP: ParseIP("abcd:2345::"), Mask: IPMask(ParseIP("ffff:ffff:ffff:ffff:8000::"))}, nil},
345 {"abcd:2345::/64", ParseIP("abcd:2345::"), &IPNet{IP: ParseIP("abcd:2345::"), Mask: IPMask(ParseIP("ffff:ffff:ffff:ffff::"))}, nil},
346 {"abcd:2345::/63", ParseIP("abcd:2345::"), &IPNet{IP: ParseIP("abcd:2345::"), Mask: IPMask(ParseIP("ffff:ffff:ffff:fffe::"))}, nil},
347 {"abcd:2345::/33", ParseIP("abcd:2345::"), &IPNet{IP: ParseIP("abcd:2345::"), Mask: IPMask(ParseIP("ffff:ffff:8000::"))}, nil},
348 {"abcd:2345::/32", ParseIP("abcd:2345::"), &IPNet{IP: ParseIP("abcd:2345::"), Mask: IPMask(ParseIP("ffff:ffff::"))}, nil},
349 {"abcd:2344::/31", ParseIP("abcd:2344::"), &IPNet{IP: ParseIP("abcd:2344::"), Mask: IPMask(ParseIP("ffff:fffe::"))}, nil},
350 {"abcd:2300::/24", ParseIP("abcd:2300::"), &IPNet{IP: ParseIP("abcd:2300::"), Mask: IPMask(ParseIP("ffff:ff00::"))}, nil},
351 {"abcd:2345::/24", ParseIP("abcd:2345::"), &IPNet{IP: ParseIP("abcd:2300::"), Mask: IPMask(ParseIP("ffff:ff00::"))}, nil},
352 {"2001:DB8::/48", ParseIP("2001:DB8::"), &IPNet{IP: ParseIP("2001:DB8::"), Mask: IPMask(ParseIP("ffff:ffff:ffff::"))}, nil},
353 {"2001:DB8::1/48", ParseIP("2001:DB8::1"), &IPNet{IP: ParseIP("2001:DB8::"), Mask: IPMask(ParseIP("ffff:ffff:ffff::"))}, nil},
354 {"192.168.1.1/255.255.255.0", nil, nil, &ParseError{Type: "CIDR address", Text: "192.168.1.1/255.255.255.0"}},
355 {"192.168.1.1/35", nil, nil, &ParseError{Type: "CIDR address", Text: "192.168.1.1/35"}},
356 {"2001:db8::1/-1", nil, nil, &ParseError{Type: "CIDR address", Text: "2001:db8::1/-1"}},
357 {"2001:db8::1/-0", nil, nil, &ParseError{Type: "CIDR address", Text: "2001:db8::1/-0"}},
358 {"-0.0.0.0/32", nil, nil, &ParseError{Type: "CIDR address", Text: "-0.0.0.0/32"}},
359 {"0.-1.0.0/32", nil, nil, &ParseError{Type: "CIDR address", Text: "0.-1.0.0/32"}},
360 {"0.0.-2.0/32", nil, nil, &ParseError{Type: "CIDR address", Text: "0.0.-2.0/32"}},
361 {"0.0.0.-3/32", nil, nil, &ParseError{Type: "CIDR address", Text: "0.0.0.-3/32"}},
362 {"0.0.0.0/-0", nil, nil, &ParseError{Type: "CIDR address", Text: "0.0.0.0/-0"}},
363 {"127.000.000.001/32", nil, nil, &ParseError{Type: "CIDR address", Text: "127.000.000.001/32"}},
364 {"", nil, nil, &ParseError{Type: "CIDR address", Text: ""}},
365 }
366
367 func TestParseCIDR(t *testing.T) {
368 for _, tt := range parseCIDRTests {
369 ip, net, err := ParseCIDR(tt.in)
370 if !reflect.DeepEqual(err, tt.err) {
371 t.Errorf("ParseCIDR(%q) = %v, %v; want %v, %v", tt.in, ip, net, tt.ip, tt.net)
372 }
373 if err == nil && (!tt.ip.Equal(ip) || !tt.net.IP.Equal(net.IP) || !reflect.DeepEqual(net.Mask, tt.net.Mask)) {
374 t.Errorf("ParseCIDR(%q) = %v, {%v, %v}; want %v, {%v, %v}", tt.in, ip, net.IP, net.Mask, tt.ip, tt.net.IP, tt.net.Mask)
375 }
376 }
377 }
378
379 var ipNetContainsTests = []struct {
380 ip IP
381 net *IPNet
382 ok bool
383 }{
384 {IPv4(172, 16, 1, 1), &IPNet{IP: IPv4(172, 16, 0, 0), Mask: CIDRMask(12, 32)}, true},
385 {IPv4(172, 24, 0, 1), &IPNet{IP: IPv4(172, 16, 0, 0), Mask: CIDRMask(13, 32)}, false},
386 {IPv4(192, 168, 0, 3), &IPNet{IP: IPv4(192, 168, 0, 0), Mask: IPv4Mask(0, 0, 255, 252)}, true},
387 {IPv4(192, 168, 0, 4), &IPNet{IP: IPv4(192, 168, 0, 0), Mask: IPv4Mask(0, 255, 0, 252)}, false},
388 {ParseIP("2001:db8:1:2::1"), &IPNet{IP: ParseIP("2001:db8:1::"), Mask: CIDRMask(47, 128)}, true},
389 {ParseIP("2001:db8:1:2::1"), &IPNet{IP: ParseIP("2001:db8:2::"), Mask: CIDRMask(47, 128)}, false},
390 {ParseIP("2001:db8:1:2::1"), &IPNet{IP: ParseIP("2001:db8:1::"), Mask: IPMask(ParseIP("ffff:0:ffff::"))}, true},
391 {ParseIP("2001:db8:1:2::1"), &IPNet{IP: ParseIP("2001:db8:1::"), Mask: IPMask(ParseIP("0:0:0:ffff::"))}, false},
392 }
393
394 func TestIPNetContains(t *testing.T) {
395 for _, tt := range ipNetContainsTests {
396 if ok := tt.net.Contains(tt.ip); ok != tt.ok {
397 t.Errorf("IPNet(%v).Contains(%v) = %v, want %v", tt.net, tt.ip, ok, tt.ok)
398 }
399 }
400 }
401
402 var ipNetStringTests = []struct {
403 in *IPNet
404 out string
405 }{
406 {&IPNet{IP: IPv4(192, 168, 1, 0), Mask: CIDRMask(26, 32)}, "192.168.1.0/26"},
407 {&IPNet{IP: IPv4(192, 168, 1, 0), Mask: IPv4Mask(255, 0, 255, 0)}, "192.168.1.0/ff00ff00"},
408 {&IPNet{IP: ParseIP("2001:db8::"), Mask: CIDRMask(55, 128)}, "2001:db8::/55"},
409 {&IPNet{IP: ParseIP("2001:db8::"), Mask: IPMask(ParseIP("8000:f123:0:cafe::"))}, "2001:db8::/8000f1230000cafe0000000000000000"},
410 {nil, "<nil>"},
411 }
412
413 func TestIPNetString(t *testing.T) {
414 for _, tt := range ipNetStringTests {
415 if out := tt.in.String(); out != tt.out {
416 t.Errorf("IPNet.String(%v) = %q, want %q", tt.in, out, tt.out)
417 }
418 }
419 }
420
421 var cidrMaskTests = []struct {
422 ones int
423 bits int
424 out IPMask
425 }{
426 {0, 32, IPv4Mask(0, 0, 0, 0)},
427 {12, 32, IPv4Mask(255, 240, 0, 0)},
428 {24, 32, IPv4Mask(255, 255, 255, 0)},
429 {32, 32, IPv4Mask(255, 255, 255, 255)},
430 {0, 128, IPMask{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}},
431 {4, 128, IPMask{0xf0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}},
432 {48, 128, IPMask{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}},
433 {128, 128, IPMask{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff}},
434 {33, 32, nil},
435 {32, 33, nil},
436 {-1, 128, nil},
437 {128, -1, nil},
438 }
439
440 func TestCIDRMask(t *testing.T) {
441 for _, tt := range cidrMaskTests {
442 if out := CIDRMask(tt.ones, tt.bits); !reflect.DeepEqual(out, tt.out) {
443 t.Errorf("CIDRMask(%v, %v) = %v, want %v", tt.ones, tt.bits, out, tt.out)
444 }
445 }
446 }
447
448 var (
449 v4addr = IP{192, 168, 0, 1}
450 v4mappedv6addr = IP{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xff, 0xff, 192, 168, 0, 1}
451 v6addr = IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0, 0, 0, 0x1, 0x23, 0, 0x12, 0, 0x1}
452 v4mask = IPMask{255, 255, 255, 0}
453 v4mappedv6mask = IPMask{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 255, 255, 255, 0}
454 v6mask = IPMask{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0, 0, 0, 0, 0, 0, 0, 0}
455 badaddr = IP{192, 168, 0}
456 badmask = IPMask{255, 255, 0}
457 v4maskzero = IPMask{0, 0, 0, 0}
458 )
459
460 var networkNumberAndMaskTests = []struct {
461 in IPNet
462 out IPNet
463 }{
464 {IPNet{IP: v4addr, Mask: v4mask}, IPNet{IP: v4addr, Mask: v4mask}},
465 {IPNet{IP: v4addr, Mask: v4mappedv6mask}, IPNet{IP: v4addr, Mask: v4mask}},
466 {IPNet{IP: v4mappedv6addr, Mask: v4mappedv6mask}, IPNet{IP: v4addr, Mask: v4mask}},
467 {IPNet{IP: v4mappedv6addr, Mask: v6mask}, IPNet{IP: v4addr, Mask: v4maskzero}},
468 {IPNet{IP: v4addr, Mask: v6mask}, IPNet{IP: v4addr, Mask: v4maskzero}},
469 {IPNet{IP: v6addr, Mask: v6mask}, IPNet{IP: v6addr, Mask: v6mask}},
470 {IPNet{IP: v6addr, Mask: v4mappedv6mask}, IPNet{IP: v6addr, Mask: v4mappedv6mask}},
471 {in: IPNet{IP: v6addr, Mask: v4mask}},
472 {in: IPNet{IP: v4addr, Mask: badmask}},
473 {in: IPNet{IP: v4mappedv6addr, Mask: badmask}},
474 {in: IPNet{IP: v6addr, Mask: badmask}},
475 {in: IPNet{IP: badaddr, Mask: v4mask}},
476 {in: IPNet{IP: badaddr, Mask: v4mappedv6mask}},
477 {in: IPNet{IP: badaddr, Mask: v6mask}},
478 {in: IPNet{IP: badaddr, Mask: badmask}},
479 }
480
481 func TestNetworkNumberAndMask(t *testing.T) {
482 for _, tt := range networkNumberAndMaskTests {
483 ip, m := networkNumberAndMask(&tt.in)
484 out := &IPNet{IP: ip, Mask: m}
485 if !reflect.DeepEqual(&tt.out, out) {
486 t.Errorf("networkNumberAndMask(%v) = %v, want %v", tt.in, out, &tt.out)
487 }
488 }
489 }
490
491 func TestSplitHostPort(t *testing.T) {
492 for _, tt := range []struct {
493 hostPort string
494 host string
495 port string
496 }{
497
498 {"localhost:http", "localhost", "http"},
499 {"localhost:80", "localhost", "80"},
500
501
502 {"localhost%lo0:http", "localhost%lo0", "http"},
503 {"localhost%lo0:80", "localhost%lo0", "80"},
504 {"[localhost%lo0]:http", "localhost%lo0", "http"},
505 {"[localhost%lo0]:80", "localhost%lo0", "80"},
506
507
508 {"127.0.0.1:http", "127.0.0.1", "http"},
509 {"127.0.0.1:80", "127.0.0.1", "80"},
510 {"[::1]:http", "::1", "http"},
511 {"[::1]:80", "::1", "80"},
512
513
514 {"[::1%lo0]:http", "::1%lo0", "http"},
515 {"[::1%lo0]:80", "::1%lo0", "80"},
516
517
518 {":http", "", "http"},
519 {":80", "", "80"},
520
521
522 {"golang.org:", "golang.org", ""},
523 {"127.0.0.1:", "127.0.0.1", ""},
524 {"[::1]:", "::1", ""},
525
526
527 {"golang.org:https%foo", "golang.org", "https%foo"},
528 } {
529 if host, port, err := SplitHostPort(tt.hostPort); host != tt.host || port != tt.port || err != nil {
530 t.Errorf("SplitHostPort(%q) = %q, %q, %v; want %q, %q, nil", tt.hostPort, host, port, err, tt.host, tt.port)
531 }
532 }
533
534 for _, tt := range []struct {
535 hostPort string
536 err string
537 }{
538 {"golang.org", "missing port in address"},
539 {"127.0.0.1", "missing port in address"},
540 {"[::1]", "missing port in address"},
541 {"[fe80::1%lo0]", "missing port in address"},
542 {"[localhost%lo0]", "missing port in address"},
543 {"localhost%lo0", "missing port in address"},
544
545 {"::1", "too many colons in address"},
546 {"fe80::1%lo0", "too many colons in address"},
547 {"fe80::1%lo0:80", "too many colons in address"},
548
549
550
551 {"[foo:bar]", "missing port in address"},
552 {"[foo:bar]baz", "missing port in address"},
553 {"[foo]bar:baz", "missing port in address"},
554
555 {"[foo]:[bar]:baz", "too many colons in address"},
556
557 {"[foo]:[bar]baz", "unexpected '[' in address"},
558 {"foo[bar]:baz", "unexpected '[' in address"},
559
560 {"foo]bar:baz", "unexpected ']' in address"},
561 } {
562 if host, port, err := SplitHostPort(tt.hostPort); err == nil {
563 t.Errorf("SplitHostPort(%q) should have failed", tt.hostPort)
564 } else {
565 e := err.(*AddrError)
566 if e.Err != tt.err {
567 t.Errorf("SplitHostPort(%q) = _, _, %q; want %q", tt.hostPort, e.Err, tt.err)
568 }
569 if host != "" || port != "" {
570 t.Errorf("SplitHostPort(%q) = %q, %q, err; want %q, %q, err on failure", tt.hostPort, host, port, "", "")
571 }
572 }
573 }
574 }
575
576 func TestJoinHostPort(t *testing.T) {
577 for _, tt := range []struct {
578 host string
579 port string
580 hostPort string
581 }{
582
583 {"localhost", "http", "localhost:http"},
584 {"localhost", "80", "localhost:80"},
585
586
587 {"localhost%lo0", "http", "localhost%lo0:http"},
588 {"localhost%lo0", "80", "localhost%lo0:80"},
589
590
591 {"127.0.0.1", "http", "127.0.0.1:http"},
592 {"127.0.0.1", "80", "127.0.0.1:80"},
593 {"::1", "http", "[::1]:http"},
594 {"::1", "80", "[::1]:80"},
595
596
597 {"::1%lo0", "http", "[::1%lo0]:http"},
598 {"::1%lo0", "80", "[::1%lo0]:80"},
599
600
601 {"", "http", ":http"},
602 {"", "80", ":80"},
603
604
605 {"golang.org", "", "golang.org:"},
606 {"127.0.0.1", "", "127.0.0.1:"},
607 {"::1", "", "[::1]:"},
608
609
610 {"golang.org", "https%foo", "golang.org:https%foo"},
611 } {
612 if hostPort := JoinHostPort(tt.host, tt.port); hostPort != tt.hostPort {
613 t.Errorf("JoinHostPort(%q, %q) = %q; want %q", tt.host, tt.port, hostPort, tt.hostPort)
614 }
615 }
616 }
617
618 var ipAddrFamilyTests = []struct {
619 in IP
620 af4 bool
621 af6 bool
622 }{
623 {IPv4bcast, true, false},
624 {IPv4allsys, true, false},
625 {IPv4allrouter, true, false},
626 {IPv4zero, true, false},
627 {IPv4(224, 0, 0, 1), true, false},
628 {IPv4(127, 0, 0, 1), true, false},
629 {IPv4(240, 0, 0, 1), true, false},
630 {IPv6unspecified, false, true},
631 {IPv6loopback, false, true},
632 {IPv6interfacelocalallnodes, false, true},
633 {IPv6linklocalallnodes, false, true},
634 {IPv6linklocalallrouters, false, true},
635 {ParseIP("ff05::a:b:c:d"), false, true},
636 {ParseIP("fe80::1:2:3:4"), false, true},
637 {ParseIP("2001:db8::123:12:1"), false, true},
638 }
639
640 func TestIPAddrFamily(t *testing.T) {
641 for _, tt := range ipAddrFamilyTests {
642 if af := tt.in.To4() != nil; af != tt.af4 {
643 t.Errorf("verifying IPv4 address family for %q = %v, want %v", tt.in, af, tt.af4)
644 }
645 if af := len(tt.in) == IPv6len && tt.in.To4() == nil; af != tt.af6 {
646 t.Errorf("verifying IPv6 address family for %q = %v, want %v", tt.in, af, tt.af6)
647 }
648 }
649 }
650
651 var ipAddrScopeTests = []struct {
652 scope func(IP) bool
653 in IP
654 ok bool
655 }{
656 {IP.IsUnspecified, IPv4zero, true},
657 {IP.IsUnspecified, IPv4(127, 0, 0, 1), false},
658 {IP.IsUnspecified, IPv6unspecified, true},
659 {IP.IsUnspecified, IPv6interfacelocalallnodes, false},
660 {IP.IsUnspecified, nil, false},
661 {IP.IsLoopback, IPv4(127, 0, 0, 1), true},
662 {IP.IsLoopback, IPv4(127, 255, 255, 254), true},
663 {IP.IsLoopback, IPv4(128, 1, 2, 3), false},
664 {IP.IsLoopback, IPv6loopback, true},
665 {IP.IsLoopback, IPv6linklocalallrouters, false},
666 {IP.IsLoopback, nil, false},
667 {IP.IsMulticast, IPv4(224, 0, 0, 0), true},
668 {IP.IsMulticast, IPv4(239, 0, 0, 0), true},
669 {IP.IsMulticast, IPv4(240, 0, 0, 0), false},
670 {IP.IsMulticast, IPv6linklocalallnodes, true},
671 {IP.IsMulticast, IP{0xff, 0x05, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, true},
672 {IP.IsMulticast, IP{0xfe, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, false},
673 {IP.IsMulticast, nil, false},
674 {IP.IsInterfaceLocalMulticast, IPv4(224, 0, 0, 0), false},
675 {IP.IsInterfaceLocalMulticast, IPv4(0xff, 0x01, 0, 0), false},
676 {IP.IsInterfaceLocalMulticast, IPv6interfacelocalallnodes, true},
677 {IP.IsInterfaceLocalMulticast, nil, false},
678 {IP.IsLinkLocalMulticast, IPv4(224, 0, 0, 0), true},
679 {IP.IsLinkLocalMulticast, IPv4(239, 0, 0, 0), false},
680 {IP.IsLinkLocalMulticast, IPv4(0xff, 0x02, 0, 0), false},
681 {IP.IsLinkLocalMulticast, IPv6linklocalallrouters, true},
682 {IP.IsLinkLocalMulticast, IP{0xff, 0x05, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, false},
683 {IP.IsLinkLocalMulticast, nil, false},
684 {IP.IsLinkLocalUnicast, IPv4(169, 254, 0, 0), true},
685 {IP.IsLinkLocalUnicast, IPv4(169, 255, 0, 0), false},
686 {IP.IsLinkLocalUnicast, IPv4(0xfe, 0x80, 0, 0), false},
687 {IP.IsLinkLocalUnicast, IP{0xfe, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, true},
688 {IP.IsLinkLocalUnicast, IP{0xfe, 0xc0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, false},
689 {IP.IsLinkLocalUnicast, nil, false},
690 {IP.IsGlobalUnicast, IPv4(240, 0, 0, 0), true},
691 {IP.IsGlobalUnicast, IPv4(232, 0, 0, 0), false},
692 {IP.IsGlobalUnicast, IPv4(169, 254, 0, 0), false},
693 {IP.IsGlobalUnicast, IPv4bcast, false},
694 {IP.IsGlobalUnicast, IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0, 0, 0, 0x1, 0x23, 0, 0x12, 0, 0x1}, true},
695 {IP.IsGlobalUnicast, IP{0xfe, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, false},
696 {IP.IsGlobalUnicast, IP{0xff, 0x05, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, false},
697 {IP.IsGlobalUnicast, nil, false},
698 {IP.IsPrivate, nil, false},
699 {IP.IsPrivate, IPv4(1, 1, 1, 1), false},
700 {IP.IsPrivate, IPv4(9, 255, 255, 255), false},
701 {IP.IsPrivate, IPv4(10, 0, 0, 0), true},
702 {IP.IsPrivate, IPv4(10, 255, 255, 255), true},
703 {IP.IsPrivate, IPv4(11, 0, 0, 0), false},
704 {IP.IsPrivate, IPv4(172, 15, 255, 255), false},
705 {IP.IsPrivate, IPv4(172, 16, 0, 0), true},
706 {IP.IsPrivate, IPv4(172, 16, 255, 255), true},
707 {IP.IsPrivate, IPv4(172, 23, 18, 255), true},
708 {IP.IsPrivate, IPv4(172, 31, 255, 255), true},
709 {IP.IsPrivate, IPv4(172, 31, 0, 0), true},
710 {IP.IsPrivate, IPv4(172, 32, 0, 0), false},
711 {IP.IsPrivate, IPv4(192, 167, 255, 255), false},
712 {IP.IsPrivate, IPv4(192, 168, 0, 0), true},
713 {IP.IsPrivate, IPv4(192, 168, 255, 255), true},
714 {IP.IsPrivate, IPv4(192, 169, 0, 0), false},
715 {IP.IsPrivate, IP{0xfb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff}, false},
716 {IP.IsPrivate, IP{0xfc, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, true},
717 {IP.IsPrivate, IP{0xfc, 0xff, 0x12, 0, 0, 0, 0, 0x44, 0, 0, 0, 0, 0, 0, 0, 0}, true},
718 {IP.IsPrivate, IP{0xfd, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff}, true},
719 {IP.IsPrivate, IP{0xfe, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}, false},
720 }
721
722 func name(f any) string {
723 return runtime.FuncForPC(reflect.ValueOf(f).Pointer()).Name()
724 }
725
726 func TestIPAddrScope(t *testing.T) {
727 for _, tt := range ipAddrScopeTests {
728 if ok := tt.scope(tt.in); ok != tt.ok {
729 t.Errorf("%s(%q) = %v, want %v", name(tt.scope), tt.in, ok, tt.ok)
730 }
731 ip := tt.in.To4()
732 if ip == nil {
733 continue
734 }
735 if ok := tt.scope(ip); ok != tt.ok {
736 t.Errorf("%s(%q) = %v, want %v", name(tt.scope), ip, ok, tt.ok)
737 }
738 }
739 }
740
741 func BenchmarkIPEqual(b *testing.B) {
742 b.Run("IPv4", func(b *testing.B) {
743 benchmarkIPEqual(b, IPv4len)
744 })
745 b.Run("IPv6", func(b *testing.B) {
746 benchmarkIPEqual(b, IPv6len)
747 })
748 }
749
750 func benchmarkIPEqual(b *testing.B, size int) {
751 ips := make([]IP, 1000)
752 for i := range ips {
753 ips[i] = make(IP, size)
754 rand.Read(ips[i])
755 }
756
757 for i := 0; i < b.N/2; i++ {
758 x := ips[i%len(ips)]
759 y := ips[i%len(ips)]
760 x.Equal(y)
761 }
762
763 for i := 0; i < b.N/2; i++ {
764 x := ips[i%len(ips)]
765 y := ips[(i+1)%len(ips)]
766 x.Equal(y)
767 }
768 }
769
View as plain text