162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0 262306a36Sopenharmony_ci#include <linux/bpf.h> 362306a36Sopenharmony_ci 462306a36Sopenharmony_ci#include <bpf/bpf_helpers.h> 562306a36Sopenharmony_ci 662306a36Sopenharmony_cistruct { 762306a36Sopenharmony_ci __uint(type, BPF_MAP_TYPE_PROG_ARRAY); 862306a36Sopenharmony_ci __uint(max_entries, 3); 962306a36Sopenharmony_ci __uint(key_size, sizeof(__u32)); 1062306a36Sopenharmony_ci __uint(value_size, sizeof(__u32)); 1162306a36Sopenharmony_ci} jmp_table SEC(".maps"); 1262306a36Sopenharmony_ci 1362306a36Sopenharmony_ci#define TAIL_FUNC(x) \ 1462306a36Sopenharmony_ci SEC("tc") \ 1562306a36Sopenharmony_ci int classifier_##x(struct __sk_buff *skb) \ 1662306a36Sopenharmony_ci { \ 1762306a36Sopenharmony_ci return x; \ 1862306a36Sopenharmony_ci } 1962306a36Sopenharmony_ciTAIL_FUNC(0) 2062306a36Sopenharmony_ciTAIL_FUNC(1) 2162306a36Sopenharmony_ciTAIL_FUNC(2) 2262306a36Sopenharmony_ci 2362306a36Sopenharmony_ciSEC("tc") 2462306a36Sopenharmony_ciint entry(struct __sk_buff *skb) 2562306a36Sopenharmony_ci{ 2662306a36Sopenharmony_ci /* Multiple locations to make sure we patch 2762306a36Sopenharmony_ci * all of them. 2862306a36Sopenharmony_ci */ 2962306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 0); 3062306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 0); 3162306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 0); 3262306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 0); 3362306a36Sopenharmony_ci 3462306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 1); 3562306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 1); 3662306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 1); 3762306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 1); 3862306a36Sopenharmony_ci 3962306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 2); 4062306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 2); 4162306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 2); 4262306a36Sopenharmony_ci bpf_tail_call_static(skb, &jmp_table, 2); 4362306a36Sopenharmony_ci 4462306a36Sopenharmony_ci return 3; 4562306a36Sopenharmony_ci} 4662306a36Sopenharmony_ci 4762306a36Sopenharmony_cichar __license[] SEC("license") = "GPL"; 48