1 0fbe79b3 2021-01-18 op /*
  2 0fbe79b3 2021-01-18 op  * Copyright (c) 2021 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 0fbe79b3 2021-01-18 op #include <errno.h>
 18 0fbe79b3 2021-01-18 op #include <stdlib.h>
 19 0fbe79b3 2021-01-18 op #include <string.h>
 20 0fbe79b3 2021-01-18 op 
 21 0fbe79b3 2021-01-18 op #include "gmid.h"
 22 0fbe79b3 2021-01-18 op 
 23 0fbe79b3 2021-01-18 op struct etm {			/* extension to mime */
 24 0fbe79b3 2021-01-18 op 	const char	*mime;
 25 0fbe79b3 2021-01-18 op 	const char	*ext;
 26 0fbe79b3 2021-01-18 op };
 27 0fbe79b3 2021-01-18 op 
 28 0fbe79b3 2021-01-18 op struct mimes {
 29 0fbe79b3 2021-01-18 op 	struct etm	*t;
 30 0fbe79b3 2021-01-18 op 	size_t		len;
 31 0fbe79b3 2021-01-18 op 	size_t		cap;
 32 0fbe79b3 2021-01-18 op };
 33 0fbe79b3 2021-01-18 op 
 34 0fbe79b3 2021-01-18 op struct mimes mimes;
 35 0fbe79b3 2021-01-18 op 
 36 0fbe79b3 2021-01-18 op void
 37 0fbe79b3 2021-01-18 op init_mime(void)
 38 0fbe79b3 2021-01-18 op {
 39 0fbe79b3 2021-01-18 op 	mimes.len = 0;
 40 0fbe79b3 2021-01-18 op 	mimes.cap = 2;
 41 0fbe79b3 2021-01-18 op 
 42 0fbe79b3 2021-01-18 op 	if ((mimes.t = calloc(mimes.cap, sizeof(struct etm))) == NULL)
 43 0fbe79b3 2021-01-18 op 		fatal("calloc: %s", strerror(errno));
 44 0fbe79b3 2021-01-18 op }
 45 0fbe79b3 2021-01-18 op 
 46 0fbe79b3 2021-01-18 op /* register mime for the given extension */
 47 0fbe79b3 2021-01-18 op void
 48 0fbe79b3 2021-01-18 op add_mime(const char *mime, const char *ext)
 49 0fbe79b3 2021-01-18 op {
 50 0fbe79b3 2021-01-18 op 	if (mimes.len == mimes.cap) {
 51 0fbe79b3 2021-01-18 op 		mimes.cap *= 1.5;
 52 0fbe79b3 2021-01-18 op 		mimes.t = realloc(mimes.t, mimes.cap * sizeof(struct etm));
 53 0fbe79b3 2021-01-18 op 		if (mimes.t == NULL)
 54 0fbe79b3 2021-01-18 op 			fatal("realloc: %s", strerror(errno));
 55 0fbe79b3 2021-01-18 op 	}
 56 0fbe79b3 2021-01-18 op 
 57 0fbe79b3 2021-01-18 op 	mimes.t[mimes.len].mime = mime;
 58 0fbe79b3 2021-01-18 op 	mimes.t[mimes.len].ext  = ext;
 59 0fbe79b3 2021-01-18 op 	mimes.len++;
 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 0fbe79b3 2021-01-18 op void
 64 0fbe79b3 2021-01-18 op load_default_mime()
 65 0fbe79b3 2021-01-18 op {
 66 0fbe79b3 2021-01-18 op 	struct etm *i, m[] = {
 67 0fbe79b3 2021-01-18 op 		{"application/pdf",	"pdf"},
 68 0fbe79b3 2021-01-18 op 		{"image/gif",		"gif"},
 69 0fbe79b3 2021-01-18 op 		{"image/jpeg",		"jpg"},
 70 0fbe79b3 2021-01-18 op 		{"image/jpeg",		"jpeg"},
 71 0fbe79b3 2021-01-18 op 		{"image/png",		"png"},
 72 0fbe79b3 2021-01-18 op 		{"image/svg+xml",	"svg"},
 73 0fbe79b3 2021-01-18 op 		{"text/gemini",		"gemini"},
 74 0fbe79b3 2021-01-18 op 		{"text/gemini",		"gmi"},
 75 0fbe79b3 2021-01-18 op 		{"text/markdown",	"markdown"},
 76 0fbe79b3 2021-01-18 op 		{"text/markdown",	"md"},
 77 0fbe79b3 2021-01-18 op 		{"text/plain",		"txt"},
 78 0fbe79b3 2021-01-18 op 		{"text/xml",		"xml"},
 79 0fbe79b3 2021-01-18 op 		{NULL, NULL}
 80 0fbe79b3 2021-01-18 op 	};
 81 0fbe79b3 2021-01-18 op 
 82 0fbe79b3 2021-01-18 op 	for (i = m; i->mime != NULL; ++i)
 83 0fbe79b3 2021-01-18 op 		add_mime(i->mime, i->ext);
 84 0fbe79b3 2021-01-18 op }
 85 0fbe79b3 2021-01-18 op 
 86 0fbe79b3 2021-01-18 op static const char *
 87 0fbe79b3 2021-01-18 op path_ext(const char *path)
 88 0fbe79b3 2021-01-18 op {
 89 0fbe79b3 2021-01-18 op 	const char *end;
 90 0fbe79b3 2021-01-18 op 
 91 0fbe79b3 2021-01-18 op 	end = path + strlen(path)-1;
 92 0fbe79b3 2021-01-18 op 	for (; end != path; --end) {
 93 0fbe79b3 2021-01-18 op 		if (*end == '.')
 94 0fbe79b3 2021-01-18 op 			return end+1;
 95 0fbe79b3 2021-01-18 op 		if (*end == '/')
 96 0fbe79b3 2021-01-18 op 			break;
 97 0fbe79b3 2021-01-18 op 	}
 98 0fbe79b3 2021-01-18 op 
 99 0fbe79b3 2021-01-18 op 	return NULL;
100 0fbe79b3 2021-01-18 op }
101 0fbe79b3 2021-01-18 op 
102 0fbe79b3 2021-01-18 op const char *
103 0fbe79b3 2021-01-18 op mime(const char *path)
104 0fbe79b3 2021-01-18 op {
105 0fbe79b3 2021-01-18 op 	const char *ext, *def = "application/octet-stream";
106 0fbe79b3 2021-01-18 op 	struct etm *t;
107 0fbe79b3 2021-01-18 op 
108 0fbe79b3 2021-01-18 op 	if ((ext = path_ext(path)) == NULL)
109 0fbe79b3 2021-01-18 op 		return def;
110 0fbe79b3 2021-01-18 op 
111 0fbe79b3 2021-01-18 op 	for (t = mimes.t; t->mime != NULL; ++t)
112 0fbe79b3 2021-01-18 op 		if (!strcmp(ext, t->ext))
113 0fbe79b3 2021-01-18 op 			return t->mime;
114 0fbe79b3 2021-01-18 op 
115 0fbe79b3 2021-01-18 op 	return def;
116 0fbe79b3 2021-01-18 op }