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

#include <vmlinux.h>
#include <bpf/bpf_tracing.h>
#include "bpf_misc.h"

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

int tid;
int i;

SEC("tp_btf/bpf_testmod_test_raw_tp_null_tp")
int BPF_PROG(test_raw_tp_null, struct sk_buff *skb)
{
        struct task_struct *task = bpf_get_current_task_btf();

        if (task->pid != tid)
                return 0;

        /* If dead code elimination kicks in, the increment +=2 will be
         * removed. For raw_tp programs attaching to tracepoints in kernel
         * modules, we mark input arguments as PTR_MAYBE_NULL, so branch
         * prediction should never kick in.
         */
        asm volatile ("%[i] += 1; if %[ctx] != 0 goto +1; %[i] += 2;"
                        : [i]"+r"(i)
                        : [ctx]"r"(skb)
                        : "memory");
        return 0;
}