src/dirplex/conf.c

   1 /*
   2     ashd - A Sane HTTP Daemon
   3     Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>
   4
   5     This program is free software: you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation, either version 3 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #include <stdlib.h>
  20 #include <stdio.h>
  21 #include <unistd.h>
  22 #include <string.h>
  23 #include <errno.h>
  24 #include <fnmatch.h>
  25 #include <sys/stat.h>
  26
  27 #ifdef HAVE_CONFIG_H
  28 #include <config.h>
  29 #endif
  30 #include <utils.h>
  31 #include <log.h>
  32 #include <cf.h>
  33 #include <resp.h>
  34
  35 #include "dirplex.h"
  36
  37 static struct config *cflist;
  38 struct config *gconfig, *lconfig;
  39
  40 static void freerule(struct rule *rule)
  41 {
  42     freeca(rule->patterns);
  43     free(rule);
  44 }
  45
  46 static void freepattern(struct pattern *pat)
  47 {
  48     struct rule **rule;
  49     struct headmod *head;
  50
  51     for(rule = pat->rules; *rule; rule++)
  52         freerule(*rule);
  53     while((head = pat->headers) != NULL) {
  54         pat->headers = head->next;
  55         free(head->name);
  56         free(head->value);
  57         free(head);
  58     }
  59     if(pat->childnm != NULL)
  60         free(pat->childnm);
  61     freeca(pat->fchild);
  62     free(pat);
  63 }
  64
  65 static void freeconfig(struct config *cf)
  66 {
  67     struct child *ch, *nch;
  68     struct pattern *pat, *npat;
  69
  70     if(cf->prev != NULL)
  71         cf->prev->next = cf->next;
  72     if(cf->next != NULL)
  73         cf->next->prev = cf->prev;
  74     if(cf == cflist)
  75         cflist = cf->next;
  76     if(cf->path != NULL)
  77         free(cf->path);
  78     for(ch = cf->children; ch != NULL; ch = nch) {
  79         nch = ch->next;
  80         freechild(ch);
  81     }
  82     for(pat = cf->patterns; pat != NULL; pat = npat) {
  83         npat = pat->next;
  84         freepattern(pat);
  85     }
  86     freeca(cf->index);
  87     if(cf->capture != NULL)
  88         free(cf->capture);
  89     free(cf);
  90 }
  91
  92 struct child *getchild(struct config *cf, char *name)
  93 {
  94     struct child *ch;
  95
  96     for(ch = cf->children; ch; ch = ch->next) {
  97         if(!strcmp(ch->name, name))
  98             break;
  99     }
 100     return(ch);
 101 }
 102
 103 static struct rule *newrule(struct pattern *pat)
 104 {
 105     int i;
 106     struct rule *rule;
 107
 108     for(i = 0; pat->rules[i]; i++);
 109     pat->rules = srealloc(pat->rules, sizeof(*pat->rules) * (i + 2));
 110     rule = pat->rules[i] = szmalloc(sizeof(*rule));
 111     pat->rules[i + 1] = NULL;
 112     return(rule);
 113 }
 114
 115 static struct pattern *newpattern(void)
 116 {
 117     struct pattern *pat;
 118
 119     omalloc(pat);
 120     pat->rules = szmalloc(sizeof(*pat->rules));
 121     return(pat);
 122 }
 123
 124 static char **cadup(char **w)
 125 {
 126     char **ret;
 127     int i, l;
 128
 129     l = calen(w);
 130     ret = smalloc(sizeof(*ret) * (l + 1));
 131     for(i = 0; i < l; i++)
 132         ret[i] = sstrdup(w[i]);
 133     ret[i] = NULL;
 134     return(ret);
 135 }
 136
 137 static struct pattern *parsepattern(struct cfstate *s)
 138 {
 139     struct pattern *pat;
 140     struct rule *rule;
 141     struct headmod *head;
 142     int sl;
 143
 144     if(!strcmp(s->argv[0], "match")) {
 145         s->expstart = 1;
 146         pat = newpattern();
 147     } else {
 148         return(NULL);
 149     }
 150
 151     if((s->argc > 1) && !strcmp(s->argv[1], "directory"))
 152         pat->type = PT_DIR;
 153     else if((s->argc > 1) && !strcmp(s->argv[1], "notfound"))
 154         pat->type = PT_NOTFOUND;
 155     else
 156         pat->type = PT_FILE;
 157     sl = s->lno;
 158     while(1) {
 159         getcfline(s);
 160         if(!strcmp(s->argv[0], "filename")) {
 161             if(s->argc < 2) {
 162                 flog(LOG_WARNING, "%s:%i: missing pattern for `filename' match", s->file, s->lno);
 163                 continue;
 164             }
 165             rule = newrule(pat);
 166             rule->type = PAT_BASENAME;
 167             rule->patterns = cadup(s->argv + 1);
 168         } else if(!strcmp(s->argv[0], "pathname")) {
 169             if(s->argc < 2) {
 170                 flog(LOG_WARNING, "%s:%i: missing pattern for `pathname' match", s->file, s->lno);
 171                 continue;
 172             }
 173             rule = newrule(pat);
 174             rule->type = PAT_PATHNAME;
 175             rule->patterns = cadup(s->argv + 1);
 176         } else if(!strcmp(s->argv[0], "all")) {
 177             newrule(pat)->type = PAT_ALL;
 178         } else if(!strcmp(s->argv[0], "default")) {
 179             newrule(pat)->type = PAT_DEFAULT;
 180         } else if(!strcmp(s->argv[0], "local")) {
 181             newrule(pat)->type = PAT_LOCAL;
 182         } else if(!strcmp(s->argv[0], "handler")) {
 183             if(s->argc < 2) {
 184                 flog(LOG_WARNING, "%s:%i: missing child name for `handler' directive", s->file, s->lno);
 185                 continue;
 186             }
 187             if(pat->childnm != NULL)
 188                 free(pat->childnm);
 189             pat->childnm = sstrdup(s->argv[1]);
 190         } else if(!strcmp(s->argv[0], "fork")) {
 191             pat->fchild = cadup(s->argv + 1);
 192         } else if(!strcmp(s->argv[0], "set") || !strcmp(s->argv[0], "xset")) {
 193             if(s->argc < 3) {
 194                 flog(LOG_WARNING, "%s:%i: missing header name or pattern for `%s' directive", s->file, s->lno, s->argv[0]);
 195                 continue;
 196             }
 197             omalloc(head);
 198             if(!strcmp(s->argv[0], "xset"))
 199                 head->name = sprintf2("X-Ash-%s", s->argv[1]);
 200             else
 201                 head->name = sstrdup(s->argv[1]);
 202             head->value = sstrdup(s->argv[2]);
 203             head->next = pat->headers;
 204             pat->headers = head;
 205         } else if(!strcmp(s->argv[0], "end") || !strcmp(s->argv[0], "eof")) {
 206             break;
 207         } else {
 208             flog(LOG_WARNING, "%s:%i: unknown directive `%s' in pattern declaration", s->file, s->lno, s->argv[0]);
 209         }
 210     }
 211
 212     if(pat->rules[0] == NULL) {
 213         flog(LOG_WARNING, "%s:%i: missing rules in match declaration", s->file, sl);
 214         freepattern(pat);
 215         return(NULL);
 216     }
 217     if((pat->childnm == NULL) && (pat->fchild == NULL)) {
 218         flog(LOG_WARNING, "%s:%i: missing handler in match declaration", s->file, sl);
 219         freepattern(pat);
 220         return(NULL);
 221     }
 222     return(pat);
 223 }
 224
 225 static struct config *emptyconfig(void)
 226 {
 227     struct config *cf;
 228
 229     omalloc(cf);
 230     return(cf);
 231 }
 232
 233 struct config *readconfig(char *file)
 234 {
 235     struct cfstate *s;
 236     FILE *in;
 237     struct config *cf;
 238     struct child *child;
 239     struct pattern *pat;
 240
 241     if((in = fopen(file, "r")) == NULL) {
 242         flog(LOG_WARNING, "%s: %s", file, strerror(errno));
 243         return(NULL);
 244     }
 245     s = mkcfparser(in, file);
 246     cf = emptyconfig();
 247
 248     while(1) {
 249         getcfline(s);
 250         if((child = parsechild(s)) != NULL) {
 251             child->next = cf->children;
 252             cf->children = child;
 253         } else if((pat = parsepattern(s)) != NULL) {
 254             pat->next = cf->patterns;
 255             cf->patterns = pat;
 256         } else if(!strcmp(s->argv[0], "index-file")) {
 257             freeca(cf->index);
 258             cf->index = cadup(s->argv + 1);
 259         } else if(!strcmp(s->argv[0], "capture")) {
 260             if(s->argc < 2) {
 261                 flog(LOG_WARNING, "%s:%i: missing argument to capture declaration", s->file, s->lno);
 262                 continue;
 263             }
 264             if(cf->capture != NULL)
 265                 free(cf->capture);
 266             cf->capture = sstrdup(s->argv[1]);
 267             cf->caproot = 0;
 268             if((s->argc > 2) && strchr(s->argv[2], 'D'))
 269                 cf->caproot = 1;
 270         } else if(!strcmp(s->argv[0], "eof")) {
 271             break;
 272         } else {
 273             flog(LOG_WARNING, "%s:%i: unknown directive `%s'", s->file, s->lno, s->argv[0]);
 274         }
 275     }
 276
 277     freecfparser(s);
 278     fclose(in);
 279     return(cf);
 280 }
 281
 282 struct config *getconfig(char *path)
 283 {
 284     struct config *cf, *ocf;
 285     struct stat sb;
 286     char *fn;
 287     time_t mtime;
 288
 289     fn = sprintf3("%s/.htrc", path);
 290     for(cf = cflist; cf != NULL; cf = cf->next) {
 291         if(!strcmp(cf->path, path)) {
 292             if(now - cf->lastck > 5) {
 293                 cf->lastck = now;
 294                 if(stat(fn, &sb) || (sb.st_mtime != cf->mtime))
 295                     break;
 296             }
 297             return(cf);
 298         }
 299     }
 300     ocf = cf;
 301     if(access(fn, R_OK) || stat(fn, &sb)) {
 302         cf = emptyconfig();
 303         mtime = 0;
 304     } else {
 305         if((cf = readconfig(fn)) == NULL)
 306             return(NULL);
 307         mtime = sb.st_mtime;
 308     }
 309     if(ocf != NULL) {
 310         mergechildren(cf->children, ocf->children);
 311         freeconfig(ocf);
 312     }
 313     cf->path = sstrdup(path);
 314     cf->mtime = mtime;
 315     cf->lastck = now;
 316     cf->next = cflist;
 317     cf->prev = NULL;
 318     if(cflist != NULL)
 319         cflist->prev = cf;
 320     cflist = cf;
 321     return(cf);
 322 }
 323
 324 struct config **getconfigs(char *file)
 325 {
 326     static struct config **ret = NULL;
 327     struct {
 328         struct config **b;
 329         size_t s, d;
 330     } buf;
 331     struct config *cf;
 332     char *tmp, *p;
 333
 334     if(ret != NULL)
 335         free(ret);
 336     bufinit(buf);
 337     if(!strncmp(file, "./", 2))
 338         file += 2;
 339     tmp = sstrdup(file);
 340     while(1) {
 341         if((p = strrchr(tmp, '/')) == NULL)
 342             break;
 343         *p = 0;
 344         if((cf = getconfig(tmp)) != NULL)
 345             bufadd(buf, cf);
 346     }
 347     free(tmp);
 348     if((cf = getconfig(".")) != NULL)
 349         bufadd(buf, cf);
 350     if(lconfig != NULL)
 351         bufadd(buf, lconfig);
 352     if(gconfig != NULL)
 353         bufadd(buf, gconfig);
 354     bufadd(buf, NULL);
 355     return(ret = buf.b);
 356 }
 357
 358 struct child *findchild(char *file, char *name, struct config **cf)
 359 {
 360     int i;
 361     struct config **cfs;
 362     struct child *ch;
 363
 364     if(cf != NULL)
 365         *cf = NULL;
 366     cfs = getconfigs(file);
 367     for(i = 0; cfs[i] != NULL; i++) {
 368         if((ch = getchild(cfs[i], name)) != NULL) {
 369             if(cf != NULL)
 370                 *cf = cfs[i];
 371             return(ch);
 372         }
 373     }
 374     if(!strcmp(name, ".notfound"))
 375         return(notfound);
 376     return(NULL);
 377 }
 378
 379 struct pattern *findmatch(char *file, int trydefault, int type)
 380 {
 381     int i, o, c;
 382     char *bn, *ln;
 383     struct config **cfs;
 384     struct pattern *pat;
 385     struct rule *rule;
 386     size_t pl;
 387
 388     if((bn = strrchr(file, '/')) != NULL)
 389         bn++;
 390     else
 391         bn = file;
 392     cfs = getconfigs(file);
 393     for(c = 0; cfs[c] != NULL; c++) {
 394         if(cfs[c]->path == NULL) {
 395             ln = file;
 396         } else {
 397             pl = strlen(cfs[c]->path);
 398             if((strlen(file) > pl) && !strncmp(file, cfs[c]->path, pl) && (file[pl] == '/'))
 399                 ln = file + pl + 1;
 400             else
 401                 ln = file;      /* This should only happen in the base directory. */
 402         }
 403         for(pat = cfs[c]->patterns; pat != NULL; pat = pat->next) {
 404             if(pat->type != type)
 405                 continue;
 406             for(i = 0; (rule = pat->rules[i]) != NULL; i++) {
 407                 if(rule->type == PAT_BASENAME) {
 408                     for(o = 0; rule->patterns[o] != NULL; o++) {
 409                         if(!fnmatch(rule->patterns[o], bn, 0))
 410                             break;
 411                     }
 412                     if(rule->patterns[o] == NULL)
 413                         break;
 414                 } else if(rule->type == PAT_PATHNAME) {
 415                     for(o = 0; rule->patterns[o] != NULL; o++) {
 416                         if(!fnmatch(rule->patterns[o], ln, FNM_PATHNAME))
 417                             break;
 418                     }
 419                     if(rule->patterns[o] == NULL)
 420                         break;
 421                 } else if(rule->type == PAT_ALL) {
 422                 } else if(rule->type == PAT_DEFAULT) {
 423                     if(!trydefault)
 424                         break;
 425                 } else if(rule->type == PAT_LOCAL) {
 426                     if(strchr(ln, '/'))
 427                         break;
 428                 }
 429             }
 430             if(!rule)
 431                 return(pat);
 432         }
 433     }
 434     if(!trydefault)
 435         return(findmatch(file, 1, type));
 436     return(NULL);
 437 }
 438
 439 static int donotfound(struct child *ch, struct hthead *req, int fd, void (*chinit)(void *), void *idata)
 440 {
 441     simpleerror(fd, 404, "Not Found", "The requested URL has no corresponding resource.");
 442     return(0);
 443 }
 444
 445 static struct chandler i_notfound = {
 446     .handle = donotfound,
 447 };
 448
 449 static struct child s_notfound = {
 450     .name = ".notfound",
 451     .iface = &i_notfound,
 452 };
 453 struct child *notfound = &s_notfound;