Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 235266b8 authored by Aneesh Kumar K.V's avatar Aneesh Kumar K.V Committed by Linus Torvalds
Browse files

selftests/vm: move 128TB mmap boundary test to generic directory 

Architectures like PPC64 support mmap hint address based large address
space selection.  This test can be run on those architectures too.  Move
the test from the x86 selftests to selftest/vm so that other
architectures can use it too.

We also add a few new test scenarios in this patch.  We do test a few
boundary conditions before we do a high address mmap.  PPC64 uses the
address limit to validate the address in the fault path.  We had bugs in
this area w.r.t SLB fault handling before we updated the addess limit.

We also touch the allocated space to make sure we don't have any bugs in
the fault handling path.

[akpm@linux-foundation.org: restore tools/testing/selftests/vm/Makefile alpha ordering]
Link: http://lkml.kernel.org/r/20171123165226.32582-1-aneesh.kumar@linux.vnet.ibm.com


Signed-off-by: default avatarAneesh Kumar K.V <aneesh.kumar@linux.vnet.ibm.com>
Cc: "Kirill A . Shutemov" <kirill.shutemov@linux.intel.com>
Cc: Ingo Molnar <mingo@redhat.com>
Cc: Thomas Gleixner <tglx@linutronix.de>
Cc: "H. Peter Anvin" <hpa@zytor.com>
Cc: Shuah Khan <shuah@kernel.org>
Signed-off-by: default avatarAndrew Morton <akpm@linux-foundation.org>
Signed-off-by: default avatarLinus Torvalds <torvalds@linux-foundation.org>
parent e496612c

tools/testing/selftests/vm/Makefile

+3 −2
Original line number Diff line number Diff line
@@ -8,17 +8,18 @@ endif 
CFLAGS = -Wall -I ../../../../usr/include $(EXTRA_CFLAGS)

LDLIBS = -lrt

TEST_GEN_FILES = compaction_test

TEST_GEN_FILES += gup_benchmark

TEST_GEN_FILES += hugepage-mmap

TEST_GEN_FILES += hugepage-shm

TEST_GEN_FILES += map_hugetlb

TEST_GEN_FILES += mlock-random-test

TEST_GEN_FILES += mlock2-tests

TEST_GEN_FILES += on-fault-limit

TEST_GEN_FILES += thuge-gen

TEST_GEN_FILES += transhuge-stress

TEST_GEN_FILES += userfaultfd

TEST_GEN_FILES += mlock-random-test

TEST_GEN_FILES += va_128TBswitch

TEST_GEN_FILES += virtual_address_range

TEST_GEN_FILES += gup_benchmark



TEST_PROGS := run_vmtests

tools/testing/selftests/vm/run_vmtests

+11 −0
Original line number Diff line number Diff line
@@ -177,4 +177,15 @@ else 
	echo "[PASS]"

fi



echo "-----------------------------"

echo "running virtual address 128TB switch test"

echo "-----------------------------"

./va_128TBswitch

if [ $? -ne 0 ]; then

    echo "[FAIL]"

    exitcode=1

else

    echo "[PASS]"

fi



exit $exitcode

tools/testing/selftests/x86/5lvl.c→tools/testing/selftests/vm/va_128TBswitch.c

+297 −0
Original line number Diff line number Diff line 
/*

 *

 * Authors: Kirill A. Shutemov <kirill.shutemov@linux.intel.com>

 * Authors: Aneesh Kumar K.V <aneesh.kumar@linux.vnet.ibm.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License, version 2, as

 * published by the Free Software Foundation.



 * This program is distributed in the hope that it would be useful, but

 * WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

 *

 */



#include <stdio.h>

#include <sys/mman.h>

#include <string.h>



#define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))



#define PAGE_SIZE	4096

#ifdef __powerpc64__

#define PAGE_SIZE	(64 << 10)

/*

 * This will work with 16M and 2M hugepage size

 */

#define HUGETLB_SIZE	(16 << 20)

#else

#define PAGE_SIZE	(4 << 10)

#define HUGETLB_SIZE	(2 << 20)

#endif



/*

 * >= 128TB is the hint addr value we used to select

 * large address space.

 */

#define ADDR_SWITCH_HINT (1UL << 47)

#define LOW_ADDR	((void *) (1UL << 30))

#define HIGH_ADDR	((void *) (1UL << 50))

#define HIGH_ADDR	((void *) (1UL << 48))



struct testcase {

	void *addr;
@@ -17,6 +48,46 @@ struct testcase { 
};



static struct testcase testcases[] = {

	{

		/*

		 * If stack is moved, we could possibly allocate

		 * this at the requested address.

		 */

		.addr = ((void *)(ADDR_SWITCH_HINT - PAGE_SIZE)),

		.size = PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(ADDR_SWITCH_HINT - PAGE_SIZE, PAGE_SIZE)",

		.low_addr_required = 1,

	},

	{

		/*

		 * We should never allocate at the requested address or above it

		 * The len cross the 128TB boundary. Without MAP_FIXED

		 * we will always search in the lower address space.

		 */

		.addr = ((void *)(ADDR_SWITCH_HINT - PAGE_SIZE)),

		.size = 2 * PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(ADDR_SWITCH_HINT - PAGE_SIZE, (2 * PAGE_SIZE))",

		.low_addr_required = 1,

	},

	{

		/*

		 * Exact mapping at 128TB, the area is free we should get that

		 * even without MAP_FIXED.

		 */

		.addr = ((void *)(ADDR_SWITCH_HINT)),

		.size = PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(ADDR_SWITCH_HINT, PAGE_SIZE)",

		.keep_mapped = 1,

	},

	{

		.addr = (void *)(ADDR_SWITCH_HINT),

		.size = 2 * PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED,

		.msg = "mmap(ADDR_SWITCH_HINT, 2 * PAGE_SIZE, MAP_FIXED)",

	},

	{

		.addr = NULL,

		.size = 2 * PAGE_SIZE,
@@ -65,113 +136,162 @@ static struct testcase testcases[] = { 
		.msg = "mmap(-1) again",

	},

	{

		.addr = (void *)((1UL << 47) - PAGE_SIZE),

		.addr = ((void *)(ADDR_SWITCH_HINT - PAGE_SIZE)),

		.size = PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(ADDR_SWITCH_HINT - PAGE_SIZE, PAGE_SIZE)",

		.low_addr_required = 1,

	},

	{

		.addr = (void *)(ADDR_SWITCH_HINT - PAGE_SIZE),

		.size = 2 * PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap((1UL << 47), 2 * PAGE_SIZE)",

		.msg = "mmap(ADDR_SWITCH_HINT - PAGE_SIZE, 2 * PAGE_SIZE)",

		.low_addr_required = 1,

		.keep_mapped = 1,

	},

	{

		.addr = (void *)((1UL << 47) - PAGE_SIZE / 2),

		.addr = (void *)(ADDR_SWITCH_HINT - PAGE_SIZE / 2),

		.size = 2 * PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap((1UL << 47), 2 * PAGE_SIZE / 2)",

		.msg = "mmap(ADDR_SWITCH_HINT - PAGE_SIZE/2 , 2 * PAGE_SIZE)",

		.low_addr_required = 1,

		.keep_mapped = 1,

	},

	{

		.addr = (void *)((1UL << 47) - PAGE_SIZE),

		.addr = ((void *)(ADDR_SWITCH_HINT)),

		.size = PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(ADDR_SWITCH_HINT, PAGE_SIZE)",

	},

	{

		.addr = (void *)(ADDR_SWITCH_HINT),

		.size = 2 * PAGE_SIZE,

		.flags = MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED,

		.msg = "mmap((1UL << 47) - PAGE_SIZE, 2 * PAGE_SIZE, MAP_FIXED)",

		.msg = "mmap(ADDR_SWITCH_HINT, 2 * PAGE_SIZE, MAP_FIXED)",

	},

};



static struct testcase hugetlb_testcases[] = {

	{

		.addr = NULL,

		.size = 2UL << 20,

		.size = HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(NULL, MAP_HUGETLB)",

		.low_addr_required = 1,

	},

	{

		.addr = LOW_ADDR,

		.size = 2UL << 20,

		.size = HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(LOW_ADDR, MAP_HUGETLB)",

		.low_addr_required = 1,

	},

	{

		.addr = HIGH_ADDR,

		.size = 2UL << 20,

		.size = HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(HIGH_ADDR, MAP_HUGETLB)",

		.keep_mapped = 1,

	},

	{

		.addr = HIGH_ADDR,

		.size = 2UL << 20,

		.size = HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(HIGH_ADDR, MAP_HUGETLB) again",

		.keep_mapped = 1,

	},

	{

		.addr = HIGH_ADDR,

		.size = 2UL << 20,

		.size = HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED,

		.msg = "mmap(HIGH_ADDR, MAP_FIXED | MAP_HUGETLB)",

	},

	{

		.addr = (void *) -1,

		.size = 2UL << 20,

		.size = HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(-1, MAP_HUGETLB)",

		.keep_mapped = 1,

	},

	{

		.addr = (void *) -1,

		.size = 2UL << 20,

		.size = HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap(-1, MAP_HUGETLB) again",

	},

	{

		.addr = (void *)((1UL << 47) - PAGE_SIZE),

		.size = 4UL << 20,

		.addr = (void *)(ADDR_SWITCH_HINT - PAGE_SIZE),

		.size = 2 * HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS,

		.msg = "mmap((1UL << 47), 4UL << 20, MAP_HUGETLB)",

		.msg = "mmap(ADDR_SWITCH_HINT - PAGE_SIZE, 2*HUGETLB_SIZE, MAP_HUGETLB)",

		.low_addr_required = 1,

		.keep_mapped = 1,

	},

	{

		.addr = (void *)((1UL << 47) - (2UL << 20)),

		.size = 4UL << 20,

		.addr = (void *)(ADDR_SWITCH_HINT),

		.size = 2 * HUGETLB_SIZE,

		.flags = MAP_HUGETLB | MAP_PRIVATE | MAP_ANONYMOUS | MAP_FIXED,

		.msg = "mmap((1UL << 47) - (2UL << 20), 4UL << 20, MAP_FIXED | MAP_HUGETLB)",

		.msg = "mmap(ADDR_SWITCH_HINT , 2*HUGETLB_SIZE, MAP_FIXED | MAP_HUGETLB)",

	},

};



int main(int argc, char **argv)

static int run_test(struct testcase *test, int count)

{

	int i;

	void *p;

	int i, ret = 0;



	for (i = 0; i < ARRAY_SIZE(testcases); i++) {

		struct testcase *t = testcases + i;

	for (i = 0; i < count; i++) {

		struct testcase *t = test + i;



		p = mmap(t->addr, t->size, PROT_NONE, t->flags, -1, 0);

		p = mmap(t->addr, t->size, PROT_READ | PROT_WRITE, t->flags, -1, 0);



		printf("%s: %p - ", t->msg, p);



		if (p == MAP_FAILED) {

			printf("FAILED\n");

			ret = 1;

			continue;

		}



		if (t->low_addr_required && p >= (void *)(1UL << 47))

		if (t->low_addr_required && p >= (void *)(ADDR_SWITCH_HINT)) {

			printf("FAILED\n");

		else

			ret = 1;

		} else {

			/*

			 * Do a dereference of the address returned so that we catch

			 * bugs in page fault handling

			 */

			memset(p, 0, t->size);

			printf("OK\n");

		}

		if (!t->keep_mapped)

			munmap(p, t->size);

	}



	return ret;

}



static int supported_arch(void)

{

#if defined(__powerpc64__)

	return 1;

#elif defined(__x86_64__)

	return 1;

#else

	return 0;

#endif

}



int main(int argc, char **argv)

{

	int ret;



	if (!supported_arch())

		return 0;



	ret = run_test(testcases, ARRAY_SIZE(testcases));

	if (argc == 2 && !strcmp(argv[1], "--run-hugetlb"))

		ret = run_test(hugetlb_testcases, ARRAY_SIZE(hugetlb_testcases));

	return ret;

}