tools/testing/selftests/bpf/prog_tests/rcu_read_lock.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates.*/
   3
   4 #define _GNU_SOURCE
   5 #include <unistd.h>
   6 #include <sys/syscall.h>
   7 #include <sys/types.h>
   8 #include <test_progs.h>
   9 #include <bpf/btf.h>
  10 #include "rcu_read_lock.skel.h"
  11 #include "cgroup_helpers.h"
  12
  13 static unsigned long long cgroup_id;
  14
  15 static void test_success(void)
  16 {
  17         struct rcu_read_lock *skel;
  18         int err;
  19
  20         skel = rcu_read_lock__open();
  21         if (!ASSERT_OK_PTR(skel, "skel_open"))
  22                 return;
  23
  24         skel->bss->target_pid = sys_gettid();
  25
  26         bpf_program__set_autoload(skel->progs.get_cgroup_id, true);
  27         bpf_program__set_autoload(skel->progs.task_succ, true);
  28         bpf_program__set_autoload(skel->progs.two_regions, true);
  29         bpf_program__set_autoload(skel->progs.non_sleepable_1, true);
  30         bpf_program__set_autoload(skel->progs.non_sleepable_2, true);
  31         bpf_program__set_autoload(skel->progs.task_trusted_non_rcuptr, true);
  32         bpf_program__set_autoload(skel->progs.rcu_read_lock_subprog, true);
  33         bpf_program__set_autoload(skel->progs.rcu_read_lock_global_subprog, true);
  34         bpf_program__set_autoload(skel->progs.rcu_read_lock_subprog_lock, true);
  35         bpf_program__set_autoload(skel->progs.rcu_read_lock_subprog_unlock, true);
  36         err = rcu_read_lock__load(skel);
  37         if (!ASSERT_OK(err, "skel_load"))
  38                 goto out;
  39
  40         err = rcu_read_lock__attach(skel);
  41         if (!ASSERT_OK(err, "skel_attach"))
  42                 goto out;
  43
  44         syscall(SYS_getpgid);
  45
  46         ASSERT_EQ(skel->bss->task_storage_val, 2, "task_storage_val");
  47         ASSERT_EQ(skel->bss->cgroup_id, cgroup_id, "cgroup_id");
  48 out:
  49         rcu_read_lock__destroy(skel);
  50 }
  51
  52 static void test_rcuptr_acquire(void)
  53 {
  54         struct rcu_read_lock *skel;
  55         int err;
  56
  57         skel = rcu_read_lock__open();
  58         if (!ASSERT_OK_PTR(skel, "skel_open"))
  59                 return;
  60
  61         skel->bss->target_pid = sys_gettid();
  62
  63         bpf_program__set_autoload(skel->progs.task_acquire, true);
  64         err = rcu_read_lock__load(skel);
  65         if (!ASSERT_OK(err, "skel_load"))
  66                 goto out;
  67
  68         err = rcu_read_lock__attach(skel);
  69         ASSERT_OK(err, "skel_attach");
  70 out:
  71         rcu_read_lock__destroy(skel);
  72 }
  73
  74 static const char * const inproper_region_tests[] = {
  75         "miss_lock",
  76         "no_lock",
  77         "miss_unlock",
  78         "non_sleepable_rcu_mismatch",
  79         "inproper_sleepable_helper",
  80         "inproper_sleepable_kfunc",
  81         "nested_rcu_region",
  82         "rcu_read_lock_global_subprog_lock",
  83         "rcu_read_lock_global_subprog_unlock",
  84 };
  85
  86 static void test_inproper_region(void)
  87 {
  88         struct rcu_read_lock *skel;
  89         struct bpf_program *prog;
  90         int i, err;
  91
  92         for (i = 0; i < ARRAY_SIZE(inproper_region_tests); i++) {
  93                 skel = rcu_read_lock__open();
  94                 if (!ASSERT_OK_PTR(skel, "skel_open"))
  95                         return;
  96
  97                 prog = bpf_object__find_program_by_name(skel->obj, inproper_region_tests[i]);
  98                 if (!ASSERT_OK_PTR(prog, "bpf_object__find_program_by_name"))
  99                         goto out;
 100                 bpf_program__set_autoload(prog, true);
 101                 err = rcu_read_lock__load(skel);
 102                 ASSERT_ERR(err, "skel_load");
 103 out:
 104                 rcu_read_lock__destroy(skel);
 105         }
 106 }
 107
 108 static const char * const rcuptr_misuse_tests[] = {
 109         "task_untrusted_rcuptr",
 110         "cross_rcu_region",
 111 };
 112
 113 static void test_rcuptr_misuse(void)
 114 {
 115         struct rcu_read_lock *skel;
 116         struct bpf_program *prog;
 117         int i, err;
 118
 119         for (i = 0; i < ARRAY_SIZE(rcuptr_misuse_tests); i++) {
 120                 skel = rcu_read_lock__open();
 121                 if (!ASSERT_OK_PTR(skel, "skel_open"))
 122                         return;
 123
 124                 prog = bpf_object__find_program_by_name(skel->obj, rcuptr_misuse_tests[i]);
 125                 if (!ASSERT_OK_PTR(prog, "bpf_object__find_program_by_name"))
 126                         goto out;
 127                 bpf_program__set_autoload(prog, true);
 128                 err = rcu_read_lock__load(skel);
 129                 ASSERT_ERR(err, "skel_load");
 130 out:
 131                 rcu_read_lock__destroy(skel);
 132         }
 133 }
 134
 135 void test_rcu_read_lock(void)
 136 {
 137         int cgroup_fd;
 138
 139         cgroup_fd = test__join_cgroup("/rcu_read_lock");
 140         if (!ASSERT_GE(cgroup_fd, 0, "join_cgroup /rcu_read_lock"))
 141                 goto out;
 142
 143         cgroup_id = get_cgroup_id("/rcu_read_lock");
 144         if (test__start_subtest("success"))
 145                 test_success();
 146         if (test__start_subtest("rcuptr_acquire"))
 147                 test_rcuptr_acquire();
 148         if (test__start_subtest("negative_tests_inproper_region"))
 149                 test_inproper_region();
 150         if (test__start_subtest("negative_tests_rcuptr_misuse"))
 151                 test_rcuptr_misuse();
 152         close(cgroup_fd);
 153 out:;
 154 }