mirror of
https://github.com/lwip-tcpip/lwip.git
synced 2024-08-25 23:52:51 +00:00
This migrates the sendmsg TCP test from socket examples (task #14408) to socket unit tests Additionally, this adds support for testing recvmsg, creating a TCP test for both sendmsg/recvmsg (referred to as msgapi test) This also makes a small change to msgapi UDP to clear the receive buffer after verifying the previous datagram
504 lines
13 KiB
C
504 lines
13 KiB
C
#include "test_sockets.h"
|
|
|
|
#include "lwip/mem.h"
|
|
#include "lwip/opt.h"
|
|
#include "lwip/sockets.h"
|
|
#include "lwip/stats.h"
|
|
|
|
#include "lwip/tcpip.h"
|
|
|
|
|
|
/* Setups/teardown functions */
|
|
|
|
static void
|
|
sockets_setup(void)
|
|
{
|
|
}
|
|
|
|
static void
|
|
sockets_teardown(void)
|
|
{
|
|
}
|
|
|
|
#ifndef NUM_SOCKETS
|
|
#define NUM_SOCKETS MEMP_NUM_NETCONN
|
|
#endif
|
|
|
|
#if LWIP_SOCKET
|
|
static int
|
|
test_sockets_alloc_socket_nonblocking(int domain, int type)
|
|
{
|
|
int s = lwip_socket(domain, type, 0);
|
|
if (s >= 0) {
|
|
int ret = lwip_fcntl(s, F_SETFL, O_NONBLOCK);
|
|
fail_unless(ret == 0);
|
|
}
|
|
return s;
|
|
}
|
|
|
|
/* Verify basic sockets functionality
|
|
*/
|
|
START_TEST(test_sockets_basics)
|
|
{
|
|
int s, i, ret;
|
|
int s2[NUM_SOCKETS];
|
|
LWIP_UNUSED_ARG(_i);
|
|
|
|
s = lwip_socket(AF_INET, SOCK_STREAM, 0);
|
|
fail_unless(s >= 0);
|
|
lwip_close(s);
|
|
|
|
for (i = 0; i < NUM_SOCKETS; i++) {
|
|
s2[i] = lwip_socket(AF_INET, SOCK_STREAM, 0);
|
|
fail_unless(s2[i] >= 0);
|
|
}
|
|
|
|
/* all sockets used, now it should fail */
|
|
s = lwip_socket(AF_INET, SOCK_STREAM, 0);
|
|
fail_unless(s == -1);
|
|
/* close one socket */
|
|
ret = lwip_close(s2[0]);
|
|
fail_unless(ret == 0);
|
|
/* now it should succeed */
|
|
s2[0] = lwip_socket(AF_INET, SOCK_STREAM, 0);
|
|
fail_unless(s2[0] >= 0);
|
|
|
|
/* close all sockets */
|
|
for (i = 0; i < NUM_SOCKETS; i++) {
|
|
ret = lwip_close(s2[i]);
|
|
fail_unless(ret == 0);
|
|
}
|
|
}
|
|
END_TEST
|
|
|
|
static void test_sockets_allfunctions_basic_domain(int domain)
|
|
{
|
|
int s, s2, s3, ret;
|
|
struct sockaddr_storage addr, addr2;
|
|
socklen_t addrlen, addr2len;
|
|
/* listen socket */
|
|
s = lwip_socket(domain, SOCK_STREAM, 0);
|
|
fail_unless(s >= 0);
|
|
|
|
ret = lwip_listen(s, 0);
|
|
fail_unless(ret == 0);
|
|
|
|
addrlen = sizeof(addr);
|
|
ret = lwip_getsockname(s, (struct sockaddr*)&addr, &addrlen);
|
|
fail_unless(ret == 0);
|
|
|
|
s2 = test_sockets_alloc_socket_nonblocking(domain, SOCK_STREAM);
|
|
fail_unless(s2 >= 0);
|
|
/* nonblocking connect s2 to s (but use loopback address) */
|
|
if (domain == AF_INET) {
|
|
#if LWIP_IPV4
|
|
struct sockaddr_in *addr4 = (struct sockaddr_in *)&addr;
|
|
addr4->sin_addr.s_addr = PP_HTONL(INADDR_LOOPBACK);
|
|
#endif
|
|
} else {
|
|
#if LWIP_IPV6
|
|
struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *)&addr;
|
|
struct in6_addr lo6 = IN6ADDR_LOOPBACK_INIT;
|
|
addr6->sin6_addr = lo6;
|
|
#endif
|
|
}
|
|
ret = lwip_connect(s2, (struct sockaddr*)&addr, addrlen);
|
|
fail_unless(ret == -1);
|
|
fail_unless(errno == EINPROGRESS);
|
|
ret = lwip_connect(s2, (struct sockaddr*)&addr, addrlen);
|
|
fail_unless(ret == -1);
|
|
fail_unless(errno == EALREADY);
|
|
|
|
while(tcpip_thread_poll_one());
|
|
|
|
s3 = lwip_accept(s, (struct sockaddr*)&addr2, &addr2len);
|
|
fail_unless(s3 >= 0);
|
|
|
|
ret = lwip_connect(s2, (struct sockaddr*)&addr, addrlen);
|
|
fail_unless(ret == -1);
|
|
fail_unless(errno == EISCONN);
|
|
|
|
ret = lwip_close(s);
|
|
fail_unless(ret == 0);
|
|
ret = lwip_close(s2);
|
|
fail_unless(ret == 0);
|
|
ret = lwip_close(s3);
|
|
fail_unless(ret == 0);
|
|
}
|
|
|
|
/* Try to step through all sockets functions once...
|
|
*/
|
|
START_TEST(test_sockets_allfunctions_basic)
|
|
{
|
|
LWIP_UNUSED_ARG(_i);
|
|
#if LWIP_IPV4
|
|
test_sockets_allfunctions_basic_domain(AF_INET);
|
|
#endif
|
|
#if LWIP_IPV6
|
|
test_sockets_allfunctions_basic_domain(AF_INET6);
|
|
#endif
|
|
}
|
|
END_TEST
|
|
|
|
static void test_sockets_init_loopback_addr(int domain, struct sockaddr_storage *addr_st, socklen_t *sz)
|
|
{
|
|
memset(addr_st, 0, sizeof(*addr_st));
|
|
switch(domain) {
|
|
#if LWIP_IPV6
|
|
case AF_INET6: {
|
|
struct sockaddr_in6 *addr = (struct sockaddr_in6*)addr_st;
|
|
struct in6_addr lo6 = IN6ADDR_LOOPBACK_INIT;
|
|
addr->sin6_family = AF_INET6;
|
|
addr->sin6_port = 0; /* use ephemeral port */
|
|
addr->sin6_addr = lo6;
|
|
*sz = sizeof(*addr);
|
|
}
|
|
break;
|
|
#endif /* LWIP_IPV6 */
|
|
#if LWIP_IPV4
|
|
case AF_INET: {
|
|
struct sockaddr_in *addr = (struct sockaddr_in*)addr_st;
|
|
addr->sin_family = AF_INET;
|
|
addr->sin_port = 0; /* use ephemeral port */
|
|
addr->sin_addr.s_addr = PP_HTONL(INADDR_LOOPBACK);
|
|
*sz = sizeof(*addr);
|
|
}
|
|
break;
|
|
#endif /* LWIP_IPV4 */
|
|
default:
|
|
*sz = 0;
|
|
fail();
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void test_sockets_msgapi_update_iovs(struct msghdr *msg, size_t bytes)
|
|
{
|
|
int i;
|
|
|
|
/* note: this modifies the underyling iov_base and iov_len for a partial
|
|
read for an individual vector. This updates the msg->msg_iov pointer
|
|
to skip fully consumed vecotrs */
|
|
|
|
/* process fully consumed vectors */
|
|
for (i = 0; i < msg->msg_iovlen; i++) {
|
|
if (msg->msg_iov[i].iov_len <= bytes) {
|
|
/* reduce bytes by amount of this vector */
|
|
bytes -= msg->msg_iov[i].iov_len;
|
|
} else {
|
|
break; /* iov not fully consumed */
|
|
}
|
|
}
|
|
|
|
/* slide down over fully consumed vectors */
|
|
msg->msg_iov = &msg->msg_iov[i];
|
|
msg->msg_iovlen -= i;
|
|
|
|
/* update new first vector with any remaining amount */
|
|
msg->msg_iov[0].iov_base = ((u8_t *)msg->msg_iov[0].iov_base + bytes);
|
|
msg->msg_iov[0].iov_len -= bytes;
|
|
}
|
|
|
|
static void test_sockets_msgapi_tcp(int domain)
|
|
{
|
|
#define BUF_SZ (TCP_SND_BUF/4)
|
|
#define TOTAL_DATA_SZ (BUF_SZ*8) /* ~(TCP_SND_BUF*2) that accounts for integer rounding */
|
|
int listnr, s1, s2, i, ret, opt;
|
|
int bytes_written, bytes_read;
|
|
struct sockaddr_storage addr_storage;
|
|
socklen_t addr_size;
|
|
struct iovec siovs[8];
|
|
struct msghdr smsg;
|
|
u8_t * snd_buf;
|
|
struct iovec riovs[5];
|
|
struct iovec riovs_tmp[5];
|
|
struct msghdr rmsg;
|
|
u8_t * rcv_buf;
|
|
int rcv_off;
|
|
int rcv_trailer;
|
|
u8_t val;
|
|
|
|
test_sockets_init_loopback_addr(domain, &addr_storage, &addr_size);
|
|
|
|
listnr = test_sockets_alloc_socket_nonblocking(domain, SOCK_STREAM);
|
|
fail_unless(listnr >= 0);
|
|
s1 = test_sockets_alloc_socket_nonblocking(domain, SOCK_STREAM);
|
|
fail_unless(s1 >= 0);
|
|
|
|
/* setup a listener socket on loopback with ephemeral port */
|
|
ret = lwip_bind(listnr, (struct sockaddr*)&addr_storage, addr_size);
|
|
fail_unless(ret == 0);
|
|
ret = lwip_listen(listnr, 0);
|
|
fail_unless(ret == 0);
|
|
|
|
/* update address with ephemeral port */
|
|
ret = lwip_getsockname(listnr, (struct sockaddr*)&addr_storage, &addr_size);
|
|
fail_unless(ret == 0);
|
|
|
|
/* connect, won't complete until we accept it */
|
|
ret = lwip_connect(s1, (struct sockaddr*)&addr_storage, addr_size);
|
|
fail_unless(ret == -1);
|
|
fail_unless(errno == EINPROGRESS);
|
|
|
|
while (tcpip_thread_poll_one());
|
|
|
|
/* accept, creating the other side of the connection */
|
|
s2 = lwip_accept(listnr, NULL, NULL);
|
|
fail_unless(s2 >= 0);
|
|
|
|
/* double check s1 is connected */
|
|
ret = lwip_connect(s1, (struct sockaddr*)&addr_storage, addr_size);
|
|
fail_unless(ret == -1);
|
|
fail_unless(errno == EISCONN);
|
|
|
|
/* set s2 to non-blocking, not inherited from listener */
|
|
opt = lwip_fcntl(s2, F_GETFL, 0);
|
|
fail_unless(opt == 0);
|
|
opt |= O_NONBLOCK;
|
|
ret = lwip_fcntl(s2, F_SETFL, opt);
|
|
fail_unless(ret == 0);
|
|
|
|
/* we are done with listener, close it */
|
|
ret = lwip_close(listnr);
|
|
fail_unless(ret == 0);
|
|
|
|
/* allocate a buffer for a stream of incrementing hex (0x00..0xFF) which we will use
|
|
to create an input vector set that is larger than the TCP's send buffer. This will
|
|
force execution of the partial IO vector send case */
|
|
snd_buf = (u8_t*)mem_malloc(BUF_SZ);
|
|
val = 0x00;
|
|
fail_unless(snd_buf != NULL);
|
|
for (i = 0; i < BUF_SZ; i++,val++) {
|
|
snd_buf[i] = val;
|
|
}
|
|
|
|
/* send the buffer 8 times in one message, equating to TOTAL_DATA_SZ */
|
|
for (i = 0; i < 8; i++) {
|
|
siovs[i].iov_base = snd_buf;
|
|
siovs[i].iov_len = BUF_SZ;
|
|
}
|
|
|
|
/* allocate a receive buffer, same size as snd_buf for easy verification */
|
|
rcv_buf = (u8_t*)mem_malloc(BUF_SZ);
|
|
fail_unless(rcv_buf != NULL);
|
|
memset(rcv_buf, 0, BUF_SZ);
|
|
/* split across iovs */
|
|
for (i = 0; i < 4; i++) {
|
|
riovs[i].iov_base = &rcv_buf[i*(BUF_SZ/4)];
|
|
riovs[i].iov_len = BUF_SZ/4;
|
|
}
|
|
/* handling trailing bytes if buffer doesn't evenly divide by 4 */
|
|
if ((BUF_SZ % 4) != 0) {
|
|
riovs[5].iov_base = &rcv_buf[4*(BUF_SZ/4)];
|
|
riovs[5].iov_len = BUF_SZ - (4*(BUF_SZ/4));
|
|
rcv_trailer = 1;
|
|
} else {
|
|
rcv_trailer = 0;
|
|
}
|
|
/* we use a copy of riovs since we'll be modifying base and len during
|
|
receiving. This gives us an easy way to reset the iovs for next recvmsg */
|
|
memcpy(riovs_tmp, riovs, sizeof(riovs));
|
|
|
|
memset(&smsg, 0, sizeof(smsg));
|
|
smsg.msg_iov = siovs;
|
|
smsg.msg_iovlen = 8;
|
|
|
|
memset(&rmsg, 0, sizeof(rmsg));
|
|
rmsg.msg_iov = riovs_tmp;
|
|
rmsg.msg_iovlen = (rcv_trailer ? 5 : 4);
|
|
|
|
bytes_written = 0;
|
|
bytes_read = 0;
|
|
rcv_off = 0;
|
|
|
|
while (bytes_written < TOTAL_DATA_SZ && (bytes_read < TOTAL_DATA_SZ)) {
|
|
/* send data */
|
|
if (bytes_written < TOTAL_DATA_SZ) {
|
|
ret = lwip_sendmsg(s1, &smsg, 0);
|
|
/* note: since we always receive after sending, there will be open
|
|
space in the send buffer */
|
|
fail_unless(ret > 0);
|
|
|
|
bytes_written += ret;
|
|
if (bytes_written < TOTAL_DATA_SZ) {
|
|
test_sockets_msgapi_update_iovs(&smsg, (size_t)ret);
|
|
}
|
|
}
|
|
|
|
while (tcpip_thread_poll_one());
|
|
|
|
/* receive and verify data */
|
|
do {
|
|
if (bytes_read < TOTAL_DATA_SZ) {
|
|
ret = lwip_recvmsg(s2, &rmsg, 0);
|
|
fail_unless(ret > 0 || (ret == -1 && errno == EWOULDBLOCK));
|
|
|
|
if (ret > 0) {
|
|
rcv_off += ret;
|
|
/* we have received a full buffer */
|
|
if (rcv_off == BUF_SZ) {
|
|
/* note: since iovs are just pointers, compare underlying buf */
|
|
fail_unless(!memcmp(snd_buf, rcv_buf, BUF_SZ));
|
|
bytes_read += BUF_SZ;
|
|
/* reset receive state for next buffer */
|
|
rcv_off = 0;
|
|
memset(rcv_buf, 0, BUF_SZ);
|
|
memcpy(riovs_tmp, riovs, sizeof(riovs));
|
|
rmsg.msg_iov = riovs_tmp;
|
|
rmsg.msg_iovlen = (rcv_trailer ? 5 : 4);
|
|
} else { /* partial read */
|
|
test_sockets_msgapi_update_iovs(&rmsg, (size_t)ret);
|
|
}
|
|
}
|
|
} else {
|
|
break;
|
|
}
|
|
} while(ret > 0);
|
|
}
|
|
|
|
ret = lwip_close(s1);
|
|
fail_unless(ret == 0);
|
|
ret = lwip_close(s2);
|
|
fail_unless(ret == 0);
|
|
mem_free(snd_buf);
|
|
mem_free(rcv_buf);
|
|
}
|
|
|
|
static void test_sockets_msgapi_udp_send_recv_loop(int s, struct msghdr *smsg, struct msghdr *rmsg)
|
|
{
|
|
int i, ret;
|
|
|
|
/* send/receive our datagram of IO vectors 10 times */
|
|
for (i = 0; i < 10; i++) {
|
|
ret = lwip_sendmsg(s, smsg, 0);
|
|
fail_unless(ret == 4);
|
|
|
|
while (tcpip_thread_poll_one());
|
|
|
|
/* receive the datagram split across 4 buffers */
|
|
ret = lwip_recvmsg(s, rmsg, 0);
|
|
fail_unless(ret == 4);
|
|
|
|
/* verify data */
|
|
fail_unless(*((u8_t*)rmsg->msg_iov[0].iov_base) == 0xDE);
|
|
fail_unless(*((u8_t*)rmsg->msg_iov[1].iov_base) == 0xAD);
|
|
fail_unless(*((u8_t*)rmsg->msg_iov[2].iov_base) == 0xBE);
|
|
fail_unless(*((u8_t*)rmsg->msg_iov[3].iov_base) == 0xEF);
|
|
|
|
/* clear rcv_buf to ensure no data is being skipped */
|
|
*((u8_t*)rmsg->msg_iov[0].iov_base) = 0x00;
|
|
*((u8_t*)rmsg->msg_iov[1].iov_base) = 0x00;
|
|
*((u8_t*)rmsg->msg_iov[2].iov_base) = 0x00;
|
|
*((u8_t*)rmsg->msg_iov[3].iov_base) = 0x00;
|
|
}
|
|
}
|
|
|
|
static void test_sockets_msgapi_udp(int domain)
|
|
{
|
|
int s, i, ret;
|
|
struct sockaddr_storage addr_storage;
|
|
socklen_t addr_size;
|
|
struct iovec riovs[4];
|
|
struct msghdr rmsg;
|
|
u8_t rcv_buf[4];
|
|
struct iovec siovs[4];
|
|
struct msghdr smsg;
|
|
u8_t snd_buf[4] = {0xDE, 0xAD, 0xBE, 0xEF};
|
|
|
|
/* initialize IO vectors with data */
|
|
for (i = 0; i < 4; i++) {
|
|
siovs[i].iov_base = &snd_buf[i];
|
|
siovs[i].iov_len = sizeof(u8_t);
|
|
riovs[i].iov_base = &rcv_buf[i];
|
|
riovs[i].iov_len = sizeof(u8_t);
|
|
}
|
|
|
|
test_sockets_init_loopback_addr(domain, &addr_storage, &addr_size);
|
|
|
|
s = test_sockets_alloc_socket_nonblocking(domain, SOCK_DGRAM);
|
|
fail_unless(s >= 0);
|
|
|
|
ret = lwip_bind(s, (struct sockaddr*)&addr_storage, addr_size);
|
|
fail_unless(ret == 0);
|
|
|
|
/* Update addr with epehermal port */
|
|
ret = lwip_getsockname(s, (struct sockaddr*)&addr_storage, &addr_size);
|
|
fail_unless(ret == 0);
|
|
switch(domain) {
|
|
#if LWIP_IPV6
|
|
case AF_INET6:
|
|
fail_unless(addr_size == sizeof(struct sockaddr_in6));
|
|
break;
|
|
#endif /* LWIP_IPV6 */
|
|
#if LWIP_IPV4
|
|
case AF_INET:
|
|
fail_unless(addr_size == sizeof(struct sockaddr_in));
|
|
break;
|
|
#endif /* LWIP_IPV6 */
|
|
default:
|
|
fail();
|
|
break;
|
|
}
|
|
|
|
/* send and receive the datagram in 4 pieces */
|
|
memset(&smsg, 0, sizeof(smsg));
|
|
smsg.msg_iov = siovs;
|
|
smsg.msg_iovlen = 4;
|
|
memset(&rmsg, 0, sizeof(rmsg));
|
|
rmsg.msg_iov = riovs;
|
|
rmsg.msg_iovlen = 4;
|
|
|
|
/* perform a sendmsg with remote host (self) */
|
|
smsg.msg_name = &addr_storage;
|
|
smsg.msg_namelen = addr_size;
|
|
|
|
test_sockets_msgapi_udp_send_recv_loop(s, &smsg, &rmsg);
|
|
|
|
/* Connect to self, allowing us to not pass message name */
|
|
ret = lwip_connect(s, (struct sockaddr*)&addr_storage, addr_size);
|
|
fail_unless(ret == 0);
|
|
|
|
smsg.msg_name = NULL;
|
|
smsg.msg_namelen = 0;
|
|
|
|
test_sockets_msgapi_udp_send_recv_loop(s, &smsg, &rmsg);
|
|
|
|
ret = lwip_close(s);
|
|
fail_unless(ret == 0);
|
|
}
|
|
|
|
START_TEST(test_sockets_msgapis)
|
|
{
|
|
LWIP_UNUSED_ARG(_i);
|
|
#if LWIP_IPV4
|
|
test_sockets_msgapi_udp(AF_INET);
|
|
test_sockets_msgapi_tcp(AF_INET);
|
|
#endif
|
|
#if LWIP_IPV6
|
|
test_sockets_msgapi_udp(AF_INET6);
|
|
test_sockets_msgapi_tcp(AF_INET6);
|
|
#endif
|
|
}
|
|
END_TEST
|
|
|
|
/** Create the suite including all tests for this module */
|
|
Suite *
|
|
sockets_suite(void)
|
|
{
|
|
testfunc tests[] = {
|
|
TESTFUNC(test_sockets_basics),
|
|
TESTFUNC(test_sockets_allfunctions_basic),
|
|
TESTFUNC(test_sockets_msgapis),
|
|
};
|
|
return create_suite("SOCKETS", tests, sizeof(tests)/sizeof(testfunc), sockets_setup, sockets_teardown);
|
|
}
|
|
|
|
#else /* LWIP_SOCKET */
|
|
|
|
Suite *
|
|
sockets_suite(void)
|
|
{
|
|
return create_suite("SOCKETS", NULL, 0, NULL, NULL);
|
|
}
|
|
#endif /* LWIP_SOCKET */
|