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

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