// SPDX-License-Identifier: GPL-2.0

#include <linux/bpf.h>
#include <bpf/bpf_helpers.h>
#include "bpf_misc.h"

#ifdef CAN_USE_GOTOL

SEC("socket")
__description("gotol, small_imm")
__success __success_unpriv __retval(1)
__naked void gotol_small_imm(void)
{
        asm volatile ("                                 \
        call %[bpf_ktime_get_ns];                       \
        if r0 == 0 goto l0_%=;                          \
        gotol l1_%=;                                    \
l2_%=:                                                  \
        gotol l3_%=;                                    \
l1_%=:                                                  \
        r0 = 1;                                         \
        gotol l2_%=;                                    \
l0_%=:                                                  \
        r0 = 2;                                         \
l3_%=:                                                  \
        exit;                                           \
"       :
        : __imm(bpf_ktime_get_ns)
        : __clobber_all);
}

SEC("socket")
__description("gotol, large_imm")
__success __failure_unpriv __retval(40000)
__naked void gotol_large_imm(void)
{
        asm volatile ("                                 \
        gotol 1f;                                       \
0:                                                      \
        r0 = 0;                                         \
        .rept 40000;                                    \
        r0 += 1;                                        \
        .endr;                                          \
        exit;                                           \
1:      gotol 0b;                                       \
"       :
        :
        : __clobber_all);
}

#else

SEC("socket")
__description("cpuv4 is not supported by compiler or jit, use a dummy test")
__success
int dummy_test(void)
{
        return 0;
}

#endif

char _license[] SEC("license") = "GPL";