diff options
Diffstat (limited to 'tests/test_tevent_trace.c')
| -rw-r--r-- | tests/test_tevent_trace.c | 1324 |
1 files changed, 1324 insertions, 0 deletions
diff --git a/tests/test_tevent_trace.c b/tests/test_tevent_trace.c new file mode 100644 index 0000000..10e1ee0 --- /dev/null +++ b/tests/test_tevent_trace.c @@ -0,0 +1,1324 @@ +/* + * Unix SMB/CIFS implementation. + * + * testing of some tevent_req aspects + * + * Copyright (C) Pavel Březina <pbrezina@redhat.com> 2021 + * + * ** NOTE! The following LGPL license applies to the tevent + * ** library. This does NOT imply that all of Samba is released + * ** under the LGPL + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 3 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, see <http://www.gnu.org/licenses/>. + */ + +#include <errno.h> +#include <setjmp.h> +#include <stdlib.h> +#include <stdint.h> +#include <signal.h> +#include <unistd.h> + +#include <talloc.h> +#include <tevent.h> +#include <cmocka.h> + +struct test_ctx { + struct tevent_context *ev; + + bool handler_skipped; + bool reattach_reset; + + uint64_t (*get_tag)(const void *event); + void (*set_tag)(void *event, uint64_t tag); + uint64_t current_tag; + + bool attach; + bool before_handler; + bool handler_called; + bool detach; +}; + +static void fd_handler(struct tevent_context *ev, + struct tevent_fd *fde, + uint16_t flags, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_fd_get_tag(fde); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + return; +} + +static void timer_handler(struct tevent_context *ev, + struct tevent_timer *te, + struct timeval current_time, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_timer_get_tag(te); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + return; +} + +static void signal_handler(struct tevent_context *ev, + struct tevent_signal *se, + int signum, + int count, + void *siginfo, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_signal_get_tag(se); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + return; +} + +static void immediate_handler(struct tevent_context *ctx, + struct tevent_immediate *im, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_immediate_get_tag(im); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + return; +} + +static void immediate_handler_reschedule(struct tevent_context *ctx, + struct tevent_immediate *im, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_immediate_get_tag(im); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + + assert_false(tctx->reattach_reset); + tctx->reattach_reset = true; + tevent_schedule_immediate(im, tctx->ev, immediate_handler, tctx); + assert_false(tctx->reattach_reset); + assert_false(tctx->handler_skipped); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + assert_int_not_equal(tag, tctx->current_tag); + tag = tevent_immediate_get_tag(im); + assert_int_equal(tag, tctx->current_tag); + + tctx->handler_skipped = true; + tctx->reattach_reset = true; + tevent_schedule_immediate(im, tctx->ev, immediate_handler, tctx); + assert_false(tctx->reattach_reset); + assert_false(tctx->handler_skipped); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + assert_int_not_equal(tag, tctx->current_tag); + tag = tevent_immediate_get_tag(im); + assert_int_equal(tag, tctx->current_tag); +} + +static void fd_handler_free(struct tevent_context *ev, + struct tevent_fd *fde, + uint16_t flags, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_fd_get_tag(fde); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + TALLOC_FREE(fde); + assert_true(tctx->detach); + return; +} + +static void timer_handler_free(struct tevent_context *ev, + struct tevent_timer *te, + struct timeval current_time, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_timer_get_tag(te); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + TALLOC_FREE(te); + assert_true(tctx->detach); + return; +} + +static void signal_handler_free(struct tevent_context *ev, + struct tevent_signal *se, + int signum, + int count, + void *siginfo, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_signal_get_tag(se); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + TALLOC_FREE(se); + assert_true(tctx->detach); + return; +} + +static void immediate_handler_free(struct tevent_context *ctx, + struct tevent_immediate *im, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tevent_immediate_get_tag(im); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->handler_called = true; + assert_int_equal(tag, tctx->current_tag); + TALLOC_FREE(im); + assert_true(tctx->detach); + return; +} + +static void trace_event_cb(void *event, + enum tevent_event_trace_point point, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag = tctx->get_tag(event); + + switch (point) { + case TEVENT_EVENT_TRACE_ATTACH: + if (tctx->reattach_reset) { + assert_true(tctx->attach); + assert_true(tctx->detach); + tctx->attach = false; + tctx->before_handler = false; + tctx->handler_called = false; + tctx->detach = false; + tctx->handler_skipped = false; + tctx->reattach_reset = false; + } + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->attach = true; + assert_int_equal(tag, tctx->current_tag); + tag = ++tctx->current_tag; + tctx->set_tag(event, tag); + break; + case TEVENT_EVENT_TRACE_BEFORE_HANDLER: + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + tctx->before_handler = true; + assert_int_equal(tag, tctx->current_tag); + break; + case TEVENT_EVENT_TRACE_DETACH: + assert_true(tctx->attach); + if (tctx->handler_skipped) { + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + } else { + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + } + assert_false(tctx->detach); + tctx->detach = true; + assert_int_equal(tag, tctx->current_tag); + break; + } +} + +static void trace_event_cb1(void *event, + enum tevent_event_trace_point point, + void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + uint64_t tag; + + switch (point) { + case TEVENT_EVENT_TRACE_ATTACH: + tctx->attach = true; + tag = ++tctx->current_tag; + tctx->set_tag(event, tag); + break; + case TEVENT_EVENT_TRACE_BEFORE_HANDLER: + tctx->before_handler = true; + break; + case TEVENT_EVENT_TRACE_DETACH: + tctx->detach = true; + break; + } +} + +static int test_setup(void **state) +{ + struct test_ctx *tctx; + + tctx = talloc_zero(NULL, struct test_ctx); + assert_non_null(tctx); + + tctx->ev = tevent_context_init(tctx); + assert_non_null(tctx->ev); + + *state = tctx; + return 0; +} + +static int test_teardown(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + tctx->get_tag = NULL; + tctx->set_tag = NULL; + tevent_set_trace_fd_callback(tctx->ev, NULL, NULL); + tevent_set_trace_timer_callback(tctx->ev, NULL, NULL); + tevent_set_trace_signal_callback(tctx->ev, NULL, NULL); + tevent_set_trace_immediate_callback(tctx->ev, NULL, NULL); + TALLOC_FREE(tctx); + return 0; +} + +static uint64_t fd_get_tag(const void *_event) +{ + const struct tevent_fd *event = + (const struct tevent_fd *)_event; + + return tevent_fd_get_tag(event); +} + +static void fd_set_tag(void *_event, uint64_t tag) +{ + struct tevent_fd *event = + (struct tevent_fd *)_event; + + tevent_fd_set_tag(event, tag); +} + +static uint64_t timer_get_tag(const void *_event) +{ + const struct tevent_timer *event = + (const struct tevent_timer *)_event; + + return tevent_timer_get_tag(event); +} + +static void timer_set_tag(void *_event, uint64_t tag) +{ + struct tevent_timer *event = + (struct tevent_timer *)_event; + + tevent_timer_set_tag(event, tag); +} + +static uint64_t signal_get_tag(const void *_event) +{ + const struct tevent_signal *event = + (const struct tevent_signal *)_event; + + return tevent_signal_get_tag(event); +} + +static void signal_set_tag(void *_event, uint64_t tag) +{ + struct tevent_signal *event = + (struct tevent_signal *)_event; + + tevent_signal_set_tag(event, tag); +} + +static uint64_t immediate_get_tag(const void *_event) +{ + const struct tevent_immediate *event = + (const struct tevent_immediate *)_event; + + return tevent_immediate_get_tag(event); +} + +static void immediate_set_tag(void *_event, uint64_t tag) +{ + struct tevent_immediate *event = + (struct tevent_immediate *)_event; + + tevent_immediate_set_tag(event, tag); +} + +static uint64_t queue_entry_get_tag(const void *_event) +{ + const struct tevent_queue_entry *event = + (const struct tevent_queue_entry *)_event; + + return tevent_queue_entry_get_tag(event); +} + +static void queue_entry_set_tag(void *_event, uint64_t tag) +{ + struct tevent_queue_entry *event = + (struct tevent_queue_entry *)_event; + + tevent_queue_entry_set_tag(event, tag); +} + +static void test_trace_event_fd__loop(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_fd *fde; + + tctx->get_tag = fd_get_tag; + tctx->set_tag = fd_set_tag; + tevent_set_trace_fd_callback(tctx->ev, (tevent_trace_fd_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + fde = tevent_add_fd(tctx->ev, tctx, 0, TEVENT_FD_WRITE, fd_handler, tctx); + assert_non_null(fde); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + TEVENT_FD_WRITEABLE(fde); + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_false(tctx->detach); + + TALLOC_FREE(fde); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_fd__reset(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_fd *fde; + + tctx->get_tag = fd_get_tag; + tctx->set_tag = fd_set_tag; + tevent_set_trace_fd_callback(tctx->ev, (tevent_trace_fd_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + fde = tevent_add_fd(tctx->ev, tctx, 0, TEVENT_FD_WRITE, fd_handler, tctx); + assert_non_null(fde); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tctx->handler_skipped = true; + tevent_re_initialise(tctx->ev); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_fd__free(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_fd *fde; + + tctx->get_tag = fd_get_tag; + tctx->set_tag = fd_set_tag; + tevent_set_trace_fd_callback(tctx->ev, (tevent_trace_fd_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + fde = tevent_add_fd(tctx->ev, tctx, 0, TEVENT_FD_WRITE, fd_handler, tctx); + assert_non_null(fde); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tctx->handler_skipped = true; + TALLOC_FREE(fde); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_fd__free_in_handler(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_fd *fde; + + tctx->get_tag = fd_get_tag; + tctx->set_tag = fd_set_tag; + tevent_set_trace_fd_callback(tctx->ev, (tevent_trace_fd_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + fde = tevent_add_fd(tctx->ev, tctx, 0, TEVENT_FD_WRITE, fd_handler_free, tctx); + assert_non_null(fde); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + TEVENT_FD_WRITEABLE(fde); + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_timer__loop(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_timer *te; + struct timeval next; + + tctx->get_tag = timer_get_tag; + tctx->set_tag = timer_set_tag; + tevent_set_trace_timer_callback(tctx->ev, (tevent_trace_timer_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + next = tevent_timeval_current(); + te = tevent_add_timer(tctx->ev, tctx, next, timer_handler, tctx); + assert_non_null(te); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + /* timer events are self destructing after calling the handler */ + assert_true(tctx->detach); +} + +static void test_trace_event_timer__reset(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_timer *te; + struct timeval next; + + tctx->get_tag = timer_get_tag; + tctx->set_tag = timer_set_tag; + tevent_set_trace_timer_callback(tctx->ev, (tevent_trace_timer_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + next = tevent_timeval_current(); + te = tevent_add_timer(tctx->ev, tctx, next, timer_handler, tctx); + assert_non_null(te); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + assert_true(tctx->attach); + + assert_false(tctx->reattach_reset); + tctx->handler_skipped = true; + tctx->reattach_reset = true; + next = tevent_timeval_current(); + tevent_update_timer(te, next); + assert_false(tctx->reattach_reset); + assert_false(tctx->handler_skipped); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tctx->handler_skipped = true; + tevent_re_initialise(tctx->ev); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_timer__free(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_timer *te; + struct timeval next; + + tctx->get_tag = timer_get_tag; + tctx->set_tag = timer_set_tag; + tevent_set_trace_timer_callback(tctx->ev, (tevent_trace_timer_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + next = tevent_timeval_current(); + te = tevent_add_timer(tctx->ev, tctx, next, timer_handler, tctx); + assert_non_null(te); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + assert_true(tctx->attach); + + tctx->handler_skipped = true; + TALLOC_FREE(te); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_timer__free_in_handler(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_timer *te; + struct timeval next; + + tctx->get_tag = timer_get_tag; + tctx->set_tag = timer_set_tag; + tevent_set_trace_timer_callback(tctx->ev, (tevent_trace_timer_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + next = tevent_timeval_current(); + te = tevent_add_timer(tctx->ev, tctx, next, timer_handler_free, tctx); + assert_non_null(te); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_signal__loop(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_signal *se; + + tctx->get_tag = signal_get_tag; + tctx->set_tag = signal_set_tag; + tevent_set_trace_signal_callback(tctx->ev, (tevent_trace_signal_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + se = tevent_add_signal(tctx->ev, tctx, SIGUSR1, 0, signal_handler, tctx); + assert_non_null(se); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + kill(getpid(), SIGUSR1); + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_false(tctx->detach); + + TALLOC_FREE(se); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_signal__reset(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_signal *se; + + tctx->get_tag = signal_get_tag; + tctx->set_tag = signal_set_tag; + tevent_set_trace_signal_callback(tctx->ev, (tevent_trace_signal_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + se = tevent_add_signal(tctx->ev, tctx, SIGUSR1, 0, signal_handler, tctx); + assert_non_null(se); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tctx->handler_skipped = true; + TALLOC_FREE(se); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_signal__free(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_signal *se; + + tctx->get_tag = signal_get_tag; + tctx->set_tag = signal_set_tag; + tevent_set_trace_signal_callback(tctx->ev, (tevent_trace_signal_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + se = tevent_add_signal(tctx->ev, tctx, SIGUSR1, 0, signal_handler, tctx); + assert_non_null(se); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tctx->handler_skipped = true; + TALLOC_FREE(se); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_signal__free_in_handler(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_signal *se; + + tctx->get_tag = signal_get_tag; + tctx->set_tag = signal_set_tag; + tevent_set_trace_signal_callback(tctx->ev, (tevent_trace_signal_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + se = tevent_add_signal(tctx->ev, tctx, SIGUSR1, 0, signal_handler_free, tctx); + assert_non_null(se); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + kill(getpid(), SIGUSR1); + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_immediate__loop(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_immediate *im; + + tctx->get_tag = immediate_get_tag; + tctx->set_tag = immediate_set_tag; + tevent_set_trace_immediate_callback(tctx->ev, (tevent_trace_immediate_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + im = tevent_create_immediate(tctx); + assert_non_null(im); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_schedule_immediate(im, tctx->ev, immediate_handler, tctx); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + /* immediate events are self detaching */ + assert_true(tctx->detach); +} + +static void test_trace_event_immediate__reset(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_immediate *im; + + tctx->get_tag = immediate_get_tag; + tctx->set_tag = immediate_set_tag; + tevent_set_trace_immediate_callback(tctx->ev, (tevent_trace_immediate_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + im = tevent_create_immediate(tctx); + assert_non_null(im); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_schedule_immediate(im, tctx->ev, immediate_handler, tctx); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tctx->handler_skipped = true; + tevent_re_initialise(tctx->ev); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_immediate__free(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_immediate *im; + + tctx->get_tag = immediate_get_tag; + tctx->set_tag = immediate_set_tag; + tevent_set_trace_immediate_callback(tctx->ev, (tevent_trace_immediate_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + im = tevent_create_immediate(tctx); + assert_non_null(im); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_schedule_immediate(im, tctx->ev, immediate_handler, tctx); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tctx->handler_skipped = true; + TALLOC_FREE(im); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_immediate__free_in_handler(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_immediate *im; + + tctx->get_tag = immediate_get_tag; + tctx->set_tag = immediate_set_tag; + tevent_set_trace_immediate_callback(tctx->ev, (tevent_trace_immediate_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + im = tevent_create_immediate(tctx); + assert_non_null(im); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_schedule_immediate(im, tctx->ev, immediate_handler_free, tctx); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_trace_event_immediate__reschedule(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_immediate *im; + + tctx->get_tag = immediate_get_tag; + tctx->set_tag = immediate_set_tag; + tevent_set_trace_immediate_callback(tctx->ev, (tevent_trace_immediate_callback_t)trace_event_cb, tctx); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + im = tevent_create_immediate(tctx); + assert_non_null(im); + + assert_false(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_schedule_immediate(im, tctx->ev, immediate_handler, tctx); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + assert_false(tctx->reattach_reset); + tctx->handler_skipped = true; + tctx->reattach_reset = true; + tevent_schedule_immediate(im, tctx->ev, immediate_handler_reschedule, tctx); + assert_false(tctx->reattach_reset); + assert_false(tctx->handler_skipped); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_loop_once(tctx->ev); + assert_false(tctx->reattach_reset); + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + + tevent_loop_once(tctx->ev); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + /* immediate events are self detaching */ + assert_true(tctx->detach); +} + +struct dummy_request_state +{ + int i; + struct tevent_queue_entry *e; +}; + +static void queue_trigger(struct tevent_req *req, void *private_data) +{ + struct test_ctx *tctx = (struct test_ctx *)private_data; + struct dummy_request_state *state = tevent_req_data( + req, struct dummy_request_state); + + tctx->handler_called = true; + assert_int_equal(tevent_queue_entry_get_tag(state->e), state->i); + TALLOC_FREE(req); + + return; +} + +static void test_trace_queue__loop(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_queue *qa, *qb; + struct tevent_req *r1, *r2, *r3, *r4, *r5; + struct dummy_request_state *ds1, *ds2, *ds3, *ds4, *ds5; + + tevent_set_trace_queue_callback( + tctx->ev, + (tevent_trace_queue_callback_t)trace_event_cb1, + tctx); + tctx->get_tag = queue_entry_get_tag; + tctx->set_tag = queue_entry_set_tag; + + qa = tevent_queue_create(tctx->ev, "test_queue A"); + assert_non_null(qa); + qb = tevent_queue_create(tctx->ev, "test_queue B"); + assert_non_null(qb); + + r1 = tevent_req_create(tctx->ev, &ds1, struct dummy_request_state); + ds1->e = tevent_queue_add_entry(qa, + tctx->ev, + r1, + queue_trigger, + *state); + ds1->i = tctx->current_tag; + assert_int_equal(ds1->i, 1); + + r2 = tevent_req_create(tctx->ev, &ds2, struct dummy_request_state); + ds2->e = tevent_queue_add_entry(qa, + tctx->ev, + r2, + queue_trigger, + *state); + ds2->i = tctx->current_tag; + assert_int_equal(ds2->i, 2); + + r3 = tevent_req_create(tctx->ev, &ds3, struct dummy_request_state); + ds3->e = tevent_queue_add_entry(qb, + tctx->ev, + r3, + queue_trigger, + *state); + ds3->i = tctx->current_tag; + assert_int_equal(ds3->i, 3); + + r4 = tevent_req_create(tctx->ev, &ds4, struct dummy_request_state); + ds4->e = tevent_queue_add_entry(qb, + tctx->ev, + r4, + queue_trigger, + *state); + ds4->i = tctx->current_tag; + assert_int_equal(ds4->i, 4); + + r5 = tevent_req_create(tctx->ev, &ds5, struct dummy_request_state); + ds5->e = tevent_queue_add_entry(qa, + tctx->ev, + r5, + queue_trigger, + *state); + ds5->i = tctx->current_tag; + assert_int_equal(ds5->i, 5); + + tevent_loop_once(tctx->ev); + tevent_loop_once(tctx->ev); + tevent_loop_once(tctx->ev); + tevent_loop_once(tctx->ev); + tevent_loop_once(tctx->ev); +} + +static void reset_tctx(struct test_ctx *tctx) +{ + tctx->attach = false; + tctx->before_handler = false; + tctx->handler_called = false; + tctx->detach = false; +} + +static void test_trace_queue__extra(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + struct tevent_queue *qa; + struct tevent_req *r1, *r2, *r3; + struct dummy_request_state *ds1, *ds2, *ds3; + + tevent_set_trace_queue_callback( + tctx->ev, + (tevent_trace_queue_callback_t)trace_event_cb1, + tctx); + tctx->get_tag = queue_entry_get_tag; + tctx->set_tag = queue_entry_set_tag; + + qa = tevent_queue_create(tctx->ev, "test_queue A"); + assert_non_null(qa); + + /* + * r1 - this tests optimize_empty - request is triggered immediately, + * (and not even scheduled to tevent_context). The TALLOC_FREE() called + * from queue_trigger removes the request/queue_entry from the queue qa. + * So qa is empty + */ + r1 = tevent_req_create(tctx->ev, &ds1, struct dummy_request_state); + ds1->e = tevent_queue_add_optimize_empty(qa, + tctx->ev, + r1, + queue_trigger, + *state); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_true(tctx->detach); + assert_int_equal(tevent_queue_length(qa), 0); + + reset_tctx(tctx); + + /* + * Test a blocker request r2 - the trigger function is NULL. + */ + r2 = tevent_req_create(tctx->ev, &ds2, struct dummy_request_state); + ds2->e = tevent_queue_add_entry(qa, tctx->ev, r2, NULL, *state); + ds2->i = tctx->current_tag; + assert_true(tctx->attach); + assert_false(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + assert_int_equal(tevent_queue_length(qa), 1); + + /* + * This runs the tevent_queue_noop_trigger(). + * A blocker r2 is still on the queue head, with triggered==true + */ + tevent_loop_once(tctx->ev); + + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + /* tevent_queue_noop_trigger() is a noop. Does not set handler_called */ + assert_false(tctx->detach); + assert_int_equal(tevent_queue_length(qa), 1); + + /* + * Add a normal request r3. It will be blocked by r2. + */ + r3 = tevent_req_create(tctx->ev, &ds3, struct dummy_request_state); + ds3->e = tevent_queue_add_entry(qa, + tctx->ev, + r3, + queue_trigger, + *state); + ds3->i = tctx->current_tag; + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_false(tctx->detach); + assert_int_equal(tevent_queue_length(qa), 2); + + /* + * Remove the blocker r2. + */ + TALLOC_FREE(r2); + assert_true(tctx->attach); + assert_true(tctx->before_handler); + assert_false(tctx->handler_called); + assert_true(tctx->detach); + assert_int_equal(tevent_queue_length(qa), 1); + + reset_tctx(tctx); + + /* Process r3 */ + tevent_loop_once(tctx->ev); + + assert_false(tctx->attach); + assert_true(tctx->before_handler); + assert_true(tctx->handler_called); + assert_true(tctx->detach); +} + +static void test_get_set_trace_fd_callback(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + tevent_trace_fd_callback_t cb; + void *pvt; + + tevent_get_trace_fd_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); + + tevent_set_trace_fd_callback(tctx->ev, (tevent_trace_fd_callback_t)trace_event_cb, tctx); + tevent_get_trace_fd_callback(tctx->ev, &cb, &pvt); + assert_ptr_equal(cb, trace_event_cb); + assert_ptr_equal(pvt, tctx); + + tevent_set_trace_fd_callback(tctx->ev, NULL, NULL); + tevent_get_trace_fd_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); +} + +static void test_get_set_trace_timer_callback(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + tevent_trace_timer_callback_t cb; + void *pvt; + + tevent_get_trace_timer_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); + + tevent_set_trace_timer_callback(tctx->ev, (tevent_trace_timer_callback_t)trace_event_cb, tctx); + tevent_get_trace_timer_callback(tctx->ev, &cb, &pvt); + assert_ptr_equal(cb, trace_event_cb); + assert_ptr_equal(pvt, tctx); + + tevent_set_trace_timer_callback(tctx->ev, NULL, NULL); + tevent_get_trace_timer_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); +} + +static void test_get_set_trace_signal_callback(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + tevent_trace_signal_callback_t cb; + void *pvt; + + tevent_get_trace_signal_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); + + tevent_set_trace_signal_callback(tctx->ev, (tevent_trace_signal_callback_t)trace_event_cb, tctx); + tevent_get_trace_signal_callback(tctx->ev, &cb, &pvt); + assert_ptr_equal(cb, trace_event_cb); + assert_ptr_equal(pvt, tctx); + + tevent_set_trace_signal_callback(tctx->ev, NULL, NULL); + tevent_get_trace_signal_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); +} + +static void test_get_set_trace_immediate_callback(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + tevent_trace_immediate_callback_t cb; + void *pvt; + + tevent_get_trace_immediate_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); + + tevent_set_trace_immediate_callback(tctx->ev, (tevent_trace_immediate_callback_t)trace_event_cb, tctx); + tevent_get_trace_immediate_callback(tctx->ev, &cb, &pvt); + assert_ptr_equal(cb, trace_event_cb); + assert_ptr_equal(pvt, tctx); + + tevent_set_trace_immediate_callback(tctx->ev, NULL, NULL); + tevent_get_trace_immediate_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); +} + +static void test_get_set_trace_queue_callback(void **state) +{ + struct test_ctx *tctx = (struct test_ctx *)(*state); + tevent_trace_queue_callback_t cb; + void *pvt; + + tevent_get_trace_queue_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); + + tevent_set_trace_queue_callback(tctx->ev, (tevent_trace_queue_callback_t)trace_event_cb, tctx); + tevent_get_trace_queue_callback(tctx->ev, &cb, &pvt); + assert_ptr_equal(cb, trace_event_cb); + assert_ptr_equal(pvt, tctx); + + tevent_set_trace_queue_callback(tctx->ev, NULL, NULL); + tevent_get_trace_queue_callback(tctx->ev, &cb, &pvt); + assert_null(cb); + assert_null(pvt); +} + +int main(int argc, char **argv) +{ + const struct CMUnitTest tests[] = { + cmocka_unit_test_setup_teardown(test_trace_event_fd__loop, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_fd__reset, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_fd__free, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_fd__free_in_handler, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_timer__loop, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_timer__reset, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_timer__free, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_timer__free_in_handler, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_signal__loop, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_signal__reset, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_signal__free, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_signal__free_in_handler, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_immediate__loop, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_immediate__reset, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_immediate__free, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_immediate__free_in_handler, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_event_immediate__reschedule, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_queue__loop, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_trace_queue__extra, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_get_set_trace_fd_callback, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_get_set_trace_timer_callback, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_get_set_trace_signal_callback, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_get_set_trace_immediate_callback, test_setup, test_teardown), + cmocka_unit_test_setup_teardown(test_get_set_trace_queue_callback, test_setup, test_teardown), + }; + + cmocka_set_message_output(CM_OUTPUT_SUBUNIT); + + return cmocka_run_group_tests(tests, NULL, NULL); +} |