Blame


1 3e4749f7 2020-10-02 op /*
2 a555e0d6 2022-07-04 op * Copyright (c) 2020, 2021, 2022 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 52418c8d 2021-02-12 op
17 52418c8d 2021-02-12 op #include "gmid.h"
18 3e4749f7 2020-10-02 op
19 8443bff7 2021-01-25 op #include <sys/stat.h>
20 8443bff7 2021-01-25 op
21 592fd624 2020-10-07 op #include <errno.h>
22 3e4749f7 2020-10-02 op #include <fcntl.h>
23 5777923b 2021-06-29 op #include <getopt.h>
24 0046c1fe 2023-06-06 op #include <locale.h>
25 7e1df73d 2021-03-31 op #include <libgen.h>
26 bcf5d929 2021-02-01 op #include <limits.h>
27 c9e97a6e 2022-12-24 op #include <grp.h>
28 ae08ec7d 2021-01-25 op #include <pwd.h>
29 0cf902af 2020-11-03 op #include <signal.h>
30 3e4749f7 2020-10-02 op #include <string.h>
31 eae52ad4 2023-06-06 op #include <syslog.h>
32 cc68fe70 2020-10-07 op
33 eae52ad4 2023-06-06 op #include "log.h"
34 c26f2460 2023-06-08 op #include "proc.h"
35 df5058c9 2023-06-05 op
36 c26f2460 2023-06-08 op #ifndef nitems
37 c26f2460 2023-06-08 op #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
38 c26f2460 2023-06-08 op #endif
39 5777923b 2021-06-29 op
40 c26f2460 2023-06-08 op static int main_configure(struct conf *);
41 c26f2460 2023-06-08 op static void main_configure_done(struct conf *);
42 c26f2460 2023-06-08 op static void main_reload(struct conf *);
43 c26f2460 2023-06-08 op static void main_sig_handler(int, short, void *);
44 c26f2460 2023-06-08 op static int main_dispatch_server(int, struct privsep_proc *, struct imsg *);
45 c26f2460 2023-06-08 op static int main_dispatch_logger(int, struct privsep_proc *, struct imsg *);
46 c26f2460 2023-06-08 op static void __dead main_shutdown(struct conf *);
47 c26f2460 2023-06-08 op
48 c26f2460 2023-06-08 op static struct privsep_proc procs[] = {
49 c26f2460 2023-06-08 op { "server", PROC_SERVER, main_dispatch_server, server },
50 c26f2460 2023-06-08 op { "logger", PROC_LOGGER, main_dispatch_logger, logger },
51 c26f2460 2023-06-08 op };
52 c26f2460 2023-06-08 op
53 c26f2460 2023-06-08 op static const char *opts = "c:D:fI:hnP:T:Vv";
54 c26f2460 2023-06-08 op
55 e5d82d94 2022-03-19 op static const struct option longopts[] = {
56 5777923b 2021-06-29 op {"help", no_argument, NULL, 'h'},
57 5777923b 2021-06-29 op {"version", no_argument, NULL, 'V'},
58 5777923b 2021-06-29 op {NULL, 0, NULL, 0},
59 5777923b 2021-06-29 op };
60 8ad1c570 2021-05-09 op
61 b8e64ccd 2021-03-31 op struct vhosthead hosts;
62 15902770 2021-01-15 op
63 bc99d868 2021-03-19 op int sock4, sock6;
64 c26f2460 2023-06-08 op int privsep_process;
65 c26f2460 2023-06-08 op int pidfd = -1;
66 d672b8fb 2021-02-03 op
67 ca84625a 2023-06-08 op int debug, verbose;
68 ca84625a 2023-06-08 op
69 32fbc478 2022-09-08 op const char *config_path = "/etc/gmid.conf";
70 32fbc478 2022-09-08 op const char *pidfile;
71 881a9dd9 2021-01-16 op
72 15902770 2021-01-15 op struct conf conf;
73 15902770 2021-01-15 op
74 3abf91b0 2021-02-07 op static void
75 9327bc04 2021-06-29 op usage(void)
76 3abf91b0 2021-02-07 op {
77 3abf91b0 2021-02-07 op fprintf(stderr,
78 0be2a537 2021-06-29 op "Version: " GMID_STRING "\n"
79 0ac785a6 2023-06-05 op "Usage: %s [-fnv] [-c config] [-D macro=value] [-P pidfile]\n",
80 9327bc04 2021-06-29 op getprogname());
81 3abf91b0 2021-02-07 op }
82 3abf91b0 2021-02-07 op
83 47b0ff10 2023-06-08 op /* used by the server process, defined here so gg can provide its own impl. */
84 47b0ff10 2023-06-08 op void
85 47b0ff10 2023-06-08 op log_request(struct client *c, char *meta, size_t l)
86 47b0ff10 2023-06-08 op {
87 47b0ff10 2023-06-08 op char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV], b[GEMINI_URL_LEN];
88 47b0ff10 2023-06-08 op char *fmted;
89 47b0ff10 2023-06-08 op const char *t;
90 47b0ff10 2023-06-08 op size_t len;
91 47b0ff10 2023-06-08 op int ec;
92 47b0ff10 2023-06-08 op
93 47b0ff10 2023-06-08 op len = sizeof(c->addr);
94 47b0ff10 2023-06-08 op ec = getnameinfo((struct sockaddr*)&c->addr, len,
95 47b0ff10 2023-06-08 op hbuf, sizeof(hbuf),
96 47b0ff10 2023-06-08 op sbuf, sizeof(sbuf),
97 47b0ff10 2023-06-08 op NI_NUMERICHOST | NI_NUMERICSERV);
98 47b0ff10 2023-06-08 op if (ec != 0)
99 47b0ff10 2023-06-08 op fatalx("getnameinfo: %s", gai_strerror(ec));
100 47b0ff10 2023-06-08 op
101 47b0ff10 2023-06-08 op if (c->iri.schema != NULL) {
102 47b0ff10 2023-06-08 op /* serialize the IRI */
103 47b0ff10 2023-06-08 op strlcpy(b, c->iri.schema, sizeof(b));
104 47b0ff10 2023-06-08 op strlcat(b, "://", sizeof(b));
105 47b0ff10 2023-06-08 op
106 47b0ff10 2023-06-08 op /* log the decoded host name, but if it was invalid
107 47b0ff10 2023-06-08 op * use the raw one. */
108 47b0ff10 2023-06-08 op if (*c->domain != '\0')
109 47b0ff10 2023-06-08 op strlcat(b, c->domain, sizeof(b));
110 47b0ff10 2023-06-08 op else
111 47b0ff10 2023-06-08 op strlcat(b, c->iri.host, sizeof(b));
112 47b0ff10 2023-06-08 op
113 47b0ff10 2023-06-08 op if (*c->iri.path != '/')
114 47b0ff10 2023-06-08 op strlcat(b, "/", sizeof(b));
115 47b0ff10 2023-06-08 op strlcat(b, c->iri.path, sizeof(b)); /* TODO: sanitize UTF8 */
116 47b0ff10 2023-06-08 op if (*c->iri.query != '\0') { /* TODO: sanitize UTF8 */
117 47b0ff10 2023-06-08 op strlcat(b, "?", sizeof(b));
118 47b0ff10 2023-06-08 op strlcat(b, c->iri.query, sizeof(b));
119 47b0ff10 2023-06-08 op }
120 47b0ff10 2023-06-08 op } else {
121 47b0ff10 2023-06-08 op if ((t = c->req) == NULL)
122 47b0ff10 2023-06-08 op t = "";
123 47b0ff10 2023-06-08 op strlcpy(b, t, sizeof(b));
124 47b0ff10 2023-06-08 op }
125 47b0ff10 2023-06-08 op
126 47b0ff10 2023-06-08 op if ((t = memchr(meta, '\r', l)) == NULL)
127 47b0ff10 2023-06-08 op t = meta + len;
128 47b0ff10 2023-06-08 op
129 47b0ff10 2023-06-08 op ec = asprintf(&fmted, "%s:%s GET %s %.*s", hbuf, sbuf, b,
130 47b0ff10 2023-06-08 op (int)(t-meta), meta);
131 47b0ff10 2023-06-08 op if (ec == -1)
132 47b0ff10 2023-06-08 op err(1, "asprintf");
133 47b0ff10 2023-06-08 op
134 47b0ff10 2023-06-08 op proc_compose(conf.ps, PROC_LOGGER, IMSG_LOG_REQUEST,
135 47b0ff10 2023-06-08 op fmted, ec + 1);
136 47b0ff10 2023-06-08 op
137 47b0ff10 2023-06-08 op free(fmted);
138 47b0ff10 2023-06-08 op }
139 47b0ff10 2023-06-08 op
140 419a4235 2021-04-28 op static int
141 419a4235 2021-04-28 op write_pidfile(const char *pidfile)
142 419a4235 2021-04-28 op {
143 419a4235 2021-04-28 op struct flock lock;
144 419a4235 2021-04-28 op int fd;
145 d672b8fb 2021-02-03 op
146 419a4235 2021-04-28 op if (pidfile == NULL)
147 419a4235 2021-04-28 op return -1;
148 419a4235 2021-04-28 op
149 419a4235 2021-04-28 op if ((fd = open(pidfile, O_WRONLY|O_CREAT|O_CLOEXEC, 0600)) == -1)
150 df5058c9 2023-06-05 op fatal("can't open pidfile %s", pidfile);
151 419a4235 2021-04-28 op
152 419a4235 2021-04-28 op lock.l_start = 0;
153 419a4235 2021-04-28 op lock.l_len = 0;
154 419a4235 2021-04-28 op lock.l_type = F_WRLCK;
155 419a4235 2021-04-28 op lock.l_whence = SEEK_SET;
156 419a4235 2021-04-28 op
157 419a4235 2021-04-28 op if (fcntl(fd, F_SETLK, &lock) == -1)
158 df5058c9 2023-06-05 op fatalx("can't lock %s, gmid is already running?", pidfile);
159 419a4235 2021-04-28 op
160 419a4235 2021-04-28 op if (ftruncate(fd, 0) == -1)
161 df5058c9 2023-06-05 op fatal("ftruncate %s", pidfile);
162 419a4235 2021-04-28 op
163 419a4235 2021-04-28 op dprintf(fd, "%d\n", getpid());
164 419a4235 2021-04-28 op
165 419a4235 2021-04-28 op return fd;
166 419a4235 2021-04-28 op }
167 419a4235 2021-04-28 op
168 8d6ae384 2021-01-24 op int
169 8d6ae384 2021-01-24 op main(int argc, char **argv)
170 8d6ae384 2021-01-24 op {
171 c26f2460 2023-06-08 op struct privsep *ps;
172 c26f2460 2023-06-08 op const char *errstr, *title = NULL;
173 c26f2460 2023-06-08 op size_t i;
174 c26f2460 2023-06-08 op int ch, conftest = 0;
175 c26f2460 2023-06-08 op int proc_instance = 0;
176 c26f2460 2023-06-08 op int proc_id = PROC_PARENT;
177 c26f2460 2023-06-08 op int argc0 = argc;
178 501e489c 2021-01-24 op
179 0046c1fe 2023-06-06 op setlocale(LC_CTYPE, "");
180 0046c1fe 2023-06-06 op
181 eae52ad4 2023-06-06 op /* log to stderr until daemonized */
182 eae52ad4 2023-06-06 op log_init(1, LOG_DAEMON);
183 c68baad2 2023-06-06 op config_init();
184 8d6ae384 2021-01-24 op
185 5777923b 2021-06-29 op while ((ch = getopt_long(argc, argv, opts, longopts, NULL)) != -1) {
186 3e4749f7 2020-10-02 op switch (ch) {
187 0ac785a6 2023-06-05 op case 'c':
188 0ac785a6 2023-06-05 op config_path = absolutify_path(optarg);
189 0ac785a6 2023-06-05 op break;
190 f98e9045 2021-06-29 op case 'D':
191 f98e9045 2021-06-29 op if (cmdline_symset(optarg) == -1)
192 df5058c9 2023-06-05 op fatalx("could not parse macro definition: %s",
193 8a50fc03 2021-07-07 op optarg);
194 f98e9045 2021-06-29 op break;
195 0ac785a6 2023-06-05 op case 'f':
196 ca84625a 2023-06-08 op debug = 1;
197 46af8c6c 2021-01-27 op break;
198 3e4749f7 2020-10-02 op case 'h':
199 9327bc04 2021-06-29 op usage();
200 3e4749f7 2020-10-02 op return 0;
201 c26f2460 2023-06-08 op case 'I':
202 c26f2460 2023-06-08 op proc_instance = strtonum(optarg, 0, PROC_MAX_INSTANCES,
203 c26f2460 2023-06-08 op &errstr);
204 c26f2460 2023-06-08 op if (errstr != NULL)
205 c26f2460 2023-06-08 op fatalx("invalid process instance");
206 c26f2460 2023-06-08 op break;
207 15902770 2021-01-15 op case 'n':
208 f0a01fc7 2021-10-09 op conftest++;
209 721e2325 2020-11-18 op break;
210 8e8b2e25 2021-04-28 op case 'P':
211 f1f13cb7 2023-06-08 op pidfile = absolutify_path(optarg);
212 8e8b2e25 2021-04-28 op break;
213 c26f2460 2023-06-08 op case 'T':
214 c26f2460 2023-06-08 op title = optarg;
215 c26f2460 2023-06-08 op proc_id = proc_getid(procs, nitems(procs), title);
216 c26f2460 2023-06-08 op if (proc_id == PROC_MAX)
217 c26f2460 2023-06-08 op fatalx("invalid process name");
218 c26f2460 2023-06-08 op break;
219 5777923b 2021-06-29 op case 'V':
220 fdb43a4c 2021-06-29 op puts("Version: " GMID_STRING);
221 5777923b 2021-06-29 op return 0;
222 8904fa0e 2021-01-27 op case 'v':
223 ca84625a 2023-06-08 op verbose = 1;
224 8904fa0e 2021-01-27 op break;
225 3e4749f7 2020-10-02 op default:
226 9327bc04 2021-06-29 op usage();
227 3e4749f7 2020-10-02 op return 1;
228 3e4749f7 2020-10-02 op }
229 3e4749f7 2020-10-02 op }
230 3e4749f7 2020-10-02 op
231 c26f2460 2023-06-08 op if (argc - optind != 0)
232 d29a2ee2 2022-09-06 op usage();
233 d29a2ee2 2022-09-06 op
234 32fbc478 2022-09-08 op parse_conf(config_path);
235 c26f2460 2023-06-08 op if (*conf.chroot != '\0' && *conf.user == '\0')
236 c26f2460 2023-06-08 op fatalx("can't chroot without a user to switch to after.");
237 d672b8fb 2021-02-03 op
238 132cae8c 2021-01-18 op if (conftest) {
239 f0a01fc7 2021-10-09 op fprintf(stderr, "config OK\n");
240 f0a01fc7 2021-10-09 op if (conftest > 1)
241 f0a01fc7 2021-10-09 op print_conf();
242 132cae8c 2021-01-18 op return 0;
243 132cae8c 2021-01-18 op }
244 4a28dd01 2020-12-28 op
245 c26f2460 2023-06-08 op if ((ps = calloc(1, sizeof(*ps))) == NULL)
246 c26f2460 2023-06-08 op fatal("calloc");
247 c26f2460 2023-06-08 op ps->ps_env = &conf;
248 c26f2460 2023-06-08 op conf.ps = ps;
249 c26f2460 2023-06-08 op if (*conf.user) {
250 c26f2460 2023-06-08 op if (geteuid())
251 c26f2460 2023-06-08 op fatalx("need root privileges");
252 c26f2460 2023-06-08 op if ((ps->ps_pw = getpwnam(conf.user)) == NULL)
253 c26f2460 2023-06-08 op fatalx("unknown user %s", conf.user);
254 c26f2460 2023-06-08 op }
255 8a50fc03 2021-07-07 op
256 c26f2460 2023-06-08 op ps->ps_instances[PROC_SERVER] = conf.prefork;
257 c26f2460 2023-06-08 op ps->ps_instance = proc_instance;
258 c26f2460 2023-06-08 op if (title != NULL)
259 c26f2460 2023-06-08 op ps->ps_title[proc_id] = title;
260 c26f2460 2023-06-08 op
261 c26f2460 2023-06-08 op if (*conf.chroot != '\0') {
262 c26f2460 2023-06-08 op for (i = 0; i < nitems(procs); ++i)
263 c26f2460 2023-06-08 op procs[i].p_chroot = conf.chroot;
264 0170ba02 2021-01-17 op }
265 c26f2460 2023-06-08 op
266 ca84625a 2023-06-08 op log_init(debug, LOG_DAEMON);
267 ca84625a 2023-06-08 op log_setverbose(verbose);
268 c26f2460 2023-06-08 op if (title != NULL)
269 c26f2460 2023-06-08 op log_procinit(title);
270 4a28dd01 2020-12-28 op
271 c26f2460 2023-06-08 op /* only the parent returns */
272 ca84625a 2023-06-08 op proc_init(ps, procs, nitems(procs), debug, argc0, argv, proc_id);
273 3e4749f7 2020-10-02 op
274 c26f2460 2023-06-08 op log_procinit("main");
275 ca84625a 2023-06-08 op if (!debug && daemon(0, 0) == -1)
276 c26f2460 2023-06-08 op fatal("daemon");
277 3841a369 2021-04-20 op
278 8e8b2e25 2021-04-28 op pidfd = write_pidfile(pidfile);
279 8e8b2e25 2021-04-28 op
280 c26f2460 2023-06-08 op sandbox_main_process();
281 c26f2460 2023-06-08 op
282 c26f2460 2023-06-08 op event_init();
283 c26f2460 2023-06-08 op
284 c26f2460 2023-06-08 op signal(SIGPIPE, SIG_IGN);
285 b9c9123b 2021-03-20 op
286 c26f2460 2023-06-08 op signal_set(&ps->ps_evsigint, SIGINT, main_sig_handler, ps);
287 c26f2460 2023-06-08 op signal_set(&ps->ps_evsigterm, SIGTERM, main_sig_handler, ps);
288 c26f2460 2023-06-08 op signal_set(&ps->ps_evsigchld, SIGCHLD, main_sig_handler, ps);
289 c26f2460 2023-06-08 op signal_set(&ps->ps_evsighup, SIGHUP, main_sig_handler, ps);
290 ca21e100 2021-02-04 op
291 c26f2460 2023-06-08 op signal_add(&ps->ps_evsigint, NULL);
292 c26f2460 2023-06-08 op signal_add(&ps->ps_evsigterm, NULL);
293 c26f2460 2023-06-08 op signal_add(&ps->ps_evsigchld, NULL);
294 c26f2460 2023-06-08 op signal_add(&ps->ps_evsighup, NULL);
295 1d3eb470 2021-03-20 op
296 c26f2460 2023-06-08 op proc_connect(ps);
297 ca21e100 2021-02-04 op
298 c26f2460 2023-06-08 op if (main_configure(&conf) == -1)
299 c26f2460 2023-06-08 op fatal("configuration failed");
300 bc99d868 2021-03-19 op
301 c26f2460 2023-06-08 op event_dispatch();
302 c26f2460 2023-06-08 op main_shutdown(&conf);
303 c26f2460 2023-06-08 op /* NOTREACHED */
304 c26f2460 2023-06-08 op return 0;
305 c26f2460 2023-06-08 op }
306 ca21e100 2021-02-04 op
307 c26f2460 2023-06-08 op static int
308 c26f2460 2023-06-08 op main_configure(struct conf *conf)
309 c26f2460 2023-06-08 op {
310 c26f2460 2023-06-08 op struct privsep *ps = conf->ps;
311 ca21e100 2021-02-04 op
312 c26f2460 2023-06-08 op conf->reload = conf->prefork;
313 ca21e100 2021-02-04 op
314 c26f2460 2023-06-08 op if (proc_compose(ps, PROC_SERVER, IMSG_RECONF_START, NULL, 0) == -1)
315 c26f2460 2023-06-08 op return -1;
316 ca21e100 2021-02-04 op
317 5d22294a 2023-06-09 op if (config_send(conf, &hosts) == -1)
318 c26f2460 2023-06-08 op return -1;
319 c26f2460 2023-06-08 op
320 c26f2460 2023-06-08 op if (proc_compose(ps, PROC_SERVER, IMSG_RECONF_END, NULL, 0) == -1)
321 c26f2460 2023-06-08 op return -1;
322 c26f2460 2023-06-08 op
323 c26f2460 2023-06-08 op return 0;
324 c26f2460 2023-06-08 op }
325 c26f2460 2023-06-08 op
326 c26f2460 2023-06-08 op static void
327 c26f2460 2023-06-08 op main_configure_done(struct conf *conf)
328 c26f2460 2023-06-08 op {
329 c26f2460 2023-06-08 op if (conf->reload == 0) {
330 c26f2460 2023-06-08 op log_warnx("configuration already done");
331 c26f2460 2023-06-08 op return;
332 ca21e100 2021-02-04 op }
333 1d3eb470 2021-03-20 op
334 c26f2460 2023-06-08 op conf->reload--;
335 c26f2460 2023-06-08 op /* send IMSG_CTL_START? */
336 c26f2460 2023-06-08 op }
337 c26f2460 2023-06-08 op
338 c26f2460 2023-06-08 op static void
339 c26f2460 2023-06-08 op main_reload(struct conf *conf)
340 c26f2460 2023-06-08 op {
341 c26f2460 2023-06-08 op if (conf->reload) {
342 c26f2460 2023-06-08 op log_debug("%s: already in progress: %d pending",
343 c26f2460 2023-06-08 op __func__, conf->reload);
344 c26f2460 2023-06-08 op return;
345 5c485529 2022-09-10 op }
346 1d3eb470 2021-03-20 op
347 c26f2460 2023-06-08 op log_debug("%s: config file %s", __func__, config_path);
348 c26f2460 2023-06-08 op config_free();
349 c26f2460 2023-06-08 op parse_conf(config_path); /* XXX should handle error here */
350 1d3eb470 2021-03-20 op
351 c26f2460 2023-06-08 op main_configure(conf);
352 c26f2460 2023-06-08 op }
353 c26f2460 2023-06-08 op
354 c26f2460 2023-06-08 op static void
355 c26f2460 2023-06-08 op main_sig_handler(int sig, short ev, void *arg)
356 c26f2460 2023-06-08 op {
357 c26f2460 2023-06-08 op struct privsep *ps = arg;
358 c26f2460 2023-06-08 op
359 c26f2460 2023-06-08 op /*
360 c26f2460 2023-06-08 op * Normal signal handler rules don't apply here because libevent
361 c26f2460 2023-06-08 op * decouples for us.
362 c26f2460 2023-06-08 op */
363 c26f2460 2023-06-08 op
364 c26f2460 2023-06-08 op switch (sig) {
365 c26f2460 2023-06-08 op case SIGHUP:
366 c26f2460 2023-06-08 op if (privsep_process != PROC_PARENT)
367 c26f2460 2023-06-08 op return;
368 c26f2460 2023-06-08 op log_info("reload requested with SIGHUP");
369 c26f2460 2023-06-08 op main_reload(ps->ps_env);
370 c26f2460 2023-06-08 op break;
371 c26f2460 2023-06-08 op case SIGCHLD:
372 c26f2460 2023-06-08 op log_warnx("one child died, quitting");
373 c26f2460 2023-06-08 op /* fallthrough */
374 c26f2460 2023-06-08 op case SIGTERM:
375 c26f2460 2023-06-08 op case SIGINT:
376 c26f2460 2023-06-08 op main_shutdown(ps->ps_env);
377 c26f2460 2023-06-08 op break;
378 c26f2460 2023-06-08 op default:
379 c26f2460 2023-06-08 op fatalx("unexpected signal %d", sig);
380 c26f2460 2023-06-08 op }
381 c26f2460 2023-06-08 op }
382 c26f2460 2023-06-08 op
383 c26f2460 2023-06-08 op static int
384 c26f2460 2023-06-08 op main_dispatch_server(int fd, struct privsep_proc *p, struct imsg *imsg)
385 c26f2460 2023-06-08 op {
386 c26f2460 2023-06-08 op struct privsep *ps = p->p_ps;
387 c26f2460 2023-06-08 op struct conf *conf = ps->ps_env;
388 c26f2460 2023-06-08 op
389 c26f2460 2023-06-08 op switch (imsg->hdr.type) {
390 c26f2460 2023-06-08 op case IMSG_RECONF_DONE:
391 c26f2460 2023-06-08 op main_configure_done(conf);
392 c26f2460 2023-06-08 op break;
393 c26f2460 2023-06-08 op default:
394 c26f2460 2023-06-08 op return -1;
395 c26f2460 2023-06-08 op }
396 c26f2460 2023-06-08 op
397 c26f2460 2023-06-08 op return 0;
398 c26f2460 2023-06-08 op }
399 c26f2460 2023-06-08 op
400 c26f2460 2023-06-08 op static int
401 c26f2460 2023-06-08 op main_dispatch_logger(int fd, struct privsep_proc *p, struct imsg *imsg)
402 c26f2460 2023-06-08 op {
403 c26f2460 2023-06-08 op struct privsep *ps = p->p_ps;
404 c26f2460 2023-06-08 op struct conf *conf = ps->ps_env;
405 c26f2460 2023-06-08 op
406 c26f2460 2023-06-08 op switch (imsg->hdr.type) {
407 c26f2460 2023-06-08 op case IMSG_RECONF_DONE:
408 c26f2460 2023-06-08 op main_configure_done(conf);
409 c26f2460 2023-06-08 op break;
410 c26f2460 2023-06-08 op default:
411 c26f2460 2023-06-08 op return -1;
412 c26f2460 2023-06-08 op }
413 c26f2460 2023-06-08 op
414 c26f2460 2023-06-08 op return 0;
415 c26f2460 2023-06-08 op }
416 c26f2460 2023-06-08 op
417 c26f2460 2023-06-08 op static void __dead
418 c26f2460 2023-06-08 op main_shutdown(struct conf *conf)
419 c26f2460 2023-06-08 op {
420 c26f2460 2023-06-08 op proc_kill(conf->ps);
421 c26f2460 2023-06-08 op config_free();
422 c26f2460 2023-06-08 op free(conf->ps);
423 c26f2460 2023-06-08 op /* free(conf); */
424 c26f2460 2023-06-08 op
425 c26f2460 2023-06-08 op log_info("parent terminating, pid %d", getpid());
426 c26f2460 2023-06-08 op
427 8e8b2e25 2021-04-28 op if (pidfd != -1)
428 8e8b2e25 2021-04-28 op close(pidfd);
429 8e8b2e25 2021-04-28 op
430 c26f2460 2023-06-08 op exit(0);
431 3e4749f7 2020-10-02 op }