busctl: output a single sorted list of names, including activatable and activated

[elogind.git] / src / libsystemd-bus / busctl.c

diff --git a/src/libsystemd-bus/busctl.c b/src/libsystemd-bus/busctl.c
index 4e2e6aff2606e30265f6b24fc7d341bf23f78834..e6c46d78a729f202ec1883b0b636c63b6fd1e291 100644 (file)
--- a/src/libsystemd-bus/busctl.c
+++ b/src/libsystemd-bus/busctl.c
@@ -53,9 +53,15 @@ static void pager_open_if_enabled(void) {
 
 static int list_bus_names(sd_bus *bus, char **argv) {
         _cleanup_strv_free_ char **acquired = NULL, **activatable = NULL;
 
 static int list_bus_names(sd_bus *bus, char **argv) {
         _cleanup_strv_free_ char **acquired = NULL, **activatable = NULL;

+        _cleanup_free_ char **merged = NULL;
+        _cleanup_hashmap_free_ Hashmap *names = NULL;

         char **i;
         int r;
         size_t max_i = 0;
         char **i;
         int r;
         size_t max_i = 0;

+        unsigned n = 0;
+        void *v;
+        char *k;
+        Iterator iterator;

 
         assert(bus);
 
 
         assert(bus);
 


@@ -67,15 +73,37 @@ static int list_bus_names(sd_bus *bus, char **argv) {
 
         pager_open_if_enabled();
 
 
         pager_open_if_enabled();
 

-        strv_sort(acquired);
-        strv_sort(activatable);
+        names = hashmap_new(string_hash_func, string_compare_func);
+        if (!names)
+                return log_oom();

 
 

-        STRV_FOREACH(i, acquired)
+        STRV_FOREACH(i, acquired) {

                 max_i = MAX(max_i, strlen(*i));
 
                 max_i = MAX(max_i, strlen(*i));
 

-        STRV_FOREACH(i, activatable)
+                r = hashmap_put(names, *i, INT_TO_PTR(1));
+                if (r < 0) {
+                        log_error("Failed to add to hashmap: %s", strerror(-r));
+                        return r;
+                }
+        }
+
+        STRV_FOREACH(i, activatable) {

                 max_i = MAX(max_i, strlen(*i));
 
                 max_i = MAX(max_i, strlen(*i));
 

+                r = hashmap_put(names, *i, INT_TO_PTR(2));
+                if (r < 0 && r != -EEXIST) {
+                        log_error("Failed to add to hashmap: %s", strerror(-r));
+                        return r;
+                }
+        }
+
+        merged = new(char*, hashmap_size(names) + 1);
+        HASHMAP_FOREACH_KEY(v, k, names, iterator)
+                merged[n++] = k;
+
+        merged[n] = NULL;
+        strv_sort(merged);
+

         printf("%-*s %*s %-*s %-*s %-*s",
                (int) max_i, "NAME", 10, "PID", 15, "PROCESS", 16, "USER", 20, "CONNECTION");
 
         printf("%-*s %*s %-*s %-*s %-*s",
                (int) max_i, "NAME", 10, "PID", 15, "PROCESS", 16, "USER", 20, "CONNECTION");
 


@@ -84,26 +112,22 @@ static int list_bus_names(sd_bus *bus, char **argv) {
         else
                 putchar('\n');
 
         else
                 putchar('\n');
 

-        STRV_FOREACH(i, activatable) {
+        STRV_FOREACH(i, merged) {
+                _cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;
+                sd_id128_t mid;

 
 

-                /* Skip the bus driver */
-                if (streq(*i, "org.freedesktop.DBus"))
-                        continue;
+                if (hashmap_get(names, *i) == INT_TO_PTR(2)) {
+                        /* Activatable */

 
 

-                if (strv_contains(acquired, *i))
+                        printf("%-*s", (int) max_i, *i);
+                        printf("          - -               -                (activation)        ");
+                        if (arg_no_machine)
+                                putchar('\n');
+                        else
+                                puts(" -");

                         continue;
 
                         continue;
 

-                printf("%-*s", (int) max_i, *i);
-                printf("          - -               -                (activation)        ");
-                if (arg_no_machine)
-                        putchar('\n');
-                else
-                        puts(" -");
-        }
-
-        STRV_FOREACH(i, acquired) {
-                _cleanup_bus_creds_unref_ sd_bus_creds *creds = NULL;
-                sd_id128_t mid;
+                }

 
                 if (arg_no_unique && (*i)[0] == ':')
                         continue;
 
                 if (arg_no_unique && (*i)[0] == ':')
                         continue;