1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copied from arch/arm64/kernel/cpufeature.c
4  *
5  * Copyright (C) 2015 ARM Ltd.
6  * Copyright (C) 2017 SiFive
7  */
8 
9 #include <linux/acpi.h>
10 #include <linux/bitmap.h>
11 #include <linux/cpu.h>
12 #include <linux/cpuhotplug.h>
13 #include <linux/ctype.h>
14 #include <linux/log2.h>
15 #include <linux/memory.h>
16 #include <linux/module.h>
17 #include <linux/of.h>
18 #include <asm/acpi.h>
19 #include <asm/alternative.h>
20 #include <asm/cacheflush.h>
21 #include <asm/cpufeature.h>
22 #include <asm/hwcap.h>
23 #include <asm/patch.h>
24 #include <asm/processor.h>
25 #include <asm/sbi.h>
26 #include <asm/vector.h>
27 #include <asm/vendor_extensions.h>
28 
29 #define NUM_ALPHA_EXTS ('z' - 'a' + 1)
30 
31 unsigned long elf_hwcap __read_mostly;
32 
33 /* Host ISA bitmap */
34 static DECLARE_BITMAP(riscv_isa, RISCV_ISA_EXT_MAX) __read_mostly;
35 
36 /* Per-cpu ISA extensions. */
37 struct riscv_isainfo hart_isa[NR_CPUS];
38 
39 /**
40  * riscv_isa_extension_base() - Get base extension word
41  *
42  * @isa_bitmap: ISA bitmap to use
43  * Return: base extension word as unsigned long value
44  *
45  * NOTE: If isa_bitmap is NULL then Host ISA bitmap will be used.
46  */
riscv_isa_extension_base(const unsigned long * isa_bitmap)47 unsigned long riscv_isa_extension_base(const unsigned long *isa_bitmap)
48 {
49 	if (!isa_bitmap)
50 		return riscv_isa[0];
51 	return isa_bitmap[0];
52 }
53 EXPORT_SYMBOL_GPL(riscv_isa_extension_base);
54 
55 /**
56  * __riscv_isa_extension_available() - Check whether given extension
57  * is available or not
58  *
59  * @isa_bitmap: ISA bitmap to use
60  * @bit: bit position of the desired extension
61  * Return: true or false
62  *
63  * NOTE: If isa_bitmap is NULL then Host ISA bitmap will be used.
64  */
__riscv_isa_extension_available(const unsigned long * isa_bitmap,unsigned int bit)65 bool __riscv_isa_extension_available(const unsigned long *isa_bitmap, unsigned int bit)
66 {
67 	const unsigned long *bmap = (isa_bitmap) ? isa_bitmap : riscv_isa;
68 
69 	if (bit >= RISCV_ISA_EXT_MAX)
70 		return false;
71 
72 	return test_bit(bit, bmap) ? true : false;
73 }
74 EXPORT_SYMBOL_GPL(__riscv_isa_extension_available);
75 
riscv_ext_zicbom_validate(const struct riscv_isa_ext_data * data,const unsigned long * isa_bitmap)76 static int riscv_ext_zicbom_validate(const struct riscv_isa_ext_data *data,
77 				     const unsigned long *isa_bitmap)
78 {
79 	if (!riscv_cbom_block_size) {
80 		pr_err("Zicbom detected in ISA string, disabling as no cbom-block-size found\n");
81 		return -EINVAL;
82 	}
83 	if (!is_power_of_2(riscv_cbom_block_size)) {
84 		pr_err("Zicbom disabled as cbom-block-size present, but is not a power-of-2\n");
85 		return -EINVAL;
86 	}
87 	return 0;
88 }
89 
riscv_ext_zicboz_validate(const struct riscv_isa_ext_data * data,const unsigned long * isa_bitmap)90 static int riscv_ext_zicboz_validate(const struct riscv_isa_ext_data *data,
91 				     const unsigned long *isa_bitmap)
92 {
93 	if (!riscv_cboz_block_size) {
94 		pr_err("Zicboz detected in ISA string, disabling as no cboz-block-size found\n");
95 		return -EINVAL;
96 	}
97 	if (!is_power_of_2(riscv_cboz_block_size)) {
98 		pr_err("Zicboz disabled as cboz-block-size present, but is not a power-of-2\n");
99 		return -EINVAL;
100 	}
101 	return 0;
102 }
103 
riscv_ext_zca_depends(const struct riscv_isa_ext_data * data,const unsigned long * isa_bitmap)104 static int riscv_ext_zca_depends(const struct riscv_isa_ext_data *data,
105 				 const unsigned long *isa_bitmap)
106 {
107 	if (__riscv_isa_extension_available(isa_bitmap, RISCV_ISA_EXT_ZCA))
108 		return 0;
109 
110 	return -EPROBE_DEFER;
111 }
riscv_ext_zcd_validate(const struct riscv_isa_ext_data * data,const unsigned long * isa_bitmap)112 static int riscv_ext_zcd_validate(const struct riscv_isa_ext_data *data,
113 				  const unsigned long *isa_bitmap)
114 {
115 	if (__riscv_isa_extension_available(isa_bitmap, RISCV_ISA_EXT_ZCA) &&
116 	    __riscv_isa_extension_available(isa_bitmap, RISCV_ISA_EXT_d))
117 		return 0;
118 
119 	return -EPROBE_DEFER;
120 }
121 
riscv_ext_zcf_validate(const struct riscv_isa_ext_data * data,const unsigned long * isa_bitmap)122 static int riscv_ext_zcf_validate(const struct riscv_isa_ext_data *data,
123 				  const unsigned long *isa_bitmap)
124 {
125 	if (IS_ENABLED(CONFIG_64BIT))
126 		return -EINVAL;
127 
128 	if (__riscv_isa_extension_available(isa_bitmap, RISCV_ISA_EXT_ZCA) &&
129 	    __riscv_isa_extension_available(isa_bitmap, RISCV_ISA_EXT_f))
130 		return 0;
131 
132 	return -EPROBE_DEFER;
133 }
134 
135 static const unsigned int riscv_zk_bundled_exts[] = {
136 	RISCV_ISA_EXT_ZBKB,
137 	RISCV_ISA_EXT_ZBKC,
138 	RISCV_ISA_EXT_ZBKX,
139 	RISCV_ISA_EXT_ZKND,
140 	RISCV_ISA_EXT_ZKNE,
141 	RISCV_ISA_EXT_ZKR,
142 	RISCV_ISA_EXT_ZKT,
143 };
144 
145 static const unsigned int riscv_zkn_bundled_exts[] = {
146 	RISCV_ISA_EXT_ZBKB,
147 	RISCV_ISA_EXT_ZBKC,
148 	RISCV_ISA_EXT_ZBKX,
149 	RISCV_ISA_EXT_ZKND,
150 	RISCV_ISA_EXT_ZKNE,
151 	RISCV_ISA_EXT_ZKNH,
152 };
153 
154 static const unsigned int riscv_zks_bundled_exts[] = {
155 	RISCV_ISA_EXT_ZBKB,
156 	RISCV_ISA_EXT_ZBKC,
157 	RISCV_ISA_EXT_ZKSED,
158 	RISCV_ISA_EXT_ZKSH
159 };
160 
161 #define RISCV_ISA_EXT_ZVKN	\
162 	RISCV_ISA_EXT_ZVKNED,	\
163 	RISCV_ISA_EXT_ZVKNHB,	\
164 	RISCV_ISA_EXT_ZVKB,	\
165 	RISCV_ISA_EXT_ZVKT
166 
167 static const unsigned int riscv_zvkn_bundled_exts[] = {
168 	RISCV_ISA_EXT_ZVKN
169 };
170 
171 static const unsigned int riscv_zvknc_bundled_exts[] = {
172 	RISCV_ISA_EXT_ZVKN,
173 	RISCV_ISA_EXT_ZVBC
174 };
175 
176 static const unsigned int riscv_zvkng_bundled_exts[] = {
177 	RISCV_ISA_EXT_ZVKN,
178 	RISCV_ISA_EXT_ZVKG
179 };
180 
181 #define RISCV_ISA_EXT_ZVKS	\
182 	RISCV_ISA_EXT_ZVKSED,	\
183 	RISCV_ISA_EXT_ZVKSH,	\
184 	RISCV_ISA_EXT_ZVKB,	\
185 	RISCV_ISA_EXT_ZVKT
186 
187 static const unsigned int riscv_zvks_bundled_exts[] = {
188 	RISCV_ISA_EXT_ZVKS
189 };
190 
191 static const unsigned int riscv_zvksc_bundled_exts[] = {
192 	RISCV_ISA_EXT_ZVKS,
193 	RISCV_ISA_EXT_ZVBC
194 };
195 
196 static const unsigned int riscv_zvksg_bundled_exts[] = {
197 	RISCV_ISA_EXT_ZVKS,
198 	RISCV_ISA_EXT_ZVKG
199 };
200 
201 static const unsigned int riscv_zvbb_exts[] = {
202 	RISCV_ISA_EXT_ZVKB
203 };
204 
205 #define RISCV_ISA_EXT_ZVE64F_IMPLY_LIST	\
206 	RISCV_ISA_EXT_ZVE64X,		\
207 	RISCV_ISA_EXT_ZVE32F,		\
208 	RISCV_ISA_EXT_ZVE32X
209 
210 #define RISCV_ISA_EXT_ZVE64D_IMPLY_LIST	\
211 	RISCV_ISA_EXT_ZVE64F,		\
212 	RISCV_ISA_EXT_ZVE64F_IMPLY_LIST
213 
214 #define RISCV_ISA_EXT_V_IMPLY_LIST	\
215 	RISCV_ISA_EXT_ZVE64D,		\
216 	RISCV_ISA_EXT_ZVE64D_IMPLY_LIST
217 
218 static const unsigned int riscv_zve32f_exts[] = {
219 	RISCV_ISA_EXT_ZVE32X
220 };
221 
222 static const unsigned int riscv_zve64f_exts[] = {
223 	RISCV_ISA_EXT_ZVE64F_IMPLY_LIST
224 };
225 
226 static const unsigned int riscv_zve64d_exts[] = {
227 	RISCV_ISA_EXT_ZVE64D_IMPLY_LIST
228 };
229 
230 static const unsigned int riscv_v_exts[] = {
231 	RISCV_ISA_EXT_V_IMPLY_LIST
232 };
233 
234 static const unsigned int riscv_zve64x_exts[] = {
235 	RISCV_ISA_EXT_ZVE32X,
236 	RISCV_ISA_EXT_ZVE64X
237 };
238 
239 /*
240  * While the [ms]envcfg CSRs were not defined until version 1.12 of the RISC-V
241  * privileged ISA, the existence of the CSRs is implied by any extension which
242  * specifies [ms]envcfg bit(s). Hence, we define a custom ISA extension for the
243  * existence of the CSR, and treat it as a subset of those other extensions.
244  */
245 static const unsigned int riscv_xlinuxenvcfg_exts[] = {
246 	RISCV_ISA_EXT_XLINUXENVCFG
247 };
248 
249 /*
250  * Zc* spec states that:
251  * - C always implies Zca
252  * - C+F implies Zcf (RV32 only)
253  * - C+D implies Zcd
254  *
255  * These extensions will be enabled and then validated depending on the
256  * availability of F/D RV32.
257  */
258 static const unsigned int riscv_c_exts[] = {
259 	RISCV_ISA_EXT_ZCA,
260 	RISCV_ISA_EXT_ZCF,
261 	RISCV_ISA_EXT_ZCD,
262 };
263 
264 /*
265  * The canonical order of ISA extension names in the ISA string is defined in
266  * chapter 27 of the unprivileged specification.
267  *
268  * Ordinarily, for in-kernel data structures, this order is unimportant but
269  * isa_ext_arr defines the order of the ISA string in /proc/cpuinfo.
270  *
271  * The specification uses vague wording, such as should, when it comes to
272  * ordering, so for our purposes the following rules apply:
273  *
274  * 1. All multi-letter extensions must be separated from other extensions by an
275  *    underscore.
276  *
277  * 2. Additional standard extensions (starting with 'Z') must be sorted after
278  *    single-letter extensions and before any higher-privileged extensions.
279  *
280  * 3. The first letter following the 'Z' conventionally indicates the most
281  *    closely related alphabetical extension category, IMAFDQLCBKJTPVH.
282  *    If multiple 'Z' extensions are named, they must be ordered first by
283  *    category, then alphabetically within a category.
284  *
285  * 3. Standard supervisor-level extensions (starting with 'S') must be listed
286  *    after standard unprivileged extensions.  If multiple supervisor-level
287  *    extensions are listed, they must be ordered alphabetically.
288  *
289  * 4. Standard machine-level extensions (starting with 'Zxm') must be listed
290  *    after any lower-privileged, standard extensions.  If multiple
291  *    machine-level extensions are listed, they must be ordered
292  *    alphabetically.
293  *
294  * 5. Non-standard extensions (starting with 'X') must be listed after all
295  *    standard extensions. If multiple non-standard extensions are listed, they
296  *    must be ordered alphabetically.
297  *
298  * An example string following the order is:
299  *    rv64imadc_zifoo_zigoo_zafoo_sbar_scar_zxmbaz_xqux_xrux
300  *
301  * New entries to this struct should follow the ordering rules described above.
302  */
303 const struct riscv_isa_ext_data riscv_isa_ext[] = {
304 	__RISCV_ISA_EXT_DATA(i, RISCV_ISA_EXT_i),
305 	__RISCV_ISA_EXT_DATA(m, RISCV_ISA_EXT_m),
306 	__RISCV_ISA_EXT_DATA(a, RISCV_ISA_EXT_a),
307 	__RISCV_ISA_EXT_DATA(f, RISCV_ISA_EXT_f),
308 	__RISCV_ISA_EXT_DATA(d, RISCV_ISA_EXT_d),
309 	__RISCV_ISA_EXT_DATA(q, RISCV_ISA_EXT_q),
310 	__RISCV_ISA_EXT_SUPERSET(c, RISCV_ISA_EXT_c, riscv_c_exts),
311 	__RISCV_ISA_EXT_SUPERSET(v, RISCV_ISA_EXT_v, riscv_v_exts),
312 	__RISCV_ISA_EXT_DATA(h, RISCV_ISA_EXT_h),
313 	__RISCV_ISA_EXT_SUPERSET_VALIDATE(zicbom, RISCV_ISA_EXT_ZICBOM, riscv_xlinuxenvcfg_exts,
314 					  riscv_ext_zicbom_validate),
315 	__RISCV_ISA_EXT_SUPERSET_VALIDATE(zicboz, RISCV_ISA_EXT_ZICBOZ, riscv_xlinuxenvcfg_exts,
316 					  riscv_ext_zicboz_validate),
317 	__RISCV_ISA_EXT_DATA(zicntr, RISCV_ISA_EXT_ZICNTR),
318 	__RISCV_ISA_EXT_DATA(zicond, RISCV_ISA_EXT_ZICOND),
319 	__RISCV_ISA_EXT_DATA(zicsr, RISCV_ISA_EXT_ZICSR),
320 	__RISCV_ISA_EXT_DATA(zifencei, RISCV_ISA_EXT_ZIFENCEI),
321 	__RISCV_ISA_EXT_DATA(zihintntl, RISCV_ISA_EXT_ZIHINTNTL),
322 	__RISCV_ISA_EXT_DATA(zihintpause, RISCV_ISA_EXT_ZIHINTPAUSE),
323 	__RISCV_ISA_EXT_DATA(zihpm, RISCV_ISA_EXT_ZIHPM),
324 	__RISCV_ISA_EXT_DATA(zimop, RISCV_ISA_EXT_ZIMOP),
325 	__RISCV_ISA_EXT_DATA(zacas, RISCV_ISA_EXT_ZACAS),
326 	__RISCV_ISA_EXT_DATA(zawrs, RISCV_ISA_EXT_ZAWRS),
327 	__RISCV_ISA_EXT_DATA(zfa, RISCV_ISA_EXT_ZFA),
328 	__RISCV_ISA_EXT_DATA(zfh, RISCV_ISA_EXT_ZFH),
329 	__RISCV_ISA_EXT_DATA(zfhmin, RISCV_ISA_EXT_ZFHMIN),
330 	__RISCV_ISA_EXT_DATA(zca, RISCV_ISA_EXT_ZCA),
331 	__RISCV_ISA_EXT_DATA_VALIDATE(zcb, RISCV_ISA_EXT_ZCB, riscv_ext_zca_depends),
332 	__RISCV_ISA_EXT_DATA_VALIDATE(zcd, RISCV_ISA_EXT_ZCD, riscv_ext_zcd_validate),
333 	__RISCV_ISA_EXT_DATA_VALIDATE(zcf, RISCV_ISA_EXT_ZCF, riscv_ext_zcf_validate),
334 	__RISCV_ISA_EXT_DATA_VALIDATE(zcmop, RISCV_ISA_EXT_ZCMOP, riscv_ext_zca_depends),
335 	__RISCV_ISA_EXT_DATA(zba, RISCV_ISA_EXT_ZBA),
336 	__RISCV_ISA_EXT_DATA(zbb, RISCV_ISA_EXT_ZBB),
337 	__RISCV_ISA_EXT_DATA(zbc, RISCV_ISA_EXT_ZBC),
338 	__RISCV_ISA_EXT_DATA(zbkb, RISCV_ISA_EXT_ZBKB),
339 	__RISCV_ISA_EXT_DATA(zbkc, RISCV_ISA_EXT_ZBKC),
340 	__RISCV_ISA_EXT_DATA(zbkx, RISCV_ISA_EXT_ZBKX),
341 	__RISCV_ISA_EXT_DATA(zbs, RISCV_ISA_EXT_ZBS),
342 	__RISCV_ISA_EXT_BUNDLE(zk, riscv_zk_bundled_exts),
343 	__RISCV_ISA_EXT_BUNDLE(zkn, riscv_zkn_bundled_exts),
344 	__RISCV_ISA_EXT_DATA(zknd, RISCV_ISA_EXT_ZKND),
345 	__RISCV_ISA_EXT_DATA(zkne, RISCV_ISA_EXT_ZKNE),
346 	__RISCV_ISA_EXT_DATA(zknh, RISCV_ISA_EXT_ZKNH),
347 	__RISCV_ISA_EXT_DATA(zkr, RISCV_ISA_EXT_ZKR),
348 	__RISCV_ISA_EXT_BUNDLE(zks, riscv_zks_bundled_exts),
349 	__RISCV_ISA_EXT_DATA(zkt, RISCV_ISA_EXT_ZKT),
350 	__RISCV_ISA_EXT_DATA(zksed, RISCV_ISA_EXT_ZKSED),
351 	__RISCV_ISA_EXT_DATA(zksh, RISCV_ISA_EXT_ZKSH),
352 	__RISCV_ISA_EXT_DATA(ztso, RISCV_ISA_EXT_ZTSO),
353 	__RISCV_ISA_EXT_SUPERSET(zvbb, RISCV_ISA_EXT_ZVBB, riscv_zvbb_exts),
354 	__RISCV_ISA_EXT_DATA(zvbc, RISCV_ISA_EXT_ZVBC),
355 	__RISCV_ISA_EXT_SUPERSET(zve32f, RISCV_ISA_EXT_ZVE32F, riscv_zve32f_exts),
356 	__RISCV_ISA_EXT_DATA(zve32x, RISCV_ISA_EXT_ZVE32X),
357 	__RISCV_ISA_EXT_SUPERSET(zve64d, RISCV_ISA_EXT_ZVE64D, riscv_zve64d_exts),
358 	__RISCV_ISA_EXT_SUPERSET(zve64f, RISCV_ISA_EXT_ZVE64F, riscv_zve64f_exts),
359 	__RISCV_ISA_EXT_SUPERSET(zve64x, RISCV_ISA_EXT_ZVE64X, riscv_zve64x_exts),
360 	__RISCV_ISA_EXT_DATA(zvfh, RISCV_ISA_EXT_ZVFH),
361 	__RISCV_ISA_EXT_DATA(zvfhmin, RISCV_ISA_EXT_ZVFHMIN),
362 	__RISCV_ISA_EXT_DATA(zvkb, RISCV_ISA_EXT_ZVKB),
363 	__RISCV_ISA_EXT_DATA(zvkg, RISCV_ISA_EXT_ZVKG),
364 	__RISCV_ISA_EXT_BUNDLE(zvkn, riscv_zvkn_bundled_exts),
365 	__RISCV_ISA_EXT_BUNDLE(zvknc, riscv_zvknc_bundled_exts),
366 	__RISCV_ISA_EXT_DATA(zvkned, RISCV_ISA_EXT_ZVKNED),
367 	__RISCV_ISA_EXT_BUNDLE(zvkng, riscv_zvkng_bundled_exts),
368 	__RISCV_ISA_EXT_DATA(zvknha, RISCV_ISA_EXT_ZVKNHA),
369 	__RISCV_ISA_EXT_DATA(zvknhb, RISCV_ISA_EXT_ZVKNHB),
370 	__RISCV_ISA_EXT_BUNDLE(zvks, riscv_zvks_bundled_exts),
371 	__RISCV_ISA_EXT_BUNDLE(zvksc, riscv_zvksc_bundled_exts),
372 	__RISCV_ISA_EXT_DATA(zvksed, RISCV_ISA_EXT_ZVKSED),
373 	__RISCV_ISA_EXT_DATA(zvksh, RISCV_ISA_EXT_ZVKSH),
374 	__RISCV_ISA_EXT_BUNDLE(zvksg, riscv_zvksg_bundled_exts),
375 	__RISCV_ISA_EXT_DATA(zvkt, RISCV_ISA_EXT_ZVKT),
376 	__RISCV_ISA_EXT_DATA(smaia, RISCV_ISA_EXT_SMAIA),
377 	__RISCV_ISA_EXT_DATA(smstateen, RISCV_ISA_EXT_SMSTATEEN),
378 	__RISCV_ISA_EXT_DATA(ssaia, RISCV_ISA_EXT_SSAIA),
379 	__RISCV_ISA_EXT_DATA(sscofpmf, RISCV_ISA_EXT_SSCOFPMF),
380 	__RISCV_ISA_EXT_DATA(sstc, RISCV_ISA_EXT_SSTC),
381 	__RISCV_ISA_EXT_DATA(svinval, RISCV_ISA_EXT_SVINVAL),
382 	__RISCV_ISA_EXT_DATA(svnapot, RISCV_ISA_EXT_SVNAPOT),
383 	__RISCV_ISA_EXT_DATA(svpbmt, RISCV_ISA_EXT_SVPBMT),
384 	__RISCV_ISA_EXT_DATA(svvptc, RISCV_ISA_EXT_SVVPTC),
385 };
386 
387 const size_t riscv_isa_ext_count = ARRAY_SIZE(riscv_isa_ext);
388 
riscv_isa_set_ext(const struct riscv_isa_ext_data * ext,unsigned long * bitmap)389 static void riscv_isa_set_ext(const struct riscv_isa_ext_data *ext, unsigned long *bitmap)
390 {
391 	if (ext->id != RISCV_ISA_EXT_INVALID)
392 		set_bit(ext->id, bitmap);
393 
394 	for (int i = 0; i < ext->subset_ext_size; i++) {
395 		if (ext->subset_ext_ids[i] != RISCV_ISA_EXT_INVALID)
396 			set_bit(ext->subset_ext_ids[i], bitmap);
397 	}
398 }
399 
riscv_get_isa_ext_data(unsigned int ext_id)400 static const struct riscv_isa_ext_data *riscv_get_isa_ext_data(unsigned int ext_id)
401 {
402 	for (int i = 0; i < riscv_isa_ext_count; i++) {
403 		if (riscv_isa_ext[i].id == ext_id)
404 			return &riscv_isa_ext[i];
405 	}
406 
407 	return NULL;
408 }
409 
410 /*
411  * "Resolve" a source ISA bitmap into one that matches kernel configuration as
412  * well as correct extension dependencies. Some extensions depends on specific
413  * kernel configuration to be usable (V needs CONFIG_RISCV_ISA_V for instance)
414  * and this function will actually validate all the extensions provided in
415  * source_isa into the resolved_isa based on extensions validate() callbacks.
416  */
riscv_resolve_isa(unsigned long * source_isa,unsigned long * resolved_isa,unsigned long * this_hwcap,unsigned long * isa2hwcap)417 static void __init riscv_resolve_isa(unsigned long *source_isa,
418 				     unsigned long *resolved_isa, unsigned long *this_hwcap,
419 				     unsigned long *isa2hwcap)
420 {
421 	bool loop;
422 	const struct riscv_isa_ext_data *ext;
423 	DECLARE_BITMAP(prev_resolved_isa, RISCV_ISA_EXT_MAX);
424 	int max_loop_count = riscv_isa_ext_count, ret;
425 	unsigned int bit;
426 
427 	do {
428 		loop = false;
429 		if (max_loop_count-- < 0) {
430 			pr_err("Failed to reach a stable ISA state\n");
431 			return;
432 		}
433 		bitmap_copy(prev_resolved_isa, resolved_isa, RISCV_ISA_EXT_MAX);
434 		for_each_set_bit(bit, source_isa, RISCV_ISA_EXT_MAX) {
435 			ext = riscv_get_isa_ext_data(bit);
436 
437 			if (ext && ext->validate) {
438 				ret = ext->validate(ext, resolved_isa);
439 				if (ret == -EPROBE_DEFER) {
440 					loop = true;
441 					continue;
442 				} else if (ret) {
443 					/* Disable the extension entirely */
444 					clear_bit(bit, source_isa);
445 					continue;
446 				}
447 			}
448 
449 			set_bit(bit, resolved_isa);
450 			/* No need to keep it in source isa now that it is enabled */
451 			clear_bit(bit, source_isa);
452 
453 			/* Single letter extensions get set in hwcap */
454 			if (bit < RISCV_ISA_EXT_BASE)
455 				*this_hwcap |= isa2hwcap[bit];
456 		}
457 	} while (loop && memcmp(prev_resolved_isa, resolved_isa, sizeof(prev_resolved_isa)));
458 }
459 
match_isa_ext(const char * name,const char * name_end,unsigned long * bitmap)460 static void __init match_isa_ext(const char *name, const char *name_end, unsigned long *bitmap)
461 {
462 	for (int i = 0; i < riscv_isa_ext_count; i++) {
463 		const struct riscv_isa_ext_data *ext = &riscv_isa_ext[i];
464 
465 		if ((name_end - name == strlen(ext->name)) &&
466 		    !strncasecmp(name, ext->name, name_end - name)) {
467 			riscv_isa_set_ext(ext, bitmap);
468 			break;
469 		}
470 	}
471 }
472 
riscv_parse_isa_string(const char * isa,unsigned long * bitmap)473 static void __init riscv_parse_isa_string(const char *isa, unsigned long *bitmap)
474 {
475 	/*
476 	 * For all possible cpus, we have already validated in
477 	 * the boot process that they at least contain "rv" and
478 	 * whichever of "32"/"64" this kernel supports, and so this
479 	 * section can be skipped.
480 	 */
481 	isa += 4;
482 
483 	while (*isa) {
484 		const char *ext = isa++;
485 		const char *ext_end = isa;
486 		bool ext_err = false;
487 
488 		switch (*ext) {
489 		case 'x':
490 		case 'X':
491 			if (acpi_disabled)
492 				pr_warn_once("Vendor extensions are ignored in riscv,isa. Use riscv,isa-extensions instead.");
493 			/*
494 			 * To skip an extension, we find its end.
495 			 * As multi-letter extensions must be split from other multi-letter
496 			 * extensions with an "_", the end of a multi-letter extension will
497 			 * either be the null character or the "_" at the start of the next
498 			 * multi-letter extension.
499 			 */
500 			for (; *isa && *isa != '_'; ++isa)
501 				;
502 			ext_err = true;
503 			break;
504 		case 's':
505 			/*
506 			 * Workaround for invalid single-letter 's' & 'u' (QEMU).
507 			 * No need to set the bit in riscv_isa as 's' & 'u' are
508 			 * not valid ISA extensions. It works unless the first
509 			 * multi-letter extension in the ISA string begins with
510 			 * "Su" and is not prefixed with an underscore.
511 			 */
512 			if (ext[-1] != '_' && ext[1] == 'u') {
513 				++isa;
514 				ext_err = true;
515 				break;
516 			}
517 			fallthrough;
518 		case 'S':
519 		case 'z':
520 		case 'Z':
521 			/*
522 			 * Before attempting to parse the extension itself, we find its end.
523 			 * As multi-letter extensions must be split from other multi-letter
524 			 * extensions with an "_", the end of a multi-letter extension will
525 			 * either be the null character or the "_" at the start of the next
526 			 * multi-letter extension.
527 			 *
528 			 * Next, as the extensions version is currently ignored, we
529 			 * eliminate that portion. This is done by parsing backwards from
530 			 * the end of the extension, removing any numbers. This may be a
531 			 * major or minor number however, so the process is repeated if a
532 			 * minor number was found.
533 			 *
534 			 * ext_end is intended to represent the first character *after* the
535 			 * name portion of an extension, but will be decremented to the last
536 			 * character itself while eliminating the extensions version number.
537 			 * A simple re-increment solves this problem.
538 			 */
539 			for (; *isa && *isa != '_'; ++isa)
540 				if (unlikely(!isalnum(*isa)))
541 					ext_err = true;
542 
543 			ext_end = isa;
544 			if (unlikely(ext_err))
545 				break;
546 
547 			if (!isdigit(ext_end[-1]))
548 				break;
549 
550 			while (isdigit(*--ext_end))
551 				;
552 
553 			if (tolower(ext_end[0]) != 'p' || !isdigit(ext_end[-1])) {
554 				++ext_end;
555 				break;
556 			}
557 
558 			while (isdigit(*--ext_end))
559 				;
560 
561 			++ext_end;
562 			break;
563 		default:
564 			/*
565 			 * Things are a little easier for single-letter extensions, as they
566 			 * are parsed forwards.
567 			 *
568 			 * After checking that our starting position is valid, we need to
569 			 * ensure that, when isa was incremented at the start of the loop,
570 			 * that it arrived at the start of the next extension.
571 			 *
572 			 * If we are already on a non-digit, there is nothing to do. Either
573 			 * we have a multi-letter extension's _, or the start of an
574 			 * extension.
575 			 *
576 			 * Otherwise we have found the current extension's major version
577 			 * number. Parse past it, and a subsequent p/minor version number
578 			 * if present. The `p` extension must not appear immediately after
579 			 * a number, so there is no fear of missing it.
580 			 *
581 			 */
582 			if (unlikely(!isalpha(*ext))) {
583 				ext_err = true;
584 				break;
585 			}
586 
587 			if (!isdigit(*isa))
588 				break;
589 
590 			while (isdigit(*++isa))
591 				;
592 
593 			if (tolower(*isa) != 'p')
594 				break;
595 
596 			if (!isdigit(*++isa)) {
597 				--isa;
598 				break;
599 			}
600 
601 			while (isdigit(*++isa))
602 				;
603 
604 			break;
605 		}
606 
607 		/*
608 		 * The parser expects that at the start of an iteration isa points to the
609 		 * first character of the next extension. As we stop parsing an extension
610 		 * on meeting a non-alphanumeric character, an extra increment is needed
611 		 * where the succeeding extension is a multi-letter prefixed with an "_".
612 		 */
613 		if (*isa == '_')
614 			++isa;
615 
616 		if (unlikely(ext_err))
617 			continue;
618 
619 		match_isa_ext(ext, ext_end, bitmap);
620 	}
621 }
622 
riscv_fill_hwcap_from_isa_string(unsigned long * isa2hwcap)623 static void __init riscv_fill_hwcap_from_isa_string(unsigned long *isa2hwcap)
624 {
625 	struct device_node *node;
626 	const char *isa;
627 	int rc;
628 	struct acpi_table_header *rhct;
629 	acpi_status status;
630 	unsigned int cpu;
631 	u64 boot_vendorid;
632 	u64 boot_archid;
633 
634 	if (!acpi_disabled) {
635 		status = acpi_get_table(ACPI_SIG_RHCT, 0, &rhct);
636 		if (ACPI_FAILURE(status))
637 			return;
638 	}
639 
640 	boot_vendorid = riscv_get_mvendorid();
641 	boot_archid = riscv_get_marchid();
642 
643 	for_each_possible_cpu(cpu) {
644 		struct riscv_isainfo *isainfo = &hart_isa[cpu];
645 		unsigned long this_hwcap = 0;
646 		DECLARE_BITMAP(source_isa, RISCV_ISA_EXT_MAX) = { 0 };
647 
648 		if (acpi_disabled) {
649 			node = of_cpu_device_node_get(cpu);
650 			if (!node) {
651 				pr_warn("Unable to find cpu node\n");
652 				continue;
653 			}
654 
655 			rc = of_property_read_string(node, "riscv,isa", &isa);
656 			of_node_put(node);
657 			if (rc) {
658 				pr_warn("Unable to find \"riscv,isa\" devicetree entry\n");
659 				continue;
660 			}
661 		} else {
662 			rc = acpi_get_riscv_isa(rhct, cpu, &isa);
663 			if (rc < 0) {
664 				pr_warn("Unable to get ISA for the hart - %d\n", cpu);
665 				continue;
666 			}
667 		}
668 
669 		riscv_parse_isa_string(isa, source_isa);
670 
671 		/*
672 		 * These ones were as they were part of the base ISA when the
673 		 * port & dt-bindings were upstreamed, and so can be set
674 		 * unconditionally where `i` is in riscv,isa on DT systems.
675 		 */
676 		if (acpi_disabled) {
677 			set_bit(RISCV_ISA_EXT_ZICSR, source_isa);
678 			set_bit(RISCV_ISA_EXT_ZIFENCEI, source_isa);
679 			set_bit(RISCV_ISA_EXT_ZICNTR, source_isa);
680 			set_bit(RISCV_ISA_EXT_ZIHPM, source_isa);
681 		}
682 
683 		/*
684 		 * "V" in ISA strings is ambiguous in practice: it should mean
685 		 * just the standard V-1.0 but vendors aren't well behaved.
686 		 * Many vendors with T-Head CPU cores which implement the 0.7.1
687 		 * version of the vector specification put "v" into their DTs.
688 		 * CPU cores with the ratified spec will contain non-zero
689 		 * marchid.
690 		 */
691 		if (acpi_disabled && boot_vendorid == THEAD_VENDOR_ID && boot_archid == 0x0) {
692 			this_hwcap &= ~isa2hwcap[RISCV_ISA_EXT_v];
693 			clear_bit(RISCV_ISA_EXT_v, source_isa);
694 		}
695 
696 		riscv_resolve_isa(source_isa, isainfo->isa, &this_hwcap, isa2hwcap);
697 
698 		/*
699 		 * All "okay" hart should have same isa. Set HWCAP based on
700 		 * common capabilities of every "okay" hart, in case they don't
701 		 * have.
702 		 */
703 		if (elf_hwcap)
704 			elf_hwcap &= this_hwcap;
705 		else
706 			elf_hwcap = this_hwcap;
707 
708 		if (bitmap_empty(riscv_isa, RISCV_ISA_EXT_MAX))
709 			bitmap_copy(riscv_isa, isainfo->isa, RISCV_ISA_EXT_MAX);
710 		else
711 			bitmap_and(riscv_isa, riscv_isa, isainfo->isa, RISCV_ISA_EXT_MAX);
712 	}
713 
714 	if (!acpi_disabled && rhct)
715 		acpi_put_table((struct acpi_table_header *)rhct);
716 }
717 
riscv_fill_cpu_vendor_ext(struct device_node * cpu_node,int cpu)718 static void __init riscv_fill_cpu_vendor_ext(struct device_node *cpu_node, int cpu)
719 {
720 	if (!IS_ENABLED(CONFIG_RISCV_ISA_VENDOR_EXT))
721 		return;
722 
723 	for (int i = 0; i < riscv_isa_vendor_ext_list_size; i++) {
724 		struct riscv_isa_vendor_ext_data_list *ext_list = riscv_isa_vendor_ext_list[i];
725 
726 		for (int j = 0; j < ext_list->ext_data_count; j++) {
727 			const struct riscv_isa_ext_data ext = ext_list->ext_data[j];
728 			struct riscv_isavendorinfo *isavendorinfo = &ext_list->per_hart_isa_bitmap[cpu];
729 
730 			if (of_property_match_string(cpu_node, "riscv,isa-extensions",
731 						     ext.property) < 0)
732 				continue;
733 
734 			/*
735 			 * Assume that subset extensions are all members of the
736 			 * same vendor.
737 			 */
738 			if (ext.subset_ext_size)
739 				for (int k = 0; k < ext.subset_ext_size; k++)
740 					set_bit(ext.subset_ext_ids[k], isavendorinfo->isa);
741 
742 			set_bit(ext.id, isavendorinfo->isa);
743 		}
744 	}
745 }
746 
747 /*
748  * Populate all_harts_isa_bitmap for each vendor with all of the extensions that
749  * are shared across CPUs for that vendor.
750  */
riscv_fill_vendor_ext_list(int cpu)751 static void __init riscv_fill_vendor_ext_list(int cpu)
752 {
753 	if (!IS_ENABLED(CONFIG_RISCV_ISA_VENDOR_EXT))
754 		return;
755 
756 	for (int i = 0; i < riscv_isa_vendor_ext_list_size; i++) {
757 		struct riscv_isa_vendor_ext_data_list *ext_list = riscv_isa_vendor_ext_list[i];
758 
759 		if (!ext_list->is_initialized) {
760 			bitmap_copy(ext_list->all_harts_isa_bitmap.isa,
761 				    ext_list->per_hart_isa_bitmap[cpu].isa,
762 				    RISCV_ISA_VENDOR_EXT_MAX);
763 			ext_list->is_initialized = true;
764 		} else {
765 			bitmap_and(ext_list->all_harts_isa_bitmap.isa,
766 				   ext_list->all_harts_isa_bitmap.isa,
767 				   ext_list->per_hart_isa_bitmap[cpu].isa,
768 				   RISCV_ISA_VENDOR_EXT_MAX);
769 		}
770 	}
771 }
772 
riscv_fill_hwcap_from_ext_list(unsigned long * isa2hwcap)773 static int __init riscv_fill_hwcap_from_ext_list(unsigned long *isa2hwcap)
774 {
775 	unsigned int cpu;
776 
777 	for_each_possible_cpu(cpu) {
778 		unsigned long this_hwcap = 0;
779 		struct device_node *cpu_node;
780 		struct riscv_isainfo *isainfo = &hart_isa[cpu];
781 		DECLARE_BITMAP(source_isa, RISCV_ISA_EXT_MAX) = { 0 };
782 
783 		cpu_node = of_cpu_device_node_get(cpu);
784 		if (!cpu_node) {
785 			pr_warn("Unable to find cpu node\n");
786 			continue;
787 		}
788 
789 		if (!of_property_present(cpu_node, "riscv,isa-extensions")) {
790 			of_node_put(cpu_node);
791 			continue;
792 		}
793 
794 		for (int i = 0; i < riscv_isa_ext_count; i++) {
795 			const struct riscv_isa_ext_data *ext = &riscv_isa_ext[i];
796 
797 			if (of_property_match_string(cpu_node, "riscv,isa-extensions",
798 						     ext->property) < 0)
799 				continue;
800 
801 			riscv_isa_set_ext(ext, source_isa);
802 		}
803 
804 		riscv_resolve_isa(source_isa, isainfo->isa, &this_hwcap, isa2hwcap);
805 		riscv_fill_cpu_vendor_ext(cpu_node, cpu);
806 
807 		of_node_put(cpu_node);
808 
809 		/*
810 		 * All "okay" harts should have same isa. Set HWCAP based on
811 		 * common capabilities of every "okay" hart, in case they don't.
812 		 */
813 		if (elf_hwcap)
814 			elf_hwcap &= this_hwcap;
815 		else
816 			elf_hwcap = this_hwcap;
817 
818 		if (bitmap_empty(riscv_isa, RISCV_ISA_EXT_MAX))
819 			bitmap_copy(riscv_isa, isainfo->isa, RISCV_ISA_EXT_MAX);
820 		else
821 			bitmap_and(riscv_isa, riscv_isa, isainfo->isa, RISCV_ISA_EXT_MAX);
822 
823 		riscv_fill_vendor_ext_list(cpu);
824 	}
825 
826 	if (bitmap_empty(riscv_isa, RISCV_ISA_EXT_MAX))
827 		return -ENOENT;
828 
829 	return 0;
830 }
831 
832 #ifdef CONFIG_RISCV_ISA_FALLBACK
833 bool __initdata riscv_isa_fallback = true;
834 #else
835 bool __initdata riscv_isa_fallback;
riscv_isa_fallback_setup(char * __unused)836 static int __init riscv_isa_fallback_setup(char *__unused)
837 {
838 	riscv_isa_fallback = true;
839 	return 1;
840 }
841 early_param("riscv_isa_fallback", riscv_isa_fallback_setup);
842 #endif
843 
riscv_fill_hwcap(void)844 void __init riscv_fill_hwcap(void)
845 {
846 	char print_str[NUM_ALPHA_EXTS + 1];
847 	unsigned long isa2hwcap[26] = {0};
848 	int i, j;
849 
850 	isa2hwcap['i' - 'a'] = COMPAT_HWCAP_ISA_I;
851 	isa2hwcap['m' - 'a'] = COMPAT_HWCAP_ISA_M;
852 	isa2hwcap['a' - 'a'] = COMPAT_HWCAP_ISA_A;
853 	isa2hwcap['f' - 'a'] = COMPAT_HWCAP_ISA_F;
854 	isa2hwcap['d' - 'a'] = COMPAT_HWCAP_ISA_D;
855 	isa2hwcap['c' - 'a'] = COMPAT_HWCAP_ISA_C;
856 	isa2hwcap['v' - 'a'] = COMPAT_HWCAP_ISA_V;
857 
858 	if (!acpi_disabled) {
859 		riscv_fill_hwcap_from_isa_string(isa2hwcap);
860 	} else {
861 		int ret = riscv_fill_hwcap_from_ext_list(isa2hwcap);
862 
863 		if (ret && riscv_isa_fallback) {
864 			pr_info("Falling back to deprecated \"riscv,isa\"\n");
865 			riscv_fill_hwcap_from_isa_string(isa2hwcap);
866 		}
867 	}
868 
869 	/*
870 	 * We don't support systems with F but without D, so mask those out
871 	 * here.
872 	 */
873 	if ((elf_hwcap & COMPAT_HWCAP_ISA_F) && !(elf_hwcap & COMPAT_HWCAP_ISA_D)) {
874 		pr_info("This kernel does not support systems with F but not D\n");
875 		elf_hwcap &= ~COMPAT_HWCAP_ISA_F;
876 	}
877 
878 	if (__riscv_isa_extension_available(NULL, RISCV_ISA_EXT_ZVE32X)) {
879 		/*
880 		 * This cannot fail when called on the boot hart
881 		 */
882 		riscv_v_setup_vsize();
883 	}
884 
885 	if (elf_hwcap & COMPAT_HWCAP_ISA_V) {
886 		/*
887 		 * ISA string in device tree might have 'v' flag, but
888 		 * CONFIG_RISCV_ISA_V is disabled in kernel.
889 		 * Clear V flag in elf_hwcap if CONFIG_RISCV_ISA_V is disabled.
890 		 */
891 		if (!IS_ENABLED(CONFIG_RISCV_ISA_V))
892 			elf_hwcap &= ~COMPAT_HWCAP_ISA_V;
893 	}
894 
895 	memset(print_str, 0, sizeof(print_str));
896 	for (i = 0, j = 0; i < NUM_ALPHA_EXTS; i++)
897 		if (riscv_isa[0] & BIT_MASK(i))
898 			print_str[j++] = (char)('a' + i);
899 	pr_info("riscv: base ISA extensions %s\n", print_str);
900 
901 	memset(print_str, 0, sizeof(print_str));
902 	for (i = 0, j = 0; i < NUM_ALPHA_EXTS; i++)
903 		if (elf_hwcap & BIT_MASK(i))
904 			print_str[j++] = (char)('a' + i);
905 	pr_info("riscv: ELF capabilities %s\n", print_str);
906 }
907 
riscv_get_elf_hwcap(void)908 unsigned long riscv_get_elf_hwcap(void)
909 {
910 	unsigned long hwcap;
911 
912 	hwcap = (elf_hwcap & ((1UL << RISCV_ISA_EXT_BASE) - 1));
913 
914 	if (!riscv_v_vstate_ctrl_user_allowed())
915 		hwcap &= ~COMPAT_HWCAP_ISA_V;
916 
917 	return hwcap;
918 }
919 
riscv_user_isa_enable(void)920 void riscv_user_isa_enable(void)
921 {
922 	if (riscv_cpu_has_extension_unlikely(smp_processor_id(), RISCV_ISA_EXT_ZICBOZ))
923 		csr_set(CSR_ENVCFG, ENVCFG_CBZE);
924 }
925 
926 #ifdef CONFIG_RISCV_ALTERNATIVE
927 /*
928  * Alternative patch sites consider 48 bits when determining when to patch
929  * the old instruction sequence with the new. These bits are broken into a
930  * 16-bit vendor ID and a 32-bit patch ID. A non-zero vendor ID means the
931  * patch site is for an erratum, identified by the 32-bit patch ID. When
932  * the vendor ID is zero, the patch site is for a cpufeature. cpufeatures
933  * further break down patch ID into two 16-bit numbers. The lower 16 bits
934  * are the cpufeature ID and the upper 16 bits are used for a value specific
935  * to the cpufeature and patch site. If the upper 16 bits are zero, then it
936  * implies no specific value is specified. cpufeatures that want to control
937  * patching on a per-site basis will provide non-zero values and implement
938  * checks here. The checks return true when patching should be done, and
939  * false otherwise.
940  */
riscv_cpufeature_patch_check(u16 id,u16 value)941 static bool riscv_cpufeature_patch_check(u16 id, u16 value)
942 {
943 	if (!value)
944 		return true;
945 
946 	switch (id) {
947 	case RISCV_ISA_EXT_ZICBOZ:
948 		/*
949 		 * Zicboz alternative applications provide the maximum
950 		 * supported block size order, or zero when it doesn't
951 		 * matter. If the current block size exceeds the maximum,
952 		 * then the alternative cannot be applied.
953 		 */
954 		return riscv_cboz_block_size <= (1U << value);
955 	}
956 
957 	return false;
958 }
959 
riscv_cpufeature_patch_func(struct alt_entry * begin,struct alt_entry * end,unsigned int stage)960 void __init_or_module riscv_cpufeature_patch_func(struct alt_entry *begin,
961 						  struct alt_entry *end,
962 						  unsigned int stage)
963 {
964 	struct alt_entry *alt;
965 	void *oldptr, *altptr;
966 	u16 id, value, vendor;
967 
968 	if (stage == RISCV_ALTERNATIVES_EARLY_BOOT)
969 		return;
970 
971 	for (alt = begin; alt < end; alt++) {
972 		id = PATCH_ID_CPUFEATURE_ID(alt->patch_id);
973 		vendor = PATCH_ID_CPUFEATURE_ID(alt->vendor_id);
974 
975 		/*
976 		 * Any alternative with a patch_id that is less than
977 		 * RISCV_ISA_EXT_MAX is interpreted as a standard extension.
978 		 *
979 		 * Any alternative with patch_id that is greater than or equal
980 		 * to RISCV_VENDOR_EXT_ALTERNATIVES_BASE is interpreted as a
981 		 * vendor extension.
982 		 */
983 		if (id < RISCV_ISA_EXT_MAX) {
984 			/*
985 			 * This patch should be treated as errata so skip
986 			 * processing here.
987 			 */
988 			if (alt->vendor_id != 0)
989 				continue;
990 
991 			if (!__riscv_isa_extension_available(NULL, id))
992 				continue;
993 
994 			value = PATCH_ID_CPUFEATURE_VALUE(alt->patch_id);
995 			if (!riscv_cpufeature_patch_check(id, value))
996 				continue;
997 		} else if (id >= RISCV_VENDOR_EXT_ALTERNATIVES_BASE) {
998 			if (!__riscv_isa_vendor_extension_available(VENDOR_EXT_ALL_CPUS, vendor,
999 								    id - RISCV_VENDOR_EXT_ALTERNATIVES_BASE))
1000 				continue;
1001 		} else {
1002 			WARN(1, "This extension id:%d is not in ISA extension list", id);
1003 			continue;
1004 		}
1005 
1006 		oldptr = ALT_OLD_PTR(alt);
1007 		altptr = ALT_ALT_PTR(alt);
1008 
1009 		mutex_lock(&text_mutex);
1010 		patch_text_nosync(oldptr, altptr, alt->alt_len);
1011 		riscv_alternative_fix_offsets(oldptr, alt->alt_len, oldptr - altptr);
1012 		mutex_unlock(&text_mutex);
1013 	}
1014 }
1015 #endif
1016