blob: 03c4499720f02cb88e5c9af2a6d145c4edf339d6 [file] [log] [blame]
// Copyright 2009 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 (
"flag"
"fmt"
"io"
"strings"
"syscall"
"testing"
)
// If an IPv6 tunnel is running, we can try dialing a real IPv6 address.
var testIPv6 = flag.Bool("ipv6", false, "assume ipv6 tunnel is present")
// fd is already connected to the destination, port 80.
// Run an HTTP request to fetch the appropriate page.
func fetchGoogle(t *testing.T, fd Conn, network, addr string) {
req := []byte("GET /robots.txt HTTP/1.0\r\nHost: www.google.com\r\n\r\n")
n, err := fd.Write(req)
buf := make([]byte, 1000)
n, err = io.ReadFull(fd, buf)
if n < 1000 {
t.Errorf("fetchGoogle: short HTTP read from %s %s - %v", network, addr, err)
return
}
}
func doDial(t *testing.T, network, addr string) {
fd, err := Dial(network, addr)
if err != nil {
t.Errorf("Dial(%q, %q, %q) = _, %v", network, "", addr, err)
return
}
fetchGoogle(t, fd, network, addr)
fd.Close()
}
func TestLookupCNAME(t *testing.T) {
if testing.Short() || !*testExternal {
t.Logf("skipping test to avoid external network")
return
}
cname, err := LookupCNAME("www.google.com")
if !strings.HasSuffix(cname, ".l.google.com.") || err != nil {
t.Errorf(`LookupCNAME("www.google.com.") = %q, %v, want "*.l.google.com.", nil`, cname, err)
}
}
var googleaddrsipv4 = []string{
"%d.%d.%d.%d:80",
"www.google.com:80",
"%d.%d.%d.%d:http",
"www.google.com:http",
"%03d.%03d.%03d.%03d:0080",
"[::ffff:%d.%d.%d.%d]:80",
"[::ffff:%02x%02x:%02x%02x]:80",
"[0:0:0:0:0000:ffff:%d.%d.%d.%d]:80",
"[0:0:0:0:000000:ffff:%d.%d.%d.%d]:80",
"[0:0:0:0:0:ffff::%d.%d.%d.%d]:80",
}
func TestDialGoogleIPv4(t *testing.T) {
if testing.Short() || !*testExternal {
t.Logf("skipping test to avoid external network")
return
}
// Insert an actual IPv4 address for google.com
// into the table.
addrs, err := LookupIP("www.google.com")
if err != nil {
t.Fatalf("lookup www.google.com: %v", err)
}
var ip IP
for _, addr := range addrs {
if x := addr.To4(); x != nil {
ip = x
break
}
}
if ip == nil {
t.Fatalf("no IPv4 addresses for www.google.com")
}
for i, s := range googleaddrsipv4 {
if strings.Contains(s, "%") {
googleaddrsipv4[i] = fmt.Sprintf(s, ip[0], ip[1], ip[2], ip[3])
}
}
for i := 0; i < len(googleaddrsipv4); i++ {
addr := googleaddrsipv4[i]
if addr == "" {
continue
}
t.Logf("-- %s --", addr)
doDial(t, "tcp", addr)
if addr[0] != '[' {
doDial(t, "tcp4", addr)
if supportsIPv6 {
// make sure syscall.SocketDisableIPv6 flag works.
syscall.SocketDisableIPv6 = true
doDial(t, "tcp", addr)
doDial(t, "tcp4", addr)
syscall.SocketDisableIPv6 = false
}
}
}
}
var googleaddrsipv6 = []string{
"[%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:80",
"ipv6.google.com:80",
"[%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:http",
"ipv6.google.com:http",
}
func TestDialGoogleIPv6(t *testing.T) {
if testing.Short() || !*testExternal {
t.Logf("skipping test to avoid external network")
return
}
// Only run tcp6 if the kernel will take it.
if !*testIPv6 || !supportsIPv6 {
return
}
// Insert an actual IPv6 address for ipv6.google.com
// into the table.
addrs, err := LookupIP("ipv6.google.com")
if err != nil {
t.Fatalf("lookup ipv6.google.com: %v", err)
}
var ip IP
for _, addr := range addrs {
if x := addr.To16(); x != nil {
ip = x
break
}
}
if ip == nil {
t.Fatalf("no IPv6 addresses for ipv6.google.com")
}
for i, s := range googleaddrsipv6 {
if strings.Contains(s, "%") {
googleaddrsipv6[i] = fmt.Sprintf(s, ip[0], ip[1], ip[2], ip[3], ip[4], ip[5], ip[6], ip[7], ip[8], ip[9], ip[10], ip[11], ip[12], ip[13], ip[14], ip[15])
}
}
for i := 0; i < len(googleaddrsipv6); i++ {
addr := googleaddrsipv6[i]
if addr == "" {
continue
}
t.Logf("-- %s --", addr)
doDial(t, "tcp", addr)
doDial(t, "tcp6", addr)
}
}