mirror of
https://gitlab.isc.org/isc-projects/bind9
synced 2025-08-30 14:07:59 +00:00
fffd444440
After the loopmgr work has been merged, we can now cleanup the TCP and TLS protocols a little bit, because there are stronger guarantees that the sockets will be kept on the respective loops/threads. We only need asynchronous call for listening sockets (start, stop) and reading from the TCP (because the isc_nm_read() might be called from read callback again. This commit does the following changes (they are intertwined together): 1. Cleanup most of the asynchronous events in the TCP code, and add comments for the events that needs to be kept asynchronous. 2. Remove isc_nm_resumeread() from the netmgr API, and replace isc_nm_resumeread() calls with existing isc_nm_read() calls. 3. Remove isc_nm_pauseread() from the netmgr API, and replace isc_nm_pauseread() calls with a new isc_nm_read_stop() call. 4. Disable the isc_nm_cancelread() for the streaming protocols, only the datagram-like protocols can use isc_nm_cancelread(). 5. Add isc_nmhandle_close() that can be used to shutdown the socket earlier than after the last detach. Formerly, the socket would be closed only after all reading and sending would be finished and the last reference would be detached. The new isc_nmhandle_close() can be used to close the underlying socket earlier, so all the other asynchronous calls would call their respective callbacks immediately. Co-authored-by: Ondřej Surý <ondrej@isc.org> Co-authored-by: Artem Boldariev <artem@isc.org>
141 lines
3.2 KiB
C
141 lines
3.2 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 <sched.h> /* IWYU pragma: keep */
|
|
#include <setjmp.h>
|
|
#include <signal.h>
|
|
#include <stdarg.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
#define UNIT_TESTING
|
|
#include <cmocka.h>
|
|
|
|
#include <isc/loop.h>
|
|
#include <isc/nonce.h>
|
|
#include <isc/os.h>
|
|
#include <isc/quota.h>
|
|
#include <isc/refcount.h>
|
|
#include <isc/sockaddr.h>
|
|
#include <isc/thread.h>
|
|
#include <isc/util.h>
|
|
#include <isc/uv.h>
|
|
|
|
#include "uv_wrap.h"
|
|
#define KEEP_BEFORE
|
|
|
|
#include "netmgr_common.h"
|
|
|
|
#include <tests/isc.h>
|
|
|
|
#if HAVE_LIBNGHTTP2
|
|
ISC_LOOP_TEST_IMPL(tls_noop) {
|
|
stream_noop(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_noresponse) {
|
|
stream_noresponse(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_timeout_recovery) {
|
|
stream_timeout_recovery(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_recv_one) {
|
|
stream_recv_one(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_recv_two) {
|
|
stream_recv_two(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_recv_send) {
|
|
stream_recv_send(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_recv_send_sendback) {
|
|
allow_send_back = true;
|
|
stream_recv_send(arg);
|
|
}
|
|
|
|
/* TLS quota */
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_recv_one_quota) {
|
|
atomic_store(&check_listener_quota, true);
|
|
stream_recv_one(arg);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_recv_two_quota) {
|
|
atomic_store(&check_listener_quota, true);
|
|
stream_recv_two(arg);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_recv_send_quota) {
|
|
atomic_store(&check_listener_quota, true);
|
|
stream_recv_send(arg);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tls_recv_send_quota_sendback) {
|
|
allow_send_back = true;
|
|
atomic_store(&check_listener_quota, true);
|
|
stream_recv_send(arg);
|
|
}
|
|
#endif
|
|
|
|
ISC_TEST_LIST_START
|
|
|
|
#if HAVE_LIBNGHTTP2
|
|
/* TLS */
|
|
ISC_TEST_ENTRY_CUSTOM(tls_noop, stream_noop_setup, stream_noop_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_noresponse, stream_noresponse_setup,
|
|
stream_noresponse_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_timeout_recovery, stream_timeout_recovery_setup,
|
|
stream_timeout_recovery_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_recv_one, stream_recv_one_setup,
|
|
stream_recv_one_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_recv_two, stream_recv_two_setup,
|
|
stream_recv_two_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_recv_send, stream_recv_send_setup,
|
|
stream_recv_send_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_recv_send_sendback, stream_recv_send_setup,
|
|
stream_recv_send_teardown)
|
|
|
|
/* TLS quota */
|
|
ISC_TEST_ENTRY_CUSTOM(tls_recv_one_quota, stream_recv_one_setup,
|
|
stream_recv_one_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_recv_two_quota, stream_recv_two_setup,
|
|
stream_recv_two_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_recv_send_quota, stream_recv_send_setup,
|
|
stream_recv_send_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tls_recv_send_quota_sendback, stream_recv_send_setup,
|
|
stream_recv_send_teardown)
|
|
#endif
|
|
|
|
ISC_TEST_LIST_END
|
|
|
|
static int
|
|
tls_setup(void **state __attribute__((__unused__))) {
|
|
stream_use_TLS = true;
|
|
stream = true;
|
|
|
|
return (0);
|
|
}
|
|
|
|
ISC_TEST_MAIN_CUSTOM(tls_setup, NULL)
|