Commit f2327985 authored by Elizabeth Figura's avatar Elizabeth Figura Committed by Greg Kroah-Hartman
Browse files

selftests: ntsync: Add some tests for wakeup signaling with WINESYNC_IOC_WAIT_ANY. 



Test contended "wait-for-any" waits, to make sure that scheduling and wakeup
logic works correctly.

Signed-off-by: default avatarElizabeth Figura <zfigura@codeweavers.com>
Link: https://lore.kernel.org/r/20241213193511.457338-21-zfigura@codeweavers.com


Signed-off-by: default avatarGreg Kroah-Hartman <gregkh@linuxfoundation.org>
parent d168f689

tools/testing/selftests/drivers/ntsync/ntsync.c

+143 −0
Original line number Diff line number Diff line
@@ -532,4 +532,147 @@ TEST(test_wait_all) 
	close(fd);

}



struct wake_args {

	int fd;

	int obj;

};



struct wait_args {

	int fd;

	unsigned long request;

	struct ntsync_wait_args *args;

	int ret;

	int err;

};



static void *wait_thread(void *arg)

{

	struct wait_args *args = arg;



	args->ret = ioctl(args->fd, args->request, args->args);

	args->err = errno;

	return NULL;

}



static __u64 get_abs_timeout(unsigned int ms)

{

	struct timespec timeout;

	clock_gettime(CLOCK_MONOTONIC, &timeout);

	return (timeout.tv_sec * 1000000000) + timeout.tv_nsec + (ms * 1000000);

}



static int wait_for_thread(pthread_t thread, unsigned int ms)

{

	struct timespec timeout;



	clock_gettime(CLOCK_REALTIME, &timeout);

	timeout.tv_nsec += ms * 1000000;

	timeout.tv_sec += (timeout.tv_nsec / 1000000000);

	timeout.tv_nsec %= 1000000000;

	return pthread_timedjoin_np(thread, NULL, &timeout);

}



TEST(wake_any)

{

	struct ntsync_mutex_args mutex_args = {0};

	struct ntsync_wait_args wait_args = {0};

	struct ntsync_sem_args sem_args = {0};

	struct wait_args thread_args;

	int objs[2], fd, ret;

	__u32 count, index;

	pthread_t thread;



	fd = open("/dev/ntsync", O_CLOEXEC | O_RDONLY);

	ASSERT_LE(0, fd);



	sem_args.count = 0;

	sem_args.max = 3;

	objs[0] = ioctl(fd, NTSYNC_IOC_CREATE_SEM, &sem_args);

	EXPECT_LE(0, objs[0]);



	mutex_args.owner = 123;

	mutex_args.count = 1;

	objs[1] = ioctl(fd, NTSYNC_IOC_CREATE_MUTEX, &mutex_args);

	EXPECT_LE(0, objs[1]);



	/* test waking the semaphore */



	wait_args.timeout = get_abs_timeout(1000);

	wait_args.objs = (uintptr_t)objs;

	wait_args.count = 2;

	wait_args.owner = 456;

	wait_args.index = 0xdeadbeef;

	thread_args.fd = fd;

	thread_args.args = &wait_args;

	thread_args.request = NTSYNC_IOC_WAIT_ANY;

	ret = pthread_create(&thread, NULL, wait_thread, &thread_args);

	EXPECT_EQ(0, ret);



	ret = wait_for_thread(thread, 100);

	EXPECT_EQ(ETIMEDOUT, ret);



	count = 1;

	ret = release_sem(objs[0], &count);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(0, count);

	check_sem_state(objs[0], 0, 3);



	ret = wait_for_thread(thread, 100);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(0, thread_args.ret);

	EXPECT_EQ(0, wait_args.index);



	/* test waking the mutex */



	/* first grab it again for owner 123 */

	ret = wait_any(fd, 1, &objs[1], 123, &index);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(0, index);



	wait_args.timeout = get_abs_timeout(1000);

	wait_args.owner = 456;

	ret = pthread_create(&thread, NULL, wait_thread, &thread_args);

	EXPECT_EQ(0, ret);



	ret = wait_for_thread(thread, 100);

	EXPECT_EQ(ETIMEDOUT, ret);



	ret = unlock_mutex(objs[1], 123, &count);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(2, count);



	ret = pthread_tryjoin_np(thread, NULL);

	EXPECT_EQ(EBUSY, ret);



	ret = unlock_mutex(objs[1], 123, &count);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(1, mutex_args.count);

	check_mutex_state(objs[1], 1, 456);



	ret = wait_for_thread(thread, 100);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(0, thread_args.ret);

	EXPECT_EQ(1, wait_args.index);



	/* delete an object while it's being waited on */



	wait_args.timeout = get_abs_timeout(200);

	wait_args.owner = 123;

	ret = pthread_create(&thread, NULL, wait_thread, &thread_args);

	EXPECT_EQ(0, ret);



	ret = wait_for_thread(thread, 100);

	EXPECT_EQ(ETIMEDOUT, ret);



	close(objs[0]);

	close(objs[1]);



	ret = wait_for_thread(thread, 200);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(-1, thread_args.ret);

	EXPECT_EQ(ETIMEDOUT, thread_args.err);



	close(fd);

}



TEST_HARNESS_MAIN