139 lines · c
1// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB2 3/*4 * ibumad BPF sample kernel side5 *6 * This program is free software; you can redistribute it and/or7 * modify it under the terms of version 2 of the GNU General Public8 * License as published by the Free Software Foundation.9 *10 * Copyright(c) 2018 Ira Weiny, Intel Corporation11 */12 13#define KBUILD_MODNAME "ibumad_count_pkts_by_class"14#include <uapi/linux/bpf.h>15 16#include <bpf/bpf_helpers.h>17 18 19struct {20 __uint(type, BPF_MAP_TYPE_ARRAY);21 __type(key, u32); /* class; u32 required */22 __type(value, u64); /* count of mads read */23 __uint(max_entries, 256); /* Room for all Classes */24} read_count SEC(".maps");25 26struct {27 __uint(type, BPF_MAP_TYPE_ARRAY);28 __type(key, u32); /* class; u32 required */29 __type(value, u64); /* count of mads written */30 __uint(max_entries, 256); /* Room for all Classes */31} write_count SEC(".maps");32 33#undef DEBUG34#ifndef DEBUG35#undef bpf_printk36#define bpf_printk(fmt, ...)37#endif38 39/* Taken from the current format defined in40 * include/trace/events/ib_umad.h41 * and42 * /sys/kernel/tracing/events/ib_umad/ib_umad_read/format43 * /sys/kernel/tracing/events/ib_umad/ib_umad_write/format44 */45struct ib_umad_rw_args {46 u64 pad;47 u8 port_num;48 u8 sl;49 u8 path_bits;50 u8 grh_present;51 u32 id;52 u32 status;53 u32 timeout_ms;54 u32 retires;55 u32 length;56 u32 qpn;57 u32 qkey;58 u8 gid_index;59 u8 hop_limit;60 u16 lid;61 u16 attr_id;62 u16 pkey_index;63 u8 base_version;64 u8 mgmt_class;65 u8 class_version;66 u8 method;67 u32 flow_label;68 u16 mad_status;69 u16 class_specific;70 u32 attr_mod;71 u64 tid;72 u8 gid[16];73 u32 dev_index;74 u8 traffic_class;75};76 77SEC("tracepoint/ib_umad/ib_umad_read_recv")78int on_ib_umad_read_recv(struct ib_umad_rw_args *ctx)79{80 u64 zero = 0, *val;81 u8 class = ctx->mgmt_class;82 83 bpf_printk("ib_umad read recv : class 0x%x\n", class);84 85 val = bpf_map_lookup_elem(&read_count, &class);86 if (!val) {87 bpf_map_update_elem(&read_count, &class, &zero, BPF_NOEXIST);88 val = bpf_map_lookup_elem(&read_count, &class);89 if (!val)90 return 0;91 }92 93 (*val) += 1;94 95 return 0;96}97SEC("tracepoint/ib_umad/ib_umad_read_send")98int on_ib_umad_read_send(struct ib_umad_rw_args *ctx)99{100 u64 zero = 0, *val;101 u8 class = ctx->mgmt_class;102 103 bpf_printk("ib_umad read send : class 0x%x\n", class);104 105 val = bpf_map_lookup_elem(&read_count, &class);106 if (!val) {107 bpf_map_update_elem(&read_count, &class, &zero, BPF_NOEXIST);108 val = bpf_map_lookup_elem(&read_count, &class);109 if (!val)110 return 0;111 }112 113 (*val) += 1;114 115 return 0;116}117SEC("tracepoint/ib_umad/ib_umad_write")118int on_ib_umad_write(struct ib_umad_rw_args *ctx)119{120 u64 zero = 0, *val;121 u8 class = ctx->mgmt_class;122 123 bpf_printk("ib_umad write : class 0x%x\n", class);124 125 val = bpf_map_lookup_elem(&write_count, &class);126 if (!val) {127 bpf_map_update_elem(&write_count, &class, &zero, BPF_NOEXIST);128 val = bpf_map_lookup_elem(&write_count, &class);129 if (!val)130 return 0;131 }132 133 (*val) += 1;134 135 return 0;136}137 138char _license[] SEC("license") = "GPL";139