diff options
author | Yu Watanabe <watanabe.yu+github@gmail.com> | 2022-10-18 23:08:24 +0900 |
---|---|---|
committer | Yu Watanabe <watanabe.yu+github@gmail.com> | 2022-10-24 18:55:11 +0900 |
commit | 5905bddce27ec76793bec4ae458b6d471698b372 (patch) | |
tree | 8aa7cc421422322bc171774923506cba5ef0c143 /src/systemctl | |
parent | 4856f63846fc794711e1b8ec970e4c56494cd320 (diff) | |
download | systemd-5905bddce27ec76793bec4ae458b6d471698b372.tar.gz |
systemctl: fix potential memleak on failure in determine_default()
And make verb_set_default() return zero on success.
Diffstat (limited to 'src/systemctl')
-rw-r--r-- | src/systemctl/systemctl-set-default.c | 16 |
1 files changed, 7 insertions, 9 deletions
diff --git a/src/systemctl/systemctl-set-default.c b/src/systemctl/systemctl-set-default.c index 06845be401..c2dbf97649 100644 --- a/src/systemctl/systemctl-set-default.c +++ b/src/systemctl/systemctl-set-default.c @@ -109,9 +109,8 @@ int verb_set_default(int argc, char *argv[], void *userdata) { if (install_client_side()) { r = unit_file_set_default(arg_scope, UNIT_FILE_FORCE, arg_root, unit, &changes, &n_changes); install_changes_dump(r, "set default", changes, n_changes, arg_quiet); - - if (r > 0) - r = 0; + if (r < 0) + goto finish; } else { _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL; @@ -134,10 +133,9 @@ int verb_set_default(int argc, char *argv[], void *userdata) { /* Try to reload if enabled */ if (!arg_no_reload) { r = daemon_reload(ACTION_RELOAD, /* graceful= */ false); - if (r > 0) - r = 0; - } else - r = 0; + if (r < 0) + goto finish; + } } emit_cmdline_warning(); @@ -147,7 +145,7 @@ int verb_set_default(int argc, char *argv[], void *userdata) { r = determine_default(&final); if (r < 0) - return r; + goto finish; if (!streq(final, unit)) log_notice("Note: \"%s\" is the default unit (possibly a runtime override).", final); @@ -156,5 +154,5 @@ int verb_set_default(int argc, char *argv[], void *userdata) { finish: install_changes_free(changes, n_changes); - return r; + return r < 0 ? r : 0; } |