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

selftests: ntsync: Add some tests for NTSYNC_IOC_WAIT_ALL. 



Test basic synchronous functionality of NTSYNC_IOC_WAIT_ALL, and when objects
are considered simultaneously signaled.

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


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

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

+91 −2
Original line number Diff line number Diff line
@@ -73,7 +73,8 @@ static int unlock_mutex(int mutex, __u32 owner, __u32 *count) 
	return ret;

}



static int wait_any(int fd, __u32 count, const int *objs, __u32 owner, __u32 *index)

static int wait_objs(int fd, unsigned long request, __u32 count,

		     const int *objs, __u32 owner, __u32 *index)

{

	struct ntsync_wait_args args = {0};

	struct timespec timeout;
@@ -86,11 +87,21 @@ static int wait_any(int fd, __u32 count, const int *objs, __u32 owner, __u32 *in 
	args.objs = (uintptr_t)objs;

	args.owner = owner;

	args.index = 0xdeadbeef;

	ret = ioctl(fd, NTSYNC_IOC_WAIT_ANY, &args);

	ret = ioctl(fd, request, &args);

	*index = args.index;

	return ret;

}



static int wait_any(int fd, __u32 count, const int *objs, __u32 owner, __u32 *index)

{

	return wait_objs(fd, NTSYNC_IOC_WAIT_ANY, count, objs, owner, index);

}



static int wait_all(int fd, __u32 count, const int *objs, __u32 owner, __u32 *index)

{

	return wait_objs(fd, NTSYNC_IOC_WAIT_ALL, count, objs, owner, index);

}



TEST(semaphore_state)

{

	struct ntsync_sem_args sem_args;
@@ -443,4 +454,82 @@ TEST(test_wait_any) 
	close(fd);

}



TEST(test_wait_all)

{

	struct ntsync_mutex_args mutex_args = {0};

	struct ntsync_sem_args sem_args = {0};

	__u32 owner, index, count;

	int objs[2], fd, ret;



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

	ASSERT_LE(0, fd);



	sem_args.count = 2;

	sem_args.max = 3;

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

	EXPECT_LE(0, objs[0]);



	mutex_args.owner = 0;

	mutex_args.count = 0;

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

	EXPECT_LE(0, objs[1]);



	ret = wait_all(fd, 2, objs, 123, &index);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(0, index);

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

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



	ret = wait_all(fd, 2, objs, 456, &index);

	EXPECT_EQ(-1, ret);

	EXPECT_EQ(ETIMEDOUT, errno);

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

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



	ret = wait_all(fd, 2, objs, 123, &index);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(0, index);

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

	check_mutex_state(objs[1], 2, 123);



	ret = wait_all(fd, 2, objs, 123, &index);

	EXPECT_EQ(-1, ret);

	EXPECT_EQ(ETIMEDOUT, errno);

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

	check_mutex_state(objs[1], 2, 123);



	count = 3;

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

	EXPECT_EQ(0, ret);

	EXPECT_EQ(0, count);



	ret = wait_all(fd, 2, objs, 123, &index);

	EXPECT_EQ(0, ret);

	EXPECT_EQ(0, index);

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

	check_mutex_state(objs[1], 3, 123);



	owner = 123;

	ret = ioctl(objs[1], NTSYNC_IOC_MUTEX_KILL, &owner);

	EXPECT_EQ(0, ret);



	ret = wait_all(fd, 2, objs, 123, &index);

	EXPECT_EQ(-1, ret);

	EXPECT_EQ(EOWNERDEAD, errno);

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

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



	close(objs[1]);



	/* test waiting on the same object twice */

	objs[1] = objs[0];

	ret = wait_all(fd, 2, objs, 123, &index);

	EXPECT_EQ(-1, ret);

	EXPECT_EQ(EINVAL, errno);



	close(objs[0]);



	close(fd);

}



TEST_HARNESS_MAIN