1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2013 Lennart Poettering
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.
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.
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/>.
27 static int prepare_handler(sd_event_source *s, void *userdata) {
28 log_info("preparing %c", PTR_TO_INT(userdata));
32 static bool got_a, got_b, got_c, got_unref;
33 static unsigned got_d;
35 static int unref_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
36 sd_event_source_unref(s);
41 static int io_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
43 log_info("got IO on %c", PTR_TO_INT(userdata));
45 if (userdata == INT_TO_PTR('a')) {
46 assert_se(sd_event_source_set_enabled(s, SD_EVENT_OFF) >= 0);
49 } else if (userdata == INT_TO_PTR('b')) {
52 } else if (userdata == INT_TO_PTR('d')) {
55 assert_se(sd_event_source_set_enabled(s, SD_EVENT_ONESHOT) >= 0);
57 assert_se(sd_event_source_set_enabled(s, SD_EVENT_OFF) >= 0);
59 assert_not_reached("Yuck!");
64 static int child_handler(sd_event_source *s, const siginfo_t *si, void *userdata) {
69 log_info("got child on %c", PTR_TO_INT(userdata));
71 assert_se(userdata == INT_TO_PTR('f'));
73 assert_se(sd_event_exit(sd_event_source_get_event(s), 0) >= 0);
74 sd_event_source_unref(s);
79 static int signal_handler(sd_event_source *s, const struct signalfd_siginfo *si, void *userdata) {
80 sd_event_source *p = NULL;
87 log_info("got signal on %c", PTR_TO_INT(userdata));
89 assert_se(userdata == INT_TO_PTR('e'));
91 assert_se(sigemptyset(&ss) >= 0);
92 assert_se(sigaddset(&ss, SIGCHLD) >= 0);
93 assert_se(sigprocmask(SIG_BLOCK, &ss, NULL) >= 0);
101 assert_se(sd_event_add_child(sd_event_source_get_event(s), &p, pid, WEXITED, child_handler, INT_TO_PTR('f')) >= 0);
102 assert_se(sd_event_source_set_enabled(p, SD_EVENT_ONESHOT) >= 0);
104 sd_event_source_unref(s);
109 static int defer_handler(sd_event_source *s, void *userdata) {
110 sd_event_source *p = NULL;
115 log_info("got defer on %c", PTR_TO_INT(userdata));
117 assert_se(userdata == INT_TO_PTR('d'));
119 assert_se(sigemptyset(&ss) >= 0);
120 assert_se(sigaddset(&ss, SIGUSR1) >= 0);
121 assert_se(sigprocmask(SIG_BLOCK, &ss, NULL) >= 0);
122 assert_se(sd_event_add_signal(sd_event_source_get_event(s), &p, SIGUSR1, signal_handler, INT_TO_PTR('e')) >= 0);
123 assert_se(sd_event_source_set_enabled(p, SD_EVENT_ONESHOT) >= 0);
126 sd_event_source_unref(s);
131 static bool do_quit = false;
133 static int time_handler(sd_event_source *s, uint64_t usec, void *userdata) {
134 log_info("got timer on %c", PTR_TO_INT(userdata));
136 if (userdata == INT_TO_PTR('c')) {
141 assert_se(sd_event_add_defer(sd_event_source_get_event(s), &p, defer_handler, INT_TO_PTR('d')) >= 0);
142 assert_se(sd_event_source_set_enabled(p, SD_EVENT_ONESHOT) >= 0);
148 assert_not_reached("Huh?");
153 static bool got_exit = false;
155 static int exit_handler(sd_event_source *s, void *userdata) {
156 log_info("got quit handler on %c", PTR_TO_INT(userdata));
163 int main(int argc, char *argv[]) {
165 sd_event_source *w = NULL, *x = NULL, *y = NULL, *z = NULL, *q = NULL, *t = NULL;
166 static const char ch = 'x';
167 int a[2] = { -1, -1 }, b[2] = { -1, -1}, d[2] = { -1, -1}, k[2] = { -1, -1 };
169 assert_se(pipe(a) >= 0);
170 assert_se(pipe(b) >= 0);
171 assert_se(pipe(d) >= 0);
172 assert_se(pipe(k) >= 0);
174 assert_se(sd_event_default(&e) >= 0);
176 assert_se(sd_event_set_watchdog(e, true) >= 0);
178 /* Test whether we cleanly can destroy an io event source from its own handler */
180 assert_se(sd_event_add_io(e, &t, k[0], EPOLLIN, unref_handler, NULL) >= 0);
181 assert_se(write(k[1], &ch, 1) == 1);
182 assert_se(sd_event_run(e, (uint64_t) -1) >= 1);
183 assert_se(got_unref);
185 got_a = false, got_b = false, got_c = false, got_d = 0;
187 /* Add a oneshot handler, trigger it, re-enable it, and trigger
189 assert_se(sd_event_add_io(e, &w, d[0], EPOLLIN, io_handler, INT_TO_PTR('d')) >= 0);
190 assert_se(sd_event_source_set_enabled(w, SD_EVENT_ONESHOT) >= 0);
191 assert_se(write(d[1], &ch, 1) >= 0);
192 assert_se(sd_event_run(e, (uint64_t) -1) >= 1);
193 assert_se(got_d == 1);
194 assert_se(write(d[1], &ch, 1) >= 0);
195 assert_se(sd_event_run(e, (uint64_t) -1) >= 1);
196 assert_se(got_d == 2);
198 assert_se(sd_event_add_io(e, &x, a[0], EPOLLIN, io_handler, INT_TO_PTR('a')) >= 0);
199 assert_se(sd_event_add_io(e, &y, b[0], EPOLLIN, io_handler, INT_TO_PTR('b')) >= 0);
200 assert_se(sd_event_add_time(e, &z, CLOCK_MONOTONIC, 0, 0, time_handler, INT_TO_PTR('c')) >= 0);
201 assert_se(sd_event_add_exit(e, &q, exit_handler, INT_TO_PTR('g')) >= 0);
203 assert_se(sd_event_source_set_priority(x, 99) >= 0);
204 assert_se(sd_event_source_set_enabled(y, SD_EVENT_ONESHOT) >= 0);
205 assert_se(sd_event_source_set_prepare(x, prepare_handler) >= 0);
206 assert_se(sd_event_source_set_priority(z, 50) >= 0);
207 assert_se(sd_event_source_set_enabled(z, SD_EVENT_ONESHOT) >= 0);
208 assert_se(sd_event_source_set_prepare(z, prepare_handler) >= 0);
210 /* Test for floating event sources */
211 assert_se(sigprocmask_many(SIG_BLOCK, SIGRTMIN+1, -1) == 0);
212 assert_se(sd_event_add_signal(e, NULL, SIGRTMIN+1, NULL, NULL) >= 0);
214 assert_se(write(a[1], &ch, 1) >= 0);
215 assert_se(write(b[1], &ch, 1) >= 0);
217 assert_se(!got_a && !got_b && !got_c);
219 assert_se(sd_event_run(e, (uint64_t) -1) >= 1);
221 assert_se(!got_a && got_b && !got_c);
223 assert_se(sd_event_run(e, (uint64_t) -1) >= 1);
225 assert_se(!got_a && got_b && got_c);
227 assert_se(sd_event_run(e, (uint64_t) -1) >= 1);
229 assert_se(got_a && got_b && got_c);
231 sd_event_source_unref(x);
232 sd_event_source_unref(y);
235 assert_se(sd_event_source_set_time(z, now(CLOCK_MONOTONIC) + 200 * USEC_PER_MSEC) >= 0);
236 assert_se(sd_event_source_set_enabled(z, SD_EVENT_ONESHOT) >= 0);
238 assert_se(sd_event_loop(e) >= 0);
240 sd_event_source_unref(z);
241 sd_event_source_unref(q);
243 sd_event_source_unref(w);