ratequeue: Add some missing includes.
[ashd.git] / src / patplex.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 <string.h>
22 #include <unistd.h>
23 #include <signal.h>
24 #include <errno.h>
25 #include <ctype.h>
26 #include <regex.h>
27 #include <limits.h>
28 #include <sys/wait.h>
29
30 #ifdef HAVE_CONFIG_H
31 #include <config.h>
32 #endif
33 #include <utils.h>
34 #include <log.h>
35 #include <req.h>
36 #include <proc.h>
37 #include <resp.h>
38 #include <cf.h>
39
40 #define PAT_REST 0
41 #define PAT_URL 1
42 #define PAT_METHOD 2
43 #define PAT_HEADER 3
44 #define PAT_ALL 4
45
46 #define PATFL_MSS 1
47 #define PATFL_UNQ 2
48
49 #define HND_CHILD 1
50 #define HND_REPARSE 2
51
52 struct config {
53     struct child *children;
54     struct pattern *patterns;
55 };
56
57 struct rule {
58     int type;
59     int fl;
60     char *header;
61     regex_t *pattern;
62 };
63
64 struct headmod {
65     struct headmod *next;
66     char *name, *value;
67 };
68
69 struct pattern {
70     struct pattern *next;
71     struct headmod *headers;
72     char *childnm;
73     struct rule **rules;
74     char *restpat;
75     int handler, prio, disable;
76 };
77
78 static struct config *gconfig, *lconfig;
79 static volatile int reload = 0;
80
81 static void freepattern(struct pattern *pat)
82 {
83     struct rule **rule;
84     struct headmod *head;
85     
86     for(rule = pat->rules; *rule; rule++) {
87         if((*rule)->header != NULL)
88             free((*rule)->header);
89         if((*rule)->pattern != NULL) {
90             regfree((*rule)->pattern);
91             free((*rule)->pattern);
92         }
93         free(*rule);
94     }
95     while((head = pat->headers) != NULL) {
96         pat->headers = head->next;
97         free(head->name);
98         free(head->value);
99         free(head);
100     }
101     if(pat->childnm != NULL)
102         free(pat->childnm);
103     free(pat);
104 }
105
106 static void freeconfig(struct config *cf)
107 {
108     struct child *ch, *nch;
109     struct pattern *pat, *npat;
110     
111     for(ch = cf->children; ch != NULL; ch = nch) {
112         nch = ch->next;
113         freechild(ch);
114     }
115     for(pat = cf->patterns; pat != NULL; pat = npat) {
116         npat = pat->next;
117         freepattern(pat);
118     }
119     free(cf);
120 }
121
122 static struct child *getchild(struct config *cf, char *name)
123 {
124     struct child *ch;
125     
126     for(ch = cf->children; ch; ch = ch->next) {
127         if(!strcmp(ch->name, name))
128             break;
129     }
130     return(ch);
131 }
132
133 static struct rule *newrule(struct pattern *pat)
134 {
135     int i;
136     struct rule *rule;
137     
138     for(i = 0; pat->rules[i]; i++);
139     pat->rules = srealloc(pat->rules, sizeof(*pat->rules) * (i + 2));
140     rule = pat->rules[i] = szmalloc(sizeof(*rule));
141     pat->rules[i + 1] = NULL;
142     return(rule);
143 }
144
145 static struct pattern *newpattern(void)
146 {
147     struct pattern *pat;
148     
149     omalloc(pat);
150     pat->rules = szmalloc(sizeof(*pat->rules));
151     return(pat);
152 }
153
154 static regex_t *regalloc(char *regex, int flags)
155 {
156     regex_t *ret;
157     int res;
158     char errbuf[256];
159     
160     omalloc(ret);
161     if((res = regcomp(ret, regex, flags | REG_EXTENDED)) != 0) {
162         regerror(res, ret, errbuf, sizeof(errbuf));
163         flog(LOG_WARNING, "%s: %s", regex, errbuf);
164         free(ret);
165         return(NULL);
166     }
167     return(ret);
168 }
169
170 static struct pattern *parsepattern(struct cfstate *s)
171 {
172     struct pattern *pat;
173     int sl;
174     struct rule *rule;
175     struct headmod *head;
176     regex_t *regex;
177     int rxfl;
178     
179     if(!strcmp(s->argv[0], "match")) {
180         s->expstart = 1;
181         pat = newpattern();
182     } else {
183         return(NULL);
184     }
185     
186     sl = s->lno;
187     while(1) {
188         getcfline(s);
189         if(!strcmp(s->argv[0], "point") ||
190            !strcmp(s->argv[0], "url") ||
191            !strcmp(s->argv[0], "method")) {
192             if(s->argc < 2) {
193                 flog(LOG_WARNING, "%s:%i: missing pattern for `%s' match", s->file, s->lno, s->argv[0]);
194                 continue;
195             }
196             rxfl = 0;
197             if(s->argc >= 3) {
198                 if(strchr(s->argv[2], 'i'))
199                     rxfl |= REG_ICASE;
200             }
201             if((regex = regalloc(s->argv[1], rxfl)) == NULL) {
202                 flog(LOG_WARNING, "%s:%i: invalid regex for `%s' match", s->file, s->lno, s->argv[0]);
203                 continue;
204             }
205             rule = newrule(pat);
206             if(!strcmp(s->argv[0], "point"))
207                 rule->type = PAT_REST;
208             else if(!strcmp(s->argv[0], "url"))
209                 rule->type = PAT_URL;
210             else if(!strcmp(s->argv[0], "method"))
211                 rule->type = PAT_METHOD;
212             rule->pattern = regex;
213             if(s->argc >= 3) {
214                 if(strchr(s->argv[2], 's'))
215                     rule->fl |= PATFL_MSS;
216                 if(strchr(s->argv[2], 'q'))
217                     rule->fl |= PATFL_UNQ;
218             }
219         } else if(!strcmp(s->argv[0], "header")) {
220             if(s->argc < 3) {
221                 flog(LOG_WARNING, "%s:%i: missing header name or pattern for `header' match", s->file, s->lno);
222                 continue;
223             }
224             rxfl = 0;
225             if(s->argc >= 4) {
226                 if(strchr(s->argv[3], 'i'))
227                     rxfl |= REG_ICASE;
228             }
229             if((regex = regalloc(s->argv[2], rxfl)) == NULL) {
230                 flog(LOG_WARNING, "%s:%i: invalid regex for `header' match", s->file, s->lno);
231                 continue;
232             }
233             rule = newrule(pat);
234             rule->type = PAT_HEADER;
235             rule->header = sstrdup(s->argv[1]);
236             rule->pattern = regex;
237             if(s->argc >= 4) {
238                 if(strchr(s->argv[3], 's'))
239                     rule->fl |= PATFL_MSS;
240             }
241         } else if(!strcmp(s->argv[0], "all")) {
242             newrule(pat)->type = PAT_ALL;
243         } else if(!strcmp(s->argv[0], "default")) {
244             newrule(pat)->type = PAT_ALL;
245             pat->prio = -10;
246         } else if(!strcmp(s->argv[0], "order") || !strcmp(s->argv[0], "priority")) {
247             if(s->argc < 2) {
248                 flog(LOG_WARNING, "%s:%i: missing specification for `%s' directive", s->file, s->lno, s->argv[0]);
249                 continue;
250             }
251             pat->prio = atoi(s->argv[1]);
252         } else if(!strcmp(s->argv[0], "handler")) {
253             if(s->argc < 2) {
254                 flog(LOG_WARNING, "%s:%i: missing child name for `handler' directive", s->file, s->lno);
255                 continue;
256             }
257             if(pat->childnm != NULL)
258                 free(pat->childnm);
259             pat->childnm = sstrdup(s->argv[1]);
260             pat->handler = HND_CHILD;
261         } else if(!strcmp(s->argv[0], "reparse")) {
262             pat->handler = HND_REPARSE;
263         } else if(!strcmp(s->argv[0], "restpat")) {
264             if(s->argc < 2) {
265                 flog(LOG_WARNING, "%s:%i: missing pattern for `restpat' directive", s->file, s->lno);
266                 continue;
267             }
268             if(pat->restpat != NULL)
269                 free(pat->restpat);
270             pat->restpat = sstrdup(s->argv[1]);
271         } else if(!strcmp(s->argv[0], "set") || !strcmp(s->argv[0], "xset")) {
272             if(s->argc < 3) {
273                 flog(LOG_WARNING, "%s:%i: missing header name or pattern for `%s' directive", s->file, s->lno, s->argv[0]);
274                 continue;
275             }
276             omalloc(head);
277             if(!strcmp(s->argv[0], "xset"))
278                 head->name = sprintf2("X-Ash-%s", s->argv[1]);
279             else
280                 head->name = sstrdup(s->argv[1]);
281             head->value = sstrdup(s->argv[2]);
282             head->next = pat->headers;
283             pat->headers = head;
284         } else if(!strcmp(s->argv[0], "end") || !strcmp(s->argv[0], "eof")) {
285             break;
286         } else {
287             flog(LOG_WARNING, "%s:%i: unknown directive `%s' in pattern declaration", s->file, s->lno, s->argv[0]);
288         }
289     }
290     
291     if(pat->rules[0] == NULL) {
292         flog(LOG_WARNING, "%s:%i: missing rules in match declaration", s->file, sl);
293         freepattern(pat);
294         return(NULL);
295     }
296     if(pat->handler == 0) {
297         flog(LOG_WARNING, "%s:%i: missing handler in match declaration", s->file, sl);
298         freepattern(pat);
299         return(NULL);
300     }
301     return(pat);
302 }
303
304 static struct config *readconfig(char *filename)
305 {
306     struct cfstate *s;
307     struct config *cf;
308     struct child *ch;
309     struct pattern *pat;
310     FILE *in;
311     
312     if((in = fopen(filename, "r")) == NULL) {
313         flog(LOG_WARNING, "%s: %s", filename, strerror(errno));
314         return(NULL);
315     }
316     s = mkcfparser(in, filename);
317     omalloc(cf);
318     
319     while(1) {
320         getcfline(s);
321         if((ch = parsechild(s)) != NULL) {
322             ch->next = cf->children;
323             cf->children = ch;
324         } else if((pat = parsepattern(s)) != NULL) {
325             pat->next = cf->patterns;
326             cf->patterns = pat;
327         } else if(!strcmp(s->argv[0], "eof")) {
328             break;
329         } else {
330             flog(LOG_WARNING, "%s:%i: unknown directive `%s'", s->file, s->lno, s->argv[0]);
331         }
332     }
333     
334     freecfparser(s);
335     fclose(in);
336     return(cf);
337 }
338
339 static void exprestpat(struct hthead *req, struct pattern *pat, char **mstr)
340 {
341     char *p, *p2, *hdr;
342     int mc;
343     struct charbuf buf;
344     
345     if(mstr == NULL)
346         mc = 0;
347     else
348         for(mc = 0; mstr[mc]; mc++);
349     bufinit(buf);
350     for(p = pat->restpat; *p; ) {
351         if(*p == '$') {
352             p++;
353             if((*p >= '0') && (*p <= '9')) {
354                 if(*p - '0' < mc)
355                     bufcatstr(buf, mstr[*p - '0']);
356                 p++;
357             } else if(*p == '_') {
358                 bufcatstr(buf, req->rest);
359                 p++;
360             } else if(*p == '$') {
361                 bufadd(buf, '$');
362                 p++;
363             } else if(*p == '{') {
364                 if((p2 = strchr(p, '}')) == NULL) {
365                     p++;
366                 } else {
367                     hdr = getheader(req, sprintf3("%.*s", p2 - p - 1, p + 1));
368                     if(hdr)
369                         bufcatstr(buf, hdr);
370                 }
371             } else if(!*p) {
372             }
373         } else {
374             bufadd(buf, *(p++));
375         }
376     }
377     bufadd(buf, 0);
378     replrest(req, buf.b);
379     buffree(buf);
380 }
381
382 static void qoffsets(char *buf, int *obuf, char *pstr, int unquote)
383 {
384     int i, o, d1, d2;
385     
386     if(unquote) {
387         i = o = 0;
388         while(pstr[i]) {
389             obuf[o] = i;
390             if((pstr[i] == '%') && ((d1 = hexdigit(pstr[i + 1])) >= 0) && ((d2 = hexdigit(pstr[i + 2])) >= 0)) {
391                 buf[o] = (d1 << 4) | d2;
392                 i += 3;
393             } else {
394                 buf[o] = pstr[i];
395                 i++;
396             }
397             o++;
398         }
399         buf[o] = 0;
400         obuf[o] = i;
401     } else {
402         for(i = 0; pstr[i]; i++) {
403             buf[i] = pstr[i];
404             obuf[i] = i;
405         }
406         buf[i] = 0;
407         obuf[i] = i;
408     }
409 }
410
411 struct match {
412     struct pattern *pat;
413     char **mstr;
414     int rmo;
415 };
416
417 static void freematch(struct match *match)
418 {
419     freeca(match->mstr);
420     free(match);
421 }
422
423 static struct match *findmatch(struct config *cf, struct hthead *req, struct match *match)
424 {
425     int i, o;
426     struct pattern *pat;
427     struct rule *rule;
428     int rmo;
429     regex_t *rx;
430     char *pstr;
431     char **mstr;
432     regmatch_t gr[10];
433     
434     mstr = NULL;
435     for(pat = cf->patterns; pat != NULL; pat = pat->next) {
436         if(pat->disable || (match && (pat->prio <= match->pat->prio)))
437             continue;
438         rmo = -1;
439         for(i = 0; (rule = pat->rules[i]) != NULL; i++) {
440             rx = NULL;
441             if(rule->type == PAT_REST) {
442                 rx = rule->pattern;
443                 pstr = req->rest;
444             } else if(rule->type == PAT_URL) {
445                 rx = rule->pattern;
446                 pstr = req->url;
447             } else if(rule->type == PAT_METHOD) {
448                 rx = rule->pattern;
449                 pstr = req->method;
450             } else if(rule->type == PAT_HEADER) {
451                 rx = rule->pattern;
452                 if(!(pstr = getheader(req, rule->header)))
453                     break;
454             }
455             if(rx != NULL) {
456                 char pbuf[strlen(pstr) + 1];
457                 int  obuf[strlen(pstr) + 1];
458                 qoffsets(pbuf, obuf, pstr, !!(rule->fl & PATFL_UNQ));
459                 if(regexec(rx, pbuf, 10, gr, 0))
460                     break;
461                 else if(rule->type == PAT_REST)
462                     rmo = obuf[gr[0].rm_eo];
463                 if(rule->fl & PATFL_MSS) {
464                     if(mstr) {
465                         flog(LOG_WARNING, "two pattern rules marked with `s' flag found (for handler %s)", pat->childnm);
466                         freeca(mstr);
467                     }
468                     for(o = 0; o < 10; o++) {
469                         if(gr[o].rm_so < 0)
470                             break;
471                     }
472                     mstr = szmalloc((o + 1) * sizeof(*mstr));
473                     for(o = 0; o < 10; o++) {
474                         if(gr[o].rm_so < 0)
475                             break;
476                         mstr[o] = smalloc(obuf[gr[o].rm_eo] - obuf[gr[o].rm_so] + 1);
477                         memcpy(mstr[o], pstr + obuf[gr[o].rm_so], obuf[gr[o].rm_eo] - obuf[gr[o].rm_so]);
478                         mstr[o][obuf[gr[o].rm_eo] - obuf[gr[o].rm_so]] = 0;
479                     }
480                 }
481             } else if(rule->type == PAT_ALL) {
482             }
483         }
484         if(!rule) {
485             if(match)
486                 freematch(match);
487             omalloc(match);
488             match->pat = pat;
489             match->mstr = mstr;
490             match->rmo = rmo;
491         }
492     }
493     return(match);
494 }
495
496 static void execmatch(struct hthead *req, struct match *match)
497 {
498     struct headmod *head;
499     
500     if(match->pat->restpat)
501         exprestpat(req, match->pat, match->mstr);
502     else if(match->rmo != -1)
503         replrest(req, req->rest + match->rmo);
504     for(head = match->pat->headers; head != NULL; head = head->next) {
505         headrmheader(req, head->name);
506         headappheader(req, head->name, head->value);
507     }
508 }
509
510 static void childerror(struct hthead *req, int fd)
511 {
512     if(errno == EAGAIN)
513         simpleerror(fd, 500, "Server Error", "The request handler is overloaded.");
514     else
515         simpleerror(fd, 500, "Server Error", "The request handler crashed.");
516 }
517
518 static void serve(struct hthead *req, int fd)
519 {
520     struct match *match;
521     struct child *ch;
522     
523     match = NULL;
524     match = findmatch(lconfig, req, match);
525     if(gconfig != NULL)
526         match = findmatch(gconfig, req, match);
527     if(match == NULL) {
528         simpleerror(fd, 404, "Not Found", "The requested resource could not be found on this server.");
529         return;
530     }
531     execmatch(req, match);
532     switch(match->pat->handler) {
533     case HND_CHILD:
534         ch = NULL;
535         if(ch == NULL)
536             ch = getchild(lconfig, match->pat->childnm);
537         if((ch == NULL) && (gconfig != NULL))
538             ch = getchild(gconfig, match->pat->childnm);
539         if(ch == NULL) {
540             flog(LOG_ERR, "child %s requested, but was not declared", match->pat->childnm);
541             simpleerror(fd, 500, "Configuration Error", "The server is erroneously configured. Handler %s was requested, but not declared.", match->pat->childnm);
542             break;
543         }
544         if(childhandle(ch, req, fd, NULL, NULL))
545             childerror(req, fd);
546         break;
547     case HND_REPARSE:
548         match->pat->disable = 1;
549         serve(req, fd);
550         match->pat->disable = 0;
551         break;
552     default:
553         abort();
554     }
555     freematch(match);
556 }
557
558 static void reloadconf(char *nm)
559 {
560     struct config *cf;
561     
562     if((cf = readconfig(nm)) == NULL) {
563         flog(LOG_WARNING, "could not reload configuration file `%s'", nm);
564         return;
565     }
566     mergechildren(cf->children, lconfig->children);
567     freeconfig(lconfig);
568     lconfig = cf;
569 }
570
571 static void chldhandler(int sig)
572 {
573     pid_t pid;
574     int st;
575     
576     while((pid = waitpid(-1, &st, WNOHANG)) > 0) {
577         if(WCOREDUMP(st))
578             flog(LOG_WARNING, "child process %i dumped core", pid);
579     }
580 }
581
582 static void sighandler(int sig)
583 {
584     if(sig == SIGHUP)
585         reload = 1;
586 }
587
588 static void usage(FILE *out)
589 {
590     fprintf(out, "usage: patplex [-hN] CONFIGFILE\n");
591 }
592
593 int main(int argc, char **argv)
594 {
595     int c;
596     int nodef;
597     char *gcf, *lcf;
598     struct hthead *req;
599     int fd;
600     
601     nodef = 0;
602     while((c = getopt(argc, argv, "hN")) >= 0) {
603         switch(c) {
604         case 'h':
605             usage(stdout);
606             exit(0);
607         case 'N':
608             nodef = 1;
609             break;
610         default:
611             usage(stderr);
612             exit(1);
613         }
614     }
615     if(argc - optind < 1) {
616         usage(stderr);
617         exit(1);
618     }
619     if(!nodef) {
620         if((gcf = findstdconf("ashd/patplex.rc")) != NULL) {
621             gconfig = readconfig(gcf);
622             free(gcf);
623         }
624     }
625     if((strchr(lcf = argv[optind], '/')) == NULL) {
626         if((lcf = findstdconf(sprintf3("ashd/%s", lcf))) == NULL) {
627             flog(LOG_ERR, "could not find requested configuration file `%s'", argv[optind]);
628             exit(1);
629         }
630     }
631     if((lconfig = readconfig(lcf)) == NULL) {
632         flog(LOG_ERR, "could not read `%s'", lcf);
633         exit(1);
634     }
635     signal(SIGCHLD, chldhandler);
636     signal(SIGHUP, sighandler);
637     signal(SIGPIPE, sighandler);
638     while(1) {
639         if(reload) {
640             reloadconf(lcf);
641             reload = 0;
642         }
643         if((fd = recvreq(0, &req)) < 0) {
644             if(errno == EINTR)
645                 continue;
646             if(errno != 0)
647                 flog(LOG_ERR, "recvreq: %s", strerror(errno));
648             break;
649         }
650         serve(req, fd);
651         freehthead(req);
652         close(fd);
653     }
654     return(0);
655 }