X-Git-Url: http://www.dolda2000.com/gitweb/?a=blobdiff_plain;f=src%2Fdirplex%2Fdirplex.c;h=14320ada0fc56fd8ebba1cc18eecd226d57e560d;hb=acc2d159e6f946ed6abc7c0e843a483d6478bee3;hp=f8ac25a9c5c0fa4b081ff96503020da0ab36c5da;hpb=300d73d9ee0fe2fb09643a2a146711f1a8d7edd3;p=ashd.git

diff --git a/src/dirplex/dirplex.c b/src/dirplex/dirplex.c
index f8ac25a..14320ad 100644
--- a/src/dirplex/dirplex.c
+++ b/src/dirplex/dirplex.c
@@ -25,6 +25,7 @@
 #include <ctype.h>
 #include <dirent.h>
 #include <time.h>
+#include <sys/wait.h>
 #include <sys/signal.h>
 
 #ifdef HAVE_CONFIG_H
@@ -58,8 +59,15 @@ static void handle(struct hthead *req, int fd, char *path, struct pattern *pat)
 {
     struct child *ch;
     struct config *ccf;
+    struct headmod *head;
     char *twd;
 
+    for(head = pat->headers; head != NULL; head = head->next) {
+	headrmheader(req, head->name);
+	headappheader(req, head->name, head->value);
+    }
+    if(!strncmp(path, "./", 2) && path[2])
+	path += 2;
     if(pat->fchild) {
 	headappheader(req, "X-Ash-File", path);
 	stdforkserve(pat->fchild, req, fd, NULL, NULL);
@@ -69,8 +77,7 @@ static void handle(struct hthead *req, int fd, char *path, struct pattern *pat)
 	    simpleerror(fd, 500, "Configuration Error", "The server is erroneously configured. Handler %s was requested, but not declared.", pat->childnm);
 	    return;
 	}
-	twd = NULL;
-	if((twd = ccf->path) != NULL) {
+	if((twd = ccf?ccf->path:NULL) != NULL) {
 	    if(!strcmp(twd, ".")) {
 		twd = NULL;
 	    } else if(strncmp(path, twd, strlen(twd)) || (path[strlen(twd)] != '/')) {
@@ -88,12 +95,25 @@ static void handle(struct hthead *req, int fd, char *path, struct pattern *pat)
     }
 }
 
+static void handle404(struct hthead *req, int fd, char *path)
+{
+    struct child *ch;
+    struct config *ccf;
+    char *tmp;
+    
+    tmp = sstrdup(path);
+    ch = findchild(tmp, ".notfound", &ccf);
+    if(childhandle(ch, req, fd, chinit, ccf?ccf->path:NULL))
+	simpleerror(fd, 500, "Server Error", "The request handler crashed.");
+    free(tmp);
+}
+
 static void handlefile(struct hthead *req, int fd, char *path)
 {
     struct pattern *pat;
 
     if((pat = findmatch(path, 0, 0)) == NULL) {
-	simpleerror(fd, 404, "Not Found", "The requested URL has no corresponding resource.");
+	handle404(req, fd, path);
 	return;
     }
     handle(req, fd, path, pat);
@@ -184,8 +204,8 @@ static int checkentry(struct hthead *req, int fd, char *path, char *rest, char *
     char *newpath;
     int rv;
     
-    if(!el == '.') {
-	simpleerror(fd, 404, "Not Found", "The requested URL has no corresponding resource.");
+    if(*el == '.') {
+	handle404(req, fd, sprintf3("%s/", path));
 	return(1);
     }
     if(!stat(sprintf3("%s/%s", path, el), &sb)) {
@@ -205,7 +225,7 @@ static int checkentry(struct hthead *req, int fd, char *path, char *rest, char *
 	    free(newpath);
 	    return(1);
 	}
-	simpleerror(fd, 404, "Not Found", "The requested URL has no corresponding resource.");
+	handle404(req, fd, sprintf3("%s/", path));
 	return(1);
     }
     if(!strchr(el, '.') && ((newpath = findfile(path, el, NULL)) != NULL)) {
@@ -236,7 +256,7 @@ static int checkdir(struct hthead *req, int fd, char *path, char *rest)
 	if(*rest == '/')
 	    rest++;
 	replrest(req, rest);
-	if(childhandle(ch, req, fd, chinit, ccf->path))
+	if(childhandle(ch, req, fd, chinit, ccf?ccf->path:NULL))
 	    simpleerror(fd, 500, "Server Error", "The request handler crashed.");
 	return(1);
     }
@@ -272,14 +292,15 @@ static int checkpath(struct hthead *req, int fd, char *path, char *rest)
 	goto out;
     }
     if(strchr(el, '/') || (!*el && *rest)) {
-	simpleerror(fd, 404, "Not Found", "The requested URL has no corresponding resource.");
+	handle404(req, fd, sprintf3("%s/", path));
 	rv = 1;
 	goto out;
     }
     if(!*el) {
 	replrest(req, rest);
 	handledir(req, fd, path);
-	return(1);
+	rv = 1;
+	goto out;
     }
     rv = checkentry(req, fd, path, rest, el);
     
@@ -293,7 +314,20 @@ static void serve(struct hthead *req, int fd)
 {
     now = time(NULL);
     if(!checkpath(req, fd, ".", req->rest))
-	simpleerror(fd, 404, "Not Found", "The requested URL has no corresponding resource.");
+	handle404(req, fd, ".");
+}
+
+static void chldhandler(int sig)
+{
+    pid_t pid;
+    
+    do {
+	pid = waitpid(-1, NULL, WNOHANG);
+    } while(pid > 0);
+}
+
+static void sighandler(int sig)
+{
 }
 
 static void usage(FILE *out)
@@ -355,7 +389,8 @@ int main(int argc, char **argv)
 	flog(LOG_ERR, "could not change directory to %s: %s", argv[optind], strerror(errno));
 	exit(1);
     }
-    signal(SIGCHLD, SIG_IGN);
+    signal(SIGCHLD, chldhandler);
+    signal(SIGPIPE, sighandler);
     while(1) {
 	if((fd = recvreq(0, &req)) < 0) {
 	    if(errno != 0)