Add test for sendmmsg

test/src/apps/network/sendmmsg.c

@@ -0,0 +1,130 @@
+// SPDX-License-Identifier: MPL-2.0
+
+#define _GNU_SOURCE
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <fcntl.h>
+#include <stdbool.h>
+#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()

test/src/apps/network/udp_broadcast.c

@@ -1,5 +1,6 @@
 // SPDX-License-Identifier: MPL-2.0
 
+#define _GNU_SOURCE
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>

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

test/src/syscall/ltp/testcases/all.txt

@@ -1324,8 +1324,8 @@ sendfile08_64
 # sendmsg02
 # sendmsg03
 
-# sendmmsg01
-# sendmmsg02
+sendmmsg01
+sendmmsg02
 
 # sendto01
 # sendto02