[GitHub/mt8127/android_kernel_alcatel_ttab.git] / arch / mips / kernel / mcount.S

/*
 * MIPS specific _mcount support
 *
 * This file is subject to the terms and conditions of the GNU General Public
 * License.  See the file "COPYING" in the main directory of this archive for
 * more details.
 *
 * Copyright (C) 2009 Lemote Inc. & DSLab, Lanzhou University, China
 * Copyright (C) 2010 DSLab, Lanzhou University, China
 * Author: Wu Zhangjin <wuzhangjin@gmail.com>
 */

#include <asm/regdef.h>
#include <asm/stackframe.h>
#include <asm/ftrace.h>

	.text
	.set noreorder
	.set noat

	.macro MCOUNT_SAVE_REGS
	PTR_SUBU	sp, PT_SIZE
	PTR_S	ra, PT_R31(sp)
	PTR_S	AT, PT_R1(sp)
	PTR_S	a0, PT_R4(sp)
	PTR_S	a1, PT_R5(sp)
	PTR_S	a2, PT_R6(sp)
	PTR_S	a3, PT_R7(sp)
#ifdef CONFIG_64BIT
	PTR_S	a4, PT_R8(sp)
	PTR_S	a5, PT_R9(sp)
	PTR_S	a6, PT_R10(sp)
	PTR_S	a7, PT_R11(sp)
#endif
	.endm

	.macro MCOUNT_RESTORE_REGS
	PTR_L	ra, PT_R31(sp)
	PTR_L	AT, PT_R1(sp)
	PTR_L	a0, PT_R4(sp)
	PTR_L	a1, PT_R5(sp)
	PTR_L	a2, PT_R6(sp)
	PTR_L	a3, PT_R7(sp)
#ifdef CONFIG_64BIT
	PTR_L	a4, PT_R8(sp)
	PTR_L	a5, PT_R9(sp)
	PTR_L	a6, PT_R10(sp)
	PTR_L	a7, PT_R11(sp)
#endif
	PTR_ADDIU	sp, PT_SIZE
	.endm

	.macro RETURN_BACK
	jr ra
	 move ra, AT
	.endm

/*
 * The -mmcount-ra-address option of gcc 4.5 uses register $12 to pass
 * the location of the parent's return address.
 */
#define MCOUNT_RA_ADDRESS_REG	$12

#ifdef CONFIG_DYNAMIC_FTRACE

NESTED(ftrace_caller, PT_SIZE, ra)
	.globl _mcount
_mcount:
	b	ftrace_stub
#ifdef CONFIG_32BIT
	 addiu sp,sp,8
#else
	 nop
#endif

	/* When tracing is activated, it calls ftrace_caller+8 (aka here) */
	lw	t1, function_trace_stop
	bnez	t1, ftrace_stub
	 nop

	MCOUNT_SAVE_REGS
#ifdef KBUILD_MCOUNT_RA_ADDRESS
	PTR_S	MCOUNT_RA_ADDRESS_REG, PT_R12(sp)
#endif

	move	a0, ra		/* arg1: self return address */
	.globl ftrace_call
ftrace_call:
	nop	/* a placeholder for the call to a real tracing function */
	 move	a1, AT		/* arg2: parent's return address */

#ifdef CONFIG_FUNCTION_GRAPH_TRACER
	.globl ftrace_graph_call
ftrace_graph_call:
	nop
	 nop
#endif

	MCOUNT_RESTORE_REGS
	.globl ftrace_stub
ftrace_stub:
	RETURN_BACK
	END(ftrace_caller)

#else	/* ! CONFIG_DYNAMIC_FTRACE */

NESTED(_mcount, PT_SIZE, ra)
	lw	t1, function_trace_stop
	bnez	t1, ftrace_stub
	 nop
	PTR_LA	t1, ftrace_stub
	PTR_L	t2, ftrace_trace_function /* Prepare t2 for (1) */
	bne	t1, t2, static_trace
	 nop

#ifdef	CONFIG_FUNCTION_GRAPH_TRACER
	PTR_L	t3, ftrace_graph_return
	bne	t1, t3, ftrace_graph_caller
	 nop
	PTR_LA	t1, ftrace_graph_entry_stub
	PTR_L	t3, ftrace_graph_entry
	bne	t1, t3, ftrace_graph_caller
	 nop
#endif
	b	ftrace_stub
	 nop

static_trace:
	MCOUNT_SAVE_REGS

	move	a0, ra		/* arg1: self return address */
	jalr	t2		/* (1) call *ftrace_trace_function */
	 move	a1, AT		/* arg2: parent's return address */

	MCOUNT_RESTORE_REGS
	.globl ftrace_stub
ftrace_stub:
	RETURN_BACK
	END(_mcount)

#endif	/* ! CONFIG_DYNAMIC_FTRACE */

#ifdef CONFIG_FUNCTION_GRAPH_TRACER

NESTED(ftrace_graph_caller, PT_SIZE, ra)
#ifndef CONFIG_DYNAMIC_FTRACE
	MCOUNT_SAVE_REGS
#endif

	/* arg1: Get the location of the parent's return address */
#ifdef KBUILD_MCOUNT_RA_ADDRESS
#ifdef CONFIG_DYNAMIC_FTRACE
	PTR_L	a0, PT_R12(sp)
#else
	move	a0, MCOUNT_RA_ADDRESS_REG
#endif
	bnez	a0, 1f	/* non-leaf func: stored in MCOUNT_RA_ADDRESS_REG */
	 nop
#endif
	PTR_LA	a0, PT_R1(sp)	/* leaf func: the location in current stack */
1:

	/* arg2: Get self return address */
#ifdef CONFIG_DYNAMIC_FTRACE
	PTR_L	a1, PT_R31(sp)
#else
	move	a1, ra
#endif

	/* arg3: Get frame pointer of current stack */
#ifdef CONFIG_FRAME_POINTER
	move	a2, fp
#else /* ! CONFIG_FRAME_POINTER */
#ifdef CONFIG_64BIT
	PTR_LA	a2, PT_SIZE(sp)
#else
	PTR_LA	a2, (PT_SIZE+8)(sp)
#endif
#endif

	jal	prepare_ftrace_return
	 nop
	MCOUNT_RESTORE_REGS
	RETURN_BACK
	END(ftrace_graph_caller)

	.align	2
	.globl	return_to_handler
return_to_handler:
	PTR_SUBU	sp, PT_SIZE
	PTR_S	v0, PT_R2(sp)

	jal	ftrace_return_to_handler
	 PTR_S	v1, PT_R3(sp)

	/* restore the real parent address: v0 -> ra */
	move	ra, v0

	PTR_L	v0, PT_R2(sp)
	PTR_L	v1, PT_R3(sp)
	jr	ra
	 PTR_ADDIU	sp, PT_SIZE
#endif /* CONFIG_FUNCTION_GRAPH_TRACER */

	.set at
	.set reorder
Commit	Line	Data
d2bb0762 WZ	1	/*
	2	* MIPS specific _mcount support
	3	*
	4	* This file is subject to the terms and conditions of the GNU General Public
	5	* License. See the file "COPYING" in the main directory of this archive for
	6	* more details.
	7	*
	8	* Copyright (C) 2009 Lemote Inc. & DSLab, Lanzhou University, China
042e571d	9	* Copyright (C) 2010 DSLab, Lanzhou University, China
f7a904df	10	* Author: Wu Zhangjin <wuzhangjin@gmail.com>
d2bb0762 WZ	11	*/
	12
	13	#include <asm/regdef.h>
	14	#include <asm/stackframe.h>
	15	#include <asm/ftrace.h>
	16
	17	.text
	18	.set noreorder
	19	.set noat
	20
	21	.macro MCOUNT_SAVE_REGS
	22	PTR_SUBU sp, PT_SIZE
	23	PTR_S ra, PT_R31(sp)
	24	PTR_S AT, PT_R1(sp)
	25	PTR_S a0, PT_R4(sp)
	26	PTR_S a1, PT_R5(sp)
	27	PTR_S a2, PT_R6(sp)
	28	PTR_S a3, PT_R7(sp)
	29	#ifdef CONFIG_64BIT
	30	PTR_S a4, PT_R8(sp)
	31	PTR_S a5, PT_R9(sp)
	32	PTR_S a6, PT_R10(sp)
	33	PTR_S a7, PT_R11(sp)
	34	#endif
	35	.endm
	36
	37	.macro MCOUNT_RESTORE_REGS
	38	PTR_L ra, PT_R31(sp)
	39	PTR_L AT, PT_R1(sp)
	40	PTR_L a0, PT_R4(sp)
	41	PTR_L a1, PT_R5(sp)
	42	PTR_L a2, PT_R6(sp)
	43	PTR_L a3, PT_R7(sp)
	44	#ifdef CONFIG_64BIT
	45	PTR_L a4, PT_R8(sp)
	46	PTR_L a5, PT_R9(sp)
	47	PTR_L a6, PT_R10(sp)
	48	PTR_L a7, PT_R11(sp)
d2bb0762	49	#endif
ad8c3969 DD	50	PTR_ADDIU sp, PT_SIZE
ad8c3969 DD	51	.endm
d2bb0762 WZ	52
	53	.macro RETURN_BACK
	54	jr ra
	55	move ra, AT
	56	.endm
	57
56b49cd3 WZ	58	/*
	59	* The -mmcount-ra-address option of gcc 4.5 uses register $12 to pass
	60	* the location of the parent's return address.
	61	*/
	62	#define MCOUNT_RA_ADDRESS_REG $12
	63
538f1952 WZ	64	#ifdef CONFIG_DYNAMIC_FTRACE
	65
	66	NESTED(ftrace_caller, PT_SIZE, ra)
	67	.globl _mcount
	68	_mcount:
	69	b ftrace_stub
ad8c3969 DD	70	#ifdef CONFIG_32BIT
	71	addiu sp,sp,8
	72	#else
	73	nop
	74	#endif
58b69401 AC	75
58b69401 AC	76	/* When tracing is activated, it calls ftrace_caller+8 (aka here) */
fc49a3be WZ	77	lw t1, function_trace_stop
fc49a3be WZ	78	bnez t1, ftrace_stub
538f1952 WZ	79	nop
	80
	81	MCOUNT_SAVE_REGS
7326c4e5	82	#ifdef KBUILD_MCOUNT_RA_ADDRESS
56b49cd3	83	PTR_S MCOUNT_RA_ADDRESS_REG, PT_R12(sp)
7326c4e5	84	#endif
538f1952	85
042e571d	86	move a0, ra /* arg1: self return address */
538f1952 WZ	87	.globl ftrace_call
	88	ftrace_call:
	89	nop /* a placeholder for the call to a real tracing function */
042e571d	90	move a1, AT /* arg2: parent's return address */
538f1952	91
e17ff5fe WZ	92	#ifdef CONFIG_FUNCTION_GRAPH_TRACER
	93	.globl ftrace_graph_call
	94	ftrace_graph_call:
	95	nop
	96	nop
	97	#endif
	98
538f1952 WZ	99	MCOUNT_RESTORE_REGS
	100	.globl ftrace_stub
	101	ftrace_stub:
	102	RETURN_BACK
	103	END(ftrace_caller)
	104
	105	#else /* ! CONFIG_DYNAMIC_FTRACE */
	106
d2bb0762	107	NESTED(_mcount, PT_SIZE, ra)
fc49a3be WZ	108	lw t1, function_trace_stop
fc49a3be WZ	109	bnez t1, ftrace_stub
69a7d1b3	110	nop
fc49a3be WZ	111	PTR_LA t1, ftrace_stub
	112	PTR_L t2, ftrace_trace_function /* Prepare t2 for (1) */
	113	bne t1, t2, static_trace
d2bb0762	114	nop
29c5d346 WZ	115
29c5d346 WZ	116	#ifdef CONFIG_FUNCTION_GRAPH_TRACER
fc49a3be WZ	117	PTR_L t3, ftrace_graph_return
fc49a3be WZ	118	bne t1, t3, ftrace_graph_caller
29c5d346	119	nop
fc49a3be WZ	120	PTR_LA t1, ftrace_graph_entry_stub
	121	PTR_L t3, ftrace_graph_entry
	122	bne t1, t3, ftrace_graph_caller
29c5d346 WZ	123	nop
29c5d346 WZ	124	#endif
d2bb0762 WZ	125	b ftrace_stub
	126	nop
	127
	128	static_trace:
	129	MCOUNT_SAVE_REGS
	130
042e571d	131	move a0, ra /* arg1: self return address */
fc49a3be	132	jalr t2 /* (1) call ftrace_trace_function /
042e571d	133	move a1, AT /* arg2: parent's return address */
d2bb0762 WZ	134
	135	MCOUNT_RESTORE_REGS
	136	.globl ftrace_stub
	137	ftrace_stub:
	138	RETURN_BACK
	139	END(_mcount)
	140
538f1952 WZ	141	#endif /* ! CONFIG_DYNAMIC_FTRACE */
538f1952 WZ	142
29c5d346 WZ	143	#ifdef CONFIG_FUNCTION_GRAPH_TRACER
	144
	145	NESTED(ftrace_graph_caller, PT_SIZE, ra)
cbe555b0	146	#ifndef CONFIG_DYNAMIC_FTRACE
29c5d346	147	MCOUNT_SAVE_REGS
e17ff5fe	148	#endif
7326c4e5	149
cbe555b0	150	/* arg1: Get the location of the parent's return address */
7326c4e5	151	#ifdef KBUILD_MCOUNT_RA_ADDRESS
cbe555b0 WZ	152	#ifdef CONFIG_DYNAMIC_FTRACE
	153	PTR_L a0, PT_R12(sp)
	154	#else
56b49cd3	155	move a0, MCOUNT_RA_ADDRESS_REG
cbe555b0	156	#endif
56b49cd3	157	bnez a0, 1f /* non-leaf func: stored in MCOUNT_RA_ADDRESS_REG */
7326c4e5	158	nop
cbe555b0 WZ	159	#endif
	160	PTR_LA a0, PT_R1(sp) /* leaf func: the location in current stack */
	161	1:
	162
	163	/* arg2: Get self return address */
	164	#ifdef CONFIG_DYNAMIC_FTRACE
	165	PTR_L a1, PT_R31(sp)
7326c4e5	166	#else
cbe555b0	167	move a1, ra
7326c4e5	168	#endif
cbe555b0 WZ	169
cbe555b0 WZ	170	/* arg3: Get frame pointer of current stack */
046199ca	171	#ifdef CONFIG_FRAME_POINTER
65ab2826	172	move a2, fp
cbe555b0	173	#else /* ! CONFIG_FRAME_POINTER */
046199ca	174	#ifdef CONFIG_64BIT
65ab2826	175	PTR_LA a2, PT_SIZE(sp)
046199ca	176	#else
65ab2826	177	PTR_LA a2, (PT_SIZE+8)(sp)
046199ca WZ	178	#endif
046199ca WZ	179	#endif
29c5d346	180
cbe555b0 WZ	181	jal prepare_ftrace_return
cbe555b0 WZ	182	nop
29c5d346 WZ	183	MCOUNT_RESTORE_REGS
	184	RETURN_BACK
	185	END(ftrace_graph_caller)
	186
	187	.align 2
	188	.globl return_to_handler
	189	return_to_handler:
	190	PTR_SUBU sp, PT_SIZE
	191	PTR_S v0, PT_R2(sp)
	192
	193	jal ftrace_return_to_handler
	194	PTR_S v1, PT_R3(sp)
	195
	196	/* restore the real parent address: v0 -> ra */
	197	move ra, v0
	198
	199	PTR_L v0, PT_R2(sp)
	200	PTR_L v1, PT_R3(sp)
	201	jr ra
	202	PTR_ADDIU sp, PT_SIZE
	203	#endif /* CONFIG_FUNCTION_GRAPH_TRACER */
	204
d2bb0762 WZ	205	.set at
d2bb0762 WZ	206	.set reorder