LXR linux/samples/bpf/ibumad

   1// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
   2
   3/*
   4 * ibumad BPF sample kernel side
   5 *
   6 * This program is free software; you can redistribute it and/or
   7 * modify it under the terms of version 2 of the GNU General Public
   8 * License as published by the Free Software Foundation.
   9 *
  10 * Copyright(c) 2018 Ira Weiny, Intel Corporation
  11 */
  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 DEBUG
  34#ifndef DEBUG
  35#undef bpf_printk
  36#define bpf_printk(fmt, ...)
  37#endif
  38
  39/* Taken from the current format defined in
  40 * include/trace/events/ib_umad.h
  41 * and
  42 * /sys/kernel/debug/tracing/events/ib_umad/ib_umad_read/format
  43 * /sys/kernel/debug/tracing/events/ib_umad/ib_umad_write/format
  44 */
  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