xref: /linux/tools/testing/selftests/bpf/progs/dynptr_fail.c (revision c36461825469a9ceee2346a2e89286c522525da7)
1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2022 Facebook */
3 
4 #include <errno.h>
5 #include <string.h>
6 #include <stdbool.h>
7 #include <linux/bpf.h>
8 #include <bpf/bpf_helpers.h>
9 #include <bpf/bpf_tracing.h>
10 #include <linux/if_ether.h>
11 #include "bpf_misc.h"
12 #include "bpf_kfuncs.h"
13 
14 char _license[] SEC("license") = "GPL";
15 
16 struct test_info {
17 	int x;
18 	struct bpf_dynptr ptr;
19 };
20 
21 struct {
22 	__uint(type, BPF_MAP_TYPE_ARRAY);
23 	__uint(max_entries, 1);
24 	__type(key, __u32);
25 	__type(value, struct bpf_dynptr);
26 } array_map1 SEC(".maps");
27 
28 struct {
29 	__uint(type, BPF_MAP_TYPE_ARRAY);
30 	__uint(max_entries, 1);
31 	__type(key, __u32);
32 	__type(value, struct test_info);
33 } array_map2 SEC(".maps");
34 
35 struct {
36 	__uint(type, BPF_MAP_TYPE_ARRAY);
37 	__uint(max_entries, 1);
38 	__type(key, __u32);
39 	__type(value, __u32);
40 } array_map3 SEC(".maps");
41 
42 struct {
43 	__uint(type, BPF_MAP_TYPE_ARRAY);
44 	__uint(max_entries, 1);
45 	__type(key, __u32);
46 	__type(value, __u64);
47 } array_map4 SEC(".maps");
48 
49 struct sample {
50 	int pid;
51 	long value;
52 	char comm[16];
53 };
54 
55 struct {
56 	__uint(type, BPF_MAP_TYPE_RINGBUF);
57 	__uint(max_entries, 4096);
58 } ringbuf SEC(".maps");
59 
60 int err, val;
61 
62 static int get_map_val_dynptr(struct bpf_dynptr *ptr)
63 {
64 	__u32 key = 0, *map_val;
65 
66 	bpf_map_update_elem(&array_map3, &key, &val, 0);
67 
68 	map_val = bpf_map_lookup_elem(&array_map3, &key);
69 	if (!map_val)
70 		return -ENOENT;
71 
72 	bpf_dynptr_from_mem(map_val, sizeof(*map_val), 0, ptr);
73 
74 	return 0;
75 }
76 
77 /* Every bpf_ringbuf_reserve_dynptr call must have a corresponding
78  * bpf_ringbuf_submit/discard_dynptr call
79  */
80 SEC("?raw_tp")
81 __failure __msg("Unreleased reference id=1")
82 int ringbuf_missing_release1(void *ctx)
83 {
84 	struct bpf_dynptr ptr = {};
85 
86 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
87 
88 	/* missing a call to bpf_ringbuf_discard/submit_dynptr */
89 
90 	return 0;
91 }
92 
93 SEC("?raw_tp")
94 __failure __msg("Unreleased reference id=3")
95 int ringbuf_missing_release2(void *ctx)
96 {
97 	struct bpf_dynptr ptr1, ptr2;
98 	struct sample *sample;
99 
100 	bpf_ringbuf_reserve_dynptr(&ringbuf, sizeof(*sample), 0, &ptr1);
101 	bpf_ringbuf_reserve_dynptr(&ringbuf, sizeof(*sample), 0, &ptr2);
102 
103 	sample = bpf_dynptr_data(&ptr1, 0, sizeof(*sample));
104 	if (!sample) {
105 		bpf_ringbuf_discard_dynptr(&ptr1, 0);
106 		bpf_ringbuf_discard_dynptr(&ptr2, 0);
107 		return 0;
108 	}
109 
110 	bpf_ringbuf_submit_dynptr(&ptr1, 0);
111 
112 	/* missing a call to bpf_ringbuf_discard/submit_dynptr on ptr2 */
113 
114 	return 0;
115 }
116 
117 static int missing_release_callback_fn(__u32 index, void *data)
118 {
119 	struct bpf_dynptr ptr;
120 
121 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
122 
123 	/* missing a call to bpf_ringbuf_discard/submit_dynptr */
124 
125 	return 0;
126 }
127 
128 /* Any dynptr initialized within a callback must have bpf_dynptr_put called */
129 SEC("?raw_tp")
130 __failure __msg("Unreleased reference id")
131 int ringbuf_missing_release_callback(void *ctx)
132 {
133 	bpf_loop(10, missing_release_callback_fn, NULL, 0);
134 	return 0;
135 }
136 
137 /* Can't call bpf_ringbuf_submit/discard_dynptr on a non-initialized dynptr */
138 SEC("?raw_tp")
139 __failure __msg("Expected an initialized dynptr as R1")
140 int ringbuf_release_uninit_dynptr(void *ctx)
141 {
142 	struct bpf_dynptr ptr;
143 
144 	/* this should fail */
145 	bpf_ringbuf_submit_dynptr(&ptr, 0);
146 
147 	return 0;
148 }
149 
150 /* A dynptr can't be used after it has been invalidated */
151 SEC("?raw_tp")
152 __failure __msg("Expected an initialized dynptr as R3")
153 int use_after_invalid(void *ctx)
154 {
155 	struct bpf_dynptr ptr;
156 	char read_data[64];
157 
158 	bpf_ringbuf_reserve_dynptr(&ringbuf, sizeof(read_data), 0, &ptr);
159 
160 	bpf_dynptr_read(read_data, sizeof(read_data), &ptr, 0, 0);
161 
162 	bpf_ringbuf_submit_dynptr(&ptr, 0);
163 
164 	/* this should fail */
165 	bpf_dynptr_read(read_data, sizeof(read_data), &ptr, 0, 0);
166 
167 	return 0;
168 }
169 
170 /* Can't call non-dynptr ringbuf APIs on a dynptr ringbuf sample */
171 SEC("?raw_tp")
172 __failure __msg("type=mem expected=ringbuf_mem")
173 int ringbuf_invalid_api(void *ctx)
174 {
175 	struct bpf_dynptr ptr;
176 	struct sample *sample;
177 
178 	bpf_ringbuf_reserve_dynptr(&ringbuf, sizeof(*sample), 0, &ptr);
179 	sample = bpf_dynptr_data(&ptr, 0, sizeof(*sample));
180 	if (!sample)
181 		goto done;
182 
183 	sample->pid = 123;
184 
185 	/* invalid API use. need to use dynptr API to submit/discard */
186 	bpf_ringbuf_submit(sample, 0);
187 
188 done:
189 	bpf_ringbuf_discard_dynptr(&ptr, 0);
190 	return 0;
191 }
192 
193 /* Can't add a dynptr to a map */
194 SEC("?raw_tp")
195 __failure __msg("invalid read from stack")
196 int add_dynptr_to_map1(void *ctx)
197 {
198 	struct bpf_dynptr ptr;
199 	int key = 0;
200 
201 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
202 
203 	/* this should fail */
204 	bpf_map_update_elem(&array_map1, &key, &ptr, 0);
205 
206 	bpf_ringbuf_submit_dynptr(&ptr, 0);
207 
208 	return 0;
209 }
210 
211 /* Can't add a struct with an embedded dynptr to a map */
212 SEC("?raw_tp")
213 __failure __msg("invalid read from stack")
214 int add_dynptr_to_map2(void *ctx)
215 {
216 	struct test_info x;
217 	int key = 0;
218 
219 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &x.ptr);
220 
221 	/* this should fail */
222 	bpf_map_update_elem(&array_map2, &key, &x, 0);
223 
224 	bpf_ringbuf_submit_dynptr(&x.ptr, 0);
225 
226 	return 0;
227 }
228 
229 /* A data slice can't be accessed out of bounds */
230 SEC("?raw_tp")
231 __failure __msg("value is outside of the allowed memory range")
232 int data_slice_out_of_bounds_ringbuf(void *ctx)
233 {
234 	struct bpf_dynptr ptr;
235 	void *data;
236 
237 	bpf_ringbuf_reserve_dynptr(&ringbuf, 8, 0, &ptr);
238 
239 	data  = bpf_dynptr_data(&ptr, 0, 8);
240 	if (!data)
241 		goto done;
242 
243 	/* can't index out of bounds of the data slice */
244 	val = *((char *)data + 8);
245 
246 done:
247 	bpf_ringbuf_submit_dynptr(&ptr, 0);
248 	return 0;
249 }
250 
251 /* A data slice can't be accessed out of bounds */
252 SEC("?tc")
253 __failure __msg("value is outside of the allowed memory range")
254 int data_slice_out_of_bounds_skb(struct __sk_buff *skb)
255 {
256 	struct bpf_dynptr ptr;
257 	struct ethhdr *hdr;
258 	char buffer[sizeof(*hdr)] = {};
259 
260 	bpf_dynptr_from_skb(skb, 0, &ptr);
261 
262 	hdr = bpf_dynptr_slice_rdwr(&ptr, 0, buffer, sizeof(buffer));
263 	if (!hdr)
264 		return SK_DROP;
265 
266 	/* this should fail */
267 	*(__u8*)(hdr + 1) = 1;
268 
269 	return SK_PASS;
270 }
271 
272 /* A metadata slice can't be accessed out of bounds */
273 SEC("?tc")
274 __failure __msg("value is outside of the allowed memory range")
275 int data_slice_out_of_bounds_skb_meta(struct __sk_buff *skb)
276 {
277 	struct bpf_dynptr meta;
278 	__u8 *md;
279 
280 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
281 
282 	md = bpf_dynptr_slice_rdwr(&meta, 0, NULL, sizeof(*md));
283 	if (!md)
284 		return SK_DROP;
285 
286 	/* this should fail */
287 	*(md + 1) = 42;
288 
289 	return SK_PASS;
290 }
291 
292 SEC("?raw_tp")
293 __failure __msg("value is outside of the allowed memory range")
294 int data_slice_out_of_bounds_map_value(void *ctx)
295 {
296 	__u32 map_val;
297 	struct bpf_dynptr ptr;
298 	void *data;
299 
300 	get_map_val_dynptr(&ptr);
301 
302 	data  = bpf_dynptr_data(&ptr, 0, sizeof(map_val));
303 	if (!data)
304 		return 0;
305 
306 	/* can't index out of bounds of the data slice */
307 	val = *((char *)data + (sizeof(map_val) + 1));
308 
309 	return 0;
310 }
311 
312 /* A data slice can't be used after it has been released */
313 SEC("?raw_tp")
314 __failure __msg("invalid mem access 'scalar'")
315 int data_slice_use_after_release1(void *ctx)
316 {
317 	struct bpf_dynptr ptr;
318 	struct sample *sample;
319 
320 	bpf_ringbuf_reserve_dynptr(&ringbuf, sizeof(*sample), 0, &ptr);
321 	sample = bpf_dynptr_data(&ptr, 0, sizeof(*sample));
322 	if (!sample)
323 		goto done;
324 
325 	sample->pid = 123;
326 
327 	bpf_ringbuf_submit_dynptr(&ptr, 0);
328 
329 	/* this should fail */
330 	val = sample->pid;
331 
332 	return 0;
333 
334 done:
335 	bpf_ringbuf_discard_dynptr(&ptr, 0);
336 	return 0;
337 }
338 
339 /* A data slice can't be used after it has been released.
340  *
341  * This tests the case where the data slice tracks a dynptr (ptr2)
342  * that is at a non-zero offset from the frame pointer (ptr1 is at fp,
343  * ptr2 is at fp - 16).
344  */
345 SEC("?raw_tp")
346 __failure __msg("invalid mem access 'scalar'")
347 int data_slice_use_after_release2(void *ctx)
348 {
349 	struct bpf_dynptr ptr1, ptr2;
350 	struct sample *sample;
351 
352 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr1);
353 	bpf_ringbuf_reserve_dynptr(&ringbuf, sizeof(*sample), 0, &ptr2);
354 
355 	sample = bpf_dynptr_data(&ptr2, 0, sizeof(*sample));
356 	if (!sample)
357 		goto done;
358 
359 	sample->pid = 23;
360 
361 	bpf_ringbuf_submit_dynptr(&ptr2, 0);
362 
363 	/* this should fail */
364 	sample->pid = 23;
365 
366 	bpf_ringbuf_submit_dynptr(&ptr1, 0);
367 
368 	return 0;
369 
370 done:
371 	bpf_ringbuf_discard_dynptr(&ptr2, 0);
372 	bpf_ringbuf_discard_dynptr(&ptr1, 0);
373 	return 0;
374 }
375 
376 /* A data slice must be first checked for NULL */
377 SEC("?raw_tp")
378 __failure __msg("invalid mem access 'mem_or_null'")
379 int data_slice_missing_null_check1(void *ctx)
380 {
381 	struct bpf_dynptr ptr;
382 	void *data;
383 
384 	bpf_ringbuf_reserve_dynptr(&ringbuf, 8, 0, &ptr);
385 
386 	data  = bpf_dynptr_data(&ptr, 0, 8);
387 
388 	/* missing if (!data) check */
389 
390 	/* this should fail */
391 	*(__u8 *)data = 3;
392 
393 	bpf_ringbuf_submit_dynptr(&ptr, 0);
394 	return 0;
395 }
396 
397 /* A data slice can't be dereferenced if it wasn't checked for null */
398 SEC("?raw_tp")
399 __failure __msg("invalid mem access 'mem_or_null'")
400 int data_slice_missing_null_check2(void *ctx)
401 {
402 	struct bpf_dynptr ptr;
403 	__u64 *data1, *data2;
404 
405 	bpf_ringbuf_reserve_dynptr(&ringbuf, 16, 0, &ptr);
406 
407 	data1 = bpf_dynptr_data(&ptr, 0, 8);
408 	data2 = bpf_dynptr_data(&ptr, 0, 8);
409 	if (data1)
410 		/* this should fail */
411 		*data2 = 3;
412 
413 	bpf_ringbuf_discard_dynptr(&ptr, 0);
414 	return 0;
415 }
416 
417 /* Can't pass in a dynptr as an arg to a helper function that doesn't take in a
418  * dynptr argument
419  */
420 SEC("?raw_tp")
421 __failure __msg("invalid read from stack")
422 int invalid_helper1(void *ctx)
423 {
424 	struct bpf_dynptr ptr;
425 
426 	get_map_val_dynptr(&ptr);
427 
428 	/* this should fail */
429 	bpf_strncmp((const char *)&ptr, sizeof(ptr), "hello!");
430 
431 	return 0;
432 }
433 
434 /* A dynptr can't be passed into a helper function at a non-zero offset */
435 SEC("?raw_tp")
436 __failure __msg("cannot pass in dynptr at an offset=-8")
437 int invalid_helper2(void *ctx)
438 {
439 	struct bpf_dynptr ptr;
440 	char read_data[64];
441 
442 	get_map_val_dynptr(&ptr);
443 
444 	/* this should fail */
445 	bpf_dynptr_read(read_data, sizeof(read_data), (void *)&ptr + 8, 0, 0);
446 	return 0;
447 }
448 
449 /* A bpf_dynptr is invalidated if it's been written into */
450 SEC("?raw_tp")
451 __failure __msg("Expected an initialized dynptr as R1")
452 int invalid_write1(void *ctx)
453 {
454 	struct bpf_dynptr ptr;
455 	void *data;
456 	__u8 x = 0;
457 
458 	get_map_val_dynptr(&ptr);
459 
460 	memcpy(&ptr, &x, sizeof(x));
461 
462 	/* this should fail */
463 	data = bpf_dynptr_data(&ptr, 0, 1);
464 	__sink(data);
465 
466 	return 0;
467 }
468 
469 /*
470  * A bpf_dynptr can't be used as a dynptr if it has been written into at a fixed
471  * offset
472  */
473 SEC("?raw_tp")
474 __failure __msg("cannot overwrite referenced dynptr")
475 int invalid_write2(void *ctx)
476 {
477 	struct bpf_dynptr ptr;
478 	char read_data[64];
479 	__u8 x = 0;
480 
481 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr);
482 
483 	memcpy((void *)&ptr + 8, &x, sizeof(x));
484 
485 	/* this should fail */
486 	bpf_dynptr_read(read_data, sizeof(read_data), &ptr, 0, 0);
487 
488 	bpf_ringbuf_submit_dynptr(&ptr, 0);
489 
490 	return 0;
491 }
492 
493 /*
494  * A bpf_dynptr can't be used as a dynptr if it has been written into at a
495  * non-const offset
496  */
497 SEC("?raw_tp")
498 __failure __msg("cannot overwrite referenced dynptr")
499 int invalid_write3(void *ctx)
500 {
501 	struct bpf_dynptr ptr;
502 	char stack_buf[16];
503 	unsigned long len;
504 	__u8 x = 0;
505 
506 	bpf_ringbuf_reserve_dynptr(&ringbuf, 8, 0, &ptr);
507 
508 	memcpy(stack_buf, &val, sizeof(val));
509 	len = stack_buf[0] & 0xf;
510 
511 	memcpy((void *)&ptr + len, &x, sizeof(x));
512 
513 	/* this should fail */
514 	bpf_ringbuf_submit_dynptr(&ptr, 0);
515 
516 	return 0;
517 }
518 
519 static int invalid_write4_callback(__u32 index, void *data)
520 {
521 	*(__u32 *)data = 123;
522 
523 	return 0;
524 }
525 
526 /* If the dynptr is written into in a callback function, it should
527  * be invalidated as a dynptr
528  */
529 SEC("?raw_tp")
530 __failure __msg("cannot overwrite referenced dynptr")
531 int invalid_write4(void *ctx)
532 {
533 	struct bpf_dynptr ptr;
534 
535 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr);
536 
537 	bpf_loop(10, invalid_write4_callback, &ptr, 0);
538 
539 	/* this should fail */
540 	bpf_ringbuf_submit_dynptr(&ptr, 0);
541 
542 	return 0;
543 }
544 
545 /* A globally-defined bpf_dynptr can't be used (it must reside as a stack frame) */
546 struct bpf_dynptr global_dynptr;
547 
548 SEC("?raw_tp")
549 __failure __msg("type=map_value expected=fp")
550 int global(void *ctx)
551 {
552 	/* this should fail */
553 	bpf_ringbuf_reserve_dynptr(&ringbuf, 16, 0, &global_dynptr);
554 
555 	bpf_ringbuf_discard_dynptr(&global_dynptr, 0);
556 
557 	return 0;
558 }
559 
560 /* A direct read should fail */
561 SEC("?raw_tp")
562 __failure __msg("invalid read from stack")
563 int invalid_read1(void *ctx)
564 {
565 	struct bpf_dynptr ptr;
566 
567 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr);
568 
569 	/* this should fail */
570 	val = *(int *)&ptr;
571 
572 	bpf_ringbuf_discard_dynptr(&ptr, 0);
573 
574 	return 0;
575 }
576 
577 /* A direct read at an offset should fail */
578 SEC("?raw_tp")
579 __failure __msg("cannot pass in dynptr at an offset")
580 int invalid_read2(void *ctx)
581 {
582 	struct bpf_dynptr ptr;
583 	char read_data[64];
584 
585 	get_map_val_dynptr(&ptr);
586 
587 	/* this should fail */
588 	bpf_dynptr_read(read_data, sizeof(read_data), (void *)&ptr + 1, 0, 0);
589 
590 	return 0;
591 }
592 
593 /* A direct read at an offset into the lower stack slot should fail */
594 SEC("?raw_tp")
595 __failure __msg("invalid read from stack")
596 int invalid_read3(void *ctx)
597 {
598 	struct bpf_dynptr ptr1, ptr2;
599 
600 	bpf_ringbuf_reserve_dynptr(&ringbuf, 16, 0, &ptr1);
601 	bpf_ringbuf_reserve_dynptr(&ringbuf, 16, 0, &ptr2);
602 
603 	/* this should fail */
604 	memcpy(&val, (void *)&ptr1 + 8, sizeof(val));
605 
606 	bpf_ringbuf_discard_dynptr(&ptr1, 0);
607 	bpf_ringbuf_discard_dynptr(&ptr2, 0);
608 
609 	return 0;
610 }
611 
612 static int invalid_read4_callback(__u32 index, void *data)
613 {
614 	/* this should fail */
615 	val = *(__u32 *)data;
616 
617 	return 0;
618 }
619 
620 /* A direct read within a callback function should fail */
621 SEC("?raw_tp")
622 __failure __msg("invalid read from stack")
623 int invalid_read4(void *ctx)
624 {
625 	struct bpf_dynptr ptr;
626 
627 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr);
628 
629 	bpf_loop(10, invalid_read4_callback, &ptr, 0);
630 
631 	bpf_ringbuf_submit_dynptr(&ptr, 0);
632 
633 	return 0;
634 }
635 
636 /* Initializing a dynptr on an offset should fail */
637 SEC("?raw_tp")
638 __failure __msg("cannot pass in dynptr at an offset=0")
639 int invalid_offset(void *ctx)
640 {
641 	struct bpf_dynptr ptr;
642 
643 	/* this should fail */
644 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr + 1);
645 
646 	bpf_ringbuf_discard_dynptr(&ptr, 0);
647 
648 	return 0;
649 }
650 
651 /* Can't release a dynptr twice */
652 SEC("?raw_tp")
653 __failure __msg("Expected an initialized dynptr as R1")
654 int release_twice(void *ctx)
655 {
656 	struct bpf_dynptr ptr;
657 
658 	bpf_ringbuf_reserve_dynptr(&ringbuf, 16, 0, &ptr);
659 
660 	bpf_ringbuf_discard_dynptr(&ptr, 0);
661 
662 	/* this second release should fail */
663 	bpf_ringbuf_discard_dynptr(&ptr, 0);
664 
665 	return 0;
666 }
667 
668 static int release_twice_callback_fn(__u32 index, void *data)
669 {
670 	/* this should fail */
671 	bpf_ringbuf_discard_dynptr(data, 0);
672 
673 	return 0;
674 }
675 
676 /* Test that releasing a dynptr twice, where one of the releases happens
677  * within a callback function, fails
678  */
679 SEC("?raw_tp")
680 __failure __msg("Expected an initialized dynptr as R1")
681 int release_twice_callback(void *ctx)
682 {
683 	struct bpf_dynptr ptr;
684 
685 	bpf_ringbuf_reserve_dynptr(&ringbuf, 32, 0, &ptr);
686 
687 	bpf_ringbuf_discard_dynptr(&ptr, 0);
688 
689 	bpf_loop(10, release_twice_callback_fn, &ptr, 0);
690 
691 	return 0;
692 }
693 
694 /* Reject unsupported local mem types for dynptr_from_mem API */
695 SEC("?raw_tp")
696 __failure __msg("Unsupported reg type fp for bpf_dynptr_from_mem data")
697 int dynptr_from_mem_invalid_api(void *ctx)
698 {
699 	struct bpf_dynptr ptr;
700 	int x = 0;
701 
702 	/* this should fail */
703 	bpf_dynptr_from_mem(&x, sizeof(x), 0, &ptr);
704 
705 	return 0;
706 }
707 
708 /* Cannot create dynptr from dynptr data */
709 SEC("?raw_tp")
710 __failure __msg("Unsupported reg type mem for bpf_dynptr_from_mem data")
711 int dynptr_from_dynptr_data(void *ctx)
712 {
713 	struct bpf_dynptr ptr, ptr2;
714 	__u8 *data;
715 
716 	if (get_map_val_dynptr(&ptr))
717 		return 0;
718 
719 	data = bpf_dynptr_data(&ptr, 0, sizeof(__u32));
720 	if (!data)
721 		return 0;
722 
723 	/* this should fail */
724 	bpf_dynptr_from_mem(data, sizeof(__u32), 0, &ptr2);
725 
726 	return 0;
727 }
728 
729 /* Cannot create dynptr from dynptr slice */
730 SEC("?tc")
731 __failure __msg("Unsupported reg type mem for bpf_dynptr_from_mem data")
732 int dynptr_from_dynptr_slice(struct __sk_buff *skb)
733 {
734 	struct bpf_dynptr ptr, ptr2;
735 	struct ethhdr *hdr;
736 	char buffer[sizeof(*hdr)] = {};
737 
738 	bpf_dynptr_from_skb(skb, 0, &ptr);
739 
740 	hdr = bpf_dynptr_slice_rdwr(&ptr, 0, buffer, sizeof(buffer));
741 	if (!hdr)
742 		return SK_DROP;
743 
744 	/* this should fail */
745 	bpf_dynptr_from_mem(hdr, sizeof(*hdr), 0, &ptr2);
746 
747 	return SK_PASS;
748 }
749 
750 SEC("?tc")
751 __failure __msg("cannot overwrite referenced dynptr") __log_level(2)
752 int dynptr_pruning_overwrite(struct __sk_buff *ctx)
753 {
754 	asm volatile (
755 		"r9 = 0xeB9F;				\
756 		 r6 = %[ringbuf] ll;			\
757 		 r1 = r6;				\
758 		 r2 = 8;				\
759 		 r3 = 0;				\
760 		 r4 = r10;				\
761 		 r4 += -16;				\
762 		 call %[bpf_ringbuf_reserve_dynptr];	\
763 		 if r0 == 0 goto pjmp1;			\
764 		 goto pjmp2;				\
765 	pjmp1:						\
766 		 *(u64 *)(r10 - 16) = r9;		\
767 	pjmp2:						\
768 		 r1 = r10;				\
769 		 r1 += -16;				\
770 		 r2 = 0;				\
771 		 call %[bpf_ringbuf_discard_dynptr];	"
772 		:
773 		: __imm(bpf_ringbuf_reserve_dynptr),
774 		  __imm(bpf_ringbuf_discard_dynptr),
775 		  __imm_addr(ringbuf)
776 		: __clobber_all
777 	);
778 	return 0;
779 }
780 
781 SEC("?tc")
782 __success __msg("12: safe") __log_level(2)
783 int dynptr_pruning_stacksafe(struct __sk_buff *ctx)
784 {
785 	asm volatile (
786 		"r9 = 0xeB9F;				\
787 		 r6 = %[ringbuf] ll;			\
788 		 r1 = r6;				\
789 		 r2 = 8;				\
790 		 r3 = 0;				\
791 		 r4 = r10;				\
792 		 r4 += -16;				\
793 		 call %[bpf_ringbuf_reserve_dynptr];	\
794 		 if r0 == 0 goto stjmp1;		\
795 		 goto stjmp2;				\
796 	stjmp1:						\
797 		 r9 = r9;				\
798 	stjmp2:						\
799 		 r1 = r10;				\
800 		 r1 += -16;				\
801 		 r2 = 0;				\
802 		 call %[bpf_ringbuf_discard_dynptr];	"
803 		:
804 		: __imm(bpf_ringbuf_reserve_dynptr),
805 		  __imm(bpf_ringbuf_discard_dynptr),
806 		  __imm_addr(ringbuf)
807 		: __clobber_all
808 	);
809 	return 0;
810 }
811 
812 SEC("?tc")
813 __failure __msg("cannot overwrite referenced dynptr") __log_level(2)
814 int dynptr_pruning_type_confusion(struct __sk_buff *ctx)
815 {
816 	asm volatile (
817 		"r6 = %[array_map4] ll;			\
818 		 r7 = %[ringbuf] ll;			\
819 		 r1 = r6;				\
820 		 r2 = r10;				\
821 		 r2 += -8;				\
822 		 r9 = 0;				\
823 		 *(u64 *)(r2 + 0) = r9;			\
824 		 r3 = r10;				\
825 		 r3 += -24;				\
826 		 r9 = 0xeB9FeB9F;			\
827 		 *(u64 *)(r10 - 16) = r9;		\
828 		 *(u64 *)(r10 - 24) = r9;		\
829 		 r9 = 0;				\
830 		 r4 = 0;				\
831 		 r8 = r2;				\
832 		 call %[bpf_map_update_elem];		\
833 		 r1 = r6;				\
834 		 r2 = r8;				\
835 		 call %[bpf_map_lookup_elem];		\
836 		 if r0 != 0 goto tjmp1;			\
837 		 exit;					\
838 	tjmp1:						\
839 		 r8 = r0;				\
840 		 r1 = r7;				\
841 		 r2 = 8;				\
842 		 r3 = 0;				\
843 		 r4 = r10;				\
844 		 r4 += -16;				\
845 		 r0 = *(u64 *)(r0 + 0);			\
846 		 call %[bpf_ringbuf_reserve_dynptr];	\
847 		 if r0 == 0 goto tjmp2;			\
848 		 r8 = r8;				\
849 		 r8 = r8;				\
850 		 r8 = r8;				\
851 		 r8 = r8;				\
852 		 r8 = r8;				\
853 		 r8 = r8;				\
854 		 r8 = r8;				\
855 		 goto tjmp3;				\
856 	tjmp2:						\
857 		 *(u64 *)(r10 - 8) = r9;		\
858 		 *(u64 *)(r10 - 16) = r9;		\
859 		 r1 = r8;				\
860 		 r1 += 8;				\
861 		 r2 = 0;				\
862 		 r3 = 0;				\
863 		 r4 = r10;				\
864 		 r4 += -16;				\
865 		 call %[bpf_dynptr_from_mem];		\
866 	tjmp3:						\
867 		 r1 = r10;				\
868 		 r1 += -16;				\
869 		 r2 = 0;				\
870 		 call %[bpf_ringbuf_discard_dynptr];	"
871 		:
872 		: __imm(bpf_map_update_elem),
873 		  __imm(bpf_map_lookup_elem),
874 		  __imm(bpf_ringbuf_reserve_dynptr),
875 		  __imm(bpf_dynptr_from_mem),
876 		  __imm(bpf_ringbuf_discard_dynptr),
877 		  __imm_addr(array_map4),
878 		  __imm_addr(ringbuf)
879 		: __clobber_all
880 	);
881 	return 0;
882 }
883 
884 SEC("?tc")
885 __failure __msg("dynptr has to be at a constant offset") __log_level(2)
886 int dynptr_var_off_overwrite(struct __sk_buff *ctx)
887 {
888 	asm volatile (
889 		"r9 = 16;				\
890 		 *(u32 *)(r10 - 4) = r9;		\
891 		 r8 = *(u32 *)(r10 - 4);		\
892 		 if r8 >= 0 goto vjmp1;			\
893 		 r0 = 1;				\
894 		 exit;					\
895 	vjmp1:						\
896 		 if r8 <= 16 goto vjmp2;		\
897 		 r0 = 1;				\
898 		 exit;					\
899 	vjmp2:						\
900 		 r8 &= 16;				\
901 		 r1 = %[ringbuf] ll;			\
902 		 r2 = 8;				\
903 		 r3 = 0;				\
904 		 r4 = r10;				\
905 		 r4 += -32;				\
906 		 r4 += r8;				\
907 		 call %[bpf_ringbuf_reserve_dynptr];	\
908 		 r9 = 0xeB9F;				\
909 		 *(u64 *)(r10 - 16) = r9;		\
910 		 r1 = r10;				\
911 		 r1 += -32;				\
912 		 r1 += r8;				\
913 		 r2 = 0;				\
914 		 call %[bpf_ringbuf_discard_dynptr];	"
915 		:
916 		: __imm(bpf_ringbuf_reserve_dynptr),
917 		  __imm(bpf_ringbuf_discard_dynptr),
918 		  __imm_addr(ringbuf)
919 		: __clobber_all
920 	);
921 	return 0;
922 }
923 
924 SEC("?tc")
925 __failure __msg("cannot overwrite referenced dynptr") __log_level(2)
926 int dynptr_partial_slot_invalidate(struct __sk_buff *ctx)
927 {
928 	asm volatile (
929 		"r6 = %[ringbuf] ll;			\
930 		 r7 = %[array_map4] ll;			\
931 		 r1 = r7;				\
932 		 r2 = r10;				\
933 		 r2 += -8;				\
934 		 r9 = 0;				\
935 		 *(u64 *)(r2 + 0) = r9;			\
936 		 r3 = r2;				\
937 		 r4 = 0;				\
938 		 r8 = r2;				\
939 		 call %[bpf_map_update_elem];		\
940 		 r1 = r7;				\
941 		 r2 = r8;				\
942 		 call %[bpf_map_lookup_elem];		\
943 		 if r0 != 0 goto sjmp1;			\
944 		 exit;					\
945 	sjmp1:						\
946 		 r7 = r0;				\
947 		 r1 = r6;				\
948 		 r2 = 8;				\
949 		 r3 = 0;				\
950 		 r4 = r10;				\
951 		 r4 += -24;				\
952 		 call %[bpf_ringbuf_reserve_dynptr];	\
953 		 *(u64 *)(r10 - 16) = r9;		\
954 		 r1 = r7;				\
955 		 r2 = 8;				\
956 		 r3 = 0;				\
957 		 r4 = r10;				\
958 		 r4 += -16;				\
959 		 call %[bpf_dynptr_from_mem];		\
960 		 r1 = r10;				\
961 		 r1 += -512;				\
962 		 r2 = 488;				\
963 		 r3 = r10;				\
964 		 r3 += -24;				\
965 		 r4 = 0;				\
966 		 r5 = 0;				\
967 		 call %[bpf_dynptr_read];		\
968 		 r8 = 1;				\
969 		 if r0 != 0 goto sjmp2;			\
970 		 r8 = 0;				\
971 	sjmp2:						\
972 		 r1 = r10;				\
973 		 r1 += -24;				\
974 		 r2 = 0;				\
975 		 call %[bpf_ringbuf_discard_dynptr];	"
976 		:
977 		: __imm(bpf_map_update_elem),
978 		  __imm(bpf_map_lookup_elem),
979 		  __imm(bpf_ringbuf_reserve_dynptr),
980 		  __imm(bpf_ringbuf_discard_dynptr),
981 		  __imm(bpf_dynptr_from_mem),
982 		  __imm(bpf_dynptr_read),
983 		  __imm_addr(ringbuf),
984 		  __imm_addr(array_map4)
985 		: __clobber_all
986 	);
987 	return 0;
988 }
989 
990 /* Test that it is allowed to overwrite unreferenced dynptr. */
991 SEC("?raw_tp")
992 __success
993 int dynptr_overwrite_unref(void *ctx)
994 {
995 	struct bpf_dynptr ptr;
996 
997 	if (get_map_val_dynptr(&ptr))
998 		return 0;
999 	if (get_map_val_dynptr(&ptr))
1000 		return 0;
1001 	if (get_map_val_dynptr(&ptr))
1002 		return 0;
1003 
1004 	return 0;
1005 }
1006 
1007 /* Test that slices are invalidated on reinitializing a dynptr. */
1008 SEC("?raw_tp")
1009 __failure __msg("invalid mem access 'scalar'")
1010 int dynptr_invalidate_slice_reinit(void *ctx)
1011 {
1012 	struct bpf_dynptr ptr;
1013 	__u8 *p;
1014 
1015 	if (get_map_val_dynptr(&ptr))
1016 		return 0;
1017 	p = bpf_dynptr_data(&ptr, 0, 1);
1018 	if (!p)
1019 		return 0;
1020 	if (get_map_val_dynptr(&ptr))
1021 		return 0;
1022 	/* this should fail */
1023 	return *p;
1024 }
1025 
1026 /* Invalidation of dynptr slices on destruction of dynptr should not miss
1027  * mem_or_null pointers.
1028  */
1029 SEC("?raw_tp")
1030 __failure __msg("R{{[0-9]+}} type=scalar expected=percpu_ptr_")
1031 int dynptr_invalidate_slice_or_null(void *ctx)
1032 {
1033 	struct bpf_dynptr ptr;
1034 	__u8 *p;
1035 
1036 	if (get_map_val_dynptr(&ptr))
1037 		return 0;
1038 
1039 	p = bpf_dynptr_data(&ptr, 0, 1);
1040 	*(__u8 *)&ptr = 0;
1041 	/* this should fail */
1042 	bpf_this_cpu_ptr(p);
1043 	return 0;
1044 }
1045 
1046 /* Destruction of dynptr should also any slices obtained from it */
1047 SEC("?raw_tp")
1048 __failure __msg("R{{[0-9]+}} invalid mem access 'scalar'")
1049 int dynptr_invalidate_slice_failure(void *ctx)
1050 {
1051 	struct bpf_dynptr ptr1;
1052 	struct bpf_dynptr ptr2;
1053 	__u8 *p1, *p2;
1054 
1055 	if (get_map_val_dynptr(&ptr1))
1056 		return 0;
1057 	if (get_map_val_dynptr(&ptr2))
1058 		return 0;
1059 
1060 	p1 = bpf_dynptr_data(&ptr1, 0, 1);
1061 	if (!p1)
1062 		return 0;
1063 	p2 = bpf_dynptr_data(&ptr2, 0, 1);
1064 	if (!p2)
1065 		return 0;
1066 
1067 	*(__u8 *)&ptr1 = 0;
1068 	/* this should fail */
1069 	return *p1;
1070 }
1071 
1072 /* Invalidation of slices should be scoped and should not prevent dereferencing
1073  * slices of another dynptr after destroying unrelated dynptr
1074  */
1075 SEC("?raw_tp")
1076 __success
1077 int dynptr_invalidate_slice_success(void *ctx)
1078 {
1079 	struct bpf_dynptr ptr1;
1080 	struct bpf_dynptr ptr2;
1081 	__u8 *p1, *p2;
1082 
1083 	if (get_map_val_dynptr(&ptr1))
1084 		return 1;
1085 	if (get_map_val_dynptr(&ptr2))
1086 		return 1;
1087 
1088 	p1 = bpf_dynptr_data(&ptr1, 0, 1);
1089 	if (!p1)
1090 		return 1;
1091 	p2 = bpf_dynptr_data(&ptr2, 0, 1);
1092 	if (!p2)
1093 		return 1;
1094 
1095 	*(__u8 *)&ptr1 = 0;
1096 	return *p2;
1097 }
1098 
1099 /* Overwriting referenced dynptr should be rejected */
1100 SEC("?raw_tp")
1101 __failure __msg("cannot overwrite referenced dynptr")
1102 int dynptr_overwrite_ref(void *ctx)
1103 {
1104 	struct bpf_dynptr ptr;
1105 
1106 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr);
1107 	/* this should fail */
1108 	if (get_map_val_dynptr(&ptr))
1109 		bpf_ringbuf_discard_dynptr(&ptr, 0);
1110 	return 0;
1111 }
1112 
1113 /* Reject writes to dynptr slot from bpf_dynptr_read */
1114 SEC("?raw_tp")
1115 __failure __msg("potential write to dynptr at off=-16")
1116 int dynptr_read_into_slot(void *ctx)
1117 {
1118 	union {
1119 		struct {
1120 			char _pad[48];
1121 			struct bpf_dynptr ptr;
1122 		};
1123 		char buf[64];
1124 	} data;
1125 
1126 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &data.ptr);
1127 	/* this should fail */
1128 	bpf_dynptr_read(data.buf, sizeof(data.buf), &data.ptr, 0, 0);
1129 
1130 	return 0;
1131 }
1132 
1133 /* bpf_dynptr_slice()s are read-only and cannot be written to */
1134 SEC("?tc")
1135 __failure __msg("R{{[0-9]+}} cannot write into rdonly_mem")
1136 int skb_invalid_slice_write(struct __sk_buff *skb)
1137 {
1138 	struct bpf_dynptr ptr;
1139 	struct ethhdr *hdr;
1140 	char buffer[sizeof(*hdr)] = {};
1141 
1142 	bpf_dynptr_from_skb(skb, 0, &ptr);
1143 
1144 	hdr = bpf_dynptr_slice(&ptr, 0, buffer, sizeof(buffer));
1145 	if (!hdr)
1146 		return SK_DROP;
1147 
1148 	/* this should fail */
1149 	hdr->h_proto = 1;
1150 
1151 	return SK_PASS;
1152 }
1153 
1154 /* bpf_dynptr_slice()s are read-only and cannot be written to */
1155 SEC("?tc")
1156 __failure __msg("R{{[0-9]+}} cannot write into rdonly_mem")
1157 int skb_meta_invalid_slice_write(struct __sk_buff *skb)
1158 {
1159 	struct bpf_dynptr meta;
1160 	__u8 *md;
1161 
1162 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1163 
1164 	md = bpf_dynptr_slice(&meta, 0, NULL, sizeof(*md));
1165 	if (!md)
1166 		return SK_DROP;
1167 
1168 	/* this should fail */
1169 	*md = 42;
1170 
1171 	return SK_PASS;
1172 }
1173 
1174 /* The read-only data slice is invalidated whenever a helper changes packet data */
1175 SEC("?tc")
1176 __failure __msg("invalid mem access 'scalar'")
1177 int skb_invalid_data_slice1(struct __sk_buff *skb)
1178 {
1179 	struct bpf_dynptr ptr;
1180 	struct ethhdr *hdr;
1181 	char buffer[sizeof(*hdr)] = {};
1182 
1183 	bpf_dynptr_from_skb(skb, 0, &ptr);
1184 
1185 	hdr = bpf_dynptr_slice(&ptr, 0, buffer, sizeof(buffer));
1186 	if (!hdr)
1187 		return SK_DROP;
1188 
1189 	val = hdr->h_proto;
1190 
1191 	if (bpf_skb_pull_data(skb, skb->len))
1192 		return SK_DROP;
1193 
1194 	/* this should fail */
1195 	val = hdr->h_proto;
1196 
1197 	return SK_PASS;
1198 }
1199 
1200 /* The read-write data slice is invalidated whenever a helper changes packet data */
1201 SEC("?tc")
1202 __failure __msg("invalid mem access 'scalar'")
1203 int skb_invalid_data_slice2(struct __sk_buff *skb)
1204 {
1205 	struct bpf_dynptr ptr;
1206 	struct ethhdr *hdr;
1207 	char buffer[sizeof(*hdr)] = {};
1208 
1209 	bpf_dynptr_from_skb(skb, 0, &ptr);
1210 
1211 	hdr = bpf_dynptr_slice_rdwr(&ptr, 0, buffer, sizeof(buffer));
1212 	if (!hdr)
1213 		return SK_DROP;
1214 
1215 	hdr->h_proto = 123;
1216 
1217 	if (bpf_skb_pull_data(skb, skb->len))
1218 		return SK_DROP;
1219 
1220 	/* this should fail */
1221 	hdr->h_proto = 1;
1222 
1223 	return SK_PASS;
1224 }
1225 
1226 /* The read-only data slice is invalidated whenever bpf_dynptr_write() is called */
1227 SEC("?tc")
1228 __failure __msg("invalid mem access 'scalar'")
1229 int skb_invalid_data_slice3(struct __sk_buff *skb)
1230 {
1231 	char write_data[64] = "hello there, world!!";
1232 	struct bpf_dynptr ptr;
1233 	struct ethhdr *hdr;
1234 	char buffer[sizeof(*hdr)] = {};
1235 
1236 	bpf_dynptr_from_skb(skb, 0, &ptr);
1237 
1238 	hdr = bpf_dynptr_slice(&ptr, 0, buffer, sizeof(buffer));
1239 	if (!hdr)
1240 		return SK_DROP;
1241 
1242 	val = hdr->h_proto;
1243 
1244 	bpf_dynptr_write(&ptr, 0, write_data, sizeof(write_data), 0);
1245 
1246 	/* this should fail */
1247 	val = hdr->h_proto;
1248 
1249 	return SK_PASS;
1250 }
1251 
1252 /* The read-write data slice is invalidated whenever bpf_dynptr_write() is called */
1253 SEC("?tc")
1254 __failure __msg("invalid mem access 'scalar'")
1255 int skb_invalid_data_slice4(struct __sk_buff *skb)
1256 {
1257 	char write_data[64] = "hello there, world!!";
1258 	struct bpf_dynptr ptr;
1259 	struct ethhdr *hdr;
1260 	char buffer[sizeof(*hdr)] = {};
1261 
1262 	bpf_dynptr_from_skb(skb, 0, &ptr);
1263 	hdr = bpf_dynptr_slice_rdwr(&ptr, 0, buffer, sizeof(buffer));
1264 	if (!hdr)
1265 		return SK_DROP;
1266 
1267 	hdr->h_proto = 123;
1268 
1269 	bpf_dynptr_write(&ptr, 0, write_data, sizeof(write_data), 0);
1270 
1271 	/* this should fail */
1272 	hdr->h_proto = 1;
1273 
1274 	return SK_PASS;
1275 }
1276 
1277 /* Read-only skb data slice is invalidated on write to skb metadata */
1278 SEC("?tc")
1279 __failure __msg("invalid mem access 'scalar'")
1280 int ro_skb_slice_invalid_after_metadata_write(struct __sk_buff *skb)
1281 {
1282 	struct bpf_dynptr data, meta;
1283 	__u8 *d;
1284 
1285 	bpf_dynptr_from_skb(skb, 0, &data);
1286 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1287 
1288 	d = bpf_dynptr_slice(&data, 0, NULL, sizeof(*d));
1289 	if (!d)
1290 		return SK_DROP;
1291 
1292 	bpf_dynptr_write(&meta, 0, "x", 1, 0);
1293 
1294 	/* this should fail */
1295 	val = *d;
1296 
1297 	return SK_PASS;
1298 }
1299 
1300 /* Read-write skb data slice is invalidated on write to skb metadata */
1301 SEC("?tc")
1302 __failure __msg("invalid mem access 'scalar'")
1303 int rw_skb_slice_invalid_after_metadata_write(struct __sk_buff *skb)
1304 {
1305 	struct bpf_dynptr data, meta;
1306 	__u8 *d;
1307 
1308 	bpf_dynptr_from_skb(skb, 0, &data);
1309 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1310 
1311 	d = bpf_dynptr_slice_rdwr(&data, 0, NULL, sizeof(*d));
1312 	if (!d)
1313 		return SK_DROP;
1314 
1315 	bpf_dynptr_write(&meta, 0, "x", 1, 0);
1316 
1317 	/* this should fail */
1318 	*d = 42;
1319 
1320 	return SK_PASS;
1321 }
1322 
1323 /* Read-only skb metadata slice is invalidated on write to skb data */
1324 SEC("?tc")
1325 __failure __msg("invalid mem access 'scalar'")
1326 int ro_skb_meta_slice_invalid_after_payload_write(struct __sk_buff *skb)
1327 {
1328 	struct bpf_dynptr data, meta;
1329 	__u8 *md;
1330 
1331 	bpf_dynptr_from_skb(skb, 0, &data);
1332 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1333 
1334 	md = bpf_dynptr_slice(&meta, 0, NULL, sizeof(*md));
1335 	if (!md)
1336 		return SK_DROP;
1337 
1338 	bpf_dynptr_write(&data, 0, "x", 1, 0);
1339 
1340 	/* this should fail */
1341 	val = *md;
1342 
1343 	return SK_PASS;
1344 }
1345 
1346 /* Read-write skb metadata slice is invalidated on write to skb data slice */
1347 SEC("?tc")
1348 __failure __msg("invalid mem access 'scalar'")
1349 int rw_skb_meta_slice_invalid_after_payload_write(struct __sk_buff *skb)
1350 {
1351 	struct bpf_dynptr data, meta;
1352 	__u8 *md;
1353 
1354 	bpf_dynptr_from_skb(skb, 0, &data);
1355 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1356 
1357 	md = bpf_dynptr_slice_rdwr(&meta, 0, NULL, sizeof(*md));
1358 	if (!md)
1359 		return SK_DROP;
1360 
1361 	bpf_dynptr_write(&data, 0, "x", 1, 0);
1362 
1363 	/* this should fail */
1364 	*md = 42;
1365 
1366 	return SK_PASS;
1367 }
1368 
1369 /* Read-only skb metadata slice is invalidated whenever a helper changes packet data */
1370 SEC("?tc")
1371 __failure __msg("invalid mem access 'scalar'")
1372 int ro_skb_meta_slice_invalid_after_payload_helper(struct __sk_buff *skb)
1373 {
1374 	struct bpf_dynptr meta;
1375 	__u8 *md;
1376 
1377 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1378 
1379 	md = bpf_dynptr_slice(&meta, 0, NULL, sizeof(*md));
1380 	if (!md)
1381 		return SK_DROP;
1382 
1383 	if (bpf_skb_pull_data(skb, skb->len))
1384 		return SK_DROP;
1385 
1386 	/* this should fail */
1387 	val = *md;
1388 
1389 	return SK_PASS;
1390 }
1391 
1392 /* Read-write skb metadata slice is invalidated whenever a helper changes packet data */
1393 SEC("?tc")
1394 __failure __msg("invalid mem access 'scalar'")
1395 int rw_skb_meta_slice_invalid_after_payload_helper(struct __sk_buff *skb)
1396 {
1397 	struct bpf_dynptr meta;
1398 	__u8 *md;
1399 
1400 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1401 
1402 	md = bpf_dynptr_slice_rdwr(&meta, 0, NULL, sizeof(*md));
1403 	if (!md)
1404 		return SK_DROP;
1405 
1406 	if (bpf_skb_pull_data(skb, skb->len))
1407 		return SK_DROP;
1408 
1409 	/* this should fail */
1410 	*md = 42;
1411 
1412 	return SK_PASS;
1413 }
1414 
1415 /* Read-only skb metadata slice is invalidated on write to skb metadata */
1416 SEC("?tc")
1417 __failure __msg("invalid mem access 'scalar'")
1418 int ro_skb_meta_slice_invalid_after_metadata_write(struct __sk_buff *skb)
1419 {
1420 	struct bpf_dynptr meta;
1421 	__u8 *md;
1422 
1423 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1424 
1425 	md = bpf_dynptr_slice(&meta, 0, NULL, sizeof(*md));
1426 	if (!md)
1427 		return SK_DROP;
1428 
1429 	bpf_dynptr_write(&meta, 0, "x", 1, 0);
1430 
1431 	/* this should fail */
1432 	val = *md;
1433 
1434 	return SK_PASS;
1435 }
1436 
1437 /* Read-write skb metadata slice is invalidated on write to skb metadata */
1438 SEC("?tc")
1439 __failure __msg("invalid mem access 'scalar'")
1440 int rw_skb_meta_slice_invalid_after_metadata_write(struct __sk_buff *skb)
1441 {
1442 	struct bpf_dynptr meta;
1443 	__u8 *md;
1444 
1445 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1446 
1447 	md = bpf_dynptr_slice_rdwr(&meta, 0, NULL, sizeof(*md));
1448 	if (!md)
1449 		return SK_DROP;
1450 
1451 	bpf_dynptr_write(&meta, 0, "x", 1, 0);
1452 
1453 	/* this should fail */
1454 	*md = 42;
1455 
1456 	return SK_PASS;
1457 }
1458 
1459 /* The read-only data slice is invalidated whenever a helper changes packet data */
1460 SEC("?xdp")
1461 __failure __msg("invalid mem access 'scalar'")
1462 int xdp_invalid_data_slice1(struct xdp_md *xdp)
1463 {
1464 	struct bpf_dynptr ptr;
1465 	struct ethhdr *hdr;
1466 	char buffer[sizeof(*hdr)] = {};
1467 
1468 	bpf_dynptr_from_xdp(xdp, 0, &ptr);
1469 	hdr = bpf_dynptr_slice(&ptr, 0, buffer, sizeof(buffer));
1470 	if (!hdr)
1471 		return SK_DROP;
1472 
1473 	val = hdr->h_proto;
1474 
1475 	if (bpf_xdp_adjust_head(xdp, 0 - (int)sizeof(*hdr)))
1476 		return XDP_DROP;
1477 
1478 	/* this should fail */
1479 	val = hdr->h_proto;
1480 
1481 	return XDP_PASS;
1482 }
1483 
1484 /* The read-write data slice is invalidated whenever a helper changes packet data */
1485 SEC("?xdp")
1486 __failure __msg("invalid mem access 'scalar'")
1487 int xdp_invalid_data_slice2(struct xdp_md *xdp)
1488 {
1489 	struct bpf_dynptr ptr;
1490 	struct ethhdr *hdr;
1491 	char buffer[sizeof(*hdr)] = {};
1492 
1493 	bpf_dynptr_from_xdp(xdp, 0, &ptr);
1494 	hdr = bpf_dynptr_slice_rdwr(&ptr, 0, buffer, sizeof(buffer));
1495 	if (!hdr)
1496 		return SK_DROP;
1497 
1498 	hdr->h_proto = 9;
1499 
1500 	if (bpf_xdp_adjust_head(xdp, 0 - (int)sizeof(*hdr)))
1501 		return XDP_DROP;
1502 
1503 	/* this should fail */
1504 	hdr->h_proto = 1;
1505 
1506 	return XDP_PASS;
1507 }
1508 
1509 /* Only supported prog type can create skb-type dynptrs */
1510 SEC("?xdp")
1511 __failure __msg("calling kernel function bpf_dynptr_from_skb is not allowed")
1512 int skb_invalid_ctx(void *ctx)
1513 {
1514 	struct bpf_dynptr ptr;
1515 
1516 	/* this should fail */
1517 	bpf_dynptr_from_skb(ctx, 0, &ptr);
1518 
1519 	return 0;
1520 }
1521 
1522 /* Only supported prog type can create skb_meta-type dynptrs */
1523 SEC("?raw_tp")
1524 __failure __msg("calling kernel function bpf_dynptr_from_skb_meta is not allowed")
1525 int skb_meta_invalid_ctx(void *ctx)
1526 {
1527 	struct bpf_dynptr meta;
1528 
1529 	/* this should fail */
1530 	bpf_dynptr_from_skb_meta(ctx, 0, &meta);
1531 
1532 	return 0;
1533 }
1534 
1535 SEC("fentry/skb_tx_error")
1536 __failure __msg("must be referenced or trusted")
1537 int BPF_PROG(skb_invalid_ctx_fentry, void *skb)
1538 {
1539 	struct bpf_dynptr ptr;
1540 
1541 	/* this should fail */
1542 	bpf_dynptr_from_skb(skb, 0, &ptr);
1543 
1544 	return 0;
1545 }
1546 
1547 SEC("fexit/skb_tx_error")
1548 __failure __msg("must be referenced or trusted")
1549 int BPF_PROG(skb_invalid_ctx_fexit, void *skb)
1550 {
1551 	struct bpf_dynptr ptr;
1552 
1553 	/* this should fail */
1554 	bpf_dynptr_from_skb(skb, 0, &ptr);
1555 
1556 	return 0;
1557 }
1558 
1559 /* Reject writes to dynptr slot for uninit arg */
1560 SEC("?raw_tp")
1561 __failure __msg("potential write to dynptr at off=-16")
1562 int uninit_write_into_slot(void *ctx)
1563 {
1564 	struct {
1565 		char buf[64];
1566 		struct bpf_dynptr ptr;
1567 	} data;
1568 
1569 	bpf_ringbuf_reserve_dynptr(&ringbuf, 80, 0, &data.ptr);
1570 	/* this should fail */
1571 	bpf_get_current_comm(data.buf, 80);
1572 
1573 	return 0;
1574 }
1575 
1576 /* Only supported prog type can create xdp-type dynptrs */
1577 SEC("?raw_tp")
1578 __failure __msg("calling kernel function bpf_dynptr_from_xdp is not allowed")
1579 int xdp_invalid_ctx(void *ctx)
1580 {
1581 	struct bpf_dynptr ptr;
1582 
1583 	/* this should fail */
1584 	bpf_dynptr_from_xdp(ctx, 0, &ptr);
1585 
1586 	return 0;
1587 }
1588 
1589 __u32 hdr_size = sizeof(struct ethhdr);
1590 /* Can't pass in variable-sized len to bpf_dynptr_slice */
1591 SEC("?tc")
1592 __failure __msg("unbounded memory access")
1593 int dynptr_slice_var_len1(struct __sk_buff *skb)
1594 {
1595 	struct bpf_dynptr ptr;
1596 	struct ethhdr *hdr;
1597 	char buffer[sizeof(*hdr)] = {};
1598 
1599 	bpf_dynptr_from_skb(skb, 0, &ptr);
1600 
1601 	/* this should fail */
1602 	hdr = bpf_dynptr_slice(&ptr, 0, buffer, hdr_size);
1603 	if (!hdr)
1604 		return SK_DROP;
1605 
1606 	return SK_PASS;
1607 }
1608 
1609 /* Can't pass in variable-sized len to bpf_dynptr_slice */
1610 SEC("?tc")
1611 __failure __msg("must be a known constant")
1612 int dynptr_slice_var_len2(struct __sk_buff *skb)
1613 {
1614 	char buffer[sizeof(struct ethhdr)] = {};
1615 	struct bpf_dynptr ptr;
1616 	struct ethhdr *hdr;
1617 
1618 	bpf_dynptr_from_skb(skb, 0, &ptr);
1619 
1620 	if (hdr_size <= sizeof(buffer)) {
1621 		/* this should fail */
1622 		hdr = bpf_dynptr_slice_rdwr(&ptr, 0, buffer, hdr_size);
1623 		if (!hdr)
1624 			return SK_DROP;
1625 		hdr->h_proto = 12;
1626 	}
1627 
1628 	return SK_PASS;
1629 }
1630 
1631 static int callback(__u32 index, void *data)
1632 {
1633         *(__u32 *)data = 123;
1634 
1635         return 0;
1636 }
1637 
1638 /* A commuted add should preserve the parent id of a dynptr data slice. */
1639 SEC("?raw_tp")
1640 __failure __msg("invalid mem access 'scalar'")
1641 int dynptr_slice_commuted_invalidate(void *ctx)
1642 {
1643 	struct bpf_dynptr ptr;
1644 	__u32 *slice, *derived;
1645 
1646 	bpf_ringbuf_reserve_dynptr(&ringbuf, sizeof(__u32), 0, &ptr);
1647 
1648 	slice = bpf_dynptr_data(&ptr, 0, sizeof(__u32));
1649 	if (!slice)
1650 		goto done;
1651 
1652 	asm volatile ("%[dst] = 0;"
1653 		"%[dst] += %[src];"
1654 		"%[src] = 0;"
1655 		: [dst]"=&r"(derived), [src]"+r"(slice)
1656 		:
1657 		: "memory");
1658 
1659 	bpf_ringbuf_discard_dynptr(&ptr, 0);
1660 	val = *derived;
1661 	return 0;
1662 
1663 done:
1664 	bpf_ringbuf_discard_dynptr(&ptr, 0);
1665 	return 0;
1666 }
1667 
1668 /* If the dynptr is written into in a callback function, its data
1669  * slices should be invalidated as well.
1670  */
1671 SEC("?raw_tp")
1672 __failure __msg("invalid mem access 'scalar'")
1673 int invalid_data_slices(void *ctx)
1674 {
1675 	struct bpf_dynptr ptr;
1676 	__u32 *slice;
1677 
1678 	if (get_map_val_dynptr(&ptr))
1679 		return 0;
1680 
1681 	slice = bpf_dynptr_data(&ptr, 0, sizeof(__u32));
1682 	if (!slice)
1683 		return 0;
1684 
1685 	bpf_loop(10, callback, &ptr, 0);
1686 
1687 	/* this should fail */
1688 	*slice = 1;
1689 
1690 	return 0;
1691 }
1692 
1693 /* Program types that don't allow writes to packet data should fail if
1694  * bpf_dynptr_slice_rdwr is called
1695  */
1696 SEC("cgroup_skb/ingress")
1697 __failure __msg("the prog does not allow writes to packet data")
1698 int invalid_slice_rdwr_rdonly(struct __sk_buff *skb)
1699 {
1700 	char buffer[sizeof(struct ethhdr)] = {};
1701 	struct bpf_dynptr ptr;
1702 	struct ethhdr *hdr;
1703 
1704 	bpf_dynptr_from_skb(skb, 0, &ptr);
1705 
1706 	/* this should fail since cgroup_skb doesn't allow
1707 	 * changing packet data
1708 	 */
1709 	hdr = bpf_dynptr_slice_rdwr(&ptr, 0, buffer, sizeof(buffer));
1710 	__sink(hdr);
1711 
1712 	return 0;
1713 }
1714 
1715 /* bpf_dynptr_adjust can only be called on initialized dynptrs */
1716 SEC("?raw_tp")
1717 __failure __msg("Expected an initialized dynptr as R1")
1718 int dynptr_adjust_invalid(void *ctx)
1719 {
1720 	struct bpf_dynptr ptr = {};
1721 
1722 	/* this should fail */
1723 	bpf_dynptr_adjust(&ptr, 1, 2);
1724 
1725 	return 0;
1726 }
1727 
1728 /* bpf_dynptr_is_null can only be called on initialized dynptrs */
1729 SEC("?raw_tp")
1730 __failure __msg("Expected an initialized dynptr as R1")
1731 int dynptr_is_null_invalid(void *ctx)
1732 {
1733 	struct bpf_dynptr ptr = {};
1734 
1735 	/* this should fail */
1736 	bpf_dynptr_is_null(&ptr);
1737 
1738 	return 0;
1739 }
1740 
1741 /* bpf_dynptr_is_rdonly can only be called on initialized dynptrs */
1742 SEC("?raw_tp")
1743 __failure __msg("Expected an initialized dynptr as R1")
1744 int dynptr_is_rdonly_invalid(void *ctx)
1745 {
1746 	struct bpf_dynptr ptr = {};
1747 
1748 	/* this should fail */
1749 	bpf_dynptr_is_rdonly(&ptr);
1750 
1751 	return 0;
1752 }
1753 
1754 /* bpf_dynptr_size can only be called on initialized dynptrs */
1755 SEC("?raw_tp")
1756 __failure __msg("Expected an initialized dynptr as R1")
1757 int dynptr_size_invalid(void *ctx)
1758 {
1759 	struct bpf_dynptr ptr = {};
1760 
1761 	/* this should fail */
1762 	bpf_dynptr_size(&ptr);
1763 
1764 	return 0;
1765 }
1766 
1767 /* Only initialized dynptrs can be cloned */
1768 SEC("?raw_tp")
1769 __failure __msg("Expected an initialized dynptr as R1")
1770 int clone_invalid1(void *ctx)
1771 {
1772 	struct bpf_dynptr ptr1 = {};
1773 	struct bpf_dynptr ptr2;
1774 
1775 	/* this should fail */
1776 	bpf_dynptr_clone(&ptr1, &ptr2);
1777 
1778 	return 0;
1779 }
1780 
1781 /* Can't overwrite an existing dynptr when cloning */
1782 SEC("?xdp")
1783 __failure __msg("cannot overwrite referenced dynptr")
1784 int clone_invalid2(struct xdp_md *xdp)
1785 {
1786 	struct bpf_dynptr ptr1;
1787 	struct bpf_dynptr clone;
1788 
1789 	bpf_dynptr_from_xdp(xdp, 0, &ptr1);
1790 
1791 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &clone);
1792 
1793 	/* this should fail */
1794 	bpf_dynptr_clone(&ptr1, &clone);
1795 
1796 	bpf_ringbuf_submit_dynptr(&clone, 0);
1797 
1798 	return 0;
1799 }
1800 
1801 /* Invalidating a dynptr should invalidate its clones */
1802 SEC("?raw_tp")
1803 __failure __msg("Expected an initialized dynptr as R3")
1804 int clone_invalidate1(void *ctx)
1805 {
1806 	struct bpf_dynptr clone;
1807 	struct bpf_dynptr ptr;
1808 	char read_data[64];
1809 
1810 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
1811 
1812 	bpf_dynptr_clone(&ptr, &clone);
1813 
1814 	bpf_ringbuf_submit_dynptr(&ptr, 0);
1815 
1816 	/* this should fail */
1817 	bpf_dynptr_read(read_data, sizeof(read_data), &clone, 0, 0);
1818 
1819 	return 0;
1820 }
1821 
1822 /* Invalidating a dynptr should invalidate its parent */
1823 SEC("?raw_tp")
1824 __failure __msg("Expected an initialized dynptr as R3")
1825 int clone_invalidate2(void *ctx)
1826 {
1827 	struct bpf_dynptr ptr;
1828 	struct bpf_dynptr clone;
1829 	char read_data[64];
1830 
1831 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
1832 
1833 	bpf_dynptr_clone(&ptr, &clone);
1834 
1835 	bpf_ringbuf_submit_dynptr(&clone, 0);
1836 
1837 	/* this should fail */
1838 	bpf_dynptr_read(read_data, sizeof(read_data), &ptr, 0, 0);
1839 
1840 	return 0;
1841 }
1842 
1843 /* Invalidating a dynptr should invalidate its siblings */
1844 SEC("?raw_tp")
1845 __failure __msg("Expected an initialized dynptr as R3")
1846 int clone_invalidate3(void *ctx)
1847 {
1848 	struct bpf_dynptr ptr;
1849 	struct bpf_dynptr clone1;
1850 	struct bpf_dynptr clone2;
1851 	char read_data[64];
1852 
1853 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
1854 
1855 	bpf_dynptr_clone(&ptr, &clone1);
1856 
1857 	bpf_dynptr_clone(&ptr, &clone2);
1858 
1859 	bpf_ringbuf_submit_dynptr(&clone2, 0);
1860 
1861 	/* this should fail */
1862 	bpf_dynptr_read(read_data, sizeof(read_data), &clone1, 0, 0);
1863 
1864 	return 0;
1865 }
1866 
1867 /* Invalidating a dynptr should invalidate any data slices
1868  * of its clones
1869  */
1870 SEC("?raw_tp")
1871 __failure __msg("invalid mem access 'scalar'")
1872 int clone_invalidate4(void *ctx)
1873 {
1874 	struct bpf_dynptr ptr;
1875 	struct bpf_dynptr clone;
1876 	int *data;
1877 
1878 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
1879 
1880 	bpf_dynptr_clone(&ptr, &clone);
1881 	data = bpf_dynptr_data(&clone, 0, sizeof(val));
1882 	if (!data)
1883 		return 0;
1884 
1885 	bpf_ringbuf_submit_dynptr(&ptr, 0);
1886 
1887 	/* this should fail */
1888 	*data = 123;
1889 
1890 	return 0;
1891 }
1892 
1893 /* Invalidating a dynptr should invalidate any data slices
1894  * of its parent
1895  */
1896 SEC("?raw_tp")
1897 __failure __msg("invalid mem access 'scalar'")
1898 int clone_invalidate5(void *ctx)
1899 {
1900 	struct bpf_dynptr ptr;
1901 	struct bpf_dynptr clone;
1902 	int *data;
1903 
1904 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
1905 	data = bpf_dynptr_data(&ptr, 0, sizeof(val));
1906 	if (!data)
1907 		return 0;
1908 
1909 	bpf_dynptr_clone(&ptr, &clone);
1910 
1911 	bpf_ringbuf_submit_dynptr(&clone, 0);
1912 
1913 	/* this should fail */
1914 	*data = 123;
1915 
1916 	return 0;
1917 }
1918 
1919 /* Invalidating a dynptr should invalidate any data slices
1920  * of its sibling
1921  */
1922 SEC("?raw_tp")
1923 __failure __msg("invalid mem access 'scalar'")
1924 int clone_invalidate6(void *ctx)
1925 {
1926 	struct bpf_dynptr ptr;
1927 	struct bpf_dynptr clone1;
1928 	struct bpf_dynptr clone2;
1929 	int *data;
1930 
1931 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
1932 
1933 	bpf_dynptr_clone(&ptr, &clone1);
1934 
1935 	bpf_dynptr_clone(&ptr, &clone2);
1936 
1937 	data = bpf_dynptr_data(&clone1, 0, sizeof(val));
1938 	if (!data)
1939 		return 0;
1940 
1941 	bpf_ringbuf_submit_dynptr(&clone2, 0);
1942 
1943 	/* this should fail */
1944 	*data = 123;
1945 
1946 	return 0;
1947 }
1948 
1949 /* A skb clone's data slices should be invalid anytime packet data changes */
1950 SEC("?tc")
1951 __failure __msg("invalid mem access 'scalar'")
1952 int clone_skb_packet_data(struct __sk_buff *skb)
1953 {
1954 	char buffer[sizeof(__u32)] = {};
1955 	struct bpf_dynptr clone;
1956 	struct bpf_dynptr ptr;
1957 	__u32 *data;
1958 
1959 	bpf_dynptr_from_skb(skb, 0, &ptr);
1960 
1961 	bpf_dynptr_clone(&ptr, &clone);
1962 	data = bpf_dynptr_slice_rdwr(&clone, 0, buffer, sizeof(buffer));
1963 	if (!data)
1964 		return XDP_DROP;
1965 
1966 	if (bpf_skb_pull_data(skb, skb->len))
1967 		return SK_DROP;
1968 
1969 	/* this should fail */
1970 	*data = 123;
1971 
1972 	return 0;
1973 }
1974 
1975 /* A skb clone's metadata slice becomes invalid anytime packet data changes */
1976 SEC("?tc")
1977 __failure __msg("invalid mem access 'scalar'")
1978 int clone_skb_packet_meta(struct __sk_buff *skb)
1979 {
1980 	struct bpf_dynptr clone, meta;
1981 	__u8 *md;
1982 
1983 	bpf_dynptr_from_skb_meta(skb, 0, &meta);
1984 	bpf_dynptr_clone(&meta, &clone);
1985 	md = bpf_dynptr_slice_rdwr(&clone, 0, NULL, sizeof(*md));
1986 	if (!md)
1987 		return SK_DROP;
1988 
1989 	if (bpf_skb_pull_data(skb, skb->len))
1990 		return SK_DROP;
1991 
1992 	/* this should fail */
1993 	*md = 42;
1994 
1995 	return 0;
1996 }
1997 
1998 /* A xdp clone's data slices should be invalid anytime packet data changes */
1999 SEC("?xdp")
2000 __failure __msg("invalid mem access 'scalar'")
2001 int clone_xdp_packet_data(struct xdp_md *xdp)
2002 {
2003 	char buffer[sizeof(__u32)] = {};
2004 	struct bpf_dynptr clone;
2005 	struct bpf_dynptr ptr;
2006 	struct ethhdr *hdr;
2007 	__u32 *data;
2008 
2009 	bpf_dynptr_from_xdp(xdp, 0, &ptr);
2010 
2011 	bpf_dynptr_clone(&ptr, &clone);
2012 	data = bpf_dynptr_slice_rdwr(&clone, 0, buffer, sizeof(buffer));
2013 	if (!data)
2014 		return XDP_DROP;
2015 
2016 	if (bpf_xdp_adjust_head(xdp, 0 - (int)sizeof(*hdr)))
2017 		return XDP_DROP;
2018 
2019 	/* this should fail */
2020 	*data = 123;
2021 
2022 	return 0;
2023 }
2024 
2025 /* Buffers that are provided must be sufficiently long */
2026 SEC("?cgroup_skb/egress")
2027 __failure __msg("memory, len pair leads to invalid memory access")
2028 int test_dynptr_skb_small_buff(struct __sk_buff *skb)
2029 {
2030 	struct bpf_dynptr ptr;
2031 	char buffer[8] = {};
2032 	__u64 *data;
2033 
2034 	if (bpf_dynptr_from_skb(skb, 0, &ptr)) {
2035 		err = 1;
2036 		return 1;
2037 	}
2038 
2039 	/* This may return NULL. SKB may require a buffer */
2040 	data = bpf_dynptr_slice(&ptr, 0, buffer, 9);
2041 
2042 	return !!data;
2043 }
2044 
2045 __noinline long global_call_bpf_dynptr(const struct bpf_dynptr *dynptr)
2046 {
2047 	long ret = 0;
2048 	/* Avoid leaving this global function empty to avoid having the compiler
2049 	 * optimize away the call to this global function.
2050 	 */
2051 	__sink(ret);
2052 	return ret;
2053 }
2054 
2055 SEC("?raw_tp")
2056 __failure __msg("R1 expected pointer to stack or const struct bpf_dynptr")
2057 int test_dynptr_reg_type(void *ctx)
2058 {
2059 	struct task_struct *current = NULL;
2060 	/* R1 should be holding a PTR_TO_BTF_ID, so this shouldn't be a
2061 	 * reg->type that can be passed to a function accepting a
2062 	 * ARG_PTR_TO_DYNPTR | MEM_RDONLY. process_dynptr_func() should catch
2063 	 * this.
2064 	 */
2065 	global_call_bpf_dynptr((const struct bpf_dynptr *)current);
2066 	return 0;
2067 }
2068 
2069 /* Overwriting a referenced dynptr is allowed if a clone still holds the ref */
2070 SEC("?raw_tp")
2071 __success
2072 int dynptr_overwrite_ref_with_clone(void *ctx)
2073 {
2074 	struct bpf_dynptr ptr, clone;
2075 
2076 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr);
2077 
2078 	bpf_dynptr_clone(&ptr, &clone);
2079 
2080 	/* Overwrite the original - clone still holds the ref */
2081 	*(volatile __u8 *)&ptr = 0;
2082 
2083 	bpf_ringbuf_discard_dynptr(&clone, 0);
2084 
2085 	return 0;
2086 }
2087 
2088 /* Overwriting the last referenced dynptr should still be rejected */
2089 SEC("?raw_tp")
2090 __failure __msg("cannot overwrite referenced dynptr")
2091 int dynptr_overwrite_ref_last_clone(void *ctx)
2092 {
2093 	struct bpf_dynptr ptr, clone;
2094 
2095 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr);
2096 
2097 	bpf_dynptr_clone(&ptr, &clone);
2098 
2099 	/* Overwrite the original - clone still holds the ref, OK */
2100 	*(volatile __u8 *)&ptr = 0;
2101 
2102 	/* Overwrite the last holder - this should fail */
2103 	*(volatile __u8 *)&clone = 0;
2104 
2105 	return 0;
2106 }
2107 
2108 /* Overwriting a clone should be allowed if the original still holds the ref */
2109 SEC("?raw_tp")
2110 __success
2111 int dynptr_overwrite_clone_with_original(void *ctx)
2112 {
2113 	struct bpf_dynptr ptr, clone;
2114 
2115 	bpf_ringbuf_reserve_dynptr(&ringbuf, 64, 0, &ptr);
2116 
2117 	bpf_dynptr_clone(&ptr, &clone);
2118 
2119 	/* Overwrite the clone - original still holds the ref */
2120 	*(volatile __u8 *)&clone = 0;
2121 
2122 	bpf_ringbuf_discard_dynptr(&ptr, 0);
2123 
2124 	return 0;
2125 }
2126 
2127 /* Data slices from the destroyed dynptr should be invalidated */
2128 SEC("?raw_tp")
2129 __failure __msg("invalid mem access 'scalar'")
2130 int dynptr_overwrite_ref_invalidate_slice(void *ctx)
2131 {
2132 	struct bpf_dynptr ptr, clone;
2133 	int *data;
2134 
2135 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
2136 
2137 	data = bpf_dynptr_data(&ptr, 0, sizeof(val));
2138 	if (!data)
2139 		return 0;
2140 
2141 	bpf_dynptr_clone(&ptr, &clone);
2142 
2143 	/* Overwrite the original - clone holds the ref */
2144 	*(volatile __u8 *)&ptr = 0;
2145 
2146 	/* data was from the original dynptr, should be invalid now */
2147 	*data = 123;
2148 
2149 	return 0;
2150 }
2151 
2152 /*
2153  * Data slices from a dynptr clone should remain valid after
2154  * overwriting the original dynptr
2155  */
2156 SEC("?raw_tp")
2157 __success
2158 int dynptr_overwrite_ref_clone_slice_valid(void *ctx)
2159 {
2160 	struct bpf_dynptr ptr, clone;
2161 	int *data;
2162 
2163 	bpf_ringbuf_reserve_dynptr(&ringbuf, val, 0, &ptr);
2164 
2165 	bpf_dynptr_clone(&ptr, &clone);
2166 
2167 	data = bpf_dynptr_data(&clone, 0, sizeof(val));
2168 	if (!data) {
2169 		bpf_ringbuf_discard_dynptr(&clone, 0);
2170 		return 0;
2171 	}
2172 
2173 	/* Overwrite the original - clone holds the ref */
2174 	*(volatile __u8 *)&ptr = 0;
2175 
2176 	/* data is from the clone, should still be valid */
2177 	*data = 123;
2178 
2179 	bpf_ringbuf_discard_dynptr(&clone, 0);
2180 
2181 	return 0;
2182 }
2183