/* SPDX-License-Identifier: GPL-2.0 */
#include <signal.h>
#include <stdlib.h>
#include <linux/compiler.h>
#include <unistd.h>
#include "../tests.h"

/* We want to check these symbols in perf script */
noinline void leaf(void);
noinline void parent(void);

static volatile sig_atomic_t done asm("leafloop_done");

static void sighandler(int sig __maybe_unused)
{
        done = 1;
}

#if defined(__aarch64__)
/*
 * Write leaf() in assembly so it stays as a minimal leaf function with no
 * stack frame and won't get silently broken in the future by any Perf wide
 * compilation options like -fstack-protector-all.
 */
asm(
        ".pushsection .text,\"ax\",%progbits\n"
        ".global leaf\n"
        ".type leaf, %function\n"
        "leaf:\n"
        "       adrp    x1, leafloop_done\n"
        "       ldr     w2, [x1, #:lo12:leafloop_done]\n"
        "       cbz     w2, leaf\n"
        "       ret\n"
        ".size leaf, .-leaf\n"
        ".popsection\n"
);

#else

noinline void leaf(void)
{
        while (!done)
                ;
}

#endif

noinline void parent(void)
{
        leaf();
}

static int leafloop(int argc, const char **argv)
{
        int sec = 1;

        if (argc > 0)
                sec = atoi(argv[0]);

        signal(SIGINT, sighandler);
        signal(SIGALRM, sighandler);
        alarm(sec);

        parent();
        return 0;
}

DEFINE_WORKLOAD(leafloop);