]>
Commit | Line | Data |
---|---|---|
53e1b683 | 1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
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" |
e4de7287 | 9 | #include "tmpfile-util.h" |
3ffd4af2 | 10 | #include "util.h" |
106ecd76 RC |
11 | |
12 | static void test_fdset_new_fill(void) { | |
13 | int fd = -1; | |
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 | ||
25 | static void test_fdset_put_dup(void) { | |
26 | _cleanup_close_ int fd = -1; | |
27 | int copyfd = -1; | |
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 | ||
44 | static void test_fdset_cloexec(void) { | |
45 | int fd = -1; | |
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 | ||
70 | static void test_fdset_close_others(void) { | |
71 | int fd = -1; | |
72 | int copyfd = -1; | |
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 | ||
2de61bbe RC |
94 | static void test_fdset_remove(void) { |
95 | _cleanup_close_ int fd = -1; | |
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 | ||
d7aeffea RC |
114 | static void test_fdset_iterate(void) { |
115 | int fd = -1; | |
116 | FDSet *fdset = NULL; | |
117 | char name[] = "/tmp/test-fdset_iterate.XXXXXX"; | |
118 | Iterator i; | |
119 | int c = 0; | |
120 | int a; | |
121 | ||
646853bd | 122 | fd = mkostemp_safe(name); |
d7aeffea RC |
123 | assert_se(fd >= 0); |
124 | ||
125 | fdset = fdset_new(); | |
126 | assert_se(fdset); | |
127 | assert_se(fdset_put(fdset, fd) >= 0); | |
128 | assert_se(fdset_put(fdset, fd) >= 0); | |
129 | assert_se(fdset_put(fdset, fd) >= 0); | |
130 | ||
131 | FDSET_FOREACH(a, fdset, i) { | |
132 | c++; | |
133 | assert_se(a == fd); | |
134 | } | |
135 | assert_se(c == 1); | |
136 | ||
137 | fdset_free(fdset); | |
138 | ||
139 | unlink(name); | |
140 | } | |
141 | ||
0805e9a9 RC |
142 | static void test_fdset_isempty(void) { |
143 | int fd; | |
144 | _cleanup_fdset_free_ FDSet *fdset = NULL; | |
145 | char name[] = "/tmp/test-fdset_isempty.XXXXXX"; | |
146 | ||
646853bd | 147 | fd = mkostemp_safe(name); |
0805e9a9 RC |
148 | assert_se(fd >= 0); |
149 | ||
150 | fdset = fdset_new(); | |
151 | assert_se(fdset); | |
152 | ||
153 | assert_se(fdset_isempty(fdset)); | |
154 | assert_se(fdset_put(fdset, fd) >= 0); | |
155 | assert_se(!fdset_isempty(fdset)); | |
156 | ||
157 | unlink(name); | |
158 | } | |
159 | ||
160 | static void test_fdset_steal_first(void) { | |
161 | int fd; | |
162 | _cleanup_fdset_free_ FDSet *fdset = NULL; | |
163 | char name[] = "/tmp/test-fdset_steal_first.XXXXXX"; | |
164 | ||
646853bd | 165 | fd = mkostemp_safe(name); |
0805e9a9 RC |
166 | assert_se(fd >= 0); |
167 | ||
168 | fdset = fdset_new(); | |
169 | assert_se(fdset); | |
170 | ||
171 | assert_se(fdset_steal_first(fdset) < 0); | |
172 | assert_se(fdset_put(fdset, fd) >= 0); | |
173 | assert_se(fdset_steal_first(fdset) == fd); | |
174 | assert_se(fdset_steal_first(fdset) < 0); | |
175 | assert_se(fdset_put(fdset, fd) >= 0); | |
176 | ||
177 | unlink(name); | |
178 | } | |
179 | ||
180 | static void test_fdset_new_array(void) { | |
181 | int fds[] = {10, 11, 12, 13}; | |
182 | _cleanup_fdset_free_ FDSet *fdset = NULL; | |
183 | ||
184 | assert_se(fdset_new_array(&fdset, fds, 4) >= 0); | |
185 | assert_se(fdset_size(fdset) == 4); | |
186 | assert_se(fdset_contains(fdset, 10)); | |
187 | assert_se(fdset_contains(fdset, 11)); | |
188 | assert_se(fdset_contains(fdset, 12)); | |
189 | assert_se(fdset_contains(fdset, 13)); | |
190 | } | |
191 | ||
106ecd76 RC |
192 | int main(int argc, char *argv[]) { |
193 | test_fdset_new_fill(); | |
194 | test_fdset_put_dup(); | |
195 | test_fdset_cloexec(); | |
196 | test_fdset_close_others(); | |
2de61bbe | 197 | test_fdset_remove(); |
d7aeffea | 198 | test_fdset_iterate(); |
0805e9a9 RC |
199 | test_fdset_isempty(); |
200 | test_fdset_steal_first(); | |
201 | test_fdset_new_array(); | |
4630bbb7 RC |
202 | |
203 | return 0; | |
106ecd76 | 204 | } |