Properly handle an edge case previously not handled correctly: a
socket can have a tcp connection that has entered time wait attached to it, in the event that shutdown() is called on the socket and the FINs properly exchange before close(). In this case we don't detach or free the inpcb, just leave the tcptw detached and freed, but we must release the inpcb lock (which we didn't previously). MFC after: 3 months
This commit is contained in:
parent
05603d9017
commit
34af7bae80
@ -1823,10 +1823,14 @@ tcp_twclose(struct tcptw *tw, int reuse)
|
||||
struct inpcb *inp;
|
||||
|
||||
/*
|
||||
* At this point, we should have an inpcb<->twtcp pair, with no
|
||||
* associated socket. Validate that this is the case.
|
||||
* At this point, we are in one of two situations:
|
||||
*
|
||||
* XXXRW: This comment stale -- could still have socket ...?
|
||||
* (1) We have no socket, just an inpcb<->twtcp pair. Release it all
|
||||
* after validating.
|
||||
*
|
||||
* (2) We have a socket, which we may or may now own the reference
|
||||
* for. If we own the reference, release all the state after
|
||||
* validating. If not, leave it for the socket close to clean up.
|
||||
*/
|
||||
inp = tw->tw_inpcb;
|
||||
KASSERT((inp->inp_vflag & INP_TIMEWAIT), ("tcp_twclose: !timewait"));
|
||||
@ -1840,9 +1844,13 @@ tcp_twclose(struct tcptw *tw, int reuse)
|
||||
inp->inp_vflag |= INP_DROPPED;
|
||||
|
||||
so = inp->inp_socket;
|
||||
if (so != NULL && inp->inp_vflag & INP_SOCKREF) {
|
||||
KASSERT(so->so_state & SS_PROTOREF,
|
||||
("tcp_twclose: !SS_PROTOREF"));
|
||||
if (so != NULL) {
|
||||
if (inp->inp_vflag & INP_SOCKREF) {
|
||||
/*
|
||||
* If a socket is present, and we own the only
|
||||
* reference, we need to tear down the socket and the
|
||||
* inpcb.
|
||||
*/
|
||||
inp->inp_vflag &= ~INP_SOCKREF;
|
||||
#ifdef INET6
|
||||
if (inp->inp_vflag & INP_IPV6PROTO) {
|
||||
@ -1855,17 +1863,26 @@ tcp_twclose(struct tcptw *tw, int reuse)
|
||||
#endif
|
||||
ACCEPT_LOCK();
|
||||
SOCK_LOCK(so);
|
||||
KASSERT(so->so_state & SS_PROTOREF,
|
||||
("tcp_twclose: INP_SOCKREF && !SS_PROTOREF"));
|
||||
so->so_state &= ~SS_PROTOREF;
|
||||
sofree(so);
|
||||
} else if (so == NULL) {
|
||||
} else {
|
||||
/*
|
||||
* If we don't own the only reference, the socket and
|
||||
* inpcb need to be left around to be handled by
|
||||
* tcp_usr_detach() later.
|
||||
*/
|
||||
INP_UNLOCK(inp);
|
||||
}
|
||||
} else {
|
||||
#ifdef INET6
|
||||
if (inp->inp_vflag & INP_IPV6PROTO)
|
||||
in6_pcbfree(inp);
|
||||
else
|
||||
#endif
|
||||
in_pcbfree(inp);
|
||||
} else
|
||||
printf("tcp_twclose: so != NULL but !INP_SOCKREF");
|
||||
}
|
||||
tcpstat.tcps_closed++;
|
||||
crfree(tw->tw_cred);
|
||||
tw->tw_cred = NULL;
|
||||
|
@ -1823,10 +1823,14 @@ tcp_twclose(struct tcptw *tw, int reuse)
|
||||
struct inpcb *inp;
|
||||
|
||||
/*
|
||||
* At this point, we should have an inpcb<->twtcp pair, with no
|
||||
* associated socket. Validate that this is the case.
|
||||
* At this point, we are in one of two situations:
|
||||
*
|
||||
* XXXRW: This comment stale -- could still have socket ...?
|
||||
* (1) We have no socket, just an inpcb<->twtcp pair. Release it all
|
||||
* after validating.
|
||||
*
|
||||
* (2) We have a socket, which we may or may now own the reference
|
||||
* for. If we own the reference, release all the state after
|
||||
* validating. If not, leave it for the socket close to clean up.
|
||||
*/
|
||||
inp = tw->tw_inpcb;
|
||||
KASSERT((inp->inp_vflag & INP_TIMEWAIT), ("tcp_twclose: !timewait"));
|
||||
@ -1840,9 +1844,13 @@ tcp_twclose(struct tcptw *tw, int reuse)
|
||||
inp->inp_vflag |= INP_DROPPED;
|
||||
|
||||
so = inp->inp_socket;
|
||||
if (so != NULL && inp->inp_vflag & INP_SOCKREF) {
|
||||
KASSERT(so->so_state & SS_PROTOREF,
|
||||
("tcp_twclose: !SS_PROTOREF"));
|
||||
if (so != NULL) {
|
||||
if (inp->inp_vflag & INP_SOCKREF) {
|
||||
/*
|
||||
* If a socket is present, and we own the only
|
||||
* reference, we need to tear down the socket and the
|
||||
* inpcb.
|
||||
*/
|
||||
inp->inp_vflag &= ~INP_SOCKREF;
|
||||
#ifdef INET6
|
||||
if (inp->inp_vflag & INP_IPV6PROTO) {
|
||||
@ -1855,17 +1863,26 @@ tcp_twclose(struct tcptw *tw, int reuse)
|
||||
#endif
|
||||
ACCEPT_LOCK();
|
||||
SOCK_LOCK(so);
|
||||
KASSERT(so->so_state & SS_PROTOREF,
|
||||
("tcp_twclose: INP_SOCKREF && !SS_PROTOREF"));
|
||||
so->so_state &= ~SS_PROTOREF;
|
||||
sofree(so);
|
||||
} else if (so == NULL) {
|
||||
} else {
|
||||
/*
|
||||
* If we don't own the only reference, the socket and
|
||||
* inpcb need to be left around to be handled by
|
||||
* tcp_usr_detach() later.
|
||||
*/
|
||||
INP_UNLOCK(inp);
|
||||
}
|
||||
} else {
|
||||
#ifdef INET6
|
||||
if (inp->inp_vflag & INP_IPV6PROTO)
|
||||
in6_pcbfree(inp);
|
||||
else
|
||||
#endif
|
||||
in_pcbfree(inp);
|
||||
} else
|
||||
printf("tcp_twclose: so != NULL but !INP_SOCKREF");
|
||||
}
|
||||
tcpstat.tcps_closed++;
|
||||
crfree(tw->tw_cred);
|
||||
tw->tw_cred = NULL;
|
||||
|
@ -169,8 +169,11 @@ tcp_usr_detach(struct socket *so)
|
||||
if (inp->inp_vflag & INP_DROPPED) {
|
||||
/*
|
||||
* Connection was in time wait and has been dropped;
|
||||
* the calling path is via tcp_twclose(), which will
|
||||
* free the tcptw, so we can discard the remainder.
|
||||
* the calling path is either via tcp_twclose(), or
|
||||
* as a result of an eventual soclose() after
|
||||
* tcp_twclose() has been called. In either case,
|
||||
* tcp_twclose() has detached the tcptw from the
|
||||
* inpcb, so we just detach and free the inpcb.
|
||||
*
|
||||
* XXXRW: Would it be cleaner to free the tcptw
|
||||
* here?
|
||||
|
Loading…
Reference in New Issue
Block a user