src/bus-proxyd/test-bus-policy.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2014 Daniel Mack
   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 <sys/socket.h>
  23 #include <sys/un.h>
  24 #include <sys/types.h>
  25 #include <fcntl.h>
  26 #include <unistd.h>
  27 #include <string.h>
  28 #include <errno.h>
  29 #include <sys/poll.h>
  30 #include <stddef.h>
  31 #include <getopt.h>
  32
  33 #include "log.h"
  34 #include "util.h"
  35 #include "sd-bus.h"
  36 #include "bus-internal.h"
  37 #include "bus-message.h"
  38 #include "bus-util.h"
  39 #include "bus-internal.h"
  40 #include "build.h"
  41 #include "strv.h"
  42 #include "def.h"
  43 #include "capability.h"
  44
  45 #include <bus-proxyd/bus-policy.h>
  46
  47 static int test_policy_load(Policy *p, const char *name)
  48 {
  49         _cleanup_free_ char *path = NULL;
  50         int r = 0;
  51
  52         path = strjoin(TEST_DIR, "/bus-policy/", name, NULL);
  53         assert_se(path);
  54
  55         if (access(path, R_OK) == 0)
  56                 policy_load(p, STRV_MAKE(path));
  57         else
  58                 r = -ENOENT;
  59
  60         return r;
  61 }
  62
  63 int main(int argc, char *argv[]) {
  64
  65         Policy p = {};
  66         struct ucred ucred = {};
  67         char **names_strv;
  68         Hashmap *names_hash;
  69
  70         /* Ownership tests */
  71         assert_se(test_policy_load(&p, "ownerships.conf") == 0);
  72
  73         ucred.uid = 0;
  74         assert_se(policy_check_own(&p, &ucred, "org.test.test1") == true);
  75         ucred.uid = 1;
  76         assert_se(policy_check_own(&p, &ucred, "org.test.test1") == true);
  77
  78         ucred.uid = 0;
  79         assert_se(policy_check_own(&p, &ucred, "org.test.test2") == true);
  80         ucred.uid = 1;
  81         assert_se(policy_check_own(&p, &ucred, "org.test.test2") == false);
  82
  83         ucred.uid = 0;
  84         assert_se(policy_check_own(&p, &ucred, "org.test.test3") == false);
  85         ucred.uid = 1;
  86         assert_se(policy_check_own(&p, &ucred, "org.test.test3") == false);
  87
  88         ucred.uid = 0;
  89         assert_se(policy_check_own(&p, &ucred, "org.test.test4") == false);
  90         ucred.uid = 1;
  91         assert_se(policy_check_own(&p, &ucred, "org.test.test4") == true);
  92
  93         policy_free(&p);
  94
  95         /* Signaltest */
  96         assert_se(test_policy_load(&p, "signals.conf") == 0);
  97         names_strv = STRV_MAKE("bli.bla.blubb");
  98
  99         ucred.uid = 0;
 100         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_SIGNAL, NULL, "/an/object/path", NULL) == true);
 101
 102         ucred.uid = 1;
 103         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_SIGNAL, NULL, "/an/object/path", NULL) == false);
 104
 105         policy_free(&p);
 106
 107         /* Method calls */
 108         assert_se(test_policy_load(&p, "methods.conf") == 0);
 109         names_strv = STRV_MAKE("org.test.test1");
 110         policy_dump(&p);
 111
 112         ucred.uid = 0;
 113
 114         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_METHOD_CALL, "/an/object/path", "bli.bla.blubb", "Member") == false);
 115         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_METHOD_CALL, "/an/object/path", "bli.bla.blubb", "Member") == false);
 116         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_METHOD_CALL, "/an/object/path", "org.test.int1", "Member") == true);
 117         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_METHOD_CALL, "/an/object/path", "org.test.int2", "Member") == true);
 118
 119         names_hash = hashmap_new(&string_hash_ops);
 120         assert(names_hash != NULL);
 121         assert_se(hashmap_put(names_hash, "org.test.test3", NULL) >= 0);
 122         assert_se(policy_check_recv(&p, &ucred, names_hash, SD_BUS_MESSAGE_METHOD_CALL, "/an/object/path", "org.test.int3", "Member111") == true);
 123
 124         policy_free(&p);
 125
 126         /* User and groups */
 127         assert_se(test_policy_load(&p, "hello.conf") == 0);
 128         policy_dump(&p);
 129
 130         ucred.uid = 0;
 131         assert_se(policy_check_hello(&p, &ucred) == true);
 132
 133         ucred.uid = 1;
 134         assert_se(policy_check_hello(&p, &ucred) == false);
 135
 136         ucred.uid = 0;
 137         ucred.gid = 1;
 138         assert_se(policy_check_hello(&p, &ucred) == false);
 139
 140         policy_free(&p);
 141
 142         return EXIT_SUCCESS;
 143 }