Retro68/gcc/libgo/go/net/net_test.go

307 lines
6.4 KiB
Go
Raw Normal View History

2012-03-27 23:13:14 +00:00
// 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 (
"io"
2014-09-21 17:33:12 +00:00
"os"
2012-03-27 23:13:14 +00:00
"runtime"
"testing"
"time"
)
2017-04-10 11:32:00 +00:00
func TestCloseRead(t *testing.T) {
switch runtime.GOOS {
case "nacl", "plan9":
t.Skipf("not supported on %s", runtime.GOOS)
2012-03-27 23:13:14 +00:00
}
2017-04-10 11:32:00 +00:00
for _, network := range []string{"tcp", "unix", "unixpacket"} {
if !testableNetwork(network) {
t.Logf("skipping %s test", network)
continue
2012-03-27 23:13:14 +00:00
}
2017-04-10 11:32:00 +00:00
ln, err := newLocalListener(network)
2012-03-27 23:13:14 +00:00
if err != nil {
2017-04-10 11:32:00 +00:00
t.Fatal(err)
2012-03-27 23:13:14 +00:00
}
2017-04-10 11:32:00 +00:00
switch network {
case "unix", "unixpacket":
defer os.Remove(ln.Addr().String())
2012-03-27 23:13:14 +00:00
}
2017-04-10 11:32:00 +00:00
defer ln.Close()
2012-03-27 23:13:14 +00:00
2017-04-10 11:32:00 +00:00
c, err := Dial(ln.Addr().Network(), ln.Addr().String())
if err != nil {
t.Fatal(err)
}
switch network {
case "unix", "unixpacket":
defer os.Remove(c.LocalAddr().String())
}
defer c.Close()
2012-03-27 23:13:14 +00:00
2017-04-10 11:32:00 +00:00
switch c := c.(type) {
case *TCPConn:
err = c.CloseRead()
case *UnixConn:
err = c.CloseRead()
}
if err != nil {
if perr := parseCloseError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
var b [1]byte
n, err := c.Read(b[:])
if n != 0 || err == nil {
t.Fatalf("got (%d, %v); want (0, error)", n, err)
}
2012-03-27 23:13:14 +00:00
}
}
2017-04-10 11:32:00 +00:00
func TestCloseWrite(t *testing.T) {
2014-09-21 17:33:12 +00:00
switch runtime.GOOS {
2017-04-10 11:32:00 +00:00
case "nacl", "plan9":
t.Skipf("not supported on %s", runtime.GOOS)
2014-09-21 17:33:12 +00:00
}
2017-04-10 11:32:00 +00:00
handler := func(ls *localServer, ln Listener) {
2014-09-21 17:33:12 +00:00
c, err := ln.Accept()
if err != nil {
2017-04-10 11:32:00 +00:00
t.Error(err)
2015-08-28 15:33:40 +00:00
return
2014-09-21 17:33:12 +00:00
}
2017-04-10 11:32:00 +00:00
defer c.Close()
var b [1]byte
n, err := c.Read(b[:])
2014-09-21 17:33:12 +00:00
if n != 0 || err != io.EOF {
2017-04-10 11:32:00 +00:00
t.Errorf("got (%d, %v); want (0, io.EOF)", n, err)
return
}
switch c := c.(type) {
case *TCPConn:
err = c.CloseWrite()
case *UnixConn:
err = c.CloseWrite()
}
if err != nil {
if perr := parseCloseError(err); perr != nil {
t.Error(perr)
}
t.Error(err)
return
}
n, err = c.Write(b[:])
if err == nil {
t.Errorf("got (%d, %v); want (any, error)", n, err)
2015-08-28 15:33:40 +00:00
return
2014-09-21 17:33:12 +00:00
}
}
2017-04-10 11:32:00 +00:00
for _, network := range []string{"tcp", "unix", "unixpacket"} {
if !testableNetwork(network) {
t.Logf("skipping %s test", network)
continue
}
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
ls, err := newLocalServer(network)
if err != nil {
t.Fatal(err)
}
defer ls.teardown()
if err := ls.buildup(handler); err != nil {
t.Fatal(err)
}
2012-03-27 23:13:14 +00:00
2017-04-10 11:32:00 +00:00
c, err := Dial(ls.Listener.Addr().Network(), ls.Listener.Addr().String())
if err != nil {
t.Fatal(err)
}
switch network {
case "unix", "unixpacket":
defer os.Remove(c.LocalAddr().String())
}
defer c.Close()
switch c := c.(type) {
case *TCPConn:
err = c.CloseWrite()
case *UnixConn:
err = c.CloseWrite()
}
if err != nil {
if perr := parseCloseError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
var b [1]byte
n, err := c.Read(b[:])
if n != 0 || err != io.EOF {
t.Fatalf("got (%d, %v); want (0, io.EOF)", n, err)
}
n, err = c.Write(b[:])
2012-03-27 23:13:14 +00:00
if err == nil {
2017-04-10 11:32:00 +00:00
t.Fatalf("got (%d, %v); want (any, error)", n, err)
}
2012-03-27 23:13:14 +00:00
}
}
2017-04-10 11:32:00 +00:00
func TestConnClose(t *testing.T) {
for _, network := range []string{"tcp", "unix", "unixpacket"} {
if !testableNetwork(network) {
t.Logf("skipping %s test", network)
continue
}
2012-03-27 23:13:14 +00:00
2017-04-10 11:32:00 +00:00
ln, err := newLocalListener(network)
if err != nil {
t.Fatal(err)
}
switch network {
case "unix", "unixpacket":
defer os.Remove(ln.Addr().String())
}
defer ln.Close()
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
c, err := Dial(ln.Addr().Network(), ln.Addr().String())
if err != nil {
t.Fatal(err)
}
switch network {
case "unix", "unixpacket":
defer os.Remove(c.LocalAddr().String())
}
defer c.Close()
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
if err := c.Close(); err != nil {
if perr := parseCloseError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
var b [1]byte
n, err := c.Read(b[:])
if n != 0 || err == nil {
t.Fatalf("got (%d, %v); want (0, error)", n, err)
}
2014-09-21 17:33:12 +00:00
}
2017-04-10 11:32:00 +00:00
}
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
func TestListenerClose(t *testing.T) {
for _, network := range []string{"tcp", "unix", "unixpacket"} {
if !testableNetwork(network) {
t.Logf("skipping %s test", network)
continue
2014-09-21 17:33:12 +00:00
}
2017-04-10 11:32:00 +00:00
ln, err := newLocalListener(network)
if err != nil {
t.Fatal(err)
}
switch network {
case "unix", "unixpacket":
defer os.Remove(ln.Addr().String())
}
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
dst := ln.Addr().String()
if err := ln.Close(); err != nil {
if perr := parseCloseError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
c, err := ln.Accept()
if err == nil {
c.Close()
t.Fatal("should fail")
}
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
if network == "tcp" {
// We will have two TCP FSMs inside the
// kernel here. There's no guarantee that a
// signal comes from the far end FSM will be
// delivered immediately to the near end FSM,
// especially on the platforms that allow
// multiple consumer threads to pull pending
// established connections at the same time by
// enabling SO_REUSEPORT option such as Linux,
// DragonFly BSD. So we need to give some time
// quantum to the kernel.
//
// Note that net.inet.tcp.reuseport_ext=1 by
// default on DragonFly BSD.
time.Sleep(time.Millisecond)
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
cc, err := Dial("tcp", dst)
if err == nil {
t.Error("Dial to closed TCP listener succeeded.")
cc.Close()
}
}
2014-09-21 17:33:12 +00:00
}
}
2017-04-10 11:32:00 +00:00
func TestPacketConnClose(t *testing.T) {
for _, network := range []string{"udp", "unixgram"} {
if !testableNetwork(network) {
t.Logf("skipping %s test", network)
continue
}
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
c, err := newLocalPacketListener(network)
if err != nil {
t.Fatal(err)
}
switch network {
case "unixgram":
defer os.Remove(c.LocalAddr().String())
}
defer c.Close()
2014-09-21 17:33:12 +00:00
2017-04-10 11:32:00 +00:00
if err := c.Close(); err != nil {
if perr := parseCloseError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
var b [1]byte
n, _, err := c.ReadFrom(b[:])
if n != 0 || err == nil {
t.Fatalf("got (%d, %v); want (0, error)", n, err)
}
2014-09-21 17:33:12 +00:00
}
2017-04-10 11:32:00 +00:00
}
// nacl was previous failing to reuse an address.
func TestListenCloseListen(t *testing.T) {
const maxTries = 10
for tries := 0; tries < maxTries; tries++ {
ln, err := newLocalListener("tcp")
if err != nil {
t.Fatal(err)
}
addr := ln.Addr().String()
if err := ln.Close(); err != nil {
if perr := parseCloseError(err); perr != nil {
t.Error(perr)
}
t.Fatal(err)
}
ln, err = Listen("tcp", addr)
if err == nil {
// Success. nacl couldn't do this before.
ln.Close()
return
}
t.Errorf("failed on try %d/%d: %v", tries+1, maxTries, err)
2014-09-21 17:33:12 +00:00
}
2017-04-10 11:32:00 +00:00
t.Fatalf("failed to listen/close/listen on same address after %d tries", maxTries)
2014-09-21 17:33:12 +00:00
}