mirror of
https://mirrors.bfsu.edu.cn/git/linux.git
synced 2024-11-30 23:54:04 +08:00
7cf245a37e
Fix all files in samples/bpf to include libbpf header files with the bpf/
prefix, to be consistent with external users of the library. Also ensure
that all includes of exported libbpf header files (those that are exported
on 'make install' of the library) use bracketed includes instead of quoted.
To make sure no new files are introduced that doesn't include the bpf/
prefix in its include, remove tools/lib/bpf from the include path entirely,
and use tools/lib instead.
Fixes: 6910d7d386 ("selftests/bpf: Ensure bpf_helper_defs.h are taken from selftests dir")
Signed-off-by: Toke Høiland-Jørgensen <toke@redhat.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Acked-by: Jesper Dangaard Brouer <brouer@redhat.com>
Acked-by: Andrii Nakryiko <andriin@fb.com>
Link: https://lore.kernel.org/bpf/157952560911.1683545.8795966751309534150.stgit@toke.dk
139 lines
3.0 KiB
C
139 lines
3.0 KiB
C
// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
|
|
|
|
/**
|
|
* ibumad BPF sample kernel side
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of version 2 of the GNU General Public
|
|
* License as published by the Free Software Foundation.
|
|
*
|
|
* Copyright(c) 2018 Ira Weiny, Intel Corporation
|
|
*/
|
|
|
|
#define KBUILD_MODNAME "ibumad_count_pkts_by_class"
|
|
#include <uapi/linux/bpf.h>
|
|
|
|
#include <bpf/bpf_helpers.h>
|
|
|
|
|
|
struct bpf_map_def SEC("maps") read_count = {
|
|
.type = BPF_MAP_TYPE_ARRAY,
|
|
.key_size = sizeof(u32), /* class; u32 required */
|
|
.value_size = sizeof(u64), /* count of mads read */
|
|
.max_entries = 256, /* Room for all Classes */
|
|
};
|
|
|
|
struct bpf_map_def SEC("maps") write_count = {
|
|
.type = BPF_MAP_TYPE_ARRAY,
|
|
.key_size = sizeof(u32), /* class; u32 required */
|
|
.value_size = sizeof(u64), /* count of mads written */
|
|
.max_entries = 256, /* Room for all Classes */
|
|
};
|
|
|
|
#undef DEBUG
|
|
#ifndef DEBUG
|
|
#undef bpf_printk
|
|
#define bpf_printk(fmt, ...)
|
|
#endif
|
|
|
|
/* Taken from the current format defined in
|
|
* include/trace/events/ib_umad.h
|
|
* and
|
|
* /sys/kernel/debug/tracing/events/ib_umad/ib_umad_read/format
|
|
* /sys/kernel/debug/tracing/events/ib_umad/ib_umad_write/format
|
|
*/
|
|
struct ib_umad_rw_args {
|
|
u64 pad;
|
|
u8 port_num;
|
|
u8 sl;
|
|
u8 path_bits;
|
|
u8 grh_present;
|
|
u32 id;
|
|
u32 status;
|
|
u32 timeout_ms;
|
|
u32 retires;
|
|
u32 length;
|
|
u32 qpn;
|
|
u32 qkey;
|
|
u8 gid_index;
|
|
u8 hop_limit;
|
|
u16 lid;
|
|
u16 attr_id;
|
|
u16 pkey_index;
|
|
u8 base_version;
|
|
u8 mgmt_class;
|
|
u8 class_version;
|
|
u8 method;
|
|
u32 flow_label;
|
|
u16 mad_status;
|
|
u16 class_specific;
|
|
u32 attr_mod;
|
|
u64 tid;
|
|
u8 gid[16];
|
|
u32 dev_index;
|
|
u8 traffic_class;
|
|
};
|
|
|
|
SEC("tracepoint/ib_umad/ib_umad_read_recv")
|
|
int on_ib_umad_read_recv(struct ib_umad_rw_args *ctx)
|
|
{
|
|
u64 zero = 0, *val;
|
|
u8 class = ctx->mgmt_class;
|
|
|
|
bpf_printk("ib_umad read recv : class 0x%x\n", class);
|
|
|
|
val = bpf_map_lookup_elem(&read_count, &class);
|
|
if (!val) {
|
|
bpf_map_update_elem(&read_count, &class, &zero, BPF_NOEXIST);
|
|
val = bpf_map_lookup_elem(&read_count, &class);
|
|
if (!val)
|
|
return 0;
|
|
}
|
|
|
|
(*val) += 1;
|
|
|
|
return 0;
|
|
}
|
|
SEC("tracepoint/ib_umad/ib_umad_read_send")
|
|
int on_ib_umad_read_send(struct ib_umad_rw_args *ctx)
|
|
{
|
|
u64 zero = 0, *val;
|
|
u8 class = ctx->mgmt_class;
|
|
|
|
bpf_printk("ib_umad read send : class 0x%x\n", class);
|
|
|
|
val = bpf_map_lookup_elem(&read_count, &class);
|
|
if (!val) {
|
|
bpf_map_update_elem(&read_count, &class, &zero, BPF_NOEXIST);
|
|
val = bpf_map_lookup_elem(&read_count, &class);
|
|
if (!val)
|
|
return 0;
|
|
}
|
|
|
|
(*val) += 1;
|
|
|
|
return 0;
|
|
}
|
|
SEC("tracepoint/ib_umad/ib_umad_write")
|
|
int on_ib_umad_write(struct ib_umad_rw_args *ctx)
|
|
{
|
|
u64 zero = 0, *val;
|
|
u8 class = ctx->mgmt_class;
|
|
|
|
bpf_printk("ib_umad write : class 0x%x\n", class);
|
|
|
|
val = bpf_map_lookup_elem(&write_count, &class);
|
|
if (!val) {
|
|
bpf_map_update_elem(&write_count, &class, &zero, BPF_NOEXIST);
|
|
val = bpf_map_lookup_elem(&write_count, &class);
|
|
if (!val)
|
|
return 0;
|
|
}
|
|
|
|
(*val) += 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
char _license[] SEC("license") = "GPL";
|