From nobody Wed May 14 16:21:29 2025 Received: from smtp.kernel.org (aws-us-west-2-korg-mail-1.web.codeaurora.org [10.30.226.201]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id 1377726AC3 for <mptcp@lists.linux.dev>; Fri, 28 Mar 2025 04:01:44 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=10.30.226.201 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1743134505; cv=none; b=kLuJTBdyZmXKHgCmkBpJyAZwha8OMHqi2f0hHN8zc6BY1LwTLkEHTHNbWIyiaSmc60t7UQzyVBLw9zVQTfGk/w8Nl811uzXJPH5y8qZloeVEdK38nTJHrPQ10xFd7nHqysabU/ilZCqyjaNKhcv7j+I9HTAY7VVTV5nZW7GiQBY= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1743134505; c=relaxed/simple; bh=RiBbyTjYRJEaIcJbhWn7VLTL4x5wpbkqXseWb7Fg16E=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=mtF16MvPEyQ1gc4ZClVwBMhMlFEnKyFRQ2pm/kX1SsapQdCtmpHyKldc1W3q9g+AyQKImjzuE7EJ6vE0XScNNSjAzBt1bKOVq+h2NmkSVo1YuuhhO3DNmC5NWkclStvEyqy8Laofhi1jCJCbr1vhUEIUWxsXenKyJ7JMqir0cGE= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dkim=pass (2048-bit key) header.d=kernel.org header.i=@kernel.org header.b=qFhBN2lO; arc=none smtp.client-ip=10.30.226.201 Authentication-Results: smtp.subspace.kernel.org; dkim=pass (2048-bit key) header.d=kernel.org header.i=@kernel.org header.b="qFhBN2lO" Received: by smtp.kernel.org (Postfix) with ESMTPSA id 5AC58C4CEE4; Fri, 28 Mar 2025 04:01:43 +0000 (UTC) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=kernel.org; s=k20201202; t=1743134504; bh=RiBbyTjYRJEaIcJbhWn7VLTL4x5wpbkqXseWb7Fg16E=; h=From:To:Cc:Subject:Date:In-Reply-To:References:From; b=qFhBN2lOmBhcq/7592wQcdt2HFiH6MWeyuyEi3EsNAJgSHea6bwiKFHQEZf+bbaAh BHO+hNh82JSFXGcuQ6u8kQorZTNBcZ3Nqb5YXa1+Olh5rO+5m6pz00DuB5e/qB64qU MZSb8vQRKATShMHbp0R0Xa2zLD33ogFxRXT6B3vE1caeFETw8qzMYHYzAESE8J/jeE /gQsIFK0y/yH0Sta9peFYBnN6ShYdqqto5RWoWM9DZYh7lPhyHFmvD40/UCZGLmQ7n SnMHX8IbMPJBEbDWgo/gce5aDvIhF7DC9aZc7M7pkwjyXWvqaeLMZJvFl1UN4E86j7 GUzdPsFOBxEnw== From: Geliang Tang <geliang@kernel.org> To: mptcp@lists.linux.dev Cc: Geliang Tang <tanggeliang@kylinos.cn> Subject: [PATCH mptcp-next v6 2/5] mptcp: pm: add accept_new_subflow() interface Date: Fri, 28 Mar 2025 12:01:32 +0800 Message-ID: <2c5a2dc15229df637c5523fd63b2b5dd88619615.1743133948.git.tanggeliang@kylinos.cn> X-Mailer: git-send-email 2.43.0 In-Reply-To: <cover.1743133948.git.tanggeliang@kylinos.cn> References: <cover.1743133948.git.tanggeliang@kylinos.cn> Precedence: bulk X-Mailing-List: mptcp@lists.linux.dev List-Id: <mptcp.lists.linux.dev> List-Subscribe: <mailto:mptcp+subscribe@lists.linux.dev> List-Unsubscribe: <mailto:mptcp+unsubscribe@lists.linux.dev> MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable Content-Type: text/plain; charset="utf-8" From: Geliang Tang <tanggeliang@kylinos.cn> The helper mptcp_pm_is_userspace() is used to distinguish userspace PM operations from in-kernel PM in mptcp_can_accept_new_subflow(). It seems reasonable to add a mandatory .accept_new_subflow interface for struct mptcp_pm_ops. The boolean type 'allow' parameter is used to control whether to allow new subflow unconditionally. A new helper mptcp_pm_accept_new_subflow() is added in pm.c to ensure that pm->ops is only used in pm.c. This interface of the in-kernel PM is implemented as a wrapper of READ_ONCE(pm->accept_subflow), and this interface of the userspace PM is implemented as a wrapper of mptcp_userspace_pm_active(). To reduce the dependency on mptcp_pm_is_userspace() to identify userspace PM from in-kernel PM by using mptcp_pm_accept_new_subflow() helper in mptcp_pm_allow_new_subflow(). Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn> --- include/net/mptcp.h | 1 + net/mptcp/pm.c | 38 +++++++++++++------------------------- net/mptcp/pm_kernel.c | 16 ++++++++++++++++ net/mptcp/pm_userspace.c | 7 +++++++ net/mptcp/protocol.h | 25 +++++++++++++++++++++++++ net/mptcp/subflow.c | 6 ++---- 6 files changed, 64 insertions(+), 29 deletions(-) diff --git a/include/net/mptcp.h b/include/net/mptcp.h index 4cf59e83c1c5..549b8075cc86 100644 --- a/include/net/mptcp.h +++ b/include/net/mptcp.h @@ -123,6 +123,7 @@ struct mptcp_pm_ops { struct mptcp_pm_addr_entry *skc); bool (*get_priority)(struct mptcp_sock *msk, struct mptcp_addr_info *skc); + bool (*accept_new_subflow)(struct mptcp_sock *msk, bool allow); =20 char name[MPTCP_PM_NAME_MAX]; struct module *owner; diff --git a/net/mptcp/pm.c b/net/mptcp/pm.c index 4ed6a2ccb112..a96f3495d220 100644 --- a/net/mptcp/pm.c +++ b/net/mptcp/pm.c @@ -460,38 +460,25 @@ void mptcp_pm_new_connection(struct mptcp_sock *msk, = const struct sock *ssk, int mptcp_event(MPTCP_EVENT_CREATED, msk, ssk, GFP_ATOMIC); } =20 +bool mptcp_pm_accept_new_subflow(struct mptcp_sock *msk, bool allow) +{ + return msk->pm.ops->accept_new_subflow(msk, allow); +} + bool mptcp_pm_allow_new_subflow(struct mptcp_sock *msk) { struct mptcp_pm_data *pm =3D &msk->pm; - unsigned int subflows_max; int ret =3D 0; =20 - if (mptcp_pm_is_userspace(msk)) { - if (mptcp_userspace_pm_active(msk)) { - spin_lock_bh(&pm->lock); - pm->subflows++; - spin_unlock_bh(&pm->lock); - return true; - } - return false; - } - - subflows_max =3D mptcp_pm_get_subflows_max(msk); - - pr_debug("msk=3D%p subflows=3D%d max=3D%d allow=3D%d\n", msk, pm->subflow= s, - subflows_max, READ_ONCE(pm->accept_subflow)); - - /* try to avoid acquiring the lock below */ - if (!READ_ONCE(pm->accept_subflow)) + if (!mptcp_pm_accept_new_subflow(msk, true)) return false; =20 - spin_lock_bh(&pm->lock); - if (READ_ONCE(pm->accept_subflow)) { - ret =3D pm->subflows < subflows_max; - if (ret && ++pm->subflows =3D=3D subflows_max) - WRITE_ONCE(pm->accept_subflow, false); + ret =3D mptcp_pm_accept_new_subflow(msk, false); + if (ret) { + spin_lock_bh(&pm->lock); + pm->subflows++; + spin_unlock_bh(&pm->lock); } - spin_unlock_bh(&pm->lock); =20 return ret; } @@ -1060,7 +1047,8 @@ struct mptcp_pm_ops *mptcp_pm_find(const char *name) =20 int mptcp_pm_validate(struct mptcp_pm_ops *pm_ops) { - if (!pm_ops->get_local_id || !pm_ops->get_priority) { + if (!pm_ops->get_local_id || !pm_ops->get_priority || + !pm_ops->accept_new_subflow) { pr_err("%s does not implement required ops\n", pm_ops->name); return -EINVAL; } diff --git a/net/mptcp/pm_kernel.c b/net/mptcp/pm_kernel.c index 8935df61aa50..eaf73c4cedd3 100644 --- a/net/mptcp/pm_kernel.c +++ b/net/mptcp/pm_kernel.c @@ -1399,6 +1399,21 @@ static struct pernet_operations mptcp_pm_pernet_ops = =3D { .size =3D sizeof(struct pm_nl_pernet), }; =20 +static bool mptcp_pm_kernel_accept_new_subflow(struct mptcp_sock *msk, + bool allow) +{ + bool ret =3D false; + + if (READ_ONCE(msk->pm.accept_subflow)) { + if (allow) + return true; + + ret =3D mptcp_pm_accept_subflow(msk); + } + + return ret; +} + static void mptcp_pm_kernel_init(struct mptcp_sock *msk) { bool subflows_allowed =3D !!mptcp_pm_get_subflows_max(msk); @@ -1422,6 +1437,7 @@ static void mptcp_pm_kernel_init(struct mptcp_sock *m= sk) struct mptcp_pm_ops mptcp_pm_kernel =3D { .get_local_id =3D mptcp_pm_kernel_get_local_id, .get_priority =3D mptcp_pm_kernel_get_priority, + .accept_new_subflow =3D mptcp_pm_kernel_accept_new_subflow, .init =3D mptcp_pm_kernel_init, .name =3D "kernel", .owner =3D THIS_MODULE, diff --git a/net/mptcp/pm_userspace.c b/net/mptcp/pm_userspace.c index d9b0a93e42f6..31da3b9f4cc8 100644 --- a/net/mptcp/pm_userspace.c +++ b/net/mptcp/pm_userspace.c @@ -683,6 +683,12 @@ int mptcp_userspace_pm_get_addr(u8 id, struct mptcp_pm= _addr_entry *addr, return ret; } =20 +static bool mptcp_pm_userspace_accept_new_subflow(struct mptcp_sock *msk, + bool allow) +{ + return mptcp_userspace_pm_active(msk); +} + static void mptcp_pm_userspace_release(struct mptcp_sock *msk) { mptcp_userspace_pm_free_local_addr_list(msk); @@ -691,6 +697,7 @@ static void mptcp_pm_userspace_release(struct mptcp_soc= k *msk) static struct mptcp_pm_ops mptcp_pm_userspace =3D { .get_local_id =3D mptcp_pm_userspace_get_local_id, .get_priority =3D mptcp_pm_userspace_get_priority, + .accept_new_subflow =3D mptcp_pm_userspace_accept_new_subflow, .release =3D mptcp_pm_userspace_release, .name =3D "userspace", .owner =3D THIS_MODULE, diff --git a/net/mptcp/protocol.h b/net/mptcp/protocol.h index 3d51898a2781..134b46b53014 100644 --- a/net/mptcp/protocol.h +++ b/net/mptcp/protocol.h @@ -1008,6 +1008,7 @@ bool mptcp_pm_addr_families_match(const struct sock *= sk, void mptcp_pm_subflow_chk_stale(const struct mptcp_sock *msk, struct sock = *ssk); void mptcp_pm_new_connection(struct mptcp_sock *msk, const struct sock *ss= k, int server_side); void mptcp_pm_fully_established(struct mptcp_sock *msk, const struct sock = *ssk); +bool mptcp_pm_accept_new_subflow(struct mptcp_sock *msk, bool allow); bool mptcp_pm_allow_new_subflow(struct mptcp_sock *msk); void mptcp_pm_connection_closed(struct mptcp_sock *msk); void mptcp_pm_subflow_established(struct mptcp_sock *msk); @@ -1184,6 +1185,30 @@ static inline void mptcp_pm_close_subflow(struct mpt= cp_sock *msk) spin_unlock_bh(&msk->pm.lock); } =20 +static inline bool __mptcp_pm_accept_subflow(struct mptcp_sock *msk) +{ + unsigned int subflows_max =3D mptcp_pm_get_subflows_max(msk); + struct mptcp_pm_data *pm =3D &msk->pm; + bool ret; + + ret =3D pm->subflows < subflows_max; + if (ret && pm->subflows + 1 =3D=3D subflows_max) + WRITE_ONCE(pm->accept_subflow, false); + + return ret; +} + +static inline bool mptcp_pm_accept_subflow(struct mptcp_sock *msk) +{ + bool ret; + + spin_lock_bh(&msk->pm.lock); + ret =3D __mptcp_pm_accept_subflow(msk); + spin_unlock_bh(&msk->pm.lock); + + return ret; +} + void mptcp_sockopt_sync_locked(struct mptcp_sock *msk, struct sock *ssk); =20 static inline struct mptcp_ext *mptcp_get_ext(const struct sk_buff *skb) diff --git a/net/mptcp/subflow.c b/net/mptcp/subflow.c index 409bd415ef1d..b13a679bdf05 100644 --- a/net/mptcp/subflow.c +++ b/net/mptcp/subflow.c @@ -58,12 +58,10 @@ static void subflow_generate_hmac(u64 key1, u64 key2, u= 32 nonce1, u32 nonce2, mptcp_crypto_hmac_sha(key1, key2, msg, 8, hmac); } =20 -static bool mptcp_can_accept_new_subflow(const struct mptcp_sock *msk) +static bool mptcp_can_accept_new_subflow(struct mptcp_sock *msk) { return mptcp_is_fully_established((void *)msk) && - ((mptcp_pm_is_userspace(msk) && - mptcp_userspace_pm_active(msk)) || - READ_ONCE(msk->pm.accept_subflow)); + mptcp_pm_accept_new_subflow(msk, true); } =20 /* validate received token and create truncated hmac and nonce for SYN-ACK= */ --=20 2.43.0