sock_cloexec.go

Documentation: net

     1  // Copyright 2013 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  // This file implements sysSocket for platforms that provide a fast path for
     6  // setting SetNonblock and CloseOnExec.
     7  
     8  //go:build dragonfly || freebsd || linux || netbsd || openbsd || solaris
     9  
    10  package net
    11  
    12  import (
    13  	"internal/poll"
    14  	"os"
    15  	"syscall"
    16  )
    17  
    18  // Wrapper around the socket system call that marks the returned file
    19  // descriptor as nonblocking and close-on-exec.
    20  func sysSocket(family, sotype, proto int) (int, error) {
    21  	s, err := socketFunc(family, sotype|syscall.SOCK_NONBLOCK|syscall.SOCK_CLOEXEC, proto)
    22  	// TODO: We can remove the fallback on Linux and *BSD,
    23  	// as currently supported versions all support accept4
    24  	// with SOCK_CLOEXEC, but Solaris does not. See issue #59359.
    25  	switch err {
    26  	case nil:
    27  		return s, nil
    28  	default:
    29  		return -1, os.NewSyscallError("socket", err)
    30  	case syscall.EPROTONOSUPPORT, syscall.EINVAL:
    31  	}
    32  
    33  	// See ../syscall/exec_unix.go for description of ForkLock.
    34  	syscall.ForkLock.RLock()
    35  	s, err = socketFunc(family, sotype, proto)
    36  	if err == nil {
    37  		syscall.CloseOnExec(s)
    38  	}
    39  	syscall.ForkLock.RUnlock()
    40  	if err != nil {
    41  		return -1, os.NewSyscallError("socket", err)
    42  	}
    43  	if err = syscall.SetNonblock(s, true); err != nil {
    44  		poll.CloseFunc(s)
    45  		return -1, os.NewSyscallError("setnonblock", err)
    46  	}
    47  	return s, nil
    48  }
    49
View as plain text