Nils O. Selåsdal
108 lines
2.5 KiB
C
108 lines
2.5 KiB
C
#include <check.h>
|
|
#include <signal.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include "ucore/iomux_signal.h"
|
|
|
|
|
|
static enum IOMUX_TYPE g_iomux_type = IOMUX_TYPE_DEFAULT;
|
|
static int g_usr1_seen;
|
|
static int g_usr2_seen;
|
|
static void set_iomux_type_select(void)
|
|
{
|
|
g_iomux_type = IOMUX_TYPE_SELECT;
|
|
g_usr2_seen = g_usr1_seen = 0;
|
|
}
|
|
|
|
static void set_iomux_type_epoll(void)
|
|
{
|
|
g_iomux_type = IOMUX_TYPE_EPOLL;
|
|
g_usr2_seen = g_usr1_seen = 0;
|
|
}
|
|
|
|
|
|
static void usr1_handler(struct IOMux *mux, int signo, void *cookie)
|
|
{
|
|
ck_assert_int_eq(signo, SIGUSR1);
|
|
g_usr1_seen++;
|
|
raise(SIGUSR2);
|
|
}
|
|
|
|
static void usr2_handler(struct IOMux *mux, int signo, void *cookie)
|
|
{
|
|
int *val = cookie;
|
|
int rc;
|
|
|
|
ck_assert_int_eq(signo, SIGUSR2);
|
|
ck_assert_int_eq(*val, 0x1234);
|
|
g_usr2_seen++;
|
|
|
|
rc = iomux_unregister_unixsignal(mux, SIGUSR2);
|
|
ck_assert_int_eq(rc, 0);
|
|
|
|
rc = iomux_unregister_unixsignal(mux, SIGUSR1);
|
|
ck_assert_int_eq(rc, 0);
|
|
}
|
|
|
|
START_TEST (test_iomux_signal_simple)
|
|
|
|
|
|
struct IOMux *mux = iomux_create(g_iomux_type);
|
|
int rc;
|
|
int cookie = 0x1234;
|
|
|
|
//can't handle SIGKILL
|
|
rc = iomux_register_unixsignal(mux, SIGKILL, usr1_handler, NULL);
|
|
ck_assert_int_ne(rc, 0);
|
|
|
|
rc = iomux_register_unixsignal(mux, SIGUSR1, usr1_handler, NULL);
|
|
ck_assert_int_eq(rc, 0);
|
|
|
|
//not priorly registred
|
|
rc = iomux_unregister_unixsignal(mux, SIGINT);
|
|
ck_assert_int_ne(rc, 0);
|
|
|
|
rc = iomux_unregister_unixsignal(mux, SIGUSR1);
|
|
ck_assert_int_eq(rc, 0);
|
|
|
|
rc = iomux_register_unixsignal(mux, SIGUSR1, usr1_handler, NULL);
|
|
ck_assert_int_eq(rc, 0);
|
|
|
|
rc = iomux_register_unixsignal(mux, SIGUSR2, usr2_handler, &cookie);
|
|
ck_assert_int_eq(rc, 0);
|
|
|
|
//already registred
|
|
rc = iomux_register_unixsignal(mux, SIGUSR1, usr1_handler, NULL);
|
|
ck_assert_int_ne(rc, 0);
|
|
|
|
raise(SIGUSR1);
|
|
iomux_run(mux);
|
|
|
|
|
|
iomux_delete(mux);
|
|
|
|
ck_assert_int_eq(g_usr1_seen, 1);
|
|
ck_assert_int_eq(g_usr2_seen, 1);
|
|
END_TEST
|
|
|
|
|
|
Suite *iomux_signal_suite(void)
|
|
{
|
|
Suite *s = suite_create("iomux_signal");
|
|
TCase *select_tc = tcase_create("iomux signal select tests");
|
|
TCase *epoll_tc = tcase_create("iomux signal epoll tests");
|
|
|
|
tcase_add_test(select_tc, test_iomux_signal_simple);
|
|
tcase_add_test(epoll_tc, test_iomux_signal_simple);
|
|
|
|
tcase_add_checked_fixture(select_tc, set_iomux_type_select, NULL);
|
|
|
|
tcase_add_checked_fixture(epoll_tc, set_iomux_type_epoll, NULL);
|
|
|
|
suite_add_tcase(s, select_tc);
|
|
suite_add_tcase(s, epoll_tc);
|
|
|
|
return s;
|
|
}
|
|
|