X-Git-Url: http://git.dolda2000.com/gitweb/?a=blobdiff_plain;f=lib%2Fmtio.c;fp=lib%2Fmtio.c;h=2a463690be8e7c8d67c195f85b535c32e39e7173;hb=83723896cdbe2fb064748e45611e9b9c829c1d72;hp=0000000000000000000000000000000000000000;hpb=f2b26d44416abd494ce430659cfbbf254d9c3c55;p=ashd.git

diff --git a/lib/mtio.c b/lib/mtio.c
new file mode 100644
index 0000000..2a46369
--- /dev/null
+++ b/lib/mtio.c
@@ -0,0 +1,124 @@
+/*
+    ashd - A Sane HTTP Daemon
+    Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>
+
+    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 3 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, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <time.h>
+#include <errno.h>
+#include <sys/select.h>
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+#include <log.h>
+#include <utils.h>
+#include <mt.h>
+#include <mtio.h>
+
+struct blocker {
+    struct blocker *n, *p;
+    int fd;
+    int ev;
+    time_t to;
+    struct muth *th;
+};
+
+static struct blocker *blockers;
+
+int block(int fd, int ev, time_t to)
+{
+    struct blocker *bl;
+    int rv;
+    
+    omalloc(bl);
+    bl->fd = fd;
+    bl->ev = ev;
+    if(to > 0)
+	bl->to = time(NULL) + to;
+    bl->th = current;
+    bl->n = blockers;
+    if(blockers)
+	blockers->p = bl;
+    blockers = bl;
+    rv = yield();
+    if(bl->n)
+	bl->n->p = bl->p;
+    if(bl->p)
+	bl->p->n = bl->n;
+    if(bl == blockers)
+	blockers = bl->n;
+    return(rv);
+}
+
+void ioloop(void)
+{
+    int ret;
+    fd_set rfds, wfds, efds;
+    struct blocker *bl, *nbl;
+    struct timeval toval;
+    time_t now, timeout;
+    int maxfd;
+    int ev;
+    
+    while(blockers != NULL) {
+	FD_ZERO(&rfds);
+	FD_ZERO(&wfds);
+	FD_ZERO(&efds);
+	maxfd = 0;
+	now = time(NULL);
+	timeout = 0;
+	for(bl = blockers; bl; bl = bl->n) {
+	    if(bl->ev & EV_READ)
+		FD_SET(bl->fd, &rfds);
+	    if(bl->ev & EV_WRITE)
+		FD_SET(bl->fd, &wfds);
+	    FD_SET(bl->fd, &efds);
+	    if(bl->fd > maxfd)
+		maxfd = bl->fd;
+	    if((bl->to != 0) && ((timeout == 0) || (timeout > bl->to)))
+		timeout = bl->to;
+	}
+	toval.tv_sec = timeout - now;
+	toval.tv_usec = 0;
+	ret = select(maxfd + 1, &rfds, &wfds, &efds, timeout?(&toval):NULL);
+	if(ret < 0) {
+	    if(errno != EINTR) {
+		flog(LOG_CRIT, "ioloop: select errored out: %s", strerror(errno));
+		/* To avoid CPU hogging in case it's bad, which it
+		 * probably is. */
+		sleep(1);
+	    }
+	}
+	now = time(NULL);
+	for(bl = blockers; bl; bl = nbl) {
+	    nbl = bl->n;
+	    ev = 0;
+	    if(FD_ISSET(bl->fd, &rfds))
+		ev |= EV_READ;
+	    if(FD_ISSET(bl->fd, &wfds))
+		ev |= EV_WRITE;
+	    if(FD_ISSET(bl->fd, &efds))
+		ev = -1;
+	    if(ev != 0)
+		resume(bl->th, ev);
+	    else if((bl->to != 0) && (bl->to <= now))
+		resume(bl->th, 0);
+	}
+    }
+}