2 ce1bfad9 2024-03-30 thomas * Copyright (c) 2024 Stefan Sperling <stsp@openbsd.org>
4 ce1bfad9 2024-03-30 thomas * Permission to use, copy, modify, and distribute this software for any
5 ce1bfad9 2024-03-30 thomas * purpose with or without fee is hereby granted, provided that the above
6 ce1bfad9 2024-03-30 thomas * copyright notice and this permission notice appear in all copies.
8 ce1bfad9 2024-03-30 thomas * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 ce1bfad9 2024-03-30 thomas * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 ce1bfad9 2024-03-30 thomas * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 ce1bfad9 2024-03-30 thomas * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 ce1bfad9 2024-03-30 thomas * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 ce1bfad9 2024-03-30 thomas * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 ce1bfad9 2024-03-30 thomas * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 b2ce1dae 2024-03-30 thomas #include "got_compat.h"
19 ce1bfad9 2024-03-30 thomas #include <sys/types.h>
20 ce1bfad9 2024-03-30 thomas #include <sys/queue.h>
21 ce1bfad9 2024-03-30 thomas #include <sys/socket.h>
22 ce1bfad9 2024-03-30 thomas #include <sys/wait.h>
24 ce1bfad9 2024-03-30 thomas #include <errno.h>
25 ce1bfad9 2024-03-30 thomas #include <event.h>
26 ce1bfad9 2024-03-30 thomas #include <limits.h>
27 ce1bfad9 2024-03-30 thomas #include <signal.h>
28 ce1bfad9 2024-03-30 thomas #include <stdio.h>
29 ce1bfad9 2024-03-30 thomas #include <stdlib.h>
30 ce1bfad9 2024-03-30 thomas #include <string.h>
31 ce1bfad9 2024-03-30 thomas #include <imsg.h>
32 ce1bfad9 2024-03-30 thomas #include <unistd.h>
34 ce1bfad9 2024-03-30 thomas #include "got_error.h"
35 ce1bfad9 2024-03-30 thomas #include "got_path.h"
37 ce1bfad9 2024-03-30 thomas #include "gotd.h"
38 ce1bfad9 2024-03-30 thomas #include "log.h"
39 ce1bfad9 2024-03-30 thomas #include "notify.h"
41 ce1bfad9 2024-03-30 thomas #ifndef nitems
42 ce1bfad9 2024-03-30 thomas #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
45 ce1bfad9 2024-03-30 thomas static struct gotd_notify {
46 ce1bfad9 2024-03-30 thomas pid_t pid;
47 ce1bfad9 2024-03-30 thomas const char *title;
48 ce1bfad9 2024-03-30 thomas struct gotd_imsgev parent_iev;
49 ce1bfad9 2024-03-30 thomas struct gotd_repolist *repos;
50 ce1bfad9 2024-03-30 thomas const char *default_sender;
51 ce1bfad9 2024-03-30 thomas } gotd_notify;
53 ce1bfad9 2024-03-30 thomas struct gotd_notify_session {
54 ce1bfad9 2024-03-30 thomas STAILQ_ENTRY(gotd_notify_session) entry;
55 ce1bfad9 2024-03-30 thomas uint32_t id;
56 ce1bfad9 2024-03-30 thomas struct gotd_imsgev iev;
58 ce1bfad9 2024-03-30 thomas STAILQ_HEAD(gotd_notify_sessions, gotd_notify_session);
60 ce1bfad9 2024-03-30 thomas static struct gotd_notify_sessions gotd_notify_sessions[GOTD_CLIENT_TABLE_SIZE];
61 ce1bfad9 2024-03-30 thomas static SIPHASH_KEY sessions_hash_key;
63 ce1bfad9 2024-03-30 thomas static void gotd_notify_shutdown(void);
65 ce1bfad9 2024-03-30 thomas static uint64_t
66 ce1bfad9 2024-03-30 thomas session_hash(uint32_t session_id)
68 ce1bfad9 2024-03-30 thomas return SipHash24(&sessions_hash_key, &session_id, sizeof(session_id));
71 ce1bfad9 2024-03-30 thomas static void
72 ce1bfad9 2024-03-30 thomas add_session(struct gotd_notify_session *session)
74 ce1bfad9 2024-03-30 thomas uint64_t slot;
76 ce1bfad9 2024-03-30 thomas slot = session_hash(session->id) % nitems(gotd_notify_sessions);
77 ce1bfad9 2024-03-30 thomas STAILQ_INSERT_HEAD(&gotd_notify_sessions[slot], session, entry);
80 ce1bfad9 2024-03-30 thomas static struct gotd_notify_session *
81 ce1bfad9 2024-03-30 thomas find_session(uint32_t session_id)
83 ce1bfad9 2024-03-30 thomas uint64_t slot;
84 ce1bfad9 2024-03-30 thomas struct gotd_notify_session *s;
86 ce1bfad9 2024-03-30 thomas slot = session_hash(session_id) % nitems(gotd_notify_sessions);
87 ce1bfad9 2024-03-30 thomas STAILQ_FOREACH(s, &gotd_notify_sessions[slot], entry) {
88 ce1bfad9 2024-03-30 thomas if (s->id == session_id)
92 ce1bfad9 2024-03-30 thomas return NULL;
95 ce1bfad9 2024-03-30 thomas static struct gotd_notify_session *
96 ce1bfad9 2024-03-30 thomas find_session_by_fd(int fd)
98 ce1bfad9 2024-03-30 thomas uint64_t slot;
99 ce1bfad9 2024-03-30 thomas struct gotd_notify_session *s;
101 ce1bfad9 2024-03-30 thomas for (slot = 0; slot < nitems(gotd_notify_sessions); slot++) {
102 ce1bfad9 2024-03-30 thomas STAILQ_FOREACH(s, &gotd_notify_sessions[slot], entry) {
103 ce1bfad9 2024-03-30 thomas if (s->iev.ibuf.fd == fd)
104 ce1bfad9 2024-03-30 thomas return s;
108 ce1bfad9 2024-03-30 thomas return NULL;
111 ce1bfad9 2024-03-30 thomas static void
112 ce1bfad9 2024-03-30 thomas remove_session(struct gotd_notify_session *session)
114 ce1bfad9 2024-03-30 thomas uint64_t slot;
116 ce1bfad9 2024-03-30 thomas slot = session_hash(session->id) % nitems(gotd_notify_sessions);
117 ce1bfad9 2024-03-30 thomas STAILQ_REMOVE(&gotd_notify_sessions[slot], session,
118 ce1bfad9 2024-03-30 thomas gotd_notify_session, entry);
119 ce1bfad9 2024-03-30 thomas free(session);
122 ce1bfad9 2024-03-30 thomas static uint32_t
123 ce1bfad9 2024-03-30 thomas get_session_id(void)
125 ce1bfad9 2024-03-30 thomas int duplicate = 0;
126 ce1bfad9 2024-03-30 thomas uint32_t id;
129 ce1bfad9 2024-03-30 thomas id = arc4random();
130 ce1bfad9 2024-03-30 thomas duplicate = (find_session(id) != NULL);
131 ce1bfad9 2024-03-30 thomas } while (duplicate || id == 0);
133 ce1bfad9 2024-03-30 thomas return id;
136 ce1bfad9 2024-03-30 thomas static void
137 ce1bfad9 2024-03-30 thomas gotd_notify_sighdlr(int sig, short event, void *arg)
140 ce1bfad9 2024-03-30 thomas * Normal signal handler rules don't apply because libevent
141 ce1bfad9 2024-03-30 thomas * decouples for us.
144 ce1bfad9 2024-03-30 thomas switch (sig) {
145 ce1bfad9 2024-03-30 thomas case SIGHUP:
146 ce1bfad9 2024-03-30 thomas log_info("%s: ignoring SIGHUP", __func__);
148 ce1bfad9 2024-03-30 thomas case SIGUSR1:
149 ce1bfad9 2024-03-30 thomas log_info("%s: ignoring SIGUSR1", __func__);
151 ce1bfad9 2024-03-30 thomas case SIGTERM:
152 ce1bfad9 2024-03-30 thomas case SIGINT:
153 ce1bfad9 2024-03-30 thomas gotd_notify_shutdown();
154 ce1bfad9 2024-03-30 thomas /* NOTREACHED */
157 ce1bfad9 2024-03-30 thomas fatalx("unexpected signal");
161 ce1bfad9 2024-03-30 thomas static void
162 f0b0cfba 2024-04-25 thomas.ad run_notification_helper(const char *prog, const char **argv, int fd,
163 f0b0cfba 2024-04-25 thomas.ad const char *user, const char *pass)
165 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
166 ce1bfad9 2024-03-30 thomas pid_t pid;
167 ce1bfad9 2024-03-30 thomas int child_status;
169 ce1bfad9 2024-03-30 thomas pid = fork();
170 ce1bfad9 2024-03-30 thomas if (pid == -1) {
171 ce1bfad9 2024-03-30 thomas err = got_error_from_errno("fork");
172 ce1bfad9 2024-03-30 thomas log_warn("%s", err->msg);
174 ce1bfad9 2024-03-30 thomas } else if (pid == 0) {
175 ce1bfad9 2024-03-30 thomas signal(SIGQUIT, SIG_DFL);
176 ce1bfad9 2024-03-30 thomas signal(SIGINT, SIG_DFL);
177 ce1bfad9 2024-03-30 thomas signal(SIGCHLD, SIG_DFL);
179 ce1bfad9 2024-03-30 thomas if (dup2(fd, STDIN_FILENO) == -1) {
180 ce1bfad9 2024-03-30 thomas fprintf(stderr, "%s: dup2: %s\n", getprogname(),
181 ce1bfad9 2024-03-30 thomas strerror(errno));
182 ce1bfad9 2024-03-30 thomas _exit(1);
185 ce1bfad9 2024-03-30 thomas closefrom(STDERR_FILENO + 1);
187 f0b0cfba 2024-04-25 thomas.ad if (user != NULL && pass != NULL) {
188 f0b0cfba 2024-04-25 thomas.ad setenv("GOT_NOTIFY_HTTP_USER", user, 1);
189 f0b0cfba 2024-04-25 thomas.ad setenv("GOT_NOTIFY_HTTP_PASS", pass, 1);
192 ce1bfad9 2024-03-30 thomas if (execv(prog, (char *const *)argv) == -1) {
193 ce1bfad9 2024-03-30 thomas fprintf(stderr, "%s: exec %s: %s\n", getprogname(),
194 ce1bfad9 2024-03-30 thomas prog, strerror(errno));
195 ce1bfad9 2024-03-30 thomas _exit(1);
198 ce1bfad9 2024-03-30 thomas /* not reached */
201 ce1bfad9 2024-03-30 thomas if (waitpid(pid, &child_status, 0) == -1) {
202 ce1bfad9 2024-03-30 thomas err = got_error_from_errno("waitpid");
203 ce1bfad9 2024-03-30 thomas goto done;
206 ce1bfad9 2024-03-30 thomas if (!WIFEXITED(child_status)) {
207 ce1bfad9 2024-03-30 thomas err = got_error(GOT_ERR_PRIVSEP_DIED);
208 ce1bfad9 2024-03-30 thomas goto done;
211 ce1bfad9 2024-03-30 thomas if (WEXITSTATUS(child_status) != 0)
212 ce1bfad9 2024-03-30 thomas err = got_error(GOT_ERR_PRIVSEP_EXIT);
215 ce1bfad9 2024-03-30 thomas log_warnx("%s: child %s pid %d: %s", gotd_notify.title,
216 ce1bfad9 2024-03-30 thomas prog, pid, err->msg);
219 ce1bfad9 2024-03-30 thomas static void
220 ce1bfad9 2024-03-30 thomas notify_email(struct gotd_notification_target *target, const char *subject_line,
223 ce1bfad9 2024-03-30 thomas const char *argv[13];
224 ce1bfad9 2024-03-30 thomas int i = 0;
226 ce1bfad9 2024-03-30 thomas argv[i++] = GOTD_PATH_PROG_NOTIFY_EMAIL;
228 ce1bfad9 2024-03-30 thomas argv[i++] = "-f";
229 ce1bfad9 2024-03-30 thomas if (target->conf.email.sender)
230 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.sender;
232 ce1bfad9 2024-03-30 thomas argv[i++] = gotd_notify.default_sender;
234 ce1bfad9 2024-03-30 thomas if (target->conf.email.responder) {
235 ce1bfad9 2024-03-30 thomas argv[i++] = "-r";
236 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.responder;
239 ce1bfad9 2024-03-30 thomas if (target->conf.email.hostname) {
240 ce1bfad9 2024-03-30 thomas argv[i++] = "-h";
241 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.hostname;
244 ce1bfad9 2024-03-30 thomas if (target->conf.email.port) {
245 ce1bfad9 2024-03-30 thomas argv[i++] = "-p";
246 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.port;
249 ce1bfad9 2024-03-30 thomas argv[i++] = "-s";
250 ce1bfad9 2024-03-30 thomas argv[i++] = subject_line;
252 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.recipient;
254 ce1bfad9 2024-03-30 thomas argv[i] = NULL;
256 f0b0cfba 2024-04-25 thomas.ad run_notification_helper(GOTD_PATH_PROG_NOTIFY_EMAIL, argv, fd,
257 f0b0cfba 2024-04-25 thomas.ad NULL, NULL);
260 ce1bfad9 2024-03-30 thomas static void
261 23022bc0 2024-04-25 thomas.ad notify_http(struct gotd_notification_target *target, const char *repo, int fd)
263 23022bc0 2024-04-25 thomas.ad const char *argv[10];
264 94a3f4e9 2024-03-30 thomas int argc = 0;
266 94a3f4e9 2024-03-30 thomas argv[argc++] = GOTD_PATH_PROG_NOTIFY_HTTP;
267 94a3f4e9 2024-03-30 thomas if (target->conf.http.tls)
268 94a3f4e9 2024-03-30 thomas argv[argc++] = "-c";
270 23022bc0 2024-04-25 thomas.ad argv[argc++] = "-r";
271 23022bc0 2024-04-25 thomas.ad argv[argc++] = repo;
272 94a3f4e9 2024-03-30 thomas argv[argc++] = "-h";
273 94a3f4e9 2024-03-30 thomas argv[argc++] = target->conf.http.hostname;
274 94a3f4e9 2024-03-30 thomas argv[argc++] = "-p";
275 94a3f4e9 2024-03-30 thomas argv[argc++] = target->conf.http.port;
277 94a3f4e9 2024-03-30 thomas argv[argc++] = target->conf.http.path;
279 94a3f4e9 2024-03-30 thomas argv[argc] = NULL;
281 f0b0cfba 2024-04-25 thomas.ad run_notification_helper(GOTD_PATH_PROG_NOTIFY_HTTP, argv, fd,
282 f0b0cfba 2024-04-25 thomas.ad target->conf.http.user, target->conf.http.password);
285 ce1bfad9 2024-03-30 thomas static const struct got_error *
286 ce1bfad9 2024-03-30 thomas send_notification(struct imsg *imsg, struct gotd_imsgev *iev)
288 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
289 ce1bfad9 2024-03-30 thomas struct gotd_imsg_notify inotify;
290 ce1bfad9 2024-03-30 thomas size_t datalen;
291 ce1bfad9 2024-03-30 thomas struct gotd_repo *repo;
292 ce1bfad9 2024-03-30 thomas struct gotd_notification_target *target;
295 ce1bfad9 2024-03-30 thomas datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
296 ce1bfad9 2024-03-30 thomas if (datalen != sizeof(inotify))
297 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_LEN);
299 ce1bfad9 2024-03-30 thomas memcpy(&inotify, imsg->data, datalen);
301 ce1bfad9 2024-03-30 thomas repo = gotd_find_repo_by_name(inotify.repo_name, gotd_notify.repos);
302 ce1bfad9 2024-03-30 thomas if (repo == NULL)
303 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_MSG);
305 ce1bfad9 2024-03-30 thomas fd = imsg_get_fd(imsg);
306 ce1bfad9 2024-03-30 thomas if (fd == -1)
307 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_NO_FD);
309 ce1bfad9 2024-03-30 thomas if (lseek(fd, 0, SEEK_SET) == -1) {
310 ce1bfad9 2024-03-30 thomas err = got_error_from_errno("lseek");
311 ce1bfad9 2024-03-30 thomas goto done;
314 ce1bfad9 2024-03-30 thomas STAILQ_FOREACH(target, &repo->notification_targets, entry) {
315 ce1bfad9 2024-03-30 thomas switch (target->type) {
316 ce1bfad9 2024-03-30 thomas case GOTD_NOTIFICATION_VIA_EMAIL:
317 ce1bfad9 2024-03-30 thomas notify_email(target, inotify.subject_line, fd);
319 ce1bfad9 2024-03-30 thomas case GOTD_NOTIFICATION_VIA_HTTP:
320 23022bc0 2024-04-25 thomas.ad notify_http(target, repo->name, fd);
325 ce1bfad9 2024-03-30 thomas if (gotd_imsg_compose_event(iev, GOTD_IMSG_NOTIFICATION_SENT,
326 ce1bfad9 2024-03-30 thomas PROC_NOTIFY, -1, NULL, 0) == -1) {
327 ce1bfad9 2024-03-30 thomas err = got_error_from_errno("imsg compose NOTIFY");
328 ce1bfad9 2024-03-30 thomas goto done;
331 ce1bfad9 2024-03-30 thomas close(fd);
332 ce1bfad9 2024-03-30 thomas return err;
335 ce1bfad9 2024-03-30 thomas static void
336 ce1bfad9 2024-03-30 thomas notify_dispatch_session(int fd, short event, void *arg)
338 ce1bfad9 2024-03-30 thomas struct gotd_imsgev *iev = arg;
339 ce1bfad9 2024-03-30 thomas struct imsgbuf *ibuf = &iev->ibuf;
340 ce1bfad9 2024-03-30 thomas ssize_t n;
341 ce1bfad9 2024-03-30 thomas int shut = 0;
342 ce1bfad9 2024-03-30 thomas struct imsg imsg;
344 ce1bfad9 2024-03-30 thomas if (event & EV_READ) {
345 ce1bfad9 2024-03-30 thomas if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
346 ce1bfad9 2024-03-30 thomas fatal("imsg_read error");
347 ce1bfad9 2024-03-30 thomas if (n == 0) {
348 ce1bfad9 2024-03-30 thomas /* Connection closed. */
349 ce1bfad9 2024-03-30 thomas shut = 1;
350 ce1bfad9 2024-03-30 thomas goto done;
354 ce1bfad9 2024-03-30 thomas if (event & EV_WRITE) {
355 ce1bfad9 2024-03-30 thomas n = msgbuf_write(&ibuf->w);
356 7ede118f 2024-04-29 thomas if (n == -1 && errno != EAGAIN && errno != EPIPE)
357 ce1bfad9 2024-03-30 thomas fatal("msgbuf_write");
358 7ede118f 2024-04-29 thomas if (n == 0 || (n == -1 && errno == EPIPE)) {
359 ce1bfad9 2024-03-30 thomas /* Connection closed. */
360 ce1bfad9 2024-03-30 thomas shut = 1;
361 ce1bfad9 2024-03-30 thomas goto done;
365 ce1bfad9 2024-03-30 thomas for (;;) {
366 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
368 ce1bfad9 2024-03-30 thomas if ((n = imsg_get(ibuf, &imsg)) == -1)
369 ce1bfad9 2024-03-30 thomas fatal("%s: imsg_get error", __func__);
370 ce1bfad9 2024-03-30 thomas if (n == 0) /* No more messages. */
373 ce1bfad9 2024-03-30 thomas switch (imsg.hdr.type) {
374 ce1bfad9 2024-03-30 thomas case GOTD_IMSG_NOTIFY:
375 ce1bfad9 2024-03-30 thomas err = send_notification(&imsg, iev);
378 ce1bfad9 2024-03-30 thomas log_debug("unexpected imsg %d", imsg.hdr.type);
381 ce1bfad9 2024-03-30 thomas imsg_free(&imsg);
384 ce1bfad9 2024-03-30 thomas log_warnx("%s: %s", __func__, err->msg);
387 ce1bfad9 2024-03-30 thomas if (!shut) {
388 ce1bfad9 2024-03-30 thomas gotd_imsg_event_add(iev);
390 ce1bfad9 2024-03-30 thomas struct gotd_notify_session *session;
392 ce1bfad9 2024-03-30 thomas /* This pipe is dead. Remove its event handler */
393 ce1bfad9 2024-03-30 thomas event_del(&iev->ev);
394 ce1bfad9 2024-03-30 thomas imsg_clear(&iev->ibuf);
396 ce1bfad9 2024-03-30 thomas session = find_session_by_fd(fd);
397 ce1bfad9 2024-03-30 thomas if (session)
398 ce1bfad9 2024-03-30 thomas remove_session(session);
402 ce1bfad9 2024-03-30 thomas static const struct got_error *
403 ce1bfad9 2024-03-30 thomas recv_session(struct imsg *imsg)
405 ce1bfad9 2024-03-30 thomas struct gotd_notify_session *session;
406 ce1bfad9 2024-03-30 thomas size_t datalen;
409 ce1bfad9 2024-03-30 thomas datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
410 ce1bfad9 2024-03-30 thomas if (datalen != 0)
411 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_LEN);
413 ce1bfad9 2024-03-30 thomas fd = imsg_get_fd(imsg);
414 ce1bfad9 2024-03-30 thomas if (fd == -1)
415 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_NO_FD);
417 ce1bfad9 2024-03-30 thomas session = calloc(1, sizeof(*session));
418 ce1bfad9 2024-03-30 thomas if (session == NULL)
419 ce1bfad9 2024-03-30 thomas return got_error_from_errno("calloc");
421 ce1bfad9 2024-03-30 thomas session->id = get_session_id();
422 ce1bfad9 2024-03-30 thomas imsg_init(&session->iev.ibuf, fd);
423 ce1bfad9 2024-03-30 thomas session->iev.handler = notify_dispatch_session;
424 ce1bfad9 2024-03-30 thomas session->iev.events = EV_READ;
425 ce1bfad9 2024-03-30 thomas session->iev.handler_arg = NULL;
426 ce1bfad9 2024-03-30 thomas event_set(&session->iev.ev, session->iev.ibuf.fd, EV_READ,
427 ce1bfad9 2024-03-30 thomas notify_dispatch_session, &session->iev);
428 ce1bfad9 2024-03-30 thomas gotd_imsg_event_add(&session->iev);
429 ce1bfad9 2024-03-30 thomas add_session(session);
431 ce1bfad9 2024-03-30 thomas return NULL;
434 ce1bfad9 2024-03-30 thomas static void
435 ce1bfad9 2024-03-30 thomas notify_dispatch(int fd, short event, void *arg)
437 ce1bfad9 2024-03-30 thomas struct gotd_imsgev *iev = arg;
438 ce1bfad9 2024-03-30 thomas struct imsgbuf *ibuf = &iev->ibuf;
439 ce1bfad9 2024-03-30 thomas ssize_t n;
440 ce1bfad9 2024-03-30 thomas int shut = 0;
441 ce1bfad9 2024-03-30 thomas struct imsg imsg;
443 ce1bfad9 2024-03-30 thomas if (event & EV_READ) {
444 ce1bfad9 2024-03-30 thomas if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
445 ce1bfad9 2024-03-30 thomas fatal("imsg_read error");
446 ce1bfad9 2024-03-30 thomas if (n == 0) {
447 ce1bfad9 2024-03-30 thomas /* Connection closed. */
448 ce1bfad9 2024-03-30 thomas shut = 1;
449 ce1bfad9 2024-03-30 thomas goto done;
453 ce1bfad9 2024-03-30 thomas if (event & EV_WRITE) {
454 ce1bfad9 2024-03-30 thomas n = msgbuf_write(&ibuf->w);
455 6d3ee6a5 2024-04-29 thomas if (n == -1 && errno != EAGAIN)
456 ce1bfad9 2024-03-30 thomas fatal("msgbuf_write");
457 6d3ee6a5 2024-04-29 thomas if (n == 0) {
458 ce1bfad9 2024-03-30 thomas /* Connection closed. */
459 ce1bfad9 2024-03-30 thomas shut = 1;
460 ce1bfad9 2024-03-30 thomas goto done;
464 ce1bfad9 2024-03-30 thomas for (;;) {
465 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
467 ce1bfad9 2024-03-30 thomas if ((n = imsg_get(ibuf, &imsg)) == -1)
468 ce1bfad9 2024-03-30 thomas fatal("%s: imsg_get error", __func__);
469 ce1bfad9 2024-03-30 thomas if (n == 0) /* No more messages. */
472 ce1bfad9 2024-03-30 thomas switch (imsg.hdr.type) {
473 ce1bfad9 2024-03-30 thomas case GOTD_IMSG_CONNECT_SESSION:
474 ce1bfad9 2024-03-30 thomas err = recv_session(&imsg);
477 ce1bfad9 2024-03-30 thomas log_debug("unexpected imsg %d", imsg.hdr.type);
480 ce1bfad9 2024-03-30 thomas imsg_free(&imsg);
483 ce1bfad9 2024-03-30 thomas log_warnx("%s: %s", __func__, err->msg);
486 ce1bfad9 2024-03-30 thomas if (!shut) {
487 ce1bfad9 2024-03-30 thomas gotd_imsg_event_add(iev);
489 ce1bfad9 2024-03-30 thomas /* This pipe is dead. Remove its event handler */
490 ce1bfad9 2024-03-30 thomas event_del(&iev->ev);
491 ce1bfad9 2024-03-30 thomas event_loopexit(NULL);
497 ce1bfad9 2024-03-30 thomas notify_main(const char *title, struct gotd_repolist *repos,
498 ce1bfad9 2024-03-30 thomas const char *default_sender)
500 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
501 ce1bfad9 2024-03-30 thomas struct event evsigint, evsigterm, evsighup, evsigusr1;
503 ce1bfad9 2024-03-30 thomas arc4random_buf(&sessions_hash_key, sizeof(sessions_hash_key));
505 ce1bfad9 2024-03-30 thomas gotd_notify.title = title;
506 ce1bfad9 2024-03-30 thomas gotd_notify.repos = repos;
507 ce1bfad9 2024-03-30 thomas gotd_notify.default_sender = default_sender;
508 ce1bfad9 2024-03-30 thomas gotd_notify.pid = getpid();
510 ce1bfad9 2024-03-30 thomas signal_set(&evsigint, SIGINT, gotd_notify_sighdlr, NULL);
511 ce1bfad9 2024-03-30 thomas signal_set(&evsigterm, SIGTERM, gotd_notify_sighdlr, NULL);
512 ce1bfad9 2024-03-30 thomas signal_set(&evsighup, SIGHUP, gotd_notify_sighdlr, NULL);
513 ce1bfad9 2024-03-30 thomas signal_set(&evsigusr1, SIGUSR1, gotd_notify_sighdlr, NULL);
514 ce1bfad9 2024-03-30 thomas signal(SIGPIPE, SIG_IGN);
516 ce1bfad9 2024-03-30 thomas signal_add(&evsigint, NULL);
517 ce1bfad9 2024-03-30 thomas signal_add(&evsigterm, NULL);
518 ce1bfad9 2024-03-30 thomas signal_add(&evsighup, NULL);
519 ce1bfad9 2024-03-30 thomas signal_add(&evsigusr1, NULL);
521 ce1bfad9 2024-03-30 thomas imsg_init(&gotd_notify.parent_iev.ibuf, GOTD_FILENO_MSG_PIPE);
522 ce1bfad9 2024-03-30 thomas gotd_notify.parent_iev.handler = notify_dispatch;
523 ce1bfad9 2024-03-30 thomas gotd_notify.parent_iev.events = EV_READ;
524 ce1bfad9 2024-03-30 thomas gotd_notify.parent_iev.handler_arg = NULL;
525 ce1bfad9 2024-03-30 thomas event_set(&gotd_notify.parent_iev.ev, gotd_notify.parent_iev.ibuf.fd,
526 ce1bfad9 2024-03-30 thomas EV_READ, notify_dispatch, &gotd_notify.parent_iev);
527 ce1bfad9 2024-03-30 thomas gotd_imsg_event_add(&gotd_notify.parent_iev);
529 ce1bfad9 2024-03-30 thomas event_dispatch();
532 ce1bfad9 2024-03-30 thomas log_warnx("%s: %s", title, err->msg);
533 ce1bfad9 2024-03-30 thomas gotd_notify_shutdown();
537 ce1bfad9 2024-03-30 thomas gotd_notify_shutdown(void)
539 b1a47061 2024-03-30 thomas log_debug("%s: shutting down", gotd_notify.title);