forked from Minki/linux
999d82cbc0
Change the bpf programs test_btf_haskv.c and test_btf_nokv.c to have two sections, and enhance test_btf.c test_file feature to test btf func_info returned by the kernel. Signed-off-by: Yonghong Song <yhs@fb.com> Signed-off-by: Martin KaFai Lau <kafai@fb.com> Signed-off-by: Alexei Starovoitov <ast@kernel.org>
58 lines
1.0 KiB
C
58 lines
1.0 KiB
C
/* SPDX-License-Identifier: GPL-2.0 */
|
|
/* Copyright (c) 2018 Facebook */
|
|
#include <linux/bpf.h>
|
|
#include "bpf_helpers.h"
|
|
|
|
int _version SEC("version") = 1;
|
|
|
|
struct ipv_counts {
|
|
unsigned int v4;
|
|
unsigned int v6;
|
|
};
|
|
|
|
struct bpf_map_def SEC("maps") btf_map = {
|
|
.type = BPF_MAP_TYPE_ARRAY,
|
|
.key_size = sizeof(int),
|
|
.value_size = sizeof(struct ipv_counts),
|
|
.max_entries = 4,
|
|
};
|
|
|
|
BPF_ANNOTATE_KV_PAIR(btf_map, int, struct ipv_counts);
|
|
|
|
struct dummy_tracepoint_args {
|
|
unsigned long long pad;
|
|
struct sock *sock;
|
|
};
|
|
|
|
__attribute__((noinline))
|
|
static int test_long_fname_2(struct dummy_tracepoint_args *arg)
|
|
{
|
|
struct ipv_counts *counts;
|
|
int key = 0;
|
|
|
|
if (!arg->sock)
|
|
return 0;
|
|
|
|
counts = bpf_map_lookup_elem(&btf_map, &key);
|
|
if (!counts)
|
|
return 0;
|
|
|
|
counts->v6++;
|
|
|
|
return 0;
|
|
}
|
|
|
|
__attribute__((noinline))
|
|
static int test_long_fname_1(struct dummy_tracepoint_args *arg)
|
|
{
|
|
return test_long_fname_2(arg);
|
|
}
|
|
|
|
SEC("dummy_tracepoint")
|
|
int _dummy_tracepoint(struct dummy_tracepoint_args *arg)
|
|
{
|
|
return test_long_fname_1(arg);
|
|
}
|
|
|
|
char _license[] SEC("license") = "GPL";
|