etc: Add environment option to run init.d/ashd silently.
[ashd.git] / src / dirplex / dirplex.c
index 358d0b1..ee7f650 100644 (file)
@@ -25,6 +25,7 @@
 #include <ctype.h>
 #include <dirent.h>
 #include <time.h>
+#include <fnmatch.h>
 #include <sys/wait.h>
 #include <sys/signal.h>
 
@@ -55,12 +56,27 @@ static void chinit(void *idata)
     }
 }
 
+static void childerror(struct hthead *req, int fd)
+{
+    if(errno == EAGAIN)
+       simpleerror(fd, 500, "Server Error", "The request handler is overloaded.");
+    else
+       simpleerror(fd, 500, "Server Error", "The request handler crashed.");
+}
+
 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);
@@ -70,8 +86,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)] != '/')) {
@@ -85,7 +100,24 @@ static void handle(struct hthead *req, int fd, char *path, struct pattern *pat)
        }
        headappheader(req, "X-Ash-File", path);
        if(childhandle(ch, req, fd, chinit, twd))
-           simpleerror(fd, 500, "Server Error", "The request handler crashed.");
+           childerror(req, fd);
+    }
+}
+
+static void handle404(struct hthead *req, int fd, char *path)
+{
+    struct child *ch;
+    struct config *ccf;
+    struct pattern *pat;
+    
+    char tmp[strlen(path) + 1];
+    strcpy(tmp, path);
+    if((pat = findmatch(tmp, 0, PT_NOTFOUND)) != NULL) {
+       handle(req, fd, tmp, pat);
+    } else {
+       ch = findchild(tmp, ".notfound", &ccf);
+       if(childhandle(ch, req, fd, chinit, ccf?ccf->path:NULL))
+           childerror(req, fd);
     }
 }
 
@@ -93,13 +125,34 @@ 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.");
+    if((pat = findmatch(path, 0, PT_FILE)) == NULL) {
+       handle404(req, fd, path);
        return;
     }
     handle(req, fd, path, pat);
 }
 
+static int checkaccess(char *path, char *name)
+{
+    int i, o;
+    struct config **cfs;
+    
+    if(*name == '.') {
+       cfs = getconfigs(sprintf3("%s/", path));
+       for(i = 0; cfs[i] != NULL; i++) {
+           if(cfs[i]->dotallow != NULL) {
+               for(o = 0; cfs[i]->dotallow[o] != NULL; o++) {
+                   if(!fnmatch(cfs[i]->dotallow[o], name, 0))
+                       return(1);
+               }
+               break;
+           }
+       }
+       return(0);
+    }
+    return(1);
+}
+
 static char *findfile(char *path, char *name, struct stat *sb)
 {
     DIR *dir;
@@ -124,12 +177,20 @@ static char *findfile(char *path, char *name, struct stat *sb)
            continue;
        if(strncmp(dent->d_name, name, strlen(name)))
            continue;
-       fp = sprintf3("%s/%s", path, dent->d_name);
-       if(stat(fp, sb))
+       fp = sprintf2("%s/%s", path, dent->d_name);
+       if(stat(fp, sb)) {
+           free(fp);
+           continue;
+       }
+       if(!S_ISREG(sb->st_mode)) {
+           free(fp);
            continue;
-       if(!S_ISREG(sb->st_mode))
+       }
+       if(!checkaccess(path, dent->d_name)) {
+           free(fp);
            continue;
-       ret = sstrdup(fp);
+       }
+       ret = fp;
        break;
     }
     closedir(dir);
@@ -167,7 +228,7 @@ static void handledir(struct hthead *req, int fd, char *path)
            break;
        }
     }
-    if((pat = findmatch(cpath, 0, 1)) != NULL) {
+    if((pat = findmatch(cpath, 0, PT_DIR)) != NULL) {
        handle(req, fd, cpath, pat);
        goto out;
     }
@@ -177,26 +238,24 @@ out:
     free(cpath);
 }
 
-static int checkpath(struct hthead *req, int fd, char *path, char *rest);
+static int checkpath(struct hthead *req, int fd, char *path, char *rest, int final);
 
-static int checkentry(struct hthead *req, int fd, char *path, char *rest, char *el)
+static int checkentry(struct hthead *req, int fd, char *path, char *rest, char *el, int final)
 {
     struct stat sb;
     char *newpath;
     int rv;
     
-    if(!el == '.') {
-       simpleerror(fd, 404, "Not Found", "The requested URL has no corresponding resource.");
-       return(1);
-    }
     if(!stat(sprintf3("%s/%s", path, el), &sb)) {
+       if(!checkaccess(path, el))
+           return(0);
        if(S_ISDIR(sb.st_mode)) {
            if(!*rest) {
                stdredir(req, fd, 301, sprintf3("%s/", el));
                return(1);
            }
            newpath = sprintf2("%s/%s", path, el);
-           rv = checkpath(req, fd, newpath, rest + 1);
+           rv = checkpath(req, fd, newpath, rest + 1, final);
            free(newpath);
            return(rv);
        } else if(S_ISREG(sb.st_mode)) {
@@ -206,7 +265,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)) {
@@ -220,12 +279,14 @@ static int checkentry(struct hthead *req, int fd, char *path, char *rest, char *
 
 static int checkdir(struct hthead *req, int fd, char *path, char *rest)
 {
-    char *cpath;
+    char *cpath, *newpath;
     struct config *cf, *ccf;
     struct child *ch;
+    struct stat sb;
+    int rv;
     
     cf = getconfig(path);
-    if(cf->capture != NULL) {
+    if((cf->capture != NULL) && (cf->caproot || !cf->path || strcmp(cf->path, "."))) {
        cpath = sprintf2("%s/", path);
        if((ch = findchild(cpath, cf->capture, &ccf)) == NULL) {
            free(cpath);
@@ -237,14 +298,30 @@ 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))
-           simpleerror(fd, 500, "Server Error", "The request handler crashed.");
+       if(childhandle(ch, req, fd, chinit, ccf?ccf->path:NULL))
+           childerror(req, fd);
        return(1);
     }
+    if(cf->reparse != NULL) {
+       newpath = (cf->reparse[0] == '/')?sstrdup(cf->reparse):sprintf2("%s/%s", path, cf->reparse);
+       rv = stat(newpath, &sb);
+       if(!rv && S_ISDIR(sb.st_mode)) {
+           rv = checkpath(req, fd, newpath, rest, !cf->parsecomb);
+       } else if(!rv && S_ISREG(sb.st_mode)) {
+           replrest(req, rest);
+           handlefile(req, fd, newpath);
+           rv = 1;
+       } else {
+           rv = !cf->parsecomb;
+       }
+       free(newpath);
+       if(rv)
+           return(rv);
+    }
     return(0);
 }
 
-static int checkpath(struct hthead *req, int fd, char *path, char *rest)
+static int checkpath(struct hthead *req, int fd, char *path, char *rest, int final)
 {
     char *p, *el;
     int rv;
@@ -273,8 +350,7 @@ 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.");
-       rv = 1;
+       rv = 0;
        goto out;
     }
     if(!*el) {
@@ -283,9 +359,13 @@ static int checkpath(struct hthead *req, int fd, char *path, char *rest)
        rv = 1;
        goto out;
     }
-    rv = checkentry(req, fd, path, rest, el);
+    rv = checkentry(req, fd, path, rest, el, final);
     
 out:
+    if(final && !rv) {
+       handle404(req, fd, sprintf3("%s/", path));
+       rv = 1;
+    }
     if(el != NULL)
        free(el);
     return(rv);
@@ -294,17 +374,18 @@ out:
 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.");
+    checkpath(req, fd, ".", req->rest, 1);
 }
 
 static void chldhandler(int sig)
 {
     pid_t pid;
+    int st;
     
-    do {
-       pid = waitpid(-1, NULL, WNOHANG);
-    } while(pid > 0);
+    while((pid = waitpid(-1, &st, WNOHANG)) > 0) {
+       if(WCOREDUMP(st))
+           flog(LOG_WARNING, "child process %i dumped core", pid);
+    }
 }
 
 static void sighandler(int sig)