tools/testing/selftests/bpf/progs/test_btf_newkv.c

   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 /* just to validate we can handle maps in multiple sections */
  14 struct bpf_map_def SEC("maps") btf_map_legacy = {
  15         .type = BPF_MAP_TYPE_ARRAY,
  16         .key_size = sizeof(int),
  17         .value_size = sizeof(long long),
  18         .max_entries = 4,
  19 };
  20
  21 BPF_ANNOTATE_KV_PAIR(btf_map_legacy, int, struct ipv_counts);
  22
  23 struct {
  24         __uint(type, BPF_MAP_TYPE_ARRAY);
  25         __uint(max_entries, 4);
  26         __type(key, int);
  27         __type(value, struct ipv_counts);
  28 } btf_map SEC(".maps");
  29
  30 struct dummy_tracepoint_args {
  31         unsigned long long pad;
  32         struct sock *sock;
  33 };
  34
  35 __attribute__((noinline))
  36 static int test_long_fname_2(struct dummy_tracepoint_args *arg)
  37 {
  38         struct ipv_counts *counts;
  39         int key = 0;
  40
  41         if (!arg->sock)
  42                 return 0;
  43
  44         counts = bpf_map_lookup_elem(&btf_map, &key);
  45         if (!counts)
  46                 return 0;
  47
  48         counts->v6++;
  49
  50         /* just verify we can reference both maps */
  51         counts = bpf_map_lookup_elem(&btf_map_legacy, &key);
  52         if (!counts)
  53                 return 0;
  54
  55         return 0;
  56 }
  57
  58 __attribute__((noinline))
  59 static int test_long_fname_1(struct dummy_tracepoint_args *arg)
  60 {
  61         return test_long_fname_2(arg);
  62 }
  63
  64 SEC("dummy_tracepoint")
  65 int _dummy_tracepoint(struct dummy_tracepoint_args *arg)
  66 {
  67         return test_long_fname_1(arg);
  68 }
  69
  70 char _license[] SEC("license") = "GPL";