src/libsystemd/sd-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 *name = NULL, *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         assert_se(asprintf(&name, "deine-mutter-%u", (unsigned) getpid()) >= 0);
  55
  56         bus_ref = bus_kernel_create_bus(name, false, &bus_name);
  57         if (bus_ref == -ENOENT)
  58                 return EXIT_TEST_SKIP;
  59
  60         assert_se(bus_ref >= 0);
  61
  62         address = strappend("kernel:path=", bus_name);
  63         assert_se(address);
  64
  65         r = sd_bus_new(&a);
  66         assert_se(r >= 0);
  67
  68         r = sd_bus_new(&b);
  69         assert_se(r >= 0);
  70
  71         r = sd_bus_set_address(a, address);
  72         assert_se(r >= 0);
  73
  74         r = sd_bus_set_address(b, address);
  75         assert_se(r >= 0);
  76
  77         r = sd_bus_start(a);
  78         assert_se(r >= 0);
  79
  80         r = sd_bus_start(b);
  81         assert_se(r >= 0);
  82
  83         r = sd_bus_message_new_method_call(b, ":1.1", "/a/path", "an.inter.face", "AMethod", &m);
  84         assert_se(r >= 0);
  85
  86         r = sd_bus_message_open_container(m, 'r', "aysay");
  87         assert_se(r >= 0);
  88
  89         r = sd_bus_message_append_array_space(m, 'y', FIRST_ARRAY, (void**) &p);
  90         assert_se(r >= 0);
  91
  92         p[0] = '<';
  93         memset(p+1, 'L', FIRST_ARRAY-2);
  94         p[FIRST_ARRAY-1] = '>';
  95
  96         r = sd_memfd_new_and_map(&f, STRING_SIZE, (void**) &s);
  97         assert_se(r >= 0);
  98
  99         s[0] = '<';
 100         for (i = 1; i < STRING_SIZE-2; i++)
 101                 s[i] = '0' + (i % 10);
 102         s[STRING_SIZE-2] = '>';
 103         s[STRING_SIZE-1] = 0;
 104         munmap(s, STRING_SIZE);
 105
 106         r = sd_memfd_get_size(f, &sz);
 107         assert_se(r >= 0);
 108         assert_se(sz == STRING_SIZE);
 109
 110         r = sd_bus_message_append_string_memfd(m, f);
 111         assert_se(r >= 0);
 112
 113         sd_memfd_free(f);
 114
 115         r = sd_memfd_new_and_map(&f, SECOND_ARRAY, (void**) &p);
 116         assert_se(r >= 0);
 117
 118         p[0] = '<';
 119         memset(p+1, 'P', SECOND_ARRAY-2);
 120         p[SECOND_ARRAY-1] = '>';
 121         munmap(p, SECOND_ARRAY);
 122
 123         r = sd_memfd_get_size(f, &sz);
 124         assert_se(r >= 0);
 125         assert_se(sz == SECOND_ARRAY);
 126
 127         r = sd_bus_message_append_array_memfd(m, 'y', f);
 128         assert_se(r >= 0);
 129
 130         sd_memfd_free(f);
 131
 132         r = sd_bus_message_close_container(m);
 133         assert_se(r >= 0);
 134
 135         r = sd_bus_message_append(m, "u", 4711);
 136         assert_se(r >= 0);
 137
 138         r = bus_message_seal(m, 55, 99*USEC_PER_SEC);
 139         assert_se(r >= 0);
 140
 141         bus_message_dump(m, stdout, true);
 142
 143         r = sd_bus_send(b, m, NULL);
 144         assert_se(r >= 0);
 145
 146         sd_bus_message_unref(m);
 147
 148         r = sd_bus_process(a, &m);
 149         assert_se(r > 0);
 150
 151         bus_message_dump(m, stdout, true);
 152         sd_bus_message_rewind(m, true);
 153
 154         r = sd_bus_message_enter_container(m, 'r', "aysay");
 155         assert_se(r > 0);
 156
 157         r = sd_bus_message_read_array(m, 'y', (const void**) &p, &l);
 158         assert_se(r > 0);
 159         assert_se(l == FIRST_ARRAY);
 160
 161         assert_se(p[0] == '<');
 162         for (i = 1; i < l-1; i++)
 163                 assert_se(p[i] == 'L');
 164         assert_se(p[l-1] == '>');
 165
 166         r = sd_bus_message_read(m, "s", &s);
 167         assert_se(r > 0);
 168
 169         assert_se(s[0] == '<');
 170         for (i = 1; i < STRING_SIZE-2; i++)
 171                 assert_se(s[i] == (char) ('0' + (i % 10)));
 172         assert_se(s[STRING_SIZE-2] == '>');
 173         assert_se(s[STRING_SIZE-1] == 0);
 174
 175         r = sd_bus_message_read_array(m, 'y', (const void**) &p, &l);
 176         assert_se(r > 0);
 177         assert_se(l == SECOND_ARRAY);
 178
 179         assert_se(p[0] == '<');
 180         for (i = 1; i < l-1; i++)
 181                 assert_se(p[i] == 'P');
 182         assert_se(p[l-1] == '>');
 183
 184         r = sd_bus_message_exit_container(m);
 185         assert_se(r > 0);
 186
 187         r = sd_bus_message_read(m, "u", &u32);
 188         assert_se(r > 0);
 189         assert_se(u32 == 4711);
 190
 191         sd_bus_message_unref(m);
 192
 193         sd_bus_unref(a);
 194         sd_bus_unref(b);
 195
 196         return 0;
 197 }