static int
queue_proc_init(struct passwd *pw, int server, const char *conf)
{
uint32_t version;
int fd;
fd = fork_proc_backend("queue", conf, "queue-proc");
if (fd == -1)
fatalx("queue-proc: exiting");
imsg_init(&ibuf, fd);
version = PROC_QUEUE_API_VERSION;
imsg_compose(&ibuf, PROC_QUEUE_INIT, 0, 0, -1,
&version, sizeof(version));
queue_api_on_close(queue_proc_close);
queue_api_on_message_create(queue_proc_message_create);
queue_api_on_message_commit(queue_proc_message_commit);
queue_api_on_message_delete(queue_proc_message_delete);
queue_api_on_message_fd_r(queue_proc_message_fd_r);
queue_api_on_message_corrupt(queue_proc_message_corrupt);
queue_api_on_envelope_create(queue_proc_envelope_create);
queue_api_on_envelope_delete(queue_proc_envelope_delete);
queue_api_on_envelope_update(queue_proc_envelope_update);
queue_api_on_envelope_load(queue_proc_envelope_load);
queue_api_on_envelope_walk(queue_proc_envelope_walk);
queue_proc_call();
queue_proc_end();
return (1);
}
static int
queue_proc_envelope_walk(uint64_t *evpid, char *buf, size_t len)
{
int r;
imsg_compose(&ibuf, PROC_QUEUE_ENVELOPE_WALK, 0, 0, -1, NULL, 0);
queue_proc_call();
queue_proc_read(&r, sizeof(r));
if (r > 0) {
queue_proc_read(evpid, sizeof(*evpid));
if (rlen > len) {
log_warnx("warn: queue-proc: buf too small");
fatalx("queue-proc: exiting");
}
if (r != (int)rlen) {
log_warnx("warn: queue-proc: len mismatch");
fatalx("queue-proc: exiting");
}
queue_proc_read(buf, rlen);
}
queue_proc_end();
return (r);
}
static int
queue_proc_init(struct passwd *pw, int server)
{
int sp[2];
uint32_t version;
errno = 0;
if (socketpair(AF_UNIX, SOCK_STREAM, PF_UNSPEC, sp) < 0) {
log_warn("warn: queue-proc: socketpair");
return (0);
}
if ((pid = fork()) == -1) {
log_warn("warn: queue-proc: fork");
goto err;
}
if (pid == 0) {
/* child process */
dup2(sp[0], STDIN_FILENO);
if (closefrom(STDERR_FILENO + 1) < 0)
exit(1);
execl(execpath, "queue_ramproc", NULL);
err(1, "execl");
}
/* parent process */
close(sp[0]);
imsg_init(&ibuf, sp[1]);
version = PROC_QUEUE_API_VERSION;
imsg_compose(&ibuf, PROC_QUEUE_INIT, 0, 0, -1,
&version, sizeof(version));
queue_api_on_message_create(queue_proc_message_create);
queue_api_on_message_commit(queue_proc_message_commit);
queue_api_on_message_delete(queue_proc_message_delete);
queue_api_on_message_fd_r(queue_proc_message_fd_r);
queue_api_on_message_corrupt(queue_proc_message_corrupt);
queue_api_on_envelope_create(queue_proc_envelope_create);
queue_api_on_envelope_delete(queue_proc_envelope_delete);
queue_api_on_envelope_update(queue_proc_envelope_update);
queue_api_on_envelope_load(queue_proc_envelope_load);
queue_api_on_envelope_walk(queue_proc_envelope_walk);
queue_proc_call();
queue_proc_end();
return (1);
err:
close(sp[0]);
close(sp[1]);
return (0);
}
static int
queue_proc_message_fd_r(uint32_t msgid)
{
imsg_compose(&ibuf, PROC_QUEUE_MESSAGE_FD_R, 0, 0, -1, &msgid,
sizeof(msgid));
queue_proc_call();
queue_proc_end();
return (imsg.fd);
}
static int
queue_proc_close(void)
{
int r;
imsg_compose(&ibuf, PROC_QUEUE_MESSAGE_CORRUPT, 0, 0, -1, NULL, 0);
queue_proc_call();
queue_proc_read(&r, sizeof(r));
queue_proc_end();
return (r);
}
static int
queue_proc_envelope_delete(uint64_t evpid)
{
int r;
imsg_compose(&ibuf, PROC_QUEUE_ENVELOPE_DELETE, 0, 0, -1, &evpid,
sizeof(evpid));
queue_proc_call();
queue_proc_read(&r, sizeof(r));
queue_proc_end();
return (r);
}
static int
queue_proc_message_corrupt(uint32_t msgid)
{
int r;
imsg_compose(&ibuf, PROC_QUEUE_MESSAGE_CORRUPT, 0, 0, -1, &msgid,
sizeof(msgid));
queue_proc_call();
queue_proc_read(&r, sizeof(r));
queue_proc_end();
return (r);
}
static int
queue_proc_message_create(uint32_t *msgid)
{
int r;
imsg_compose(&ibuf, PROC_QUEUE_MESSAGE_CREATE, 0, 0, -1, NULL, 0);
queue_proc_call();
queue_proc_read(&r, sizeof(r));
if (r == 1)
queue_proc_read(msgid, sizeof(*msgid));
queue_proc_end();
return (r);
}
static int
queue_proc_envelope_update(uint64_t evpid, const char *buf, size_t len)
{
struct ibuf *b;
int r;
b = imsg_create(&ibuf, PROC_QUEUE_ENVELOPE_UPDATE, 0, 0,
len + sizeof(evpid));
if (imsg_add(b, &evpid, sizeof(evpid)) == -1 ||
imsg_add(b, buf, len) == -1)
return (0);
imsg_close(&ibuf, b);
queue_proc_call();
queue_proc_read(&r, sizeof(r));
queue_proc_end();
return (r);
}
static int
queue_proc_message_commit(uint32_t msgid, const char *path)
{
int r, fd;
fd = open(path, O_RDONLY);
if (fd == -1) {
log_warn("queue-proc: open: %s", path);
return (0);
}
imsg_compose(&ibuf, PROC_QUEUE_MESSAGE_COMMIT, 0, 0, fd, &msgid,
sizeof(msgid));
queue_proc_call();
queue_proc_read(&r, sizeof(r));
queue_proc_end();
return (r);
}
static int
queue_proc_envelope_load(uint64_t evpid, char *buf, size_t len)
{
int r;
imsg_compose(&ibuf, PROC_QUEUE_ENVELOPE_LOAD, 0, 0, -1, &evpid,
sizeof(evpid));
queue_proc_call();
if (rlen > len) {
log_warnx("warn: queue-proc: buf too small");
fatalx("queue-proc: exiting");
}
r = rlen;
queue_proc_read(buf, rlen);
queue_proc_end();
return (r);
}
static int
queue_proc_envelope_create(uint32_t msgid, const char *buf, size_t len,
uint64_t *evpid)
{
struct ibuf *b;
int r;
msgid = evpid_to_msgid(*evpid);
b = imsg_create(&ibuf, PROC_QUEUE_ENVELOPE_CREATE, 0, 0,
sizeof(msgid) + len);
if (imsg_add(b, &msgid, sizeof(msgid)) == -1 ||
imsg_add(b, buf, len) == -1)
return (0);
imsg_close(&ibuf, b);
queue_proc_call();
queue_proc_read(&r, sizeof(r));
if (r == 1)
queue_proc_read(evpid, sizeof(*evpid));
queue_proc_end();
return (r);
}