mirror of
https://gitlab.isc.org/isc-projects/bind9
synced 2025-08-22 10:10:06 +00:00
1032681af0
Change the internal type used for isc_tid unit to isc_tid_t to hide the specific integer type being used for the 'tid'. Internally, the signed integer type is being used. This allows us to have negatively indexed arrays that works both for threads with assigned tid and the threads with unassigned tid. This should be used only in specific situations.
111 lines
2.4 KiB
C
111 lines
2.4 KiB
C
/*
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* SPDX-License-Identifier: MPL-2.0
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
*/
|
|
|
|
#include <inttypes.h>
|
|
#include <sched.h> /* IWYU pragma: keep */
|
|
#include <setjmp.h>
|
|
#include <stdarg.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#define UNIT_TESTING
|
|
#include <cmocka.h>
|
|
|
|
#include <isc/async.h>
|
|
#include <isc/atomic.h>
|
|
#include <isc/lib.h>
|
|
#include <isc/loop.h>
|
|
#include <isc/os.h>
|
|
#include <isc/result.h>
|
|
#include <isc/tid.h>
|
|
#include <isc/util.h>
|
|
|
|
#include "async.c"
|
|
|
|
#include <tests/isc.h>
|
|
|
|
static atomic_uint scheduled = 0;
|
|
|
|
static void
|
|
async_cb(void *arg) {
|
|
isc_tid_t tid = isc_tid();
|
|
|
|
UNUSED(arg);
|
|
|
|
atomic_fetch_add(&scheduled, 1);
|
|
|
|
if (tid > 0) {
|
|
isc_loop_t *loop = isc_loop_get(loopmgr, tid - 1);
|
|
isc_async_run(loop, async_cb, loopmgr);
|
|
} else {
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
}
|
|
|
|
static void
|
|
async_setup_cb(void *arg) {
|
|
isc_tid_t tid = isc_loopmgr_nloops(loopmgr) - 1;
|
|
isc_loop_t *loop = isc_loop_get(loopmgr, tid);
|
|
|
|
UNUSED(arg);
|
|
|
|
isc_async_run(loop, async_cb, loopmgr);
|
|
}
|
|
|
|
ISC_RUN_TEST_IMPL(isc_async_run) {
|
|
isc_loop_setup(isc_loop_main(loopmgr), async_setup_cb, loopmgr);
|
|
isc_loopmgr_run(loopmgr);
|
|
assert_int_equal(atomic_load(&scheduled), loopmgr->nloops);
|
|
}
|
|
|
|
static char string[32] = "";
|
|
int n1 = 1, n2 = 2, n3 = 3, n4 = 4, n5 = 5;
|
|
|
|
static void
|
|
append(void *arg) {
|
|
char value[32];
|
|
sprintf(value, "%d", *(int *)arg);
|
|
strlcat(string, value, 10);
|
|
}
|
|
|
|
static void
|
|
async_multiple(void *arg) {
|
|
isc_loop_t *loop = isc_loop();
|
|
|
|
UNUSED(arg);
|
|
|
|
isc_async_run(loop, append, &n1);
|
|
isc_async_run(loop, append, &n2);
|
|
isc_async_run(loop, append, &n3);
|
|
isc_async_run(loop, append, &n4);
|
|
isc_async_run(loop, append, &n5);
|
|
isc_loopmgr_shutdown(loopmgr);
|
|
}
|
|
|
|
ISC_RUN_TEST_IMPL(isc_async_multiple) {
|
|
string[0] = '\0';
|
|
isc_loop_setup(isc_loop_main(loopmgr), async_multiple, loopmgr);
|
|
isc_loopmgr_run(loopmgr);
|
|
assert_string_equal(string, "12345");
|
|
}
|
|
|
|
ISC_TEST_LIST_START
|
|
ISC_TEST_ENTRY_CUSTOM(isc_async_run, setup_loopmgr, teardown_loopmgr)
|
|
ISC_TEST_ENTRY_CUSTOM(isc_async_multiple, setup_loopmgr, teardown_loopmgr)
|
|
ISC_TEST_LIST_END
|
|
|
|
ISC_TEST_MAIN
|