#include "error.h"
#include "fd.h"
#include "exec.h"
+#include "daemon.h"
+#include "signal.h"
+#include "df.h"
struct gengetopt_args_info conf;
char *dss_error_txt = NULL;
+static FILE *logfile;
+int signal_pipe;
DEFINE_DSS_ERRLIST;
COMMAND(create) \
COMMAND(prune) \
COMMAND(run)
-#define COMMAND(x) int com_ ##x(int, char * const * const);
+#define COMMAND(x) int com_ ##x(void);
COMMANDS
#undef COMMAND
-#define COMMAND(x) if (conf.x ##_given) return com_ ##x(argc, argv);
-int call_command_handler(int argc, char * const * const argv)
+#define COMMAND(x) if (conf.x ##_given) return com_ ##x();
+int call_command_handler(void)
{
COMMANDS
DSS_EMERG_LOG("BUG: did not find command handler\n");
unsigned interval;
};
+__printf_2_3 void dss_log(int ll, const char* fmt,...)
+{
+ va_list argp;
+ FILE *outfd;
+ struct tm *tm;
+ time_t t1;
+ char str[255] = "";
+
+ if (ll < conf.loglevel_arg)
+ return;
+ outfd = logfile? logfile : stderr;
+ time(&t1);
+ tm = localtime(&t1);
+ strftime(str, sizeof(str), "%b %d %H:%M:%S", tm);
+ fprintf(outfd, "%s ", str);
+ if (conf.loglevel_arg <= INFO)
+ fprintf(outfd, "%i: ", ll);
+ va_start(argp, fmt);
+ vfprintf(outfd, fmt, argp);
+ va_end(argp);
+}
+
+/**
+ * Print a message either to stdout or to the log file.
+ */
+__printf_1_2 void dss_msg(const char* fmt,...)
+{
+ FILE *outfd = conf.daemon_given? logfile : stdout;
+ va_list argp;
+ va_start(argp, fmt);
+ vfprintf(outfd, fmt, argp);
+ va_end(argp);
+}
+
int is_snapshot(const char *dirname, int64_t now, struct snapshot *s)
{
int i, ret;
DSS_DEBUG_LOG("Waiting for process %d to terminate\n", (int)pid);
for (;;) {
- ret = waitpid(pid, status, 0);
- if (ret >= 0 || errno != EINTR)
+ pause();
+ ret = next_signal();
+ if (ret < 0)
break;
+ if (!ret)
+ continue;
+ if (ret == SIGCHLD) {
+ ret = waitpid(pid, status, 0);
+ if (ret >= 0)
+ break;
+ if (errno != EINTR) /* error */
+ break;
+ }
+ DSS_WARNING_LOG("sending SIGTERM to pid %d\n", (int)pid);
+ kill(pid, SIGTERM);
}
if (ret < 0) {
ret = -ERRNO_TO_DSS_ERROR(errno);
return ret;
}
-int remove_redundant_snapshot(struct snapshot_list *sl,
- int dry_run, pid_t *pid)
+int remove_redundant_snapshot(struct snapshot_list *sl, pid_t *pid)
{
int ret, i, interval;
struct snapshot *s;
prev = s;
}
assert(victim);
- if (dry_run) {
- printf("%s would be removed (interval = %i)\n",
+ if (conf.dry_run_given) {
+ dss_msg("%s would be removed (interval = %i)\n",
victim->name, victim->interval);
continue;
}
return 0;
}
-int remove_old_snapshot(struct snapshot_list *sl, int dry_run, pid_t *pid)
+int remove_old_snapshot(struct snapshot_list *sl, pid_t *pid)
{
int i, ret;
struct snapshot *s;
FOR_EACH_SNAPSHOT(s, i, sl) {
if (s->interval <= conf.num_intervals_arg)
continue;
- if (dry_run) {
- printf("%s would be removed (interval = %i)\n",
+ if (conf.dry_run_given) {
+ dss_msg("%s would be removed (interval = %i)\n",
s->name, s->interval);
continue;
}
return 1;
}
-int com_run(int argc, char * const * argv)
+int com_run(void)
{
+ int ret;
+
+ if (conf.dry_run_given) {
+ make_err_msg("dry_run not supported by this command");
+ return -E_SYNTAX;
+ }
+ ret = install_sighandler(SIGHUP);
+ if (ret < 0)
+ return ret;
return 42;
}
-int com_prune(int argc, char * const * argv)
+void log_disk_space(struct disk_space *ds)
+{
+ DSS_INFO_LOG("free: %uM/%uM (%u%%), %u%% inodes unused\n",
+ ds->free_mb, ds->total_mb, ds->percent_free,
+ ds->percent_free_inodes);
+}
+
+int com_prune(void)
{
- int ret, dry_run = 0;
+ int ret;
struct snapshot_list sl;
pid_t pid;
+ struct disk_space ds;
- if (argc > 2) {
- make_err_msg("too many arguments");
- return -E_SYNTAX;
- }
- if (argc == 2) {
- if (strcmp(argv[1], "-d")) {
- make_err_msg("%s", argv[1]);
- return -E_SYNTAX;
- }
- dry_run = 1;
- }
+ ret = get_disk_space(".", &ds);
+ if (ret < 0)
+ return ret;
+ log_disk_space(&ds);
for (;;) {
get_snapshot_list(&sl);
- ret = remove_old_snapshot(&sl, dry_run, &pid);
+ ret = remove_old_snapshot(&sl, &pid);
free_snapshot_list(&sl);
if (ret < 0)
return ret;
}
for (;;) {
get_snapshot_list(&sl);
- ret = remove_redundant_snapshot(&sl, dry_run, &pid);
+ ret = remove_redundant_snapshot(&sl, &pid);
free_snapshot_list(&sl);
if (ret < 0)
return ret;
return ret;
}
-int com_create(int argc, __a_unused char * const * argv)
+int com_create(void)
{
int ret, status, es;
char **rsync_argv;
int64_t snapshot_num;
pid_t pid;
- if (argc != 1) {
- ret = -E_SYNTAX;
- make_err_msg("create: no args expected, %d given", argc - 1);
- return ret;
- }
create_rsync_argv(&rsync_argv, &snapshot_num);
+ if (conf.dry_run_given) {
+ int i;
+ char *msg = NULL;
+ for (i = 0; rsync_argv[i]; i++) {
+ char *tmp = msg;
+ msg = make_message("%s%s%s", tmp? tmp : "",
+ tmp? " " : "", rsync_argv[i]);
+ free(tmp);
+ }
+ dss_msg("%s\n", msg);
+ free(msg);
+ return 1;
+ }
DSS_NOTICE_LOG("creating snapshot %lli\n", (long long)snapshot_num);
ret = create_snapshot(rsync_argv, &pid);
if (ret < 0)
return ret;
}
-int com_ls(int argc, __a_unused char * const * argv)
+int com_ls(void)
{
- int i, ret;
+ int i;
struct snapshot_list sl;
struct snapshot *s;
- if (argc != 1) {
- ret = -E_SYNTAX;
- make_err_msg("ls: no args expected, %d given", argc - 1);
- return ret;
- }
get_snapshot_list(&sl);
FOR_EACH_SNAPSHOT(s, i, &sl)
- printf("%u\t%s\n", s->interval, s->name);
+ dss_msg("%u\t%s\n", s->interval, s->name);
free_snapshot_list(&sl);
return 1;
}
exit(status);
}
-__printf_2_3 void dss_log(int ll, const char* fmt,...)
-{
- va_list argp;
- if (ll < conf.loglevel_arg)
- return;
- va_start(argp, fmt);
- vfprintf(stderr, fmt, argp);
- va_end(argp);
-}
-
int read_config_file(void)
{
int ret;
return 1;
}
+static void setup_signal_handling(void)
+{
+ int ret;
+
+ DSS_NOTICE_LOG("setting up signal handlers\n");
+ signal_pipe = signal_init(); /* always successful */
+ ret = install_sighandler(SIGINT);
+ if (ret < 0)
+ goto err;
+ ret = install_sighandler(SIGTERM);
+ if (ret < 0)
+ goto err;
+ ret = install_sighandler(SIGCHLD);
+ if (ret < 0)
+ goto err;
+ return;
+err:
+ DSS_EMERG_LOG("could not install signal handlers\n");
+ exit(EXIT_FAILURE);
+}
+
+
int main(int argc, char **argv)
{
int ret;
ret = check_config();
if (ret < 0)
goto out;
+ if (conf.logfile_given) {
+ logfile = open_log(conf.logfile_arg);
+ log_welcome(conf.loglevel_arg);
+ }
ret = dss_chdir(conf.dest_dir_arg);
if (ret < 0)
goto out;
- ret = call_command_handler(conf.inputs_num, conf.inputs);
+ if (conf.daemon_given)
+ daemon_init();
+ setup_signal_handling();
+ ret = call_command_handler();
out:
if (ret < 0)
log_err_msg(EMERG, -ret);
projects / dss.git / blobdiff