1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2019 Facebook */ 3 #include <test_progs.h> 4 #include "test_pkt_access.skel.h" 5 #include "fentry_test.skel.h" 6 #include "fexit_test.skel.h" 7 8 BPF_EMBED_OBJ(pkt_access, "test_pkt_access.o"); 9 BPF_EMBED_OBJ(fentry, "fentry_test.o"); 10 BPF_EMBED_OBJ(fexit, "fexit_test.o"); 11 12 void test_fentry_fexit(void) 13 { 14 struct test_pkt_access *pkt_skel = NULL; 15 struct fentry_test *fentry_skel = NULL; 16 struct fexit_test *fexit_skel = NULL; 17 __u64 *fentry_res, *fexit_res; 18 __u32 duration = 0, retval; 19 int err, pkt_fd, i; 20 21 pkt_skel = test_pkt_access__open_and_load(&pkt_access_embed); 22 if (CHECK(!pkt_skel, "pkt_skel_load", "pkt_access skeleton failed\n")) 23 return; 24 fentry_skel = fentry_test__open_and_load(&fentry_embed); 25 if (CHECK(!fentry_skel, "fentry_skel_load", "fentry skeleton failed\n")) 26 goto close_prog; 27 fexit_skel = fexit_test__open_and_load(&fexit_embed); 28 if (CHECK(!fexit_skel, "fexit_skel_load", "fexit skeleton failed\n")) 29 goto close_prog; 30 31 err = fentry_test__attach(fentry_skel); 32 if (CHECK(err, "fentry_attach", "fentry attach failed: %d\n", err)) 33 goto close_prog; 34 err = fexit_test__attach(fexit_skel); 35 if (CHECK(err, "fexit_attach", "fexit attach failed: %d\n", err)) 36 goto close_prog; 37 38 pkt_fd = bpf_program__fd(pkt_skel->progs.test_pkt_access); 39 err = bpf_prog_test_run(pkt_fd, 1, &pkt_v6, sizeof(pkt_v6), 40 NULL, NULL, &retval, &duration); 41 CHECK(err || retval, "ipv6", 42 "err %d errno %d retval %d duration %d\n", 43 err, errno, retval, duration); 44 45 fentry_res = (__u64 *)fentry_skel->bss; 46 fexit_res = (__u64 *)fexit_skel->bss; 47 printf("%lld\n", fentry_skel->bss->test1_result); 48 for (i = 0; i < 6; i++) { 49 CHECK(fentry_res[i] != 1, "result", 50 "fentry_test%d failed err %lld\n", i + 1, fentry_res[i]); 51 CHECK(fexit_res[i] != 1, "result", 52 "fexit_test%d failed err %lld\n", i + 1, fexit_res[i]); 53 } 54 55 close_prog: 56 test_pkt_access__destroy(pkt_skel); 57 fentry_test__destroy(fentry_skel); 58 fexit_test__destroy(fexit_skel); 59 } 60