libgo/go/net/ip_test.go

   1 // Copyright 2009 The Go Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 package net
   6
   7 import (
   8         "bytes"
   9         "reflect"
  10         "testing"
  11         "os"
  12 )
  13
  14 func isEqual(a, b []byte) bool {
  15         if a == nil && b == nil {
  16                 return true
  17         }
  18         if a == nil || b == nil {
  19                 return false
  20         }
  21         return bytes.Equal(a, b)
  22 }
  23
  24 var parseiptests = []struct {
  25         in  string
  26         out IP
  27 }{
  28         {"127.0.1.2", IPv4(127, 0, 1, 2)},
  29         {"127.0.0.1", IPv4(127, 0, 0, 1)},
  30         {"127.0.0.256", nil},
  31         {"abc", nil},
  32         {"::ffff:127.0.0.1", IPv4(127, 0, 0, 1)},
  33         {"2001:4860:0:2001::68",
  34                 IP{0x20, 0x01, 0x48, 0x60, 0, 0, 0x20, 0x01,
  35                         0, 0, 0, 0, 0, 0, 0x00, 0x68,
  36                 },
  37         },
  38         {"::ffff:4a7d:1363", IPv4(74, 125, 19, 99)},
  39 }
  40
  41 func TestParseIP(t *testing.T) {
  42         for _, tt := range parseiptests {
  43                 if out := ParseIP(tt.in); !isEqual(out, tt.out) {
  44                         t.Errorf("ParseIP(%#q) = %v, want %v", tt.in, out, tt.out)
  45                 }
  46         }
  47 }
  48
  49 var ipstringtests = []struct {
  50         in  IP
  51         out string
  52 }{
  53         // cf. RFC 5952 (A Recommendation for IPv6 Address Text Representation)
  54         {IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0,
  55                 0, 0, 0x1, 0x23, 0, 0x12, 0, 0x1},
  56                 "2001:db8::123:12:1"},
  57         {IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0,
  58                 0, 0, 0, 0, 0, 0, 0, 0x1},
  59                 "2001:db8::1"},
  60         {IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0x1,
  61                 0, 0, 0, 0x1, 0, 0, 0, 0x1},
  62                 "2001:db8:0:1:0:1:0:1"},
  63         {IP{0x20, 0x1, 0xd, 0xb8, 0, 0x1, 0, 0,
  64                 0, 0x1, 0, 0, 0, 0x1, 0, 0},
  65                 "2001:db8:1:0:1:0:1:0"},
  66         {IP{0x20, 0x1, 0, 0, 0, 0, 0, 0,
  67                 0, 0x1, 0, 0, 0, 0, 0, 0x1},
  68                 "2001::1:0:0:1"},
  69         {IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0,
  70                 0, 0x1, 0, 0, 0, 0, 0, 0},
  71                 "2001:db8:0:0:1::"},
  72         {IP{0x20, 0x1, 0xd, 0xb8, 0, 0, 0, 0,
  73                 0, 0x1, 0, 0, 0, 0, 0, 0x1},
  74                 "2001:db8::1:0:0:1"},
  75         {IP{0x20, 0x1, 0xD, 0xB8, 0, 0, 0, 0,
  76                 0, 0xA, 0, 0xB, 0, 0xC, 0, 0xD},
  77                 "2001:db8::a:b:c:d"},
  78 }
  79
  80 func TestIPString(t *testing.T) {
  81         for _, tt := range ipstringtests {
  82                 if out := tt.in.String(); out != tt.out {
  83                         t.Errorf("IP.String(%v) = %#q, want %#q", tt.in, out, tt.out)
  84                 }
  85         }
  86 }
  87
  88 var parsecidrtests = []struct {
  89         in   string
  90         ip   IP
  91         mask IPMask
  92         err  os.Error
  93 }{
  94         {"135.104.0.0/32", IPv4(135, 104, 0, 0), IPv4Mask(255, 255, 255, 255), nil},
  95         {"0.0.0.0/24", IPv4(0, 0, 0, 0), IPv4Mask(255, 255, 255, 0), nil},
  96         {"135.104.0.0/24", IPv4(135, 104, 0, 0), IPv4Mask(255, 255, 255, 0), nil},
  97         {"135.104.0.1/32", IPv4(135, 104, 0, 1), IPv4Mask(255, 255, 255, 255), nil},
  98         {"135.104.0.1/24", nil, nil, &ParseError{"CIDR address", "135.104.0.1/24"}},
  99         {"::1/128", ParseIP("::1"), IPMask(ParseIP("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff")), nil},
 100         {"abcd:2345::/127", ParseIP("abcd:2345::"), IPMask(ParseIP("ffff:ffff:ffff:ffff:ffff:ffff:ffff:fffe")), nil},
 101         {"abcd:2345::/65", ParseIP("abcd:2345::"), IPMask(ParseIP("ffff:ffff:ffff:ffff:8000::")), nil},
 102         {"abcd:2345::/64", ParseIP("abcd:2345::"), IPMask(ParseIP("ffff:ffff:ffff:ffff::")), nil},
 103         {"abcd:2345::/63", ParseIP("abcd:2345::"), IPMask(ParseIP("ffff:ffff:ffff:fffe::")), nil},
 104         {"abcd:2345::/33", ParseIP("abcd:2345::"), IPMask(ParseIP("ffff:ffff:8000::")), nil},
 105         {"abcd:2345::/32", ParseIP("abcd:2345::"), IPMask(ParseIP("ffff:ffff::")), nil},
 106         {"abcd:2344::/31", ParseIP("abcd:2344::"), IPMask(ParseIP("ffff:fffe::")), nil},
 107         {"abcd:2300::/24", ParseIP("abcd:2300::"), IPMask(ParseIP("ffff:ff00::")), nil},
 108         {"abcd:2345::/24", nil, nil, &ParseError{"CIDR address", "abcd:2345::/24"}},
 109         {"2001:DB8::/48", ParseIP("2001:DB8::"), IPMask(ParseIP("ffff:ffff:ffff::")), nil},
 110 }
 111
 112 func TestParseCIDR(t *testing.T) {
 113         for _, tt := range parsecidrtests {
 114                 if ip, mask, err := ParseCIDR(tt.in); !isEqual(ip, tt.ip) || !isEqual(mask, tt.mask) || !reflect.DeepEqual(err, tt.err) {
 115                         t.Errorf("ParseCIDR(%q) = %v, %v, %v; want %v, %v, %v", tt.in, ip, mask, err, tt.ip, tt.mask, tt.err)
 116                 }
 117         }
 118 }
 119
 120 var splitjointests = []struct {
 121         Host string
 122         Port string
 123         Join string
 124 }{
 125         {"www.google.com", "80", "www.google.com:80"},
 126         {"127.0.0.1", "1234", "127.0.0.1:1234"},
 127         {"::1", "80", "[::1]:80"},
 128 }
 129
 130 func TestSplitHostPort(t *testing.T) {
 131         for _, tt := range splitjointests {
 132                 if host, port, err := SplitHostPort(tt.Join); host != tt.Host || port != tt.Port || err != nil {
 133                         t.Errorf("SplitHostPort(%q) = %q, %q, %v; want %q, %q, nil", tt.Join, host, port, err, tt.Host, tt.Port)
 134                 }
 135         }
 136 }
 137
 138 func TestJoinHostPort(t *testing.T) {
 139         for _, tt := range splitjointests {
 140                 if join := JoinHostPort(tt.Host, tt.Port); join != tt.Join {
 141                         t.Errorf("JoinHostPort(%q, %q) = %q; want %q", tt.Host, tt.Port, join, tt.Join)
 142                 }
 143         }
 144 }