Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
1// SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)
2/* Copyright (C) 2019 Netronome Systems, Inc. */
3/* Copyright (C) 2020 Facebook, Inc. */
4#include <stdlib.h>
5#include <string.h>
6#include <errno.h>
7#include <bpf/bpf.h>
8#include <bpf/libbpf.h>
9#include "testing_helpers.h"
10
11int parse_num_list(const char *s, bool **num_set, int *num_set_len)
12{
13 int i, set_len = 0, new_len, num, start = 0, end = -1;
14 bool *set = NULL, *tmp, parsing_end = false;
15 char *next;
16
17 while (s[0]) {
18 errno = 0;
19 num = strtol(s, &next, 10);
20 if (errno)
21 return -errno;
22
23 if (parsing_end)
24 end = num;
25 else
26 start = num;
27
28 if (!parsing_end && *next == '-') {
29 s = next + 1;
30 parsing_end = true;
31 continue;
32 } else if (*next == ',') {
33 parsing_end = false;
34 s = next + 1;
35 end = num;
36 } else if (*next == '\0') {
37 parsing_end = false;
38 s = next;
39 end = num;
40 } else {
41 return -EINVAL;
42 }
43
44 if (start > end)
45 return -EINVAL;
46
47 if (end + 1 > set_len) {
48 new_len = end + 1;
49 tmp = realloc(set, new_len);
50 if (!tmp) {
51 free(set);
52 return -ENOMEM;
53 }
54 for (i = set_len; i < start; i++)
55 tmp[i] = false;
56 set = tmp;
57 set_len = new_len;
58 }
59 for (i = start; i <= end; i++)
60 set[i] = true;
61 }
62
63 if (!set)
64 return -EINVAL;
65
66 *num_set = set;
67 *num_set_len = set_len;
68
69 return 0;
70}
71
72__u32 link_info_prog_id(const struct bpf_link *link, struct bpf_link_info *info)
73{
74 __u32 info_len = sizeof(*info);
75 int err;
76
77 memset(info, 0, sizeof(*info));
78 err = bpf_obj_get_info_by_fd(bpf_link__fd(link), info, &info_len);
79 if (err) {
80 printf("failed to get link info: %d\n", -errno);
81 return 0;
82 }
83 return info->prog_id;
84}
85
86int extra_prog_load_log_flags = 0;
87
88int bpf_prog_test_load(const char *file, enum bpf_prog_type type,
89 struct bpf_object **pobj, int *prog_fd)
90{
91 LIBBPF_OPTS(bpf_object_open_opts, opts,
92 .kernel_log_level = extra_prog_load_log_flags,
93 );
94 struct bpf_object *obj;
95 struct bpf_program *prog;
96 __u32 flags;
97 int err;
98
99 obj = bpf_object__open_file(file, &opts);
100 if (!obj)
101 return -errno;
102
103 prog = bpf_object__next_program(obj, NULL);
104 if (!prog) {
105 err = -ENOENT;
106 goto err_out;
107 }
108
109 if (type != BPF_PROG_TYPE_UNSPEC)
110 bpf_program__set_type(prog, type);
111
112 flags = bpf_program__flags(prog) | BPF_F_TEST_RND_HI32;
113 bpf_program__set_flags(prog, flags);
114
115 err = bpf_object__load(obj);
116 if (err)
117 goto err_out;
118
119 *pobj = obj;
120 *prog_fd = bpf_program__fd(prog);
121
122 return 0;
123err_out:
124 bpf_object__close(obj);
125 return err;
126}
127
128int bpf_test_load_program(enum bpf_prog_type type, const struct bpf_insn *insns,
129 size_t insns_cnt, const char *license,
130 __u32 kern_version, char *log_buf,
131 size_t log_buf_sz)
132{
133 LIBBPF_OPTS(bpf_prog_load_opts, opts,
134 .kern_version = kern_version,
135 .prog_flags = BPF_F_TEST_RND_HI32,
136 .log_level = extra_prog_load_log_flags,
137 .log_buf = log_buf,
138 .log_size = log_buf_sz,
139 );
140
141 return bpf_prog_load(type, NULL, license, insns, insns_cnt, &opts);
142}