Differential D4363 Diff 15259 head/tests/sys/aio/aio_test.c

Changeset View

Standalone View

head/tests/sys/aio/aio_test.c

Show First 20 Lines • Show All 716 Lines • ▼ Show 20 Lines	if (nread == -1) {
atf_tc_fail("aio_waitcomplete() failed: %s", strerror(errno));		atf_tc_fail("aio_waitcomplete() failed: %s", strerror(errno));
}		}
atf_tc_fail("aio_read() of too large read size returned: %zd", nread);		atf_tc_fail("aio_read() of too large read size returned: %zd", nread);

finished:		finished:
close(fd);		close(fd);
}		}

		/*
		* This tests for a bug where arriving socket data can wakeup multiple
		* AIO read requests resulting in an uncancellable request.
		*/
		ATF_TC_WITHOUT_HEAD(aio_socket_two_reads);
		ATF_TC_BODY(aio_socket_two_reads, tc)
		{
		struct ioreq {
		struct aiocb iocb;
		char buffer[1024];
		} ioreq[2];
		struct aiocb *iocb;
		unsigned i;
		int s[2];
		char c;

		ATF_REQUIRE_KERNEL_MODULE("aio");
		#if __FreeBSD_version < 1100101
		aft_tc_skip("kernel version %d is too old (%d required)",
		__FreeBSD_version, 1100101);
		#endif

		ATF_REQUIRE(socketpair(PF_UNIX, SOCK_STREAM, 0, s) != -1);

		/* Queue two read requests. */
		memset(&ioreq, 0, sizeof(ioreq));
		for (i = 0; i < nitems(ioreq); i++) {
		ioreq[i].iocb.aio_nbytes = sizeof(ioreq[i].buffer);
		ioreq[i].iocb.aio_fildes = s[0];
		ioreq[i].iocb.aio_buf = ioreq[i].buffer;
		ATF_REQUIRE(aio_read(&ioreq[i].iocb) == 0);
		}

		/* Send a single byte. This should complete one request. */
		c = 0xc3;
		ATF_REQUIRE(write(s[1], &c, sizeof(c)) == 1);

		ATF_REQUIRE(aio_waitcomplete(&iocb, NULL) == 1);

		/* Determine which request completed and verify the data was read. */
		if (iocb == &ioreq[0].iocb)
		i = 0;
		else
		i = 1;
		ATF_REQUIRE(ioreq[i].buffer[0] == c);

		i ^= 1;

		/*
		* Try to cancel the other request. On broken systems this
		* will fail and the process will hang on exit.
		*/
		ATF_REQUIRE(aio_error(&ioreq[i].iocb) == EINPROGRESS);
		ATF_REQUIRE(aio_cancel(s[0], &ioreq[i].iocb) == AIO_CANCELED);

		close(s[1]);
		close(s[0]);
		}

ATF_TP_ADD_TCS(tp)		ATF_TP_ADD_TCS(tp)
{		{

ATF_TP_ADD_TC(tp, aio_file_test);		ATF_TP_ADD_TC(tp, aio_file_test);
ATF_TP_ADD_TC(tp, aio_fifo_test);		ATF_TP_ADD_TC(tp, aio_fifo_test);
ATF_TP_ADD_TC(tp, aio_unix_socketpair_test);		ATF_TP_ADD_TC(tp, aio_unix_socketpair_test);
ATF_TP_ADD_TC(tp, aio_pty_test);		ATF_TP_ADD_TC(tp, aio_pty_test);
ATF_TP_ADD_TC(tp, aio_pipe_test);		ATF_TP_ADD_TC(tp, aio_pipe_test);
ATF_TP_ADD_TC(tp, aio_md_test);		ATF_TP_ADD_TC(tp, aio_md_test);
ATF_TP_ADD_TC(tp, aio_large_read_test);		ATF_TP_ADD_TC(tp, aio_large_read_test);
		ATF_TP_ADD_TC(tp, aio_socket_two_reads);

return (atf_no_error());		return (atf_no_error());
}		}