src/test/test-namespace.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 Zbigniew Jędrzejewski-Szmek
   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
  24 #include "fd-util.h"
  25 #include "namespace.h"
  26 #include "process-util.h"
  27 #include "string-util.h"
  28 #include "util.h"
  29
  30 static void test_tmpdir(const char *id, const char *A, const char *B) {
  31         _cleanup_free_ char *a, *b;
  32         struct stat x, y;
  33         char *c, *d;
  34
  35         assert_se(setup_tmp_dirs(id, &a, &b) == 0);
  36         assert_se(startswith(a, A));
  37         assert_se(startswith(b, B));
  38
  39         assert_se(stat(a, &x) >= 0);
  40         assert_se(stat(b, &y) >= 0);
  41
  42         assert_se(S_ISDIR(x.st_mode));
  43         assert_se(S_ISDIR(y.st_mode));
  44
  45         assert_se((x.st_mode & 01777) == 0700);
  46         assert_se((y.st_mode & 01777) == 0700);
  47
  48         c = strjoina(a, "/tmp");
  49         d = strjoina(b, "/tmp");
  50
  51         assert_se(stat(c, &x) >= 0);
  52         assert_se(stat(d, &y) >= 0);
  53
  54         assert_se(S_ISDIR(x.st_mode));
  55         assert_se(S_ISDIR(y.st_mode));
  56
  57         assert_se((x.st_mode & 01777) == 01777);
  58         assert_se((y.st_mode & 01777) == 01777);
  59
  60         assert_se(rmdir(c) >= 0);
  61         assert_se(rmdir(d) >= 0);
  62
  63         assert_se(rmdir(a) >= 0);
  64         assert_se(rmdir(b) >= 0);
  65 }
  66
  67 static void test_netns(void) {
  68         _cleanup_close_pair_ int s[2] = { -1, -1 };
  69         pid_t pid1, pid2, pid3;
  70         int r, n = 0;
  71         siginfo_t si;
  72
  73         if (geteuid() > 0)
  74                 return;
  75
  76         assert_se(socketpair(AF_UNIX, SOCK_DGRAM, 0, s) >= 0);
  77
  78         pid1 = fork();
  79         assert_se(pid1 >= 0);
  80
  81         if (pid1 == 0) {
  82                 r = setup_netns(s);
  83                 assert_se(r >= 0);
  84                 _exit(r);
  85         }
  86
  87         pid2 = fork();
  88         assert_se(pid2 >= 0);
  89
  90         if (pid2 == 0) {
  91                 r = setup_netns(s);
  92                 assert_se(r >= 0);
  93                 exit(r);
  94         }
  95
  96         pid3 = fork();
  97         assert_se(pid3 >= 0);
  98
  99         if (pid3 == 0) {
 100                 r = setup_netns(s);
 101                 assert_se(r >= 0);
 102                 exit(r);
 103         }
 104
 105         r = wait_for_terminate(pid1, &si);
 106         assert_se(r >= 0);
 107         assert_se(si.si_code == CLD_EXITED);
 108         n += si.si_status;
 109
 110         r = wait_for_terminate(pid2, &si);
 111         assert_se(r >= 0);
 112         assert_se(si.si_code == CLD_EXITED);
 113         n += si.si_status;
 114
 115         r = wait_for_terminate(pid3, &si);
 116         assert_se(r >= 0);
 117         assert_se(si.si_code == CLD_EXITED);
 118         n += si.si_status;
 119
 120         assert_se(n == 1);
 121 }
 122
 123 int main(int argc, char *argv[]) {
 124         sd_id128_t bid;
 125         char boot_id[SD_ID128_STRING_MAX];
 126         _cleanup_free_ char *x = NULL, *y = NULL, *z = NULL, *zz = NULL;
 127
 128         assert_se(sd_id128_get_boot(&bid) >= 0);
 129         sd_id128_to_string(bid, boot_id);
 130
 131         x = strjoin("/tmp/systemd-private-", boot_id, "-abcd.service-", NULL);
 132         y = strjoin("/var/tmp/systemd-private-", boot_id, "-abcd.service-", NULL);
 133         assert_se(x && y);
 134
 135         test_tmpdir("abcd.service", x, y);
 136
 137         z = strjoin("/tmp/systemd-private-", boot_id, "-sys-devices-pci0000:00-0000:00:1a.0-usb3-3\\x2d1-3\\x2d1:1.0-bluetooth-hci0.device-", NULL);
 138         zz = strjoin("/var/tmp/systemd-private-", boot_id, "-sys-devices-pci0000:00-0000:00:1a.0-usb3-3\\x2d1-3\\x2d1:1.0-bluetooth-hci0.device-", NULL);
 139
 140         assert_se(z && zz);
 141
 142         test_tmpdir("sys-devices-pci0000:00-0000:00:1a.0-usb3-3\\x2d1-3\\x2d1:1.0-bluetooth-hci0.device", z, zz);
 143
 144         test_netns();
 145
 146         return 0;
 147 }