default_uamor &= ~(0x3ul << pkeyshift(execute_only_key));
        }
 
+       if (unlikely(num_pkey <= 3)) {
+               /*
+                * Insufficient number of keys to support
+                * KUAP/KUEP feature.
+                */
+               disable_kuep = true;
+               disable_kuap = true;
+               WARN(1, "Disabling kernel user protection due to low (%d) max supported keys\n", num_pkey);
+       } else {
+               /*  handle key which is used by kernel for KAUP */
+               reserved_allocation_mask |= (0x1 << 3);
+               /*
+                * Mark access for kup_key in default amr so that
+                * we continue to operate with that AMR in
+                * copy_to/from_user().
+                */
+               default_amr   &= ~(0x3ul << pkeyshift(3));
+               default_iamr  &= ~(0x1ul << pkeyshift(3));
+               default_uamor &= ~(0x3ul << pkeyshift(3));
+       }
+
        /*
         * Allow access for only key 0. And prevent any other modification.
         */
        reserved_allocation_mask |= (0x1 << 1);
        default_uamor &= ~(0x3ul << pkeyshift(1));
 
-       /*  handle key which is used by kernel for KAUP */
-       reserved_allocation_mask |= (0x1 << 3);
-       /*
-        * Mark access for KUAP key in default amr so that
-        * we continue to operate with that AMR in
-        * copy_to/from_user().
-        */
-       default_amr   &= ~(0x3ul << pkeyshift(3));
-       default_iamr  &= ~(0x1ul << pkeyshift(3));
-       default_uamor &= ~(0x3ul << pkeyshift(3));
-
-
        /*
         * Prevent the usage of OS reserved keys. Update UAMOR
         * for those keys. Also mark the rest of the bits in the
 
 unsigned long kernstart_virt_addr __ro_after_init = KERNELBASE;
 EXPORT_SYMBOL_GPL(kernstart_virt_addr);
 
-static bool disable_kuep = !IS_ENABLED(CONFIG_PPC_KUEP);
-static bool disable_kuap = !IS_ENABLED(CONFIG_PPC_KUAP);
+bool disable_kuep = !IS_ENABLED(CONFIG_PPC_KUEP);
+bool disable_kuap = !IS_ENABLED(CONFIG_PPC_KUAP);
 
 static int __init parse_nosmep(char *p)
 {