emehta-contrib/on_db_save.c

145 lines
3.2 KiB
C

#include "atheme.h"
#include "conf.h"
#include "datastream.h"
#ifndef _WIN32
DECLARE_MODULE_V1
(
"contrib/on_db_save", false, _modinit, _moddeinit,
"",
"Atheme Development Group <http://www.atheme.org>"
);
static char *command = NULL;
static void on_db_save(void *unused);
static struct update_command_state {
connection_t *out, *err;
pid_t pid;
int running;
} update_command_proc;
void _modinit(module_t *m)
{
hook_add_event("db_saved");
hook_add_db_saved(on_db_save);
add_dupstr_conf_item("db_update_command", &conf_gi_table, 0, &command, NULL);
}
void _moddeinit(module_unload_intent_t intent)
{
hook_del_db_saved(on_db_save);
del_conf_item("db_update_command", &conf_gi_table);
}
static void update_command_finished(pid_t pid, int status, void *data)
{
if (!WIFEXITED(status) || WEXITSTATUS(status) != 0)
slog(LG_ERROR, "ERROR: Database update command failed with error %d", WEXITSTATUS(status));
update_command_proc.running = 0;
}
static void update_command_recvq_handler(connection_t *cptr, int err)
{
char buf[BUFSIZE];
int count;
count = recvq_getline(cptr, buf, sizeof(buf) - 1);
if (count <= 0)
return;
if (buf[count-1] == '\n')
count--;
if (count == 0)
buf[count++] = ' ';
buf[count] = '\0';
if (err)
{
slog(LG_ERROR, "ERROR: database update command said: %s", buf);
}
else
slog(LG_DEBUG, "db update command stdout: %s", buf);
}
static void update_command_stdout_handler(connection_t *cptr)
{
update_command_recvq_handler(cptr, 0);
}
static void update_command_stderr_handler(connection_t *cptr)
{
update_command_recvq_handler(cptr, 1);
}
static void on_db_save(void *unused)
{
int stdout_pipes[2], stderr_pipes[2];
pid_t pid;
int errno1;
if (!command)
return;
if (update_command_proc.running)
{
slog(LG_ERROR, "ERROR: database update command is still running");
return;
}
if (pipe(stdout_pipes) == -1)
{
int err = errno;
slog(LG_ERROR, "ERROR: Couldn't create pipe for database update command: %s", strerror(err));
return;
}
if (pipe(stderr_pipes) == -1)
{
int err = errno;
slog(LG_ERROR, "ERROR: Couldn't create pipe for database update command: %s", strerror(err));
close(stdout_pipes[0]);
close(stdout_pipes[1]);
return;
}
pid = fork();
switch (pid)
{
case -1:
errno1 = errno;
slog(LG_ERROR, "Failed to fork for database update command: %s", strerror(errno1));
return;
case 0:
connection_close_all_fds();
close(stdout_pipes[0]);
close(stderr_pipes[0]);
dup2(stdout_pipes[1], 1);
dup2(stderr_pipes[1], 2);
close(stdout_pipes[1]);
close(stderr_pipes[1]);
execl("/bin/sh", "sh", "-c", command, NULL);
write(2, "Failed to exec /bin/sh\n", 23);
_exit(255);
return;
default:
close(stdout_pipes[1]);
close(stderr_pipes[1]);
update_command_proc.out = connection_add("update_command_stdout", stdout_pipes[0], 0, recvq_put, NULL);
update_command_proc.err = connection_add("update_command_stderr", stderr_pipes[0], 0, recvq_put, NULL);
update_command_proc.out->recvq_handler = update_command_stdout_handler;
update_command_proc.err->recvq_handler = update_command_stderr_handler;
update_command_proc.pid = pid;
update_command_proc.running = 1;
childproc_add(pid, "db_update", update_command_finished, NULL);
break;
}
}
#endif