src/libsystemd-bus/test-bus-marshal.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2013 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <assert.h>
  23 #include <stdlib.h>
  24 #include <byteswap.h>
  25
  26 #ifdef HAVE_GLIB
  27 #include <gio/gio.h>
  28 #endif
  29
  30 #include <dbus.h>
  31
  32 #include "log.h"
  33 #include "util.h"
  34
  35 #include "sd-bus.h"
  36 #include "bus-message.h"
  37 #include "bus-util.h"
  38
  39 int main(int argc, char *argv[]) {
  40         _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
  41         int r, boolean;
  42         const char *x, *y, *z, *a, *b, *c, *d;
  43         uint8_t u, v;
  44         void *buffer = NULL;
  45         size_t sz;
  46         char *h;
  47         const int32_t integer_array[] = { -1, -2, 0, 1, 2 }, *return_array;
  48         char *s;
  49
  50         r = sd_bus_message_new_method_call(NULL, "foobar.waldo", "/", "foobar.waldo", "Piep", &m);
  51         assert_se(r >= 0);
  52
  53         r = sd_bus_message_append(m, "s", "a string");
  54         assert_se(r >= 0);
  55
  56         r = sd_bus_message_append(m, "s", NULL);
  57         assert_se(r < 0);
  58
  59         r = sd_bus_message_append(m, "as", 2, "string #1", "string #2");
  60         assert_se(r >= 0);
  61
  62         r = sd_bus_message_append(m, "sass", "foobar", 5, "foo", "bar", "waldo", "piep", "pap", "after");
  63         assert_se(r >= 0);
  64
  65         r = sd_bus_message_append(m, "a{yv}", 2, 3, "s", "foo", 5, "s", "waldo");
  66         assert_se(r >= 0);
  67
  68         r = sd_bus_message_append(m, "ba(ss)", 255, 3, "aaa", "1", "bbb", "2", "ccc", "3");
  69         assert_se(r >= 0);
  70
  71         r = sd_bus_message_open_container(m, 'a', "s");
  72         assert_se(r >= 0);
  73
  74         r = sd_bus_message_append_basic(m, 's', "foobar");
  75         assert_se(r >= 0);
  76
  77         r = sd_bus_message_append_basic(m, 's', "waldo");
  78         assert_se(r >= 0);
  79
  80         r = sd_bus_message_close_container(m);
  81         assert_se(r >= 0);
  82
  83         r = sd_bus_message_append_string_space(m, 5, &s);
  84         assert_se(r >= 0);
  85         strcpy(s, "hallo");
  86
  87         r = sd_bus_message_append_array(m, 'i', integer_array, sizeof(integer_array));
  88         assert_se(r >= 0);
  89
  90         r = bus_message_seal(m, 4711);
  91         assert_se(r >= 0);
  92
  93         bus_message_dump(m);
  94
  95         r = bus_message_get_blob(m, &buffer, &sz);
  96         assert_se(r >= 0);
  97
  98         h = hexmem(buffer, sz);
  99         assert_se(h);
 100
 101         log_info("message size = %lu, contents =\n%s", (unsigned long) sz, h);
 102         free(h);
 103
 104 #ifdef HAVE_GLIB
 105         {
 106                 GDBusMessage *g;
 107                 char *p;
 108
 109 #if !defined(GLIB_VERSION_2_36)
 110                 g_type_init();
 111 #endif
 112
 113                 g = g_dbus_message_new_from_blob(buffer, sz, 0, NULL);
 114                 p = g_dbus_message_print(g, 0);
 115                 log_info("%s", p);
 116                 g_free(p);
 117                 g_object_unref(g);
 118         }
 119 #endif
 120
 121         {
 122                 DBusMessage *w;
 123                 DBusError error;
 124
 125                 dbus_error_init(&error);
 126
 127                 w = dbus_message_demarshal(buffer, sz, &error);
 128                 if (!w) {
 129                         log_error("%s", error.message);
 130                 } else
 131                         dbus_message_unref(w);
 132         }
 133
 134         m = sd_bus_message_unref(m);
 135
 136         r = bus_message_from_malloc(buffer, sz, NULL, 0, NULL, NULL, &m);
 137         assert_se(r >= 0);
 138
 139         bus_message_dump(m);
 140
 141         assert_se(sd_bus_message_rewind(m, true) >= 0);
 142
 143         r = sd_bus_message_read(m, "sas", &x, 2, &y, &z);
 144         assert_se(r > 0);
 145         assert_se(streq(x, "a string"));
 146         assert_se(streq(y, "string #1"));
 147         assert_se(streq(z, "string #2"));
 148
 149         r = sd_bus_message_read(m, "sass", &x, 5, &y, &z, &a, &b, &c, &d);
 150         assert_se(r > 0);
 151         assert_se(streq(x, "foobar"));
 152         assert_se(streq(y, "foo"));
 153         assert_se(streq(z, "bar"));
 154         assert_se(streq(a, "waldo"));
 155         assert_se(streq(b, "piep"));
 156         assert_se(streq(c, "pap"));
 157         assert_se(streq(d, "after"));
 158
 159         r = sd_bus_message_read(m, "a{yv}", 2, &u, "s", &x, &v, "s", &y);
 160         assert_se(r > 0);
 161         assert_se(u == 3);
 162         assert_se(streq(x, "foo"));
 163         assert_se(v == 5);
 164         assert_se(streq(y, "waldo"));
 165
 166         r = sd_bus_message_read(m, "ba(ss)", &boolean, 3, &x, &y, &a, &b, &c, &d);
 167         assert_se(r > 0);
 168         assert_se(boolean);
 169         assert_se(streq(x, "aaa"));
 170         assert_se(streq(y, "1"));
 171         assert_se(streq(a, "bbb"));
 172         assert_se(streq(b, "2"));
 173         assert_se(streq(c, "ccc"));
 174         assert_se(streq(d, "3"));
 175
 176         r = sd_bus_message_read(m, "as", 2, &x, &y);
 177         assert_se(r > 0);
 178         assert_se(streq(x, "foobar"));
 179         assert_se(streq(y, "waldo"));
 180
 181         r = sd_bus_message_read_basic(m, 's', &s);
 182         assert_se(r > 0);
 183         assert_se(streq(s, "hallo"));
 184
 185         r = sd_bus_message_read_array(m, 'i', (const void**) &return_array, &sz);
 186         assert_se(r > 0);
 187         assert_se(sz == sizeof(integer_array));
 188         assert_se(memcmp(integer_array, return_array, sz) == 0);
 189
 190         r = sd_bus_message_peek_type(m, NULL, NULL);
 191         assert_se(r == 0);
 192
 193         return 0;
 194 }