From ab9206c2327b31aa65a2b1e65d3e3e7722350eed Mon Sep 17 00:00:00 2001
From: Maximilian Fridrich <m.fridrich@commend.com>
Date: Thu, 12 Dec 2024 16:17:44 +0100
Subject: [PATCH] test/tcp,udp: Add TOS/TCLASS tests

---
 test/tcp.c  | 60 +++++++++++++++++++++++++++++++++++++++++++
 test/test.c |  2 ++
 test/test.h |  2 ++
 test/udp.c  | 74 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 138 insertions(+)

diff --git a/test/tcp.c b/test/tcp.c
index 2619da449..35f7640f4 100644
--- a/test/tcp.c
+++ b/test/tcp.c
@@ -227,3 +227,63 @@ int test_tcp(void)
 
 	return err;
 }
+
+
+static int tcp_tos(const char *addr)
+{
+	struct tcp_test *tt;
+	struct sa srv;
+	int err;
+
+	tt = mem_zalloc(sizeof(*tt), destructor);
+	if (!tt)
+		return ENOMEM;
+
+	err = sa_set_str(&srv, addr, 0);
+	if (err)
+		goto out;
+
+	err = tcp_listen(&tt->ts, &srv, tcp_server_conn_handler, tt);
+	if (err)
+		goto out;
+
+	tcp_settos(tt->ts, 184);
+	tcp_settos(tt->ts, 120);
+
+	err = tcp_local_get(tt->ts, &srv);
+	if (err)
+		goto out;
+
+	err = tcp_connect(&tt->tc, &srv, tcp_client_estab_handler,
+			  tcp_client_recv_handler, tcp_client_close_handler,
+			  tt);
+	if (err)
+		goto out;
+
+	err = re_main_timeout(500);
+	if (err)
+		goto out;
+
+	if (tt->err)
+		err = tt->err;
+
+ out:
+	mem_deref(tt);
+
+	return err;
+}
+
+
+int test_tcp_tos(void)
+{
+	int err;
+
+	err = tcp_tos("127.0.0.1");
+	TEST_ERR(err);
+
+	err = tcp_tos("::1");
+	TEST_ERR(err);
+
+ out:
+	return err;
+}
diff --git a/test/test.c b/test/test.c
index 1ec985d46..9494e77cf 100644
--- a/test/test.c
+++ b/test/test.c
@@ -216,6 +216,7 @@ static const struct test tests[] = {
 	TEST(test_sys_fs_fopen),
 	TEST(test_sys_getenv),
 	TEST(test_tcp),
+	TEST(test_tcp_tos),
 	TEST(test_telev),
 	TEST(test_text2pcap),
 #ifdef USE_TLS
@@ -236,6 +237,7 @@ static const struct test tests[] = {
 	TEST(test_turn),
 	TEST(test_turn_tcp),
 	TEST(test_udp),
+	TEST(test_udp_tos),
 	TEST(test_unixsock),
 	TEST(test_uri),
 	TEST(test_uri_encode),
diff --git a/test/test.h b/test/test.h
index 303b84166..6af831927 100644
--- a/test/test.h
+++ b/test/test.h
@@ -332,6 +332,7 @@ int test_sys_fs_isfile(void);
 int test_sys_fs_fopen(void);
 int test_sys_getenv(void);
 int test_tcp(void);
+int test_tcp_tos(void);
 int test_telev(void);
 int test_text2pcap(void);
 int test_thread(void);
@@ -343,6 +344,7 @@ int test_turn(void);
 int test_turn_tcp(void);
 int test_turn_thread(void);
 int test_udp(void);
+int test_udp_tos(void);
 int test_unixsock(void);
 int test_uri(void);
 int test_uri_encode(void);
diff --git a/test/udp.c b/test/udp.c
index 22a8d3243..d3ecc46b6 100644
--- a/test/udp.c
+++ b/test/udp.c
@@ -213,3 +213,77 @@ int test_udp(void)
 
 	return err;
 }
+
+
+static int udp_tos(const char *addr)
+{
+	struct udp_test *ut;
+	int layer = 0;
+	int err;
+
+	ut = mem_zalloc(sizeof(*ut), destructor);
+	if (!ut)
+		return ENOMEM;
+
+	err  = sa_set_str(&ut->cli, addr, 0);
+	err |= sa_set_str(&ut->srv, addr, 0);
+	if (err)
+		goto out;
+
+	err  = udp_listen(&ut->usc, &ut->cli, udp_recv_client, ut);
+	err |= udp_listen(&ut->uss, &ut->srv, udp_recv_server, ut);
+	if (err)
+		goto out;
+
+	udp_settos(ut->usc, 184);
+	udp_settos(ut->uss, 120);
+
+	udp_rxbuf_presz_set(ut->uss, 16);
+
+	err  = udp_local_get(ut->usc, &ut->cli);
+	err |= udp_local_get(ut->uss, &ut->srv);
+	if (err)
+		goto out;
+
+	err = udp_register_helper(&ut->uh, ut->usc, layer,
+				  udp_helper_send, udp_helper_recv, ut);
+	if (err)
+		goto out;
+
+	/* Send from connected client UDP socket */
+	err = udp_connect(ut->usc, &ut->srv);
+	if (err)
+		goto out;
+
+	/* Start test */
+	err = send_data(ut->usc, &ut->srv, data0);
+	if (err)
+		goto out;
+
+	err = re_main_timeout(100);
+	if (err)
+		goto out;
+
+	if (ut->err)
+		err = ut->err;
+
+ out:
+	mem_deref(ut);
+
+	return err;
+}
+
+
+int test_udp_tos(void)
+{
+	int err;
+
+	err = udp_tos("127.0.0.1");
+	TEST_ERR(err);
+
+	err = udp_tos("::1");
+	TEST_ERR(err);
+
+ out:
+	return err;
+}