]>
Commit | Line | Data |
---|---|---|
53e1b683 | 1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
70cb8b7b | 2 | |
dccca82b LP |
3 | #include <errno.h> |
4 | #include <sys/socket.h> | |
5 | ||
62cc1c55 | 6 | #include "sd-bus.h" |
70cb8b7b LP |
7 | |
8 | #include "macro.h" | |
317bb217 | 9 | #include "tests.h" |
70cb8b7b LP |
10 | |
11 | static bool track_cb_called_x = false; | |
12 | static bool track_cb_called_y = false; | |
13 | ||
14 | static int track_cb_x(sd_bus_track *t, void *userdata) { | |
15 | ||
16 | log_error("TRACK CB X"); | |
17 | ||
18 | assert_se(!track_cb_called_x); | |
19 | track_cb_called_x = true; | |
20 | ||
21 | /* This means b's name disappeared. Let's now disconnect, to make sure the track handling on disconnect works | |
22 | * as it should. */ | |
23 | ||
24 | assert_se(shutdown(sd_bus_get_fd(sd_bus_track_get_bus(t)), SHUT_RDWR) >= 0); | |
25 | return 1; | |
26 | } | |
27 | ||
28 | static int track_cb_y(sd_bus_track *t, void *userdata) { | |
29 | int r; | |
30 | ||
31 | log_error("TRACK CB Y"); | |
32 | ||
33 | assert_se(!track_cb_called_y); | |
34 | track_cb_called_y = true; | |
35 | ||
36 | /* We got disconnected, let's close everything */ | |
37 | ||
38 | r = sd_event_exit(sd_bus_get_event(sd_bus_track_get_bus(t)), EXIT_SUCCESS); | |
39 | assert_se(r >= 0); | |
40 | ||
41 | return 0; | |
42 | } | |
43 | ||
44 | int main(int argc, char *argv[]) { | |
45 | _cleanup_(sd_event_unrefp) sd_event *event = NULL; | |
46 | _cleanup_(sd_bus_track_unrefp) sd_bus_track *x = NULL, *y = NULL; | |
47 | _cleanup_(sd_bus_unrefp) sd_bus *a = NULL, *b = NULL; | |
f9c8c01f | 48 | bool use_system_bus = false; |
70cb8b7b LP |
49 | const char *unique; |
50 | int r; | |
51 | ||
6d7c4033 ZJS |
52 | test_setup_logging(LOG_INFO); |
53 | ||
70cb8b7b LP |
54 | r = sd_event_default(&event); |
55 | assert_se(r >= 0); | |
56 | ||
6349cda2 | 57 | r = sd_bus_open_user(&a); |
ab4a88eb | 58 | if (IN_SET(r, -ECONNREFUSED, -ENOENT, -ENOMEDIUM)) { |
f9c8c01f YW |
59 | r = sd_bus_open_system(&a); |
60 | if (IN_SET(r, -ECONNREFUSED, -ENOENT)) | |
61 | return log_tests_skipped("Failed to connect to bus"); | |
62 | use_system_bus = true; | |
63 | } | |
70cb8b7b LP |
64 | assert_se(r >= 0); |
65 | ||
66 | r = sd_bus_attach_event(a, event, SD_EVENT_PRIORITY_NORMAL); | |
67 | assert_se(r >= 0); | |
68 | ||
f9c8c01f YW |
69 | if (use_system_bus) |
70 | r = sd_bus_open_system(&b); | |
71 | else | |
72 | r = sd_bus_open_user(&b); | |
70cb8b7b LP |
73 | assert_se(r >= 0); |
74 | ||
75 | r = sd_bus_attach_event(b, event, SD_EVENT_PRIORITY_NORMAL); | |
76 | assert_se(r >= 0); | |
77 | ||
78 | /* Watch b's name from a */ | |
79 | r = sd_bus_track_new(a, &x, track_cb_x, NULL); | |
80 | assert_se(r >= 0); | |
81 | ||
82 | r = sd_bus_get_unique_name(b, &unique); | |
83 | assert_se(r >= 0); | |
84 | ||
85 | r = sd_bus_track_add_name(x, unique); | |
86 | assert_se(r >= 0); | |
87 | ||
88 | /* Watch's a's own name from a */ | |
89 | r = sd_bus_track_new(a, &y, track_cb_y, NULL); | |
90 | assert_se(r >= 0); | |
91 | ||
92 | r = sd_bus_get_unique_name(a, &unique); | |
93 | assert_se(r >= 0); | |
94 | ||
95 | r = sd_bus_track_add_name(y, unique); | |
96 | assert_se(r >= 0); | |
97 | ||
98 | /* Now make b's name disappear */ | |
99 | sd_bus_close(b); | |
100 | ||
101 | r = sd_event_loop(event); | |
102 | assert_se(r >= 0); | |
103 | ||
104 | assert_se(track_cb_called_x); | |
105 | assert_se(track_cb_called_y); | |
106 | ||
107 | return 0; | |
108 | } |