xref: /xnu-11215.81.4/tests/socket_poll_close_25786011.c (revision d4514f0bc1d3f944c22d92e68b646ac3fb40d452)
1*d4514f0bSApple OSS Distributions #include <darwintest.h>
2*d4514f0bSApple OSS Distributions #include <poll.h>
3*d4514f0bSApple OSS Distributions #include <sys/socket.h>
4*d4514f0bSApple OSS Distributions #include <unistd.h>
5*d4514f0bSApple OSS Distributions 
6*d4514f0bSApple OSS Distributions T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true));
7*d4514f0bSApple OSS Distributions 
8*d4514f0bSApple OSS Distributions T_DECL(socket_poll_close_25786011, "Tests an invalid poll call to a socket and then calling close.", T_META_LTEPHASE(LTE_POSTINIT), T_META_TAG_VM_PREFERRED)
9*d4514f0bSApple OSS Distributions {
10*d4514f0bSApple OSS Distributions 	int my_socket, ret;
11*d4514f0bSApple OSS Distributions 
12*d4514f0bSApple OSS Distributions 	my_socket = socket(PF_LOCAL, SOCK_STREAM, 0);
13*d4514f0bSApple OSS Distributions 	T_WITH_ERRNO; T_ASSERT_TRUE(my_socket > 0, "create socket");
14*d4514f0bSApple OSS Distributions 
15*d4514f0bSApple OSS Distributions 	/*
16*d4514f0bSApple OSS Distributions 	 * Setup a pollfd that we know will return an error when we try
17*d4514f0bSApple OSS Distributions 	 * to create a knote for it. We specify a BSD vnode specific event
18*d4514f0bSApple OSS Distributions 	 * for a socket.
19*d4514f0bSApple OSS Distributions 	 */
20*d4514f0bSApple OSS Distributions 	struct pollfd my_pollfd = {
21*d4514f0bSApple OSS Distributions 		.fd = my_socket,
22*d4514f0bSApple OSS Distributions 		.events = POLLEXTEND
23*d4514f0bSApple OSS Distributions 	};
24*d4514f0bSApple OSS Distributions 
25*d4514f0bSApple OSS Distributions 	/*
26*d4514f0bSApple OSS Distributions 	 * Previously the call to kevent_register() in the kernel from this call
27*d4514f0bSApple OSS Distributions 	 * would leak an iocount reference on the fileproc, which would cause any
28*d4514f0bSApple OSS Distributions 	 * subsequent calls to close() on the associated fd to block indefinitely.
29*d4514f0bSApple OSS Distributions 	 */
30*d4514f0bSApple OSS Distributions 	ret = poll(&my_pollfd, 1, 0);
31*d4514f0bSApple OSS Distributions 	T_WITH_ERRNO; T_ASSERT_TRUE(ret == 1, "poll returned %d", ret);
32*d4514f0bSApple OSS Distributions 
33*d4514f0bSApple OSS Distributions 	ret = close(my_socket);
34*d4514f0bSApple OSS Distributions 	T_ASSERT_POSIX_ZERO(ret, "close on socket with fd %d\n", my_socket);
35*d4514f0bSApple OSS Distributions 
36*d4514f0bSApple OSS Distributions 	T_PASS("socket_poll_close_25786011 PASSED");
37*d4514f0bSApple OSS Distributions }
38