bpf/prog_tests/core_retro.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 // Copyright (c) 2020 Facebook
0003 #define _GNU_SOURCE
0004 #include <test_progs.h>
0005 #include "test_core_retro.skel.h"
0006
0007 void test_core_retro(void)
0008 {
0009     int err, zero = 0, res, my_pid = getpid();
0010     struct test_core_retro *skel;
0011
0012     /* load program */
0013     skel = test_core_retro__open_and_load();
0014     if (!ASSERT_OK_PTR(skel, "skel_load"))
0015         goto out_close;
0016
0017     err = bpf_map__update_elem(skel->maps.exp_tgid_map, &zero, sizeof(zero),
0018                    &my_pid, sizeof(my_pid), 0);
0019     if (!ASSERT_OK(err, "map_update"))
0020         goto out_close;
0021
0022     /* attach probe */
0023     err = test_core_retro__attach(skel);
0024     if (!ASSERT_OK(err, "attach_kprobe"))
0025         goto out_close;
0026
0027     /* trigger */
0028     usleep(1);
0029
0030     err = bpf_map__lookup_elem(skel->maps.results, &zero, sizeof(zero), &res, sizeof(res), 0);
0031     if (!ASSERT_OK(err, "map_lookup"))
0032         goto out_close;
0033
0034     ASSERT_EQ(res, my_pid, "pid_check");
0035
0036 out_close:
0037     test_core_retro__destroy(skel);
0038 }