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