a9cdfc29 |
/*
|
0ba367ec |
* $Id$
|
a9cdfc29 |
*
|
0ba367ec |
* Copyright (C) 2005 iptelorg GmbH
*
|
a0553f4e |
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
|
0ba367ec |
*
|
a0553f4e |
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
0ba367ec |
*/
/*
* tcp io wait common stuff used by tcp_main.c & tcp_read.c
* All the functions are inline because of speed reasons and because they are
* used only from 2 places.
* You also have to define:
|
a0553f4e |
* int handle_io(struct fd_map* fm, short events, int idx) (see below)
|
0ba367ec |
* (this could be trivially replaced by a callback pointer entry attached
* to the io_wait handler if more flexibility rather then performance
* is needed)
* fd_type - define to some enum of you choice and define also
* FD_TYPE_DEFINED (if you don't do it fd_type will be defined
* to int). 0 has a special not set/not init. meaning
* (a lot of sanity checks and the sigio_rt code are based on
* this assumption)
* local_malloc (defaults to pkg_malloc)
* local_free (defaults to pkg_free)
|
a9cdfc29 |
*
|
0ba367ec |
*/
|
a9cdfc29 |
/*
|
0ba367ec |
* History:
* --------
* 2005-06-13 created by andrei
|
741a9937 |
* 2005-06-26 added kqueue (andrei)
|
9eda5956 |
* 2005-07-01 added /dev/poll (andrei)
|
19db15d7 |
* 2006-05-30 sigio 64 bit workarround enabled for kernels < 2.6.5 (andrei)
|
5702d448 |
* 2007-11-22 when handle_io() is called in a loop check & stop if the fd was
* removed inside handle_io() (andrei)
|
a0553f4e |
* 2007-11-29 support for write (POLLOUT); added io_watch_chg() (andrei)
|
5d364ef9 |
* 2008-02-04 POLLRDHUP & EPOLLRDHUP support (automatically enabled if POLLIN
* is set) (andrei)
|
05c080a5 |
* 2010-06-17 re-enabled & enhanced the EV_ERROR for kqueue (andrei)
|
0ba367ec |
*/
#ifndef _io_wait_h
#define _io_wait_h
#include <errno.h>
#include <string.h>
#ifdef HAVE_SIGIO_RT
#define __USE_GNU /* or else F_SETSIG won't be included */
#include <sys/types.h> /* recv */
#include <sys/socket.h> /* recv */
#include <signal.h> /* sigprocmask, sigwait a.s.o */
#endif
|
5d364ef9 |
#define _GNU_SOURCE /* for POLLRDHUP on linux */
#include <sys/poll.h>
#include <fcntl.h>
|
0ba367ec |
#ifdef HAVE_EPOLL
#include <sys/epoll.h>
#endif
|
741a9937 |
#ifdef HAVE_KQUEUE
#include <sys/types.h> /* needed on freebsd */
#include <sys/event.h>
#include <sys/time.h>
#endif
|
9eda5956 |
#ifdef HAVE_DEVPOLL
#include <sys/devpoll.h>
#endif
|
0ba367ec |
#ifdef HAVE_SELECT
|
26376604 |
/* needed on openbsd for select*/
|
a9cdfc29 |
#include <sys/time.h>
#include <sys/types.h>
|
26376604 |
#include <unistd.h>
/* needed according to POSIX for select*/
|
0ba367ec |
#include <sys/select.h>
#endif
#include "dprint.h"
#include "poll_types.h" /* poll_types*/
#ifdef HAVE_SIGIO_RT
#include "pt.h" /* mypid() */
#endif
|
8880cb34 |
#include "compiler_opt.h"
|
0ba367ec |
|
5d364ef9 |
#ifdef HAVE_EPOLL
/* fix defines for EPOLL */
#if defined POLLRDHUP && ! defined EPOLLRDHUP
#define EPOLLRDHUP POLLRDHUP /* should work on all linuxes */
#endif /* POLLRDHUP && EPOLLRDHUP */
#endif /* HAVE_EPOLL */
|
19db15d7 |
extern int _os_ver; /* os version number, needed to select bugs workarrounds */
|
0ba367ec |
#if 0
enum fd_types; /* this should be defined from the including file,
|
a9cdfc29 |
see tcp_main.c for an example,
|
0ba367ec |
0 has a special meaning: not used/empty*/
#endif
#ifndef FD_TYPE_DEFINED
typedef int fd_type;
#define FD_TYPE_DEFINED
#endif
/* maps a fd to some other structure; used in almost all cases
* except epoll and maybe kqueue or /dev/poll */
struct fd_map{
int fd; /* fd no */
fd_type type; /* "data" type */
void* data; /* pointer to the corresponding structure */
|
a0553f4e |
short events; /* events we are interested int */
|
0ba367ec |
};
|
741a9937 |
#ifdef HAVE_KQUEUE
#ifndef KQ_CHANGES_ARRAY_SIZE
|
a0553f4e |
#define KQ_CHANGES_ARRAY_SIZE 256
|
741a9937 |
#ifdef __OS_netbsd
#define KEV_UDATA_CAST (intptr_t)
#else
#define KEV_UDATA_CAST
#endif
#endif
#endif
|
0ba367ec |
/* handler structure */
struct io_wait_handler{
|
89deaf49 |
enum poll_types poll_method;
int flags;
struct fd_map* fd_hash;
|
a9cdfc29 |
int fd_no; /* current index used in fd_array and the passed size for
|
99682621 |
ep_array (for kq_array at least
max(twice the size, kq_changes_size) should be
be passed). */
|
89deaf49 |
int max_fd_no; /* maximum fd no, is also the size of fd_array,
fd_hash and ep_array*/
/* common stuff for POLL, SIGIO_RT and SELECT
* since poll support is always compiled => this will always be compiled */
struct pollfd* fd_array; /* used also by devpoll as devpoll array */
int crt_fd_array_idx; /* crt idx for which handle_io is called
(updated also by del -> internal optimization) */
/* end of common stuff */
|
0ba367ec |
#ifdef HAVE_EPOLL
int epfd; /* epoll ctrl fd */
|
89deaf49 |
struct epoll_event* ep_array;
|
0ba367ec |
#endif
#ifdef HAVE_SIGIO_RT
sigset_t sset; /* signal mask for sigio & sigrtmin */
int signo; /* real time signal used */
#endif
|
741a9937 |
#ifdef HAVE_KQUEUE
|
89deaf49 |
int kq_fd;
|
741a9937 |
struct kevent* kq_array; /* used for the eventlist*/
struct kevent* kq_changes; /* used for the changelist */
size_t kq_nchanges;
|
99682621 |
size_t kq_array_size; /* array size */
|
741a9937 |
size_t kq_changes_size; /* size of the changes array */
#endif
|
9eda5956 |
#ifdef HAVE_DEVPOLL
int dpoll_fd;
#endif
|
0ba367ec |
#ifdef HAVE_SELECT
|
a0553f4e |
fd_set master_rset; /* read set */
fd_set master_wset; /* write set */
|
0ba367ec |
int max_fd_select; /* maximum select used fd */
#endif
};
typedef struct io_wait_handler io_wait_h;
/* get the corresponding fd_map structure pointer */
#define get_fd_map(h, fd) (&(h)->fd_hash[(fd)])
/* remove a fd_map structure from the hash; the pointer must be returned
* by get_fd_map or hash_fd_map*/
#define unhash_fd_map(pfm) \
do{ \
(pfm)->type=0 /*F_NONE */; \
(pfm)->fd=-1; \
}while(0)
/* add a fd_map structure to the fd hash */
static inline struct fd_map* hash_fd_map( io_wait_h* h,
int fd,
|
a0553f4e |
short events,
|
0ba367ec |
fd_type type,
void* data)
{
h->fd_hash[fd].fd=fd;
|
a0553f4e |
h->fd_hash[fd].events=events;
|
0ba367ec |
h->fd_hash[fd].type=type;
h->fd_hash[fd].data=data;
return &h->fd_hash[fd];
}
#ifdef HANDLE_IO_INLINE
/* generic handle io routine, this must be defined in the including file
* (faster then registering a callback pointer)
*
|
a0553f4e |
* params: fm - pointer to a fd hash entry
* events - combinations of POLLIN, POLLOUT, POLLERR & POLLHUP
* idx - index in the fd_array (or -1 if not known)
|
0ba367ec |
* return: -1 on error
|
a9cdfc29 |
* 0 on EAGAIN or when by some other way it is known that no more
|
0ba367ec |
* io events are queued on the fd (the receive buffer is empty).
* Usefull to detect when there are no more io events queued for
* sigio_rt, epoll_et, kqueue.
* >0 on successfull read from the fd (when there might be more io
* queued -- the receive buffer might still be non-empty)
*/
|
a0553f4e |
inline static int handle_io(struct fd_map* fm, short events, int idx);
|
0ba367ec |
#else
|
a0553f4e |
int handle_io(struct fd_map* fm, short events, int idx);
|
0ba367ec |
#endif
|
741a9937 |
#ifdef HAVE_KQUEUE
/*
* kqueue specific function: register a change
* (adds a change to the kevent change array, and if full flushes it first)
|
a0553f4e |
*
* TODO: check if the event already exists in the change list or if it's
* complementary to an event in the list (e.g. EVFILT_WRITE, EV_DELETE
* and EVFILT_WRITE, EV_ADD for the same fd).
|
741a9937 |
* returns: -1 on error, 0 on success
*/
|
a9cdfc29 |
static inline int kq_ev_change(io_wait_h* h, int fd, int filter, int flag,
|
741a9937 |
void* data)
{
int n;
|
b0bd3201 |
int r;
|
741a9937 |
struct timespec tspec;
if (h->kq_nchanges>=h->kq_changes_size){
/* changes array full ! */
LOG(L_WARN, "WARNING: kq_ev_change: kqueue changes array full"
" trying to flush...\n");
tspec.tv_sec=0;
tspec.tv_nsec=0;
again:
n=kevent(h->kq_fd, h->kq_changes, h->kq_nchanges, 0, 0, &tspec);
|
b0bd3201 |
if (unlikely(n == -1)){
|
8966d0a1 |
if (unlikely(errno == EINTR)) goto again;
else {
/* for a detailed explanation of what follows see
io_wait_loop_kqueue EV_ERROR case */
if (unlikely(!(errno == EBADF || errno == ENOENT)))
BUG("kq_ev_change: kevent flush changes failed"
" (unexpected error): %s [%d]\n",
strerror(errno), errno);
/* ignore error even if it's not a EBADF/ENOENT */
|
b0bd3201 |
/* one of the file descriptors is bad, probably already
closed => try to apply changes one-by-one */
for (r = 0; r < h->kq_nchanges; r++) {
retry2:
n = kevent(h->kq_fd, &h->kq_changes[r], 1, 0, 0, &tspec);
if (n==-1) {
|
8966d0a1 |
if (unlikely(errno == EINTR))
|
b0bd3201 |
goto retry2;
|
8966d0a1 |
/* for a detailed explanation of what follows see
io_wait_loop_kqueue EV_ERROR case */
if (unlikely(!(errno == EBADF || errno == ENOENT)))
BUG("kq_ev_change: kevent flush changes failed:"
|
c7694799 |
" (unexpected error) %s [%d] (%d/%lu)\n",
|
8966d0a1 |
strerror(errno), errno,
|
c7694799 |
r, (unsigned long)h->kq_nchanges);
|
8966d0a1 |
continue; /* skip over it */
|
b0bd3201 |
}
}
}
|
741a9937 |
}
h->kq_nchanges=0; /* changes array is empty */
}
EV_SET(&h->kq_changes[h->kq_nchanges], fd, filter, flag, 0, 0,
KEV_UDATA_CAST data);
h->kq_nchanges++;
return 0;
}
#endif
|
0ba367ec |
/* generic io_watch_add function
|
a0553f4e |
* Params:
* h - pointer to initialized io_wait handle
* fd - fd to watch
* events - bitmap with the fd events for which the fd should be watched
* (combination of POLLIN and POLLOUT)
* type - fd type (non 0 value, returned in the call to handle_io)
* data - pointer/private data returned in the handle_io call
|
0ba367ec |
* returns 0 on success, -1 on error
|
a0553f4e |
*
|
5702d448 |
* WARNING: handle_io() can be called immediately (from io_watch_add()) so
* make sure that any dependent init. (e.g. data stuff) is made before
* calling io_watch_add
|
0ba367ec |
*
* this version should be faster than pointers to poll_method specific
* functions (it avoids functions calls, the overhead being only an extra
* switch())*/
inline static int io_watch_add( io_wait_h* h,
int fd,
|
a0553f4e |
short events,
|
0ba367ec |
fd_type type,
void* data)
{
/* helper macros */
|
a0553f4e |
#define fd_array_setup(ev) \
|
0ba367ec |
do{ \
h->fd_array[h->fd_no].fd=fd; \
|
a0553f4e |
h->fd_array[h->fd_no].events=(ev); /* useless for select */ \
|
0ba367ec |
h->fd_array[h->fd_no].revents=0; /* useless for select */ \
}while(0)
#define set_fd_flags(f) \
do{ \
flags=fcntl(fd, F_GETFL); \
if (flags==-1){ \
LOG(L_ERR, "ERROR: io_watch_add: fnctl: GETFL failed:" \
" %s [%d]\n", strerror(errno), errno); \
goto error; \
} \
if (fcntl(fd, F_SETFL, flags|(f))==-1){ \
LOG(L_ERR, "ERROR: io_watch_add: fnctl: SETFL" \
" failed: %s [%d]\n", strerror(errno), errno); \
goto error; \
} \
}while(0)
struct fd_map* e;
int flags;
#ifdef HAVE_EPOLL
struct epoll_event ep_event;
#endif
|
9eda5956 |
#ifdef HAVE_DEVPOLL
struct pollfd pfd;
#endif
|
3665e5c1 |
#if defined(HAVE_SIGIO_RT) || defined (HAVE_EPOLL)
|
0054114a |
int n;
|
3665e5c1 |
#endif
#if defined(HAVE_SIGIO_RT)
|
3135b4bb |
int idx;
int check_io;
struct pollfd pf;
|
0ba367ec |
|
c742b6a4 |
check_io=0; /* set to 1 if we need to check for pre-existing queued
|
3135b4bb |
io/data on the fd */
idx=-1;
#endif
e=0;
|
a0553f4e |
/* sanity checks */
if (unlikely(fd==-1)){
|
0ba367ec |
LOG(L_CRIT, "BUG: io_watch_add: fd is -1!\n");
goto error;
}
|
a0553f4e |
if (unlikely((events&(POLLIN|POLLOUT))==0)){
LOG(L_CRIT, "BUG: io_watch_add: invalid events: 0x%0x\n", events);
goto error;
}
|
741a9937 |
/* check if not too big */
|
a0553f4e |
if (unlikely(h->fd_no>=h->max_fd_no)){
|
0ba367ec |
LOG(L_CRIT, "ERROR: io_watch_add: maximum fd number exceeded:"
" %d/%d\n", h->fd_no, h->max_fd_no);
goto error;
}
DBG("DBG: io_watch_add(%p, %d, %d, %p), fd_no=%d\n",
h, fd, type, data, h->fd_no);
/* hash sanity check */
e=get_fd_map(h, fd);
|
a0553f4e |
if (unlikely(e && (e->type!=0 /*F_NONE*/))){
|
0ba367ec |
LOG(L_ERR, "ERROR: io_watch_add: trying to overwrite entry %d"
|
316130a7 |
" watched for %x in the hash(%d, %d, %p) with (%d, %d, %p)\n",
fd, events, e->fd, e->type, e->data, fd, type, data);
e=0;
|
0ba367ec |
goto error;
}
|
a0553f4e |
if (unlikely((e=hash_fd_map(h, fd, events, type, data))==0)){
|
0ba367ec |
LOG(L_ERR, "ERROR: io_watch_add: failed to hash the fd %d\n", fd);
goto error;
}
switch(h->poll_method){ /* faster then pointer to functions */
case POLL_POLL:
|
5d364ef9 |
#ifdef POLLRDHUP
/* listen to POLLRDHUP by default (if POLLIN) */
events|=((int)!(events & POLLIN) - 1) & POLLRDHUP;
#endif /* POLLRDHUP */
|
a0553f4e |
fd_array_setup(events);
|
0ba367ec |
set_fd_flags(O_NONBLOCK);
break;
#ifdef HAVE_SELECT
case POLL_SELECT:
|
a0553f4e |
fd_array_setup(events);
if (likely(events & POLLIN))
FD_SET(fd, &h->master_rset);
if (unlikely(events & POLLOUT))
FD_SET(fd, &h->master_wset);
|
0ba367ec |
if (h->max_fd_select<fd) h->max_fd_select=fd;
break;
#endif
#ifdef HAVE_SIGIO_RT
case POLL_SIGIO_RT:
|
a0553f4e |
fd_array_setup(events);
|
a9cdfc29 |
/* re-set O_ASYNC might be needed, if not done from
|
3135b4bb |
* io_watch_del (or if somebody wants to add a fd which has
|
a0553f4e |
* already O_ASYNC/F_SETSIG set on a duplicate)
|
3135b4bb |
*/
|
0ba367ec |
/* set async & signal */
if (fcntl(fd, F_SETOWN, my_pid())==-1){
LOG(L_ERR, "ERROR: io_watch_add: fnctl: SETOWN"
" failed: %s [%d]\n", strerror(errno), errno);
goto error;
}
if (fcntl(fd, F_SETSIG, h->signo)==-1){
LOG(L_ERR, "ERROR: io_watch_add: fnctl: SETSIG"
" failed: %s [%d]\n", strerror(errno), errno);
goto error;
}
/* set both non-blocking and async */
set_fd_flags(O_ASYNC| O_NONBLOCK);
#ifdef EXTRA_DEBUG
DBG("io_watch_add: sigio_rt on f %d, signal %d to pid %d\n",
|
eab43383 |
fd, h->signo, my_pid());
|
0ba367ec |
#endif
/* empty socket receive buffer, if buffer is already full
|
3135b4bb |
* no more space to put packets
* => no more signals are ever generated
* also when moving fds, the freshly moved fd might have
* already some bytes queued, we want to get them now
* and not later -- andrei */
idx=h->fd_no;
check_io=1;
|
0ba367ec |
break;
#endif
#ifdef HAVE_EPOLL
case POLL_EPOLL_LT:
|
5d364ef9 |
ep_event.events=
#ifdef POLLRDHUP
/* listen for EPOLLRDHUP too */
((EPOLLIN|EPOLLRDHUP) & ((int)!(events & POLLIN)-1) ) |
#else /* POLLRDHUP */
(EPOLLIN & ((int)!(events & POLLIN)-1) ) |
#endif /* POLLRDHUP */
(EPOLLOUT & ((int)!(events & POLLOUT)-1) );
|
0ba367ec |
ep_event.data.ptr=e;
again1:
n=epoll_ctl(h->epfd, EPOLL_CTL_ADD, fd, &ep_event);
|
a0553f4e |
if (unlikely(n==-1)){
|
0ba367ec |
if (errno==EAGAIN) goto again1;
LOG(L_ERR, "ERROR: io_watch_add: epoll_ctl failed: %s [%d]\n",
strerror(errno), errno);
goto error;
}
break;
case POLL_EPOLL_ET:
set_fd_flags(O_NONBLOCK);
|
5d364ef9 |
ep_event.events=
#ifdef POLLRDHUP
/* listen for EPOLLRDHUP too */
((EPOLLIN|EPOLLRDHUP) & ((int)!(events & POLLIN)-1) ) |
#else /* POLLRDHUP */
(EPOLLIN & ((int)!(events & POLLIN)-1) ) |
#endif /* POLLRDHUP */
(EPOLLOUT & ((int)!(events & POLLOUT)-1) ) |
EPOLLET;
|
0ba367ec |
ep_event.data.ptr=e;
again2:
n=epoll_ctl(h->epfd, EPOLL_CTL_ADD, fd, &ep_event);
|
a0553f4e |
if (unlikely(n==-1)){
|
0ba367ec |
if (errno==EAGAIN) goto again2;
LOG(L_ERR, "ERROR: io_watch_add: epoll_ctl failed: %s [%d]\n",
strerror(errno), errno);
goto error;
}
break;
|
741a9937 |
#endif
#ifdef HAVE_KQUEUE
case POLL_KQUEUE:
|
8880cb34 |
if (likely( events & POLLIN)){
|
a0553f4e |
if (unlikely(kq_ev_change(h, fd, EVFILT_READ, EV_ADD, e)==-1))
|
e5be1a06 |
goto error;
|
a0553f4e |
}
if (unlikely( events & POLLOUT)){
if (unlikely(kq_ev_change(h, fd, EVFILT_WRITE, EV_ADD, e)==-1))
|
316130a7 |
{
if (likely(events & POLLIN)){
kq_ev_change(h, fd, EVFILT_READ, EV_DELETE, 0);
}
|
e5be1a06 |
goto error;
|
316130a7 |
}
|
a0553f4e |
}
|
741a9937 |
break;
|
0ba367ec |
#endif
|
9eda5956 |
#ifdef HAVE_DEVPOLL
case POLL_DEVPOLL:
pfd.fd=fd;
|
a0553f4e |
pfd.events=events;
|
9eda5956 |
pfd.revents=0;
again_devpoll:
if (write(h->dpoll_fd, &pfd, sizeof(pfd))==-1){
if (errno==EAGAIN) goto again_devpoll;
LOG(L_ERR, "ERROR: io_watch_add: /dev/poll write failed:"
"%s [%d]\n", strerror(errno), errno);
goto error;
}
break;
#endif
|
0ba367ec |
default:
LOG(L_CRIT, "BUG: io_watch_add: no support for poll method "
" %s (%d)\n", poll_method_str[h->poll_method],
h->poll_method);
goto error;
}
|
741a9937 |
h->fd_no++; /* "activate" changes, for epoll/kqueue/devpoll it
|
0ba367ec |
has only informative value */
|
6e154d5d |
#if defined(HAVE_SIGIO_RT)
|
3135b4bb |
if (check_io){
/* handle possible pre-existing events */
pf.fd=fd;
|
a0553f4e |
pf.events=events;
|
3135b4bb |
check_io_again:
|
a43212de |
n=0;
|
a9cdfc29 |
while(e->type && ((n=poll(&pf, 1, 0))>0) &&
|
a0553f4e |
(handle_io(e, pf.revents, idx)>0) &&
|
503bc281 |
(pf.revents & (e->events|POLLERR|POLLHUP)));
|
a0553f4e |
if (unlikely(e->type && (n==-1))){
|
3135b4bb |
if (errno==EINTR) goto check_io_again;
LOG(L_ERR, "ERROR: io_watch_add: check_io poll: %s [%d]\n",
strerror(errno), errno);
}
}
#endif
|
0ba367ec |
return 0;
error:
|
3135b4bb |
if (e) unhash_fd_map(e);
|
0ba367ec |
return -1;
#undef fd_array_setup
|
a9cdfc29 |
#undef set_fd_flags
|
0ba367ec |
}
|
741a9937 |
#define IO_FD_CLOSING 16
|
a9cdfc29 |
/* parameters: h - handler
|
741a9937 |
* fd - file descriptor
* index - index in the fd_array if known, -1 if not
* (if index==-1 fd_array will be searched for the
|
a9cdfc29 |
* corresponding fd* entry -- slower but unavoidable in
|
741a9937 |
* some cases). index is not used (no fd_array) for epoll,
* /dev/poll and kqueue
|
a9cdfc29 |
* flags - optimization flags, e.g. IO_FD_CLOSING, the fd was
|
741a9937 |
* or will shortly be closed, in some cases we can avoid
* extra remove operations (e.g.: epoll, kqueue, sigio)
|
0ba367ec |
* returns 0 if ok, -1 on error */
|
741a9937 |
inline static int io_watch_del(io_wait_h* h, int fd, int idx, int flags)
|
0ba367ec |
{
#define fix_fd_array \
do{\
|
a0553f4e |
if (unlikely(idx==-1)){ \
|
0ba367ec |
/* fix idx if -1 and needed */ \
for (idx=0; (idx<h->fd_no) && \
(h->fd_array[idx].fd!=fd); idx++); \
} \
|
a0553f4e |
if (likely(idx<h->fd_no)){ \
|
0ba367ec |
memmove(&h->fd_array[idx], &h->fd_array[idx+1], \
(h->fd_no-(idx+1))*sizeof(*(h->fd_array))); \
|
89deaf49 |
if ((idx<=h->crt_fd_array_idx) && (h->crt_fd_array_idx>=0)) \
h->crt_fd_array_idx--; \
|
0ba367ec |
} \
}while(0)
struct fd_map* e;
|
a0553f4e |
int events;
|
0ba367ec |
#ifdef HAVE_EPOLL
int n;
struct epoll_event ep_event;
#endif
|
9eda5956 |
#ifdef HAVE_DEVPOLL
struct pollfd pfd;
#endif
|
3135b4bb |
#ifdef HAVE_SIGIO_RT
int fd_flags;
#endif
|
0ba367ec |
|
a0553f4e |
if (unlikely((fd<0) || (fd>=h->max_fd_no))){
|
0ba367ec |
LOG(L_CRIT, "BUG: io_watch_del: invalid fd %d, not in [0, %d) \n",
fd, h->fd_no);
goto error;
}
|
3135b4bb |
DBG("DBG: io_watch_del (%p, %d, %d, 0x%x) fd_no=%d called\n",
h, fd, idx, flags, h->fd_no);
|
0ba367ec |
e=get_fd_map(h, fd);
/* more sanity checks */
|
a0553f4e |
if (unlikely(e==0)){
|
0ba367ec |
LOG(L_CRIT, "BUG: io_watch_del: no corresponding hash entry for %d\n",
fd);
goto error;
}
|
a0553f4e |
if (unlikely(e->type==0 /*F_NONE*/)){
|
0ba367ec |
LOG(L_ERR, "ERROR: io_watch_del: trying to delete already erased"
|
316130a7 |
" entry %d in the hash(%d, %d, %p) flags %x)\n",
fd, e->fd, e->type, e->data, flags);
|
0ba367ec |
goto error;
}
|
a0553f4e |
events=e->events;
|
0ba367ec |
switch(h->poll_method){
case POLL_POLL:
fix_fd_array;
break;
#ifdef HAVE_SELECT
case POLL_SELECT:
|
a0553f4e |
if (likely(events & POLLIN))
FD_CLR(fd, &h->master_rset);
if (unlikely(events & POLLOUT))
FD_CLR(fd, &h->master_wset);
if (unlikely(h->max_fd_select && (h->max_fd_select==fd)))
|
0ba367ec |
/* we don't know the prev. max, so we just decrement it */
|
a9cdfc29 |
h->max_fd_select--;
|
a0553f4e |
fix_fd_array;
|
0ba367ec |
break;
#endif
#ifdef HAVE_SIGIO_RT
case POLL_SIGIO_RT:
|
3135b4bb |
/* the O_ASYNC flag must be reset all the time, the fd
* can be changed only if O_ASYNC is reset (if not and
* the fd is a duplicate, you will get signals from the dup. fd
|
a31a6e8c |
* and not from the original, even if the dup. fd was closed
|
3135b4bb |
* and the signals re-set on the original) -- andrei
*/
/*if (!(flags & IO_FD_CLOSING)){*/
/* reset ASYNC */
|
a9cdfc29 |
fd_flags=fcntl(fd, F_GETFL);
if (unlikely(fd_flags==-1)){
LOG(L_ERR, "ERROR: io_watch_del: fnctl: GETFL failed:"
" %s [%d]\n", strerror(errno), errno);
goto error;
}
if (unlikely(fcntl(fd, F_SETFL, fd_flags&(~O_ASYNC))==-1)){
LOG(L_ERR, "ERROR: io_watch_del: fnctl: SETFL"
" failed: %s [%d]\n", strerror(errno), errno);
goto error;
}
|
2d8cd170 |
fix_fd_array; /* only on success */
|
0ba367ec |
break;
#endif
#ifdef HAVE_EPOLL
case POLL_EPOLL_LT:
case POLL_EPOLL_ET:
|
3135b4bb |
/* epoll doesn't seem to automatically remove sockets,
|
a0553f4e |
* if the socket is a duplicate/moved and the original
|
3135b4bb |
* is still open. The fd is removed from the epoll set
|
a9cdfc29 |
* only when the original (and all the copies?) is/are
|
3135b4bb |
* closed. This is probably a bug in epoll. --andrei */
#ifdef EPOLL_NO_CLOSE_BUG
|
741a9937 |
if (!(flags & IO_FD_CLOSING)){
|
3135b4bb |
#endif
|
a0553f4e |
again_epoll:
|
741a9937 |
n=epoll_ctl(h->epfd, EPOLL_CTL_DEL, fd, &ep_event);
|
a0553f4e |
if (unlikely(n==-1)){
if (errno==EAGAIN) goto again_epoll;
|
741a9937 |
LOG(L_ERR, "ERROR: io_watch_del: removing fd from epoll "
"list failed: %s [%d]\n", strerror(errno), errno);
goto error;
}
|
3135b4bb |
#ifdef EPOLL_NO_CLOSE_BUG
|
741a9937 |
}
|
3135b4bb |
#endif
|
741a9937 |
break;
#endif
#ifdef HAVE_KQUEUE
case POLL_KQUEUE:
if (!(flags & IO_FD_CLOSING)){
|
a0553f4e |
if (likely(events & POLLIN)){
if (unlikely(kq_ev_change(h, fd, EVFILT_READ,
EV_DELETE, 0) ==-1)){
/* try to delete the write filter anyway */
if (events & POLLOUT){
kq_ev_change(h, fd, EVFILT_WRITE, EV_DELETE, 0);
}
goto error;
}
}
if (unlikely(events & POLLOUT)){
if (unlikely(kq_ev_change(h, fd, EVFILT_WRITE,
EV_DELETE, 0) ==-1))
goto error;
}
|
0ba367ec |
}
break;
|
9eda5956 |
#endif
#ifdef HAVE_DEVPOLL
case POLL_DEVPOLL:
/* for /dev/poll the closed fds _must_ be removed
(they are not removed automatically on close()) */
pfd.fd=fd;
pfd.events=POLLREMOVE;
pfd.revents=0;
again_devpoll:
if (write(h->dpoll_fd, &pfd, sizeof(pfd))==-1){
if (errno==EINTR) goto again_devpoll;
LOG(L_ERR, "ERROR: io_watch_del: removing fd from "
|
a9cdfc29 |
"/dev/poll failed: %s [%d]\n",
|
9eda5956 |
strerror(errno), errno);
goto error;
}
break;
|
0ba367ec |
#endif
default:
LOG(L_CRIT, "BUG: io_watch_del: no support for poll method "
|
a9cdfc29 |
" %s (%d)\n", poll_method_str[h->poll_method],
|
0ba367ec |
h->poll_method);
goto error;
}
|
2d8cd170 |
unhash_fd_map(e); /* only on success */
|
0ba367ec |
h->fd_no--;
return 0;
error:
return -1;
#undef fix_fd_array
}
|
a9cdfc29 |
/* parameters: h - handler
|
a0553f4e |
* fd - file descriptor
* events - new events to watch for
* idx - index in the fd_array if known, -1 if not
* (if index==-1 fd_array will be searched for the
|
a9cdfc29 |
* corresponding fd* entry -- slower but unavoidable in
|
a0553f4e |
* some cases). index is not used (no fd_array) for epoll,
* /dev/poll and kqueue
* returns 0 if ok, -1 on error */
inline static int io_watch_chg(io_wait_h* h, int fd, short events, int idx )
{
#define fd_array_chg(ev) \
do{\
if (unlikely(idx==-1)){ \
/* fix idx if -1 and needed */ \
for (idx=0; (idx<h->fd_no) && \
(h->fd_array[idx].fd!=fd); idx++); \
} \
if (likely(idx<h->fd_no)){ \
h->fd_array[idx].events=(ev); \
} \
}while(0)
struct fd_map* e;
int add_events;
int del_events;
|
a43212de |
#ifdef HAVE_DEVPOLL
struct pollfd pfd;
#endif
|
a0553f4e |
#ifdef HAVE_EPOLL
int n;
struct epoll_event ep_event;
#endif
if (unlikely((fd<0) || (fd>=h->max_fd_no))){
LOG(L_CRIT, "BUG: io_watch_chg: invalid fd %d, not in [0, %d) \n",
fd, h->fd_no);
goto error;
}
if (unlikely((events&(POLLIN|POLLOUT))==0)){
LOG(L_CRIT, "BUG: io_watch_chg: invalid events: 0x%0x\n", events);
goto error;
}
DBG("DBG: io_watch_chg (%p, %d, 0x%x, 0x%x) fd_no=%d called\n",
h, fd, events, idx, h->fd_no);
e=get_fd_map(h, fd);
/* more sanity checks */
if (unlikely(e==0)){
LOG(L_CRIT, "BUG: io_watch_chg: no corresponding hash entry for %d\n",
fd);
goto error;
}
if (unlikely(e->type==0 /*F_NONE*/)){
LOG(L_ERR, "ERROR: io_watch_chg: trying to change an already erased"
" entry %d in the hash(%d, %d, %p) )\n",
fd, e->fd, e->type, e->data);
goto error;
}
add_events=events & ~e->events;
del_events=e->events & ~events;
switch(h->poll_method){
case POLL_POLL:
|
5d364ef9 |
#ifdef POLLRDHUP
|
b7ef1576 |
fd_array_chg(events |
|
2d8cd170 |
/* listen to POLLRDHUP by default (if POLLIN) */
|
b7ef1576 |
(((int)!(events & POLLIN) - 1) & POLLRDHUP)
|
2d8cd170 |
);
|
b7ef1576 |
#else /* POLLRDHUP */
fd_array_chg(events);
#endif /* POLLRDHUP */
|
a0553f4e |
break;
#ifdef HAVE_SELECT
case POLL_SELECT:
fd_array_chg(events);
if (unlikely(del_events & POLLIN))
FD_CLR(fd, &h->master_rset);
else if (unlikely(add_events & POLLIN))
FD_SET(fd, &h->master_rset);
if (likely(del_events & POLLOUT))
FD_CLR(fd, &h->master_wset);
else if (likely(add_events & POLLOUT))
FD_SET(fd, &h->master_wset);
break;
#endif
#ifdef HAVE_SIGIO_RT
case POLL_SIGIO_RT:
fd_array_chg(events);
|
a43212de |
/* no need for check_io, since SIGIO_RT listens by default for all
* the events */
|
a0553f4e |
break;
#endif
#ifdef HAVE_EPOLL
case POLL_EPOLL_LT:
|
5d364ef9 |
ep_event.events=
#ifdef POLLRDHUP
/* listen for EPOLLRDHUP too */
((EPOLLIN|EPOLLRDHUP) & ((int)!(events & POLLIN)-1) ) |
#else /* POLLRDHUP */
(EPOLLIN & ((int)!(events & POLLIN)-1) ) |
#endif /* POLLRDHUP */
(EPOLLOUT & ((int)!(events & POLLOUT)-1) );
|
a0553f4e |
ep_event.data.ptr=e;
again_epoll_lt:
n=epoll_ctl(h->epfd, EPOLL_CTL_MOD, fd, &ep_event);
if (unlikely(n==-1)){
if (errno==EAGAIN) goto again_epoll_lt;
LOG(L_ERR, "ERROR: io_watch_chg: modifying epoll events"
" failed: %s [%d]\n", strerror(errno), errno);
goto error;
}
break;
case POLL_EPOLL_ET:
|
5d364ef9 |
ep_event.events=
#ifdef POLLRDHUP
/* listen for EPOLLRDHUP too */
((EPOLLIN|EPOLLRDHUP) & ((int)!(events & POLLIN)-1) ) |
#else /* POLLRDHUP */
(EPOLLIN & ((int)!(events & POLLIN)-1) ) |
#endif /* POLLRDHUP */
(EPOLLOUT & ((int)!(events & POLLOUT)-1) ) |
EPOLLET;
|
a0553f4e |
ep_event.data.ptr=e;
again_epoll_et:
n=epoll_ctl(h->epfd, EPOLL_CTL_MOD, fd, &ep_event);
if (unlikely(n==-1)){
if (errno==EAGAIN) goto again_epoll_et;
LOG(L_ERR, "ERROR: io_watch_chg: modifying epoll events"
" failed: %s [%d]\n", strerror(errno), errno);
goto error;
}
break;
#endif
#ifdef HAVE_KQUEUE
case POLL_KQUEUE:
if (unlikely(del_events & POLLIN)){
if (unlikely(kq_ev_change(h, fd, EVFILT_READ,
EV_DELETE, 0) ==-1))
goto error;
}else if (unlikely(add_events & POLLIN)){
if (unlikely(kq_ev_change(h, fd, EVFILT_READ, EV_ADD, e) ==-1))
goto error;
}
if (likely(del_events & POLLOUT)){
if (unlikely(kq_ev_change(h, fd, EVFILT_WRITE,
EV_DELETE, 0) ==-1))
goto error;
}else if (likely(add_events & POLLOUT)){
if (unlikely(kq_ev_change(h, fd, EVFILT_WRITE, EV_ADD, e)==-1))
goto error;
}
break;
#endif
#ifdef HAVE_DEVPOLL
case POLL_DEVPOLL:
/* for /dev/poll the closed fds _must_ be removed
(they are not removed automatically on close()) */
pfd.fd=fd;
pfd.events=POLLREMOVE;
pfd.revents=0;
again_devpoll1:
if (unlikely(write(h->dpoll_fd, &pfd, sizeof(pfd))==-1)){
if (errno==EINTR) goto again_devpoll1;
LOG(L_ERR, "ERROR: io_watch_chg: removing fd from "
|
a9cdfc29 |
"/dev/poll failed: %s [%d]\n",
|
a0553f4e |
strerror(errno), errno);
goto error;
}
again_devpoll2:
pfd.events=events;
pfd.revents=0;
if (unlikely(write(h->dpoll_fd, &pfd, sizeof(pfd))==-1)){
if (errno==EINTR) goto again_devpoll2;
LOG(L_ERR, "ERROR: io_watch_chg: re-adding fd to "
|
a9cdfc29 |
"/dev/poll failed: %s [%d]\n",
|
a0553f4e |
strerror(errno), errno);
|
2d8cd170 |
/* error re-adding the fd => mark it as removed/unhash */
unhash_fd_map(e);
|
a0553f4e |
goto error;
}
break;
#endif
default:
LOG(L_CRIT, "BUG: io_watch_chg: no support for poll method "
|
a9cdfc29 |
" %s (%d)\n", poll_method_str[h->poll_method],
|
a0553f4e |
h->poll_method);
goto error;
}
|
2d8cd170 |
e->events=events; /* only on success */
|
a0553f4e |
return 0;
error:
return -1;
#undef fix_fd_array
}
|
89deaf49 |
|
a9cdfc29 |
/* io_wait_loop_x style function.
|
0ba367ec |
* wait for io using poll()
* params: h - io_wait handle
* t - timeout in s
* repeat - if !=0 handle_io will be called until it returns <=0
|
9eda5956 |
* returns: number of IO events handled on success (can be 0), -1 on error
|
0ba367ec |
*/
inline static int io_wait_loop_poll(io_wait_h* h, int t, int repeat)
{
int n, r;
int ret;
|
5702d448 |
struct fd_map* fm;
|
89deaf49 |
|
0ba367ec |
again:
ret=n=poll(h->fd_array, h->fd_no, t*1000);
if (n==-1){
if (errno==EINTR) goto again; /* signal, ignore it */
else{
LOG(L_ERR, "ERROR:io_wait_loop_poll: poll: %s [%d]\n",
strerror(errno), errno);
goto error;
}
}
for (r=0; (r<h->fd_no) && n; r++){
|
a0553f4e |
fm=get_fd_map(h, h->fd_array[r].fd);
if (h->fd_array[r].revents & (fm->events|POLLERR|POLLHUP)){
|
0ba367ec |
n--;
/* sanity checks */
|
a0553f4e |
if (unlikely((h->fd_array[r].fd >= h->max_fd_no)||
(h->fd_array[r].fd < 0))){
|
0ba367ec |
LOG(L_CRIT, "BUG: io_wait_loop_poll: bad fd %d "
"(no in the 0 - %d range)\n",
h->fd_array[r].fd, h->max_fd_no);
/* try to continue anyway */
h->fd_array[r].events=0; /* clear the events */
continue;
}
|
89deaf49 |
h->crt_fd_array_idx=r;
|
a0553f4e |
/* repeat handle_io if repeat, fd still watched (not deleted
* inside handle_io), handle_io returns that there's still
* IO and the fd is still watched for the triggering event */
|
a9cdfc29 |
while(fm->type &&
|
a0553f4e |
(handle_io(fm, h->fd_array[r].revents, r) > 0) &&
|
503bc281 |
repeat && ((fm->events|POLLERR|POLLHUP) &
h->fd_array[r].revents));
|
a9cdfc29 |
r=h->crt_fd_array_idx; /* can change due to io_watch_del(fd)
|
89deaf49 |
array shifting */
|
0ba367ec |
}
}
error:
return ret;
}
#ifdef HAVE_SELECT
/* wait for io using select */
inline static int io_wait_loop_select(io_wait_h* h, int t, int repeat)
{
|
a0553f4e |
fd_set sel_rset;
fd_set sel_wset;
|
0ba367ec |
int n, ret;
struct timeval timeout;
int r;
|
5702d448 |
struct fd_map* fm;
|
a0553f4e |
int revents;
|
0ba367ec |
again:
|
a0553f4e |
sel_rset=h->master_rset;
sel_wset=h->master_wset;
|
0ba367ec |
timeout.tv_sec=t;
timeout.tv_usec=0;
|
a0553f4e |
ret=n=select(h->max_fd_select+1, &sel_rset, &sel_wset, 0, &timeout);
|
0ba367ec |
if (n<0){
if (errno==EINTR) goto again; /* just a signal */
LOG(L_ERR, "ERROR: io_wait_loop_select: select: %s [%d]\n",
strerror(errno), errno);
n=0;
/* continue */
}
/* use poll fd array */
|
89deaf49 |
for(r=0; (r<h->fd_no) && n; r++){
|
a0553f4e |
revents=0;
if (likely(FD_ISSET(h->fd_array[r].fd, &sel_rset)))
revents|=POLLIN;
if (unlikely(FD_ISSET(h->fd_array[r].fd, &sel_wset)))
revents|=POLLOUT;
|
89deaf49 |
if (unlikely(revents)){
h->crt_fd_array_idx=r;
|
5702d448 |
fm=get_fd_map(h, h->fd_array[r].fd);
|
a9cdfc29 |
while(fm->type && (fm->events & revents) &&
|
a0553f4e |
(handle_io(fm, revents, r)>0) && repeat);
|
a9cdfc29 |
r=h->crt_fd_array_idx; /* can change due to io_watch_del(fd)
|
89deaf49 |
array shifting */
|
0ba367ec |
n--;
}
};
return ret;
}
#endif
#ifdef HAVE_EPOLL
inline static int io_wait_loop_epoll(io_wait_h* h, int t, int repeat)
{
int n, r;
|
5702d448 |
struct fd_map* fm;
|
a0553f4e |
int revents;
|
0ba367ec |
again:
n=epoll_wait(h->epfd, h->ep_array, h->fd_no, t*1000);
|
a0553f4e |
if (unlikely(n==-1)){
|
0ba367ec |
if (errno==EINTR) goto again; /* signal, ignore it */
else{
|
3135b4bb |
LOG(L_ERR, "ERROR:io_wait_loop_epoll: "
|
a9cdfc29 |
"epoll_wait(%d, %p, %d, %d): %s [%d]\n",
|
3135b4bb |
h->epfd, h->ep_array, h->fd_no, t*1000,
strerror(errno), errno);
|
0ba367ec |
goto error;
}
}
|
8d7f99dd |
#if 0
if (n>1){
for(r=0; r<n; r++){
LOG(L_ERR, "WARNING: ep_array[%d]= %x, %p\n",
r, h->ep_array[r].events, h->ep_array[r].data.ptr);
}
}
#endif
|
0ba367ec |
for (r=0; r<n; r++){
|
503bc281 |
revents= (POLLIN & (!(h->ep_array[r].events & (EPOLLIN|EPOLLPRI))
-1)) |
|
a0553f4e |
(POLLOUT & (!(h->ep_array[r].events & EPOLLOUT)-1)) |
(POLLERR & (!(h->ep_array[r].events & EPOLLERR)-1)) |
|
5d364ef9 |
(POLLHUP & (!(h->ep_array[r].events & EPOLLHUP)-1))
#ifdef POLLRDHUP
| (POLLRDHUP & (!(h->ep_array[r].events & EPOLLRDHUP)-1))
#endif
;
|
a0553f4e |
if (likely(revents)){
|
5702d448 |
fm=(struct fd_map*)h->ep_array[r].data.ptr;
|
a9cdfc29 |
while(fm->type && ((fm->events|POLLERR|POLLHUP) & revents) &&
|
a0553f4e |
(handle_io(fm, revents, -1)>0) && repeat);
|
8d7f99dd |
}else{
LOG(L_ERR, "ERROR:io_wait_loop_epoll: unexpected event %x"
" on %d/%d, data=%p\n", h->ep_array[r].events,
r+1, n, h->ep_array[r].data.ptr);
}
|
0ba367ec |
}
error:
return n;
}
#endif
|
741a9937 |
#ifdef HAVE_KQUEUE
inline static int io_wait_loop_kqueue(io_wait_h* h, int t, int repeat)
{
int n, r;
struct timespec tspec;
|
5702d448 |
struct fd_map* fm;
|
b0bd3201 |
int orig_changes;
int apply_changes;
|
8880cb34 |
int revents;
|
741a9937 |
tspec.tv_sec=t;
tspec.tv_nsec=0;
|
b0bd3201 |
orig_changes=h->kq_nchanges;
apply_changes=orig_changes;
do {
|
741a9937 |
again:
|
b0bd3201 |
n=kevent(h->kq_fd, h->kq_changes, apply_changes, h->kq_array,
|
99682621 |
h->kq_array_size, &tspec);
|
a0553f4e |
if (unlikely(n==-1)){
|
8966d0a1 |
if (unlikely(errno==EINTR)) goto again; /* signal, ignore it */
else {
/* for a detailed explanation of what follows see below
the EV_ERROR case */
if (unlikely(!(errno==EBADF || errno==ENOENT)))
BUG("io_wait_loop_kqueue: kevent: unexpected error"
" %s [%d]\n", strerror(errno), errno);
|
b0bd3201 |
/* some of the FDs in kq_changes are bad (already closed)
and there is not enough space in kq_array to return all
of them back */
|
99682621 |
apply_changes = h->kq_array_size;
|
b0bd3201 |
goto again;
|
741a9937 |
}
}
|
b0bd3201 |
/* remove applied changes */
h->kq_nchanges -= apply_changes;
if (unlikely(apply_changes < orig_changes)) {
orig_changes -= apply_changes;
memmove(&h->kq_changes[0], &h->kq_changes[apply_changes],
sizeof(h->kq_changes[0])*h->kq_nchanges);
|
a9cdfc29 |
apply_changes = (orig_changes < h->kq_array_size) ? orig_changes :
h->kq_array_size;
|
b0bd3201 |
} else {
orig_changes = 0;
apply_changes = 0;
}
|
741a9937 |
for (r=0; r<n; r++){
|
2f2e593b |
#ifdef EXTRA_DEBUG
DBG("DBG: kqueue: event %d/%d: fd=%d, udata=%lx, flags=0x%x\n",
r, n, h->kq_array[r].ident, (long)h->kq_array[r].udata,
h->kq_array[r].flags);
#endif
|
8966d0a1 |
if (unlikely((h->kq_array[r].flags & EV_ERROR) ||
h->kq_array[r].udata == 0)){
|
05c080a5 |
/* error in changes: we ignore it if it has to do with a
bad fd or update==0. It can be caused by trying to remove an
already closed fd: race between adding something to the
|
8966d0a1 |
changes array, close() and applying the changes (EBADF).
E.g. for ser tcp: tcp_main sends a fd to child for reading
|
05c080a5 |
=> deletes it from the watched fds => the changes array
will contain an EV_DELETE for it. Before the changes
are applied (they are at the end of the main io_wait loop,
after all the fd events were processed), a CON_ERR sent
to tcp_main by a sender (send fail) is processed and causes
the fd to be closed. When the changes are applied =>
error for the EV_DELETE attempt of a closed fd.
|
8966d0a1 |
Something similar can happen when a fd is scheduled
for removal, is close()'ed before being removed and
re-opened(a new sock. get the same fd). When the
watched fd changes will be applied the fd will be valid
(so no EBADF), but it's not already watch => ENOENT.
We report a BUG for the other errors (there's nothing
|
a9cdfc29 |
constructive we can do if we get an error we don't know
|
8966d0a1 |
how to handle), but apart from that we ignore it in the
idea that it is better apply the rest of the changes,
rather then dropping all of them.
|
05c080a5 |
*/
/*
example EV_ERROR for trying to delete a read watched fd,
that was already closed:
{
ident = 63, [fd]
filter = -1, [EVFILT_READ]
flags = 16384, [EV_ERROR]
fflags = 0,
data = 9, [errno = EBADF]
udata = 0x0
}
*/
|
8966d0a1 |
if (h->kq_array[r].data != EBADF &&
h->kq_array[r].data != ENOENT)
BUG("io_wait_loop_kqueue: kevent unexpected error on "
"fd %ld udata %lx: %s [%ld]\n",
(long)h->kq_array[r].ident,
(long)h->kq_array[r].udata,
|
741a9937 |
strerror(h->kq_array[r].data),
|
2f2e593b |
(long)h->kq_array[r].data);
|
05c080a5 |
}else{
|
5702d448 |
fm=(struct fd_map*)h->kq_array[r].udata;
|
a0553f4e |
if (likely(h->kq_array[r].filter==EVFILT_READ)){
|
05c080a5 |
revents=POLLIN |
(((int)!(h->kq_array[r].flags & EV_EOF)-1)&POLLHUP) |
|
8966d0a1 |
(((int)!((h->kq_array[r].flags & EV_EOF) &&
h->kq_array[r].fflags != 0) - 1)&POLLERR);
|
a9cdfc29 |
while(fm->type && (fm->events & revents) &&
|
a0553f4e |
(handle_io(fm, revents, -1)>0) && repeat);
}else if (h->kq_array[r].filter==EVFILT_WRITE){
|
05c080a5 |
revents=POLLOUT |
(((int)!(h->kq_array[r].flags & EV_EOF)-1)&POLLHUP) |
|
8966d0a1 |
(((int)!((h->kq_array[r].flags & EV_EOF) &&
h->kq_array[r].fflags != 0) - 1)&POLLERR);
|
a9cdfc29 |
while(fm->type && (fm->events & revents) &&
|
a0553f4e |
(handle_io(fm, revents, -1)>0) && repeat);
|
8966d0a1 |
}else{
BUG("io_wait_loop_kqueue: unknown filter: kqueue: event "
"%d/%d: fd=%d, filter=%d, flags=0x%x, fflags=0x%x,"
" data=%lx, udata=%lx\n",
|
c7694799 |
r, n, (int)h->kq_array[r].ident, (int)h->kq_array[r].filter,
|
a9cdfc29 |
h->kq_array[r].flags, h->kq_array[r].fflags,
|
c7694799 |
(unsigned long)h->kq_array[r].data,
(unsigned long)h->kq_array[r].udata);
|
a0553f4e |
}
|
05c080a5 |
}
|
741a9937 |
}
|
b0bd3201 |
} while(unlikely(orig_changes));
|
741a9937 |
return n;
}
#endif
|
0ba367ec |
#ifdef HAVE_SIGIO_RT
/* sigio rt version has no repeat (it doesn't make sense)*/
inline static int io_wait_loop_sigio_rt(io_wait_h* h, int t)
{
int n;
int ret;
struct timespec ts;
siginfo_t siginfo;
int sigio_band;
int sigio_fd;
struct fd_map* fm;
|
a0553f4e |
int revents;
|
018226b0 |
#ifdef SIGINFO64_WORKARROUND
int* pi;
#endif
|
0ba367ec |
ret=1; /* 1 event per call normally */
ts.tv_sec=t;
ts.tv_nsec=0;
|
a0553f4e |
if (unlikely(!sigismember(&h->sset, h->signo) ||
!sigismember(&h->sset, SIGIO))) {
|
0ba367ec |
LOG(L_CRIT, "BUG: io_wait_loop_sigio_rt: the signal mask"
" is not properly set!\n");
goto error;
}
again:
n=sigtimedwait(&h->sset, &siginfo, &ts);
|
a0553f4e |
if (unlikely(n==-1)){
|
0ba367ec |
if (errno==EINTR) goto again; /* some other signal, ignore it */
else if (errno==EAGAIN){ /* timeout */
ret=0;
goto end;
}else{
LOG(L_ERR, "ERROR: io_wait_loop_sigio_rt: sigtimed_wait"
" %s [%d]\n", strerror(errno), errno);
goto error;
}
}
|
a0553f4e |
if (likely(n!=SIGIO)){
|
0ba367ec |
#ifdef SIGINFO64_WORKARROUND
/* on linux siginfo.si_band is defined as long in userspace
|
19db15d7 |
* and as int in kernel (< 2.6.5) => on 64 bits things will break!
|
0ba367ec |
* (si_band will include si_fd, and si_fd will contain
|
19db15d7 |
* garbage).
|
0ba367ec |
* see /usr/src/linux/include/asm-generic/siginfo.h and
* /usr/include/bits/siginfo.h
|
19db15d7 |
* On newer kernels this is fixed (si_band is long in the kernel too).
|
0ba367ec |
* -- andrei */
|
19db15d7 |
if ((_os_ver<0x020605) && (sizeof(siginfo.si_band)>sizeof(int))){
|
018226b0 |
pi=(int*)(void*)&siginfo.si_band; /* avoid type punning warnings */
sigio_band=*pi;
sigio_fd=*(pi+1);
|
0ba367ec |
}else
#endif
{
sigio_band=siginfo.si_band;
sigio_fd=siginfo.si_fd;
}
|
a0553f4e |
if (unlikely(siginfo.si_code==SI_SIGIO)){
|
0ba367ec |
/* old style, we don't know the event (linux 2.2.?) */
LOG(L_WARN, "WARNING: io_wait_loop_sigio_rt: old style sigio"
" interface\n");
fm=get_fd_map(h, sigio_fd);
/* we can have queued signals generated by fds not watched
* any more, or by fds in transition, to a child => ignore them*/
if (fm->type)
|
a0553f4e |
handle_io(fm, POLLIN|POLLOUT, -1);
|
0ba367ec |
}else{
|
503bc281 |
/* si_code contains the SIGPOLL reason: POLL_IN, POLL_OUT,
* POLL_MSG, POLL_ERR, POLL_PRI or POLL_HUP
* and si_band the translated poll event bitmap:
* POLLIN|POLLRDNORM (=POLL_IN),
* POLLOUT|POLLWRNORM|POLLWRBAND (=POLL_OUT),
* POLLIN|POLLRDNORM|POLLMSG (=POLL_MSG),
* POLLERR (=POLL_ERR),
* POLLPRI|POLLRDBAND (=POLL_PRI),
|
a9cdfc29 |
* POLLHUP|POLLERR (=POLL_HUP)
|
503bc281 |
* [linux 2.6.22 fs/fcntl.c:447]
*/
|
0ba367ec |
#ifdef EXTRA_DEBUG
DBG("io_wait_loop_sigio_rt: siginfo: signal=%d (%d),"
" si_code=%d, si_band=0x%x,"
" si_fd=%d\n",
|
a9cdfc29 |
siginfo.si_signo, n, siginfo.si_code,
|
0ba367ec |
(unsigned)sigio_band,
sigio_fd);
#endif
|
8d7f99dd |
/* on some errors (e.g. when receving TCP RST), sigio_band will
|
503bc281 |
* be set to 0x08 (POLLERR) or 0x18 (POLLERR|POLLHUP - on stream
* unix socket close) , so better catch all events --andrei */
if (likely(sigio_band)){
|
0ba367ec |
fm=get_fd_map(h, sigio_fd);
|
503bc281 |
revents=sigio_band;
/* fix revents==POLLPRI case */
revents |= (!(revents & POLLPRI)-1) & POLLIN;
|
0ba367ec |
/* we can have queued signals generated by fds not watched
|
a9cdfc29 |
* any more, or by fds in transition, to a child
|
0ba367ec |
* => ignore them */
|
503bc281 |
if (fm->type && ((fm->events|POLLERR|POLLHUP) & revents))
|
a0553f4e |
handle_io(fm, revents, -1);
|
8d7f99dd |
else
|
503bc281 |
DBG("WARNING: io_wait_loop_sigio_rt: ignoring event"
" %x on fd %d, watching for %x, si_code=%x "
"(fm->type=%d, fm->fd=%d, fm->data=%p)\n",
sigio_band, sigio_fd, fm->events, siginfo.si_code,
fm->type, fm->fd, fm->data);
|
8d7f99dd |
}else{
LOG(L_ERR, "ERROR: io_wait_loop_sigio_rt: unexpected event"
" on fd %d: %x\n", sigio_fd, sigio_band);
|
0ba367ec |
}
}
}else{
|
a9cdfc29 |
/* signal queue overflow
|
0ba367ec |
* TODO: increase signal queue size: 2.4x /proc/.., 2.6x -rlimits */
LOG(L_WARN, "WARNING: io_wait_loop_sigio_rt: signal queue overflowed"
"- falling back to poll\n");
/* clear real-time signal queue
* both SIG_IGN and SIG_DFL are needed , it doesn't work
* only with SIG_DFL */
if (signal(h->signo, SIG_IGN)==SIG_ERR){
LOG(L_CRIT, "BUG: do_poll: couldn't reset signal to IGN\n");
}
if (signal(h->signo, SIG_DFL)==SIG_ERR){
LOG(L_CRIT, "BUG: do_poll: couldn't reset signal to DFL\n");
}
/* falling back to normal poll */
ret=io_wait_loop_poll(h, -1, 1);
}
end:
return ret;
error:
return -1;
}
#endif
|
9eda5956 |
#ifdef HAVE_DEVPOLL
inline static int io_wait_loop_devpoll(io_wait_h* h, int t, int repeat)
{
int n, r;
int ret;
struct dvpoll dpoll;
|
5702d448 |
struct fd_map* fm;
|
9eda5956 |
dpoll.dp_timeout=t*1000;
dpoll.dp_nfds=h->fd_no;
dpoll.dp_fds=h->fd_array;
again:
ret=n=ioctl(h->dpoll_fd, DP_POLL, &dpoll);
|
a0553f4e |
if (unlikely(n==-1)){
|
9eda5956 |
if (errno==EINTR) goto again; /* signal, ignore it */
else{
LOG(L_ERR, "ERROR:io_wait_loop_devpoll: ioctl: %s [%d]\n",
strerror(errno), errno);
goto error;
}
}
for (r=0; r< n; r++){
if (h->fd_array[r].revents & (POLLNVAL|POLLERR)){
LOG(L_ERR, "ERROR: io_wait_loop_devpoll: pollinval returned"
" for fd %d, revents=%x\n",
h->fd_array[r].fd, h->fd_array[r].revents);
}
/* POLLIN|POLLHUP just go through */
|
5702d448 |
fm=get_fd_map(h, h->fd_array[r].fd);
|
a0553f4e |
while(fm->type && (fm->events & h->fd_array[r].revents) &&
(handle_io(fm, h->fd_array[r].revents, r) > 0) && repeat);
|
9eda5956 |
}
error:
return ret;
}
#endif
|
0ba367ec |
/* init */
/* initializes the static vars/arrays
* params: h - pointer to the io_wait_h that will be initialized
* max_fd - maximum allowed fd number
* poll_m - poll method (0 for automatic best fit)
*/
int init_io_wait(io_wait_h* h, int max_fd, enum poll_types poll_method);
/* destroys everything init_io_wait allocated */
void destroy_io_wait(io_wait_h* h);
#endif
|