From a6c0353b9268d5b780fb7ff05a10cb5031446e5d Mon Sep 17 00:00:00 2001 From: Lennart Poettering Date: Tue, 30 Jul 2013 02:28:22 +0200 Subject: [PATCH] core: open up SendSIGHUP property for transient units --- man/systemd-run.xml | 12 +++++++++++ src/core/dbus-kill.c | 48 +++++++++++++++++++++++++++++++++++++++-- src/core/dbus-kill.h | 2 +- src/core/dbus-scope.c | 4 ++++ src/core/dbus-service.c | 4 ++++ src/run/run.c | 18 +++++++++++++--- 6 files changed, 82 insertions(+), 6 deletions(-) diff --git a/man/systemd-run.xml b/man/systemd-run.xml index 707d88d76..e1e885178 100644 --- a/man/systemd-run.xml +++ b/man/systemd-run.xml @@ -160,6 +160,18 @@ along with systemd; If not, see . + + + + + When terminating the scope unit send a SIGHUP + immediately after SIGTERM. This is useful to indicate to + shells and shell-like processes that the connection has been + sewered. Also see SendSIGHUP= in + systemd.kill5. + + + All command-line arguments after the first non-option diff --git a/src/core/dbus-kill.c b/src/core/dbus-kill.c index e970ea329..beae7da85 100644 --- a/src/core/dbus-kill.c +++ b/src/core/dbus-kill.c @@ -25,12 +25,56 @@ #include "dbus-kill.h" #include "dbus-common.h" -DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_kill_append_mode, kill_mode, KillMode); +static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_kill_append_mode, kill_mode, KillMode); const BusProperty bus_kill_context_properties[] = { { "KillMode", bus_kill_append_mode, "s", offsetof(KillContext, kill_mode) }, { "KillSignal", bus_property_append_int, "i", offsetof(KillContext, kill_signal) }, { "SendSIGKILL", bus_property_append_bool, "b", offsetof(KillContext, send_sigkill) }, { "SendSIGHUP", bus_property_append_bool, "b", offsetof(KillContext, send_sighup) }, - { NULL, } + {} }; + +int bus_kill_context_set_transient_property( + Unit *u, + KillContext *c, + const char *name, + DBusMessageIter *i, + UnitSetPropertiesMode mode, + DBusError *error) { + + assert(u); + assert(c); + assert(name); + assert(i); + + if (streq(name, "SendSIGHUP")) { + + if (dbus_message_iter_get_arg_type(i) != DBUS_TYPE_BOOLEAN) + return -EINVAL; + + if (mode != UNIT_CHECK) { + dbus_bool_t b; + dbus_message_iter_get_basic(i, &b); + c->send_sighup = b; + } + + return 1; + + } else if (streq(name, "SendSIGKILL")) { + + if (dbus_message_iter_get_arg_type(i) != DBUS_TYPE_BOOLEAN) + return -EINVAL; + + if (mode != UNIT_CHECK) { + dbus_bool_t b; + dbus_message_iter_get_basic(i, &b); + c->send_sigkill = b; + } + + return 1; + + } + + return 0; +} diff --git a/src/core/dbus-kill.h b/src/core/dbus-kill.h index 8c8bff592..7676d98e9 100644 --- a/src/core/dbus-kill.h +++ b/src/core/dbus-kill.h @@ -34,4 +34,4 @@ extern const BusProperty bus_kill_context_properties[]; -int bus_kill_append_mode(DBusMessageIter *i, const char *property, void *data); +int bus_kill_context_set_transient_property(Unit *u, KillContext *c, const char *name, DBusMessageIter *i, UnitSetPropertiesMode mode, DBusError *error); diff --git a/src/core/dbus-scope.c b/src/core/dbus-scope.c index 771820c2d..497e4520d 100644 --- a/src/core/dbus-scope.c +++ b/src/core/dbus-scope.c @@ -170,6 +170,10 @@ int bus_scope_set_property( r = bus_scope_set_transient_property(s, name, i, mode, error); if (r != 0) return r; + + r = bus_kill_context_set_transient_property(u, &s->kill_context, name, i, mode, error); + if (r != 0) + return r; } return 0; diff --git a/src/core/dbus-service.c b/src/core/dbus-service.c index 1a44e1fd1..85b13f01e 100644 --- a/src/core/dbus-service.c +++ b/src/core/dbus-service.c @@ -324,6 +324,10 @@ int bus_service_set_property( r = bus_service_set_transient_property(s, name, i, mode, error); if (r != 0) return r; + + r = bus_kill_context_set_transient_property(u, &s->kill_context, name, i, mode, error); + if (r != 0) + return r; } return 0; diff --git a/src/run/run.c b/src/run/run.c index c1a0ffb13..c5d314bdf 100644 --- a/src/run/run.c +++ b/src/run/run.c @@ -35,6 +35,7 @@ static bool arg_remain_after_exit = false; static const char *arg_unit = NULL; static const char *arg_description = NULL; static const char *arg_slice = NULL; +static bool arg_send_sighup = false; static int help(void) { @@ -47,7 +48,8 @@ static int help(void) { " --unit=UNIT Run under the specified unit name\n" " --description=TEXT Description for unit\n" " --slice=SLICE Run in the specified slice\n" - " -r --remain-after-exit Leave service around until explicitly stopped\n", + " -r --remain-after-exit Leave service around until explicitly stopped\n" + " --send-sighup Send SIGHUP when terminating\n", program_invocation_short_name); return 0; @@ -61,7 +63,8 @@ static int parse_argv(int argc, char *argv[]) { ARG_SCOPE, ARG_UNIT, ARG_DESCRIPTION, - ARG_SLICE + ARG_SLICE, + ARG_SEND_SIGHUP, }; static const struct option options[] = { @@ -72,7 +75,8 @@ static int parse_argv(int argc, char *argv[]) { { "unit", required_argument, NULL, ARG_UNIT }, { "description", required_argument, NULL, ARG_DESCRIPTION }, { "slice", required_argument, NULL, ARG_SLICE }, - { "remain-after-exit", required_argument, NULL, 'r' }, + { "remain-after-exit", no_argument, NULL, 'r' }, + { "send-sighup", no_argument, NULL, ARG_SEND_SIGHUP }, { NULL, 0, NULL, 0 }, }; @@ -114,6 +118,10 @@ static int parse_argv(int argc, char *argv[]) { arg_slice = optarg; break; + case ARG_SEND_SIGHUP: + arg_send_sighup = true; + break; + case 'r': arg_remain_after_exit = true; break; @@ -174,6 +182,10 @@ static int message_start_transient_unit_new(sd_bus *bus, const char *name, sd_bu return r; } + r = sd_bus_message_append(m, "(sv)", "SendSIGHUP", "b", arg_send_sighup); + if (r < 0) + return r; + *ret = m; m = NULL; -- 2.30.2