/* $NetBSD: profile.h,v 1.2 2026/03/09 20:26:36 skrll Exp $ */

/*-
 * Copyright (c) 2014 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Matt Thomas of 3am Software Foundry.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef _RISCV_PROFILE_H_
#define _RISCV_PROFILE_H_

#include <machine/sysreg.h>

#define _MCOUNT_DECL void mcount

#define MCOUNT_ASM_NAME "_mcount"
#define PLTSYM

#ifdef _LP64
#define MCOUNT  MCOUNT_ARCH("8", "sd", "ld")
#else
#define MCOUNT  MCOUNT_ARCH("4", "sw", "lw")
#endif

#ifdef __PIC__
#define _PLT    "@plt"
#else
#define _PLT    /* nothing */   
#endif


#define MCOUNT_ARCH(_SZREG, _REG_S, _REG_L)     __asm(                  \
"       .text\n"                                                        \
"       .align  2\n"                                                    \
"       .type   " MCOUNT_ASM_NAME ",@function\n"                        \
"       .global " MCOUNT_ASM_NAME "\n"                                  \
MCOUNT_ASM_NAME ":\n"                                                   \
        /*                                                              \
         * Preserve registers that could be trashed during mcount, i.e. \
         * caller saved registers.                                      \
         */                                                             \
"       addi    sp, sp, -(16 * " _SZREG ")\n"                           \
"       " _REG_S "      ra, ( 0 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      t0, ( 1 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      t1, ( 2 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      t2, ( 3 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      a0, ( 4 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      a1, ( 5 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      a2, ( 6 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      a3, ( 7 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      a4, ( 8 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      a5, ( 9 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      a6, (10 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      a7, (11 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      t3, (12 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      t4, (13 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      t5, (14 * " _SZREG ")(sp)\n"                    \
"       " _REG_S "      t6, (15 * " _SZREG ")(sp)\n"                    \
"       mv      a1, ra\n"                                               \
"       call    mcount " _PLT "\n"                                      \
         /* restore caller saved registers */                           \
"       " _REG_L "      ra, ( 0 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      t0, ( 1 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      t1, ( 2 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      t2, ( 3 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      a0, ( 4 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      a1, ( 5 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      a2, ( 6 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      a3, ( 7 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      a4, ( 8 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      a5, ( 9 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      a6, (10 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      a7, (11 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      t3, (12 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      t4, (13 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      t5, (14 * " _SZREG ")(sp)\n"                    \
"       " _REG_L "      t6, (15 * " _SZREG ")(sp)\n"                    \
"       addi    sp, sp, (16 * " _SZREG ")\n"                            \
"       jr      ra\n");


#ifdef _KERNEL

#define MCOUNT_ENTER                                            \
   s = (csr_sstatus_read() & SR_SIE) != 0;                      \
   csr_sstatus_clear(SR_SIE);   // DISABLE_INTERRUPTS

#define MCOUNT_EXIT                                             \
   if (s)                                                       \
        csr_sstatus_set(SR_SIE); // ENABLE_INTERRUPTS

#endif

#endif /* _RISCV_PROFILE_H_ */