[<prev] [next>] [<thread-prev] [thread-next>] [day] [month] [year] [list]
Message-ID: <CAGHK07B_cPZaNNkiCwM-t5B_wxe5tihuReUHcG8mM6fpSW2Cyg@mail.gmail.com>
Date: Wed, 16 Jun 2021 16:40:22 +1000
From: Jonathan Maxwell <jmaxwell37@...il.com>
To: Maciej Żenczykowski <zenczykowski@...il.com>
Cc: Maciej Żenczykowski <maze@...gle.com>,
Linux Network Development Mailing List
<netdev@...r.kernel.org>, "David S . Miller" <davem@...emloft.net>,
Hideaki YOSHIFUJI <yoshfuji@...ux-ipv6.org>,
David Ahern <dsahern@...nel.org>,
Jakub Kicinski <kuba@...nel.org>,
Eric Dumazet <edumazet@...gle.com>
Subject: Re: [PATCH] inet_diag: add support for tw_mark
On Wed, Jun 16, 2021 at 4:06 PM Maciej Żenczykowski
<zenczykowski@...il.com> wrote:
>
> From: Maciej Żenczykowski <maze@...gle.com>
>
> Timewait sockets have included mark since approx 4.18.
>
> Cc: Eric Dumazet <edumazet@...gle.com>
> Cc: Jon Maxwell <jmaxwell37@...il.com>
> Fixes: 00483690552c ("tcp: Add mark for TIMEWAIT sockets")
> Signed-off-by: Maciej Żenczykowski <maze@...gle.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@...il.com>
Powered by blists - more mailing lists