src/test/test-execute.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 <grp.h>
  21 #include <pwd.h>
  22 #include <stdio.h>
  23 #include <sys/prctl.h>
  24 #include <sys/types.h>
  25
  26 #include "fileio.h"
  27 #include "fs-util.h"
  28 #include "macro.h"
  29 #include "manager.h"
  30 #include "mkdir.h"
  31 #include "path-util.h"
  32 #include "rm-rf.h"
  33 #ifdef HAVE_SECCOMP
  34 #include "seccomp-util.h"
  35 #endif
  36 #include "test-helper.h"
  37 #include "unit.h"
  38 #include "util.h"
  39 #include "virt.h"
  40
  41 typedef void (*test_function_t)(Manager *m);
  42
  43 static void check(Manager *m, Unit *unit, int status_expected, int code_expected) {
  44         Service *service = NULL;
  45         usec_t ts;
  46         usec_t timeout = 2 * USEC_PER_SEC;
  47
  48         assert_se(m);
  49         assert_se(unit);
  50
  51         service = SERVICE(unit);
  52         printf("%s\n", unit->id);
  53         exec_context_dump(&service->exec_context, stdout, "\t");
  54         ts = now(CLOCK_MONOTONIC);
  55         while (service->state != SERVICE_DEAD && service->state != SERVICE_FAILED) {
  56                 int r;
  57                 usec_t n;
  58
  59                 r = sd_event_run(m->event, 100 * USEC_PER_MSEC);
  60                 assert_se(r >= 0);
  61
  62                 n = now(CLOCK_MONOTONIC);
  63                 if (ts + timeout < n) {
  64                         log_error("Test timeout when testing %s", unit->id);
  65                         exit(EXIT_FAILURE);
  66                 }
  67         }
  68         exec_status_dump(&service->main_exec_status, stdout, "\t");
  69         assert_se(service->main_exec_status.status == status_expected);
  70         assert_se(service->main_exec_status.code == code_expected);
  71 }
  72
  73 static void test(Manager *m, const char *unit_name, int status_expected, int code_expected) {
  74         Unit *unit;
  75
  76         assert_se(unit_name);
  77
  78         assert_se(manager_load_unit(m, unit_name, NULL, NULL, &unit) >= 0);
  79         assert_se(UNIT_VTABLE(unit)->start(unit) >= 0);
  80         check(m, unit, status_expected, code_expected);
  81 }
  82
  83 static void test_exec_workingdirectory(Manager *m) {
  84         assert_se(mkdir_p("/tmp/test-exec_workingdirectory", 0755) >= 0);
  85
  86         test(m, "exec-workingdirectory.service", 0, CLD_EXITED);
  87
  88         (void) rm_rf("/tmp/test-exec_workingdirectory", REMOVE_ROOT|REMOVE_PHYSICAL);
  89 }
  90
  91 static void test_exec_personality(Manager *m) {
  92 #if defined(__x86_64__)
  93         test(m, "exec-personality-x86-64.service", 0, CLD_EXITED);
  94
  95 #elif defined(__s390__)
  96         test(m, "exec-personality-s390.service", 0, CLD_EXITED);
  97
  98 #elif defined(__powerpc64__)
  99 #  if __BYTE_ORDER == __BIG_ENDIAN
 100         test(m, "exec-personality-ppc64.service", 0, CLD_EXITED);
 101 #  else
 102         test(m, "exec-personality-ppc64le.service", 0, CLD_EXITED);
 103 #  endif
 104
 105 #elif defined(__aarch64__)
 106         test(m, "exec-personality-aarch64.service", 0, CLD_EXITED);
 107
 108 #elif defined(__i386__)
 109         test(m, "exec-personality-x86.service", 0, CLD_EXITED);
 110 #endif
 111 }
 112
 113 static void test_exec_ignoresigpipe(Manager *m) {
 114         test(m, "exec-ignoresigpipe-yes.service", 0, CLD_EXITED);
 115         test(m, "exec-ignoresigpipe-no.service", SIGPIPE, CLD_KILLED);
 116 }
 117
 118 static void test_exec_privatetmp(Manager *m) {
 119         assert_se(touch("/tmp/test-exec_privatetmp") >= 0);
 120
 121         test(m, "exec-privatetmp-yes.service", 0, CLD_EXITED);
 122         test(m, "exec-privatetmp-no.service", 0, CLD_EXITED);
 123
 124         unlink("/tmp/test-exec_privatetmp");
 125 }
 126
 127 static void test_exec_privatedevices(Manager *m) {
 128         if (detect_container() > 0) {
 129                 log_notice("testing in container, skipping private device tests");
 130                 return;
 131         }
 132         test(m, "exec-privatedevices-yes.service", 0, CLD_EXITED);
 133         test(m, "exec-privatedevices-no.service", 0, CLD_EXITED);
 134 }
 135
 136 static void test_exec_privatedevices_capabilities(Manager *m) {
 137         if (detect_container() > 0) {
 138                 log_notice("testing in container, skipping private device tests");
 139                 return;
 140         }
 141         test(m, "exec-privatedevices-yes-capability-mknod.service", 0, CLD_EXITED);
 142         test(m, "exec-privatedevices-no-capability-mknod.service", 0, CLD_EXITED);
 143 }
 144
 145 static void test_exec_readonlypaths(Manager *m) {
 146         test(m, "exec-readonlypaths.service", 0, CLD_EXITED);
 147         test(m, "exec-readonlypaths-mount-propagation.service", 0, CLD_EXITED);
 148 }
 149
 150 static void test_exec_readwritepaths(Manager *m) {
 151         test(m, "exec-readwritepaths-mount-propagation.service", 0, CLD_EXITED);
 152 }
 153
 154 static void test_exec_inaccessiblepaths(Manager *m) {
 155         test(m, "exec-inaccessiblepaths-mount-propagation.service", 0, CLD_EXITED);
 156 }
 157
 158 static void test_exec_systemcallfilter(Manager *m) {
 159 #ifdef HAVE_SECCOMP
 160         if (!is_seccomp_available())
 161                 return;
 162         test(m, "exec-systemcallfilter-not-failing.service", 0, CLD_EXITED);
 163         test(m, "exec-systemcallfilter-not-failing2.service", 0, CLD_EXITED);
 164         test(m, "exec-systemcallfilter-failing.service", SIGSYS, CLD_KILLED);
 165         test(m, "exec-systemcallfilter-failing2.service", SIGSYS, CLD_KILLED);
 166
 167 #endif
 168 }
 169
 170 static void test_exec_systemcallerrornumber(Manager *m) {
 171 #ifdef HAVE_SECCOMP
 172         if (is_seccomp_available())
 173                 test(m, "exec-systemcallerrornumber.service", 1, CLD_EXITED);
 174 #endif
 175 }
 176
 177 static void test_exec_systemcall_system_mode_with_user(Manager *m) {
 178 #ifdef HAVE_SECCOMP
 179         if (!is_seccomp_available())
 180                 return;
 181         if (getpwnam("nobody"))
 182                 test(m, "exec-systemcallfilter-system-user.service", 0, CLD_EXITED);
 183         else if (getpwnam("nfsnobody"))
 184                 test(m, "exec-systemcallfilter-system-user-nfsnobody.service", 0, CLD_EXITED);
 185         else
 186                 log_error_errno(errno, "Skipping test_exec_systemcall_system_mode_with_user, could not find nobody/nfsnobody user: %m");
 187 #endif
 188 }
 189
 190 static void test_exec_user(Manager *m) {
 191         if (getpwnam("nobody"))
 192                 test(m, "exec-user.service", 0, CLD_EXITED);
 193         else if (getpwnam("nfsnobody"))
 194                 test(m, "exec-user-nfsnobody.service", 0, CLD_EXITED);
 195         else
 196                 log_error_errno(errno, "Skipping test_exec_user, could not find nobody/nfsnobody user: %m");
 197 }
 198
 199 static void test_exec_group(Manager *m) {
 200         if (getgrnam("nobody"))
 201                 test(m, "exec-group.service", 0, CLD_EXITED);
 202         else if (getgrnam("nfsnobody"))
 203                 test(m, "exec-group-nfsnobody.service", 0, CLD_EXITED);
 204         else
 205                 log_error_errno(errno, "Skipping test_exec_group, could not find nobody/nfsnobody group: %m");
 206 }
 207
 208 static void test_exec_environment(Manager *m) {
 209         test(m, "exec-environment.service", 0, CLD_EXITED);
 210         test(m, "exec-environment-multiple.service", 0, CLD_EXITED);
 211         test(m, "exec-environment-empty.service", 0, CLD_EXITED);
 212 }
 213
 214 static void test_exec_environmentfile(Manager *m) {
 215         static const char e[] =
 216                 "VAR1='word1 word2'\n"
 217                 "VAR2=word3 \n"
 218                 "# comment1\n"
 219                 "\n"
 220                 "; comment2\n"
 221                 " ; # comment3\n"
 222                 "line without an equal\n"
 223                 "VAR3='$word 5 6'\n";
 224         int r;
 225
 226         r = write_string_file("/tmp/test-exec_environmentfile.conf", e, WRITE_STRING_FILE_CREATE);
 227         assert_se(r == 0);
 228
 229         test(m, "exec-environmentfile.service", 0, CLD_EXITED);
 230
 231         unlink("/tmp/test-exec_environmentfile.conf");
 232 }
 233
 234 static void test_exec_passenvironment(Manager *m) {
 235         /* test-execute runs under MANAGER_USER which, by default, forwards all
 236          * variables present in the environment, but only those that are
 237          * present _at the time it is created_!
 238          *
 239          * So these PassEnvironment checks are still expected to work, since we
 240          * are ensuring the variables are not present at manager creation (they
 241          * are unset explicitly in main) and are only set here.
 242          *
 243          * This is still a good approximation of how a test for MANAGER_SYSTEM
 244          * would work.
 245          */
 246         assert_se(setenv("VAR1", "word1 word2", 1) == 0);
 247         assert_se(setenv("VAR2", "word3", 1) == 0);
 248         assert_se(setenv("VAR3", "$word 5 6", 1) == 0);
 249         test(m, "exec-passenvironment.service", 0, CLD_EXITED);
 250         test(m, "exec-passenvironment-repeated.service", 0, CLD_EXITED);
 251         test(m, "exec-passenvironment-empty.service", 0, CLD_EXITED);
 252         assert_se(unsetenv("VAR1") == 0);
 253         assert_se(unsetenv("VAR2") == 0);
 254         assert_se(unsetenv("VAR3") == 0);
 255         test(m, "exec-passenvironment-absent.service", 0, CLD_EXITED);
 256 }
 257
 258 static void test_exec_umask(Manager *m) {
 259         test(m, "exec-umask-default.service", 0, CLD_EXITED);
 260         test(m, "exec-umask-0177.service", 0, CLD_EXITED);
 261 }
 262
 263 static void test_exec_runtimedirectory(Manager *m) {
 264         test(m, "exec-runtimedirectory.service", 0, CLD_EXITED);
 265         test(m, "exec-runtimedirectory-mode.service", 0, CLD_EXITED);
 266         if (getgrnam("nobody"))
 267                 test(m, "exec-runtimedirectory-owner.service", 0, CLD_EXITED);
 268         else if (getgrnam("nfsnobody"))
 269                 test(m, "exec-runtimedirectory-owner-nfsnobody.service", 0, CLD_EXITED);
 270         else
 271                 log_error_errno(errno, "Skipping test_exec_runtimedirectory-owner, could not find nobody/nfsnobody group: %m");
 272 }
 273
 274 static void test_exec_capabilityboundingset(Manager *m) {
 275         int r;
 276
 277         /* We use capsh to test if the capabilities are
 278          * properly set, so be sure that it exists */
 279         r = find_binary("capsh", NULL);
 280         if (r < 0) {
 281                 log_error_errno(r, "Skipping test_exec_capabilityboundingset, could not find capsh binary: %m");
 282                 return;
 283         }
 284
 285         test(m, "exec-capabilityboundingset-simple.service", 0, CLD_EXITED);
 286         test(m, "exec-capabilityboundingset-reset.service", 0, CLD_EXITED);
 287         test(m, "exec-capabilityboundingset-merge.service", 0, CLD_EXITED);
 288         test(m, "exec-capabilityboundingset-invert.service", 0, CLD_EXITED);
 289 }
 290
 291 static void test_exec_capabilityambientset(Manager *m) {
 292         int r;
 293
 294         /* Check if the kernel has support for ambient capabilities. Run
 295          * the tests only if that's the case. Clearing all ambient
 296          * capabilities is fine, since we are expecting them to be unset
 297          * in the first place for the tests. */
 298         r = prctl(PR_CAP_AMBIENT, PR_CAP_AMBIENT_CLEAR_ALL, 0, 0, 0);
 299         if (r >= 0 || errno != EINVAL) {
 300                 if (getpwnam("nobody")) {
 301                         test(m, "exec-capabilityambientset.service", 0, CLD_EXITED);
 302                         test(m, "exec-capabilityambientset-merge.service", 0, CLD_EXITED);
 303                 } else if (getpwnam("nfsnobody")) {
 304                         test(m, "exec-capabilityambientset-nfsnobody.service", 0, CLD_EXITED);
 305                         test(m, "exec-capabilityambientset-merge-nfsnobody.service", 0, CLD_EXITED);
 306                 } else
 307                         log_error_errno(errno, "Skipping test_exec_capabilityambientset, could not find nobody/nfsnobody user: %m");
 308         } else
 309                 log_error_errno(errno, "Skipping test_exec_capabilityambientset, the kernel does not support ambient capabilities: %m");
 310 }
 311
 312 static void test_exec_privatenetwork(Manager *m) {
 313         int r;
 314
 315         r = find_binary("ip", NULL);
 316         if (r < 0) {
 317                 log_error_errno(r, "Skipping test_exec_privatenetwork, could not find ip binary: %m");
 318                 return;
 319         }
 320
 321         test(m, "exec-privatenetwork-yes.service", 0, CLD_EXITED);
 322 }
 323
 324 static void test_exec_oomscoreadjust(Manager *m) {
 325         test(m, "exec-oomscoreadjust-positive.service", 0, CLD_EXITED);
 326         test(m, "exec-oomscoreadjust-negative.service", 0, CLD_EXITED);
 327 }
 328
 329 static void test_exec_ioschedulingclass(Manager *m) {
 330         test(m, "exec-ioschedulingclass-none.service", 0, CLD_EXITED);
 331         test(m, "exec-ioschedulingclass-idle.service", 0, CLD_EXITED);
 332         test(m, "exec-ioschedulingclass-realtime.service", 0, CLD_EXITED);
 333         test(m, "exec-ioschedulingclass-best-effort.service", 0, CLD_EXITED);
 334 }
 335
 336 static void test_exec_spec_interpolation(Manager *m) {
 337         test(m, "exec-spec-interpolation.service", 0, CLD_EXITED);
 338 }
 339
 340 static int run_tests(UnitFileScope scope, test_function_t *tests) {
 341         test_function_t *test = NULL;
 342         Manager *m = NULL;
 343         int r;
 344
 345         assert_se(tests);
 346
 347         r = manager_new(scope, true, &m);
 348         if (MANAGER_SKIP_TEST(r)) {
 349                 log_notice_errno(r, "Skipping test: manager_new: %m");
 350                 return EXIT_TEST_SKIP;
 351         }
 352         assert_se(r >= 0);
 353         assert_se(manager_startup(m, NULL, NULL) >= 0);
 354
 355         for (test = tests; test && *test; test++)
 356                 (*test)(m);
 357
 358         manager_free(m);
 359
 360         return 0;
 361 }
 362
 363 int main(int argc, char *argv[]) {
 364         test_function_t user_tests[] = {
 365                 test_exec_workingdirectory,
 366                 test_exec_personality,
 367                 test_exec_ignoresigpipe,
 368                 test_exec_privatetmp,
 369                 test_exec_privatedevices,
 370                 test_exec_privatedevices_capabilities,
 371                 test_exec_readonlypaths,
 372                 test_exec_readwritepaths,
 373                 test_exec_inaccessiblepaths,
 374                 test_exec_privatenetwork,
 375                 test_exec_systemcallfilter,
 376                 test_exec_systemcallerrornumber,
 377                 test_exec_user,
 378                 test_exec_group,
 379                 test_exec_environment,
 380                 test_exec_environmentfile,
 381                 test_exec_passenvironment,
 382                 test_exec_umask,
 383                 test_exec_runtimedirectory,
 384                 test_exec_capabilityboundingset,
 385                 test_exec_capabilityambientset,
 386                 test_exec_oomscoreadjust,
 387                 test_exec_ioschedulingclass,
 388                 test_exec_spec_interpolation,
 389                 NULL,
 390         };
 391         test_function_t system_tests[] = {
 392                 test_exec_systemcall_system_mode_with_user,
 393                 NULL,
 394         };
 395         int r;
 396
 397         log_parse_environment();
 398         log_open();
 399
 400         /* It is needed otherwise cgroup creation fails */
 401         if (getuid() != 0) {
 402                 printf("Skipping test: not root\n");
 403                 return EXIT_TEST_SKIP;
 404         }
 405
 406         assert_se(setenv("XDG_RUNTIME_DIR", "/tmp/", 1) == 0);
 407         assert_se(set_unit_path(TEST_DIR "/test-execute/") >= 0);
 408
 409         /* Unset VAR1, VAR2 and VAR3 which are used in the PassEnvironment test
 410          * cases, otherwise (and if they are present in the environment),
 411          * `manager_default_environment` will copy them into the default
 412          * environment which is passed to each created job, which will make the
 413          * tests that expect those not to be present to fail.
 414          */
 415         assert_se(unsetenv("VAR1") == 0);
 416         assert_se(unsetenv("VAR2") == 0);
 417         assert_se(unsetenv("VAR3") == 0);
 418
 419         r = run_tests(UNIT_FILE_USER, user_tests);
 420         if (r != 0)
 421                 return r;
 422
 423         return run_tests(UNIT_FILE_SYSTEM, system_tests);
 424 }