1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef _ASM_X86_PKRU_H 3 #define _ASM_X86_PKRU_H 4 5 #include <asm/cpufeature.h> 6 7 #define PKRU_AD_BIT 0x1u 8 #define PKRU_WD_BIT 0x2u 9 #define PKRU_BITS_PER_PKEY 2 10 11 #ifdef CONFIG_X86_INTEL_MEMORY_PROTECTION_KEYS 12 extern u32 init_pkru_value; 13 #define pkru_get_init_value() READ_ONCE(init_pkru_value) 14 #else 15 #define init_pkru_value 0 16 #define pkru_get_init_value() 0 17 #endif 18 __pkru_allows_read(u32 pkru,u16 pkey)19 static inline bool __pkru_allows_read(u32 pkru, u16 pkey) 20 { 21 int pkru_pkey_bits = pkey * PKRU_BITS_PER_PKEY; 22 return !(pkru & (PKRU_AD_BIT << pkru_pkey_bits)); 23 } 24 __pkru_allows_write(u32 pkru,u16 pkey)25 static inline bool __pkru_allows_write(u32 pkru, u16 pkey) 26 { 27 int pkru_pkey_bits = pkey * PKRU_BITS_PER_PKEY; 28 /* 29 * Access-disable disables writes too so we need to check 30 * both bits here. 31 */ 32 return !(pkru & ((PKRU_AD_BIT|PKRU_WD_BIT) << pkru_pkey_bits)); 33 } 34 read_pkru(void)35 static inline u32 read_pkru(void) 36 { 37 if (cpu_feature_enabled(X86_FEATURE_OSPKE)) 38 return rdpkru(); 39 return 0; 40 } 41 write_pkru(u32 pkru)42 static inline void write_pkru(u32 pkru) 43 { 44 if (!cpu_feature_enabled(X86_FEATURE_OSPKE)) 45 return; 46 /* 47 * WRPKRU is relatively expensive compared to RDPKRU. 48 * Avoid WRPKRU when it would not change the value. 49 */ 50 if (pkru != rdpkru()) 51 wrpkru(pkru); 52 } 53 pkru_write_default(void)54 static inline void pkru_write_default(void) 55 { 56 if (!cpu_feature_enabled(X86_FEATURE_OSPKE)) 57 return; 58 59 wrpkru(pkru_get_init_value()); 60 } 61 62 #endif 63