root/tools/testing/selftests/bpf/progs/test_btf_haskv.c

DEFINITIONS

1. test_long_fname_2
2. test_long_fname_1
3. SEC

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 /* Copyright (c) 2018 Facebook */
   3 #include <linux/bpf.h>
   4 #include "bpf_helpers.h"
   5 
   6 int _version SEC("version") = 1;
   7 
   8 struct ipv_counts {
   9         unsigned int v4;
  10         unsigned int v6;
  11 };
  12 
  13 struct 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 
  20 BPF_ANNOTATE_KV_PAIR(btf_map, int, struct ipv_counts);
  21 
  22 struct dummy_tracepoint_args {
  23         unsigned long long pad;
  24         struct sock *sock;
  25 };
  26 
  27 __attribute__((noinline))
  28 static 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))
  46 static int test_long_fname_1(struct dummy_tracepoint_args *arg)
  47 {
  48         return test_long_fname_2(arg);
  49 }
  50 
  51 SEC("dummy_tracepoint")
  52 int _dummy_tracepoint(struct dummy_tracepoint_args *arg)
  53 {
  54         return test_long_fname_1(arg);
  55 }
  56 
  57 char _license[] SEC("license") = "GPL";