// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2026 Meta Platforms, Inc. and affiliates. */

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

/* linear chain main -> A -> B */
__naked __noinline __used
static unsigned long linear_b(void)
{
        asm volatile (
                "r0 = 42;"
                "exit;"
        );
}

__naked __noinline __used
static unsigned long linear_a(void)
{
        asm volatile (
                "call linear_b;"
                "exit;"
        );
}

SEC("?raw_tp")
__success __log_level(2)
__msg("topo_order[0] = linear_b")
__msg("topo_order[1] = linear_a")
__msg("topo_order[2] = topo_linear")
__naked int topo_linear(void)
{
        asm volatile (
                "call linear_a;"
                "exit;"
        );
}

/* diamond main -> A, main -> B, A -> C, B -> C */
__naked __noinline __used
static unsigned long diamond_c(void)
{
        asm volatile (
                "r0 = 1;"
                "exit;"
        );
}

__naked __noinline __used
static unsigned long diamond_b(void)
{
        asm volatile (
                "call diamond_c;"
                "exit;"
        );
}

__naked __noinline __used
static unsigned long diamond_a(void)
{
        asm volatile (
                "call diamond_c;"
                "exit;"
        );
}

SEC("?raw_tp")
__success __log_level(2)
__msg("topo_order[0] = diamond_c")
__msg("topo_order[3] = topo_diamond")
__naked int topo_diamond(void)
{
        asm volatile (
                "call diamond_a;"
                "call diamond_b;"
                "exit;"
        );
}

/* main -> global_a (global) -> static_leaf (static, leaf) */
__naked __noinline __used
static unsigned long static_leaf(void)
{
        asm volatile (
                "r0 = 7;"
                "exit;"
        );
}

__noinline __used
int global_a(int x)
{
        return static_leaf();
}

SEC("?raw_tp")
__success __log_level(2)
__msg("topo_order[0] = static_leaf")
__msg("topo_order[1] = global_a")
__msg("topo_order[2] = topo_mixed")
__naked int topo_mixed(void)
{
        asm volatile (
                "r1 = 0;"
                "call global_a;"
                "exit;"
        );
}

/*
 * shared static callee from global and main:
 * main -> shared_leaf (static)
 * main -> global_b (global) -> shared_leaf (static)
 */
__naked __noinline __used
static unsigned long shared_leaf(void)
{
        asm volatile (
                "r0 = 99;"
                "exit;"
        );
}

__noinline __used
int global_b(int x)
{
        return shared_leaf();
}

SEC("?raw_tp")
__success __log_level(2)
__msg("topo_order[0] = shared_leaf")
__msg("topo_order[1] = global_b")
__msg("topo_order[2] = topo_shared")
__naked int topo_shared(void)
{
        asm volatile (
                "call shared_leaf;"
                "r1 = 0;"
                "call global_b;"
                "exit;"
        );
}

/* duplicate calls to the same subprog */
__naked __noinline __used
static unsigned long dup_leaf(void)
{
        asm volatile (
                "r0 = 0;"
                "exit;"
        );
}

SEC("?raw_tp")
__success __log_level(2)
__msg("topo_order[0] = dup_leaf")
__msg("topo_order[1] = topo_dup_calls")
__naked int topo_dup_calls(void)
{
        asm volatile (
                "call dup_leaf;"
                "call dup_leaf;"
                "exit;"
        );
}

/* main calls bpf_loop() with loop_cb as the callback */
static int loop_cb(int idx, void *ctx)
{
        return 0;
}

SEC("?raw_tp")
__success __log_level(2)
__msg("topo_order[0] = loop_cb")
__msg("topo_order[1] = topo_loop_cb")
int topo_loop_cb(void)
{
        bpf_loop(1, loop_cb, NULL, 0);
        return 0;
}

/*
 * bpf_loop callback calling another subprog
 * main -> bpf_loop(callback=loop_cb2) -> loop_cb2 -> loop_cb2_leaf
 */
__naked __noinline __used
static unsigned long loop_cb2_leaf(void)
{
        asm volatile (
                "r0 = 0;"
                "exit;"
        );
}

static int loop_cb2(int idx, void *ctx)
{
        return loop_cb2_leaf();
}

SEC("?raw_tp")
__success __log_level(2)
__msg("topo_order[0] = loop_cb2_leaf")
__msg("topo_order[1] = loop_cb2")
__msg("topo_order[2] = topo_loop_cb_chain")
int topo_loop_cb_chain(void)
{
        bpf_loop(1, loop_cb2, NULL, 0);
        return 0;
}

/* no calls (single subprog) */
SEC("?raw_tp")
__success __log_level(2)
__msg("topo_order[0] = topo_no_calls")
__naked int topo_no_calls(void)
{
        asm volatile (
                "r0 = 0;"
                "exit;"
        );
}

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