1
2
3
4
5
6#ifndef _RDMA_TOOL_RES_H_
7#define _RDMA_TOOL_RES_H_
8
9#include "rdma.h"
10
11int _res_send_msg(struct rd *rd, uint32_t command, mnl_cb_t callback);
12int _res_send_idx_msg(struct rd *rd, uint32_t command, mnl_cb_t callback,
13 uint32_t idx, uint32_t id);
14
15int res_pd_parse_cb(const struct nlmsghdr *nlh, void *data);
16int res_pd_idx_parse_cb(const struct nlmsghdr *nlh, void *data);
17int res_mr_parse_cb(const struct nlmsghdr *nlh, void *data);
18int res_mr_idx_parse_cb(const struct nlmsghdr *nlh, void *data);
19int res_cq_parse_cb(const struct nlmsghdr *nlh, void *data);
20int res_cq_idx_parse_cb(const struct nlmsghdr *nlh, void *data);
21int res_cm_id_parse_cb(const struct nlmsghdr *nlh, void *data);
22int res_cm_id_idx_parse_cb(const struct nlmsghdr *nlh, void *data);
23int res_qp_parse_cb(const struct nlmsghdr *nlh, void *data);
24int res_qp_idx_parse_cb(const struct nlmsghdr *nlh, void *data);
25
26static inline uint32_t res_get_command(uint32_t command, struct rd *rd)
27{
28 if (!rd->show_raw)
29 return command;
30
31 switch (command) {
32 case RDMA_NLDEV_CMD_RES_QP_GET:
33 return RDMA_NLDEV_CMD_RES_QP_GET_RAW;
34 case RDMA_NLDEV_CMD_RES_CQ_GET:
35 return RDMA_NLDEV_CMD_RES_CQ_GET_RAW;
36 case RDMA_NLDEV_CMD_RES_MR_GET:
37 return RDMA_NLDEV_CMD_RES_MR_GET_RAW;
38 default:
39 return command;
40 }
41}
42
43#define RES_FUNC(name, command, valid_filters, strict_port, id) \
44 static inline int _##name(struct rd *rd) \
45 { \
46 uint32_t idx, _command; \
47 int ret; \
48 _command = res_get_command(command, rd); \
49 if (id) { \
50 ret = rd_doit_index(rd, &idx); \
51 if (ret) { \
52 rd->suppress_errors = true; \
53 ret = _res_send_idx_msg(rd, _command, \
54 name##_idx_parse_cb, \
55 idx, id); \
56 if (!ret || rd->show_raw) \
57 return ret; \
58
59
60 \
61 } \
62 } \
63 return _res_send_msg(rd, _command, name##_parse_cb); \
64 } \
65 static inline int name(struct rd *rd) \
66 { \
67 int ret = rd_build_filter(rd, valid_filters); \
68 if (ret) \
69 return ret; \
70 if ((uintptr_t)valid_filters != (uintptr_t)NULL) { \
71 ret = rd_set_arg_to_devname(rd); \
72 if (ret) \
73 return ret; \
74 } \
75 if (strict_port) \
76 return rd_exec_dev(rd, _##name); \
77 else \
78 return rd_exec_link(rd, _##name, strict_port); \
79 }
80
81static const
82struct filters pd_valid_filters[MAX_NUMBER_OF_FILTERS] = {
83 { .name = "dev", .is_number = false },
84 { .name = "users", .is_number = true },
85 { .name = "pid", .is_number = true },
86 { .name = "ctxn", .is_number = true },
87 { .name = "pdn", .is_number = true, .is_doit = true },
88 { .name = "ctxn", .is_number = true }
89};
90
91RES_FUNC(res_pd, RDMA_NLDEV_CMD_RES_PD_GET, pd_valid_filters, true,
92 RDMA_NLDEV_ATTR_RES_PDN);
93
94static const
95struct filters mr_valid_filters[MAX_NUMBER_OF_FILTERS] = {
96 { .name = "dev", .is_number = false },
97 { .name = "rkey", .is_number = true },
98 { .name = "lkey", .is_number = true },
99 { .name = "mrlen", .is_number = true },
100 { .name = "pid", .is_number = true },
101 { .name = "mrn", .is_number = true, .is_doit = true },
102 { .name = "pdn", .is_number = true }
103};
104
105RES_FUNC(res_mr, RDMA_NLDEV_CMD_RES_MR_GET, mr_valid_filters, true,
106 RDMA_NLDEV_ATTR_RES_MRN);
107
108static const
109struct filters cq_valid_filters[MAX_NUMBER_OF_FILTERS] = {
110 { .name = "dev", .is_number = false },
111 { .name = "users", .is_number = true },
112 { .name = "poll-ctx", .is_number = false },
113 { .name = "pid", .is_number = true },
114 { .name = "cqn", .is_number = true, .is_doit = true },
115 { .name = "ctxn", .is_number = true }
116};
117
118RES_FUNC(res_cq, RDMA_NLDEV_CMD_RES_CQ_GET, cq_valid_filters, true,
119 RDMA_NLDEV_ATTR_RES_CQN);
120
121static const
122struct filters cm_id_valid_filters[MAX_NUMBER_OF_FILTERS] = {
123 { .name = "link", .is_number = false },
124 { .name = "lqpn", .is_number = true },
125 { .name = "qp-type", .is_number = false },
126 { .name = "state", .is_number = false },
127 { .name = "ps", .is_number = false },
128 { .name = "dev-type", .is_number = false },
129 { .name = "transport-type", .is_number = false },
130 { .name = "pid", .is_number = true },
131 { .name = "src-addr", .is_number = false },
132 { .name = "src-port", .is_number = true },
133 { .name = "dst-addr", .is_number = false },
134 { .name = "dst-port", .is_number = true },
135 { .name = "cm-idn", .is_number = true, .is_doit = true }
136};
137
138RES_FUNC(res_cm_id, RDMA_NLDEV_CMD_RES_CM_ID_GET, cm_id_valid_filters, false,
139 RDMA_NLDEV_ATTR_RES_CM_IDN);
140
141static const struct
142filters qp_valid_filters[MAX_NUMBER_OF_FILTERS] = {
143 { .name = "link", .is_number = false },
144 { .name = "lqpn", .is_number = true, .is_doit = true },
145 { .name = "rqpn", .is_number = true },
146 { .name = "pid", .is_number = true },
147 { .name = "sq-psn", .is_number = true },
148 { .name = "rq-psn", .is_number = true },
149 { .name = "type", .is_number = false },
150 { .name = "path-mig-state", .is_number = false },
151 { .name = "state", .is_number = false },
152 { .name = "pdn", .is_number = true },
153};
154
155RES_FUNC(res_qp, RDMA_NLDEV_CMD_RES_QP_GET, qp_valid_filters, false,
156 RDMA_NLDEV_ATTR_RES_LQPN);
157
158char *get_task_name(uint32_t pid);
159void print_dev(struct rd *rd, uint32_t idx, const char *name);
160void print_link(struct rd *rd, uint32_t idx, const char *name, uint32_t port,
161 struct nlattr **nla_line);
162void print_key(struct rd *rd, const char *name, uint64_t val,
163 struct nlattr *nlattr);
164void res_print_uint(struct rd *rd, const char *name, uint64_t val,
165 struct nlattr *nlattr);
166void print_comm(struct rd *rd, const char *str, struct nlattr **nla_line);
167const char *qp_types_to_str(uint8_t idx);
168void print_qp_type(struct rd *rd, uint32_t val);
169#endif
170
