nsz Git - musl/blob - src/network/recvmmsg.c

   1 #define _GNU_SOURCE
   2 #include <sys/socket.h>
   3 #include <limits.h>
   4 #include <errno.h>
   5 #include <time.h>
   6 #include "syscall.h"
   7
   8 #define IS32BIT(x) !((x)+0x80000000ULL>>32)
   9 #define CLAMP(x) (int)(IS32BIT(x) ? (x) : 0x7fffffffU+((0ULL+(x))>>63))
  10
  11 int recvmmsg(int fd, struct mmsghdr *msgvec, unsigned int vlen, unsigned int flags, struct timespec *timeout)
  12 {
  13 #if LONG_MAX > INT_MAX
  14         struct mmsghdr *mh = msgvec;
  15         unsigned int i;
  16         for (i = vlen; i; i--, mh++)
  17                 mh->msg_hdr.__pad1 = mh->msg_hdr.__pad2 = 0;
  18 #endif
  19 #ifdef SYS_recvmmsg_time64
  20         time_t s = timeout ? timeout->tv_sec : 0;
  21         long ns = timeout ? timeout->tv_nsec : 0;
  22         int r = -ENOSYS;
  23         if (SYS_recvmmsg == SYS_recvmmsg_time64 || !IS32BIT(s))
  24                 r = __syscall_cp(SYS_recvmmsg_time64, fd, msgvec, vlen, flags,
  25                         ((long long[]){s, ns}));
  26         if (SYS_recvmmsg == SYS_recvmmsg_time64 || r!=-ENOSYS)
  27                 return __syscall_ret(r);
  28         return syscall_cp(SYS_recvmmsg, fd, msgvec, vlen, flags,
  29                 timeout ? ((long[]){CLAMP(s), ns}) : 0);
  30 #else
  31         return syscall_cp(SYS_recvmmsg, fd, msgvec, vlen, flags, timeout);
  32 #endif
  33 }