| // 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 ( |
| "fmt" |
| "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 { |
| t.Run(tt.snet+" "+tt.saddr+"<-"+tt.taddr, func(t *testing.T) { |
| if !testableListenArgs(tt.snet, tt.saddr, tt.taddr) { |
| t.Skip("not testable") |
| } |
| |
| 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 := (&streamListener{Listener: ln}).newLocalServer() |
| 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) { ls.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) |
| } |
| } |
| }) |
| } |
| } |
| |
| // TestUnixAndUnixpacketServer tests concurrent accept-read-write |
| // servers |
| func TestUnixAndUnixpacketServer(t *testing.T) { |
| var unixAndUnixpacketServerTests = []struct { |
| network, address string |
| }{ |
| {"unix", testUnixAddr(t)}, |
| {"unix", "@nettest/go/unix"}, |
| |
| {"unixpacket", testUnixAddr(t)}, |
| {"unixpacket", "@nettest/go/unixpacket"}, |
| } |
| |
| 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 := (&streamListener{Listener: ln}).newLocalServer() |
| 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) { ls.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) |
| } |
| |
| if addr := c.LocalAddr(); addr != nil { |
| t.Logf("connected %s->%s", addr, lss[i].Listener.Addr()) |
| } |
| |
| 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 { |
| i, tt := i, tt |
| t.Run(fmt.Sprint(i), func(t *testing.T) { |
| if !testableListenArgs(tt.snet, tt.saddr, tt.taddr) { |
| t.Skipf("skipping %s %s<-%s test", tt.snet, tt.saddr, tt.taddr) |
| } |
| t.Logf("%s %s<-%s", tt.snet, tt.saddr, tt.taddr) |
| |
| c1, err := ListenPacket(tt.snet, tt.saddr) |
| if err != nil { |
| if perr := parseDialError(err); perr != nil { |
| t.Error(perr) |
| } |
| t.Fatal(err) |
| } |
| |
| ls := (&packetListener{PacketConn: c1}).newLocalServer() |
| 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 trch != nil || tpch != nil { |
| select { |
| case err, ok := <-trch: |
| if !ok { |
| trch = nil |
| } |
| if err != nil { |
| t.Errorf("client: %v", err) |
| } |
| case err, ok := <-tpch: |
| if !ok { |
| tpch = nil |
| } |
| if err != nil { |
| t.Errorf("server: %v", err) |
| } |
| } |
| } |
| }) |
| } |
| } |
| |
| func TestUnixgramServer(t *testing.T) { |
| var unixgramServerTests = []struct { |
| saddr string // server endpoint |
| caddr string // client endpoint |
| dial bool // test with Dial |
| }{ |
| {saddr: testUnixAddr(t), caddr: testUnixAddr(t)}, |
| {saddr: testUnixAddr(t), caddr: testUnixAddr(t), dial: true}, |
| |
| {saddr: "@nettest/go/unixgram/server", caddr: "@nettest/go/unixgram/client"}, |
| } |
| |
| for i, tt := range unixgramServerTests { |
| i, tt := i, tt |
| t.Run(fmt.Sprint(i), func(t *testing.T) { |
| if !testableListenArgs("unixgram", tt.saddr, "") { |
| t.Skipf("skipping unixgram %s<-%s test", tt.saddr, tt.caddr) |
| } |
| t.Logf("unixgram %s<-%s", tt.saddr, tt.caddr) |
| |
| c1, err := ListenPacket("unixgram", tt.saddr) |
| if err != nil { |
| if perr := parseDialError(err); perr != nil { |
| t.Error(perr) |
| } |
| t.Fatal(err) |
| } |
| |
| ls := (&packetListener{PacketConn: c1}).newLocalServer() |
| 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 trch != nil || tpch != nil { |
| select { |
| case err, ok := <-trch: |
| if !ok { |
| trch = nil |
| } |
| if err != nil { |
| t.Errorf("client: %v", err) |
| } |
| case err, ok := <-tpch: |
| if !ok { |
| tpch = nil |
| } |
| if err != nil { |
| t.Errorf("server: %v", err) |
| } |
| } |
| } |
| }) |
| } |
| } |
