1251875Speter/* Licensed to the Apache Software Foundation (ASF) under one or more 2251875Speter * contributor license agreements. See the NOTICE file distributed with 3251875Speter * this work for additional information regarding copyright ownership. 4251875Speter * The ASF licenses this file to You under the Apache License, Version 2.0 5251875Speter * (the "License"); you may not use this file except in compliance with 6251875Speter * the License. You may obtain a copy of the License at 7251875Speter * 8251875Speter * http://www.apache.org/licenses/LICENSE-2.0 9251875Speter * 10251875Speter * Unless required by applicable law or agreed to in writing, software 11251875Speter * distributed under the License is distributed on an "AS IS" BASIS, 12251875Speter * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13251875Speter * See the License for the specific language governing permissions and 14251875Speter * limitations under the License. 15251875Speter */ 16251875Speter 17251875Speter#include "apr_arch_file_io.h" 18251875Speter#include "apr_arch_networkio.h" 19251875Speter#include "apr_poll.h" 20251875Speter#include "apr_errno.h" 21251875Speter#include "apr_support.h" 22251875Speter 23251875Speter/* The only case where we don't use wait_for_io_or_timeout is on 24251875Speter * pre-BONE BeOS, so this check should be sufficient and simpler */ 25266735Speter#if !defined(BEOS_R5) && !defined(OS2) && APR_FILES_AS_SOCKETS 26251875Speter#define USE_WAIT_FOR_IO 27251875Speter#endif 28251875Speter 29251875Speter#ifdef USE_WAIT_FOR_IO 30251875Speter 31251875Speter#ifdef WAITIO_USES_POLL 32251875Speter 33251875Speter#ifdef HAVE_POLL_H 34251875Speter#include <poll.h> 35251875Speter#endif 36251875Speter#ifdef HAVE_SYS_POLL_H 37251875Speter#include <sys/poll.h> 38251875Speter#endif 39251875Speter 40251875Speterapr_status_t apr_wait_for_io_or_timeout(apr_file_t *f, apr_socket_t *s, 41251875Speter int for_read) 42251875Speter{ 43251875Speter struct pollfd pfd; 44251875Speter int rc, timeout; 45251875Speter 46251875Speter timeout = f ? f->timeout / 1000 : s->timeout / 1000; 47251875Speter pfd.fd = f ? f->filedes : s->socketdes; 48251875Speter pfd.events = for_read ? POLLIN : POLLOUT; 49251875Speter 50251875Speter do { 51251875Speter rc = poll(&pfd, 1, timeout); 52251875Speter } while (rc == -1 && errno == EINTR); 53251875Speter if (rc == 0) { 54251875Speter return APR_TIMEUP; 55251875Speter } 56251875Speter else if (rc > 0) { 57251875Speter return APR_SUCCESS; 58251875Speter } 59251875Speter else { 60251875Speter return errno; 61251875Speter } 62251875Speter} 63251875Speter 64251875Speter#else /* !WAITIO_USES_POLL */ 65251875Speter 66251875Speterapr_status_t apr_wait_for_io_or_timeout(apr_file_t *f, apr_socket_t *s, 67251875Speter int for_read) 68251875Speter{ 69251875Speter apr_interval_time_t timeout; 70251875Speter apr_pollfd_t pfd; 71251875Speter int type = for_read ? APR_POLLIN : APR_POLLOUT; 72251875Speter apr_pollset_t *pollset; 73251875Speter apr_status_t status; 74251875Speter 75251875Speter /* TODO - timeout should be less each time through this loop */ 76251875Speter if (f) { 77251875Speter pfd.desc_type = APR_POLL_FILE; 78251875Speter pfd.desc.f = f; 79251875Speter 80251875Speter pollset = f->pollset; 81251875Speter if (pollset == NULL) { 82251875Speter status = apr_pollset_create(&(f->pollset), 1, f->pool, 0); 83251875Speter if (status != APR_SUCCESS) { 84251875Speter return status; 85251875Speter } 86251875Speter pollset = f->pollset; 87251875Speter } 88251875Speter timeout = f->timeout; 89251875Speter } 90251875Speter else { 91251875Speter pfd.desc_type = APR_POLL_SOCKET; 92251875Speter pfd.desc.s = s; 93251875Speter 94251875Speter pollset = s->pollset; 95251875Speter timeout = s->timeout; 96251875Speter } 97251875Speter pfd.reqevents = type; 98251875Speter 99251875Speter /* Remove the object if it was in the pollset, then add in the new 100251875Speter * object with the correct reqevents value. Ignore the status result 101251875Speter * on the remove, because it might not be in there (yet). 102251875Speter */ 103251875Speter (void) apr_pollset_remove(pollset, &pfd); 104251875Speter 105251875Speter /* ### check status code */ 106251875Speter (void) apr_pollset_add(pollset, &pfd); 107251875Speter 108251875Speter do { 109251875Speter int numdesc; 110251875Speter const apr_pollfd_t *pdesc; 111251875Speter 112251875Speter status = apr_pollset_poll(pollset, timeout, &numdesc, &pdesc); 113251875Speter 114251875Speter if (numdesc == 1 && (pdesc[0].rtnevents & type) != 0) { 115251875Speter return APR_SUCCESS; 116251875Speter } 117251875Speter } while (APR_STATUS_IS_EINTR(status)); 118251875Speter 119251875Speter return status; 120251875Speter} 121251875Speter#endif /* WAITIO_USES_POLL */ 122251875Speter 123251875Speter#endif /* USE_WAIT_FOR_IO */ 124