git: 74703901d8bb - main - tcp: use a TCP flag to check if connection has been close(2)d
- Go to: [ bottom of page ] [ top of archives ] [ this month ]
Date: Mon, 04 Jul 2022 19:41:14 UTC
The branch main has been updated by glebius:
URL: https://cgit.FreeBSD.org/src/commit/?id=74703901d8bbc3bc7a29df648bc3c131c87393c2
commit 74703901d8bbc3bc7a29df648bc3c131c87393c2
Author: Gleb Smirnoff <glebius@FreeBSD.org>
AuthorDate: 2022-07-04 19:40:51 +0000
Commit: Gleb Smirnoff <glebius@FreeBSD.org>
CommitDate: 2022-07-04 19:40:51 +0000
tcp: use a TCP flag to check if connection has been close(2)d
The flag SS_NOFDREF is a private flag of the socket layer. It also
is supposed to be read with SOCK_LOCK(), which we don't own here.
Reviewed by: rrs, tuexen
Differential revision: https://reviews.freebsd.org/D35663
---
sys/netinet/tcp_input.c | 3 +--
sys/netinet/tcp_stacks/bbr.c | 54 ++++++++++++++++---------------------------
sys/netinet/tcp_stacks/rack.c | 29 ++++++++++-------------
sys/netinet/tcp_usrreq.c | 1 +
sys/netinet/tcp_var.h | 2 +-
5 files changed, 35 insertions(+), 54 deletions(-)
diff --git a/sys/netinet/tcp_input.c b/sys/netinet/tcp_input.c
index a51e62616bea..f2d4f56ac582 100644
--- a/sys/netinet/tcp_input.c
+++ b/sys/netinet/tcp_input.c
@@ -2315,8 +2315,7 @@ tcp_do_segment(struct mbuf *m, struct tcphdr *th, struct socket *so,
* If new data are received on a connection after the
* user processes are gone, then RST the other end.
*/
- if ((so->so_state & SS_NOFDREF) &&
- tp->t_state > TCPS_CLOSE_WAIT && tlen) {
+ if ((tp->t_flags & TF_CLOSED) && tlen) {
if ((s = tcp_log_addrs(inc, th, NULL, NULL))) {
log(LOG_DEBUG, "%s; %s: %s: Received %d bytes of data "
"after socket was closed, "
diff --git a/sys/netinet/tcp_stacks/bbr.c b/sys/netinet/tcp_stacks/bbr.c
index 67d0b7a5c64c..6278bb228c4f 100644
--- a/sys/netinet/tcp_stacks/bbr.c
+++ b/sys/netinet/tcp_stacks/bbr.c
@@ -9493,15 +9493,12 @@ bbr_do_fin_wait_1(struct mbuf *m, struct tcphdr *th, struct socket *so,
/*
* If new data are received on a connection after the user processes
* are gone, then RST the other end.
+ * We call a new function now so we might continue and setup
+ * to reset at all data being ack'd.
*/
- if ((so->so_state & SS_NOFDREF) && tlen) {
- /*
- * We call a new function now so we might continue and setup
- * to reset at all data being ack'd.
- */
- if (bbr_check_data_after_close(m, bbr, tp, &tlen, th, so))
- return (1);
- }
+ if ((tp->t_flags & TF_CLOSED) && tlen &&
+ bbr_check_data_after_close(m, bbr, tp, &tlen, th, so))
+ return (1);
/*
* If last ACK falls within this segment's sequence numbers, record
* its timestamp. NOTE: 1) That the test incorporates suggestions
@@ -9620,15 +9617,12 @@ bbr_do_closing(struct mbuf *m, struct tcphdr *th, struct socket *so,
/*
* If new data are received on a connection after the user processes
* are gone, then RST the other end.
+ * We call a new function now so we might continue and setup
+ * to reset at all data being ack'd.
*/
- if ((so->so_state & SS_NOFDREF) && tlen) {
- /*
- * We call a new function now so we might continue and setup
- * to reset at all data being ack'd.
- */
- if (bbr_check_data_after_close(m, bbr, tp, &tlen, th, so))
- return (1);
- }
+ if ((tp->t_flags & TF_CLOSED) && tlen &&
+ bbr_check_data_after_close(m, bbr, tp, &tlen, th, so))
+ return (1);
/*
* If last ACK falls within this segment's sequence numbers, record
* its timestamp. NOTE: 1) That the test incorporates suggestions
@@ -9733,15 +9727,12 @@ bbr_do_lastack(struct mbuf *m, struct tcphdr *th, struct socket *so,
/*
* If new data are received on a connection after the user processes
* are gone, then RST the other end.
+ * We call a new function now so we might continue and setup
+ * to reset at all data being ack'd.
*/
- if ((so->so_state & SS_NOFDREF) && tlen) {
- /*
- * We call a new function now so we might continue and setup
- * to reset at all data being ack'd.
- */
- if (bbr_check_data_after_close(m, bbr, tp, &tlen, th, so))
- return (1);
- }
+ if ((tp->t_flags & TF_CLOSED) && tlen &&
+ bbr_check_data_after_close(m, bbr, tp, &tlen, th, so))
+ return (1);
/*
* If last ACK falls within this segment's sequence numbers, record
* its timestamp. NOTE: 1) That the test incorporates suggestions
@@ -9850,17 +9841,12 @@ bbr_do_fin_wait_2(struct mbuf *m, struct tcphdr *th, struct socket *so,
* If new data are received on a connection after the user processes
* are gone, then we may RST the other end depending on the outcome
* of bbr_check_data_after_close.
+ * We call a new function now so we might continue and setup
+ * to reset at all data being ack'd.
*/
- if ((so->so_state & SS_NOFDREF) &&
- tlen) {
- /*
- * We call a new function now so we might continue and setup
- * to reset at all data being ack'd.
- */
- if (bbr_check_data_after_close(m, bbr, tp, &tlen, th, so))
- return (1);
- }
- INP_WLOCK_ASSERT(tp->t_inpcb);
+ if (tp->t_state > TCPS_CLOSE_WAIT && tlen &&
+ bbr_check_data_after_close(m, bbr, tp, &tlen, th, so))
+ return (1);
/*
* If last ACK falls within this segment's sequence numbers, record
* its timestamp. NOTE: 1) That the test incorporates suggestions
diff --git a/sys/netinet/tcp_stacks/rack.c b/sys/netinet/tcp_stacks/rack.c
index 3b40138d0fd5..ff43fbd02de2 100644
--- a/sys/netinet/tcp_stacks/rack.c
+++ b/sys/netinet/tcp_stacks/rack.c
@@ -11855,10 +11855,9 @@ rack_do_fin_wait_1(struct mbuf *m, struct tcphdr *th, struct socket *so,
* If new data are received on a connection after the user processes
* are gone, then RST the other end.
*/
- if ((so->so_state & SS_NOFDREF) && tlen) {
- if (rack_check_data_after_close(m, tp, &tlen, th, so))
- return (1);
- }
+ if ((tp->t_flags & TF_CLOSED) && tlen &&
+ rack_check_data_after_close(m, tp, &tlen, th, so))
+ return (1);
/*
* If last ACK falls within this segment's sequence numbers, record
* its timestamp. NOTE: 1) That the test incorporates suggestions
@@ -11983,10 +11982,9 @@ rack_do_closing(struct mbuf *m, struct tcphdr *th, struct socket *so,
* If new data are received on a connection after the user processes
* are gone, then RST the other end.
*/
- if ((so->so_state & SS_NOFDREF) && tlen) {
- if (rack_check_data_after_close(m, tp, &tlen, th, so))
- return (1);
- }
+ if ((tp->t_flags & TF_CLOSED) && tlen &&
+ rack_check_data_after_close(m, tp, &tlen, th, so))
+ return (1);
/*
* If last ACK falls within this segment's sequence numbers, record
* its timestamp. NOTE: 1) That the test incorporates suggestions
@@ -12097,10 +12095,9 @@ rack_do_lastack(struct mbuf *m, struct tcphdr *th, struct socket *so,
* If new data are received on a connection after the user processes
* are gone, then RST the other end.
*/
- if ((so->so_state & SS_NOFDREF) && tlen) {
- if (rack_check_data_after_close(m, tp, &tlen, th, so))
- return (1);
- }
+ if ((tp->t_flags & TF_CLOSED) && tlen &&
+ rack_check_data_after_close(m, tp, &tlen, th, so))
+ return (1);
/*
* If last ACK falls within this segment's sequence numbers, record
* its timestamp. NOTE: 1) That the test incorporates suggestions
@@ -12212,11 +12209,9 @@ rack_do_fin_wait_2(struct mbuf *m, struct tcphdr *th, struct socket *so,
* If new data are received on a connection after the user processes
* are gone, then RST the other end.
*/
- if ((so->so_state & SS_NOFDREF) &&
- tlen) {
- if (rack_check_data_after_close(m, tp, &tlen, th, so))
- return (1);
- }
+ if ((tp->t_flags & TF_CLOSED) && tlen &&
+ rack_check_data_after_close(m, tp, &tlen, th, so))
+ return (1);
/*
* If last ACK falls within this segment's sequence numbers, record
* its timestamp. NOTE: 1) That the test incorporates suggestions
diff --git a/sys/netinet/tcp_usrreq.c b/sys/netinet/tcp_usrreq.c
index e05ef963375c..6e4171bd1264 100644
--- a/sys/netinet/tcp_usrreq.c
+++ b/sys/netinet/tcp_usrreq.c
@@ -1406,6 +1406,7 @@ tcp_usr_close(struct socket *so)
if (!(inp->inp_flags & INP_TIMEWAIT) &&
!(inp->inp_flags & INP_DROPPED)) {
tp = intotcpcb(inp);
+ tp->t_flags |= TF_CLOSED;
TCPDEBUG1();
tcp_disconnect(tp);
TCPDEBUG2(PRU_CLOSE);
diff --git a/sys/netinet/tcp_var.h b/sys/netinet/tcp_var.h
index be9ff06f3a85..6d6c711f25b7 100644
--- a/sys/netinet/tcp_var.h
+++ b/sys/netinet/tcp_var.h
@@ -524,7 +524,7 @@ tcp_unlock_or_drop(struct tcpcb *tp, int tcp_output_retval)
#define TF_FORCEDATA 0x00800000 /* force out a byte */
#define TF_TSO 0x01000000 /* TSO enabled on this connection */
#define TF_TOE 0x02000000 /* this connection is offloaded */
-#define TF_UNUSED0 0x04000000 /* unused */
+#define TF_CLOSED 0x04000000 /* close(2) called on socket */
#define TF_UNUSED1 0x08000000 /* unused */
#define TF_LRD 0x10000000 /* Lost Retransmission Detection */
#define TF_CONGRECOVERY 0x20000000 /* congestion recovery mode */