diff mbox series

inet_diag: add support for tw_mark

Message ID 20210616060604.3639340-1-zenczykowski@gmail.com (mailing list archive)
State Accepted
Commit 1b3fc771769c9f9418b23dd5676ab25a215d247d
Delegated to: Netdev Maintainers
Headers show
Series inet_diag: add support for tw_mark | expand

Checks

Context Check Description
netdev/cover_letter success Link
netdev/fixes_present success Link
netdev/patch_count success Link
netdev/tree_selection success Guessed tree name to be net-next
netdev/subject_prefix warning Target tree name not specified in the subject
netdev/cc_maintainers success CCed 7 of 7 maintainers
netdev/source_inline success Was 0 now: 0
netdev/verify_signedoff success Link
netdev/module_param success Was 0 now: 0
netdev/build_32bit success Errors and warnings before: 0 this patch: 0
netdev/kdoc success Errors and warnings before: 0 this patch: 0
netdev/verify_fixes success Link
netdev/checkpatch success total: 0 errors, 0 warnings, 0 checks, 36 lines checked
netdev/build_allmodconfig_warn success Errors and warnings before: 0 this patch: 0
netdev/header_inline success Link

Commit Message

Maciej Żenczykowski June 16, 2021, 6:06 a.m. UTC
From: Maciej Żenczykowski <maze@google.com>

Timewait sockets have included mark since approx 4.18.

Cc: Eric Dumazet <edumazet@google.com>
Cc: Jon Maxwell <jmaxwell37@gmail.com>
Fixes: 00483690552c ("tcp: Add mark for TIMEWAIT sockets")
Signed-off-by: Maciej Żenczykowski <maze@google.com>
---
 net/ipv4/inet_diag.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

Comments

Eric Dumazet June 16, 2021, 6:18 a.m. UTC | #1
On Wed, Jun 16, 2021 at 8:06 AM Maciej Żenczykowski
<zenczykowski@gmail.com> wrote:
>
> From: Maciej Żenczykowski <maze@google.com>
>
> Timewait sockets have included mark since approx 4.18.
>
> Cc: Eric Dumazet <edumazet@google.com>
> Cc: Jon Maxwell <jmaxwell37@gmail.com>
> Fixes: 00483690552c ("tcp: Add mark for TIMEWAIT sockets")
> Signed-off-by: Maciej Żenczykowski <maze@google.com>
> ---
>

SGTM, thanks Maciej !

Reviewed-by: Eric Dumazet <edumazet@google.com>
Jonathan Maxwell June 16, 2021, 6:40 a.m. UTC | #2
On Wed, Jun 16, 2021 at 4:06 PM Maciej Żenczykowski
<zenczykowski@gmail.com> wrote:
>
> From: Maciej Żenczykowski <maze@google.com>
>
> Timewait sockets have included mark since approx 4.18.
>
> Cc: Eric Dumazet <edumazet@google.com>
> Cc: Jon Maxwell <jmaxwell37@gmail.com>
> Fixes: 00483690552c ("tcp: Add mark for TIMEWAIT sockets")
> Signed-off-by: Maciej Żenczykowski <maze@google.com>
> ---
>  net/ipv4/inet_diag.c | 12 ++++++++++--
>  1 file changed, 10 insertions(+), 2 deletions(-)
>
> diff --git a/net/ipv4/inet_diag.c b/net/ipv4/inet_diag.c
> index 93474b1bea4e..e65f4ef024a4 100644
> --- a/net/ipv4/inet_diag.c
> +++ b/net/ipv4/inet_diag.c
> @@ -416,7 +416,7 @@ EXPORT_SYMBOL_GPL(inet_sk_diag_fill);
>  static int inet_twsk_diag_fill(struct sock *sk,
>                                struct sk_buff *skb,
>                                struct netlink_callback *cb,
> -                              u16 nlmsg_flags)
> +                              u16 nlmsg_flags, bool net_admin)
>  {
>         struct inet_timewait_sock *tw = inet_twsk(sk);
>         struct inet_diag_msg *r;
> @@ -444,6 +444,12 @@ static int inet_twsk_diag_fill(struct sock *sk,
>         r->idiag_uid          = 0;
>         r->idiag_inode        = 0;
>
> +       if (net_admin && nla_put_u32(skb, INET_DIAG_MARK,
> +                                    tw->tw_mark)) {
> +               nlmsg_cancel(skb, nlh);
> +               return -EMSGSIZE;
> +       }
> +
>         nlmsg_end(skb, nlh);
>         return 0;
>  }
> @@ -494,7 +500,7 @@ static int sk_diag_fill(struct sock *sk, struct sk_buff *skb,
>                         u16 nlmsg_flags, bool net_admin)
>  {
>         if (sk->sk_state == TCP_TIME_WAIT)
> -               return inet_twsk_diag_fill(sk, skb, cb, nlmsg_flags);
> +               return inet_twsk_diag_fill(sk, skb, cb, nlmsg_flags, net_admin);
>
>         if (sk->sk_state == TCP_NEW_SYN_RECV)
>                 return inet_req_diag_fill(sk, skb, cb, nlmsg_flags, net_admin);
> @@ -801,6 +807,8 @@ int inet_diag_bc_sk(const struct nlattr *bc, struct sock *sk)
>                 entry.mark = sk->sk_mark;
>         else if (sk->sk_state == TCP_NEW_SYN_RECV)
>                 entry.mark = inet_rsk(inet_reqsk(sk))->ir_mark;
> +       else if (sk->sk_state == TCP_TIME_WAIT)
> +               entry.mark = inet_twsk(sk)->tw_mark;
>         else
>                 entry.mark = 0;
>  #ifdef CONFIG_SOCK_CGROUP_DATA
> --
> 2.32.0.272.g935e593368-goog
>

Thanks for adding that to the inet_diag module.

Reviewed-by: Jon Maxwell <jmaxwell37@gmail.com>
patchwork-bot+netdevbpf@kernel.org June 16, 2021, 7:50 a.m. UTC | #3
Hello:

This patch was applied to netdev/net-next.git (refs/heads/master):

On Tue, 15 Jun 2021 23:06:04 -0700 you wrote:
> From: Maciej Żenczykowski <maze@google.com>
> 
> Timewait sockets have included mark since approx 4.18.
> 
> Cc: Eric Dumazet <edumazet@google.com>
> Cc: Jon Maxwell <jmaxwell37@gmail.com>
> Fixes: 00483690552c ("tcp: Add mark for TIMEWAIT sockets")
> Signed-off-by: Maciej Żenczykowski <maze@google.com>
> 
> [...]

Here is the summary with links:
  - inet_diag: add support for tw_mark
    https://git.kernel.org/netdev/net-next/c/1b3fc771769c

You are awesome, thank you!
--
Deet-doot-dot, I am a bot.
https://korg.docs.kernel.org/patchwork/pwbot.html
diff mbox series

Patch

diff --git a/net/ipv4/inet_diag.c b/net/ipv4/inet_diag.c
index 93474b1bea4e..e65f4ef024a4 100644
--- a/net/ipv4/inet_diag.c
+++ b/net/ipv4/inet_diag.c
@@ -416,7 +416,7 @@  EXPORT_SYMBOL_GPL(inet_sk_diag_fill);
 static int inet_twsk_diag_fill(struct sock *sk,
 			       struct sk_buff *skb,
 			       struct netlink_callback *cb,
-			       u16 nlmsg_flags)
+			       u16 nlmsg_flags, bool net_admin)
 {
 	struct inet_timewait_sock *tw = inet_twsk(sk);
 	struct inet_diag_msg *r;
@@ -444,6 +444,12 @@  static int inet_twsk_diag_fill(struct sock *sk,
 	r->idiag_uid	      = 0;
 	r->idiag_inode	      = 0;
 
+	if (net_admin && nla_put_u32(skb, INET_DIAG_MARK,
+				     tw->tw_mark)) {
+		nlmsg_cancel(skb, nlh);
+		return -EMSGSIZE;
+	}
+
 	nlmsg_end(skb, nlh);
 	return 0;
 }
@@ -494,7 +500,7 @@  static int sk_diag_fill(struct sock *sk, struct sk_buff *skb,
 			u16 nlmsg_flags, bool net_admin)
 {
 	if (sk->sk_state == TCP_TIME_WAIT)
-		return inet_twsk_diag_fill(sk, skb, cb, nlmsg_flags);
+		return inet_twsk_diag_fill(sk, skb, cb, nlmsg_flags, net_admin);
 
 	if (sk->sk_state == TCP_NEW_SYN_RECV)
 		return inet_req_diag_fill(sk, skb, cb, nlmsg_flags, net_admin);
@@ -801,6 +807,8 @@  int inet_diag_bc_sk(const struct nlattr *bc, struct sock *sk)
 		entry.mark = sk->sk_mark;
 	else if (sk->sk_state == TCP_NEW_SYN_RECV)
 		entry.mark = inet_rsk(inet_reqsk(sk))->ir_mark;
+	else if (sk->sk_state == TCP_TIME_WAIT)
+		entry.mark = inet_twsk(sk)->tw_mark;
 	else
 		entry.mark = 0;
 #ifdef CONFIG_SOCK_CGROUP_DATA