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