#include <sys/select.h>
#include <signal.h>
#include <stdint.h>
#include <errno.h>
#include "syscall.h"

#define IS32BIT(x) !((x)+0x80000000ULL>>32)
#define CLAMP(x) (int)(IS32BIT(x) ? (x) : 0x7fffffffU+((0ULL+(x))>>63))

int pselect(int n, fd_set *restrict rfds, fd_set *restrict wfds, fd_set *restrict efds, const struct timespec *restrict ts, const sigset_t *restrict mask)
{
#ifdef __EMSCRIPTEN__
	// Emscripten does not support true async signals so we just implement pselect
	// in terms of select here in userspace.
	struct timeval tv_timeout;
	if (ts) {
		tv_timeout.tv_sec = ts->tv_sec;
		tv_timeout.tv_usec = ts->tv_nsec / 1000;
	}
	sigset_t origmask;
	if (mask) pthread_sigmask(SIG_SETMASK, mask, &origmask);
	int rtn = select(n, rfds, wfds, efds, ts ? &tv_timeout : NULL);
	if (mask) pthread_sigmask(SIG_SETMASK, &origmask, NULL);
	return rtn;
#else
	syscall_arg_t data[2] = { (uintptr_t)mask, _NSIG/8 };
	time_t s = ts ? ts->tv_sec : 0;
	long ns = ts ? ts->tv_nsec : 0;
#ifdef SYS_pselect6_time64
	int r = -ENOSYS;
	if (SYS_pselect6 == SYS_pselect6_time64 || !IS32BIT(s))
		r = __syscall_cp(SYS_pselect6_time64, n, rfds, wfds, efds,
			ts ? ((long long[]){s, ns}) : 0, data);
	if (SYS_pselect6 == SYS_pselect6_time64 || r!=-ENOSYS)
		return __syscall_ret(r);
	s = CLAMP(s);
#endif
	return syscall_cp(SYS_pselect6, n, rfds, wfds, efds,
		ts ? ((long[]){s, ns}) : 0, data);
#endif
}