X-Git-Url: http://git.tuebingen.mpg.de/?p=dss.git;a=blobdiff_plain;f=dss.c;h=631b36dfe12795da7ccaef38ea343b6342a14dda;hp=bf30f893fa36fb9749d00ab706311709a9d4a42f;hb=79489a96d14274b78a654bbcf60d8eb8289cae50;hpb=17eea857c175123d68e7dffc28134befa4c0aa5f diff --git a/dss.c b/dss.c index bf30f89..631b36d 100644 --- a/dss.c +++ b/dss.c @@ -1,5 +1,5 @@ /* - * Copyright (C) 2008-2010 Andre Noll + * Copyright (C) 2008-2011 Andre Noll * * Licensed under the GPL v2. For licencing details see COPYING. */ @@ -33,6 +33,7 @@ #include "df.h" #include "time.h" #include "snap.h" +#include "ipc.h" /** Command line and config file options. */ static struct gengetopt_args_info conf; @@ -170,7 +171,10 @@ static void dump_dss_config(const char *msg) COMMAND(ls) \ COMMAND(create) \ COMMAND(prune) \ - COMMAND(run) + COMMAND(run) \ + COMMAND(kill) \ + COMMAND(reload) \ + #define COMMAND(x) static int com_ ##x(void); COMMANDS #undef COMMAND @@ -238,6 +242,35 @@ static char *get_config_file_name(void) return config_file; } +static int send_signal(int sig) +{ + pid_t pid; + char *config_file = get_config_file_name(); + int ret = get_dss_pid(config_file, &pid); + + free(config_file); + if (ret < 0) + return ret; + if (conf.dry_run_given) { + dss_msg("%d\n", (int)pid); + return 0; + } + ret = kill(pid, sig); + if (ret < 0) + return -ERRNO_TO_DSS_ERROR(errno); + return 1; +} + +static int com_kill(void) +{ + return send_signal(SIGTERM); +} + +static int com_reload(void) +{ + return send_signal(SIGHUP); +} + static void dss_get_snapshot_list(struct snapshot_list *sl) { get_snapshot_list(sl, conf.unit_interval_arg, conf.num_intervals_arg); @@ -299,29 +332,22 @@ static int next_snapshot_is_due(void) return 0; } -static int pre_create_hook(void) +static void pre_create_hook(void) { - int ret, fds[3] = {0, 0, 0}; - assert(snapshot_creation_status == HS_READY); /* make sure that the next snapshot time will be recomputed */ invalidate_next_snapshot_time(); DSS_DEBUG_LOG("executing %s\n", conf.pre_create_hook_arg); - ret = dss_exec_cmdline_pid(&create_pid, - conf.pre_create_hook_arg, fds); - if (ret < 0) - return ret; + dss_exec_cmdline_pid(&create_pid, conf.pre_create_hook_arg); snapshot_creation_status = HS_PRE_RUNNING; - return ret; } -static int pre_remove_hook(struct snapshot *s, const char *why) +static void pre_remove_hook(struct snapshot *s, const char *why) { - int ret, fds[3] = {0, 0, 0}; char *cmd; if (!s) - return 0; + return; DSS_DEBUG_LOG("%s snapshot %s\n", why, s->name); assert(snapshot_removal_status == HS_READY); assert(remove_pid == 0); @@ -334,18 +360,14 @@ static int pre_remove_hook(struct snapshot *s, const char *why) cmd = make_message("%s %s/%s", conf.pre_remove_hook_arg, conf.dest_dir_arg, s->name); DSS_DEBUG_LOG("executing %s\n", cmd); - ret = dss_exec_cmdline_pid(&remove_pid, cmd, fds); + dss_exec_cmdline_pid(&remove_pid, cmd); free(cmd); - if (ret < 0) - return ret; snapshot_removal_status = HS_PRE_RUNNING; - return ret; } static int exec_rm(void) { struct snapshot *s = snapshot_currently_being_removed; - int fds[3] = {0, 0, 0}; char *new_name = being_deleted_name(s); char *argv[] = {"rm", "-rf", new_name, NULL}; int ret; @@ -357,9 +379,7 @@ static int exec_rm(void) ret = dss_rename(s->name, new_name); if (ret < 0) goto out; - ret = dss_exec(&remove_pid, argv[0], argv, fds); - if (ret < 0) - goto out; + dss_exec(&remove_pid, argv[0], argv); snapshot_removal_status = HS_RUNNING; out: free(new_name); @@ -575,31 +595,24 @@ static int try_to_free_disk_space(void) ret = -ERRNO_TO_DSS_ERROR(ENOSPC); goto out; remove: - ret = pre_remove_hook(victim, why); + pre_remove_hook(victim, why); out: free_snapshot_list(&sl); return ret; } -static int post_create_hook(void) +static void post_create_hook(void) { - int ret, fds[3] = {0, 0, 0}; - char *cmd; - - cmd = make_message("%s %s/%s", conf.post_create_hook_arg, + char *cmd = make_message("%s %s/%s", conf.post_create_hook_arg, conf.dest_dir_arg, path_to_last_complete_snapshot); DSS_NOTICE_LOG("executing %s\n", cmd); - ret = dss_exec_cmdline_pid(&create_pid, cmd, fds); + dss_exec_cmdline_pid(&create_pid, cmd); free(cmd); - if (ret < 0) - return ret; snapshot_creation_status = HS_POST_RUNNING; - return ret; } -static int post_remove_hook(void) +static void post_remove_hook(void) { - int ret, fds[3] = {0, 0, 0}; char *cmd; struct snapshot *s = snapshot_currently_being_removed; @@ -608,12 +621,9 @@ static int post_remove_hook(void) cmd = make_message("%s %s/%s", conf.post_remove_hook_arg, conf.dest_dir_arg, s->name); DSS_NOTICE_LOG("executing %s\n", cmd); - ret = dss_exec_cmdline_pid(&remove_pid, cmd, fds); + dss_exec_cmdline_pid(&remove_pid, cmd); free(cmd); - if (ret < 0) - return ret; snapshot_removal_status = HS_POST_RUNNING; - return ret; } static void dss_kill(pid_t pid, int sig, const char *msg) @@ -1148,14 +1158,12 @@ static void free_rsync_argv(char **argv) static int create_snapshot(char **argv) { - int ret, fds[3] = {0, 0, 0}; + int ret; ret = rename_resume_snap(current_snapshot_creation_time); if (ret < 0) return ret; - ret = dss_exec(&create_pid, argv[0], argv, fds); - if (ret < 0) - return ret; + dss_exec(&create_pid, argv[0], argv); snapshot_creation_status = HS_RUNNING; return ret; } @@ -1197,9 +1205,7 @@ static int select_loop(void) continue; } if (snapshot_removal_status == HS_SUCCESS) { - ret = post_remove_hook(); - if (ret < 0) - goto out; + post_remove_hook(); continue; } ret = try_to_free_disk_space(); @@ -1214,9 +1220,7 @@ static int select_loop(void) case HS_READY: if (!next_snapshot_is_due()) continue; - ret = pre_create_hook(); - if (ret < 0) - goto out; + pre_create_hook(); continue; case HS_PRE_RUNNING: case HS_RUNNING: @@ -1239,9 +1243,7 @@ static int select_loop(void) goto out; continue; case HS_SUCCESS: - ret = post_create_hook(); - if (ret < 0) - goto out; + post_create_hook(); continue; } } @@ -1251,18 +1253,30 @@ out: static void exit_hook(int exit_code) { - int fds[3] = {0, 0, 0}; char *argv[] = {conf.exit_hook_arg, dss_strerror(-exit_code), NULL}; pid_t pid; DSS_NOTICE_LOG("executing %s %s\n", argv[0], argv[1]); - dss_exec(&pid, conf.exit_hook_arg, argv, fds); + dss_exec(&pid, conf.exit_hook_arg, argv); +} + +static void lock_dss_or_die(void) +{ + char *config_file = get_config_file_name(); + int ret = lock_dss(config_file); + + free(config_file); + if (ret < 0) { + DSS_EMERG_LOG("failed to lock: %s\n", dss_strerror(-ret)); + exit(EXIT_FAILURE); + } } static int com_run(void) { int ret; + lock_dss_or_die(); if (conf.dry_run_given) { DSS_ERROR_LOG("dry_run not supported by this command\n"); return -E_SYNTAX; @@ -1285,6 +1299,7 @@ static int com_prune(void) struct disk_space ds; const char *why; + lock_dss_or_die(); ret = get_disk_space(".", &ds); if (ret < 0) return ret; @@ -1307,9 +1322,7 @@ rm: ret = 0; goto out; } - ret = pre_remove_hook(victim, why); - if (ret < 0) - goto out; + pre_remove_hook(victim, why); if (snapshot_removal_status == HS_PRE_RUNNING) { ret = wait_for_remove_process(); if (ret < 0) @@ -1325,9 +1338,7 @@ rm: goto out; if (snapshot_removal_status != HS_SUCCESS) goto out; - ret = post_remove_hook(); - if (ret < 0) - goto out; + post_remove_hook(); if (snapshot_removal_status != HS_POST_RUNNING) goto out; ret = wait_for_remove_process(); @@ -1344,6 +1355,7 @@ static int com_create(void) int ret, status; char **rsync_argv; + lock_dss_or_die(); if (conf.dry_run_given) { int i; char *msg = NULL; @@ -1359,9 +1371,7 @@ static int com_create(void) free(msg); return 1; } - ret = pre_create_hook(); - if (ret < 0) - return ret; + pre_create_hook(); if (create_pid) { ret = wait_for_process(create_pid, &status); if (ret < 0)