aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tools/testing/selftests/bpf/progs/verifier_bitfield_write.c
blob: 623f130a31989159b47d343221d013a45842270e (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
// SPDX-License-Identifier: GPL-2.0

#include <linux/bpf.h>
#include <stdint.h>

#include <bpf/bpf_helpers.h>
#include <bpf/bpf_core_read.h>

#include "bpf_misc.h"

struct core_reloc_bitfields {
	/* unsigned bitfields */
	uint8_t		ub1: 1;
	uint8_t		ub2: 2;
	uint32_t	ub7: 7;
	/* signed bitfields */
	int8_t		sb4: 4;
	int32_t		sb20: 20;
	/* non-bitfields */
	uint32_t	u32;
	int32_t		s32;
} __attribute__((preserve_access_index));

SEC("tc")
__description("single CO-RE bitfield roundtrip")
__btf_path("btf__core_reloc_bitfields.bpf.o")
__success
__retval(3)
int single_field_roundtrip(struct __sk_buff *ctx)
{
	struct core_reloc_bitfields bitfields;

	__builtin_memset(&bitfields, 0, sizeof(bitfields));
	BPF_CORE_WRITE_BITFIELD(&bitfields, ub2, 3);
	return BPF_CORE_READ_BITFIELD(&bitfields, ub2);
}

SEC("tc")
__description("multiple CO-RE bitfield roundtrip")
__btf_path("btf__core_reloc_bitfields.bpf.o")
__success
__retval(0x3FD)
int multiple_field_roundtrip(struct __sk_buff *ctx)
{
	struct core_reloc_bitfields bitfields;
	uint8_t ub2;
	int8_t sb4;

	__builtin_memset(&bitfields, 0, sizeof(bitfields));
	BPF_CORE_WRITE_BITFIELD(&bitfields, ub2, 1);
	BPF_CORE_WRITE_BITFIELD(&bitfields, sb4, -1);

	ub2 = BPF_CORE_READ_BITFIELD(&bitfields, ub2);
	sb4 = BPF_CORE_READ_BITFIELD(&bitfields, sb4);

	return (((uint8_t)sb4) << 2) | ub2;
}

SEC("tc")
__description("adjacent CO-RE bitfield roundtrip")
__btf_path("btf__core_reloc_bitfields.bpf.o")
__success
__retval(7)
int adjacent_field_roundtrip(struct __sk_buff *ctx)
{
	struct core_reloc_bitfields bitfields;
	uint8_t ub1, ub2;

	__builtin_memset(&bitfields, 0, sizeof(bitfields));
	BPF_CORE_WRITE_BITFIELD(&bitfields, ub1, 1);
	BPF_CORE_WRITE_BITFIELD(&bitfields, ub2, 3);

	ub1 = BPF_CORE_READ_BITFIELD(&bitfields, ub1);
	ub2 = BPF_CORE_READ_BITFIELD(&bitfields, ub2);

	return (ub2 << 1) | ub1;
}

SEC("tc")
__description("multibyte CO-RE bitfield roundtrip")
__btf_path("btf__core_reloc_bitfields.bpf.o")
__success
__retval(0x21)
int multibyte_field_roundtrip(struct __sk_buff *ctx)
{
	struct core_reloc_bitfields bitfields;
	uint32_t ub7;
	uint8_t ub1;

	__builtin_memset(&bitfields, 0, sizeof(bitfields));
	BPF_CORE_WRITE_BITFIELD(&bitfields, ub1, 1);
	BPF_CORE_WRITE_BITFIELD(&bitfields, ub7, 16);

	ub1 = BPF_CORE_READ_BITFIELD(&bitfields, ub1);
	ub7 = BPF_CORE_READ_BITFIELD(&bitfields, ub7);

	return (ub7 << 1) | ub1;
}

char _license[] SEC("license") = "GPL";