/* Copyright 2000-2004 The Apache Software Foundation
		 *
		 * Licensed under the Apache License, Version 2.0 (the "License");
		 * you may not use this file except in compliance with the License.
		 * You may obtain a copy of the License at
		 *
		 *     http://www.apache.org/licenses/LICENSE-2.0
		 *
		 * Unless required by applicable law or agreed to in writing, software
		 * distributed under the License is distributed on an "AS IS" BASIS,
		 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		 * See the License for the specific language governing permissions and
		 * limitations under the License.
		 */
		
		#include "apr.h"
		#include "apr_poll.h"
		#include "apr_time.h"
		#include "apr_portable.h"
		#include "apr_arch_networkio.h"
		#include "apr_arch_file_io.h"
		#if HAVE_POLL_H
		#include <poll.h>
		#endif
		#if HAVE_SYS_POLL_H
		#include <sys/poll.h>
		#endif
		
		
		#ifdef NETWARE
		#define HAS_SOCKETS(dt) (dt == APR_POLL_SOCKET) ? 1 : 0
		#define HAS_PIPES(dt) (dt == APR_POLL_FILE) ? 1 : 0
		#endif
		
		
		#ifdef HAVE_POLL    /* We can just use poll to do our socket polling. */
		
		static apr_int16_t get_event(apr_int16_t event)
          54    {
          54        apr_int16_t rv = 0;
		
          54        if (event & APR_POLLIN)
          50            rv |= POLLIN;        
          54        if (event & APR_POLLPRI)
      ######            rv |= POLLPRI;        
          54        if (event & APR_POLLOUT)
           4            rv |= POLLOUT;       
          54        if (event & APR_POLLERR)
      ######            rv |= POLLERR;        
          54        if (event & APR_POLLHUP)
      ######            rv |= POLLHUP;        
          54        if (event & APR_POLLNVAL)
      ######            rv |= POLLNVAL;        
		
          54        return rv;
		}
		
		static apr_int16_t get_revent(apr_int16_t event)
          13    {
          13        apr_int16_t rv = 0;
		
          13        if (event & POLLIN)
           4            rv |= APR_POLLIN;
          13        if (event & POLLPRI)
      ######            rv |= APR_POLLPRI;
          13        if (event & POLLOUT)
           9            rv |= APR_POLLOUT;
          13        if (event & POLLERR)
      ######            rv |= APR_POLLERR;
          13        if (event & POLLHUP)
      ######            rv |= APR_POLLHUP;
          13        if (event & POLLNVAL)
      ######            rv |= APR_POLLNVAL;
		
          13        return rv;
		}        
		
		#define SMALL_POLLSET_LIMIT  8
		
		APR_DECLARE(apr_status_t) apr_poll(apr_pollfd_t *aprset, apr_int32_t num,
		                      apr_int32_t *nsds, apr_interval_time_t timeout)
      ######    {
      ######        int i, num_to_poll;
		#ifdef HAVE_VLA
		    /* XXX: I trust that this is a segv when insufficient stack exists? */
      ######        struct pollfd pollset[num];
		#elif defined(HAVE_ALLOCA)
		    struct pollfd *pollset = alloca(sizeof(struct pollfd) * num);
		    if (!pollset)
		        return APR_ENOMEM;
		#else
		    struct pollfd tmp_pollset[SMALL_POLLSET_LIMIT];
		    struct pollfd *pollset;
		
		    if (num <= SMALL_POLLSET_LIMIT) {
		        pollset = tmp_pollset;
		    }
		    else {
		        /* This does require O(n) to copy the descriptors to the internal
		         * mapping.
		         */
		        pollset = malloc(sizeof(struct pollfd) * num);
		        /* The other option is adding an apr_pool_abort() fn to invoke
		         * the pool's out of memory handler
		         */
		        if (!pollset)
		            return APR_ENOMEM;
		    }
		#endif
      ######        for (i = 0; i < num; i++) {
      ######            if (aprset[i].desc_type == APR_POLL_SOCKET) {
      ######                pollset[i].fd = aprset[i].desc.s->socketdes;
		        }
      ######            else if (aprset[i].desc_type == APR_POLL_FILE) {
      ######                pollset[i].fd = aprset[i].desc.f->filedes;
		        }
		        else {
      ######                break;
		        }
      ######            pollset[i].events = get_event(aprset[i].reqevents);
		    }
      ######        num_to_poll = i;
		
      ######        if (timeout > 0) {
      ######            timeout /= 1000; /* convert microseconds to milliseconds */
		    }
		
      ######        i = poll(pollset, num_to_poll, timeout);
      ######        (*nsds) = i;
		
      ######        for (i = 0; i < num; i++) {
      ######            aprset[i].rtnevents = get_revent(pollset[i].revents);
		    }
		    
		#if !defined(HAVE_VLA) && !defined(HAVE_ALLOCA)
		    if (num > SMALL_POLLSET_LIMIT) {
		        free(pollset);
		    }
		#endif
		
      ######        if ((*nsds) < 0) {
      ######            return apr_get_netos_error();
		    }
      ######        if ((*nsds) == 0) {
      ######            return APR_TIMEUP;
		    }
      ######        return APR_SUCCESS;
		}
		
		
		#else    /* Use select to mimic poll */
		
		APR_DECLARE(apr_status_t) apr_poll(apr_pollfd_t *aprset, int num, apr_int32_t *nsds, 
				    apr_interval_time_t timeout)
		{
		    fd_set readset, writeset, exceptset;
		    int rv, i;
		    int maxfd = -1;
		    struct timeval tv, *tvptr;
		#ifdef NETWARE
		    apr_datatype_e set_type = APR_NO_DESC;
		#endif
		
		    if (timeout < 0) {
		        tvptr = NULL;
		    }
		    else {
		        tv.tv_sec = (long)apr_time_sec(timeout);
		        tv.tv_usec = (long)apr_time_usec(timeout);
		        tvptr = &tv;
		    }
		
		    FD_ZERO(&readset);
		    FD_ZERO(&writeset);
		    FD_ZERO(&exceptset);
		
		    for (i = 0; i < num; i++) {
		        apr_os_sock_t fd;
		
		        aprset[i].rtnevents = 0;
		
		        if (aprset[i].desc_type == APR_POLL_SOCKET) {
		#ifdef NETWARE
		            if (HAS_PIPES(set_type)) {
		                return APR_EBADF;
		            }
		            else {
		                set_type = APR_POLL_SOCKET;
		            }
		#endif
		            fd = aprset[i].desc.s->socketdes;
		        }
		        else if (aprset[i].desc_type == APR_POLL_FILE) {
		#if !APR_FILES_AS_SOCKETS
		            return APR_EBADF;
		#else
		#ifdef NETWARE
		            if (aprset[i].desc.f->is_pipe && !HAS_SOCKETS(set_type)) {
		                set_type = APR_POLL_FILE;
		            }
		            else
		                return APR_EBADF;
		#endif /* NETWARE */
		
		            fd = aprset[i].desc.f->filedes;
		
		#endif /* APR_FILES_AS_SOCKETS */
		        }
		        else {
		            break;
		        }
		#if !defined(WIN32) && !defined(NETWARE) /* socket sets handled with array of handles */
		        if (fd >= FD_SETSIZE) {
		            /* XXX invent new error code so application has a clue */
		            return APR_EBADF;
		        }
		#endif
		        if (aprset[i].reqevents & APR_POLLIN) {
		            FD_SET(fd, &readset);
		        }
		        if (aprset[i].reqevents & APR_POLLOUT) {
		            FD_SET(fd, &writeset);
		        }
		        if (aprset[i].reqevents & 
		            (APR_POLLPRI | APR_POLLERR | APR_POLLHUP | APR_POLLNVAL)) {
		            FD_SET(fd, &exceptset);
		        }
		        if ((int)fd > maxfd) {
		            maxfd = (int)fd;
		        }
		    }
		
		#ifdef NETWARE
		    if (HAS_PIPES(set_type)) {
		        rv = pipe_select(maxfd + 1, &readset, &writeset, &exceptset, tvptr);
		    }
		    else {
		#endif
		
		    rv = select(maxfd + 1, &readset, &writeset, &exceptset, tvptr);
		
		#ifdef NETWARE
		    }
		#endif
		
		    (*nsds) = rv;
		    if ((*nsds) == 0) {
		        return APR_TIMEUP;
		    }
		    if ((*nsds) < 0) {
		        return apr_get_netos_error();
		    }
		
		    for (i = 0; i < num; i++) {
		        apr_os_sock_t fd;
		
		        if (aprset[i].desc_type == APR_POLL_SOCKET) {
		            fd = aprset[i].desc.s->socketdes;
		        }
		        else if (aprset[i].desc_type == APR_POLL_FILE) {
		#if !APR_FILES_AS_SOCKETS
		            return APR_EBADF;
		#else
		            fd = aprset[i].desc.f->filedes;
		#endif
		        }
		        else {
		            break;
		        }
		        if (FD_ISSET(fd, &readset)) {
		            aprset[i].rtnevents |= APR_POLLIN;
		        }
		        if (FD_ISSET(fd, &writeset)) {
		            aprset[i].rtnevents |= APR_POLLOUT;
		        }
		        if (FD_ISSET(fd, &exceptset)) {
		            aprset[i].rtnevents |= APR_POLLERR;
		        }
		    }
		
		    return APR_SUCCESS;
		}
		
		#endif 
		
		struct apr_pollset_t {
		    apr_uint32_t nelts;
		    apr_uint32_t nalloc;
		#ifdef HAVE_POLL
		    struct pollfd *pollset;
		#else
		    fd_set readset, writeset, exceptset;
		    int maxfd;
		#endif
		    apr_pollfd_t *query_set;
		    apr_pollfd_t *result_set;
		
		#ifdef NETWARE
		    int set_type;
		#endif
		};
		
		APR_DECLARE(apr_status_t) apr_pollset_create(apr_pollset_t **pollset,
		                                             apr_uint32_t size,
		                                             apr_pool_t *p,
		                                             apr_uint32_t flags)
           2    {
		#if !defined(HAVE_POLL) && defined(FD_SETSIZE)
		    if (size > FD_SETSIZE) {
		        *pollset = NULL;
		        return APR_EINVAL;
		    }
		#endif
           2        *pollset = apr_palloc(p, sizeof(**pollset));
           2        (*pollset)->nelts = 0;
           2        (*pollset)->nalloc = size;
		#ifdef HAVE_POLL
           2        (*pollset)->pollset = apr_palloc(p, size * sizeof(struct pollfd));
		#else
		    FD_ZERO(&((*pollset)->readset));
		    FD_ZERO(&((*pollset)->writeset));
		    FD_ZERO(&((*pollset)->exceptset));
		    (*pollset)->maxfd = 0;
		#ifdef NETWARE
		    (*pollset)->set_type = APR_NO_DESC;
		#endif
		#endif
           2        (*pollset)->query_set = apr_palloc(p, size * sizeof(apr_pollfd_t));
           2        (*pollset)->result_set = apr_palloc(p, size * sizeof(apr_pollfd_t));
		
           2        return APR_SUCCESS;
		}
		
		APR_DECLARE(apr_status_t) apr_pollset_destroy(apr_pollset_t *pollset)
      ######    {
		    /* A no-op function for now.  If we later implement /dev/poll
		     * support, we'll need to close the /dev/poll fd here
		     */
      ######        return APR_SUCCESS;
		}
		
		APR_DECLARE(apr_status_t) apr_pollset_add(apr_pollset_t *pollset,
		                                          const apr_pollfd_t *descriptor)
          54    {
		#ifndef HAVE_POLL
		    apr_os_sock_t fd;
		#endif
		
          54        if (pollset->nelts == pollset->nalloc) {
      ######            return APR_ENOMEM;
		    }
		
          54        pollset->query_set[pollset->nelts] = *descriptor;
		#ifdef HAVE_POLL
		
          54        if (descriptor->desc_type == APR_POLL_SOCKET) {
          54            pollset->pollset[pollset->nelts].fd = descriptor->desc.s->socketdes;
		    }
		    else {
      ######            pollset->pollset[pollset->nelts].fd = descriptor->desc.f->filedes;
		    }
		
          54        pollset->pollset[pollset->nelts].events = get_event(descriptor->reqevents);
		#else
		    if (descriptor->desc_type == APR_POLL_SOCKET) {
		#ifdef NETWARE
		        /* NetWare can't handle mixed descriptor types in select() */
		        if (HAS_PIPES(pollset->set_type)) {
		            return APR_EBADF;
		        }
		        else {
		            pollset->set_type = APR_POLL_SOCKET;
		        }
		#endif
		        fd = descriptor->desc.s->socketdes;
		    }
		    else {
		#if !APR_FILES_AS_SOCKETS
		        return APR_EBADF;
		#else
		#ifdef NETWARE
		        /* NetWare can't handle mixed descriptor types in select() */
		        if (descriptor->desc.f->is_pipe && !HAS_SOCKETS(pollset->set_type)) {
		            pollset->set_type = APR_POLL_FILE;
		            fd = descriptor->desc.f->filedes;
		        }
		        else {
		            return APR_EBADF;
		        }
		#else
		        fd = descriptor->desc.f->filedes;
		#endif
		#endif
		    }
		#if !defined(WIN32) && !defined(NETWARE) /* socket sets handled with array of handles */
		    if (fd >= FD_SETSIZE) {
		        /* XXX invent new error code so application has a clue */
		        return APR_EBADF;
		    }
		#endif
		    if (descriptor->reqevents & APR_POLLIN) {
		        FD_SET(fd, &(pollset->readset));
		    }
		    if (descriptor->reqevents & APR_POLLOUT) {
		        FD_SET(fd, &(pollset->writeset));
		    }
		    if (descriptor->reqevents &
		        (APR_POLLPRI | APR_POLLERR | APR_POLLHUP | APR_POLLNVAL)) {
		        FD_SET(fd, &(pollset->exceptset));
		    }
		    if ((int)fd > pollset->maxfd) {
		        pollset->maxfd = (int)fd;
		    }
		#endif
          54        pollset->nelts++;
          54        return APR_SUCCESS;
		}
		
		APR_DECLARE(apr_status_t) apr_pollset_remove(apr_pollset_t *pollset,
		                                             const apr_pollfd_t *descriptor)
           2    {
           2        apr_uint32_t i;
		#ifndef HAVE_POLL
		    apr_os_sock_t fd;
		#endif
		
		#ifdef HAVE_POLL
           4        for (i = 0; i < pollset->nelts; i++) {
           4            if (descriptor->desc.s == pollset->query_set[i].desc.s) {
		            /* Found an instance of the fd: remove this and any other copies */
           2                apr_uint32_t dst = i;
           2                apr_uint32_t old_nelts = pollset->nelts;
           2                pollset->nelts--;
           5                for (i++; i < old_nelts; i++) {
           3                    if (descriptor->desc.s == pollset->query_set[i].desc.s) {
      ######                        pollset->nelts--;
		                }
		                else {
           3                        pollset->pollset[dst] = pollset->pollset[i];
           3                        pollset->query_set[dst] = pollset->query_set[i];
           3                        dst++;
		                }
		            }
           2                return APR_SUCCESS;
		        }
		    }
		
		#else /* no poll */
		    if (descriptor->desc_type == APR_POLL_SOCKET) {
		        fd = descriptor->desc.s->socketdes;
		    }
		    else {
		#if !APR_FILES_AS_SOCKETS
		        return APR_EBADF;
		#else
		        fd = descriptor->desc.f->filedes;
		#endif
		    }
		
		    for (i = 0; i < pollset->nelts; i++) {
		        if (descriptor->desc.s == pollset->query_set[i].desc.s) {
		            /* Found an instance of the fd: remove this and any other copies */
		            apr_uint32_t dst = i;
		            apr_uint32_t old_nelts = pollset->nelts;
		            pollset->nelts--;
		            for (i++; i < old_nelts; i++) {
		                if (descriptor->desc.s == pollset->query_set[i].desc.s) {
		                    pollset->nelts--;
		                }
		                else {
		                    pollset->query_set[dst] = pollset->query_set[i];
		                    dst++;
		                }
		            }
		            FD_CLR(fd, &(pollset->readset));
		            FD_CLR(fd, &(pollset->writeset));
		            FD_CLR(fd, &(pollset->exceptset));
		            if (((int)fd == pollset->maxfd) && (pollset->maxfd > 0)) {
		                pollset->maxfd--;
		            }
		            return APR_SUCCESS;
		        }
		    }
		#endif /* no poll */
		
      ######        return APR_NOTFOUND;
		}
		
		#ifdef HAVE_POLL
		APR_DECLARE(apr_status_t) apr_pollset_poll(apr_pollset_t *pollset,
		                                           apr_interval_time_t timeout,
		                                           apr_int32_t *num,
		                                           const apr_pollfd_t **descriptors)
          10    {
          10        int rv;
          10        apr_uint32_t i, j;
		
          10        if (timeout > 0) {
           3            timeout /= 1000;
		    }
          10        rv = poll(pollset->pollset, pollset->nelts, timeout);
          10        (*num) = rv;
          10        if (rv < 0) {
      ######            return apr_get_netos_error();
		    }
          10        if (rv == 0) {
           4            return APR_TIMEUP;
		    }
           6        j = 0;
         165        for (i = 0; i < pollset->nelts; i++) {
         159            if (pollset->pollset[i].revents != 0) {
          13                pollset->result_set[j] = pollset->query_set[i];
          13                pollset->result_set[j].rtnevents =
		                get_revent(pollset->pollset[i].revents);
          13                j++;
		        }
		    }
           6        if (descriptors)
           6            *descriptors = pollset->result_set;
           6        return APR_SUCCESS;
		}
		
		#else /* no poll */
		
		APR_DECLARE(apr_status_t) apr_pollset_poll(apr_pollset_t *pollset,
		                                           apr_interval_time_t timeout,
		                                           apr_int32_t *num,
		                                           const apr_pollfd_t **descriptors)
		{
		    int rv;
		    apr_uint32_t i, j;
		    struct timeval tv, *tvptr;
		    fd_set readset, writeset, exceptset;
		
		    if (timeout < 0) {
		        tvptr = NULL;
		    }
		    else {
		        tv.tv_sec = (long)apr_time_sec(timeout);
		        tv.tv_usec = (long)apr_time_usec(timeout);
		        tvptr = &tv;
		    }
		
		    memcpy(&readset, &(pollset->readset), sizeof(fd_set));
		    memcpy(&writeset, &(pollset->writeset), sizeof(fd_set));
		    memcpy(&exceptset, &(pollset->exceptset), sizeof(fd_set));
		
		#ifdef NETWARE
		    if (HAS_PIPES(pollset->set_type)) {
		        rv = pipe_select(pollset->maxfd + 1, &readset, &writeset, &exceptset, tvptr);
		    }
		    else
		#endif
		    rv = select(pollset->maxfd + 1, &readset, &writeset, &exceptset, tvptr);
		
		    (*num) = rv;
		    if (rv < 0) {
		        return apr_get_netos_error();
		    }
		    if (rv == 0) {
		        return APR_TIMEUP;
		    }
		    j = 0;
		    for (i = 0; i < pollset->nelts; i++) {
		        apr_os_sock_t fd;
		        if (pollset->query_set[i].desc_type == APR_POLL_SOCKET) {
		            fd = pollset->query_set[i].desc.s->socketdes;
		        }
		        else {
		#if !APR_FILES_AS_SOCKETS
		            return APR_EBADF;
		#else
		            fd = pollset->query_set[i].desc.f->filedes;
		#endif
		        }
		        if (FD_ISSET(fd, &readset) || FD_ISSET(fd, &writeset) ||
		            FD_ISSET(fd, &exceptset)) {
		            pollset->result_set[j] = pollset->query_set[i];
		            pollset->result_set[j].rtnevents = 0;
		            if (FD_ISSET(fd, &readset)) {
		                pollset->result_set[j].rtnevents |= APR_POLLIN;
		            }
		            if (FD_ISSET(fd, &writeset)) {
		                pollset->result_set[j].rtnevents |= APR_POLLOUT;
		            }
		            if (FD_ISSET(fd, &exceptset)) {
		                pollset->result_set[j].rtnevents |= APR_POLLERR;
		            }
		            j++;
		        }
		    }
		
		    if (descriptors)
		        *descriptors = pollset->result_set;
		    return APR_SUCCESS;
		}
		
		#endif /* no poll */