xref: /freebsd/sys/netinet/tcp_reass.c (revision c6db8143eda5c775467145ac73e8ebec47afdd8f)
1 /*-
2  * Copyright (c) 1982, 1986, 1988, 1990, 1993, 1994, 1995
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 4. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  *
29  *	@(#)tcp_input.c	8.12 (Berkeley) 5/24/95
30  */
31 
32 #include <sys/cdefs.h>
33 __FBSDID("$FreeBSD$");
34 
35 #include "opt_inet.h"
36 #include "opt_inet6.h"
37 #include "opt_tcpdebug.h"
38 
39 #include <sys/param.h>
40 #include <sys/kernel.h>
41 #include <sys/malloc.h>
42 #include <sys/mbuf.h>
43 #include <sys/socket.h>
44 #include <sys/socketvar.h>
45 #include <sys/sysctl.h>
46 #include <sys/syslog.h>
47 #include <sys/systm.h>
48 
49 #include <vm/uma.h>
50 
51 #include <net/if.h>
52 #include <net/if_var.h>
53 #include <net/route.h>
54 #include <net/vnet.h>
55 
56 #include <netinet/in.h>
57 #include <netinet/in_pcb.h>
58 #include <netinet/in_systm.h>
59 #include <netinet/in_var.h>
60 #include <netinet/ip.h>
61 #include <netinet/ip_var.h>
62 #include <netinet/ip_options.h>
63 #include <netinet/ip6.h>
64 #include <netinet6/in6_pcb.h>
65 #include <netinet6/ip6_var.h>
66 #include <netinet6/nd6.h>
67 #include <netinet/tcp.h>
68 #include <netinet/tcp_fsm.h>
69 #include <netinet/tcp_seq.h>
70 #include <netinet/tcp_timer.h>
71 #include <netinet/tcp_var.h>
72 #include <netinet6/tcp6_var.h>
73 #include <netinet/tcpip.h>
74 
75 void
76 tcp_reass_flush(struct tcpcb *tp)
77 {
78 	struct mbuf *m;
79 
80 	INP_WLOCK_ASSERT(tp->t_inpcb);
81 
82 	while ((m = tp->t_segq) != NULL) {
83 		tp->t_segq = m->m_nextpkt;
84 		tp->t_segqlen -= m->m_pkthdr.len;
85 		m_freem(m);
86 	}
87 
88 	KASSERT((tp->t_segqlen == 0),
89 	    ("TCP reass queue %p length is %d instead of 0 after flush.",
90 	    tp, tp->t_segqlen));
91 }
92 
93 #define	M_TCPHDR(m)	((struct tcphdr *)((m)->m_pkthdr.pkt_tcphdr))
94 
95 int
96 tcp_reass(struct tcpcb *tp, struct tcphdr *th, int *tlenp, struct mbuf *m)
97 {
98 	struct socket *so = tp->t_inpcb->inp_socket;
99 	struct mbuf *mq, *mp;
100 	int flags, wakeup;
101 
102 	INP_WLOCK_ASSERT(tp->t_inpcb);
103 
104 	/*
105 	 * XXX: tcp_reass() is rather inefficient with its data structures
106 	 * and should be rewritten (see NetBSD for optimizations).
107 	 */
108 
109 	/*
110 	 * Call with th==NULL after become established to
111 	 * force pre-ESTABLISHED data up to user socket.
112 	 */
113 	if (th == NULL)
114 		goto present;
115 
116 	M_ASSERTPKTHDR(m);
117 	KASSERT(*tlenp == m->m_pkthdr.len, ("%s: tlenp %u len %u", __func__,
118 	    *tlenp, m->m_pkthdr.len));
119 
120 	/*
121 	 * Limit the number of segments that can be queued to reduce the
122 	 * potential for mbuf exhaustion. For best performance, we want to be
123 	 * able to queue a full window's worth of segments. The size of the
124 	 * socket receive buffer determines our advertised window and grows
125 	 * automatically when socket buffer autotuning is enabled. Use it as the
126 	 * basis for our queue limit.
127 	 * Always let the missing segment through which caused this queue.
128 	 * NB: Access to the socket buffer is left intentionally unlocked as we
129 	 * can tolerate stale information here.
130 	 */
131 	if ((th->th_seq != tp->rcv_nxt || !TCPS_HAVEESTABLISHED(tp->t_state)) &&
132 	    tp->t_segqlen + m->m_pkthdr.len >= sbspace(&so->so_rcv)) {
133 		char *s;
134 
135 		TCPSTAT_INC(tcps_rcvreassfull);
136 		*tlenp = 0;
137 		if ((s = tcp_log_addrs(&tp->t_inpcb->inp_inc, th, NULL,
138 		    NULL))) {
139 			log(LOG_DEBUG, "%s; %s: queue limit reached, "
140 			    "segment dropped\n", s, __func__);
141 			free(s, M_TCPLOG);
142 		}
143 		m_freem(m);
144 		return (0);
145 	}
146 
147 	/*
148 	 * Find a segment which begins after this one does.
149 	 */
150 	mp = NULL;
151 	for (mq = tp->t_segq; mq != NULL; mq = mq->m_nextpkt) {
152 		if (SEQ_GT(M_TCPHDR(mq)->th_seq, th->th_seq))
153 			break;
154 		mp = mq;
155 	}
156 
157 	/*
158 	 * If there is a preceding segment, it may provide some of
159 	 * our data already.  If so, drop the data from the incoming
160 	 * segment.  If it provides all of our data, drop us.
161 	 */
162 	if (mp != NULL) {
163 		int i;
164 
165 		/* conversion to int (in i) handles seq wraparound */
166 		i = M_TCPHDR(mp)->th_seq + mp->m_pkthdr.len - th->th_seq;
167 		if (i > 0) {
168 			if (i >= *tlenp) {
169 				TCPSTAT_INC(tcps_rcvduppack);
170 				TCPSTAT_ADD(tcps_rcvdupbyte, *tlenp);
171 				m_freem(m);
172 				/*
173 				 * Try to present any queued data
174 				 * at the left window edge to the user.
175 				 * This is needed after the 3-WHS
176 				 * completes.
177 				 */
178 				goto present;	/* ??? */
179 			}
180 			m_adj(m, i);
181 			*tlenp -= i;
182 			th->th_seq += i;
183 		}
184 	}
185 	tp->t_rcvoopack++;
186 	TCPSTAT_INC(tcps_rcvoopack);
187 	TCPSTAT_ADD(tcps_rcvoobyte, *tlenp);
188 
189 	/*
190 	 * While we overlap succeeding segments trim them or,
191 	 * if they are completely covered, dequeue them.
192 	 */
193 	while (mq) {
194 		struct mbuf *nq;
195 		int i;
196 
197 		i = (th->th_seq + *tlenp) - M_TCPHDR(mq)->th_seq;
198 		if (i <= 0)
199 			break;
200 		if (i < mq->m_pkthdr.len) {
201 			M_TCPHDR(mq)->th_seq += i;
202 			m_adj(mq, i);
203 			tp->t_segqlen -= i;
204 			break;
205 		}
206 
207 		nq = mq->m_nextpkt;
208 		tp->t_segqlen -= mq->m_pkthdr.len;
209 		m_freem(mq);
210 		if (mp)
211 			mp->m_nextpkt = nq;
212 		else
213 			tp->t_segq = nq;
214 		mq = nq;
215 	}
216 
217 	/*
218 	 * Insert the new segment queue entry into place.  Try to collapse
219 	 * mbuf chains if segments are adjacent.
220 	 */
221 	if (mp) {
222 		if (M_TCPHDR(mp)->th_seq + mp->m_pkthdr.len == th->th_seq)
223 			m_catpkt(mp, m);
224 		else {
225 			m->m_nextpkt = mp->m_nextpkt;
226 			mp->m_nextpkt = m;
227 			m->m_pkthdr.pkt_tcphdr = th;
228 		}
229 	} else {
230 		mq = tp->t_segq;
231 		tp->t_segq = m;
232 		if (mq && th->th_seq + *tlenp == M_TCPHDR(mq)->th_seq) {
233 			m->m_nextpkt = mq->m_nextpkt;
234 			mq->m_nextpkt = NULL;
235 			m_catpkt(m, mq);
236 		} else
237 			m->m_nextpkt = mq;
238 		m->m_pkthdr.pkt_tcphdr = th;
239 	}
240 	tp->t_segqlen += *tlenp;
241 
242 present:
243 	/*
244 	 * Present data to user, advancing rcv_nxt through
245 	 * completed sequence space.
246 	 */
247 	if (!TCPS_HAVEESTABLISHED(tp->t_state))
248 		return (0);
249 
250 	flags = 0;
251 	wakeup = 0;
252 	SOCKBUF_LOCK(&so->so_rcv);
253 	while ((mq = tp->t_segq) != NULL &&
254 	    M_TCPHDR(mq)->th_seq == tp->rcv_nxt) {
255 		tp->t_segq = mq->m_nextpkt;
256 
257 		tp->rcv_nxt += mq->m_pkthdr.len;
258 		tp->t_segqlen -= mq->m_pkthdr.len;
259 		flags = M_TCPHDR(mq)->th_flags & TH_FIN;
260 
261 		if (so->so_rcv.sb_state & SBS_CANTRCVMORE)
262 			m_freem(mq);
263 		else {
264 			mq->m_nextpkt = NULL;
265 			sbappendstream_locked(&so->so_rcv, mq);
266 			wakeup = 1;
267 		}
268 	}
269 	ND6_HINT(tp);
270 	if (wakeup)
271 		sorwakeup_locked(so);
272 	else
273 		SOCKBUF_UNLOCK(&so->so_rcv);
274 	return (flags);
275 }
276