/* $OpenBSD: profile.h,v 1.2 2021/02/17 12:11:45 kettenis Exp $ */
/*
 * Copyright (c) 2015 Dale Rahn <drahn@dalerahn.com>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#define	_MCOUNT_DECL void _mcount

#define MCOUNT_ASM_NAME "__mcount"

#ifdef __PIC__
#define	PLTSYM		"" /* XXX -aarch64 defaults to PLT? */
#else
#define	PLTSYM		""
#endif

#define MCOUNT							\
__asm__ (".text;"						\
	 ".align 3;"						\
	 ".globl " MCOUNT_ASM_NAME ";"				\
	 ".type " MCOUNT_ASM_NAME ",@function;"			\
	 MCOUNT_ASM_NAME ":;"					\
	 "	stp	x0, x1, [sp, #-160]!;"			\
	 "	stp	x2, x3, [sp, #16];"			\
	 "	stp	x4, x5, [sp, #32];"			\
	 "	stp	x6, x7, [sp, #48];"			\
	 "	stp	x8, x9, [sp, #64];"			\
	 "	stp	x10,x11,[sp, #80];"			\
	 "	stp	x12,x13,[sp, #96];"			\
	 "	stp	x14,x15,[sp, #112];"			\
	 "	stp	x16,x17,[sp, #128];"			\
	 "	stp	x29,lr, [sp, #144];"			\
	 /* load from pc at 8 off frame pointer */		\
	 "	ldr	x0, [x29, #8];"				\
	 "	mov	x1, lr;"				\
	 "	bl	" __STRING(_mcount) PLTSYM ";"		\
	 /* restore argument registers */			\
	 "	ldp	x2, x3, [sp, #16];"			\
	 "	ldp	x4, x5, [sp, #32];"			\
	 "	ldp	x6, x7, [sp, #48];"			\
	 "	ldp	x8, x9, [sp, #64];"			\
	 "	ldp	x10,x11,[sp, #80];"			\
	 "	ldp	x12,x13,[sp, #96];"			\
	 "	ldp	x14,x15,[sp, #112];"			\
	 "	ldp	x16,x17,[sp, #128];"			\
	 "	ldp	x29,lr, [sp, #144];"			\
	 "	ldp	x0, x1, [sp], #160;"			\
	 "	ret;");

#ifdef _KERNEL
// Change this to dair read/set, then restore.
#define MCOUNT_ENTER						\
__asm__ ("mrs %x0, daif; msr daifset, #0x3": "=r"(s));
#define	MCOUNT_EXIT						\
__asm__ ("msr daif, %x0":: "r"(s));
	
#endif // _KERNEL