From: gilles Date: Sat, 17 Oct 2015 18:00:32 +0000 (+0000) Subject: both of these are deprecated X-Git-Url: http://artulab.com/gitweb/?a=commitdiff_plain;h=598cc9e195b72e9f0aac6209310f546b354c0a33;p=openbsd both of these are deprecated --- diff --git a/usr.sbin/smtpd/queue_api.c b/usr.sbin/smtpd/queue_api.c deleted file mode 100644 index d66bc142691..00000000000 --- a/usr.sbin/smtpd/queue_api.c +++ /dev/null @@ -1,426 +0,0 @@ -/* $OpenBSD: queue_api.c,v 1.8 2015/10/14 21:30:40 gilles Exp $ */ - -/* - * Copyright (c) 2013 Eric Faurot - * - * Permission to use, copy, modify, and distribute this software for any - * purpose with or without fee is hereby granted, provided that the above - * copyright notice and this permission notice appear in all copies. - * - * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES - * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF - * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR - * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES - * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN - * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF - * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. - */ - -#include -#include -#include -#include - -#include -#include - -#include -#include -#include -#include -#include -#include - -#include "smtpd-defines.h" -#include "smtpd-api.h" -#include "log.h" - -static int (*handler_close)(void); -static int (*handler_message_create)(uint32_t *); -static int (*handler_message_commit)(uint32_t, const char *); -static int (*handler_message_delete)(uint32_t); -static int (*handler_message_fd_r)(uint32_t); -static int (*handler_message_corrupt)(uint32_t); -static int (*handler_envelope_create)(uint32_t, const char *, size_t, uint64_t *); -static int (*handler_envelope_delete)(uint64_t); -static int (*handler_envelope_update)(uint64_t, const char *, size_t); -static int (*handler_envelope_load)(uint64_t, char *, size_t); -static int (*handler_envelope_walk)(uint64_t *, char *, size_t); - -static struct imsgbuf ibuf; -static struct imsg imsg; -static size_t rlen; -static char *rdata; -static struct ibuf *buf; -static const char *rootpath = PATH_SPOOL; -static const char *user = SMTPD_QUEUE_USER; - -static void -queue_msg_get(void *dst, size_t len) -{ - if (len > rlen) { - log_warnx("warn: queue-proc: bad msg len"); - fatalx("queue-proc: exiting"); - } - - if (len == 0) - return; - - if (dst) - memmove(dst, rdata, len); - - rlen -= len; - rdata += len; -} - -static void -queue_msg_end(void) -{ - if (rlen) { - log_warnx("warn: queue-proc: bogus data"); - fatalx("queue-proc: exiting"); - } - imsg_free(&imsg); -} - -static void -queue_msg_add(const void *data, size_t len) -{ - if (buf == NULL) - buf = imsg_create(&ibuf, PROC_QUEUE_OK, 0, 0, 1024); - if (buf == NULL) { - log_warnx("warn: queue-api: imsg_create failed"); - fatalx("queue-api: exiting"); - } - if (imsg_add(buf, data, len) == -1) { - log_warnx("warn: queue-api: imsg_add failed"); - fatalx("queue-api: exiting"); - } -} - -static void -queue_msg_close(void) -{ - imsg_close(&ibuf, buf); - buf = NULL; -} - -static void -queue_msg_dispatch(void) -{ - uint64_t evpid; - uint32_t msgid, version; - size_t n, m; - char buffer[8192], path[PATH_MAX]; - int r, fd; - FILE *ifile, *ofile; - - switch (imsg.hdr.type) { - case PROC_QUEUE_INIT: - queue_msg_get(&version, sizeof(version)); - queue_msg_end(); - - if (version != PROC_QUEUE_API_VERSION) { - log_warnx("warn: queue-api: bad API version"); - fatalx("queue-api: exiting"); - } - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, -1, NULL, 0); - break; - - case PROC_QUEUE_CLOSE: - queue_msg_end(); - - if (handler_close) - r = handler_close(); - else - r = 1; - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_QUEUE_MESSAGE_CREATE: - queue_msg_end(); - - r = handler_message_create(&msgid); - - queue_msg_add(&r, sizeof(r)); - if (r == 1) - queue_msg_add(&msgid, sizeof(msgid)); - queue_msg_close(); - break; - - case PROC_QUEUE_MESSAGE_DELETE: - queue_msg_get(&msgid, sizeof(msgid)); - queue_msg_end(); - - r = handler_message_delete(msgid); - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_QUEUE_MESSAGE_COMMIT: - queue_msg_get(&msgid, sizeof(msgid)); - queue_msg_end(); - - /* XXX needs more love */ - r = -1; - (void)snprintf(path, sizeof path, "/tmp/message.XXXXXXXXXX"); - fd = mkstemp(path); - if (fd == -1) { - log_warn("warn: queue-api: mkstemp"); - } - else { - ifile = fdopen(imsg.fd, "r"); - ofile = fdopen(fd, "w"); - m = n = 0; - if (ifile && ofile) { - while (!feof(ifile)) { - n = fread(buffer, 1, sizeof(buffer), - ifile); - m = fwrite(buffer, 1, n, ofile); - if (m != n) - break; - fflush(ofile); - } - if (m != n) - r = 0; - else - r = handler_message_commit(msgid, path); - } - if (ifile) - fclose(ifile); - else - close(imsg.fd); - if (ofile) - fclose(ofile); - else - close(fd); - } - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_QUEUE_MESSAGE_FD_R: - queue_msg_get(&msgid, sizeof(msgid)); - queue_msg_end(); - - fd = handler_message_fd_r(msgid); - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, fd, NULL, 0); - break; - - case PROC_QUEUE_MESSAGE_CORRUPT: - queue_msg_get(&msgid, sizeof(msgid)); - queue_msg_end(); - - r = handler_message_corrupt(msgid); - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_QUEUE_ENVELOPE_CREATE: - queue_msg_get(&msgid, sizeof(msgid)); - r = handler_envelope_create(msgid, rdata, rlen, &evpid); - queue_msg_get(NULL, rlen); - queue_msg_end(); - - queue_msg_add(&r, sizeof(r)); - if (r == 1) - queue_msg_add(&evpid, sizeof(evpid)); - queue_msg_close(); - break; - - case PROC_QUEUE_ENVELOPE_DELETE: - queue_msg_get(&evpid, sizeof(evpid)); - queue_msg_end(); - - r = handler_envelope_delete(evpid); - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_QUEUE_ENVELOPE_LOAD: - queue_msg_get(&evpid, sizeof(evpid)); - queue_msg_end(); - - r = handler_envelope_load(evpid, buffer, sizeof(buffer)); - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, -1, buffer, r); - break; - - case PROC_QUEUE_ENVELOPE_UPDATE: - queue_msg_get(&evpid, sizeof(evpid)); - r = handler_envelope_update(evpid, rdata, rlen); - queue_msg_get(NULL, rlen); - queue_msg_end(); - - imsg_compose(&ibuf, PROC_QUEUE_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_QUEUE_ENVELOPE_WALK: - queue_msg_end(); - - r = handler_envelope_walk(&evpid, buffer, sizeof(buffer)); - - queue_msg_add(&r, sizeof(r)); - if (r > 0) { - queue_msg_add(&evpid, sizeof(evpid)); - queue_msg_add(buffer, r); - } - queue_msg_close(); - break; - - default: - log_warnx("warn: queue-api: bad message %d", imsg.hdr.type); - fatalx("queue-api: exiting"); - } -} - -void -queue_api_on_close(int(*cb)(void)) -{ - handler_close = cb; -} - -void -queue_api_on_message_create(int(*cb)(uint32_t *)) -{ - handler_message_create = cb; -} - -void -queue_api_on_message_commit(int(*cb)(uint32_t, const char *)) -{ - handler_message_commit = cb; -} - -void -queue_api_on_message_delete(int(*cb)(uint32_t)) -{ - handler_message_delete = cb; -} - -void -queue_api_on_message_fd_r(int(*cb)(uint32_t)) -{ - handler_message_fd_r = cb; -} - -void -queue_api_on_message_corrupt(int(*cb)(uint32_t)) -{ - handler_message_corrupt = cb; -} - -void -queue_api_on_envelope_create(int(*cb)(uint32_t, const char *, size_t, uint64_t *)) -{ - handler_envelope_create = cb; -} - -void -queue_api_on_envelope_delete(int(*cb)(uint64_t)) -{ - handler_envelope_delete = cb; -} - -void -queue_api_on_envelope_update(int(*cb)(uint64_t, const char *, size_t)) -{ - handler_envelope_update = cb; -} - -void -queue_api_on_envelope_load(int(*cb)(uint64_t, char *, size_t)) -{ - handler_envelope_load = cb; -} - -void -queue_api_on_envelope_walk(int(*cb)(uint64_t *, char *, size_t)) -{ - handler_envelope_walk = cb; -} - -void -queue_api_no_chroot(void) -{ - rootpath = NULL; -} - -void -queue_api_set_chroot(const char *path) -{ - rootpath = path; -} - -void -queue_api_set_user(const char *username) -{ - user = username; -} - -int -queue_api_dispatch(void) -{ - struct passwd *pw = NULL; - ssize_t n; - - if (user) { - pw = getpwnam(user); - if (pw == NULL) { - log_warn("queue-api: getpwnam"); - fatalx("queue-api: exiting"); - } - } - - if (rootpath) { - if (chroot(rootpath) == -1) { - log_warn("queue-api: chroot"); - fatalx("queue-api: exiting"); - } - if (chdir("/") == -1) { - log_warn("queue-api: chdir"); - fatalx("queue-api: exiting"); - } - } - - if (pw && - (setgroups(1, &pw->pw_gid) || - setresgid(pw->pw_gid, pw->pw_gid, pw->pw_gid) || - setresuid(pw->pw_uid, pw->pw_uid, pw->pw_uid))) { - log_warn("queue-api: cannot drop privileges"); - fatalx("queue-api: exiting"); - } - - imsg_init(&ibuf, 0); - - while (1) { - n = imsg_get(&ibuf, &imsg); - if (n == -1) { - log_warn("warn: queue-api: imsg_get"); - break; - } - - if (n) { - rdata = imsg.data; - rlen = imsg.hdr.len - IMSG_HEADER_SIZE; - queue_msg_dispatch(); - imsg_flush(&ibuf); - continue; - } - - n = imsg_read(&ibuf); - if (n == -1) { - log_warn("warn: queue-api: imsg_read"); - break; - } - if (n == 0) { - log_warnx("warn: queue-api: pipe closed"); - break; - } - } - - return (1); -} diff --git a/usr.sbin/smtpd/scheduler_api.c b/usr.sbin/smtpd/scheduler_api.c deleted file mode 100644 index ab23d29eb93..00000000000 --- a/usr.sbin/smtpd/scheduler_api.c +++ /dev/null @@ -1,481 +0,0 @@ -/* $OpenBSD: scheduler_api.c,v 1.7 2015/01/20 17:37:54 deraadt Exp $ */ - -/* - * Copyright (c) 2013 Eric Faurot - * - * Permission to use, copy, modify, and distribute this software for any - * purpose with or without fee is hereby granted, provided that the above - * copyright notice and this permission notice appear in all copies. - * - * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES - * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF - * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR - * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES - * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN - * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF - * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. - */ - -#include -#include -#include -#include - -#include -#include - -#include -#include -#include -#include - -#include "smtpd-defines.h" -#include "smtpd-api.h" -#include "log.h" - -static int (*handler_init)(void); -static int (*handler_insert)(struct scheduler_info *); -static size_t (*handler_commit)(uint32_t); -static size_t (*handler_rollback)(uint32_t); -static int (*handler_update)(struct scheduler_info *); -static int (*handler_delete)(uint64_t); -static int (*handler_hold)(uint64_t, uint64_t); -static int (*handler_release)(int, uint64_t, int); -static int (*handler_batch)(int, int *, size_t *, uint64_t *, int *); -static size_t (*handler_messages)(uint32_t, uint32_t *, size_t); -static size_t (*handler_envelopes)(uint64_t, struct evpstate *, size_t); -static int (*handler_schedule)(uint64_t); -static int (*handler_remove)(uint64_t); -static int (*handler_suspend)(uint64_t); -static int (*handler_resume)(uint64_t); - -#define MAX_BATCH_SIZE 1024 - -static struct imsgbuf ibuf; -static struct imsg imsg; -static size_t rlen; -static char *rdata; -static struct ibuf *buf; -static const char *rootpath = PATH_CHROOT; -static const char *user = SMTPD_USER; - -static void -scheduler_msg_get(void *dst, size_t len) -{ - if (len > rlen) { - log_warnx("warn: scheduler-proc: bad msg len"); - fatalx("scheduler-proc: exiting"); - } - - if (len == 0) - return; - - if (dst) - memmove(dst, rdata, len); - - rlen -= len; - rdata += len; -} - -static void -scheduler_msg_end(void) -{ - if (rlen) { - log_warnx("warn: scheduler-proc: bogus data"); - fatalx("scheduler-proc: exiting"); - } - imsg_free(&imsg); -} - -static void -scheduler_msg_add(const void *data, size_t len) -{ - if (buf == NULL) - buf = imsg_create(&ibuf, PROC_SCHEDULER_OK, 0, 0, 1024); - if (buf == NULL) { - log_warnx("warn: table-api: imsg_create failed"); - fatalx("table-api: exiting"); - } - if (imsg_add(buf, data, len) == -1) { - log_warnx("warn: table-api: imsg_add failed"); - fatalx("table-api: exiting"); - } -} - -static void -scheduler_msg_close(void) -{ - imsg_close(&ibuf, buf); - buf = NULL; -} - -static void -scheduler_msg_dispatch(void) -{ - size_t n, sz, count; - struct evpstate evpstates[MAX_BATCH_SIZE]; - uint64_t evpid, evpids[MAX_BATCH_SIZE], u64; - uint32_t msgids[MAX_BATCH_SIZE], version, msgid; - struct scheduler_info info; - int typemask, r, type, types[MAX_BATCH_SIZE]; - int delay; - - switch (imsg.hdr.type) { - case PROC_SCHEDULER_INIT: - log_debug("scheduler-api: PROC_SCHEDULER_INIT"); - scheduler_msg_get(&version, sizeof(version)); - scheduler_msg_end(); - - if (version != PROC_SCHEDULER_API_VERSION) { - log_warnx("warn: scheduler-api: bad API version"); - fatalx("scheduler-api: exiting"); - } - - r = handler_init(); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_SCHEDULER_INSERT: - log_debug("scheduler-api: PROC_SCHEDULER_INSERT"); - scheduler_msg_get(&info, sizeof(info)); - scheduler_msg_end(); - - r = handler_insert(&info); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_SCHEDULER_COMMIT: - log_debug("scheduler-api: PROC_SCHEDULER_COMMIT"); - scheduler_msg_get(&msgid, sizeof(msgid)); - scheduler_msg_end(); - - n = handler_commit(msgid); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &n, sizeof(n)); - break; - - case PROC_SCHEDULER_ROLLBACK: - log_debug("scheduler-api: PROC_SCHEDULER_ROLLBACK"); - scheduler_msg_get(&msgid, sizeof(msgid)); - scheduler_msg_end(); - - n = handler_rollback(msgid); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &n, sizeof(n)); - break; - - case PROC_SCHEDULER_UPDATE: - log_debug("scheduler-api: PROC_SCHEDULER_UPDATE"); - scheduler_msg_get(&info, sizeof(info)); - scheduler_msg_end(); - - r = handler_update(&info); - - scheduler_msg_add(&r, sizeof(r)); - if (r == 1) - scheduler_msg_add(&info, sizeof(info)); - scheduler_msg_close(); - break; - - case PROC_SCHEDULER_DELETE: - log_debug("scheduler-api: PROC_SCHEDULER_DELETE"); - scheduler_msg_get(&evpid, sizeof(evpid)); - scheduler_msg_end(); - - r = handler_delete(evpid); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_SCHEDULER_HOLD: - log_debug("scheduler-api: PROC_SCHEDULER_HOLD"); - scheduler_msg_get(&evpid, sizeof(evpid)); - scheduler_msg_get(&u64, sizeof(u64)); - scheduler_msg_end(); - - r = handler_hold(evpid, u64); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_SCHEDULER_RELEASE: - log_debug("scheduler-api: PROC_SCHEDULER_RELEASE"); - scheduler_msg_get(&type, sizeof(type)); - scheduler_msg_get(&u64, sizeof(u64)); - scheduler_msg_get(&r, sizeof(r)); - scheduler_msg_end(); - - r = handler_release(type, u64, r); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_SCHEDULER_BATCH: - log_debug("scheduler-api: PROC_SCHEDULER_BATCH"); - scheduler_msg_get(&typemask, sizeof(typemask)); - scheduler_msg_get(&count, sizeof(count)); - scheduler_msg_end(); - - if (count > MAX_BATCH_SIZE) - count = MAX_BATCH_SIZE; - - r = handler_batch(typemask, &delay, &count, evpids, types); - scheduler_msg_add(&r, sizeof(r)); - scheduler_msg_add(&delay, sizeof(delay)); - scheduler_msg_add(&count, sizeof(count)); - if (r > 0) { - scheduler_msg_add(evpids, sizeof(*evpids) * count); - scheduler_msg_add(types, sizeof(*types) * count); - } - scheduler_msg_close(); - break; - - case PROC_SCHEDULER_MESSAGES: - log_debug("scheduler-api: PROC_SCHEDULER_MESSAGES"); - scheduler_msg_get(&msgid, sizeof(msgid)); - scheduler_msg_get(&sz, sizeof(sz)); - scheduler_msg_end(); - - if (sz > MAX_BATCH_SIZE) - sz = MAX_BATCH_SIZE; - - n = handler_messages(msgid, msgids, sz); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, msgids, - n * sizeof(*msgids)); - break; - - case PROC_SCHEDULER_ENVELOPES: - log_debug("scheduler-api: PROC_SCHEDULER_ENVELOPES"); - scheduler_msg_get(&evpid, sizeof(evpid)); - scheduler_msg_get(&sz, sizeof(sz)); - scheduler_msg_end(); - - if (sz > MAX_BATCH_SIZE) - sz = MAX_BATCH_SIZE; - - n = handler_envelopes(evpid, evpstates, sz); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, evpstates, - n * sizeof(*evpstates)); - break; - - case PROC_SCHEDULER_SCHEDULE: - log_debug("scheduler-api: PROC_SCHEDULER_SCHEDULE"); - scheduler_msg_get(&evpid, sizeof(evpid)); - scheduler_msg_end(); - - r = handler_schedule(evpid); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_SCHEDULER_REMOVE: - log_debug("scheduler-api: PROC_SCHEDULER_REMOVE"); - scheduler_msg_get(&evpid, sizeof(evpid)); - scheduler_msg_end(); - - r = handler_remove(evpid); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_SCHEDULER_SUSPEND: - log_debug("scheduler-api: PROC_SCHEDULER_SUSPEND"); - scheduler_msg_get(&evpid, sizeof(evpid)); - scheduler_msg_end(); - - r = handler_suspend(evpid); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - case PROC_SCHEDULER_RESUME: - log_debug("scheduler-api: PROC_SCHEDULER_RESUME"); - scheduler_msg_get(&evpid, sizeof(evpid)); - scheduler_msg_end(); - - r = handler_resume(evpid); - - imsg_compose(&ibuf, PROC_SCHEDULER_OK, 0, 0, -1, &r, sizeof(r)); - break; - - default: - log_warnx("warn: scheduler-api: bad message %d", imsg.hdr.type); - fatalx("scheduler-api: exiting"); - } -} - -void -scheduler_api_on_init(int(*cb)(void)) -{ - handler_init = cb; -} - -void -scheduler_api_on_insert(int(*cb)(struct scheduler_info *)) -{ - handler_insert = cb; -} - -void -scheduler_api_on_commit(size_t(*cb)(uint32_t)) -{ - handler_commit = cb; -} - -void -scheduler_api_on_rollback(size_t(*cb)(uint32_t)) -{ - handler_rollback = cb; -} - -void -scheduler_api_on_update(int(*cb)(struct scheduler_info *)) -{ - handler_update = cb; -} - -void -scheduler_api_on_delete(int(*cb)(uint64_t)) -{ - handler_delete = cb; -} - -void -scheduler_api_on_batch(int(*cb)(int, int *, size_t *, uint64_t *, int *)) -{ - handler_batch = cb; -} - -void -scheduler_api_on_messages(size_t(*cb)(uint32_t, uint32_t *, size_t)) -{ - handler_messages = cb; -} - -void -scheduler_api_on_envelopes(size_t(*cb)(uint64_t, struct evpstate *, size_t)) -{ - handler_envelopes = cb; -} - -void -scheduler_api_on_schedule(int(*cb)(uint64_t)) -{ - handler_schedule = cb; -} - -void -scheduler_api_on_remove(int(*cb)(uint64_t)) -{ - handler_remove = cb; -} - -void -scheduler_api_on_suspend(int(*cb)(uint64_t)) -{ - handler_suspend = cb; -} - -void -scheduler_api_on_resume(int(*cb)(uint64_t)) -{ - handler_resume = cb; -} - -void -scheduler_api_on_hold(int(*cb)(uint64_t, uint64_t)) -{ - handler_hold = cb; -} - -void -scheduler_api_on_release(int(*cb)(int, uint64_t, int)) -{ - handler_release = cb; -} - -void -scheduler_api_no_chroot(void) -{ - rootpath = NULL; -} - -void -scheduler_api_set_chroot(const char *path) -{ - rootpath = path; -} - -void -scheduler_api_set_user(const char *username) -{ - user = username; -} - -int -scheduler_api_dispatch(void) -{ - struct passwd *pw = NULL; - ssize_t n; - - if (user) { - pw = getpwnam(user); - if (pw == NULL) { - log_warn("queue-api: getpwnam"); - fatalx("queue-api: exiting"); - } - } - - if (rootpath) { - if (chroot(rootpath) == -1) { - log_warn("queue-api: chroot"); - fatalx("queue-api: exiting"); - } - if (chdir("/") == -1) { - log_warn("queue-api: chdir"); - fatalx("queue-api: exiting"); - } - } - - if (pw && - (setgroups(1, &pw->pw_gid) || - setresgid(pw->pw_gid, pw->pw_gid, pw->pw_gid) || - setresuid(pw->pw_uid, pw->pw_uid, pw->pw_uid))) { - log_warn("queue-api: cannot drop privileges"); - fatalx("queue-api: exiting"); - } - - imsg_init(&ibuf, 0); - - while (1) { - n = imsg_get(&ibuf, &imsg); - if (n == -1) { - log_warn("warn: scheduler-api: imsg_get"); - break; - } - - if (n) { - rdata = imsg.data; - rlen = imsg.hdr.len - IMSG_HEADER_SIZE; - scheduler_msg_dispatch(); - imsg_flush(&ibuf); - continue; - } - - n = imsg_read(&ibuf); - if (n == -1) { - log_warn("warn: scheduler-api: imsg_read"); - break; - } - if (n == 0) { - log_warnx("warn: scheduler-api: pipe closed"); - break; - } - } - - return (1); -}