xref: /linux/tools/testing/selftests/bpf/testing_helpers.c (revision 26fbb4c8c7c3ee9a4c3b4de555a8587b5a19154e)
1 // SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)
2 /* Copyright (C) 2020 Facebook, Inc. */
3 #include <stdlib.h>
4 #include <errno.h>
5 #include "testing_helpers.h"
6 
7 int parse_num_list(const char *s, bool **num_set, int *num_set_len)
8 {
9 	int i, set_len = 0, new_len, num, start = 0, end = -1;
10 	bool *set = NULL, *tmp, parsing_end = false;
11 	char *next;
12 
13 	while (s[0]) {
14 		errno = 0;
15 		num = strtol(s, &next, 10);
16 		if (errno)
17 			return -errno;
18 
19 		if (parsing_end)
20 			end = num;
21 		else
22 			start = num;
23 
24 		if (!parsing_end && *next == '-') {
25 			s = next + 1;
26 			parsing_end = true;
27 			continue;
28 		} else if (*next == ',') {
29 			parsing_end = false;
30 			s = next + 1;
31 			end = num;
32 		} else if (*next == '\0') {
33 			parsing_end = false;
34 			s = next;
35 			end = num;
36 		} else {
37 			return -EINVAL;
38 		}
39 
40 		if (start > end)
41 			return -EINVAL;
42 
43 		if (end + 1 > set_len) {
44 			new_len = end + 1;
45 			tmp = realloc(set, new_len);
46 			if (!tmp) {
47 				free(set);
48 				return -ENOMEM;
49 			}
50 			for (i = set_len; i < start; i++)
51 				tmp[i] = false;
52 			set = tmp;
53 			set_len = new_len;
54 		}
55 		for (i = start; i <= end; i++)
56 			set[i] = true;
57 	}
58 
59 	if (!set)
60 		return -EINVAL;
61 
62 	*num_set = set;
63 	*num_set_len = set_len;
64 
65 	return 0;
66 }
67 
68 __u32 link_info_prog_id(const struct bpf_link *link, struct bpf_link_info *info)
69 {
70 	__u32 info_len = sizeof(*info);
71 	int err;
72 
73 	memset(info, 0, sizeof(*info));
74 	err = bpf_obj_get_info_by_fd(bpf_link__fd(link), info, &info_len);
75 	if (err) {
76 		printf("failed to get link info: %d\n", -errno);
77 		return 0;
78 	}
79 	return info->prog_id;
80 }
81