xref: /linux-6.15/arch/powerpc/lib/code-patching.c (revision ccc8fcf7)
12874c5fdSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-or-later
2aaddd3eaSMichael Ellerman /*
3aaddd3eaSMichael Ellerman  *  Copyright 2008 Michael Ellerman, IBM Corporation.
4aaddd3eaSMichael Ellerman  */
5aaddd3eaSMichael Ellerman 
6aaddd3eaSMichael Ellerman #include <linux/kernel.h>
771f6e58eSNaveen N. Rao #include <linux/kprobes.h>
8ae0dc736SMichael Ellerman #include <linux/vmalloc.h>
9ae0dc736SMichael Ellerman #include <linux/init.h>
1027ac792cSAndrea Righi #include <linux/mm.h>
1137bc3e5fSBalbir Singh #include <linux/cpuhotplug.h>
1237bc3e5fSBalbir Singh #include <linux/slab.h>
137c0f6ba6SLinus Torvalds #include <linux/uaccess.h>
14aaddd3eaSMichael Ellerman 
1537bc3e5fSBalbir Singh #include <asm/tlbflush.h>
1637bc3e5fSBalbir Singh #include <asm/page.h>
1737bc3e5fSBalbir Singh #include <asm/code-patching.h>
18252eb558SChristophe Leroy #include <asm/setup.h>
1975346251SJordan Niethe #include <asm/inst.h>
20aaddd3eaSMichael Ellerman 
2194afd069SJordan Niethe static int __patch_instruction(struct ppc_inst *exec_addr, struct ppc_inst instr,
2294afd069SJordan Niethe 			       struct ppc_inst *patch_addr)
23aaddd3eaSMichael Ellerman {
24ef296729SRussell Currey 	int err = 0;
25b6e37968SSteven Rostedt 
26650b55b7SJordan Niethe 	if (!ppc_inst_prefixed(instr)) {
2794afd069SJordan Niethe 		__put_user_asm(ppc_inst_val(instr), patch_addr, err, "stw");
28650b55b7SJordan Niethe 	} else {
2916ef9767SMichael Ellerman 		__put_user_asm(ppc_inst_as_u64(instr), patch_addr, err, "std");
30650b55b7SJordan Niethe 	}
31650b55b7SJordan Niethe 
32b6e37968SSteven Rostedt 	if (err)
33b6e37968SSteven Rostedt 		return err;
3437bc3e5fSBalbir Singh 
358cf4c057SChristophe Leroy 	asm ("dcbst 0, %0; sync; icbi 0,%1; sync; isync" :: "r" (patch_addr),
368cf4c057SChristophe Leroy 							    "r" (exec_addr));
3737bc3e5fSBalbir Singh 
38b6e37968SSteven Rostedt 	return 0;
39aaddd3eaSMichael Ellerman }
40aaddd3eaSMichael Ellerman 
4194afd069SJordan Niethe int raw_patch_instruction(struct ppc_inst *addr, struct ppc_inst instr)
428cf4c057SChristophe Leroy {
438cf4c057SChristophe Leroy 	return __patch_instruction(addr, instr, addr);
448cf4c057SChristophe Leroy }
458cf4c057SChristophe Leroy 
4637bc3e5fSBalbir Singh #ifdef CONFIG_STRICT_KERNEL_RWX
4737bc3e5fSBalbir Singh static DEFINE_PER_CPU(struct vm_struct *, text_poke_area);
4837bc3e5fSBalbir Singh 
4937bc3e5fSBalbir Singh static int text_area_cpu_up(unsigned int cpu)
5037bc3e5fSBalbir Singh {
5137bc3e5fSBalbir Singh 	struct vm_struct *area;
5237bc3e5fSBalbir Singh 
5337bc3e5fSBalbir Singh 	area = get_vm_area(PAGE_SIZE, VM_ALLOC);
5437bc3e5fSBalbir Singh 	if (!area) {
5537bc3e5fSBalbir Singh 		WARN_ONCE(1, "Failed to create text area for cpu %d\n",
5637bc3e5fSBalbir Singh 			cpu);
5737bc3e5fSBalbir Singh 		return -1;
5837bc3e5fSBalbir Singh 	}
5937bc3e5fSBalbir Singh 	this_cpu_write(text_poke_area, area);
6037bc3e5fSBalbir Singh 
6137bc3e5fSBalbir Singh 	return 0;
6237bc3e5fSBalbir Singh }
6337bc3e5fSBalbir Singh 
6437bc3e5fSBalbir Singh static int text_area_cpu_down(unsigned int cpu)
6537bc3e5fSBalbir Singh {
6637bc3e5fSBalbir Singh 	free_vm_area(this_cpu_read(text_poke_area));
6737bc3e5fSBalbir Singh 	return 0;
6837bc3e5fSBalbir Singh }
6937bc3e5fSBalbir Singh 
7037bc3e5fSBalbir Singh /*
7137bc3e5fSBalbir Singh  * Run as a late init call. This allows all the boot time patching to be done
7237bc3e5fSBalbir Singh  * simply by patching the code, and then we're called here prior to
7337bc3e5fSBalbir Singh  * mark_rodata_ro(), which happens after all init calls are run. Although
7437bc3e5fSBalbir Singh  * BUG_ON() is rude, in this case it should only happen if ENOMEM, and we judge
7537bc3e5fSBalbir Singh  * it as being preferable to a kernel that will crash later when someone tries
7637bc3e5fSBalbir Singh  * to use patch_instruction().
7737bc3e5fSBalbir Singh  */
7837bc3e5fSBalbir Singh static int __init setup_text_poke_area(void)
7937bc3e5fSBalbir Singh {
8037bc3e5fSBalbir Singh 	BUG_ON(!cpuhp_setup_state(CPUHP_AP_ONLINE_DYN,
8137bc3e5fSBalbir Singh 		"powerpc/text_poke:online", text_area_cpu_up,
8237bc3e5fSBalbir Singh 		text_area_cpu_down));
8337bc3e5fSBalbir Singh 
8437bc3e5fSBalbir Singh 	return 0;
8537bc3e5fSBalbir Singh }
8637bc3e5fSBalbir Singh late_initcall(setup_text_poke_area);
8737bc3e5fSBalbir Singh 
8837bc3e5fSBalbir Singh /*
8937bc3e5fSBalbir Singh  * This can be called for kernel text or a module.
9037bc3e5fSBalbir Singh  */
9137bc3e5fSBalbir Singh static int map_patch_area(void *addr, unsigned long text_poke_addr)
9237bc3e5fSBalbir Singh {
9337bc3e5fSBalbir Singh 	unsigned long pfn;
9437bc3e5fSBalbir Singh 	int err;
9537bc3e5fSBalbir Singh 
96*ccc8fcf7SChristophe Leroy 	if (is_vmalloc_or_module_addr(addr))
9737bc3e5fSBalbir Singh 		pfn = vmalloc_to_pfn(addr);
9837bc3e5fSBalbir Singh 	else
9937bc3e5fSBalbir Singh 		pfn = __pa_symbol(addr) >> PAGE_SHIFT;
10037bc3e5fSBalbir Singh 
101c766ee72SChristophe Leroy 	err = map_kernel_page(text_poke_addr, (pfn << PAGE_SHIFT), PAGE_KERNEL);
10237bc3e5fSBalbir Singh 
10337bc3e5fSBalbir Singh 	pr_devel("Mapped addr %lx with pfn %lx:%d\n", text_poke_addr, pfn, err);
10437bc3e5fSBalbir Singh 	if (err)
10537bc3e5fSBalbir Singh 		return -1;
10637bc3e5fSBalbir Singh 
10737bc3e5fSBalbir Singh 	return 0;
10837bc3e5fSBalbir Singh }
10937bc3e5fSBalbir Singh 
11037bc3e5fSBalbir Singh static inline int unmap_patch_area(unsigned long addr)
11137bc3e5fSBalbir Singh {
11237bc3e5fSBalbir Singh 	pte_t *ptep;
11337bc3e5fSBalbir Singh 	pmd_t *pmdp;
11437bc3e5fSBalbir Singh 	pud_t *pudp;
1152fb47060SMike Rapoport 	p4d_t *p4dp;
11637bc3e5fSBalbir Singh 	pgd_t *pgdp;
11737bc3e5fSBalbir Singh 
11837bc3e5fSBalbir Singh 	pgdp = pgd_offset_k(addr);
11937bc3e5fSBalbir Singh 	if (unlikely(!pgdp))
12037bc3e5fSBalbir Singh 		return -EINVAL;
12137bc3e5fSBalbir Singh 
1222fb47060SMike Rapoport 	p4dp = p4d_offset(pgdp, addr);
1232fb47060SMike Rapoport 	if (unlikely(!p4dp))
1242fb47060SMike Rapoport 		return -EINVAL;
1252fb47060SMike Rapoport 
1262fb47060SMike Rapoport 	pudp = pud_offset(p4dp, addr);
12737bc3e5fSBalbir Singh 	if (unlikely(!pudp))
12837bc3e5fSBalbir Singh 		return -EINVAL;
12937bc3e5fSBalbir Singh 
13037bc3e5fSBalbir Singh 	pmdp = pmd_offset(pudp, addr);
13137bc3e5fSBalbir Singh 	if (unlikely(!pmdp))
13237bc3e5fSBalbir Singh 		return -EINVAL;
13337bc3e5fSBalbir Singh 
13437bc3e5fSBalbir Singh 	ptep = pte_offset_kernel(pmdp, addr);
13537bc3e5fSBalbir Singh 	if (unlikely(!ptep))
13637bc3e5fSBalbir Singh 		return -EINVAL;
13737bc3e5fSBalbir Singh 
13837bc3e5fSBalbir Singh 	pr_devel("clearing mm %p, pte %p, addr %lx\n", &init_mm, ptep, addr);
13937bc3e5fSBalbir Singh 
14037bc3e5fSBalbir Singh 	/*
14137bc3e5fSBalbir Singh 	 * In hash, pte_clear flushes the tlb, in radix, we have to
14237bc3e5fSBalbir Singh 	 */
14337bc3e5fSBalbir Singh 	pte_clear(&init_mm, addr, ptep);
14437bc3e5fSBalbir Singh 	flush_tlb_kernel_range(addr, addr + PAGE_SIZE);
14537bc3e5fSBalbir Singh 
14637bc3e5fSBalbir Singh 	return 0;
14737bc3e5fSBalbir Singh }
14837bc3e5fSBalbir Singh 
14994afd069SJordan Niethe static int do_patch_instruction(struct ppc_inst *addr, struct ppc_inst instr)
15037bc3e5fSBalbir Singh {
15137bc3e5fSBalbir Singh 	int err;
15294afd069SJordan Niethe 	struct ppc_inst *patch_addr = NULL;
15337bc3e5fSBalbir Singh 	unsigned long flags;
15437bc3e5fSBalbir Singh 	unsigned long text_poke_addr;
15537bc3e5fSBalbir Singh 	unsigned long kaddr = (unsigned long)addr;
15637bc3e5fSBalbir Singh 
15737bc3e5fSBalbir Singh 	/*
15837bc3e5fSBalbir Singh 	 * During early early boot patch_instruction is called
15937bc3e5fSBalbir Singh 	 * when text_poke_area is not ready, but we still need
16037bc3e5fSBalbir Singh 	 * to allow patching. We just do the plain old patching
16137bc3e5fSBalbir Singh 	 */
1628183d99fSChristophe Leroy 	if (!this_cpu_read(text_poke_area))
1638cf4c057SChristophe Leroy 		return raw_patch_instruction(addr, instr);
16437bc3e5fSBalbir Singh 
16537bc3e5fSBalbir Singh 	local_irq_save(flags);
16637bc3e5fSBalbir Singh 
16737bc3e5fSBalbir Singh 	text_poke_addr = (unsigned long)__this_cpu_read(text_poke_area)->addr;
16837bc3e5fSBalbir Singh 	if (map_patch_area(addr, text_poke_addr)) {
16937bc3e5fSBalbir Singh 		err = -1;
17037bc3e5fSBalbir Singh 		goto out;
17137bc3e5fSBalbir Singh 	}
17237bc3e5fSBalbir Singh 
17394afd069SJordan Niethe 	patch_addr = (struct ppc_inst *)(text_poke_addr + (kaddr & ~PAGE_MASK));
17437bc3e5fSBalbir Singh 
1758cf4c057SChristophe Leroy 	__patch_instruction(addr, instr, patch_addr);
17637bc3e5fSBalbir Singh 
17737bc3e5fSBalbir Singh 	err = unmap_patch_area(text_poke_addr);
17837bc3e5fSBalbir Singh 	if (err)
17937bc3e5fSBalbir Singh 		pr_warn("failed to unmap %lx\n", text_poke_addr);
18037bc3e5fSBalbir Singh 
18137bc3e5fSBalbir Singh out:
18237bc3e5fSBalbir Singh 	local_irq_restore(flags);
18337bc3e5fSBalbir Singh 
18437bc3e5fSBalbir Singh 	return err;
18537bc3e5fSBalbir Singh }
18637bc3e5fSBalbir Singh #else /* !CONFIG_STRICT_KERNEL_RWX */
18737bc3e5fSBalbir Singh 
18894afd069SJordan Niethe static int do_patch_instruction(struct ppc_inst *addr, struct ppc_inst instr)
18937bc3e5fSBalbir Singh {
1908cf4c057SChristophe Leroy 	return raw_patch_instruction(addr, instr);
19137bc3e5fSBalbir Singh }
19237bc3e5fSBalbir Singh 
19337bc3e5fSBalbir Singh #endif /* CONFIG_STRICT_KERNEL_RWX */
194b45ba4a5SChristophe Leroy 
19594afd069SJordan Niethe int patch_instruction(struct ppc_inst *addr, struct ppc_inst instr)
196b45ba4a5SChristophe Leroy {
197b45ba4a5SChristophe Leroy 	/* Make sure we aren't patching a freed init section */
198b45ba4a5SChristophe Leroy 	if (init_mem_is_free && init_section_contains(addr, 4)) {
199b45ba4a5SChristophe Leroy 		pr_debug("Skipping init section patching addr: 0x%px\n", addr);
200b45ba4a5SChristophe Leroy 		return 0;
201b45ba4a5SChristophe Leroy 	}
202b45ba4a5SChristophe Leroy 	return do_patch_instruction(addr, instr);
203b45ba4a5SChristophe Leroy }
20437bc3e5fSBalbir Singh NOKPROBE_SYMBOL(patch_instruction);
20537bc3e5fSBalbir Singh 
20694afd069SJordan Niethe int patch_branch(struct ppc_inst *addr, unsigned long target, int flags)
207e7a57273SMichael Ellerman {
20894afd069SJordan Niethe 	struct ppc_inst instr;
2097c95d889SJordan Niethe 
2107c95d889SJordan Niethe 	create_branch(&instr, addr, target, flags);
2117c95d889SJordan Niethe 	return patch_instruction(addr, instr);
212e7a57273SMichael Ellerman }
213e7a57273SMichael Ellerman 
214ebfa50dfSAnju T bool is_offset_in_branch_range(long offset)
215ebfa50dfSAnju T {
216ebfa50dfSAnju T 	/*
217ebfa50dfSAnju T 	 * Powerpc branch instruction is :
218ebfa50dfSAnju T 	 *
219ebfa50dfSAnju T 	 *  0         6                 30   31
220ebfa50dfSAnju T 	 *  +---------+----------------+---+---+
221ebfa50dfSAnju T 	 *  | opcode  |     LI         |AA |LK |
222ebfa50dfSAnju T 	 *  +---------+----------------+---+---+
223ebfa50dfSAnju T 	 *  Where AA = 0 and LK = 0
224ebfa50dfSAnju T 	 *
225ebfa50dfSAnju T 	 * LI is a signed 24 bits integer. The real branch offset is computed
226ebfa50dfSAnju T 	 * by: imm32 = SignExtend(LI:'0b00', 32);
227ebfa50dfSAnju T 	 *
228ebfa50dfSAnju T 	 * So the maximum forward branch should be:
229ebfa50dfSAnju T 	 *   (0x007fffff << 2) = 0x01fffffc =  0x1fffffc
230ebfa50dfSAnju T 	 * The maximum backward branch should be:
231ebfa50dfSAnju T 	 *   (0xff800000 << 2) = 0xfe000000 = -0x2000000
232ebfa50dfSAnju T 	 */
233ebfa50dfSAnju T 	return (offset >= -0x2000000 && offset <= 0x1fffffc && !(offset & 0x3));
234ebfa50dfSAnju T }
235ebfa50dfSAnju T 
23651c9c084SAnju T /*
23751c9c084SAnju T  * Helper to check if a given instruction is a conditional branch
23851c9c084SAnju T  * Derived from the conditional checks in analyse_instr()
23951c9c084SAnju T  */
24094afd069SJordan Niethe bool is_conditional_branch(struct ppc_inst instr)
24151c9c084SAnju T {
2428094892dSJordan Niethe 	unsigned int opcode = ppc_inst_primary_opcode(instr);
24351c9c084SAnju T 
24451c9c084SAnju T 	if (opcode == 16)       /* bc, bca, bcl, bcla */
24551c9c084SAnju T 		return true;
24651c9c084SAnju T 	if (opcode == 19) {
247777e26f0SJordan Niethe 		switch ((ppc_inst_val(instr) >> 1) & 0x3ff) {
24851c9c084SAnju T 		case 16:        /* bclr, bclrl */
24951c9c084SAnju T 		case 528:       /* bcctr, bcctrl */
25051c9c084SAnju T 		case 560:       /* bctar, bctarl */
25151c9c084SAnju T 			return true;
25251c9c084SAnju T 		}
25351c9c084SAnju T 	}
25451c9c084SAnju T 	return false;
25551c9c084SAnju T }
25671f6e58eSNaveen N. Rao NOKPROBE_SYMBOL(is_conditional_branch);
25751c9c084SAnju T 
25894afd069SJordan Niethe int create_branch(struct ppc_inst *instr,
25994afd069SJordan Niethe 		  const struct ppc_inst *addr,
260e7a57273SMichael Ellerman 		  unsigned long target, int flags)
261aaddd3eaSMichael Ellerman {
262053a858eSMichael Ellerman 	long offset;
263aaddd3eaSMichael Ellerman 
26494afd069SJordan Niethe 	*instr = ppc_inst(0);
265053a858eSMichael Ellerman 	offset = target;
266aaddd3eaSMichael Ellerman 	if (! (flags & BRANCH_ABSOLUTE))
267053a858eSMichael Ellerman 		offset = offset - (unsigned long)addr;
268053a858eSMichael Ellerman 
269053a858eSMichael Ellerman 	/* Check we can represent the target in the instruction format */
270ebfa50dfSAnju T 	if (!is_offset_in_branch_range(offset))
2717c95d889SJordan Niethe 		return 1;
272aaddd3eaSMichael Ellerman 
273aaddd3eaSMichael Ellerman 	/* Mask out the flags and target, so they don't step on each other. */
27494afd069SJordan Niethe 	*instr = ppc_inst(0x48000000 | (flags & 0x3) | (offset & 0x03FFFFFC));
275aaddd3eaSMichael Ellerman 
2767c95d889SJordan Niethe 	return 0;
277aaddd3eaSMichael Ellerman }
278411781a2SMichael Ellerman 
27994afd069SJordan Niethe int create_cond_branch(struct ppc_inst *instr, const struct ppc_inst *addr,
280411781a2SMichael Ellerman 		       unsigned long target, int flags)
281411781a2SMichael Ellerman {
282411781a2SMichael Ellerman 	long offset;
283411781a2SMichael Ellerman 
284411781a2SMichael Ellerman 	offset = target;
285411781a2SMichael Ellerman 	if (! (flags & BRANCH_ABSOLUTE))
286411781a2SMichael Ellerman 		offset = offset - (unsigned long)addr;
287411781a2SMichael Ellerman 
288411781a2SMichael Ellerman 	/* Check we can represent the target in the instruction format */
289411781a2SMichael Ellerman 	if (offset < -0x8000 || offset > 0x7FFF || offset & 0x3)
2907c95d889SJordan Niethe 		return 1;
291411781a2SMichael Ellerman 
292411781a2SMichael Ellerman 	/* Mask out the flags and target, so they don't step on each other. */
29394afd069SJordan Niethe 	*instr = ppc_inst(0x40000000 | (flags & 0x3FF0003) | (offset & 0xFFFC));
294411781a2SMichael Ellerman 
2957c95d889SJordan Niethe 	return 0;
296411781a2SMichael Ellerman }
297411781a2SMichael Ellerman 
29894afd069SJordan Niethe static unsigned int branch_opcode(struct ppc_inst instr)
299411781a2SMichael Ellerman {
3008094892dSJordan Niethe 	return ppc_inst_primary_opcode(instr) & 0x3F;
301411781a2SMichael Ellerman }
302411781a2SMichael Ellerman 
30394afd069SJordan Niethe static int instr_is_branch_iform(struct ppc_inst instr)
304411781a2SMichael Ellerman {
305411781a2SMichael Ellerman 	return branch_opcode(instr) == 18;
306411781a2SMichael Ellerman }
307411781a2SMichael Ellerman 
30894afd069SJordan Niethe static int instr_is_branch_bform(struct ppc_inst instr)
309411781a2SMichael Ellerman {
310411781a2SMichael Ellerman 	return branch_opcode(instr) == 16;
311411781a2SMichael Ellerman }
312411781a2SMichael Ellerman 
31394afd069SJordan Niethe int instr_is_relative_branch(struct ppc_inst instr)
314411781a2SMichael Ellerman {
315777e26f0SJordan Niethe 	if (ppc_inst_val(instr) & BRANCH_ABSOLUTE)
316411781a2SMichael Ellerman 		return 0;
317411781a2SMichael Ellerman 
318411781a2SMichael Ellerman 	return instr_is_branch_iform(instr) || instr_is_branch_bform(instr);
319411781a2SMichael Ellerman }
320411781a2SMichael Ellerman 
32194afd069SJordan Niethe int instr_is_relative_link_branch(struct ppc_inst instr)
322b9eab08dSJosh Poimboeuf {
323777e26f0SJordan Niethe 	return instr_is_relative_branch(instr) && (ppc_inst_val(instr) & BRANCH_SET_LINK);
324b9eab08dSJosh Poimboeuf }
325b9eab08dSJosh Poimboeuf 
32694afd069SJordan Niethe static unsigned long branch_iform_target(const struct ppc_inst *instr)
327411781a2SMichael Ellerman {
328411781a2SMichael Ellerman 	signed long imm;
329411781a2SMichael Ellerman 
330777e26f0SJordan Niethe 	imm = ppc_inst_val(*instr) & 0x3FFFFFC;
331411781a2SMichael Ellerman 
332411781a2SMichael Ellerman 	/* If the top bit of the immediate value is set this is negative */
333411781a2SMichael Ellerman 	if (imm & 0x2000000)
334411781a2SMichael Ellerman 		imm -= 0x4000000;
335411781a2SMichael Ellerman 
336777e26f0SJordan Niethe 	if ((ppc_inst_val(*instr) & BRANCH_ABSOLUTE) == 0)
337411781a2SMichael Ellerman 		imm += (unsigned long)instr;
338411781a2SMichael Ellerman 
339411781a2SMichael Ellerman 	return (unsigned long)imm;
340411781a2SMichael Ellerman }
341411781a2SMichael Ellerman 
34294afd069SJordan Niethe static unsigned long branch_bform_target(const struct ppc_inst *instr)
343411781a2SMichael Ellerman {
344411781a2SMichael Ellerman 	signed long imm;
345411781a2SMichael Ellerman 
346777e26f0SJordan Niethe 	imm = ppc_inst_val(*instr) & 0xFFFC;
347411781a2SMichael Ellerman 
348411781a2SMichael Ellerman 	/* If the top bit of the immediate value is set this is negative */
349411781a2SMichael Ellerman 	if (imm & 0x8000)
350411781a2SMichael Ellerman 		imm -= 0x10000;
351411781a2SMichael Ellerman 
352777e26f0SJordan Niethe 	if ((ppc_inst_val(*instr) & BRANCH_ABSOLUTE) == 0)
353411781a2SMichael Ellerman 		imm += (unsigned long)instr;
354411781a2SMichael Ellerman 
355411781a2SMichael Ellerman 	return (unsigned long)imm;
356411781a2SMichael Ellerman }
357411781a2SMichael Ellerman 
35894afd069SJordan Niethe unsigned long branch_target(const struct ppc_inst *instr)
359411781a2SMichael Ellerman {
360f8faaffaSJordan Niethe 	if (instr_is_branch_iform(ppc_inst_read(instr)))
361411781a2SMichael Ellerman 		return branch_iform_target(instr);
362f8faaffaSJordan Niethe 	else if (instr_is_branch_bform(ppc_inst_read(instr)))
363411781a2SMichael Ellerman 		return branch_bform_target(instr);
364411781a2SMichael Ellerman 
365411781a2SMichael Ellerman 	return 0;
366411781a2SMichael Ellerman }
367411781a2SMichael Ellerman 
36894afd069SJordan Niethe int instr_is_branch_to_addr(const struct ppc_inst *instr, unsigned long addr)
369411781a2SMichael Ellerman {
370f8faaffaSJordan Niethe 	if (instr_is_branch_iform(ppc_inst_read(instr)) ||
371f8faaffaSJordan Niethe 	    instr_is_branch_bform(ppc_inst_read(instr)))
372411781a2SMichael Ellerman 		return branch_target(instr) == addr;
373411781a2SMichael Ellerman 
374411781a2SMichael Ellerman 	return 0;
375411781a2SMichael Ellerman }
376411781a2SMichael Ellerman 
37794afd069SJordan Niethe int translate_branch(struct ppc_inst *instr, const struct ppc_inst *dest,
37894afd069SJordan Niethe 		     const struct ppc_inst *src)
379411781a2SMichael Ellerman {
380411781a2SMichael Ellerman 	unsigned long target;
381411781a2SMichael Ellerman 	target = branch_target(src);
382411781a2SMichael Ellerman 
383f8faaffaSJordan Niethe 	if (instr_is_branch_iform(ppc_inst_read(src)))
384f8faaffaSJordan Niethe 		return create_branch(instr, dest, target,
385f8faaffaSJordan Niethe 				     ppc_inst_val(ppc_inst_read(src)));
386f8faaffaSJordan Niethe 	else if (instr_is_branch_bform(ppc_inst_read(src)))
387f8faaffaSJordan Niethe 		return create_cond_branch(instr, dest, target,
388f8faaffaSJordan Niethe 					  ppc_inst_val(ppc_inst_read(src)));
389411781a2SMichael Ellerman 
3907c95d889SJordan Niethe 	return 1;
391411781a2SMichael Ellerman }
392ae0dc736SMichael Ellerman 
3931e8341aeSKevin Hao #ifdef CONFIG_PPC_BOOK3E_64
3941e8341aeSKevin Hao void __patch_exception(int exc, unsigned long addr)
3951e8341aeSKevin Hao {
3961e8341aeSKevin Hao 	extern unsigned int interrupt_base_book3e;
3971e8341aeSKevin Hao 	unsigned int *ibase = &interrupt_base_book3e;
3981e8341aeSKevin Hao 
3991e8341aeSKevin Hao 	/* Our exceptions vectors start with a NOP and -then- a branch
4001e8341aeSKevin Hao 	 * to deal with single stepping from userspace which stops on
4011e8341aeSKevin Hao 	 * the second instruction. Thus we need to patch the second
4021e8341aeSKevin Hao 	 * instruction of the exception, not the first one
4031e8341aeSKevin Hao 	 */
4041e8341aeSKevin Hao 
40594afd069SJordan Niethe 	patch_branch((struct ppc_inst *)(ibase + (exc / 4) + 1), addr, 0);
4061e8341aeSKevin Hao }
4071e8341aeSKevin Hao #endif
408ae0dc736SMichael Ellerman 
409ae0dc736SMichael Ellerman #ifdef CONFIG_CODE_PATCHING_SELFTEST
410ae0dc736SMichael Ellerman 
411ae0dc736SMichael Ellerman static void __init test_trampoline(void)
412ae0dc736SMichael Ellerman {
413ae0dc736SMichael Ellerman 	asm ("nop;\n");
414ae0dc736SMichael Ellerman }
415ae0dc736SMichael Ellerman 
416ae0dc736SMichael Ellerman #define check(x)	\
417ae0dc736SMichael Ellerman 	if (!(x)) printk("code-patching: test failed at line %d\n", __LINE__);
418ae0dc736SMichael Ellerman 
419ae0dc736SMichael Ellerman static void __init test_branch_iform(void)
420ae0dc736SMichael Ellerman {
4217c95d889SJordan Niethe 	int err;
42294afd069SJordan Niethe 	struct ppc_inst instr;
423ae0dc736SMichael Ellerman 	unsigned long addr;
424ae0dc736SMichael Ellerman 
425ae0dc736SMichael Ellerman 	addr = (unsigned long)&instr;
426ae0dc736SMichael Ellerman 
427ae0dc736SMichael Ellerman 	/* The simplest case, branch to self, no flags */
42875346251SJordan Niethe 	check(instr_is_branch_iform(ppc_inst(0x48000000)));
429ae0dc736SMichael Ellerman 	/* All bits of target set, and flags */
43075346251SJordan Niethe 	check(instr_is_branch_iform(ppc_inst(0x4bffffff)));
431ae0dc736SMichael Ellerman 	/* High bit of opcode set, which is wrong */
43275346251SJordan Niethe 	check(!instr_is_branch_iform(ppc_inst(0xcbffffff)));
433ae0dc736SMichael Ellerman 	/* Middle bits of opcode set, which is wrong */
43475346251SJordan Niethe 	check(!instr_is_branch_iform(ppc_inst(0x7bffffff)));
435ae0dc736SMichael Ellerman 
436ae0dc736SMichael Ellerman 	/* Simplest case, branch to self with link */
43775346251SJordan Niethe 	check(instr_is_branch_iform(ppc_inst(0x48000001)));
438ae0dc736SMichael Ellerman 	/* All bits of targets set */
43975346251SJordan Niethe 	check(instr_is_branch_iform(ppc_inst(0x4bfffffd)));
440ae0dc736SMichael Ellerman 	/* Some bits of targets set */
44175346251SJordan Niethe 	check(instr_is_branch_iform(ppc_inst(0x4bff00fd)));
442ae0dc736SMichael Ellerman 	/* Must be a valid branch to start with */
44375346251SJordan Niethe 	check(!instr_is_branch_iform(ppc_inst(0x7bfffffd)));
444ae0dc736SMichael Ellerman 
445ae0dc736SMichael Ellerman 	/* Absolute branch to 0x100 */
44675346251SJordan Niethe 	instr = ppc_inst(0x48000103);
447ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, 0x100));
448ae0dc736SMichael Ellerman 	/* Absolute branch to 0x420fc */
44975346251SJordan Niethe 	instr = ppc_inst(0x480420ff);
450ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, 0x420fc));
451ae0dc736SMichael Ellerman 	/* Maximum positive relative branch, + 20MB - 4B */
45275346251SJordan Niethe 	instr = ppc_inst(0x49fffffc);
453ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr + 0x1FFFFFC));
454ae0dc736SMichael Ellerman 	/* Smallest negative relative branch, - 4B */
45575346251SJordan Niethe 	instr = ppc_inst(0x4bfffffc);
456ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr - 4));
457ae0dc736SMichael Ellerman 	/* Largest negative relative branch, - 32 MB */
45875346251SJordan Niethe 	instr = ppc_inst(0x4a000000);
459ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr - 0x2000000));
460ae0dc736SMichael Ellerman 
461ae0dc736SMichael Ellerman 	/* Branch to self, with link */
4627c95d889SJordan Niethe 	err = create_branch(&instr, &instr, addr, BRANCH_SET_LINK);
463ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr));
464ae0dc736SMichael Ellerman 
465ae0dc736SMichael Ellerman 	/* Branch to self - 0x100, with link */
4667c95d889SJordan Niethe 	err = create_branch(&instr, &instr, addr - 0x100, BRANCH_SET_LINK);
467ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr - 0x100));
468ae0dc736SMichael Ellerman 
469ae0dc736SMichael Ellerman 	/* Branch to self + 0x100, no link */
4707c95d889SJordan Niethe 	err = create_branch(&instr, &instr, addr + 0x100, 0);
471ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr + 0x100));
472ae0dc736SMichael Ellerman 
473ae0dc736SMichael Ellerman 	/* Maximum relative negative offset, - 32 MB */
4747c95d889SJordan Niethe 	err = create_branch(&instr, &instr, addr - 0x2000000, BRANCH_SET_LINK);
475ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr - 0x2000000));
476ae0dc736SMichael Ellerman 
477ae0dc736SMichael Ellerman 	/* Out of range relative negative offset, - 32 MB + 4*/
4787c95d889SJordan Niethe 	err = create_branch(&instr, &instr, addr - 0x2000004, BRANCH_SET_LINK);
4797c95d889SJordan Niethe 	check(err);
480ae0dc736SMichael Ellerman 
481ae0dc736SMichael Ellerman 	/* Out of range relative positive offset, + 32 MB */
4827c95d889SJordan Niethe 	err = create_branch(&instr, &instr, addr + 0x2000000, BRANCH_SET_LINK);
4837c95d889SJordan Niethe 	check(err);
484ae0dc736SMichael Ellerman 
485ae0dc736SMichael Ellerman 	/* Unaligned target */
4867c95d889SJordan Niethe 	err = create_branch(&instr, &instr, addr + 3, BRANCH_SET_LINK);
4877c95d889SJordan Niethe 	check(err);
488ae0dc736SMichael Ellerman 
489ae0dc736SMichael Ellerman 	/* Check flags are masked correctly */
4907c95d889SJordan Niethe 	err = create_branch(&instr, &instr, addr, 0xFFFFFFFC);
491ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr));
492217862d9SJordan Niethe 	check(ppc_inst_equal(instr, ppc_inst(0x48000000)));
493ae0dc736SMichael Ellerman }
494ae0dc736SMichael Ellerman 
495ae0dc736SMichael Ellerman static void __init test_create_function_call(void)
496ae0dc736SMichael Ellerman {
49794afd069SJordan Niethe 	struct ppc_inst *iptr;
498ae0dc736SMichael Ellerman 	unsigned long dest;
49994afd069SJordan Niethe 	struct ppc_inst instr;
500ae0dc736SMichael Ellerman 
501ae0dc736SMichael Ellerman 	/* Check we can create a function call */
50294afd069SJordan Niethe 	iptr = (struct ppc_inst *)ppc_function_entry(test_trampoline);
503ae0dc736SMichael Ellerman 	dest = ppc_function_entry(test_create_function_call);
5047c95d889SJordan Niethe 	create_branch(&instr, iptr, dest, BRANCH_SET_LINK);
5057c95d889SJordan Niethe 	patch_instruction(iptr, instr);
506ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(iptr, dest));
507ae0dc736SMichael Ellerman }
508ae0dc736SMichael Ellerman 
509ae0dc736SMichael Ellerman static void __init test_branch_bform(void)
510ae0dc736SMichael Ellerman {
5117c95d889SJordan Niethe 	int err;
512ae0dc736SMichael Ellerman 	unsigned long addr;
51394afd069SJordan Niethe 	struct ppc_inst *iptr, instr;
51494afd069SJordan Niethe 	unsigned int flags;
515ae0dc736SMichael Ellerman 
516ae0dc736SMichael Ellerman 	iptr = &instr;
517ae0dc736SMichael Ellerman 	addr = (unsigned long)iptr;
518ae0dc736SMichael Ellerman 
519ae0dc736SMichael Ellerman 	/* The simplest case, branch to self, no flags */
52075346251SJordan Niethe 	check(instr_is_branch_bform(ppc_inst(0x40000000)));
521ae0dc736SMichael Ellerman 	/* All bits of target set, and flags */
52275346251SJordan Niethe 	check(instr_is_branch_bform(ppc_inst(0x43ffffff)));
523ae0dc736SMichael Ellerman 	/* High bit of opcode set, which is wrong */
52475346251SJordan Niethe 	check(!instr_is_branch_bform(ppc_inst(0xc3ffffff)));
525ae0dc736SMichael Ellerman 	/* Middle bits of opcode set, which is wrong */
52675346251SJordan Niethe 	check(!instr_is_branch_bform(ppc_inst(0x7bffffff)));
527ae0dc736SMichael Ellerman 
528ae0dc736SMichael Ellerman 	/* Absolute conditional branch to 0x100 */
52975346251SJordan Niethe 	instr = ppc_inst(0x43ff0103);
530ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, 0x100));
531ae0dc736SMichael Ellerman 	/* Absolute conditional branch to 0x20fc */
53275346251SJordan Niethe 	instr = ppc_inst(0x43ff20ff);
533ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, 0x20fc));
534ae0dc736SMichael Ellerman 	/* Maximum positive relative conditional branch, + 32 KB - 4B */
53575346251SJordan Niethe 	instr = ppc_inst(0x43ff7ffc);
536ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr + 0x7FFC));
537ae0dc736SMichael Ellerman 	/* Smallest negative relative conditional branch, - 4B */
53875346251SJordan Niethe 	instr = ppc_inst(0x43fffffc);
539ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr - 4));
540ae0dc736SMichael Ellerman 	/* Largest negative relative conditional branch, - 32 KB */
54175346251SJordan Niethe 	instr = ppc_inst(0x43ff8000);
542ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr - 0x8000));
543ae0dc736SMichael Ellerman 
544ae0dc736SMichael Ellerman 	/* All condition code bits set & link */
545ae0dc736SMichael Ellerman 	flags = 0x3ff000 | BRANCH_SET_LINK;
546ae0dc736SMichael Ellerman 
547ae0dc736SMichael Ellerman 	/* Branch to self */
5487c95d889SJordan Niethe 	err = create_cond_branch(&instr, iptr, addr, flags);
549ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr));
550ae0dc736SMichael Ellerman 
551ae0dc736SMichael Ellerman 	/* Branch to self - 0x100 */
5527c95d889SJordan Niethe 	err = create_cond_branch(&instr, iptr, addr - 0x100, flags);
553ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr - 0x100));
554ae0dc736SMichael Ellerman 
555ae0dc736SMichael Ellerman 	/* Branch to self + 0x100 */
5567c95d889SJordan Niethe 	err = create_cond_branch(&instr, iptr, addr + 0x100, flags);
557ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr + 0x100));
558ae0dc736SMichael Ellerman 
559ae0dc736SMichael Ellerman 	/* Maximum relative negative offset, - 32 KB */
5607c95d889SJordan Niethe 	err = create_cond_branch(&instr, iptr, addr - 0x8000, flags);
561ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr - 0x8000));
562ae0dc736SMichael Ellerman 
563ae0dc736SMichael Ellerman 	/* Out of range relative negative offset, - 32 KB + 4*/
5647c95d889SJordan Niethe 	err = create_cond_branch(&instr, iptr, addr - 0x8004, flags);
5657c95d889SJordan Niethe 	check(err);
566ae0dc736SMichael Ellerman 
567ae0dc736SMichael Ellerman 	/* Out of range relative positive offset, + 32 KB */
5687c95d889SJordan Niethe 	err = create_cond_branch(&instr, iptr, addr + 0x8000, flags);
5697c95d889SJordan Niethe 	check(err);
570ae0dc736SMichael Ellerman 
571ae0dc736SMichael Ellerman 	/* Unaligned target */
5727c95d889SJordan Niethe 	err = create_cond_branch(&instr, iptr, addr + 3, flags);
5737c95d889SJordan Niethe 	check(err);
574ae0dc736SMichael Ellerman 
575ae0dc736SMichael Ellerman 	/* Check flags are masked correctly */
5767c95d889SJordan Niethe 	err = create_cond_branch(&instr, iptr, addr, 0xFFFFFFFC);
577ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(&instr, addr));
578217862d9SJordan Niethe 	check(ppc_inst_equal(instr, ppc_inst(0x43FF0000)));
579ae0dc736SMichael Ellerman }
580ae0dc736SMichael Ellerman 
581ae0dc736SMichael Ellerman static void __init test_translate_branch(void)
582ae0dc736SMichael Ellerman {
583ae0dc736SMichael Ellerman 	unsigned long addr;
5840b582db5SJordan Niethe 	void *p, *q;
58594afd069SJordan Niethe 	struct ppc_inst instr;
586ae0dc736SMichael Ellerman 	void *buf;
587ae0dc736SMichael Ellerman 
588ae0dc736SMichael Ellerman 	buf = vmalloc(PAGE_ALIGN(0x2000000 + 1));
589ae0dc736SMichael Ellerman 	check(buf);
590ae0dc736SMichael Ellerman 	if (!buf)
591ae0dc736SMichael Ellerman 		return;
592ae0dc736SMichael Ellerman 
593ae0dc736SMichael Ellerman 	/* Simple case, branch to self moved a little */
594ae0dc736SMichael Ellerman 	p = buf;
595ae0dc736SMichael Ellerman 	addr = (unsigned long)p;
596ae0dc736SMichael Ellerman 	patch_branch(p, addr, 0);
597ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
5980b582db5SJordan Niethe 	q = p + 4;
5997c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6007c95d889SJordan Niethe 	patch_instruction(q, instr);
601ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
602ae0dc736SMichael Ellerman 
603ae0dc736SMichael Ellerman 	/* Maximum negative case, move b . to addr + 32 MB */
604ae0dc736SMichael Ellerman 	p = buf;
605ae0dc736SMichael Ellerman 	addr = (unsigned long)p;
606ae0dc736SMichael Ellerman 	patch_branch(p, addr, 0);
607ae0dc736SMichael Ellerman 	q = buf + 0x2000000;
6087c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6097c95d889SJordan Niethe 	patch_instruction(q, instr);
610ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
611ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
612f8faaffaSJordan Niethe 	check(ppc_inst_equal(ppc_inst_read(q), ppc_inst(0x4a000000)));
613ae0dc736SMichael Ellerman 
614ae0dc736SMichael Ellerman 	/* Maximum positive case, move x to x - 32 MB + 4 */
615ae0dc736SMichael Ellerman 	p = buf + 0x2000000;
616ae0dc736SMichael Ellerman 	addr = (unsigned long)p;
617ae0dc736SMichael Ellerman 	patch_branch(p, addr, 0);
618ae0dc736SMichael Ellerman 	q = buf + 4;
6197c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6207c95d889SJordan Niethe 	patch_instruction(q, instr);
621ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
622ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
623f8faaffaSJordan Niethe 	check(ppc_inst_equal(ppc_inst_read(q), ppc_inst(0x49fffffc)));
624ae0dc736SMichael Ellerman 
625ae0dc736SMichael Ellerman 	/* Jump to x + 16 MB moved to x + 20 MB */
626ae0dc736SMichael Ellerman 	p = buf;
627ae0dc736SMichael Ellerman 	addr = 0x1000000 + (unsigned long)buf;
628ae0dc736SMichael Ellerman 	patch_branch(p, addr, BRANCH_SET_LINK);
629ae0dc736SMichael Ellerman 	q = buf + 0x1400000;
6307c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6317c95d889SJordan Niethe 	patch_instruction(q, instr);
632ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
633ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
634ae0dc736SMichael Ellerman 
635ae0dc736SMichael Ellerman 	/* Jump to x + 16 MB moved to x - 16 MB + 4 */
636ae0dc736SMichael Ellerman 	p = buf + 0x1000000;
637ae0dc736SMichael Ellerman 	addr = 0x2000000 + (unsigned long)buf;
638ae0dc736SMichael Ellerman 	patch_branch(p, addr, 0);
639ae0dc736SMichael Ellerman 	q = buf + 4;
6407c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6417c95d889SJordan Niethe 	patch_instruction(q, instr);
642ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
643ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
644ae0dc736SMichael Ellerman 
645ae0dc736SMichael Ellerman 
646ae0dc736SMichael Ellerman 	/* Conditional branch tests */
647ae0dc736SMichael Ellerman 
648ae0dc736SMichael Ellerman 	/* Simple case, branch to self moved a little */
649ae0dc736SMichael Ellerman 	p = buf;
650ae0dc736SMichael Ellerman 	addr = (unsigned long)p;
6517c95d889SJordan Niethe 	create_cond_branch(&instr, p, addr, 0);
6527c95d889SJordan Niethe 	patch_instruction(p, instr);
653ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
6540b582db5SJordan Niethe 	q = buf + 4;
6557c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6567c95d889SJordan Niethe 	patch_instruction(q, instr);
657ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
658ae0dc736SMichael Ellerman 
659ae0dc736SMichael Ellerman 	/* Maximum negative case, move b . to addr + 32 KB */
660ae0dc736SMichael Ellerman 	p = buf;
661ae0dc736SMichael Ellerman 	addr = (unsigned long)p;
6627c95d889SJordan Niethe 	create_cond_branch(&instr, p, addr, 0xFFFFFFFC);
6637c95d889SJordan Niethe 	patch_instruction(p, instr);
664ae0dc736SMichael Ellerman 	q = buf + 0x8000;
6657c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6667c95d889SJordan Niethe 	patch_instruction(q, instr);
667ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
668ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
669f8faaffaSJordan Niethe 	check(ppc_inst_equal(ppc_inst_read(q), ppc_inst(0x43ff8000)));
670ae0dc736SMichael Ellerman 
671ae0dc736SMichael Ellerman 	/* Maximum positive case, move x to x - 32 KB + 4 */
672ae0dc736SMichael Ellerman 	p = buf + 0x8000;
673ae0dc736SMichael Ellerman 	addr = (unsigned long)p;
6747c95d889SJordan Niethe 	create_cond_branch(&instr, p, addr, 0xFFFFFFFC);
6757c95d889SJordan Niethe 	patch_instruction(p, instr);
676ae0dc736SMichael Ellerman 	q = buf + 4;
6777c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6787c95d889SJordan Niethe 	patch_instruction(q, instr);
679ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
680ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
681f8faaffaSJordan Niethe 	check(ppc_inst_equal(ppc_inst_read(q), ppc_inst(0x43ff7ffc)));
682ae0dc736SMichael Ellerman 
683ae0dc736SMichael Ellerman 	/* Jump to x + 12 KB moved to x + 20 KB */
684ae0dc736SMichael Ellerman 	p = buf;
685ae0dc736SMichael Ellerman 	addr = 0x3000 + (unsigned long)buf;
6867c95d889SJordan Niethe 	create_cond_branch(&instr, p, addr, BRANCH_SET_LINK);
6877c95d889SJordan Niethe 	patch_instruction(p, instr);
688ae0dc736SMichael Ellerman 	q = buf + 0x5000;
6897c95d889SJordan Niethe 	translate_branch(&instr, q, p);
6907c95d889SJordan Niethe 	patch_instruction(q, instr);
691ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
692ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
693ae0dc736SMichael Ellerman 
694ae0dc736SMichael Ellerman 	/* Jump to x + 8 KB moved to x - 8 KB + 4 */
695ae0dc736SMichael Ellerman 	p = buf + 0x2000;
696ae0dc736SMichael Ellerman 	addr = 0x4000 + (unsigned long)buf;
6977c95d889SJordan Niethe 	create_cond_branch(&instr, p, addr, 0);
6987c95d889SJordan Niethe 	patch_instruction(p, instr);
699ae0dc736SMichael Ellerman 	q = buf + 4;
7007c95d889SJordan Niethe 	translate_branch(&instr, q, p);
7017c95d889SJordan Niethe 	patch_instruction(q, instr);
702ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(p, addr));
703ae0dc736SMichael Ellerman 	check(instr_is_branch_to_addr(q, addr));
704ae0dc736SMichael Ellerman 
705ae0dc736SMichael Ellerman 	/* Free the buffer we were using */
706ae0dc736SMichael Ellerman 	vfree(buf);
707ae0dc736SMichael Ellerman }
708ae0dc736SMichael Ellerman 
709f77f8ff7SJordan Niethe #ifdef CONFIG_PPC64
710f77f8ff7SJordan Niethe static void __init test_prefixed_patching(void)
711f77f8ff7SJordan Niethe {
712f77f8ff7SJordan Niethe 	extern unsigned int code_patching_test1[];
713f77f8ff7SJordan Niethe 	extern unsigned int code_patching_test1_expected[];
714f77f8ff7SJordan Niethe 	extern unsigned int end_code_patching_test1[];
715f77f8ff7SJordan Niethe 
716f77f8ff7SJordan Niethe 	__patch_instruction((struct ppc_inst *)code_patching_test1,
717f77f8ff7SJordan Niethe 			    ppc_inst_prefix(OP_PREFIX << 26, 0x00000000),
718f77f8ff7SJordan Niethe 			    (struct ppc_inst *)code_patching_test1);
719f77f8ff7SJordan Niethe 
720f77f8ff7SJordan Niethe 	check(!memcmp(code_patching_test1,
721f77f8ff7SJordan Niethe 		      code_patching_test1_expected,
722f77f8ff7SJordan Niethe 		      sizeof(unsigned int) *
723f77f8ff7SJordan Niethe 		      (end_code_patching_test1 - code_patching_test1)));
724f77f8ff7SJordan Niethe }
725f77f8ff7SJordan Niethe #else
726f77f8ff7SJordan Niethe static inline void test_prefixed_patching(void) {}
727f77f8ff7SJordan Niethe #endif
728f77f8ff7SJordan Niethe 
729ae0dc736SMichael Ellerman static int __init test_code_patching(void)
730ae0dc736SMichael Ellerman {
731ae0dc736SMichael Ellerman 	printk(KERN_DEBUG "Running code patching self-tests ...\n");
732ae0dc736SMichael Ellerman 
733ae0dc736SMichael Ellerman 	test_branch_iform();
734ae0dc736SMichael Ellerman 	test_branch_bform();
735ae0dc736SMichael Ellerman 	test_create_function_call();
736ae0dc736SMichael Ellerman 	test_translate_branch();
737f77f8ff7SJordan Niethe 	test_prefixed_patching();
738ae0dc736SMichael Ellerman 
739ae0dc736SMichael Ellerman 	return 0;
740ae0dc736SMichael Ellerman }
741ae0dc736SMichael Ellerman late_initcall(test_code_patching);
742ae0dc736SMichael Ellerman 
743ae0dc736SMichael Ellerman #endif /* CONFIG_CODE_PATCHING_SELFTEST */
744