/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ /*** This file is part of systemd. Copyright 2012 Lennart Poettering systemd is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. systemd is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with systemd; If not, see <http://www.gnu.org/licenses/>. ***/ #include <unistd.h> #include <dbus/dbus.h> #include "macro.h" #include "util.h" #include "dbus-common.h" static int inhibit(DBusConnection *bus, const char *what) { DBusMessage *m, *reply; DBusError error; const char *who = "Test Tool", *reason = "Just because!", *mode = "block"; int fd; dbus_error_init(&error); m = dbus_message_new_method_call( "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)); reply = dbus_connection_send_with_reply_and_block(bus, m, -1, &error); assert(reply); 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; } static void print_inhibitors(DBusConnection *bus) { DBusMessage *m, *reply; DBusError error; unsigned n = 0; DBusMessageIter iter, sub, sub2; dbus_error_init(&error); m = dbus_message_new_method_call( "org.freedesktop.login1", "/org/freedesktop/login1", "org.freedesktop.login1.Manager", "ListInhibitors"); assert(m); reply = dbus_connection_send_with_reply_and_block(bus, m, -1, &error); assert(reply); 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); n++; } printf("%u inhibitors\n", n); dbus_message_unref(m); dbus_message_unref(reply); } int main(int argc, char*argv[]) { DBusConnection *bus; int fd1, fd2; bus = dbus_bus_get_private(DBUS_BUS_SYSTEM, NULL); assert(bus); print_inhibitors(bus); fd1 = inhibit(bus, "sleep"); assert(fd1 >= 0); print_inhibitors(bus); fd2 = inhibit(bus, "idle:shutdown"); assert(fd2 >= 0); print_inhibitors(bus); close_nointr_nofail(fd1); sleep(1); print_inhibitors(bus); close_nointr_nofail(fd2); sleep(1); print_inhibitors(bus); return 0; }