// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) 2021. Huawei Technologies Co., Ltd */
#include "vmlinux.h"
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>

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

SEC("struct_ops/test_1")
int BPF_PROG(test_1, struct bpf_dummy_ops_state *state)
{
        int ret;

        /* Check that 'state' nullable status is detected correctly.
         * If 'state' argument would be assumed non-null by verifier
         * the code below would be deleted as dead (which it shouldn't).
         * Hide it from the compiler behind 'asm' block to avoid
         * unnecessary optimizations.
         */
        asm volatile (
                "if %[state] != 0 goto +2;"
                "r0 = 0xf2f3f4f5;"
                "exit;"
        ::[state]"p"(state));

        ret = state->val;
        state->val = 0x5a;
        return ret;
}

__u64 test_2_args[5];

SEC("struct_ops/test_2")
int BPF_PROG(test_2, struct bpf_dummy_ops_state *state, int a1, unsigned short a2,
             char a3, unsigned long a4)
{
        test_2_args[0] = state->val;
        test_2_args[1] = a1;
        test_2_args[2] = a2;
        test_2_args[3] = a3;
        test_2_args[4] = a4;
        return 0;
}

SEC("struct_ops.s/test_sleepable")
int BPF_PROG(test_sleepable, struct bpf_dummy_ops_state *state)
{
        return 0;
}

SEC(".struct_ops")
struct bpf_dummy_ops dummy_1 = {
        .test_1 = (void *)test_1,
        .test_2 = (void *)test_2,
        .test_sleepable = (void *)test_sleepable,
};