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/bootparam.h>
19
20#ifdef CONFIG_AMD_MEM_ENCRYPT
21
22extern u64 sme_me_mask;
23extern u64 sev_status;
24extern bool sev_enabled;
25
26void sme_encrypt_execute(unsigned long encrypted_kernel_vaddr,
27			 unsigned long decrypted_kernel_vaddr,
28			 unsigned long kernel_len,
29			 unsigned long encryption_wa,
30			 unsigned long encryption_pgd);
31
32void __init sme_early_encrypt(resource_size_t paddr,
33			      unsigned long size);
34void __init sme_early_decrypt(resource_size_t paddr,
35			      unsigned long size);
36
37void __init sme_map_bootdata(char *real_mode_data);
38void __init sme_unmap_bootdata(char *real_mode_data);
39
40void __init sme_early_init(void);
41
42void __init sme_encrypt_kernel(struct boot_params *bp);
43void __init sme_enable(struct boot_params *bp);
44
45int __init early_set_memory_decrypted(unsigned long vaddr, unsigned long size);
46int __init early_set_memory_encrypted(unsigned long vaddr, unsigned long size);
47
48void __init mem_encrypt_free_decrypted_mem(void);
49
50void __init sev_es_init_vc_handling(void);
51bool sme_active(void);
52bool sev_active(void);
53bool sev_es_active(void);
54
55void __init mem_encrypt_init(void);
56
57#define __bss_decrypted __section(".bss..decrypted")
58
59#else	/* !CONFIG_AMD_MEM_ENCRYPT */
60
61#define sme_me_mask	0ULL
62
63static inline void __init sme_early_encrypt(resource_size_t paddr,
64					    unsigned long size) { }
65static inline void __init sme_early_decrypt(resource_size_t paddr,
66					    unsigned long size) { }
67
68static inline void __init sme_map_bootdata(char *real_mode_data) { }
69static inline void __init sme_unmap_bootdata(char *real_mode_data) { }
70
71static inline void __init sme_early_init(void) { }
72
73static inline void __init sme_encrypt_kernel(struct boot_params *bp) { }
74static inline void __init sme_enable(struct boot_params *bp) { }
75
76static inline void sev_es_init_vc_handling(void) { }
77static inline bool sme_active(void) { return false; }
78static inline bool sev_active(void) { return false; }
79static inline bool sev_es_active(void) { return false; }
80
81static inline int __init
82early_set_memory_decrypted(unsigned long vaddr, unsigned long size) { return 0; }
83static inline int __init
84early_set_memory_encrypted(unsigned long vaddr, unsigned long size) { return 0; }
85
86static inline void mem_encrypt_free_decrypted_mem(void) { }
87
88static inline void mem_encrypt_init(void) { }
89
90#define __bss_decrypted
91
92#endif	/* CONFIG_AMD_MEM_ENCRYPT */
93
94/*
95 * The __sme_pa() and __sme_pa_nodebug() macros are meant for use when
96 * writing to or comparing values from the cr3 register.  Having the
97 * encryption mask set in cr3 enables the PGD entry to be encrypted and
98 * avoid special case handling of PGD allocations.
99 */
100#define __sme_pa(x)		(__pa(x) | sme_me_mask)
101#define __sme_pa_nodebug(x)	(__pa_nodebug(x) | sme_me_mask)
102
103extern char __start_bss_decrypted[], __end_bss_decrypted[], __start_bss_decrypted_unused[];
104
105static inline bool mem_encrypt_active(void)
106{
107	return sme_me_mask;
108}
109
110static inline u64 sme_get_me_mask(void)
111{
112	return sme_me_mask;
113}
114
115#endif	/* __ASSEMBLY__ */
116
117#endif	/* __X86_MEM_ENCRYPT_H__ */
118