1 3e4749f7 2020-10-02 op /*
  2 3e4749f7 2020-10-02 op  * Copyright (c) 2020 Omar Polo <op@omarpolo.com>
  3 3e4749f7 2020-10-02 op  *
  4 3e4749f7 2020-10-02 op  * Permission to use, copy, modify, and distribute this software for any
  5 3e4749f7 2020-10-02 op  * purpose with or without fee is hereby granted, provided that the above
  6 3e4749f7 2020-10-02 op  * copyright notice and this permission notice appear in all copies.
  7 3e4749f7 2020-10-02 op  *
  8 3e4749f7 2020-10-02 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 3e4749f7 2020-10-02 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 3e4749f7 2020-10-02 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 3e4749f7 2020-10-02 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 3e4749f7 2020-10-02 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 3e4749f7 2020-10-02 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 3e4749f7 2020-10-02 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 3e4749f7 2020-10-02 op  */
 16 3e4749f7 2020-10-02 op 
 17 f28f9311 2020-10-14 op #include <sys/mman.h>
 18 3e4749f7 2020-10-02 op #include <sys/stat.h>
 19 3e4749f7 2020-10-02 op 
 20 3e4749f7 2020-10-02 op #include <assert.h>
 21 3e4749f7 2020-10-02 op #include <err.h>
 22 592fd624 2020-10-07 op #include <errno.h>
 23 3e4749f7 2020-10-02 op #include <fcntl.h>
 24 231bfcdc 2020-12-21 op #include <limits.h>
 25 709d6e5e 2021-01-10 op #include <netdb.h>
 26 0cf902af 2020-11-03 op #include <signal.h>
 27 80bbcad5 2021-01-10 op #include <stdarg.h>
 28 3e4749f7 2020-10-02 op #include <string.h>
 29 3e4749f7 2020-10-02 op 
 30 488f059a 2020-12-24 op #include "gmid.h"
 31 cc68fe70 2020-10-07 op 
 32 15902770 2021-01-15 op struct vhost hosts[HOSTSLEN];
 33 15902770 2021-01-15 op 
 34 120381c9 2020-11-06 op int connected_clients;
 35 15902770 2021-01-15 op int goterror;
 36 592fd624 2020-10-07 op 
 37 15902770 2021-01-15 op struct conf conf;
 38 15902770 2021-01-15 op 
 39 33d32d1f 2020-12-25 op struct etm {			/* file extension to mime */
 40 33d32d1f 2020-12-25 op 	const char	*mime;
 41 33d32d1f 2020-12-25 op 	const char	*ext;
 42 33d32d1f 2020-12-25 op } filetypes[] = {
 43 33d32d1f 2020-12-25 op 	{"application/pdf",	"pdf"},
 44 33d32d1f 2020-12-25 op 
 45 33d32d1f 2020-12-25 op 	{"image/gif",		"gif"},
 46 33d32d1f 2020-12-25 op 	{"image/jpeg",		"jpg"},
 47 33d32d1f 2020-12-25 op 	{"image/jpeg",		"jpeg"},
 48 33d32d1f 2020-12-25 op 	{"image/png",		"png"},
 49 33d32d1f 2020-12-25 op 	{"image/svg+xml",	"svg"},
 50 33d32d1f 2020-12-25 op 
 51 33d32d1f 2020-12-25 op 	{"text/gemini",		"gemini"},
 52 33d32d1f 2020-12-25 op 	{"text/gemini",		"gmi"},
 53 33d32d1f 2020-12-25 op 	{"text/markdown",	"markdown"},
 54 33d32d1f 2020-12-25 op 	{"text/markdown",	"md"},
 55 33d32d1f 2020-12-25 op 	{"text/plain",		"txt"},
 56 33d32d1f 2020-12-25 op 	{"text/xml",		"xml"},
 57 33d32d1f 2020-12-25 op 
 58 33d32d1f 2020-12-25 op 	{NULL, NULL}
 59 33d32d1f 2020-12-25 op };
 60 33d32d1f 2020-12-25 op 
 61 80bbcad5 2021-01-10 op __attribute__ ((format (printf, 1, 2)))
 62 0d7a38c4 2021-01-11 op __attribute__ ((__noreturn__))
 63 0d7a38c4 2021-01-11 op static inline void
 64 80bbcad5 2021-01-10 op fatal(const char *fmt, ...)
 65 80bbcad5 2021-01-10 op {
 66 80bbcad5 2021-01-10 op 	va_list ap;
 67 80bbcad5 2021-01-10 op 
 68 80bbcad5 2021-01-10 op 	va_start(ap, fmt);
 69 80bbcad5 2021-01-10 op 
 70 15902770 2021-01-15 op 	if (conf.foreground) {
 71 80bbcad5 2021-01-10 op 		vfprintf(stderr, fmt, ap);
 72 80bbcad5 2021-01-10 op 		fprintf(stderr, "\n");
 73 80bbcad5 2021-01-10 op 	} else
 74 80bbcad5 2021-01-10 op 		vsyslog(LOG_DAEMON | LOG_CRIT, fmt, ap);
 75 80bbcad5 2021-01-10 op 
 76 80bbcad5 2021-01-10 op 	va_end(ap);
 77 80bbcad5 2021-01-10 op 	exit(1);
 78 80bbcad5 2021-01-10 op }
 79 80bbcad5 2021-01-10 op 
 80 dafb57b8 2021-01-15 op void
 81 709d6e5e 2021-01-10 op logs(int priority, struct client *c,
 82 709d6e5e 2021-01-10 op     const char *fmt, ...)
 83 80bbcad5 2021-01-10 op {
 84 709d6e5e 2021-01-10 op 	char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV];
 85 709d6e5e 2021-01-10 op 	char *fmted, *s;
 86 709d6e5e 2021-01-10 op 	size_t len;
 87 709d6e5e 2021-01-10 op 	int ec;
 88 80bbcad5 2021-01-10 op 	va_list ap;
 89 80bbcad5 2021-01-10 op 
 90 80bbcad5 2021-01-10 op 	va_start(ap, fmt);
 91 80bbcad5 2021-01-10 op 
 92 dafb57b8 2021-01-15 op 	if (c == NULL) {
 93 dafb57b8 2021-01-15 op 		strncpy(hbuf, "<internal>", sizeof(hbuf));
 94 dafb57b8 2021-01-15 op 		sbuf[0] = '\0';
 95 dafb57b8 2021-01-15 op 	} else {
 96 dafb57b8 2021-01-15 op 		len = sizeof(c->addr);
 97 dafb57b8 2021-01-15 op 		ec = getnameinfo((struct sockaddr*)&c->addr, len,
 98 f7b816dc 2021-01-15 op 		    hbuf, sizeof(hbuf),
 99 f7b816dc 2021-01-15 op 		    sbuf, sizeof(sbuf),
100 f7b816dc 2021-01-15 op 		    NI_NUMERICHOST | NI_NUMERICSERV);
101 dafb57b8 2021-01-15 op 		if (ec != 0)
102 dafb57b8 2021-01-15 op 			fatal("getnameinfo: %s", gai_strerror(ec));
103 dafb57b8 2021-01-15 op 	}
104 80bbcad5 2021-01-10 op 
105 709d6e5e 2021-01-10 op 	if (vasprintf(&fmted, fmt, ap) == -1)
106 709d6e5e 2021-01-10 op 		fatal("vasprintf: %s", strerror(errno));
107 709d6e5e 2021-01-10 op 
108 15902770 2021-01-15 op 	if (conf.foreground)
109 709d6e5e 2021-01-10 op 		fprintf(stderr, "%s:%s %s\n", hbuf, sbuf, fmted);
110 709d6e5e 2021-01-10 op 	else {
111 709d6e5e 2021-01-10 op 		if (asprintf(&s, "%s:%s %s", hbuf, sbuf, fmted) == -1)
112 709d6e5e 2021-01-10 op 			fatal("asprintf: %s", strerror(errno));
113 709d6e5e 2021-01-10 op 		syslog(priority | LOG_DAEMON, "%s", s);
114 709d6e5e 2021-01-10 op 		free(s);
115 709d6e5e 2021-01-10 op 	}
116 80bbcad5 2021-01-10 op 
117 709d6e5e 2021-01-10 op 	free(fmted);
118 709d6e5e 2021-01-10 op 
119 80bbcad5 2021-01-10 op 	va_end(ap);
120 80bbcad5 2021-01-10 op }
121 80bbcad5 2021-01-10 op 
122 120381c9 2020-11-06 op void
123 4a28dd01 2020-12-28 op sig_handler(int sig)
124 120381c9 2020-11-06 op {
125 120381c9 2020-11-06 op 	(void)sig;
126 120381c9 2020-11-06 op }
127 120381c9 2020-11-06 op 
128 72f653b6 2020-11-06 op int
129 72f653b6 2020-11-06 op starts_with(const char *str, const char *prefix)
130 72f653b6 2020-11-06 op {
131 72f653b6 2020-11-06 op 	size_t i;
132 72f653b6 2020-11-06 op 
133 72f653b6 2020-11-06 op 	for (i = 0; prefix[i] != '\0'; ++i)
134 72f653b6 2020-11-06 op 		if (str[i] != prefix[i])
135 72f653b6 2020-11-06 op 			return 0;
136 72f653b6 2020-11-06 op 	return 1;
137 72f653b6 2020-11-06 op }
138 72f653b6 2020-11-06 op 
139 3e4749f7 2020-10-02 op int
140 592fd624 2020-10-07 op start_reply(struct pollfd *pfd, struct client *client, int code, const char *reason)
141 3e4749f7 2020-10-02 op {
142 f7b816dc 2021-01-15 op 	char buf[1030]; 	/* status + ' ' + max reply len + \r\n\0 */
143 3e4749f7 2020-10-02 op 	int len;
144 3e4749f7 2020-10-02 op 
145 592fd624 2020-10-07 op 	client->code = code;
146 592fd624 2020-10-07 op 	client->meta = reason;
147 592fd624 2020-10-07 op 	client->state = S_INITIALIZING;
148 592fd624 2020-10-07 op 
149 3e4749f7 2020-10-02 op 	len = snprintf(buf, sizeof(buf), "%d %s\r\n", code, reason);
150 3e4749f7 2020-10-02 op 	assert(len < (int)sizeof(buf));
151 f7b816dc 2021-01-15 op 
152 f7b816dc 2021-01-15 op 	switch (tls_write(client->ctx, buf, len)) {
153 f7b816dc 2021-01-15 op 	case TLS_WANT_POLLIN:
154 592fd624 2020-10-07 op 		pfd->events = POLLIN;
155 592fd624 2020-10-07 op 		return 0;
156 f7b816dc 2021-01-15 op 	case TLS_WANT_POLLOUT:
157 592fd624 2020-10-07 op 		pfd->events = POLLOUT;
158 592fd624 2020-10-07 op 		return 0;
159 f7b816dc 2021-01-15 op 	default:
160 f7b816dc 2021-01-15 op 		return 1;
161 592fd624 2020-10-07 op 	}
162 3e4749f7 2020-10-02 op }
163 3e4749f7 2020-10-02 op 
164 f28f9311 2020-10-14 op ssize_t
165 f28f9311 2020-10-14 op filesize(int fd)
166 f28f9311 2020-10-14 op {
167 f28f9311 2020-10-14 op 	ssize_t len;
168 f28f9311 2020-10-14 op 
169 f28f9311 2020-10-14 op 	if ((len = lseek(fd, 0, SEEK_END)) == -1)
170 f28f9311 2020-10-14 op 		return -1;
171 f28f9311 2020-10-14 op 	if (lseek(fd, 0, SEEK_SET) == -1)
172 f28f9311 2020-10-14 op 		return -1;
173 f28f9311 2020-10-14 op 	return len;
174 f28f9311 2020-10-14 op }
175 f28f9311 2020-10-14 op 
176 cc68fe70 2020-10-07 op const char *
177 cc68fe70 2020-10-07 op path_ext(const char *path)
178 cc68fe70 2020-10-07 op {
179 cc68fe70 2020-10-07 op 	const char *end;
180 cc68fe70 2020-10-07 op 
181 f7b816dc 2021-01-15 op 	end = path + strlen(path)-1;
182 cc68fe70 2020-10-07 op 	for (; end != path; --end) {
183 cc68fe70 2020-10-07 op 		if (*end == '.')
184 cc68fe70 2020-10-07 op 			return end+1;
185 cc68fe70 2020-10-07 op 		if (*end == '/')
186 cc68fe70 2020-10-07 op 			break;
187 cc68fe70 2020-10-07 op 	}
188 cc68fe70 2020-10-07 op 
189 cc68fe70 2020-10-07 op 	return NULL;
190 cc68fe70 2020-10-07 op }
191 cc68fe70 2020-10-07 op 
192 cc68fe70 2020-10-07 op const char *
193 cc68fe70 2020-10-07 op mime(const char *path)
194 cc68fe70 2020-10-07 op {
195 cc68fe70 2020-10-07 op 	const char *ext, *def = "application/octet-stream";
196 cc68fe70 2020-10-07 op 	struct etm *t;
197 cc68fe70 2020-10-07 op 
198 cc68fe70 2020-10-07 op 	if ((ext = path_ext(path)) == NULL)
199 cc68fe70 2020-10-07 op 		return def;
200 cc68fe70 2020-10-07 op 
201 cc68fe70 2020-10-07 op 	for (t = filetypes; t->mime != NULL; ++t)
202 cc68fe70 2020-10-07 op 		if (!strcmp(ext, t->ext))
203 cc68fe70 2020-10-07 op 			return t->mime;
204 cc68fe70 2020-10-07 op 
205 cc68fe70 2020-10-07 op 	return def;
206 3e4749f7 2020-10-02 op }
207 3e4749f7 2020-10-02 op 
208 f28f9311 2020-10-14 op int
209 d7802bb4 2020-12-02 op check_path(struct client *c, const char *path, int *fd)
210 3e4749f7 2020-10-02 op {
211 72342dc9 2020-11-06 op 	struct stat sb;
212 3e4749f7 2020-10-02 op 
213 f28f9311 2020-10-14 op 	assert(path != NULL);
214 15902770 2021-01-15 op 	if ((*fd = openat(c->host->dirfd, *path ? path : ".",
215 f7b816dc 2021-01-15 op 	    O_RDONLY | O_NOFOLLOW | O_CLOEXEC)) == -1) {
216 a5d310bc 2020-11-10 op 		return FILE_MISSING;
217 a5d310bc 2020-11-10 op 	}
218 a5d310bc 2020-11-10 op 
219 a5d310bc 2020-11-10 op 	if (fstat(*fd, &sb) == -1) {
220 d7802bb4 2020-12-02 op 		LOGN(c, "failed stat for %s: %s", path, strerror(errno));
221 a5d310bc 2020-11-10 op 		return FILE_MISSING;
222 a5d310bc 2020-11-10 op 	}
223 a5d310bc 2020-11-10 op 
224 a5d310bc 2020-11-10 op 	if (S_ISDIR(sb.st_mode))
225 a5d310bc 2020-11-10 op 		return FILE_DIRECTORY;
226 a5d310bc 2020-11-10 op 
227 a5d310bc 2020-11-10 op 	if (sb.st_mode & S_IXUSR)
228 a5d310bc 2020-11-10 op 		return FILE_EXECUTABLE;
229 a5d310bc 2020-11-10 op 
230 a5d310bc 2020-11-10 op 	return FILE_EXISTS;
231 a5d310bc 2020-11-10 op }
232 a5d310bc 2020-11-10 op 
233 a5d310bc 2020-11-10 op int
234 33d32d1f 2020-12-25 op open_file(char *fpath, char *query, struct pollfd *fds, struct client *c)
235 a5d310bc 2020-11-10 op {
236 d7802bb4 2020-12-02 op 	switch (check_path(c, fpath, &c->fd)) {
237 a5d310bc 2020-11-10 op 	case FILE_EXECUTABLE:
238 15902770 2021-01-15 op 		if (c->host->cgi != NULL && starts_with(fpath, c->host->cgi))
239 a5d310bc 2020-11-10 op 			return start_cgi(fpath, "", query, fds, c);
240 3e4749f7 2020-10-02 op 
241 a5d310bc 2020-11-10 op 		/* fallthrough */
242 a5d310bc 2020-11-10 op 
243 a5d310bc 2020-11-10 op 	case FILE_EXISTS:
244 a5d310bc 2020-11-10 op 		if ((c->len = filesize(c->fd)) == -1) {
245 d7802bb4 2020-12-02 op 			LOGE(c, "failed to get file size for %s", fpath);
246 a5d310bc 2020-11-10 op 			goodbye(fds, c);
247 72342dc9 2020-11-06 op 			return 0;
248 a5d310bc 2020-11-10 op 		}
249 72342dc9 2020-11-06 op 
250 a5d310bc 2020-11-10 op 		if ((c->buf = mmap(NULL, c->len, PROT_READ, MAP_PRIVATE,
251 a5d310bc 2020-11-10 op 			    c->fd, 0)) == MAP_FAILED) {
252 a5d310bc 2020-11-10 op 			warn("mmap: %s", fpath);
253 a5d310bc 2020-11-10 op 			goodbye(fds, c);
254 a5d310bc 2020-11-10 op 			return 0;
255 a5d310bc 2020-11-10 op 		}
256 a5d310bc 2020-11-10 op 		c->i = c->buf;
257 a5d310bc 2020-11-10 op 		return start_reply(fds, c, SUCCESS, mime(fpath));
258 a5d310bc 2020-11-10 op 
259 a5d310bc 2020-11-10 op 	case FILE_DIRECTORY:
260 d7802bb4 2020-12-02 op 		LOGD(c, "%s is a directory, trying %s/index.gmi", fpath, fpath);
261 f28f9311 2020-10-14 op 		close(c->fd);
262 f28f9311 2020-10-14 op 		c->fd = -1;
263 f28f9311 2020-10-14 op 		send_dir(fpath, fds, c);
264 f28f9311 2020-10-14 op 		return 0;
265 3e4749f7 2020-10-02 op 
266 a5d310bc 2020-11-10 op 	case FILE_MISSING:
267 15902770 2021-01-15 op 		if (c->host->cgi != NULL && starts_with(fpath, c->host->cgi))
268 a5d310bc 2020-11-10 op 			return check_for_cgi(fpath, query, fds, c);
269 72342dc9 2020-11-06 op 
270 a5d310bc 2020-11-10 op 		if (!start_reply(fds, c, NOT_FOUND, "not found"))
271 a5d310bc 2020-11-10 op 			return 0;
272 f28f9311 2020-10-14 op 		goodbye(fds, c);
273 f28f9311 2020-10-14 op 		return 0;
274 f28f9311 2020-10-14 op 
275 a5d310bc 2020-11-10 op 	default:
276 a5d310bc 2020-11-10 op 		/* unreachable */
277 a5d310bc 2020-11-10 op 		abort();
278 592fd624 2020-10-07 op 	}
279 72342dc9 2020-11-06 op }
280 72342dc9 2020-11-06 op 
281 72342dc9 2020-11-06 op void
282 75d233f0 2020-11-06 op send_file(char *path, char *query, struct pollfd *fds, struct client *c)
283 f28f9311 2020-10-14 op {
284 f28f9311 2020-10-14 op 	ssize_t ret, len;
285 592fd624 2020-10-07 op 
286 f28f9311 2020-10-14 op 	if (c->fd == -1) {
287 75d233f0 2020-11-06 op 		if (!open_file(path, query, fds, c))
288 f28f9311 2020-10-14 op 			return;
289 f28f9311 2020-10-14 op 		c->state = S_SENDING;
290 f28f9311 2020-10-14 op 	}
291 592fd624 2020-10-07 op 
292 f28f9311 2020-10-14 op 	len = (c->buf + c->len) - c->i;
293 f28f9311 2020-10-14 op 
294 f28f9311 2020-10-14 op 	while (len > 0) {
295 f28f9311 2020-10-14 op 		switch (ret = tls_write(c->ctx, c->i, len)) {
296 f28f9311 2020-10-14 op 		case -1:
297 d7802bb4 2020-12-02 op 			LOGE(c, "tls_write: %s", tls_error(c->ctx));
298 f28f9311 2020-10-14 op 			goodbye(fds, c);
299 f28f9311 2020-10-14 op 			return;
300 f28f9311 2020-10-14 op 
301 f28f9311 2020-10-14 op 		case TLS_WANT_POLLIN:
302 f28f9311 2020-10-14 op 			fds->events = POLLIN;
303 f28f9311 2020-10-14 op 			return;
304 f28f9311 2020-10-14 op 
305 f28f9311 2020-10-14 op 		case TLS_WANT_POLLOUT:
306 f28f9311 2020-10-14 op 			fds->events = POLLOUT;
307 f28f9311 2020-10-14 op 			return;
308 f28f9311 2020-10-14 op 
309 f28f9311 2020-10-14 op 		default:
310 f28f9311 2020-10-14 op 			c->i += ret;
311 f28f9311 2020-10-14 op 			len -= ret;
312 f28f9311 2020-10-14 op 			break;
313 3e4749f7 2020-10-02 op 		}
314 3e4749f7 2020-10-02 op 	}
315 f28f9311 2020-10-14 op 
316 f28f9311 2020-10-14 op 	goodbye(fds, c);
317 3e4749f7 2020-10-02 op }
318 3e4749f7 2020-10-02 op 
319 3e4749f7 2020-10-02 op void
320 592fd624 2020-10-07 op send_dir(char *path, struct pollfd *fds, struct client *client)
321 3e4749f7 2020-10-02 op {
322 3e4749f7 2020-10-02 op 	char fpath[PATHBUF];
323 3e4749f7 2020-10-02 op 	size_t len;
324 3e4749f7 2020-10-02 op 
325 3e4749f7 2020-10-02 op 	bzero(fpath, PATHBUF);
326 3e4749f7 2020-10-02 op 
327 9c56b0a7 2020-10-14 op 	if (path[0] != '.')
328 3e4749f7 2020-10-02 op 		fpath[0] = '.';
329 3e4749f7 2020-10-02 op 
330 3e4749f7 2020-10-02 op 	/* this cannot fail since sizeof(fpath) > maxlen of path */
331 3e4749f7 2020-10-02 op 	strlcat(fpath, path, PATHBUF);
332 3e4749f7 2020-10-02 op 	len = strlen(fpath);
333 3e4749f7 2020-10-02 op 
334 3e4749f7 2020-10-02 op 	/* add a trailing / in case. */
335 3e4749f7 2020-10-02 op 	if (fpath[len-1] != '/') {
336 3e4749f7 2020-10-02 op 		fpath[len] = '/';
337 3e4749f7 2020-10-02 op 	}
338 3e4749f7 2020-10-02 op 
339 3e4749f7 2020-10-02 op 	strlcat(fpath, "index.gmi", sizeof(fpath));
340 3e4749f7 2020-10-02 op 
341 75d233f0 2020-11-06 op 	send_file(fpath, NULL, fds, client);
342 3e4749f7 2020-10-02 op }
343 3e4749f7 2020-10-02 op 
344 3e4749f7 2020-10-02 op void
345 9862b637 2021-01-13 op handle_handshake(struct pollfd *fds, struct client *c)
346 3e4749f7 2020-10-02 op {
347 15902770 2021-01-15 op 	struct vhost *h;
348 15902770 2021-01-15 op 	const char *servname;
349 15902770 2021-01-15 op 
350 9862b637 2021-01-13 op 	switch (tls_handshake(c->ctx)) {
351 9862b637 2021-01-13 op 	case 0:  /* success */
352 9862b637 2021-01-13 op 	case -1: /* already handshaked */
353 9862b637 2021-01-13 op 		break;
354 9862b637 2021-01-13 op 	case TLS_WANT_POLLIN:
355 9862b637 2021-01-13 op 		fds->events = POLLIN;
356 9862b637 2021-01-13 op 		return;
357 9862b637 2021-01-13 op 	case TLS_WANT_POLLOUT:
358 9862b637 2021-01-13 op 		fds->events = POLLOUT;
359 9862b637 2021-01-13 op 		return;
360 9862b637 2021-01-13 op 	default:
361 9862b637 2021-01-13 op 		/* unreachable */
362 9862b637 2021-01-13 op 		abort();
363 9862b637 2021-01-13 op 	}
364 9862b637 2021-01-13 op 
365 15902770 2021-01-15 op 	servname = tls_conn_servername(c->ctx);
366 15902770 2021-01-15 op 	if (servname == NULL)
367 f7b816dc 2021-01-15 op 		goto hostnotfound;
368 9862b637 2021-01-13 op 
369 15902770 2021-01-15 op 	for (h = hosts; h->domain != NULL; ++h) {
370 15902770 2021-01-15 op 		if (!strcmp(h->domain, servname) || !strcmp(h->domain, "*"))
371 15902770 2021-01-15 op 			break;
372 15902770 2021-01-15 op 	}
373 15902770 2021-01-15 op 
374 15902770 2021-01-15 op 	if (h->domain != NULL) {
375 15902770 2021-01-15 op 		c->state = S_OPEN;
376 15902770 2021-01-15 op 		c->host = h;
377 15902770 2021-01-15 op 		handle_open_conn(fds, c);
378 15902770 2021-01-15 op 		return;
379 15902770 2021-01-15 op 	}
380 15902770 2021-01-15 op 
381 f7b816dc 2021-01-15 op hostnotfound:
382 15902770 2021-01-15 op 	/* XXX: check the correct response */
383 15902770 2021-01-15 op 	if (!start_reply(fds, c, BAD_REQUEST, "Wrong host or missing SNI"))
384 15902770 2021-01-15 op 		return;
385 15902770 2021-01-15 op 	goodbye(fds, c);
386 9862b637 2021-01-13 op }
387 9862b637 2021-01-13 op 
388 9862b637 2021-01-13 op void
389 9862b637 2021-01-13 op handle_open_conn(struct pollfd *fds, struct client *c)
390 9862b637 2021-01-13 op {
391 592fd624 2020-10-07 op 	char buf[GEMINI_URL_LEN];
392 9862b637 2021-01-13 op 	const char *parse_err = "invalid request";
393 3c1cf9d0 2021-01-11 op 	struct iri iri;
394 3e4749f7 2020-10-02 op 
395 9862b637 2021-01-13 op 	bzero(buf, sizeof(buf));
396 9862b637 2021-01-13 op 
397 9862b637 2021-01-13 op 	switch (tls_read(c->ctx, buf, sizeof(buf)-1)) {
398 9862b637 2021-01-13 op 	case -1:
399 9862b637 2021-01-13 op 		LOGE(c, "tls_read: %s", tls_error(c->ctx));
400 9862b637 2021-01-13 op 		goodbye(fds, c);
401 9862b637 2021-01-13 op 		return;
402 9862b637 2021-01-13 op 
403 9862b637 2021-01-13 op 	case TLS_WANT_POLLIN:
404 9862b637 2021-01-13 op 		fds->events = POLLIN;
405 9862b637 2021-01-13 op 		return;
406 9862b637 2021-01-13 op 
407 9862b637 2021-01-13 op 	case TLS_WANT_POLLOUT:
408 9862b637 2021-01-13 op 		fds->events = POLLOUT;
409 9862b637 2021-01-13 op 		return;
410 9862b637 2021-01-13 op 	}
411 9862b637 2021-01-13 op 
412 9862b637 2021-01-13 op 	if (!trim_req_iri(buf) || !parse_iri(buf, &iri, &parse_err)) {
413 9862b637 2021-01-13 op 		if (!start_reply(fds, c, BAD_REQUEST, parse_err))
414 592fd624 2020-10-07 op 			return;
415 9862b637 2021-01-13 op 		goodbye(fds, c);
416 9862b637 2021-01-13 op 		return;
417 9862b637 2021-01-13 op 	}
418 3e4749f7 2020-10-02 op 
419 b777bf4b 2021-01-15 op 	if (strcmp(iri.schema, "gemini") || iri.port_no != conf.port) {
420 9862b637 2021-01-13 op 		if (!start_reply(fds, c, PROXY_REFUSED, "won't proxy request"))
421 592fd624 2020-10-07 op 			return;
422 9862b637 2021-01-13 op 		goodbye(fds, c);
423 9862b637 2021-01-13 op 		return;
424 9862b637 2021-01-13 op 	}
425 3e4749f7 2020-10-02 op 
426 9862b637 2021-01-13 op 	LOGI(c, "GET %s%s%s",
427 f7b816dc 2021-01-15 op 	    *iri.path ? iri.path : "/",
428 f7b816dc 2021-01-15 op 	    *iri.query ? "?" : "",
429 f7b816dc 2021-01-15 op 	    *iri.query ? iri.query : "");
430 3e4749f7 2020-10-02 op 
431 9862b637 2021-01-13 op 	send_file(iri.path, iri.query, fds, c);
432 9862b637 2021-01-13 op }
433 592fd624 2020-10-07 op 
434 9862b637 2021-01-13 op void
435 9862b637 2021-01-13 op handle(struct pollfd *fds, struct client *client)
436 9862b637 2021-01-13 op {
437 9862b637 2021-01-13 op 	switch (client->state) {
438 9862b637 2021-01-13 op 	case S_HANDSHAKE:
439 9862b637 2021-01-13 op 		handle_handshake(fds, client);
440 9862b637 2021-01-13 op 		break;
441 592fd624 2020-10-07 op 
442 9862b637 2021-01-13 op 	case S_OPEN:
443 9862b637 2021-01-13 op                 handle_open_conn(fds, client);
444 592fd624 2020-10-07 op 		break;
445 592fd624 2020-10-07 op 
446 592fd624 2020-10-07 op 	case S_INITIALIZING:
447 592fd624 2020-10-07 op 		if (!start_reply(fds, client, client->code, client->meta))
448 592fd624 2020-10-07 op 			return;
449 592fd624 2020-10-07 op 
450 592fd624 2020-10-07 op 		if (client->code != SUCCESS) {
451 592fd624 2020-10-07 op 			/* we don't need a body */
452 592fd624 2020-10-07 op 			goodbye(fds, client);
453 592fd624 2020-10-07 op 			return;
454 592fd624 2020-10-07 op 		}
455 592fd624 2020-10-07 op 
456 592fd624 2020-10-07 op 		client->state = S_SENDING;
457 592fd624 2020-10-07 op 
458 592fd624 2020-10-07 op 		/* fallthrough */
459 592fd624 2020-10-07 op 
460 592fd624 2020-10-07 op 	case S_SENDING:
461 aff8d190 2020-11-06 op 		if (client->child != -1)
462 aff8d190 2020-11-06 op 			handle_cgi(fds, client);
463 aff8d190 2020-11-06 op 		else
464 aff8d190 2020-11-06 op 			send_file(NULL, NULL, fds, client);
465 592fd624 2020-10-07 op 		break;
466 592fd624 2020-10-07 op 
467 592fd624 2020-10-07 op 	case S_CLOSING:
468 592fd624 2020-10-07 op 		goodbye(fds, client);
469 592fd624 2020-10-07 op 		break;
470 592fd624 2020-10-07 op 
471 592fd624 2020-10-07 op 	default:
472 592fd624 2020-10-07 op 		/* unreachable */
473 592fd624 2020-10-07 op 		abort();
474 592fd624 2020-10-07 op 	}
475 3e4749f7 2020-10-02 op }
476 3e4749f7 2020-10-02 op 
477 592fd624 2020-10-07 op void
478 592fd624 2020-10-07 op mark_nonblock(int fd)
479 592fd624 2020-10-07 op {
480 592fd624 2020-10-07 op 	int flags;
481 592fd624 2020-10-07 op 
482 592fd624 2020-10-07 op 	if ((flags = fcntl(fd, F_GETFL)) == -1)
483 80bbcad5 2021-01-10 op 		fatal("fcntl(F_GETFL): %s", strerror(errno));
484 592fd624 2020-10-07 op 	if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1)
485 80bbcad5 2021-01-10 op 		fatal("fcntl(F_SETFL): %s", strerror(errno));
486 592fd624 2020-10-07 op }
487 592fd624 2020-10-07 op 
488 3e4749f7 2020-10-02 op int
489 9468027b 2020-10-15 op make_socket(int port, int family)
490 3e4749f7 2020-10-02 op {
491 3e4749f7 2020-10-02 op 	int sock, v;
492 9468027b 2020-10-15 op 	struct sockaddr_in addr4;
493 9468027b 2020-10-15 op 	struct sockaddr_in6 addr6;
494 9468027b 2020-10-15 op 	struct sockaddr *addr;
495 9468027b 2020-10-15 op 	socklen_t len;
496 3e4749f7 2020-10-02 op 
497 9468027b 2020-10-15 op         switch (family) {
498 9468027b 2020-10-15 op 	case AF_INET:
499 9468027b 2020-10-15 op 		bzero(&addr4, sizeof(addr4));
500 9468027b 2020-10-15 op 		addr4.sin_family = family;
501 9468027b 2020-10-15 op 		addr4.sin_port = htons(port);
502 9468027b 2020-10-15 op 		addr4.sin_addr.s_addr = INADDR_ANY;
503 9468027b 2020-10-15 op 		addr = (struct sockaddr*)&addr4;
504 9468027b 2020-10-15 op 		len = sizeof(addr4);
505 9468027b 2020-10-15 op 		break;
506 9468027b 2020-10-15 op 
507 9468027b 2020-10-15 op 	case AF_INET6:
508 9468027b 2020-10-15 op 		bzero(&addr6, sizeof(addr6));
509 9468027b 2020-10-15 op 		addr6.sin6_family = AF_INET6;
510 9468027b 2020-10-15 op 		addr6.sin6_port = htons(port);
511 9468027b 2020-10-15 op 		addr6.sin6_addr = in6addr_any;
512 9468027b 2020-10-15 op 		addr = (struct sockaddr*)&addr6;
513 9468027b 2020-10-15 op 		len = sizeof(addr6);
514 9468027b 2020-10-15 op 		break;
515 9468027b 2020-10-15 op 
516 9468027b 2020-10-15 op 	default:
517 9468027b 2020-10-15 op 		/* unreachable */
518 9468027b 2020-10-15 op 		abort();
519 9468027b 2020-10-15 op 	}
520 9468027b 2020-10-15 op 
521 9468027b 2020-10-15 op 	if ((sock = socket(family, SOCK_STREAM, 0)) == -1)
522 80bbcad5 2021-01-10 op 		fatal("socket: %s", strerror(errno));
523 3e4749f7 2020-10-02 op 
524 3e4749f7 2020-10-02 op 	v = 1;
525 3e4749f7 2020-10-02 op 	if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &v, sizeof(v)) == -1)
526 80bbcad5 2021-01-10 op 		fatal("setsockopt(SO_REUSEADDR): %s", strerror(errno));
527 3e4749f7 2020-10-02 op 
528 3e4749f7 2020-10-02 op 	v = 1;
529 3e4749f7 2020-10-02 op 	if (setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, &v, sizeof(v)) == -1)
530 80bbcad5 2021-01-10 op 		fatal("setsockopt(SO_REUSEPORT): %s", strerror(errno));
531 3e4749f7 2020-10-02 op 
532 592fd624 2020-10-07 op 	mark_nonblock(sock);
533 592fd624 2020-10-07 op 
534 9468027b 2020-10-15 op 	if (bind(sock, addr, len) == -1)
535 80bbcad5 2021-01-10 op 		fatal("bind: %s", strerror(errno));
536 3e4749f7 2020-10-02 op 
537 3e4749f7 2020-10-02 op 	if (listen(sock, 16) == -1)
538 80bbcad5 2021-01-10 op 		fatal("listen: %s", strerror(errno));
539 3e4749f7 2020-10-02 op 
540 3e4749f7 2020-10-02 op 	return sock;
541 3e4749f7 2020-10-02 op }
542 3e4749f7 2020-10-02 op 
543 3e4749f7 2020-10-02 op void
544 592fd624 2020-10-07 op do_accept(int sock, struct tls *ctx, struct pollfd *fds, struct client *clients)
545 3e4749f7 2020-10-02 op {
546 592fd624 2020-10-07 op 	int i, fd;
547 709d6e5e 2021-01-10 op 	struct sockaddr_storage addr;
548 3e4749f7 2020-10-02 op 	socklen_t len;
549 3e4749f7 2020-10-02 op 
550 592fd624 2020-10-07 op 	len = sizeof(addr);
551 592fd624 2020-10-07 op 	if ((fd = accept(sock, (struct sockaddr*)&addr, &len)) == -1) {
552 592fd624 2020-10-07 op 		if (errno == EWOULDBLOCK)
553 592fd624 2020-10-07 op 			return;
554 80bbcad5 2021-01-10 op 		fatal("accept: %s", strerror(errno));
555 592fd624 2020-10-07 op 	}
556 3e4749f7 2020-10-02 op 
557 592fd624 2020-10-07 op 	mark_nonblock(fd);
558 3e4749f7 2020-10-02 op 
559 592fd624 2020-10-07 op 	for (i = 0; i < MAX_USERS; ++i) {
560 592fd624 2020-10-07 op 		if (fds[i].fd == -1) {
561 592fd624 2020-10-07 op 			bzero(&clients[i], sizeof(struct client));
562 592fd624 2020-10-07 op 			if (tls_accept_socket(ctx, &clients[i].ctx, fd) == -1)
563 592fd624 2020-10-07 op 				break; /* goodbye fd! */
564 592fd624 2020-10-07 op 
565 592fd624 2020-10-07 op 			fds[i].fd = fd;
566 592fd624 2020-10-07 op 			fds[i].events = POLLIN;
567 592fd624 2020-10-07 op 
568 9862b637 2021-01-13 op 			clients[i].state = S_HANDSHAKE;
569 592fd624 2020-10-07 op 			clients[i].fd = -1;
570 72342dc9 2020-11-06 op 			clients[i].child = -1;
571 f28f9311 2020-10-14 op 			clients[i].buf = MAP_FAILED;
572 2c3a40fa 2020-11-06 op 			clients[i].af = AF_INET;
573 709d6e5e 2021-01-10 op 			clients[i].addr = addr;
574 592fd624 2020-10-07 op 
575 120381c9 2020-11-06 op 			connected_clients++;
576 592fd624 2020-10-07 op 			return;
577 3e4749f7 2020-10-02 op 		}
578 592fd624 2020-10-07 op 	}
579 3e4749f7 2020-10-02 op 
580 592fd624 2020-10-07 op 	close(fd);
581 592fd624 2020-10-07 op }
582 3e4749f7 2020-10-02 op 
583 592fd624 2020-10-07 op void
584 592fd624 2020-10-07 op goodbye(struct pollfd *pfd, struct client *c)
585 592fd624 2020-10-07 op {
586 592fd624 2020-10-07 op 	c->state = S_CLOSING;
587 592fd624 2020-10-07 op 
588 f7b816dc 2021-01-15 op 	switch (tls_close(c->ctx)) {
589 f7b816dc 2021-01-15 op 	case TLS_WANT_POLLIN:
590 592fd624 2020-10-07 op 		pfd->events = POLLIN;
591 592fd624 2020-10-07 op 		return;
592 f7b816dc 2021-01-15 op 	case TLS_WANT_POLLOUT:
593 592fd624 2020-10-07 op 		pfd->events = POLLOUT;
594 592fd624 2020-10-07 op 		return;
595 592fd624 2020-10-07 op 	}
596 592fd624 2020-10-07 op 
597 120381c9 2020-11-06 op 	connected_clients--;
598 120381c9 2020-11-06 op 
599 592fd624 2020-10-07 op 	tls_free(c->ctx);
600 592fd624 2020-10-07 op 	c->ctx = NULL;
601 592fd624 2020-10-07 op 
602 f28f9311 2020-10-14 op 	if (c->buf != MAP_FAILED)
603 f28f9311 2020-10-14 op 		munmap(c->buf, c->len);
604 f28f9311 2020-10-14 op 
605 592fd624 2020-10-07 op 	if (c->fd != -1)
606 592fd624 2020-10-07 op 		close(c->fd);
607 592fd624 2020-10-07 op 
608 592fd624 2020-10-07 op 	close(pfd->fd);
609 592fd624 2020-10-07 op 	pfd->fd = -1;
610 3e4749f7 2020-10-02 op }
611 3e4749f7 2020-10-02 op 
612 3e4749f7 2020-10-02 op void
613 33756bd2 2021-01-10 op loop(struct tls *ctx, int sock4, int sock6)
614 592fd624 2020-10-07 op {
615 16ace04d 2021-01-08 op 	int i;
616 592fd624 2020-10-07 op 	struct client clients[MAX_USERS];
617 592fd624 2020-10-07 op 	struct pollfd fds[MAX_USERS];
618 592fd624 2020-10-07 op 
619 592fd624 2020-10-07 op 	for (i = 0; i < MAX_USERS; ++i) {
620 592fd624 2020-10-07 op 		fds[i].fd = -1;
621 592fd624 2020-10-07 op 		fds[i].events = POLLIN;
622 592fd624 2020-10-07 op 		bzero(&clients[i], sizeof(struct client));
623 592fd624 2020-10-07 op 	}
624 592fd624 2020-10-07 op 
625 33756bd2 2021-01-10 op 	fds[0].fd = sock4;
626 33756bd2 2021-01-10 op 	fds[1].fd = sock6;
627 592fd624 2020-10-07 op 
628 592fd624 2020-10-07 op 	for (;;) {
629 16ace04d 2021-01-08 op 		if (poll(fds, MAX_USERS, INFTIM) == -1) {
630 120381c9 2020-11-06 op 			if (errno == EINTR) {
631 d7802bb4 2020-12-02 op                                 warnx("connected clients: %d",
632 d7802bb4 2020-12-02 op 				    connected_clients);
633 120381c9 2020-11-06 op 				continue;
634 120381c9 2020-11-06 op 			}
635 80bbcad5 2021-01-10 op 			fatal("poll: %s", strerror(errno));
636 120381c9 2020-11-06 op 		}
637 592fd624 2020-10-07 op 
638 592fd624 2020-10-07 op 		for (i = 0; i < MAX_USERS; i++) {
639 592fd624 2020-10-07 op 			if (fds[i].revents == 0)
640 592fd624 2020-10-07 op 				continue;
641 592fd624 2020-10-07 op 
642 592fd624 2020-10-07 op 			if (fds[i].revents & (POLLERR|POLLNVAL))
643 80bbcad5 2021-01-10 op 				fatal("bad fd %d: %s", fds[i].fd,
644 d7802bb4 2020-12-02 op 				    strerror(errno));
645 592fd624 2020-10-07 op 
646 592fd624 2020-10-07 op 			if (fds[i].revents & POLLHUP) {
647 aff8d190 2020-11-06 op 				/* fds[i] may be the fd of the stdin
648 aff8d190 2020-11-06 op 				 * of a cgi script that has exited. */
649 aff8d190 2020-11-06 op 				if (!clients[i].waiting_on_child) {
650 aff8d190 2020-11-06 op 					goodbye(&fds[i], &clients[i]);
651 aff8d190 2020-11-06 op 					continue;
652 aff8d190 2020-11-06 op 				}
653 592fd624 2020-10-07 op 			}
654 592fd624 2020-10-07 op 
655 33756bd2 2021-01-10 op 			if (fds[i].fd == sock4)
656 33756bd2 2021-01-10 op 				do_accept(sock4, ctx, fds, clients);
657 33756bd2 2021-01-10 op 			else if (fds[i].fd == sock6)
658 33756bd2 2021-01-10 op 				do_accept(sock6, ctx, fds, clients);
659 33756bd2 2021-01-10 op 			else
660 33756bd2 2021-01-10 op 				handle(&fds[i], &clients[i]);
661 592fd624 2020-10-07 op 		}
662 592fd624 2020-10-07 op 	}
663 231bfcdc 2020-12-21 op }
664 231bfcdc 2020-12-21 op 
665 231bfcdc 2020-12-21 op char *
666 231bfcdc 2020-12-21 op absolutify_path(const char *path)
667 231bfcdc 2020-12-21 op {
668 231bfcdc 2020-12-21 op 	char *wd, *r;
669 231bfcdc 2020-12-21 op 
670 231bfcdc 2020-12-21 op 	if (*path == '/')
671 231bfcdc 2020-12-21 op 		return strdup(path);
672 231bfcdc 2020-12-21 op 
673 231bfcdc 2020-12-21 op 	wd = getwd(NULL);
674 231bfcdc 2020-12-21 op 	if (asprintf(&r, "%s/%s", wd, path) == -1)
675 231bfcdc 2020-12-21 op 		err(1, "asprintf");
676 231bfcdc 2020-12-21 op 	free(wd);
677 231bfcdc 2020-12-21 op 	return r;
678 15902770 2021-01-15 op }
679 15902770 2021-01-15 op 
680 15902770 2021-01-15 op void
681 15902770 2021-01-15 op yyerror(const char *msg)
682 15902770 2021-01-15 op {
683 15902770 2021-01-15 op 	goterror = 1;
684 15902770 2021-01-15 op 	fprintf(stderr, "%d: %s\n", yylineno, msg);
685 15902770 2021-01-15 op }
686 15902770 2021-01-15 op 
687 15902770 2021-01-15 op int
688 15902770 2021-01-15 op parse_portno(const char *p)
689 15902770 2021-01-15 op {
690 15902770 2021-01-15 op 	char *ep;
691 15902770 2021-01-15 op 	long lval;
692 15902770 2021-01-15 op 
693 15902770 2021-01-15 op 	errno = 0;
694 15902770 2021-01-15 op 	lval = strtol(p, &ep, 10);
695 15902770 2021-01-15 op 	if (p[0] == '\0' || *ep != '\0')
696 15902770 2021-01-15 op 		errx(1, "not a number: %s", p);
697 15902770 2021-01-15 op 	if (lval < 0 || lval > UINT16_MAX)
698 15902770 2021-01-15 op 		errx(1, "port number out of range for domain %s: %ld", p, lval);
699 15902770 2021-01-15 op 	return lval;
700 592fd624 2020-10-07 op }
701 592fd624 2020-10-07 op 
702 592fd624 2020-10-07 op void
703 15902770 2021-01-15 op parse_conf(const char *path)
704 15902770 2021-01-15 op {
705 15902770 2021-01-15 op 	if ((yyin = fopen(path, "r")) == NULL)
706 15902770 2021-01-15 op 		err(1, "cannot open config %s", path);
707 15902770 2021-01-15 op 	yyparse();
708 15902770 2021-01-15 op 	fclose(yyin);
709 15902770 2021-01-15 op 
710 15902770 2021-01-15 op 	if (goterror)
711 15902770 2021-01-15 op 		exit(1);
712 15902770 2021-01-15 op }
713 15902770 2021-01-15 op 
714 15902770 2021-01-15 op void
715 15902770 2021-01-15 op load_vhosts(struct tls_config *tlsconf)
716 15902770 2021-01-15 op {
717 15902770 2021-01-15 op 	struct vhost *h;
718 15902770 2021-01-15 op 
719 15902770 2021-01-15 op 	/* we need to set something, then we can add how many key we want */
720 15902770 2021-01-15 op 	if (tls_config_set_keypair_file(tlsconf, hosts->cert, hosts->key))
721 15902770 2021-01-15 op 		errx(1, "tls_config_set_keypair_file failed");
722 15902770 2021-01-15 op 
723 15902770 2021-01-15 op 	for (h = hosts; h->domain != NULL; ++h) {
724 15902770 2021-01-15 op 		if (tls_config_add_keypair_file(tlsconf, h->cert, h->key) == -1)
725 15902770 2021-01-15 op 			errx(1, "failed to load the keypair (%s, %s)",
726 15902770 2021-01-15 op 			    h->cert, h->key);
727 15902770 2021-01-15 op 
728 15902770 2021-01-15 op 		if ((h->dirfd = open(h->dir, O_RDONLY | O_DIRECTORY)) == -1)
729 15902770 2021-01-15 op 			err(1, "open %s for domain %s", h->dir, h->domain);
730 15902770 2021-01-15 op 	}
731 15902770 2021-01-15 op }
732 15902770 2021-01-15 op 
733 15902770 2021-01-15 op void
734 3e4749f7 2020-10-02 op usage(const char *me)
735 3e4749f7 2020-10-02 op {
736 3e4749f7 2020-10-02 op 	fprintf(stderr,
737 15902770 2021-01-15 op 	    "USAGE: %s [-n] [-c config] | [-6fh] [-C cert] [-d root] [-K key] "
738 15902770 2021-01-15 op 	    "[-p port] [-x cgi-bin]\n",
739 3e4749f7 2020-10-02 op 	    me);
740 3e4749f7 2020-10-02 op }
741 3e4749f7 2020-10-02 op 
742 3e4749f7 2020-10-02 op int
743 3e4749f7 2020-10-02 op main(int argc, char **argv)
744 3e4749f7 2020-10-02 op {
745 3e4749f7 2020-10-02 op 	struct tls *ctx = NULL;
746 15902770 2021-01-15 op 	struct tls_config *tlsconf;
747 15902770 2021-01-15 op 	int sock4, sock6, ch;
748 15902770 2021-01-15 op 	const char *config_path = NULL;
749 15902770 2021-01-15 op 	size_t i;
750 dafb57b8 2021-01-15 op 	int conftest = 0;
751 120381c9 2020-11-06 op 
752 15902770 2021-01-15 op 	bzero(hosts, sizeof(hosts));
753 15902770 2021-01-15 op 	for (i = 0; i < HOSTSLEN; ++i)
754 15902770 2021-01-15 op 		hosts[i].dirfd = -1;
755 51d876f0 2020-12-21 op 
756 15902770 2021-01-15 op 	conf.foreground = 1;
757 15902770 2021-01-15 op 	conf.port = 1965;
758 15902770 2021-01-15 op 	conf.ipv6 = 0;
759 5bc3c98e 2021-01-15 op 	conf.protos = TLS_PROTOCOL_TLSv1_2 | TLS_PROTOCOL_TLSv1_3;
760 85dff1f9 2021-01-11 op 
761 15902770 2021-01-15 op 	connected_clients = 0;
762 15902770 2021-01-15 op 
763 15902770 2021-01-15 op 	while ((ch = getopt(argc, argv, "6C:c:d:fhK:np:x:")) != -1) {
764 3e4749f7 2020-10-02 op 		switch (ch) {
765 85dff1f9 2021-01-11 op 		case '6':
766 15902770 2021-01-15 op 			conf.ipv6 = 1;
767 85dff1f9 2021-01-11 op 			break;
768 85dff1f9 2021-01-11 op 
769 15902770 2021-01-15 op 		case 'C':
770 15902770 2021-01-15 op 			hosts[0].cert = optarg;
771 15902770 2021-01-15 op 			break;
772 15902770 2021-01-15 op 
773 3e4749f7 2020-10-02 op 		case 'c':
774 15902770 2021-01-15 op 			config_path = optarg;
775 3e4749f7 2020-10-02 op 			break;
776 3e4749f7 2020-10-02 op 
777 3e4749f7 2020-10-02 op 		case 'd':
778 15902770 2021-01-15 op 			free((char*)hosts[0].dir);
779 15902770 2021-01-15 op 			if ((hosts[0].dir = absolutify_path(optarg)) == NULL)
780 231bfcdc 2020-12-21 op 				err(1, "absolutify_path");
781 3e4749f7 2020-10-02 op 			break;
782 3e4749f7 2020-10-02 op 
783 d7802bb4 2020-12-02 op 		case 'f':
784 15902770 2021-01-15 op 			conf.foreground = 1;
785 d7802bb4 2020-12-02 op 			break;
786 d7802bb4 2020-12-02 op 
787 3e4749f7 2020-10-02 op 		case 'h':
788 3e4749f7 2020-10-02 op 			usage(*argv);
789 3e4749f7 2020-10-02 op 			return 0;
790 3e4749f7 2020-10-02 op 
791 15902770 2021-01-15 op 		case 'K':
792 15902770 2021-01-15 op 			hosts[0].key = optarg;
793 3e4749f7 2020-10-02 op 			break;
794 3e4749f7 2020-10-02 op 
795 15902770 2021-01-15 op 		case 'n':
796 15902770 2021-01-15 op 			conftest = 1;
797 721e2325 2020-11-18 op 			break;
798 721e2325 2020-11-18 op 
799 15902770 2021-01-15 op 		case 'p':
800 15902770 2021-01-15 op 			conf.port = parse_portno(optarg);
801 15902770 2021-01-15 op 
802 72342dc9 2020-11-06 op 		case 'x':
803 15902770 2021-01-15 op 			/* drop the starting / (if any) */
804 15902770 2021-01-15 op 			if (*optarg == '/')
805 15902770 2021-01-15 op 				optarg++;
806 15902770 2021-01-15 op 			hosts[0].cgi = optarg;
807 72342dc9 2020-11-06 op 			break;
808 72342dc9 2020-11-06 op 
809 3e4749f7 2020-10-02 op 		default:
810 3e4749f7 2020-10-02 op 			usage(*argv);
811 3e4749f7 2020-10-02 op 			return 1;
812 3e4749f7 2020-10-02 op 		}
813 3e4749f7 2020-10-02 op 	}
814 3e4749f7 2020-10-02 op 
815 15902770 2021-01-15 op 	if (config_path != NULL) {
816 15902770 2021-01-15 op 		if (hosts[0].cert != NULL || hosts[0].key != NULL ||
817 15902770 2021-01-15 op 		    hosts[0].dir != NULL)
818 15902770 2021-01-15 op 			errx(1, "can't specify options in conf mode");
819 15902770 2021-01-15 op 		parse_conf(config_path);
820 15902770 2021-01-15 op 	} else {
821 15902770 2021-01-15 op 		if (hosts[0].cert == NULL || hosts[0].key == NULL ||
822 15902770 2021-01-15 op 		    hosts[0].dir == NULL)
823 15902770 2021-01-15 op 			errx(1, "missing cert, key or root directory to serve");
824 15902770 2021-01-15 op 		hosts[0].domain = "*";
825 15902770 2021-01-15 op 	}
826 15902770 2021-01-15 op 
827 15902770 2021-01-15 op 	if (conftest)
828 15902770 2021-01-15 op                 errx(0, "config OK");
829 15902770 2021-01-15 op 
830 4a28dd01 2020-12-28 op 	signal(SIGPIPE, SIG_IGN);
831 4a28dd01 2020-12-28 op 	signal(SIGCHLD, SIG_IGN);
832 4a28dd01 2020-12-28 op 
833 4a28dd01 2020-12-28 op #ifdef SIGINFO
834 4a28dd01 2020-12-28 op 	signal(SIGINFO, sig_handler);
835 4a28dd01 2020-12-28 op #endif
836 4a28dd01 2020-12-28 op 	signal(SIGUSR2, sig_handler);
837 4a28dd01 2020-12-28 op 
838 15902770 2021-01-15 op 	if (!conf.foreground)
839 4a28dd01 2020-12-28 op 		signal(SIGHUP, SIG_IGN);
840 4a28dd01 2020-12-28 op 
841 15902770 2021-01-15 op 	if ((tlsconf = tls_config_new()) == NULL)
842 3e4749f7 2020-10-02 op 		err(1, "tls_config_new");
843 0d8ca45a 2020-10-03 op 
844 677afbd3 2020-12-02 op 	/* optionally accept client certs, but don't try to verify them */
845 15902770 2021-01-15 op 	tls_config_verify_client_optional(tlsconf);
846 15902770 2021-01-15 op 	tls_config_insecure_noverifycert(tlsconf);
847 677afbd3 2020-12-02 op 
848 5bc3c98e 2021-01-15 op 	if (tls_config_set_protocols(tlsconf, conf.protos) == -1)
849 0d8ca45a 2020-10-03 op 		err(1, "tls_config_set_protocols");
850 3e4749f7 2020-10-02 op 
851 15902770 2021-01-15 op 	load_vhosts(tlsconf);
852 3e4749f7 2020-10-02 op 
853 3e4749f7 2020-10-02 op 	if ((ctx = tls_server()) == NULL)
854 3e4749f7 2020-10-02 op 		err(1, "tls_server");
855 3e4749f7 2020-10-02 op 
856 15902770 2021-01-15 op 	if (tls_configure(ctx, tlsconf) == -1)
857 3e4749f7 2020-10-02 op 		errx(1, "tls_configure: %s", tls_error(ctx));
858 3e4749f7 2020-10-02 op 
859 15902770 2021-01-15 op 	sock4 = make_socket(conf.port, AF_INET);
860 15902770 2021-01-15 op 	if (conf.ipv6)
861 15902770 2021-01-15 op 		sock6 = make_socket(conf.port, AF_INET6);
862 85dff1f9 2021-01-11 op 	else
863 85dff1f9 2021-01-11 op 		sock6 = -1;
864 3e4749f7 2020-10-02 op 
865 15902770 2021-01-15 op 	if (!conf.foreground && daemon(0, 1) == -1)
866 d7802bb4 2020-12-02 op 		exit(1);
867 d7802bb4 2020-12-02 op 
868 15902770 2021-01-15 op 	sandbox();
869 3e4749f7 2020-10-02 op 
870 33756bd2 2021-01-10 op 	loop(ctx, sock4, sock6);
871 3e4749f7 2020-10-02 op 
872 33756bd2 2021-01-10 op 	close(sock4);
873 33756bd2 2021-01-10 op 	close(sock6);
874 3e4749f7 2020-10-02 op 	tls_free(ctx);
875 15902770 2021-01-15 op 	tls_config_free(tlsconf);
876 0d7a38c4 2021-01-11 op 
877 0d7a38c4 2021-01-11 op 	return 0;
878 3e4749f7 2020-10-02 op }