99 lines · c
1// SPDX-License-Identifier: GPL-2.02/* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */3 4#define _GNU_SOURCE5#include <sys/wait.h>6#include <test_progs.h>7#include <unistd.h>8 9#include "task_kfunc_failure.skel.h"10#include "task_kfunc_success.skel.h"11 12static struct task_kfunc_success *open_load_task_kfunc_skel(void)13{14 struct task_kfunc_success *skel;15 int err;16 17 skel = task_kfunc_success__open();18 if (!ASSERT_OK_PTR(skel, "skel_open"))19 return NULL;20 21 skel->bss->pid = getpid();22 23 err = task_kfunc_success__load(skel);24 if (!ASSERT_OK(err, "skel_load"))25 goto cleanup;26 27 return skel;28 29cleanup:30 task_kfunc_success__destroy(skel);31 return NULL;32}33 34static void run_success_test(const char *prog_name)35{36 struct task_kfunc_success *skel;37 int status;38 pid_t child_pid;39 struct bpf_program *prog;40 struct bpf_link *link = NULL;41 42 skel = open_load_task_kfunc_skel();43 if (!ASSERT_OK_PTR(skel, "open_load_skel"))44 return;45 46 if (!ASSERT_OK(skel->bss->err, "pre_spawn_err"))47 goto cleanup;48 49 prog = bpf_object__find_program_by_name(skel->obj, prog_name);50 if (!ASSERT_OK_PTR(prog, "bpf_object__find_program_by_name"))51 goto cleanup;52 53 link = bpf_program__attach(prog);54 if (!ASSERT_OK_PTR(link, "attached_link"))55 goto cleanup;56 57 child_pid = fork();58 if (!ASSERT_GT(child_pid, -1, "child_pid"))59 goto cleanup;60 if (child_pid == 0)61 _exit(0);62 waitpid(child_pid, &status, 0);63 64 ASSERT_OK(skel->bss->err, "post_wait_err");65 66cleanup:67 bpf_link__destroy(link);68 task_kfunc_success__destroy(skel);69}70 71static const char * const success_tests[] = {72 "test_task_acquire_release_argument",73 "test_task_acquire_release_current",74 "test_task_acquire_leave_in_map",75 "test_task_xchg_release",76 "test_task_map_acquire_release",77 "test_task_current_acquire_release",78 "test_task_from_pid_arg",79 "test_task_from_pid_current",80 "test_task_from_pid_invalid",81 "task_kfunc_acquire_trusted_walked",82 "test_task_kfunc_flavor_relo",83 "test_task_kfunc_flavor_relo_not_found",84};85 86void test_task_kfunc(void)87{88 int i;89 90 for (i = 0; i < ARRAY_SIZE(success_tests); i++) {91 if (!test__start_subtest(success_tests[i]))92 continue;93 94 run_success_test(success_tests[i]);95 }96 97 RUN_TESTS(task_kfunc_failure);98}99