X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Fupdate-utmp%2Fupdate-utmp.c;h=a62c538ecd65ffbda89ba7c1cef6b23eb27ffe6f;hp=67c5788fe7663032a0077b90bb9c09e6c696a381;hb=47c649b5de480b832bb11e0d52ba58d962d61fb2;hpb=55f2dca329afd0dcdc4793ce3e945cb8af653937 diff --git a/src/update-utmp/update-utmp.c b/src/update-utmp/update-utmp.c index 67c5788fe..a62c538ec 100644 --- a/src/update-utmp/update-utmp.c +++ b/src/update-utmp/update-utmp.c @@ -25,68 +25,47 @@ #include #include -#include - #ifdef HAVE_AUDIT #include #endif +#include "sd-bus.h" + #include "log.h" #include "macro.h" #include "util.h" #include "special.h" #include "utmp-wtmp.h" -#include "dbus-common.h" +#include "bus-util.h" +#include "bus-error.h" typedef struct Context { - DBusConnection *bus; + sd_bus *bus; #ifdef HAVE_AUDIT int audit_fd; #endif } Context; static usec_t get_startup_time(Context *c) { - const char - *interface = "org.freedesktop.systemd1.Manager", - *property = "StartupTimestamp"; - + _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; usec_t t = 0; - DBusMessage *reply = NULL; - DBusMessageIter iter, sub; + int r; assert(c); - if (bus_method_call_with_reply ( + r = sd_bus_get_property_trivial( c->bus, "org.freedesktop.systemd1", "/org/freedesktop/systemd1", - "org.freedesktop.DBus.Properties", - "Get", - &reply, - NULL, - DBUS_TYPE_STRING, &interface, - DBUS_TYPE_STRING, &property, - DBUS_TYPE_INVALID)) - goto finish; - - if (!dbus_message_iter_init(reply, &iter) || - dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_VARIANT) { - log_error("Failed to parse reply."); - goto finish; - } - - dbus_message_iter_recurse(&iter, &sub); - - if (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_UINT64) { - log_error("Failed to parse reply."); - goto finish; + "org.freedesktop.systemd1.Manager", + "UserspaceTimestamp", + &error, + 't', &t); + if (r < 0) { + log_error("Failed to get timestamp: %s", bus_error_message(&error, -r)); + return 0; } - dbus_message_iter_get_basic(&sub, &t); - -finish: - if (reply) - dbus_message_unref(reply); return t; } @@ -104,97 +83,56 @@ static int get_current_runlevel(Context *c) { { '3', SPECIAL_RUNLEVEL3_TARGET }, { '4', SPECIAL_RUNLEVEL4_TARGET }, { '2', SPECIAL_RUNLEVEL2_TARGET }, - { 'S', SPECIAL_RESCUE_TARGET }, + { '1', SPECIAL_RESCUE_TARGET }, }; - const char - *interface = "org.freedesktop.systemd1.Unit", - *property = "ActiveState"; - DBusMessage *reply = NULL; - int r = 0; + _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL; + int r; unsigned i; - DBusError error; assert(c); - dbus_error_init(&error); - for (i = 0; i < ELEMENTSOF(table); i++) { - const char *path = NULL, *state; - DBusMessageIter iter, sub; + _cleanup_bus_message_unref_ sd_bus_message *reply = NULL; + _cleanup_free_ char *state = NULL; + const char *path = NULL; - r = bus_method_call_with_reply ( + r = sd_bus_call_method( c->bus, "org.freedesktop.systemd1", "/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", - "GetUnit", + "LoadUnit", + &error, &reply, - NULL, - DBUS_TYPE_STRING, &table[i].special, - DBUS_TYPE_INVALID); - if (r == -ENOMEM) - goto finish; - if (r) + "s", table[i].special); + if (r < 0) { + log_warning("Failed to get runlevel: %s", bus_error_message(&error, -r)); continue; - - if (!dbus_message_get_args(reply, &error, - DBUS_TYPE_OBJECT_PATH, &path, - DBUS_TYPE_INVALID)) { - log_error("Failed to parse reply: %s", bus_error_message(&error)); - r = -EIO; - goto finish; } - dbus_message_unref(reply); - r = bus_method_call_with_reply ( + r = sd_bus_message_read(reply, "o", &path); + if (r < 0) + return bus_log_parse_error(r); + + r = sd_bus_get_property_string( c->bus, "org.freedesktop.systemd1", path, - "org.freedesktop.DBus.Properties", - "Get", - &reply, - NULL, - DBUS_TYPE_STRING, &interface, - DBUS_TYPE_STRING, &property, - DBUS_TYPE_INVALID); - if (r) - goto finish; - - if (!dbus_message_iter_init(reply, &iter) || - dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_VARIANT) { - log_error("Failed to parse reply."); - r = -EIO; - goto finish; - } - - dbus_message_iter_recurse(&iter, &sub); - - if (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_STRING) { - log_error("Failed to parse reply."); - r = -EIO; - goto finish; + "org.freedesktop.systemd1.Unit", + "ActiveState", + &error, + &state); + if (r < 0) { + log_warning("Failed to get state: %s", bus_error_message(&error, -r)); + return r; } - dbus_message_iter_get_basic(&sub, &state); - if (streq(state, "active") || streq(state, "reloading")) - r = table[i].runlevel; - - dbus_message_unref(reply); - reply = NULL; - - if (r) - break; + return table[i].runlevel; } -finish: - if (reply) - dbus_message_unref(reply); - - dbus_error_free(&error); - - return r; + return 0; } static int on_reboot(Context *c) { @@ -312,12 +250,8 @@ static int on_runlevel(Context *c) { int main(int argc, char *argv[]) { int r; - DBusError error; - Context c; + Context c = {}; - dbus_error_init(&error); - - zero(c); #ifdef HAVE_AUDIT c.audit_fd = -1; #endif @@ -345,9 +279,9 @@ int main(int argc, char *argv[]) { errno != EAFNOSUPPORT && errno != EPROTONOSUPPORT) log_error("Failed to connect to audit log: %m"); #endif - - if (bus_connect(DBUS_BUS_SYSTEM, &c.bus, NULL, &error) < 0) { - log_error("Failed to get D-Bus connection: %s", bus_error_message(&error)); + r = bus_open_system_systemd(&c.bus); + if (r < 0) { + log_error("Failed to get D-Bus connection: %s", strerror(-r)); r = -EIO; goto finish; } @@ -373,14 +307,8 @@ finish: audit_close(c.audit_fd); #endif - if (c.bus) { - dbus_connection_flush(c.bus); - dbus_connection_close(c.bus); - dbus_connection_unref(c.bus); - } - - dbus_error_free(&error); - dbus_shutdown(); + if (c.bus) + sd_bus_unref(c.bus); return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; }