src/libelogind/sd-bus/test-bus-match.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2013 Lennart Poettering
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include "bus-match.h"
  21 #include "bus-message.h"
  22 #include "bus-slot.h"
  23 #include "bus-util.h"
  24 #include "log.h"
  25 #include "macro.h"
  26
  27 static bool mask[32];
  28
  29 static int filter(sd_bus_message *m, void *userdata, sd_bus_error *ret_error) {
  30         log_info("Ran %u", PTR_TO_UINT(userdata));
  31         assert_se(PTR_TO_UINT(userdata) < ELEMENTSOF(mask));
  32         mask[PTR_TO_UINT(userdata)] = true;
  33         return 0;
  34 }
  35
  36 static bool mask_contains(unsigned a[], unsigned n) {
  37         unsigned i, j;
  38
  39         for (i = 0; i < ELEMENTSOF(mask); i++) {
  40                 bool found = false;
  41
  42                 for (j = 0; j < n; j++)
  43                         if (a[j] == i) {
  44                                 found = true;
  45                                 break;
  46                         }
  47
  48                 if (found != mask[i])
  49                         return false;
  50         }
  51
  52         return true;
  53 }
  54
  55 static int match_add(sd_bus_slot *slots, struct bus_match_node *root, const char *match, int value) {
  56         struct bus_match_component *components = NULL;
  57         unsigned n_components = 0;
  58         sd_bus_slot *s;
  59         int r;
  60
  61         s = slots + value;
  62         zero(*s);
  63
  64         r = bus_match_parse(match, &components, &n_components);
  65         if (r < 0)
  66                 return r;
  67
  68         s->userdata = INT_TO_PTR(value);
  69         s->match_callback.callback = filter;
  70
  71         r = bus_match_add(root, components, n_components, &s->match_callback);
  72         bus_match_parse_free(components, n_components);
  73
  74         return r;
  75 }
  76
  77 static void test_match_scope(const char *match, enum bus_match_scope scope) {
  78         struct bus_match_component *components = NULL;
  79         unsigned n_components = 0;
  80
  81         assert_se(bus_match_parse(match, &components, &n_components) >= 0);
  82         assert_se(bus_match_get_scope(components, n_components) == scope);
  83         bus_match_parse_free(components, n_components);
  84 }
  85
  86 int main(int argc, char *argv[]) {
  87         struct bus_match_node root = {
  88                 .type = BUS_MATCH_ROOT,
  89         };
  90
  91         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
  92         _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
  93         enum bus_match_node_type i;
  94         sd_bus_slot slots[19];
  95         int r;
  96
  97         r = sd_bus_open_system(&bus);
  98         if (r < 0)
  99                 return EXIT_TEST_SKIP;
 100
 101         assert_se(match_add(slots, &root, "arg2='wal\\'do',sender='foo',type='signal',interface='bar.x',", 1) >= 0);
 102         assert_se(match_add(slots, &root, "arg2='wal\\'do2',sender='foo',type='signal',interface='bar.x',", 2) >= 0);
 103         assert_se(match_add(slots, &root, "arg3='test',sender='foo',type='signal',interface='bar.x',", 3) >= 0);
 104         assert_se(match_add(slots, &root, "arg3='test',sender='foo',type='method_call',interface='bar.x',", 4) >= 0);
 105         assert_se(match_add(slots, &root, "", 5) >= 0);
 106         assert_se(match_add(slots, &root, "interface='quux.x'", 6) >= 0);
 107         assert_se(match_add(slots, &root, "interface='bar.x'", 7) >= 0);
 108         assert_se(match_add(slots, &root, "member='waldo',path='/foo/bar'", 8) >= 0);
 109         assert_se(match_add(slots, &root, "path='/foo/bar'", 9) >= 0);
 110         assert_se(match_add(slots, &root, "path_namespace='/foo'", 10) >= 0);
 111         assert_se(match_add(slots, &root, "path_namespace='/foo/quux'", 11) >= 0);
 112         assert_se(match_add(slots, &root, "arg1='two'", 12) >= 0);
 113         assert_se(match_add(slots, &root, "member='waldo',arg2path='/prefix/'", 13) >= 0);
 114         assert_se(match_add(slots, &root, "member=waldo,path='/foo/bar',arg3namespace='prefix'", 14) >= 0);
 115         assert_se(match_add(slots, &root, "arg4has='pi'", 15) >= 0);
 116         assert_se(match_add(slots, &root, "arg4has='pa'", 16) >= 0);
 117         assert_se(match_add(slots, &root, "arg4has='po'", 17) >= 0);
 118         assert_se(match_add(slots, &root, "arg4='pi'", 18) >= 0);
 119
 120         bus_match_dump(&root, 0);
 121
 122         assert_se(sd_bus_message_new_signal(bus, &m, "/foo/bar", "bar.x", "waldo") >= 0);
 123         assert_se(sd_bus_message_append(m, "ssssas", "one", "two", "/prefix/three", "prefix.four", 3, "pi", "pa", "po") >= 0);
 124         assert_se(bus_message_seal(m, 1, 0) >= 0);
 125
 126         zero(mask);
 127         assert_se(bus_match_run(NULL, &root, m) == 0);
 128         assert_se(mask_contains((unsigned[]) { 9, 8, 7, 5, 10, 12, 13, 14, 15, 16, 17 }, 11));
 129
 130         assert_se(bus_match_remove(&root, &slots[8].match_callback) >= 0);
 131         assert_se(bus_match_remove(&root, &slots[13].match_callback) >= 0);
 132
 133         bus_match_dump(&root, 0);
 134
 135         zero(mask);
 136         assert_se(bus_match_run(NULL, &root, m) == 0);
 137         assert_se(mask_contains((unsigned[]) { 9, 5, 10, 12, 14, 7, 15, 16, 17 }, 9));
 138
 139         for (i = 0; i < _BUS_MATCH_NODE_TYPE_MAX; i++) {
 140                 char buf[32];
 141                 const char *x;
 142
 143                 assert_se(x = bus_match_node_type_to_string(i, buf, sizeof(buf)));
 144
 145                 if (i >= BUS_MATCH_MESSAGE_TYPE)
 146                         assert_se(bus_match_node_type_from_string(x, strlen(x)) == i);
 147         }
 148
 149         bus_match_free(&root);
 150
 151         test_match_scope("interface='foobar'", BUS_MATCH_GENERIC);
 152         test_match_scope("", BUS_MATCH_GENERIC);
 153         test_match_scope("interface='org.freedesktop.DBus.Local'", BUS_MATCH_LOCAL);
 154         test_match_scope("sender='org.freedesktop.DBus.Local'", BUS_MATCH_LOCAL);
 155         test_match_scope("member='gurke',path='/org/freedesktop/DBus/Local'", BUS_MATCH_LOCAL);
 156         test_match_scope("arg2='piep',sender='org.freedesktop.DBus',member='waldo'", BUS_MATCH_DRIVER);
 157
 158         return 0;
 159 }