linux/tools/testing/selftests/bpf/progs/test_btf_haskv.c
<<
>>
Prefs
   1/* SPDX-License-Identifier: GPL-2.0 */
   2/* Copyright (c) 2018 Facebook */
   3#include <linux/bpf.h>
   4#include "bpf_helpers.h"
   5
   6int _version SEC("version") = 1;
   7
   8struct ipv_counts {
   9        unsigned int v4;
  10        unsigned int v6;
  11};
  12
  13struct bpf_map_def SEC("maps") btf_map = {
  14        .type = BPF_MAP_TYPE_ARRAY,
  15        .key_size = sizeof(int),
  16        .value_size = sizeof(struct ipv_counts),
  17        .max_entries = 4,
  18};
  19
  20BPF_ANNOTATE_KV_PAIR(btf_map, int, struct ipv_counts);
  21
  22struct dummy_tracepoint_args {
  23        unsigned long long pad;
  24        struct sock *sock;
  25};
  26
  27__attribute__((noinline))
  28static int test_long_fname_2(struct dummy_tracepoint_args *arg)
  29{
  30        struct ipv_counts *counts;
  31        int key = 0;
  32
  33        if (!arg->sock)
  34                return 0;
  35
  36        counts = bpf_map_lookup_elem(&btf_map, &key);
  37        if (!counts)
  38                return 0;
  39
  40        counts->v6++;
  41
  42        return 0;
  43}
  44
  45__attribute__((noinline))
  46static int test_long_fname_1(struct dummy_tracepoint_args *arg)
  47{
  48        return test_long_fname_2(arg);
  49}
  50
  51SEC("dummy_tracepoint")
  52int _dummy_tracepoint(struct dummy_tracepoint_args *arg)
  53{
  54        return test_long_fname_1(arg);
  55}
  56
  57char _license[] SEC("license") = "GPL";
  58