1 /* SPDX-License-Identifier: GPL-2.0-only */
2 /*
3  * AMD Memory Encryption Support
4  *
5  * Copyright (C) 2016 Advanced Micro Devices, Inc.
6  *
7  * Author: Tom Lendacky <thomas.lendacky@amd.com>
8  */
9 
10 #ifndef __X86_MEM_ENCRYPT_H__
11 #define __X86_MEM_ENCRYPT_H__
12 
13 #ifndef __ASSEMBLY__
14 
15 #include <linux/init.h>
16 #include <linux/cc_platform.h>
17 
18 #include <asm/asm.h>
19 struct boot_params;
20 
21 #ifdef CONFIG_X86_MEM_ENCRYPT
22 void __init mem_encrypt_init(void);
23 void __init mem_encrypt_setup_arch(void);
24 #else
mem_encrypt_init(void)25 static inline void mem_encrypt_init(void) { }
mem_encrypt_setup_arch(void)26 static inline void __init mem_encrypt_setup_arch(void) { }
27 #endif
28 
29 #ifdef CONFIG_AMD_MEM_ENCRYPT
30 
31 extern u64 sme_me_mask;
32 extern u64 sev_status;
33 
34 void sme_encrypt_execute(unsigned long encrypted_kernel_vaddr,
35 			 unsigned long decrypted_kernel_vaddr,
36 			 unsigned long kernel_len,
37 			 unsigned long encryption_wa,
38 			 unsigned long encryption_pgd);
39 
40 void __init sme_early_encrypt(resource_size_t paddr,
41 			      unsigned long size);
42 void __init sme_early_decrypt(resource_size_t paddr,
43 			      unsigned long size);
44 
45 void __init sme_map_bootdata(char *real_mode_data);
46 void __init sme_unmap_bootdata(char *real_mode_data);
47 
48 void __init sme_early_init(void);
49 
50 void sme_encrypt_kernel(struct boot_params *bp);
51 void sme_enable(struct boot_params *bp);
52 
53 int __init early_set_memory_decrypted(unsigned long vaddr, unsigned long size);
54 int __init early_set_memory_encrypted(unsigned long vaddr, unsigned long size);
55 void __init early_set_mem_enc_dec_hypercall(unsigned long vaddr,
56 					    unsigned long size, bool enc);
57 
58 void __init mem_encrypt_free_decrypted_mem(void);
59 
60 void __init sev_es_init_vc_handling(void);
61 
sme_get_me_mask(void)62 static inline u64 sme_get_me_mask(void)
63 {
64 	return RIP_REL_REF(sme_me_mask);
65 }
66 
67 #define __bss_decrypted __section(".bss..decrypted")
68 
69 #else	/* !CONFIG_AMD_MEM_ENCRYPT */
70 
71 #define sme_me_mask	0ULL
72 #define sev_status	0ULL
73 
sme_early_encrypt(resource_size_t paddr,unsigned long size)74 static inline void __init sme_early_encrypt(resource_size_t paddr,
75 					    unsigned long size) { }
sme_early_decrypt(resource_size_t paddr,unsigned long size)76 static inline void __init sme_early_decrypt(resource_size_t paddr,
77 					    unsigned long size) { }
78 
sme_map_bootdata(char * real_mode_data)79 static inline void __init sme_map_bootdata(char *real_mode_data) { }
sme_unmap_bootdata(char * real_mode_data)80 static inline void __init sme_unmap_bootdata(char *real_mode_data) { }
81 
sme_early_init(void)82 static inline void __init sme_early_init(void) { }
83 
sme_encrypt_kernel(struct boot_params * bp)84 static inline void sme_encrypt_kernel(struct boot_params *bp) { }
sme_enable(struct boot_params * bp)85 static inline void sme_enable(struct boot_params *bp) { }
86 
sev_es_init_vc_handling(void)87 static inline void sev_es_init_vc_handling(void) { }
88 
89 static inline int __init
early_set_memory_decrypted(unsigned long vaddr,unsigned long size)90 early_set_memory_decrypted(unsigned long vaddr, unsigned long size) { return 0; }
91 static inline int __init
early_set_memory_encrypted(unsigned long vaddr,unsigned long size)92 early_set_memory_encrypted(unsigned long vaddr, unsigned long size) { return 0; }
93 static inline void __init
early_set_mem_enc_dec_hypercall(unsigned long vaddr,unsigned long size,bool enc)94 early_set_mem_enc_dec_hypercall(unsigned long vaddr, unsigned long size, bool enc) {}
95 
mem_encrypt_free_decrypted_mem(void)96 static inline void mem_encrypt_free_decrypted_mem(void) { }
97 
sme_get_me_mask(void)98 static inline u64 sme_get_me_mask(void) { return 0; }
99 
100 #define __bss_decrypted
101 
102 #endif	/* CONFIG_AMD_MEM_ENCRYPT */
103 
104 void add_encrypt_protection_map(void);
105 
106 /*
107  * The __sme_pa() and __sme_pa_nodebug() macros are meant for use when
108  * writing to or comparing values from the cr3 register.  Having the
109  * encryption mask set in cr3 enables the PGD entry to be encrypted and
110  * avoid special case handling of PGD allocations.
111  */
112 #define __sme_pa(x)		(__pa(x) | sme_me_mask)
113 #define __sme_pa_nodebug(x)	(__pa_nodebug(x) | sme_me_mask)
114 
115 extern char __start_bss_decrypted[], __end_bss_decrypted[], __start_bss_decrypted_unused[];
116 
117 #endif	/* __ASSEMBLY__ */
118 
119 #endif	/* __X86_MEM_ENCRYPT_H__ */
120