4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version
7 * 2 of the License, or (at your option) any later version.
9 * Authors: Leon Romanovsky <leonro@mellanox.com>
15 int rd_argc(struct rd
*rd
)
20 char *rd_argv(struct rd
*rd
)
27 int strcmpx(const char *str1
, const char *str2
)
29 if (strlen(str1
) > strlen(str2
))
31 return strncmp(str1
, str2
, strlen(str1
));
34 static bool rd_argv_match(struct rd
*rd
, const char *pattern
)
38 return strcmpx(rd_argv(rd
), pattern
) == 0;
41 void rd_arg_inc(struct rd
*rd
)
49 bool rd_no_arg(struct rd
*rd
)
51 return rd_argc(rd
) == 0;
55 * Possible input:output
56 * dev/port | first port | is_dump_all
59 * mlx5_1/0 | 0 | false
60 * mlx5_1/1 | 1 | false
61 * mlx5_1/- | 0 | false
63 * In strict mode, /- will return error.
65 static int get_port_from_argv(struct rd
*rd
, uint32_t *port
,
66 bool *is_dump_all
, bool strict_port
)
73 slash
= strchr(rd_argv(rd
), '/');
74 /* if no port found, return 0 */
83 if (isdigit(*slash
)) {
87 if (!*port
&& strlen(slash
))
93 static struct dev_map
*dev_map_alloc(const char *dev_name
)
95 struct dev_map
*dev_map
;
97 dev_map
= calloc(1, sizeof(*dev_map
));
100 dev_map
->dev_name
= strdup(dev_name
);
101 if (!dev_map
->dev_name
) {
109 static void dev_map_cleanup(struct rd
*rd
)
111 struct dev_map
*dev_map
, *tmp
;
113 list_for_each_entry_safe(dev_map
, tmp
,
114 &rd
->dev_map_list
, list
) {
115 list_del(&dev_map
->list
);
116 free(dev_map
->dev_name
);
121 static int add_filter(struct rd
*rd
, char *key
, char *value
,
122 const struct filters valid_filters
[])
124 char cset
[] = "1234567890,-";
125 struct filter_entry
*fe
;
126 bool key_found
= false;
130 fe
= calloc(1, sizeof(*fe
));
134 while (idx
< MAX_NUMBER_OF_FILTERS
&& valid_filters
[idx
].name
) {
135 if (!strcmpx(key
, valid_filters
[idx
].name
)) {
142 pr_err("Unsupported filter option: %s\n", key
);
148 * Check the filter validity, not optimal, but works
150 * Actually, there are three types of filters
151 * numeric - for example PID or QPN
152 * string - for example states
153 * link - user requested to filter on specific link
154 * e.g. mlx5_1/1, mlx5_1/-, mlx5_1 ...
156 if (valid_filters
[idx
].is_number
&&
157 strspn(value
, cset
) != strlen(value
)) {
158 pr_err("%s filter accepts \"%s\" characters only\n", key
, cset
);
163 fe
->key
= strdup(key
);
164 fe
->value
= strdup(value
);
165 if (!fe
->key
|| !fe
->value
) {
170 for (idx
= 0; idx
< strlen(fe
->value
); idx
++)
171 fe
->value
[idx
] = tolower(fe
->value
[idx
]);
173 list_add_tail(&fe
->list
, &rd
->filter_list
);
184 int rd_build_filter(struct rd
*rd
, const struct filters valid_filters
[])
189 if (!valid_filters
|| !rd_argc(rd
))
192 if (rd_argc(rd
) == 1) {
193 pr_err("No filter data was supplied to filter option %s\n", rd_argv(rd
));
198 if (rd_argc(rd
) % 2) {
199 pr_err("There is filter option without data\n");
204 while (idx
!= rd_argc(rd
)) {
206 * We can do micro-optimization and skip "dev"
207 * and "link" filters, but it is not worth of it.
209 ret
= add_filter(rd
, *(rd
->argv
+ idx
),
210 *(rd
->argv
+ idx
+ 1), valid_filters
);
220 bool rd_check_is_key_exist(struct rd
*rd
, const char *key
)
222 struct filter_entry
*fe
;
224 list_for_each_entry(fe
, &rd
->filter_list
, list
) {
225 if (!strcmpx(fe
->key
, key
))
233 * Check if string entry is filtered:
234 * * key doesn't exist -> user didn't request -> not filtered
236 bool rd_check_is_string_filtered(struct rd
*rd
,
237 const char *key
, const char *val
)
239 bool key_is_filtered
= false;
240 struct filter_entry
*fe
;
244 list_for_each_entry(fe
, &rd
->filter_list
, list
) {
245 if (!strcmpx(fe
->key
, key
)) {
246 /* We found the key */
247 p
= strdup(fe
->value
);
248 key_is_filtered
= true;
251 * Something extremely wrong if we fail
252 * to allocate small amount of bytes.
254 pr_err("Found key, but failed to allocate memory to store value\n");
255 return key_is_filtered
;
259 * Need to check if value in range
260 * It can come in the following formats
261 * and their permutations:
265 str
= strtok(p
, ",");
267 if (strlen(str
) == strlen(val
) &&
268 !strcasecmp(str
, val
)) {
269 key_is_filtered
= false;
272 str
= strtok(NULL
, ",");
280 return key_is_filtered
;
284 * Check if key is filtered:
285 * key doesn't exist -> user didn't request -> not filtered
287 bool rd_check_is_filtered(struct rd
*rd
, const char *key
, uint32_t val
)
289 bool key_is_filtered
= false;
290 struct filter_entry
*fe
;
292 list_for_each_entry(fe
, &rd
->filter_list
, list
) {
293 uint32_t left_val
= 0, fe_value
= 0;
294 bool range_check
= false;
297 if (!strcmpx(fe
->key
, key
)) {
298 /* We found the key */
299 key_is_filtered
= true;
301 * Need to check if value in range
302 * It can come in the following formats
303 * (and their permutations):
308 * numb1,numb2-numb3,numb4-numb5
312 fe_value
= strtol(p
, &p
, 10);
313 if (fe_value
== val
||
314 (range_check
&& left_val
< val
&&
316 key_is_filtered
= false;
333 return key_is_filtered
;
336 static void filters_cleanup(struct rd
*rd
)
338 struct filter_entry
*fe
, *tmp
;
340 list_for_each_entry_safe(fe
, tmp
,
341 &rd
->filter_list
, list
) {
349 static const enum mnl_attr_data_type nldev_policy
[RDMA_NLDEV_ATTR_MAX
] = {
350 [RDMA_NLDEV_ATTR_DEV_INDEX
] = MNL_TYPE_U32
,
351 [RDMA_NLDEV_ATTR_DEV_NAME
] = MNL_TYPE_NUL_STRING
,
352 [RDMA_NLDEV_ATTR_PORT_INDEX
] = MNL_TYPE_U32
,
353 [RDMA_NLDEV_ATTR_CAP_FLAGS
] = MNL_TYPE_U64
,
354 [RDMA_NLDEV_ATTR_FW_VERSION
] = MNL_TYPE_NUL_STRING
,
355 [RDMA_NLDEV_ATTR_NODE_GUID
] = MNL_TYPE_U64
,
356 [RDMA_NLDEV_ATTR_SYS_IMAGE_GUID
] = MNL_TYPE_U64
,
357 [RDMA_NLDEV_ATTR_LID
] = MNL_TYPE_U32
,
358 [RDMA_NLDEV_ATTR_SM_LID
] = MNL_TYPE_U32
,
359 [RDMA_NLDEV_ATTR_LMC
] = MNL_TYPE_U8
,
360 [RDMA_NLDEV_ATTR_PORT_STATE
] = MNL_TYPE_U8
,
361 [RDMA_NLDEV_ATTR_PORT_PHYS_STATE
] = MNL_TYPE_U8
,
362 [RDMA_NLDEV_ATTR_DEV_NODE_TYPE
] = MNL_TYPE_U8
,
363 [RDMA_NLDEV_ATTR_RES_SUMMARY
] = MNL_TYPE_NESTED
,
364 [RDMA_NLDEV_ATTR_RES_SUMMARY_ENTRY
] = MNL_TYPE_NESTED
,
365 [RDMA_NLDEV_ATTR_RES_SUMMARY_ENTRY_NAME
] = MNL_TYPE_NUL_STRING
,
366 [RDMA_NLDEV_ATTR_RES_SUMMARY_ENTRY_CURR
] = MNL_TYPE_U64
,
367 [RDMA_NLDEV_ATTR_RES_QP
] = MNL_TYPE_NESTED
,
368 [RDMA_NLDEV_ATTR_RES_QP_ENTRY
] = MNL_TYPE_NESTED
,
369 [RDMA_NLDEV_ATTR_RES_LQPN
] = MNL_TYPE_U32
,
370 [RDMA_NLDEV_ATTR_RES_RQPN
] = MNL_TYPE_U32
,
371 [RDMA_NLDEV_ATTR_RES_RQ_PSN
] = MNL_TYPE_U32
,
372 [RDMA_NLDEV_ATTR_RES_SQ_PSN
] = MNL_TYPE_U32
,
373 [RDMA_NLDEV_ATTR_RES_PATH_MIG_STATE
] = MNL_TYPE_U8
,
374 [RDMA_NLDEV_ATTR_RES_TYPE
] = MNL_TYPE_U8
,
375 [RDMA_NLDEV_ATTR_RES_STATE
] = MNL_TYPE_U8
,
376 [RDMA_NLDEV_ATTR_RES_PID
] = MNL_TYPE_U32
,
377 [RDMA_NLDEV_ATTR_RES_KERN_NAME
] = MNL_TYPE_NUL_STRING
,
378 [RDMA_NLDEV_ATTR_RES_CM_ID
] = MNL_TYPE_NESTED
,
379 [RDMA_NLDEV_ATTR_RES_CM_ID_ENTRY
] = MNL_TYPE_NESTED
,
380 [RDMA_NLDEV_ATTR_RES_PS
] = MNL_TYPE_U32
,
381 [RDMA_NLDEV_ATTR_RES_SRC_ADDR
] = MNL_TYPE_UNSPEC
,
382 [RDMA_NLDEV_ATTR_RES_DST_ADDR
] = MNL_TYPE_UNSPEC
,
383 [RDMA_NLDEV_ATTR_RES_CQ
] = MNL_TYPE_NESTED
,
384 [RDMA_NLDEV_ATTR_RES_CQ_ENTRY
] = MNL_TYPE_NESTED
,
385 [RDMA_NLDEV_ATTR_RES_CQE
] = MNL_TYPE_U32
,
386 [RDMA_NLDEV_ATTR_RES_USECNT
] = MNL_TYPE_U64
,
387 [RDMA_NLDEV_ATTR_RES_POLL_CTX
] = MNL_TYPE_U8
,
388 [RDMA_NLDEV_ATTR_RES_MR
] = MNL_TYPE_NESTED
,
389 [RDMA_NLDEV_ATTR_RES_MR_ENTRY
] = MNL_TYPE_NESTED
,
390 [RDMA_NLDEV_ATTR_RES_RKEY
] = MNL_TYPE_U32
,
391 [RDMA_NLDEV_ATTR_RES_LKEY
] = MNL_TYPE_U32
,
392 [RDMA_NLDEV_ATTR_RES_IOVA
] = MNL_TYPE_U64
,
393 [RDMA_NLDEV_ATTR_RES_MRLEN
] = MNL_TYPE_U64
,
396 int rd_attr_cb(const struct nlattr
*attr
, void *data
)
398 const struct nlattr
**tb
= data
;
401 if (mnl_attr_type_valid(attr
, RDMA_NLDEV_ATTR_MAX
) < 0)
402 /* We received uknown attribute */
405 type
= mnl_attr_get_type(attr
);
407 if (mnl_attr_validate(attr
, nldev_policy
[type
]) < 0)
414 int rd_dev_init_cb(const struct nlmsghdr
*nlh
, void *data
)
416 struct nlattr
*tb
[RDMA_NLDEV_ATTR_MAX
] = {};
417 struct dev_map
*dev_map
;
418 struct rd
*rd
= data
;
419 const char *dev_name
;
421 mnl_attr_parse(nlh
, 0, rd_attr_cb
, tb
);
422 if (!tb
[RDMA_NLDEV_ATTR_DEV_NAME
] || !tb
[RDMA_NLDEV_ATTR_DEV_INDEX
])
424 if (!tb
[RDMA_NLDEV_ATTR_PORT_INDEX
]) {
425 pr_err("This tool doesn't support switches yet\n");
429 dev_name
= mnl_attr_get_str(tb
[RDMA_NLDEV_ATTR_DEV_NAME
]);
431 dev_map
= dev_map_alloc(dev_name
);
433 /* The main function will cleanup the allocations */
435 list_add_tail(&dev_map
->list
, &rd
->dev_map_list
);
437 dev_map
->num_ports
= mnl_attr_get_u32(tb
[RDMA_NLDEV_ATTR_PORT_INDEX
]);
438 dev_map
->idx
= mnl_attr_get_u32(tb
[RDMA_NLDEV_ATTR_DEV_INDEX
]);
442 void rd_free(struct rd
*rd
)
451 int rd_set_arg_to_devname(struct rd
*rd
)
455 while (!rd_no_arg(rd
)) {
456 if (rd_argv_match(rd
, "dev") || rd_argv_match(rd
, "link")) {
459 pr_err("No device name was supplied\n");
470 int rd_exec_link(struct rd
*rd
, int (*cb
)(struct rd
*rd
), bool strict_port
)
472 struct dev_map
*dev_map
;
477 jsonw_start_array(rd
->jw
);
479 list_for_each_entry(dev_map
, &rd
->dev_map_list
, list
) {
480 rd
->dev_idx
= dev_map
->idx
;
481 port
= (strict_port
) ? 1 : 0;
482 for (; port
< dev_map
->num_ports
+ 1; port
++) {
493 dev_map
= dev_map_lookup(rd
, true);
494 ret
= get_port_from_argv(rd
, &port
, &is_dump_all
, strict_port
);
495 if (!dev_map
|| port
> dev_map
->num_ports
|| (!port
&& ret
)) {
496 pr_err("Wrong device name\n");
501 rd
->dev_idx
= dev_map
->idx
;
503 for (; rd
->port_idx
< dev_map
->num_ports
+ 1; rd
->port_idx
++) {
509 * We got request to show link for devname
518 jsonw_end_array(rd
->jw
);
522 int rd_exec_dev(struct rd
*rd
, int (*cb
)(struct rd
*rd
))
524 struct dev_map
*dev_map
;
528 jsonw_start_array(rd
->jw
);
530 list_for_each_entry(dev_map
, &rd
->dev_map_list
, list
) {
531 rd
->dev_idx
= dev_map
->idx
;
537 dev_map
= dev_map_lookup(rd
, false);
539 pr_err("Wrong device name - %s\n", rd_argv(rd
));
544 rd
->dev_idx
= dev_map
->idx
;
549 jsonw_end_array(rd
->jw
);
553 int rd_exec_cmd(struct rd
*rd
, const struct rd_cmd
*cmds
, const char *str
)
555 const struct rd_cmd
*c
;
557 /* First argument in objs table is default variant */
559 return cmds
->func(rd
);
561 for (c
= cmds
+ 1; c
->cmd
; ++c
) {
562 if (rd_argv_match(rd
, c
->cmd
)) {
563 /* Move to next argument */
569 pr_err("Unknown %s '%s'.\n", str
, rd_argv(rd
));
573 void rd_prepare_msg(struct rd
*rd
, uint32_t cmd
, uint32_t *seq
, uint16_t flags
)
577 rd
->nlh
= mnl_nlmsg_put_header(rd
->buff
);
578 rd
->nlh
->nlmsg_type
= RDMA_NL_GET_TYPE(RDMA_NL_NLDEV
, cmd
);
579 rd
->nlh
->nlmsg_seq
= *seq
;
580 rd
->nlh
->nlmsg_flags
= flags
;
583 int rd_send_msg(struct rd
*rd
)
587 rd
->nl
= mnl_socket_open(NETLINK_RDMA
);
589 pr_err("Failed to open NETLINK_RDMA socket\n");
593 ret
= mnl_socket_bind(rd
->nl
, 0, MNL_SOCKET_AUTOPID
);
595 pr_err("Failed to bind socket with err %d\n", ret
);
599 ret
= mnl_socket_sendto(rd
->nl
, rd
->nlh
, rd
->nlh
->nlmsg_len
);
601 pr_err("Failed to send to socket with err %d\n", ret
);
607 mnl_socket_close(rd
->nl
);
611 int rd_recv_msg(struct rd
*rd
, mnl_cb_t callback
, void *data
, unsigned int seq
)
615 char buf
[MNL_SOCKET_BUFFER_SIZE
];
617 portid
= mnl_socket_get_portid(rd
->nl
);
619 ret
= mnl_socket_recvfrom(rd
->nl
, buf
, sizeof(buf
));
623 ret
= mnl_cb_run(buf
, ret
, seq
, portid
, callback
, data
);
626 mnl_socket_close(rd
->nl
);
630 static struct dev_map
*_dev_map_lookup(struct rd
*rd
, const char *dev_name
)
632 struct dev_map
*dev_map
;
634 list_for_each_entry(dev_map
, &rd
->dev_map_list
, list
)
635 if (strcmp(dev_name
, dev_map
->dev_name
) == 0)
641 struct dev_map
*dev_map_lookup(struct rd
*rd
, bool allow_port_index
)
643 struct dev_map
*dev_map
;
650 dev_name
= strdup(rd_argv(rd
));
651 if (allow_port_index
) {
652 slash
= strrchr(dev_name
, '/');
657 dev_map
= _dev_map_lookup(rd
, dev_name
);