From c8d4f3b5658c5d872c80d936dd2e9d9f278052d0 Mon Sep 17 00:00:00 2001
From: Artem Boldariev <artem@boldariev.com>
Date: Wed, 21 Jun 2023 16:33:13 +0300
Subject: [PATCH] Extend PROXY Stream tests to include checks for PROXY over
 TLS

This commit extends the PROXY stream unit test suite with checks
related to PROXY over TLS support within the transport.
---
 tests/isc/proxystream_test.c | 115 +++++++++++++++++++++++++++++++++++
 1 file changed, 115 insertions(+)

diff --git a/tests/isc/proxystream_test.c b/tests/isc/proxystream_test.c
index 72e72e8d59..0d9a1bab21 100644
--- a/tests/isc/proxystream_test.c
+++ b/tests/isc/proxystream_test.c
@@ -118,6 +118,80 @@ ISC_LOOP_TEST_IMPL(proxystream_recv_send_quota_sendback) {
 	stream_recv_send(arg);
 }
 
+/* PROXY over TLS (as used by, e.g., dnsdist) */
+
+/* TCP */
+ISC_LOOP_TEST_IMPL(proxystreamtls_noop) {
+	stream_noop(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_noresponse) {
+	stream_noresponse(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_shutdownconnect) {
+	stream_shutdownconnect(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_shutdownread) {
+	stream_shutdownread(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_timeout_recovery) {
+	stream_timeout_recovery(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_recv_one) {
+	stream_recv_one(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_recv_two) {
+	stream_recv_two(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_recv_send) {
+	stream_recv_send(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_recv_send_sendback) {
+	allow_send_back = true;
+	stream_recv_send(arg);
+	return;
+}
+
+/* TCP Quota */
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_recv_one_quota) {
+	atomic_store(&check_listener_quota, true);
+	stream_recv_one(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_recv_two_quota) {
+	atomic_store(&check_listener_quota, true);
+	stream_recv_two(arg);
+	return;
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_recv_send_quota) {
+	atomic_store(&check_listener_quota, true);
+	stream_recv_send(arg);
+}
+
+ISC_LOOP_TEST_IMPL(proxystreamtls_recv_send_quota_sendback) {
+	atomic_store(&check_listener_quota, true);
+	allow_send_back = true;
+	stream_recv_send(arg);
+}
+
 ISC_TEST_LIST_START
 
 /* Stream */
@@ -154,6 +228,47 @@ ISC_TEST_ENTRY_CUSTOM(proxystream_recv_send_quota_sendback,
 		      proxystream_recv_send_setup,
 		      proxystream_recv_send_teardown)
 
+/* PROXY over TLS */
+
+/* Stream */
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_noop, proxystreamtls_noop_setup,
+		      proxystreamtls_noop_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_noresponse,
+		      proxystreamtls_noresponse_setup,
+		      proxystreamtls_noresponse_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_shutdownconnect,
+		      proxystreamtls_shutdownconnect_setup,
+		      proxystreamtls_shutdownconnect_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_shutdownread,
+		      proxystreamtls_shutdownread_setup,
+		      proxystreamtls_shutdownread_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_timeout_recovery,
+		      proxystreamtls_timeout_recovery_setup,
+		      proxystreamtls_timeout_recovery_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_recv_one, proxystreamtls_recv_one_setup,
+		      proxystreamtls_recv_one_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_recv_two, proxystreamtls_recv_two_setup,
+		      proxystreamtls_recv_two_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_recv_send, proxystreamtls_recv_send_setup,
+		      proxystreamtls_recv_send_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_recv_send_sendback,
+		      proxystreamtls_recv_send_setup,
+		      proxystreamtls_recv_send_teardown)
+
+/* Stream Quota */
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_recv_one_quota,
+		      proxystreamtls_recv_one_setup,
+		      proxystreamtls_recv_one_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_recv_two_quota,
+		      proxystreamtls_recv_two_setup,
+		      proxystreamtls_recv_two_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_recv_send_quota,
+		      proxystreamtls_recv_send_setup,
+		      proxystreamtls_recv_send_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxystreamtls_recv_send_quota_sendback,
+		      proxystreamtls_recv_send_setup,
+		      proxystreamtls_recv_send_teardown)
+
 ISC_TEST_LIST_END
 
 static int