diff options
Diffstat (limited to 'src/s_bsd_devpoll.c')
| -rw-r--r-- | src/s_bsd_devpoll.c | 186 |
1 files changed, 186 insertions, 0 deletions
diff --git a/src/s_bsd_devpoll.c b/src/s_bsd_devpoll.c new file mode 100644 index 0000000..49d4eee --- /dev/null +++ b/src/s_bsd_devpoll.c @@ -0,0 +1,186 @@ +/* + * ircd-hybrid: an advanced Internet Relay Chat Daemon(ircd). + * s_bsd_devpoll.c: /dev/poll compatible network routines. + * + * Originally by Adrian Chadd <adrian@creative.net.au> + * Copyright (C) 2002 Hybrid Development Team + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 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 General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 + * USA + * + * $Id$ + */ + +#include "stdinc.h" +#if USE_IOPOLL_MECHANISM == __IOPOLL_MECHANISM_DEVPOLL +#include <sys/ioctl.h> +/* HPUX uses devpoll.h and not sys/devpoll.h */ +#ifdef HAVE_DEVPOLL_H +# include <devpoll.h> +#else +# ifdef HAVE_SYS_DEVPOLL_H +# include <sys/devpoll.h> +# else +# error "No devpoll.h found! Try ./configuring and letting the script choose for you." +# endif +#endif +#include "fdlist.h" +#include "ircd.h" +#include "s_bsd.h" +#include "log.h" + +static fde_t dpfd; + +/* + * init_netio + * + * This is a needed exported function which will be called to initialise + * the network loop code. + */ +void +init_netio(void) +{ + int fd; + + if ((fd = open("/dev/poll", O_RDWR)) < 0) + { + ilog(LOG_TYPE_IRCD, "init_netio: Couldn't open /dev/poll - %d: %s", + errno, strerror(errno)); + exit(115); /* Whee! */ + } + + fd_open(&dpfd, fd, 0, "/dev/poll file descriptor"); +} + +/* + * Write an update to the devpoll filter. + * See, we end up having to do a seperate (?) remove before we do an + * add of a new polltype, so we have to have this function seperate from + * the others. + */ +static void +devpoll_write_update(int fd, int events) +{ + struct pollfd pfd; + + /* Build the pollfd entry */ + pfd.revents = 0; + pfd.fd = fd; + pfd.events = events; + + /* Write the thing to our poll fd */ + if (write(dpfd.fd, &pfd, sizeof(pfd)) != sizeof(pfd)) + ilog(LOG_TYPE_IRCD, "devpoll_write_update: dpfd write failed %d: %s", + errno, strerror(errno)); +} + +/* + * comm_setselect + * + * This is a needed exported function which will be called to register + * and deregister interest in a pending IO state for a given FD. + */ +void +comm_setselect(fde_t *F, unsigned int type, PF *handler, + void *client_data, time_t timeout) +{ + int new_events; + + if ((type & COMM_SELECT_READ)) + { + F->read_handler = handler; + F->read_data = client_data; + } + + if ((type & COMM_SELECT_WRITE)) + { + F->write_handler = handler; + F->write_data = client_data; + } + + new_events = (F->read_handler ? POLLIN : 0) | + (F->write_handler ? POLLOUT : 0); + + if (timeout != 0) + F->timeout = CurrentTime + (timeout / 1000); + + if (new_events != F->evcache) + { + devpoll_write_update(F->fd, POLLREMOVE); + if ((F->evcache = new_events)) + devpoll_write_update(F->fd, new_events); + } +} + +/* + * comm_select + * + * Called to do the new-style IO, courtesy of squid (like most of this + * new IO code). This routine handles the stuff we've hidden in + * comm_setselect and fd_table[] and calls callbacks for IO ready + * events. + */ +void +comm_select(void) +{ + int num, i; + struct pollfd pollfds[128]; + struct dvpoll dopoll; + PF *hdl; + fde_t *F; + + dopoll.dp_timeout = SELECT_DELAY; + dopoll.dp_nfds = 128; + dopoll.dp_fds = &pollfds[0]; + num = ioctl(dpfd.fd, DP_POLL, &dopoll); + + set_time(); + + if (num < 0) + { +#ifdef HAVE_USLEEP + usleep(50000); /* avoid 99% CPU in comm_select */ +#endif + return; + } + + for (i = 0; i < num; i++) + { + F = lookup_fd(dopoll.dp_fds[i].fd); + if (F == NULL || !F->flags.open) + continue; + + if ((dopoll.dp_fds[i].revents & POLLIN)) + if ((hdl = F->read_handler) != NULL) + { + F->read_handler = NULL; + hdl(F, F->read_data); + if (!F->flags.open) + continue; + } + + if ((dopoll.dp_fds[i].revents & POLLOUT)) + if ((hdl = F->write_handler) != NULL) + { + F->write_handler = NULL; + hdl(F, F->write_data); + if (!F->flags.open) + continue; + } + + comm_setselect(F, 0, NULL, NULL, 0); + } +} +#endif |