htparser: Add a `Server' header if none previously present.
[ashd.git] / lib / cf.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 <ctype.h>
24 #include <glob.h>
25 #include <libgen.h>
26 #include <errno.h>
27
28 #ifdef HAVE_CONFIG_H
29 #include <config.h>
30 #endif
31 #include <utils.h>
32 #include <cf.h>
33 #include <mt.h>
34 #include <proc.h>
35 #include <log.h>
36
37 #define CH_SOCKET 0
38 #define CH_FORK 1
39
40 static int parsefile(struct cfstate *s, FILE *in);
41
42 static int doinclude(struct cfstate *s, char *spec)
43 {
44     int rv, i;
45     FILE *inc;
46     glob_t globm;
47     char *fbk, *dir, *fspec;
48
49     rv = 0;
50     fbk = s->file;
51     if(spec[0] == '/') {
52         fspec = spec;
53     } else {
54         dir = sstrdup(fbk);
55         fspec = sprintf3("%s/%s", dirname(dir), spec);
56         free(dir);
57     }
58     if(glob(fspec, 0, NULL, &globm))
59         return(0);
60     for(i = 0; i < globm.gl_pathc; i++) {
61         if((inc = fopen(globm.gl_pathv[i], "r")) != NULL) {
62             s->file = globm.gl_pathv[i];
63             if(parsefile(s, inc)) {
64                 fclose(inc);
65                 rv = 1;
66                 goto out;
67             }
68             fclose(inc);
69             inc = NULL;
70         }
71     }
72     
73 out:
74     globfree(&globm);
75     s->file = fbk;
76     return(rv);
77 }
78
79 static int parsefile(struct cfstate *s, FILE *in)
80 {
81     int i;
82     char line[1024];
83     int eof, argc;
84     int ind, indst[80], indl;
85     char *p, **w;
86     
87     s->lno = 0;
88     indst[indl = 0] = 0;
89     eof = 0;
90     while(1) {
91         if(fgets(line, sizeof(line), in) == NULL) {
92             eof = 1;
93             line[0] = 0;
94         }
95         s->lno++;
96         if(line[0]) {
97             for(p = line + strlen(line) - 1; p >= line; p--) {
98                 if(isspace(*p))
99                     *p = 0;
100                 else
101                     break;
102             }
103         }
104         for(ind = 0, p = line; *p; p++) {
105             if(*p == ' ') {
106                 ind++;
107             } else if(*p == '\t') {
108                 ind = ind - (ind % 8) + 8;
109             } else {
110                 break;
111             }
112         }
113         if(!eof && (!*p || (*p == '#')))
114             continue;
115         
116     reindent:
117         if(ind > indst[indl]) {
118             indst[++indl] = ind;
119             if(!s->expstart) {
120                 s->res = tokenize("start");
121                 if(yield())
122                     return(1);
123             } else {
124                 s->expstart = 0;
125             }
126         } else {
127             if(s->expstart) {
128                 s->res = tokenize("end");
129                 if(yield())
130                     return(1);
131                 s->expstart = 0;
132             }
133             while(ind < indst[indl]) {
134                 indl--;
135                 s->res = tokenize("end");
136                 if(yield())
137                     return(1);
138             }
139             if(ind > indst[indl]) {
140                 flog(LOG_WARNING, "%s:%i: unexpected indentation level", s->file, s->lno);
141                 goto reindent;
142             }
143         }
144         
145         if(eof)
146             return(0);
147         
148         argc = calen(w = tokenize(line));
149         if(argc < 1) {
150             /* Shouldn't happen, but... */
151             freeca(w);
152             continue;
153         }
154         
155         if(indl == 0) {
156             if(!strcmp(w[0], "include")) {
157                 for(i = 1; i < argc; i++) {
158                     if(doinclude(s, w[i])) {
159                         freeca(w);
160                         return(1);
161                     }
162                 }
163                 freeca(w);
164                 continue;
165             }
166         }
167         
168         if(!strcmp(w[0], "start") ||
169            !strcmp(w[0], "end") || 
170            !strcmp(w[0], "eof")) {
171             flog(LOG_WARNING, "%s:%i: illegal directive: %s", s->file, s->lno, w[0]);
172         } else {
173             s->res = w;
174             if(yield())
175                 return(1);
176         }
177     }
178 }
179
180 static void parsefn(struct muth *mt, va_list args)
181 {
182     vavar(struct cfstate *, s);
183     vavar(FILE *, in);
184     vavar(char *, file);
185     
186     s->file = sstrdup(file);
187     if(parsefile(s, in))
188         goto out;
189     do {
190         s->res = tokenize("eof");
191     } while(!yield());
192     
193 out:
194     free(s->file);
195 }
196
197 char **getcfline(struct cfstate *s)
198 {
199     freeca(s->argv);
200     if(s->res == NULL)
201         resume(s->pf, 0);
202     s->argc = calen(s->argv = s->res);
203     s->res = NULL;
204     return(s->argv);
205 }
206
207 struct cfstate *mkcfparser(FILE *in, char *name)
208 {
209     struct cfstate *s;
210     
211     omalloc(s);
212     s->pf = mustart(parsefn, s, in, name);
213     return(s);
214 }
215
216 void freecfparser(struct cfstate *s)
217 {
218     resume(s->pf, -1);
219     freeca(s->argv);
220     freeca(s->res);
221     free(s);
222 }
223
224 char *findstdconf(char *name)
225 {
226     char *path, *p, *p2, *t;
227     
228     if((path = getenv("PATH")) == NULL)
229         return(NULL);
230     path = sstrdup(path);
231     for(p = strtok(path, ":"); p != NULL; p = strtok(NULL, ":")) {
232         if((p2 = strrchr(p, '/')) == NULL)
233             continue;
234         *p2 = 0;
235         if(!access(t = sprintf2("%s/etc/%s", p, name), R_OK)) {
236             free(path);
237             return(t);
238         }
239         free(t);
240     }
241     free(path);
242     return(NULL);
243 }
244
245 static struct child *newchild(char *name, int type)
246 {
247     struct child *ch;
248     
249     omalloc(ch);
250     ch->name = sstrdup(name);
251     ch->type = type;
252     ch->fd = -1;
253     return(ch);
254 }
255
256 void freechild(struct child *ch)
257 {
258     if(ch->fd != -1)
259         close(ch->fd);
260     if(ch->name != NULL)
261         free(ch->name);
262     if(ch->argv != NULL)
263         freeca(ch->argv);
264     free(ch);
265 }
266
267 void skipcfblock(struct cfstate *s)
268 {
269     char **w;
270     
271     while(1) {
272         w = getcfline(s);
273         if(!strcmp(w[0], "end") || !strcmp(w[0], "eof"))
274             return;
275     }
276 }
277
278 struct child *parsechild(struct cfstate *s)
279 {
280     struct child *ch;
281     int i;
282     int sl;
283     
284     sl = s->lno;
285     if(!strcmp(s->argv[0], "child")) {
286         s->expstart = 1;
287         if(s->argc < 2) {
288             flog(LOG_WARNING, "%s:%i: missing name in child declaration", s->file, s->lno);
289             skipcfblock(s);
290             return(NULL);
291         }
292         ch = newchild(s->argv[1], CH_SOCKET);
293     } else if(!strcmp(s->argv[0], "fchild")) {
294         s->expstart = 1;
295         if(s->argc < 2) {
296             flog(LOG_WARNING, "%s:%i: missing name in child declaration", s->file, s->lno);
297             skipcfblock(s);
298             return(NULL);
299         }
300         ch = newchild(s->argv[1], CH_FORK);
301     } else {
302         return(NULL);
303     }
304     
305     while(1) {
306         getcfline(s);
307         if(!strcmp(s->argv[0], "exec")) {
308             if(s->argc < 2) {
309                 flog(LOG_WARNING, "%s:%i: too few parameters to `exec'", s->file, s->lno);
310                 continue;
311             }
312             ch->argv = szmalloc(sizeof(*ch->argv) * s->argc);
313             for(i = 0; i < s->argc - 1; i++)
314                 ch->argv[i] = sstrdup(s->argv[i + 1]);
315         } else if(!strcmp(s->argv[0], "end") || !strcmp(s->argv[0], "eof")) {
316             break;
317         } else {
318             flog(LOG_WARNING, "%s:%i: unknown directive `%s' in child declaration", s->file, s->lno, s->argv[0]);
319         }
320     }
321     if(ch->argv == NULL) {
322         flog(LOG_WARNING, "%s:%i: missing `exec' in child declaration %s", s->file, sl, ch->name);
323         freechild(ch);
324         return(NULL);
325     }
326     return(ch);
327 }
328
329 int childhandle(struct child *ch, struct hthead *req, int fd, void (*chinit)(void *), void *idata)
330 {
331     if(ch->type == CH_SOCKET) {
332         if(ch->fd < 0)
333             ch->fd = stdmkchild(ch->argv, chinit, idata);
334         if(sendreq(ch->fd, req, fd)) {
335             if(errno == EPIPE) {
336                 /* Assume that the child has crashed and restart it. */
337                 close(ch->fd);
338                 ch->fd = stdmkchild(ch->argv, chinit, idata);
339                 if(!sendreq(ch->fd, req, fd))
340                     return(0);
341             }
342             flog(LOG_ERR, "could not pass on request to child %s: %s", ch->name, strerror(errno));
343             close(ch->fd);
344             ch->fd = -1;
345             return(-1);
346         }
347     } else if(ch->type == CH_FORK) {
348         if(stdforkserve(ch->argv, req, fd, chinit, idata) < 0)
349             return(-1);
350     }
351     return(0);
352 }