lib/mtio-select.c

   1 /*
   2     ashd - A Sane HTTP Daemon
   3     Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>
   4
   5     This program is free software: you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation, either version 3 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #include <stdlib.h>
  20 #include <string.h>
  21 #include <time.h>
  22 #include <unistd.h>
  23 #include <errno.h>
  24 #include <sys/select.h>
  25
  26 #ifdef HAVE_CONFIG_H
  27 #include <config.h>
  28 #endif
  29 #include <log.h>
  30 #include <utils.h>
  31 #include <mt.h>
  32 #include <mtio.h>
  33
  34 static struct blocker *blockers;
  35 static int exitstatus;
  36
  37 struct blocker {
  38     struct blocker *n, *p;
  39     struct iterator *it;
  40     int fd;
  41     int ev, rev, id;
  42     time_t to;
  43     struct muth *th;
  44 };
  45
  46 struct iterator {
  47     struct blocker *bl;
  48 };
  49
  50 static void addblock(struct blocker *bl)
  51 {
  52     bl->n = blockers;
  53     if(blockers)
  54         blockers->p = bl;
  55     blockers = bl;
  56 }
  57
  58 static void remblock(struct blocker *bl)
  59 {
  60     if(bl->n)
  61         bl->n->p = bl->p;
  62     if(bl->p)
  63         bl->p->n = bl->n;
  64     if(bl == blockers)
  65         blockers = bl->n;
  66     if(bl->it) {
  67         if((bl->it->bl = bl->n) != NULL)
  68             bl->it->bl->it = bl->it;
  69         bl->it = NULL;
  70     }
  71 }
  72
  73 struct selected mblock(time_t to, int n, struct selected *spec)
  74 {
  75     int i, id;
  76     struct blocker bls[n];
  77
  78     to = (to > 0)?(time(NULL) + to):0;
  79     for(i = 0; i < n; i++) {
  80         bls[i] = (struct blocker){
  81             .fd = spec[i].fd,
  82             .ev = spec[i].ev,
  83             .id = i,
  84             .to = to,
  85             .th = current,
  86         };
  87         addblock(&bls[i]);
  88     }
  89     id = yield();
  90     for(i = 0; i < n; i++)
  91         remblock(&bls[i]);
  92     if(id < 0)
  93         return((struct selected){.fd = -1, .ev = -1});
  94     return((struct selected){.fd = bls[id].fd, .ev = bls[id].rev});
  95 }
  96
  97 int block(int fd, int ev, time_t to)
  98 {
  99     struct blocker bl;
 100     int rv;
 101
 102     if(fd >= FD_SETSIZE) {
 103         flog(LOG_ERR, "tried to use more file descriptors than select() can handle: fd %i", fd);
 104         errno = EMFILE;
 105         return(-1);
 106     }
 107     bl = (struct blocker) {
 108         .fd = fd,
 109         .ev = ev,
 110         .id = -1,
 111         .to = (to > 0)?(time(NULL) + to):0,
 112         .th = current,
 113     };
 114     addblock(&bl);
 115     rv = yield();
 116     remblock(&bl);
 117     return(rv);
 118 }
 119
 120 int ioloop(void)
 121 {
 122     int ret;
 123     fd_set rfds, wfds, efds;
 124     struct blocker *bl;
 125     struct iterator it;
 126     struct timeval toval;
 127     time_t now, timeout;
 128     int maxfd;
 129     int ev;
 130
 131     exitstatus = 0;
 132     while(blockers != NULL) {
 133         FD_ZERO(&rfds);
 134         FD_ZERO(&wfds);
 135         FD_ZERO(&efds);
 136         maxfd = 0;
 137         now = time(NULL);
 138         timeout = 0;
 139         for(bl = blockers; bl; bl = bl->n) {
 140             if(bl->ev & EV_READ)
 141                 FD_SET(bl->fd, &rfds);
 142             if(bl->ev & EV_WRITE)
 143                 FD_SET(bl->fd, &wfds);
 144             FD_SET(bl->fd, &efds);
 145             if(bl->fd > maxfd)
 146                 maxfd = bl->fd;
 147             if((bl->to != 0) && ((timeout == 0) || (timeout > bl->to)))
 148                 timeout = bl->to;
 149         }
 150         if(exitstatus)
 151             return(exitstatus);
 152         toval.tv_sec = timeout - now;
 153         toval.tv_usec = 0;
 154         ret = select(maxfd + 1, &rfds, &wfds, &efds, timeout?(&toval):NULL);
 155         if(ret < 0) {
 156             if(errno != EINTR) {
 157                 flog(LOG_CRIT, "ioloop: select errored out: %s", strerror(errno));
 158                 /* To avoid CPU hogging in case it's bad, which it
 159                  * probably is. */
 160                 sleep(1);
 161             }
 162         } else {
 163             now = time(NULL);
 164             for(bl = it.bl = blockers; bl; bl = it.bl) {
 165                 if((it.bl = bl->n) != NULL)
 166                     it.bl->it = &it;
 167                 ev = 0;
 168                 if(FD_ISSET(bl->fd, &rfds))
 169                     ev |= EV_READ;
 170                 if(FD_ISSET(bl->fd, &wfds))
 171                     ev |= EV_WRITE;
 172                 if(FD_ISSET(bl->fd, &efds))
 173                     ev = -1;
 174                 if((ev < 0) || (ev & bl->ev)) {
 175                     if(bl->id < 0) {
 176                         resume(bl->th, ev);
 177                     } else {
 178                         bl->rev = ev;
 179                         resume(bl->th, bl->id);
 180                     }
 181                 } else if((bl->to != 0) && (bl->to <= now)) {
 182                     if(bl->id < 0) {
 183                         resume(bl->th, 0);
 184                     } else {
 185                         bl->rev = 0;
 186                         resume(bl->th, bl->id);
 187                     }
 188                 }
 189                 if(it.bl)
 190                     it.bl->it = NULL;
 191             }
 192         }
 193     }
 194     return(0);
 195 }
 196
 197 void exitioloop(int status)
 198 {
 199     exitstatus = status;
 200 }