1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2021 Facebook */ 3 #include <linux/bpf.h> 4 #include <bpf/bpf_helpers.h> 5 #include "bpf_tcp_helpers.h" 6 7 extern const int bpf_prog_active __ksym; 8 extern __u64 bpf_kfunc_call_test1(struct sock *sk, __u32 a, __u64 b, 9 __u32 c, __u64 d) __ksym; 10 extern struct sock *bpf_kfunc_call_test3(struct sock *sk) __ksym; 11 int active_res = -1; 12 int sk_state_res = -1; 13 14 int __noinline f1(struct __sk_buff *skb) 15 { 16 struct bpf_sock *sk = skb->sk; 17 int *active; 18 19 if (!sk) 20 return -1; 21 22 sk = bpf_sk_fullsock(sk); 23 if (!sk) 24 return -1; 25 26 active = (int *)bpf_per_cpu_ptr(&bpf_prog_active, 27 bpf_get_smp_processor_id()); 28 if (active) 29 active_res = *active; 30 31 sk_state_res = bpf_kfunc_call_test3((struct sock *)sk)->sk_state; 32 33 return (__u32)bpf_kfunc_call_test1((struct sock *)sk, 1, 2, 3, 4); 34 } 35 36 SEC("tc") 37 int kfunc_call_test1(struct __sk_buff *skb) 38 { 39 return f1(skb); 40 } 41 42 char _license[] SEC("license") = "GPL"; 43