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