tools/testing/selftests/bpf/progs/fexit_test.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright (c) 2019 Facebook */
   3 #include <linux/bpf.h>
   4 #include <bpf/bpf_helpers.h>
   5 #include <bpf/bpf_tracing.h>
   6
   7 char _license[] SEC("license") = "GPL";
   8
   9 __u64 test1_result = 0;
  10 SEC("fexit/bpf_fentry_test1")
  11 int BPF_PROG(test1, int a, int ret)
  12 {
  13         test1_result = a == 1 && ret == 2;
  14         return 0;
  15 }
  16
  17 __u64 test2_result = 0;
  18 SEC("fexit/bpf_fentry_test2")
  19 int BPF_PROG(test2, int a, __u64 b, int ret)
  20 {
  21         test2_result = a == 2 && b == 3 && ret == 5;
  22         return 0;
  23 }
  24
  25 __u64 test3_result = 0;
  26 SEC("fexit/bpf_fentry_test3")
  27 int BPF_PROG(test3, char a, int b, __u64 c, int ret)
  28 {
  29         test3_result = a == 4 && b == 5 && c == 6 && ret == 15;
  30         return 0;
  31 }
  32
  33 __u64 test4_result = 0;
  34 SEC("fexit/bpf_fentry_test4")
  35 int BPF_PROG(test4, void *a, char b, int c, __u64 d, int ret)
  36 {
  37         test4_result = a == (void *)7 && b == 8 && c == 9 && d == 10 &&
  38                 ret == 34;
  39         return 0;
  40 }
  41
  42 __u64 test5_result = 0;
  43 SEC("fexit/bpf_fentry_test5")
  44 int BPF_PROG(test5, __u64 a, void *b, short c, int d, __u64 e, int ret)
  45 {
  46         test5_result = a == 11 && b == (void *)12 && c == 13 && d == 14 &&
  47                 e == 15 && ret == 65;
  48         return 0;
  49 }
  50
  51 __u64 test6_result = 0;
  52 SEC("fexit/bpf_fentry_test6")
  53 int BPF_PROG(test6, __u64 a, void *b, short c, int d, void *e, __u64 f, int ret)
  54 {
  55         test6_result = a == 16 && b == (void *)17 && c == 18 && d == 19 &&
  56                 e == (void *)20 && f == 21 && ret == 111;
  57         return 0;
  58 }
  59
  60 struct bpf_fentry_test_t {
  61         struct bpf_fentry_test *a;
  62 };
  63
  64 __u64 test7_result = 0;
  65 SEC("fexit/bpf_fentry_test7")
  66 int BPF_PROG(test7, struct bpf_fentry_test_t *arg)
  67 {
  68         if (arg == 0)
  69                 test7_result = 1;
  70         return 0;
  71 }
  72
  73 __u64 test8_result = 0;
  74 SEC("fexit/bpf_fentry_test8")
  75 int BPF_PROG(test8, struct bpf_fentry_test_t *arg)
  76 {
  77         if (arg->a == 0)
  78                 test8_result = 1;
  79         return 0;
  80 }