mirror of
https://github.com/torvalds/linux.git
synced 2026-10-15 20:30: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>
68 lines
1.7 KiB
C
68 lines
1.7 KiB
C
// SPDX-License-Identifier: GPL-2.0-only
|
|
/* Copyright (c) 2026 Christian Brauner */
|
|
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <sys/xattr.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <test_progs.h>
|
|
|
|
#include "sock_read_xattr.skel.h"
|
|
|
|
static const char xattr_value[] = "bpf_sock_value";
|
|
static const char xattr_name[] = "user.bpf_test";
|
|
|
|
static void test_read_sock_xattr(void)
|
|
{
|
|
struct sockaddr_in addr = {};
|
|
struct sock_read_xattr *skel = NULL;
|
|
struct bpf_link *link = NULL;
|
|
int sock_fd = -1, err;
|
|
|
|
sock_fd = socket(AF_INET, SOCK_STREAM, 0);
|
|
if (!ASSERT_OK_FD(sock_fd, "socket"))
|
|
return;
|
|
|
|
err = fsetxattr(sock_fd, xattr_name, xattr_value, sizeof(xattr_value), 0);
|
|
if (!ASSERT_OK(err, "fsetxattr"))
|
|
goto out;
|
|
|
|
skel = sock_read_xattr__open_and_load();
|
|
if (!ASSERT_OK_PTR(skel, "sock_read_xattr__open_and_load"))
|
|
goto out;
|
|
|
|
skel->bss->monitored_pid = sys_gettid();
|
|
|
|
/* Only attach the functional program; the verifier-only programs
|
|
* above are not pid-gated and would clobber the shared globals.
|
|
*/
|
|
link = bpf_program__attach(skel->progs.read_sock_xattr);
|
|
if (!ASSERT_OK_PTR(link, "attach read_sock_xattr"))
|
|
goto out;
|
|
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = htons(1234);
|
|
addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
|
|
/* Only the lsm/socket_connect hook matters; the connect may fail. */
|
|
connect(sock_fd, (struct sockaddr *)&addr, sizeof(addr));
|
|
|
|
ASSERT_EQ(skel->data->read_ret, sizeof(xattr_value), "read_ret");
|
|
ASSERT_STREQ(skel->bss->value, xattr_value, "value");
|
|
|
|
out:
|
|
bpf_link__destroy(link);
|
|
if (sock_fd >= 0)
|
|
close(sock_fd);
|
|
sock_read_xattr__destroy(skel);
|
|
}
|
|
|
|
void test_sock_xattr(void)
|
|
{
|
|
RUN_TESTS(sock_read_xattr);
|
|
|
|
if (test__start_subtest("read_sock_xattr"))
|
|
test_read_sock_xattr();
|
|
}
|