1 // SPDX-License-Identifier: GPL-2.0
2 #include <test_progs.h>
3 #include <network_helpers.h>
4 
5 /* test_tailcall_1 checks basic functionality by patching multiple locations
6  * in a single program for a single tail call slot with nop->jmp, jmp->nop
7  * and jmp->jmp rewrites. Also checks for nop->nop.
8  */
9 static void test_tailcall_1(void)
10 {
11 	int err, map_fd, prog_fd, main_fd, i, j;
12 	struct bpf_map *prog_array;
13 	struct bpf_program *prog;
14 	struct bpf_object *obj;
15 	char prog_name[32];
16 	char buff[128] = {};
17 	LIBBPF_OPTS(bpf_test_run_opts, topts,
18 		.data_in = buff,
19 		.data_size_in = sizeof(buff),
20 		.repeat = 1,
21 	);
22 
23 	err = bpf_prog_test_load("tailcall1.bpf.o", BPF_PROG_TYPE_SCHED_CLS, &obj,
24 				 &prog_fd);
25 	if (CHECK_FAIL(err))
26 		return;
27 
28 	prog = bpf_object__find_program_by_name(obj, "entry");
29 	if (CHECK_FAIL(!prog))
30 		goto out;
31 
32 	main_fd = bpf_program__fd(prog);
33 	if (CHECK_FAIL(main_fd < 0))
34 		goto out;
35 
36 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
37 	if (CHECK_FAIL(!prog_array))
38 		goto out;
39 
40 	map_fd = bpf_map__fd(prog_array);
41 	if (CHECK_FAIL(map_fd < 0))
42 		goto out;
43 
44 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
45 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", i);
46 
47 		prog = bpf_object__find_program_by_name(obj, prog_name);
48 		if (CHECK_FAIL(!prog))
49 			goto out;
50 
51 		prog_fd = bpf_program__fd(prog);
52 		if (CHECK_FAIL(prog_fd < 0))
53 			goto out;
54 
55 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
56 		if (CHECK_FAIL(err))
57 			goto out;
58 	}
59 
60 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
61 		err = bpf_prog_test_run_opts(main_fd, &topts);
62 		ASSERT_OK(err, "tailcall");
63 		ASSERT_EQ(topts.retval, i, "tailcall retval");
64 
65 		err = bpf_map_delete_elem(map_fd, &i);
66 		if (CHECK_FAIL(err))
67 			goto out;
68 	}
69 
70 	err = bpf_prog_test_run_opts(main_fd, &topts);
71 	ASSERT_OK(err, "tailcall");
72 	ASSERT_EQ(topts.retval, 3, "tailcall retval");
73 
74 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
75 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", i);
76 
77 		prog = bpf_object__find_program_by_name(obj, prog_name);
78 		if (CHECK_FAIL(!prog))
79 			goto out;
80 
81 		prog_fd = bpf_program__fd(prog);
82 		if (CHECK_FAIL(prog_fd < 0))
83 			goto out;
84 
85 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
86 		if (CHECK_FAIL(err))
87 			goto out;
88 	}
89 
90 	err = bpf_prog_test_run_opts(main_fd, &topts);
91 	ASSERT_OK(err, "tailcall");
92 	ASSERT_OK(topts.retval, "tailcall retval");
93 
94 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
95 		j = bpf_map__max_entries(prog_array) - 1 - i;
96 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", j);
97 
98 		prog = bpf_object__find_program_by_name(obj, prog_name);
99 		if (CHECK_FAIL(!prog))
100 			goto out;
101 
102 		prog_fd = bpf_program__fd(prog);
103 		if (CHECK_FAIL(prog_fd < 0))
104 			goto out;
105 
106 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
107 		if (CHECK_FAIL(err))
108 			goto out;
109 	}
110 
111 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
112 		j = bpf_map__max_entries(prog_array) - 1 - i;
113 
114 		err = bpf_prog_test_run_opts(main_fd, &topts);
115 		ASSERT_OK(err, "tailcall");
116 		ASSERT_EQ(topts.retval, j, "tailcall retval");
117 
118 		err = bpf_map_delete_elem(map_fd, &i);
119 		if (CHECK_FAIL(err))
120 			goto out;
121 	}
122 
123 	err = bpf_prog_test_run_opts(main_fd, &topts);
124 	ASSERT_OK(err, "tailcall");
125 	ASSERT_EQ(topts.retval, 3, "tailcall retval");
126 
127 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
128 		err = bpf_map_delete_elem(map_fd, &i);
129 		if (CHECK_FAIL(err >= 0 || errno != ENOENT))
130 			goto out;
131 
132 		err = bpf_prog_test_run_opts(main_fd, &topts);
133 		ASSERT_OK(err, "tailcall");
134 		ASSERT_EQ(topts.retval, 3, "tailcall retval");
135 	}
136 
137 out:
138 	bpf_object__close(obj);
139 }
140 
141 /* test_tailcall_2 checks that patching multiple programs for a single
142  * tail call slot works. It also jumps through several programs and tests
143  * the tail call limit counter.
144  */
145 static void test_tailcall_2(void)
146 {
147 	int err, map_fd, prog_fd, main_fd, i;
148 	struct bpf_map *prog_array;
149 	struct bpf_program *prog;
150 	struct bpf_object *obj;
151 	char prog_name[32];
152 	char buff[128] = {};
153 	LIBBPF_OPTS(bpf_test_run_opts, topts,
154 		.data_in = buff,
155 		.data_size_in = sizeof(buff),
156 		.repeat = 1,
157 	);
158 
159 	err = bpf_prog_test_load("tailcall2.bpf.o", BPF_PROG_TYPE_SCHED_CLS, &obj,
160 				 &prog_fd);
161 	if (CHECK_FAIL(err))
162 		return;
163 
164 	prog = bpf_object__find_program_by_name(obj, "entry");
165 	if (CHECK_FAIL(!prog))
166 		goto out;
167 
168 	main_fd = bpf_program__fd(prog);
169 	if (CHECK_FAIL(main_fd < 0))
170 		goto out;
171 
172 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
173 	if (CHECK_FAIL(!prog_array))
174 		goto out;
175 
176 	map_fd = bpf_map__fd(prog_array);
177 	if (CHECK_FAIL(map_fd < 0))
178 		goto out;
179 
180 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
181 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", i);
182 
183 		prog = bpf_object__find_program_by_name(obj, prog_name);
184 		if (CHECK_FAIL(!prog))
185 			goto out;
186 
187 		prog_fd = bpf_program__fd(prog);
188 		if (CHECK_FAIL(prog_fd < 0))
189 			goto out;
190 
191 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
192 		if (CHECK_FAIL(err))
193 			goto out;
194 	}
195 
196 	err = bpf_prog_test_run_opts(main_fd, &topts);
197 	ASSERT_OK(err, "tailcall");
198 	ASSERT_EQ(topts.retval, 2, "tailcall retval");
199 
200 	i = 2;
201 	err = bpf_map_delete_elem(map_fd, &i);
202 	if (CHECK_FAIL(err))
203 		goto out;
204 
205 	err = bpf_prog_test_run_opts(main_fd, &topts);
206 	ASSERT_OK(err, "tailcall");
207 	ASSERT_EQ(topts.retval, 1, "tailcall retval");
208 
209 	i = 0;
210 	err = bpf_map_delete_elem(map_fd, &i);
211 	if (CHECK_FAIL(err))
212 		goto out;
213 
214 	err = bpf_prog_test_run_opts(main_fd, &topts);
215 	ASSERT_OK(err, "tailcall");
216 	ASSERT_EQ(topts.retval, 3, "tailcall retval");
217 out:
218 	bpf_object__close(obj);
219 }
220 
221 static void test_tailcall_count(const char *which)
222 {
223 	int err, map_fd, prog_fd, main_fd, data_fd, i, val;
224 	struct bpf_map *prog_array, *data_map;
225 	struct bpf_program *prog;
226 	struct bpf_object *obj;
227 	char buff[128] = {};
228 	LIBBPF_OPTS(bpf_test_run_opts, topts,
229 		.data_in = buff,
230 		.data_size_in = sizeof(buff),
231 		.repeat = 1,
232 	);
233 
234 	err = bpf_prog_test_load(which, BPF_PROG_TYPE_SCHED_CLS, &obj,
235 			    &prog_fd);
236 	if (CHECK_FAIL(err))
237 		return;
238 
239 	prog = bpf_object__find_program_by_name(obj, "entry");
240 	if (CHECK_FAIL(!prog))
241 		goto out;
242 
243 	main_fd = bpf_program__fd(prog);
244 	if (CHECK_FAIL(main_fd < 0))
245 		goto out;
246 
247 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
248 	if (CHECK_FAIL(!prog_array))
249 		goto out;
250 
251 	map_fd = bpf_map__fd(prog_array);
252 	if (CHECK_FAIL(map_fd < 0))
253 		goto out;
254 
255 	prog = bpf_object__find_program_by_name(obj, "classifier_0");
256 	if (CHECK_FAIL(!prog))
257 		goto out;
258 
259 	prog_fd = bpf_program__fd(prog);
260 	if (CHECK_FAIL(prog_fd < 0))
261 		goto out;
262 
263 	i = 0;
264 	err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
265 	if (CHECK_FAIL(err))
266 		goto out;
267 
268 	err = bpf_prog_test_run_opts(main_fd, &topts);
269 	ASSERT_OK(err, "tailcall");
270 	ASSERT_EQ(topts.retval, 1, "tailcall retval");
271 
272 	data_map = bpf_object__find_map_by_name(obj, "tailcall.bss");
273 	if (CHECK_FAIL(!data_map || !bpf_map__is_internal(data_map)))
274 		goto out;
275 
276 	data_fd = bpf_map__fd(data_map);
277 	if (CHECK_FAIL(data_fd < 0))
278 		goto out;
279 
280 	i = 0;
281 	err = bpf_map_lookup_elem(data_fd, &i, &val);
282 	ASSERT_OK(err, "tailcall count");
283 	ASSERT_EQ(val, 33, "tailcall count");
284 
285 	i = 0;
286 	err = bpf_map_delete_elem(map_fd, &i);
287 	if (CHECK_FAIL(err))
288 		goto out;
289 
290 	err = bpf_prog_test_run_opts(main_fd, &topts);
291 	ASSERT_OK(err, "tailcall");
292 	ASSERT_OK(topts.retval, "tailcall retval");
293 out:
294 	bpf_object__close(obj);
295 }
296 
297 /* test_tailcall_3 checks that the count value of the tail call limit
298  * enforcement matches with expectations. JIT uses direct jump.
299  */
300 static void test_tailcall_3(void)
301 {
302 	test_tailcall_count("tailcall3.bpf.o");
303 }
304 
305 /* test_tailcall_6 checks that the count value of the tail call limit
306  * enforcement matches with expectations. JIT uses indirect jump.
307  */
308 static void test_tailcall_6(void)
309 {
310 	test_tailcall_count("tailcall6.bpf.o");
311 }
312 
313 /* test_tailcall_4 checks that the kernel properly selects indirect jump
314  * for the case where the key is not known. Latter is passed via global
315  * data to select different targets we can compare return value of.
316  */
317 static void test_tailcall_4(void)
318 {
319 	int err, map_fd, prog_fd, main_fd, data_fd, i;
320 	struct bpf_map *prog_array, *data_map;
321 	struct bpf_program *prog;
322 	struct bpf_object *obj;
323 	static const int zero = 0;
324 	char buff[128] = {};
325 	char prog_name[32];
326 	LIBBPF_OPTS(bpf_test_run_opts, topts,
327 		.data_in = buff,
328 		.data_size_in = sizeof(buff),
329 		.repeat = 1,
330 	);
331 
332 	err = bpf_prog_test_load("tailcall4.bpf.o", BPF_PROG_TYPE_SCHED_CLS, &obj,
333 				 &prog_fd);
334 	if (CHECK_FAIL(err))
335 		return;
336 
337 	prog = bpf_object__find_program_by_name(obj, "entry");
338 	if (CHECK_FAIL(!prog))
339 		goto out;
340 
341 	main_fd = bpf_program__fd(prog);
342 	if (CHECK_FAIL(main_fd < 0))
343 		goto out;
344 
345 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
346 	if (CHECK_FAIL(!prog_array))
347 		goto out;
348 
349 	map_fd = bpf_map__fd(prog_array);
350 	if (CHECK_FAIL(map_fd < 0))
351 		goto out;
352 
353 	data_map = bpf_object__find_map_by_name(obj, "tailcall.bss");
354 	if (CHECK_FAIL(!data_map || !bpf_map__is_internal(data_map)))
355 		goto out;
356 
357 	data_fd = bpf_map__fd(data_map);
358 	if (CHECK_FAIL(data_fd < 0))
359 		goto out;
360 
361 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
362 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", i);
363 
364 		prog = bpf_object__find_program_by_name(obj, prog_name);
365 		if (CHECK_FAIL(!prog))
366 			goto out;
367 
368 		prog_fd = bpf_program__fd(prog);
369 		if (CHECK_FAIL(prog_fd < 0))
370 			goto out;
371 
372 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
373 		if (CHECK_FAIL(err))
374 			goto out;
375 	}
376 
377 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
378 		err = bpf_map_update_elem(data_fd, &zero, &i, BPF_ANY);
379 		if (CHECK_FAIL(err))
380 			goto out;
381 
382 		err = bpf_prog_test_run_opts(main_fd, &topts);
383 		ASSERT_OK(err, "tailcall");
384 		ASSERT_EQ(topts.retval, i, "tailcall retval");
385 	}
386 
387 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
388 		err = bpf_map_update_elem(data_fd, &zero, &i, BPF_ANY);
389 		if (CHECK_FAIL(err))
390 			goto out;
391 
392 		err = bpf_map_delete_elem(map_fd, &i);
393 		if (CHECK_FAIL(err))
394 			goto out;
395 
396 		err = bpf_prog_test_run_opts(main_fd, &topts);
397 		ASSERT_OK(err, "tailcall");
398 		ASSERT_EQ(topts.retval, 3, "tailcall retval");
399 	}
400 out:
401 	bpf_object__close(obj);
402 }
403 
404 /* test_tailcall_5 probes similarly to test_tailcall_4 that the kernel generates
405  * an indirect jump when the keys are const but different from different branches.
406  */
407 static void test_tailcall_5(void)
408 {
409 	int err, map_fd, prog_fd, main_fd, data_fd, i, key[] = { 1111, 1234, 5678 };
410 	struct bpf_map *prog_array, *data_map;
411 	struct bpf_program *prog;
412 	struct bpf_object *obj;
413 	static const int zero = 0;
414 	char buff[128] = {};
415 	char prog_name[32];
416 	LIBBPF_OPTS(bpf_test_run_opts, topts,
417 		.data_in = buff,
418 		.data_size_in = sizeof(buff),
419 		.repeat = 1,
420 	);
421 
422 	err = bpf_prog_test_load("tailcall5.bpf.o", BPF_PROG_TYPE_SCHED_CLS, &obj,
423 				 &prog_fd);
424 	if (CHECK_FAIL(err))
425 		return;
426 
427 	prog = bpf_object__find_program_by_name(obj, "entry");
428 	if (CHECK_FAIL(!prog))
429 		goto out;
430 
431 	main_fd = bpf_program__fd(prog);
432 	if (CHECK_FAIL(main_fd < 0))
433 		goto out;
434 
435 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
436 	if (CHECK_FAIL(!prog_array))
437 		goto out;
438 
439 	map_fd = bpf_map__fd(prog_array);
440 	if (CHECK_FAIL(map_fd < 0))
441 		goto out;
442 
443 	data_map = bpf_object__find_map_by_name(obj, "tailcall.bss");
444 	if (CHECK_FAIL(!data_map || !bpf_map__is_internal(data_map)))
445 		goto out;
446 
447 	data_fd = bpf_map__fd(data_map);
448 	if (CHECK_FAIL(data_fd < 0))
449 		goto out;
450 
451 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
452 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", i);
453 
454 		prog = bpf_object__find_program_by_name(obj, prog_name);
455 		if (CHECK_FAIL(!prog))
456 			goto out;
457 
458 		prog_fd = bpf_program__fd(prog);
459 		if (CHECK_FAIL(prog_fd < 0))
460 			goto out;
461 
462 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
463 		if (CHECK_FAIL(err))
464 			goto out;
465 	}
466 
467 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
468 		err = bpf_map_update_elem(data_fd, &zero, &key[i], BPF_ANY);
469 		if (CHECK_FAIL(err))
470 			goto out;
471 
472 		err = bpf_prog_test_run_opts(main_fd, &topts);
473 		ASSERT_OK(err, "tailcall");
474 		ASSERT_EQ(topts.retval, i, "tailcall retval");
475 	}
476 
477 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
478 		err = bpf_map_update_elem(data_fd, &zero, &key[i], BPF_ANY);
479 		if (CHECK_FAIL(err))
480 			goto out;
481 
482 		err = bpf_map_delete_elem(map_fd, &i);
483 		if (CHECK_FAIL(err))
484 			goto out;
485 
486 		err = bpf_prog_test_run_opts(main_fd, &topts);
487 		ASSERT_OK(err, "tailcall");
488 		ASSERT_EQ(topts.retval, 3, "tailcall retval");
489 	}
490 out:
491 	bpf_object__close(obj);
492 }
493 
494 /* test_tailcall_bpf2bpf_1 purpose is to make sure that tailcalls are working
495  * correctly in correlation with BPF subprograms
496  */
497 static void test_tailcall_bpf2bpf_1(void)
498 {
499 	int err, map_fd, prog_fd, main_fd, i;
500 	struct bpf_map *prog_array;
501 	struct bpf_program *prog;
502 	struct bpf_object *obj;
503 	char prog_name[32];
504 	LIBBPF_OPTS(bpf_test_run_opts, topts,
505 		.data_in = &pkt_v4,
506 		.data_size_in = sizeof(pkt_v4),
507 		.repeat = 1,
508 	);
509 
510 	err = bpf_prog_test_load("tailcall_bpf2bpf1.bpf.o", BPF_PROG_TYPE_SCHED_CLS,
511 				 &obj, &prog_fd);
512 	if (CHECK_FAIL(err))
513 		return;
514 
515 	prog = bpf_object__find_program_by_name(obj, "entry");
516 	if (CHECK_FAIL(!prog))
517 		goto out;
518 
519 	main_fd = bpf_program__fd(prog);
520 	if (CHECK_FAIL(main_fd < 0))
521 		goto out;
522 
523 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
524 	if (CHECK_FAIL(!prog_array))
525 		goto out;
526 
527 	map_fd = bpf_map__fd(prog_array);
528 	if (CHECK_FAIL(map_fd < 0))
529 		goto out;
530 
531 	/* nop -> jmp */
532 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
533 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", i);
534 
535 		prog = bpf_object__find_program_by_name(obj, prog_name);
536 		if (CHECK_FAIL(!prog))
537 			goto out;
538 
539 		prog_fd = bpf_program__fd(prog);
540 		if (CHECK_FAIL(prog_fd < 0))
541 			goto out;
542 
543 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
544 		if (CHECK_FAIL(err))
545 			goto out;
546 	}
547 
548 	err = bpf_prog_test_run_opts(main_fd, &topts);
549 	ASSERT_OK(err, "tailcall");
550 	ASSERT_EQ(topts.retval, 1, "tailcall retval");
551 
552 	/* jmp -> nop, call subprog that will do tailcall */
553 	i = 1;
554 	err = bpf_map_delete_elem(map_fd, &i);
555 	if (CHECK_FAIL(err))
556 		goto out;
557 
558 	err = bpf_prog_test_run_opts(main_fd, &topts);
559 	ASSERT_OK(err, "tailcall");
560 	ASSERT_OK(topts.retval, "tailcall retval");
561 
562 	/* make sure that subprog can access ctx and entry prog that
563 	 * called this subprog can properly return
564 	 */
565 	i = 0;
566 	err = bpf_map_delete_elem(map_fd, &i);
567 	if (CHECK_FAIL(err))
568 		goto out;
569 
570 	err = bpf_prog_test_run_opts(main_fd, &topts);
571 	ASSERT_OK(err, "tailcall");
572 	ASSERT_EQ(topts.retval, sizeof(pkt_v4) * 2, "tailcall retval");
573 out:
574 	bpf_object__close(obj);
575 }
576 
577 /* test_tailcall_bpf2bpf_2 checks that the count value of the tail call limit
578  * enforcement matches with expectations when tailcall is preceded with
579  * bpf2bpf call.
580  */
581 static void test_tailcall_bpf2bpf_2(void)
582 {
583 	int err, map_fd, prog_fd, main_fd, data_fd, i, val;
584 	struct bpf_map *prog_array, *data_map;
585 	struct bpf_program *prog;
586 	struct bpf_object *obj;
587 	char buff[128] = {};
588 	LIBBPF_OPTS(bpf_test_run_opts, topts,
589 		.data_in = buff,
590 		.data_size_in = sizeof(buff),
591 		.repeat = 1,
592 	);
593 
594 	err = bpf_prog_test_load("tailcall_bpf2bpf2.bpf.o", BPF_PROG_TYPE_SCHED_CLS,
595 				 &obj, &prog_fd);
596 	if (CHECK_FAIL(err))
597 		return;
598 
599 	prog = bpf_object__find_program_by_name(obj, "entry");
600 	if (CHECK_FAIL(!prog))
601 		goto out;
602 
603 	main_fd = bpf_program__fd(prog);
604 	if (CHECK_FAIL(main_fd < 0))
605 		goto out;
606 
607 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
608 	if (CHECK_FAIL(!prog_array))
609 		goto out;
610 
611 	map_fd = bpf_map__fd(prog_array);
612 	if (CHECK_FAIL(map_fd < 0))
613 		goto out;
614 
615 	prog = bpf_object__find_program_by_name(obj, "classifier_0");
616 	if (CHECK_FAIL(!prog))
617 		goto out;
618 
619 	prog_fd = bpf_program__fd(prog);
620 	if (CHECK_FAIL(prog_fd < 0))
621 		goto out;
622 
623 	i = 0;
624 	err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
625 	if (CHECK_FAIL(err))
626 		goto out;
627 
628 	err = bpf_prog_test_run_opts(main_fd, &topts);
629 	ASSERT_OK(err, "tailcall");
630 	ASSERT_EQ(topts.retval, 1, "tailcall retval");
631 
632 	data_map = bpf_object__find_map_by_name(obj, "tailcall.bss");
633 	if (CHECK_FAIL(!data_map || !bpf_map__is_internal(data_map)))
634 		goto out;
635 
636 	data_fd = bpf_map__fd(data_map);
637 	if (CHECK_FAIL(data_fd < 0))
638 		goto out;
639 
640 	i = 0;
641 	err = bpf_map_lookup_elem(data_fd, &i, &val);
642 	ASSERT_OK(err, "tailcall count");
643 	ASSERT_EQ(val, 33, "tailcall count");
644 
645 	i = 0;
646 	err = bpf_map_delete_elem(map_fd, &i);
647 	if (CHECK_FAIL(err))
648 		goto out;
649 
650 	err = bpf_prog_test_run_opts(main_fd, &topts);
651 	ASSERT_OK(err, "tailcall");
652 	ASSERT_OK(topts.retval, "tailcall retval");
653 out:
654 	bpf_object__close(obj);
655 }
656 
657 /* test_tailcall_bpf2bpf_3 checks that non-trivial amount of stack (up to
658  * 256 bytes) can be used within bpf subprograms that have the tailcalls
659  * in them
660  */
661 static void test_tailcall_bpf2bpf_3(void)
662 {
663 	int err, map_fd, prog_fd, main_fd, i;
664 	struct bpf_map *prog_array;
665 	struct bpf_program *prog;
666 	struct bpf_object *obj;
667 	char prog_name[32];
668 	LIBBPF_OPTS(bpf_test_run_opts, topts,
669 		.data_in = &pkt_v4,
670 		.data_size_in = sizeof(pkt_v4),
671 		.repeat = 1,
672 	);
673 
674 	err = bpf_prog_test_load("tailcall_bpf2bpf3.bpf.o", BPF_PROG_TYPE_SCHED_CLS,
675 				 &obj, &prog_fd);
676 	if (CHECK_FAIL(err))
677 		return;
678 
679 	prog = bpf_object__find_program_by_name(obj, "entry");
680 	if (CHECK_FAIL(!prog))
681 		goto out;
682 
683 	main_fd = bpf_program__fd(prog);
684 	if (CHECK_FAIL(main_fd < 0))
685 		goto out;
686 
687 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
688 	if (CHECK_FAIL(!prog_array))
689 		goto out;
690 
691 	map_fd = bpf_map__fd(prog_array);
692 	if (CHECK_FAIL(map_fd < 0))
693 		goto out;
694 
695 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
696 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", i);
697 
698 		prog = bpf_object__find_program_by_name(obj, prog_name);
699 		if (CHECK_FAIL(!prog))
700 			goto out;
701 
702 		prog_fd = bpf_program__fd(prog);
703 		if (CHECK_FAIL(prog_fd < 0))
704 			goto out;
705 
706 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
707 		if (CHECK_FAIL(err))
708 			goto out;
709 	}
710 
711 	err = bpf_prog_test_run_opts(main_fd, &topts);
712 	ASSERT_OK(err, "tailcall");
713 	ASSERT_EQ(topts.retval, sizeof(pkt_v4) * 3, "tailcall retval");
714 
715 	i = 1;
716 	err = bpf_map_delete_elem(map_fd, &i);
717 	if (CHECK_FAIL(err))
718 		goto out;
719 
720 	err = bpf_prog_test_run_opts(main_fd, &topts);
721 	ASSERT_OK(err, "tailcall");
722 	ASSERT_EQ(topts.retval, sizeof(pkt_v4), "tailcall retval");
723 
724 	i = 0;
725 	err = bpf_map_delete_elem(map_fd, &i);
726 	if (CHECK_FAIL(err))
727 		goto out;
728 
729 	err = bpf_prog_test_run_opts(main_fd, &topts);
730 	ASSERT_OK(err, "tailcall");
731 	ASSERT_EQ(topts.retval, sizeof(pkt_v4) * 2, "tailcall retval");
732 out:
733 	bpf_object__close(obj);
734 }
735 
736 #include "tailcall_bpf2bpf4.skel.h"
737 
738 /* test_tailcall_bpf2bpf_4 checks that tailcall counter is correctly preserved
739  * across tailcalls combined with bpf2bpf calls. for making sure that tailcall
740  * counter behaves correctly, bpf program will go through following flow:
741  *
742  * entry -> entry_subprog -> tailcall0 -> bpf_func0 -> subprog0 ->
743  * -> tailcall1 -> bpf_func1 -> subprog1 -> tailcall2 -> bpf_func2 ->
744  * subprog2 [here bump global counter] --------^
745  *
746  * We go through first two tailcalls and start counting from the subprog2 where
747  * the loop begins. At the end of the test make sure that the global counter is
748  * equal to 31, because tailcall counter includes the first two tailcalls
749  * whereas global counter is incremented only on loop presented on flow above.
750  *
751  * The noise parameter is used to insert bpf_map_update calls into the logic
752  * to force verifier to patch instructions. This allows us to ensure jump
753  * logic remains correct with instruction movement.
754  */
755 static void test_tailcall_bpf2bpf_4(bool noise)
756 {
757 	int err, map_fd, prog_fd, main_fd, data_fd, i;
758 	struct tailcall_bpf2bpf4__bss val;
759 	struct bpf_map *prog_array, *data_map;
760 	struct bpf_program *prog;
761 	struct bpf_object *obj;
762 	char prog_name[32];
763 	LIBBPF_OPTS(bpf_test_run_opts, topts,
764 		.data_in = &pkt_v4,
765 		.data_size_in = sizeof(pkt_v4),
766 		.repeat = 1,
767 	);
768 
769 	err = bpf_prog_test_load("tailcall_bpf2bpf4.bpf.o", BPF_PROG_TYPE_SCHED_CLS,
770 				 &obj, &prog_fd);
771 	if (CHECK_FAIL(err))
772 		return;
773 
774 	prog = bpf_object__find_program_by_name(obj, "entry");
775 	if (CHECK_FAIL(!prog))
776 		goto out;
777 
778 	main_fd = bpf_program__fd(prog);
779 	if (CHECK_FAIL(main_fd < 0))
780 		goto out;
781 
782 	prog_array = bpf_object__find_map_by_name(obj, "jmp_table");
783 	if (CHECK_FAIL(!prog_array))
784 		goto out;
785 
786 	map_fd = bpf_map__fd(prog_array);
787 	if (CHECK_FAIL(map_fd < 0))
788 		goto out;
789 
790 	for (i = 0; i < bpf_map__max_entries(prog_array); i++) {
791 		snprintf(prog_name, sizeof(prog_name), "classifier_%d", i);
792 
793 		prog = bpf_object__find_program_by_name(obj, prog_name);
794 		if (CHECK_FAIL(!prog))
795 			goto out;
796 
797 		prog_fd = bpf_program__fd(prog);
798 		if (CHECK_FAIL(prog_fd < 0))
799 			goto out;
800 
801 		err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
802 		if (CHECK_FAIL(err))
803 			goto out;
804 	}
805 
806 	data_map = bpf_object__find_map_by_name(obj, "tailcall.bss");
807 	if (CHECK_FAIL(!data_map || !bpf_map__is_internal(data_map)))
808 		goto out;
809 
810 	data_fd = bpf_map__fd(data_map);
811 	if (CHECK_FAIL(data_fd < 0))
812 		goto out;
813 
814 	i = 0;
815 	val.noise = noise;
816 	val.count = 0;
817 	err = bpf_map_update_elem(data_fd, &i, &val, BPF_ANY);
818 	if (CHECK_FAIL(err))
819 		goto out;
820 
821 	err = bpf_prog_test_run_opts(main_fd, &topts);
822 	ASSERT_OK(err, "tailcall");
823 	ASSERT_EQ(topts.retval, sizeof(pkt_v4) * 3, "tailcall retval");
824 
825 	i = 0;
826 	err = bpf_map_lookup_elem(data_fd, &i, &val);
827 	ASSERT_OK(err, "tailcall count");
828 	ASSERT_EQ(val.count, 31, "tailcall count");
829 
830 out:
831 	bpf_object__close(obj);
832 }
833 
834 #include "tailcall_bpf2bpf6.skel.h"
835 
836 /* Tail call counting works even when there is data on stack which is
837  * not aligned to 8 bytes.
838  */
839 static void test_tailcall_bpf2bpf_6(void)
840 {
841 	struct tailcall_bpf2bpf6 *obj;
842 	int err, map_fd, prog_fd, main_fd, data_fd, i, val;
843 	LIBBPF_OPTS(bpf_test_run_opts, topts,
844 		.data_in = &pkt_v4,
845 		.data_size_in = sizeof(pkt_v4),
846 		.repeat = 1,
847 	);
848 
849 	obj = tailcall_bpf2bpf6__open_and_load();
850 	if (!ASSERT_OK_PTR(obj, "open and load"))
851 		return;
852 
853 	main_fd = bpf_program__fd(obj->progs.entry);
854 	if (!ASSERT_GE(main_fd, 0, "entry prog fd"))
855 		goto out;
856 
857 	map_fd = bpf_map__fd(obj->maps.jmp_table);
858 	if (!ASSERT_GE(map_fd, 0, "jmp_table map fd"))
859 		goto out;
860 
861 	prog_fd = bpf_program__fd(obj->progs.classifier_0);
862 	if (!ASSERT_GE(prog_fd, 0, "classifier_0 prog fd"))
863 		goto out;
864 
865 	i = 0;
866 	err = bpf_map_update_elem(map_fd, &i, &prog_fd, BPF_ANY);
867 	if (!ASSERT_OK(err, "jmp_table map update"))
868 		goto out;
869 
870 	err = bpf_prog_test_run_opts(main_fd, &topts);
871 	ASSERT_OK(err, "entry prog test run");
872 	ASSERT_EQ(topts.retval, 0, "tailcall retval");
873 
874 	data_fd = bpf_map__fd(obj->maps.bss);
875 	if (!ASSERT_GE(data_fd, 0, "bss map fd"))
876 		goto out;
877 
878 	i = 0;
879 	err = bpf_map_lookup_elem(data_fd, &i, &val);
880 	ASSERT_OK(err, "bss map lookup");
881 	ASSERT_EQ(val, 1, "done flag is set");
882 
883 out:
884 	tailcall_bpf2bpf6__destroy(obj);
885 }
886 
887 void test_tailcalls(void)
888 {
889 	if (test__start_subtest("tailcall_1"))
890 		test_tailcall_1();
891 	if (test__start_subtest("tailcall_2"))
892 		test_tailcall_2();
893 	if (test__start_subtest("tailcall_3"))
894 		test_tailcall_3();
895 	if (test__start_subtest("tailcall_4"))
896 		test_tailcall_4();
897 	if (test__start_subtest("tailcall_5"))
898 		test_tailcall_5();
899 	if (test__start_subtest("tailcall_6"))
900 		test_tailcall_6();
901 	if (test__start_subtest("tailcall_bpf2bpf_1"))
902 		test_tailcall_bpf2bpf_1();
903 	if (test__start_subtest("tailcall_bpf2bpf_2"))
904 		test_tailcall_bpf2bpf_2();
905 	if (test__start_subtest("tailcall_bpf2bpf_3"))
906 		test_tailcall_bpf2bpf_3();
907 	if (test__start_subtest("tailcall_bpf2bpf_4"))
908 		test_tailcall_bpf2bpf_4(false);
909 	if (test__start_subtest("tailcall_bpf2bpf_5"))
910 		test_tailcall_bpf2bpf_4(true);
911 	if (test__start_subtest("tailcall_bpf2bpf_6"))
912 		test_tailcall_bpf2bpf_6();
913 }
914