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

   1 // SPDX-License-Identifier: GPL-2.0
   2 #include <linux/bpf.h>
   3 #include <bpf/bpf_helpers.h>
   4 #include "bpf_legacy.h"
   5 #include "bpf_misc.h"
   6
   7 struct {
   8         __uint(type, BPF_MAP_TYPE_PROG_ARRAY);
   9         __uint(max_entries, 2);
  10         __uint(key_size, sizeof(__u32));
  11         __uint(value_size, sizeof(__u32));
  12 } jmp_table SEC(".maps");
  13
  14 __noinline
  15 int subprog_tail2(struct __sk_buff *skb)
  16 {
  17         volatile char arr[64] = {};
  18
  19         if (load_word(skb, 0) || load_half(skb, 0))
  20                 bpf_tail_call_static(skb, &jmp_table, 10);
  21         else
  22                 bpf_tail_call_static(skb, &jmp_table, 1);
  23
  24         __sink(arr[sizeof(arr) - 1]);
  25
  26         return skb->len;
  27 }
  28
  29 static __noinline
  30 int subprog_tail(struct __sk_buff *skb)
  31 {
  32         volatile char arr[64] = {};
  33
  34         bpf_tail_call_static(skb, &jmp_table, 0);
  35
  36         __sink(arr[sizeof(arr) - 1]);
  37
  38         return skb->len * 2;
  39 }
  40
  41 SEC("tc")
  42 int classifier_0(struct __sk_buff *skb)
  43 {
  44         volatile char arr[128] = {};
  45
  46         __sink(arr[sizeof(arr) - 1]);
  47
  48         return subprog_tail2(skb);
  49 }
  50
  51 SEC("tc")
  52 int classifier_1(struct __sk_buff *skb)
  53 {
  54         volatile char arr[128] = {};
  55
  56         __sink(arr[sizeof(arr) - 1]);
  57
  58         return skb->len * 3;
  59 }
  60
  61 SEC("tc")
  62 int entry(struct __sk_buff *skb)
  63 {
  64         volatile char arr[128] = {};
  65
  66         __sink(arr[sizeof(arr) - 1]);
  67
  68         return subprog_tail(skb);
  69 }
  70
  71 char __license[] SEC("license") = "GPL";