13bc253c2SGeliang Tang // SPDX-License-Identifier: GPL-2.0 23bc253c2SGeliang Tang /* Multipath TCP 33bc253c2SGeliang Tang * 43bc253c2SGeliang Tang * Copyright (c) 2020, Tessares SA. 53bc253c2SGeliang Tang * Copyright (c) 2022, SUSE. 63bc253c2SGeliang Tang * 73bc253c2SGeliang Tang * Author: Nicolas Rybowski <nicolas.rybowski@tessares.net> 83bc253c2SGeliang Tang */ 93bc253c2SGeliang Tang 103bc253c2SGeliang Tang #define pr_fmt(fmt) "MPTCP: " fmt 113bc253c2SGeliang Tang 123bc253c2SGeliang Tang #include <linux/bpf.h> 133bc253c2SGeliang Tang #include "protocol.h" 143bc253c2SGeliang Tang bpf_mptcp_sock_from_subflow(struct sock * sk)153bc253c2SGeliang Tangstruct mptcp_sock *bpf_mptcp_sock_from_subflow(struct sock *sk) 163bc253c2SGeliang Tang { 173bc253c2SGeliang Tang if (sk && sk_fullsock(sk) && sk->sk_protocol == IPPROTO_TCP && sk_is_mptcp(sk)) 183bc253c2SGeliang Tang return mptcp_sk(mptcp_subflow_ctx(sk)->conn); 193bc253c2SGeliang Tang 203bc253c2SGeliang Tang return NULL; 213bc253c2SGeliang Tang } 22*0dd061a6SGeliang Tang 23*0dd061a6SGeliang Tang BTF_SET8_START(bpf_mptcp_fmodret_ids) 24*0dd061a6SGeliang Tang BTF_ID_FLAGS(func, update_socket_protocol) 25*0dd061a6SGeliang Tang BTF_SET8_END(bpf_mptcp_fmodret_ids) 26*0dd061a6SGeliang Tang 27*0dd061a6SGeliang Tang static const struct btf_kfunc_id_set bpf_mptcp_fmodret_set = { 28*0dd061a6SGeliang Tang .owner = THIS_MODULE, 29*0dd061a6SGeliang Tang .set = &bpf_mptcp_fmodret_ids, 30*0dd061a6SGeliang Tang }; 31*0dd061a6SGeliang Tang bpf_mptcp_kfunc_init(void)32*0dd061a6SGeliang Tangstatic int __init bpf_mptcp_kfunc_init(void) 33*0dd061a6SGeliang Tang { 34*0dd061a6SGeliang Tang return register_btf_fmodret_id_set(&bpf_mptcp_fmodret_set); 35*0dd061a6SGeliang Tang } 36*0dd061a6SGeliang Tang late_initcall(bpf_mptcp_kfunc_init); 37