Commit 2f02ca0b authored by Hubert Miś's avatar Hubert Miś Committed by Jukka Rissanen
Browse files

tests: net: socket: SO_PROTOCOL socket option 



This patch adds tests to verify getting SO_PROTOCOL socket option for
TCP, UDP, and TLS sockets.

Signed-off-by: default avatarHubert Miś <hubert.mis@nordicsemi.no>
parent ce3dcf23

tests/net/socket/tcp/src/main.c

+29 −0
Original line number Diff line number Diff line
@@ -541,6 +541,34 @@ void test_so_type(void) 
	k_sleep(TCP_TEARDOWN_TIMEOUT);

}



void test_so_protocol(void)

{

	struct sockaddr_in bind_addr4;

	struct sockaddr_in6 bind_addr6;

	int sock1, sock2, rv;

	int optval;

	socklen_t optlen = sizeof(optval);



	prepare_sock_tcp_v4(CONFIG_NET_CONFIG_MY_IPV4_ADDR, ANY_PORT,

			    &sock1, &bind_addr4);

	prepare_sock_tcp_v6(CONFIG_NET_CONFIG_MY_IPV6_ADDR, ANY_PORT,

			    &sock2, &bind_addr6);



	rv = getsockopt(sock1, SOL_SOCKET, SO_PROTOCOL, &optval, &optlen);

	zassert_equal(rv, 0, "getsockopt failed (%d)", errno);

	zassert_equal(optval, IPPROTO_TCP, "getsockopt got invalid protocol");

	zassert_equal(optlen, sizeof(optval), "getsockopt got invalid size");



	rv = getsockopt(sock2, SOL_SOCKET, SO_PROTOCOL, &optval, &optlen);

	zassert_equal(rv, 0, "getsockopt failed (%d)", errno);

	zassert_equal(optval, IPPROTO_TCP, "getsockopt got invalid protocol");

	zassert_equal(optlen, sizeof(optval), "getsockopt got invalid size");



	test_close(sock1);

	test_close(sock2);

	k_sleep(TCP_TEARDOWN_TIMEOUT);

}



void test_v4_so_rcvtimeo(void)

{

	int c_sock;
@@ -766,6 +794,7 @@ void test_main(void) 
		ztest_unit_test(test_open_close_immediately),

		ztest_user_unit_test(test_v4_accept_timeout),

		ztest_unit_test(test_so_type),

		ztest_unit_test(test_so_protocol),

		ztest_unit_test(test_v4_so_rcvtimeo),

		ztest_unit_test(test_v6_so_rcvtimeo),

		ztest_user_unit_test(test_socket_permission)

tests/net/socket/tls/src/main.c

+32 −1
Original line number Diff line number Diff line
@@ -53,6 +53,36 @@ void test_so_type(void) 
	k_sleep(TCP_TEARDOWN_TIMEOUT);

}



void test_so_protocol(void)

{

	struct sockaddr_in bind_addr4;

	struct sockaddr_in6 bind_addr6;

	int sock1, sock2, rv;

	int optval;

	socklen_t optlen = sizeof(optval);



	prepare_sock_tls_v4(CONFIG_NET_CONFIG_MY_IPV4_ADDR, ANY_PORT,

			    &sock1, &bind_addr4, IPPROTO_TLS_1_2);

	prepare_sock_tls_v6(CONFIG_NET_CONFIG_MY_IPV6_ADDR, ANY_PORT,

			    &sock2, &bind_addr6, IPPROTO_TLS_1_1);



	rv = getsockopt(sock1, SOL_SOCKET, SO_PROTOCOL, &optval, &optlen);

	zassert_equal(rv, 0, "getsockopt failed (%d)", errno);

	zassert_equal(optval, IPPROTO_TLS_1_2,

		      "getsockopt got invalid protocol");

	zassert_equal(optlen, sizeof(optval), "getsockopt got invalid size");



	rv = getsockopt(sock2, SOL_SOCKET, SO_PROTOCOL, &optval, &optlen);

	zassert_equal(rv, 0, "getsockopt failed (%d)", errno);

	zassert_equal(optval, IPPROTO_TLS_1_1,

		      "getsockopt got invalid protocol");

	zassert_equal(optlen, sizeof(optval), "getsockopt got invalid size");



	test_close(sock1);

	test_close(sock2);

	k_sleep(TCP_TEARDOWN_TIMEOUT);

}



void test_main(void)

{

	if (IS_ENABLED(CONFIG_NET_TC_THREAD_COOPERATIVE)) {
@@ -64,7 +94,8 @@ void test_main(void) 


	ztest_test_suite(

		socket_tls,

		ztest_unit_test(test_so_type)

		ztest_unit_test(test_so_type),

		ztest_unit_test(test_so_protocol)

		);



	ztest_run_test_suite(socket_tls);

tests/net/socket/udp/src/main.c

+30 −0
Original line number Diff line number Diff line
@@ -845,6 +845,35 @@ void test_so_rcvtimeo(void) 
	zassert_equal(rv, 0, "close failed");

}



void test_so_protocol(void)

{

	struct sockaddr_in bind_addr4;

	struct sockaddr_in6 bind_addr6;

	int sock1, sock2, rv;

	int optval;

	socklen_t optsize = sizeof(optval);



	prepare_sock_udp_v4(CONFIG_NET_CONFIG_MY_IPV4_ADDR, 55555,

			    &sock1, &bind_addr4);

	prepare_sock_udp_v6(CONFIG_NET_CONFIG_MY_IPV6_ADDR, 55555,

			    &sock2, &bind_addr6);



	rv = getsockopt(sock1, SOL_SOCKET, SO_PROTOCOL, &optval, &optsize);

	zassert_equal(rv, 0, "getsockopt failed (%d)", errno);

	zassert_equal(optval, IPPROTO_UDP, "getsockopt got invalid protocol");

	zassert_equal(optsize, sizeof(optval), "getsockopt got invalid size");



	rv = getsockopt(sock2, SOL_SOCKET, SO_PROTOCOL, &optval, &optsize);

	zassert_equal(rv, 0, "getsockopt failed (%d)", errno);

	zassert_equal(optval, IPPROTO_UDP, "getsockopt got invalid protocol");

	zassert_equal(optsize, sizeof(optval), "getsockopt got invalid size");



	rv = close(sock1);

	zassert_equal(rv, 0, "close failed");

	rv = close(sock2);

	zassert_equal(rv, 0, "close failed");

}



static void comm_sendmsg_with_txtime(int client_sock,

				     struct sockaddr *client_addr,

				     socklen_t client_addrlen,
@@ -1061,6 +1090,7 @@ void test_main(void) 
			 ztest_unit_test(test_so_priority),

			 ztest_unit_test(test_so_txtime),

			 ztest_unit_test(test_so_rcvtimeo),

			 ztest_unit_test(test_so_protocol),

			 ztest_unit_test(test_v4_sendmsg_recvfrom),

			 ztest_user_unit_test(test_v4_sendmsg_recvfrom),

			 ztest_unit_test(test_v4_sendmsg_recvfrom_no_aux_data),

CRA Git | Maintained and supported by SUSTech CRA and CCSE