X-Git-Url: http://www.dolda2000.com/gitweb/?a=blobdiff_plain;f=src%2Fhtparser.c;h=c78f064c33f7775e1e7882586a510dc76876830b;hb=a0327573bb6032e02e988a0d0aed709b4be7a422;hp=3a788ac899bee541cb997ba729cf039955c2cc2e;hpb=f4cdf91946475f428f3861275ce9d07f2da73a93;p=ashd.git diff --git a/src/htparser.c b/src/htparser.c index 3a788ac..c78f064 100644 --- a/src/htparser.c +++ b/src/htparser.c @@ -23,7 +23,9 @@ #include #include #include +#include #include +#include #ifdef HAVE_CONFIG_H #include @@ -31,6 +33,8 @@ #include #include #include +#include +#include #define EV_READ 1 #define EV_WRITE 2 @@ -39,12 +43,14 @@ struct blocker { struct blocker *n, *p; int fd; int ev; + time_t to; struct muth *th; }; static struct blocker *blockers; +int plex; -static int block(int fd, int ev) +static int block(int fd, int ev, time_t to) { struct blocker *bl; int rv; @@ -52,6 +58,8 @@ static int block(int fd, int ev) omalloc(bl); bl->fd = fd; bl->ev = ev; + if(to > 0) + bl->to = time(NULL) + to; bl->th = current; bl->n = blockers; if(blockers) @@ -115,6 +123,331 @@ static int listensock6(int port) return(fd); } +static size_t readhead(int fd, struct charbuf *buf) +{ + int nl; + size_t off; + + int get1(void) + { + int ret; + + while(!(off < buf->d)) { + sizebuf(*buf, buf->d + 1024); + ret = recv(fd, buf->b + buf->d, buf->s - buf->d, MSG_DONTWAIT); + if(ret <= 0) { + if((ret < 0) && (errno == EAGAIN)) { + if(block(fd, EV_READ, 60) <= 0) + return(-1); + continue; + } + return(-1); + } + buf->d += ret; + } + return(buf->b[off++]); + } + + nl = 0; + off = 0; + while(1) { + switch(get1()) { + case '\n': + if(nl) + return(off); + nl = 1; + break; + case '\r': + break; + case -1: + return(-1); + default: + nl = 0; + break; + } + } +} + +#define SKIPNL(ptr) ({ \ + int __buf__; \ + if(*(ptr) == '\r') \ + *((ptr)++) = 0; \ + if(*(ptr) != '\n') { \ + __buf__ = 0; \ + } else { \ + *((ptr)++) = 0; \ + __buf__ = 1; \ + } \ + __buf__;}) +static struct hthead *parserawreq(char *buf) +{ + char *p, *p2, *nl; + char *method, *url, *ver; + struct hthead *req; + + if((nl = strchr(buf, '\n')) == NULL) + return(NULL); + if(((p = strchr(buf, ' ')) == NULL) || (p > nl)) + return(NULL); + method = buf; + *(p++) = 0; + if(((p2 = strchr(p, ' ')) == NULL) || (p2 > nl)) + return(NULL); + url = p; + p = p2; + *(p++) = 0; + if(strncmp(p, "HTTP/", 5)) + return(NULL); + ver = (p += 5); + for(; ((*p >= '0') && (*p <= '9')) || (*p == '.'); p++); + if(!SKIPNL(p)) + return(NULL); + + req = mkreq(method, url, ver); + while(1) { + if(SKIPNL(p)) { + if(*p) + goto fail; + break; + } + if((nl = strchr(p, '\n')) == NULL) + goto fail; + if(((p2 = strchr(p, ':')) == NULL) || (p2 > nl)) + goto fail; + *(p2++) = 0; + for(; (*p2 == ' ') || (*p2 == '\t'); p2++); + for(nl = p2; (*nl != '\r') && (*nl != '\n'); nl++); + if(!SKIPNL(nl)) + goto fail; + if(strncasecmp(p, "x-ash-", 6)) + headappheader(req, p, p2); + p = nl; + } + return(req); + +fail: + freehthead(req); + return(NULL); +} + +static struct hthead *parserawresp(char *buf) +{ + char *p, *p2, *nl; + char *msg, *ver; + int code; + struct hthead *resp; + + if((nl = strchr(buf, '\n')) == NULL) + return(NULL); + p = strchr(buf, '\r'); + if((p != NULL) && (p < nl)) + nl = p; + if(strncmp(buf, "HTTP/", 5)) + return(NULL); + ver = p = buf + 5; + for(; ((*p >= '0') && (*p <= '9')) || (*p == '.'); p++); + if(*p != ' ') + return(NULL); + *(p++) = 0; + if(((p2 = strchr(p, ' ')) == NULL) || (p2 > nl)) + return(NULL); + *(p2++) = 0; + code = atoi(p); + if((code < 100) || (code >= 600)) + return(NULL); + if(p2 >= nl) + return(NULL); + msg = p2; + p = nl; + if(!SKIPNL(p)) + return(NULL); + + resp = mkresp(code, msg, ver); + while(1) { + if(SKIPNL(p)) { + if(*p) + goto fail; + break; + } + if((nl = strchr(p, '\n')) == NULL) + goto fail; + if(((p2 = strchr(p, ':')) == NULL) || (p2 > nl)) + goto fail; + *(p2++) = 0; + for(; (*p2 == ' ') || (*p2 == '\t'); p2++); + for(nl = p2; (*nl != '\r') && (*nl != '\n'); nl++); + if(!SKIPNL(nl)) + goto fail; + headappheader(resp, p, p2); + p = nl; + } + return(resp); + +fail: + freehthead(resp); + return(NULL); +} + +static off_t passdata(int src, int dst, struct charbuf *buf, off_t max) +{ + size_t dataoff, smax; + off_t sent; + int eof, ret; + + sent = 0; + eof = 0; + while(!eof || (buf->d > 0)) { + if(!eof && (buf->d < buf->s) && ((max < 0) || (sent + buf->d < max))) { + while(1) { + ret = recv(src, buf->b + buf->d, buf->s - buf->d, MSG_DONTWAIT); + if((ret < 0) && (errno == EAGAIN)) { + } else if(ret < 0) { + return(-1); + } else if(ret == 0) { + eof = 1; + break; + } else { + buf->d += ret; + break; + } + if(buf->d > 0) + break; + if(block(src, EV_READ, 0) <= 0) + return(-1); + } + } + for(dataoff = 0; (dataoff < buf->d) && ((max < 0) || (sent < max));) { + if(block(dst, EV_WRITE, 120) <= 0) + return(-1); + smax = buf->d - dataoff; + if(sent + smax > max) + smax = max - sent; + ret = send(dst, buf->b + dataoff, smax, MSG_NOSIGNAL | MSG_DONTWAIT); + if(ret < 0) + return(-1); + dataoff += ret; + sent += ret; + } + bufeat(*buf, dataoff); + } + return(sent); +} + +static void serve(struct muth *muth, va_list args) +{ + vavar(int, fd); + vavar(struct sockaddr_storage, name); + int cfd; + char old; + char *hd; + struct charbuf inbuf, outbuf; + struct hthead *req, *resp; + off_t dlen, sent; + size_t headoff; + char nmbuf[256]; + + bufinit(inbuf); + bufinit(outbuf); + cfd = -1; + req = NULL; + while(1) { + /* + * First, find and decode the header: + */ + if((headoff = readhead(fd, &inbuf)) < 0) + goto out; + if(headoff > 65536) { + /* We cannot handle arbitrarily large headers, as they + * need to fit within a single Unix datagram. This is + * probably a safe limit, and larger packets than this are + * most likely erroneous (or malicious) anyway. */ + goto out; + } + old = inbuf.b[headoff]; + inbuf.b[headoff] = 0; + if((req = parserawreq(inbuf.b)) == NULL) + goto out; + inbuf.b[headoff] = old; + bufeat(inbuf, headoff); + + /* + * Add metainformation and then send the request to the root + * multiplexer: + */ + if(name.ss_family == AF_INET) { + headappheader(req, "X-Ash-Address", inet_ntop(AF_INET, &((struct sockaddr_in *)&name)->sin_addr, nmbuf, sizeof(nmbuf))); + headappheader(req, "X-Ash-Port", sprintf3("%i", ntohs(((struct sockaddr_in *)&name)->sin_port))); + } else if(name.ss_family == AF_INET6) { + headappheader(req, "X-Ash-Address", inet_ntop(AF_INET6, &((struct sockaddr_in6 *)&name)->sin6_addr, nmbuf, sizeof(nmbuf))); + headappheader(req, "X-Ash-Port", sprintf3("%i", ntohs(((struct sockaddr_in6 *)&name)->sin6_port))); + } + cfd = sendreq(plex, req); + + /* + * If there is message data, pass it: + */ + if((hd = getheader(req, "content-length")) != NULL) { + dlen = atoo(hd); + if(dlen > 0) + passdata(fd, cfd, &inbuf, dlen); + } + + /* + * Find and decode the response header: + */ + outbuf.d = 0; + headoff = readhead(cfd, &outbuf); + hd = memcpy(smalloc(headoff + 1), outbuf.b, headoff); + hd[headoff] = 0; + if((resp = parserawresp(hd)) == NULL) + goto out; + + /* + * Pass the actual output: + */ + sizebuf(outbuf, 65536); + sent = passdata(cfd, fd, &outbuf, -1); + sent -= headoff; + + /* + * Check for connection expiry + */ + if(strcasecmp(req->method, "head")) { + if((hd = getheader(resp, "content-length")) != NULL) { + if(sent != atoo(hd)) { + /* Exit because of error */ + goto out; + } + } else { + if(((hd = getheader(resp, "transfer-encoding")) == NULL) || !strcasecmp(hd, "identity")) + break; + } + if(((hd = getheader(req, "connection")) != NULL) && !strcasecmp(hd, "close")) + break; + if(((hd = getheader(resp, "connection")) != NULL) && !strcasecmp(hd, "close")) + break; + } + + close(cfd); + cfd = -1; + freehthead(req); + req = NULL; + freehthead(resp); + resp = NULL; + } + +out: + if(cfd >= 0) + close(cfd); + if(req != NULL) + freehthead(req); + if(resp != NULL) + freehthead(resp); + buffree(inbuf); + buffree(outbuf); + close(fd); +} + static void listenloop(struct muth *muth, va_list args) { vavar(int, ss); @@ -124,12 +457,17 @@ static void listenloop(struct muth *muth, va_list args) while(1) { namelen = sizeof(name); - block(ss, EV_READ); + block(ss, EV_READ, 0); ns = accept(ss, (struct sockaddr *)&name, &namelen); - block(ns, EV_WRITE); - write(ns, "test\n", 5); - close(ns); + if(ns < 0) { + flog(LOG_ERR, "accept: %s", strerror(errno)); + goto out; + } + mustart(serve, ns, name); } + +out: + close(ss); } static void ioloop(void) @@ -137,14 +475,18 @@ static 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(1) { + 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); @@ -153,8 +495,12 @@ static void ioloop(void) FD_SET(bl->fd, &efds); if(bl->fd > maxfd) maxfd = bl->fd; + if((bl->to != 0) && ((timeout == 0) || (timeout > bl->to))) + timeout = bl->to; } - ret = select(maxfd + 1, &rfds, &wfds, &efds, NULL); + 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)); @@ -163,6 +509,7 @@ static void ioloop(void) sleep(1); } } + now = time(NULL); for(bl = blockers; bl; bl = nbl) { nbl = bl->n; ev = 0; @@ -172,7 +519,10 @@ static void ioloop(void) ev |= EV_WRITE; if(FD_ISSET(bl->fd, &efds)) ev = -1; - resume(bl->th, ev); + if(ev != 0) + resume(bl->th, ev); + else if((bl->to != 0) && (bl->to <= now)) + resume(bl->th, 0); } } } @@ -181,6 +531,14 @@ int main(int argc, char **argv) { int fd; + if(argc < 2) { + fprintf(stderr, "usage: htparser ROOT [ARGS...]\n"); + exit(1); + } + if((plex = stdmkchild(argv + 1)) < 0) { + flog(LOG_ERR, "could not spawn root multiplexer: %s", strerror(errno)); + return(1); + } if((fd = listensock6(8080)) < 0) { flog(LOG_ERR, "could not listen on IPv6: %s", strerror(errno)); return(1);