map_excl.c

 1// SPDX-License-Identifier: GPL-2.0
 2/* Copyright (C) 2025 Google LLC. */
 3#define _GNU_SOURCE
 4#include <unistd.h>
 5#include <sys/syscall.h>
 6#include <test_progs.h>
 7#include <bpf/btf.h>
 8
 9#include "map_excl.skel.h"
10
11static void test_map_excl_allowed(void)
12{
13	struct map_excl *skel = map_excl__open();
14	int err;
15
16	err = bpf_map__set_exclusive_program(skel->maps.excl_map, skel->progs.should_have_access);
17	if (!ASSERT_OK(err, "bpf_map__set_exclusive_program"))
18		goto out;
19
20	bpf_program__set_autoload(skel->progs.should_have_access, true);
21	bpf_program__set_autoload(skel->progs.should_not_have_access, false);
22
23	err = map_excl__load(skel);
24	ASSERT_OK(err, "map_excl__load");
25out:
26	map_excl__destroy(skel);
27}
28
29static void test_map_excl_denied(void)
30{
31	struct map_excl *skel = map_excl__open();
32	int err;
33
34	err = bpf_map__set_exclusive_program(skel->maps.excl_map, skel->progs.should_have_access);
35	if (!ASSERT_OK(err, "bpf_map__make_exclusive"))
36		goto out;
37
38	bpf_program__set_autoload(skel->progs.should_have_access, false);
39	bpf_program__set_autoload(skel->progs.should_not_have_access, true);
40
41	err = map_excl__load(skel);
42	ASSERT_EQ(err, -EACCES, "exclusive map access not denied\n");
43out:
44	map_excl__destroy(skel);
45
46}
47
48void test_map_excl(void)
49{
50	if (test__start_subtest("map_excl_allowed"))
51		test_map_excl_allowed();
52	if (test__start_subtest("map_excl_denied"))
53		test_map_excl_denied();
54}