src/libsystemd-bus/test-event.c

	test-event.c revision 6203e07a83214a55bb1f88508fcda2005c601dea
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani/***
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  This file is part of systemd.
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  Copyright 2013 Lennart Poettering
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  systemd is free software; you can redistribute it and/or modify it
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  under the terms of the GNU Lesser General Public License as published by
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  the Free Software Foundation; either version 2.1 of the License, or
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  (at your option) any later version.
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  systemd is distributed in the hope that it will be useful, but
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  WITHOUT ANY WARRANTY; without even the implied warranty of
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  Lesser General Public License for more details.
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  You should have received a copy of the GNU Lesser General Public License
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani  along with systemd; If not, see <http://www.gnu.org/licenses/>.
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani***/
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani#include "sd-event.h"
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani#include "log.h"
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani#include "util.h"
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahanistatic int prepare_handler(sd_event_source *s, void *userdata) {
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        log_info("preparing %c", PTR_TO_INT(userdata));
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        return 1;
0037c2dc54ca5bc5a3ec428619ff136066467565Beniamino Galvani}
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahanistatic bool got_a, got_b, got_c;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahanistatic unsigned got_d;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
6b3fd9a14d378a35d95ec4be65065708d092bee2Torstein Husebøstatic int io_handler(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        log_info("got IO on %c", PTR_TO_INT(userdata));
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        if (userdata == INT_TO_PTR('a')) {
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                assert_se(sd_event_source_set_enabled(s, SD_EVENT_OFF) >= 0);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                assert_se(!got_a);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                got_a = true;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        } else if (userdata == INT_TO_PTR('b')) {
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                assert_se(!got_b);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                got_b = true;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        } else if (userdata == INT_TO_PTR('d')) {
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                got_d++;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                if (got_d < 2)
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                        assert_se(sd_event_source_set_enabled(s, SD_EVENT_ONESHOT) >= 0);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                else
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                        assert_se(sd_event_source_set_enabled(s, SD_EVENT_OFF) >= 0);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        } else
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                assert_not_reached("Yuck!");
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        return 1;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani}
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahanistatic int child_handler(sd_event_source *s, const siginfo_t *si, void *userdata) {
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert(s);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert(si);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        log_info("got child on %c", PTR_TO_INT(userdata));
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert(userdata == INT_TO_PTR('f'));
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
9c8e3101ceef00342263d57dce3fa61956824985Lennart Poettering        assert_se(sd_event_exit(sd_event_source_get_event(s), 0) >= 0);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        sd_event_source_unref(s);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        return 1;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani}
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahanistatic int signal_handler(sd_event_source *s, const struct signalfd_siginfo *si, void *userdata) {
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        sd_event_source *p;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        sigset_t ss;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        pid_t pid;
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert(s);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert(si);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        log_info("got signal on %c", PTR_TO_INT(userdata));
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert(userdata == INT_TO_PTR('e'));
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert_se(sigemptyset(&ss) >= 0);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert_se(sigaddset(&ss, SIGCHLD) >= 0);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert_se(sigprocmask(SIG_BLOCK, &ss, NULL) >= 0);
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        pid = fork();
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani        assert_se(pid >= 0);
7a6f1457462840ed52f2a332e2bd56ca210cff2dTom Gundersen
0037c2dc54ca5bc5a3ec428619ff136066467565Beniamino Galvani        if (pid == 0)
ad1ad5c8e36ea795034fcdac660b15d7c141d55bSusant Sahani                _exit(0);

        assert_se(sd_event_add_child(sd_event_source_get_event(s), pid, WEXITED, child_handler, INT_TO_PTR('f'), &p) >= 0);
        assert_se(sd_event_source_set_enabled(p, SD_EVENT_ONESHOT) >= 0);

        sd_event_source_unref(s);

        return 1;
}

static int defer_handler(sd_event_source *s, void *userdata) {
        sd_event_source *p;
        sigset_t ss;

        assert(s);

        log_info("got defer on %c", PTR_TO_INT(userdata));

        assert(userdata == INT_TO_PTR('d'));

        assert_se(sigemptyset(&ss) >= 0);
        assert_se(sigaddset(&ss, SIGUSR1) >= 0);
        assert_se(sigprocmask(SIG_BLOCK, &ss, NULL) >= 0);
        assert_se(sd_event_add_signal(sd_event_source_get_event(s), SIGUSR1, signal_handler, INT_TO_PTR('e'), &p) >= 0);
        assert_se(sd_event_source_set_enabled(p, SD_EVENT_ONESHOT) >= 0);
        raise(SIGUSR1);

        sd_event_source_unref(s);

        return 1;
}

static bool do_quit = false;

static int time_handler(sd_event_source *s, uint64_t usec, void *userdata) {
        log_info("got timer on %c", PTR_TO_INT(userdata));

        if (userdata == INT_TO_PTR('c')) {

                if (do_quit) {
                        sd_event_source *p;

                        assert_se(sd_event_add_defer(sd_event_source_get_event(s), defer_handler, INT_TO_PTR('d'), &p) >= 0);
                        assert_se(sd_event_source_set_enabled(p, SD_EVENT_ONESHOT) >= 0);
                } else {
                        assert(!got_c);
                        got_c = true;
                }
        } else
                assert_not_reached("Huh?");

        return 2;
}

static bool got_exit = false;

static int exit_handler(sd_event_source *s, void *userdata) {
        log_info("got quit handler on %c", PTR_TO_INT(userdata));

        got_exit = true;

        return 3;
}

int main(int argc, char *argv[]) {
        sd_event *e = NULL;
        sd_event_source *w = NULL, *x = NULL, *y = NULL, *z = NULL, *q = NULL;
        static const char ch = 'x';
        int a[2] = { -1, -1 }, b[2] = { -1, -1}, d[2] = { -1, -1};

        assert_se(pipe(a) >= 0);
        assert_se(pipe(b) >= 0);
        assert_se(pipe(d) >= 0);

        assert_se(sd_event_default(&e) >= 0);

        assert_se(sd_event_set_watchdog(e, true) >= 0);

        got_a = false, got_b = false, got_c = false, got_d = 0;

        /* Add a oneshot handler, trigger it, re-enable it, and trigger
         * it again. */
        assert_se(sd_event_add_io(e, d[0], EPOLLIN, io_handler, INT_TO_PTR('d'), &w) >= 0);
        assert_se(sd_event_source_set_enabled(w, SD_EVENT_ONESHOT) >= 0);
        assert_se(write(d[1], &ch, 1) >= 0);
        assert_se(sd_event_run(e, (uint64_t) -1) >= 1);
        assert_se(got_d == 1);
        assert_se(write(d[1], &ch, 1) >= 0);
        assert_se(sd_event_run(e, (uint64_t) -1) >= 1);
        assert_se(got_d == 2);

        assert_se(sd_event_add_io(e, a[0], EPOLLIN, io_handler, INT_TO_PTR('a'), &x) >= 0);
        assert_se(sd_event_add_io(e, b[0], EPOLLIN, io_handler, INT_TO_PTR('b'), &y) >= 0);
        assert_se(sd_event_add_monotonic(e, 0, 0, time_handler, INT_TO_PTR('c'), &z) >= 0);
        assert_se(sd_event_add_exit(e, exit_handler, INT_TO_PTR('g'), &q) >= 0);

        assert_se(sd_event_source_set_priority(x, 99) >= 0);
        assert_se(sd_event_source_set_enabled(y, SD_EVENT_ONESHOT) >= 0);
        assert_se(sd_event_source_set_prepare(x, prepare_handler) >= 0);
        assert_se(sd_event_source_set_priority(z, 50) >= 0);
        assert_se(sd_event_source_set_enabled(z, SD_EVENT_ONESHOT) >= 0);
        assert_se(sd_event_source_set_prepare(z, prepare_handler) >= 0);

        assert_se(write(a[1], &ch, 1) >= 0);
        assert_se(write(b[1], &ch, 1) >= 0);

        assert_se(!got_a && !got_b && !got_c);

        assert_se(sd_event_run(e, (uint64_t) -1) >= 1);

        assert_se(!got_a && got_b && !got_c);

        assert_se(sd_event_run(e, (uint64_t) -1) >= 1);

        assert_se(!got_a && got_b && got_c);

        assert_se(sd_event_run(e, (uint64_t) -1) >= 1);

        assert_se(got_a && got_b && got_c);

        sd_event_source_unref(x);
        sd_event_source_unref(y);

        do_quit = true;
        assert_se(sd_event_source_set_time(z, now(CLOCK_MONOTONIC) + 200 * USEC_PER_MSEC) >= 0);
        assert_se(sd_event_source_set_enabled(z, SD_EVENT_ONESHOT) >= 0);

        assert_se(sd_event_loop(e) >= 0);

        sd_event_source_unref(z);
        sd_event_source_unref(q);

        sd_event_source_unref(w);

        sd_event_unref(e);

        close_pipe(a);
        close_pipe(b);

        return 0;
}