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