Blame


1 0fbe79b3 2021-01-18 op /*
2 a555e0d6 2022-07-04 op * Copyright (c) 2021, 2022 Omar Polo <op@omarpolo.com>
3 0fbe79b3 2021-01-18 op *
4 0fbe79b3 2021-01-18 op * Permission to use, copy, modify, and distribute this software for any
5 0fbe79b3 2021-01-18 op * purpose with or without fee is hereby granted, provided that the above
6 0fbe79b3 2021-01-18 op * copyright notice and this permission notice appear in all copies.
7 0fbe79b3 2021-01-18 op *
8 0fbe79b3 2021-01-18 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 0fbe79b3 2021-01-18 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 0fbe79b3 2021-01-18 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 0fbe79b3 2021-01-18 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 0fbe79b3 2021-01-18 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 0fbe79b3 2021-01-18 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 0fbe79b3 2021-01-18 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 0fbe79b3 2021-01-18 op */
16 0fbe79b3 2021-01-18 op
17 52418c8d 2021-02-12 op #include "gmid.h"
18 52418c8d 2021-02-12 op
19 0fbe79b3 2021-01-18 op #include <errno.h>
20 0fbe79b3 2021-01-18 op #include <stdlib.h>
21 0fbe79b3 2021-01-18 op #include <string.h>
22 0fbe79b3 2021-01-18 op
23 0fbe79b3 2021-01-18 op void
24 b2a6b613 2021-01-21 op init_mime(struct mime *mime)
25 0fbe79b3 2021-01-18 op {
26 b2a6b613 2021-01-21 op mime->len = 0;
27 b2a6b613 2021-01-21 op mime->cap = 16;
28 0fbe79b3 2021-01-18 op
29 b2a6b613 2021-01-21 op mime->t = calloc(mime->cap, sizeof(struct etm));
30 b2a6b613 2021-01-21 op if (mime->t == NULL)
31 0fbe79b3 2021-01-18 op fatal("calloc: %s", strerror(errno));
32 0fbe79b3 2021-01-18 op }
33 0fbe79b3 2021-01-18 op
34 0fbe79b3 2021-01-18 op /* register mime for the given extension */
35 d8d170aa 2022-04-08 op int
36 b2a6b613 2021-01-21 op add_mime(struct mime *mime, const char *mt, const char *ext)
37 0fbe79b3 2021-01-18 op {
38 d8d170aa 2022-04-08 op char *mimetype, *extension;
39 d8d170aa 2022-04-08 op struct etm *t;
40 d8d170aa 2022-04-08 op size_t newcap;
41 95210bb3 2021-01-21 op
42 b2a6b613 2021-01-21 op if (mime->len == mime->cap) {
43 d8d170aa 2022-04-08 op newcap = mime->cap * 1.5;
44 d8d170aa 2022-04-08 op t = recallocarray(mime->t, mime->cap, newcap,
45 95210bb3 2021-01-21 op sizeof(struct etm));
46 d8d170aa 2022-04-08 op if (t == NULL)
47 d8d170aa 2022-04-08 op return -1;
48 d8d170aa 2022-04-08 op mime->t = t;
49 d8d170aa 2022-04-08 op mime->cap = newcap;
50 0fbe79b3 2021-01-18 op }
51 0fbe79b3 2021-01-18 op
52 d8d170aa 2022-04-08 op if ((mimetype = strdup(mt)) == NULL)
53 d8d170aa 2022-04-08 op return -1;
54 d8d170aa 2022-04-08 op if ((extension = strdup(ext)) == NULL) {
55 d8d170aa 2022-04-08 op free(mimetype);
56 d8d170aa 2022-04-08 op return -1;
57 d8d170aa 2022-04-08 op }
58 d8d170aa 2022-04-08 op
59 d8d170aa 2022-04-08 op mime->t[mime->len].mime = mimetype;
60 d8d170aa 2022-04-08 op mime->t[mime->len].ext = extension;
61 b2a6b613 2021-01-21 op mime->len++;
62 d8d170aa 2022-04-08 op return 0;
63 0fbe79b3 2021-01-18 op }
64 0fbe79b3 2021-01-18 op
65 0fbe79b3 2021-01-18 op /* load a default set of common mime-extension associations */
66 d8d170aa 2022-04-08 op int
67 b2a6b613 2021-01-21 op load_default_mime(struct mime *mime)
68 0fbe79b3 2021-01-18 op {
69 d8d170aa 2022-04-08 op const struct mapping {
70 d8d170aa 2022-04-08 op const char *mime;
71 d8d170aa 2022-04-08 op const char *ext;
72 d8d170aa 2022-04-08 op } m[] = {
73 0fbe79b3 2021-01-18 op {"application/pdf", "pdf"},
74 0fbe79b3 2021-01-18 op {"image/gif", "gif"},
75 0fbe79b3 2021-01-18 op {"image/jpeg", "jpg"},
76 0fbe79b3 2021-01-18 op {"image/jpeg", "jpeg"},
77 0fbe79b3 2021-01-18 op {"image/png", "png"},
78 0fbe79b3 2021-01-18 op {"image/svg+xml", "svg"},
79 0fbe79b3 2021-01-18 op {"text/gemini", "gemini"},
80 0fbe79b3 2021-01-18 op {"text/gemini", "gmi"},
81 0fbe79b3 2021-01-18 op {"text/markdown", "markdown"},
82 0fbe79b3 2021-01-18 op {"text/markdown", "md"},
83 0fbe79b3 2021-01-18 op {"text/plain", "txt"},
84 0bfdcb17 2021-04-22 op {"text/x-patch", "diff"},
85 4d820b6c 2021-04-21 op {"text/x-patch", "patch"},
86 0fbe79b3 2021-01-18 op {"text/xml", "xml"},
87 0fbe79b3 2021-01-18 op {NULL, NULL}
88 d8d170aa 2022-04-08 op }, *i;
89 0fbe79b3 2021-01-18 op
90 d8d170aa 2022-04-08 op for (i = m; i->mime != NULL; ++i) {
91 d8d170aa 2022-04-08 op if (add_mime(mime, i->mime, i->ext) == -1)
92 d8d170aa 2022-04-08 op return -1;
93 d8d170aa 2022-04-08 op }
94 d8d170aa 2022-04-08 op
95 d8d170aa 2022-04-08 op return 0;
96 0fbe79b3 2021-01-18 op }
97 0fbe79b3 2021-01-18 op
98 0fbe79b3 2021-01-18 op static const char *
99 0fbe79b3 2021-01-18 op path_ext(const char *path)
100 0fbe79b3 2021-01-18 op {
101 0fbe79b3 2021-01-18 op const char *end;
102 0fbe79b3 2021-01-18 op
103 0fbe79b3 2021-01-18 op end = path + strlen(path)-1;
104 0fbe79b3 2021-01-18 op for (; end != path; --end) {
105 0fbe79b3 2021-01-18 op if (*end == '.')
106 0fbe79b3 2021-01-18 op return end+1;
107 0fbe79b3 2021-01-18 op if (*end == '/')
108 0fbe79b3 2021-01-18 op break;
109 0fbe79b3 2021-01-18 op }
110 0fbe79b3 2021-01-18 op
111 0fbe79b3 2021-01-18 op return NULL;
112 0fbe79b3 2021-01-18 op }
113 0fbe79b3 2021-01-18 op
114 18bd8391 2022-04-08 op static int
115 18bd8391 2022-04-08 op mime_comp(const void *a, const void *b)
116 18bd8391 2022-04-08 op {
117 18bd8391 2022-04-08 op const struct etm *x = a, *y = b;
118 18bd8391 2022-04-08 op
119 18bd8391 2022-04-08 op return strcmp(x->ext, y->ext);
120 18bd8391 2022-04-08 op }
121 18bd8391 2022-04-08 op
122 18bd8391 2022-04-08 op void
123 18bd8391 2022-04-08 op sort_mime(struct mime *m)
124 18bd8391 2022-04-08 op {
125 18bd8391 2022-04-08 op qsort(m->t, m->len, sizeof(*m->t), mime_comp);
126 18bd8391 2022-04-08 op }
127 18bd8391 2022-04-08 op
128 18bd8391 2022-04-08 op static int
129 18bd8391 2022-04-08 op mime_find(const void *a, const void *b)
130 18bd8391 2022-04-08 op {
131 18bd8391 2022-04-08 op const char *ext = a;
132 18bd8391 2022-04-08 op const struct etm *x = b;
133 18bd8391 2022-04-08 op
134 18bd8391 2022-04-08 op return strcmp(ext, x->ext);
135 18bd8391 2022-04-08 op }
136 18bd8391 2022-04-08 op
137 0fbe79b3 2021-01-18 op const char *
138 6119e13e 2021-01-19 op mime(struct vhost *host, const char *path)
139 0fbe79b3 2021-01-18 op {
140 6119e13e 2021-01-19 op const char *def, *ext;
141 0fbe79b3 2021-01-18 op struct etm *t;
142 0fbe79b3 2021-01-18 op
143 c8b74339 2021-01-24 op def = vhost_default_mime(host, path);
144 6119e13e 2021-01-19 op
145 0fbe79b3 2021-01-18 op if ((ext = path_ext(path)) == NULL)
146 6119e13e 2021-01-19 op return def;
147 0fbe79b3 2021-01-18 op
148 18bd8391 2022-04-08 op t = bsearch(ext, conf.mime.t, conf.mime.len, sizeof(*conf.mime.t),
149 18bd8391 2022-04-08 op mime_find);
150 18bd8391 2022-04-08 op if (t != NULL)
151 18bd8391 2022-04-08 op return t->mime;
152 8fb0df4d 2022-04-08 op if (!strcmp(ext, "gmi") || !strcmp(ext, "gemini"))
153 8fb0df4d 2022-04-08 op return "text/gemini";
154 6119e13e 2021-01-19 op return def;
155 0fbe79b3 2021-01-18 op }
156 d8d170aa 2022-04-08 op
157 d8d170aa 2022-04-08 op void
158 d8d170aa 2022-04-08 op free_mime(struct mime *m)
159 d8d170aa 2022-04-08 op {
160 d8d170aa 2022-04-08 op struct etm *t;
161 d8d170aa 2022-04-08 op
162 d8d170aa 2022-04-08 op for (t = m->t; t->mime != NULL; ++t) {
163 d8d170aa 2022-04-08 op free(t->mime);
164 d8d170aa 2022-04-08 op free(t->ext);
165 d8d170aa 2022-04-08 op }
166 d8d170aa 2022-04-08 op
167 d8d170aa 2022-04-08 op free(m->t);
168 d8d170aa 2022-04-08 op }