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_ciint selector = 0;
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci#define TAIL_FUNC(x)				\
1662306a36Sopenharmony_ci	SEC("tc")				\
1762306a36Sopenharmony_ci	int classifier_##x(struct __sk_buff *skb)	\
1862306a36Sopenharmony_ci	{					\
1962306a36Sopenharmony_ci		return x;			\
2062306a36Sopenharmony_ci	}
2162306a36Sopenharmony_ciTAIL_FUNC(0)
2262306a36Sopenharmony_ciTAIL_FUNC(1)
2362306a36Sopenharmony_ciTAIL_FUNC(2)
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ciSEC("tc")
2662306a36Sopenharmony_ciint entry(struct __sk_buff *skb)
2762306a36Sopenharmony_ci{
2862306a36Sopenharmony_ci	int idx = 0;
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ci	if (selector == 1234)
3162306a36Sopenharmony_ci		idx = 1;
3262306a36Sopenharmony_ci	else if (selector == 5678)
3362306a36Sopenharmony_ci		idx = 2;
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_ci	bpf_tail_call(skb, &jmp_table, idx);
3662306a36Sopenharmony_ci	return 3;
3762306a36Sopenharmony_ci}
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_cichar __license[] SEC("license") = "GPL";
40