123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234 |
- /* -*- mode:c; c-file-style:"k&r"; c-basic-offset: 4; tab-width:4; indent-tabs-mode:nil; mode:auto-fill; fill-column:78; -*- */
- /* vim: set ts=4 sw=4 et tw=78 fo=cqt wm=0: */
- /* Copyright (C) 2014 Stony Brook University
- This file is part of Graphene Library OS.
- Graphene Library OS is free software: you can redistribute it and/or
- modify it under the terms of the GNU Lesser General Public License
- as published by the Free Software Foundation, either version 3 of the
- License, or (at your option) any later version.
- Graphene Library OS 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 Lesser General Public License for more details.
- You should have received a copy of the GNU Lesser General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>. */
- /*
- * db_object.c
- *
- * This file contains APIs for closing or polling PAL handles.
- */
- #include "pal_defs.h"
- #include "pal_freebsd_defs.h"
- #include "pal.h"
- #include "pal_internal.h"
- #include "pal_freebsd.h"
- #include "pal_error.h"
- #include "pal_debug.h"
- #include "api.h"
- #include <time.h>
- #include <sys/poll.h>
- #include <sys/wait.h>
- #include <atomic.h>
- #include <cmpxchg.h>
- #include <errno.h>
- #define DEFAULT_QUANTUM 500
- /* internally to wait for one object. Also used as a shortcut to wait
- on events and semaphores */
- static int _DkObjectWaitOne (PAL_HANDLE handle, int timeout)
- {
- /* only for all these handle which has a file descriptor, or
- a eventfd. events and semaphores will skip this part */
- if (handle->__in.flags & HAS_FDS) {
- struct pollfd fds[MAX_FDS];
- int off[MAX_FDS];
- int nfds = 0;
- for (int i = 0 ; i < MAX_FDS ; i++) {
- int events = 0;
- if ((handle->__in.flags & RFD(i)) &&
- !(handle->__in.flags & ERROR(i)))
- events |= POLLIN;
- if ((handle->__in.flags & WFD(i)) &&
- !(handle->__in.flags & WRITEABLE(i)) &&
- !(handle->__in.flags & ERROR(i)))
- events |= POLLOUT;
- if (events) {
- fds[nfds].fd = handle->__in.fds[i];
- fds[nfds].events = events|POLLHUP|POLLERR;
- fds[nfds].revents = 0;
- off[nfds] = i;
- nfds++;
- }
- }
- if (!nfds)
- return -PAL_ERROR_TRYAGAIN;
- int ret = INLINE_SYSCALL(poll, 3, &fds, nfds,
- timeout ? timeout : -1);
- if (IS_ERR(ret))
- switch (ERRNO(ret)) {
- case EINTR:
- return -PAL_ERROR_INTERRUPTED;
- default:
- return unix_to_pal_error(ERRNO(ret));
- }
- if (!ret)
- return -PAL_ERROR_TRYAGAIN;
- for (int i = 0 ; i < nfds ; i++) {
- if (!fds[i].revents)
- continue;
- if (fds[i].revents & POLLOUT)
- handle->__in.flags |= WRITEABLE(off[i]);
- if (fds[i].revents & (POLLHUP|POLLERR))
- handle->__in.flags |= ERROR(off[i]);
- }
- return 0;
- }
- const struct handle_ops * ops = HANDLE_OPS(handle);
- if (!ops->wait)
- return -PAL_ERROR_NOTSUPPORT;
- return ops->wait(handle, timeout);
- }
- /* _DkObjectsWaitAny for internal use. The function wait for any of the handle
- in the handle array. timeout can be set for the wait. */
- int _DkObjectsWaitAny (int count, PAL_HANDLE * handleArray, int timeout,
- PAL_HANDLE * polled)
- {
- if (count <= 0)
- return 0;
- if (count == 1) {
- *polled = handleArray[0];
- return _DkObjectWaitOne(handleArray[0], timeout);
- }
- int i, j, ret, maxfds = 0, nfds = 0;
- /* we are not gonna to allow any polling on muliple synchronous
- objects, doing this is simply violating the division of
- labor between PAL and library OS */
- for (i = 0 ; i < count ; i++) {
- PAL_HANDLE hdl = handleArray[i];
- if (!hdl)
- continue;
- if (!(hdl->__in.flags & HAS_FDS))
- return -PAL_ERROR_NOTSUPPORT;
- /* eliminate repeated entries */
- for (j = 0 ; j < i ; j++)
- if (hdl == handleArray[j])
- break;
- if (j == i) {
- for (j = 0 ; j < MAX_FDS ; j++)
- if (hdl->__in.flags & (RFD(j)|WFD(j)))
- maxfds++;
- }
- }
- struct pollfd * fds = __alloca(sizeof(struct pollfd) * maxfds);
- PAL_HANDLE * hdls = __alloca(sizeof(PAL_HANDLE) * maxfds);
- for (i = 0 ; i < count ; i++) {
- PAL_HANDLE hdl = handleArray[i];
- if (!hdl)
- continue;
- for (j = 0 ; j < i ; j++)
- if (hdl == handleArray[j])
- break;
- if (j < i)
- continue;
- for (j = 0 ; j < MAX_FDS ; j++) {
- int events = 0;
- if ((hdl->__in.flags & RFD(j)) &&
- !(hdl->__in.flags & ERROR(j)))
- events |= POLLIN;
- if ((hdl->__in.flags & WFD(j)) &&
- !(hdl->__in.flags & WRITEABLE(j)) &&
- !(hdl->__in.flags & ERROR(j)))
- events |= POLLOUT;
- if (events && hdl->__in.fds[j] != PAL_IDX_POISON) {
- fds[nfds].fd = hdl->__in.fds[j];
- fds[nfds].events = events|POLLHUP|POLLERR;
- fds[nfds].revents = 0;
- hdls[nfds] = hdl;
- nfds++;
- }
- }
- }
- if (!nfds)
- return -PAL_ERROR_TRYAGAIN;
- ret = INLINE_SYSCALL(poll, 3, fds, nfds, timeout ? timeout : -1);
- if (IS_ERR(ret))
- switch (ERRNO(ret)) {
- case EINTR:
- return -PAL_ERROR_INTERRUPTED;
- default:
- return unix_to_pal_error(ERRNO(ret));
- }
- if (!ret)
- return -PAL_ERROR_TRYAGAIN;
- PAL_HANDLE polled_hdl = NULL;
- for (i = 0 ; i < nfds ; i++) {
- if (!fds[i].revents)
- continue;
- PAL_HANDLE hdl = hdls[i];
- if (polled_hdl) {
- if (hdl != polled_hdl)
- continue;
- } else {
- polled_hdl = hdl;
- }
- for (j = 0 ; j < MAX_FDS ; j++)
- if ((hdl->__in.flags & (RFD(j)|WFD(j))) &&
- hdl->__in.fds[j] == fds[i].fd)
- break;
- if (j == MAX_FDS)
- continue;
- if (fds[i].revents & POLLOUT)
- hdl->__in.flags |= WRITEABLE(j);
- if (fds[i].revents & (POLLHUP|POLLERR))
- hdl->__in.flags |= ERROR(j);
- }
- *polled = polled_hdl;
- return polled_hdl ? 0 : -PAL_ERROR_TRYAGAIN;
- }
|