src/libsystemd/sd-bus/test-bus-gvariant.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 #if HAVE_GLIB
  21 #include <glib.h>
  22 #endif
  23
  24 #include "sd-bus.h"
  25
  26 #include "alloc-util.h"
  27 #include "bus-dump.h"
  28 #include "bus-gvariant.h"
  29 #include "bus-internal.h"
  30 #include "bus-message.h"
  31 #include "bus-util.h"
  32 #include "macro.h"
  33 #include "util.h"
  34
  35 static void test_bus_gvariant_is_fixed_size(void) {
  36         assert_se(bus_gvariant_is_fixed_size("") > 0);
  37         assert_se(bus_gvariant_is_fixed_size("()") > 0);
  38         assert_se(bus_gvariant_is_fixed_size("y") > 0);
  39         assert_se(bus_gvariant_is_fixed_size("u") > 0);
  40         assert_se(bus_gvariant_is_fixed_size("b") > 0);
  41         assert_se(bus_gvariant_is_fixed_size("n") > 0);
  42         assert_se(bus_gvariant_is_fixed_size("q") > 0);
  43         assert_se(bus_gvariant_is_fixed_size("i") > 0);
  44         assert_se(bus_gvariant_is_fixed_size("t") > 0);
  45         assert_se(bus_gvariant_is_fixed_size("d") > 0);
  46         assert_se(bus_gvariant_is_fixed_size("s") == 0);
  47         assert_se(bus_gvariant_is_fixed_size("o") == 0);
  48         assert_se(bus_gvariant_is_fixed_size("g") == 0);
  49         assert_se(bus_gvariant_is_fixed_size("h") > 0);
  50         assert_se(bus_gvariant_is_fixed_size("ay") == 0);
  51         assert_se(bus_gvariant_is_fixed_size("v") == 0);
  52         assert_se(bus_gvariant_is_fixed_size("(u)") > 0);
  53         assert_se(bus_gvariant_is_fixed_size("(uuuuy)") > 0);
  54         assert_se(bus_gvariant_is_fixed_size("(uusuuy)") == 0);
  55         assert_se(bus_gvariant_is_fixed_size("a{ss}") == 0);
  56         assert_se(bus_gvariant_is_fixed_size("((u)yyy(b(iiii)))") > 0);
  57         assert_se(bus_gvariant_is_fixed_size("((u)yyy(b(iiivi)))") == 0);
  58 }
  59
  60 static void test_bus_gvariant_get_size(void) {
  61         assert_se(bus_gvariant_get_size("") == 0);
  62         assert_se(bus_gvariant_get_size("()") == 1);
  63         assert_se(bus_gvariant_get_size("y") == 1);
  64         assert_se(bus_gvariant_get_size("u") == 4);
  65         assert_se(bus_gvariant_get_size("b") == 1);
  66         assert_se(bus_gvariant_get_size("n") == 2);
  67         assert_se(bus_gvariant_get_size("q") == 2);
  68         assert_se(bus_gvariant_get_size("i") == 4);
  69         assert_se(bus_gvariant_get_size("t") == 8);
  70         assert_se(bus_gvariant_get_size("d") == 8);
  71         assert_se(bus_gvariant_get_size("s") < 0);
  72         assert_se(bus_gvariant_get_size("o") < 0);
  73         assert_se(bus_gvariant_get_size("g") < 0);
  74         assert_se(bus_gvariant_get_size("h") == 4);
  75         assert_se(bus_gvariant_get_size("ay") < 0);
  76         assert_se(bus_gvariant_get_size("v") < 0);
  77         assert_se(bus_gvariant_get_size("(u)") == 4);
  78         assert_se(bus_gvariant_get_size("(uuuuy)") == 20);
  79         assert_se(bus_gvariant_get_size("(uusuuy)") < 0);
  80         assert_se(bus_gvariant_get_size("a{ss}") < 0);
  81         assert_se(bus_gvariant_get_size("((u)yyy(b(iiii)))") == 28);
  82         assert_se(bus_gvariant_get_size("((u)yyy(b(iiivi)))") < 0);
  83         assert_se(bus_gvariant_get_size("((b)(t))") == 16);
  84         assert_se(bus_gvariant_get_size("((b)(b)(t))") == 16);
  85         assert_se(bus_gvariant_get_size("(bt)") == 16);
  86         assert_se(bus_gvariant_get_size("((t)(b))") == 16);
  87         assert_se(bus_gvariant_get_size("(tb)") == 16);
  88         assert_se(bus_gvariant_get_size("((b)(b))") == 2);
  89         assert_se(bus_gvariant_get_size("((t)(t))") == 16);
  90 }
  91
  92 static void test_bus_gvariant_get_alignment(void) {
  93         assert_se(bus_gvariant_get_alignment("") == 1);
  94         assert_se(bus_gvariant_get_alignment("()") == 1);
  95         assert_se(bus_gvariant_get_alignment("y") == 1);
  96         assert_se(bus_gvariant_get_alignment("b") == 1);
  97         assert_se(bus_gvariant_get_alignment("u") == 4);
  98         assert_se(bus_gvariant_get_alignment("s") == 1);
  99         assert_se(bus_gvariant_get_alignment("o") == 1);
 100         assert_se(bus_gvariant_get_alignment("g") == 1);
 101         assert_se(bus_gvariant_get_alignment("v") == 8);
 102         assert_se(bus_gvariant_get_alignment("h") == 4);
 103         assert_se(bus_gvariant_get_alignment("i") == 4);
 104         assert_se(bus_gvariant_get_alignment("t") == 8);
 105         assert_se(bus_gvariant_get_alignment("x") == 8);
 106         assert_se(bus_gvariant_get_alignment("q") == 2);
 107         assert_se(bus_gvariant_get_alignment("n") == 2);
 108         assert_se(bus_gvariant_get_alignment("d") == 8);
 109         assert_se(bus_gvariant_get_alignment("ay") == 1);
 110         assert_se(bus_gvariant_get_alignment("as") == 1);
 111         assert_se(bus_gvariant_get_alignment("au") == 4);
 112         assert_se(bus_gvariant_get_alignment("an") == 2);
 113         assert_se(bus_gvariant_get_alignment("ans") == 2);
 114         assert_se(bus_gvariant_get_alignment("ant") == 8);
 115         assert_se(bus_gvariant_get_alignment("(ss)") == 1);
 116         assert_se(bus_gvariant_get_alignment("(ssu)") == 4);
 117         assert_se(bus_gvariant_get_alignment("a(ssu)") == 4);
 118         assert_se(bus_gvariant_get_alignment("(u)") == 4);
 119         assert_se(bus_gvariant_get_alignment("(uuuuy)") == 4);
 120         assert_se(bus_gvariant_get_alignment("(uusuuy)") == 4);
 121         assert_se(bus_gvariant_get_alignment("a{ss}") == 1);
 122         assert_se(bus_gvariant_get_alignment("((u)yyy(b(iiii)))") == 4);
 123         assert_se(bus_gvariant_get_alignment("((u)yyy(b(iiivi)))") == 8);
 124         assert_se(bus_gvariant_get_alignment("((b)(t))") == 8);
 125         assert_se(bus_gvariant_get_alignment("((b)(b)(t))") == 8);
 126         assert_se(bus_gvariant_get_alignment("(bt)") == 8);
 127         assert_se(bus_gvariant_get_alignment("((t)(b))") == 8);
 128         assert_se(bus_gvariant_get_alignment("(tb)") == 8);
 129         assert_se(bus_gvariant_get_alignment("((b)(b))") == 1);
 130         assert_se(bus_gvariant_get_alignment("((t)(t))") == 8);
 131 }
 132
 133 static void test_marshal(void) {
 134         _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL, *n = NULL;
 135         _cleanup_(sd_bus_flush_close_unrefp) sd_bus *bus = NULL;
 136         _cleanup_free_ void *blob;
 137         size_t sz;
 138         int r;
 139
 140         r = sd_bus_open_user(&bus);
 141         if (r < 0)
 142                 exit(EXIT_TEST_SKIP);
 143
 144         bus->message_version = 2; /* dirty hack to enable gvariant */
 145
 146         assert_se(sd_bus_message_new_method_call(bus, &m, "a.service.name", "/an/object/path/which/is/really/really/long/so/that/we/hit/the/eight/bit/boundary/by/quite/some/margin/to/test/this/stuff/that/it/really/works", "an.interface.name", "AMethodName") >= 0);
 147
 148         assert_cc(sizeof(struct bus_header) == 16);
 149
 150         assert_se(sd_bus_message_append(m,
 151                                         "a(usv)", 3,
 152                                         4711, "first-string-parameter", "(st)", "X", (uint64_t) 1111,
 153                                         4712, "second-string-parameter", "(a(si))", 2, "Y", 5, "Z", 6,
 154                                         4713, "third-string-parameter", "(uu)", 1, 2) >= 0);
 155
 156         assert_se(bus_message_seal(m, 4711, 0) >= 0);
 157
 158 #if HAVE_GLIB
 159         {
 160                 GVariant *v;
 161                 char *t;
 162
 163 #if !defined(GLIB_VERSION_2_36)
 164                 g_type_init();
 165 #endif
 166
 167                 v = g_variant_new_from_data(G_VARIANT_TYPE("(yyyyuta{tv})"), m->header, sizeof(struct bus_header) + m->fields_size, false, NULL, NULL);
 168                 assert_se(g_variant_is_normal_form(v));
 169                 t = g_variant_print(v, TRUE);
 170                 printf("%s\n", t);
 171                 g_free(t);
 172                 g_variant_unref(v);
 173
 174                 v = g_variant_new_from_data(G_VARIANT_TYPE("(a(usv))"), m->body.data, m->user_body_size, false, NULL, NULL);
 175                 assert_se(g_variant_is_normal_form(v));
 176                 t = g_variant_print(v, TRUE);
 177                 printf("%s\n", t);
 178                 g_free(t);
 179                 g_variant_unref(v);
 180         }
 181 #endif
 182
 183         assert_se(bus_message_dump(m, NULL, BUS_MESSAGE_DUMP_WITH_HEADER) >= 0);
 184
 185         assert_se(bus_message_get_blob(m, &blob, &sz) >= 0);
 186
 187 #if HAVE_GLIB
 188         {
 189                 GVariant *v;
 190                 char *t;
 191
 192                 v = g_variant_new_from_data(G_VARIANT_TYPE("(yyyyuta{tv}v)"), blob, sz, false, NULL, NULL);
 193                 assert_se(g_variant_is_normal_form(v));
 194                 t = g_variant_print(v, TRUE);
 195                 printf("%s\n", t);
 196                 g_free(t);
 197                 g_variant_unref(v);
 198         }
 199 #endif
 200
 201         assert_se(bus_message_from_malloc(bus, blob, sz, NULL, 0, NULL, &n) >= 0);
 202         blob = NULL;
 203
 204         assert_se(bus_message_dump(n, NULL, BUS_MESSAGE_DUMP_WITH_HEADER) >= 0);
 205
 206         m = sd_bus_message_unref(m);
 207
 208         assert_se(sd_bus_message_new_method_call(bus, &m, "a.x", "/a/x", "a.x", "Ax") >= 0);
 209
 210         assert_se(sd_bus_message_append(m, "as", 0) >= 0);
 211
 212         assert_se(bus_message_seal(m, 4712, 0) >= 0);
 213         assert_se(bus_message_dump(m, NULL, BUS_MESSAGE_DUMP_WITH_HEADER) >= 0);
 214 }
 215
 216 int main(int argc, char *argv[]) {
 217
 218         test_bus_gvariant_is_fixed_size();
 219         test_bus_gvariant_get_size();
 220         test_bus_gvariant_get_alignment();
 221         test_marshal();
 222
 223         return 0;
 224 }