src/test/test-fd-util.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 /***
   3   This file is part of systemd.
   4
   5   Copyright 2010 Lennart Poettering
   6
   7   systemd is free software; you can redistribute it and/or modify it
   8   under the terms of the GNU Lesser General Public License as published by
   9   the Free Software Foundation; either version 2.1 of the License, or
  10   (at your option) any later version.
  11
  12   systemd is distributed in the hope that it will be useful, but
  13   WITHOUT ANY WARRANTY; without even the implied warranty of
  14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15   Lesser General Public License for more details.
  16
  17   You should have received a copy of the GNU Lesser General Public License
  18   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  19 ***/
  20
  21 #include <fcntl.h>
  22 #include <unistd.h>
  23
  24 #include "alloc-util.h"
  25 #include "fd-util.h"
  26 #include "fileio.h"
  27 #include "macro.h"
  28 #include "random-util.h"
  29 #include "string-util.h"
  30 #include "util.h"
  31
  32 static void test_close_many(void) {
  33         int fds[3];
  34         char name0[] = "/tmp/test-close-many.XXXXXX";
  35         char name1[] = "/tmp/test-close-many.XXXXXX";
  36         char name2[] = "/tmp/test-close-many.XXXXXX";
  37
  38         fds[0] = mkostemp_safe(name0);
  39         fds[1] = mkostemp_safe(name1);
  40         fds[2] = mkostemp_safe(name2);
  41
  42         close_many(fds, 2);
  43
  44         assert_se(fcntl(fds[0], F_GETFD) == -1);
  45         assert_se(fcntl(fds[1], F_GETFD) == -1);
  46         assert_se(fcntl(fds[2], F_GETFD) >= 0);
  47
  48         safe_close(fds[2]);
  49
  50         unlink(name0);
  51         unlink(name1);
  52         unlink(name2);
  53 }
  54
  55 static void test_close_nointr(void) {
  56         char name[] = "/tmp/test-test-close_nointr.XXXXXX";
  57         int fd;
  58
  59         fd = mkostemp_safe(name);
  60         assert_se(fd >= 0);
  61         assert_se(close_nointr(fd) >= 0);
  62         assert_se(close_nointr(fd) < 0);
  63
  64         unlink(name);
  65 }
  66
  67 static void test_same_fd(void) {
  68         _cleanup_close_pair_ int p[2] = { -1, -1 };
  69         _cleanup_close_ int a = -1, b = -1, c = -1;
  70
  71         assert_se(pipe2(p, O_CLOEXEC) >= 0);
  72         assert_se((a = dup(p[0])) >= 0);
  73         assert_se((b = open("/dev/null", O_RDONLY|O_CLOEXEC)) >= 0);
  74         assert_se((c = dup(a)) >= 0);
  75
  76         assert_se(same_fd(p[0], p[0]) > 0);
  77         assert_se(same_fd(p[1], p[1]) > 0);
  78         assert_se(same_fd(a, a) > 0);
  79         assert_se(same_fd(b, b) > 0);
  80
  81         assert_se(same_fd(a, p[0]) > 0);
  82         assert_se(same_fd(p[0], a) > 0);
  83         assert_se(same_fd(c, p[0]) > 0);
  84         assert_se(same_fd(p[0], c) > 0);
  85         assert_se(same_fd(a, c) > 0);
  86         assert_se(same_fd(c, a) > 0);
  87
  88         assert_se(same_fd(p[0], p[1]) == 0);
  89         assert_se(same_fd(p[1], p[0]) == 0);
  90         assert_se(same_fd(p[0], b) == 0);
  91         assert_se(same_fd(b, p[0]) == 0);
  92         assert_se(same_fd(p[1], a) == 0);
  93         assert_se(same_fd(a, p[1]) == 0);
  94         assert_se(same_fd(p[1], b) == 0);
  95         assert_se(same_fd(b, p[1]) == 0);
  96
  97         assert_se(same_fd(a, b) == 0);
  98         assert_se(same_fd(b, a) == 0);
  99 }
 100
 101 static void test_open_serialization_fd(void) {
 102         _cleanup_close_ int fd = -1;
 103
 104         fd = open_serialization_fd("test");
 105         assert_se(fd >= 0);
 106
 107         write(fd, "test\n", 5);
 108 }
 109
 110 static void test_acquire_data_fd_one(unsigned flags) {
 111         char wbuffer[196*1024 - 7];
 112         char rbuffer[sizeof(wbuffer)];
 113         int fd;
 114
 115         fd = acquire_data_fd("foo", 3, flags);
 116         assert_se(fd >= 0);
 117
 118         zero(rbuffer);
 119         assert_se(read(fd, rbuffer, sizeof(rbuffer)) == 3);
 120         assert_se(streq(rbuffer, "foo"));
 121
 122         fd = safe_close(fd);
 123
 124         fd = acquire_data_fd("", 0, flags);
 125         assert_se(fd >= 0);
 126
 127         zero(rbuffer);
 128         assert_se(read(fd, rbuffer, sizeof(rbuffer)) == 0);
 129         assert_se(streq(rbuffer, ""));
 130
 131         fd = safe_close(fd);
 132
 133         random_bytes(wbuffer, sizeof(wbuffer));
 134
 135         fd = acquire_data_fd(wbuffer, sizeof(wbuffer), flags);
 136         assert_se(fd >= 0);
 137
 138         zero(rbuffer);
 139         assert_se(read(fd, rbuffer, sizeof(rbuffer)) == sizeof(rbuffer));
 140         assert_se(memcmp(rbuffer, wbuffer, sizeof(rbuffer)) == 0);
 141
 142         fd = safe_close(fd);
 143 }
 144
 145 static void test_acquire_data_fd(void) {
 146
 147         test_acquire_data_fd_one(0);
 148         test_acquire_data_fd_one(ACQUIRE_NO_DEV_NULL);
 149         test_acquire_data_fd_one(ACQUIRE_NO_MEMFD);
 150         test_acquire_data_fd_one(ACQUIRE_NO_DEV_NULL|ACQUIRE_NO_MEMFD);
 151         test_acquire_data_fd_one(ACQUIRE_NO_PIPE);
 152         test_acquire_data_fd_one(ACQUIRE_NO_DEV_NULL|ACQUIRE_NO_PIPE);
 153         test_acquire_data_fd_one(ACQUIRE_NO_MEMFD|ACQUIRE_NO_PIPE);
 154         test_acquire_data_fd_one(ACQUIRE_NO_DEV_NULL|ACQUIRE_NO_MEMFD|ACQUIRE_NO_PIPE);
 155         test_acquire_data_fd_one(ACQUIRE_NO_DEV_NULL|ACQUIRE_NO_MEMFD|ACQUIRE_NO_PIPE|ACQUIRE_NO_TMPFILE);
 156 }
 157
 158 int main(int argc, char *argv[]) {
 159         test_close_many();
 160         test_close_nointr();
 161         test_same_fd();
 162         test_open_serialization_fd();
 163         test_acquire_data_fd();
 164
 165         return 0;
 166 }