X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Flogin%2Ftest-inhibit.c;h=a3cf9d293b4d5c1b39040d5f50a7da6e66a1b6a6;hp=7b6deffc3ede7406bcc1d3fd7adba5ba2adf136d;hb=a5c28308446550e0966825472babdf123fc976ac;hpb=4943c1c94ba751c98763f4232b4350481b22c90a diff --git a/src/login/test-inhibit.c b/src/login/test-inhibit.c index 7b6deffc3..a3cf9d293 100644 --- a/src/login/test-inhibit.c +++ b/src/login/test-inhibit.c @@ -1,5 +1,3 @@ -/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ - /*** This file is part of systemd. @@ -21,117 +19,92 @@ #include -#include +#include "sd-bus.h" +#include "bus-util.h" +#include "fd-util.h" #include "macro.h" #include "util.h" -#include "dbus-common.h" -static int inhibit(DBusConnection *bus, const char *what) { - DBusMessage *m, *reply; - DBusError error; +static int inhibit(sd_bus *bus, const char *what) { + _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL; + _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; const char *who = "Test Tool", *reason = "Just because!", *mode = "block"; int fd; + int r; - dbus_error_init(&error); - - m = dbus_message_new_method_call( + r = sd_bus_call_method(bus, "org.freedesktop.login1", "/org/freedesktop/login1", "org.freedesktop.login1.Manager", - "Inhibit"); - assert(m); - - assert_se(dbus_message_append_args(m, - DBUS_TYPE_STRING, &what, - DBUS_TYPE_STRING, &who, - DBUS_TYPE_STRING, &reason, - DBUS_TYPE_STRING, &mode, - DBUS_TYPE_INVALID)); + "Inhibit", + &error, + &reply, + "ssss", what, who, reason, mode); + assert_se(r >= 0); - reply = dbus_connection_send_with_reply_and_block(bus, m, -1, &error); - assert(reply); + r = sd_bus_message_read_basic(reply, SD_BUS_TYPE_UNIX_FD, &fd); + assert_se(r >= 0); + assert_se(fd >= 0); - assert(dbus_message_get_args(reply, &error, - DBUS_TYPE_UNIX_FD, &fd, - DBUS_TYPE_INVALID)); - - dbus_message_unref(m); - dbus_message_unref(reply); - - return fd; + return dup(fd); } -static void print_inhibitors(DBusConnection *bus) { - DBusMessage *m, *reply; - DBusError error; +static void print_inhibitors(sd_bus *bus) { + _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL; + _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL; + const char *what, *who, *why, *mode; + uint32_t uid, pid; unsigned n = 0; - DBusMessageIter iter, sub, sub2; - - dbus_error_init(&error); + int r; - m = dbus_message_new_method_call( + r = sd_bus_call_method(bus, "org.freedesktop.login1", "/org/freedesktop/login1", "org.freedesktop.login1.Manager", - "ListInhibitors"); - assert(m); + "ListInhibitors", + &error, + &reply, + ""); + assert_se(r >= 0); - reply = dbus_connection_send_with_reply_and_block(bus, m, -1, &error); - assert(reply); + r = sd_bus_message_enter_container(reply, SD_BUS_TYPE_ARRAY, "(ssssuu)"); + assert_se(r >= 0); - assert(dbus_message_iter_init(reply, &iter)); - dbus_message_iter_recurse(&iter, &sub); - - while (dbus_message_iter_get_arg_type(&sub) != DBUS_TYPE_INVALID) { - const char *what, *who, *why, *mode; - dbus_uint32_t uid, pid; - - dbus_message_iter_recurse(&sub, &sub2); - - assert_se(bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &what, true) >= 0); - assert_se(bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &who, true) >= 0); - assert_se(bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &why, true) >= 0); - assert_se(bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_STRING, &mode, true) >= 0); - assert_se(bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_UINT32, &uid, true) >= 0); - assert_se(bus_iter_get_basic_and_next(&sub2, DBUS_TYPE_UINT32, &pid, false) >= 0); - - printf("what=<%s> who=<%s> why=<%s> mode=<%s> uid=<%lu> pid=<%lu>\n", - what, who, why, mode, (unsigned long) uid, (unsigned long) pid); - - dbus_message_iter_next(&sub); + while ((r = sd_bus_message_read(reply, "(ssssuu)", &what, &who, &why, &mode, &uid, &pid)) > 0) { + printf("what=<%s> who=<%s> why=<%s> mode=<%s> uid=<%"PRIu32"> pid=<%"PRIu32">\n", + what, who, why, mode, uid, pid); n++; } + assert_se(r >= 0); printf("%u inhibitors\n", n); - - dbus_message_unref(m); - dbus_message_unref(reply); } int main(int argc, char*argv[]) { - DBusConnection *bus; + _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL; int fd1, fd2; + int r; - bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, NULL); - assert(bus); + r = sd_bus_open_system(&bus); + assert_se(r >= 0); print_inhibitors(bus); fd1 = inhibit(bus, "sleep"); - assert(fd1 >= 0); + assert_se(fd1 >= 0); print_inhibitors(bus); fd2 = inhibit(bus, "idle:shutdown"); - assert(fd2 >= 0); + assert_se(fd2 >= 0); print_inhibitors(bus); - close_nointr_nofail(fd1); + safe_close(fd1); sleep(1); print_inhibitors(bus); - close_nointr_nofail(fd2); + safe_close(fd2); sleep(1); print_inhibitors(bus);