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 ce1bfad9 2024-03-30 thomas run_notification_helper(const char *prog, const char **argv, int fd)
164 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
165 ce1bfad9 2024-03-30 thomas pid_t pid;
166 ce1bfad9 2024-03-30 thomas int child_status;
168 ce1bfad9 2024-03-30 thomas pid = fork();
169 ce1bfad9 2024-03-30 thomas if (pid == -1) {
170 ce1bfad9 2024-03-30 thomas err = got_error_from_errno("fork");
171 ce1bfad9 2024-03-30 thomas log_warn("%s", err->msg);
173 ce1bfad9 2024-03-30 thomas } else if (pid == 0) {
174 ce1bfad9 2024-03-30 thomas signal(SIGQUIT, SIG_DFL);
175 ce1bfad9 2024-03-30 thomas signal(SIGINT, SIG_DFL);
176 ce1bfad9 2024-03-30 thomas signal(SIGCHLD, SIG_DFL);
178 ce1bfad9 2024-03-30 thomas if (dup2(fd, STDIN_FILENO) == -1) {
179 ce1bfad9 2024-03-30 thomas fprintf(stderr, "%s: dup2: %s\n", getprogname(),
180 ce1bfad9 2024-03-30 thomas strerror(errno));
181 ce1bfad9 2024-03-30 thomas _exit(1);
184 ce1bfad9 2024-03-30 thomas closefrom(STDERR_FILENO + 1);
186 ce1bfad9 2024-03-30 thomas if (execv(prog, (char *const *)argv) == -1) {
187 ce1bfad9 2024-03-30 thomas fprintf(stderr, "%s: exec %s: %s\n", getprogname(),
188 ce1bfad9 2024-03-30 thomas prog, strerror(errno));
189 ce1bfad9 2024-03-30 thomas _exit(1);
192 ce1bfad9 2024-03-30 thomas /* not reached */
195 ce1bfad9 2024-03-30 thomas if (waitpid(pid, &child_status, 0) == -1) {
196 ce1bfad9 2024-03-30 thomas err = got_error_from_errno("waitpid");
197 ce1bfad9 2024-03-30 thomas goto done;
200 ce1bfad9 2024-03-30 thomas if (!WIFEXITED(child_status)) {
201 ce1bfad9 2024-03-30 thomas err = got_error(GOT_ERR_PRIVSEP_DIED);
202 ce1bfad9 2024-03-30 thomas goto done;
205 ce1bfad9 2024-03-30 thomas if (WEXITSTATUS(child_status) != 0)
206 ce1bfad9 2024-03-30 thomas err = got_error(GOT_ERR_PRIVSEP_EXIT);
209 ce1bfad9 2024-03-30 thomas log_warnx("%s: child %s pid %d: %s", gotd_notify.title,
210 ce1bfad9 2024-03-30 thomas prog, pid, err->msg);
213 ce1bfad9 2024-03-30 thomas static void
214 ce1bfad9 2024-03-30 thomas notify_email(struct gotd_notification_target *target, const char *subject_line,
217 ce1bfad9 2024-03-30 thomas const char *argv[13];
218 ce1bfad9 2024-03-30 thomas int i = 0;
220 ce1bfad9 2024-03-30 thomas argv[i++] = GOTD_PATH_PROG_NOTIFY_EMAIL;
222 ce1bfad9 2024-03-30 thomas argv[i++] = "-f";
223 ce1bfad9 2024-03-30 thomas if (target->conf.email.sender)
224 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.sender;
226 ce1bfad9 2024-03-30 thomas argv[i++] = gotd_notify.default_sender;
228 ce1bfad9 2024-03-30 thomas if (target->conf.email.responder) {
229 ce1bfad9 2024-03-30 thomas argv[i++] = "-r";
230 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.responder;
233 ce1bfad9 2024-03-30 thomas if (target->conf.email.hostname) {
234 ce1bfad9 2024-03-30 thomas argv[i++] = "-h";
235 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.hostname;
238 ce1bfad9 2024-03-30 thomas if (target->conf.email.port) {
239 ce1bfad9 2024-03-30 thomas argv[i++] = "-p";
240 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.port;
243 ce1bfad9 2024-03-30 thomas argv[i++] = "-s";
244 ce1bfad9 2024-03-30 thomas argv[i++] = subject_line;
246 ce1bfad9 2024-03-30 thomas argv[i++] = target->conf.email.recipient;
248 ce1bfad9 2024-03-30 thomas argv[i] = NULL;
250 ce1bfad9 2024-03-30 thomas run_notification_helper(GOTD_PATH_PROG_NOTIFY_EMAIL, argv, fd);
253 ce1bfad9 2024-03-30 thomas static void
254 94a3f4e9 2024-03-30 thomas notify_http(struct gotd_notification_target *target, int fd)
256 94a3f4e9 2024-03-30 thomas const char *argv[8];
257 94a3f4e9 2024-03-30 thomas int argc = 0;
259 94a3f4e9 2024-03-30 thomas argv[argc++] = GOTD_PATH_PROG_NOTIFY_HTTP;
260 94a3f4e9 2024-03-30 thomas if (target->conf.http.tls)
261 94a3f4e9 2024-03-30 thomas argv[argc++] = "-c";
263 94a3f4e9 2024-03-30 thomas argv[argc++] = "-h";
264 94a3f4e9 2024-03-30 thomas argv[argc++] = target->conf.http.hostname;
265 94a3f4e9 2024-03-30 thomas argv[argc++] = "-p";
266 94a3f4e9 2024-03-30 thomas argv[argc++] = target->conf.http.port;
268 94a3f4e9 2024-03-30 thomas argv[argc++] = target->conf.http.path;
270 94a3f4e9 2024-03-30 thomas argv[argc] = NULL;
272 ce1bfad9 2024-03-30 thomas run_notification_helper(GOTD_PATH_PROG_NOTIFY_HTTP, argv, fd);
275 ce1bfad9 2024-03-30 thomas static const struct got_error *
276 ce1bfad9 2024-03-30 thomas send_notification(struct imsg *imsg, struct gotd_imsgev *iev)
278 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
279 ce1bfad9 2024-03-30 thomas struct gotd_imsg_notify inotify;
280 ce1bfad9 2024-03-30 thomas size_t datalen;
281 ce1bfad9 2024-03-30 thomas struct gotd_repo *repo;
282 ce1bfad9 2024-03-30 thomas struct gotd_notification_target *target;
285 ce1bfad9 2024-03-30 thomas datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
286 ce1bfad9 2024-03-30 thomas if (datalen != sizeof(inotify))
287 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_LEN);
289 ce1bfad9 2024-03-30 thomas memcpy(&inotify, imsg->data, datalen);
291 ce1bfad9 2024-03-30 thomas repo = gotd_find_repo_by_name(inotify.repo_name, gotd_notify.repos);
292 ce1bfad9 2024-03-30 thomas if (repo == NULL)
293 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_MSG);
295 ce1bfad9 2024-03-30 thomas fd = imsg_get_fd(imsg);
296 ce1bfad9 2024-03-30 thomas if (fd == -1)
297 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_NO_FD);
299 ce1bfad9 2024-03-30 thomas if (lseek(fd, 0, SEEK_SET) == -1) {
300 ce1bfad9 2024-03-30 thomas err = got_error_from_errno("lseek");
301 ce1bfad9 2024-03-30 thomas goto done;
304 ce1bfad9 2024-03-30 thomas STAILQ_FOREACH(target, &repo->notification_targets, entry) {
305 ce1bfad9 2024-03-30 thomas switch (target->type) {
306 ce1bfad9 2024-03-30 thomas case GOTD_NOTIFICATION_VIA_EMAIL:
307 ce1bfad9 2024-03-30 thomas notify_email(target, inotify.subject_line, fd);
309 ce1bfad9 2024-03-30 thomas case GOTD_NOTIFICATION_VIA_HTTP:
310 94a3f4e9 2024-03-30 thomas notify_http(target, fd);
315 ce1bfad9 2024-03-30 thomas if (gotd_imsg_compose_event(iev, GOTD_IMSG_NOTIFICATION_SENT,
316 ce1bfad9 2024-03-30 thomas PROC_NOTIFY, -1, NULL, 0) == -1) {
317 ce1bfad9 2024-03-30 thomas err = got_error_from_errno("imsg compose NOTIFY");
318 ce1bfad9 2024-03-30 thomas goto done;
321 ce1bfad9 2024-03-30 thomas close(fd);
322 ce1bfad9 2024-03-30 thomas return err;
325 ce1bfad9 2024-03-30 thomas static void
326 ce1bfad9 2024-03-30 thomas notify_dispatch_session(int fd, short event, void *arg)
328 ce1bfad9 2024-03-30 thomas struct gotd_imsgev *iev = arg;
329 ce1bfad9 2024-03-30 thomas struct imsgbuf *ibuf = &iev->ibuf;
330 ce1bfad9 2024-03-30 thomas ssize_t n;
331 ce1bfad9 2024-03-30 thomas int shut = 0;
332 ce1bfad9 2024-03-30 thomas struct imsg imsg;
334 ce1bfad9 2024-03-30 thomas if (event & EV_READ) {
335 ce1bfad9 2024-03-30 thomas if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
336 ce1bfad9 2024-03-30 thomas fatal("imsg_read error");
337 ce1bfad9 2024-03-30 thomas if (n == 0) {
338 ce1bfad9 2024-03-30 thomas /* Connection closed. */
339 ce1bfad9 2024-03-30 thomas shut = 1;
340 ce1bfad9 2024-03-30 thomas goto done;
344 ce1bfad9 2024-03-30 thomas if (event & EV_WRITE) {
345 ce1bfad9 2024-03-30 thomas n = msgbuf_write(&ibuf->w);
346 ce1bfad9 2024-03-30 thomas if (n == -1 && errno != EAGAIN)
347 ce1bfad9 2024-03-30 thomas fatal("msgbuf_write");
348 ce1bfad9 2024-03-30 thomas if (n == 0) {
349 ce1bfad9 2024-03-30 thomas /* Connection closed. */
350 ce1bfad9 2024-03-30 thomas shut = 1;
351 ce1bfad9 2024-03-30 thomas goto done;
355 ce1bfad9 2024-03-30 thomas for (;;) {
356 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
358 ce1bfad9 2024-03-30 thomas if ((n = imsg_get(ibuf, &imsg)) == -1)
359 ce1bfad9 2024-03-30 thomas fatal("%s: imsg_get error", __func__);
360 ce1bfad9 2024-03-30 thomas if (n == 0) /* No more messages. */
363 ce1bfad9 2024-03-30 thomas switch (imsg.hdr.type) {
364 ce1bfad9 2024-03-30 thomas case GOTD_IMSG_NOTIFY:
365 ce1bfad9 2024-03-30 thomas err = send_notification(&imsg, iev);
368 ce1bfad9 2024-03-30 thomas log_debug("unexpected imsg %d", imsg.hdr.type);
371 ce1bfad9 2024-03-30 thomas imsg_free(&imsg);
374 ce1bfad9 2024-03-30 thomas log_warnx("%s: %s", __func__, err->msg);
377 ce1bfad9 2024-03-30 thomas if (!shut) {
378 ce1bfad9 2024-03-30 thomas gotd_imsg_event_add(iev);
380 ce1bfad9 2024-03-30 thomas struct gotd_notify_session *session;
382 ce1bfad9 2024-03-30 thomas /* This pipe is dead. Remove its event handler */
383 ce1bfad9 2024-03-30 thomas event_del(&iev->ev);
384 ce1bfad9 2024-03-30 thomas imsg_clear(&iev->ibuf);
386 ce1bfad9 2024-03-30 thomas session = find_session_by_fd(fd);
387 ce1bfad9 2024-03-30 thomas if (session)
388 ce1bfad9 2024-03-30 thomas remove_session(session);
392 ce1bfad9 2024-03-30 thomas static const struct got_error *
393 ce1bfad9 2024-03-30 thomas recv_session(struct imsg *imsg)
395 ce1bfad9 2024-03-30 thomas struct gotd_notify_session *session;
396 ce1bfad9 2024-03-30 thomas size_t datalen;
399 ce1bfad9 2024-03-30 thomas datalen = imsg->hdr.len - IMSG_HEADER_SIZE;
400 ce1bfad9 2024-03-30 thomas if (datalen != 0)
401 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_LEN);
403 ce1bfad9 2024-03-30 thomas fd = imsg_get_fd(imsg);
404 ce1bfad9 2024-03-30 thomas if (fd == -1)
405 ce1bfad9 2024-03-30 thomas return got_error(GOT_ERR_PRIVSEP_NO_FD);
407 ce1bfad9 2024-03-30 thomas session = calloc(1, sizeof(*session));
408 ce1bfad9 2024-03-30 thomas if (session == NULL)
409 ce1bfad9 2024-03-30 thomas return got_error_from_errno("calloc");
411 ce1bfad9 2024-03-30 thomas session->id = get_session_id();
412 ce1bfad9 2024-03-30 thomas imsg_init(&session->iev.ibuf, fd);
413 ce1bfad9 2024-03-30 thomas session->iev.handler = notify_dispatch_session;
414 ce1bfad9 2024-03-30 thomas session->iev.events = EV_READ;
415 ce1bfad9 2024-03-30 thomas session->iev.handler_arg = NULL;
416 ce1bfad9 2024-03-30 thomas event_set(&session->iev.ev, session->iev.ibuf.fd, EV_READ,
417 ce1bfad9 2024-03-30 thomas notify_dispatch_session, &session->iev);
418 ce1bfad9 2024-03-30 thomas gotd_imsg_event_add(&session->iev);
419 ce1bfad9 2024-03-30 thomas add_session(session);
421 ce1bfad9 2024-03-30 thomas return NULL;
424 ce1bfad9 2024-03-30 thomas static void
425 ce1bfad9 2024-03-30 thomas notify_dispatch(int fd, short event, void *arg)
427 ce1bfad9 2024-03-30 thomas struct gotd_imsgev *iev = arg;
428 ce1bfad9 2024-03-30 thomas struct imsgbuf *ibuf = &iev->ibuf;
429 ce1bfad9 2024-03-30 thomas ssize_t n;
430 ce1bfad9 2024-03-30 thomas int shut = 0;
431 ce1bfad9 2024-03-30 thomas struct imsg imsg;
433 ce1bfad9 2024-03-30 thomas if (event & EV_READ) {
434 ce1bfad9 2024-03-30 thomas if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
435 ce1bfad9 2024-03-30 thomas fatal("imsg_read error");
436 ce1bfad9 2024-03-30 thomas if (n == 0) {
437 ce1bfad9 2024-03-30 thomas /* Connection closed. */
438 ce1bfad9 2024-03-30 thomas shut = 1;
439 ce1bfad9 2024-03-30 thomas goto done;
443 ce1bfad9 2024-03-30 thomas if (event & EV_WRITE) {
444 ce1bfad9 2024-03-30 thomas n = msgbuf_write(&ibuf->w);
445 ce1bfad9 2024-03-30 thomas if (n == -1 && errno != EAGAIN)
446 ce1bfad9 2024-03-30 thomas fatal("msgbuf_write");
447 ce1bfad9 2024-03-30 thomas if (n == 0) {
448 ce1bfad9 2024-03-30 thomas /* Connection closed. */
449 ce1bfad9 2024-03-30 thomas shut = 1;
450 ce1bfad9 2024-03-30 thomas goto done;
454 ce1bfad9 2024-03-30 thomas for (;;) {
455 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
457 ce1bfad9 2024-03-30 thomas if ((n = imsg_get(ibuf, &imsg)) == -1)
458 ce1bfad9 2024-03-30 thomas fatal("%s: imsg_get error", __func__);
459 ce1bfad9 2024-03-30 thomas if (n == 0) /* No more messages. */
462 ce1bfad9 2024-03-30 thomas switch (imsg.hdr.type) {
463 ce1bfad9 2024-03-30 thomas case GOTD_IMSG_CONNECT_SESSION:
464 ce1bfad9 2024-03-30 thomas err = recv_session(&imsg);
467 ce1bfad9 2024-03-30 thomas log_debug("unexpected imsg %d", imsg.hdr.type);
470 ce1bfad9 2024-03-30 thomas imsg_free(&imsg);
473 ce1bfad9 2024-03-30 thomas log_warnx("%s: %s", __func__, err->msg);
476 ce1bfad9 2024-03-30 thomas if (!shut) {
477 ce1bfad9 2024-03-30 thomas gotd_imsg_event_add(iev);
479 ce1bfad9 2024-03-30 thomas /* This pipe is dead. Remove its event handler */
480 ce1bfad9 2024-03-30 thomas event_del(&iev->ev);
481 ce1bfad9 2024-03-30 thomas event_loopexit(NULL);
487 ce1bfad9 2024-03-30 thomas notify_main(const char *title, struct gotd_repolist *repos,
488 ce1bfad9 2024-03-30 thomas const char *default_sender)
490 ce1bfad9 2024-03-30 thomas const struct got_error *err = NULL;
491 ce1bfad9 2024-03-30 thomas struct event evsigint, evsigterm, evsighup, evsigusr1;
493 ce1bfad9 2024-03-30 thomas arc4random_buf(&sessions_hash_key, sizeof(sessions_hash_key));
495 ce1bfad9 2024-03-30 thomas gotd_notify.title = title;
496 ce1bfad9 2024-03-30 thomas gotd_notify.repos = repos;
497 ce1bfad9 2024-03-30 thomas gotd_notify.default_sender = default_sender;
498 ce1bfad9 2024-03-30 thomas gotd_notify.pid = getpid();
500 ce1bfad9 2024-03-30 thomas signal_set(&evsigint, SIGINT, gotd_notify_sighdlr, NULL);
501 ce1bfad9 2024-03-30 thomas signal_set(&evsigterm, SIGTERM, gotd_notify_sighdlr, NULL);
502 ce1bfad9 2024-03-30 thomas signal_set(&evsighup, SIGHUP, gotd_notify_sighdlr, NULL);
503 ce1bfad9 2024-03-30 thomas signal_set(&evsigusr1, SIGUSR1, gotd_notify_sighdlr, NULL);
504 ce1bfad9 2024-03-30 thomas signal(SIGPIPE, SIG_IGN);
506 ce1bfad9 2024-03-30 thomas signal_add(&evsigint, NULL);
507 ce1bfad9 2024-03-30 thomas signal_add(&evsigterm, NULL);
508 ce1bfad9 2024-03-30 thomas signal_add(&evsighup, NULL);
509 ce1bfad9 2024-03-30 thomas signal_add(&evsigusr1, NULL);
511 ce1bfad9 2024-03-30 thomas imsg_init(&gotd_notify.parent_iev.ibuf, GOTD_FILENO_MSG_PIPE);
512 ce1bfad9 2024-03-30 thomas gotd_notify.parent_iev.handler = notify_dispatch;
513 ce1bfad9 2024-03-30 thomas gotd_notify.parent_iev.events = EV_READ;
514 ce1bfad9 2024-03-30 thomas gotd_notify.parent_iev.handler_arg = NULL;
515 ce1bfad9 2024-03-30 thomas event_set(&gotd_notify.parent_iev.ev, gotd_notify.parent_iev.ibuf.fd,
516 ce1bfad9 2024-03-30 thomas EV_READ, notify_dispatch, &gotd_notify.parent_iev);
517 ce1bfad9 2024-03-30 thomas gotd_imsg_event_add(&gotd_notify.parent_iev);
519 ce1bfad9 2024-03-30 thomas event_dispatch();
522 ce1bfad9 2024-03-30 thomas log_warnx("%s: %s", title, err->msg);
523 ce1bfad9 2024-03-30 thomas gotd_notify_shutdown();
527 ce1bfad9 2024-03-30 thomas gotd_notify_shutdown(void)
529 b1a47061 2024-03-30 thomas log_debug("%s: shutting down", gotd_notify.title);
