mirror of
https://github.com/torvalds/linux.git
synced 2026-10-16 04:39:02 +02:00
d717b7e84f
Add a selftest that loads the kfunc in sleepable and non-sleepable lsm/socket_connect programs and checks that a value set via fsetxattr() on a socket is read back. Link: https://patch.msgid.link/20260617-work-bpf-sock-xattr-v1-2-a1276f7c9da3@kernel.org Reviewed-by: John Fastabend <john.fastabend@gmail.com> Signed-off-by: Christian Brauner (Amutable) <brauner@kernel.org>
55 lines
1.2 KiB
C
55 lines
1.2 KiB
C
// 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;
|
|
}
|