accept4: don't fall back to accept if we got unknown flags

[musl] / src / network / accept4.c

diff --git a/src/network/accept4.c b/src/network/accept4.c
index 6b5c16c..765a38e 100644 (file)
--- a/src/network/accept4.c
+++ b/src/network/accept4.c
@@ -1,9 +1,23 @@
 #define _GNU_SOURCE
 #include <sys/socket.h>
+#include <errno.h>
+#include <fcntl.h>
 #include "syscall.h"
-#include "libc.h"
 
 int accept4(int fd, struct sockaddr *restrict addr, socklen_t *restrict len, int flg)
 {
-       return socketcall_cp(accept4, fd, addr, len, flg, 0, 0);
+       if (!flg) return accept(fd, addr, len);
+       int ret = socketcall_cp(accept4, fd, addr, len, flg, 0, 0);
+       if (ret>=0 || (errno != ENOSYS && errno != EINVAL)) return ret;
+       if (flg & ~(SOCK_CLOEXEC|SOCK_NONBLOCK)) {
+               errno = EINVAL;
+               return -1;
+       }
+       ret = accept(fd, addr, len);
+       if (ret<0) return ret;
+       if (flg & SOCK_CLOEXEC)
+               __syscall(SYS_fcntl, ret, F_SETFD, FD_CLOEXEC);
+       if (flg & SOCK_NONBLOCK)
+               __syscall(SYS_fcntl, ret, F_SETFL, O_NONBLOCK);
+       return ret;
 }