bpf: Allow the new syncookie helpers to work with SKBs

This commit allows the new BPF helpers to work in SKB context (in TC
BPF programs): bpf_tcp_raw_{gen,check}_syncookie_ipv{4,6}.

Using these helpers in TC BPF programs is not recommended, because it's
unlikely that the BPF program will provide any substantional speedup
compared to regular SYN cookies or synproxy, after the SKB is already
created.

Signed-off-by: Maxim Mikityanskiy <maximmi@nvidia.com>
Reviewed-by: Tariq Toukan <tariqt@nvidia.com>
Link: https://lore.kernel.org/r/20220615134847.3753567-6-maximmi@nvidia.com
Signed-off-by: Alexei Starovoitov <ast@kernel.org>

diff --git a/net/core/filter.c b/net/core/filter.c
index b62d4126a5616d6683cd4d698ad5635375a05390..423f47db84c60c9e850b3e6c155c43aa4bc5d152 100644 (file)
--- a/net/core/filter.c
+++ b/net/core/filter.c
@@ -7915,6 +7915,16 @@ tc_cls_act_func_proto(enum bpf_func_id func_id, const struct bpf_prog *prog)
                return &bpf_sk_assign_proto;
        case BPF_FUNC_skb_set_tstamp:
                return &bpf_skb_set_tstamp_proto;
+#ifdef CONFIG_SYN_COOKIES
+       case BPF_FUNC_tcp_raw_gen_syncookie_ipv4:
+               return &bpf_tcp_raw_gen_syncookie_ipv4_proto;
+       case BPF_FUNC_tcp_raw_gen_syncookie_ipv6:
+               return &bpf_tcp_raw_gen_syncookie_ipv6_proto;
+       case BPF_FUNC_tcp_raw_check_syncookie_ipv4:
+               return &bpf_tcp_raw_check_syncookie_ipv4_proto;
+       case BPF_FUNC_tcp_raw_check_syncookie_ipv6:
+               return &bpf_tcp_raw_check_syncookie_ipv6_proto;
+#endif
 #endif
        default:
                return bpf_sk_base_func_proto(func_id);