File: security_bpf_map.c

package info (click to toggle)
linux 6.17.9-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,734,892 kB
  • sloc: ansic: 26,684,085; asm: 271,195; sh: 147,401; python: 75,980; makefile: 57,306; perl: 36,943; xml: 19,562; cpp: 5,899; yacc: 4,909; lex: 2,943; awk: 1,556; sed: 29; ruby: 25
file content (69 lines) | stat: -rw-r--r-- 1,397 bytes parent folder | download | duplicates (9)
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
// SPDX-License-Identifier: GPL-2.0-only

#include "vmlinux.h"
#include <bpf/bpf_tracing.h>
#include <bpf/bpf_helpers.h>

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

#define EPERM 1 /* Operation not permitted */

/* From include/linux/mm.h. */
#define FMODE_WRITE	0x2

struct map;

struct {
	__uint(type, BPF_MAP_TYPE_ARRAY);
	__type(key, __u32);
	__type(value, __u32);
	__uint(max_entries, 1);
} prot_status_map SEC(".maps");

struct {
	__uint(type, BPF_MAP_TYPE_HASH);
	__type(key, __u32);
	__type(value, __u32);
	__uint(max_entries, 3);
} prot_map SEC(".maps");

struct {
	__uint(type, BPF_MAP_TYPE_HASH);
	__type(key, __u32);
	__type(value, __u32);
	__uint(max_entries, 3);
} not_prot_map SEC(".maps");

SEC("fmod_ret/security_bpf_map")
int BPF_PROG(fmod_bpf_map, struct bpf_map *map, int fmode)
{
	__u32 key = 0;
	__u32 *status_ptr = bpf_map_lookup_elem(&prot_status_map, &key);

	if (!status_ptr || !*status_ptr)
		return 0;

	if (map == &prot_map) {
		/* Allow read-only access */
		if (fmode & FMODE_WRITE)
			return -EPERM;
	}

	return 0;
}

/*
 * This program keeps references to maps. This is needed to prevent
 * optimizing them out.
 */
SEC("fentry/bpf_fentry_test1")
int BPF_PROG(fentry_dummy1, int a)
{
	__u32 key = 0;
	__u32 val1 = a;
	__u32 val2 = a + 1;

	bpf_map_update_elem(&prot_map, &key, &val1, BPF_ANY);
	bpf_map_update_elem(&not_prot_map, &key, &val2, BPF_ANY);
	return 0;
}