mirror of
https://gitlab.isc.org/isc-projects/bind9
synced 2025-08-29 05:28:00 +00:00
af257140e6
In rare circumstances, the UDP port for the listening socket and the UDP port for the connecting socket might be the same. Because we use the "reuse" port socket option, this isn't caught when binding the socket, and thus the connected client socket could send a datagram to itself, completely bypassing the server. This doesn't happen under normal operation mode because `named` is listening on a privileged port (53), and even if not, it doesn't usually talk to itself as the tests do. Pick an arbitrary port for listening (9153-9156) that is outside the ephemeral port range for the network manager related unit tests (except the `doh_test).
143 lines
3.3 KiB
C
143 lines
3.3 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/tcp.c"
|
|
#include "netmgr_common.h"
|
|
|
|
#include <tests/isc.h>
|
|
|
|
/* TCP */
|
|
ISC_LOOP_TEST_IMPL(tcp_noop) {
|
|
stream_noop(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_noresponse) {
|
|
stream_noresponse(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_timeout_recovery) {
|
|
stream_timeout_recovery(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_recv_one) {
|
|
stream_recv_one(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_recv_two) {
|
|
stream_recv_two(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_recv_send) {
|
|
stream_recv_send(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_recv_send_sendback) {
|
|
allow_send_back = true;
|
|
stream_recv_send(arg);
|
|
return;
|
|
}
|
|
|
|
/* TCP Quota */
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_recv_one_quota) {
|
|
atomic_store(&check_listener_quota, true);
|
|
stream_recv_one(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_recv_two_quota) {
|
|
atomic_store(&check_listener_quota, true);
|
|
stream_recv_two(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_recv_send_quota) {
|
|
atomic_store(&check_listener_quota, true);
|
|
stream_recv_send(arg);
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_recv_send_quota_sendback) {
|
|
atomic_store(&check_listener_quota, true);
|
|
allow_send_back = true;
|
|
stream_recv_send(arg);
|
|
}
|
|
|
|
ISC_TEST_LIST_START
|
|
|
|
/* TCP */
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_noop, stream_noop_setup, stream_noop_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_noresponse, stream_noresponse_setup,
|
|
stream_noresponse_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_timeout_recovery, stream_timeout_recovery_setup,
|
|
stream_timeout_recovery_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_recv_one, stream_recv_one_setup,
|
|
stream_recv_one_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_recv_two, stream_recv_two_setup,
|
|
stream_recv_two_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_recv_send, stream_recv_send_setup,
|
|
stream_recv_send_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_recv_send_sendback, stream_recv_send_setup,
|
|
stream_recv_send_teardown)
|
|
|
|
/* TCP Quota */
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_recv_one_quota, stream_recv_one_setup,
|
|
stream_recv_one_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_recv_two_quota, stream_recv_two_setup,
|
|
stream_recv_two_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_recv_send_quota, stream_recv_send_setup,
|
|
stream_recv_send_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_recv_send_quota_sendback, stream_recv_send_setup,
|
|
stream_recv_send_teardown)
|
|
|
|
ISC_TEST_LIST_END
|
|
|
|
static int
|
|
tcp_setup(void **state __attribute__((__unused__))) {
|
|
stream_port = TCP_TEST_PORT;
|
|
stream_use_TLS = false;
|
|
stream = true;
|
|
|
|
return (0);
|
|
}
|
|
|
|
ISC_TEST_MAIN_CUSTOM(tcp_setup, NULL)
|