Source file
src/net/external_test.go
1
2
3
4
5 package net
6
7 import (
8 "fmt"
9 "internal/testenv"
10 "io"
11 "strings"
12 "testing"
13 )
14
15 func TestResolveGoogle(t *testing.T) {
16 testenv.MustHaveExternalNetwork(t)
17
18 if !supportsIPv4() || !supportsIPv6() || !*testIPv4 || !*testIPv6 {
19 t.Skip("both IPv4 and IPv6 are required")
20 }
21
22 for _, network := range []string{"tcp", "tcp4", "tcp6"} {
23 addr, err := ResolveTCPAddr(network, "www.google.com:http")
24 if err != nil {
25 t.Error(err)
26 continue
27 }
28 switch {
29 case network == "tcp" && addr.IP.To4() == nil:
30 fallthrough
31 case network == "tcp4" && addr.IP.To4() == nil:
32 t.Errorf("got %v; want an IPv4 address on %s", addr, network)
33 case network == "tcp6" && (addr.IP.To16() == nil || addr.IP.To4() != nil):
34 t.Errorf("got %v; want an IPv6 address on %s", addr, network)
35 }
36 }
37 }
38
39 var dialGoogleTests = []struct {
40 dial func(string, string) (Conn, error)
41 unreachableNetwork string
42 networks []string
43 addrs []string
44 }{
45 {
46 dial: (&Dialer{DualStack: true}).Dial,
47 networks: []string{"tcp", "tcp4", "tcp6"},
48 addrs: []string{"www.google.com:http"},
49 },
50 {
51 dial: Dial,
52 unreachableNetwork: "tcp6",
53 networks: []string{"tcp", "tcp4"},
54 },
55 {
56 dial: Dial,
57 unreachableNetwork: "tcp4",
58 networks: []string{"tcp", "tcp6"},
59 },
60 }
61
62 func TestDialGoogle(t *testing.T) {
63 testenv.MustHaveExternalNetwork(t)
64
65 if !supportsIPv4() || !supportsIPv6() || !*testIPv4 || !*testIPv6 {
66 t.Skip("both IPv4 and IPv6 are required")
67 }
68
69 var err error
70 dialGoogleTests[1].addrs, dialGoogleTests[2].addrs, err = googleLiteralAddrs()
71 if err != nil {
72 t.Error(err)
73 }
74 for _, tt := range dialGoogleTests {
75 for _, network := range tt.networks {
76 disableSocketConnect(tt.unreachableNetwork)
77 for _, addr := range tt.addrs {
78 if err := fetchGoogle(tt.dial, network, addr); err != nil {
79 t.Error(err)
80 }
81 }
82 enableSocketConnect()
83 }
84 }
85 }
86
87 var (
88 literalAddrs4 = [...]string{
89 "%d.%d.%d.%d:80",
90 "www.google.com:80",
91 "%d.%d.%d.%d:http",
92 "www.google.com:http",
93 "[::ffff:%d.%d.%d.%d]:80",
94 "[::ffff:%02x%02x:%02x%02x]:80",
95 "[0:0:0:0:0000:ffff:%d.%d.%d.%d]:80",
96 "[0:0:0:0:000000:ffff:%d.%d.%d.%d]:80",
97 "[0:0:0:0::ffff:%d.%d.%d.%d]:80",
98 }
99 literalAddrs6 = [...]string{
100 "[%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:80",
101 "ipv6.google.com:80",
102 "[%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:http",
103 "ipv6.google.com:http",
104 }
105 )
106
107 func googleLiteralAddrs() (lits4, lits6 []string, err error) {
108 ips, err := LookupIP("www.google.com")
109 if err != nil {
110 return nil, nil, err
111 }
112 if len(ips) == 0 {
113 return nil, nil, nil
114 }
115 var ip4, ip6 IP
116 for _, ip := range ips {
117 if ip4 == nil && ip.To4() != nil {
118 ip4 = ip.To4()
119 }
120 if ip6 == nil && ip.To16() != nil && ip.To4() == nil {
121 ip6 = ip.To16()
122 }
123 if ip4 != nil && ip6 != nil {
124 break
125 }
126 }
127 if ip4 != nil {
128 for i, lit4 := range literalAddrs4 {
129 if strings.Contains(lit4, "%") {
130 literalAddrs4[i] = fmt.Sprintf(lit4, ip4[0], ip4[1], ip4[2], ip4[3])
131 }
132 }
133 lits4 = literalAddrs4[:]
134 }
135 if ip6 != nil {
136 for i, lit6 := range literalAddrs6 {
137 if strings.Contains(lit6, "%") {
138 literalAddrs6[i] = fmt.Sprintf(lit6, ip6[0], ip6[1], ip6[2], ip6[3], ip6[4], ip6[5], ip6[6], ip6[7], ip6[8], ip6[9], ip6[10], ip6[11], ip6[12], ip6[13], ip6[14], ip6[15])
139 }
140 }
141 lits6 = literalAddrs6[:]
142 }
143 return
144 }
145
146 func fetchGoogle(dial func(string, string) (Conn, error), network, address string) error {
147 c, err := dial(network, address)
148 if err != nil {
149 return err
150 }
151 defer c.Close()
152 req := []byte("GET /robots.txt HTTP/1.0\r\nHost: www.google.com\r\n\r\n")
153 if _, err := c.Write(req); err != nil {
154 return err
155 }
156 b := make([]byte, 1000)
157 n, err := io.ReadFull(c, b)
158 if err != nil {
159 return err
160 }
161 if n < 1000 {
162 return fmt.Errorf("short read from %s:%s->%s", network, c.RemoteAddr(), c.LocalAddr())
163 }
164 return nil
165 }
166
View as plain text