[apple/xnu.git] / tests / recvmsg_x_test.c

/*
 * Copyright (c) 2020 Apple Inc. All rights reserved.
 *
 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
 *
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. The rights granted to you under the License
 * may not be used to create, or enable the creation or redistribution of,
 * unlawful or unlicensed copies of an Apple operating system, or to
 * circumvent, violate, or enable the circumvention or violation of, any
 * terms of an Apple operating system software license agreement.
 *
 * Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this file.
 *
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 *
 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
 */

/* -*- compile-command: "xcrun --sdk iphoneos.internal make recvmsg_x_test" -*- */


#include <sys/errno.h>
#include <sys/fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <stdbool.h>
#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <unistd.h>

#include <darwintest.h>
#include <darwintest_utils.h>

#define NMSGS       5
#define BUFFERLEN   1000

T_GLOBAL_META(T_META_NAMESPACE("xnu.net"));

static void
sendPackets(int s, struct sockaddr *dst, unsigned int numMsg, size_t bufferLen)
{
	ssize_t count = 0;
	struct msghdr msg = {};
	struct iovec vec = {};
	char *bytes = calloc(1, bufferLen);
	if (bytes == NULL) {
		err(EX_OSERR, "calloc()");
	}

	vec.iov_base = bytes;
	vec.iov_len = bufferLen;

	msg.msg_name = (void *)dst;
	msg.msg_namelen = dst->sa_len;
	msg.msg_iov = &vec;
	msg.msg_iovlen = 1;
	msg.msg_flags = 0;

	for (unsigned int i = 0; i < numMsg; i++) {
		ssize_t n;
		T_QUIET; T_EXPECT_POSIX_SUCCESS(n = sendmsg(s, &msg, 0), "sendmsg()");
		T_LOG("Sent %ld bytes\n", n);
		count += 1;
	}

	// Wait a bit to make sure the packets reach the receiver
	usleep(100000);

	T_LOG("Sent %ld packet\n", count);

	free(bytes);
}

static void
recvPackets_x(int s, unsigned int numMsg, size_t buflen, socklen_t cmsgLen)
{
	struct msghdr_x *msgList;
	struct sockaddr_in *srcAddrs;
	struct iovec *vec;
	char *buffers;
	char *cmsgBuf;

	T_QUIET; T_ASSERT_NOTNULL(msgList = calloc(numMsg, sizeof(struct msghdr_x)), "msgList calloc()");
	T_QUIET; T_ASSERT_NOTNULL(srcAddrs = calloc(numMsg, sizeof(struct sockaddr_in)), "srcAddrs calloc()");
	T_QUIET; T_ASSERT_NOTNULL(vec = calloc(numMsg, sizeof(struct iovec)), "vec calloc()");
	T_QUIET; T_ASSERT_NOTNULL(buffers = calloc(numMsg, buflen), "buffers calloc()");
	T_QUIET; T_ASSERT_NOTNULL(cmsgBuf = calloc(numMsg, ALIGN(cmsgLen)), "cmsgBuf calloc()");

	u_int count = 0;
	while (true) {
		/*
		 * Wrap around when we've exhausted the list
		 */
		if ((count % numMsg) == 0) {
			for (unsigned int i = 0; i < numMsg; i++) {
				struct msghdr_x *msg = &msgList[i];
				msg->msg_name = &srcAddrs[i];
				msg->msg_namelen = sizeof(srcAddrs[i]);
				vec[i].iov_base = buffers + (i * buflen);
				vec[i].iov_len = buflen;
				msg->msg_iov = &vec[i];
				msg->msg_iovlen = 1;
				msg->msg_control = cmsgBuf + (i * ALIGN(cmsgLen));
				msg->msg_controllen = cmsgLen;
				msg->msg_flags = 0;

				T_QUIET; T_EXPECT_TRUE((uintptr_t)msg->msg_control % sizeof(uint32_t) == 0, NULL);
			}
		}

		ssize_t n = recvmsg_x(s, msgList + (count % numMsg), numMsg - (count % numMsg), 0);
		if (n < 0) {
			if (errno == EINTR) {
				T_LOG("recvmsg_x(): %s", strerror(errno));
				continue;
			}
			if (errno == EWOULDBLOCK) {
				T_LOG("recvmsg_x(): %s", strerror(errno));
				break;
			}
			T_FAIL("recvmsg_x() failed: %s", strerror(errno));
		}
		T_LOG("recvmsg_x returned %ld packets\n", n);

		for (unsigned int i = count; i < count + (u_int)n; i++) {
			struct msghdr_x *msg = &msgList[i % numMsg];

			T_LOG("Received packet #%d %lu bytes with recvmsg_x(), msg_namelen = %u, msg_controllen = %d -> %d, msg_flags = 0x%x\n",
			    i + 1, msg->msg_datalen, msg->msg_namelen, cmsgLen, msg->msg_controllen, msg->msg_flags);

			struct cmsghdr *cmsg;

			for (cmsg = CMSG_FIRSTHDR(msg); cmsg; cmsg = CMSG_NXTHDR(msg, cmsg)) {
				T_QUIET; T_EXPECT_TRUE((uintptr_t)cmsg % sizeof(uint32_t) == 0, NULL);

				T_LOG("level = %d, type = %d, length = %d\n", cmsg->cmsg_level, cmsg->cmsg_type, cmsg->cmsg_len);
			}
		}

		count += (u_int)n;
	}

	free(msgList);
	free(srcAddrs);
	free(vec);
	free(buffers);
	free(cmsgBuf);
}

T_DECL(recvmsg_x_test, "exercise revcmsg_x() with various parameter")
{
	struct sockaddr_in addr = {
		.sin_len = sizeof(addr),
		.sin_family = AF_INET,
		.sin_addr.s_addr = htonl(0x7f000001),
		.sin_port = 0
	};

	int recvSocket;
	T_QUIET; T_EXPECT_POSIX_SUCCESS(recvSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP), "socket()");
	T_QUIET; T_EXPECT_POSIX_SUCCESS(bind(recvSocket, (const struct sockaddr *)&addr, sizeof(addr)), "bind()");

	socklen_t addrLen = sizeof(addr);
	T_QUIET; T_EXPECT_POSIX_SUCCESS(getsockname(recvSocket, (struct sockaddr *)&addr, &addrLen), "getsockname()");

	int one = 1;
	T_QUIET; T_EXPECT_POSIX_SUCCESS(setsockopt(recvSocket, IPPROTO_IP, IP_RECVPKTINFO, (void *)&one, sizeof(one)), "setsockopt(IP_RECVPKTINFO)");

	int flags = fcntl(recvSocket, F_GETFL, 0);
	T_QUIET; T_EXPECT_POSIX_SUCCESS(fcntl(recvSocket, F_SETFL, flags | O_NONBLOCK), "fcntl()");

	int sendSocket;
	T_QUIET; T_EXPECT_POSIX_SUCCESS(sendSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP), "sendSocket socket()");

	for (int dontTrunc = 0; dontTrunc <= 1; dontTrunc++) {
		T_QUIET; T_EXPECT_POSIX_SUCCESS(setsockopt(recvSocket, SOL_SOCKET, SO_DONTTRUNC, (void *)&dontTrunc, sizeof(dontTrunc)), "setsockopt(SO_DONTTRUNC)");

		T_LOG("\n================= recvmsg_x() test =================\n");
		sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
		recvPackets_x(recvSocket, NMSGS, BUFFERLEN, 50);

		T_LOG("\n================= recvmsg_x() test =================\n");
		sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
		recvPackets_x(recvSocket, NMSGS, BUFFERLEN * 2, 50);

		T_LOG("\n================= recvmsg_x() test =================\n");
		sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
		recvPackets_x(recvSocket, NMSGS, BUFFERLEN / 2, 50);

		T_LOG("\n================= recvmsg_x() test =================\n");
		sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
		recvPackets_x(recvSocket, NMSGS, BUFFERLEN, 10);

		T_LOG("\n================= recvmsg_x() test =================\n");
		sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
		recvPackets_x(recvSocket, NMSGS, BUFFERLEN / 2, 10);
	}

	close(sendSocket);
	close(recvSocket);

	T_LOG("\n================= PASS =================\n");
}
Commit	Line	Data
c3c9b80d A	1	/*
	2	* Copyright (c) 2020 Apple Inc. All rights reserved.
	3	*
	4	* @APPLE_OSREFERENCE_LICENSE_HEADER_START@
	5	*
	6	* This file contains Original Code and/or Modifications of Original Code
	7	* as defined in and that are subject to the Apple Public Source License
	8	* Version 2.0 (the 'License'). You may not use this file except in
	9	* compliance with the License. The rights granted to you under the License
	10	* may not be used to create, or enable the creation or redistribution of,
	11	* unlawful or unlicensed copies of an Apple operating system, or to
	12	* circumvent, violate, or enable the circumvention or violation of, any
	13	* terms of an Apple operating system software license agreement.
	14	*
	15	* Please obtain a copy of the License at
	16	* http://www.opensource.apple.com/apsl/ and read it before using this file.
	17	*
	18	* The Original Code and all software distributed under the License are
	19	* distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
	20	* EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
	21	* INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
	22	* FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
	23	* Please see the License for the specific language governing rights and
	24	* limitations under the License.
	25	*
	26	* @APPLE_OSREFERENCE_LICENSE_HEADER_END@
	27	*/
	28
	29	/* -- compile-command: "xcrun --sdk iphoneos.internal make recvmsg_x_test" -- */
	30
	31
	32	#include <sys/errno.h>
	33	#include <sys/fcntl.h>
	34	#include <sys/socket.h>
	35	#include <netinet/in.h>
	36	#include <stdbool.h>
	37	#include <err.h>
	38	#include <stdio.h>
	39	#include <stdlib.h>
	40	#include <string.h>
	41	#include <sysexits.h>
	42	#include <unistd.h>
	43
	44	#include <darwintest.h>
	45	#include <darwintest_utils.h>
	46
	47	#define NMSGS 5
	48	#define BUFFERLEN 1000
	49
	50	T_GLOBAL_META(T_META_NAMESPACE("xnu.net"));
	51
	52	static void
	53	sendPackets(int s, struct sockaddr *dst, unsigned int numMsg, size_t bufferLen)
	54	{
	55	ssize_t count = 0;
	56	struct msghdr msg = {};
	57	struct iovec vec = {};
	58	char *bytes = calloc(1, bufferLen);
	59	if (bytes == NULL) {
	60	err(EX_OSERR, "calloc()");
	61	}
	62
	63	vec.iov_base = bytes;
	64	vec.iov_len = bufferLen;
65
66	msg.msg_name = (void *)dst;
67	msg.msg_namelen = dst->sa_len;
68	msg.msg_iov = &vec;
69	msg.msg_iovlen = 1;
70	msg.msg_flags = 0;
71
72	for (unsigned int i = 0; i < numMsg; i++) {
73	ssize_t n;
74	T_QUIET; T_EXPECT_POSIX_SUCCESS(n = sendmsg(s, &msg, 0), "sendmsg()");
75	T_LOG("Sent %ld bytes\n", n);
76	count += 1;
77	}
78
79	// Wait a bit to make sure the packets reach the receiver
80	usleep(100000);
81
82	T_LOG("Sent %ld packet\n", count);
83
84	free(bytes);
85	}
86
87	static void
88	recvPackets_x(int s, unsigned int numMsg, size_t buflen, socklen_t cmsgLen)
89	{
90	struct msghdr_x *msgList;
91	struct sockaddr_in *srcAddrs;
92	struct iovec *vec;
93	char *buffers;
94	char *cmsgBuf;
95
96	T_QUIET; T_ASSERT_NOTNULL(msgList = calloc(numMsg, sizeof(struct msghdr_x)), "msgList calloc()");
97	T_QUIET; T_ASSERT_NOTNULL(srcAddrs = calloc(numMsg, sizeof(struct sockaddr_in)), "srcAddrs calloc()");
98	T_QUIET; T_ASSERT_NOTNULL(vec = calloc(numMsg, sizeof(struct iovec)), "vec calloc()");
99	T_QUIET; T_ASSERT_NOTNULL(buffers = calloc(numMsg, buflen), "buffers calloc()");
100	T_QUIET; T_ASSERT_NOTNULL(cmsgBuf = calloc(numMsg, ALIGN(cmsgLen)), "cmsgBuf calloc()");
101
102	u_int count = 0;
103	while (true) {
104	/*
105	* Wrap around when we've exhausted the list
106	*/
107	if ((count % numMsg) == 0) {
108	for (unsigned int i = 0; i < numMsg; i++) {
109	struct msghdr_x *msg = &msgList[i];
110	msg->msg_name = &srcAddrs[i];
111	msg->msg_namelen = sizeof(srcAddrs[i]);
112	vec[i].iov_base = buffers + (i * buflen);
113	vec[i].iov_len = buflen;
114	msg->msg_iov = &vec[i];
115	msg->msg_iovlen = 1;
116	msg->msg_control = cmsgBuf + (i * ALIGN(cmsgLen));
117	msg->msg_controllen = cmsgLen;
118	msg->msg_flags = 0;
119
120	T_QUIET; T_EXPECT_TRUE((uintptr_t)msg->msg_control % sizeof(uint32_t) == 0, NULL);
121	}
122	}
123
124	ssize_t n = recvmsg_x(s, msgList + (count % numMsg), numMsg - (count % numMsg), 0);
125	if (n < 0) {
126	if (errno == EINTR) {
127	T_LOG("recvmsg_x(): %s", strerror(errno));
128	continue;
129	}
130	if (errno == EWOULDBLOCK) {
131	T_LOG("recvmsg_x(): %s", strerror(errno));
132	break;
133	}
134	T_FAIL("recvmsg_x() failed: %s", strerror(errno));
135	}
136	T_LOG("recvmsg_x returned %ld packets\n", n);
137
138	for (unsigned int i = count; i < count + (u_int)n; i++) {
139	struct msghdr_x *msg = &msgList[i % numMsg];
140
141	T_LOG("Received packet #%d %lu bytes with recvmsg_x(), msg_namelen = %u, msg_controllen = %d -> %d, msg_flags = 0x%x\n",
142	i + 1, msg->msg_datalen, msg->msg_namelen, cmsgLen, msg->msg_controllen, msg->msg_flags);
143
144	struct cmsghdr *cmsg;
145
146	for (cmsg = CMSG_FIRSTHDR(msg); cmsg; cmsg = CMSG_NXTHDR(msg, cmsg)) {
147	T_QUIET; T_EXPECT_TRUE((uintptr_t)cmsg % sizeof(uint32_t) == 0, NULL);
148
149	T_LOG("level = %d, type = %d, length = %d\n", cmsg->cmsg_level, cmsg->cmsg_type, cmsg->cmsg_len);
150	}
151	}
152
153	count += (u_int)n;
154	}
155
156	free(msgList);
157	free(srcAddrs);
158	free(vec);
159	free(buffers);
160	free(cmsgBuf);
161	}
162
163	T_DECL(recvmsg_x_test, "exercise revcmsg_x() with various parameter")
164	{
165	struct sockaddr_in addr = {
166	.sin_len = sizeof(addr),
167	.sin_family = AF_INET,
168	.sin_addr.s_addr = htonl(0x7f000001),
169	.sin_port = 0
170	};
171
172	int recvSocket;
173	T_QUIET; T_EXPECT_POSIX_SUCCESS(recvSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP), "socket()");
174	T_QUIET; T_EXPECT_POSIX_SUCCESS(bind(recvSocket, (const struct sockaddr *)&addr, sizeof(addr)), "bind()");
175
176	socklen_t addrLen = sizeof(addr);
177	T_QUIET; T_EXPECT_POSIX_SUCCESS(getsockname(recvSocket, (struct sockaddr *)&addr, &addrLen), "getsockname()");
178
179	int one = 1;
180	T_QUIET; T_EXPECT_POSIX_SUCCESS(setsockopt(recvSocket, IPPROTO_IP, IP_RECVPKTINFO, (void *)&one, sizeof(one)), "setsockopt(IP_RECVPKTINFO)");
181
182	int flags = fcntl(recvSocket, F_GETFL, 0);
183	T_QUIET; T_EXPECT_POSIX_SUCCESS(fcntl(recvSocket, F_SETFL, flags \| O_NONBLOCK), "fcntl()");
184
185	int sendSocket;
186	T_QUIET; T_EXPECT_POSIX_SUCCESS(sendSocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP), "sendSocket socket()");
187
188	for (int dontTrunc = 0; dontTrunc <= 1; dontTrunc++) {
189	T_QUIET; T_EXPECT_POSIX_SUCCESS(setsockopt(recvSocket, SOL_SOCKET, SO_DONTTRUNC, (void *)&dontTrunc, sizeof(dontTrunc)), "setsockopt(SO_DONTTRUNC)");
190
191	T_LOG("\n================= recvmsg_x() test =================\n");
192	sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
193	recvPackets_x(recvSocket, NMSGS, BUFFERLEN, 50);
194
195	T_LOG("\n================= recvmsg_x() test =================\n");
196	sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
197	recvPackets_x(recvSocket, NMSGS, BUFFERLEN * 2, 50);
198
199	T_LOG("\n================= recvmsg_x() test =================\n");
200	sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
201	recvPackets_x(recvSocket, NMSGS, BUFFERLEN / 2, 50);
202
203	T_LOG("\n================= recvmsg_x() test =================\n");
204	sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
205	recvPackets_x(recvSocket, NMSGS, BUFFERLEN, 10);
206
207	T_LOG("\n================= recvmsg_x() test =================\n");
208	sendPackets(sendSocket, (struct sockaddr *)&addr, NMSGS, BUFFERLEN);
209	recvPackets_x(recvSocket, NMSGS, BUFFERLEN / 2, 10);
210	}
211
212	close(sendSocket);
213	close(recvSocket);
214
215	T_LOG("\n================= PASS =================\n");
216	}