164 lines
3.9 KiB
C
164 lines
3.9 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 <signal.h>
|
|
#include <stdarg.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
/*
|
|
* 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>
|
|
|
|
#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_shutdownconnect) {
|
|
stream_shutdownconnect(arg);
|
|
return;
|
|
}
|
|
|
|
ISC_LOOP_TEST_IMPL(tcp_shutdownread) {
|
|
stream_shutdownread(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_shutdownconnect, stream_shutdownconnect_setup,
|
|
stream_shutdownconnect_teardown)
|
|
ISC_TEST_ENTRY_CUSTOM(tcp_shutdownread, stream_shutdownread_setup,
|
|
stream_shutdownread_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 ISC_ATTR_UNUSED) {
|
|
stream_port = TCP_TEST_PORT;
|
|
stream_use_TLS = false;
|
|
stream = true;
|
|
|
|
return 0;
|
|
}
|
|
|
|
ISC_TEST_MAIN_CUSTOM(tcp_setup, NULL)
|