op public repos

Blame

Date:: Fri Nov 6 12:01:31 2020 UTC
Message:: implementing CGI – NOT READY YET! This is a first try at implementing CGI scripting. The idea is that, if CGI is explicitly enabled by the user, when a user requires an executable file instead of serving it to the client, that file will be executed and its output fed to the client. There are various pieces that are still lacking, the firsts that comes to mind are: - performance: the handle_cgi just loops ignoring the WANT_POLLIN/POLLOUT and blocking if the child process hasn’t outputted anything. - we don’t parse query variable (yet) - we need to set more variables in the child environment side question: it’s better to set the variables using setenv() or by providing an explicit environment? - document what environment the CGI script will get - improve the horrible unveil/pledge(cgi ? …) but now I can serve “hello-world”-tier script from gmid!
Actions:: History | Blob | Raw File
  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/socket.h>
 19 3e4749f7 2020-10-02 op #include <sys/stat.h>
 20 3e4749f7 2020-10-02 op 
 21 2c3a40fa 2020-11-06 op #include <arpa/inet.h>
 22 3e4749f7 2020-10-02 op #include <netinet/in.h>
 23 3e4749f7 2020-10-02 op 
 24 3e4749f7 2020-10-02 op #include <assert.h>
 25 3e4749f7 2020-10-02 op #include <err.h>
 26 592fd624 2020-10-07 op #include <errno.h>
 27 3e4749f7 2020-10-02 op #include <fcntl.h>
 28 592fd624 2020-10-07 op #include <poll.h>
 29 0cf902af 2020-11-03 op #include <signal.h>
 30 3e4749f7 2020-10-02 op #include <stdio.h>
 31 3e4749f7 2020-10-02 op #include <stdlib.h>
 32 3e4749f7 2020-10-02 op #include <string.h>
 33 3e4749f7 2020-10-02 op #include <tls.h>
 34 3e4749f7 2020-10-02 op #include <unistd.h>
 35 3e4749f7 2020-10-02 op 
 36 3e4749f7 2020-10-02 op #ifndef __OpenBSD__
 37 3e4749f7 2020-10-02 op # define pledge(a, b) 0
 38 3e4749f7 2020-10-02 op # define unveil(a, b) 0
 39 3e4749f7 2020-10-02 op #endif /* __OpenBSD__ */
 40 3e4749f7 2020-10-02 op 
 41 592fd624 2020-10-07 op #ifndef INFTIM
 42 592fd624 2020-10-07 op # define INFTIM -1
 43 592fd624 2020-10-07 op #endif /* INFTIM */
 44 592fd624 2020-10-07 op 
 45 3e4749f7 2020-10-02 op #define GEMINI_URL_LEN (1024+3)	/* URL max len + \r\n + \0 */
 46 3e4749f7 2020-10-02 op 
 47 4d4f0e19 2020-10-03 op /* large enough to hold a copy of a gemini URL and still have extra room */
 48 d95b77a0 2020-10-07 op #define PATHBUF		2048
 49 3e4749f7 2020-10-02 op 
 50 3e4749f7 2020-10-02 op #define SUCCESS		20
 51 72342dc9 2020-11-06 op #define TEMP_FAILURE	40
 52 3e4749f7 2020-10-02 op #define NOT_FOUND	51
 53 3e4749f7 2020-10-02 op #define BAD_REQUEST	59
 54 3e4749f7 2020-10-02 op 
 55 592fd624 2020-10-07 op #ifndef MAX_USERS
 56 592fd624 2020-10-07 op #define MAX_USERS	64
 57 592fd624 2020-10-07 op #endif
 58 592fd624 2020-10-07 op 
 59 592fd624 2020-10-07 op enum {
 60 592fd624 2020-10-07 op 	S_OPEN,
 61 592fd624 2020-10-07 op 	S_INITIALIZING,
 62 592fd624 2020-10-07 op 	S_SENDING,
 63 592fd624 2020-10-07 op 	S_CLOSING,
 64 592fd624 2020-10-07 op };
 65 592fd624 2020-10-07 op 
 66 592fd624 2020-10-07 op struct client {
 67 592fd624 2020-10-07 op 	struct tls	*ctx;
 68 f28f9311 2020-10-14 op 	int		 state;
 69 f28f9311 2020-10-14 op 	int		 code;
 70 592fd624 2020-10-07 op 	const char	*meta;
 71 f28f9311 2020-10-14 op 	int		 fd;
 72 72342dc9 2020-11-06 op 	pid_t		 child;
 73 f28f9311 2020-10-14 op 	void		*buf, *i;
 74 f28f9311 2020-10-14 op 	ssize_t		 len, off;
 75 2c3a40fa 2020-11-06 op 	int		 af;
 76 2c3a40fa 2020-11-06 op 	struct in_addr	 addr;
 77 592fd624 2020-10-07 op };
 78 592fd624 2020-10-07 op 
 79 cc68fe70 2020-10-07 op struct etm {			/* file extension to mime */
 80 cc68fe70 2020-10-07 op 	const char	*mime;
 81 cc68fe70 2020-10-07 op 	const char	*ext;
 82 cc68fe70 2020-10-07 op } filetypes[] = {
 83 cc68fe70 2020-10-07 op 	{"application/pdf",	"pdf"},
 84 cc68fe70 2020-10-07 op 
 85 cc68fe70 2020-10-07 op 	{"image/gif",		"gif"},
 86 cc68fe70 2020-10-07 op 	{"image/jpeg",		"jpg"},
 87 cc68fe70 2020-10-07 op 	{"image/jpeg",		"jpeg"},
 88 cc68fe70 2020-10-07 op 	{"image/png",		"png"},
 89 cc68fe70 2020-10-07 op 	{"image/svg+xml",	"svg"},
 90 cc68fe70 2020-10-07 op 
 91 cc68fe70 2020-10-07 op 	{"text/gemini",		"gemini"},
 92 cc68fe70 2020-10-07 op 	{"text/gemini",		"gmi"},
 93 cc68fe70 2020-10-07 op 	{"text/markdown",	"markdown"},
 94 cc68fe70 2020-10-07 op 	{"text/markdown",	"md"},
 95 cc68fe70 2020-10-07 op 	{"text/plain",		"txt"},
 96 932b001a 2020-11-06 op 	{"text/xml",		"xml"},
 97 cc68fe70 2020-10-07 op 
 98 cc68fe70 2020-10-07 op 	{NULL, NULL}
 99 cc68fe70 2020-10-07 op };
100 cc68fe70 2020-10-07 op 
101 2c3a40fa 2020-11-06 op #define LOG(c, fmt, ...)						\
102 2c3a40fa 2020-11-06 op 	do {								\
103 2c3a40fa 2020-11-06 op 		char buf[INET_ADDRSTRLEN];				\
104 2c3a40fa 2020-11-06 op 		if (inet_ntop((c)->af, &(c)->addr, buf, sizeof(buf)) == NULL) \
105 2c3a40fa 2020-11-06 op 			err(1, "inet_ntop");				\
106 2c3a40fa 2020-11-06 op 		dprintf(logfd, "[%s] " fmt "\n", buf, __VA_ARGS__);	\
107 2c3a40fa 2020-11-06 op 	} while (0)
108 3e4749f7 2020-10-02 op 
109 72342dc9 2020-11-06 op const char *dir;
110 2c3a40fa 2020-11-06 op int dirfd, logfd;
111 72342dc9 2020-11-06 op int cgi;
112 2c3a40fa 2020-11-06 op 
113 592fd624 2020-10-07 op char		*url_after_proto(char*);
114 592fd624 2020-10-07 op char		*url_start_of_request(char*);
115 2c3a40fa 2020-11-06 op int		 url_trim(struct client*, char*);
116 592fd624 2020-10-07 op void		 adjust_path(char*);
117 592fd624 2020-10-07 op int		 path_isdir(char*);
118 f28f9311 2020-10-14 op ssize_t		 filesize(int);
119 592fd624 2020-10-07 op 
120 592fd624 2020-10-07 op int		 start_reply(struct pollfd*, struct client*, int, const char*);
121 cc68fe70 2020-10-07 op const char	*path_ext(const char*);
122 cc68fe70 2020-10-07 op const char	*mime(const char*);
123 f28f9311 2020-10-14 op int		 open_file(char*, struct pollfd*, struct client*);
124 72342dc9 2020-11-06 op void		 start_cgi(const char*, struct pollfd*, struct client*);
125 72342dc9 2020-11-06 op void		 handle_cgi(struct pollfd*, struct client*);
126 592fd624 2020-10-07 op void		 send_file(char*, struct pollfd*, struct client*);
127 592fd624 2020-10-07 op void		 send_dir(char*, struct pollfd*, struct client*);
128 592fd624 2020-10-07 op void		 handle(struct pollfd*, struct client*);
129 592fd624 2020-10-07 op 
130 592fd624 2020-10-07 op void		 mark_nonblock(int);
131 592fd624 2020-10-07 op int		 make_soket(int);
132 592fd624 2020-10-07 op void		 do_accept(int, struct tls*, struct pollfd*, struct client*);
133 592fd624 2020-10-07 op void		 goodbye(struct pollfd*, struct client*);
134 592fd624 2020-10-07 op void		 loop(struct tls*, int);
135 592fd624 2020-10-07 op 
136 592fd624 2020-10-07 op void		 usage(const char*);
137 592fd624 2020-10-07 op 
138 3e4749f7 2020-10-02 op char *
139 3e4749f7 2020-10-02 op url_after_proto(char *url)
140 3e4749f7 2020-10-02 op {
141 3e4749f7 2020-10-02 op 	char *s;
142 3e4749f7 2020-10-02 op 	const char *proto = "gemini";
143 3e4749f7 2020-10-02 op 	const char *marker = "://";
144 3e4749f7 2020-10-02 op 
145 3e4749f7 2020-10-02 op 	if ((s = strstr(url, marker)) == NULL)
146 3e4749f7 2020-10-02 op 		return url;
147 3e4749f7 2020-10-02 op 
148 3e4749f7 2020-10-02 op 	/* not a gemini:// URL */
149 3e4749f7 2020-10-02 op 
150 3e4749f7 2020-10-02 op 	if (s - strlen(proto) < url)
151 3e4749f7 2020-10-02 op 		return NULL;
152 3e4749f7 2020-10-02 op 	/* TODO: */
153 3e4749f7 2020-10-02 op 	/* if (strcmp(s - strlen(proto), proto)) */
154 3e4749f7 2020-10-02 op 	/* 	return NULL; */
155 3e4749f7 2020-10-02 op 
156 3e4749f7 2020-10-02 op 	/* a valid gemini:// URL */
157 3e4749f7 2020-10-02 op 	return s + strlen(marker);
158 3e4749f7 2020-10-02 op }
159 3e4749f7 2020-10-02 op 
160 3e4749f7 2020-10-02 op char *
161 3e4749f7 2020-10-02 op url_start_of_request(char *url)
162 3e4749f7 2020-10-02 op {
163 3e4749f7 2020-10-02 op 	char *s, *t;
164 3e4749f7 2020-10-02 op 
165 3e4749f7 2020-10-02 op 	if ((s = url_after_proto(url)) == NULL)
166 3e4749f7 2020-10-02 op 		return NULL;
167 3e4749f7 2020-10-02 op 
168 3e4749f7 2020-10-02 op 	if ((t = strstr(s, "/")) == NULL)
169 3e4749f7 2020-10-02 op 		return s + strlen(s);
170 3e4749f7 2020-10-02 op 	return t;
171 3e4749f7 2020-10-02 op }
172 3e4749f7 2020-10-02 op 
173 3e4749f7 2020-10-02 op int
174 2c3a40fa 2020-11-06 op url_trim(struct client *c, char *url)
175 3e4749f7 2020-10-02 op {
176 3e4749f7 2020-10-02 op 	const char *e = "\r\n";
177 3e4749f7 2020-10-02 op 	char *s;
178 3e4749f7 2020-10-02 op 
179 3e4749f7 2020-10-02 op 	if ((s = strstr(url, e)) == NULL)
180 3e4749f7 2020-10-02 op 		return 0;
181 3e4749f7 2020-10-02 op 	s[0] = '\0';
182 3e4749f7 2020-10-02 op 	s[1] = '\0';
183 3e4749f7 2020-10-02 op 
184 3e4749f7 2020-10-02 op 	if (s[2] != '\0') {
185 2c3a40fa 2020-11-06 op 		LOG(c, "%s", "request longer than 1024 bytes\n");
186 3e4749f7 2020-10-02 op 		return 0;
187 3e4749f7 2020-10-02 op 	}
188 3e4749f7 2020-10-02 op 
189 3e4749f7 2020-10-02 op 	return 1;
190 3e4749f7 2020-10-02 op }
191 3e4749f7 2020-10-02 op 
192 3e4749f7 2020-10-02 op void
193 3e4749f7 2020-10-02 op adjust_path(char *path)
194 3e4749f7 2020-10-02 op {
195 3e4749f7 2020-10-02 op 	char *s;
196 3e4749f7 2020-10-02 op 	size_t len;
197 3e4749f7 2020-10-02 op 
198 f28f9311 2020-10-14 op 	/* /.. -> / */
199 3e4749f7 2020-10-02 op 	len = strlen(path);
200 3e4749f7 2020-10-02 op 	if (len >= 3) {
201 3e4749f7 2020-10-02 op 		if (!strcmp(&path[len-3], "/..")) {
202 3e4749f7 2020-10-02 op 			path[len-2] = '\0';
203 3e4749f7 2020-10-02 op 		}
204 3e4749f7 2020-10-02 op 	}
205 3e4749f7 2020-10-02 op 
206 3e4749f7 2020-10-02 op 	/* if the path is only `..` trim out and exit */
207 3e4749f7 2020-10-02 op 	if (!strcmp(path, "..")) {
208 3e4749f7 2020-10-02 op 		path[0] = '\0';
209 3e4749f7 2020-10-02 op 		return;
210 3e4749f7 2020-10-02 op 	}
211 3e4749f7 2020-10-02 op 
212 3e4749f7 2020-10-02 op 	/* remove every ../ in the path */
213 3e4749f7 2020-10-02 op 	while (1) {
214 3e4749f7 2020-10-02 op 		if ((s = strstr(path, "../")) == NULL)
215 3e4749f7 2020-10-02 op 			return;
216 67328d23 2020-10-03 op 		memmove(s, s+3, strlen(s)+1);	/* copy also the \0 */
217 3e4749f7 2020-10-02 op 	}
218 3e4749f7 2020-10-02 op }
219 3e4749f7 2020-10-02 op 
220 3e4749f7 2020-10-02 op int
221 3e4749f7 2020-10-02 op path_isdir(char *path)
222 3e4749f7 2020-10-02 op {
223 3e4749f7 2020-10-02 op 	if (*path == '\0')
224 3e4749f7 2020-10-02 op 		return 1;
225 4d4f0e19 2020-10-03 op 	return path[strlen(path)-1] == '/';
226 3e4749f7 2020-10-02 op }
227 3e4749f7 2020-10-02 op 
228 592fd624 2020-10-07 op int
229 592fd624 2020-10-07 op start_reply(struct pollfd *pfd, struct client *client, int code, const char *reason)
230 3e4749f7 2020-10-02 op {
231 3e4749f7 2020-10-02 op 	char buf[1030] = {0}; 	/* status + ' ' + max reply len + \r\n\0 */
232 3e4749f7 2020-10-02 op 	int len;
233 592fd624 2020-10-07 op 	int ret;
234 3e4749f7 2020-10-02 op 
235 592fd624 2020-10-07 op 	client->code = code;
236 592fd624 2020-10-07 op 	client->meta = reason;
237 592fd624 2020-10-07 op 	client->state = S_INITIALIZING;
238 592fd624 2020-10-07 op 
239 3e4749f7 2020-10-02 op 	len = snprintf(buf, sizeof(buf), "%d %s\r\n", code, reason);
240 3e4749f7 2020-10-02 op 	assert(len < (int)sizeof(buf));
241 592fd624 2020-10-07 op 	ret = tls_write(client->ctx, buf, len);
242 592fd624 2020-10-07 op 	if (ret == TLS_WANT_POLLIN) {
243 592fd624 2020-10-07 op 		pfd->events = POLLIN;
244 592fd624 2020-10-07 op 		return 0;
245 592fd624 2020-10-07 op 	}
246 592fd624 2020-10-07 op 
247 592fd624 2020-10-07 op 	if (ret == TLS_WANT_POLLOUT) {
248 592fd624 2020-10-07 op 		pfd->events = POLLOUT;
249 592fd624 2020-10-07 op 		return 0;
250 592fd624 2020-10-07 op 	}
251 592fd624 2020-10-07 op 
252 592fd624 2020-10-07 op 	return 1;
253 3e4749f7 2020-10-02 op }
254 3e4749f7 2020-10-02 op 
255 f28f9311 2020-10-14 op ssize_t
256 f28f9311 2020-10-14 op filesize(int fd)
257 f28f9311 2020-10-14 op {
258 f28f9311 2020-10-14 op 	ssize_t len;
259 f28f9311 2020-10-14 op 
260 f28f9311 2020-10-14 op 	if ((len = lseek(fd, 0, SEEK_END)) == -1)
261 f28f9311 2020-10-14 op 		return -1;
262 f28f9311 2020-10-14 op 	if (lseek(fd, 0, SEEK_SET) == -1)
263 f28f9311 2020-10-14 op 		return -1;
264 f28f9311 2020-10-14 op 	return len;
265 f28f9311 2020-10-14 op }
266 f28f9311 2020-10-14 op 
267 cc68fe70 2020-10-07 op const char *
268 cc68fe70 2020-10-07 op path_ext(const char *path)
269 cc68fe70 2020-10-07 op {
270 cc68fe70 2020-10-07 op 	const char *end;
271 cc68fe70 2020-10-07 op 
272 cc68fe70 2020-10-07 op 	end = path + strlen(path)-1; /* the last byte before the NUL */
273 cc68fe70 2020-10-07 op 	for (; end != path; --end) {
274 cc68fe70 2020-10-07 op 		if (*end == '.')
275 cc68fe70 2020-10-07 op 			return end+1;
276 cc68fe70 2020-10-07 op 		if (*end == '/')
277 cc68fe70 2020-10-07 op 			break;
278 cc68fe70 2020-10-07 op 	}
279 cc68fe70 2020-10-07 op 
280 cc68fe70 2020-10-07 op 	return NULL;
281 cc68fe70 2020-10-07 op }
282 cc68fe70 2020-10-07 op 
283 cc68fe70 2020-10-07 op const char *
284 cc68fe70 2020-10-07 op mime(const char *path)
285 cc68fe70 2020-10-07 op {
286 cc68fe70 2020-10-07 op 	const char *ext, *def = "application/octet-stream";
287 cc68fe70 2020-10-07 op 	struct etm *t;
288 cc68fe70 2020-10-07 op 
289 cc68fe70 2020-10-07 op 	if ((ext = path_ext(path)) == NULL)
290 cc68fe70 2020-10-07 op 		return def;
291 cc68fe70 2020-10-07 op 
292 cc68fe70 2020-10-07 op 	for (t = filetypes; t->mime != NULL; ++t)
293 cc68fe70 2020-10-07 op 		if (!strcmp(ext, t->ext))
294 cc68fe70 2020-10-07 op 			return t->mime;
295 cc68fe70 2020-10-07 op 
296 cc68fe70 2020-10-07 op 	return def;
297 3e4749f7 2020-10-02 op }
298 3e4749f7 2020-10-02 op 
299 f28f9311 2020-10-14 op int
300 f28f9311 2020-10-14 op open_file(char *path, struct pollfd *fds, struct client *c)
301 3e4749f7 2020-10-02 op {
302 3e4749f7 2020-10-02 op 	char fpath[PATHBUF];
303 72342dc9 2020-11-06 op 	struct stat sb;
304 3e4749f7 2020-10-02 op 
305 f28f9311 2020-10-14 op 	assert(path != NULL);
306 3e4749f7 2020-10-02 op 
307 f28f9311 2020-10-14 op 	bzero(fpath, sizeof(fpath));
308 3e4749f7 2020-10-02 op 
309 f28f9311 2020-10-14 op 	if (*path != '.')
310 f28f9311 2020-10-14 op 		fpath[0] = '.';
311 f28f9311 2020-10-14 op 	strlcat(fpath, path, PATHBUF);
312 3e4749f7 2020-10-02 op 
313 f28f9311 2020-10-14 op 	if ((c->fd = openat(dirfd, fpath, O_RDONLY | O_NOFOLLOW)) == -1) {
314 2c3a40fa 2020-11-06 op 		LOG(c, "open failed: %s", fpath);
315 f28f9311 2020-10-14 op 		if (!start_reply(fds, c, NOT_FOUND, "not found"))
316 f28f9311 2020-10-14 op 			return 0;
317 f28f9311 2020-10-14 op 		goodbye(fds, c);
318 f28f9311 2020-10-14 op 		return 0;
319 f28f9311 2020-10-14 op 	}
320 3e4749f7 2020-10-02 op 
321 72342dc9 2020-11-06 op 	if (fstat(c->fd, &sb) == -1) {
322 72342dc9 2020-11-06 op 		LOG(c, "fstat failed for %s", fpath);
323 72342dc9 2020-11-06 op 		if (!start_reply(fds, c, TEMP_FAILURE, "internal server error"))
324 72342dc9 2020-11-06 op 			return 0;
325 72342dc9 2020-11-06 op 		goodbye(fds, c);
326 72342dc9 2020-11-06 op 		return 0;
327 72342dc9 2020-11-06 op 	}
328 72342dc9 2020-11-06 op 
329 72342dc9 2020-11-06 op 	if (S_ISDIR(sb.st_mode)) {
330 2c3a40fa 2020-11-06 op 		LOG(c, "%s is a directory, trying %s/index.gmi", fpath, fpath);
331 f28f9311 2020-10-14 op 		close(c->fd);
332 f28f9311 2020-10-14 op 		c->fd = -1;
333 f28f9311 2020-10-14 op 		send_dir(fpath, fds, c);
334 f28f9311 2020-10-14 op 		return 0;
335 f28f9311 2020-10-14 op 	}
336 3e4749f7 2020-10-02 op 
337 72342dc9 2020-11-06 op 	if (cgi && (sb.st_mode & S_IXUSR)) {
338 72342dc9 2020-11-06 op 		start_cgi(fpath, fds, c);
339 72342dc9 2020-11-06 op 		return 0;
340 72342dc9 2020-11-06 op 	}
341 72342dc9 2020-11-06 op 
342 f28f9311 2020-10-14 op 	if ((c->len = filesize(c->fd)) == -1) {
343 2c3a40fa 2020-11-06 op 		LOG(c, "failed to get file size for %s", fpath);
344 f28f9311 2020-10-14 op 		goodbye(fds, c);
345 f28f9311 2020-10-14 op 		return 0;
346 f28f9311 2020-10-14 op 	}
347 f28f9311 2020-10-14 op 
348 f28f9311 2020-10-14 op 	if ((c->buf = mmap(NULL, c->len, PROT_READ, MAP_PRIVATE,
349 f28f9311 2020-10-14 op 		    c->fd, 0)) == MAP_FAILED) {
350 f28f9311 2020-10-14 op 		warn("mmap: %s", fpath);
351 f28f9311 2020-10-14 op 		goodbye(fds, c);
352 f28f9311 2020-10-14 op 		return 0;
353 592fd624 2020-10-07 op 	}
354 f28f9311 2020-10-14 op 	c->i = c->buf;
355 3e4749f7 2020-10-02 op 
356 f28f9311 2020-10-14 op 	return start_reply(fds, c, SUCCESS, mime(fpath));
357 72342dc9 2020-11-06 op }
358 72342dc9 2020-11-06 op 
359 72342dc9 2020-11-06 op void
360 72342dc9 2020-11-06 op start_cgi(const char *path, struct pollfd *fds, struct client *c)
361 72342dc9 2020-11-06 op {
362 72342dc9 2020-11-06 op 	pid_t pid;
363 72342dc9 2020-11-06 op 	int p[2];
364 72342dc9 2020-11-06 op 
365 72342dc9 2020-11-06 op 	if (pipe(p) == -1)
366 72342dc9 2020-11-06 op 		goto err;
367 72342dc9 2020-11-06 op 
368 72342dc9 2020-11-06 op 	switch (pid = fork()) {
369 72342dc9 2020-11-06 op 	case -1:
370 72342dc9 2020-11-06 op 		goto err;
371 72342dc9 2020-11-06 op 
372 72342dc9 2020-11-06 op 	case 0: {		/* child */
373 72342dc9 2020-11-06 op                 char *expath;
374 72342dc9 2020-11-06 op 		char addr[INET_ADDRSTRLEN];
375 72342dc9 2020-11-06 op 		char *argv[] = { NULL, NULL, NULL };
376 72342dc9 2020-11-06 op 		char *envp[] = {
377 72342dc9 2020-11-06 op 			/* inherited */
378 72342dc9 2020-11-06 op 			"PATH=",
379 72342dc9 2020-11-06 op 
380 72342dc9 2020-11-06 op 			/* CGI */
381 72342dc9 2020-11-06 op 			"SERVER_SOFTWARE=gmid",
382 72342dc9 2020-11-06 op 			/* "SERVER_NAME=example.com", */
383 72342dc9 2020-11-06 op 			/* "GATEWAY_INTERFACE=CGI/version" */
384 72342dc9 2020-11-06 op 			"SERVER_PROTOCOL=gemini",
385 72342dc9 2020-11-06 op 			"SERVER_PORT=1965",
386 72342dc9 2020-11-06 op 			/* "PATH_INFO=" */
387 72342dc9 2020-11-06 op 			/* "PATH_TRANSLATED=" */
388 72342dc9 2020-11-06 op 			/* "SCRIPT_NAME=" */
389 72342dc9 2020-11-06 op 			/* "QUERY_STRING=" */
390 72342dc9 2020-11-06 op                         "REMOTE_ADDR=",
391 72342dc9 2020-11-06 op 			NULL,
392 72342dc9 2020-11-06 op 		};
393 72342dc9 2020-11-06 op 		size_t i;
394 72342dc9 2020-11-06 op 
395 72342dc9 2020-11-06 op 		close(p[0]);	/* close the read end */
396 72342dc9 2020-11-06 op 		if (dup2(p[1], 1) == -1)
397 72342dc9 2020-11-06 op 			goto childerr;
398 72342dc9 2020-11-06 op 
399 72342dc9 2020-11-06 op 		if (inet_ntop(c->af, &c->addr, addr, sizeof(addr)) == NULL)
400 72342dc9 2020-11-06 op                         goto childerr;
401 72342dc9 2020-11-06 op 
402 72342dc9 2020-11-06 op 		/* skip the ./ at the start of path*/
403 72342dc9 2020-11-06 op 		if (asprintf(&expath, "%s%s", dir, path+2) == -1)
404 72342dc9 2020-11-06 op 			goto childerr;
405 72342dc9 2020-11-06 op 		argv[0] = argv[1] = expath;
406 72342dc9 2020-11-06 op 
407 72342dc9 2020-11-06 op 		/* fix the envp */
408 72342dc9 2020-11-06 op 		for (i = 0; envp[i] != NULL; ++i) {
409 72342dc9 2020-11-06 op 			if (!strcmp(envp[i], "PATH="))
410 72342dc9 2020-11-06 op 				envp[i] = getenv("PATH");
411 72342dc9 2020-11-06 op 			else if (!strcmp(envp[i], "REMOTE_ADDR="))
412 72342dc9 2020-11-06 op 				envp[i] = addr;
413 72342dc9 2020-11-06 op 			/* else if (!strcmp(envp[i], "PATH_INFO")) */
414 72342dc9 2020-11-06 op 			/* ... */
415 72342dc9 2020-11-06 op 		}
416 72342dc9 2020-11-06 op 
417 72342dc9 2020-11-06 op 		execvpe(expath, argv, envp);
418 72342dc9 2020-11-06 op 		goto childerr;
419 72342dc9 2020-11-06 op 	}
420 72342dc9 2020-11-06 op 
421 72342dc9 2020-11-06 op 	default:		/* parent */
422 72342dc9 2020-11-06 op 		close(p[1]);	/* close the write end */
423 72342dc9 2020-11-06 op 		close(c->fd);
424 72342dc9 2020-11-06 op 		c->fd = p[0];
425 72342dc9 2020-11-06 op 		c->child = pid;
426 72342dc9 2020-11-06 op 		handle_cgi(fds, c);
427 72342dc9 2020-11-06 op 		return;
428 72342dc9 2020-11-06 op 	}
429 72342dc9 2020-11-06 op 
430 72342dc9 2020-11-06 op err:
431 72342dc9 2020-11-06 op 	if (!start_reply(fds, c, TEMP_FAILURE, "internal server error"))
432 72342dc9 2020-11-06 op 		return;
433 72342dc9 2020-11-06 op 	goodbye(fds, c);
434 72342dc9 2020-11-06 op 	return;
435 72342dc9 2020-11-06 op 
436 72342dc9 2020-11-06 op childerr:
437 72342dc9 2020-11-06 op 	dprintf(p[1], "%d internal server error\r\n", TEMP_FAILURE);
438 72342dc9 2020-11-06 op 	close(p[1]);
439 72342dc9 2020-11-06 op         _exit(1);
440 72342dc9 2020-11-06 op }
441 72342dc9 2020-11-06 op 
442 72342dc9 2020-11-06 op void
443 72342dc9 2020-11-06 op handle_cgi(struct pollfd *fds, struct client *c)
444 72342dc9 2020-11-06 op {
445 72342dc9 2020-11-06 op 	char buf[1024];
446 72342dc9 2020-11-06 op 	ssize_t r, todo;
447 72342dc9 2020-11-06 op 
448 72342dc9 2020-11-06 op 	while (1) {
449 72342dc9 2020-11-06 op 		r = read(c->fd, buf, sizeof(buf));
450 72342dc9 2020-11-06 op 		if (r == -1 || r == 0)
451 72342dc9 2020-11-06 op 			break;
452 72342dc9 2020-11-06 op 		todo = r;
453 72342dc9 2020-11-06 op 		while (todo > 0) {
454 72342dc9 2020-11-06 op 			switch (r = tls_write(c->ctx, buf, todo)) {
455 72342dc9 2020-11-06 op 			case -1:
456 72342dc9 2020-11-06 op 				goto end;
457 72342dc9 2020-11-06 op 
458 72342dc9 2020-11-06 op 			case TLS_WANT_POLLOUT:
459 72342dc9 2020-11-06 op 			case TLS_WANT_POLLIN:
460 72342dc9 2020-11-06 op 				/* evil! */
461 72342dc9 2020-11-06 op 				continue;
462 72342dc9 2020-11-06 op 
463 72342dc9 2020-11-06 op 			default:
464 72342dc9 2020-11-06 op 				todo -= r;
465 72342dc9 2020-11-06 op 			}
466 72342dc9 2020-11-06 op 		}
467 72342dc9 2020-11-06 op 	}
468 72342dc9 2020-11-06 op 
469 72342dc9 2020-11-06 op end:
470 72342dc9 2020-11-06 op 	goodbye(fds, c);
471 f28f9311 2020-10-14 op }
472 592fd624 2020-10-07 op 
473 f28f9311 2020-10-14 op void
474 f28f9311 2020-10-14 op send_file(char *path, struct pollfd *fds, struct client *c)
475 f28f9311 2020-10-14 op {
476 f28f9311 2020-10-14 op 	ssize_t ret, len;
477 592fd624 2020-10-07 op 
478 f28f9311 2020-10-14 op 	if (c->fd == -1) {
479 f28f9311 2020-10-14 op 		if (!open_file(path, fds, c))
480 f28f9311 2020-10-14 op 			return;
481 f28f9311 2020-10-14 op 		c->state = S_SENDING;
482 f28f9311 2020-10-14 op 	}
483 592fd624 2020-10-07 op 
484 f28f9311 2020-10-14 op 	len = (c->buf + c->len) - c->i;
485 f28f9311 2020-10-14 op 
486 f28f9311 2020-10-14 op 	while (len > 0) {
487 f28f9311 2020-10-14 op 		switch (ret = tls_write(c->ctx, c->i, len)) {
488 f28f9311 2020-10-14 op 		case -1:
489 2c3a40fa 2020-11-06 op 			LOG(c, "tls_write: %s", tls_error(c->ctx));
490 f28f9311 2020-10-14 op 			goodbye(fds, c);
491 f28f9311 2020-10-14 op 			return;
492 f28f9311 2020-10-14 op 
493 f28f9311 2020-10-14 op 		case TLS_WANT_POLLIN:
494 f28f9311 2020-10-14 op 			fds->events = POLLIN;
495 f28f9311 2020-10-14 op 			return;
496 f28f9311 2020-10-14 op 
497 f28f9311 2020-10-14 op 		case TLS_WANT_POLLOUT:
498 f28f9311 2020-10-14 op 			fds->events = POLLOUT;
499 f28f9311 2020-10-14 op 			return;
500 f28f9311 2020-10-14 op 
501 f28f9311 2020-10-14 op 		default:
502 f28f9311 2020-10-14 op 			c->i += ret;
503 f28f9311 2020-10-14 op 			len -= ret;
504 f28f9311 2020-10-14 op 			break;
505 3e4749f7 2020-10-02 op 		}
506 3e4749f7 2020-10-02 op 	}
507 f28f9311 2020-10-14 op 
508 f28f9311 2020-10-14 op 	goodbye(fds, c);
509 3e4749f7 2020-10-02 op }
510 3e4749f7 2020-10-02 op 
511 3e4749f7 2020-10-02 op void
512 592fd624 2020-10-07 op send_dir(char *path, struct pollfd *fds, struct client *client)
513 3e4749f7 2020-10-02 op {
514 3e4749f7 2020-10-02 op 	char fpath[PATHBUF];
515 3e4749f7 2020-10-02 op 	size_t len;
516 3e4749f7 2020-10-02 op 
517 3e4749f7 2020-10-02 op 	bzero(fpath, PATHBUF);
518 3e4749f7 2020-10-02 op 
519 9c56b0a7 2020-10-14 op 	if (path[0] != '.')
520 3e4749f7 2020-10-02 op 		fpath[0] = '.';
521 3e4749f7 2020-10-02 op 
522 3e4749f7 2020-10-02 op 	/* this cannot fail since sizeof(fpath) > maxlen of path */
523 3e4749f7 2020-10-02 op 	strlcat(fpath, path, PATHBUF);
524 3e4749f7 2020-10-02 op 	len = strlen(fpath);
525 3e4749f7 2020-10-02 op 
526 3e4749f7 2020-10-02 op 	/* add a trailing / in case. */
527 3e4749f7 2020-10-02 op 	if (fpath[len-1] != '/') {
528 3e4749f7 2020-10-02 op 		fpath[len] = '/';
529 3e4749f7 2020-10-02 op 	}
530 3e4749f7 2020-10-02 op 
531 3e4749f7 2020-10-02 op 	strlcat(fpath, "index.gmi", sizeof(fpath));
532 3e4749f7 2020-10-02 op 
533 592fd624 2020-10-07 op 	send_file(fpath, fds, client);
534 3e4749f7 2020-10-02 op }
535 3e4749f7 2020-10-02 op 
536 3e4749f7 2020-10-02 op void
537 592fd624 2020-10-07 op handle(struct pollfd *fds, struct client *client)
538 3e4749f7 2020-10-02 op {
539 592fd624 2020-10-07 op 	char buf[GEMINI_URL_LEN];
540 3e4749f7 2020-10-02 op 	char *path;
541 3e4749f7 2020-10-02 op 
542 592fd624 2020-10-07 op 	switch (client->state) {
543 592fd624 2020-10-07 op 	case S_OPEN:
544 592fd624 2020-10-07 op 		bzero(buf, GEMINI_URL_LEN);
545 592fd624 2020-10-07 op 		switch (tls_read(client->ctx, buf, sizeof(buf)-1)) {
546 592fd624 2020-10-07 op 		case -1:
547 2c3a40fa 2020-11-06 op 			LOG(client, "tls_read: %s", tls_error(client->ctx));
548 592fd624 2020-10-07 op 			goodbye(fds, client);
549 592fd624 2020-10-07 op 			return;
550 3e4749f7 2020-10-02 op 
551 592fd624 2020-10-07 op 		case TLS_WANT_POLLIN:
552 592fd624 2020-10-07 op 			fds->events = POLLIN;
553 592fd624 2020-10-07 op 			return;
554 3e4749f7 2020-10-02 op 
555 592fd624 2020-10-07 op 		case TLS_WANT_POLLOUT:
556 592fd624 2020-10-07 op 			fds->events = POLLOUT;
557 592fd624 2020-10-07 op 			return;
558 592fd624 2020-10-07 op 		}
559 3e4749f7 2020-10-02 op 
560 2c3a40fa 2020-11-06 op 		if (!url_trim(client, buf)) {
561 592fd624 2020-10-07 op 			if (!start_reply(fds, client, BAD_REQUEST, "bad request"))
562 592fd624 2020-10-07 op 				return;
563 592fd624 2020-10-07 op 			goodbye(fds, client);
564 592fd624 2020-10-07 op 			return;
565 592fd624 2020-10-07 op 		}
566 3e4749f7 2020-10-02 op 
567 592fd624 2020-10-07 op 		if ((path = url_start_of_request(buf)) == NULL) {
568 592fd624 2020-10-07 op 			if (!start_reply(fds, client, BAD_REQUEST, "bad request"))
569 592fd624 2020-10-07 op 				return;
570 592fd624 2020-10-07 op 			goodbye(fds, client);
571 592fd624 2020-10-07 op 			return;
572 592fd624 2020-10-07 op 		}
573 592fd624 2020-10-07 op 
574 592fd624 2020-10-07 op 		adjust_path(path);
575 2c3a40fa 2020-11-06 op 		LOG(client, "get %s", path);
576 592fd624 2020-10-07 op 
577 592fd624 2020-10-07 op 		if (path_isdir(path))
578 592fd624 2020-10-07 op 			send_dir(path, fds, client);
579 592fd624 2020-10-07 op 		else
580 592fd624 2020-10-07 op 			send_file(path, fds, client);
581 592fd624 2020-10-07 op 		break;
582 592fd624 2020-10-07 op 
583 592fd624 2020-10-07 op 	case S_INITIALIZING:
584 592fd624 2020-10-07 op 		if (!start_reply(fds, client, client->code, client->meta))
585 592fd624 2020-10-07 op 			return;
586 592fd624 2020-10-07 op 
587 592fd624 2020-10-07 op 		if (client->code != SUCCESS) {
588 592fd624 2020-10-07 op 			/* we don't need a body */
589 592fd624 2020-10-07 op 			goodbye(fds, client);
590 592fd624 2020-10-07 op 			return;
591 592fd624 2020-10-07 op 		}
592 592fd624 2020-10-07 op 
593 592fd624 2020-10-07 op 		client->state = S_SENDING;
594 592fd624 2020-10-07 op 
595 592fd624 2020-10-07 op 		/* fallthrough */
596 592fd624 2020-10-07 op 
597 592fd624 2020-10-07 op 	case S_SENDING:
598 592fd624 2020-10-07 op 		send_file(NULL, fds, client);
599 592fd624 2020-10-07 op 		break;
600 592fd624 2020-10-07 op 
601 592fd624 2020-10-07 op 	case S_CLOSING:
602 592fd624 2020-10-07 op 		goodbye(fds, client);
603 592fd624 2020-10-07 op 		break;
604 592fd624 2020-10-07 op 
605 592fd624 2020-10-07 op 	default:
606 592fd624 2020-10-07 op 		/* unreachable */
607 592fd624 2020-10-07 op 		abort();
608 592fd624 2020-10-07 op 	}
609 3e4749f7 2020-10-02 op }
610 3e4749f7 2020-10-02 op 
611 592fd624 2020-10-07 op void
612 592fd624 2020-10-07 op mark_nonblock(int fd)
613 592fd624 2020-10-07 op {
614 592fd624 2020-10-07 op 	int flags;
615 592fd624 2020-10-07 op 
616 592fd624 2020-10-07 op 	if ((flags = fcntl(fd, F_GETFL)) == -1)
617 592fd624 2020-10-07 op 		err(1, "fcntl(F_GETFL)");
618 592fd624 2020-10-07 op 	if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1)
619 592fd624 2020-10-07 op 		err(1, "fcntl(F_SETFL)");
620 592fd624 2020-10-07 op }
621 592fd624 2020-10-07 op 
622 3e4749f7 2020-10-02 op int
623 9468027b 2020-10-15 op make_socket(int port, int family)
624 3e4749f7 2020-10-02 op {
625 3e4749f7 2020-10-02 op 	int sock, v;
626 9468027b 2020-10-15 op 	struct sockaddr_in addr4;
627 9468027b 2020-10-15 op 	struct sockaddr_in6 addr6;
628 9468027b 2020-10-15 op 	struct sockaddr *addr;
629 9468027b 2020-10-15 op 	socklen_t len;
630 3e4749f7 2020-10-02 op 
631 9468027b 2020-10-15 op         switch (family) {
632 9468027b 2020-10-15 op 	case AF_INET:
633 9468027b 2020-10-15 op 		bzero(&addr4, sizeof(addr4));
634 9468027b 2020-10-15 op 		addr4.sin_family = family;
635 9468027b 2020-10-15 op 		addr4.sin_port = htons(port);
636 9468027b 2020-10-15 op 		addr4.sin_addr.s_addr = INADDR_ANY;
637 9468027b 2020-10-15 op 		addr = (struct sockaddr*)&addr4;
638 9468027b 2020-10-15 op 		len = sizeof(addr4);
639 9468027b 2020-10-15 op 		break;
640 9468027b 2020-10-15 op 
641 9468027b 2020-10-15 op 	case AF_INET6:
642 9468027b 2020-10-15 op 		bzero(&addr6, sizeof(addr6));
643 9468027b 2020-10-15 op 		addr6.sin6_family = AF_INET6;
644 9468027b 2020-10-15 op 		addr6.sin6_port = htons(port);
645 9468027b 2020-10-15 op 		addr6.sin6_addr = in6addr_any;
646 9468027b 2020-10-15 op 		addr = (struct sockaddr*)&addr6;
647 9468027b 2020-10-15 op 		len = sizeof(addr6);
648 9468027b 2020-10-15 op 		break;
649 9468027b 2020-10-15 op 
650 9468027b 2020-10-15 op 	default:
651 9468027b 2020-10-15 op 		/* unreachable */
652 9468027b 2020-10-15 op 		abort();
653 9468027b 2020-10-15 op 	}
654 9468027b 2020-10-15 op 
655 9468027b 2020-10-15 op 	if ((sock = socket(family, SOCK_STREAM, 0)) == -1)
656 f28f9311 2020-10-14 op 		err(1, "socket");
657 3e4749f7 2020-10-02 op 
658 3e4749f7 2020-10-02 op 	v = 1;
659 3e4749f7 2020-10-02 op 	if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &v, sizeof(v)) == -1)
660 3e4749f7 2020-10-02 op 		err(1, "setsockopt(SO_REUSEADDR)");
661 3e4749f7 2020-10-02 op 
662 3e4749f7 2020-10-02 op 	v = 1;
663 3e4749f7 2020-10-02 op 	if (setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, &v, sizeof(v)) == -1)
664 3e4749f7 2020-10-02 op 		err(1, "setsockopt(SO_REUSEPORT)");
665 3e4749f7 2020-10-02 op 
666 592fd624 2020-10-07 op 	mark_nonblock(sock);
667 592fd624 2020-10-07 op 
668 9468027b 2020-10-15 op 	if (bind(sock, addr, len) == -1)
669 f28f9311 2020-10-14 op 		err(1, "bind");
670 3e4749f7 2020-10-02 op 
671 3e4749f7 2020-10-02 op 	if (listen(sock, 16) == -1)
672 f28f9311 2020-10-14 op 		err(1, "listen");
673 3e4749f7 2020-10-02 op 
674 3e4749f7 2020-10-02 op 	return sock;
675 3e4749f7 2020-10-02 op }
676 3e4749f7 2020-10-02 op 
677 3e4749f7 2020-10-02 op void
678 592fd624 2020-10-07 op do_accept(int sock, struct tls *ctx, struct pollfd *fds, struct client *clients)
679 3e4749f7 2020-10-02 op {
680 592fd624 2020-10-07 op 	int i, fd;
681 592fd624 2020-10-07 op 	struct sockaddr_in addr;
682 3e4749f7 2020-10-02 op 	socklen_t len;
683 3e4749f7 2020-10-02 op 
684 592fd624 2020-10-07 op 	len = sizeof(addr);
685 592fd624 2020-10-07 op 	if ((fd = accept(sock, (struct sockaddr*)&addr, &len)) == -1) {
686 592fd624 2020-10-07 op 		if (errno == EWOULDBLOCK)
687 592fd624 2020-10-07 op 			return;
688 592fd624 2020-10-07 op 		err(1, "accept");
689 592fd624 2020-10-07 op 	}
690 3e4749f7 2020-10-02 op 
691 592fd624 2020-10-07 op 	mark_nonblock(fd);
692 3e4749f7 2020-10-02 op 
693 592fd624 2020-10-07 op 	for (i = 0; i < MAX_USERS; ++i) {
694 592fd624 2020-10-07 op 		if (fds[i].fd == -1) {
695 592fd624 2020-10-07 op 			bzero(&clients[i], sizeof(struct client));
696 592fd624 2020-10-07 op 			if (tls_accept_socket(ctx, &clients[i].ctx, fd) == -1)
697 592fd624 2020-10-07 op 				break; /* goodbye fd! */
698 592fd624 2020-10-07 op 
699 592fd624 2020-10-07 op 			fds[i].fd = fd;
700 592fd624 2020-10-07 op 			fds[i].events = POLLIN;
701 592fd624 2020-10-07 op 
702 592fd624 2020-10-07 op 			clients[i].state = S_OPEN;
703 592fd624 2020-10-07 op 			clients[i].fd = -1;
704 72342dc9 2020-11-06 op 			clients[i].child = -1;
705 f28f9311 2020-10-14 op 			clients[i].buf = MAP_FAILED;
706 2c3a40fa 2020-11-06 op 			clients[i].af = AF_INET;
707 2c3a40fa 2020-11-06 op 			clients[i].addr = addr.sin_addr;
708 592fd624 2020-10-07 op 
709 592fd624 2020-10-07 op 			return;
710 3e4749f7 2020-10-02 op 		}
711 592fd624 2020-10-07 op 	}
712 3e4749f7 2020-10-02 op 
713 592fd624 2020-10-07 op 	close(fd);
714 592fd624 2020-10-07 op }
715 3e4749f7 2020-10-02 op 
716 592fd624 2020-10-07 op void
717 592fd624 2020-10-07 op goodbye(struct pollfd *pfd, struct client *c)
718 592fd624 2020-10-07 op {
719 592fd624 2020-10-07 op 	ssize_t ret;
720 592fd624 2020-10-07 op 
721 592fd624 2020-10-07 op 	c->state = S_CLOSING;
722 592fd624 2020-10-07 op 
723 592fd624 2020-10-07 op 	ret = tls_close(c->ctx);
724 592fd624 2020-10-07 op 	if (ret == TLS_WANT_POLLIN) {
725 592fd624 2020-10-07 op 		pfd->events = POLLIN;
726 592fd624 2020-10-07 op 		return;
727 3e4749f7 2020-10-02 op 	}
728 592fd624 2020-10-07 op 	if (ret == TLS_WANT_POLLOUT) {
729 592fd624 2020-10-07 op 		pfd->events = POLLOUT;
730 592fd624 2020-10-07 op 		return;
731 592fd624 2020-10-07 op 	}
732 592fd624 2020-10-07 op 
733 592fd624 2020-10-07 op 	tls_free(c->ctx);
734 592fd624 2020-10-07 op 	c->ctx = NULL;
735 592fd624 2020-10-07 op 
736 f28f9311 2020-10-14 op 	if (c->buf != MAP_FAILED)
737 f28f9311 2020-10-14 op 		munmap(c->buf, c->len);
738 f28f9311 2020-10-14 op 
739 592fd624 2020-10-07 op 	if (c->fd != -1)
740 592fd624 2020-10-07 op 		close(c->fd);
741 592fd624 2020-10-07 op 
742 592fd624 2020-10-07 op 	close(pfd->fd);
743 592fd624 2020-10-07 op 	pfd->fd = -1;
744 3e4749f7 2020-10-02 op }
745 3e4749f7 2020-10-02 op 
746 3e4749f7 2020-10-02 op void
747 592fd624 2020-10-07 op loop(struct tls *ctx, int sock)
748 592fd624 2020-10-07 op {
749 592fd624 2020-10-07 op 	int i, todo;
750 592fd624 2020-10-07 op 	struct client clients[MAX_USERS];
751 592fd624 2020-10-07 op 	struct pollfd fds[MAX_USERS];
752 592fd624 2020-10-07 op 
753 592fd624 2020-10-07 op 	for (i = 0; i < MAX_USERS; ++i) {
754 592fd624 2020-10-07 op 		fds[i].fd = -1;
755 592fd624 2020-10-07 op 		fds[i].events = POLLIN;
756 592fd624 2020-10-07 op 		bzero(&clients[i], sizeof(struct client));
757 592fd624 2020-10-07 op 	}
758 592fd624 2020-10-07 op 
759 592fd624 2020-10-07 op 	fds[0].fd = sock;
760 592fd624 2020-10-07 op 
761 592fd624 2020-10-07 op 	for (;;) {
762 f28f9311 2020-10-14 op 		if ((todo = poll(fds, MAX_USERS, INFTIM)) == -1)
763 592fd624 2020-10-07 op 			err(1, "poll");
764 592fd624 2020-10-07 op 
765 592fd624 2020-10-07 op 		for (i = 0; i < MAX_USERS; i++) {
766 592fd624 2020-10-07 op 			assert(i < MAX_USERS);
767 592fd624 2020-10-07 op 
768 592fd624 2020-10-07 op 			if (fds[i].revents == 0)
769 592fd624 2020-10-07 op 				continue;
770 592fd624 2020-10-07 op 
771 592fd624 2020-10-07 op 			if (fds[i].revents & (POLLERR|POLLNVAL))
772 592fd624 2020-10-07 op 				err(1, "bad fd %d", fds[i].fd);
773 592fd624 2020-10-07 op 
774 592fd624 2020-10-07 op 			if (fds[i].revents & POLLHUP) {
775 592fd624 2020-10-07 op 				goodbye(&fds[i], &clients[i]);
776 592fd624 2020-10-07 op 				continue;
777 592fd624 2020-10-07 op 			}
778 592fd624 2020-10-07 op 
779 592fd624 2020-10-07 op 			todo--;
780 592fd624 2020-10-07 op 
781 592fd624 2020-10-07 op 			if (i == 0) { /* new client */
782 592fd624 2020-10-07 op 				do_accept(sock, ctx, fds, clients);
783 592fd624 2020-10-07 op 				continue;
784 592fd624 2020-10-07 op 			}
785 592fd624 2020-10-07 op 
786 592fd624 2020-10-07 op 			handle(&fds[i], &clients[i]);
787 592fd624 2020-10-07 op 		}
788 592fd624 2020-10-07 op 	}
789 592fd624 2020-10-07 op }
790 592fd624 2020-10-07 op 
791 592fd624 2020-10-07 op void
792 3e4749f7 2020-10-02 op usage(const char *me)
793 3e4749f7 2020-10-02 op {
794 3e4749f7 2020-10-02 op 	fprintf(stderr,
795 3e4749f7 2020-10-02 op 	    "USAGE: %s [-h] [-c cert.pem] [-d docs] [-k key.pem]\n",
796 3e4749f7 2020-10-02 op 	    me);
797 3e4749f7 2020-10-02 op }
798 3e4749f7 2020-10-02 op 
799 3e4749f7 2020-10-02 op int
800 3e4749f7 2020-10-02 op main(int argc, char **argv)
801 3e4749f7 2020-10-02 op {
802 72342dc9 2020-11-06 op 	const char *cert = "cert.pem", *key = "key.pem";
803 3e4749f7 2020-10-02 op 	struct tls *ctx = NULL;
804 3e4749f7 2020-10-02 op 	struct tls_config *conf;
805 3e4749f7 2020-10-02 op 	int sock, ch;
806 3e4749f7 2020-10-02 op 
807 0cf902af 2020-11-03 op 	signal(SIGPIPE, SIG_IGN);
808 72342dc9 2020-11-06 op 	signal(SIGCHLD, SIG_IGN);
809 0cf902af 2020-11-03 op 
810 72342dc9 2020-11-06 op 	dir = "docs/";
811 2c3a40fa 2020-11-06 op 	logfd = 2;		/* stderr */
812 72342dc9 2020-11-06 op 	cgi = 0;
813 2c3a40fa 2020-11-06 op 
814 72342dc9 2020-11-06 op 	while ((ch = getopt(argc, argv, "c:d:hk:l:x")) != -1) {
815 3e4749f7 2020-10-02 op 		switch (ch) {
816 3e4749f7 2020-10-02 op 		case 'c':
817 3e4749f7 2020-10-02 op 			cert = optarg;
818 3e4749f7 2020-10-02 op 			break;
819 3e4749f7 2020-10-02 op 
820 3e4749f7 2020-10-02 op 		case 'd':
821 3e4749f7 2020-10-02 op 			dir = optarg;
822 3e4749f7 2020-10-02 op 			break;
823 3e4749f7 2020-10-02 op 
824 3e4749f7 2020-10-02 op 		case 'h':
825 3e4749f7 2020-10-02 op 			usage(*argv);
826 3e4749f7 2020-10-02 op 			return 0;
827 3e4749f7 2020-10-02 op 
828 3e4749f7 2020-10-02 op 		case 'k':
829 3e4749f7 2020-10-02 op 			key = optarg;
830 3e4749f7 2020-10-02 op 			break;
831 3e4749f7 2020-10-02 op 
832 2c3a40fa 2020-11-06 op 		case 'l':
833 2c3a40fa 2020-11-06 op 			/* open log file or create it with 644 */
834 2c3a40fa 2020-11-06 op 			if ((logfd = open(optarg, O_WRONLY | O_CREAT,
835 2c3a40fa 2020-11-06 op 					S_IRUSR | S_IWUSR | S_IRGRP | S_IWOTH)) == -1)
836 2c3a40fa 2020-11-06 op 				err(1, "%s", optarg);
837 2c3a40fa 2020-11-06 op 			break;
838 2c3a40fa 2020-11-06 op 
839 72342dc9 2020-11-06 op 		case 'x':
840 72342dc9 2020-11-06 op 			cgi = 1;
841 72342dc9 2020-11-06 op 			break;
842 72342dc9 2020-11-06 op 
843 3e4749f7 2020-10-02 op 		default:
844 3e4749f7 2020-10-02 op 			usage(*argv);
845 3e4749f7 2020-10-02 op 			return 1;
846 3e4749f7 2020-10-02 op 		}
847 3e4749f7 2020-10-02 op 	}
848 3e4749f7 2020-10-02 op 
849 3e4749f7 2020-10-02 op 	if ((conf = tls_config_new()) == NULL)
850 3e4749f7 2020-10-02 op 		err(1, "tls_config_new");
851 0d8ca45a 2020-10-03 op 
852 0d8ca45a 2020-10-03 op 	if (tls_config_set_protocols(conf,
853 0d8ca45a 2020-10-03 op 	    TLS_PROTOCOL_TLSv1_2 | TLS_PROTOCOL_TLSv1_3) == -1)
854 0d8ca45a 2020-10-03 op 		err(1, "tls_config_set_protocols");
855 3e4749f7 2020-10-02 op 
856 4d4f0e19 2020-10-03 op 	if (tls_config_set_cert_file(conf, cert) == -1)
857 4d4f0e19 2020-10-03 op 		err(1, "tls_config_set_cert_file: %s", cert);
858 3e4749f7 2020-10-02 op 
859 4d4f0e19 2020-10-03 op 	if (tls_config_set_key_file(conf, key) == -1)
860 4d4f0e19 2020-10-03 op 		err(1, "tls_config_set_key_file: %s", key);
861 3e4749f7 2020-10-02 op 
862 3e4749f7 2020-10-02 op 	if ((ctx = tls_server()) == NULL)
863 3e4749f7 2020-10-02 op 		err(1, "tls_server");
864 3e4749f7 2020-10-02 op 
865 3e4749f7 2020-10-02 op 	if (tls_configure(ctx, conf) == -1)
866 3e4749f7 2020-10-02 op 		errx(1, "tls_configure: %s", tls_error(ctx));
867 3e4749f7 2020-10-02 op 
868 9468027b 2020-10-15 op 	sock = make_socket(1965, AF_INET);
869 3e4749f7 2020-10-02 op 
870 3e4749f7 2020-10-02 op 	if ((dirfd = open(dir, O_RDONLY | O_DIRECTORY)) == -1)
871 3e4749f7 2020-10-02 op 		err(1, "open: %s", dir);
872 3e4749f7 2020-10-02 op 
873 72342dc9 2020-11-06 op 	if (unveil(dir, cgi ? "rx" : "r") == -1)
874 3e4749f7 2020-10-02 op 		err(1, "unveil");
875 3e4749f7 2020-10-02 op 
876 72342dc9 2020-11-06 op 	if (pledge(cgi ? "stdio rpath inet proc exec" : "stdio rpath inet", NULL) == -1)
877 3e4749f7 2020-10-02 op 		err(1, "pledge");
878 3e4749f7 2020-10-02 op 
879 3e4749f7 2020-10-02 op 	loop(ctx, sock);
880 3e4749f7 2020-10-02 op 
881 3e4749f7 2020-10-02 op 	close(sock);
882 3e4749f7 2020-10-02 op 	tls_free(ctx);
883 3e4749f7 2020-10-02 op 	tls_config_free(conf);
884 3e4749f7 2020-10-02 op }