[RFC PATCH v3 2/2] selftests/vm: add mempinfd test

[Zhou Wang]

This test gets a fd from new mempinfd syscall and creates multiple threads
to do pin/unpin memory.

Signed-off-by: Zhou Wang <wangzhou1@xxxxxxxxxxxxx>
Suggested-by: Barry Song <song.bao.hua@xxxxxxxxxxxxx>
---
 tools/testing/selftests/vm/Makefile   |   1 +
 tools/testing/selftests/vm/mempinfd.c | 131 ++++++++++++++++++++++++++++++++++
 2 files changed, 132 insertions(+)
 create mode 100644 tools/testing/selftests/vm/mempinfd.c

diff --git a/tools/testing/selftests/vm/Makefile b/tools/testing/selftests/vm/Makefile
index d42115e..2d5b509 100644
--- a/tools/testing/selftests/vm/Makefile
+++ b/tools/testing/selftests/vm/Makefile
@@ -42,6 +42,7 @@ TEST_GEN_FILES += on-fault-limit
 TEST_GEN_FILES += thuge-gen
 TEST_GEN_FILES += transhuge-stress
 TEST_GEN_FILES += userfaultfd
+TEST_GEN_FILES += mempinfd
 
 ifeq ($(MACHINE),x86_64)
 CAN_BUILD_I386 := $(shell ./../x86/check_cc.sh $(CC) ../x86/trivial_32bit_program.c -m32)
diff --git a/tools/testing/selftests/vm/mempinfd.c b/tools/testing/selftests/vm/mempinfd.c
new file mode 100644
index 0000000..51d5cbf
--- /dev/null
+++ b/tools/testing/selftests/vm/mempinfd.c
@@ -0,0 +1,131 @@
+// SPDX-License-Identifier: GPL-2.0-only
+/* Copyright (c) 2021 HiSilicon Limited. */
+#define _GNU_SOURCE
+#include <linux/mempinfd.h>
+#include <malloc.h>
+#include <pthread.h>
+#include <stdio.h>
+#include <sys/ioctl.h>
+#include <sys/syscall.h>
+
+#include "../kselftest.h"
+
+#ifdef __NR_mempinfd
+
+#define DEF_PIN_SIZE		(4096 * 1024)
+#define MAX_THREAD_NUM		20
+#define DEF_THREAD_NUM		1
+#define DEF_TIMES		10000
+
+struct test_data {
+	int fd;
+	unsigned long mem_size;
+	unsigned long times;
+};
+
+static void *do_pin_test(void *data)
+{
+	struct mem_pin_address addr;
+	struct test_data *d = data;
+	unsigned long times;
+	int ret, fd;
+	int i = 0;
+	void *p;
+
+	p = malloc(d->mem_size);
+	if (!p) {
+		fprintf(stderr, "fail to allocate memory\n");
+		return NULL;
+	}
+
+	addr.addr = (__u64)p;
+	addr.size = d->mem_size;
+	times = d->mem_size;
+	fd = d->fd;
+
+	while (i++ < times) {
+		ret = ioctl(fd, MEM_CMD_PIN, &addr);
+		if (ret) {
+			fprintf(stderr, "fail to pin memory\n");
+			return NULL;
+		}
+
+		usleep(1000);
+
+		ret = ioctl(fd, MEM_CMD_UNPIN, &addr);
+		if (ret) {
+			fprintf(stderr, "fail to unpin memory\n");
+			return NULL;
+		}
+	}
+
+	free(p);
+
+	return NULL;
+}
+
+int main(int argc, char **argv)
+{
+	unsigned long thread_num = DEF_THREAD_NUM;
+	unsigned long mem_size = DEF_PIN_SIZE;
+	unsigned long times = DEF_TIMES;
+	pthread_t threads[MAX_THREAD_NUM];
+	struct test_data data;
+	int fd, opt, i;
+	int ret = 0;
+
+	while ((opt = getopt(argc, argv, "s:n:t:")) != -1) {
+		switch (opt) {
+		case 's':
+			mem_size = atoi(optarg);
+			break;
+		case 'n':
+			thread_num = atoi(optarg);
+			if (thread_num > MAX_THREAD_NUM)
+				return -1;
+			break;
+		case 't':
+			times = atoi(optarg);
+			break;
+		default:
+			return -1;
+		}
+	}
+
+	fd = syscall(__NR_mempinfd);
+	if (fd < 0) {
+		fprintf(stderr, "mempinfd syscall not available in this kernel\n");
+		return -1;
+	}
+
+	data.fd = fd;
+	data.mem_size = mem_size;
+	data.times = times;
+
+	for (i = 0; i < thread_num; i++) {
+		ret = pthread_create(&threads[i], NULL, do_pin_test, &data);
+		if (ret) {
+			fprintf(stderr, "fail to create thread %d: %d\n",
+				i, -errno);
+			return -1;
+		}
+	}
+
+	for (i = 0; i < thread_num; i++)
+		pthread_join(threads[i], NULL);
+
+	close(fd);
+
+	return 0;
+}
+
+#else /* __NR_mempinfd */
+
+#warning "missing __NR_mempinfd definition"
+int main(void)
+{
+	printf("skip: Skipping mempinfd test (missing __NR_mempinfd)\n");
+	return KSFT_SKIP;
+}
+
+#endif /* __NR_mempinfd */
-- 
2.8.1