bpf/prog_tests/fentry_fexit.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 /* Copyright (c) 2019 Facebook */
0003 #include <test_progs.h>
0004 #include "fentry_test.lskel.h"
0005 #include "fexit_test.lskel.h"
0006
0007 void test_fentry_fexit(void)
0008 {
0009     struct fentry_test_lskel *fentry_skel = NULL;
0010     struct fexit_test_lskel *fexit_skel = NULL;
0011     __u64 *fentry_res, *fexit_res;
0012     int err, prog_fd, i;
0013     LIBBPF_OPTS(bpf_test_run_opts, topts);
0014
0015     fentry_skel = fentry_test_lskel__open_and_load();
0016     if (!ASSERT_OK_PTR(fentry_skel, "fentry_skel_load"))
0017         goto close_prog;
0018     fexit_skel = fexit_test_lskel__open_and_load();
0019     if (!ASSERT_OK_PTR(fexit_skel, "fexit_skel_load"))
0020         goto close_prog;
0021
0022     err = fentry_test_lskel__attach(fentry_skel);
0023     if (!ASSERT_OK(err, "fentry_attach"))
0024         goto close_prog;
0025     err = fexit_test_lskel__attach(fexit_skel);
0026     if (!ASSERT_OK(err, "fexit_attach"))
0027         goto close_prog;
0028
0029     prog_fd = fexit_skel->progs.test1.prog_fd;
0030     err = bpf_prog_test_run_opts(prog_fd, &topts);
0031     ASSERT_OK(err, "ipv6 test_run");
0032     ASSERT_OK(topts.retval, "ipv6 test retval");
0033
0034     fentry_res = (__u64 *)fentry_skel->bss;
0035     fexit_res = (__u64 *)fexit_skel->bss;
0036     printf("%lld\n", fentry_skel->bss->test1_result);
0037     for (i = 0; i < 8; i++) {
0038         ASSERT_EQ(fentry_res[i], 1, "fentry result");
0039         ASSERT_EQ(fexit_res[i], 1, "fexit result");
0040     }
0041
0042 close_prog:
0043     fentry_test_lskel__destroy(fentry_skel);
0044     fexit_test_lskel__destroy(fexit_skel);
0045 }