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

   1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
   2
   3 #include <unistd.h>
   4
   5 #include "sd-bus.h"
   6
   7 #include "bus-internal.h"
   8 #include "bus-message.h"
   9 #include "process-util.h"
  10 #include "tests.h"
  11
  12 static bool use_system_bus = false;
  13
  14 static void test_bus_new(void) {
  15         _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL;
  16
  17         assert_se(sd_bus_new(&bus) == 0);
  18         assert_se(bus->n_ref == 1);
  19 }
  20
  21 static void test_bus_fork(void) {
  22         _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL;
  23         int r;
  24
  25         assert_se(sd_bus_new(&bus) == 0);
  26         assert_se(bus->n_ref == 1);
  27
  28         /* Check that after a fork the cleanup functions return NULL */
  29         r = safe_fork("(bus-fork-test)", FORK_WAIT|FORK_LOG, NULL);
  30         if (r == 0) {
  31                 assert_se(bus);
  32                 ASSERT_RETURN_EXPECTED_SE(sd_bus_is_ready(bus) == -ECHILD);
  33                 assert_se(sd_bus_flush_close_unref(bus) == NULL);
  34                 assert_se(sd_bus_close_unref(bus) == NULL);
  35                 assert_se(sd_bus_unref(bus) == NULL);
  36                 sd_bus_close(bus);
  37                 assert_se(bus->n_ref == 1);
  38                 _exit(EXIT_SUCCESS);
  39         }
  40
  41         assert_se(r >= 0);
  42         assert_se(bus->n_ref == 1);
  43 }
  44
  45 static int test_bus_open(void) {
  46         _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
  47         int r;
  48
  49         r = sd_bus_open_user(&bus);
  50         if (IN_SET(r, -ECONNREFUSED, -ENOENT, -ENOMEDIUM)) {
  51                 r = sd_bus_open_system(&bus);
  52                 if (IN_SET(r, -ECONNREFUSED, -ENOENT))
  53                         return r;
  54                 use_system_bus = true;
  55         }
  56
  57         assert_se(r >= 0);
  58         assert_se(bus->n_ref >= 1); /* we send a hello message when opening, so the count is above 1 */
  59
  60         return 0;
  61 }
  62
  63 static void test_bus_new_method_call(void) {
  64         sd_bus *bus = NULL;
  65         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
  66
  67         assert_se(use_system_bus ? sd_bus_open_system(&bus) >= 0 : sd_bus_open_user(&bus) >= 0);
  68
  69         assert_se(sd_bus_message_new_method_call(bus, &m, "a.service.name", "/an/object/path", "an.interface.name", "AMethodName") >= 0);
  70
  71         assert_se(m->n_ref == 1); /* We hold the only reference to the message */
  72         assert_se(bus->n_ref >= 2);
  73         sd_bus_flush_close_unref(bus);
  74         assert_se(m->n_ref == 1);
  75 }
  76
  77 static void test_bus_new_signal(void) {
  78         sd_bus *bus = NULL;
  79         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
  80
  81         assert_se(use_system_bus ? sd_bus_open_system(&bus) >= 0 : sd_bus_open_user(&bus) >= 0);
  82
  83         assert_se(sd_bus_message_new_signal(bus, &m, "/an/object/path", "an.interface.name", "Name") >= 0);
  84
  85         assert_se(m->n_ref == 1); /* We hold the only reference to the message */
  86         assert_se(bus->n_ref >= 2);
  87         sd_bus_flush_close_unref(bus);
  88         assert_se(m->n_ref == 1);
  89 }
  90
  91 int main(int argc, char **argv) {
  92         test_setup_logging(LOG_INFO);
  93
  94         test_bus_new();
  95         test_bus_fork();
  96
  97         if (test_bus_open() < 0)
  98                 return log_tests_skipped("Failed to connect to bus");
  99
 100         test_bus_new_method_call();
 101         test_bus_new_signal();
 102
 103         return EXIT_SUCCESS;
 104 }