the fcntl function is heavy, so make the syscall directly instead. also, avoid the code size and runtime overhead of querying the old flags, since it's reasonable to assume nothing will be set on a newly-created socket. this code is only used on old kernels which lack proper atomic close-on-exec support, so future changes that might invalidate such an assumption do not need to be considered.
22 lines
585 B
C
22 lines
585 B
C
#include <sys/socket.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#include "syscall.h"
|
|
|
|
int socket(int domain, int type, int protocol)
|
|
{
|
|
int s = socketcall(socket, domain, type, protocol, 0, 0, 0);
|
|
if (s<0 && (errno==EINVAL || errno==EPROTONOSUPPORT)
|
|
&& (type&(SOCK_CLOEXEC|SOCK_NONBLOCK))) {
|
|
s = socketcall(socket, domain,
|
|
type & ~(SOCK_CLOEXEC|SOCK_NONBLOCK),
|
|
protocol, 0, 0, 0);
|
|
if (s < 0) return s;
|
|
if (type & SOCK_CLOEXEC)
|
|
__syscall(SYS_fcntl, s, F_SETFD, FD_CLOEXEC);
|
|
if (type & SOCK_NONBLOCK)
|
|
__syscall(SYS_fcntl, s, F_SETFL, O_NONBLOCK);
|
|
}
|
|
return s;
|
|
}
|