Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
torvalds
GitHub Repository: torvalds/linux
 Path: blob/master/samples/bpf/ibumad_kern.c
25924 views
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


19
struct {

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


26
struct {

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/tracing/events/ib_umad/ib_umad_read/format

43
 * /sys/kernel/tracing/events/ib_umad/ib_umad_write/format

44
 */

45
struct 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


77
SEC("tracepoint/ib_umad/ib_umad_read_recv")

78
int 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
}

97
SEC("tracepoint/ib_umad/ib_umad_read_send")

98
int 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
}

117
SEC("tracepoint/ib_umad/ib_umad_write")

118
int 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


138
char _license[] SEC("license") = "GPL";

139


140