cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

ibumad_kern.c (2989B)

      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";