1
/*
2
 * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3
 *
4
 * Permission to use, copy, modify, and distribute this software for any
5
 * purpose with or without fee is hereby granted, provided that the above
6
 * copyright notice and this permission notice appear in all copies.
7
 *
8
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15
 */
16

17
#include "gmid.h"
18

19
#include <errno.h>
20
#include <stdlib.h>
21
#include <string.h>
22

23
void
24
init_mime(struct mime *mime)
25
{
26
	mime->len = 0;
27
	mime->cap = 16;
28

29
	mime->t = calloc(mime->cap, sizeof(struct etm));
30
	if (mime->t == NULL)
31
		fatal("calloc: %s", strerror(errno));
32
}
33

34
/* register mime for the given extension */
35
int
36
add_mime(struct mime *mime, const char *mt, const char *ext)
37
{
38
	char *mimetype, *extension;
39
	struct etm *t;
40
	size_t newcap;
41

42
	if (mime->len == mime->cap) {
43
		newcap = mime->cap * 1.5;
44
		t = recallocarray(mime->t, mime->cap, newcap,
45
		    sizeof(struct etm));
46
		if (t == NULL)
47
			return -1;
48
		mime->t = t;
49
		mime->cap = newcap;
50
	}
51

52
	if ((mimetype = strdup(mt)) == NULL)
53
		return -1;
54
	if ((extension = strdup(ext)) == NULL) {
55
		free(mimetype);
56
		return -1;
57
	}
58

59
	mime->t[mime->len].mime = mimetype;
60
	mime->t[mime->len].ext  = extension;
61
	mime->len++;
62
	return 0;
63
}
64

65
/* load a default set of common mime-extension associations */
66
int
67
load_default_mime(struct mime *mime)
68
{
69
	const struct mapping {
70
		const char *mime;
71
		const char *ext;
72
	} m[] = {
73
		{"application/pdf",	"pdf"},
74
		{"image/gif",		"gif"},
75
		{"image/jpeg",		"jpg"},
76
		{"image/jpeg",		"jpeg"},
77
		{"image/png",		"png"},
78
		{"image/svg+xml",	"svg"},
79
		{"text/gemini",		"gemini"},
80
		{"text/gemini",		"gmi"},
81
		{"text/markdown",	"markdown"},
82
		{"text/markdown",	"md"},
83
		{"text/plain",		"txt"},
84
		{"text/x-patch",	"diff"},
85
		{"text/x-patch",	"patch"},
86
		{"text/xml",		"xml"},
87
		{NULL, NULL}
88
	}, *i;
89

90
	for (i = m; i->mime != NULL; ++i) {
91
		if (add_mime(mime, i->mime, i->ext) == -1)
92
			return -1;
93
	}
94

95
	return 0;
96
}
97

98
static const char *
99
path_ext(const char *path)
100
{
101
	const char *end;
102

103
	end = path + strlen(path)-1;
104
	for (; end != path; --end) {
105
		if (*end == '.')
106
			return end+1;
107
		if (*end == '/')
108
			break;
109
	}
110

111
	return NULL;
112
}
113

114
const char *
115
mime(struct vhost *host, const char *path)
116
{
117
	const char *def, *ext;
118
	struct etm *t;
119

120
	def = vhost_default_mime(host, path);
121

122
	if ((ext = path_ext(path)) == NULL)
123
		return def;
124

125
	for (t = conf.mime.t; t->mime != NULL; ++t)
126
		if (!strcmp(ext, t->ext))
127
			return t->mime;
128

129
	return def;
130
}
131

132
void
133
free_mime(struct mime *m)
134
{
135
	struct etm *t;
136

137
	for (t = m->t; t->mime != NULL; ++t) {
138
		free(t->mime);
139
		free(t->ext);
140
	}
141

142
	free(m->t);
143
}