src/libsystemd-bus/test-bus-zero-copy.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 <fcntl.h>
  23 #include <sys/mman.h>
  24
  25 #include "util.h"
  26 #include "log.h"
  27
  28 #include "sd-bus.h"
  29 #include "sd-memfd.h"
  30 #include "bus-message.h"
  31 #include "bus-error.h"
  32 #include "bus-kernel.h"
  33 #include "bus-dump.h"
  34
  35 #define FIRST_ARRAY 17
  36 #define SECOND_ARRAY 33
  37
  38 #define STRING_SIZE 123
  39
  40 int main(int argc, char *argv[]) {
  41         _cleanup_free_ char *bus_name = NULL, *address = NULL;
  42         uint8_t *p;
  43         sd_bus *a, *b;
  44         int r, bus_ref;
  45         sd_bus_message *m;
  46         sd_memfd *f;
  47         uint64_t sz;
  48         uint32_t u32;
  49         size_t i, l;
  50         char *s;
  51
  52         log_set_max_level(LOG_DEBUG);
  53
  54         bus_ref = bus_kernel_create_bus("deine-mutter", &bus_name);
  55         if (bus_ref == -ENOENT)
  56                 return EXIT_TEST_SKIP;
  57
  58         assert_se(bus_ref >= 0);
  59
  60         address = strappend("kernel:path=", bus_name);
  61         assert_se(address);
  62
  63         r = sd_bus_new(&a);
  64         assert_se(r >= 0);
  65
  66         r = sd_bus_new(&b);
  67         assert_se(r >= 0);
  68
  69         r = sd_bus_set_address(a, address);
  70         assert_se(r >= 0);
  71
  72         r = sd_bus_set_address(b, address);
  73         assert_se(r >= 0);
  74
  75         r = sd_bus_start(a);
  76         assert_se(r >= 0);
  77
  78         r = sd_bus_start(b);
  79         assert_se(r >= 0);
  80
  81         r = sd_bus_message_new_method_call(b, ":1.1", "/a/path", "an.inter.face", "AMethod", &m);
  82         assert_se(r >= 0);
  83
  84         r = sd_bus_message_open_container(m, 'r', "aysay");
  85         assert_se(r >= 0);
  86
  87         r = sd_bus_message_append_array_space(m, 'y', FIRST_ARRAY, (void**) &p);
  88         assert_se(r >= 0);
  89
  90         p[0] = '<';
  91         memset(p+1, 'L', FIRST_ARRAY-2);
  92         p[FIRST_ARRAY-1] = '>';
  93
  94         r = sd_memfd_new_and_map(&f, STRING_SIZE, (void**) &s);
  95         assert_se(r >= 0);
  96
  97         s[0] = '<';
  98         for (i = 1; i < STRING_SIZE-2; i++)
  99                 s[i] = '0' + (i % 10);
 100         s[STRING_SIZE-2] = '>';
 101         s[STRING_SIZE-1] = 0;
 102         munmap(s, STRING_SIZE);
 103
 104         r = sd_memfd_get_size(f, &sz);
 105         assert_se(r >= 0);
 106         assert_se(sz == STRING_SIZE);
 107
 108         r = sd_bus_message_append_string_memfd(m, f);
 109         assert_se(r >= 0);
 110
 111         sd_memfd_free(f);
 112
 113         r = sd_memfd_new_and_map(&f, SECOND_ARRAY, (void**) &p);
 114         assert_se(r >= 0);
 115
 116         p[0] = '<';
 117         memset(p+1, 'P', SECOND_ARRAY-2);
 118         p[SECOND_ARRAY-1] = '>';
 119         munmap(p, SECOND_ARRAY);
 120
 121         r = sd_memfd_get_size(f, &sz);
 122         assert_se(r >= 0);
 123         assert_se(sz == SECOND_ARRAY);
 124
 125         r = sd_bus_message_append_array_memfd(m, 'y', f);
 126         assert_se(r >= 0);
 127
 128         sd_memfd_free(f);
 129
 130         r = sd_bus_message_close_container(m);
 131         assert_se(r >= 0);
 132
 133         r = sd_bus_message_append(m, "u", 4711);
 134         assert_se(r >= 0);
 135
 136         r = bus_message_seal(m, 55, 99*USEC_PER_SEC);
 137         assert_se(r >= 0);
 138
 139         bus_message_dump(m, stdout, true);
 140
 141         r = sd_bus_send(b, m, NULL);
 142         assert_se(r >= 0);
 143
 144         sd_bus_message_unref(m);
 145
 146         r = sd_bus_process(a, &m);
 147         assert_se(r > 0);
 148
 149         bus_message_dump(m, stdout, true);
 150         sd_bus_message_rewind(m, true);
 151
 152         r = sd_bus_message_enter_container(m, 'r', "aysay");
 153         assert_se(r > 0);
 154
 155         r = sd_bus_message_read_array(m, 'y', (const void**) &p, &l);
 156         assert_se(r > 0);
 157         assert_se(l == FIRST_ARRAY);
 158
 159         assert_se(p[0] == '<');
 160         for (i = 1; i < l-1; i++)
 161                 assert_se(p[i] == 'L');
 162         assert_se(p[l-1] == '>');
 163
 164         r = sd_bus_message_read(m, "s", &s);
 165         assert_se(r > 0);
 166
 167         assert_se(s[0] == '<');
 168         for (i = 1; i < STRING_SIZE-2; i++)
 169                 assert_se(s[i] == (char) ('0' + (i % 10)));
 170         assert_se(s[STRING_SIZE-2] == '>');
 171         assert_se(s[STRING_SIZE-1] == 0);
 172
 173         r = sd_bus_message_read_array(m, 'y', (const void**) &p, &l);
 174         assert_se(r > 0);
 175         assert_se(l == SECOND_ARRAY);
 176
 177         assert_se(p[0] == '<');
 178         for (i = 1; i < l-1; i++)
 179                 assert_se(p[i] == 'P');
 180         assert_se(p[l-1] == '>');
 181
 182         r = sd_bus_message_exit_container(m);
 183         assert_se(r > 0);
 184
 185         r = sd_bus_message_read(m, "u", &u32);
 186         assert_se(r > 0);
 187         assert_se(u32 == 4711);
 188
 189         sd_bus_message_unref(m);
 190
 191         sd_bus_unref(a);
 192         sd_bus_unref(b);
 193
 194         return 0;
 195 }