/* SPDX-License-Identifier: GPL-2.0 */

#include <linux/export.h>
#include <linux/linkage.h>
#include <asm/asm-offsets.h>
#include <asm/nospec-insn.h>

#define R2_OFF 0
#define R5_OFF (R2_OFF + (5 - 2 + 1) * 8)
#define R14_OFF (R5_OFF + 8)
#define RETADDR_OFF (R14_OFF + 8)
#define R15_OFF (RETADDR_OFF + 8)
#define BACKCHAIN_OFF (R15_OFF + 8)
#define FRAME_SIZE (BACKCHAIN_OFF + 8)
#define FRAME_OFF (STACK_FRAME_OVERHEAD - FRAME_SIZE)
#if (FRAME_OFF + BACKCHAIN_OFF) != __SF_BACKCHAIN
#error Stack frame layout calculation is broken
#endif

        GEN_BR_THUNK %r1

SYM_FUNC_START(arch_bpf_timed_may_goto)
        /*
         * This function has a special ABI: the parameters are in %r12 and
         * %r13; the return value is in %r12; all GPRs except %r0, %r1, and
         * %r12 are callee-saved; and the return address is in %r0.
         */
        stmg %r2,%r5,FRAME_OFF+R2_OFF(%r15)
        stg %r14,FRAME_OFF+R14_OFF(%r15)
        stg %r0,FRAME_OFF+RETADDR_OFF(%r15)
        stg %r15,FRAME_OFF+R15_OFF(%r15)
        lgr %r1,%r15
        lay %r15,-FRAME_SIZE(%r15)
        stg %r1,__SF_BACKCHAIN(%r15)

        lay %r2,0(%r12,%r13)
        brasl %r14,bpf_check_timed_may_goto
        lgr %r12,%r2

        lg %r15,FRAME_SIZE+FRAME_OFF+R15_OFF(%r15)
        lmg %r2,%r5,FRAME_OFF+R2_OFF(%r15)
        lg %r14,FRAME_OFF+R14_OFF(%r15)
        lg %r1,FRAME_OFF+RETADDR_OFF(%r15)
        BR_EX %r1
SYM_FUNC_END(arch_bpf_timed_may_goto)