[thirdparty/systemd.git] / src / test / test-fdset.c

/* SPDX-License-Identifier: LGPL-2.1-or-later */

#include <fcntl.h>
#include <unistd.h>

#include "fd-util.h"
#include "fdset.h"
#include "macro.h"
#include "tests.h"
#include "tmpfile-util.h"

TEST(fdset_new_fill) {
        int fd = -EBADF;
        _cleanup_fdset_free_ FDSet *fdset = NULL;
        char name[] = "/tmp/test-fdset_new_fill.XXXXXX";

        fd = mkostemp_safe(name);
        assert_se(fd >= 0);
        assert_se(fdset_new_fill(&fdset) >= 0);
        assert_se(fdset_contains(fdset, fd));

        unlink(name);
}

TEST(fdset_put_dup) {
        _cleanup_close_ int fd = -EBADF;
        int copyfd = -EBADF;
        _cleanup_fdset_free_ FDSet *fdset = NULL;
        char name[] = "/tmp/test-fdset_put_dup.XXXXXX";

        fd = mkostemp_safe(name);
        assert_se(fd >= 0);

        fdset = fdset_new();
        assert_se(fdset);
        copyfd = fdset_put_dup(fdset, fd);
        assert_se(copyfd >= 0 && copyfd != fd);
        assert_se(fdset_contains(fdset, copyfd));
        assert_se(!fdset_contains(fdset, fd));

        unlink(name);
}

TEST(fdset_cloexec) {
        int fd = -EBADF;
        _cleanup_fdset_free_ FDSet *fdset = NULL;
        int flags = -1;
        char name[] = "/tmp/test-fdset_cloexec.XXXXXX";

        fd = mkostemp_safe(name);
        assert_se(fd >= 0);

        fdset = fdset_new();
        assert_se(fdset);
        assert_se(fdset_put(fdset, fd));

        assert_se(fdset_cloexec(fdset, false) >= 0);
        flags = fcntl(fd, F_GETFD);
        assert_se(flags >= 0);
        assert_se(!(flags & FD_CLOEXEC));

        assert_se(fdset_cloexec(fdset, true) >= 0);
        flags = fcntl(fd, F_GETFD);
        assert_se(flags >= 0);
        assert_se(flags & FD_CLOEXEC);

        unlink(name);
}

TEST(fdset_close_others) {
        int fd = -EBADF;
        int copyfd = -EBADF;
        _cleanup_fdset_free_ FDSet *fdset = NULL;
        int flags = -1;
        char name[] = "/tmp/test-fdset_close_others.XXXXXX";

        fd = mkostemp_safe(name);
        assert_se(fd >= 0);

        fdset = fdset_new();
        assert_se(fdset);
        copyfd = fdset_put_dup(fdset, fd);
        assert_se(copyfd >= 0);

        assert_se(fdset_close_others(fdset) >= 0);
        flags = fcntl(fd, F_GETFD);
        assert_se(flags < 0);
        flags = fcntl(copyfd, F_GETFD);
        assert_se(flags >= 0);

        unlink(name);
}

TEST(fdset_remove) {
        _cleanup_close_ int fd = -EBADF;
        FDSet *fdset = NULL;
        char name[] = "/tmp/test-fdset_remove.XXXXXX";

        fd = mkostemp_safe(name);
        assert_se(fd >= 0);

        fdset = fdset_new();
        assert_se(fdset);
        assert_se(fdset_put(fdset, fd) >= 0);
        assert_se(fdset_remove(fdset, fd) >= 0);
        assert_se(!fdset_contains(fdset, fd));
        fdset_free(fdset);

        assert_se(fcntl(fd, F_GETFD) >= 0);

        unlink(name);
}

TEST(fdset_iterate) {
        int fd = -EBADF;
        FDSet *fdset = NULL;
        char name[] = "/tmp/test-fdset_iterate.XXXXXX";
        int c = 0;
        int a;

        fd = mkostemp_safe(name);
        assert_se(fd >= 0);

        fdset = fdset_new();
        assert_se(fdset);
        assert_se(fdset_put(fdset, fd) >= 0);
        assert_se(fdset_put(fdset, fd) >= 0);
        assert_se(fdset_put(fdset, fd) >= 0);

        FDSET_FOREACH(a, fdset) {
                c++;
                assert_se(a == fd);
        }
        assert_se(c == 1);

        fdset_free(fdset);

        unlink(name);
}

TEST(fdset_isempty) {
        int fd;
        _cleanup_fdset_free_ FDSet *fdset = NULL;
        char name[] = "/tmp/test-fdset_isempty.XXXXXX";

        fd = mkostemp_safe(name);
        assert_se(fd >= 0);

        fdset = fdset_new();
        assert_se(fdset);

        assert_se(fdset_isempty(fdset));
        assert_se(fdset_put(fdset, fd) >= 0);
        assert_se(!fdset_isempty(fdset));

        unlink(name);
}

TEST(fdset_steal_first) {
        int fd;
        _cleanup_fdset_free_ FDSet *fdset = NULL;
        char name[] = "/tmp/test-fdset_steal_first.XXXXXX";

        fd = mkostemp_safe(name);
        assert_se(fd >= 0);

        fdset = fdset_new();
        assert_se(fdset);

        assert_se(fdset_steal_first(fdset) < 0);
        assert_se(fdset_put(fdset, fd) >= 0);
        assert_se(fdset_steal_first(fdset) == fd);
        assert_se(fdset_steal_first(fdset) < 0);
        assert_se(fdset_put(fdset, fd) >= 0);

        unlink(name);
}

TEST(fdset_new_array) {
        int fds[] = {10, 11, 12, 13};
        _cleanup_fdset_free_ FDSet *fdset = NULL;

        assert_se(fdset_new_array(&fdset, fds, 4) >= 0);
        assert_se(fdset_size(fdset) == 4);
        assert_se(fdset_contains(fdset, 10));
        assert_se(fdset_contains(fdset, 11));
        assert_se(fdset_contains(fdset, 12));
        assert_se(fdset_contains(fdset, 13));
}

DEFINE_TEST_MAIN(LOG_INFO);
Commit	Line	Data
db9ecf05	1	/* SPDX-License-Identifier: LGPL-2.1-or-later */
106ecd76 RC	2
	3	#include <fcntl.h>
	4	#include <unistd.h>
	5
3ffd4af2	6	#include "fd-util.h"
106ecd76	7	#include "fdset.h"
106ecd76	8	#include "macro.h"
4f7452a8	9	#include "tests.h"
e4de7287	10	#include "tmpfile-util.h"
106ecd76	11
4f7452a8	12	TEST(fdset_new_fill) {
254d1313	13	int fd = -EBADF;
106ecd76 RC	14	_cleanup_fdset_free_ FDSet *fdset = NULL;
	15	char name[] = "/tmp/test-fdset_new_fill.XXXXXX";
	16
646853bd	17	fd = mkostemp_safe(name);
106ecd76 RC	18	assert_se(fd >= 0);
	19	assert_se(fdset_new_fill(&fdset) >= 0);
	20	assert_se(fdset_contains(fdset, fd));
	21
	22	unlink(name);
	23	}
	24
4f7452a8	25	TEST(fdset_put_dup) {
254d1313 ZJS	26	_cleanup_close_ int fd = -EBADF;
254d1313 ZJS	27	int copyfd = -EBADF;
106ecd76 RC	28	_cleanup_fdset_free_ FDSet *fdset = NULL;
	29	char name[] = "/tmp/test-fdset_put_dup.XXXXXX";
	30
646853bd	31	fd = mkostemp_safe(name);
106ecd76 RC	32	assert_se(fd >= 0);
	33
	34	fdset = fdset_new();
	35	assert_se(fdset);
	36	copyfd = fdset_put_dup(fdset, fd);
	37	assert_se(copyfd >= 0 && copyfd != fd);
	38	assert_se(fdset_contains(fdset, copyfd));
	39	assert_se(!fdset_contains(fdset, fd));
	40
	41	unlink(name);
	42	}
	43
4f7452a8	44	TEST(fdset_cloexec) {
254d1313	45	int fd = -EBADF;
106ecd76 RC	46	_cleanup_fdset_free_ FDSet *fdset = NULL;
	47	int flags = -1;
	48	char name[] = "/tmp/test-fdset_cloexec.XXXXXX";
	49
646853bd	50	fd = mkostemp_safe(name);
106ecd76 RC	51	assert_se(fd >= 0);
	52
	53	fdset = fdset_new();
	54	assert_se(fdset);
	55	assert_se(fdset_put(fdset, fd));
	56
	57	assert_se(fdset_cloexec(fdset, false) >= 0);
	58	flags = fcntl(fd, F_GETFD);
	59	assert_se(flags >= 0);
	60	assert_se(!(flags & FD_CLOEXEC));
	61
	62	assert_se(fdset_cloexec(fdset, true) >= 0);
	63	flags = fcntl(fd, F_GETFD);
	64	assert_se(flags >= 0);
	65	assert_se(flags & FD_CLOEXEC);
	66
	67	unlink(name);
	68	}
	69
4f7452a8	70	TEST(fdset_close_others) {
254d1313 ZJS	71	int fd = -EBADF;
254d1313 ZJS	72	int copyfd = -EBADF;
106ecd76 RC	73	_cleanup_fdset_free_ FDSet *fdset = NULL;
	74	int flags = -1;
	75	char name[] = "/tmp/test-fdset_close_others.XXXXXX";
	76
646853bd	77	fd = mkostemp_safe(name);
106ecd76 RC	78	assert_se(fd >= 0);
	79
	80	fdset = fdset_new();
	81	assert_se(fdset);
	82	copyfd = fdset_put_dup(fdset, fd);
	83	assert_se(copyfd >= 0);
	84
	85	assert_se(fdset_close_others(fdset) >= 0);
	86	flags = fcntl(fd, F_GETFD);
	87	assert_se(flags < 0);
	88	flags = fcntl(copyfd, F_GETFD);
	89	assert_se(flags >= 0);
	90
	91	unlink(name);
	92	}
	93
4f7452a8	94	TEST(fdset_remove) {
254d1313	95	_cleanup_close_ int fd = -EBADF;
2de61bbe RC	96	FDSet *fdset = NULL;
	97	char name[] = "/tmp/test-fdset_remove.XXXXXX";
	98
646853bd	99	fd = mkostemp_safe(name);
2de61bbe RC	100	assert_se(fd >= 0);
	101
	102	fdset = fdset_new();
	103	assert_se(fdset);
	104	assert_se(fdset_put(fdset, fd) >= 0);
	105	assert_se(fdset_remove(fdset, fd) >= 0);
	106	assert_se(!fdset_contains(fdset, fd));
	107	fdset_free(fdset);
	108
	109	assert_se(fcntl(fd, F_GETFD) >= 0);
	110
	111	unlink(name);
	112	}
	113
4f7452a8	114	TEST(fdset_iterate) {
254d1313	115	int fd = -EBADF;
d7aeffea RC	116	FDSet *fdset = NULL;
d7aeffea RC	117	char name[] = "/tmp/test-fdset_iterate.XXXXXX";
d7aeffea RC	118	int c = 0;
	119	int a;
	120
646853bd	121	fd = mkostemp_safe(name);
d7aeffea RC	122	assert_se(fd >= 0);
	123
	124	fdset = fdset_new();
	125	assert_se(fdset);
	126	assert_se(fdset_put(fdset, fd) >= 0);
	127	assert_se(fdset_put(fdset, fd) >= 0);
	128	assert_se(fdset_put(fdset, fd) >= 0);
	129
90e74a66	130	FDSET_FOREACH(a, fdset) {
d7aeffea RC	131	c++;
	132	assert_se(a == fd);
	133	}
	134	assert_se(c == 1);
	135
	136	fdset_free(fdset);
	137
	138	unlink(name);
	139	}
	140
4f7452a8	141	TEST(fdset_isempty) {
0805e9a9 RC	142	int fd;
	143	_cleanup_fdset_free_ FDSet *fdset = NULL;
	144	char name[] = "/tmp/test-fdset_isempty.XXXXXX";
	145
646853bd	146	fd = mkostemp_safe(name);
0805e9a9 RC	147	assert_se(fd >= 0);
	148
	149	fdset = fdset_new();
	150	assert_se(fdset);
	151
	152	assert_se(fdset_isempty(fdset));
	153	assert_se(fdset_put(fdset, fd) >= 0);
	154	assert_se(!fdset_isempty(fdset));
	155
	156	unlink(name);
	157	}
	158
4f7452a8	159	TEST(fdset_steal_first) {
0805e9a9 RC	160	int fd;
	161	_cleanup_fdset_free_ FDSet *fdset = NULL;
	162	char name[] = "/tmp/test-fdset_steal_first.XXXXXX";
	163
646853bd	164	fd = mkostemp_safe(name);
0805e9a9 RC	165	assert_se(fd >= 0);
	166
	167	fdset = fdset_new();
	168	assert_se(fdset);
	169
	170	assert_se(fdset_steal_first(fdset) < 0);
	171	assert_se(fdset_put(fdset, fd) >= 0);
	172	assert_se(fdset_steal_first(fdset) == fd);
	173	assert_se(fdset_steal_first(fdset) < 0);
	174	assert_se(fdset_put(fdset, fd) >= 0);
	175
	176	unlink(name);
	177	}
	178
4f7452a8	179	TEST(fdset_new_array) {
0805e9a9 RC	180	int fds[] = {10, 11, 12, 13};
	181	_cleanup_fdset_free_ FDSet *fdset = NULL;
	182
	183	assert_se(fdset_new_array(&fdset, fds, 4) >= 0);
	184	assert_se(fdset_size(fdset) == 4);
	185	assert_se(fdset_contains(fdset, 10));
	186	assert_se(fdset_contains(fdset, 11));
	187	assert_se(fdset_contains(fdset, 12));
	188	assert_se(fdset_contains(fdset, 13));
	189	}
	190
4f7452a8	191	DEFINE_TEST_MAIN(LOG_INFO);