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