X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=src%2Fhtparser.c;h=001c6c1b13c8e71215dd24ca5ae10db49f9e96cf;hb=32e24c19cca5538c922542c7a6078efc53f5638b;hp=5d7a9324cfeef4aa909a366124a0ce6bb4435477;hpb=af34331c8aae9bacdae3655546b2fb6eb2fbe23f;p=ashd.git

diff --git a/src/htparser.c b/src/htparser.c
index 5d7a932..001c6c1 100644
--- a/src/htparser.c
+++ b/src/htparser.c
@@ -365,8 +365,10 @@ static void serve(struct muth *muth, va_list args)
 	 */
 	if((hd = getheader(req, "content-length")) != NULL) {
 	    dlen = atoo(hd);
-	    if(dlen > 0)
-		passdata(fd, cfd, &inbuf, dlen);
+	    if(dlen > 0) {
+		if(passdata(fd, cfd, &inbuf, dlen) < 0)
+		    goto out;
+	    }
 	}
 	/* Make sure to send EOF */
 	shutdown(cfd, SHUT_WR);
@@ -386,7 +388,8 @@ static void serve(struct muth *muth, va_list args)
 	 * Pass the actual output:
 	 */
 	sizebuf(outbuf, 65536);
-	sent = passdata(cfd, fd, &outbuf, -1);
+	if((sent = passdata(cfd, fd, &outbuf, -1)) < 0)
+	    goto out;
 	sent -= headoff;
 	
 	/*
@@ -450,6 +453,28 @@ out:
     close(ss);
 }
 
+static void plexwatch(struct muth *muth, va_list args)
+{
+    vavar(int, fd);
+    char *buf;
+    int ret;
+    
+    while(1) {
+	block(fd, EV_READ, 0);
+	buf = smalloc(65536);
+	ret = recv(fd, buf, 65536, 0);
+	if(ret < 0) {
+	    flog(LOG_WARNING, "received error on rootplex read channel: %s", strerror(errno));
+	    exit(1);
+	} else if(ret == 0) {
+	    exit(0);
+	}
+	/* Maybe I'd like to implement some protocol in this direction
+	 * some day... */
+	free(buf);
+    }
+}
+
 int main(int argc, char **argv)
 {
     int fd;
@@ -475,6 +500,7 @@ int main(int argc, char **argv)
     } else {
 	mustart(listenloop, fd);
     }
+    mustart(plexwatch, plex);
     ioloop();
     return(0);
 }