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

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 /***
   3 ***/
   4
   5 #include <pthread.h>
   6 #include <stdlib.h>
   7
   8 #include "sd-bus.h"
   9
  10 #include "alloc-util.h"
  11 #include "bus-dump.h"
  12 #include "bus-internal.h"
  13 #include "bus-message.h"
  14 #include "bus-util.h"
  15 #include "log.h"
  16 #include "macro.h"
  17 #include "strv.h"
  18 #include "util.h"
  19
  20 struct context {
  21         int fds[2];
  22         bool quit;
  23         char *something;
  24         char *automatic_string_property;
  25         uint32_t automatic_integer_property;
  26 };
  27
  28 static int something_handler(sd_bus_message *m, void *userdata, sd_bus_error *error) {
  29         struct context *c = userdata;
  30         const char *s;
  31         char *n = NULL;
  32         int r;
  33
  34         r = sd_bus_message_read(m, "s", &s);
  35         assert_se(r > 0);
  36
  37         n = strjoin("<<<", s, ">>>");
  38         assert_se(n);
  39
  40         free(c->something);
  41         c->something = n;
  42
  43         log_info("AlterSomething() called, got %s, returning %s", s, n);
  44
  45         /* This should fail, since the return type doesn't match */
  46         assert_se(sd_bus_reply_method_return(m, "u", 4711) == -ENOMSG);
  47
  48         r = sd_bus_reply_method_return(m, "s", n);
  49         assert_se(r >= 0);
  50
  51         return 1;
  52 }
  53
  54 static int exit_handler(sd_bus_message *m, void *userdata, sd_bus_error *error) {
  55         struct context *c = userdata;
  56         int r;
  57
  58         c->quit = true;
  59
  60         log_info("Exit called");
  61
  62         r = sd_bus_reply_method_return(m, "");
  63         assert_se(r >= 0);
  64
  65         return 1;
  66 }
  67
  68 static int get_handler(sd_bus *bus, const char *path, const char *interface, const char *property, sd_bus_message *reply, void *userdata, sd_bus_error *error) {
  69         struct context *c = userdata;
  70         int r;
  71
  72         log_info("property get for %s called, returning \"%s\".", property, c->something);
  73
  74         r = sd_bus_message_append(reply, "s", c->something);
  75         assert_se(r >= 0);
  76
  77         return 1;
  78 }
  79
  80 static int set_handler(sd_bus *bus, const char *path, const char *interface, const char *property, sd_bus_message *value, void *userdata, sd_bus_error *error) {
  81         struct context *c = userdata;
  82         const char *s;
  83         char *n;
  84         int r;
  85
  86         log_info("property set for %s called", property);
  87
  88         r = sd_bus_message_read(value, "s", &s);
  89         assert_se(r >= 0);
  90
  91         n = strdup(s);
  92         assert_se(n);
  93
  94         free(c->something);
  95         c->something = n;
  96
  97         return 1;
  98 }
  99
 100 static int value_handler(sd_bus *bus, const char *path, const char *interface, const char *property, sd_bus_message *reply, void *userdata, sd_bus_error *error) {
 101         _cleanup_free_ char *s = NULL;
 102         const char *x;
 103         int r;
 104
 105         assert_se(asprintf(&s, "object %p, path %s", userdata, path) >= 0);
 106         r = sd_bus_message_append(reply, "s", s);
 107         assert_se(r >= 0);
 108
 109         assert_se(x = startswith(path, "/value/"));
 110
 111         assert_se(PTR_TO_UINT(userdata) == 30);
 112
 113         return 1;
 114 }
 115
 116 static int notify_test(sd_bus_message *m, void *userdata, sd_bus_error *error) {
 117         int r;
 118
 119         assert_se(sd_bus_emit_properties_changed(sd_bus_message_get_bus(m), m->path, "org.freedesktop.systemd.ValueTest", "Value", NULL) >= 0);
 120
 121         r = sd_bus_reply_method_return(m, NULL);
 122         assert_se(r >= 0);
 123
 124         return 1;
 125 }
 126
 127 static int notify_test2(sd_bus_message *m, void *userdata, sd_bus_error *error) {
 128         int r;
 129
 130         assert_se(sd_bus_emit_properties_changed_strv(sd_bus_message_get_bus(m), m->path, "org.freedesktop.systemd.ValueTest", NULL) >= 0);
 131
 132         r = sd_bus_reply_method_return(m, NULL);
 133         assert_se(r >= 0);
 134
 135         return 1;
 136 }
 137
 138 static int emit_interfaces_added(sd_bus_message *m, void *userdata, sd_bus_error *error) {
 139         int r;
 140
 141         assert_se(sd_bus_emit_interfaces_added(sd_bus_message_get_bus(m), "/value/a/x", "org.freedesktop.systemd.ValueTest", NULL) >= 0);
 142
 143         r = sd_bus_reply_method_return(m, NULL);
 144         assert_se(r >= 0);
 145
 146         return 1;
 147 }
 148
 149 static int emit_interfaces_removed(sd_bus_message *m, void *userdata, sd_bus_error *error) {
 150         int r;
 151
 152         assert_se(sd_bus_emit_interfaces_removed(sd_bus_message_get_bus(m), "/value/a/x", "org.freedesktop.systemd.ValueTest", NULL) >= 0);
 153
 154         r = sd_bus_reply_method_return(m, NULL);
 155         assert_se(r >= 0);
 156
 157         return 1;
 158 }
 159
 160 static int emit_object_added(sd_bus_message *m, void *userdata, sd_bus_error *error) {
 161         int r;
 162
 163         assert_se(sd_bus_emit_object_added(sd_bus_message_get_bus(m), "/value/a/x") >= 0);
 164
 165         r = sd_bus_reply_method_return(m, NULL);
 166         assert_se(r >= 0);
 167
 168         return 1;
 169 }
 170
 171 static int emit_object_removed(sd_bus_message *m, void *userdata, sd_bus_error *error) {
 172         int r;
 173
 174         assert_se(sd_bus_emit_object_removed(sd_bus_message_get_bus(m), "/value/a/x") >= 0);
 175
 176         r = sd_bus_reply_method_return(m, NULL);
 177         assert_se(r >= 0);
 178
 179         return 1;
 180 }
 181
 182 static const sd_bus_vtable vtable[] = {
 183         SD_BUS_VTABLE_START(0),
 184         SD_BUS_METHOD("AlterSomething", "s", "s", something_handler, 0),
 185         SD_BUS_METHOD("Exit", "", "", exit_handler, 0),
 186         SD_BUS_WRITABLE_PROPERTY("Something", "s", get_handler, set_handler, 0, 0),
 187         SD_BUS_WRITABLE_PROPERTY("AutomaticStringProperty", "s", NULL, NULL, offsetof(struct context, automatic_string_property), 0),
 188         SD_BUS_WRITABLE_PROPERTY("AutomaticIntegerProperty", "u", NULL, NULL, offsetof(struct context, automatic_integer_property), 0),
 189         SD_BUS_METHOD("NoOperation", NULL, NULL, NULL, 0),
 190         SD_BUS_METHOD("EmitInterfacesAdded", NULL, NULL, emit_interfaces_added, 0),
 191         SD_BUS_METHOD("EmitInterfacesRemoved", NULL, NULL, emit_interfaces_removed, 0),
 192         SD_BUS_METHOD("EmitObjectAdded", NULL, NULL, emit_object_added, 0),
 193         SD_BUS_METHOD("EmitObjectRemoved", NULL, NULL, emit_object_removed, 0),
 194         SD_BUS_VTABLE_END
 195 };
 196
 197 static const sd_bus_vtable vtable2[] = {
 198         SD_BUS_VTABLE_START(0),
 199         SD_BUS_METHOD("NotifyTest", "", "", notify_test, 0),
 200         SD_BUS_METHOD("NotifyTest2", "", "", notify_test2, 0),
 201         SD_BUS_PROPERTY("Value", "s", value_handler, 10, SD_BUS_VTABLE_PROPERTY_EMITS_CHANGE),
 202         SD_BUS_PROPERTY("Value2", "s", value_handler, 10, SD_BUS_VTABLE_PROPERTY_EMITS_INVALIDATION),
 203         SD_BUS_PROPERTY("Value3", "s", value_handler, 10, SD_BUS_VTABLE_PROPERTY_CONST),
 204         SD_BUS_PROPERTY("Value4", "s", value_handler, 10, 0),
 205         SD_BUS_PROPERTY("AnExplicitProperty", "s", NULL, offsetof(struct context, something), SD_BUS_VTABLE_PROPERTY_EXPLICIT|SD_BUS_VTABLE_PROPERTY_EMITS_INVALIDATION),
 206         SD_BUS_VTABLE_END
 207 };
 208
 209 static int enumerator_callback(sd_bus *bus, const char *path, void *userdata, char ***nodes, sd_bus_error *error) {
 210
 211         if (object_path_startswith("/value", path))
 212                 assert_se(*nodes = strv_new("/value/a", "/value/b", "/value/c", NULL));
 213
 214         return 1;
 215 }
 216
 217 static int enumerator2_callback(sd_bus *bus, const char *path, void *userdata, char ***nodes, sd_bus_error *error) {
 218
 219         if (object_path_startswith("/value/a", path))
 220                 assert_se(*nodes = strv_new("/value/a/x", "/value/a/y", "/value/a/z", NULL));
 221
 222         return 1;
 223 }
 224
 225 static void *server(void *p) {
 226         struct context *c = p;
 227         sd_bus *bus = NULL;
 228         sd_id128_t id;
 229         int r;
 230
 231         c->quit = false;
 232
 233         assert_se(sd_id128_randomize(&id) >= 0);
 234
 235         assert_se(sd_bus_new(&bus) >= 0);
 236         assert_se(sd_bus_set_fd(bus, c->fds[0], c->fds[0]) >= 0);
 237         assert_se(sd_bus_set_server(bus, 1, id) >= 0);
 238
 239         assert_se(sd_bus_add_object_vtable(bus, NULL, "/foo", "org.freedesktop.systemd.test", vtable, c) >= 0);
 240         assert_se(sd_bus_add_object_vtable(bus, NULL, "/foo", "org.freedesktop.systemd.test2", vtable, c) >= 0);
 241         assert_se(sd_bus_add_fallback_vtable(bus, NULL, "/value", "org.freedesktop.systemd.ValueTest", vtable2, NULL, UINT_TO_PTR(20)) >= 0);
 242         assert_se(sd_bus_add_node_enumerator(bus, NULL, "/value", enumerator_callback, NULL) >= 0);
 243         assert_se(sd_bus_add_node_enumerator(bus, NULL, "/value/a", enumerator2_callback, NULL) >= 0);
 244         assert_se(sd_bus_add_object_manager(bus, NULL, "/value") >= 0);
 245         assert_se(sd_bus_add_object_manager(bus, NULL, "/value/a") >= 0);
 246
 247         assert_se(sd_bus_start(bus) >= 0);
 248
 249         log_error("Entering event loop on server");
 250
 251         while (!c->quit) {
 252                 log_error("Loop!");
 253
 254                 r = sd_bus_process(bus, NULL);
 255                 if (r < 0) {
 256                         log_error_errno(r, "Failed to process requests: %m");
 257                         goto fail;
 258                 }
 259
 260                 if (r == 0) {
 261                         r = sd_bus_wait(bus, (uint64_t) -1);
 262                         if (r < 0) {
 263                                 log_error_errno(r, "Failed to wait: %m");
 264                                 goto fail;
 265                         }
 266
 267                         continue;
 268                 }
 269         }
 270
 271         r = 0;
 272
 273 fail:
 274         if (bus) {
 275                 sd_bus_flush(bus);
 276                 sd_bus_unref(bus);
 277         }
 278
 279         return INT_TO_PTR(r);
 280 }
 281
 282 static int client(struct context *c) {
 283         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
 284         _cleanup_(sd_bus_unrefp) sd_bus *bus = NULL;
 285         _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
 286         const char *s;
 287         int r;
 288
 289         assert_se(sd_bus_new(&bus) >= 0);
 290         assert_se(sd_bus_set_fd(bus, c->fds[1], c->fds[1]) >= 0);
 291         assert_se(sd_bus_start(bus) >= 0);
 292
 293         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "NoOperation", &error, NULL, NULL);
 294         assert_se(r >= 0);
 295
 296         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "AlterSomething", &error, &reply, "s", "hallo");
 297         assert_se(r >= 0);
 298
 299         r = sd_bus_message_read(reply, "s", &s);
 300         assert_se(r >= 0);
 301         assert_se(streq(s, "<<<hallo>>>"));
 302
 303         sd_bus_message_unref(reply);
 304         reply = NULL;
 305
 306         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Doesntexist", &error, &reply, "");
 307         assert_se(r < 0);
 308         assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_UNKNOWN_METHOD));
 309
 310         sd_bus_error_free(&error);
 311
 312         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "AlterSomething", &error, &reply, "as", 1, "hallo");
 313         assert_se(r < 0);
 314         assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_INVALID_ARGS));
 315
 316         sd_bus_error_free(&error);
 317
 318         r = sd_bus_get_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Something", &error, &reply, "s");
 319         assert_se(r >= 0);
 320
 321         r = sd_bus_message_read(reply, "s", &s);
 322         assert_se(r >= 0);
 323         assert_se(streq(s, "<<<hallo>>>"));
 324
 325         sd_bus_message_unref(reply);
 326         reply = NULL;
 327
 328         r = sd_bus_set_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Something", &error, "s", "test");
 329         assert_se(r >= 0);
 330
 331         r = sd_bus_get_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Something", &error, &reply, "s");
 332         assert_se(r >= 0);
 333
 334         r = sd_bus_message_read(reply, "s", &s);
 335         assert_se(r >= 0);
 336         assert_se(streq(s, "test"));
 337
 338         sd_bus_message_unref(reply);
 339         reply = NULL;
 340
 341         r = sd_bus_set_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "AutomaticIntegerProperty", &error, "u", 815);
 342         assert_se(r >= 0);
 343
 344         assert_se(c->automatic_integer_property == 815);
 345
 346         r = sd_bus_set_property(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "AutomaticStringProperty", &error, "s", "Du Dödel, Du!");
 347         assert_se(r >= 0);
 348
 349         assert_se(streq(c->automatic_string_property, "Du Dödel, Du!"));
 350
 351         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, "");
 352         assert_se(r >= 0);
 353
 354         r = sd_bus_message_read(reply, "s", &s);
 355         assert_se(r >= 0);
 356         fputs(s, stdout);
 357
 358         sd_bus_message_unref(reply);
 359         reply = NULL;
 360
 361         r = sd_bus_get_property(bus, "org.freedesktop.systemd.test", "/value/xuzz", "org.freedesktop.systemd.ValueTest", "Value", &error, &reply, "s");
 362         assert_se(r >= 0);
 363
 364         r = sd_bus_message_read(reply, "s", &s);
 365         assert_se(r >= 0);
 366         log_info("read %s", s);
 367
 368         sd_bus_message_unref(reply);
 369         reply = NULL;
 370
 371         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, "");
 372         assert_se(r >= 0);
 373
 374         r = sd_bus_message_read(reply, "s", &s);
 375         assert_se(r >= 0);
 376         fputs(s, stdout);
 377
 378         sd_bus_message_unref(reply);
 379         reply = NULL;
 380
 381         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, "");
 382         assert_se(r >= 0);
 383
 384         r = sd_bus_message_read(reply, "s", &s);
 385         assert_se(r >= 0);
 386         fputs(s, stdout);
 387
 388         sd_bus_message_unref(reply);
 389         reply = NULL;
 390
 391         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value/a", "org.freedesktop.DBus.Introspectable", "Introspect", &error, &reply, "");
 392         assert_se(r >= 0);
 393
 394         r = sd_bus_message_read(reply, "s", &s);
 395         assert_se(r >= 0);
 396         fputs(s, stdout);
 397
 398         sd_bus_message_unref(reply);
 399         reply = NULL;
 400
 401         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.DBus.Properties", "GetAll", &error, &reply, "s", "");
 402         assert_se(r >= 0);
 403
 404         bus_message_dump(reply, stdout, BUS_MESSAGE_DUMP_WITH_HEADER);
 405
 406         sd_bus_message_unref(reply);
 407         reply = NULL;
 408
 409         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value/a", "org.freedesktop.DBus.Properties", "GetAll", &error, &reply, "s", "org.freedesktop.systemd.ValueTest2");
 410         assert_se(r < 0);
 411         assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_UNKNOWN_INTERFACE));
 412         sd_bus_error_free(&error);
 413
 414         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.DBus.ObjectManager", "GetManagedObjects", &error, &reply, "");
 415         assert_se(r < 0);
 416         assert_se(sd_bus_error_has_name(&error, SD_BUS_ERROR_UNKNOWN_METHOD));
 417         sd_bus_error_free(&error);
 418
 419         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value", "org.freedesktop.DBus.ObjectManager", "GetManagedObjects", &error, &reply, "");
 420         assert_se(r >= 0);
 421
 422         bus_message_dump(reply, stdout, BUS_MESSAGE_DUMP_WITH_HEADER);
 423
 424         sd_bus_message_unref(reply);
 425         reply = NULL;
 426
 427         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value/a", "org.freedesktop.systemd.ValueTest", "NotifyTest", &error, NULL, "");
 428         assert_se(r >= 0);
 429
 430         r = sd_bus_process(bus, &reply);
 431         assert_se(r > 0);
 432
 433         assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.Properties", "PropertiesChanged"));
 434         bus_message_dump(reply, stdout, BUS_MESSAGE_DUMP_WITH_HEADER);
 435
 436         sd_bus_message_unref(reply);
 437         reply = NULL;
 438
 439         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/value/a", "org.freedesktop.systemd.ValueTest", "NotifyTest2", &error, NULL, "");
 440         assert_se(r >= 0);
 441
 442         r = sd_bus_process(bus, &reply);
 443         assert_se(r > 0);
 444
 445         assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.Properties", "PropertiesChanged"));
 446         bus_message_dump(reply, stdout, BUS_MESSAGE_DUMP_WITH_HEADER);
 447
 448         sd_bus_message_unref(reply);
 449         reply = NULL;
 450
 451         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "EmitInterfacesAdded", &error, NULL, "");
 452         assert_se(r >= 0);
 453
 454         r = sd_bus_process(bus, &reply);
 455         assert_se(r > 0);
 456
 457         assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.ObjectManager", "InterfacesAdded"));
 458         bus_message_dump(reply, stdout, BUS_MESSAGE_DUMP_WITH_HEADER);
 459
 460         sd_bus_message_unref(reply);
 461         reply = NULL;
 462
 463         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "EmitInterfacesRemoved", &error, NULL, "");
 464         assert_se(r >= 0);
 465
 466         r = sd_bus_process(bus, &reply);
 467         assert_se(r > 0);
 468
 469         assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.ObjectManager", "InterfacesRemoved"));
 470         bus_message_dump(reply, stdout, BUS_MESSAGE_DUMP_WITH_HEADER);
 471
 472         sd_bus_message_unref(reply);
 473         reply = NULL;
 474
 475         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "EmitObjectAdded", &error, NULL, "");
 476         assert_se(r >= 0);
 477
 478         r = sd_bus_process(bus, &reply);
 479         assert_se(r > 0);
 480
 481         assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.ObjectManager", "InterfacesAdded"));
 482         bus_message_dump(reply, stdout, BUS_MESSAGE_DUMP_WITH_HEADER);
 483
 484         sd_bus_message_unref(reply);
 485         reply = NULL;
 486
 487         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "EmitObjectRemoved", &error, NULL, "");
 488         assert_se(r >= 0);
 489
 490         r = sd_bus_process(bus, &reply);
 491         assert_se(r > 0);
 492
 493         assert_se(sd_bus_message_is_signal(reply, "org.freedesktop.DBus.ObjectManager", "InterfacesRemoved"));
 494         bus_message_dump(reply, stdout, BUS_MESSAGE_DUMP_WITH_HEADER);
 495
 496         sd_bus_message_unref(reply);
 497         reply = NULL;
 498
 499         r = sd_bus_call_method(bus, "org.freedesktop.systemd.test", "/foo", "org.freedesktop.systemd.test", "Exit", &error, NULL, "");
 500         assert_se(r >= 0);
 501
 502         sd_bus_flush(bus);
 503
 504         return 0;
 505 }
 506
 507 int main(int argc, char *argv[]) {
 508         struct context c = {};
 509         pthread_t s;
 510         void *p;
 511         int r, q;
 512
 513         c.automatic_integer_property = 4711;
 514         assert_se(c.automatic_string_property = strdup("dudeldu"));
 515
 516         assert_se(socketpair(AF_UNIX, SOCK_STREAM, 0, c.fds) >= 0);
 517
 518         r = pthread_create(&s, NULL, server, &c);
 519         if (r != 0)
 520                 return -r;
 521
 522         r = client(&c);
 523
 524         q = pthread_join(s, &p);
 525         if (q != 0)
 526                 return -q;
 527
 528         if (r < 0)
 529                 return r;
 530
 531         if (PTR_TO_INT(p) < 0)
 532                 return PTR_TO_INT(p);
 533
 534         free(c.something);
 535         free(c.automatic_string_property);
 536
 537         return EXIT_SUCCESS;
 538 }