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

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