From b548631afcd317fd0f762b324c086d1580f83a50 Mon Sep 17 00:00:00 2001 From: Lennart Poettering Date: Wed, 3 Feb 2010 12:37:42 +0100 Subject: [PATCH] dbus: implement start/stop/restart/reload/cancel D-Bus calls --- dbus-job.c | 39 +++++++++++++++++++----- dbus-unit.c | 85 +++++++++++++++++++++++++++++++++++++++++++++++++++-- job.c | 7 +++++ job.h | 6 +++- 4 files changed, 127 insertions(+), 10 deletions(-) diff --git a/dbus-job.c b/dbus-job.c index fd84aa6a..896dff78 100644 --- a/dbus-job.c +++ b/dbus-job.c @@ -9,9 +9,10 @@ static const char introspection[] = DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE "" " " + " " " " " " - " " + " " " " " " BUS_PROPERTIES_INTERFACE @@ -86,16 +87,40 @@ static int bus_job_append_unit(Manager *m, DBusMessageIter *i, const char *prope } static DBusHandlerResult bus_job_message_dispatch(Job *j, DBusMessage *message) { - const BusProperty properties[] = { - { "org.freedesktop.systemd1.Job", "Id", bus_property_append_uint32, "u", &j->id }, - { "org.freedesktop.systemd1.Job", "State", bus_job_append_state, "s", j }, - { "org.freedesktop.systemd1.Job", "Type", bus_job_append_type, "s", j }, - { "org.freedesktop.systemd1.Job", "Unit", bus_job_append_unit, "(so)", j }, + { "org.freedesktop.systemd1.Job", "Id", bus_property_append_uint32, "u", &j->id }, + { "org.freedesktop.systemd1.Job", "State", bus_job_append_state, "s", j }, + { "org.freedesktop.systemd1.Job", "JobType", bus_job_append_type, "s", j }, + { "org.freedesktop.systemd1.Job", "Unit", bus_job_append_unit, "(so)", j }, { NULL, NULL, NULL, NULL, NULL } }; - return bus_default_message_handler(j->manager, message, introspection, properties); + DBusMessage *reply = NULL; + Manager *m = j->manager; + + if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Job", "Cancel")) { + if (!(reply = dbus_message_new_method_return(message))) + goto oom; + + job_free(j); + + } else + return bus_default_message_handler(j->manager, message, introspection, properties); + + if (reply) { + if (!dbus_connection_send(m->bus, reply, NULL)) + goto oom; + + dbus_message_unref(reply); + } + + return DBUS_HANDLER_RESULT_HANDLED; + +oom: + if (reply) + dbus_message_unref(reply); + + return DBUS_HANDLER_RESULT_NEED_MEMORY; } DBusHandlerResult bus_job_message_handler(DBusConnection *connection, DBusMessage *message, void *data) { diff --git a/dbus-unit.c b/dbus-unit.c index 0bdf1620..db5a8848 100644 --- a/dbus-unit.c +++ b/dbus-unit.c @@ -8,8 +8,23 @@ static const char introspection[] = DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE "" - " " " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " + " " " " " " " " @@ -191,7 +206,73 @@ static DBusHandlerResult bus_unit_message_dispatch(Unit *u, DBusMessage *message { NULL, NULL, NULL, NULL, NULL } }; - return bus_default_message_handler(u->meta.manager, message, introspection, properties); + DBusMessage *reply = NULL; + Manager *m = u->meta.manager; + DBusError error; + JobType job_type = _JOB_TYPE_INVALID; + + dbus_error_init(&error); + + if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Start")) + job_type = JOB_START; + else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Stop")) + job_type = JOB_STOP; + else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Reload")) + job_type = JOB_RELOAD; + else if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Unit", "Restart")) + job_type = JOB_RESTART; + else + return bus_default_message_handler(u->meta.manager, message, introspection, properties); + + if (job_type != _JOB_TYPE_INVALID) { + const char *smode; + JobMode mode; + Job *j; + int r; + char *path; + + if (!dbus_message_get_args( + message, + &error, + DBUS_TYPE_STRING, &smode, + DBUS_TYPE_INVALID)) + return bus_send_error_reply(m, message, &error, -EINVAL); + + if ((mode = job_mode_from_string(smode)) == _JOB_MODE_INVALID) + return bus_send_error_reply(m, message, NULL, -EINVAL); + + if ((r = manager_add_job(m, job_type, u, mode, true, &j)) < 0) + return bus_send_error_reply(m, message, NULL, r); + + if (!(reply = dbus_message_new_method_return(message))) + goto oom; + + if (!(path = job_dbus_path(j))) + goto oom; + + if (!dbus_message_append_args( + reply, + DBUS_TYPE_OBJECT_PATH, &path, + DBUS_TYPE_INVALID)) + goto oom; + } + + if (reply) { + if (!dbus_connection_send(m->bus, reply, NULL)) + goto oom; + + dbus_message_unref(reply); + } + + return DBUS_HANDLER_RESULT_HANDLED; + +oom: + if (reply) + dbus_message_unref(reply); + + dbus_error_free(&error); + + return DBUS_HANDLER_RESULT_NEED_MEMORY; } static DBusHandlerResult bus_unit_message_handler(DBusConnection *connection, DBusMessage *message, void *data) { diff --git a/job.c b/job.c index d02551af..75756e74 100644 --- a/job.c +++ b/job.c @@ -492,3 +492,10 @@ static const char* const job_type_table[_JOB_TYPE_MAX] = { }; DEFINE_STRING_TABLE_LOOKUP(job_type, JobType); + +static const char* const job_mode_table[_JOB_MODE_MAX] = { + [JOB_FAIL] = "fail", + [JOB_REPLACE] = "replace" +}; + +DEFINE_STRING_TABLE_LOOKUP(job_mode, JobMode); diff --git a/job.h b/job.h index 83401e8c..132c46db 100644 --- a/job.h +++ b/job.h @@ -46,7 +46,8 @@ enum JobState { enum JobMode { JOB_FAIL, JOB_REPLACE, - _JOB_MODE_MAX + _JOB_MODE_MAX, + _JOB_MODE_INVALID = -1 }; struct JobDependency { @@ -114,6 +115,9 @@ JobType job_type_from_string(const char *s); const char* job_state_to_string(JobState t); JobState job_state_from_string(const char *s); +const char* job_mode_to_string(JobMode t); +JobMode job_mode_from_string(const char *s); + char *job_dbus_path(Job *j); #endif -- 2.39.5