src/net/unixsock_readmsg_test.go

   1 // Copyright 2021 The Go Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 //go:build unix
   6
   7 package net
   8
   9 import (
  10         "internal/syscall/unix"
  11         "os"
  12         "syscall"
  13         "testing"
  14         "time"
  15 )
  16
  17 func TestUnixConnReadMsgUnixSCMRightsCloseOnExec(t *testing.T) {
  18         if !testableNetwork("unix") {
  19                 t.Skip("not unix system")
  20         }
  21
  22         scmFile, err := os.Open(os.DevNull)
  23         if err != nil {
  24                 t.Fatalf("file open: %v", err)
  25         }
  26         defer scmFile.Close()
  27
  28         rights := syscall.UnixRights(int(scmFile.Fd()))
  29         fds, err := syscall.Socketpair(syscall.AF_LOCAL, syscall.SOCK_STREAM, 0)
  30         if err != nil {
  31                 t.Fatalf("Socketpair: %v", err)
  32         }
  33
  34         writeFile := os.NewFile(uintptr(fds[0]), "write-socket")
  35         defer writeFile.Close()
  36         readFile := os.NewFile(uintptr(fds[1]), "read-socket")
  37         defer readFile.Close()
  38
  39         cw, err := FileConn(writeFile)
  40         if err != nil {
  41                 t.Fatalf("FileConn: %v", err)
  42         }
  43         defer cw.Close()
  44         cr, err := FileConn(readFile)
  45         if err != nil {
  46                 t.Fatalf("FileConn: %v", err)
  47         }
  48         defer cr.Close()
  49
  50         ucw, ok := cw.(*UnixConn)
  51         if !ok {
  52                 t.Fatalf("got %T; want UnixConn", cw)
  53         }
  54         ucr, ok := cr.(*UnixConn)
  55         if !ok {
  56                 t.Fatalf("got %T; want UnixConn", cr)
  57         }
  58
  59         oob := make([]byte, syscall.CmsgSpace(4))
  60         err = ucw.SetWriteDeadline(time.Now().Add(5 * time.Second))
  61         if err != nil {
  62                 t.Fatalf("Can't set unix connection timeout: %v", err)
  63         }
  64         _, _, err = ucw.WriteMsgUnix(nil, rights, nil)
  65         if err != nil {
  66                 t.Fatalf("UnixConn readMsg: %v", err)
  67         }
  68         err = ucr.SetReadDeadline(time.Now().Add(5 * time.Second))
  69         if err != nil {
  70                 t.Fatalf("Can't set unix connection timeout: %v", err)
  71         }
  72         _, oobn, _, _, err := ucr.ReadMsgUnix(nil, oob)
  73         if err != nil {
  74                 t.Fatalf("UnixConn readMsg: %v", err)
  75         }
  76
  77         scms, err := syscall.ParseSocketControlMessage(oob[:oobn])
  78         if err != nil {
  79                 t.Fatalf("ParseSocketControlMessage: %v", err)
  80         }
  81         if len(scms) != 1 {
  82                 t.Fatalf("got scms = %#v; expected 1 SocketControlMessage", scms)
  83         }
  84         scm := scms[0]
  85         gotFDs, err := syscall.ParseUnixRights(&scm)
  86         if err != nil {
  87                 t.Fatalf("syscall.ParseUnixRights: %v", err)
  88         }
  89         if len(gotFDs) != 1 {
  90                 t.Fatalf("got FDs %#v: wanted only 1 fd", gotFDs)
  91         }
  92         defer func() {
  93                 if err := syscall.Close(gotFDs[0]); err != nil {
  94                         t.Fatalf("fail to close gotFDs: %v", err)
  95                 }
  96         }()
  97
  98         flags, err := unix.Fcntl(gotFDs[0], syscall.F_GETFD, 0)
  99         if err != nil {
 100                 t.Fatalf("Can't get flags of fd:%#v, with err:%v", gotFDs[0], err)
 101         }
 102         if flags&syscall.FD_CLOEXEC == 0 {
 103                 t.Fatalf("got flags %#x, want %#x (FD_CLOEXEC) set", flags, syscall.FD_CLOEXEC)
 104         }
 105 }