src/test/test-condition.c

   1 /***
   2   This file is part of systemd
   3
   4   Copyright 2014 Ronny Chevalier
   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 #include "sd-id128.h"
  21
  22 #include "alloc-util.h"
  23 #include "apparmor-util.h"
  24 #include "architecture.h"
  25 #include "audit-util.h"
  26 #include "condition.h"
  27 #include "hostname-util.h"
  28 #include "ima-util.h"
  29 #include "log.h"
  30 #include "macro.h"
  31 #include "selinux-util.h"
  32 #include "smack-util.h"
  33 #include "util.h"
  34
  35 static void test_condition_test_path(void) {
  36         Condition *condition;
  37
  38         condition = condition_new(CONDITION_PATH_EXISTS, "/bin/sh", false, false);
  39         assert_se(condition_test(condition));
  40         condition_free(condition);
  41
  42         condition = condition_new(CONDITION_PATH_EXISTS, "/bin/s?", false, false);
  43         assert_se(!condition_test(condition));
  44         condition_free(condition);
  45
  46         condition = condition_new(CONDITION_PATH_EXISTS_GLOB, "/bin/s?", false, false);
  47         assert_se(condition_test(condition));
  48         condition_free(condition);
  49
  50         condition = condition_new(CONDITION_PATH_EXISTS_GLOB, "/bin/s?", false, true);
  51         assert_se(!condition_test(condition));
  52         condition_free(condition);
  53
  54         condition = condition_new(CONDITION_PATH_EXISTS, "/thiscertainlywontexist", false, false);
  55         assert_se(!condition_test(condition));
  56         condition_free(condition);
  57
  58         condition = condition_new(CONDITION_PATH_EXISTS, "/thiscertainlywontexist", false, true);
  59         assert_se(condition_test(condition));
  60         condition_free(condition);
  61
  62         condition = condition_new(CONDITION_PATH_IS_DIRECTORY, "/bin", false, false);
  63         assert_se(condition_test(condition));
  64         condition_free(condition);
  65
  66         condition = condition_new(CONDITION_DIRECTORY_NOT_EMPTY, "/bin", false, false);
  67         assert_se(condition_test(condition));
  68         condition_free(condition);
  69
  70         condition = condition_new(CONDITION_FILE_NOT_EMPTY, "/bin/sh", false, false);
  71         assert_se(condition_test(condition));
  72         condition_free(condition);
  73
  74         condition = condition_new(CONDITION_FILE_IS_EXECUTABLE, "/bin/sh", false, false);
  75         assert_se(condition_test(condition));
  76         condition_free(condition);
  77
  78         condition = condition_new(CONDITION_FILE_IS_EXECUTABLE, "/etc/passwd", false, false);
  79         assert_se(!condition_test(condition));
  80         condition_free(condition);
  81
  82         condition = condition_new(CONDITION_PATH_IS_MOUNT_POINT, "/proc", false, false);
  83         assert_se(condition_test(condition));
  84         condition_free(condition);
  85
  86         condition = condition_new(CONDITION_PATH_IS_MOUNT_POINT, "/", false, false);
  87         assert_se(condition_test(condition));
  88         condition_free(condition);
  89
  90         condition = condition_new(CONDITION_PATH_IS_MOUNT_POINT, "/bin", false, false);
  91         assert_se(!condition_test(condition));
  92         condition_free(condition);
  93
  94         condition = condition_new(CONDITION_PATH_IS_READ_WRITE, "/tmp", false, false);
  95         assert_se(condition_test(condition));
  96         condition_free(condition);
  97
  98         condition = condition_new(CONDITION_PATH_IS_SYMBOLIC_LINK, "/dev/stdout", false, false);
  99         assert_se(condition_test(condition));
 100         condition_free(condition);
 101 }
 102
 103 static void test_condition_test_ac_power(void) {
 104         Condition *condition;
 105
 106         condition = condition_new(CONDITION_AC_POWER, "true", false, false);
 107         assert_se(condition_test(condition) == on_ac_power());
 108         condition_free(condition);
 109
 110         condition = condition_new(CONDITION_AC_POWER, "false", false, false);
 111         assert_se(condition_test(condition) != on_ac_power());
 112         condition_free(condition);
 113
 114         condition = condition_new(CONDITION_AC_POWER, "false", false, true);
 115         assert_se(condition_test(condition) == on_ac_power());
 116         condition_free(condition);
 117 }
 118
 119 static void test_condition_test_host(void) {
 120         Condition *condition;
 121         sd_id128_t id;
 122         int r;
 123         char sid[SD_ID128_STRING_MAX];
 124         _cleanup_free_ char *hostname = NULL;
 125
 126         r = sd_id128_get_machine(&id);
 127         assert_se(r >= 0);
 128         assert_se(sd_id128_to_string(id, sid));
 129
 130         condition = condition_new(CONDITION_HOST, sid, false, false);
 131         assert_se(condition_test(condition));
 132         condition_free(condition);
 133
 134         condition = condition_new(CONDITION_HOST, "garbage value jjjjjjjjjjjjjj", false, false);
 135         assert_se(!condition_test(condition));
 136         condition_free(condition);
 137
 138         condition = condition_new(CONDITION_HOST, sid, false, true);
 139         assert_se(!condition_test(condition));
 140         condition_free(condition);
 141
 142         hostname = gethostname_malloc();
 143         assert_se(hostname);
 144
 145         condition = condition_new(CONDITION_HOST, hostname, false, false);
 146         assert_se(condition_test(condition));
 147         condition_free(condition);
 148 }
 149
 150 static void test_condition_test_architecture(void) {
 151         Condition *condition;
 152         const char *sa;
 153         int a;
 154
 155         a = uname_architecture();
 156         assert_se(a >= 0);
 157
 158         sa = architecture_to_string(a);
 159         assert_se(sa);
 160
 161         condition = condition_new(CONDITION_ARCHITECTURE, sa, false, false);
 162         assert_se(condition_test(condition));
 163         condition_free(condition);
 164
 165         condition = condition_new(CONDITION_ARCHITECTURE, "garbage value", false, false);
 166         assert_se(condition_test(condition) < 0);
 167         condition_free(condition);
 168
 169         condition = condition_new(CONDITION_ARCHITECTURE, sa, false, true);
 170         assert_se(!condition_test(condition));
 171         condition_free(condition);
 172 }
 173
 174 static void test_condition_test_kernel_command_line(void) {
 175         Condition *condition;
 176
 177         condition = condition_new(CONDITION_KERNEL_COMMAND_LINE, "thisreallyshouldntbeonthekernelcommandline", false, false);
 178         assert_se(!condition_test(condition));
 179         condition_free(condition);
 180
 181         condition = condition_new(CONDITION_KERNEL_COMMAND_LINE, "andthis=neither", false, false);
 182         assert_se(!condition_test(condition));
 183         condition_free(condition);
 184 }
 185
 186 static void test_condition_test_null(void) {
 187         Condition *condition;
 188
 189         condition = condition_new(CONDITION_NULL, NULL, false, false);
 190         assert_se(condition_test(condition));
 191         condition_free(condition);
 192
 193         condition = condition_new(CONDITION_NULL, NULL, false, true);
 194         assert_se(!condition_test(condition));
 195         condition_free(condition);
 196 }
 197
 198 static void test_condition_test_security(void) {
 199         Condition *condition;
 200
 201         condition = condition_new(CONDITION_SECURITY, "garbage oifdsjfoidsjoj", false, false);
 202         assert_se(!condition_test(condition));
 203         condition_free(condition);
 204
 205         condition = condition_new(CONDITION_SECURITY, "selinux", false, true);
 206         assert_se(condition_test(condition) != mac_selinux_use());
 207         condition_free(condition);
 208
 209         condition = condition_new(CONDITION_SECURITY, "ima", false, false);
 210         assert_se(condition_test(condition) == use_ima());
 211         condition_free(condition);
 212
 213         condition = condition_new(CONDITION_SECURITY, "apparmor", false, false);
 214         assert_se(condition_test(condition) == mac_apparmor_use());
 215         condition_free(condition);
 216
 217         condition = condition_new(CONDITION_SECURITY, "smack", false, false);
 218         assert_se(condition_test(condition) == mac_smack_use());
 219         condition_free(condition);
 220
 221         condition = condition_new(CONDITION_SECURITY, "audit", false, false);
 222         assert_se(condition_test(condition) == use_audit());
 223         condition_free(condition);
 224 }
 225
 226
 227 int main(int argc, char *argv[]) {
 228         log_parse_environment();
 229         log_open();
 230
 231         test_condition_test_path();
 232         test_condition_test_ac_power();
 233         test_condition_test_host();
 234         test_condition_test_architecture();
 235         test_condition_test_kernel_command_line();
 236         test_condition_test_null();
 237         test_condition_test_security();
 238
 239         return 0;
 240 }