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