dirplex, patplex: Added ability to set arbitrary headers based on match.
[ashd.git] / 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     sl = s->lno;
154     while(1) {
155         getcfline(s);
156         if(!strcmp(s->argv[0], "filename")) {
157             if(s->argc < 2) {
158                 flog(LOG_WARNING, "%s:%i: missing pattern for `filename' match", s->file, s->lno);
159                 continue;
160             }
161             rule = newrule(pat);
162             rule->type = PAT_BASENAME;
163             rule->patterns = cadup(s->argv + 1);
164         } else if(!strcmp(s->argv[0], "pathname")) {
165             if(s->argc < 2) {
166                 flog(LOG_WARNING, "%s:%i: missing pattern for `pathname' match", s->file, s->lno);
167                 continue;
168             }
169             rule = newrule(pat);
170             rule->type = PAT_PATHNAME;
171             rule->patterns = cadup(s->argv + 1);
172         } else if(!strcmp(s->argv[0], "all")) {
173             newrule(pat)->type = PAT_ALL;
174         } else if(!strcmp(s->argv[0], "default")) {
175             newrule(pat)->type = PAT_DEFAULT;
176         } else if(!strcmp(s->argv[0], "local")) {
177             newrule(pat)->type = PAT_LOCAL;
178         } else if(!strcmp(s->argv[0], "handler")) {
179             if(s->argc < 2) {
180                 flog(LOG_WARNING, "%s:%i: missing child name for `handler' directive", s->file, s->lno);
181                 continue;
182             }
183             if(pat->childnm != NULL)
184                 free(pat->childnm);
185             pat->childnm = sstrdup(s->argv[1]);
186         } else if(!strcmp(s->argv[0], "fork")) {
187             pat->fchild = cadup(s->argv + 1);
188         } else if(!strcmp(s->argv[0], "set")) {
189             if(s->argc < 3) {
190                 flog(LOG_WARNING, "%s:%i: missing header name or pattern for `set' directive", s->file, s->lno);
191                 continue;
192             }
193             omalloc(head);
194             head->name = sstrdup(s->argv[1]);
195             head->value = sstrdup(s->argv[2]);
196             head->next = pat->headers;
197             pat->headers = head;
198         } else if(!strcmp(s->argv[0], "end") || !strcmp(s->argv[0], "eof")) {
199             break;
200         } else {
201             flog(LOG_WARNING, "%s:%i: unknown directive `%s' in pattern declaration", s->file, s->lno, s->argv[0]);
202         }
203     }
204     
205     if(pat->rules[0] == NULL) {
206         flog(LOG_WARNING, "%s:%i: missing rules in match declaration", s->file, sl);
207         freepattern(pat);
208         return(NULL);
209     }
210     if((pat->childnm == NULL) && (pat->fchild == NULL)) {
211         flog(LOG_WARNING, "%s:%i: missing handler in match declaration", s->file, sl);
212         freepattern(pat);
213         return(NULL);
214     }
215     return(pat);
216 }
217
218 static struct config *emptyconfig(void)
219 {
220     struct config *cf;
221     
222     omalloc(cf);
223     return(cf);
224 }
225
226 struct config *readconfig(char *file)
227 {
228     struct cfstate *s;
229     FILE *in;
230     struct config *cf;
231     struct child *child;
232     struct pattern *pat;
233     
234     if((in = fopen(file, "r")) == NULL) {
235         flog(LOG_WARNING, "%s: %s", file, strerror(errno));
236         return(NULL);
237     }
238     s = mkcfparser(in, file);
239     cf = emptyconfig();
240     
241     while(1) {
242         getcfline(s);
243         if((child = parsechild(s)) != NULL) {
244             child->next = cf->children;
245             cf->children = child;
246         } else if((pat = parsepattern(s)) != NULL) {
247             pat->next = cf->patterns;
248             cf->patterns = pat;
249         } else if(!strcmp(s->argv[0], "index-file")) {
250             freeca(cf->index);
251             cf->index = NULL;
252             if(s->argc > 1)
253                 cf->index = cadup(s->argv + 1);
254         } else if(!strcmp(s->argv[0], "capture")) {
255             if(s->argc < 2) {
256                 flog(LOG_WARNING, "%s:%i: missing argument to capture declaration", s->file, s->lno);
257                 continue;
258             }
259             if(cf->capture != NULL)
260                 free(cf->capture);
261             cf->capture = sstrdup(s->argv[1]);
262         } else if(!strcmp(s->argv[0], "eof")) {
263             break;
264         } else {
265             flog(LOG_WARNING, "%s:%i: unknown directive `%s'", s->file, s->lno, s->argv[0]);
266         }
267     }
268     
269     freecfparser(s);
270     fclose(in);
271     return(cf);
272 }
273
274 struct config *getconfig(char *path)
275 {
276     struct config *cf, *ocf;
277     struct stat sb;
278     char *fn;
279     time_t mtime;
280     
281     fn = sprintf3("%s/.htrc", path);
282     for(cf = cflist; cf != NULL; cf = cf->next) {
283         if(!strcmp(cf->path, path)) {
284             if(now - cf->lastck > 5) {
285                 cf->lastck = now;
286                 if(stat(fn, &sb) || (sb.st_mtime != cf->mtime))
287                     break;
288             }
289             return(cf);
290         }
291     }
292     ocf = cf;
293     if(access(fn, R_OK) || stat(fn, &sb)) {
294         cf = emptyconfig();
295         mtime = 0;
296     } else {
297         if((cf = readconfig(fn)) == NULL)
298             return(NULL);
299         mtime = sb.st_mtime;
300     }
301     if(ocf != NULL) {
302         mergechildren(cf->children, ocf->children);
303         freeconfig(ocf);
304     }
305     cf->path = sstrdup(path);
306     cf->mtime = mtime;
307     cf->lastck = now;
308     cf->next = cflist;
309     cf->prev = NULL;
310     if(cflist != NULL)
311         cflist->prev = cf;
312     cflist = cf;
313     return(cf);
314 }
315
316 struct config **getconfigs(char *file)
317 {
318     static struct config **ret = NULL;
319     struct {
320         struct config **b;
321         size_t s, d;
322     } buf;
323     struct config *cf;
324     char *tmp, *p;
325     
326     if(ret != NULL)
327         free(ret);
328     bufinit(buf);
329     tmp = sstrdup(file);
330     while(1) {
331         if((p = strrchr(tmp, '/')) == NULL)
332             break;
333         *p = 0;
334         if((cf = getconfig(tmp)) != NULL)
335             bufadd(buf, cf);
336     }
337     free(tmp);
338     if((cf = getconfig(".")) != NULL)
339         bufadd(buf, cf);
340     if(lconfig != NULL)
341         bufadd(buf, lconfig);
342     if(gconfig != NULL)
343         bufadd(buf, gconfig);
344     bufadd(buf, NULL);
345     return(ret = buf.b);
346 }
347
348 struct child *findchild(char *file, char *name, struct config **cf)
349 {
350     int i;
351     struct config **cfs;
352     struct child *ch;
353     
354     if(cf != NULL)
355         *cf = NULL;
356     cfs = getconfigs(file);
357     for(i = 0; cfs[i] != NULL; i++) {
358         if((ch = getchild(cfs[i], name)) != NULL) {
359             if(cf != NULL)
360                 *cf = cfs[i];
361             return(ch);
362         }
363     }
364     if(!strcmp(name, ".notfound"))
365         return(notfound);
366     return(NULL);
367 }
368
369 struct pattern *findmatch(char *file, int trydefault, int dir)
370 {
371     int i, o, c;
372     char *bn;
373     struct config **cfs;
374     struct pattern *pat;
375     struct rule *rule;
376     size_t pl;
377     
378     if((bn = strrchr(file, '/')) != NULL)
379         bn++;
380     else
381         bn = file;
382     cfs = getconfigs(file);
383     for(c = 0; cfs[c] != NULL; c++) {
384         for(pat = cfs[c]->patterns; pat != NULL; pat = pat->next) {
385             if(!dir && (pat->type == PT_DIR))
386                 continue;
387             if(dir && (pat->type != PT_DIR))
388                 continue;
389             for(i = 0; (rule = pat->rules[i]) != NULL; i++) {
390                 if(rule->type == PAT_BASENAME) {
391                     for(o = 0; rule->patterns[o] != NULL; o++) {
392                         if(!fnmatch(rule->patterns[o], bn, 0))
393                             break;
394                     }
395                     if(rule->patterns[o] == NULL)
396                         break;
397                 } else if(rule->type == PAT_PATHNAME) {
398                     for(o = 0; rule->patterns[o] != NULL; o++) {
399                         if(!fnmatch(rule->patterns[o], file, FNM_PATHNAME))
400                             break;
401                     }
402                     if(rule->patterns[o] == NULL)
403                         break;
404                 } else if(rule->type == PAT_ALL) {
405                 } else if(rule->type == PAT_DEFAULT) {
406                     if(!trydefault)
407                         break;
408                 } else if(rule->type == PAT_LOCAL) {
409                     if(cfs[c]->path == NULL)
410                         break;
411                     pl = strlen(cfs[c]->path);
412                     if(!((strlen(file) > pl) && !strncmp(file, cfs[c]->path, pl) && (file[pl] == '/') && !strchr(file + pl + 1, '/')))
413                         break;
414                 }
415             }
416             if(!rule)
417                 return(pat);
418         }
419     }
420     if(!trydefault)
421         return(findmatch(file, 1, dir));
422     return(NULL);
423 }
424
425 static int donotfound(struct child *ch, struct hthead *req, int fd, void (*chinit)(void *), void *idata)
426 {
427     simpleerror(fd, 404, "Not Found", "The requested URL has no corresponding resource.");
428     return(0);
429 }
430
431 static struct chandler i_notfound = {
432     .handle = donotfound,
433 };
434
435 static struct child s_notfound = {
436     .name = ".notfound",
437     .iface = &i_notfound,
438 };
439 struct child *notfound = &s_notfound;