1 files changed, 8 insertions, 0 deletions

diff --git a/include/linux/mem_encrypt.h b/include/linux/mem_encrypt.h
index 570f4fcff13f..1255f09f5e42 100644
--- a/include/linux/mem_encrypt.h
+++ b/include/linux/mem_encrypt.h
@@ -35,6 +35,14 @@ static inline unsigned long sme_get_me_mask(void)
         return sme_me_mask;
 }
 
+/*
+ * The __sme_set() and __sme_clr() macros are useful for adding or removing
+ * the encryption mask from a value (e.g. when dealing with pagetable
+ * entries).
+ */
+#define __sme_set(x)            ((unsigned long)(x) | sme_me_mask)
+#define __sme_clr(x)            ((unsigned long)(x) & ~sme_me_mask)
+
 #endif  /* __ASSEMBLY__ */
 
 #endif  /* __MEM_ENCRYPT_H__ */