Quelle  raw_tp_null.c   Sprache: C

// 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;
}