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
  38 int main(int argc, char *argv[]) {
  39         _cleanup_bus_message_unref_ sd_bus_message *m = NULL;
  40         int r, boolean;
  41         const char *x, *y, *z, *a, *b, *c, *d;
  42         uint8_t u, v;
  43         void *buffer = NULL;
  44         size_t sz;
  45         char *h;
  46         const int32_t integer_array[] = { -1, -2, 0, 1, 2 }, *return_array;
  47
  48         r = sd_bus_message_new_method_call(NULL, "foobar.waldo", "/", "foobar.waldo", "Piep", &m);
  49         assert_se(r >= 0);
  50
  51         r = sd_bus_message_append(m, "s", "a string");
  52         assert_se(r >= 0);
  53
  54         r = sd_bus_message_append(m, "s", NULL);
  55         assert_se(r < 0);
  56
  57         r = sd_bus_message_append(m, "as", 2, "string #1", "string #2");
  58         assert_se(r >= 0);
  59
  60         r = sd_bus_message_append(m, "sass", "foobar", 5, "foo", "bar", "waldo", "piep", "pap", "after");
  61         assert_se(r >= 0);
  62
  63         r = sd_bus_message_append(m, "a{yv}", 2, 3, "s", "foo", 5, "s", "waldo");
  64         assert_se(r >= 0);
  65
  66         r = sd_bus_message_append(m, "ba(ss)", 255, 3, "aaa", "1", "bbb", "2", "ccc", "3");
  67         assert_se(r >= 0);
  68
  69         r = sd_bus_message_open_container(m, 'a', "s");
  70         assert_se(r >= 0);
  71
  72         r = sd_bus_message_append_basic(m, 's', "foobar");
  73         assert_se(r >= 0);
  74
  75         r = sd_bus_message_append_basic(m, 's', "waldo");
  76         assert_se(r >= 0);
  77
  78         r = sd_bus_message_close_container(m);
  79         assert_se(r >= 0);
  80
  81         r = sd_bus_message_append_array(m, 'i', integer_array, sizeof(integer_array));
  82         assert_se(r >= 0);
  83
  84         r = bus_message_seal(m, 4711);
  85         assert_se(r >= 0);
  86
  87         bus_message_dump(m);
  88
  89         r = bus_message_get_blob(m, &buffer, &sz);
  90         assert_se(r >= 0);
  91
  92         h = hexmem(buffer, sz);
  93         assert_se(h);
  94
  95         log_info("message size = %lu, contents =\n%s", (unsigned long) sz, h);
  96         free(h);
  97
  98 #ifdef HAVE_GLIB
  99         {
 100                 GDBusMessage *g;
 101                 char *p;
 102
 103 #if !defined(GLIB_VERSION_2_36)
 104                 g_type_init();
 105 #endif
 106
 107                 g = g_dbus_message_new_from_blob(buffer, sz, 0, NULL);
 108                 p = g_dbus_message_print(g, 0);
 109                 log_info("%s", p);
 110                 g_free(p);
 111                 g_object_unref(g);
 112         }
 113 #endif
 114
 115         {
 116                 DBusMessage *w;
 117                 DBusError error;
 118
 119                 dbus_error_init(&error);
 120
 121                 w = dbus_message_demarshal(buffer, sz, &error);
 122                 if (!w) {
 123                         log_error("%s", error.message);
 124                 } else
 125                         dbus_message_unref(w);
 126         }
 127
 128         m = sd_bus_message_unref(m);
 129
 130         r = bus_message_from_malloc(buffer, sz, NULL, 0, NULL, NULL, &m);
 131         assert_se(r >= 0);
 132
 133         bus_message_dump(m);
 134
 135         assert_se(sd_bus_message_rewind(m, true) >= 0);
 136
 137         r = sd_bus_message_read(m, "sas", &x, 2, &y, &z);
 138         assert_se(r > 0);
 139         assert_se(streq(x, "a string"));
 140         assert_se(streq(y, "string #1"));
 141         assert_se(streq(z, "string #2"));
 142
 143         r = sd_bus_message_read(m, "sass", &x, 5, &y, &z, &a, &b, &c, &d);
 144         assert_se(r > 0);
 145         assert_se(streq(x, "foobar"));
 146         assert_se(streq(y, "foo"));
 147         assert_se(streq(z, "bar"));
 148         assert_se(streq(a, "waldo"));
 149         assert_se(streq(b, "piep"));
 150         assert_se(streq(c, "pap"));
 151         assert_se(streq(d, "after"));
 152
 153         r = sd_bus_message_read(m, "a{yv}", 2, &u, "s", &x, &v, "s", &y);
 154         assert_se(r > 0);
 155         assert_se(u == 3);
 156         assert_se(streq(x, "foo"));
 157         assert_se(v == 5);
 158         assert_se(streq(y, "waldo"));
 159
 160         r = sd_bus_message_read(m, "ba(ss)", &boolean, 3, &x, &y, &a, &b, &c, &d);
 161         assert_se(r > 0);
 162         assert_se(boolean);
 163         assert_se(streq(x, "aaa"));
 164         assert_se(streq(y, "1"));
 165         assert_se(streq(a, "bbb"));
 166         assert_se(streq(b, "2"));
 167         assert_se(streq(c, "ccc"));
 168         assert_se(streq(d, "3"));
 169
 170         r = sd_bus_message_read(m, "as", 2, &x, &y);
 171         assert_se(r > 0);
 172         assert_se(streq(x, "foobar"));
 173         assert_se(streq(y, "waldo"));
 174
 175         r = sd_bus_message_read_array(m, 'i', (const void**) &return_array, &sz);
 176         assert_se(r > 0);
 177         assert_se(sz == sizeof(integer_array));
 178         assert_se(memcmp(integer_array, return_array, sz) == 0);
 179
 180         r = sd_bus_message_peek_type(m, NULL, NULL);
 181         assert_se(r == 0);
 182
 183         return 0;
 184 }