blob: 2e998e23a8a0aba1bb3f3687d7fc0d9bfa6235db [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 (
"os"
"testing"
)
var tcpServerTests = []struct {
snet, saddr string // server endpoint
tnet, taddr string // target endpoint for client
}{
{snet: "tcp", saddr: ":0", tnet: "tcp", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "0.0.0.0:0", tnet: "tcp", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "[::ffff:0.0.0.0]:0", tnet: "tcp", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "[::]:0", tnet: "tcp", taddr: "::1"},
{snet: "tcp", saddr: ":0", tnet: "tcp", taddr: "::1"},
{snet: "tcp", saddr: "0.0.0.0:0", tnet: "tcp", taddr: "::1"},
{snet: "tcp", saddr: "[::ffff:0.0.0.0]:0", tnet: "tcp", taddr: "::1"},
{snet: "tcp", saddr: "[::]:0", tnet: "tcp", taddr: "127.0.0.1"},
{snet: "tcp", saddr: ":0", tnet: "tcp4", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "0.0.0.0:0", tnet: "tcp4", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "[::ffff:0.0.0.0]:0", tnet: "tcp4", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "[::]:0", tnet: "tcp6", taddr: "::1"},
{snet: "tcp", saddr: ":0", tnet: "tcp6", taddr: "::1"},
{snet: "tcp", saddr: "0.0.0.0:0", tnet: "tcp6", taddr: "::1"},
{snet: "tcp", saddr: "[::ffff:0.0.0.0]:0", tnet: "tcp6", taddr: "::1"},
{snet: "tcp", saddr: "[::]:0", tnet: "tcp4", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "127.0.0.1:0", tnet: "tcp", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "[::ffff:127.0.0.1]:0", tnet: "tcp", taddr: "127.0.0.1"},
{snet: "tcp", saddr: "[::1]:0", tnet: "tcp", taddr: "::1"},
{snet: "tcp4", saddr: ":0", tnet: "tcp4", taddr: "127.0.0.1"},
{snet: "tcp4", saddr: "0.0.0.0:0", tnet: "tcp4", taddr: "127.0.0.1"},
{snet: "tcp4", saddr: "[::ffff:0.0.0.0]:0", tnet: "tcp4", taddr: "127.0.0.1"},
{snet: "tcp4", saddr: "127.0.0.1:0", tnet: "tcp4", taddr: "127.0.0.1"},
{snet: "tcp6", saddr: ":0", tnet: "tcp6", taddr: "::1"},
{snet: "tcp6", saddr: "[::]:0", tnet: "tcp6", taddr: "::1"},
{snet: "tcp6", saddr: "[::1]:0", tnet: "tcp6", taddr: "::1"},
}
// TestTCPServer tests concurrent accept-read-write servers.
func TestTCPServer(t *testing.T) {
const N = 3
for i, tt := range tcpServerTests {
if !testableListenArgs(tt.snet, tt.saddr, tt.taddr) {
t.Logf("skipping %s test", tt.snet+" "+tt.saddr+"<-"+tt.taddr)
continue
}
ln, err := Listen(tt.snet, tt.saddr)
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
var lss []*localServer
var tpchs []chan error
defer func() {
for _, ls := range lss {
ls.teardown()
}
}()
for i := 0; i < N; i++ {
ls, err := (&streamListener{Listener: ln}).newLocalServer()
if err != nil {
t.Fatal(err)
}
lss = append(lss, ls)
tpchs = append(tpchs, make(chan error, 1))
}
for i := 0; i < N; i++ {
ch := tpchs[i]
handler := func(ls *localServer, ln Listener) { transponder(ln, ch) }
if err := lss[i].buildup(handler); err != nil {
t.Fatal(err)
}
}
var trchs []chan error
for i := 0; i < N; i++ {
_, port, err := SplitHostPort(lss[i].Listener.Addr().String())
if err != nil {
t.Fatal(err)
}
d := Dialer{Timeout: someTimeout}
c, err := d.Dial(tt.tnet, JoinHostPort(tt.taddr, port))
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
defer c.Close()
trchs = append(trchs, make(chan error, 1))
go transceiver(c, []byte("TCP SERVER TEST"), trchs[i])
}
for _, ch := range trchs {
for err := range ch {
t.Errorf("#%d: %v", i, err)
}
}
for _, ch := range tpchs {
for err := range ch {
t.Errorf("#%d: %v", i, err)
}
}
}
}
var unixAndUnixpacketServerTests = []struct {
network, address string
}{
{"unix", testUnixAddr()},
{"unix", "@nettest/go/unix"},
{"unixpacket", testUnixAddr()},
{"unixpacket", "@nettest/go/unixpacket"},
}
// TestUnixAndUnixpacketServer tests concurrent accept-read-write
// servers
func TestUnixAndUnixpacketServer(t *testing.T) {
const N = 3
for i, tt := range unixAndUnixpacketServerTests {
if !testableListenArgs(tt.network, tt.address, "") {
t.Logf("skipping %s test", tt.network+" "+tt.address)
continue
}
ln, err := Listen(tt.network, tt.address)
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
var lss []*localServer
var tpchs []chan error
defer func() {
for _, ls := range lss {
ls.teardown()
}
}()
for i := 0; i < N; i++ {
ls, err := (&streamListener{Listener: ln}).newLocalServer()
if err != nil {
t.Fatal(err)
}
lss = append(lss, ls)
tpchs = append(tpchs, make(chan error, 1))
}
for i := 0; i < N; i++ {
ch := tpchs[i]
handler := func(ls *localServer, ln Listener) { transponder(ln, ch) }
if err := lss[i].buildup(handler); err != nil {
t.Fatal(err)
}
}
var trchs []chan error
for i := 0; i < N; i++ {
d := Dialer{Timeout: someTimeout}
c, err := d.Dial(lss[i].Listener.Addr().Network(), lss[i].Listener.Addr().String())
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
defer os.Remove(c.LocalAddr().String())
defer c.Close()
trchs = append(trchs, make(chan error, 1))
go transceiver(c, []byte("UNIX AND UNIXPACKET SERVER TEST"), trchs[i])
}
for _, ch := range trchs {
for err := range ch {
t.Errorf("#%d: %v", i, err)
}
}
for _, ch := range tpchs {
for err := range ch {
t.Errorf("#%d: %v", i, err)
}
}
}
}
var udpServerTests = []struct {
snet, saddr string // server endpoint
tnet, taddr string // target endpoint for client
dial bool // test with Dial
}{
{snet: "udp", saddr: ":0", tnet: "udp", taddr: "127.0.0.1"},
{snet: "udp", saddr: "0.0.0.0:0", tnet: "udp", taddr: "127.0.0.1"},
{snet: "udp", saddr: "[::ffff:0.0.0.0]:0", tnet: "udp", taddr: "127.0.0.1"},
{snet: "udp", saddr: "[::]:0", tnet: "udp", taddr: "::1"},
{snet: "udp", saddr: ":0", tnet: "udp", taddr: "::1"},
{snet: "udp", saddr: "0.0.0.0:0", tnet: "udp", taddr: "::1"},
{snet: "udp", saddr: "[::ffff:0.0.0.0]:0", tnet: "udp", taddr: "::1"},
{snet: "udp", saddr: "[::]:0", tnet: "udp", taddr: "127.0.0.1"},
{snet: "udp", saddr: ":0", tnet: "udp4", taddr: "127.0.0.1"},
{snet: "udp", saddr: "0.0.0.0:0", tnet: "udp4", taddr: "127.0.0.1"},
{snet: "udp", saddr: "[::ffff:0.0.0.0]:0", tnet: "udp4", taddr: "127.0.0.1"},
{snet: "udp", saddr: "[::]:0", tnet: "udp6", taddr: "::1"},
{snet: "udp", saddr: ":0", tnet: "udp6", taddr: "::1"},
{snet: "udp", saddr: "0.0.0.0:0", tnet: "udp6", taddr: "::1"},
{snet: "udp", saddr: "[::ffff:0.0.0.0]:0", tnet: "udp6", taddr: "::1"},
{snet: "udp", saddr: "[::]:0", tnet: "udp4", taddr: "127.0.0.1"},
{snet: "udp", saddr: "127.0.0.1:0", tnet: "udp", taddr: "127.0.0.1"},
{snet: "udp", saddr: "[::ffff:127.0.0.1]:0", tnet: "udp", taddr: "127.0.0.1"},
{snet: "udp", saddr: "[::1]:0", tnet: "udp", taddr: "::1"},
{snet: "udp4", saddr: ":0", tnet: "udp4", taddr: "127.0.0.1"},
{snet: "udp4", saddr: "0.0.0.0:0", tnet: "udp4", taddr: "127.0.0.1"},
{snet: "udp4", saddr: "[::ffff:0.0.0.0]:0", tnet: "udp4", taddr: "127.0.0.1"},
{snet: "udp4", saddr: "127.0.0.1:0", tnet: "udp4", taddr: "127.0.0.1"},
{snet: "udp6", saddr: ":0", tnet: "udp6", taddr: "::1"},
{snet: "udp6", saddr: "[::]:0", tnet: "udp6", taddr: "::1"},
{snet: "udp6", saddr: "[::1]:0", tnet: "udp6", taddr: "::1"},
{snet: "udp", saddr: "127.0.0.1:0", tnet: "udp", taddr: "127.0.0.1", dial: true},
{snet: "udp", saddr: "[::1]:0", tnet: "udp", taddr: "::1", dial: true},
}
func TestUDPServer(t *testing.T) {
for i, tt := range udpServerTests {
if !testableListenArgs(tt.snet, tt.saddr, tt.taddr) {
t.Logf("skipping %s test", tt.snet+" "+tt.saddr+"<-"+tt.taddr)
continue
}
c1, err := ListenPacket(tt.snet, tt.saddr)
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
ls, err := (&packetListener{PacketConn: c1}).newLocalServer()
if err != nil {
t.Fatal(err)
}
defer ls.teardown()
tpch := make(chan error, 1)
handler := func(ls *localPacketServer, c PacketConn) { packetTransponder(c, tpch) }
if err := ls.buildup(handler); err != nil {
t.Fatal(err)
}
trch := make(chan error, 1)
_, port, err := SplitHostPort(ls.PacketConn.LocalAddr().String())
if err != nil {
t.Fatal(err)
}
if tt.dial {
d := Dialer{Timeout: someTimeout}
c2, err := d.Dial(tt.tnet, JoinHostPort(tt.taddr, port))
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
defer c2.Close()
go transceiver(c2, []byte("UDP SERVER TEST"), trch)
} else {
c2, err := ListenPacket(tt.tnet, JoinHostPort(tt.taddr, "0"))
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
defer c2.Close()
dst, err := ResolveUDPAddr(tt.tnet, JoinHostPort(tt.taddr, port))
if err != nil {
t.Fatal(err)
}
go packetTransceiver(c2, []byte("UDP SERVER TEST"), dst, trch)
}
for err := range trch {
t.Errorf("#%d: %v", i, err)
}
for err := range tpch {
t.Errorf("#%d: %v", i, err)
}
}
}
var unixgramServerTests = []struct {
saddr string // server endpoint
caddr string // client endpoint
dial bool // test with Dial
}{
{saddr: testUnixAddr(), caddr: testUnixAddr()},
{saddr: testUnixAddr(), caddr: testUnixAddr(), dial: true},
{saddr: "@nettest/go/unixgram/server", caddr: "@nettest/go/unixgram/client"},
}
func TestUnixgramServer(t *testing.T) {
for i, tt := range unixgramServerTests {
if !testableListenArgs("unixgram", tt.saddr, "") {
t.Logf("skipping %s test", "unixgram "+tt.saddr+"<-"+tt.caddr)
continue
}
c1, err := ListenPacket("unixgram", tt.saddr)
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
ls, err := (&packetListener{PacketConn: c1}).newLocalServer()
if err != nil {
t.Fatal(err)
}
defer ls.teardown()
tpch := make(chan error, 1)
handler := func(ls *localPacketServer, c PacketConn) { packetTransponder(c, tpch) }
if err := ls.buildup(handler); err != nil {
t.Fatal(err)
}
trch := make(chan error, 1)
if tt.dial {
d := Dialer{Timeout: someTimeout, LocalAddr: &UnixAddr{Net: "unixgram", Name: tt.caddr}}
c2, err := d.Dial("unixgram", ls.PacketConn.LocalAddr().String())
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
defer os.Remove(c2.LocalAddr().String())
defer c2.Close()
go transceiver(c2, []byte(c2.LocalAddr().String()), trch)
} else {
c2, err := ListenPacket("unixgram", tt.caddr)
if err != nil {
if perr := parseDialError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
defer os.Remove(c2.LocalAddr().String())
defer c2.Close()
go packetTransceiver(c2, []byte("UNIXGRAM SERVER TEST"), ls.PacketConn.LocalAddr(), trch)
}
for err := range trch {
t.Errorf("#%d: %v", i, err)
}
for err := range tpch {
t.Errorf("#%d: %v", i, err)
}
}
}