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 struct etm *t;
39 d8d170aa 2022-04-08 op size_t newcap;
40 95210bb3 2021-01-21 op
41 b2a6b613 2021-01-21 op if (mime->len == mime->cap) {
42 d8d170aa 2022-04-08 op newcap = mime->cap * 1.5;
43 d8d170aa 2022-04-08 op t = recallocarray(mime->t, mime->cap, newcap,
44 95210bb3 2021-01-21 op sizeof(struct etm));
45 d8d170aa 2022-04-08 op if (t == NULL)
46 d8d170aa 2022-04-08 op return -1;
47 d8d170aa 2022-04-08 op mime->t = t;
48 d8d170aa 2022-04-08 op mime->cap = newcap;
49 0fbe79b3 2021-01-18 op }
50 0fbe79b3 2021-01-18 op
51 aa9543b9 2022-09-10 op t = &mime->t[mime->len];
52 aa9543b9 2022-09-10 op if (strlcpy(t->mime, mt, sizeof(t->mime)) >= sizeof(t->mime))
53 d8d170aa 2022-04-08 op return -1;
54 aa9543b9 2022-09-10 op if (strlcpy(t->ext, ext, sizeof(t->ext)) >= sizeof(t->ext))
55 d8d170aa 2022-04-08 op return -1;
56 b2a6b613 2021-01-21 op mime->len++;
57 d8d170aa 2022-04-08 op return 0;
58 0fbe79b3 2021-01-18 op }
59 0fbe79b3 2021-01-18 op
60 0fbe79b3 2021-01-18 op /* load a default set of common mime-extension associations */
61 d8d170aa 2022-04-08 op int
62 b2a6b613 2021-01-21 op load_default_mime(struct mime *mime)
63 0fbe79b3 2021-01-18 op {
64 d8d170aa 2022-04-08 op const struct mapping {
65 d8d170aa 2022-04-08 op const char *mime;
66 d8d170aa 2022-04-08 op const char *ext;
67 d8d170aa 2022-04-08 op } m[] = {
68 0fbe79b3 2021-01-18 op {"application/pdf", "pdf"},
69 0fbe79b3 2021-01-18 op {"image/gif", "gif"},
70 0fbe79b3 2021-01-18 op {"image/jpeg", "jpg"},
71 0fbe79b3 2021-01-18 op {"image/jpeg", "jpeg"},
72 0fbe79b3 2021-01-18 op {"image/png", "png"},
73 0fbe79b3 2021-01-18 op {"image/svg+xml", "svg"},
74 0fbe79b3 2021-01-18 op {"text/gemini", "gemini"},
75 0fbe79b3 2021-01-18 op {"text/gemini", "gmi"},
76 0fbe79b3 2021-01-18 op {"text/markdown", "markdown"},
77 0fbe79b3 2021-01-18 op {"text/markdown", "md"},
78 0fbe79b3 2021-01-18 op {"text/plain", "txt"},
79 0bfdcb17 2021-04-22 op {"text/x-patch", "diff"},
80 4d820b6c 2021-04-21 op {"text/x-patch", "patch"},
81 0fbe79b3 2021-01-18 op {"text/xml", "xml"},
82 0fbe79b3 2021-01-18 op {NULL, NULL}
83 d8d170aa 2022-04-08 op }, *i;
84 0fbe79b3 2021-01-18 op
85 cd5826b8 2022-09-10 op /* don't load the default if `types' was used. */
86 cd5826b8 2022-09-10 op if (mime->len != 0)
87 cd5826b8 2022-09-10 op return 0;
88 cd5826b8 2022-09-10 op
89 d8d170aa 2022-04-08 op for (i = m; i->mime != NULL; ++i) {
90 d8d170aa 2022-04-08 op if (add_mime(mime, i->mime, i->ext) == -1)
91 d8d170aa 2022-04-08 op return -1;
92 d8d170aa 2022-04-08 op }
93 d8d170aa 2022-04-08 op
94 d8d170aa 2022-04-08 op return 0;
95 0fbe79b3 2021-01-18 op }
96 0fbe79b3 2021-01-18 op
97 0fbe79b3 2021-01-18 op static const char *
98 0fbe79b3 2021-01-18 op path_ext(const char *path)
99 0fbe79b3 2021-01-18 op {
100 0fbe79b3 2021-01-18 op const char *end;
101 0fbe79b3 2021-01-18 op
102 0fbe79b3 2021-01-18 op end = path + strlen(path)-1;
103 0fbe79b3 2021-01-18 op for (; end != path; --end) {
104 0fbe79b3 2021-01-18 op if (*end == '.')
105 0fbe79b3 2021-01-18 op return end+1;
106 0fbe79b3 2021-01-18 op if (*end == '/')
107 0fbe79b3 2021-01-18 op break;
108 0fbe79b3 2021-01-18 op }
109 0fbe79b3 2021-01-18 op
110 0fbe79b3 2021-01-18 op return NULL;
111 0fbe79b3 2021-01-18 op }
112 0fbe79b3 2021-01-18 op
113 18bd8391 2022-04-08 op static int
114 18bd8391 2022-04-08 op mime_comp(const void *a, const void *b)
115 18bd8391 2022-04-08 op {
116 18bd8391 2022-04-08 op const struct etm *x = a, *y = b;
117 18bd8391 2022-04-08 op
118 18bd8391 2022-04-08 op return strcmp(x->ext, y->ext);
119 18bd8391 2022-04-08 op }
120 18bd8391 2022-04-08 op
121 18bd8391 2022-04-08 op void
122 18bd8391 2022-04-08 op sort_mime(struct mime *m)
123 18bd8391 2022-04-08 op {
124 18bd8391 2022-04-08 op qsort(m->t, m->len, sizeof(*m->t), mime_comp);
125 18bd8391 2022-04-08 op }
126 18bd8391 2022-04-08 op
127 18bd8391 2022-04-08 op static int
128 18bd8391 2022-04-08 op mime_find(const void *a, const void *b)
129 18bd8391 2022-04-08 op {
130 18bd8391 2022-04-08 op const char *ext = a;
131 18bd8391 2022-04-08 op const struct etm *x = b;
132 18bd8391 2022-04-08 op
133 18bd8391 2022-04-08 op return strcmp(ext, x->ext);
134 18bd8391 2022-04-08 op }
135 18bd8391 2022-04-08 op
136 0fbe79b3 2021-01-18 op const char *
137 6119e13e 2021-01-19 op mime(struct vhost *host, const char *path)
138 0fbe79b3 2021-01-18 op {
139 6119e13e 2021-01-19 op const char *def, *ext;
140 0fbe79b3 2021-01-18 op struct etm *t;
141 0fbe79b3 2021-01-18 op
142 c8b74339 2021-01-24 op def = vhost_default_mime(host, path);
143 6119e13e 2021-01-19 op
144 0fbe79b3 2021-01-18 op if ((ext = path_ext(path)) == NULL)
145 6119e13e 2021-01-19 op return def;
146 0fbe79b3 2021-01-18 op
147 18bd8391 2022-04-08 op t = bsearch(ext, conf.mime.t, conf.mime.len, sizeof(*conf.mime.t),
148 18bd8391 2022-04-08 op mime_find);
149 18bd8391 2022-04-08 op if (t != NULL)
150 18bd8391 2022-04-08 op return t->mime;
151 8fb0df4d 2022-04-08 op if (!strcmp(ext, "gmi") || !strcmp(ext, "gemini"))
152 8fb0df4d 2022-04-08 op return "text/gemini";
153 6119e13e 2021-01-19 op return def;
154 0fbe79b3 2021-01-18 op }
155 d8d170aa 2022-04-08 op
156 d8d170aa 2022-04-08 op void
157 d8d170aa 2022-04-08 op free_mime(struct mime *m)
158 d8d170aa 2022-04-08 op {
159 d8d170aa 2022-04-08 op free(m->t);
160 d8d170aa 2022-04-08 op }