toaruos/libc/poll/poll.c
2019-12-05 17:49:36 +09:00

54 lines
1.1 KiB
C

#include <poll.h>
#include <stdio.h>
#include <errno.h>
#include <sys/fswait.h>
extern char * _argv_0;
int poll(struct pollfd *fds, nfds_t nfds, int timeout) {
int count_pollin = 0;
for (nfds_t i = 0; i < nfds; ++i) {
if (fds[i].events & POLLIN) {
count_pollin++;
}
fds[i].revents = 0;
}
for (nfds_t i = 0; i < nfds; ++i) {
if (fds[i].events & (~POLLIN)) {
fprintf(stderr, "%s: poll: unsupported bit set in fds (this implementation only supports POLLIN)\n", _argv_0);
return -EINVAL;
}
}
int fswait_fds[count_pollin];
int fswait_backref[count_pollin];
int fswait_results[count_pollin];
int j = 0;
for (nfds_t i = 0; i < nfds; ++i) {
if (fds[i].events & POLLIN) {
fswait_fds[j] = fds[i].fd;
fswait_backref[j] = i;
j++;
}
}
int ret = fswait3(count_pollin, fswait_fds, timeout, fswait_results);
if (ret >= 0 && ret < count_pollin) {
int count = 0;
for (int i = 0; i < count_pollin; ++i) {
if (fswait_results[i]) {
fds[fswait_backref[i]].revents = POLLIN;
count++;
}
}
return count;
} else if (ret == count_pollin) {
return 0;
} else {
return ret; /* Error */
}
}