diff mbox series

[17/29] selftests/mm: Drop global mem_fd in uffd tests

Message ID	20230330160806.3107497-1-peterx@redhat.com
State	New
Headers	Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 2620:137:e000::1:20 as permitted sender) client-ip=2620:137:e000::1:20; From: Peter Xu <peterx@redhat.com> To: linux-kernel@vger.kernel.org, linux-mm@kvack.org Cc: Mike Kravetz <mike.kravetz@oracle.com>, Mike Rapoport <rppt@linux.vnet.ibm.com>, Andrea Arcangeli <aarcange@redhat.com>, peterx@redhat.com, Leonardo Bras Soares Passos <lsoaresp@redhat.com>, Andrew Morton <akpm@linux-foundation.org>, David Hildenbrand <david@redhat.com>, Axel Rasmussen <axelrasmussen@google.com>, Nadav Amit <nadav.amit@gmail.com> Subject: [PATCH 17/29] selftests/mm: Drop global mem_fd in uffd tests Date: Thu, 30 Mar 2023 12:08:06 -0400 Message-Id: <20230330160806.3107497-1-peterx@redhat.com> In-Reply-To: <20230330155707.3106228-1-peterx@redhat.com> References: <20230330155707.3106228-1-peterx@redhat.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit Precedence: bulk
Series	selftests/mm: Split / Refactor userfault test \| [00/29] selftests/mm: Split / Refactor userfault test [01/29] Revert "userfaultfd: don't fail on unrecognized features" [02/29] selftests/mm: Update .gitignore with two missing tests [03/29] selftests/mm: Dump a summary in run_vmtests.sh [04/29] selftests/mm: Merge util.h into vm_util.h [05/29] selftests/mm: Use TEST_GEN_PROGS where proper [06/29] selftests/mm: Link vm_util.c always [07/29] selftests/mm: Merge default_huge_page_size() into one [08/29] selftests/mm: Use PM_* macros in vm_utils.h [09/29] selftests/mm: Reuse pagemap_get_entry() in vm_util.h [10/29] selftests/mm: Test UFFDIO_ZEROPAGE only when !hugetlb [11/29] selftests/mm: Drop test_uffdio_zeropage_eexist [12/29] selftests/mm: Create uffd-common.[ch] [13/29] selftests/mm: Split uffd tests into uffd-stress and uffd-unit-tests [14/29] selftests/mm: uffd_[un]register() [15/29] selftests/mm: uffd_open_{dev\|sys}() [16/29] selftests/mm: UFFDIO_API test [17/29] selftests/mm: Drop global mem_fd in uffd tests [18/29] selftests/mm: Drop global hpage_size in uffd tests [19/29] selftests/mm: Let uffd_handle_page_fault() takes wp parameter [20/29] selftests/mm: Allow allocate_area() to fail properly [21/29] selftests/mm: Add framework for uffd-unit-test [22/29] selftests/mm: Move uffd pagemap test to unit test [23/29] selftests/mm: Move uffd minor test to unit test [24/29] selftests/mm: Move uffd sig/events tests into uffd unit tests [25/29] selftests/mm: Move zeropage test into uffd unit tests [26/29] selftests/mm: Workaround no way to detect uffd-minor + wp [27/29] selftests/mm: Allow uffd test to skip properly with no privilege [28/29] selftests/mm: Drop sys/dev test in uffd-stress test [29/29] selftests/mm: Add shmem-private test to uffd-stress

Commit Message

Peter Xu March 30, 2023, 4:08 p.m. UTC

  Drop it by creating the memfd dynamically in the tests.

Signed-off-by: Peter Xu <peterx@redhat.com>
---
 tools/testing/selftests/mm/uffd-common.c | 28 +++++++++++++++++++++++-
 tools/testing/selftests/mm/uffd-common.h |  2 +-
 tools/testing/selftests/mm/uffd-stress.c | 15 -------------
 3 files changed, 28 insertions(+), 17 deletions(-)

Comments

Mike Rapoport April 11, 2023, 10:39 a.m. UTC | #1

On Thu, Mar 30, 2023 at 12:08:06PM -0400, Peter Xu wrote:
> Drop it by creating the memfd dynamically in the tests.
> 
> Signed-off-by: Peter Xu <peterx@redhat.com>

Reviewed-by: Mike Rapoport (IBM) <rppt@kernel.org>

> ---
>  tools/testing/selftests/mm/uffd-common.c | 28 +++++++++++++++++++++++-
>  tools/testing/selftests/mm/uffd-common.h |  2 +-
>  tools/testing/selftests/mm/uffd-stress.c | 15 -------------
>  3 files changed, 28 insertions(+), 17 deletions(-)
> 
> diff --git a/tools/testing/selftests/mm/uffd-common.c b/tools/testing/selftests/mm/uffd-common.c
> index 3a9b5c1aca9d..a2b6e4957d0f 100644
> --- a/tools/testing/selftests/mm/uffd-common.c
> +++ b/tools/testing/selftests/mm/uffd-common.c
> @@ -12,12 +12,32 @@
>  volatile bool test_uffdio_copy_eexist = true;
>  unsigned long nr_cpus, nr_pages, nr_pages_per_cpu, page_size, hpage_size;
>  char *area_src, *area_src_alias, *area_dst, *area_dst_alias, *area_remap;
> -int mem_fd, uffd = -1, uffd_flags, finished, *pipefd, test_type;
> +int uffd = -1, uffd_flags, finished, *pipefd, test_type;
>  bool map_shared, test_collapse, test_dev_userfaultfd;
>  bool test_uffdio_wp = true, test_uffdio_minor = false;
>  unsigned long long *count_verify;
>  uffd_test_ops_t *uffd_test_ops;
> 
> +static int uffd_mem_fd_create(off_t mem_size, bool hugetlb)
> +{
> +	unsigned int memfd_flags = 0;
> +	int mem_fd;
> +
> +	if (hugetlb)
> +		memfd_flags = MFD_HUGETLB;
> +	mem_fd = memfd_create("uffd-test", memfd_flags);
> +	if (mem_fd < 0)
> +		err("memfd_create");
> +	if (ftruncate(mem_fd, mem_size))
> +		err("ftruncate");
> +	if (fallocate(mem_fd,
> +		      FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE, 0,
> +		      mem_size))
> +		err("fallocate");
> +
> +	return mem_fd;
> +}
> +
>  static void anon_release_pages(char *rel_area)
>  {
>  	if (madvise(rel_area, nr_pages * page_size, MADV_DONTNEED))
> @@ -51,6 +71,7 @@ static void hugetlb_allocate_area(void **alloc_area, bool is_src)
>  	off_t offset = is_src ? 0 : size;
>  	void *area_alias = NULL;
>  	char **alloc_area_alias;
> +	int mem_fd = uffd_mem_fd_create(size * 2, true);
> 
>  	*alloc_area = mmap(NULL, size, PROT_READ | PROT_WRITE,
>  			   (map_shared ? MAP_SHARED : MAP_PRIVATE) |
> @@ -73,6 +94,8 @@ static void hugetlb_allocate_area(void **alloc_area, bool is_src)
>  	}
>  	if (area_alias)
>  		*alloc_area_alias = area_alias;
> +
> +	close(mem_fd);
>  }
> 
>  static void hugetlb_alias_mapping(__u64 *start, size_t len, unsigned long offset)
> @@ -95,6 +118,7 @@ static void shmem_allocate_area(void **alloc_area, bool is_src)
>  	size_t bytes = nr_pages * page_size;
>  	unsigned long offset = is_src ? 0 : bytes;
>  	char *p = NULL, *p_alias = NULL;
> +	int mem_fd = uffd_mem_fd_create(bytes * 2, false);
> 
>  	if (test_collapse) {
>  		p = BASE_PMD_ADDR;
> @@ -124,6 +148,8 @@ static void shmem_allocate_area(void **alloc_area, bool is_src)
>  		area_src_alias = area_alias;
>  	else
>  		area_dst_alias = area_alias;
> +
> +	close(mem_fd);
>  }
> 
>  static void shmem_alias_mapping(__u64 *start, size_t len, unsigned long offset)
> diff --git a/tools/testing/selftests/mm/uffd-common.h b/tools/testing/selftests/mm/uffd-common.h
> index 11f770391bd9..0dfab7057295 100644
> --- a/tools/testing/selftests/mm/uffd-common.h
> +++ b/tools/testing/selftests/mm/uffd-common.h
> @@ -87,7 +87,7 @@ typedef struct uffd_test_ops uffd_test_ops_t;
> 
>  extern unsigned long nr_cpus, nr_pages, nr_pages_per_cpu, page_size, hpage_size;
>  extern char *area_src, *area_src_alias, *area_dst, *area_dst_alias, *area_remap;
> -extern int mem_fd, uffd, uffd_flags, finished, *pipefd, test_type;
> +extern int uffd, uffd_flags, finished, *pipefd, test_type;
>  extern bool map_shared, test_collapse, test_dev_userfaultfd;
>  extern bool test_uffdio_wp, test_uffdio_minor;
>  extern unsigned long long *count_verify;
> diff --git a/tools/testing/selftests/mm/uffd-stress.c b/tools/testing/selftests/mm/uffd-stress.c
> index e6d39a755082..4eca1a0276c2 100644
> --- a/tools/testing/selftests/mm/uffd-stress.c
> +++ b/tools/testing/selftests/mm/uffd-stress.c
> @@ -1090,21 +1090,6 @@ int main(int argc, char **argv)
>  	}
>  	nr_pages = nr_pages_per_cpu * nr_cpus;
> 
> -	if (test_type == TEST_SHMEM || test_type == TEST_HUGETLB) {
> -		unsigned int memfd_flags = 0;
> -
> -		if (test_type == TEST_HUGETLB)
> -			memfd_flags = MFD_HUGETLB;
> -		mem_fd = memfd_create(argv[0], memfd_flags);
> -		if (mem_fd < 0)
> -			err("memfd_create");
> -		if (ftruncate(mem_fd, nr_pages * page_size * 2))
> -			err("ftruncate");
> -		if (fallocate(mem_fd,
> -			      FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE, 0,
> -			      nr_pages * page_size * 2))
> -			err("fallocate");
> -	}
>  	printf("nr_pages: %lu, nr_pages_per_cpu: %lu\n",
>  	       nr_pages, nr_pages_per_cpu);
>  	return userfaultfd_stress();
> -- 
> 2.39.1
>

diff mbox series

Patch

diff --git a/tools/testing/selftests/mm/uffd-common.c b/tools/testing/selftests/mm/uffd-common.c
index 3a9b5c1aca9d..a2b6e4957d0f 100644
--- a/tools/testing/selftests/mm/uffd-common.c
+++ b/tools/testing/selftests/mm/uffd-common.c
@@ -12,12 +12,32 @@ 
 volatile bool test_uffdio_copy_eexist = true;
 unsigned long nr_cpus, nr_pages, nr_pages_per_cpu, page_size, hpage_size;
 char *area_src, *area_src_alias, *area_dst, *area_dst_alias, *area_remap;
-int mem_fd, uffd = -1, uffd_flags, finished, *pipefd, test_type;
+int uffd = -1, uffd_flags, finished, *pipefd, test_type;
 bool map_shared, test_collapse, test_dev_userfaultfd;
 bool test_uffdio_wp = true, test_uffdio_minor = false;
 unsigned long long *count_verify;
 uffd_test_ops_t *uffd_test_ops;
 
+static int uffd_mem_fd_create(off_t mem_size, bool hugetlb)
+{
+	unsigned int memfd_flags = 0;
+	int mem_fd;
+
+	if (hugetlb)
+		memfd_flags = MFD_HUGETLB;
+	mem_fd = memfd_create("uffd-test", memfd_flags);
+	if (mem_fd < 0)
+		err("memfd_create");
+	if (ftruncate(mem_fd, mem_size))
+		err("ftruncate");
+	if (fallocate(mem_fd,
+		      FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE, 0,
+		      mem_size))
+		err("fallocate");
+
+	return mem_fd;
+}
+
 static void anon_release_pages(char *rel_area)
 {
 	if (madvise(rel_area, nr_pages * page_size, MADV_DONTNEED))
@@ -51,6 +71,7 @@  static void hugetlb_allocate_area(void **alloc_area, bool is_src)
 	off_t offset = is_src ? 0 : size;
 	void *area_alias = NULL;
 	char **alloc_area_alias;
+	int mem_fd = uffd_mem_fd_create(size * 2, true);
 
 	*alloc_area = mmap(NULL, size, PROT_READ | PROT_WRITE,
 			   (map_shared ? MAP_SHARED : MAP_PRIVATE) |
@@ -73,6 +94,8 @@  static void hugetlb_allocate_area(void **alloc_area, bool is_src)
 	}
 	if (area_alias)
 		*alloc_area_alias = area_alias;
+
+	close(mem_fd);
 }
 
 static void hugetlb_alias_mapping(__u64 *start, size_t len, unsigned long offset)
@@ -95,6 +118,7 @@  static void shmem_allocate_area(void **alloc_area, bool is_src)
 	size_t bytes = nr_pages * page_size;
 	unsigned long offset = is_src ? 0 : bytes;
 	char *p = NULL, *p_alias = NULL;
+	int mem_fd = uffd_mem_fd_create(bytes * 2, false);
 
 	if (test_collapse) {
 		p = BASE_PMD_ADDR;
@@ -124,6 +148,8 @@  static void shmem_allocate_area(void **alloc_area, bool is_src)
 		area_src_alias = area_alias;
 	else
 		area_dst_alias = area_alias;
+
+	close(mem_fd);
 }
 
 static void shmem_alias_mapping(__u64 *start, size_t len, unsigned long offset)
diff --git a/tools/testing/selftests/mm/uffd-common.h b/tools/testing/selftests/mm/uffd-common.h
index 11f770391bd9..0dfab7057295 100644
--- a/tools/testing/selftests/mm/uffd-common.h
+++ b/tools/testing/selftests/mm/uffd-common.h
@@ -87,7 +87,7 @@  typedef struct uffd_test_ops uffd_test_ops_t;
 
 extern unsigned long nr_cpus, nr_pages, nr_pages_per_cpu, page_size, hpage_size;
 extern char *area_src, *area_src_alias, *area_dst, *area_dst_alias, *area_remap;
-extern int mem_fd, uffd, uffd_flags, finished, *pipefd, test_type;
+extern int uffd, uffd_flags, finished, *pipefd, test_type;
 extern bool map_shared, test_collapse, test_dev_userfaultfd;
 extern bool test_uffdio_wp, test_uffdio_minor;
 extern unsigned long long *count_verify;
diff --git a/tools/testing/selftests/mm/uffd-stress.c b/tools/testing/selftests/mm/uffd-stress.c
index e6d39a755082..4eca1a0276c2 100644
--- a/tools/testing/selftests/mm/uffd-stress.c
+++ b/tools/testing/selftests/mm/uffd-stress.c
@@ -1090,21 +1090,6 @@  int main(int argc, char **argv)
 	}
 	nr_pages = nr_pages_per_cpu * nr_cpus;
 
-	if (test_type == TEST_SHMEM || test_type == TEST_HUGETLB) {
-		unsigned int memfd_flags = 0;
-
-		if (test_type == TEST_HUGETLB)
-			memfd_flags = MFD_HUGETLB;
-		mem_fd = memfd_create(argv[0], memfd_flags);
-		if (mem_fd < 0)
-			err("memfd_create");
-		if (ftruncate(mem_fd, nr_pages * page_size * 2))
-			err("ftruncate");
-		if (fallocate(mem_fd,
-			      FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE, 0,
-			      nr_pages * page_size * 2))
-			err("fallocate");
-	}
 	printf("nr_pages: %lu, nr_pages_per_cpu: %lu\n",
 	       nr_pages, nr_pages_per_cpu);
 	return userfaultfd_stress();