Blame


1 0126d91d 2022-09-07 op /*
2 0126d91d 2022-09-07 op * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
3 0126d91d 2022-09-07 op *
4 0126d91d 2022-09-07 op * Permission to use, copy, modify, and distribute this software for any
5 0126d91d 2022-09-07 op * purpose with or without fee is hereby granted, provided that the above
6 0126d91d 2022-09-07 op * copyright notice and this permission notice appear in all copies.
7 0126d91d 2022-09-07 op *
8 0126d91d 2022-09-07 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 0126d91d 2022-09-07 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 0126d91d 2022-09-07 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 0126d91d 2022-09-07 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 0126d91d 2022-09-07 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 0126d91d 2022-09-07 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 0126d91d 2022-09-07 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 0126d91d 2022-09-07 op */
16 0126d91d 2022-09-07 op
17 0126d91d 2022-09-07 op #include "gmid.h"
18 0126d91d 2022-09-07 op
19 0126d91d 2022-09-07 op #include <sys/stat.h>
20 0126d91d 2022-09-07 op #include <sys/wait.h>
21 0126d91d 2022-09-07 op
22 0126d91d 2022-09-07 op #include <errno.h>
23 0126d91d 2022-09-07 op #include <fcntl.h>
24 0126d91d 2022-09-07 op #include <getopt.h>
25 0126d91d 2022-09-07 op #include <libgen.h>
26 0126d91d 2022-09-07 op #include <signal.h>
27 0126d91d 2022-09-07 op #include <string.h>
28 0126d91d 2022-09-07 op #include <unistd.h>
29 0126d91d 2022-09-07 op
30 0126d91d 2022-09-07 op struct imsgbuf ibuf, logibuf;
31 0126d91d 2022-09-07 op struct conf conf;
32 0126d91d 2022-09-07 op
33 0126d91d 2022-09-07 op struct fcgi fcgi[FCGI_MAX]; /* just because it's referenced */
34 0126d91d 2022-09-07 op struct vhosthead hosts;
35 0126d91d 2022-09-07 op
36 0126d91d 2022-09-07 op
37 0126d91d 2022-09-07 op static const struct option opts[] = {
38 0126d91d 2022-09-07 op {"help", no_argument, NULL, 'h'},
39 0126d91d 2022-09-07 op {"version", no_argument, NULL, 'V'},
40 0126d91d 2022-09-07 op {NULL, 0, NULL, 0},
41 0126d91d 2022-09-07 op };
42 0126d91d 2022-09-07 op
43 0126d91d 2022-09-07 op void
44 0126d91d 2022-09-07 op load_local_cert(struct vhost *h, const char *hostname, const char *dir)
45 0126d91d 2022-09-07 op {
46 0126d91d 2022-09-07 op char *cert, *key;
47 0126d91d 2022-09-07 op
48 0126d91d 2022-09-07 op if (asprintf(&cert, "%s/%s.cert.pem", dir, hostname) == -1)
49 0126d91d 2022-09-07 op errx(1, "asprintf");
50 0126d91d 2022-09-07 op if (asprintf(&key, "%s/%s.key.pem", dir, hostname) == -1)
51 0126d91d 2022-09-07 op errx(1, "asprintf");
52 0126d91d 2022-09-07 op
53 0126d91d 2022-09-07 op if (access(cert, R_OK) == -1 || access(key, R_OK) == -1)
54 0126d91d 2022-09-07 op gen_certificate(hostname, cert, key);
55 0126d91d 2022-09-07 op
56 0126d91d 2022-09-07 op h->cert = cert;
57 0126d91d 2022-09-07 op h->key = key;
58 0126d91d 2022-09-07 op h->domain = hostname;
59 0126d91d 2022-09-07 op }
60 0126d91d 2022-09-07 op
61 0126d91d 2022-09-07 op /* wrapper around dirname(3). dn must be PATH_MAX+1 at least. */
62 0126d91d 2022-09-07 op static void
63 0126d91d 2022-09-07 op pdirname(const char *path, char *dn)
64 0126d91d 2022-09-07 op {
65 0126d91d 2022-09-07 op char p[PATH_MAX+1];
66 0126d91d 2022-09-07 op char *t;
67 0126d91d 2022-09-07 op
68 0126d91d 2022-09-07 op strlcpy(p, path, sizeof(p));
69 0126d91d 2022-09-07 op t = dirname(p);
70 0126d91d 2022-09-07 op memmove(dn, t, strlen(t)+1);
71 0126d91d 2022-09-07 op }
72 0126d91d 2022-09-07 op
73 0126d91d 2022-09-07 op static void
74 0126d91d 2022-09-07 op mkdirs(const char *path, mode_t mode)
75 0126d91d 2022-09-07 op {
76 0126d91d 2022-09-07 op char dname[PATH_MAX+1];
77 0126d91d 2022-09-07 op
78 0126d91d 2022-09-07 op pdirname(path, dname);
79 0126d91d 2022-09-07 op if (!strcmp(dname, "/"))
80 0126d91d 2022-09-07 op return;
81 0126d91d 2022-09-07 op mkdirs(dname, mode);
82 0126d91d 2022-09-07 op if (mkdir(path, mode) != 0 && errno != EEXIST)
83 0126d91d 2022-09-07 op fatal("can't mkdir %s: %s", path, strerror(errno));
84 0126d91d 2022-09-07 op }
85 0126d91d 2022-09-07 op
86 0126d91d 2022-09-07 op /* $XDG_DATA_HOME/gmid */
87 0126d91d 2022-09-07 op char *
88 0126d91d 2022-09-07 op data_dir(void)
89 0126d91d 2022-09-07 op {
90 0126d91d 2022-09-07 op const char *home, *xdg;
91 0126d91d 2022-09-07 op char *t;
92 0126d91d 2022-09-07 op
93 0126d91d 2022-09-07 op if ((xdg = getenv("XDG_DATA_HOME")) == NULL) {
94 0126d91d 2022-09-07 op if ((home = getenv("HOME")) == NULL)
95 0126d91d 2022-09-07 op errx(1, "XDG_DATA_HOME and HOME both empty");
96 0126d91d 2022-09-07 op if (asprintf(&t, "%s/.local/share/gmid", home) == -1)
97 0126d91d 2022-09-07 op err(1, "asprintf");
98 0126d91d 2022-09-07 op } else {
99 0126d91d 2022-09-07 op if (asprintf(&t, "%s/gmid", xdg) == -1)
100 0126d91d 2022-09-07 op err(1, "asprintf");
101 0126d91d 2022-09-07 op }
102 0126d91d 2022-09-07 op
103 0126d91d 2022-09-07 op mkdirs(t, 0755);
104 0126d91d 2022-09-07 op return t;
105 0126d91d 2022-09-07 op }
106 0126d91d 2022-09-07 op
107 0126d91d 2022-09-07 op static void
108 0126d91d 2022-09-07 op logger_init(void)
109 0126d91d 2022-09-07 op {
110 0126d91d 2022-09-07 op int p[2];
111 0126d91d 2022-09-07 op
112 0126d91d 2022-09-07 op if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, p) == -1)
113 0126d91d 2022-09-07 op err(1, "socketpair");
114 0126d91d 2022-09-07 op
115 0126d91d 2022-09-07 op switch (fork()) {
116 0126d91d 2022-09-07 op case -1:
117 0126d91d 2022-09-07 op err(1, "fork");
118 0126d91d 2022-09-07 op case 0:
119 0126d91d 2022-09-07 op close(p[0]);
120 0126d91d 2022-09-07 op setproctitle("logger");
121 0126d91d 2022-09-07 op imsg_init(&logibuf, p[1]);
122 0126d91d 2022-09-07 op _exit(logger_main(p[1], &logibuf));
123 0126d91d 2022-09-07 op default:
124 0126d91d 2022-09-07 op close(p[1]);
125 0126d91d 2022-09-07 op imsg_init(&logibuf, p[0]);
126 0126d91d 2022-09-07 op return;
127 0126d91d 2022-09-07 op }
128 0126d91d 2022-09-07 op }
129 0126d91d 2022-09-07 op
130 0126d91d 2022-09-07 op static int
131 0126d91d 2022-09-07 op serve(const char *host, int port, const char *dir, struct tls *ctx)
132 0126d91d 2022-09-07 op {
133 0126d91d 2022-09-07 op struct addrinfo hints, *res, *res0;
134 0126d91d 2022-09-07 op int error, saved_errno, sock = -1;
135 0126d91d 2022-09-07 op const char *cause = NULL;
136 0126d91d 2022-09-07 op char service[32];
137 0126d91d 2022-09-07 op
138 0126d91d 2022-09-07 op if (snprintf(service, sizeof(service), "%d", port) < 0)
139 0126d91d 2022-09-07 op fatal("snprintf");
140 0126d91d 2022-09-07 op
141 0126d91d 2022-09-07 op memset(&hints, 0, sizeof(hints));
142 0126d91d 2022-09-07 op hints.ai_family = AF_UNSPEC;
143 0126d91d 2022-09-07 op hints.ai_socktype = SOCK_STREAM;
144 0126d91d 2022-09-07 op hints.ai_flags = AI_PASSIVE;
145 0126d91d 2022-09-07 op error = getaddrinfo(host, service, &hints, &res0);
146 0126d91d 2022-09-07 op if (error)
147 0126d91d 2022-09-07 op fatal("%s", gai_strerror(error));
148 0126d91d 2022-09-07 op for (res = res0; res; res = res->ai_next) {
149 0126d91d 2022-09-07 op sock = socket(res->ai_family, res->ai_socktype,
150 0126d91d 2022-09-07 op res->ai_protocol);
151 0126d91d 2022-09-07 op if (sock == -1) {
152 0126d91d 2022-09-07 op cause = "socket";
153 0126d91d 2022-09-07 op continue;
154 0126d91d 2022-09-07 op }
155 0126d91d 2022-09-07 op
156 0126d91d 2022-09-07 op if (bind(sock, res->ai_addr, res->ai_addrlen) == -1) {
157 0126d91d 2022-09-07 op cause = "bind";
158 0126d91d 2022-09-07 op saved_errno = errno;
159 0126d91d 2022-09-07 op close(sock);
160 0126d91d 2022-09-07 op errno = saved_errno;
161 0126d91d 2022-09-07 op continue;
162 0126d91d 2022-09-07 op }
163 0126d91d 2022-09-07 op
164 0126d91d 2022-09-07 op if (listen(sock, 5) == -1)
165 0126d91d 2022-09-07 op fatal("listen");
166 0126d91d 2022-09-07 op
167 0126d91d 2022-09-07 op /*
168 0126d91d 2022-09-07 op * for the time being, we're happy as soon as
169 0126d91d 2022-09-07 op * something binds.
170 0126d91d 2022-09-07 op */
171 0126d91d 2022-09-07 op break;
172 0126d91d 2022-09-07 op }
173 0126d91d 2022-09-07 op
174 0126d91d 2022-09-07 op if (sock == -1)
175 0126d91d 2022-09-07 op fatal("%s", cause);
176 0126d91d 2022-09-07 op freeaddrinfo(res0);
177 0126d91d 2022-09-07 op
178 0126d91d 2022-09-07 op log_notice(NULL, "serving %s on port %d", dir, port);
179 0126d91d 2022-09-07 op loop(ctx, sock, -1, NULL);
180 0126d91d 2022-09-07 op return 0;
181 0126d91d 2022-09-07 op }
182 0126d91d 2022-09-07 op
183 0126d91d 2022-09-07 op static __dead void
184 0126d91d 2022-09-07 op usage(void)
185 0126d91d 2022-09-07 op {
186 0126d91d 2022-09-07 op fprintf(stderr,
187 c5b4db93 2022-09-10 op "Version: " GE_STRING "\n"
188 0126d91d 2022-09-07 op "Usage: %s [-hVv] [-d certs-dir] [-H hostname] [-p port] [dir]\n",
189 0126d91d 2022-09-07 op getprogname());
190 0126d91d 2022-09-07 op exit(1);
191 0126d91d 2022-09-07 op }
192 0126d91d 2022-09-07 op
193 0126d91d 2022-09-07 op int
194 0126d91d 2022-09-07 op main(int argc, char **argv)
195 0126d91d 2022-09-07 op {
196 0126d91d 2022-09-07 op struct tls_config *tlsconf;
197 0126d91d 2022-09-07 op struct tls *ctx;
198 0126d91d 2022-09-07 op struct vhost *host;
199 0126d91d 2022-09-07 op struct location *loc;
200 0126d91d 2022-09-07 op const char *errstr, *certs_dir = NULL, *hostname = "localhost";
201 0126d91d 2022-09-07 op char path[PATH_MAX];
202 0126d91d 2022-09-07 op int ch;
203 0126d91d 2022-09-07 op
204 0126d91d 2022-09-07 op logger_init();
205 0126d91d 2022-09-07 op conf.port = 1965;
206 0126d91d 2022-09-07 op
207 0126d91d 2022-09-07 op while ((ch = getopt_long(argc, argv, "d:H:hp:Vv", opts, NULL)) != -1) {
208 0126d91d 2022-09-07 op switch (ch) {
209 0126d91d 2022-09-07 op case 'd':
210 0126d91d 2022-09-07 op certs_dir = optarg;
211 0126d91d 2022-09-07 op break;
212 0126d91d 2022-09-07 op case 'H':
213 0126d91d 2022-09-07 op hostname = optarg;
214 0126d91d 2022-09-07 op break;
215 0126d91d 2022-09-07 op case 'h':
216 0126d91d 2022-09-07 op usage();
217 0126d91d 2022-09-07 op break;
218 0126d91d 2022-09-07 op case 'p':
219 0126d91d 2022-09-07 op conf.port = strtonum(optarg, 0, UINT16_MAX, &errstr);
220 0126d91d 2022-09-07 op if (errstr)
221 0126d91d 2022-09-07 op fatal("port number is %s: %s", errstr, optarg);
222 0126d91d 2022-09-07 op break;
223 0126d91d 2022-09-07 op case 'V':
224 c5b4db93 2022-09-10 op puts("Version: " GE_STRING);
225 0126d91d 2022-09-07 op return 0;
226 0126d91d 2022-09-07 op default:
227 0126d91d 2022-09-07 op usage();
228 0126d91d 2022-09-07 op break;
229 0126d91d 2022-09-07 op }
230 0126d91d 2022-09-07 op }
231 0126d91d 2022-09-07 op argc -= optind;
232 0126d91d 2022-09-07 op argv += optind;
233 0126d91d 2022-09-07 op
234 0126d91d 2022-09-07 op if (argc > 1)
235 0126d91d 2022-09-07 op usage();
236 0126d91d 2022-09-07 op
237 0126d91d 2022-09-07 op /* prepare the configuration */
238 0126d91d 2022-09-07 op conf.verbose = 1;
239 0126d91d 2022-09-07 op init_mime(&conf.mime);
240 0126d91d 2022-09-07 op
241 0126d91d 2022-09-07 op if (certs_dir == NULL)
242 0126d91d 2022-09-07 op certs_dir = data_dir();
243 0126d91d 2022-09-07 op
244 0126d91d 2022-09-07 op if (load_default_mime(&conf.mime) == -1)
245 0126d91d 2022-09-07 op fatal("can't load default mime types");
246 0126d91d 2022-09-07 op sort_mime(&conf.mime);
247 0126d91d 2022-09-07 op
248 0126d91d 2022-09-07 op /* set up the implicit vhost and location */
249 0126d91d 2022-09-07 op
250 0126d91d 2022-09-07 op host = xcalloc(1, sizeof(*host));
251 0126d91d 2022-09-07 op TAILQ_INSERT_HEAD(&hosts, host, vhosts);
252 0126d91d 2022-09-07 op
253 0126d91d 2022-09-07 op loc = xcalloc(1, sizeof(*loc));
254 0126d91d 2022-09-07 op loc->fcgi = -1;
255 0126d91d 2022-09-07 op TAILQ_INSERT_HEAD(&host->locations, loc, locations);
256 0126d91d 2022-09-07 op
257 0126d91d 2022-09-07 op load_local_cert(host, hostname, certs_dir);
258 0126d91d 2022-09-07 op
259 0126d91d 2022-09-07 op host->domain = "*";
260 0126d91d 2022-09-07 op loc->auto_index = 1;
261 0126d91d 2022-09-07 op loc->match = "*";
262 0126d91d 2022-09-07 op
263 0126d91d 2022-09-07 op if (*argv == NULL) {
264 0126d91d 2022-09-07 op if (getcwd(path, sizeof(path)) == NULL)
265 0126d91d 2022-09-07 op fatal("getcwd");
266 0126d91d 2022-09-07 op loc->dir = path;
267 0126d91d 2022-09-07 op } else
268 0126d91d 2022-09-07 op loc->dir = absolutify_path(*argv);
269 0126d91d 2022-09-07 op
270 0126d91d 2022-09-07 op if ((loc->dirfd = open(loc->dir, O_RDONLY|O_DIRECTORY)) == -1)
271 0126d91d 2022-09-07 op fatal("can't open %s", loc->dir);
272 0126d91d 2022-09-07 op
273 0126d91d 2022-09-07 op /* setup tls */
274 0126d91d 2022-09-07 op
275 0126d91d 2022-09-07 op if ((tlsconf = tls_config_new()) == NULL)
276 0126d91d 2022-09-07 op fatal("tls_config_new"); /* XXX: fatalx */
277 0126d91d 2022-09-07 op
278 0126d91d 2022-09-07 op /* optionally accept client certs but don't try to verify them */
279 0126d91d 2022-09-07 op tls_config_verify_client_optional(tlsconf);
280 0126d91d 2022-09-07 op tls_config_insecure_noverifycert(tlsconf);
281 0126d91d 2022-09-07 op
282 0126d91d 2022-09-07 op if ((ctx = tls_server()) == NULL)
283 0126d91d 2022-09-07 op fatal("tls_server failure"); /* XXX: fatalx */
284 0126d91d 2022-09-07 op
285 0126d91d 2022-09-07 op if (tls_config_set_keypair_file(tlsconf, host->cert, host->key))
286 0126d91d 2022-09-07 op fatal("can't load the keypair (%s, %s)",
287 0126d91d 2022-09-07 op host->cert, host->key);
288 0126d91d 2022-09-07 op
289 0126d91d 2022-09-07 op if (tls_configure(ctx, tlsconf) == -1)
290 0126d91d 2022-09-07 op fatal("tls_configure: %s", tls_error(ctx));
291 0126d91d 2022-09-07 op
292 0126d91d 2022-09-07 op /* start the server */
293 0126d91d 2022-09-07 op signal(SIGPIPE, SIG_IGN);
294 0126d91d 2022-09-07 op setproctitle("%s", loc->dir);
295 0126d91d 2022-09-07 op return serve(hostname, conf.port, loc->dir, ctx);
296 0126d91d 2022-09-07 op }