From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S1755016AbdKIOww (ORCPT ); Thu, 9 Nov 2017 09:52:52 -0500 Received: from mail-pf0-f195.google.com ([209.85.192.195]:53692 "EHLO mail-pf0-f195.google.com" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S1753826AbdKIOws (ORCPT ); Thu, 9 Nov 2017 09:52:48 -0500 X-Google-Smtp-Source: ABhQp+RVkJQdeDNXF2Em5PYTi+9sC+OLgy9ZFIMXKZCjfMkhkwAPDM27kb7bN+TUPRLPvuA50PLIsA== Message-ID: <1510239161.2849.123.camel@edumazet-glaptop3.roam.corp.google.com> Subject: Re: [PATCH net-next 2/2] tcp: handle TCP_TIME_WAIT/TCP_NEW_SYN_RECV in tcp_set_state tracepoint From: Eric Dumazet To: Yafang Shao Cc: davem@davemloft.net, kuznet@ms2.inr.ac.ru, yoshfuji@linux-ipv6.org, rostedt@goodmis.org, mingo@redhat.com, netdev@vger.kernel.org, linux-kernel@vger.kernel.org Date: Thu, 09 Nov 2017 06:52:41 -0800 In-Reply-To: <1510237564-6013-2-git-send-email-laoar.shao@gmail.com> References: <1510237564-6013-1-git-send-email-laoar.shao@gmail.com> <1510237564-6013-2-git-send-email-laoar.shao@gmail.com> Content-Type: text/plain; charset="UTF-8" X-Mailer: Evolution 3.10.4-0ubuntu2 Mime-Version: 1.0 Content-Transfer-Encoding: 7bit Sender: linux-kernel-owner@vger.kernel.org List-ID: X-Mailing-List: linux-kernel@vger.kernel.org On Thu, 2017-11-09 at 14:26 +0000, Yafang Shao wrote: > When TCP connetion in TCP_TIME_WAIT or TCP_NEW_SYN_RECV state, it can't > get the sport/dport/saddr/daddr from inet_sock. > > trace_tcp_set_state may be called when the oldstate in these two states. > > Signed-off-by: Yafang Shao > --- > include/trace/events/tcp.h | 33 ++++++++++++++++++++++++++------- > 1 file changed, 26 insertions(+), 7 deletions(-) > > diff --git a/include/trace/events/tcp.h b/include/trace/events/tcp.h > index 07cccca..1982a71 100644 > --- a/include/trace/events/tcp.h > +++ b/include/trace/events/tcp.h > @@ -196,7 +196,6 @@ > ), > > TP_fast_assign( > - struct inet_sock *inet = inet_sk(sk); > struct in6_addr *pin6; > __be32 *p32; > > @@ -204,14 +203,34 @@ > __entry->oldstate = oldstate; > __entry->newstate = newstate; > > - __entry->sport = ntohs(inet->inet_sport); > - __entry->dport = ntohs(inet->inet_dport); > + if (oldstate == TCP_TIME_WAIT) { > + __entry->sport = ntohs(inet_twsk(sk)->tw_sport); > + __entry->dport = ntohs(inet_twsk(sk)->tw_dport); > > - p32 = (__be32 *) __entry->saddr; > - *p32 = inet->inet_saddr; > + p32 = (__be32 *) __entry->saddr; > + *p32 = inet_twsk(sk)->tw_rcv_saddr; > > - p32 = (__be32 *) __entry->daddr; > - *p32 = inet->inet_daddr; > + p32 = (__be32 *) __entry->daddr; > + *p32 = inet_twsk(sk)->tw_daddr; > + } else if (oldstate == TCP_NEW_SYN_RECV) { > + __entry->sport = inet_rsk(inet_reqsk(sk))->ir_num; > + __entry->dport = ntohs(inet_rsk(inet_reqsk(sk))->ir_rmt_port); > + > + p32 = (__be32 *) __entry->saddr; > + *p32 = inet_rsk(inet_reqsk(sk))->ir_loc_addr; > + > + p32 = (__be32 *) __entry->daddr; > + *p32 = inet_rsk(inet_reqsk(sk))->ir_rmt_addr; > + } else { > + __entry->sport = ntohs(inet_sk(sk)->inet_sport); > + __entry->dport = ntohs(inet_sk(sk)->inet_dport); > + > + p32 = (__be32 *) __entry->saddr; > + *p32 = inet_sk(sk)->inet_saddr; > + > + p32 = (__be32 *) __entry->daddr; > + *p32 = inet_sk(sk)->inet_daddr; > + } Wow. Since all three variants of sockets (full sockets, request sockets, timewait sockets) are all hashed into ehash table these days, they all have the fields at the same offset For IPv4, that would be : __sk_common.skc_daddr (or inet_daddr) __sk_common.skc_rcv_saddr (or inet_rcv_saddr ) __sk_common.skc_dport (or inet_dport) __sk_common.skc_num (or inet_num) Look at __inet_lookup_established() and INET_MATCH() : They deal with the three variants, without having to look at sk_state. If you were using the fields that are common to all sockets, no need to add all this unnecessary complexity.