123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463 |
- // Copyright 2013 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.
- //go:build !js && !plan9 && !windows
- package net
- import (
- "bytes"
- "internal/testenv"
- "os"
- "reflect"
- "runtime"
- "syscall"
- "testing"
- "time"
- )
- func TestReadUnixgramWithUnnamedSocket(t *testing.T) {
- if !testableNetwork("unixgram") {
- t.Skip("unixgram test")
- }
- if runtime.GOOS == "openbsd" {
- testenv.SkipFlaky(t, 15157)
- }
- addr := testUnixAddr(t)
- la, err := ResolveUnixAddr("unixgram", addr)
- if err != nil {
- t.Fatal(err)
- }
- c, err := ListenUnixgram("unixgram", la)
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- c.Close()
- os.Remove(addr)
- }()
- off := make(chan bool)
- data := [5]byte{1, 2, 3, 4, 5}
- go func() {
- defer func() { off <- true }()
- s, err := syscall.Socket(syscall.AF_UNIX, syscall.SOCK_DGRAM, 0)
- if err != nil {
- t.Error(err)
- return
- }
- defer syscall.Close(s)
- rsa := &syscall.SockaddrUnix{Name: addr}
- if err := syscall.Sendto(s, data[:], 0, rsa); err != nil {
- t.Error(err)
- return
- }
- }()
- <-off
- b := make([]byte, 64)
- c.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
- n, from, err := c.ReadFrom(b)
- if err != nil {
- t.Fatal(err)
- }
- if from != nil {
- t.Fatalf("unexpected peer address: %v", from)
- }
- if !bytes.Equal(b[:n], data[:]) {
- t.Fatalf("got %v; want %v", b[:n], data[:])
- }
- }
- func TestUnixgramZeroBytePayload(t *testing.T) {
- if !testableNetwork("unixgram") {
- t.Skip("unixgram test")
- }
- c1 := newLocalPacketListener(t, "unixgram")
- defer os.Remove(c1.LocalAddr().String())
- defer c1.Close()
- c2, err := Dial("unixgram", c1.LocalAddr().String())
- if err != nil {
- t.Fatal(err)
- }
- defer os.Remove(c2.LocalAddr().String())
- defer c2.Close()
- for _, genericRead := range []bool{false, true} {
- n, err := c2.Write(nil)
- if err != nil {
- t.Fatal(err)
- }
- if n != 0 {
- t.Errorf("got %d; want 0", n)
- }
- c1.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
- var b [1]byte
- var peer Addr
- if genericRead {
- _, err = c1.(Conn).Read(b[:])
- } else {
- _, peer, err = c1.ReadFrom(b[:])
- }
- switch err {
- case nil: // ReadFrom succeeds
- if peer != nil { // peer is connected-mode
- t.Fatalf("unexpected peer address: %v", peer)
- }
- default: // Read may timeout, it depends on the platform
- if !isDeadlineExceeded(err) {
- t.Fatal(err)
- }
- }
- }
- }
- func TestUnixgramZeroByteBuffer(t *testing.T) {
- if !testableNetwork("unixgram") {
- t.Skip("unixgram test")
- }
- // issue 4352: Recvfrom failed with "address family not
- // supported by protocol family" if zero-length buffer provided
- c1 := newLocalPacketListener(t, "unixgram")
- defer os.Remove(c1.LocalAddr().String())
- defer c1.Close()
- c2, err := Dial("unixgram", c1.LocalAddr().String())
- if err != nil {
- t.Fatal(err)
- }
- defer os.Remove(c2.LocalAddr().String())
- defer c2.Close()
- b := []byte("UNIXGRAM ZERO BYTE BUFFER TEST")
- for _, genericRead := range []bool{false, true} {
- n, err := c2.Write(b)
- if err != nil {
- t.Fatal(err)
- }
- if n != len(b) {
- t.Errorf("got %d; want %d", n, len(b))
- }
- c1.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
- var peer Addr
- if genericRead {
- _, err = c1.(Conn).Read(nil)
- } else {
- _, peer, err = c1.ReadFrom(nil)
- }
- switch err {
- case nil: // ReadFrom succeeds
- if peer != nil { // peer is connected-mode
- t.Fatalf("unexpected peer address: %v", peer)
- }
- default: // Read may timeout, it depends on the platform
- if !isDeadlineExceeded(err) {
- t.Fatal(err)
- }
- }
- }
- }
- func TestUnixgramWrite(t *testing.T) {
- if !testableNetwork("unixgram") {
- t.Skip("unixgram test")
- }
- addr := testUnixAddr(t)
- laddr, err := ResolveUnixAddr("unixgram", addr)
- if err != nil {
- t.Fatal(err)
- }
- c, err := ListenPacket("unixgram", addr)
- if err != nil {
- t.Fatal(err)
- }
- defer os.Remove(addr)
- defer c.Close()
- testUnixgramWriteConn(t, laddr)
- testUnixgramWritePacketConn(t, laddr)
- }
- func testUnixgramWriteConn(t *testing.T, raddr *UnixAddr) {
- c, err := Dial("unixgram", raddr.String())
- if err != nil {
- t.Fatal(err)
- }
- defer c.Close()
- b := []byte("CONNECTED-MODE SOCKET")
- if _, err := c.(*UnixConn).WriteToUnix(b, raddr); err == nil {
- t.Fatal("should fail")
- } else if err.(*OpError).Err != ErrWriteToConnected {
- t.Fatalf("should fail as ErrWriteToConnected: %v", err)
- }
- if _, err = c.(*UnixConn).WriteTo(b, raddr); err == nil {
- t.Fatal("should fail")
- } else if err.(*OpError).Err != ErrWriteToConnected {
- t.Fatalf("should fail as ErrWriteToConnected: %v", err)
- }
- if _, _, err = c.(*UnixConn).WriteMsgUnix(b, nil, raddr); err == nil {
- t.Fatal("should fail")
- } else if err.(*OpError).Err != ErrWriteToConnected {
- t.Fatalf("should fail as ErrWriteToConnected: %v", err)
- }
- if _, err := c.Write(b); err != nil {
- t.Fatal(err)
- }
- }
- func testUnixgramWritePacketConn(t *testing.T, raddr *UnixAddr) {
- addr := testUnixAddr(t)
- c, err := ListenPacket("unixgram", addr)
- if err != nil {
- t.Fatal(err)
- }
- defer os.Remove(addr)
- defer c.Close()
- b := []byte("UNCONNECTED-MODE SOCKET")
- if _, err := c.(*UnixConn).WriteToUnix(b, raddr); err != nil {
- t.Fatal(err)
- }
- if _, err := c.WriteTo(b, raddr); err != nil {
- t.Fatal(err)
- }
- if _, _, err := c.(*UnixConn).WriteMsgUnix(b, nil, raddr); err != nil {
- t.Fatal(err)
- }
- if _, err := c.(*UnixConn).Write(b); err == nil {
- t.Fatal("should fail")
- }
- }
- func TestUnixConnLocalAndRemoteNames(t *testing.T) {
- if !testableNetwork("unix") {
- t.Skip("unix test")
- }
- handler := func(ls *localServer, ln Listener) {}
- for _, laddr := range []string{"", testUnixAddr(t)} {
- laddr := laddr
- taddr := testUnixAddr(t)
- ta, err := ResolveUnixAddr("unix", taddr)
- if err != nil {
- t.Fatal(err)
- }
- ln, err := ListenUnix("unix", ta)
- if err != nil {
- t.Fatal(err)
- }
- ls := (&streamListener{Listener: ln}).newLocalServer()
- defer ls.teardown()
- if err := ls.buildup(handler); err != nil {
- t.Fatal(err)
- }
- la, err := ResolveUnixAddr("unix", laddr)
- if err != nil {
- t.Fatal(err)
- }
- c, err := DialUnix("unix", la, ta)
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- c.Close()
- if la != nil {
- defer os.Remove(laddr)
- }
- }()
- if _, err := c.Write([]byte("UNIXCONN LOCAL AND REMOTE NAME TEST")); err != nil {
- t.Fatal(err)
- }
- switch runtime.GOOS {
- case "android", "linux":
- if laddr == "" {
- laddr = "@" // autobind feature
- }
- }
- var connAddrs = [3]struct{ got, want Addr }{
- {ln.Addr(), ta},
- {c.LocalAddr(), &UnixAddr{Name: laddr, Net: "unix"}},
- {c.RemoteAddr(), ta},
- }
- for _, ca := range connAddrs {
- if !reflect.DeepEqual(ca.got, ca.want) {
- t.Fatalf("got %#v, expected %#v", ca.got, ca.want)
- }
- }
- }
- }
- func TestUnixgramConnLocalAndRemoteNames(t *testing.T) {
- if !testableNetwork("unixgram") {
- t.Skip("unixgram test")
- }
- for _, laddr := range []string{"", testUnixAddr(t)} {
- laddr := laddr
- taddr := testUnixAddr(t)
- ta, err := ResolveUnixAddr("unixgram", taddr)
- if err != nil {
- t.Fatal(err)
- }
- c1, err := ListenUnixgram("unixgram", ta)
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- c1.Close()
- os.Remove(taddr)
- }()
- var la *UnixAddr
- if laddr != "" {
- if la, err = ResolveUnixAddr("unixgram", laddr); err != nil {
- t.Fatal(err)
- }
- }
- c2, err := DialUnix("unixgram", la, ta)
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- c2.Close()
- if la != nil {
- defer os.Remove(laddr)
- }
- }()
- switch runtime.GOOS {
- case "android", "linux":
- if laddr == "" {
- laddr = "@" // autobind feature
- }
- }
- var connAddrs = [4]struct{ got, want Addr }{
- {c1.LocalAddr(), ta},
- {c1.RemoteAddr(), nil},
- {c2.LocalAddr(), &UnixAddr{Name: laddr, Net: "unixgram"}},
- {c2.RemoteAddr(), ta},
- }
- for _, ca := range connAddrs {
- if !reflect.DeepEqual(ca.got, ca.want) {
- t.Fatalf("got %#v; want %#v", ca.got, ca.want)
- }
- }
- }
- }
- func TestUnixUnlink(t *testing.T) {
- if !testableNetwork("unix") {
- t.Skip("unix test")
- }
- name := testUnixAddr(t)
- listen := func(t *testing.T) *UnixListener {
- l, err := Listen("unix", name)
- if err != nil {
- t.Fatal(err)
- }
- return l.(*UnixListener)
- }
- checkExists := func(t *testing.T, desc string) {
- if _, err := os.Stat(name); err != nil {
- t.Fatalf("unix socket does not exist %s: %v", desc, err)
- }
- }
- checkNotExists := func(t *testing.T, desc string) {
- if _, err := os.Stat(name); err == nil {
- t.Fatalf("unix socket does exist %s: %v", desc, err)
- }
- }
- // Listener should remove on close.
- t.Run("Listen", func(t *testing.T) {
- l := listen(t)
- checkExists(t, "after Listen")
- l.Close()
- checkNotExists(t, "after Listener close")
- })
- // FileListener should not.
- t.Run("FileListener", func(t *testing.T) {
- l := listen(t)
- f, _ := l.File()
- l1, _ := FileListener(f)
- checkExists(t, "after FileListener")
- f.Close()
- checkExists(t, "after File close")
- l1.Close()
- checkExists(t, "after FileListener close")
- l.Close()
- checkNotExists(t, "after Listener close")
- })
- // Only first call to l.Close should remove.
- t.Run("SecondClose", func(t *testing.T) {
- l := listen(t)
- checkExists(t, "after Listen")
- l.Close()
- checkNotExists(t, "after Listener close")
- if err := os.WriteFile(name, []byte("hello world"), 0666); err != nil {
- t.Fatalf("cannot recreate socket file: %v", err)
- }
- checkExists(t, "after writing temp file")
- l.Close()
- checkExists(t, "after second Listener close")
- os.Remove(name)
- })
- // SetUnlinkOnClose should do what it says.
- t.Run("Listen/SetUnlinkOnClose(true)", func(t *testing.T) {
- l := listen(t)
- checkExists(t, "after Listen")
- l.SetUnlinkOnClose(true)
- l.Close()
- checkNotExists(t, "after Listener close")
- })
- t.Run("Listen/SetUnlinkOnClose(false)", func(t *testing.T) {
- l := listen(t)
- checkExists(t, "after Listen")
- l.SetUnlinkOnClose(false)
- l.Close()
- checkExists(t, "after Listener close")
- os.Remove(name)
- })
- t.Run("FileListener/SetUnlinkOnClose(true)", func(t *testing.T) {
- l := listen(t)
- f, _ := l.File()
- l1, _ := FileListener(f)
- checkExists(t, "after FileListener")
- l1.(*UnixListener).SetUnlinkOnClose(true)
- f.Close()
- checkExists(t, "after File close")
- l1.Close()
- checkNotExists(t, "after FileListener close")
- l.Close()
- })
- t.Run("FileListener/SetUnlinkOnClose(false)", func(t *testing.T) {
- l := listen(t)
- f, _ := l.File()
- l1, _ := FileListener(f)
- checkExists(t, "after FileListener")
- l1.(*UnixListener).SetUnlinkOnClose(false)
- f.Close()
- checkExists(t, "after File close")
- l1.Close()
- checkExists(t, "after FileListener close")
- l.Close()
- })
- }
|