summaryrefslogtreecommitdiff
path: root/tools/testing/selftests/bpf/progs/sock_read_xattr.c
blob: c4a8eae8cc3c9f7432cf72a2f3101e6c5282f174 (plain)
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
// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2026 Christian Brauner */

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

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

char value[16];
int read_ret = -1;
__u32 monitored_pid = 0;

static __always_inline void read_xattr(struct socket *sock)
{
	struct bpf_dynptr value_ptr;

	bpf_dynptr_from_mem(value, sizeof(value), 0, &value_ptr);
	bpf_sock_read_xattr(sock, "user.bpf_test", &value_ptr);
}

SEC("lsm.s/socket_connect")
__success
int BPF_PROG(trusted_sock_ptr_sleepable, struct socket *sock)
{
	read_xattr(sock);
	return 0;
}

SEC("lsm/socket_connect")
__success
int BPF_PROG(trusted_sock_ptr_non_sleepable, struct socket *sock)
{
	read_xattr(sock);
	return 0;
}

SEC("lsm.s/socket_connect")
__success
int BPF_PROG(read_sock_xattr, struct socket *sock)
{
	struct bpf_dynptr value_ptr;
	__u32 pid = bpf_get_current_pid_tgid() >> 32;

	if (pid != monitored_pid)
		return 0;

	bpf_dynptr_from_mem(value, sizeof(value), 0, &value_ptr);
	read_ret = bpf_sock_read_xattr(sock, "user.bpf_test", &value_ptr);
	return 0;
}