4 files changed, 95 insertions, 2 deletions

diff --git a/tools/testing/selftests/powerpc/Makefile b/tools/testing/selftests/powerpc/Makefile
index 316194f26ff4..b3dbe9ef1a40 100644
--- a/tools/testing/selftests/powerpc/Makefile
+++ b/tools/testing/selftests/powerpc/Makefile
@@ -13,7 +13,7 @@ CFLAGS := -Wall -O2 -flto -Wall -Werror -DGIT_VERSION='"$(GIT_VERSION)"' -I$(CUR
 
 export CC CFLAGS
 
-TARGETS = pmu copyloops
+TARGETS = pmu copyloops mm
 
 endif
 
diff --git a/tools/testing/selftests/powerpc/copyloops/asm/ppc_asm.h b/tools/testing/selftests/powerpc/copyloops/asm/ppc_asm.h
index ccd9c84c4e3f..d1dc37425510 100644
--- a/tools/testing/selftests/powerpc/copyloops/asm/ppc_asm.h
+++ b/tools/testing/selftests/powerpc/copyloops/asm/ppc_asm.h
@@ -46,12 +46,15 @@
 #define R20 r20
 #define R21 r21
 #define R22 r22
+#define R29 r29
+#define R30 r30
+#define R31 r31
 
 #define STACKFRAMESIZE  256
-#define STK_PARAM(i)    (48 + ((i)-3)*8)
 #define STK_REG(i)      (112 + ((i)-14)*8)
 
 #define _GLOBAL(A) FUNC_START(test_ ## A)
+#define _GLOBAL_TOC(A) _GLOBAL(A)
 
 #define PPC_MTOCRF(A, B)        mtocrf A, B
 
diff --git a/tools/testing/selftests/powerpc/mm/Makefile b/tools/testing/selftests/powerpc/mm/Makefile
new file mode 100644
index 000000000000..357ccbd6bad9
--- /dev/null
+++ b/tools/testing/selftests/powerpc/mm/Makefile
@@ -0,0 +1,18 @@
+noarg:
+        $(MAKE) -C ../
+
+PROGS := hugetlb_vs_thp_test
+
+all: $(PROGS)
+
+$(PROGS): ../harness.c
+
+run_tests: all
+        @-for PROG in $(PROGS); do \
+                ./$$PROG; \
+        done;
+
+clean:
+        rm -f $(PROGS)
+
+.PHONY: all run_tests clean
diff --git a/tools/testing/selftests/powerpc/mm/hugetlb_vs_thp_test.c b/tools/testing/selftests/powerpc/mm/hugetlb_vs_thp_test.c
new file mode 100644
index 000000000000..3d8e5b033e1d
--- /dev/null
+++ b/tools/testing/selftests/powerpc/mm/hugetlb_vs_thp_test.c
@@ -0,0 +1,72 @@
+#include <stdio.h>
+#include <sys/mman.h>
+#include <unistd.h>
+
+#include "utils.h"
+
+/* This must match the huge page & THP size */
+#define SIZE    (16 * 1024 * 1024)
+
+static int test_body(void)
+{
+        void *addr;
+        char *p;
+
+        addr = (void *)0xa0000000;
+
+        p = mmap(addr, SIZE, PROT_READ | PROT_WRITE,
+                 MAP_HUGETLB | MAP_ANONYMOUS | MAP_PRIVATE, -1, 0);
+        if (p != MAP_FAILED) {
+                /*
+                 * Typically the mmap will fail because no huge pages are
+                 * allocated on the system. But if there are huge pages
+                 * allocated the mmap will succeed. That's fine too, we just
+                 * munmap here before continuing.
+                 */
+                munmap(addr, SIZE);
+        }
+
+        p = mmap(addr, SIZE, PROT_READ | PROT_WRITE,
+                 MAP_ANONYMOUS | MAP_PRIVATE, -1, 0);
+        if (p == MAP_FAILED) {
+                printf("Mapping failed @ %p\n", addr);
+                perror("mmap");
+                return 1;
+        }
+
+        /*
+         * Either a user or kernel access is sufficient to trigger the bug.
+         * A kernel access is easier to spot & debug, as it will trigger the
+         * softlockup or RCU stall detectors, and when the system is kicked
+         * into xmon we get a backtrace in the kernel.
+         *
+         * A good option is:
+         *  getcwd(p, SIZE);
+         *
+         * For the purposes of this testcase it's preferable to spin in
+         * userspace, so the harness can kill us if we get stuck. That way we
+         * see a test failure rather than a dead system.
+         */
+        *p = 0xf;
+
+        munmap(addr, SIZE);
+
+        return 0;
+}
+
+static int test_main(void)
+{
+        int i;
+
+        /* 10,000 because it's a "bunch", and completes reasonably quickly */
+        for (i = 0; i < 10000; i++)
+                if (test_body())
+                        return 1;
+
+        return 0;
+}
+
+int main(void)
+{
+        return test_harness(test_main, "hugetlb_vs_thp");
+}