rev |
line source |
nuclear@12
|
1 /* tinyweb - tiny web server library and daemon
|
nuclear@12
|
2 * Author: John Tsiombikas <nuclear@member.fsf.org>
|
nuclear@12
|
3 *
|
nuclear@12
|
4 * This program is placed in the public domain. Feel free to use it any
|
nuclear@12
|
5 * way you like. Mentions and retaining this attribution header will be
|
nuclear@12
|
6 * appreciated, but not required.
|
nuclear@12
|
7 */
|
nuclear@6
|
8 #include <stdlib.h>
|
nuclear@6
|
9 #include <string.h>
|
nuclear@6
|
10 #include "mime.h"
|
nuclear@6
|
11 #include "rbtree.h"
|
nuclear@6
|
12
|
nuclear@6
|
13 /* TODO: do proper content detection */
|
nuclear@6
|
14 struct mime_type {
|
nuclear@6
|
15 const char *suffix, *type;
|
nuclear@6
|
16 };
|
nuclear@6
|
17
|
nuclear@6
|
18 static struct mime_type def_types[] = {
|
nuclear@6
|
19 {"txt", "text/plain"},
|
nuclear@6
|
20 {"htm", "text/html"},
|
nuclear@6
|
21 {"html", "text/html"},
|
nuclear@6
|
22 {"png", "image/png"},
|
nuclear@6
|
23 {"jpg", "image/jpeg"},
|
nuclear@6
|
24 {"jpeg", "image/jpeg"},
|
nuclear@6
|
25 {"gif", "image/gif"},
|
nuclear@6
|
26 {"bmp", "image/bmp"},
|
nuclear@9
|
27 {"cgi", 0},
|
nuclear@6
|
28 {0, 0}
|
nuclear@6
|
29 };
|
nuclear@6
|
30
|
nuclear@6
|
31 static int init_types(void);
|
nuclear@6
|
32 static void del_func(struct rbnode *node, void *cls);
|
nuclear@6
|
33
|
nuclear@6
|
34 static struct rbtree *types;
|
nuclear@6
|
35
|
nuclear@6
|
36 static int init_types(void)
|
nuclear@6
|
37 {
|
nuclear@6
|
38 int i;
|
nuclear@6
|
39
|
nuclear@6
|
40 if(types) return 0;
|
nuclear@6
|
41
|
nuclear@9
|
42 if(!(types = rb_create(RB_KEY_STRING))) {
|
nuclear@6
|
43 return -1;
|
nuclear@6
|
44 }
|
nuclear@6
|
45 rb_set_delete_func(types, del_func, 0);
|
nuclear@6
|
46
|
nuclear@6
|
47 for(i=0; def_types[i].suffix; i++) {
|
nuclear@6
|
48 add_mime_type(def_types[i].suffix, def_types[i].type);
|
nuclear@6
|
49 }
|
nuclear@6
|
50 return 0;
|
nuclear@6
|
51 }
|
nuclear@6
|
52
|
nuclear@6
|
53 static void del_func(struct rbnode *node, void *cls)
|
nuclear@6
|
54 {
|
nuclear@6
|
55 free(node->key);
|
nuclear@6
|
56 free(node->data);
|
nuclear@6
|
57 }
|
nuclear@6
|
58
|
nuclear@6
|
59 int add_mime_type(const char *suffix, const char *type)
|
nuclear@6
|
60 {
|
nuclear@6
|
61 init_types();
|
nuclear@6
|
62
|
nuclear@9
|
63 return rb_insert(types, strdup(suffix), type ? strdup(type) : 0);
|
nuclear@6
|
64 }
|
nuclear@6
|
65
|
nuclear@6
|
66 const char *mime_type(const char *path)
|
nuclear@6
|
67 {
|
nuclear@6
|
68 const char *suffix;
|
nuclear@6
|
69
|
nuclear@6
|
70 init_types();
|
nuclear@6
|
71
|
nuclear@6
|
72 if((suffix = strrchr(path, '.'))) {
|
nuclear@6
|
73 struct rbnode *node = rb_find(types, (void*)(suffix + 1));
|
nuclear@6
|
74 if(node) {
|
nuclear@6
|
75 return node->data;
|
nuclear@6
|
76 }
|
nuclear@6
|
77 }
|
nuclear@6
|
78 return "text/plain";
|
nuclear@6
|
79 }
|