mirror/lwip
1
0
Fork 0
mirror of https://github.com/lwip-tcpip/lwip.git synced 2024-10-04 05:39:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_socket: convert sendmsg test to use recvmsg

Browse Source 
This converts the sendmsg test to use recvmsg for receiving, thus
exercising both sendmsg and recvmsg in a single test

This also adjusts the test naming to communicate all message APIs
(sendmsg/recvmsg) are being tested
This commit is contained in:
Joel Cunningham 2017-03-17 14:17:42 -05:00
parent aef2accfa3
commit 02e957de1e
1 changed files with 34 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
test/unit/api/test_sockets.c
View File

@ -138,49 +138,47 @@ START_TEST(test_sockets_allfunctions_basic)
} }
END_TEST END_TEST
static void test_sockets_sendmsg_udp_send_recv_loop(int s, struct msghdr *msg) static void test_sockets_msgapi_udp_send_recv_loop(int s, struct msghdr *smsg, struct msghdr *rmsg)
{ {
int i, ret; int i, ret;
u8_t buf[4];
/* send/receive our datagram of IO vectors 10 times */ /* send/receive our datagram of IO vectors 10 times */
for (i = 0; i < 10; i++) { for (i = 0; i < 10; i++) {
ret = lwip_sendmsg(s, msg, 0); ret = lwip_sendmsg(s, smsg, 0);
fail_unless(ret == 4); fail_unless(ret == 4);
while (tcpip_thread_poll_one()); while (tcpip_thread_poll_one());
ret = lwip_recv(s, buf, sizeof(buf), 0); /* receive the datagram split across 4 buffers */
ret = lwip_recvmsg(s, rmsg, 0);
fail_unless(ret == 4); fail_unless(ret == 4);
/* verify data */ /* verify data */
fail_unless(buf[0] == 0xDE); fail_unless(*((u8_t*)rmsg->msg_iov[0].iov_base) == 0xDE);
fail_unless(buf[1] == 0xAD); fail_unless(*((u8_t*)rmsg->msg_iov[1].iov_base) == 0xAD);
fail_unless(buf[2] == 0xBE); fail_unless(*((u8_t*)rmsg->msg_iov[2].iov_base) == 0xBE);
fail_unless(buf[3] == 0xEF); fail_unless(*((u8_t*)rmsg->msg_iov[3].iov_base) == 0xEF);
} }
} }
static void test_sockets_sendmsg_udp(int domain) static void test_sockets_msgapi_udp(int domain)
{ {
int s, i, ret; int s, i, ret;
struct sockaddr_storage addr_storage; struct sockaddr_storage addr_storage;
socklen_t addr_size; socklen_t addr_size;
struct iovec riovs[4];
struct iovec iovs[4]; struct msghdr rmsg = {0};
struct msghdr msg; u8_t rcv_buf[4];
u8_t bytes[4]; struct iovec siovs[4];
struct msghdr smsg = {0};
/* each datagram should be 0xDEADBEEF */ u8_t snd_buf[4] = {0xDE, 0xAD, 0xBE, 0xEF};
bytes[0] = 0xDE;
bytes[1] = 0xAD;
bytes[2] = 0xBE;
bytes[3] = 0xEF;
/* initialize IO vectors with data */ /* initialize IO vectors with data */
for (i = 0; i < 4; i++) { for (i = 0; i < 4; i++) {
iovs[i].iov_base = &bytes[i]; siovs[i].iov_base = &snd_buf[i];
iovs[i].iov_len = sizeof(char); siovs[i].iov_len = sizeof(u8_t);
riovs[i].iov_base = &rcv_buf[i];
riovs[i].iov_len = sizeof(u8_t);
} }
/* set up address to send to */ /* set up address to send to */
@ -238,39 +236,39 @@ static void test_sockets_sendmsg_udp(int domain)
break; break;
} }
msg.msg_iov = iovs; /* send and receive the datagram in 4 pieces */
msg.msg_iovlen = 4; smsg.msg_iov = siovs;
msg.msg_control = NULL; smsg.msg_iovlen = 4;
msg.msg_controllen = 0; rmsg.msg_iov = riovs;
msg.msg_flags = 0; rmsg.msg_iovlen = 4;
/* perform a sendmsg with remote host (self) */ /* perform a sendmsg with remote host (self) */
msg.msg_name = &addr_storage; smsg.msg_name = &addr_storage;
msg.msg_namelen = addr_size; smsg.msg_namelen = addr_size;
test_sockets_sendmsg_udp_send_recv_loop(s, &msg); test_sockets_msgapi_udp_send_recv_loop(s, &smsg, &rmsg);
/* Connect to self, allowing us to not pass message name */ /* Connect to self, allowing us to not pass message name */
ret = lwip_connect(s, (struct sockaddr*)&addr_storage, addr_size); ret = lwip_connect(s, (struct sockaddr*)&addr_storage, addr_size);
fail_unless(ret == 0); fail_unless(ret == 0);
msg.msg_name = NULL; smsg.msg_name = NULL;
msg.msg_namelen = 0; smsg.msg_namelen = 0;
test_sockets_sendmsg_udp_send_recv_loop(s, &msg); test_sockets_msgapi_udp_send_recv_loop(s, &smsg, &rmsg);
ret = lwip_close(s); ret = lwip_close(s);
fail_unless(ret == 0); fail_unless(ret == 0);
} }
START_TEST(test_sockets_sendmsg) START_TEST(test_sockets_msgapis)
{ {
LWIP_UNUSED_ARG(_i); LWIP_UNUSED_ARG(_i);
#if LWIP_IPV4 #if LWIP_IPV4
test_sockets_sendmsg_udp(AF_INET); test_sockets_msgapi_udp(AF_INET);
#endif #endif
#if LWIP_IPV6 #if LWIP_IPV6
test_sockets_sendmsg_udp(AF_INET6); test_sockets_msgapi_udp(AF_INET6);
#endif #endif
} }
END_TEST END_TEST
@ -282,7 +280,7 @@ sockets_suite(void)
testfunc tests[] = { testfunc tests[] = {
TESTFUNC(test_sockets_basics), TESTFUNC(test_sockets_basics),
TESTFUNC(test_sockets_allfunctions_basic), TESTFUNC(test_sockets_allfunctions_basic),
TESTFUNC(test_sockets_sendmsg), TESTFUNC(test_sockets_msgapis),
}; };
return create_suite("SOCKETS", tests, sizeof(tests)/sizeof(testfunc), sockets_setup, sockets_teardown); return create_suite("SOCKETS", tests, sizeof(tests)/sizeof(testfunc), sockets_setup, sockets_teardown);
} }