diff --git a/test/src/apps/network/sendmmsg.c b/test/src/apps/network/sendmmsg.c new file mode 100644 index 000000000..71bd7b0ce --- /dev/null +++ b/test/src/apps/network/sendmmsg.c @@ -0,0 +1,130 @@ +// SPDX-License-Identifier: MPL-2.0 + +#define _GNU_SOURCE +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include "../test.h" + +int sk_sender; +int sk_receiver; +struct sockaddr_in sender_addr; +struct sockaddr_in receiver_addr; +socklen_t sockaddr_len = sizeof(struct sockaddr_in); + +#define SENDER_PORT 13245 +#define RECEIVER_PORT 13246 + +#define NUM_MESSAGES 3 +#define MAX_BUFFER_SIZE 256 + +void set_nonblocking(int fd) +{ + int flags = CHECK(fcntl(fd, F_GETFL, 0)); + CHECK(fcntl(fd, F_SETFL, flags | O_NONBLOCK)); +} + +FN_SETUP(create_and_bind) +{ + sk_sender = CHECK(socket(AF_INET, SOCK_DGRAM, 0)); + sk_receiver = CHECK(socket(AF_INET, SOCK_DGRAM, 0)); + + memset(&sender_addr, 0, sockaddr_len); + sender_addr.sin_family = AF_INET; + sender_addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK); + sender_addr.sin_port = htons(SENDER_PORT); + CHECK(bind(sk_sender, (struct sockaddr *)&sender_addr, sockaddr_len)); + + memset(&receiver_addr, 0, sockaddr_len); + receiver_addr.sin_family = AF_INET; + receiver_addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK); + receiver_addr.sin_port = htons(RECEIVER_PORT); + CHECK(bind(sk_receiver, (struct sockaddr *)&receiver_addr, + sockaddr_len)); + + set_nonblocking(sk_sender); + set_nonblocking(sk_receiver); + + CHECK(connect(sk_sender, (struct sockaddr *)&receiver_addr, + sockaddr_len)); +} +END_SETUP() + +bool check_len(char buffers[][MAX_BUFFER_SIZE], struct mmsghdr *msgs) +{ + for (int i = 0; i < NUM_MESSAGES; ++i) { + if (i == NUM_MESSAGES - 1) { + break; + } + + if (msgs[i].msg_len != strlen(buffers[i])) { + return false; + } + } + + return true; +} + +FN_TEST(sendmmsg) +{ + struct mmsghdr msgs[NUM_MESSAGES]; + struct iovec iovs[NUM_MESSAGES]; + char buffers[NUM_MESSAGES][MAX_BUFFER_SIZE]; + + for (int i = 0; i < NUM_MESSAGES; ++i) { + if (i == NUM_MESSAGES - 1) { + msgs[i].msg_hdr.msg_name = NULL; + msgs[i].msg_hdr.msg_namelen = 0; + msgs[i].msg_hdr.msg_iov = NULL; + msgs[i].msg_hdr.msg_iovlen = 0x100; + break; + } + + memset(buffers[i], 0, MAX_BUFFER_SIZE); + sprintf(buffers[i], "The %d Hello from sender!", i); + iovs[i].iov_base = buffers[i]; + iovs[i].iov_len = strlen(buffers[i]); + + msgs[i].msg_hdr.msg_iov = &iovs[i]; + msgs[i].msg_hdr.msg_iovlen = 1; + msgs[i].msg_hdr.msg_name = NULL; + msgs[i].msg_hdr.msg_namelen = 0; + msgs[i].msg_hdr.msg_control = NULL; + msgs[i].msg_hdr.msg_controllen = 0; + msgs[i].msg_hdr.msg_flags = 0; + msgs[i].msg_len = 0x12345; + } + + TEST_RES(sendmmsg(sk_sender, msgs, NUM_MESSAGES, 0), + _ret == NUM_MESSAGES - 1 && check_len(buffers, msgs)); + + struct msghdr msghdr; + char recv_buffers[NUM_MESSAGES][MAX_BUFFER_SIZE]; + for (int i = 0; i < NUM_MESSAGES; ++i) { + if (i == NUM_MESSAGES - 1) { + TEST_ERRNO(recvmsg(sk_receiver, &msghdr, 0), EAGAIN); + break; + } + + memset(recv_buffers[i], 0, MAX_BUFFER_SIZE); + + iovs[i].iov_base = recv_buffers[i]; + iovs[i].iov_len = MAX_BUFFER_SIZE; + msghdr.msg_name = NULL; + msghdr.msg_namelen = 0; + msghdr.msg_iov = &iovs[i]; + msghdr.msg_iovlen = 1; + msghdr.msg_control = NULL; + msghdr.msg_controllen = 0; + TEST_RES(recvmsg(sk_receiver, &msghdr, 0), + _ret == strlen(buffers[i]) && + (strcmp(buffers[i], recv_buffers[i]) == 0)); + } +} +END_TEST() \ No newline at end of file diff --git a/test/src/apps/network/udp_broadcast.c b/test/src/apps/network/udp_broadcast.c index 1121bc9da..f8842fca5 100644 --- a/test/src/apps/network/udp_broadcast.c +++ b/test/src/apps/network/udp_broadcast.c @@ -1,5 +1,6 @@ // SPDX-License-Identifier: MPL-2.0 +#define _GNU_SOURCE #include #include #include diff --git a/test/src/apps/scripts/network.sh b/test/src/apps/scripts/network.sh index a6e2708b6..0c88fea08 100755 --- a/test/src/apps/scripts/network.sh +++ b/test/src/apps/scripts/network.sh @@ -38,6 +38,7 @@ sleep 0.2 ./unix_stream_err ./unix_seqpacket_err ./unix_datagram_err +./sendmmsg ./netlink_route ./rtnl_err diff --git a/test/src/syscall/ltp/testcases/all.txt b/test/src/syscall/ltp/testcases/all.txt index 73d1fc48b..a679cb5e7 100644 --- a/test/src/syscall/ltp/testcases/all.txt +++ b/test/src/syscall/ltp/testcases/all.txt @@ -1324,8 +1324,8 @@ sendfile08_64 # sendmsg02 # sendmsg03 -# sendmmsg01 -# sendmmsg02 +sendmmsg01 +sendmmsg02 # sendto01 # sendto02