From 0c5a43a8a05d74de6c3ffa4f6c42429c2861d75f Mon Sep 17 00:00:00 2001 From: Andreas Schneider Date: Tue, 17 May 2016 10:01:38 +0200 Subject: tests: Add test_connect_sendto_null_ipv4() test Signed-off-by: Andreas Schneider Reviewed-by: Ralph Boehme --- tests/test_echo_udp_sendto_recvfrom.c | 52 +++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) (limited to 'tests/test_echo_udp_sendto_recvfrom.c') diff --git a/tests/test_echo_udp_sendto_recvfrom.c b/tests/test_echo_udp_sendto_recvfrom.c index 014eece..bb22371 100644 --- a/tests/test_echo_udp_sendto_recvfrom.c +++ b/tests/test_echo_udp_sendto_recvfrom.c @@ -241,6 +241,55 @@ static void test_connect_sendto_ipv4(void **state) close(s); } +static void test_connect_sendto_null_ipv4(void **state) +{ + struct torture_address addr = { + .sa_socklen = sizeof(struct sockaddr_in), + }; + char send_buf[] = "packet.0"; + char recv_buf[64] = {0}; + ssize_t ret; + int rc; + int s; + + (void) state; /* unused */ + + addr.sa.in.sin_family = AF_INET; + addr.sa.in.sin_port = htons(torture_server_port()); + + rc = inet_pton(AF_INET, + torture_server_address(AF_INET), + &addr.sa.in.sin_addr); + assert_int_equal(rc, 1); + + s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); + assert_int_not_equal(s, -1); + + /* Now, connect */ + rc = connect(s, &addr.sa.s, addr.sa_socklen); + assert_return_code(rc, errno); + + ret = sendto(s, + send_buf, + sizeof(send_buf), + 0, + NULL, + 0); + assert_return_code(ret, errno); + + ret = recvfrom(s, + recv_buf, + sizeof(recv_buf), + 0, + NULL, + 0); + assert_return_code(ret, errno); + + assert_memory_equal(send_buf, recv_buf, sizeof(send_buf)); + + close(s); +} + int main(void) { int rc; @@ -256,6 +305,9 @@ int main(void) { cmocka_unit_test_setup_teardown(test_connect_sendto_ipv4, setup_echo_srv_udp_ipv4, teardown), + cmocka_unit_test_setup_teardown(test_connect_sendto_null_ipv4, + setup_echo_srv_udp_ipv4, + teardown), }; rc = cmocka_run_group_tests(sendto_tests, NULL, NULL); -- cgit