109 lines
2.7 KiB
C
109 lines
2.7 KiB
C
/*
|
|
* (C) 2016 by Holger Hans Peter Freyther
|
|
*
|
|
* All Rights Reserved
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation; either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#include "evpoll.h"
|
|
|
|
#include <osmocom/core/linuxlist.h>
|
|
#include <osmocom/core/select.h>
|
|
#include <osmocom/core/timer.h>
|
|
|
|
#include <sys/select.h>
|
|
|
|
/* based on osmo_select_main GPLv2+ so combined compatible with AGPLv3+ */
|
|
int evpoll(struct pollfd *fds, nfds_t nfds, int timeout)
|
|
{
|
|
struct timeval *tv, null_tv = { 0, 0} , poll_tv;
|
|
fd_set readset, writeset, exceptset;
|
|
int maxfd, rc, i;
|
|
|
|
FD_ZERO(&readset);
|
|
FD_ZERO(&writeset);
|
|
FD_ZERO(&exceptset);
|
|
|
|
/* prepare read and write fdsets */
|
|
maxfd = osmo_fd_fill_fds(&readset, &writeset, &exceptset);
|
|
|
|
for (i = 0; i < nfds; ++i) {
|
|
if (fds[i].fd < 0)
|
|
continue;
|
|
if ((fds[i].events & (POLLIN | POLLOUT | POLLERR)) == 0)
|
|
continue;
|
|
|
|
/* copy events, glib seems to map POLLPRI to exceptionset? */
|
|
if (fds[i].events & POLLIN)
|
|
FD_SET(fds[i].fd, &readset);
|
|
if (fds[i].events & POLLOUT)
|
|
FD_SET(fds[i].fd, &writeset);
|
|
if (fds[i].events & POLLERR)
|
|
FD_SET(fds[i].fd, &exceptset);
|
|
|
|
if (fds[i].fd > maxfd)
|
|
maxfd = fds[i].fd;
|
|
}
|
|
|
|
|
|
osmo_timers_check();
|
|
osmo_timers_prepare();
|
|
|
|
/*
|
|
* 0 == wake-up immediately
|
|
* -1 == block forever.. which will depend on our timers
|
|
*/
|
|
if (timeout == 0) {
|
|
tv = &null_tv;
|
|
} else if (timeout == -1)
|
|
tv = osmo_timers_nearest();
|
|
else {
|
|
poll_tv.tv_sec = timeout / 1000;
|
|
poll_tv.tv_usec = (timeout % 1000) * 1000;
|
|
|
|
tv = osmo_timers_nearest();
|
|
if (!tv)
|
|
tv = &poll_tv;
|
|
else if (timercmp(&poll_tv, tv, <))
|
|
tv = &poll_tv;
|
|
}
|
|
|
|
rc = select(maxfd+1, &readset, &writeset, &exceptset, tv);
|
|
if (rc < 0)
|
|
return 0;
|
|
|
|
/* fire timers */
|
|
osmo_timers_update();
|
|
|
|
/* call registered callback functions */
|
|
osmo_fd_disp_fds(&readset, &writeset, &exceptset);
|
|
|
|
for (i = 0; i < nfds; ++i) {
|
|
if (fds[i].fd < 0)
|
|
continue;
|
|
|
|
fds[i].revents = 0;
|
|
if (FD_ISSET(fds[i].fd, &readset))
|
|
fds[i].revents = POLLIN;
|
|
if (FD_ISSET(fds[i].fd, &writeset))
|
|
fds[i].revents |= POLLOUT;
|
|
if (FD_ISSET(fds[i].fd, &exceptset))
|
|
fds[i].revents |= POLLERR;
|
|
}
|
|
|
|
return rc;
|
|
}
|