mir/bind
2
0
Fork 0
mirror of https://gitlab.isc.org/isc-projects/bind9 synced 2025-08-22 10:10:06 +00:00
Code Issues Releases Wiki Activity
bind/tests/isc/udp_test.c

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

238 lines
6.6 KiB
C
Raw Normal View History

Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
/*
* 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> whenever including <cmocka.h> Development versions of cmocka require the intmax_t and uintmax_t types to be defined by the time the test code includes the <cmocka.h> header. These types are defined in the <stdint.h> header, which is included by the <inttypes.h> header, which in turn is already explicitly included by some of the programs in the tests/ directory. Ensure all programs in that directory that include the <cmocka.h> header also include the <inttypes.h> header to future-proof the code while keeping the change set minimal and the resulting code consistent. Also prevent explicitly including the <stdint.h> header in those programs as it is included by the <inttypes.h> header.
2023-05-18 15:12:23 +02:00
#include <inttypes.h>
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
#include <sched.h> /* IWYU pragma: keep */
#include <setjmp.h>
#include <signal.h>
#include <stdarg.h>
#include <stdlib.h>
#include <unistd.h>
unit tests: include an OpenSSL header before including cmocka.h OpenSSL 3.1.0 uses __attribute__(malloc), conflicting with a redefined malloc in cmocka.h. As a workaround, include an OpenSSL header file before including cmocka.h in the unit tests where OpenSSL is used.
2023-04-06 13:18:04 +00:00
/*
* As a workaround, include an OpenSSL header file before including cmocka.h,
* because OpenSSL 3.1.0 uses __attribute__(malloc), conflicting with a
* redefined malloc in cmocka.h.
*/
#include <openssl/err.h>
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
#define UNIT_TESTING
#include <cmocka.h>
#include <isc/async.h>
#include <isc/job.h>
Move the library init and shutdown to executables Instead of relying on unreliable order of execution of the library constructors and destructors, move them to individual binaries. The advantage is that the execution time and order will remain constant and will not depend on the dynamic load dependency solver. This requires more work, but that was mitigated by a simple requirement, any executable using libisc and libdns, must include <isc/lib.h> and <dns/lib.h> respectively (in this particular order). In turn, these two headers must not be included from within any library as they contain inlined functions marked with constructor/destructor attributes.
2025-02-04 13:17:31 +01:00
#include <isc/lib.h>
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
#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 "uv_wrap.h"
#define KEEP_BEFORE
#include "netmgr/socket.c"
#include "netmgr/udp.c"
#include "netmgr_common.h"
#include "uv.c"
#include <tests/isc.h>
/* Callbacks */
static void
mock_recv_cb(isc_nmhandle_t *handle, isc_result_t eresult, isc_region_t *region,
void *cbarg) {
UNUSED(handle);
UNUSED(eresult);
UNUSED(region);
UNUSED(cbarg);
}
Fix the streaming read callback shutdown logic When shutting down TCP sockets, the read callback calling logic was flawed, it would call either one less callback or one extra. Fix the logic in the way: 1. When isc_nm_read() has been called but isc_nm_read_stop() hasn't on the handle, the read callback will be called with ISC_R_CANCELED to cancel active reading from the socket/handle. 2. When isc_nm_read() has been called and isc_nm_read_stop() has been called on the on the handle, the read callback will be called with ISC_R_SHUTTINGDOWN to signal that the dormant (not-reading) socket is being shut down. 3. The .reading and .recv_read flags are little bit tricky. The .reading flag indicates if the outer layer is reading the data (that would be uv_tcp_t for TCP and isc_nmsocket_t (TCP) for TLSStream), the .recv_read flag indicates whether somebody is interested in the data read from the socket. Usually, you would expect that the .reading should be false when .recv_read is false, but it gets even more tricky with TLSStream as the TLS protocol might need to read from the socket even when sending data. Fix the usage of the .recv_read and .reading flags in the TLSStream to their true meaning - which mostly consist of using .recv_read everywhere and then wrapping isc_nm_read() and isc_nm_read_stop() with the .reading flag. 4. The TLS failed read helper has been modified to resemble the TCP code as much as possible, clearing and re-setting the .recv_read flag in the TCP timeout code has been fixed and .recv_read is now cleared when isc_nm_read_stop() has been called on the streaming socket. 5. The use of Network Manager in the named_controlconf, isccc_ccmsg, and isc_httpd units have been greatly simplified due to the improved design. 6. More unit tests for TCP and TLS testing the shutdown conditions have been added. Co-authored-by: Ondřej Surý <ondrej@isc.org> Co-authored-by: Artem Boldariev <artem@isc.org>
2023-04-13 17:27:50 +02:00
static void
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
udp_connect_nomemory_cb(isc_nmhandle_t *handle, isc_result_t eresult,
void *cbarg) {
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
UNUSED(handle);
UNUSED(cbarg);
isc_refcount_decrement(&active_cconnects);
assert_int_equal(eresult, ISC_R_NOMEMORY);
isc_loopmgr_shutdown(loopmgr);
}
/* UDP */
ISC_LOOP_TEST_IMPL(mock_listenudp_uv_udp_open) {
isc_result_t result = ISC_R_SUCCESS;
WILL_RETURN(uv_udp_open, UV_ENOMEM);
result = isc_nm_listenudp(netmgr, ISC_NM_LISTEN_ALL, &udp_listen_addr,
mock_recv_cb, NULL, &listen_sock);
assert_int_not_equal(result, ISC_R_SUCCESS);
assert_null(listen_sock);
RESET_RETURN;
isc_loopmgr_shutdown(loopmgr);
}
ISC_LOOP_TEST_IMPL(mock_listenudp_uv_udp_bind) {
isc_result_t result = ISC_R_SUCCESS;
WILL_RETURN(uv_udp_bind, UV_EADDRINUSE);
result = isc_nm_listenudp(netmgr, ISC_NM_LISTEN_ALL, &udp_listen_addr,
mock_recv_cb, NULL, &listen_sock);
assert_int_not_equal(result, ISC_R_SUCCESS);
assert_null(listen_sock);
RESET_RETURN;
isc_loopmgr_shutdown(loopmgr);
}
ISC_LOOP_TEST_IMPL(mock_listenudp_uv_udp_recv_start) {
isc_result_t result = ISC_R_SUCCESS;
WILL_RETURN(uv_udp_recv_start, UV_EADDRINUSE);
result = isc_nm_listenudp(netmgr, ISC_NM_LISTEN_ALL, &udp_listen_addr,
mock_recv_cb, NULL, &listen_sock);
assert_int_not_equal(result, ISC_R_SUCCESS);
assert_null(listen_sock);
RESET_RETURN;
isc_loopmgr_shutdown(loopmgr);
}
ISC_LOOP_TEST_IMPL(mock_udpconnect_uv_udp_open) {
WILL_RETURN(uv_udp_open, UV_ENOMEM);
isc_refcount_increment0(&active_cconnects);
isc_nm_udpconnect(netmgr, &udp_connect_addr, &udp_listen_addr,
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
udp_connect_nomemory_cb, NULL, UDP_T_CONNECT);
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
isc_loopmgr_shutdown(loopmgr);
RESET_RETURN;
}
ISC_LOOP_TEST_IMPL(mock_udpconnect_uv_udp_bind) {
WILL_RETURN(uv_udp_bind, UV_ENOMEM);
isc_refcount_increment0(&active_cconnects);
isc_nm_udpconnect(netmgr, &udp_connect_addr, &udp_listen_addr,
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
udp_connect_nomemory_cb, NULL, UDP_T_CONNECT);
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
isc_loopmgr_shutdown(loopmgr);
RESET_RETURN;
}
ISC_LOOP_TEST_IMPL(mock_udpconnect_uv_udp_connect) {
WILL_RETURN(uv_udp_connect, UV_ENOMEM);
isc_refcount_increment0(&active_cconnects);
isc_nm_udpconnect(netmgr, &udp_connect_addr, &udp_listen_addr,
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
udp_connect_nomemory_cb, NULL, UDP_T_CONNECT);
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
isc_loopmgr_shutdown(loopmgr);
RESET_RETURN;
}
ISC_LOOP_TEST_IMPL(mock_udpconnect_uv_recv_buffer_size) {
WILL_RETURN(uv_recv_buffer_size, UV_ENOMEM);
isc_refcount_increment0(&active_cconnects);
isc_nm_udpconnect(netmgr, &udp_connect_addr, &udp_listen_addr,
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
connect_success_cb, NULL, UDP_T_CONNECT);
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
isc_loopmgr_shutdown(loopmgr);
RESET_RETURN;
}
ISC_LOOP_TEST_IMPL(mock_udpconnect_uv_send_buffer_size) {
WILL_RETURN(uv_send_buffer_size, UV_ENOMEM);
isc_refcount_increment0(&active_cconnects);
isc_nm_udpconnect(netmgr, &udp_connect_addr, &udp_listen_addr,
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
connect_success_cb, NULL, UDP_T_CONNECT);
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
isc_loopmgr_shutdown(loopmgr);
RESET_RETURN;
}
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_noop) { udp_noop(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_noresponse) { udp_noresponse(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_timeout_recovery) { udp_timeout_recovery(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_shutdown_connect) { udp_shutdown_connect(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_shutdown_read) { udp_shutdown_read(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_cancel_read) { udp_cancel_read(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_recv_one) { udp_recv_one(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_recv_two) { udp_recv_two(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_recv_send) { udp_recv_send(arg); }
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_LOOP_TEST_IMPL(udp_double_read) { udp_double_read(arg); }
Prevent unexpected UDP client read callbacks The network manager UDP code was misinterpreting when the libuv called the udp_recv_cb with nrecv == 0 and addr == NULL -> this doesn't really mean that the "stream" has ended, but the libuv indicates that the receive buffer can be freed. This could lead to assertion failure in the code that calls isc_nm_read() from the network manager read callback due to the extra spurious callbacks. Properly handle the extra callback calls from the libuv in the client read callback, and refactor the UDP isc_nm_read() implementation to be synchronous, so no datagram is lost between the time that we stop the reading from the UDP socket and we restart it again in the asychronous udpread event. Add a unit test that tests the isc_nm_read() call from the read callback to receive two datagrams.
2022-09-15 09:48:34 +02:00
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
ISC_TEST_LIST_START
Refactor UDP tests This commit mostly moves the code around to make the parts of the UDP unit test suite reusable. That changes the unit test suite structure to resemble that of stream based unit tests. The motivation behind this is to reuse most of the code for the new PROXY over UDP uni tests suite.
2023-10-11 22:18:54 +03:00
ISC_TEST_ENTRY_CUSTOM(mock_listenudp_uv_udp_open, setup_udp_test,
teardown_udp_test)
ISC_TEST_ENTRY_CUSTOM(mock_listenudp_uv_udp_bind, setup_udp_test,
teardown_udp_test)
ISC_TEST_ENTRY_CUSTOM(mock_listenudp_uv_udp_recv_start, setup_udp_test,
teardown_udp_test)
ISC_TEST_ENTRY_CUSTOM(mock_udpconnect_uv_udp_open, setup_udp_test,
teardown_udp_test)
ISC_TEST_ENTRY_CUSTOM(mock_udpconnect_uv_udp_bind, setup_udp_test,
teardown_udp_test)
ISC_TEST_ENTRY_CUSTOM(mock_udpconnect_uv_udp_connect, setup_udp_test,
teardown_udp_test)
ISC_TEST_ENTRY_CUSTOM(mock_udpconnect_uv_recv_buffer_size, setup_udp_test,
teardown_udp_test)
ISC_TEST_ENTRY_CUSTOM(mock_udpconnect_uv_send_buffer_size, setup_udp_test,
teardown_udp_test)
ISC_TEST_ENTRY_CUSTOM(udp_noop, udp_noop_setup, udp_noop_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_noresponse, udp_noresponse_setup,
udp_noresponse_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_timeout_recovery, udp_timeout_recovery_setup,
udp_timeout_recovery_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_shutdown_read, udp_shutdown_read_setup,
udp_shutdown_read_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_cancel_read, udp_cancel_read_setup,
udp_cancel_read_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_shutdown_connect, udp_shutdown_connect_setup,
udp_shutdown_connect_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_double_read, udp_double_read_setup,
udp_double_read_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_recv_one, udp_recv_one_setup, udp_recv_one_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_recv_two, udp_recv_two_setup, udp_recv_two_teardown)
ISC_TEST_ENTRY_CUSTOM(udp_recv_send, udp_recv_send_setup,
udp_recv_send_teardown)
Split netmgr_test into separate per-transport unit tests The netmgr_test unit test has been subdivided into tcp_test, tcpdns_test, tls_test, tlsdns_test, and udp_test components. These have been updated to use the new loopmgr.
2022-07-26 13:03:48 +02:00
ISC_TEST_LIST_END
ISC_TEST_MAIN
Reference in New Issue Copy Permalink