Replace literal 0 with macro PKEY_UNRESTRICTED where pkey_*() functions
are used in mm selftests for memory protection keys.

Signed-off-by: Yury Khrustalev <yury.khrusta...@arm.com>
Suggested-by: Joey Gouly <joey.go...@arm.com>
Acked-by: Dave Hansen <dave.han...@linux.intel.com>
---
 tools/testing/selftests/mm/mseal_test.c            | 6 +++---
 tools/testing/selftests/mm/pkey-helpers.h          | 3 ++-
 tools/testing/selftests/mm/pkey_sighandler_tests.c | 4 ++--
 tools/testing/selftests/mm/protection_keys.c       | 2 +-
 4 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/tools/testing/selftests/mm/mseal_test.c 
b/tools/testing/selftests/mm/mseal_test.c
index 01675c412b2a..30ea37e8ecf8 100644
--- a/tools/testing/selftests/mm/mseal_test.c
+++ b/tools/testing/selftests/mm/mseal_test.c
@@ -218,7 +218,7 @@ bool seal_support(void)
 bool pkey_supported(void)
 {
 #if defined(__i386__) || defined(__x86_64__) /* arch */
-       int pkey = sys_pkey_alloc(0, 0);
+       int pkey = sys_pkey_alloc(0, PKEY_UNRESTRICTED);
 
        if (pkey > 0)
                return true;
@@ -1671,7 +1671,7 @@ static void test_seal_discard_ro_anon_on_pkey(bool seal)
        setup_single_address_rw(size, &ptr);
        FAIL_TEST_IF_FALSE(ptr != (void *)-1);
 
-       pkey = sys_pkey_alloc(0, 0);
+       pkey = sys_pkey_alloc(0, PKEY_UNRESTRICTED);
        FAIL_TEST_IF_FALSE(pkey > 0);
 
        ret = sys_mprotect_pkey((void *)ptr, size, PROT_READ | PROT_WRITE, 
pkey);
@@ -1683,7 +1683,7 @@ static void test_seal_discard_ro_anon_on_pkey(bool seal)
        }
 
        /* sealing doesn't take effect if PKRU allow write. */
-       set_pkey(pkey, 0);
+       set_pkey(pkey, PKEY_UNRESTRICTED);
        ret = sys_madvise(ptr, size, MADV_DONTNEED);
        FAIL_TEST_IF_FALSE(!ret);
 
diff --git a/tools/testing/selftests/mm/pkey-helpers.h 
b/tools/testing/selftests/mm/pkey-helpers.h
index f7cfe163b0ff..10fa3ca9b05b 100644
--- a/tools/testing/selftests/mm/pkey-helpers.h
+++ b/tools/testing/selftests/mm/pkey-helpers.h
@@ -12,6 +12,7 @@
 #include <stdlib.h>
 #include <ucontext.h>
 #include <sys/mman.h>
+#include <linux/mman.h>
 
 #include "../kselftest.h"
 
@@ -224,7 +225,7 @@ static inline u32 *siginfo_get_pkey_ptr(siginfo_t *si)
 static inline int kernel_has_pkeys(void)
 {
        /* try allocating a key and see if it succeeds */
-       int ret = sys_pkey_alloc(0, 0);
+       int ret = sys_pkey_alloc(0, PKEY_UNRESTRICTED);
        if (ret <= 0) {
                return 0;
        }
diff --git a/tools/testing/selftests/mm/pkey_sighandler_tests.c 
b/tools/testing/selftests/mm/pkey_sighandler_tests.c
index c593a426341c..2015ed7e0928 100644
--- a/tools/testing/selftests/mm/pkey_sighandler_tests.c
+++ b/tools/testing/selftests/mm/pkey_sighandler_tests.c
@@ -314,7 +314,7 @@ static void 
test_sigsegv_handler_with_different_pkey_for_stack(void)
        __write_pkey_reg(pkey_reg);
 
        /* Protect the new stack with MPK 1 */
-       pkey = pkey_alloc(0, 0);
+       pkey = pkey_alloc(0, PKEY_UNRESTRICTED);
        pkey_mprotect(stack, STACK_SIZE, PROT_READ | PROT_WRITE, pkey);
 
        /* Set up alternate signal stack that will use the default MPK */
@@ -487,7 +487,7 @@ static void test_pkru_sigreturn(void)
        __write_pkey_reg(pkey_reg);
 
        /* Protect the stack with MPK 2 */
-       pkey = pkey_alloc(0, 0);
+       pkey = pkey_alloc(0, PKEY_UNRESTRICTED);
        pkey_mprotect(stack, STACK_SIZE, PROT_READ | PROT_WRITE, pkey);
 
        /* Set up alternate signal stack that will use the default MPK */
diff --git a/tools/testing/selftests/mm/protection_keys.c 
b/tools/testing/selftests/mm/protection_keys.c
index 4990f7ab4cb7..cca7435a7bc5 100644
--- a/tools/testing/selftests/mm/protection_keys.c
+++ b/tools/testing/selftests/mm/protection_keys.c
@@ -491,7 +491,7 @@ int sys_pkey_alloc(unsigned long flags, unsigned long 
init_val)
 int alloc_pkey(void)
 {
        int ret;
-       unsigned long init_val = 0x0;
+       unsigned long init_val = PKEY_UNRESTRICTED;
 
        dprintf1("%s()::%d, pkey_reg: 0x%016llx shadow: %016llx\n",
                        __func__, __LINE__, __read_pkey_reg(), shadow_pkey_reg);
-- 
2.39.5


Reply via email to