mptcp: add scheduled in mptcp_subflow_context

This patch adds a new member scheduled in struct mptcp_subflow_context,
which will be set in the MPTCP scheduler context when the scheduler
picks this subflow to send data.

Add a new helper mptcp_subflow_set_scheduled() to set this flag using
WRITE_ONCE().

Reviewed-by: Mat Martineau <martineau@kernel.org>
Signed-off-by: Geliang Tang <geliang.tang@suse.com>
Signed-off-by: Mat Martineau <martineau@kernel.org>
Link: https://lore.kernel.org/r/20230821-upstream-net-next-20230818-v1-6-0c860fb256a8@kernel.org
Signed-off-by: Jakub Kicinski <kuba@kernel.org>

diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h
index 548c302a757ee62caf55697dddaef926f955ce93..e7523a40132f57431bf85949edbe55a21709c4b6 100644 (file)
--- a/net/mptcp/protocol.h
+++ b/net/mptcp/protocol.h
@@ -491,6 +491,7 @@ struct mptcp_subflow_context {
                is_mptfo : 1,       /* subflow is doing TFO */
                __unused : 9;
        enum mptcp_data_avail data_avail;
+       bool    scheduled;
        u32     remote_nonce;
        u64     thmac;
        u32     local_nonce;
@@ -663,6 +664,8 @@ void mptcp_unregister_scheduler(struct mptcp_sched_ops *sched);
 int mptcp_init_sched(struct mptcp_sock *msk,
                     struct mptcp_sched_ops *sched);
 void mptcp_release_sched(struct mptcp_sock *msk);
+void mptcp_subflow_set_scheduled(struct mptcp_subflow_context *subflow,
+                                bool scheduled);
 
 static inline bool __tcp_can_send(const struct sock *ssk)
 {

diff --git a/net/mptcp/sched.c b/net/mptcp/sched.c
index 53773668b5ee0b4be37dec8260883407b0af4975..d295b92a57899bcad273c98657c63e72977fffbe 100644 (file)
--- a/net/mptcp/sched.c
+++ b/net/mptcp/sched.c
@@ -87,3 +87,9 @@ void mptcp_release_sched(struct mptcp_sock *msk)
 
        bpf_module_put(sched, sched->owner);
 }
+
+void mptcp_subflow_set_scheduled(struct mptcp_subflow_context *subflow,
+                                bool scheduled)
+{
+       WRITE_ONCE(subflow->scheduled, scheduled);
+}