mirror of
https://github.com/golang/go
synced 2024-11-06 15:46:19 -07:00
4b3d8d1a39
This change applies the same transformation as in CL 366774, but to the net package. testUnixAddr was using os.CreateTemp to obtain a unique socket path, but then calling os.Remove on that path immediately. Since the existence of the file is what guarantees its uniqueness, that could occasionally result in testUnixAddr returning the same path for two calls, causing the tests using those paths to fail — especially if they are the same test or are run in parallel. Instead, we now create a unique, short temp directory for each call, and use a path within that directory for the socket address. For #34611 Change-Id: I8e13b606abce2479a0305f7aeecf5d54c449a032 Reviewed-on: https://go-review.googlesource.com/c/go/+/370694 Trust: Bryan Mills <bcmills@google.com> Reviewed-by: Ian Lance Taylor <iant@golang.org> Run-TryBot: Bryan Mills <bcmills@google.com> TryBot-Result: Gopher Robot <gobot@golang.org>
152 lines
3.0 KiB
Go
152 lines
3.0 KiB
Go
// Copyright 2012 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.
|
|
|
|
// This file implements API tests across platforms and will never have a build
|
|
// tag.
|
|
|
|
//go:build !js
|
|
|
|
package net
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
// The full stack test cases for IPConn have been moved to the
|
|
// following:
|
|
// golang.org/x/net/ipv4
|
|
// golang.org/x/net/ipv6
|
|
// golang.org/x/net/icmp
|
|
|
|
func packetConnTestData(t *testing.T, network string) ([]byte, func()) {
|
|
if !testableNetwork(network) {
|
|
return nil, func() { t.Logf("skipping %s test", network) }
|
|
}
|
|
return []byte("PACKETCONN TEST"), nil
|
|
}
|
|
|
|
func TestPacketConn(t *testing.T) {
|
|
var packetConnTests = []struct {
|
|
net string
|
|
addr1 string
|
|
addr2 string
|
|
}{
|
|
{"udp", "127.0.0.1:0", "127.0.0.1:0"},
|
|
{"unixgram", testUnixAddr(t), testUnixAddr(t)},
|
|
}
|
|
|
|
closer := func(c PacketConn, net, addr1, addr2 string) {
|
|
c.Close()
|
|
switch net {
|
|
case "unixgram":
|
|
os.Remove(addr1)
|
|
os.Remove(addr2)
|
|
}
|
|
}
|
|
|
|
for _, tt := range packetConnTests {
|
|
wb, skipOrFatalFn := packetConnTestData(t, tt.net)
|
|
if skipOrFatalFn != nil {
|
|
skipOrFatalFn()
|
|
continue
|
|
}
|
|
|
|
c1, err := ListenPacket(tt.net, tt.addr1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer closer(c1, tt.net, tt.addr1, tt.addr2)
|
|
c1.LocalAddr()
|
|
|
|
c2, err := ListenPacket(tt.net, tt.addr2)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer closer(c2, tt.net, tt.addr1, tt.addr2)
|
|
c2.LocalAddr()
|
|
rb2 := make([]byte, 128)
|
|
|
|
if _, err := c1.WriteTo(wb, c2.LocalAddr()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, _, err := c2.ReadFrom(rb2); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := c2.WriteTo(wb, c1.LocalAddr()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rb1 := make([]byte, 128)
|
|
if _, _, err := c1.ReadFrom(rb1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestConnAndPacketConn(t *testing.T) {
|
|
var packetConnTests = []struct {
|
|
net string
|
|
addr1 string
|
|
addr2 string
|
|
}{
|
|
{"udp", "127.0.0.1:0", "127.0.0.1:0"},
|
|
{"unixgram", testUnixAddr(t), testUnixAddr(t)},
|
|
}
|
|
|
|
closer := func(c PacketConn, net, addr1, addr2 string) {
|
|
c.Close()
|
|
switch net {
|
|
case "unixgram":
|
|
os.Remove(addr1)
|
|
os.Remove(addr2)
|
|
}
|
|
}
|
|
|
|
for _, tt := range packetConnTests {
|
|
var wb []byte
|
|
wb, skipOrFatalFn := packetConnTestData(t, tt.net)
|
|
if skipOrFatalFn != nil {
|
|
skipOrFatalFn()
|
|
continue
|
|
}
|
|
|
|
c1, err := ListenPacket(tt.net, tt.addr1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer closer(c1, tt.net, tt.addr1, tt.addr2)
|
|
c1.LocalAddr()
|
|
|
|
c2, err := Dial(tt.net, c1.LocalAddr().String())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer c2.Close()
|
|
c2.LocalAddr()
|
|
c2.RemoteAddr()
|
|
|
|
if _, err := c2.Write(wb); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rb1 := make([]byte, 128)
|
|
if _, _, err := c1.ReadFrom(rb1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var dst Addr
|
|
switch tt.net {
|
|
case "unixgram":
|
|
continue
|
|
default:
|
|
dst = c2.LocalAddr()
|
|
}
|
|
if _, err := c1.WriteTo(wb, dst); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
rb2 := make([]byte, 128)
|
|
if _, err := c2.Read(rb2); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|