1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
/*
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <bits/pthread_cancel.h>
#include <errno.h>
#include <stdio.h>
#include <sys/poll.h>
#include <sys/select.h>
#include <sys/time.h>
#include <syscall.h>
#include <AK/Vector.h>
extern "C" {
// https://pubs.opengroup.org/onlinepubs/9699919799/functions/select.html
int select(int nfds, fd_set* readfds, fd_set* writefds, fd_set* exceptfds, timeval* timeout_tv)
{
__pthread_maybe_cancel();
timespec* timeout_ts = nullptr;
timespec timeout;
if (timeout_tv) {
timeout_ts = &timeout;
TIMEVAL_TO_TIMESPEC(timeout_tv, timeout_ts);
}
return pselect(nfds, readfds, writefds, exceptfds, timeout_ts, nullptr);
}
// https://pubs.opengroup.org/onlinepubs/9699919799/functions/pselect.html
int pselect(int nfds, fd_set* readfds, fd_set* writefds, fd_set* exceptfds, timespec const* timeout, sigset_t const* sigmask)
{
__pthread_maybe_cancel();
Vector<pollfd, FD_SETSIZE> fds;
if (nfds < 0 || static_cast<size_t>(nfds) >= fds.capacity())
return EINVAL;
for (int fd = 0; fd < nfds; fd++) {
short events = 0;
if (readfds && FD_ISSET(fd, readfds))
events |= POLLIN;
if (writefds && FD_ISSET(fd, writefds))
events |= POLLOUT;
if (exceptfds && FD_ISSET(fd, exceptfds))
events |= POLLPRI;
if (!events)
continue;
fds.unchecked_append({ fd, events, 0 });
}
if (ppoll(fds.data(), fds.size(), timeout, sigmask) < 0)
return -1;
if (readfds)
FD_ZERO(readfds);
if (writefds)
FD_ZERO(writefds);
if (exceptfds)
FD_ZERO(exceptfds);
int marked_fd_count = 0;
for (auto& fd_entry : fds) {
if (fd_entry.revents == 0)
continue;
if (readfds && (fd_entry.revents & POLLIN)) {
FD_SET(fd_entry.fd, readfds);
marked_fd_count++;
}
if (writefds && (fd_entry.revents & POLLOUT)) {
FD_SET(fd_entry.fd, writefds);
marked_fd_count++;
}
if (exceptfds && (fd_entry.revents & (POLLPRI | POLLERR | POLLHUP))) {
FD_SET(fd_entry.fd, exceptfds);
marked_fd_count++;
}
}
return marked_fd_count;
}
}
|
