1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17#undef DEBUG
18
19#include <linux/string.h>
20#include <asm/pgalloc.h>
21#include <asm/pgtable.h>
22#include <asm/kup.h>
23
24phys_addr_t memstart_addr __ro_after_init = (phys_addr_t)~0ull;
25EXPORT_SYMBOL_GPL(memstart_addr);
26phys_addr_t kernstart_addr __ro_after_init;
27EXPORT_SYMBOL_GPL(kernstart_addr);
28unsigned long kernstart_virt_addr __ro_after_init = KERNELBASE;
29EXPORT_SYMBOL_GPL(kernstart_virt_addr);
30
31static bool disable_kuep = !IS_ENABLED(CONFIG_PPC_KUEP);
32static bool disable_kuap = !IS_ENABLED(CONFIG_PPC_KUAP);
33
34static int __init parse_nosmep(char *p)
35{
36 disable_kuep = true;
37 pr_warn("Disabling Kernel Userspace Execution Prevention\n");
38 return 0;
39}
40early_param("nosmep", parse_nosmep);
41
42static int __init parse_nosmap(char *p)
43{
44 disable_kuap = true;
45 pr_warn("Disabling Kernel Userspace Access Protection\n");
46 return 0;
47}
48early_param("nosmap", parse_nosmap);
49
50void __ref setup_kup(void)
51{
52 setup_kuep(disable_kuep);
53 setup_kuap(disable_kuap);
54}
55
56#define CTOR(shift) static void ctor_##shift(void *addr) \
57{ \
58 memset(addr, 0, sizeof(void *) << (shift)); \
59}
60
61CTOR(0); CTOR(1); CTOR(2); CTOR(3); CTOR(4); CTOR(5); CTOR(6); CTOR(7);
62CTOR(8); CTOR(9); CTOR(10); CTOR(11); CTOR(12); CTOR(13); CTOR(14); CTOR(15);
63
64static inline void (*ctor(int shift))(void *)
65{
66 BUILD_BUG_ON(MAX_PGTABLE_INDEX_SIZE != 15);
67
68 switch (shift) {
69 case 0: return ctor_0;
70 case 1: return ctor_1;
71 case 2: return ctor_2;
72 case 3: return ctor_3;
73 case 4: return ctor_4;
74 case 5: return ctor_5;
75 case 6: return ctor_6;
76 case 7: return ctor_7;
77 case 8: return ctor_8;
78 case 9: return ctor_9;
79 case 10: return ctor_10;
80 case 11: return ctor_11;
81 case 12: return ctor_12;
82 case 13: return ctor_13;
83 case 14: return ctor_14;
84 case 15: return ctor_15;
85 }
86 return NULL;
87}
88
89struct kmem_cache *pgtable_cache[MAX_PGTABLE_INDEX_SIZE + 1];
90EXPORT_SYMBOL_GPL(pgtable_cache);
91
92
93
94
95
96
97
98
99void pgtable_cache_add(unsigned int shift)
100{
101 char *name;
102 unsigned long table_size = sizeof(void *) << shift;
103 unsigned long align = table_size;
104
105
106
107
108
109
110
111
112 unsigned long minalign = max(MAX_PGTABLE_INDEX_SIZE + 1,
113 HUGEPD_SHIFT_MASK + 1);
114 struct kmem_cache *new;
115
116
117
118
119 BUG_ON(!is_power_of_2(minalign));
120 BUG_ON(shift > MAX_PGTABLE_INDEX_SIZE);
121
122 if (PGT_CACHE(shift))
123 return;
124
125 align = max_t(unsigned long, align, minalign);
126 name = kasprintf(GFP_KERNEL, "pgtable-2^%d", shift);
127 new = kmem_cache_create(name, table_size, align, 0, ctor(shift));
128 if (!new)
129 panic("Could not allocate pgtable cache for order %d", shift);
130
131 kfree(name);
132 pgtable_cache[shift] = new;
133
134 pr_debug("Allocated pgtable cache for order %d\n", shift);
135}
136EXPORT_SYMBOL_GPL(pgtable_cache_add);
137
138void pgtable_cache_init(void)
139{
140 pgtable_cache_add(PGD_INDEX_SIZE);
141
142 if (PMD_CACHE_INDEX)
143 pgtable_cache_add(PMD_CACHE_INDEX);
144
145
146
147
148
149 if (PUD_CACHE_INDEX)
150 pgtable_cache_add(PUD_CACHE_INDEX);
151}
152