LXR linux/tools/testing/selftests/bpf/progs/test_btf

   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
   7int _version SEC("version") = 1;
   8
   9struct ipv_counts {
  10        unsigned int v4;
  11        unsigned int v6;
  12};
  13
  14/* just to validate we can handle maps in multiple sections */
  15struct 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
  22BPF_ANNOTATE_KV_PAIR(btf_map_legacy, int, struct ipv_counts);
  23
  24struct {
  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__attribute__((noinline))
  32int test_long_fname_2(void)
  33{
  34        struct ipv_counts *counts;
  35        int key = 0;
  36
  37        counts = bpf_map_lookup_elem(&btf_map, &key);
  38        if (!counts)
  39                return 0;
  40
  41        counts->v6++;
  42
  43        /* just verify we can reference both maps */
  44        counts = bpf_map_lookup_elem(&btf_map_legacy, &key);
  45        if (!counts)
  46                return 0;
  47
  48        return 0;
  49}
  50
  51__attribute__((noinline))
  52int test_long_fname_1(void)
  53{
  54        return test_long_fname_2();
  55}
  56
  57SEC("dummy_tracepoint")
  58int _dummy_tracepoint(void *arg)
  59{
  60        return test_long_fname_1();
  61}
  62
  63char _license[] SEC("license") = "GPL";
  64