r6986: added support for <% include("somefile.ejs") %> for including common scripts
[samba.git] / source / web_server / http.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    http handling code
5
6    Copyright (C) Andrew Tridgell 2005
7    
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #include "includes.h"
24 #include "web_server/web_server.h"
25 #include "smbd/service_stream.h"
26 #include "lib/events/events.h"
27 #include "system/filesys.h"
28 #include "system/iconv.h"
29 #include "system/time.h"
30 #include "web_server/esp/esp.h"
31
32 /* state of the esp subsystem */
33 struct esp_state {
34         struct websrv_context *web;
35         struct MprVar variables[ESP_OBJ_MAX];
36         struct EspRequest *req;
37 };
38
39 /* destroy a esp session */
40 static int esp_destructor(void *ptr)
41 {
42         struct esp_state *esp = talloc_get_type(ptr, struct esp_state);
43         if (esp->req) {
44                 espDestroyRequest(esp->req);
45         }
46         espClose();
47         mprFreeAll();
48         return 0;
49 }
50
51 /*
52   output the http headers
53 */
54 static void http_output_headers(struct websrv_context *web)
55 {
56         int i;
57         char *s;
58         DATA_BLOB b;
59         const char *response_string = "Unknown Code";
60         const struct {
61                 unsigned code;
62                 const char *response_string;
63         } codes[] = {
64                 { 200, "OK" },
65                 { 301, "Moved" },
66                 { 302, "Found" },
67                 { 303, "Method" },
68                 { 304, "Not Modified" },
69                 { 400, "Bad request" },
70                 { 401, "Unauthorized" },
71                 { 403, "Forbidden" },
72                 { 404, "Not Found" },
73                 { 500, "Internal Server Error" },
74                 { 501, "Not implemented" }
75         };
76         for (i=0;i<ARRAY_SIZE(codes);i++) {
77                 if (codes[i].code == web->output.response_code) {
78                         response_string = codes[i].response_string;
79                 }
80         }
81
82         if (web->output.headers == NULL) return;
83         s = talloc_asprintf(web, "HTTP/1.0 %u %s\r\n", 
84                             web->output.response_code, response_string);
85         if (s == NULL) return;
86         for (i=0;web->output.headers[i];i++) {
87                 s = talloc_asprintf_append(s, "%s\r\n", web->output.headers[i]);
88         }
89         s = talloc_asprintf_append(s, "\r\n");
90         if (s == NULL) return;
91
92         b = web->output.content;
93         web->output.content.data = s;
94         web->output.content.length = strlen(s);
95         data_blob_append(web, &web->output.content, b.data, b.length);
96         data_blob_free(&b);
97 }
98
99 /*
100   called when esp wants to read a file to support include() calls
101 */
102 static int http_readFile(EspHandle handle, char **buf, int *len, char *path)
103 {
104         int fd = -1;
105         struct stat st;
106         *buf = NULL;
107
108         fd = open(path, O_RDONLY);
109         if (fd == -1 || fstat(fd, &st) != 0 || !S_ISREG(st.st_mode)) goto failed;
110
111         *buf = talloc_size(handle, st.st_size+1);
112         if (*buf == NULL) goto failed;
113
114         if (read(fd, *buf, st.st_size) != st.st_size) goto failed;
115
116         (*buf)[st.st_size] = 0;
117
118         close(fd);
119         *len = st.st_size;
120         return 0;
121
122 failed:
123         if (fd != -1) close(fd);
124         talloc_free(*buf);
125         *buf = NULL;
126         return -1;
127 }
128
129 /*
130   called when esp wants to output something
131 */
132 static int http_writeBlock(EspHandle handle, char *buf, int size)
133 {
134         struct websrv_context *web = talloc_get_type(handle, struct websrv_context);
135         NTSTATUS status;
136         status = data_blob_append(web, &web->output.content, buf, size);
137         if (!NT_STATUS_IS_OK(status)) return -1;
138         return size;
139 }
140
141
142 /*
143   set a http header
144 */
145 static void http_setHeader(EspHandle handle, const char *value, bool allowMultiple)
146 {
147         struct websrv_context *web = talloc_get_type(handle, struct websrv_context);
148         char *p = strchr(value, ':');
149
150         if (p && !allowMultiple && web->output.headers) {
151                 int i;
152                 for (i=0;web->output.headers[i];i++) {
153                         if (strncmp(web->output.headers[i], value, (p+1)-value) == 0) {
154                                 web->output.headers[i] = talloc_strdup(web, value);
155                                 return;
156                         }
157                 }
158         }
159
160         web->output.headers = str_list_add(web->output.headers, value);
161 }
162
163 /*
164   set a http response code
165 */
166 static void http_setResponseCode(EspHandle handle, int code)
167 {
168         struct websrv_context *web = talloc_get_type(handle, struct websrv_context);
169         web->output.response_code = code;
170 }
171
172 /*
173   redirect to another web page
174  */
175 static void http_redirect(EspHandle handle, int code, char *url)
176 {
177         struct websrv_context *web = talloc_get_type(handle, struct websrv_context);
178         const char *host = web->input.host;
179         
180         /* form the full url, unless it already looks like a url */
181         if (strchr(url, ':') == NULL) {
182                 if (host == NULL) {
183                         host = talloc_asprintf(web, "%s:%u",
184                                                socket_get_my_addr(web->conn->socket, web),
185                                                socket_get_my_port(web->conn->socket));
186                 }
187                 if (host == NULL) goto internal_error;
188                 if (url[0] != '/') {
189                         char *p = strrchr(web->input.url, '/');
190                         if (p == web->input.url) {
191                                 url = talloc_asprintf(web, "http://%s/%s", host, url);
192                         } else {
193                                 int dirlen = p - web->input.url;
194                                 url = talloc_asprintf(web, "http://%s%*.*s/%s",
195                                                       host, 
196                                                       dirlen, dirlen, web->input.url,
197                                                       url);
198                         }
199                         if (url == NULL) goto internal_error;
200                 }
201         }
202
203         http_setHeader(handle, talloc_asprintf(web, "Location: %s", url), 0);
204
205         /* make sure we give a valid redirect code */
206         if (code >= 300 && code < 400) {
207                 http_setResponseCode(handle, code);
208         } else {
209                 http_setResponseCode(handle, 302);
210         }
211         return;
212
213 internal_error:
214         http_error(web, 500, "Internal server error");
215 }
216
217
218 /* callbacks for esp processing */
219 static const struct Esp esp_control = {
220         .maxScriptSize   = 60000,
221         .writeBlock      = http_writeBlock,
222         .setHeader       = http_setHeader,
223         .redirect        = http_redirect,
224         .setResponseCode = http_setResponseCode,
225         .readFile        = http_readFile
226 };
227
228
229 /*
230   setup for a raw http level error
231 */
232 void http_error(struct websrv_context *web, int code, const char *info)
233 {
234         char *s;
235         s = talloc_asprintf(web,"<HTML><HEAD><TITLE>Error %u</TITLE></HEAD><BODY><H1>Error %u</H1>%s<p></BODY></HTML>\r\n\r\n", 
236                             code, code, info);
237         if (s == NULL) {
238                 stream_terminate_connection(web->conn, "http_error: out of memory");
239                 return;
240         }
241         http_writeBlock(web, s, strlen(s));
242         http_setResponseCode(web, code);
243         http_output_headers(web);
244         EVENT_FD_NOT_READABLE(web->conn->event.fde);
245         EVENT_FD_WRITEABLE(web->conn->event.fde);
246 }
247
248 /*
249   map a unix error code to a http error
250 */
251 void http_error_unix(struct websrv_context *web, const char *info)
252 {
253         int code = 500;
254         switch (errno) {
255         case ENOENT:
256         case EISDIR:
257                 code = 404;
258                 break;
259         case EACCES:
260                 code = 403;
261                 break;
262         }
263         http_error(web, code, info);
264 }
265
266 /*
267   return the local path for a URL
268 */
269 static const char *http_local_path(struct websrv_context *web, const char *url)
270 {
271         int i;
272         char *path;
273
274         /* check that the url is OK */
275         if (url[0] != '/') return NULL;
276
277         for (i=0;url[i];i++) {
278                 if ((!isalnum(url[i]) && !strchr("./", url[i])) ||
279                     (url[i] == '.' && strchr("/.", url[i+1]))) {
280                         return NULL;
281                 }
282         }
283
284         path = talloc_asprintf(web, "%s/%s", lp_swat_directory(), url+1);
285         if (path == NULL) return NULL;
286
287         if (directory_exist(path)) {
288                 path = talloc_asprintf_append(path, "/index.html");
289         }
290         return path;
291 }
292
293
294 /*
295   a simple file request
296 */
297 static void http_simple_request(struct websrv_context *web)
298 {
299         const char *url = web->input.url;
300         const char *path;
301         struct stat st;
302
303         path = http_local_path(web, url);
304         if (path == NULL) goto invalid;
305
306         /* looks ok */
307         web->output.fd = open(path, O_RDONLY);
308         if (web->output.fd == -1) {
309                 http_error_unix(web, path);
310                 return;
311         }
312
313         if (fstat(web->output.fd, &st) != 0 || !S_ISREG(st.st_mode)) {
314                 close(web->output.fd);
315                 goto invalid;
316         }
317
318         http_output_headers(web);
319         EVENT_FD_WRITEABLE(web->conn->event.fde);
320         return;
321
322 invalid:
323         http_error(web, 400, "Malformed URL");
324 }
325
326 /*
327   setup the standard ESP arrays
328 */
329 static void http_setup_arrays(struct esp_state *esp)
330 {
331         struct websrv_context *web = esp->web;
332         struct EspRequest *req = esp->req;
333         char *p;
334
335         espSetStringVar(req, ESP_REQUEST_OBJ, "CONTENT_LENGTH", 
336                         talloc_asprintf(esp, "%u", web->input.content_length));
337         if (web->input.query_string) {
338                 espSetStringVar(req, ESP_REQUEST_OBJ, "QUERY_STRING", 
339                                 web->input.query_string);
340         }
341         espSetStringVar(req, ESP_REQUEST_OBJ, "REQUEST_METHOD", 
342                         web->input.post_request?"POST":"GET");
343         espSetStringVar(req, ESP_REQUEST_OBJ, "REQUEST_URI", web->input.url);
344         p = strrchr(web->input.url, '/');
345         espSetStringVar(req, ESP_REQUEST_OBJ, "SCRIPT_NAME", p+1);
346
347         if (web->input.referer) {
348                 espSetStringVar(req, ESP_HEADERS_OBJ, "HTT_REFERER", web->input.referer);
349         }
350         if (web->input.user_agent) {
351                 espSetStringVar(req, ESP_HEADERS_OBJ, "USER_AGENT", web->input.user_agent);
352         }
353
354         espSetStringVar(req, ESP_SERVER_OBJ, "SERVER_ADDR", 
355                         socket_get_my_addr(web->conn->socket, esp));
356         espSetStringVar(req, ESP_SERVER_OBJ, "SERVER_PORT", 
357                         talloc_asprintf(esp, "%u", socket_get_my_port(web->conn->socket)));
358         espSetStringVar(req, ESP_SERVER_OBJ, "SERVER_PROTOCOL", "http");
359 }
360
361
362
363
364
365 /*
366   process a esp request
367 */
368 static void esp_request(struct esp_state *esp)
369 {
370         struct websrv_context *web = esp->web;
371         const char *url = web->input.url;
372         const char *path;
373         size_t size;
374         int res;
375         char *emsg = NULL, *buf;
376
377         http_setup_arrays(esp);
378
379         path = http_local_path(web, url);
380         if (path == NULL) goto invalid;
381
382         espSetStringVar(esp->req, ESP_REQUEST_OBJ, "SCRIPT_FILENAME", path);
383
384         if (http_readFile(web, &buf, &size, path) != 0) {
385                 http_error_unix(web, path);
386                 return;
387         }
388
389         res = espProcessRequest(esp->req, path, buf, &emsg);
390         if (res != 0 && emsg) {
391                 http_writeBlock(esp, emsg, strlen(emsg));
392         }
393         talloc_free(buf);
394         http_output_headers(web);
395         EVENT_FD_WRITEABLE(web->conn->event.fde);
396         return;
397
398 invalid:
399         http_error(web, 400, "Malformed URL");
400 }
401
402
403 /* 
404    handling of + and % escapes in http variables 
405 */
406 static const char *http_unescape(TALLOC_CTX *mem_ctx, const char *p)
407 {
408         char *s0 = talloc_strdup(mem_ctx, p);
409         char *s = s0;
410         if (s == NULL) return NULL;
411
412         while (*s) {
413                 unsigned v;
414                 if (*s == '+') *s = ' ';
415                 if (*s == '%' && sscanf(s+1, "%02x", &v) == 1) {
416                         *s = (char)v;
417                         memmove(s+1, s+3, strlen(s+3)+1);
418                 }
419                 s++;
420         }
421
422         return s0;
423 }
424
425 /*
426   set a form or GET variable
427 */
428 static void esp_putvar(struct esp_state *esp, const char *var, const char *value)
429 {
430         espSetStringVar(esp->req, ESP_FORM_OBJ, 
431                         http_unescape(esp, var),
432                         http_unescape(esp, value));
433 }
434
435
436 /*
437   parse the variables in a POST style request
438 */
439 static NTSTATUS http_parse_post(struct esp_state *esp)
440 {
441         DATA_BLOB b = esp->web->input.partial;
442
443         while (b.length) {
444                 char *p, *line;
445                 size_t len;
446
447                 p = memchr(b.data, '&', b.length);
448                 if (p == NULL) {
449                         len = b.length;
450                 } else {
451                         len = p - (char *)b.data;
452                 }
453                 line = talloc_strndup(esp, b.data, len);
454                 NT_STATUS_HAVE_NO_MEMORY(line);
455                                      
456                 p = strchr(line,'=');
457                 if (p) {
458                         *p = 0;
459                         esp_putvar(esp, line, p+1);
460                 }
461                 talloc_free(line);
462                 b.length -= len;
463                 b.data += len;
464                 if (b.length > 0) {
465                         b.length--;
466                         b.data++;
467                 }
468         }
469
470         return NT_STATUS_OK;
471 }
472
473 /*
474   parse the variables in a GET style request
475 */
476 static NTSTATUS http_parse_get(struct esp_state *esp)
477 {
478         struct websrv_context *web = esp->web;
479         char *p, *s, *tok;
480         char *pp;
481
482         p = strchr(web->input.url, '?');
483         web->input.query_string = p+1;
484         *p = 0;
485
486         s = talloc_strdup(esp, esp->web->input.query_string);
487         NT_STATUS_HAVE_NO_MEMORY(s);
488
489         for (tok=strtok_r(s,"&;", &pp);tok;tok=strtok_r(NULL,"&;", &pp)) {
490                 p = strchr(tok,'=');
491                 if (p) {
492                         *p = 0;
493                         esp_putvar(esp, tok, p+1);
494                 }
495         }
496         return NT_STATUS_OK;
497 }
498
499 /*
500   setup some standard variables
501 */
502 static void http_setup_vars(struct esp_state *esp)
503 {
504         int i;
505
506         for (i = 0; i < ESP_OBJ_MAX; i++) {
507                 esp->variables[i] = mprCreateUndefinedVar();
508         }
509         esp->variables[ESP_HEADERS_OBJ]     = mprCreateObjVar("headers", ESP_HASH_SIZE);
510         esp->variables[ESP_FORM_OBJ]        = mprCreateObjVar("form", ESP_HASH_SIZE);
511         esp->variables[ESP_APPLICATION_OBJ] = mprCreateObjVar("application", ESP_HASH_SIZE);
512         esp->variables[ESP_COOKIES_OBJ]     = mprCreateObjVar("cookies", ESP_HASH_SIZE);
513         esp->variables[ESP_FILES_OBJ]       = mprCreateObjVar("files", ESP_HASH_SIZE);
514         esp->variables[ESP_REQUEST_OBJ]     = mprCreateObjVar("request", ESP_HASH_SIZE);
515         esp->variables[ESP_SERVER_OBJ]      = mprCreateObjVar("server", ESP_HASH_SIZE);
516         esp->variables[ESP_SESSION_OBJ]     = mprCreateObjVar("session", ESP_HASH_SIZE);
517 }
518
519 /*
520   process a complete http request
521 */
522 void http_process_input(struct websrv_context *web)
523 {
524         NTSTATUS status;
525         struct esp_state *esp;
526         char *p;
527         int i;
528         const char *file_type = NULL;
529         const struct {
530                 const char *extension;
531                 const char *mime_type;
532         } mime_types[] = {
533                 {"gif",  "image/gif"},
534                 {"png",  "image/png"},
535                 {"jpg",  "image/jpeg"},
536                 {"txt",  "text/plain"}
537         };
538
539         esp = talloc_zero(web, struct esp_state);
540         if (esp == NULL) goto internal_error;
541
542         esp->web = web;
543
544         mprSetCtx(esp);
545
546         talloc_set_destructor(esp, esp_destructor);
547
548         if (espOpen(&esp_control) != 0) goto internal_error;
549
550         http_setup_vars(esp);
551         
552         esp->req = espCreateRequest(web, web->input.url, esp->variables);
553         if (esp->req == NULL) goto internal_error;
554
555         if (web->input.url == NULL) {
556                 http_error(web, 400, "You must specify a GET or POST request");
557                 return;
558         }
559
560         if (web->input.post_request) {
561                 status = http_parse_post(esp);
562                 if (!NT_STATUS_IS_OK(status)) {
563                         http_error(web, 400, "Malformed POST data");
564                         return;
565                 }
566         } else if (strchr(web->input.url, '?')) {
567                 status = http_parse_get(esp);
568                 if (!NT_STATUS_IS_OK(status)) {
569                         http_error(web, 400, "Malformed GET data");
570                         return;
571                 }
572         }
573
574         /* process all html files as ESP */
575         p = strrchr(web->input.url, '.');
576         for (i=0;p && i<ARRAY_SIZE(mime_types);i++) {
577                 if (strcmp(mime_types[i].extension, p+1) == 0) {
578                         file_type = mime_types[i].mime_type;
579                 }
580         }
581         if (file_type == NULL) {
582                 file_type = "text/html";
583         }
584
585         /* setup basic headers */
586         http_setResponseCode(web, 200);
587         http_setHeader(web, talloc_asprintf(esp, "Date: %s", 
588                                             http_timestring(esp, time(NULL))), 0);
589         http_setHeader(web, "Server: Samba", 0);
590         http_setHeader(web, "Connection: close", 0);
591         http_setHeader(web, talloc_asprintf(esp, "Content-Type: %s", file_type), 0);
592
593         if (strcmp(file_type, "text/html") == 0) {
594                 esp_request(esp);
595         } else {
596                 http_simple_request(web);
597         }
598         talloc_free(esp);
599         return;
600         
601 internal_error:
602         talloc_free(esp);
603         http_error(web, 500, "Internal server error");
604 }
605
606
607 /*
608   parse one line of header input
609 */
610 NTSTATUS http_parse_header(struct websrv_context *web, const char *line)
611 {
612         if (line[0] == 0) {
613                 web->input.end_of_headers = True;
614         } else if (strncasecmp(line,"GET ", 4)==0) {
615                 web->input.url = talloc_strndup(web, &line[4], strcspn(&line[4], " \t"));
616         } else if (strncasecmp(line,"POST ", 5)==0) {
617                 web->input.post_request = True;
618                 web->input.url = talloc_strndup(web, &line[5], strcspn(&line[5], " \t"));
619         } else if (strchr(line, ':') == NULL) {
620                 http_error(web, 400, "This server only accepts GET and POST requests");
621                 return NT_STATUS_INVALID_PARAMETER;
622         } else if (strncasecmp(line,"Content-Length: ", 16)==0) {
623                 web->input.content_length = strtoul(&line[16], NULL, 10);
624         } else {
625 #define PULL_HEADER(v, s) do { \
626         if (strncmp(line, s, strlen(s)) == 0) { \
627                 web->input.v = talloc_strdup(web, &line[strlen(s)]); \
628                 return NT_STATUS_OK; \
629         } \
630 } while (0)
631                 PULL_HEADER(content_type, "Content-Type: ");
632                 PULL_HEADER(user_agent, "User-Agent: ");
633                 PULL_HEADER(referer, "Referer: ");
634                 PULL_HEADER(host, "Host: ");
635                 PULL_HEADER(accept_encoding, "Accept-Encoding: ");
636         }
637
638         /* ignore all other headers for now */
639         return NT_STATUS_OK;
640 }
641
642