1 /* SPDX-License-Identifier: LGPL-2.1+ */
9 #include "alloc-util.h"
13 #include "path-util.h"
14 #include "random-util.h"
16 #include "socket-util.h"
17 #include "string-util.h"
18 #include "tmpfile-util.h"
20 static int method_foobar(sd_bus_message
*m
, void *userdata
, sd_bus_error
*ret_error
) {
21 log_info("Got Foobar() call.");
23 assert_se(sd_event_exit(sd_bus_get_event(sd_bus_message_get_bus(m
)), 0) >= 0);
24 return sd_bus_reply_method_return(m
, NULL
);
27 static int method_exit(sd_bus_message
*m
, void *userdata
, sd_bus_error
*ret_error
) {
28 log_info("Got Exit() call");
29 assert_se(sd_event_exit(sd_bus_get_event(sd_bus_message_get_bus(m
)), 1) >= 0);
30 return sd_bus_reply_method_return(m
, NULL
);
33 static const sd_bus_vtable vtable
[] = {
34 SD_BUS_VTABLE_START(0),
35 SD_BUS_METHOD("Foobar", NULL
, NULL
, method_foobar
, SD_BUS_VTABLE_UNPRIVILEGED
),
36 SD_BUS_METHOD("Exit", NULL
, NULL
, method_exit
, SD_BUS_VTABLE_UNPRIVILEGED
),
40 static void* thread_server(void *p
) {
41 _cleanup_free_
char *suffixed
= NULL
, *suffixed2
= NULL
, *d
= NULL
;
42 _cleanup_close_
int fd
= -1;
43 union sockaddr_union u
= {};
47 log_debug("Initializing server");
49 /* Let's play some games, by slowly creating the socket directory, and renaming it in the middle */
50 (void) usleep(100 * USEC_PER_MSEC
);
52 assert_se(mkdir_parents(path
, 0755) >= 0);
53 (void) usleep(100 * USEC_PER_MSEC
);
55 d
= dirname_malloc(path
);
57 assert_se(asprintf(&suffixed
, "%s.%" PRIx64
, d
, random_u64()) >= 0);
58 assert_se(rename(d
, suffixed
) >= 0);
59 (void) usleep(100 * USEC_PER_MSEC
);
61 assert_se(asprintf(&suffixed2
, "%s.%" PRIx64
, d
, random_u64()) >= 0);
62 assert_se(symlink(suffixed2
, d
) >= 0);
63 (void) usleep(100 * USEC_PER_MSEC
);
65 assert_se(symlink(basename(suffixed
), suffixed2
) >= 0);
66 (void) usleep(100 * USEC_PER_MSEC
);
68 salen
= sockaddr_un_set_path(&u
.un
, path
);
69 assert_se(salen
>= 0);
71 fd
= socket(AF_UNIX
, SOCK_STREAM
|SOCK_CLOEXEC
, 0);
74 assert_se(bind(fd
, &u
.sa
, salen
) >= 0);
75 usleep(100 * USEC_PER_MSEC
);
77 assert_se(listen(fd
, SOMAXCONN
) >= 0);
78 usleep(100 * USEC_PER_MSEC
);
80 assert_se(touch(path
) >= 0);
81 usleep(100 * USEC_PER_MSEC
);
83 log_debug("Initialized server");
86 _cleanup_(sd_bus_flush_close_unrefp
) sd_bus
*bus
= NULL
;
87 _cleanup_(sd_event_unrefp
) sd_event
*event
= NULL
;
91 assert_se(sd_id128_randomize(&id
) >= 0);
93 assert_se(sd_event_new(&event
) >= 0);
95 bus_fd
= accept4(fd
, NULL
, NULL
, SOCK_NONBLOCK
|SOCK_CLOEXEC
);
96 assert_se(bus_fd
>= 0);
98 log_debug("Accepted server connection");
100 assert_se(sd_bus_new(&bus
) >= 0);
101 assert_se(sd_bus_set_description(bus
, "server") >= 0);
102 assert_se(sd_bus_set_fd(bus
, bus_fd
, bus_fd
) >= 0);
103 assert_se(sd_bus_set_server(bus
, true, id
) >= 0);
104 /* assert_se(sd_bus_set_anonymous(bus, true) >= 0); */
106 assert_se(sd_bus_attach_event(bus
, event
, 0) >= 0);
108 assert_se(sd_bus_add_object_vtable(bus
, NULL
, "/foo", "foo.TestInterface", vtable
, NULL
) >= 0);
110 assert_se(sd_bus_start(bus
) >= 0);
112 assert_se(sd_event_loop(event
) >= 0);
114 assert_se(sd_event_get_exit_code(event
, &code
) >= 0);
120 log_debug("Server done");
125 static void* thread_client1(void *p
) {
126 _cleanup_(sd_bus_error_free
) sd_bus_error error
= SD_BUS_ERROR_NULL
;
127 _cleanup_(sd_bus_flush_close_unrefp
) sd_bus
*bus
= NULL
;
128 const char *path
= p
, *t
;
131 log_debug("Initializing client1");
133 assert_se(sd_bus_new(&bus
) >= 0);
134 assert_se(sd_bus_set_description(bus
, "client1") >= 0);
136 t
= strjoina("unix:path=", path
);
137 assert_se(sd_bus_set_address(bus
, t
) >= 0);
138 assert_se(sd_bus_set_watch_bind(bus
, true) >= 0);
139 assert_se(sd_bus_start(bus
) >= 0);
141 r
= sd_bus_call_method(bus
, "foo.bar", "/foo", "foo.TestInterface", "Foobar", &error
, NULL
, NULL
);
144 log_debug("Client1 done");
149 static int client2_callback(sd_bus_message
*m
, void *userdata
, sd_bus_error
*ret_error
) {
150 assert_se(sd_bus_message_is_method_error(m
, NULL
) == 0);
151 assert_se(sd_event_exit(sd_bus_get_event(sd_bus_message_get_bus(m
)), 0) >= 0);
155 static void* thread_client2(void *p
) {
156 _cleanup_(sd_bus_flush_close_unrefp
) sd_bus
*bus
= NULL
;
157 _cleanup_(sd_event_unrefp
) sd_event
*event
= NULL
;
158 const char *path
= p
, *t
;
160 log_debug("Initializing client2");
162 assert_se(sd_event_new(&event
) >= 0);
163 assert_se(sd_bus_new(&bus
) >= 0);
164 assert_se(sd_bus_set_description(bus
, "client2") >= 0);
166 t
= strjoina("unix:path=", path
);
167 assert_se(sd_bus_set_address(bus
, t
) >= 0);
168 assert_se(sd_bus_set_watch_bind(bus
, true) >= 0);
169 assert_se(sd_bus_attach_event(bus
, event
, 0) >= 0);
170 assert_se(sd_bus_start(bus
) >= 0);
172 assert_se(sd_bus_call_method_async(bus
, NULL
, "foo.bar", "/foo", "foo.TestInterface", "Foobar", client2_callback
, NULL
, NULL
) >= 0);
174 assert_se(sd_event_loop(event
) >= 0);
176 log_debug("Client2 done");
181 static void request_exit(const char *path
) {
182 _cleanup_(sd_bus_flush_close_unrefp
) sd_bus
*bus
= NULL
;
185 assert_se(sd_bus_new(&bus
) >= 0);
187 t
= strjoina("unix:path=", path
);
188 assert_se(sd_bus_set_address(bus
, t
) >= 0);
189 assert_se(sd_bus_set_watch_bind(bus
, true) >= 0);
190 assert_se(sd_bus_set_description(bus
, "request-exit") >= 0);
191 assert_se(sd_bus_start(bus
) >= 0);
193 assert_se(sd_bus_call_method(bus
, "foo.bar", "/foo", "foo.TestInterface", "Exit", NULL
, NULL
, NULL
) >= 0);
196 int main(int argc
, char *argv
[]) {
197 _cleanup_(rm_rf_physical_and_freep
) char *d
= NULL
;
198 pthread_t server
, client1
, client2
;
201 log_set_max_level(LOG_DEBUG
);
203 /* We use /dev/shm here rather than /tmp, since some weird distros might set up /tmp as some weird fs that
204 * doesn't support inotify properly. */
205 assert_se(mkdtemp_malloc("/dev/shm/systemd-watch-bind-XXXXXX", &d
) >= 0);
207 path
= strjoina(d
, "/this/is/a/socket");
209 assert_se(pthread_create(&server
, NULL
, thread_server
, path
) == 0);
210 assert_se(pthread_create(&client1
, NULL
, thread_client1
, path
) == 0);
211 assert_se(pthread_create(&client2
, NULL
, thread_client2
, path
) == 0);
213 assert_se(pthread_join(client1
, NULL
) == 0);
214 assert_se(pthread_join(client2
, NULL
) == 0);
218 assert_se(pthread_join(server
, NULL
) == 0);