[apple/xnu.git] / tools / tests / darwintests / socket_poll_close_25786011.c

#include <darwintest.h>
#include <poll.h>
#include <sys/socket.h>
#include <unistd.h>

T_DECL(socket_poll_close_25786011, "Tests an invalid poll call to a socket and then calling close.", T_META_LTEPHASE(LTE_POSTINIT))
{
	int my_socket, ret;

	my_socket = socket(PF_LOCAL, SOCK_STREAM, 0);
	T_WITH_ERRNO; T_ASSERT_TRUE(my_socket > 0, "create socket");

	/*
	 * Setup a pollfd that we know will return an error when we try
	 * to create a knote for it. We specify a BSD vnode specific event
	 * for a socket.
	 */
	struct pollfd my_pollfd = {
		.fd = my_socket,
	        .events = POLLEXTEND
	};

	/*
	 * Previously the call to kevent_register() in the kernel from this call
	 * would leak an iocount reference on the fileproc, which would cause any
	 * subsequent calls to close() on the associated fd to block indefinitely.
	 */
	ret = poll(&my_pollfd, 1, 0);
	T_WITH_ERRNO; T_ASSERT_TRUE(ret == 1, "poll returned %d", ret);

	ret = close(my_socket);
	T_ASSERT_POSIX_ZERO(ret, "close on socket with fd %d\n", my_socket);

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