1 // SPDX-License-Identifier: GPL-2.0 1 // SPDX-License-Identifier: GPL-2.0
2 /* Multipath TCP 2 /* Multipath TCP
3 * 3 *
4 * Copyright (c) 2020, Tessares SA. 4 * Copyright (c) 2020, Tessares SA.
5 * Copyright (c) 2022, SUSE. 5 * Copyright (c) 2022, SUSE.
6 * 6 *
7 * Author: Nicolas Rybowski <nicolas.rybowski@ 7 * Author: Nicolas Rybowski <nicolas.rybowski@tessares.net>
8 */ 8 */
9 9
10 #define pr_fmt(fmt) "MPTCP: " fmt 10 #define pr_fmt(fmt) "MPTCP: " fmt
11 11
12 #include <linux/bpf.h> 12 #include <linux/bpf.h>
13 #include "protocol.h" 13 #include "protocol.h"
14 14
15 struct mptcp_sock *bpf_mptcp_sock_from_subflow 15 struct mptcp_sock *bpf_mptcp_sock_from_subflow(struct sock *sk)
16 { 16 {
17 if (sk && sk_fullsock(sk) && sk->sk_pr 17 if (sk && sk_fullsock(sk) && sk->sk_protocol == IPPROTO_TCP && sk_is_mptcp(sk))
18 return mptcp_sk(mptcp_subflow_ 18 return mptcp_sk(mptcp_subflow_ctx(sk)->conn);
19 19
20 return NULL; 20 return NULL;
21 } 21 }
22 22
23 BTF_SET8_START(bpf_mptcp_fmodret_ids) 23 BTF_SET8_START(bpf_mptcp_fmodret_ids)
24 BTF_ID_FLAGS(func, update_socket_protocol) 24 BTF_ID_FLAGS(func, update_socket_protocol)
25 BTF_SET8_END(bpf_mptcp_fmodret_ids) 25 BTF_SET8_END(bpf_mptcp_fmodret_ids)
26 26
27 static const struct btf_kfunc_id_set bpf_mptcp 27 static const struct btf_kfunc_id_set bpf_mptcp_fmodret_set = {
28 .owner = THIS_MODULE, 28 .owner = THIS_MODULE,
29 .set = &bpf_mptcp_fmodret_ids, 29 .set = &bpf_mptcp_fmodret_ids,
30 }; 30 };
31 31
32 static int __init bpf_mptcp_kfunc_init(void) 32 static int __init bpf_mptcp_kfunc_init(void)
33 { 33 {
34 return register_btf_fmodret_id_set(&bp 34 return register_btf_fmodret_id_set(&bpf_mptcp_fmodret_set);
35 } 35 }
36 late_initcall(bpf_mptcp_kfunc_init); 36 late_initcall(bpf_mptcp_kfunc_init);
37 37
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.