Classify processes from sessions into cgroups

[elogind.git] / src / cgroups-agent / cgroups-agent.c

diff --git a/src/cgroups-agent/cgroups-agent.c b/src/cgroups-agent/cgroups-agent.c
index 7a6173e2a26fed3d5f03226036a9c80150ec315e..47a526b87f6be7cd0c8ed011524629d4c6f40cbc 100644 (file)
--- a/src/cgroups-agent/cgroups-agent.c
+++ b/src/cgroups-agent/cgroups-agent.c
@@ -19,83 +19,67 @@
   along with systemd; If not, see <http://www.gnu.org/licenses/>.
 ***/
 
-#include <dbus/dbus.h>
-
 #include <stdlib.h>
 
+#include "sd-bus.h"
 #include "log.h"
-#include "dbus-common.h"
+#include "bus-util.h"
 
 int main(int argc, char *argv[]) {
-        DBusError error;
-        DBusConnection *bus = NULL;
-        DBusMessage *m = NULL;
-        int r = EXIT_FAILURE;
-
-        dbus_error_init(&error);
+        _cleanup_bus_flush_close_unref_ sd_bus *bus = NULL;
+        int r;
 
         if (argc != 2) {
                 log_error("Incorrect number of arguments.");
-                goto finish;
+                return EXIT_FAILURE;
         }
 
         log_set_target(LOG_TARGET_AUTO);
         log_parse_environment();
         log_open();
 
+#if 0
         /* We send this event to the private D-Bus socket and then the
          * system instance will forward this to the system bus. We do
          * this to avoid an activation loop when we start dbus when we
          * are called when the dbus service is shut down. */
 
-        if (!(bus = dbus_connection_open_private("unix:path=/run/systemd/private", &error))) {
-#ifndef LEGACY
-                dbus_error_free(&error);
-
-                /* Retry with the pre v21 socket name, to ease upgrades */
-                if (!(bus = dbus_connection_open_private("unix:abstract=/org/freedesktop/systemd1/private", &error))) {
-#endif
-                        log_error("Failed to get D-Bus connection: %s", bus_error_message(&error));
-                        goto finish;
-                }
-#ifndef LEGACY
-        }
+        r = bus_open_system_systemd(&bus);
+#else
+        /* Unlike in systemd where this has to use a private socket,
+           since elogind doesn't associate control groups with services
+           and doesn't manage the dbus service, we can just use the
+           system bus.  */
+        r = sd_bus_open_system(&bus);
+#endif // 0
+
+        if (r < 0) {
+#if 0
+                /* If we couldn't connect we assume this was triggered
+                 * while systemd got restarted/transitioned from
+                 * initrd to the system, so let's ignore this */
+                log_debug_errno(r, "Failed to get D-Bus connection: %m");
+#else
+                /* If dbus isn't running or responding, there is nothing
+                 * we can do about it. */
+                log_debug_errno(r, "Failed to open system bus: %m");
 #endif
-
-        if (bus_check_peercred(bus) < 0) {
-                log_error("Bus owner not root.");
-                goto finish;
-        }
-
-        if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1/agent", "org.freedesktop.systemd1.Agent", "Released"))) {
-                log_error("Could not allocate signal message.");
-                goto finish;
+                return EXIT_FAILURE;
         }
 
-        if (!dbus_message_append_args(m,
-                                      DBUS_TYPE_STRING, &argv[1],
-                                      DBUS_TYPE_INVALID)) {
-                log_error("Could not attach group information to signal message.");
-                goto finish;
-        }
-
-        if (!dbus_connection_send(bus, m, NULL)) {
-                log_error("Failed to send signal message on private connection.");
-                goto finish;
-        }
-
-        r = EXIT_SUCCESS;
-
-finish:
-        if (bus) {
-                dbus_connection_flush(bus);
-                dbus_connection_close(bus);
-                dbus_connection_unref(bus);
+        r = sd_bus_emit_signal(bus,
+                               "/org/freedesktop/systemd1/agent",
+                               "org.freedesktop.systemd1.Agent",
+                               "Released",
+                               "s", argv[1]);
+        if (r < 0) {
+#if 0
+                log_debug_errno(r, "Failed to send signal message on private connection: %m");
+#else
+                log_debug_errno(r, "Failed to send signal message: %m");
+#endif
+                return EXIT_FAILURE;
         }
 
-        if (m)
-                dbus_message_unref(m);
-
-        dbus_error_free(&error);
-        return r;
+        return EXIT_SUCCESS;
 }