src/libsystemd/sd-bus/test-bus-cleanup.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <stdio.h>
   4
   5 #include "sd-bus.h"
   6
   7 #include "bus-internal.h"
   8 #include "bus-message.h"
   9 #include "bus-util.h"
  10 #include "refcnt.h"
  11 #include "tests.h"
  12
  13 static bool use_system_bus = false;
  14
  15 static void test_bus_new(void) {
  16         _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL;
  17
  18         assert_se(sd_bus_new(&bus) == 0);
  19         printf("after new: refcount %u\n", REFCNT_GET(bus->n_ref));
  20 }
  21
  22 static int test_bus_open(void) {
  23         _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
  24         int r;
  25
  26         r = sd_bus_open_user(&bus);
  27         if (IN_SET(r, -ECONNREFUSED, -ENOENT)) {
  28                 r = sd_bus_open_system(&bus);
  29                 if (IN_SET(r, -ECONNREFUSED, -ENOENT))
  30                         return r;
  31                 use_system_bus = true;
  32         }
  33
  34         assert_se(r >= 0);
  35         printf("after open: refcount %u\n", REFCNT_GET(bus->n_ref));
  36
  37         return 0;
  38 }
  39
  40 static void test_bus_new_method_call(void) {
  41         sd_bus *bus = NULL;
  42         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
  43
  44         assert_se(use_system_bus ? sd_bus_open_system(&bus) >= 0 : sd_bus_open_user(&bus) >= 0);
  45
  46         assert_se(sd_bus_message_new_method_call(bus, &m, "a.service.name", "/an/object/path", "an.interface.name", "AMethodName") >= 0);
  47
  48         printf("after message_new_method_call: refcount %u\n", REFCNT_GET(bus->n_ref));
  49
  50         sd_bus_flush_close_unref(bus);
  51         printf("after bus_flush_close_unref: refcount %u\n", m->n_ref);
  52 }
  53
  54 static void test_bus_new_signal(void) {
  55         sd_bus *bus = NULL;
  56         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
  57
  58         assert_se(use_system_bus ? sd_bus_open_system(&bus) >= 0 : sd_bus_open_user(&bus) >= 0);
  59
  60         assert_se(sd_bus_message_new_signal(bus, &m, "/an/object/path", "an.interface.name", "Name") >= 0);
  61
  62         printf("after message_new_signal: refcount %u\n", REFCNT_GET(bus->n_ref));
  63
  64         sd_bus_flush_close_unref(bus);
  65         printf("after bus_flush_close_unref: refcount %u\n", m->n_ref);
  66 }
  67
  68 int main(int argc, char **argv) {
  69         test_setup_logging(LOG_INFO);
  70
  71         test_bus_new();
  72
  73         if (test_bus_open() < 0)
  74                 return log_tests_skipped("Failed to connect to bus");
  75
  76         test_bus_new_method_call();
  77         test_bus_new_signal();
  78
  79         return EXIT_SUCCESS;
  80 }