1 // SPDX-License-Identifier: GPL-2.0-only
2 /* Copyright (c) 2017 Facebook
3  */
4 #include "test_progs.h"
5 #include "bpf_rlimit.h"
6 
7 int error_cnt, pass_cnt;
8 bool jit_enabled;
9 bool verifier_stats = false;
10 
11 struct ipv4_packet pkt_v4 = {
12 	.eth.h_proto = __bpf_constant_htons(ETH_P_IP),
13 	.iph.ihl = 5,
14 	.iph.protocol = IPPROTO_TCP,
15 	.iph.tot_len = __bpf_constant_htons(MAGIC_BYTES),
16 	.tcp.urg_ptr = 123,
17 	.tcp.doff = 5,
18 };
19 
20 struct ipv6_packet pkt_v6 = {
21 	.eth.h_proto = __bpf_constant_htons(ETH_P_IPV6),
22 	.iph.nexthdr = IPPROTO_TCP,
23 	.iph.payload_len = __bpf_constant_htons(MAGIC_BYTES),
24 	.tcp.urg_ptr = 123,
25 	.tcp.doff = 5,
26 };
27 
28 int bpf_find_map(const char *test, struct bpf_object *obj, const char *name)
29 {
30 	struct bpf_map *map;
31 
32 	map = bpf_object__find_map_by_name(obj, name);
33 	if (!map) {
34 		printf("%s:FAIL:map '%s' not found\n", test, name);
35 		error_cnt++;
36 		return -1;
37 	}
38 	return bpf_map__fd(map);
39 }
40 
41 static bool is_jit_enabled(void)
42 {
43 	const char *jit_sysctl = "/proc/sys/net/core/bpf_jit_enable";
44 	bool enabled = false;
45 	int sysctl_fd;
46 
47 	sysctl_fd = open(jit_sysctl, 0, O_RDONLY);
48 	if (sysctl_fd != -1) {
49 		char tmpc;
50 
51 		if (read(sysctl_fd, &tmpc, sizeof(tmpc)) == 1)
52 			enabled = (tmpc != '0');
53 		close(sysctl_fd);
54 	}
55 
56 	return enabled;
57 }
58 
59 int compare_map_keys(int map1_fd, int map2_fd)
60 {
61 	__u32 key, next_key;
62 	char val_buf[PERF_MAX_STACK_DEPTH *
63 		     sizeof(struct bpf_stack_build_id)];
64 	int err;
65 
66 	err = bpf_map_get_next_key(map1_fd, NULL, &key);
67 	if (err)
68 		return err;
69 	err = bpf_map_lookup_elem(map2_fd, &key, val_buf);
70 	if (err)
71 		return err;
72 
73 	while (bpf_map_get_next_key(map1_fd, &key, &next_key) == 0) {
74 		err = bpf_map_lookup_elem(map2_fd, &next_key, val_buf);
75 		if (err)
76 			return err;
77 
78 		key = next_key;
79 	}
80 	if (errno != ENOENT)
81 		return -1;
82 
83 	return 0;
84 }
85 
86 int compare_stack_ips(int smap_fd, int amap_fd, int stack_trace_len)
87 {
88 	__u32 key, next_key, *cur_key_p, *next_key_p;
89 	char *val_buf1, *val_buf2;
90 	int i, err = 0;
91 
92 	val_buf1 = malloc(stack_trace_len);
93 	val_buf2 = malloc(stack_trace_len);
94 	cur_key_p = NULL;
95 	next_key_p = &key;
96 	while (bpf_map_get_next_key(smap_fd, cur_key_p, next_key_p) == 0) {
97 		err = bpf_map_lookup_elem(smap_fd, next_key_p, val_buf1);
98 		if (err)
99 			goto out;
100 		err = bpf_map_lookup_elem(amap_fd, next_key_p, val_buf2);
101 		if (err)
102 			goto out;
103 		for (i = 0; i < stack_trace_len; i++) {
104 			if (val_buf1[i] != val_buf2[i]) {
105 				err = -1;
106 				goto out;
107 			}
108 		}
109 		key = *next_key_p;
110 		cur_key_p = &key;
111 		next_key_p = &next_key;
112 	}
113 	if (errno != ENOENT)
114 		err = -1;
115 
116 out:
117 	free(val_buf1);
118 	free(val_buf2);
119 	return err;
120 }
121 
122 int extract_build_id(char *build_id, size_t size)
123 {
124 	FILE *fp;
125 	char *line = NULL;
126 	size_t len = 0;
127 
128 	fp = popen("readelf -n ./urandom_read | grep 'Build ID'", "r");
129 	if (fp == NULL)
130 		return -1;
131 
132 	if (getline(&line, &len, fp) == -1)
133 		goto err;
134 	fclose(fp);
135 
136 	if (len > size)
137 		len = size;
138 	memcpy(build_id, line, len);
139 	build_id[len] = '\0';
140 	return 0;
141 err:
142 	fclose(fp);
143 	return -1;
144 }
145 
146 void *spin_lock_thread(void *arg)
147 {
148 	__u32 duration, retval;
149 	int err, prog_fd = *(u32 *) arg;
150 
151 	err = bpf_prog_test_run(prog_fd, 10000, &pkt_v4, sizeof(pkt_v4),
152 				NULL, NULL, &retval, &duration);
153 	CHECK(err || retval, "",
154 	      "err %d errno %d retval %d duration %d\n",
155 	      err, errno, retval, duration);
156 	pthread_exit(arg);
157 }
158 
159 #define DECLARE
160 #include <prog_tests/tests.h>
161 #undef DECLARE
162 
163 int main(int ac, char **av)
164 {
165 	srand(time(NULL));
166 
167 	jit_enabled = is_jit_enabled();
168 
169 	if (ac == 2 && strcmp(av[1], "-s") == 0)
170 		verifier_stats = true;
171 
172 #define CALL
173 #include <prog_tests/tests.h>
174 #undef CALL
175 
176 	printf("Summary: %d PASSED, %d FAILED\n", pass_cnt, error_cnt);
177 	return error_cnt ? EXIT_FAILURE : EXIT_SUCCESS;
178 }
179