1 /*
2  *  Copyright (C) 2001 Ben. Herrenschmidt (benh@kernel.crashing.org)
3  *
4  *  Modifications for ppc64:
5  *      Copyright (C) 2003 Dave Engebretsen <engebret@us.ibm.com>
6  *
7  *  Copyright 2008 Michael Ellerman, IBM Corporation.
8  *
9  *  This program is free software; you can redistribute it and/or
10  *  modify it under the terms of the GNU General Public License
11  *  as published by the Free Software Foundation; either version
12  *  2 of the License, or (at your option) any later version.
13  */
14 
15 #include <linux/types.h>
16 #include <linux/jump_label.h>
17 #include <linux/kernel.h>
18 #include <linux/string.h>
19 #include <linux/init.h>
20 #include <linux/sched/mm.h>
21 #include <asm/cputable.h>
22 #include <asm/code-patching.h>
23 #include <asm/page.h>
24 #include <asm/sections.h>
25 #include <asm/setup.h>
26 #include <asm/security_features.h>
27 #include <asm/firmware.h>
28 
29 struct fixup_entry {
30 	unsigned long	mask;
31 	unsigned long	value;
32 	long		start_off;
33 	long		end_off;
34 	long		alt_start_off;
35 	long		alt_end_off;
36 };
37 
38 static unsigned int *calc_addr(struct fixup_entry *fcur, long offset)
39 {
40 	/*
41 	 * We store the offset to the code as a negative offset from
42 	 * the start of the alt_entry, to support the VDSO. This
43 	 * routine converts that back into an actual address.
44 	 */
45 	return (unsigned int *)((unsigned long)fcur + offset);
46 }
47 
48 static int patch_alt_instruction(unsigned int *src, unsigned int *dest,
49 				 unsigned int *alt_start, unsigned int *alt_end)
50 {
51 	unsigned int instr;
52 
53 	instr = *src;
54 
55 	if (instr_is_relative_branch(*src)) {
56 		unsigned int *target = (unsigned int *)branch_target(src);
57 
58 		/* Branch within the section doesn't need translating */
59 		if (target < alt_start || target > alt_end) {
60 			instr = translate_branch(dest, src);
61 			if (!instr)
62 				return 1;
63 		}
64 	}
65 
66 	raw_patch_instruction(dest, instr);
67 
68 	return 0;
69 }
70 
71 static int patch_feature_section(unsigned long value, struct fixup_entry *fcur)
72 {
73 	unsigned int *start, *end, *alt_start, *alt_end, *src, *dest;
74 
75 	start = calc_addr(fcur, fcur->start_off);
76 	end = calc_addr(fcur, fcur->end_off);
77 	alt_start = calc_addr(fcur, fcur->alt_start_off);
78 	alt_end = calc_addr(fcur, fcur->alt_end_off);
79 
80 	if ((alt_end - alt_start) > (end - start))
81 		return 1;
82 
83 	if ((value & fcur->mask) == fcur->value)
84 		return 0;
85 
86 	src = alt_start;
87 	dest = start;
88 
89 	for (; src < alt_end; src++, dest++) {
90 		if (patch_alt_instruction(src, dest, alt_start, alt_end))
91 			return 1;
92 	}
93 
94 	for (; dest < end; dest++)
95 		raw_patch_instruction(dest, PPC_INST_NOP);
96 
97 	return 0;
98 }
99 
100 void do_feature_fixups(unsigned long value, void *fixup_start, void *fixup_end)
101 {
102 	struct fixup_entry *fcur, *fend;
103 
104 	fcur = fixup_start;
105 	fend = fixup_end;
106 
107 	for (; fcur < fend; fcur++) {
108 		if (patch_feature_section(value, fcur)) {
109 			WARN_ON(1);
110 			printk("Unable to patch feature section at %p - %p" \
111 				" with %p - %p\n",
112 				calc_addr(fcur, fcur->start_off),
113 				calc_addr(fcur, fcur->end_off),
114 				calc_addr(fcur, fcur->alt_start_off),
115 				calc_addr(fcur, fcur->alt_end_off));
116 		}
117 	}
118 }
119 
120 #ifdef CONFIG_PPC_BOOK3S_64
121 void do_stf_entry_barrier_fixups(enum stf_barrier_type types)
122 {
123 	unsigned int instrs[3], *dest;
124 	long *start, *end;
125 	int i;
126 
127 	start = PTRRELOC(&__start___stf_entry_barrier_fixup),
128 	end = PTRRELOC(&__stop___stf_entry_barrier_fixup);
129 
130 	instrs[0] = 0x60000000; /* nop */
131 	instrs[1] = 0x60000000; /* nop */
132 	instrs[2] = 0x60000000; /* nop */
133 
134 	i = 0;
135 	if (types & STF_BARRIER_FALLBACK) {
136 		instrs[i++] = 0x7d4802a6; /* mflr r10		*/
137 		instrs[i++] = 0x60000000; /* branch patched below */
138 		instrs[i++] = 0x7d4803a6; /* mtlr r10		*/
139 	} else if (types & STF_BARRIER_EIEIO) {
140 		instrs[i++] = 0x7e0006ac; /* eieio + bit 6 hint */
141 	} else if (types & STF_BARRIER_SYNC_ORI) {
142 		instrs[i++] = 0x7c0004ac; /* hwsync		*/
143 		instrs[i++] = 0xe94d0000; /* ld r10,0(r13)	*/
144 		instrs[i++] = 0x63ff0000; /* ori 31,31,0 speculation barrier */
145 	}
146 
147 	for (i = 0; start < end; start++, i++) {
148 		dest = (void *)start + *start;
149 
150 		pr_devel("patching dest %lx\n", (unsigned long)dest);
151 
152 		patch_instruction(dest, instrs[0]);
153 
154 		if (types & STF_BARRIER_FALLBACK)
155 			patch_branch(dest + 1, (unsigned long)&stf_barrier_fallback,
156 				     BRANCH_SET_LINK);
157 		else
158 			patch_instruction(dest + 1, instrs[1]);
159 
160 		patch_instruction(dest + 2, instrs[2]);
161 	}
162 
163 	printk(KERN_DEBUG "stf-barrier: patched %d entry locations (%s barrier)\n", i,
164 		(types == STF_BARRIER_NONE)                  ? "no" :
165 		(types == STF_BARRIER_FALLBACK)              ? "fallback" :
166 		(types == STF_BARRIER_EIEIO)                 ? "eieio" :
167 		(types == (STF_BARRIER_SYNC_ORI))            ? "hwsync"
168 		                                           : "unknown");
169 }
170 
171 void do_stf_exit_barrier_fixups(enum stf_barrier_type types)
172 {
173 	unsigned int instrs[6], *dest;
174 	long *start, *end;
175 	int i;
176 
177 	start = PTRRELOC(&__start___stf_exit_barrier_fixup),
178 	end = PTRRELOC(&__stop___stf_exit_barrier_fixup);
179 
180 	instrs[0] = 0x60000000; /* nop */
181 	instrs[1] = 0x60000000; /* nop */
182 	instrs[2] = 0x60000000; /* nop */
183 	instrs[3] = 0x60000000; /* nop */
184 	instrs[4] = 0x60000000; /* nop */
185 	instrs[5] = 0x60000000; /* nop */
186 
187 	i = 0;
188 	if (types & STF_BARRIER_FALLBACK || types & STF_BARRIER_SYNC_ORI) {
189 		if (cpu_has_feature(CPU_FTR_HVMODE)) {
190 			instrs[i++] = 0x7db14ba6; /* mtspr 0x131, r13 (HSPRG1) */
191 			instrs[i++] = 0x7db04aa6; /* mfspr r13, 0x130 (HSPRG0) */
192 		} else {
193 			instrs[i++] = 0x7db243a6; /* mtsprg 2,r13	*/
194 			instrs[i++] = 0x7db142a6; /* mfsprg r13,1    */
195 	        }
196 		instrs[i++] = 0x7c0004ac; /* hwsync		*/
197 		instrs[i++] = 0xe9ad0000; /* ld r13,0(r13)	*/
198 		instrs[i++] = 0x63ff0000; /* ori 31,31,0 speculation barrier */
199 		if (cpu_has_feature(CPU_FTR_HVMODE)) {
200 			instrs[i++] = 0x7db14aa6; /* mfspr r13, 0x131 (HSPRG1) */
201 		} else {
202 			instrs[i++] = 0x7db242a6; /* mfsprg r13,2 */
203 		}
204 	} else if (types & STF_BARRIER_EIEIO) {
205 		instrs[i++] = 0x7e0006ac; /* eieio + bit 6 hint */
206 	}
207 
208 	for (i = 0; start < end; start++, i++) {
209 		dest = (void *)start + *start;
210 
211 		pr_devel("patching dest %lx\n", (unsigned long)dest);
212 
213 		patch_instruction(dest, instrs[0]);
214 		patch_instruction(dest + 1, instrs[1]);
215 		patch_instruction(dest + 2, instrs[2]);
216 		patch_instruction(dest + 3, instrs[3]);
217 		patch_instruction(dest + 4, instrs[4]);
218 		patch_instruction(dest + 5, instrs[5]);
219 	}
220 	printk(KERN_DEBUG "stf-barrier: patched %d exit locations (%s barrier)\n", i,
221 		(types == STF_BARRIER_NONE)                  ? "no" :
222 		(types == STF_BARRIER_FALLBACK)              ? "fallback" :
223 		(types == STF_BARRIER_EIEIO)                 ? "eieio" :
224 		(types == (STF_BARRIER_SYNC_ORI))            ? "hwsync"
225 		                                           : "unknown");
226 }
227 
228 
229 void do_stf_barrier_fixups(enum stf_barrier_type types)
230 {
231 	do_stf_entry_barrier_fixups(types);
232 	do_stf_exit_barrier_fixups(types);
233 }
234 
235 void do_rfi_flush_fixups(enum l1d_flush_type types)
236 {
237 	unsigned int instrs[3], *dest;
238 	long *start, *end;
239 	int i;
240 
241 	start = PTRRELOC(&__start___rfi_flush_fixup),
242 	end = PTRRELOC(&__stop___rfi_flush_fixup);
243 
244 	instrs[0] = 0x60000000; /* nop */
245 	instrs[1] = 0x60000000; /* nop */
246 	instrs[2] = 0x60000000; /* nop */
247 
248 	if (types & L1D_FLUSH_FALLBACK)
249 		/* b .+16 to fallback flush */
250 		instrs[0] = 0x48000010;
251 
252 	i = 0;
253 	if (types & L1D_FLUSH_ORI) {
254 		instrs[i++] = 0x63ff0000; /* ori 31,31,0 speculation barrier */
255 		instrs[i++] = 0x63de0000; /* ori 30,30,0 L1d flush*/
256 	}
257 
258 	if (types & L1D_FLUSH_MTTRIG)
259 		instrs[i++] = 0x7c12dba6; /* mtspr TRIG2,r0 (SPR #882) */
260 
261 	for (i = 0; start < end; start++, i++) {
262 		dest = (void *)start + *start;
263 
264 		pr_devel("patching dest %lx\n", (unsigned long)dest);
265 
266 		patch_instruction(dest, instrs[0]);
267 		patch_instruction(dest + 1, instrs[1]);
268 		patch_instruction(dest + 2, instrs[2]);
269 	}
270 
271 	printk(KERN_DEBUG "rfi-flush: patched %d locations (%s flush)\n", i,
272 		(types == L1D_FLUSH_NONE)       ? "no" :
273 		(types == L1D_FLUSH_FALLBACK)   ? "fallback displacement" :
274 		(types &  L1D_FLUSH_ORI)        ? (types & L1D_FLUSH_MTTRIG)
275 							? "ori+mttrig type"
276 							: "ori type" :
277 		(types &  L1D_FLUSH_MTTRIG)     ? "mttrig type"
278 						: "unknown");
279 }
280 
281 void do_barrier_nospec_fixups_range(bool enable, void *fixup_start, void *fixup_end)
282 {
283 	unsigned int instr, *dest;
284 	long *start, *end;
285 	int i;
286 
287 	start = fixup_start;
288 	end = fixup_end;
289 
290 	instr = 0x60000000; /* nop */
291 
292 	if (enable) {
293 		pr_info("barrier-nospec: using ORI speculation barrier\n");
294 		instr = 0x63ff0000; /* ori 31,31,0 speculation barrier */
295 	}
296 
297 	for (i = 0; start < end; start++, i++) {
298 		dest = (void *)start + *start;
299 
300 		pr_devel("patching dest %lx\n", (unsigned long)dest);
301 		patch_instruction(dest, instr);
302 	}
303 
304 	printk(KERN_DEBUG "barrier-nospec: patched %d locations\n", i);
305 }
306 
307 void do_barrier_nospec_fixups(bool enable)
308 {
309 	void *start, *end;
310 
311 	start = PTRRELOC(&__start___barrier_nospec_fixup),
312 	end = PTRRELOC(&__stop___barrier_nospec_fixup);
313 
314 	do_barrier_nospec_fixups_range(enable, start, end);
315 }
316 
317 #endif /* CONFIG_PPC_BOOK3S_64 */
318 
319 void do_lwsync_fixups(unsigned long value, void *fixup_start, void *fixup_end)
320 {
321 	long *start, *end;
322 	unsigned int *dest;
323 
324 	if (!(value & CPU_FTR_LWSYNC))
325 		return ;
326 
327 	start = fixup_start;
328 	end = fixup_end;
329 
330 	for (; start < end; start++) {
331 		dest = (void *)start + *start;
332 		raw_patch_instruction(dest, PPC_INST_LWSYNC);
333 	}
334 }
335 
336 static void do_final_fixups(void)
337 {
338 #if defined(CONFIG_PPC64) && defined(CONFIG_RELOCATABLE)
339 	int *src, *dest;
340 	unsigned long length;
341 
342 	if (PHYSICAL_START == 0)
343 		return;
344 
345 	src = (int *)(KERNELBASE + PHYSICAL_START);
346 	dest = (int *)KERNELBASE;
347 	length = (__end_interrupts - _stext) / sizeof(int);
348 
349 	while (length--) {
350 		raw_patch_instruction(dest, *src);
351 		src++;
352 		dest++;
353 	}
354 #endif
355 }
356 
357 static unsigned long __initdata saved_cpu_features;
358 static unsigned int __initdata saved_mmu_features;
359 #ifdef CONFIG_PPC64
360 static unsigned long __initdata saved_firmware_features;
361 #endif
362 
363 void __init apply_feature_fixups(void)
364 {
365 	struct cpu_spec *spec = PTRRELOC(*PTRRELOC(&cur_cpu_spec));
366 
367 	*PTRRELOC(&saved_cpu_features) = spec->cpu_features;
368 	*PTRRELOC(&saved_mmu_features) = spec->mmu_features;
369 
370 	/*
371 	 * Apply the CPU-specific and firmware specific fixups to kernel text
372 	 * (nop out sections not relevant to this CPU or this firmware).
373 	 */
374 	do_feature_fixups(spec->cpu_features,
375 			  PTRRELOC(&__start___ftr_fixup),
376 			  PTRRELOC(&__stop___ftr_fixup));
377 
378 	do_feature_fixups(spec->mmu_features,
379 			  PTRRELOC(&__start___mmu_ftr_fixup),
380 			  PTRRELOC(&__stop___mmu_ftr_fixup));
381 
382 	do_lwsync_fixups(spec->cpu_features,
383 			 PTRRELOC(&__start___lwsync_fixup),
384 			 PTRRELOC(&__stop___lwsync_fixup));
385 
386 #ifdef CONFIG_PPC64
387 	saved_firmware_features = powerpc_firmware_features;
388 	do_feature_fixups(powerpc_firmware_features,
389 			  &__start___fw_ftr_fixup, &__stop___fw_ftr_fixup);
390 #endif
391 	do_final_fixups();
392 }
393 
394 void __init setup_feature_keys(void)
395 {
396 	/*
397 	 * Initialise jump label. This causes all the cpu/mmu_has_feature()
398 	 * checks to take on their correct polarity based on the current set of
399 	 * CPU/MMU features.
400 	 */
401 	jump_label_init();
402 	cpu_feature_keys_init();
403 	mmu_feature_keys_init();
404 }
405 
406 static int __init check_features(void)
407 {
408 	WARN(saved_cpu_features != cur_cpu_spec->cpu_features,
409 	     "CPU features changed after feature patching!\n");
410 	WARN(saved_mmu_features != cur_cpu_spec->mmu_features,
411 	     "MMU features changed after feature patching!\n");
412 #ifdef CONFIG_PPC64
413 	WARN(saved_firmware_features != powerpc_firmware_features,
414 	     "Firmware features changed after feature patching!\n");
415 #endif
416 
417 	return 0;
418 }
419 late_initcall(check_features);
420 
421 #ifdef CONFIG_FTR_FIXUP_SELFTEST
422 
423 #define check(x)	\
424 	if (!(x)) printk("feature-fixups: test failed at line %d\n", __LINE__);
425 
426 /* This must be after the text it fixes up, vmlinux.lds.S enforces that atm */
427 static struct fixup_entry fixup;
428 
429 static long calc_offset(struct fixup_entry *entry, unsigned int *p)
430 {
431 	return (unsigned long)p - (unsigned long)entry;
432 }
433 
434 static void test_basic_patching(void)
435 {
436 	extern unsigned int ftr_fixup_test1[];
437 	extern unsigned int end_ftr_fixup_test1[];
438 	extern unsigned int ftr_fixup_test1_orig[];
439 	extern unsigned int ftr_fixup_test1_expected[];
440 	int size = 4 * (end_ftr_fixup_test1 - ftr_fixup_test1);
441 
442 	fixup.value = fixup.mask = 8;
443 	fixup.start_off = calc_offset(&fixup, ftr_fixup_test1 + 1);
444 	fixup.end_off = calc_offset(&fixup, ftr_fixup_test1 + 2);
445 	fixup.alt_start_off = fixup.alt_end_off = 0;
446 
447 	/* Sanity check */
448 	check(memcmp(ftr_fixup_test1, ftr_fixup_test1_orig, size) == 0);
449 
450 	/* Check we don't patch if the value matches */
451 	patch_feature_section(8, &fixup);
452 	check(memcmp(ftr_fixup_test1, ftr_fixup_test1_orig, size) == 0);
453 
454 	/* Check we do patch if the value doesn't match */
455 	patch_feature_section(0, &fixup);
456 	check(memcmp(ftr_fixup_test1, ftr_fixup_test1_expected, size) == 0);
457 
458 	/* Check we do patch if the mask doesn't match */
459 	memcpy(ftr_fixup_test1, ftr_fixup_test1_orig, size);
460 	check(memcmp(ftr_fixup_test1, ftr_fixup_test1_orig, size) == 0);
461 	patch_feature_section(~8, &fixup);
462 	check(memcmp(ftr_fixup_test1, ftr_fixup_test1_expected, size) == 0);
463 }
464 
465 static void test_alternative_patching(void)
466 {
467 	extern unsigned int ftr_fixup_test2[];
468 	extern unsigned int end_ftr_fixup_test2[];
469 	extern unsigned int ftr_fixup_test2_orig[];
470 	extern unsigned int ftr_fixup_test2_alt[];
471 	extern unsigned int ftr_fixup_test2_expected[];
472 	int size = 4 * (end_ftr_fixup_test2 - ftr_fixup_test2);
473 
474 	fixup.value = fixup.mask = 0xF;
475 	fixup.start_off = calc_offset(&fixup, ftr_fixup_test2 + 1);
476 	fixup.end_off = calc_offset(&fixup, ftr_fixup_test2 + 2);
477 	fixup.alt_start_off = calc_offset(&fixup, ftr_fixup_test2_alt);
478 	fixup.alt_end_off = calc_offset(&fixup, ftr_fixup_test2_alt + 1);
479 
480 	/* Sanity check */
481 	check(memcmp(ftr_fixup_test2, ftr_fixup_test2_orig, size) == 0);
482 
483 	/* Check we don't patch if the value matches */
484 	patch_feature_section(0xF, &fixup);
485 	check(memcmp(ftr_fixup_test2, ftr_fixup_test2_orig, size) == 0);
486 
487 	/* Check we do patch if the value doesn't match */
488 	patch_feature_section(0, &fixup);
489 	check(memcmp(ftr_fixup_test2, ftr_fixup_test2_expected, size) == 0);
490 
491 	/* Check we do patch if the mask doesn't match */
492 	memcpy(ftr_fixup_test2, ftr_fixup_test2_orig, size);
493 	check(memcmp(ftr_fixup_test2, ftr_fixup_test2_orig, size) == 0);
494 	patch_feature_section(~0xF, &fixup);
495 	check(memcmp(ftr_fixup_test2, ftr_fixup_test2_expected, size) == 0);
496 }
497 
498 static void test_alternative_case_too_big(void)
499 {
500 	extern unsigned int ftr_fixup_test3[];
501 	extern unsigned int end_ftr_fixup_test3[];
502 	extern unsigned int ftr_fixup_test3_orig[];
503 	extern unsigned int ftr_fixup_test3_alt[];
504 	int size = 4 * (end_ftr_fixup_test3 - ftr_fixup_test3);
505 
506 	fixup.value = fixup.mask = 0xC;
507 	fixup.start_off = calc_offset(&fixup, ftr_fixup_test3 + 1);
508 	fixup.end_off = calc_offset(&fixup, ftr_fixup_test3 + 2);
509 	fixup.alt_start_off = calc_offset(&fixup, ftr_fixup_test3_alt);
510 	fixup.alt_end_off = calc_offset(&fixup, ftr_fixup_test3_alt + 2);
511 
512 	/* Sanity check */
513 	check(memcmp(ftr_fixup_test3, ftr_fixup_test3_orig, size) == 0);
514 
515 	/* Expect nothing to be patched, and the error returned to us */
516 	check(patch_feature_section(0xF, &fixup) == 1);
517 	check(memcmp(ftr_fixup_test3, ftr_fixup_test3_orig, size) == 0);
518 	check(patch_feature_section(0, &fixup) == 1);
519 	check(memcmp(ftr_fixup_test3, ftr_fixup_test3_orig, size) == 0);
520 	check(patch_feature_section(~0xF, &fixup) == 1);
521 	check(memcmp(ftr_fixup_test3, ftr_fixup_test3_orig, size) == 0);
522 }
523 
524 static void test_alternative_case_too_small(void)
525 {
526 	extern unsigned int ftr_fixup_test4[];
527 	extern unsigned int end_ftr_fixup_test4[];
528 	extern unsigned int ftr_fixup_test4_orig[];
529 	extern unsigned int ftr_fixup_test4_alt[];
530 	extern unsigned int ftr_fixup_test4_expected[];
531 	int size = 4 * (end_ftr_fixup_test4 - ftr_fixup_test4);
532 	unsigned long flag;
533 
534 	/* Check a high-bit flag */
535 	flag = 1UL << ((sizeof(unsigned long) - 1) * 8);
536 	fixup.value = fixup.mask = flag;
537 	fixup.start_off = calc_offset(&fixup, ftr_fixup_test4 + 1);
538 	fixup.end_off = calc_offset(&fixup, ftr_fixup_test4 + 5);
539 	fixup.alt_start_off = calc_offset(&fixup, ftr_fixup_test4_alt);
540 	fixup.alt_end_off = calc_offset(&fixup, ftr_fixup_test4_alt + 2);
541 
542 	/* Sanity check */
543 	check(memcmp(ftr_fixup_test4, ftr_fixup_test4_orig, size) == 0);
544 
545 	/* Check we don't patch if the value matches */
546 	patch_feature_section(flag, &fixup);
547 	check(memcmp(ftr_fixup_test4, ftr_fixup_test4_orig, size) == 0);
548 
549 	/* Check we do patch if the value doesn't match */
550 	patch_feature_section(0, &fixup);
551 	check(memcmp(ftr_fixup_test4, ftr_fixup_test4_expected, size) == 0);
552 
553 	/* Check we do patch if the mask doesn't match */
554 	memcpy(ftr_fixup_test4, ftr_fixup_test4_orig, size);
555 	check(memcmp(ftr_fixup_test4, ftr_fixup_test4_orig, size) == 0);
556 	patch_feature_section(~flag, &fixup);
557 	check(memcmp(ftr_fixup_test4, ftr_fixup_test4_expected, size) == 0);
558 }
559 
560 static void test_alternative_case_with_branch(void)
561 {
562 	extern unsigned int ftr_fixup_test5[];
563 	extern unsigned int end_ftr_fixup_test5[];
564 	extern unsigned int ftr_fixup_test5_expected[];
565 	int size = 4 * (end_ftr_fixup_test5 - ftr_fixup_test5);
566 
567 	check(memcmp(ftr_fixup_test5, ftr_fixup_test5_expected, size) == 0);
568 }
569 
570 static void test_alternative_case_with_external_branch(void)
571 {
572 	extern unsigned int ftr_fixup_test6[];
573 	extern unsigned int end_ftr_fixup_test6[];
574 	extern unsigned int ftr_fixup_test6_expected[];
575 	int size = 4 * (end_ftr_fixup_test6 - ftr_fixup_test6);
576 
577 	check(memcmp(ftr_fixup_test6, ftr_fixup_test6_expected, size) == 0);
578 }
579 
580 static void test_alternative_case_with_branch_to_end(void)
581 {
582 	extern unsigned int ftr_fixup_test7[];
583 	extern unsigned int end_ftr_fixup_test7[];
584 	extern unsigned int ftr_fixup_test7_expected[];
585 	int size = 4 * (end_ftr_fixup_test7 - ftr_fixup_test7);
586 
587 	check(memcmp(ftr_fixup_test7, ftr_fixup_test7_expected, size) == 0);
588 }
589 
590 static void test_cpu_macros(void)
591 {
592 	extern u8 ftr_fixup_test_FTR_macros[];
593 	extern u8 ftr_fixup_test_FTR_macros_expected[];
594 	unsigned long size = ftr_fixup_test_FTR_macros_expected -
595 			     ftr_fixup_test_FTR_macros;
596 
597 	/* The fixups have already been done for us during boot */
598 	check(memcmp(ftr_fixup_test_FTR_macros,
599 		     ftr_fixup_test_FTR_macros_expected, size) == 0);
600 }
601 
602 static void test_fw_macros(void)
603 {
604 #ifdef CONFIG_PPC64
605 	extern u8 ftr_fixup_test_FW_FTR_macros[];
606 	extern u8 ftr_fixup_test_FW_FTR_macros_expected[];
607 	unsigned long size = ftr_fixup_test_FW_FTR_macros_expected -
608 			     ftr_fixup_test_FW_FTR_macros;
609 
610 	/* The fixups have already been done for us during boot */
611 	check(memcmp(ftr_fixup_test_FW_FTR_macros,
612 		     ftr_fixup_test_FW_FTR_macros_expected, size) == 0);
613 #endif
614 }
615 
616 static void test_lwsync_macros(void)
617 {
618 	extern u8 lwsync_fixup_test[];
619 	extern u8 end_lwsync_fixup_test[];
620 	extern u8 lwsync_fixup_test_expected_LWSYNC[];
621 	extern u8 lwsync_fixup_test_expected_SYNC[];
622 	unsigned long size = end_lwsync_fixup_test -
623 			     lwsync_fixup_test;
624 
625 	/* The fixups have already been done for us during boot */
626 	if (cur_cpu_spec->cpu_features & CPU_FTR_LWSYNC) {
627 		check(memcmp(lwsync_fixup_test,
628 			     lwsync_fixup_test_expected_LWSYNC, size) == 0);
629 	} else {
630 		check(memcmp(lwsync_fixup_test,
631 			     lwsync_fixup_test_expected_SYNC, size) == 0);
632 	}
633 }
634 
635 static int __init test_feature_fixups(void)
636 {
637 	printk(KERN_DEBUG "Running feature fixup self-tests ...\n");
638 
639 	test_basic_patching();
640 	test_alternative_patching();
641 	test_alternative_case_too_big();
642 	test_alternative_case_too_small();
643 	test_alternative_case_with_branch();
644 	test_alternative_case_with_external_branch();
645 	test_alternative_case_with_branch_to_end();
646 	test_cpu_macros();
647 	test_fw_macros();
648 	test_lwsync_macros();
649 
650 	return 0;
651 }
652 late_initcall(test_feature_fixups);
653 
654 #endif /* CONFIG_FTR_FIXUP_SELFTEST */
655