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 "build.h"
  40 #include "strv.h"
  41 #include "def.h"
  42 #include "capability.h"
  43
  44 #include <bus-proxyd/bus-policy.h>
  45
  46 static int test_policy_load(Policy *p, const char *name)
  47 {
  48         _cleanup_free_ char *path = NULL;
  49         int r = 0;
  50
  51         path = strjoin(TEST_DIR, "/bus-policy/", name, NULL);
  52         assert_se(path);
  53
  54         if (access(path, R_OK) == 0)
  55                 policy_load(p, STRV_MAKE(path));
  56         else
  57                 r = -ENOENT;
  58
  59         return r;
  60 }
  61
  62 int main(int argc, char *argv[]) {
  63
  64         Policy p = {};
  65         struct ucred ucred = {};
  66         char **names_strv;
  67         Hashmap *names_hash;
  68
  69         /* Ownership tests */
  70         assert_se(test_policy_load(&p, "ownerships.conf") == 0);
  71
  72         ucred.uid = 0;
  73         assert_se(policy_check_own(&p, &ucred, "org.test.test1") == true);
  74         ucred.uid = 1;
  75         assert_se(policy_check_own(&p, &ucred, "org.test.test1") == true);
  76
  77         ucred.uid = 0;
  78         assert_se(policy_check_own(&p, &ucred, "org.test.test2") == true);
  79         ucred.uid = 1;
  80         assert_se(policy_check_own(&p, &ucred, "org.test.test2") == false);
  81
  82         ucred.uid = 0;
  83         assert_se(policy_check_own(&p, &ucred, "org.test.test3") == false);
  84         ucred.uid = 1;
  85         assert_se(policy_check_own(&p, &ucred, "org.test.test3") == false);
  86
  87         ucred.uid = 0;
  88         assert_se(policy_check_own(&p, &ucred, "org.test.test4") == false);
  89         ucred.uid = 1;
  90         assert_se(policy_check_own(&p, &ucred, "org.test.test4") == true);
  91
  92         policy_free(&p);
  93
  94         /* Signaltest */
  95         assert_se(test_policy_load(&p, "signals.conf") == 0);
  96         names_strv = STRV_MAKE("bli.bla.blubb");
  97
  98         ucred.uid = 0;
  99         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_SIGNAL, NULL, "/an/object/path", NULL) == true);
 100
 101         ucred.uid = 1;
 102         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_SIGNAL, NULL, "/an/object/path", NULL) == false);
 103
 104         policy_free(&p);
 105
 106         /* Method calls */
 107         assert_se(test_policy_load(&p, "methods.conf") == 0);
 108         names_strv = STRV_MAKE("org.test.test1");
 109         policy_dump(&p);
 110
 111         ucred.uid = 0;
 112
 113         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_METHOD_CALL, "/an/object/path", "bli.bla.blubb", "Member") == false);
 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", "org.test.int1", "Member") == true);
 116         assert_se(policy_check_send(&p, &ucred, names_strv, SD_BUS_MESSAGE_METHOD_CALL, "/an/object/path", "org.test.int2", "Member") == true);
 117
 118         names_hash = hashmap_new(&string_hash_ops);
 119         assert(names_hash != NULL);
 120         assert_se(hashmap_put(names_hash, "org.test.test3", NULL) >= 0);
 121         assert_se(policy_check_recv(&p, &ucred, names_hash, SD_BUS_MESSAGE_METHOD_CALL, "/an/object/path", "org.test.int3", "Member111") == true);
 122
 123         policy_free(&p);
 124
 125         /* User and groups */
 126         assert_se(test_policy_load(&p, "hello.conf") == 0);
 127         policy_dump(&p);
 128
 129         ucred.uid = 0;
 130         assert_se(policy_check_hello(&p, &ucred) == true);
 131
 132         ucred.uid = 1;
 133         assert_se(policy_check_hello(&p, &ucred) == false);
 134
 135         ucred.uid = 0;
 136         ucred.gid = 1;
 137         assert_se(policy_check_hello(&p, &ucred) == false);
 138
 139         policy_free(&p);
 140
 141         return EXIT_SUCCESS;
 142 }