xref: /freebsd-13.1/sys/netinet/tcp_subr.c (revision d64a952f)
1 /*-
2  * SPDX-License-Identifier: BSD-3-Clause
3  *
4  * Copyright (c) 1982, 1986, 1988, 1990, 1993, 1995
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  *
31  *	@(#)tcp_subr.c	8.2 (Berkeley) 5/24/95
32  */
33 
34 #include <sys/cdefs.h>
35 __FBSDID("$FreeBSD$");
36 
37 #include "opt_inet.h"
38 #include "opt_inet6.h"
39 #include "opt_ipsec.h"
40 #include "opt_kern_tls.h"
41 #include "opt_tcpdebug.h"
42 
43 #include <sys/param.h>
44 #include <sys/systm.h>
45 #include <sys/arb.h>
46 #include <sys/callout.h>
47 #include <sys/eventhandler.h>
48 #ifdef TCP_HHOOK
49 #include <sys/hhook.h>
50 #endif
51 #include <sys/kernel.h>
52 #ifdef TCP_HHOOK
53 #include <sys/khelp.h>
54 #endif
55 #ifdef KERN_TLS
56 #include <sys/ktls.h>
57 #endif
58 #include <sys/qmath.h>
59 #include <sys/stats.h>
60 #include <sys/sysctl.h>
61 #include <sys/jail.h>
62 #include <sys/malloc.h>
63 #include <sys/refcount.h>
64 #include <sys/mbuf.h>
65 #ifdef INET6
66 #include <sys/domain.h>
67 #endif
68 #include <sys/priv.h>
69 #include <sys/proc.h>
70 #include <sys/sdt.h>
71 #include <sys/socket.h>
72 #include <sys/socketvar.h>
73 #include <sys/protosw.h>
74 #include <sys/random.h>
75 
76 #include <vm/uma.h>
77 
78 #include <net/route.h>
79 #include <net/route/nhop.h>
80 #include <net/if.h>
81 #include <net/if_var.h>
82 #include <net/vnet.h>
83 
84 #include <netinet/in.h>
85 #include <netinet/in_fib.h>
86 #include <netinet/in_kdtrace.h>
87 #include <netinet/in_pcb.h>
88 #include <netinet/in_systm.h>
89 #include <netinet/in_var.h>
90 #include <netinet/ip.h>
91 #include <netinet/ip_icmp.h>
92 #include <netinet/ip_var.h>
93 #ifdef INET6
94 #include <netinet/icmp6.h>
95 #include <netinet/ip6.h>
96 #include <netinet6/in6_fib.h>
97 #include <netinet6/in6_pcb.h>
98 #include <netinet6/ip6_var.h>
99 #include <netinet6/scope6_var.h>
100 #include <netinet6/nd6.h>
101 #endif
102 
103 #include <netinet/tcp.h>
104 #include <netinet/tcp_fsm.h>
105 #include <netinet/tcp_seq.h>
106 #include <netinet/tcp_timer.h>
107 #include <netinet/tcp_var.h>
108 #include <netinet/tcp_log_buf.h>
109 #include <netinet/tcp_syncache.h>
110 #include <netinet/tcp_hpts.h>
111 #include <netinet/cc/cc.h>
112 #ifdef INET6
113 #include <netinet6/tcp6_var.h>
114 #endif
115 #include <netinet/tcpip.h>
116 #include <netinet/tcp_fastopen.h>
117 #ifdef TCPPCAP
118 #include <netinet/tcp_pcap.h>
119 #endif
120 #ifdef TCPDEBUG
121 #include <netinet/tcp_debug.h>
122 #endif
123 #ifdef INET6
124 #include <netinet6/ip6protosw.h>
125 #endif
126 #ifdef TCP_OFFLOAD
127 #include <netinet/tcp_offload.h>
128 #endif
129 #include <netinet/udp.h>
130 #include <netinet/udp_var.h>
131 
132 #include <netipsec/ipsec_support.h>
133 
134 #include <machine/in_cksum.h>
135 #include <crypto/siphash/siphash.h>
136 
137 #include <security/mac/mac_framework.h>
138 
139 VNET_DEFINE(int, tcp_mssdflt) = TCP_MSS;
140 #ifdef INET6
141 VNET_DEFINE(int, tcp_v6mssdflt) = TCP6_MSS;
142 #endif
143 
144 #ifdef NETFLIX_EXP_DETECTION
145 /*  Sack attack detection thresholds and such */
146 SYSCTL_NODE(_net_inet_tcp, OID_AUTO, sack_attack,
147     CTLFLAG_RW | CTLFLAG_MPSAFE, 0,
148     "Sack Attack detection thresholds");
149 int32_t tcp_force_detection = 0;
150 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, force_detection,
151     CTLFLAG_RW,
152     &tcp_force_detection, 0,
153     "Do we force detection even if the INP has it off?");
154 int32_t tcp_sack_to_ack_thresh = 700;	/* 70 % */
155 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, sack_to_ack_thresh,
156     CTLFLAG_RW,
157     &tcp_sack_to_ack_thresh, 700,
158     "Percentage of sacks to acks we must see above (10.1 percent is 101)?");
159 int32_t tcp_sack_to_move_thresh = 600;	/* 60 % */
160 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, move_thresh,
161     CTLFLAG_RW,
162     &tcp_sack_to_move_thresh, 600,
163     "Percentage of sack moves we must see above (10.1 percent is 101)");
164 int32_t tcp_restoral_thresh = 650;	/* 65 % (sack:2:ack -5%) */
165 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, restore_thresh,
166     CTLFLAG_RW,
167     &tcp_restoral_thresh, 550,
168     "Percentage of sack to ack percentage we must see below to restore(10.1 percent is 101)");
169 int32_t tcp_sad_decay_val = 800;
170 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, decay_per,
171     CTLFLAG_RW,
172     &tcp_sad_decay_val, 800,
173     "The decay percentage (10.1 percent equals 101 )");
174 int32_t tcp_map_minimum = 500;
175 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, nummaps,
176     CTLFLAG_RW,
177     &tcp_map_minimum, 500,
178     "Number of Map enteries before we start detection");
179 int32_t tcp_attack_on_turns_on_logging = 0;
180 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, attacks_logged,
181     CTLFLAG_RW,
182     &tcp_attack_on_turns_on_logging, 0,
183    "When we have a positive hit on attack, do we turn on logging?");
184 int32_t tcp_sad_pacing_interval = 2000;
185 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, sad_pacing_int,
186     CTLFLAG_RW,
187     &tcp_sad_pacing_interval, 2000,
188     "What is the minimum pacing interval for a classified attacker?");
189 
190 int32_t tcp_sad_low_pps = 100;
191 SYSCTL_INT(_net_inet_tcp_sack_attack, OID_AUTO, sad_low_pps,
192     CTLFLAG_RW,
193     &tcp_sad_low_pps, 100,
194     "What is the input pps that below which we do not decay?");
195 #endif
196 uint32_t tcp_ack_war_time_window = 1000;
197 SYSCTL_UINT(_net_inet_tcp, OID_AUTO, ack_war_timewindow,
198     CTLFLAG_RW,
199     &tcp_ack_war_time_window, 1000,
200    "If the tcp_stack does ack-war prevention how many milliseconds are in its time window?");
201 uint32_t tcp_ack_war_cnt = 5;
202 SYSCTL_UINT(_net_inet_tcp, OID_AUTO, ack_war_cnt,
203     CTLFLAG_RW,
204     &tcp_ack_war_cnt, 5,
205    "If the tcp_stack does ack-war prevention how many acks can be sent in its time window?");
206 
207 struct rwlock tcp_function_lock;
208 
209 static int
sysctl_net_inet_tcp_mss_check(SYSCTL_HANDLER_ARGS)210 sysctl_net_inet_tcp_mss_check(SYSCTL_HANDLER_ARGS)
211 {
212 	int error, new;
213 
214 	new = V_tcp_mssdflt;
215 	error = sysctl_handle_int(oidp, &new, 0, req);
216 	if (error == 0 && req->newptr) {
217 		if (new < TCP_MINMSS)
218 			error = EINVAL;
219 		else
220 			V_tcp_mssdflt = new;
221 	}
222 	return (error);
223 }
224 
225 SYSCTL_PROC(_net_inet_tcp, TCPCTL_MSSDFLT, mssdflt,
226     CTLFLAG_VNET | CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_NEEDGIANT,
227     &VNET_NAME(tcp_mssdflt), 0, &sysctl_net_inet_tcp_mss_check, "I",
228     "Default TCP Maximum Segment Size");
229 
230 #ifdef INET6
231 static int
sysctl_net_inet_tcp_mss_v6_check(SYSCTL_HANDLER_ARGS)232 sysctl_net_inet_tcp_mss_v6_check(SYSCTL_HANDLER_ARGS)
233 {
234 	int error, new;
235 
236 	new = V_tcp_v6mssdflt;
237 	error = sysctl_handle_int(oidp, &new, 0, req);
238 	if (error == 0 && req->newptr) {
239 		if (new < TCP_MINMSS)
240 			error = EINVAL;
241 		else
242 			V_tcp_v6mssdflt = new;
243 	}
244 	return (error);
245 }
246 
247 SYSCTL_PROC(_net_inet_tcp, TCPCTL_V6MSSDFLT, v6mssdflt,
248     CTLFLAG_VNET | CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_NEEDGIANT,
249     &VNET_NAME(tcp_v6mssdflt), 0, &sysctl_net_inet_tcp_mss_v6_check, "I",
250    "Default TCP Maximum Segment Size for IPv6");
251 #endif /* INET6 */
252 
253 /*
254  * Minimum MSS we accept and use. This prevents DoS attacks where
255  * we are forced to a ridiculous low MSS like 20 and send hundreds
256  * of packets instead of one. The effect scales with the available
257  * bandwidth and quickly saturates the CPU and network interface
258  * with packet generation and sending. Set to zero to disable MINMSS
259  * checking. This setting prevents us from sending too small packets.
260  */
261 VNET_DEFINE(int, tcp_minmss) = TCP_MINMSS;
262 SYSCTL_INT(_net_inet_tcp, OID_AUTO, minmss, CTLFLAG_VNET | CTLFLAG_RW,
263      &VNET_NAME(tcp_minmss), 0,
264     "Minimum TCP Maximum Segment Size");
265 
266 VNET_DEFINE(int, tcp_do_rfc1323) = 1;
267 SYSCTL_INT(_net_inet_tcp, TCPCTL_DO_RFC1323, rfc1323, CTLFLAG_VNET | CTLFLAG_RW,
268     &VNET_NAME(tcp_do_rfc1323), 0,
269     "Enable rfc1323 (high performance TCP) extensions");
270 
271 /*
272  * As of June 2021, several TCP stacks violate RFC 7323 from September 2014.
273  * Some stacks negotiate TS, but never send them after connection setup. Some
274  * stacks negotiate TS, but don't send them when sending keep-alive segments.
275  * These include modern widely deployed TCP stacks.
276  * Therefore tolerating violations for now...
277  */
278 VNET_DEFINE(int, tcp_tolerate_missing_ts) = 1;
279 SYSCTL_INT(_net_inet_tcp, OID_AUTO, tolerate_missing_ts, CTLFLAG_VNET | CTLFLAG_RW,
280     &VNET_NAME(tcp_tolerate_missing_ts), 0,
281     "Tolerate missing TCP timestamps");
282 
283 VNET_DEFINE(int, tcp_ts_offset_per_conn) = 1;
284 SYSCTL_INT(_net_inet_tcp, OID_AUTO, ts_offset_per_conn, CTLFLAG_VNET | CTLFLAG_RW,
285     &VNET_NAME(tcp_ts_offset_per_conn), 0,
286     "Initialize TCP timestamps per connection instead of per host pair");
287 
288 /* How many connections are pacing */
289 static volatile uint32_t number_of_tcp_connections_pacing = 0;
290 static uint32_t shadow_num_connections = 0;
291 
292 static int tcp_pacing_limit = 10000;
293 SYSCTL_INT(_net_inet_tcp, OID_AUTO, pacing_limit, CTLFLAG_RW,
294     &tcp_pacing_limit, 1000,
295     "If the TCP stack does pacing, is there a limit (-1 = no, 0 = no pacing N = number of connections)");
296 
297 SYSCTL_UINT(_net_inet_tcp, OID_AUTO, pacing_count, CTLFLAG_RD,
298     &shadow_num_connections, 0, "Number of TCP connections being paced");
299 
300 static int	tcp_log_debug = 0;
301 SYSCTL_INT(_net_inet_tcp, OID_AUTO, log_debug, CTLFLAG_RW,
302     &tcp_log_debug, 0, "Log errors caused by incoming TCP segments");
303 
304 static int	tcp_tcbhashsize;
305 SYSCTL_INT(_net_inet_tcp, OID_AUTO, tcbhashsize, CTLFLAG_RDTUN | CTLFLAG_NOFETCH,
306     &tcp_tcbhashsize, 0, "Size of TCP control-block hashtable");
307 
308 static int	do_tcpdrain = 1;
309 SYSCTL_INT(_net_inet_tcp, OID_AUTO, do_tcpdrain, CTLFLAG_RW, &do_tcpdrain, 0,
310     "Enable tcp_drain routine for extra help when low on mbufs");
311 
312 SYSCTL_UINT(_net_inet_tcp, OID_AUTO, pcbcount, CTLFLAG_VNET | CTLFLAG_RD,
313     &VNET_NAME(tcbinfo.ipi_count), 0, "Number of active PCBs");
314 
315 VNET_DEFINE_STATIC(int, icmp_may_rst) = 1;
316 #define	V_icmp_may_rst			VNET(icmp_may_rst)
317 SYSCTL_INT(_net_inet_tcp, OID_AUTO, icmp_may_rst, CTLFLAG_VNET | CTLFLAG_RW,
318     &VNET_NAME(icmp_may_rst), 0,
319     "Certain ICMP unreachable messages may abort connections in SYN_SENT");
320 
321 VNET_DEFINE_STATIC(int, tcp_isn_reseed_interval) = 0;
322 #define	V_tcp_isn_reseed_interval	VNET(tcp_isn_reseed_interval)
323 SYSCTL_INT(_net_inet_tcp, OID_AUTO, isn_reseed_interval, CTLFLAG_VNET | CTLFLAG_RW,
324     &VNET_NAME(tcp_isn_reseed_interval), 0,
325     "Seconds between reseeding of ISN secret");
326 
327 static int	tcp_soreceive_stream;
328 SYSCTL_INT(_net_inet_tcp, OID_AUTO, soreceive_stream, CTLFLAG_RDTUN,
329     &tcp_soreceive_stream, 0, "Using soreceive_stream for TCP sockets");
330 
331 VNET_DEFINE(uma_zone_t, sack_hole_zone);
332 #define	V_sack_hole_zone		VNET(sack_hole_zone)
333 VNET_DEFINE(uint32_t, tcp_map_entries_limit) = 0;	/* unlimited */
334 static int
sysctl_net_inet_tcp_map_limit_check(SYSCTL_HANDLER_ARGS)335 sysctl_net_inet_tcp_map_limit_check(SYSCTL_HANDLER_ARGS)
336 {
337 	int error;
338 	uint32_t new;
339 
340 	new = V_tcp_map_entries_limit;
341 	error = sysctl_handle_int(oidp, &new, 0, req);
342 	if (error == 0 && req->newptr) {
343 		/* only allow "0" and value > minimum */
344 		if (new > 0 && new < TCP_MIN_MAP_ENTRIES_LIMIT)
345 			error = EINVAL;
346 		else
347 			V_tcp_map_entries_limit = new;
348 	}
349 	return (error);
350 }
351 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, map_limit,
352     CTLFLAG_VNET | CTLTYPE_UINT | CTLFLAG_RW | CTLFLAG_NEEDGIANT,
353     &VNET_NAME(tcp_map_entries_limit), 0,
354     &sysctl_net_inet_tcp_map_limit_check, "IU",
355     "Total sendmap entries limit");
356 
357 VNET_DEFINE(uint32_t, tcp_map_split_limit) = 0;	/* unlimited */
358 SYSCTL_UINT(_net_inet_tcp, OID_AUTO, split_limit, CTLFLAG_VNET | CTLFLAG_RW,
359      &VNET_NAME(tcp_map_split_limit), 0,
360     "Total sendmap split entries limit");
361 
362 #ifdef TCP_HHOOK
363 VNET_DEFINE(struct hhook_head *, tcp_hhh[HHOOK_TCP_LAST+1]);
364 #endif
365 
366 #define TS_OFFSET_SECRET_LENGTH SIPHASH_KEY_LENGTH
367 VNET_DEFINE_STATIC(u_char, ts_offset_secret[TS_OFFSET_SECRET_LENGTH]);
368 #define	V_ts_offset_secret	VNET(ts_offset_secret)
369 
370 static int	tcp_default_fb_init(struct tcpcb *tp);
371 static void	tcp_default_fb_fini(struct tcpcb *tp, int tcb_is_purged);
372 static int	tcp_default_handoff_ok(struct tcpcb *tp);
373 static struct inpcb *tcp_notify(struct inpcb *, int);
374 static struct inpcb *tcp_mtudisc_notify(struct inpcb *, int);
375 static void tcp_mtudisc(struct inpcb *, int);
376 static char *	tcp_log_addr(struct in_conninfo *inc, struct tcphdr *th,
377 		    void *ip4hdr, const void *ip6hdr);
378 
379 static struct tcp_function_block tcp_def_funcblk = {
380 	.tfb_tcp_block_name = "freebsd",
381 	.tfb_tcp_output = tcp_output,
382 	.tfb_tcp_do_segment = tcp_do_segment,
383 	.tfb_tcp_ctloutput = tcp_default_ctloutput,
384 	.tfb_tcp_handoff_ok = tcp_default_handoff_ok,
385 	.tfb_tcp_fb_init = tcp_default_fb_init,
386 	.tfb_tcp_fb_fini = tcp_default_fb_fini,
387 };
388 
389 static int tcp_fb_cnt = 0;
390 struct tcp_funchead t_functions;
391 static struct tcp_function_block *tcp_func_set_ptr = &tcp_def_funcblk;
392 
393 static struct tcp_function_block *
find_tcp_functions_locked(struct tcp_function_set * fs)394 find_tcp_functions_locked(struct tcp_function_set *fs)
395 {
396 	struct tcp_function *f;
397 	struct tcp_function_block *blk=NULL;
398 
399 	TAILQ_FOREACH(f, &t_functions, tf_next) {
400 		if (strcmp(f->tf_name, fs->function_set_name) == 0) {
401 			blk = f->tf_fb;
402 			break;
403 		}
404 	}
405 	return(blk);
406 }
407 
408 static struct tcp_function_block *
find_tcp_fb_locked(struct tcp_function_block * blk,struct tcp_function ** s)409 find_tcp_fb_locked(struct tcp_function_block *blk, struct tcp_function **s)
410 {
411 	struct tcp_function_block *rblk=NULL;
412 	struct tcp_function *f;
413 
414 	TAILQ_FOREACH(f, &t_functions, tf_next) {
415 		if (f->tf_fb == blk) {
416 			rblk = blk;
417 			if (s) {
418 				*s = f;
419 			}
420 			break;
421 		}
422 	}
423 	return (rblk);
424 }
425 
426 struct tcp_function_block *
find_and_ref_tcp_functions(struct tcp_function_set * fs)427 find_and_ref_tcp_functions(struct tcp_function_set *fs)
428 {
429 	struct tcp_function_block *blk;
430 
431 	rw_rlock(&tcp_function_lock);
432 	blk = find_tcp_functions_locked(fs);
433 	if (blk)
434 		refcount_acquire(&blk->tfb_refcnt);
435 	rw_runlock(&tcp_function_lock);
436 	return(blk);
437 }
438 
439 struct tcp_function_block *
find_and_ref_tcp_fb(struct tcp_function_block * blk)440 find_and_ref_tcp_fb(struct tcp_function_block *blk)
441 {
442 	struct tcp_function_block *rblk;
443 
444 	rw_rlock(&tcp_function_lock);
445 	rblk = find_tcp_fb_locked(blk, NULL);
446 	if (rblk)
447 		refcount_acquire(&rblk->tfb_refcnt);
448 	rw_runlock(&tcp_function_lock);
449 	return(rblk);
450 }
451 
452 static struct tcp_function_block *
find_and_ref_tcp_default_fb(void)453 find_and_ref_tcp_default_fb(void)
454 {
455 	struct tcp_function_block *rblk;
456 
457 	rw_rlock(&tcp_function_lock);
458 	rblk = tcp_func_set_ptr;
459 	refcount_acquire(&rblk->tfb_refcnt);
460 	rw_runlock(&tcp_function_lock);
461 	return (rblk);
462 }
463 
464 void
tcp_switch_back_to_default(struct tcpcb * tp)465 tcp_switch_back_to_default(struct tcpcb *tp)
466 {
467 	struct tcp_function_block *tfb;
468 
469 	KASSERT(tp->t_fb != &tcp_def_funcblk,
470 	    ("%s: called by the built-in default stack", __func__));
471 
472 	/*
473 	 * Release the old stack. This function will either find a new one
474 	 * or panic.
475 	 */
476 	if (tp->t_fb->tfb_tcp_fb_fini != NULL)
477 		(*tp->t_fb->tfb_tcp_fb_fini)(tp, 0);
478 	refcount_release(&tp->t_fb->tfb_refcnt);
479 
480 	/*
481 	 * Now, we'll find a new function block to use.
482 	 * Start by trying the current user-selected
483 	 * default, unless this stack is the user-selected
484 	 * default.
485 	 */
486 	tfb = find_and_ref_tcp_default_fb();
487 	if (tfb == tp->t_fb) {
488 		refcount_release(&tfb->tfb_refcnt);
489 		tfb = NULL;
490 	}
491 	/* Does the stack accept this connection? */
492 	if (tfb != NULL && tfb->tfb_tcp_handoff_ok != NULL &&
493 	    (*tfb->tfb_tcp_handoff_ok)(tp)) {
494 		refcount_release(&tfb->tfb_refcnt);
495 		tfb = NULL;
496 	}
497 	/* Try to use that stack. */
498 	if (tfb != NULL) {
499 		/* Initialize the new stack. If it succeeds, we are done. */
500 		tp->t_fb = tfb;
501 		if (tp->t_fb->tfb_tcp_fb_init == NULL ||
502 		    (*tp->t_fb->tfb_tcp_fb_init)(tp) == 0)
503 			return;
504 
505 		/*
506 		 * Initialization failed. Release the reference count on
507 		 * the stack.
508 		 */
509 		refcount_release(&tfb->tfb_refcnt);
510 	}
511 
512 	/*
513 	 * If that wasn't feasible, use the built-in default
514 	 * stack which is not allowed to reject anyone.
515 	 */
516 	tfb = find_and_ref_tcp_fb(&tcp_def_funcblk);
517 	if (tfb == NULL) {
518 		/* there always should be a default */
519 		panic("Can't refer to tcp_def_funcblk");
520 	}
521 	if (tfb->tfb_tcp_handoff_ok != NULL) {
522 		if ((*tfb->tfb_tcp_handoff_ok) (tp)) {
523 			/* The default stack cannot say no */
524 			panic("Default stack rejects a new session?");
525 		}
526 	}
527 	tp->t_fb = tfb;
528 	if (tp->t_fb->tfb_tcp_fb_init != NULL &&
529 	    (*tp->t_fb->tfb_tcp_fb_init)(tp)) {
530 		/* The default stack cannot fail */
531 		panic("Default stack initialization failed");
532 	}
533 }
534 
535 static void
tcp_recv_udp_tunneled_packet(struct mbuf * m,int off,struct inpcb * inp,const struct sockaddr * sa,void * ctx)536 tcp_recv_udp_tunneled_packet(struct mbuf *m, int off, struct inpcb *inp,
537     const struct sockaddr *sa, void *ctx)
538 {
539 	struct ip *iph;
540 #ifdef INET6
541 	struct ip6_hdr *ip6;
542 #endif
543 	struct udphdr *uh;
544 	struct tcphdr *th;
545 	int thlen;
546 	uint16_t port;
547 
548 	TCPSTAT_INC(tcps_tunneled_pkts);
549 	if ((m->m_flags & M_PKTHDR) == 0) {
550 		/* Can't handle one that is not a pkt hdr */
551 		TCPSTAT_INC(tcps_tunneled_errs);
552 		goto out;
553 	}
554 	thlen = sizeof(struct tcphdr);
555 	if (m->m_len < off + sizeof(struct udphdr) + thlen &&
556 	    (m =  m_pullup(m, off + sizeof(struct udphdr) + thlen)) == NULL) {
557 		TCPSTAT_INC(tcps_tunneled_errs);
558 		goto out;
559 	}
560 	iph = mtod(m, struct ip *);
561 	uh = (struct udphdr *)((caddr_t)iph + off);
562 	th = (struct tcphdr *)(uh + 1);
563 	thlen = th->th_off << 2;
564 	if (m->m_len < off + sizeof(struct udphdr) + thlen) {
565 		m =  m_pullup(m, off + sizeof(struct udphdr) + thlen);
566 		if (m == NULL) {
567 			TCPSTAT_INC(tcps_tunneled_errs);
568 			goto out;
569 		} else {
570 			iph = mtod(m, struct ip *);
571 			uh = (struct udphdr *)((caddr_t)iph + off);
572 			th = (struct tcphdr *)(uh + 1);
573 		}
574 	}
575 	m->m_pkthdr.tcp_tun_port = port = uh->uh_sport;
576 	bcopy(th, uh, m->m_len - off);
577 	m->m_len -= sizeof(struct udphdr);
578 	m->m_pkthdr.len -= sizeof(struct udphdr);
579 	/*
580 	 * We use the same algorithm for
581 	 * both UDP and TCP for c-sum. So
582 	 * the code in tcp_input will skip
583 	 * the checksum. So we do nothing
584 	 * with the flag (m->m_pkthdr.csum_flags).
585 	 */
586 	switch (iph->ip_v) {
587 #ifdef INET
588 	case IPVERSION:
589 		iph->ip_len = htons(ntohs(iph->ip_len) - sizeof(struct udphdr));
590 		tcp_input_with_port(&m, &off, IPPROTO_TCP, port);
591 		break;
592 #endif
593 #ifdef INET6
594 	case IPV6_VERSION >> 4:
595 		ip6 = mtod(m, struct ip6_hdr *);
596 		ip6->ip6_plen = htons(ntohs(ip6->ip6_plen) - sizeof(struct udphdr));
597 		tcp6_input_with_port(&m, &off, IPPROTO_TCP, port);
598 		break;
599 #endif
600 	default:
601 		goto out;
602 		break;
603 	}
604 	return;
605 out:
606 	m_freem(m);
607 }
608 
609 static int
sysctl_net_inet_default_tcp_functions(SYSCTL_HANDLER_ARGS)610 sysctl_net_inet_default_tcp_functions(SYSCTL_HANDLER_ARGS)
611 {
612 	int error=ENOENT;
613 	struct tcp_function_set fs;
614 	struct tcp_function_block *blk;
615 
616 	memset(&fs, 0, sizeof(fs));
617 	rw_rlock(&tcp_function_lock);
618 	blk = find_tcp_fb_locked(tcp_func_set_ptr, NULL);
619 	if (blk) {
620 		/* Found him */
621 		strcpy(fs.function_set_name, blk->tfb_tcp_block_name);
622 		fs.pcbcnt = blk->tfb_refcnt;
623 	}
624 	rw_runlock(&tcp_function_lock);
625 	error = sysctl_handle_string(oidp, fs.function_set_name,
626 				     sizeof(fs.function_set_name), req);
627 
628 	/* Check for error or no change */
629 	if (error != 0 || req->newptr == NULL)
630 		return(error);
631 
632 	rw_wlock(&tcp_function_lock);
633 	blk = find_tcp_functions_locked(&fs);
634 	if ((blk == NULL) ||
635 	    (blk->tfb_flags & TCP_FUNC_BEING_REMOVED)) {
636 		error = ENOENT;
637 		goto done;
638 	}
639 	tcp_func_set_ptr = blk;
640 done:
641 	rw_wunlock(&tcp_function_lock);
642 	return (error);
643 }
644 
645 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, functions_default,
646     CTLTYPE_STRING | CTLFLAG_RW | CTLFLAG_NEEDGIANT,
647     NULL, 0, sysctl_net_inet_default_tcp_functions, "A",
648     "Set/get the default TCP functions");
649 
650 static int
sysctl_net_inet_list_available(SYSCTL_HANDLER_ARGS)651 sysctl_net_inet_list_available(SYSCTL_HANDLER_ARGS)
652 {
653 	int error, cnt, linesz;
654 	struct tcp_function *f;
655 	char *buffer, *cp;
656 	size_t bufsz, outsz;
657 	bool alias;
658 
659 	cnt = 0;
660 	rw_rlock(&tcp_function_lock);
661 	TAILQ_FOREACH(f, &t_functions, tf_next) {
662 		cnt++;
663 	}
664 	rw_runlock(&tcp_function_lock);
665 
666 	bufsz = (cnt+2) * ((TCP_FUNCTION_NAME_LEN_MAX * 2) + 13) + 1;
667 	buffer = malloc(bufsz, M_TEMP, M_WAITOK);
668 
669 	error = 0;
670 	cp = buffer;
671 
672 	linesz = snprintf(cp, bufsz, "\n%-32s%c %-32s %s\n", "Stack", 'D',
673 	    "Alias", "PCB count");
674 	cp += linesz;
675 	bufsz -= linesz;
676 	outsz = linesz;
677 
678 	rw_rlock(&tcp_function_lock);
679 	TAILQ_FOREACH(f, &t_functions, tf_next) {
680 		alias = (f->tf_name != f->tf_fb->tfb_tcp_block_name);
681 		linesz = snprintf(cp, bufsz, "%-32s%c %-32s %u\n",
682 		    f->tf_fb->tfb_tcp_block_name,
683 		    (f->tf_fb == tcp_func_set_ptr) ? '*' : ' ',
684 		    alias ? f->tf_name : "-",
685 		    f->tf_fb->tfb_refcnt);
686 		if (linesz >= bufsz) {
687 			error = EOVERFLOW;
688 			break;
689 		}
690 		cp += linesz;
691 		bufsz -= linesz;
692 		outsz += linesz;
693 	}
694 	rw_runlock(&tcp_function_lock);
695 	if (error == 0)
696 		error = sysctl_handle_string(oidp, buffer, outsz + 1, req);
697 	free(buffer, M_TEMP);
698 	return (error);
699 }
700 
701 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, functions_available,
702     CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_NEEDGIANT,
703     NULL, 0, sysctl_net_inet_list_available, "A",
704     "list available TCP Function sets");
705 
706 VNET_DEFINE(int, tcp_udp_tunneling_port) = TCP_TUNNELING_PORT_DEFAULT;
707 
708 #ifdef INET
709 VNET_DEFINE(struct socket *, udp4_tun_socket) = NULL;
710 #define	V_udp4_tun_socket	VNET(udp4_tun_socket)
711 #endif
712 #ifdef INET6
713 VNET_DEFINE(struct socket *, udp6_tun_socket) = NULL;
714 #define	V_udp6_tun_socket	VNET(udp6_tun_socket)
715 #endif
716 
717 static void
tcp_over_udp_stop(void)718 tcp_over_udp_stop(void)
719 {
720 	/*
721 	 * This function assumes sysctl caller holds inp_rinfo_lock()
722 	 * for writing!
723 	 */
724 #ifdef INET
725 	if (V_udp4_tun_socket != NULL) {
726 		soclose(V_udp4_tun_socket);
727 		V_udp4_tun_socket = NULL;
728 	}
729 #endif
730 #ifdef INET6
731 	if (V_udp6_tun_socket != NULL) {
732 		soclose(V_udp6_tun_socket);
733 		V_udp6_tun_socket = NULL;
734 	}
735 #endif
736 }
737 
738 static int
tcp_over_udp_start(void)739 tcp_over_udp_start(void)
740 {
741 	uint16_t port;
742 	int ret;
743 #ifdef INET
744 	struct sockaddr_in sin;
745 #endif
746 #ifdef INET6
747 	struct sockaddr_in6 sin6;
748 #endif
749 	/*
750 	 * This function assumes sysctl caller holds inp_info_rlock()
751 	 * for writing!
752 	 */
753 	port = V_tcp_udp_tunneling_port;
754 	if (ntohs(port) == 0) {
755 		/* Must have a port set */
756 		return (EINVAL);
757 	}
758 #ifdef INET
759 	if (V_udp4_tun_socket != NULL) {
760 		/* Already running -- must stop first */
761 		return (EALREADY);
762 	}
763 #endif
764 #ifdef INET6
765 	if (V_udp6_tun_socket != NULL) {
766 		/* Already running -- must stop first */
767 		return (EALREADY);
768 	}
769 #endif
770 #ifdef INET
771 	if ((ret = socreate(PF_INET, &V_udp4_tun_socket,
772 	    SOCK_DGRAM, IPPROTO_UDP,
773 	    curthread->td_ucred, curthread))) {
774 		tcp_over_udp_stop();
775 		return (ret);
776 	}
777 	/* Call the special UDP hook. */
778 	if ((ret = udp_set_kernel_tunneling(V_udp4_tun_socket,
779 	    tcp_recv_udp_tunneled_packet,
780 	    tcp_ctlinput_viaudp,
781 	    NULL))) {
782 		tcp_over_udp_stop();
783 		return (ret);
784 	}
785 	/* Ok, we have a socket, bind it to the port. */
786 	memset(&sin, 0, sizeof(struct sockaddr_in));
787 	sin.sin_len = sizeof(struct sockaddr_in);
788 	sin.sin_family = AF_INET;
789 	sin.sin_port = htons(port);
790 	if ((ret = sobind(V_udp4_tun_socket,
791 	    (struct sockaddr *)&sin, curthread))) {
792 		tcp_over_udp_stop();
793 		return (ret);
794 	}
795 #endif
796 #ifdef INET6
797 	if ((ret = socreate(PF_INET6, &V_udp6_tun_socket,
798 	    SOCK_DGRAM, IPPROTO_UDP,
799 	    curthread->td_ucred, curthread))) {
800 		tcp_over_udp_stop();
801 		return (ret);
802 	}
803 	/* Call the special UDP hook. */
804 	if ((ret = udp_set_kernel_tunneling(V_udp6_tun_socket,
805 	    tcp_recv_udp_tunneled_packet,
806 	    tcp6_ctlinput_viaudp,
807 	    NULL))) {
808 		tcp_over_udp_stop();
809 		return (ret);
810 	}
811 	/* Ok, we have a socket, bind it to the port. */
812 	memset(&sin6, 0, sizeof(struct sockaddr_in6));
813 	sin6.sin6_len = sizeof(struct sockaddr_in6);
814 	sin6.sin6_family = AF_INET6;
815 	sin6.sin6_port = htons(port);
816 	if ((ret = sobind(V_udp6_tun_socket,
817 	    (struct sockaddr *)&sin6, curthread))) {
818 		tcp_over_udp_stop();
819 		return (ret);
820 	}
821 #endif
822 	return (0);
823 }
824 
825 static int
sysctl_net_inet_tcp_udp_tunneling_port_check(SYSCTL_HANDLER_ARGS)826 sysctl_net_inet_tcp_udp_tunneling_port_check(SYSCTL_HANDLER_ARGS)
827 {
828 	int error;
829 	uint32_t old, new;
830 
831 	old = V_tcp_udp_tunneling_port;
832 	new = old;
833 	error = sysctl_handle_int(oidp, &new, 0, req);
834 	if ((error == 0) &&
835 	    (req->newptr != NULL)) {
836 		if ((new < TCP_TUNNELING_PORT_MIN) ||
837 		    (new > TCP_TUNNELING_PORT_MAX)) {
838 			error = EINVAL;
839 		} else {
840 			V_tcp_udp_tunneling_port = new;
841 			if (old != 0) {
842 				tcp_over_udp_stop();
843 			}
844 			if (new != 0) {
845 				error = tcp_over_udp_start();
846 			}
847 		}
848 	}
849 	return (error);
850 }
851 
852 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, udp_tunneling_port,
853     CTLFLAG_VNET | CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_MPSAFE,
854     &VNET_NAME(tcp_udp_tunneling_port),
855     0, &sysctl_net_inet_tcp_udp_tunneling_port_check, "IU",
856     "Tunneling port for tcp over udp");
857 
858 VNET_DEFINE(int, tcp_udp_tunneling_overhead) = TCP_TUNNELING_OVERHEAD_DEFAULT;
859 
860 static int
sysctl_net_inet_tcp_udp_tunneling_overhead_check(SYSCTL_HANDLER_ARGS)861 sysctl_net_inet_tcp_udp_tunneling_overhead_check(SYSCTL_HANDLER_ARGS)
862 {
863 	int error, new;
864 
865 	new = V_tcp_udp_tunneling_overhead;
866 	error = sysctl_handle_int(oidp, &new, 0, req);
867 	if (error == 0 && req->newptr) {
868 		if ((new < TCP_TUNNELING_OVERHEAD_MIN) ||
869 		    (new > TCP_TUNNELING_OVERHEAD_MAX))
870 			error = EINVAL;
871 		else
872 			V_tcp_udp_tunneling_overhead = new;
873 	}
874 	return (error);
875 }
876 
877 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, udp_tunneling_overhead,
878     CTLFLAG_VNET | CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_MPSAFE,
879     &VNET_NAME(tcp_udp_tunneling_overhead),
880     0, &sysctl_net_inet_tcp_udp_tunneling_overhead_check, "IU",
881     "MSS reduction when using tcp over udp");
882 
883 /*
884  * Exports one (struct tcp_function_info) for each alias/name.
885  */
886 static int
sysctl_net_inet_list_func_info(SYSCTL_HANDLER_ARGS)887 sysctl_net_inet_list_func_info(SYSCTL_HANDLER_ARGS)
888 {
889 	int cnt, error;
890 	struct tcp_function *f;
891 	struct tcp_function_info tfi;
892 
893 	/*
894 	 * We don't allow writes.
895 	 */
896 	if (req->newptr != NULL)
897 		return (EINVAL);
898 
899 	/*
900 	 * Wire the old buffer so we can directly copy the functions to
901 	 * user space without dropping the lock.
902 	 */
903 	if (req->oldptr != NULL) {
904 		error = sysctl_wire_old_buffer(req, 0);
905 		if (error)
906 			return (error);
907 	}
908 
909 	/*
910 	 * Walk the list and copy out matching entries. If INVARIANTS
911 	 * is compiled in, also walk the list to verify the length of
912 	 * the list matches what we have recorded.
913 	 */
914 	rw_rlock(&tcp_function_lock);
915 
916 	cnt = 0;
917 #ifndef INVARIANTS
918 	if (req->oldptr == NULL) {
919 		cnt = tcp_fb_cnt;
920 		goto skip_loop;
921 	}
922 #endif
923 	TAILQ_FOREACH(f, &t_functions, tf_next) {
924 #ifdef INVARIANTS
925 		cnt++;
926 #endif
927 		if (req->oldptr != NULL) {
928 			bzero(&tfi, sizeof(tfi));
929 			tfi.tfi_refcnt = f->tf_fb->tfb_refcnt;
930 			tfi.tfi_id = f->tf_fb->tfb_id;
931 			(void)strlcpy(tfi.tfi_alias, f->tf_name,
932 			    sizeof(tfi.tfi_alias));
933 			(void)strlcpy(tfi.tfi_name,
934 			    f->tf_fb->tfb_tcp_block_name, sizeof(tfi.tfi_name));
935 			error = SYSCTL_OUT(req, &tfi, sizeof(tfi));
936 			/*
937 			 * Don't stop on error, as that is the
938 			 * mechanism we use to accumulate length
939 			 * information if the buffer was too short.
940 			 */
941 		}
942 	}
943 	KASSERT(cnt == tcp_fb_cnt,
944 	    ("%s: cnt (%d) != tcp_fb_cnt (%d)", __func__, cnt, tcp_fb_cnt));
945 #ifndef INVARIANTS
946 skip_loop:
947 #endif
948 	rw_runlock(&tcp_function_lock);
949 	if (req->oldptr == NULL)
950 		error = SYSCTL_OUT(req, NULL,
951 		    (cnt + 1) * sizeof(struct tcp_function_info));
952 
953 	return (error);
954 }
955 
956 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, function_info,
957 	    CTLTYPE_OPAQUE | CTLFLAG_SKIP | CTLFLAG_RD | CTLFLAG_MPSAFE,
958 	    NULL, 0, sysctl_net_inet_list_func_info, "S,tcp_function_info",
959 	    "List TCP function block name-to-ID mappings");
960 
961 /*
962  * tfb_tcp_handoff_ok() function for the default stack.
963  * Note that we'll basically try to take all comers.
964  */
965 static int
tcp_default_handoff_ok(struct tcpcb * tp)966 tcp_default_handoff_ok(struct tcpcb *tp)
967 {
968 
969 	return (0);
970 }
971 
972 /*
973  * tfb_tcp_fb_init() function for the default stack.
974  *
975  * This handles making sure we have appropriate timers set if you are
976  * transitioning a socket that has some amount of setup done.
977  *
978  * The init() fuction from the default can *never* return non-zero i.e.
979  * it is required to always succeed since it is the stack of last resort!
980  */
981 static int
tcp_default_fb_init(struct tcpcb * tp)982 tcp_default_fb_init(struct tcpcb *tp)
983 {
984 
985 	struct socket *so;
986 
987 	INP_WLOCK_ASSERT(tp->t_inpcb);
988 
989 	KASSERT(tp->t_state >= 0 && tp->t_state < TCPS_TIME_WAIT,
990 	    ("%s: connection %p in unexpected state %d", __func__, tp,
991 	    tp->t_state));
992 
993 	/*
994 	 * Nothing to do for ESTABLISHED or LISTEN states. And, we don't
995 	 * know what to do for unexpected states (which includes TIME_WAIT).
996 	 */
997 	if (tp->t_state <= TCPS_LISTEN || tp->t_state >= TCPS_TIME_WAIT)
998 		return (0);
999 
1000 	/*
1001 	 * Make sure some kind of transmission timer is set if there is
1002 	 * outstanding data.
1003 	 */
1004 	so = tp->t_inpcb->inp_socket;
1005 	if ((!TCPS_HAVEESTABLISHED(tp->t_state) || sbavail(&so->so_snd) ||
1006 	    tp->snd_una != tp->snd_max) && !(tcp_timer_active(tp, TT_REXMT) ||
1007 	    tcp_timer_active(tp, TT_PERSIST))) {
1008 		/*
1009 		 * If the session has established and it looks like it should
1010 		 * be in the persist state, set the persist timer. Otherwise,
1011 		 * set the retransmit timer.
1012 		 */
1013 		if (TCPS_HAVEESTABLISHED(tp->t_state) && tp->snd_wnd == 0 &&
1014 		    (int32_t)(tp->snd_nxt - tp->snd_una) <
1015 		    (int32_t)sbavail(&so->so_snd))
1016 			tcp_setpersist(tp);
1017 		else
1018 			tcp_timer_activate(tp, TT_REXMT, tp->t_rxtcur);
1019 	}
1020 
1021 	/* All non-embryonic sessions get a keepalive timer. */
1022 	if (!tcp_timer_active(tp, TT_KEEP))
1023 		tcp_timer_activate(tp, TT_KEEP,
1024 		    TCPS_HAVEESTABLISHED(tp->t_state) ? TP_KEEPIDLE(tp) :
1025 		    TP_KEEPINIT(tp));
1026 
1027 	/*
1028 	 * Make sure critical variables are initialized
1029 	 * if transitioning while in Recovery.
1030 	 */
1031 	if IN_FASTRECOVERY(tp->t_flags) {
1032 		if (tp->sackhint.recover_fs == 0)
1033 			tp->sackhint.recover_fs = max(1,
1034 			    tp->snd_nxt - tp->snd_una);
1035 	}
1036 
1037 	return (0);
1038 }
1039 
1040 /*
1041  * tfb_tcp_fb_fini() function for the default stack.
1042  *
1043  * This changes state as necessary (or prudent) to prepare for another stack
1044  * to assume responsibility for the connection.
1045  */
1046 static void
tcp_default_fb_fini(struct tcpcb * tp,int tcb_is_purged)1047 tcp_default_fb_fini(struct tcpcb *tp, int tcb_is_purged)
1048 {
1049 
1050 	INP_WLOCK_ASSERT(tp->t_inpcb);
1051 	return;
1052 }
1053 
1054 /*
1055  * Target size of TCP PCB hash tables. Must be a power of two.
1056  *
1057  * Note that this can be overridden by the kernel environment
1058  * variable net.inet.tcp.tcbhashsize
1059  */
1060 #ifndef TCBHASHSIZE
1061 #define TCBHASHSIZE	0
1062 #endif
1063 
1064 /*
1065  * XXX
1066  * Callouts should be moved into struct tcp directly.  They are currently
1067  * separate because the tcpcb structure is exported to userland for sysctl
1068  * parsing purposes, which do not know about callouts.
1069  */
1070 struct tcpcb_mem {
1071 	struct	tcpcb		tcb;
1072 	struct	tcp_timer	tt;
1073 	struct	cc_var		ccv;
1074 #ifdef TCP_HHOOK
1075 	struct	osd		osd;
1076 #endif
1077 };
1078 
1079 VNET_DEFINE_STATIC(uma_zone_t, tcpcb_zone);
1080 #define	V_tcpcb_zone			VNET(tcpcb_zone)
1081 
1082 MALLOC_DEFINE(M_TCPLOG, "tcplog", "TCP address and flags print buffers");
1083 MALLOC_DEFINE(M_TCPFUNCTIONS, "tcpfunc", "TCP function set memory");
1084 
1085 static struct mtx isn_mtx;
1086 
1087 #define	ISN_LOCK_INIT()	mtx_init(&isn_mtx, "isn_mtx", NULL, MTX_DEF)
1088 #define	ISN_LOCK()	mtx_lock(&isn_mtx)
1089 #define	ISN_UNLOCK()	mtx_unlock(&isn_mtx)
1090 
1091 /*
1092  * TCP initialization.
1093  */
1094 static void
tcp_zone_change(void * tag)1095 tcp_zone_change(void *tag)
1096 {
1097 
1098 	uma_zone_set_max(V_tcbinfo.ipi_zone, maxsockets);
1099 	uma_zone_set_max(V_tcpcb_zone, maxsockets);
1100 	tcp_tw_zone_change();
1101 }
1102 
1103 static int
tcp_inpcb_init(void * mem,int size,int flags)1104 tcp_inpcb_init(void *mem, int size, int flags)
1105 {
1106 	struct inpcb *inp = mem;
1107 
1108 	INP_LOCK_INIT(inp, "inp", "tcpinp");
1109 	return (0);
1110 }
1111 
1112 /*
1113  * Take a value and get the next power of 2 that doesn't overflow.
1114  * Used to size the tcp_inpcb hash buckets.
1115  */
1116 static int
maketcp_hashsize(int size)1117 maketcp_hashsize(int size)
1118 {
1119 	int hashsize;
1120 
1121 	/*
1122 	 * auto tune.
1123 	 * get the next power of 2 higher than maxsockets.
1124 	 */
1125 	hashsize = 1 << fls(size);
1126 	/* catch overflow, and just go one power of 2 smaller */
1127 	if (hashsize < size) {
1128 		hashsize = 1 << (fls(size) - 1);
1129 	}
1130 	return (hashsize);
1131 }
1132 
1133 static volatile int next_tcp_stack_id = 1;
1134 
1135 /*
1136  * Register a TCP function block with the name provided in the names
1137  * array.  (Note that this function does NOT automatically register
1138  * blk->tfb_tcp_block_name as a stack name.  Therefore, you should
1139  * explicitly include blk->tfb_tcp_block_name in the list of names if
1140  * you wish to register the stack with that name.)
1141  *
1142  * Either all name registrations will succeed or all will fail.  If
1143  * a name registration fails, the function will update the num_names
1144  * argument to point to the array index of the name that encountered
1145  * the failure.
1146  *
1147  * Returns 0 on success, or an error code on failure.
1148  */
1149 int
register_tcp_functions_as_names(struct tcp_function_block * blk,int wait,const char * names[],int * num_names)1150 register_tcp_functions_as_names(struct tcp_function_block *blk, int wait,
1151     const char *names[], int *num_names)
1152 {
1153 	struct tcp_function *n;
1154 	struct tcp_function_set fs;
1155 	int error, i;
1156 
1157 	KASSERT(names != NULL && *num_names > 0,
1158 	    ("%s: Called with 0-length name list", __func__));
1159 	KASSERT(names != NULL, ("%s: Called with NULL name list", __func__));
1160 	KASSERT(rw_initialized(&tcp_function_lock),
1161 	    ("%s: called too early", __func__));
1162 
1163 	if ((blk->tfb_tcp_output == NULL) ||
1164 	    (blk->tfb_tcp_do_segment == NULL) ||
1165 	    (blk->tfb_tcp_ctloutput == NULL) ||
1166 	    (strlen(blk->tfb_tcp_block_name) == 0)) {
1167 		/*
1168 		 * These functions are required and you
1169 		 * need a name.
1170 		 */
1171 		*num_names = 0;
1172 		return (EINVAL);
1173 	}
1174 	if (blk->tfb_tcp_timer_stop_all ||
1175 	    blk->tfb_tcp_timer_activate ||
1176 	    blk->tfb_tcp_timer_active ||
1177 	    blk->tfb_tcp_timer_stop) {
1178 		/*
1179 		 * If you define one timer function you
1180 		 * must have them all.
1181 		 */
1182 		if ((blk->tfb_tcp_timer_stop_all == NULL) ||
1183 		    (blk->tfb_tcp_timer_activate == NULL) ||
1184 		    (blk->tfb_tcp_timer_active == NULL) ||
1185 		    (blk->tfb_tcp_timer_stop == NULL)) {
1186 			*num_names = 0;
1187 			return (EINVAL);
1188 		}
1189 	}
1190 
1191 	if (blk->tfb_flags & TCP_FUNC_BEING_REMOVED) {
1192 		*num_names = 0;
1193 		return (EINVAL);
1194 	}
1195 
1196 	refcount_init(&blk->tfb_refcnt, 0);
1197 	blk->tfb_id = atomic_fetchadd_int(&next_tcp_stack_id, 1);
1198 	for (i = 0; i < *num_names; i++) {
1199 		n = malloc(sizeof(struct tcp_function), M_TCPFUNCTIONS, wait);
1200 		if (n == NULL) {
1201 			error = ENOMEM;
1202 			goto cleanup;
1203 		}
1204 		n->tf_fb = blk;
1205 
1206 		(void)strlcpy(fs.function_set_name, names[i],
1207 		    sizeof(fs.function_set_name));
1208 		rw_wlock(&tcp_function_lock);
1209 		if (find_tcp_functions_locked(&fs) != NULL) {
1210 			/* Duplicate name space not allowed */
1211 			rw_wunlock(&tcp_function_lock);
1212 			free(n, M_TCPFUNCTIONS);
1213 			error = EALREADY;
1214 			goto cleanup;
1215 		}
1216 		(void)strlcpy(n->tf_name, names[i], sizeof(n->tf_name));
1217 		TAILQ_INSERT_TAIL(&t_functions, n, tf_next);
1218 		tcp_fb_cnt++;
1219 		rw_wunlock(&tcp_function_lock);
1220 	}
1221 	return(0);
1222 
1223 cleanup:
1224 	/*
1225 	 * Deregister the names we just added. Because registration failed
1226 	 * for names[i], we don't need to deregister that name.
1227 	 */
1228 	*num_names = i;
1229 	rw_wlock(&tcp_function_lock);
1230 	while (--i >= 0) {
1231 		TAILQ_FOREACH(n, &t_functions, tf_next) {
1232 			if (!strncmp(n->tf_name, names[i],
1233 			    TCP_FUNCTION_NAME_LEN_MAX)) {
1234 				TAILQ_REMOVE(&t_functions, n, tf_next);
1235 				tcp_fb_cnt--;
1236 				n->tf_fb = NULL;
1237 				free(n, M_TCPFUNCTIONS);
1238 				break;
1239 			}
1240 		}
1241 	}
1242 	rw_wunlock(&tcp_function_lock);
1243 	return (error);
1244 }
1245 
1246 /*
1247  * Register a TCP function block using the name provided in the name
1248  * argument.
1249  *
1250  * Returns 0 on success, or an error code on failure.
1251  */
1252 int
register_tcp_functions_as_name(struct tcp_function_block * blk,const char * name,int wait)1253 register_tcp_functions_as_name(struct tcp_function_block *blk, const char *name,
1254     int wait)
1255 {
1256 	const char *name_list[1];
1257 	int num_names, rv;
1258 
1259 	num_names = 1;
1260 	if (name != NULL)
1261 		name_list[0] = name;
1262 	else
1263 		name_list[0] = blk->tfb_tcp_block_name;
1264 	rv = register_tcp_functions_as_names(blk, wait, name_list, &num_names);
1265 	return (rv);
1266 }
1267 
1268 /*
1269  * Register a TCP function block using the name defined in
1270  * blk->tfb_tcp_block_name.
1271  *
1272  * Returns 0 on success, or an error code on failure.
1273  */
1274 int
register_tcp_functions(struct tcp_function_block * blk,int wait)1275 register_tcp_functions(struct tcp_function_block *blk, int wait)
1276 {
1277 
1278 	return (register_tcp_functions_as_name(blk, NULL, wait));
1279 }
1280 
1281 /*
1282  * Deregister all names associated with a function block. This
1283  * functionally removes the function block from use within the system.
1284  *
1285  * When called with a true quiesce argument, mark the function block
1286  * as being removed so no more stacks will use it and determine
1287  * whether the removal would succeed.
1288  *
1289  * When called with a false quiesce argument, actually attempt the
1290  * removal.
1291  *
1292  * When called with a force argument, attempt to switch all TCBs to
1293  * use the default stack instead of returning EBUSY.
1294  *
1295  * Returns 0 on success (or if the removal would succeed, or an error
1296  * code on failure.
1297  */
1298 int
deregister_tcp_functions(struct tcp_function_block * blk,bool quiesce,bool force)1299 deregister_tcp_functions(struct tcp_function_block *blk, bool quiesce,
1300     bool force)
1301 {
1302 	struct tcp_function *f;
1303 
1304 	if (blk == &tcp_def_funcblk) {
1305 		/* You can't un-register the default */
1306 		return (EPERM);
1307 	}
1308 	rw_wlock(&tcp_function_lock);
1309 	if (blk == tcp_func_set_ptr) {
1310 		/* You can't free the current default */
1311 		rw_wunlock(&tcp_function_lock);
1312 		return (EBUSY);
1313 	}
1314 	/* Mark the block so no more stacks can use it. */
1315 	blk->tfb_flags |= TCP_FUNC_BEING_REMOVED;
1316 	/*
1317 	 * If TCBs are still attached to the stack, attempt to switch them
1318 	 * to the default stack.
1319 	 */
1320 	if (force && blk->tfb_refcnt) {
1321 		struct inpcb *inp;
1322 		struct tcpcb *tp;
1323 		VNET_ITERATOR_DECL(vnet_iter);
1324 
1325 		rw_wunlock(&tcp_function_lock);
1326 
1327 		VNET_LIST_RLOCK();
1328 		VNET_FOREACH(vnet_iter) {
1329 			CURVNET_SET(vnet_iter);
1330 			INP_INFO_WLOCK(&V_tcbinfo);
1331 			CK_LIST_FOREACH(inp, V_tcbinfo.ipi_listhead, inp_list) {
1332 				INP_WLOCK(inp);
1333 				if (inp->inp_flags & INP_TIMEWAIT) {
1334 					INP_WUNLOCK(inp);
1335 					continue;
1336 				}
1337 				tp = intotcpcb(inp);
1338 				if (tp == NULL || tp->t_fb != blk) {
1339 					INP_WUNLOCK(inp);
1340 					continue;
1341 				}
1342 				tcp_switch_back_to_default(tp);
1343 				INP_WUNLOCK(inp);
1344 			}
1345 			INP_INFO_WUNLOCK(&V_tcbinfo);
1346 			CURVNET_RESTORE();
1347 		}
1348 		VNET_LIST_RUNLOCK();
1349 
1350 		rw_wlock(&tcp_function_lock);
1351 	}
1352 	if (blk->tfb_refcnt) {
1353 		/* TCBs still attached. */
1354 		rw_wunlock(&tcp_function_lock);
1355 		return (EBUSY);
1356 	}
1357 	if (quiesce) {
1358 		/* Skip removal. */
1359 		rw_wunlock(&tcp_function_lock);
1360 		return (0);
1361 	}
1362 	/* Remove any function names that map to this function block. */
1363 	while (find_tcp_fb_locked(blk, &f) != NULL) {
1364 		TAILQ_REMOVE(&t_functions, f, tf_next);
1365 		tcp_fb_cnt--;
1366 		f->tf_fb = NULL;
1367 		free(f, M_TCPFUNCTIONS);
1368 	}
1369 	rw_wunlock(&tcp_function_lock);
1370 	return (0);
1371 }
1372 
1373 void
tcp_init(void)1374 tcp_init(void)
1375 {
1376 	const char *tcbhash_tuneable;
1377 	int hashsize;
1378 
1379 	tcbhash_tuneable = "net.inet.tcp.tcbhashsize";
1380 
1381 #ifdef TCP_HHOOK
1382 	if (hhook_head_register(HHOOK_TYPE_TCP, HHOOK_TCP_EST_IN,
1383 	    &V_tcp_hhh[HHOOK_TCP_EST_IN], HHOOK_NOWAIT|HHOOK_HEADISINVNET) != 0)
1384 		printf("%s: WARNING: unable to register helper hook\n", __func__);
1385 	if (hhook_head_register(HHOOK_TYPE_TCP, HHOOK_TCP_EST_OUT,
1386 	    &V_tcp_hhh[HHOOK_TCP_EST_OUT], HHOOK_NOWAIT|HHOOK_HEADISINVNET) != 0)
1387 		printf("%s: WARNING: unable to register helper hook\n", __func__);
1388 #endif
1389 #ifdef STATS
1390 	if (tcp_stats_init())
1391 		printf("%s: WARNING: unable to initialise TCP stats\n",
1392 		    __func__);
1393 #endif
1394 	hashsize = TCBHASHSIZE;
1395 	TUNABLE_INT_FETCH(tcbhash_tuneable, &hashsize);
1396 	if (hashsize == 0) {
1397 		/*
1398 		 * Auto tune the hash size based on maxsockets.
1399 		 * A perfect hash would have a 1:1 mapping
1400 		 * (hashsize = maxsockets) however it's been
1401 		 * suggested that O(2) average is better.
1402 		 */
1403 		hashsize = maketcp_hashsize(maxsockets / 4);
1404 		/*
1405 		 * Our historical default is 512,
1406 		 * do not autotune lower than this.
1407 		 */
1408 		if (hashsize < 512)
1409 			hashsize = 512;
1410 		if (bootverbose && IS_DEFAULT_VNET(curvnet))
1411 			printf("%s: %s auto tuned to %d\n", __func__,
1412 			    tcbhash_tuneable, hashsize);
1413 	}
1414 	/*
1415 	 * We require a hashsize to be a power of two.
1416 	 * Previously if it was not a power of two we would just reset it
1417 	 * back to 512, which could be a nasty surprise if you did not notice
1418 	 * the error message.
1419 	 * Instead what we do is clip it to the closest power of two lower
1420 	 * than the specified hash value.
1421 	 */
1422 	if (!powerof2(hashsize)) {
1423 		int oldhashsize = hashsize;
1424 
1425 		hashsize = maketcp_hashsize(hashsize);
1426 		/* prevent absurdly low value */
1427 		if (hashsize < 16)
1428 			hashsize = 16;
1429 		printf("%s: WARNING: TCB hash size not a power of 2, "
1430 		    "clipped from %d to %d.\n", __func__, oldhashsize,
1431 		    hashsize);
1432 	}
1433 	in_pcbinfo_init(&V_tcbinfo, "tcp", &V_tcb, hashsize, hashsize,
1434 	    "tcp_inpcb", tcp_inpcb_init, IPI_HASHFIELDS_4TUPLE);
1435 
1436 	/*
1437 	 * These have to be type stable for the benefit of the timers.
1438 	 */
1439 	V_tcpcb_zone = uma_zcreate("tcpcb", sizeof(struct tcpcb_mem),
1440 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
1441 	uma_zone_set_max(V_tcpcb_zone, maxsockets);
1442 	uma_zone_set_warning(V_tcpcb_zone, "kern.ipc.maxsockets limit reached");
1443 
1444 	tcp_tw_init();
1445 	syncache_init();
1446 	tcp_hc_init();
1447 
1448 	TUNABLE_INT_FETCH("net.inet.tcp.sack.enable", &V_tcp_do_sack);
1449 	V_sack_hole_zone = uma_zcreate("sackhole", sizeof(struct sackhole),
1450 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
1451 
1452 	tcp_fastopen_init();
1453 
1454 	V_tcp_msl = TCPTV_MSL;
1455 
1456 	/* Skip initialization of globals for non-default instances. */
1457 	if (!IS_DEFAULT_VNET(curvnet))
1458 		return;
1459 
1460 	tcp_reass_global_init();
1461 
1462 	/* XXX virtualize those below? */
1463 	tcp_delacktime = TCPTV_DELACK;
1464 	tcp_keepinit = TCPTV_KEEP_INIT;
1465 	tcp_keepidle = TCPTV_KEEP_IDLE;
1466 	tcp_keepintvl = TCPTV_KEEPINTVL;
1467 	tcp_maxpersistidle = TCPTV_KEEP_IDLE;
1468 	tcp_rexmit_initial = TCPTV_RTOBASE;
1469 	if (tcp_rexmit_initial < 1)
1470 		tcp_rexmit_initial = 1;
1471 	tcp_rexmit_min = TCPTV_MIN;
1472 	if (tcp_rexmit_min < 1)
1473 		tcp_rexmit_min = 1;
1474 	tcp_persmin = TCPTV_PERSMIN;
1475 	tcp_persmax = TCPTV_PERSMAX;
1476 	tcp_rexmit_slop = TCPTV_CPU_VAR;
1477 	tcp_finwait2_timeout = TCPTV_FINWAIT2_TIMEOUT;
1478 	tcp_tcbhashsize = hashsize;
1479 
1480 	/* Setup the tcp function block list */
1481 	TAILQ_INIT(&t_functions);
1482 	rw_init(&tcp_function_lock, "tcp_func_lock");
1483 	register_tcp_functions(&tcp_def_funcblk, M_WAITOK);
1484 #ifdef TCP_BLACKBOX
1485 	/* Initialize the TCP logging data. */
1486 	tcp_log_init();
1487 #endif
1488 	arc4rand(&V_ts_offset_secret, sizeof(V_ts_offset_secret), 0);
1489 
1490 	if (tcp_soreceive_stream) {
1491 #ifdef INET
1492 		tcp_usrreqs.pru_soreceive = soreceive_stream;
1493 #endif
1494 #ifdef INET6
1495 		tcp6_usrreqs.pru_soreceive = soreceive_stream;
1496 #endif /* INET6 */
1497 	}
1498 
1499 #ifdef INET6
1500 #define TCP_MINPROTOHDR (sizeof(struct ip6_hdr) + sizeof(struct tcphdr))
1501 #else /* INET6 */
1502 #define TCP_MINPROTOHDR (sizeof(struct tcpiphdr))
1503 #endif /* INET6 */
1504 	if (max_protohdr < TCP_MINPROTOHDR)
1505 		max_protohdr = TCP_MINPROTOHDR;
1506 	if (max_linkhdr + TCP_MINPROTOHDR > MHLEN)
1507 		panic("tcp_init");
1508 #undef TCP_MINPROTOHDR
1509 
1510 	ISN_LOCK_INIT();
1511 	EVENTHANDLER_REGISTER(shutdown_pre_sync, tcp_fini, NULL,
1512 		SHUTDOWN_PRI_DEFAULT);
1513 	EVENTHANDLER_REGISTER(maxsockets_change, tcp_zone_change, NULL,
1514 		EVENTHANDLER_PRI_ANY);
1515 
1516 	tcp_inp_lro_direct_queue = counter_u64_alloc(M_WAITOK);
1517 	tcp_inp_lro_wokeup_queue = counter_u64_alloc(M_WAITOK);
1518 	tcp_inp_lro_compressed = counter_u64_alloc(M_WAITOK);
1519 	tcp_inp_lro_locks_taken = counter_u64_alloc(M_WAITOK);
1520 	tcp_extra_mbuf = counter_u64_alloc(M_WAITOK);
1521 	tcp_would_have_but = counter_u64_alloc(M_WAITOK);
1522 	tcp_comp_total = counter_u64_alloc(M_WAITOK);
1523 	tcp_uncomp_total = counter_u64_alloc(M_WAITOK);
1524 #ifdef TCPPCAP
1525 	tcp_pcap_init();
1526 #endif
1527 }
1528 
1529 #ifdef VIMAGE
1530 static void
tcp_destroy(void * unused __unused)1531 tcp_destroy(void *unused __unused)
1532 {
1533 	int n;
1534 #ifdef TCP_HHOOK
1535 	int error;
1536 #endif
1537 
1538 	/*
1539 	 * All our processes are gone, all our sockets should be cleaned
1540 	 * up, which means, we should be past the tcp_discardcb() calls.
1541 	 * Sleep to let all tcpcb timers really disappear and cleanup.
1542 	 */
1543 	for (;;) {
1544 		INP_LIST_RLOCK(&V_tcbinfo);
1545 		n = V_tcbinfo.ipi_count;
1546 		INP_LIST_RUNLOCK(&V_tcbinfo);
1547 		if (n == 0)
1548 			break;
1549 		pause("tcpdes", hz / 10);
1550 	}
1551 	tcp_hc_destroy();
1552 	syncache_destroy();
1553 	tcp_tw_destroy();
1554 	in_pcbinfo_destroy(&V_tcbinfo);
1555 	/* tcp_discardcb() clears the sack_holes up. */
1556 	uma_zdestroy(V_sack_hole_zone);
1557 	uma_zdestroy(V_tcpcb_zone);
1558 
1559 	/*
1560 	 * Cannot free the zone until all tcpcbs are released as we attach
1561 	 * the allocations to them.
1562 	 */
1563 	tcp_fastopen_destroy();
1564 
1565 #ifdef TCP_HHOOK
1566 	error = hhook_head_deregister(V_tcp_hhh[HHOOK_TCP_EST_IN]);
1567 	if (error != 0) {
1568 		printf("%s: WARNING: unable to deregister helper hook "
1569 		    "type=%d, id=%d: error %d returned\n", __func__,
1570 		    HHOOK_TYPE_TCP, HHOOK_TCP_EST_IN, error);
1571 	}
1572 	error = hhook_head_deregister(V_tcp_hhh[HHOOK_TCP_EST_OUT]);
1573 	if (error != 0) {
1574 		printf("%s: WARNING: unable to deregister helper hook "
1575 		    "type=%d, id=%d: error %d returned\n", __func__,
1576 		    HHOOK_TYPE_TCP, HHOOK_TCP_EST_OUT, error);
1577 	}
1578 #endif
1579 }
1580 VNET_SYSUNINIT(tcp, SI_SUB_PROTO_DOMAIN, SI_ORDER_FOURTH, tcp_destroy, NULL);
1581 #endif
1582 
1583 void
tcp_fini(void * xtp)1584 tcp_fini(void *xtp)
1585 {
1586 
1587 }
1588 
1589 /*
1590  * Fill in the IP and TCP headers for an outgoing packet, given the tcpcb.
1591  * tcp_template used to store this data in mbufs, but we now recopy it out
1592  * of the tcpcb each time to conserve mbufs.
1593  */
1594 void
tcpip_fillheaders(struct inpcb * inp,uint16_t port,void * ip_ptr,void * tcp_ptr)1595 tcpip_fillheaders(struct inpcb *inp, uint16_t port, void *ip_ptr, void *tcp_ptr)
1596 {
1597 	struct tcphdr *th = (struct tcphdr *)tcp_ptr;
1598 
1599 	INP_WLOCK_ASSERT(inp);
1600 
1601 #ifdef INET6
1602 	if ((inp->inp_vflag & INP_IPV6) != 0) {
1603 		struct ip6_hdr *ip6;
1604 
1605 		ip6 = (struct ip6_hdr *)ip_ptr;
1606 		ip6->ip6_flow = (ip6->ip6_flow & ~IPV6_FLOWINFO_MASK) |
1607 			(inp->inp_flow & IPV6_FLOWINFO_MASK);
1608 		ip6->ip6_vfc = (ip6->ip6_vfc & ~IPV6_VERSION_MASK) |
1609 			(IPV6_VERSION & IPV6_VERSION_MASK);
1610 		if (port == 0)
1611 			ip6->ip6_nxt = IPPROTO_TCP;
1612 		else
1613 			ip6->ip6_nxt = IPPROTO_UDP;
1614 		ip6->ip6_plen = htons(sizeof(struct tcphdr));
1615 		ip6->ip6_src = inp->in6p_laddr;
1616 		ip6->ip6_dst = inp->in6p_faddr;
1617 	}
1618 #endif /* INET6 */
1619 #if defined(INET6) && defined(INET)
1620 	else
1621 #endif
1622 #ifdef INET
1623 	{
1624 		struct ip *ip;
1625 
1626 		ip = (struct ip *)ip_ptr;
1627 		ip->ip_v = IPVERSION;
1628 		ip->ip_hl = 5;
1629 		ip->ip_tos = inp->inp_ip_tos;
1630 		ip->ip_len = 0;
1631 		ip->ip_id = 0;
1632 		ip->ip_off = 0;
1633 		ip->ip_ttl = inp->inp_ip_ttl;
1634 		ip->ip_sum = 0;
1635 		if (port == 0)
1636 			ip->ip_p = IPPROTO_TCP;
1637 		else
1638 			ip->ip_p = IPPROTO_UDP;
1639 		ip->ip_src = inp->inp_laddr;
1640 		ip->ip_dst = inp->inp_faddr;
1641 	}
1642 #endif /* INET */
1643 	th->th_sport = inp->inp_lport;
1644 	th->th_dport = inp->inp_fport;
1645 	th->th_seq = 0;
1646 	th->th_ack = 0;
1647 	th->th_x2 = 0;
1648 	th->th_off = 5;
1649 	th->th_flags = 0;
1650 	th->th_win = 0;
1651 	th->th_urp = 0;
1652 	th->th_sum = 0;		/* in_pseudo() is called later for ipv4 */
1653 }
1654 
1655 /*
1656  * Create template to be used to send tcp packets on a connection.
1657  * Allocates an mbuf and fills in a skeletal tcp/ip header.  The only
1658  * use for this function is in keepalives, which use tcp_respond.
1659  */
1660 struct tcptemp *
tcpip_maketemplate(struct inpcb * inp)1661 tcpip_maketemplate(struct inpcb *inp)
1662 {
1663 	struct tcptemp *t;
1664 
1665 	t = malloc(sizeof(*t), M_TEMP, M_NOWAIT);
1666 	if (t == NULL)
1667 		return (NULL);
1668 	tcpip_fillheaders(inp, 0, (void *)&t->tt_ipgen, (void *)&t->tt_t);
1669 	return (t);
1670 }
1671 
1672 /*
1673  * Send a single message to the TCP at address specified by
1674  * the given TCP/IP header.  If m == NULL, then we make a copy
1675  * of the tcpiphdr at th and send directly to the addressed host.
1676  * This is used to force keep alive messages out using the TCP
1677  * template for a connection.  If flags are given then we send
1678  * a message back to the TCP which originated the segment th,
1679  * and discard the mbuf containing it and any other attached mbufs.
1680  *
1681  * In any case the ack and sequence number of the transmitted
1682  * segment are as specified by the parameters.
1683  *
1684  * NOTE: If m != NULL, then th must point to *inside* the mbuf.
1685  */
1686 void
tcp_respond(struct tcpcb * tp,void * ipgen,struct tcphdr * th,struct mbuf * m,tcp_seq ack,tcp_seq seq,int flags)1687 tcp_respond(struct tcpcb *tp, void *ipgen, struct tcphdr *th, struct mbuf *m,
1688     tcp_seq ack, tcp_seq seq, int flags)
1689 {
1690 	struct tcpopt to;
1691 	struct inpcb *inp;
1692 	struct ip *ip;
1693 	struct mbuf *optm;
1694 	struct udphdr *uh = NULL;
1695 	struct tcphdr *nth;
1696 	u_char *optp;
1697 #ifdef INET6
1698 	struct ip6_hdr *ip6;
1699 	int isipv6;
1700 #endif /* INET6 */
1701 	int optlen, tlen, win, ulen;
1702 	bool incl_opts;
1703 	uint16_t port;
1704 
1705 	KASSERT(tp != NULL || m != NULL, ("tcp_respond: tp and m both NULL"));
1706 	NET_EPOCH_ASSERT();
1707 
1708 #ifdef INET6
1709 	isipv6 = ((struct ip *)ipgen)->ip_v == (IPV6_VERSION >> 4);
1710 	ip6 = ipgen;
1711 #endif /* INET6 */
1712 	ip = ipgen;
1713 
1714 	if (tp != NULL) {
1715 		inp = tp->t_inpcb;
1716 		KASSERT(inp != NULL, ("tcp control block w/o inpcb"));
1717 		INP_WLOCK_ASSERT(inp);
1718 	} else
1719 		inp = NULL;
1720 
1721 	if (m != NULL) {
1722 #ifdef INET6
1723 		if (isipv6 && ip6 && (ip6->ip6_nxt == IPPROTO_UDP))
1724 			port = m->m_pkthdr.tcp_tun_port;
1725 		else
1726 #endif
1727 		if (ip && (ip->ip_p == IPPROTO_UDP))
1728 			port = m->m_pkthdr.tcp_tun_port;
1729 		else
1730 			port = 0;
1731 	} else
1732 		port = tp->t_port;
1733 
1734 	incl_opts = false;
1735 	win = 0;
1736 	if (tp != NULL) {
1737 		if (!(flags & TH_RST)) {
1738 			win = sbspace(&inp->inp_socket->so_rcv);
1739 			if (win > TCP_MAXWIN << tp->rcv_scale)
1740 				win = TCP_MAXWIN << tp->rcv_scale;
1741 		}
1742 		if ((tp->t_flags & TF_NOOPT) == 0)
1743 			incl_opts = true;
1744 	}
1745 	if (m == NULL) {
1746 		m = m_gethdr(M_NOWAIT, MT_DATA);
1747 		if (m == NULL)
1748 			return;
1749 		m->m_data += max_linkhdr;
1750 #ifdef INET6
1751 		if (isipv6) {
1752 			bcopy((caddr_t)ip6, mtod(m, caddr_t),
1753 			      sizeof(struct ip6_hdr));
1754 			ip6 = mtod(m, struct ip6_hdr *);
1755 			nth = (struct tcphdr *)(ip6 + 1);
1756 			if (port) {
1757 				/* Insert a UDP header */
1758 				uh = (struct udphdr *)nth;
1759 				uh->uh_sport = htons(V_tcp_udp_tunneling_port);
1760 				uh->uh_dport = port;
1761 				nth = (struct tcphdr *)(uh + 1);
1762 			}
1763 		} else
1764 #endif /* INET6 */
1765 		{
1766 			bcopy((caddr_t)ip, mtod(m, caddr_t), sizeof(struct ip));
1767 			ip = mtod(m, struct ip *);
1768 			nth = (struct tcphdr *)(ip + 1);
1769 			if (port) {
1770 				/* Insert a UDP header */
1771 				uh = (struct udphdr *)nth;
1772 				uh->uh_sport = htons(V_tcp_udp_tunneling_port);
1773 				uh->uh_dport = port;
1774 				nth = (struct tcphdr *)(uh + 1);
1775 			}
1776 		}
1777 		bcopy((caddr_t)th, (caddr_t)nth, sizeof(struct tcphdr));
1778 		flags = TH_ACK;
1779 	} else if ((!M_WRITABLE(m)) || (port != 0)) {
1780 		struct mbuf *n;
1781 
1782 		/* Can't reuse 'm', allocate a new mbuf. */
1783 		n = m_gethdr(M_NOWAIT, MT_DATA);
1784 		if (n == NULL) {
1785 			m_freem(m);
1786 			return;
1787 		}
1788 
1789 		if (!m_dup_pkthdr(n, m, M_NOWAIT)) {
1790 			m_freem(m);
1791 			m_freem(n);
1792 			return;
1793 		}
1794 
1795 		n->m_data += max_linkhdr;
1796 		/* m_len is set later */
1797 #define xchg(a,b,type) { type t; t=a; a=b; b=t; }
1798 #ifdef INET6
1799 		if (isipv6) {
1800 			bcopy((caddr_t)ip6, mtod(n, caddr_t),
1801 			      sizeof(struct ip6_hdr));
1802 			ip6 = mtod(n, struct ip6_hdr *);
1803 			xchg(ip6->ip6_dst, ip6->ip6_src, struct in6_addr);
1804 			nth = (struct tcphdr *)(ip6 + 1);
1805 			if (port) {
1806 				/* Insert a UDP header */
1807 				uh = (struct udphdr *)nth;
1808 				uh->uh_sport = htons(V_tcp_udp_tunneling_port);
1809 				uh->uh_dport = port;
1810 				nth = (struct tcphdr *)(uh + 1);
1811 			}
1812 		} else
1813 #endif /* INET6 */
1814 		{
1815 			bcopy((caddr_t)ip, mtod(n, caddr_t), sizeof(struct ip));
1816 			ip = mtod(n, struct ip *);
1817 			xchg(ip->ip_dst.s_addr, ip->ip_src.s_addr, uint32_t);
1818 			nth = (struct tcphdr *)(ip + 1);
1819 			if (port) {
1820 				/* Insert a UDP header */
1821 				uh = (struct udphdr *)nth;
1822 				uh->uh_sport = htons(V_tcp_udp_tunneling_port);
1823 				uh->uh_dport = port;
1824 				nth = (struct tcphdr *)(uh + 1);
1825 			}
1826 		}
1827 		bcopy((caddr_t)th, (caddr_t)nth, sizeof(struct tcphdr));
1828 		xchg(nth->th_dport, nth->th_sport, uint16_t);
1829 		th = nth;
1830 		m_freem(m);
1831 		m = n;
1832 	} else {
1833 		/*
1834 		 *  reuse the mbuf.
1835 		 * XXX MRT We inherit the FIB, which is lucky.
1836 		 */
1837 		m_freem(m->m_next);
1838 		m->m_next = NULL;
1839 		m->m_data = (caddr_t)ipgen;
1840 		/* m_len is set later */
1841 #ifdef INET6
1842 		if (isipv6) {
1843 			xchg(ip6->ip6_dst, ip6->ip6_src, struct in6_addr);
1844 			nth = (struct tcphdr *)(ip6 + 1);
1845 		} else
1846 #endif /* INET6 */
1847 		{
1848 			xchg(ip->ip_dst.s_addr, ip->ip_src.s_addr, uint32_t);
1849 			nth = (struct tcphdr *)(ip + 1);
1850 		}
1851 		if (th != nth) {
1852 			/*
1853 			 * this is usually a case when an extension header
1854 			 * exists between the IPv6 header and the
1855 			 * TCP header.
1856 			 */
1857 			nth->th_sport = th->th_sport;
1858 			nth->th_dport = th->th_dport;
1859 		}
1860 		xchg(nth->th_dport, nth->th_sport, uint16_t);
1861 #undef xchg
1862 	}
1863 	tlen = 0;
1864 #ifdef INET6
1865 	if (isipv6)
1866 		tlen = sizeof (struct ip6_hdr) + sizeof (struct tcphdr);
1867 #endif
1868 #if defined(INET) && defined(INET6)
1869 	else
1870 #endif
1871 #ifdef INET
1872 		tlen = sizeof (struct tcpiphdr);
1873 #endif
1874 	if (port)
1875 		tlen += sizeof (struct udphdr);
1876 #ifdef INVARIANTS
1877 	m->m_len = 0;
1878 	KASSERT(M_TRAILINGSPACE(m) >= tlen,
1879 	    ("Not enough trailing space for message (m=%p, need=%d, have=%ld)",
1880 	    m, tlen, (long)M_TRAILINGSPACE(m)));
1881 #endif
1882 	m->m_len = tlen;
1883 	to.to_flags = 0;
1884 	if (incl_opts) {
1885 		/* Make sure we have room. */
1886 		if (M_TRAILINGSPACE(m) < TCP_MAXOLEN) {
1887 			m->m_next = m_get(M_NOWAIT, MT_DATA);
1888 			if (m->m_next) {
1889 				optp = mtod(m->m_next, u_char *);
1890 				optm = m->m_next;
1891 			} else
1892 				incl_opts = false;
1893 		} else {
1894 			optp = (u_char *) (nth + 1);
1895 			optm = m;
1896 		}
1897 	}
1898 	if (incl_opts) {
1899 		/* Timestamps. */
1900 		if (tp->t_flags & TF_RCVD_TSTMP) {
1901 			to.to_tsval = tcp_ts_getticks() + tp->ts_offset;
1902 			to.to_tsecr = tp->ts_recent;
1903 			to.to_flags |= TOF_TS;
1904 		}
1905 #if defined(IPSEC_SUPPORT) || defined(TCP_SIGNATURE)
1906 		/* TCP-MD5 (RFC2385). */
1907 		if (tp->t_flags & TF_SIGNATURE)
1908 			to.to_flags |= TOF_SIGNATURE;
1909 #endif
1910 		/* Add the options. */
1911 		tlen += optlen = tcp_addoptions(&to, optp);
1912 
1913 		/* Update m_len in the correct mbuf. */
1914 		optm->m_len += optlen;
1915 	} else
1916 		optlen = 0;
1917 #ifdef INET6
1918 	if (isipv6) {
1919 		if (uh) {
1920 			ulen = tlen - sizeof(struct ip6_hdr);
1921 			uh->uh_ulen = htons(ulen);
1922 		}
1923 		ip6->ip6_flow = 0;
1924 		ip6->ip6_vfc = IPV6_VERSION;
1925 		if (port)
1926 			ip6->ip6_nxt = IPPROTO_UDP;
1927 		else
1928 			ip6->ip6_nxt = IPPROTO_TCP;
1929 		ip6->ip6_plen = htons(tlen - sizeof(*ip6));
1930 	}
1931 #endif
1932 #if defined(INET) && defined(INET6)
1933 	else
1934 #endif
1935 #ifdef INET
1936 	{
1937 		if (uh) {
1938 			ulen = tlen - sizeof(struct ip);
1939 			uh->uh_ulen = htons(ulen);
1940 		}
1941 		ip->ip_len = htons(tlen);
1942 		ip->ip_ttl = V_ip_defttl;
1943 		if (port) {
1944 			ip->ip_p = IPPROTO_UDP;
1945 		} else {
1946 			ip->ip_p = IPPROTO_TCP;
1947 		}
1948 		if (V_path_mtu_discovery)
1949 			ip->ip_off |= htons(IP_DF);
1950 	}
1951 #endif
1952 	m->m_pkthdr.len = tlen;
1953 	m->m_pkthdr.rcvif = NULL;
1954 #ifdef MAC
1955 	if (inp != NULL) {
1956 		/*
1957 		 * Packet is associated with a socket, so allow the
1958 		 * label of the response to reflect the socket label.
1959 		 */
1960 		INP_WLOCK_ASSERT(inp);
1961 		mac_inpcb_create_mbuf(inp, m);
1962 	} else {
1963 		/*
1964 		 * Packet is not associated with a socket, so possibly
1965 		 * update the label in place.
1966 		 */
1967 		mac_netinet_tcp_reply(m);
1968 	}
1969 #endif
1970 	nth->th_seq = htonl(seq);
1971 	nth->th_ack = htonl(ack);
1972 	nth->th_x2 = 0;
1973 	nth->th_off = (sizeof (struct tcphdr) + optlen) >> 2;
1974 	nth->th_flags = flags;
1975 	if (tp != NULL)
1976 		nth->th_win = htons((u_short) (win >> tp->rcv_scale));
1977 	else
1978 		nth->th_win = htons((u_short)win);
1979 	nth->th_urp = 0;
1980 
1981 #if defined(IPSEC_SUPPORT) || defined(TCP_SIGNATURE)
1982 	if (to.to_flags & TOF_SIGNATURE) {
1983 		if (!TCPMD5_ENABLED() ||
1984 		    TCPMD5_OUTPUT(m, nth, to.to_signature) != 0) {
1985 			m_freem(m);
1986 			return;
1987 		}
1988 	}
1989 #endif
1990 
1991 #ifdef INET6
1992 	if (isipv6) {
1993 		if (port) {
1994 			m->m_pkthdr.csum_flags = CSUM_UDP_IPV6;
1995 			m->m_pkthdr.csum_data = offsetof(struct udphdr, uh_sum);
1996 			uh->uh_sum = in6_cksum_pseudo(ip6, ulen, IPPROTO_UDP, 0);
1997 			nth->th_sum = 0;
1998 		} else {
1999 			m->m_pkthdr.csum_flags = CSUM_TCP_IPV6;
2000 			m->m_pkthdr.csum_data = offsetof(struct tcphdr, th_sum);
2001 			nth->th_sum = in6_cksum_pseudo(ip6,
2002 			    tlen - sizeof(struct ip6_hdr), IPPROTO_TCP, 0);
2003 		}
2004 		ip6->ip6_hlim = in6_selecthlim(tp != NULL ? tp->t_inpcb :
2005 		    NULL, NULL);
2006 	}
2007 #endif /* INET6 */
2008 #if defined(INET6) && defined(INET)
2009 	else
2010 #endif
2011 #ifdef INET
2012 	{
2013 		if (port) {
2014 			uh->uh_sum = in_pseudo(ip->ip_src.s_addr, ip->ip_dst.s_addr,
2015 			    htons(ulen + IPPROTO_UDP));
2016 			m->m_pkthdr.csum_flags = CSUM_UDP;
2017 			m->m_pkthdr.csum_data = offsetof(struct udphdr, uh_sum);
2018 			nth->th_sum = 0;
2019 		} else {
2020 			m->m_pkthdr.csum_flags = CSUM_TCP;
2021 			m->m_pkthdr.csum_data = offsetof(struct tcphdr, th_sum);
2022 			nth->th_sum = in_pseudo(ip->ip_src.s_addr, ip->ip_dst.s_addr,
2023 			    htons((u_short)(tlen - sizeof(struct ip) + ip->ip_p)));
2024 		}
2025 	}
2026 #endif /* INET */
2027 #ifdef TCPDEBUG
2028 	if (tp == NULL || (inp->inp_socket->so_options & SO_DEBUG))
2029 		tcp_trace(TA_OUTPUT, 0, tp, mtod(m, void *), th, 0);
2030 #endif
2031 	TCP_PROBE3(debug__output, tp, th, m);
2032 	if (flags & TH_RST)
2033 		TCP_PROBE5(accept__refused, NULL, NULL, m, tp, nth);
2034 
2035 #ifdef INET6
2036 	if (isipv6) {
2037 		TCP_PROBE5(send, NULL, tp, ip6, tp, nth);
2038 		(void)ip6_output(m, NULL, NULL, 0, NULL, NULL, inp);
2039 	}
2040 #endif /* INET6 */
2041 #if defined(INET) && defined(INET6)
2042 	else
2043 #endif
2044 #ifdef INET
2045 	{
2046 		TCP_PROBE5(send, NULL, tp, ip, tp, nth);
2047 		(void)ip_output(m, NULL, NULL, 0, NULL, inp);
2048 	}
2049 #endif
2050 }
2051 
2052 /*
2053  * Create a new TCP control block, making an
2054  * empty reassembly queue and hooking it to the argument
2055  * protocol control block.  The `inp' parameter must have
2056  * come from the zone allocator set up in tcp_init().
2057  */
2058 struct tcpcb *
tcp_newtcpcb(struct inpcb * inp)2059 tcp_newtcpcb(struct inpcb *inp)
2060 {
2061 	struct tcpcb_mem *tm;
2062 	struct tcpcb *tp;
2063 #ifdef INET6
2064 	int isipv6 = (inp->inp_vflag & INP_IPV6) != 0;
2065 #endif /* INET6 */
2066 
2067 	tm = uma_zalloc(V_tcpcb_zone, M_NOWAIT | M_ZERO);
2068 	if (tm == NULL)
2069 		return (NULL);
2070 	tp = &tm->tcb;
2071 
2072 	/* Initialise cc_var struct for this tcpcb. */
2073 	tp->ccv = &tm->ccv;
2074 	tp->ccv->type = IPPROTO_TCP;
2075 	tp->ccv->ccvc.tcp = tp;
2076 	rw_rlock(&tcp_function_lock);
2077 	tp->t_fb = tcp_func_set_ptr;
2078 	refcount_acquire(&tp->t_fb->tfb_refcnt);
2079 	rw_runlock(&tcp_function_lock);
2080 	/*
2081 	 * Use the current system default CC algorithm.
2082 	 */
2083 	CC_LIST_RLOCK();
2084 	KASSERT(!STAILQ_EMPTY(&cc_list), ("cc_list is empty!"));
2085 	CC_ALGO(tp) = CC_DEFAULT();
2086 	CC_LIST_RUNLOCK();
2087 	/*
2088 	 * The tcpcb will hold a reference on its inpcb until tcp_discardcb()
2089 	 * is called.
2090 	 */
2091 	in_pcbref(inp);	/* Reference for tcpcb */
2092 	tp->t_inpcb = inp;
2093 
2094 	if (CC_ALGO(tp)->cb_init != NULL)
2095 		if (CC_ALGO(tp)->cb_init(tp->ccv) > 0) {
2096 			if (tp->t_fb->tfb_tcp_fb_fini)
2097 				(*tp->t_fb->tfb_tcp_fb_fini)(tp, 1);
2098 			in_pcbrele_wlocked(inp);
2099 			refcount_release(&tp->t_fb->tfb_refcnt);
2100 			uma_zfree(V_tcpcb_zone, tm);
2101 			return (NULL);
2102 		}
2103 
2104 #ifdef TCP_HHOOK
2105 	tp->osd = &tm->osd;
2106 	if (khelp_init_osd(HELPER_CLASS_TCP, tp->osd)) {
2107 		if (tp->t_fb->tfb_tcp_fb_fini)
2108 			(*tp->t_fb->tfb_tcp_fb_fini)(tp, 1);
2109 		in_pcbrele_wlocked(inp);
2110 		refcount_release(&tp->t_fb->tfb_refcnt);
2111 		uma_zfree(V_tcpcb_zone, tm);
2112 		return (NULL);
2113 	}
2114 #endif
2115 
2116 #ifdef VIMAGE
2117 	tp->t_vnet = inp->inp_vnet;
2118 #endif
2119 	tp->t_timers = &tm->tt;
2120 	TAILQ_INIT(&tp->t_segq);
2121 	tp->t_maxseg =
2122 #ifdef INET6
2123 		isipv6 ? V_tcp_v6mssdflt :
2124 #endif /* INET6 */
2125 		V_tcp_mssdflt;
2126 
2127 	/* Set up our timeouts. */
2128 	callout_init(&tp->t_timers->tt_rexmt, 1);
2129 	callout_init(&tp->t_timers->tt_persist, 1);
2130 	callout_init(&tp->t_timers->tt_keep, 1);
2131 	callout_init(&tp->t_timers->tt_2msl, 1);
2132 	callout_init(&tp->t_timers->tt_delack, 1);
2133 
2134 	if (V_tcp_do_rfc1323)
2135 		tp->t_flags = (TF_REQ_SCALE|TF_REQ_TSTMP);
2136 	if (V_tcp_do_sack)
2137 		tp->t_flags |= TF_SACK_PERMIT;
2138 	TAILQ_INIT(&tp->snd_holes);
2139 
2140 	/*
2141 	 * Init srtt to TCPTV_SRTTBASE (0), so we can tell that we have no
2142 	 * rtt estimate.  Set rttvar so that srtt + 4 * rttvar gives
2143 	 * reasonable initial retransmit time.
2144 	 */
2145 	tp->t_srtt = TCPTV_SRTTBASE;
2146 	tp->t_rttvar = ((tcp_rexmit_initial - TCPTV_SRTTBASE) << TCP_RTTVAR_SHIFT) / 4;
2147 	tp->t_rttmin = tcp_rexmit_min;
2148 	tp->t_rxtcur = tcp_rexmit_initial;
2149 	tp->snd_cwnd = TCP_MAXWIN << TCP_MAX_WINSHIFT;
2150 	tp->snd_ssthresh = TCP_MAXWIN << TCP_MAX_WINSHIFT;
2151 	tp->t_rcvtime = ticks;
2152 	/*
2153 	 * IPv4 TTL initialization is necessary for an IPv6 socket as well,
2154 	 * because the socket may be bound to an IPv6 wildcard address,
2155 	 * which may match an IPv4-mapped IPv6 address.
2156 	 */
2157 	inp->inp_ip_ttl = V_ip_defttl;
2158 	inp->inp_ppcb = tp;
2159 #ifdef TCPPCAP
2160 	/*
2161 	 * Init the TCP PCAP queues.
2162 	 */
2163 	tcp_pcap_tcpcb_init(tp);
2164 #endif
2165 #ifdef TCP_BLACKBOX
2166 	/* Initialize the per-TCPCB log data. */
2167 	tcp_log_tcpcbinit(tp);
2168 #endif
2169 	tp->t_pacing_rate = -1;
2170 	if (tp->t_fb->tfb_tcp_fb_init) {
2171 		if ((*tp->t_fb->tfb_tcp_fb_init)(tp)) {
2172 			refcount_release(&tp->t_fb->tfb_refcnt);
2173 			in_pcbrele_wlocked(inp);
2174 			uma_zfree(V_tcpcb_zone, tm);
2175 			return (NULL);
2176 		}
2177 	}
2178 #ifdef STATS
2179 	if (V_tcp_perconn_stats_enable == 1)
2180 		tp->t_stats = stats_blob_alloc(V_tcp_perconn_stats_dflt_tpl, 0);
2181 #endif
2182 	return (tp);		/* XXX */
2183 }
2184 
2185 /*
2186  * Switch the congestion control algorithm back to NewReno for any active
2187  * control blocks using an algorithm which is about to go away.
2188  * This ensures the CC framework can allow the unload to proceed without leaving
2189  * any dangling pointers which would trigger a panic.
2190  * Returning non-zero would inform the CC framework that something went wrong
2191  * and it would be unsafe to allow the unload to proceed. However, there is no
2192  * way for this to occur with this implementation so we always return zero.
2193  */
2194 int
tcp_ccalgounload(struct cc_algo * unload_algo)2195 tcp_ccalgounload(struct cc_algo *unload_algo)
2196 {
2197 	struct cc_algo *tmpalgo;
2198 	struct inpcb *inp;
2199 	struct tcpcb *tp;
2200 	VNET_ITERATOR_DECL(vnet_iter);
2201 
2202 	/*
2203 	 * Check all active control blocks across all network stacks and change
2204 	 * any that are using "unload_algo" back to NewReno. If "unload_algo"
2205 	 * requires cleanup code to be run, call it.
2206 	 */
2207 	VNET_LIST_RLOCK();
2208 	VNET_FOREACH(vnet_iter) {
2209 		CURVNET_SET(vnet_iter);
2210 		INP_INFO_WLOCK(&V_tcbinfo);
2211 		/*
2212 		 * New connections already part way through being initialised
2213 		 * with the CC algo we're removing will not race with this code
2214 		 * because the INP_INFO_WLOCK is held during initialisation. We
2215 		 * therefore don't enter the loop below until the connection
2216 		 * list has stabilised.
2217 		 */
2218 		CK_LIST_FOREACH(inp, &V_tcb, inp_list) {
2219 			INP_WLOCK(inp);
2220 			/* Important to skip tcptw structs. */
2221 			if (!(inp->inp_flags & INP_TIMEWAIT) &&
2222 			    (tp = intotcpcb(inp)) != NULL) {
2223 				/*
2224 				 * By holding INP_WLOCK here, we are assured
2225 				 * that the connection is not currently
2226 				 * executing inside the CC module's functions
2227 				 * i.e. it is safe to make the switch back to
2228 				 * NewReno.
2229 				 */
2230 				if (CC_ALGO(tp) == unload_algo) {
2231 					tmpalgo = CC_ALGO(tp);
2232 					if (tmpalgo->cb_destroy != NULL)
2233 						tmpalgo->cb_destroy(tp->ccv);
2234 					CC_DATA(tp) = NULL;
2235 					/*
2236 					 * NewReno may allocate memory on
2237 					 * demand for certain stateful
2238 					 * configuration as needed, but is
2239 					 * coded to never fail on memory
2240 					 * allocation failure so it is a safe
2241 					 * fallback.
2242 					 */
2243 					CC_ALGO(tp) = &newreno_cc_algo;
2244 				}
2245 			}
2246 			INP_WUNLOCK(inp);
2247 		}
2248 		INP_INFO_WUNLOCK(&V_tcbinfo);
2249 		CURVNET_RESTORE();
2250 	}
2251 	VNET_LIST_RUNLOCK();
2252 
2253 	return (0);
2254 }
2255 
2256 /*
2257  * Drop a TCP connection, reporting
2258  * the specified error.  If connection is synchronized,
2259  * then send a RST to peer.
2260  */
2261 struct tcpcb *
tcp_drop(struct tcpcb * tp,int errno)2262 tcp_drop(struct tcpcb *tp, int errno)
2263 {
2264 	struct socket *so = tp->t_inpcb->inp_socket;
2265 
2266 	NET_EPOCH_ASSERT();
2267 	INP_INFO_LOCK_ASSERT(&V_tcbinfo);
2268 	INP_WLOCK_ASSERT(tp->t_inpcb);
2269 
2270 	if (TCPS_HAVERCVDSYN(tp->t_state)) {
2271 		tcp_state_change(tp, TCPS_CLOSED);
2272 		(void) tp->t_fb->tfb_tcp_output(tp);
2273 		TCPSTAT_INC(tcps_drops);
2274 	} else
2275 		TCPSTAT_INC(tcps_conndrops);
2276 	if (errno == ETIMEDOUT && tp->t_softerror)
2277 		errno = tp->t_softerror;
2278 	so->so_error = errno;
2279 	return (tcp_close(tp));
2280 }
2281 
2282 void
tcp_discardcb(struct tcpcb * tp)2283 tcp_discardcb(struct tcpcb *tp)
2284 {
2285 	struct inpcb *inp = tp->t_inpcb;
2286 	struct socket *so = inp->inp_socket;
2287 #ifdef INET6
2288 	int isipv6 = (inp->inp_vflag & INP_IPV6) != 0;
2289 #endif /* INET6 */
2290 	int released __unused;
2291 
2292 	INP_WLOCK_ASSERT(inp);
2293 
2294 	/*
2295 	 * Make sure that all of our timers are stopped before we delete the
2296 	 * PCB.
2297 	 *
2298 	 * If stopping a timer fails, we schedule a discard function in same
2299 	 * callout, and the last discard function called will take care of
2300 	 * deleting the tcpcb.
2301 	 */
2302 	tp->t_timers->tt_draincnt = 0;
2303 	tcp_timer_stop(tp, TT_REXMT);
2304 	tcp_timer_stop(tp, TT_PERSIST);
2305 	tcp_timer_stop(tp, TT_KEEP);
2306 	tcp_timer_stop(tp, TT_2MSL);
2307 	tcp_timer_stop(tp, TT_DELACK);
2308 	if (tp->t_fb->tfb_tcp_timer_stop_all) {
2309 		/*
2310 		 * Call the stop-all function of the methods,
2311 		 * this function should call the tcp_timer_stop()
2312 		 * method with each of the function specific timeouts.
2313 		 * That stop will be called via the tfb_tcp_timer_stop()
2314 		 * which should use the async drain function of the
2315 		 * callout system (see tcp_var.h).
2316 		 */
2317 		tp->t_fb->tfb_tcp_timer_stop_all(tp);
2318 	}
2319 
2320 	/* free the reassembly queue, if any */
2321 	tcp_reass_flush(tp);
2322 
2323 #ifdef TCP_OFFLOAD
2324 	/* Disconnect offload device, if any. */
2325 	if (tp->t_flags & TF_TOE)
2326 		tcp_offload_detach(tp);
2327 #endif
2328 
2329 	tcp_free_sackholes(tp);
2330 
2331 #ifdef TCPPCAP
2332 	/* Free the TCP PCAP queues. */
2333 	tcp_pcap_drain(&(tp->t_inpkts));
2334 	tcp_pcap_drain(&(tp->t_outpkts));
2335 #endif
2336 
2337 	/* Allow the CC algorithm to clean up after itself. */
2338 	if (CC_ALGO(tp)->cb_destroy != NULL)
2339 		CC_ALGO(tp)->cb_destroy(tp->ccv);
2340 	CC_DATA(tp) = NULL;
2341 
2342 #ifdef TCP_HHOOK
2343 	khelp_destroy_osd(tp->osd);
2344 #endif
2345 #ifdef STATS
2346 	stats_blob_destroy(tp->t_stats);
2347 #endif
2348 
2349 	CC_ALGO(tp) = NULL;
2350 	inp->inp_ppcb = NULL;
2351 	if (tp->t_timers->tt_draincnt == 0) {
2352 		/* We own the last reference on tcpcb, let's free it. */
2353 #ifdef TCP_BLACKBOX
2354 		tcp_log_tcpcbfini(tp);
2355 #endif
2356 		TCPSTATES_DEC(tp->t_state);
2357 		if (tp->t_fb->tfb_tcp_fb_fini)
2358 			(*tp->t_fb->tfb_tcp_fb_fini)(tp, 1);
2359 
2360 		/*
2361 		 * If we got enough samples through the srtt filter,
2362 		 * save the rtt and rttvar in the routing entry.
2363 		 * 'Enough' is arbitrarily defined as 4 rtt samples.
2364 		 * 4 samples is enough for the srtt filter to converge
2365 		 * to within enough % of the correct value; fewer samples
2366 		 * and we could save a bogus rtt. The danger is not high
2367 		 * as tcp quickly recovers from everything.
2368 		 * XXX: Works very well but needs some more statistics!
2369 		 *
2370 		 * XXXRRS: Updating must be after the stack fini() since
2371 		 * that may be converting some internal representation of
2372 		 * say srtt etc into the general one used by other stacks.
2373 		 * Lets also at least protect against the so being NULL
2374 		 * as RW stated below.
2375 		 */
2376 		if ((tp->t_rttupdated >= 4) && (so != NULL)) {
2377 			struct hc_metrics_lite metrics;
2378 			uint32_t ssthresh;
2379 
2380 			bzero(&metrics, sizeof(metrics));
2381 			/*
2382 			 * Update the ssthresh always when the conditions below
2383 			 * are satisfied. This gives us better new start value
2384 			 * for the congestion avoidance for new connections.
2385 			 * ssthresh is only set if packet loss occurred on a session.
2386 			 *
2387 			 * XXXRW: 'so' may be NULL here, and/or socket buffer may be
2388 			 * being torn down.  Ideally this code would not use 'so'.
2389 			 */
2390 			ssthresh = tp->snd_ssthresh;
2391 			if (ssthresh != 0 && ssthresh < so->so_snd.sb_hiwat / 2) {
2392 				/*
2393 				 * convert the limit from user data bytes to
2394 				 * packets then to packet data bytes.
2395 				 */
2396 				ssthresh = (ssthresh + tp->t_maxseg / 2) / tp->t_maxseg;
2397 				if (ssthresh < 2)
2398 					ssthresh = 2;
2399 				ssthresh *= (tp->t_maxseg +
2400 #ifdef INET6
2401 					     (isipv6 ? sizeof (struct ip6_hdr) +
2402 					      sizeof (struct tcphdr) :
2403 #endif
2404 					      sizeof (struct tcpiphdr)
2405 #ifdef INET6
2406 						     )
2407 #endif
2408 					);
2409 			} else
2410 				ssthresh = 0;
2411 			metrics.rmx_ssthresh = ssthresh;
2412 
2413 			metrics.rmx_rtt = tp->t_srtt;
2414 			metrics.rmx_rttvar = tp->t_rttvar;
2415 			metrics.rmx_cwnd = tp->snd_cwnd;
2416 			metrics.rmx_sendpipe = 0;
2417 			metrics.rmx_recvpipe = 0;
2418 
2419 			tcp_hc_update(&inp->inp_inc, &metrics);
2420 		}
2421 		refcount_release(&tp->t_fb->tfb_refcnt);
2422 		tp->t_inpcb = NULL;
2423 		uma_zfree(V_tcpcb_zone, tp);
2424 		released = in_pcbrele_wlocked(inp);
2425 		KASSERT(!released, ("%s: inp %p should not have been released "
2426 			"here", __func__, inp));
2427 	}
2428 }
2429 
2430 void
tcp_timer_discard(void * ptp)2431 tcp_timer_discard(void *ptp)
2432 {
2433 	struct inpcb *inp;
2434 	struct tcpcb *tp;
2435 	struct epoch_tracker et;
2436 
2437 	tp = (struct tcpcb *)ptp;
2438 	CURVNET_SET(tp->t_vnet);
2439 	NET_EPOCH_ENTER(et);
2440 	inp = tp->t_inpcb;
2441 	KASSERT(inp != NULL, ("%s: tp %p tp->t_inpcb == NULL",
2442 		__func__, tp));
2443 	INP_WLOCK(inp);
2444 	KASSERT((tp->t_timers->tt_flags & TT_STOPPED) != 0,
2445 		("%s: tcpcb has to be stopped here", __func__));
2446 	tp->t_timers->tt_draincnt--;
2447 	if (tp->t_timers->tt_draincnt == 0) {
2448 		/* We own the last reference on this tcpcb, let's free it. */
2449 #ifdef TCP_BLACKBOX
2450 		tcp_log_tcpcbfini(tp);
2451 #endif
2452 		TCPSTATES_DEC(tp->t_state);
2453 		if (tp->t_fb->tfb_tcp_fb_fini)
2454 			(*tp->t_fb->tfb_tcp_fb_fini)(tp, 1);
2455 		refcount_release(&tp->t_fb->tfb_refcnt);
2456 		tp->t_inpcb = NULL;
2457 		uma_zfree(V_tcpcb_zone, tp);
2458 		if (in_pcbrele_wlocked(inp)) {
2459 			NET_EPOCH_EXIT(et);
2460 			CURVNET_RESTORE();
2461 			return;
2462 		}
2463 	}
2464 	INP_WUNLOCK(inp);
2465 	NET_EPOCH_EXIT(et);
2466 	CURVNET_RESTORE();
2467 }
2468 
2469 /*
2470  * Attempt to close a TCP control block, marking it as dropped, and freeing
2471  * the socket if we hold the only reference.
2472  */
2473 struct tcpcb *
tcp_close(struct tcpcb * tp)2474 tcp_close(struct tcpcb *tp)
2475 {
2476 	struct inpcb *inp = tp->t_inpcb;
2477 	struct socket *so;
2478 
2479 	INP_INFO_LOCK_ASSERT(&V_tcbinfo);
2480 	INP_WLOCK_ASSERT(inp);
2481 
2482 #ifdef TCP_OFFLOAD
2483 	if (tp->t_state == TCPS_LISTEN)
2484 		tcp_offload_listen_stop(tp);
2485 #endif
2486 	/*
2487 	 * This releases the TFO pending counter resource for TFO listen
2488 	 * sockets as well as passively-created TFO sockets that transition
2489 	 * from SYN_RECEIVED to CLOSED.
2490 	 */
2491 	if (tp->t_tfo_pending) {
2492 		tcp_fastopen_decrement_counter(tp->t_tfo_pending);
2493 		tp->t_tfo_pending = NULL;
2494 	}
2495 	in_pcbdrop(inp);
2496 	TCPSTAT_INC(tcps_closed);
2497 	if (tp->t_state != TCPS_CLOSED)
2498 		tcp_state_change(tp, TCPS_CLOSED);
2499 	KASSERT(inp->inp_socket != NULL, ("tcp_close: inp_socket NULL"));
2500 	so = inp->inp_socket;
2501 	soisdisconnected(so);
2502 	if (inp->inp_flags & INP_SOCKREF) {
2503 		KASSERT(so->so_state & SS_PROTOREF,
2504 		    ("tcp_close: !SS_PROTOREF"));
2505 		inp->inp_flags &= ~INP_SOCKREF;
2506 		INP_WUNLOCK(inp);
2507 		SOCK_LOCK(so);
2508 		so->so_state &= ~SS_PROTOREF;
2509 		sofree(so);
2510 		return (NULL);
2511 	}
2512 	return (tp);
2513 }
2514 
2515 void
tcp_drain(void)2516 tcp_drain(void)
2517 {
2518 	VNET_ITERATOR_DECL(vnet_iter);
2519 
2520 	if (!do_tcpdrain)
2521 		return;
2522 
2523 	VNET_LIST_RLOCK_NOSLEEP();
2524 	VNET_FOREACH(vnet_iter) {
2525 		CURVNET_SET(vnet_iter);
2526 		struct inpcb *inpb;
2527 		struct tcpcb *tcpb;
2528 
2529 	/*
2530 	 * Walk the tcpbs, if existing, and flush the reassembly queue,
2531 	 * if there is one...
2532 	 * XXX: The "Net/3" implementation doesn't imply that the TCP
2533 	 *      reassembly queue should be flushed, but in a situation
2534 	 *	where we're really low on mbufs, this is potentially
2535 	 *	useful.
2536 	 */
2537 		INP_INFO_WLOCK(&V_tcbinfo);
2538 		CK_LIST_FOREACH(inpb, V_tcbinfo.ipi_listhead, inp_list) {
2539 			INP_WLOCK(inpb);
2540 			if (inpb->inp_flags & INP_TIMEWAIT) {
2541 				INP_WUNLOCK(inpb);
2542 				continue;
2543 			}
2544 			if ((tcpb = intotcpcb(inpb)) != NULL) {
2545 				tcp_reass_flush(tcpb);
2546 				tcp_clean_sackreport(tcpb);
2547 #ifdef TCP_BLACKBOX
2548 				tcp_log_drain(tcpb);
2549 #endif
2550 #ifdef TCPPCAP
2551 				if (tcp_pcap_aggressive_free) {
2552 					/* Free the TCP PCAP queues. */
2553 					tcp_pcap_drain(&(tcpb->t_inpkts));
2554 					tcp_pcap_drain(&(tcpb->t_outpkts));
2555 				}
2556 #endif
2557 			}
2558 			INP_WUNLOCK(inpb);
2559 		}
2560 		INP_INFO_WUNLOCK(&V_tcbinfo);
2561 		CURVNET_RESTORE();
2562 	}
2563 	VNET_LIST_RUNLOCK_NOSLEEP();
2564 }
2565 
2566 /*
2567  * Notify a tcp user of an asynchronous error;
2568  * store error as soft error, but wake up user
2569  * (for now, won't do anything until can select for soft error).
2570  *
2571  * Do not wake up user since there currently is no mechanism for
2572  * reporting soft errors (yet - a kqueue filter may be added).
2573  */
2574 static struct inpcb *
tcp_notify(struct inpcb * inp,int error)2575 tcp_notify(struct inpcb *inp, int error)
2576 {
2577 	struct tcpcb *tp;
2578 
2579 	INP_INFO_LOCK_ASSERT(&V_tcbinfo);
2580 	INP_WLOCK_ASSERT(inp);
2581 
2582 	if ((inp->inp_flags & INP_TIMEWAIT) ||
2583 	    (inp->inp_flags & INP_DROPPED))
2584 		return (inp);
2585 
2586 	tp = intotcpcb(inp);
2587 	KASSERT(tp != NULL, ("tcp_notify: tp == NULL"));
2588 
2589 	/*
2590 	 * Ignore some errors if we are hooked up.
2591 	 * If connection hasn't completed, has retransmitted several times,
2592 	 * and receives a second error, give up now.  This is better
2593 	 * than waiting a long time to establish a connection that
2594 	 * can never complete.
2595 	 */
2596 	if (tp->t_state == TCPS_ESTABLISHED &&
2597 	    (error == EHOSTUNREACH || error == ENETUNREACH ||
2598 	     error == EHOSTDOWN)) {
2599 		if (inp->inp_route.ro_nh) {
2600 			NH_FREE(inp->inp_route.ro_nh);
2601 			inp->inp_route.ro_nh = (struct nhop_object *)NULL;
2602 		}
2603 		return (inp);
2604 	} else if (tp->t_state < TCPS_ESTABLISHED && tp->t_rxtshift > 3 &&
2605 	    tp->t_softerror) {
2606 		tp = tcp_drop(tp, error);
2607 		if (tp != NULL)
2608 			return (inp);
2609 		else
2610 			return (NULL);
2611 	} else {
2612 		tp->t_softerror = error;
2613 		return (inp);
2614 	}
2615 #if 0
2616 	wakeup( &so->so_timeo);
2617 	sorwakeup(so);
2618 	sowwakeup(so);
2619 #endif
2620 }
2621 
2622 static int
tcp_pcblist(SYSCTL_HANDLER_ARGS)2623 tcp_pcblist(SYSCTL_HANDLER_ARGS)
2624 {
2625 	struct epoch_tracker et;
2626 	struct inpcb *inp;
2627 	struct xinpgen xig;
2628 	int error;
2629 
2630 	if (req->newptr != NULL)
2631 		return (EPERM);
2632 
2633 	if (req->oldptr == NULL) {
2634 		int n;
2635 
2636 		n = V_tcbinfo.ipi_count +
2637 		    counter_u64_fetch(V_tcps_states[TCPS_SYN_RECEIVED]);
2638 		n += imax(n / 8, 10);
2639 		req->oldidx = 2 * (sizeof xig) + n * sizeof(struct xtcpcb);
2640 		return (0);
2641 	}
2642 
2643 	if ((error = sysctl_wire_old_buffer(req, 0)) != 0)
2644 		return (error);
2645 
2646 	bzero(&xig, sizeof(xig));
2647 	xig.xig_len = sizeof xig;
2648 	xig.xig_count = V_tcbinfo.ipi_count +
2649 	    counter_u64_fetch(V_tcps_states[TCPS_SYN_RECEIVED]);
2650 	xig.xig_gen = V_tcbinfo.ipi_gencnt;
2651 	xig.xig_sogen = so_gencnt;
2652 	error = SYSCTL_OUT(req, &xig, sizeof xig);
2653 	if (error)
2654 		return (error);
2655 
2656 	error = syncache_pcblist(req);
2657 	if (error)
2658 		return (error);
2659 
2660 	NET_EPOCH_ENTER(et);
2661 	for (inp = CK_LIST_FIRST(V_tcbinfo.ipi_listhead);
2662 	    inp != NULL;
2663 	    inp = CK_LIST_NEXT(inp, inp_list)) {
2664 		INP_RLOCK(inp);
2665 		if (inp->inp_gencnt <= xig.xig_gen) {
2666 			int crerr;
2667 
2668 			/*
2669 			 * XXX: This use of cr_cansee(), introduced with
2670 			 * TCP state changes, is not quite right, but for
2671 			 * now, better than nothing.
2672 			 */
2673 			if (inp->inp_flags & INP_TIMEWAIT) {
2674 				if (intotw(inp) != NULL)
2675 					crerr = cr_cansee(req->td->td_ucred,
2676 					    intotw(inp)->tw_cred);
2677 				else
2678 					crerr = EINVAL;	/* Skip this inp. */
2679 			} else
2680 				crerr = cr_canseeinpcb(req->td->td_ucred, inp);
2681 			if (crerr == 0) {
2682 				struct xtcpcb xt;
2683 
2684 				tcp_inptoxtp(inp, &xt);
2685 				INP_RUNLOCK(inp);
2686 				error = SYSCTL_OUT(req, &xt, sizeof xt);
2687 				if (error)
2688 					break;
2689 				else
2690 					continue;
2691 			}
2692 		}
2693 		INP_RUNLOCK(inp);
2694 	}
2695 	NET_EPOCH_EXIT(et);
2696 
2697 	if (!error) {
2698 		/*
2699 		 * Give the user an updated idea of our state.
2700 		 * If the generation differs from what we told
2701 		 * her before, she knows that something happened
2702 		 * while we were processing this request, and it
2703 		 * might be necessary to retry.
2704 		 */
2705 		xig.xig_gen = V_tcbinfo.ipi_gencnt;
2706 		xig.xig_sogen = so_gencnt;
2707 		xig.xig_count = V_tcbinfo.ipi_count +
2708 		    counter_u64_fetch(V_tcps_states[TCPS_SYN_RECEIVED]);
2709 		error = SYSCTL_OUT(req, &xig, sizeof xig);
2710 	}
2711 
2712 	return (error);
2713 }
2714 
2715 SYSCTL_PROC(_net_inet_tcp, TCPCTL_PCBLIST, pcblist,
2716     CTLTYPE_OPAQUE | CTLFLAG_RD | CTLFLAG_NEEDGIANT,
2717     NULL, 0, tcp_pcblist, "S,xtcpcb",
2718     "List of active TCP connections");
2719 
2720 #ifdef INET
2721 static int
tcp_getcred(SYSCTL_HANDLER_ARGS)2722 tcp_getcred(SYSCTL_HANDLER_ARGS)
2723 {
2724 	struct xucred xuc;
2725 	struct sockaddr_in addrs[2];
2726 	struct epoch_tracker et;
2727 	struct inpcb *inp;
2728 	int error;
2729 
2730 	error = priv_check(req->td, PRIV_NETINET_GETCRED);
2731 	if (error)
2732 		return (error);
2733 	error = SYSCTL_IN(req, addrs, sizeof(addrs));
2734 	if (error)
2735 		return (error);
2736 	NET_EPOCH_ENTER(et);
2737 	inp = in_pcblookup(&V_tcbinfo, addrs[1].sin_addr, addrs[1].sin_port,
2738 	    addrs[0].sin_addr, addrs[0].sin_port, INPLOOKUP_RLOCKPCB, NULL);
2739 	NET_EPOCH_EXIT(et);
2740 	if (inp != NULL) {
2741 		if (inp->inp_socket == NULL)
2742 			error = ENOENT;
2743 		if (error == 0)
2744 			error = cr_canseeinpcb(req->td->td_ucred, inp);
2745 		if (error == 0)
2746 			cru2x(inp->inp_cred, &xuc);
2747 		INP_RUNLOCK(inp);
2748 	} else
2749 		error = ENOENT;
2750 	if (error == 0)
2751 		error = SYSCTL_OUT(req, &xuc, sizeof(struct xucred));
2752 	return (error);
2753 }
2754 
2755 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, getcred,
2756     CTLTYPE_OPAQUE | CTLFLAG_RW | CTLFLAG_PRISON | CTLFLAG_NEEDGIANT,
2757     0, 0, tcp_getcred, "S,xucred",
2758     "Get the xucred of a TCP connection");
2759 #endif /* INET */
2760 
2761 #ifdef INET6
2762 static int
tcp6_getcred(SYSCTL_HANDLER_ARGS)2763 tcp6_getcred(SYSCTL_HANDLER_ARGS)
2764 {
2765 	struct epoch_tracker et;
2766 	struct xucred xuc;
2767 	struct sockaddr_in6 addrs[2];
2768 	struct inpcb *inp;
2769 	int error;
2770 #ifdef INET
2771 	int mapped = 0;
2772 #endif
2773 
2774 	error = priv_check(req->td, PRIV_NETINET_GETCRED);
2775 	if (error)
2776 		return (error);
2777 	error = SYSCTL_IN(req, addrs, sizeof(addrs));
2778 	if (error)
2779 		return (error);
2780 	if ((error = sa6_embedscope(&addrs[0], V_ip6_use_defzone)) != 0 ||
2781 	    (error = sa6_embedscope(&addrs[1], V_ip6_use_defzone)) != 0) {
2782 		return (error);
2783 	}
2784 	if (IN6_IS_ADDR_V4MAPPED(&addrs[0].sin6_addr)) {
2785 #ifdef INET
2786 		if (IN6_IS_ADDR_V4MAPPED(&addrs[1].sin6_addr))
2787 			mapped = 1;
2788 		else
2789 #endif
2790 			return (EINVAL);
2791 	}
2792 
2793 	NET_EPOCH_ENTER(et);
2794 #ifdef INET
2795 	if (mapped == 1)
2796 		inp = in_pcblookup(&V_tcbinfo,
2797 			*(struct in_addr *)&addrs[1].sin6_addr.s6_addr[12],
2798 			addrs[1].sin6_port,
2799 			*(struct in_addr *)&addrs[0].sin6_addr.s6_addr[12],
2800 			addrs[0].sin6_port, INPLOOKUP_RLOCKPCB, NULL);
2801 	else
2802 #endif
2803 		inp = in6_pcblookup(&V_tcbinfo,
2804 			&addrs[1].sin6_addr, addrs[1].sin6_port,
2805 			&addrs[0].sin6_addr, addrs[0].sin6_port,
2806 			INPLOOKUP_RLOCKPCB, NULL);
2807 	NET_EPOCH_EXIT(et);
2808 	if (inp != NULL) {
2809 		if (inp->inp_socket == NULL)
2810 			error = ENOENT;
2811 		if (error == 0)
2812 			error = cr_canseeinpcb(req->td->td_ucred, inp);
2813 		if (error == 0)
2814 			cru2x(inp->inp_cred, &xuc);
2815 		INP_RUNLOCK(inp);
2816 	} else
2817 		error = ENOENT;
2818 	if (error == 0)
2819 		error = SYSCTL_OUT(req, &xuc, sizeof(struct xucred));
2820 	return (error);
2821 }
2822 
2823 SYSCTL_PROC(_net_inet6_tcp6, OID_AUTO, getcred,
2824     CTLTYPE_OPAQUE | CTLFLAG_RW | CTLFLAG_PRISON | CTLFLAG_NEEDGIANT,
2825     0, 0, tcp6_getcred, "S,xucred",
2826     "Get the xucred of a TCP6 connection");
2827 #endif /* INET6 */
2828 
2829 #ifdef INET
2830 static void
tcp_ctlinput_with_port(int cmd,struct sockaddr * sa,void * vip,uint16_t port)2831 tcp_ctlinput_with_port(int cmd, struct sockaddr *sa, void *vip, uint16_t port)
2832 {
2833 	struct ip *ip = vip;
2834 	struct tcphdr *th;
2835 	struct in_addr faddr;
2836 	struct inpcb *inp;
2837 	struct tcpcb *tp;
2838 	struct inpcb *(*notify)(struct inpcb *, int) = tcp_notify;
2839 	struct icmp *icp;
2840 	struct in_conninfo inc;
2841 	tcp_seq icmp_tcp_seq;
2842 	int mtu;
2843 
2844 	faddr = ((struct sockaddr_in *)sa)->sin_addr;
2845 	if (sa->sa_family != AF_INET || faddr.s_addr == INADDR_ANY)
2846 		return;
2847 
2848 	if (cmd == PRC_MSGSIZE)
2849 		notify = tcp_mtudisc_notify;
2850 	else if (V_icmp_may_rst && (cmd == PRC_UNREACH_ADMIN_PROHIB ||
2851 		cmd == PRC_UNREACH_PORT || cmd == PRC_UNREACH_PROTOCOL ||
2852 		cmd == PRC_TIMXCEED_INTRANS) && ip)
2853 		notify = tcp_drop_syn_sent;
2854 
2855 	/*
2856 	 * Hostdead is ugly because it goes linearly through all PCBs.
2857 	 * XXX: We never get this from ICMP, otherwise it makes an
2858 	 * excellent DoS attack on machines with many connections.
2859 	 */
2860 	else if (cmd == PRC_HOSTDEAD)
2861 		ip = NULL;
2862 	else if ((unsigned)cmd >= PRC_NCMDS || inetctlerrmap[cmd] == 0)
2863 		return;
2864 
2865 	if (ip == NULL) {
2866 		in_pcbnotifyall(&V_tcbinfo, faddr, inetctlerrmap[cmd], notify);
2867 		return;
2868 	}
2869 
2870 	icp = (struct icmp *)((caddr_t)ip - offsetof(struct icmp, icmp_ip));
2871 	th = (struct tcphdr *)((caddr_t)ip + (ip->ip_hl << 2));
2872 	inp = in_pcblookup(&V_tcbinfo, faddr, th->th_dport, ip->ip_src,
2873 	    th->th_sport, INPLOOKUP_WLOCKPCB, NULL);
2874 	if (inp != NULL && PRC_IS_REDIRECT(cmd)) {
2875 		/* signal EHOSTDOWN, as it flushes the cached route */
2876 		inp = (*notify)(inp, EHOSTDOWN);
2877 		goto out;
2878 	}
2879 	icmp_tcp_seq = th->th_seq;
2880 	if (inp != NULL)  {
2881 		if (!(inp->inp_flags & INP_TIMEWAIT) &&
2882 		    !(inp->inp_flags & INP_DROPPED) &&
2883 		    !(inp->inp_socket == NULL)) {
2884 			tp = intotcpcb(inp);
2885 			if (tp->t_port != port) {
2886 				goto out;
2887 			}
2888 			if (SEQ_GEQ(ntohl(icmp_tcp_seq), tp->snd_una) &&
2889 			    SEQ_LT(ntohl(icmp_tcp_seq), tp->snd_max)) {
2890 				if (cmd == PRC_MSGSIZE) {
2891 					/*
2892 					 * MTU discovery:
2893 					 * If we got a needfrag set the MTU
2894 					 * in the route to the suggested new
2895 					 * value (if given) and then notify.
2896 					 */
2897 					mtu = ntohs(icp->icmp_nextmtu);
2898 					/*
2899 					 * If no alternative MTU was
2900 					 * proposed, try the next smaller
2901 					 * one.
2902 					 */
2903 					if (!mtu)
2904 						mtu = ip_next_mtu(
2905 						    ntohs(ip->ip_len), 1);
2906 					if (mtu < V_tcp_minmss +
2907 					    sizeof(struct tcpiphdr))
2908 						mtu = V_tcp_minmss +
2909 						    sizeof(struct tcpiphdr);
2910 					/*
2911 					 * Only process the offered MTU if it
2912 					 * is smaller than the current one.
2913 					 */
2914 					if (mtu < tp->t_maxseg +
2915 					    sizeof(struct tcpiphdr)) {
2916 						bzero(&inc, sizeof(inc));
2917 						inc.inc_faddr = faddr;
2918 						inc.inc_fibnum =
2919 						    inp->inp_inc.inc_fibnum;
2920 						tcp_hc_updatemtu(&inc, mtu);
2921 						tcp_mtudisc(inp, mtu);
2922 					}
2923 				} else
2924 					inp = (*notify)(inp,
2925 					    inetctlerrmap[cmd]);
2926 			}
2927 		}
2928 	} else {
2929 		bzero(&inc, sizeof(inc));
2930 		inc.inc_fport = th->th_dport;
2931 		inc.inc_lport = th->th_sport;
2932 		inc.inc_faddr = faddr;
2933 		inc.inc_laddr = ip->ip_src;
2934 		syncache_unreach(&inc, icmp_tcp_seq, port);
2935 	}
2936 out:
2937 	if (inp != NULL)
2938 		INP_WUNLOCK(inp);
2939 }
2940 
2941 void
tcp_ctlinput(int cmd,struct sockaddr * sa,void * vip)2942 tcp_ctlinput(int cmd, struct sockaddr *sa, void *vip)
2943 {
2944 	tcp_ctlinput_with_port(cmd, sa, vip, htons(0));
2945 }
2946 
2947 void
tcp_ctlinput_viaudp(int cmd,struct sockaddr * sa,void * vip,void * unused)2948 tcp_ctlinput_viaudp(int cmd, struct sockaddr *sa, void *vip, void *unused)
2949 {
2950 	/* Its a tunneled TCP over UDP icmp */
2951 	struct ip *outer_ip, *inner_ip;
2952 	struct icmp *icmp;
2953 	struct udphdr *udp;
2954 	struct tcphdr *th, ttemp;
2955 	int i_hlen, o_len;
2956 	uint16_t port;
2957 
2958 	inner_ip = (struct ip *)vip;
2959 	icmp = (struct icmp *)((caddr_t)inner_ip -
2960 	    (sizeof(struct icmp) - sizeof(struct ip)));
2961 	outer_ip = (struct ip *)((caddr_t)icmp - sizeof(struct ip));
2962 	i_hlen = inner_ip->ip_hl << 2;
2963 	o_len = ntohs(outer_ip->ip_len);
2964 	if (o_len <
2965 	    (sizeof(struct ip) + 8 + i_hlen + sizeof(struct udphdr) + offsetof(struct tcphdr, th_ack))) {
2966 		/* Not enough data present */
2967 		return;
2968 	}
2969 	/* Ok lets strip out the inner udphdr header by copying up on top of it the tcp hdr */
2970 	udp = (struct udphdr *)(((caddr_t)inner_ip) + i_hlen);
2971 	if (ntohs(udp->uh_sport) != V_tcp_udp_tunneling_port) {
2972 		return;
2973 	}
2974 	port = udp->uh_dport;
2975 	th = (struct tcphdr *)(udp + 1);
2976 	memcpy(&ttemp, th, sizeof(struct tcphdr));
2977 	memcpy(udp, &ttemp, sizeof(struct tcphdr));
2978 	/* Now adjust down the size of the outer IP header */
2979 	o_len -= sizeof(struct udphdr);
2980 	outer_ip->ip_len = htons(o_len);
2981 	/* Now call in to the normal handling code */
2982 	tcp_ctlinput_with_port(cmd, sa, vip, port);
2983 }
2984 #endif /* INET */
2985 
2986 #ifdef INET6
2987 static void
tcp6_ctlinput_with_port(int cmd,struct sockaddr * sa,void * d,uint16_t port)2988 tcp6_ctlinput_with_port(int cmd, struct sockaddr *sa, void *d, uint16_t port)
2989 {
2990 	struct in6_addr *dst;
2991 	struct inpcb *(*notify)(struct inpcb *, int) = tcp_notify;
2992 	struct ip6_hdr *ip6;
2993 	struct mbuf *m;
2994 	struct inpcb *inp;
2995 	struct tcpcb *tp;
2996 	struct icmp6_hdr *icmp6;
2997 	struct ip6ctlparam *ip6cp = NULL;
2998 	const struct sockaddr_in6 *sa6_src = NULL;
2999 	struct in_conninfo inc;
3000 	struct tcp_ports {
3001 		uint16_t th_sport;
3002 		uint16_t th_dport;
3003 	} t_ports;
3004 	tcp_seq icmp_tcp_seq;
3005 	unsigned int mtu;
3006 	unsigned int off;
3007 
3008 	if (sa->sa_family != AF_INET6 ||
3009 	    sa->sa_len != sizeof(struct sockaddr_in6))
3010 		return;
3011 
3012 	/* if the parameter is from icmp6, decode it. */
3013 	if (d != NULL) {
3014 		ip6cp = (struct ip6ctlparam *)d;
3015 		icmp6 = ip6cp->ip6c_icmp6;
3016 		m = ip6cp->ip6c_m;
3017 		ip6 = ip6cp->ip6c_ip6;
3018 		off = ip6cp->ip6c_off;
3019 		sa6_src = ip6cp->ip6c_src;
3020 		dst = ip6cp->ip6c_finaldst;
3021 	} else {
3022 		m = NULL;
3023 		ip6 = NULL;
3024 		off = 0;	/* fool gcc */
3025 		sa6_src = &sa6_any;
3026 		dst = NULL;
3027 	}
3028 
3029 	if (cmd == PRC_MSGSIZE)
3030 		notify = tcp_mtudisc_notify;
3031 	else if (V_icmp_may_rst && (cmd == PRC_UNREACH_ADMIN_PROHIB ||
3032 		cmd == PRC_UNREACH_PORT || cmd == PRC_UNREACH_PROTOCOL ||
3033 		cmd == PRC_TIMXCEED_INTRANS) && ip6 != NULL)
3034 		notify = tcp_drop_syn_sent;
3035 
3036 	/*
3037 	 * Hostdead is ugly because it goes linearly through all PCBs.
3038 	 * XXX: We never get this from ICMP, otherwise it makes an
3039 	 * excellent DoS attack on machines with many connections.
3040 	 */
3041 	else if (cmd == PRC_HOSTDEAD)
3042 		ip6 = NULL;
3043 	else if ((unsigned)cmd >= PRC_NCMDS || inet6ctlerrmap[cmd] == 0)
3044 		return;
3045 
3046 	if (ip6 == NULL) {
3047 		in6_pcbnotify(&V_tcbinfo, sa, 0,
3048 			      (const struct sockaddr *)sa6_src,
3049 			      0, cmd, NULL, notify);
3050 		return;
3051 	}
3052 
3053 	/* Check if we can safely get the ports from the tcp hdr */
3054 	if (m == NULL ||
3055 	    (m->m_pkthdr.len <
3056 		(int32_t) (off + sizeof(struct tcp_ports)))) {
3057 		return;
3058 	}
3059 	bzero(&t_ports, sizeof(struct tcp_ports));
3060 	m_copydata(m, off, sizeof(struct tcp_ports), (caddr_t)&t_ports);
3061 	inp = in6_pcblookup(&V_tcbinfo, &ip6->ip6_dst, t_ports.th_dport,
3062 	    &ip6->ip6_src, t_ports.th_sport, INPLOOKUP_WLOCKPCB, NULL);
3063 	if (inp != NULL && PRC_IS_REDIRECT(cmd)) {
3064 		/* signal EHOSTDOWN, as it flushes the cached route */
3065 		inp = (*notify)(inp, EHOSTDOWN);
3066 		goto out;
3067 	}
3068 	off += sizeof(struct tcp_ports);
3069 	if (m->m_pkthdr.len < (int32_t) (off + sizeof(tcp_seq))) {
3070 		goto out;
3071 	}
3072 	m_copydata(m, off, sizeof(tcp_seq), (caddr_t)&icmp_tcp_seq);
3073 	if (inp != NULL)  {
3074 		if (!(inp->inp_flags & INP_TIMEWAIT) &&
3075 		    !(inp->inp_flags & INP_DROPPED) &&
3076 		    !(inp->inp_socket == NULL)) {
3077 			tp = intotcpcb(inp);
3078 			if (tp->t_port != port) {
3079 				goto out;
3080 			}
3081 			if (SEQ_GEQ(ntohl(icmp_tcp_seq), tp->snd_una) &&
3082 			    SEQ_LT(ntohl(icmp_tcp_seq), tp->snd_max)) {
3083 				if (cmd == PRC_MSGSIZE) {
3084 					/*
3085 					 * MTU discovery:
3086 					 * If we got a needfrag set the MTU
3087 					 * in the route to the suggested new
3088 					 * value (if given) and then notify.
3089 					 */
3090 					mtu = ntohl(icmp6->icmp6_mtu);
3091 					/*
3092 					 * If no alternative MTU was
3093 					 * proposed, or the proposed
3094 					 * MTU was too small, set to
3095 					 * the min.
3096 					 */
3097 					if (mtu < IPV6_MMTU)
3098 						mtu = IPV6_MMTU - 8;
3099 					bzero(&inc, sizeof(inc));
3100 					inc.inc_fibnum = M_GETFIB(m);
3101 					inc.inc_flags |= INC_ISIPV6;
3102 					inc.inc6_faddr = *dst;
3103 					if (in6_setscope(&inc.inc6_faddr,
3104 						m->m_pkthdr.rcvif, NULL))
3105 						goto out;
3106 					/*
3107 					 * Only process the offered MTU if it
3108 					 * is smaller than the current one.
3109 					 */
3110 					if (mtu < tp->t_maxseg +
3111 					    sizeof (struct tcphdr) +
3112 					    sizeof (struct ip6_hdr)) {
3113 						tcp_hc_updatemtu(&inc, mtu);
3114 						tcp_mtudisc(inp, mtu);
3115 						ICMP6STAT_INC(icp6s_pmtuchg);
3116 					}
3117 				} else
3118 					inp = (*notify)(inp,
3119 					    inet6ctlerrmap[cmd]);
3120 			}
3121 		}
3122 	} else {
3123 		bzero(&inc, sizeof(inc));
3124 		inc.inc_fibnum = M_GETFIB(m);
3125 		inc.inc_flags |= INC_ISIPV6;
3126 		inc.inc_fport = t_ports.th_dport;
3127 		inc.inc_lport = t_ports.th_sport;
3128 		inc.inc6_faddr = *dst;
3129 		inc.inc6_laddr = ip6->ip6_src;
3130 		syncache_unreach(&inc, icmp_tcp_seq, port);
3131 	}
3132 out:
3133 	if (inp != NULL)
3134 		INP_WUNLOCK(inp);
3135 }
3136 
3137 void
tcp6_ctlinput(int cmd,struct sockaddr * sa,void * d)3138 tcp6_ctlinput(int cmd, struct sockaddr *sa, void *d)
3139 {
3140 	tcp6_ctlinput_with_port(cmd, sa, d, htons(0));
3141 }
3142 
3143 void
tcp6_ctlinput_viaudp(int cmd,struct sockaddr * sa,void * d,void * unused)3144 tcp6_ctlinput_viaudp(int cmd, struct sockaddr *sa, void *d, void *unused)
3145 {
3146 	struct ip6ctlparam *ip6cp;
3147 	struct mbuf *m;
3148 	struct udphdr *udp;
3149 	uint16_t port;
3150 
3151 	ip6cp = (struct ip6ctlparam *)d;
3152 	m = m_pulldown(ip6cp->ip6c_m, ip6cp->ip6c_off, sizeof(struct udphdr), NULL);
3153 	if (m == NULL) {
3154 		return;
3155 	}
3156 	udp = mtod(m, struct udphdr *);
3157 	if (ntohs(udp->uh_sport) != V_tcp_udp_tunneling_port) {
3158 		return;
3159 	}
3160 	port = udp->uh_dport;
3161 	m_adj(m, sizeof(struct udphdr));
3162 	if ((m->m_flags & M_PKTHDR) == 0) {
3163 		ip6cp->ip6c_m->m_pkthdr.len -= sizeof(struct udphdr);
3164 	}
3165 	/* Now call in to the normal handling code */
3166 	tcp6_ctlinput_with_port(cmd, sa, d, port);
3167 }
3168 
3169 #endif /* INET6 */
3170 
3171 static uint32_t
tcp_keyed_hash(struct in_conninfo * inc,u_char * key,u_int len)3172 tcp_keyed_hash(struct in_conninfo *inc, u_char *key, u_int len)
3173 {
3174 	SIPHASH_CTX ctx;
3175 	uint32_t hash[2];
3176 
3177 	KASSERT(len >= SIPHASH_KEY_LENGTH,
3178 	    ("%s: keylen %u too short ", __func__, len));
3179 	SipHash24_Init(&ctx);
3180 	SipHash_SetKey(&ctx, (uint8_t *)key);
3181 	SipHash_Update(&ctx, &inc->inc_fport, sizeof(uint16_t));
3182 	SipHash_Update(&ctx, &inc->inc_lport, sizeof(uint16_t));
3183 	switch (inc->inc_flags & INC_ISIPV6) {
3184 #ifdef INET
3185 	case 0:
3186 		SipHash_Update(&ctx, &inc->inc_faddr, sizeof(struct in_addr));
3187 		SipHash_Update(&ctx, &inc->inc_laddr, sizeof(struct in_addr));
3188 		break;
3189 #endif
3190 #ifdef INET6
3191 	case INC_ISIPV6:
3192 		SipHash_Update(&ctx, &inc->inc6_faddr, sizeof(struct in6_addr));
3193 		SipHash_Update(&ctx, &inc->inc6_laddr, sizeof(struct in6_addr));
3194 		break;
3195 #endif
3196 	}
3197 	SipHash_Final((uint8_t *)hash, &ctx);
3198 
3199 	return (hash[0] ^ hash[1]);
3200 }
3201 
3202 uint32_t
tcp_new_ts_offset(struct in_conninfo * inc)3203 tcp_new_ts_offset(struct in_conninfo *inc)
3204 {
3205 	struct in_conninfo inc_store, *local_inc;
3206 
3207 	if (!V_tcp_ts_offset_per_conn) {
3208 		memcpy(&inc_store, inc, sizeof(struct in_conninfo));
3209 		inc_store.inc_lport = 0;
3210 		inc_store.inc_fport = 0;
3211 		local_inc = &inc_store;
3212 	} else {
3213 		local_inc = inc;
3214 	}
3215 	return (tcp_keyed_hash(local_inc, V_ts_offset_secret,
3216 	    sizeof(V_ts_offset_secret)));
3217 }
3218 
3219 /*
3220  * Following is where TCP initial sequence number generation occurs.
3221  *
3222  * There are two places where we must use initial sequence numbers:
3223  * 1.  In SYN-ACK packets.
3224  * 2.  In SYN packets.
3225  *
3226  * All ISNs for SYN-ACK packets are generated by the syncache.  See
3227  * tcp_syncache.c for details.
3228  *
3229  * The ISNs in SYN packets must be monotonic; TIME_WAIT recycling
3230  * depends on this property.  In addition, these ISNs should be
3231  * unguessable so as to prevent connection hijacking.  To satisfy
3232  * the requirements of this situation, the algorithm outlined in
3233  * RFC 1948 is used, with only small modifications.
3234  *
3235  * Implementation details:
3236  *
3237  * Time is based off the system timer, and is corrected so that it
3238  * increases by one megabyte per second.  This allows for proper
3239  * recycling on high speed LANs while still leaving over an hour
3240  * before rollover.
3241  *
3242  * As reading the *exact* system time is too expensive to be done
3243  * whenever setting up a TCP connection, we increment the time
3244  * offset in two ways.  First, a small random positive increment
3245  * is added to isn_offset for each connection that is set up.
3246  * Second, the function tcp_isn_tick fires once per clock tick
3247  * and increments isn_offset as necessary so that sequence numbers
3248  * are incremented at approximately ISN_BYTES_PER_SECOND.  The
3249  * random positive increments serve only to ensure that the same
3250  * exact sequence number is never sent out twice (as could otherwise
3251  * happen when a port is recycled in less than the system tick
3252  * interval.)
3253  *
3254  * net.inet.tcp.isn_reseed_interval controls the number of seconds
3255  * between seeding of isn_secret.  This is normally set to zero,
3256  * as reseeding should not be necessary.
3257  *
3258  * Locking of the global variables isn_secret, isn_last_reseed, isn_offset,
3259  * isn_offset_old, and isn_ctx is performed using the ISN lock.  In
3260  * general, this means holding an exclusive (write) lock.
3261  */
3262 
3263 #define ISN_BYTES_PER_SECOND 1048576
3264 #define ISN_STATIC_INCREMENT 4096
3265 #define ISN_RANDOM_INCREMENT (4096 - 1)
3266 #define ISN_SECRET_LENGTH    SIPHASH_KEY_LENGTH
3267 
3268 VNET_DEFINE_STATIC(u_char, isn_secret[ISN_SECRET_LENGTH]);
3269 VNET_DEFINE_STATIC(int, isn_last);
3270 VNET_DEFINE_STATIC(int, isn_last_reseed);
3271 VNET_DEFINE_STATIC(u_int32_t, isn_offset);
3272 VNET_DEFINE_STATIC(u_int32_t, isn_offset_old);
3273 
3274 #define	V_isn_secret			VNET(isn_secret)
3275 #define	V_isn_last			VNET(isn_last)
3276 #define	V_isn_last_reseed		VNET(isn_last_reseed)
3277 #define	V_isn_offset			VNET(isn_offset)
3278 #define	V_isn_offset_old		VNET(isn_offset_old)
3279 
3280 tcp_seq
tcp_new_isn(struct in_conninfo * inc)3281 tcp_new_isn(struct in_conninfo *inc)
3282 {
3283 	tcp_seq new_isn;
3284 	u_int32_t projected_offset;
3285 
3286 	ISN_LOCK();
3287 	/* Seed if this is the first use, reseed if requested. */
3288 	if ((V_isn_last_reseed == 0) || ((V_tcp_isn_reseed_interval > 0) &&
3289 	     (((u_int)V_isn_last_reseed + (u_int)V_tcp_isn_reseed_interval*hz)
3290 		< (u_int)ticks))) {
3291 		arc4rand(&V_isn_secret, sizeof(V_isn_secret), 0);
3292 		V_isn_last_reseed = ticks;
3293 	}
3294 
3295 	/* Compute the hash and return the ISN. */
3296 	new_isn = (tcp_seq)tcp_keyed_hash(inc, V_isn_secret,
3297 	    sizeof(V_isn_secret));
3298 	V_isn_offset += ISN_STATIC_INCREMENT +
3299 		(arc4random() & ISN_RANDOM_INCREMENT);
3300 	if (ticks != V_isn_last) {
3301 		projected_offset = V_isn_offset_old +
3302 		    ISN_BYTES_PER_SECOND / hz * (ticks - V_isn_last);
3303 		if (SEQ_GT(projected_offset, V_isn_offset))
3304 			V_isn_offset = projected_offset;
3305 		V_isn_offset_old = V_isn_offset;
3306 		V_isn_last = ticks;
3307 	}
3308 	new_isn += V_isn_offset;
3309 	ISN_UNLOCK();
3310 	return (new_isn);
3311 }
3312 
3313 /*
3314  * When a specific ICMP unreachable message is received and the
3315  * connection state is SYN-SENT, drop the connection.  This behavior
3316  * is controlled by the icmp_may_rst sysctl.
3317  */
3318 struct inpcb *
tcp_drop_syn_sent(struct inpcb * inp,int errno)3319 tcp_drop_syn_sent(struct inpcb *inp, int errno)
3320 {
3321 	struct tcpcb *tp;
3322 
3323 	NET_EPOCH_ASSERT();
3324 	INP_WLOCK_ASSERT(inp);
3325 
3326 	if ((inp->inp_flags & INP_TIMEWAIT) ||
3327 	    (inp->inp_flags & INP_DROPPED))
3328 		return (inp);
3329 
3330 	tp = intotcpcb(inp);
3331 	if (tp->t_state != TCPS_SYN_SENT)
3332 		return (inp);
3333 
3334 	if (IS_FASTOPEN(tp->t_flags))
3335 		tcp_fastopen_disable_path(tp);
3336 
3337 	tp = tcp_drop(tp, errno);
3338 	if (tp != NULL)
3339 		return (inp);
3340 	else
3341 		return (NULL);
3342 }
3343 
3344 /*
3345  * When `need fragmentation' ICMP is received, update our idea of the MSS
3346  * based on the new value. Also nudge TCP to send something, since we
3347  * know the packet we just sent was dropped.
3348  * This duplicates some code in the tcp_mss() function in tcp_input.c.
3349  */
3350 static struct inpcb *
tcp_mtudisc_notify(struct inpcb * inp,int error)3351 tcp_mtudisc_notify(struct inpcb *inp, int error)
3352 {
3353 
3354 	tcp_mtudisc(inp, -1);
3355 	return (inp);
3356 }
3357 
3358 static void
tcp_mtudisc(struct inpcb * inp,int mtuoffer)3359 tcp_mtudisc(struct inpcb *inp, int mtuoffer)
3360 {
3361 	struct tcpcb *tp;
3362 	struct socket *so;
3363 
3364 	INP_WLOCK_ASSERT(inp);
3365 	if ((inp->inp_flags & INP_TIMEWAIT) ||
3366 	    (inp->inp_flags & INP_DROPPED))
3367 		return;
3368 
3369 	tp = intotcpcb(inp);
3370 	KASSERT(tp != NULL, ("tcp_mtudisc: tp == NULL"));
3371 
3372 	tcp_mss_update(tp, -1, mtuoffer, NULL, NULL);
3373 
3374 	so = inp->inp_socket;
3375 	SOCKBUF_LOCK(&so->so_snd);
3376 	/* If the mss is larger than the socket buffer, decrease the mss. */
3377 	if (so->so_snd.sb_hiwat < tp->t_maxseg)
3378 		tp->t_maxseg = so->so_snd.sb_hiwat;
3379 	SOCKBUF_UNLOCK(&so->so_snd);
3380 
3381 	TCPSTAT_INC(tcps_mturesent);
3382 	tp->t_rtttime = 0;
3383 	tp->snd_nxt = tp->snd_una;
3384 	tcp_free_sackholes(tp);
3385 	tp->snd_recover = tp->snd_max;
3386 	if (tp->t_flags & TF_SACK_PERMIT)
3387 		EXIT_FASTRECOVERY(tp->t_flags);
3388 	if (tp->t_fb->tfb_tcp_mtu_chg != NULL) {
3389 		/*
3390 		 * Conceptually the snd_nxt setting
3391 		 * and freeing sack holes should
3392 		 * be done by the default stacks
3393 		 * own tfb_tcp_mtu_chg().
3394 		 */
3395 		tp->t_fb->tfb_tcp_mtu_chg(tp);
3396 	}
3397 	tp->t_fb->tfb_tcp_output(tp);
3398 }
3399 
3400 #ifdef INET
3401 /*
3402  * Look-up the routing entry to the peer of this inpcb.  If no route
3403  * is found and it cannot be allocated, then return 0.  This routine
3404  * is called by TCP routines that access the rmx structure and by
3405  * tcp_mss_update to get the peer/interface MTU.
3406  */
3407 uint32_t
tcp_maxmtu(struct in_conninfo * inc,struct tcp_ifcap * cap)3408 tcp_maxmtu(struct in_conninfo *inc, struct tcp_ifcap *cap)
3409 {
3410 	struct nhop_object *nh;
3411 	struct ifnet *ifp;
3412 	uint32_t maxmtu = 0;
3413 
3414 	KASSERT(inc != NULL, ("tcp_maxmtu with NULL in_conninfo pointer"));
3415 
3416 	if (inc->inc_faddr.s_addr != INADDR_ANY) {
3417 		nh = fib4_lookup(inc->inc_fibnum, inc->inc_faddr, 0, NHR_NONE, 0);
3418 		if (nh == NULL)
3419 			return (0);
3420 
3421 		ifp = nh->nh_ifp;
3422 		maxmtu = nh->nh_mtu;
3423 
3424 		/* Report additional interface capabilities. */
3425 		if (cap != NULL) {
3426 			if (ifp->if_capenable & IFCAP_TSO4 &&
3427 			    ifp->if_hwassist & CSUM_TSO) {
3428 				cap->ifcap |= CSUM_TSO;
3429 				cap->tsomax = ifp->if_hw_tsomax;
3430 				cap->tsomaxsegcount = ifp->if_hw_tsomaxsegcount;
3431 				cap->tsomaxsegsize = ifp->if_hw_tsomaxsegsize;
3432 			}
3433 		}
3434 	}
3435 	return (maxmtu);
3436 }
3437 #endif /* INET */
3438 
3439 #ifdef INET6
3440 uint32_t
tcp_maxmtu6(struct in_conninfo * inc,struct tcp_ifcap * cap)3441 tcp_maxmtu6(struct in_conninfo *inc, struct tcp_ifcap *cap)
3442 {
3443 	struct nhop_object *nh;
3444 	struct in6_addr dst6;
3445 	uint32_t scopeid;
3446 	struct ifnet *ifp;
3447 	uint32_t maxmtu = 0;
3448 
3449 	KASSERT(inc != NULL, ("tcp_maxmtu6 with NULL in_conninfo pointer"));
3450 
3451 	if (inc->inc_flags & INC_IPV6MINMTU)
3452 		return (IPV6_MMTU);
3453 
3454 	if (!IN6_IS_ADDR_UNSPECIFIED(&inc->inc6_faddr)) {
3455 		in6_splitscope(&inc->inc6_faddr, &dst6, &scopeid);
3456 		nh = fib6_lookup(inc->inc_fibnum, &dst6, scopeid, NHR_NONE, 0);
3457 		if (nh == NULL)
3458 			return (0);
3459 
3460 		ifp = nh->nh_ifp;
3461 		maxmtu = nh->nh_mtu;
3462 
3463 		/* Report additional interface capabilities. */
3464 		if (cap != NULL) {
3465 			if (ifp->if_capenable & IFCAP_TSO6 &&
3466 			    ifp->if_hwassist & CSUM_TSO) {
3467 				cap->ifcap |= CSUM_TSO;
3468 				cap->tsomax = ifp->if_hw_tsomax;
3469 				cap->tsomaxsegcount = ifp->if_hw_tsomaxsegcount;
3470 				cap->tsomaxsegsize = ifp->if_hw_tsomaxsegsize;
3471 			}
3472 		}
3473 	}
3474 
3475 	return (maxmtu);
3476 }
3477 #endif /* INET6 */
3478 
3479 /*
3480  * Calculate effective SMSS per RFC5681 definition for a given TCP
3481  * connection at its current state, taking into account SACK and etc.
3482  */
3483 u_int
tcp_maxseg(const struct tcpcb * tp)3484 tcp_maxseg(const struct tcpcb *tp)
3485 {
3486 	u_int optlen;
3487 
3488 	if (tp->t_flags & TF_NOOPT)
3489 		return (tp->t_maxseg);
3490 
3491 	/*
3492 	 * Here we have a simplified code from tcp_addoptions(),
3493 	 * without a proper loop, and having most of paddings hardcoded.
3494 	 * We might make mistakes with padding here in some edge cases,
3495 	 * but this is harmless, since result of tcp_maxseg() is used
3496 	 * only in cwnd and ssthresh estimations.
3497 	 */
3498 	if (TCPS_HAVEESTABLISHED(tp->t_state)) {
3499 		if (tp->t_flags & TF_RCVD_TSTMP)
3500 			optlen = TCPOLEN_TSTAMP_APPA;
3501 		else
3502 			optlen = 0;
3503 #if defined(IPSEC_SUPPORT) || defined(TCP_SIGNATURE)
3504 		if (tp->t_flags & TF_SIGNATURE)
3505 			optlen += PADTCPOLEN(TCPOLEN_SIGNATURE);
3506 #endif
3507 		if ((tp->t_flags & TF_SACK_PERMIT) && tp->rcv_numsacks > 0) {
3508 			optlen += TCPOLEN_SACKHDR;
3509 			optlen += tp->rcv_numsacks * TCPOLEN_SACK;
3510 			optlen = PADTCPOLEN(optlen);
3511 		}
3512 	} else {
3513 		if (tp->t_flags & TF_REQ_TSTMP)
3514 			optlen = TCPOLEN_TSTAMP_APPA;
3515 		else
3516 			optlen = PADTCPOLEN(TCPOLEN_MAXSEG);
3517 		if (tp->t_flags & TF_REQ_SCALE)
3518 			optlen += PADTCPOLEN(TCPOLEN_WINDOW);
3519 #if defined(IPSEC_SUPPORT) || defined(TCP_SIGNATURE)
3520 		if (tp->t_flags & TF_SIGNATURE)
3521 			optlen += PADTCPOLEN(TCPOLEN_SIGNATURE);
3522 #endif
3523 		if (tp->t_flags & TF_SACK_PERMIT)
3524 			optlen += PADTCPOLEN(TCPOLEN_SACK_PERMITTED);
3525 	}
3526 #undef PAD
3527 	optlen = min(optlen, TCP_MAXOLEN);
3528 	return (tp->t_maxseg - optlen);
3529 }
3530 
3531 
3532 u_int
tcp_fixed_maxseg(const struct tcpcb * tp)3533 tcp_fixed_maxseg(const struct tcpcb *tp)
3534 {
3535 	int optlen;
3536 
3537 	if (tp->t_flags & TF_NOOPT)
3538 		return (tp->t_maxseg);
3539 
3540 	/*
3541 	 * Here we have a simplified code from tcp_addoptions(),
3542 	 * without a proper loop, and having most of paddings hardcoded.
3543 	 * We only consider fixed options that we would send every
3544 	 * time I.e. SACK is not considered. This is important
3545 	 * for cc modules to figure out what the modulo of the
3546 	 * cwnd should be.
3547 	 */
3548 #define	PAD(len)	((((len) / 4) + !!((len) % 4)) * 4)
3549 	if (TCPS_HAVEESTABLISHED(tp->t_state)) {
3550 		if (tp->t_flags & TF_RCVD_TSTMP)
3551 			optlen = TCPOLEN_TSTAMP_APPA;
3552 		else
3553 			optlen = 0;
3554 #if defined(IPSEC_SUPPORT) || defined(TCP_SIGNATURE)
3555 		if (tp->t_flags & TF_SIGNATURE)
3556 			optlen += PAD(TCPOLEN_SIGNATURE);
3557 #endif
3558 	} else {
3559 		if (tp->t_flags & TF_REQ_TSTMP)
3560 			optlen = TCPOLEN_TSTAMP_APPA;
3561 		else
3562 			optlen = PAD(TCPOLEN_MAXSEG);
3563 		if (tp->t_flags & TF_REQ_SCALE)
3564 			optlen += PAD(TCPOLEN_WINDOW);
3565 #if defined(IPSEC_SUPPORT) || defined(TCP_SIGNATURE)
3566 		if (tp->t_flags & TF_SIGNATURE)
3567 			optlen += PAD(TCPOLEN_SIGNATURE);
3568 #endif
3569 		if (tp->t_flags & TF_SACK_PERMIT)
3570 			optlen += PAD(TCPOLEN_SACK_PERMITTED);
3571 	}
3572 #undef PAD
3573 	optlen = min(optlen, TCP_MAXOLEN);
3574 	return (tp->t_maxseg - optlen);
3575 }
3576 
3577 
3578 
3579 static int
sysctl_drop(SYSCTL_HANDLER_ARGS)3580 sysctl_drop(SYSCTL_HANDLER_ARGS)
3581 {
3582 	/* addrs[0] is a foreign socket, addrs[1] is a local one. */
3583 	struct sockaddr_storage addrs[2];
3584 	struct inpcb *inp;
3585 	struct tcpcb *tp;
3586 	struct tcptw *tw;
3587 	struct sockaddr_in *fin, *lin;
3588 	struct epoch_tracker et;
3589 #ifdef INET6
3590 	struct sockaddr_in6 *fin6, *lin6;
3591 #endif
3592 	int error;
3593 
3594 	inp = NULL;
3595 	fin = lin = NULL;
3596 #ifdef INET6
3597 	fin6 = lin6 = NULL;
3598 #endif
3599 	error = 0;
3600 
3601 	if (req->oldptr != NULL || req->oldlen != 0)
3602 		return (EINVAL);
3603 	if (req->newptr == NULL)
3604 		return (EPERM);
3605 	if (req->newlen < sizeof(addrs))
3606 		return (ENOMEM);
3607 	error = SYSCTL_IN(req, &addrs, sizeof(addrs));
3608 	if (error)
3609 		return (error);
3610 
3611 	switch (addrs[0].ss_family) {
3612 #ifdef INET6
3613 	case AF_INET6:
3614 		fin6 = (struct sockaddr_in6 *)&addrs[0];
3615 		lin6 = (struct sockaddr_in6 *)&addrs[1];
3616 		if (fin6->sin6_len != sizeof(struct sockaddr_in6) ||
3617 		    lin6->sin6_len != sizeof(struct sockaddr_in6))
3618 			return (EINVAL);
3619 		if (IN6_IS_ADDR_V4MAPPED(&fin6->sin6_addr)) {
3620 			if (!IN6_IS_ADDR_V4MAPPED(&lin6->sin6_addr))
3621 				return (EINVAL);
3622 			in6_sin6_2_sin_in_sock((struct sockaddr *)&addrs[0]);
3623 			in6_sin6_2_sin_in_sock((struct sockaddr *)&addrs[1]);
3624 			fin = (struct sockaddr_in *)&addrs[0];
3625 			lin = (struct sockaddr_in *)&addrs[1];
3626 			break;
3627 		}
3628 		error = sa6_embedscope(fin6, V_ip6_use_defzone);
3629 		if (error)
3630 			return (error);
3631 		error = sa6_embedscope(lin6, V_ip6_use_defzone);
3632 		if (error)
3633 			return (error);
3634 		break;
3635 #endif
3636 #ifdef INET
3637 	case AF_INET:
3638 		fin = (struct sockaddr_in *)&addrs[0];
3639 		lin = (struct sockaddr_in *)&addrs[1];
3640 		if (fin->sin_len != sizeof(struct sockaddr_in) ||
3641 		    lin->sin_len != sizeof(struct sockaddr_in))
3642 			return (EINVAL);
3643 		break;
3644 #endif
3645 	default:
3646 		return (EINVAL);
3647 	}
3648 	NET_EPOCH_ENTER(et);
3649 	switch (addrs[0].ss_family) {
3650 #ifdef INET6
3651 	case AF_INET6:
3652 		inp = in6_pcblookup(&V_tcbinfo, &fin6->sin6_addr,
3653 		    fin6->sin6_port, &lin6->sin6_addr, lin6->sin6_port,
3654 		    INPLOOKUP_WLOCKPCB, NULL);
3655 		break;
3656 #endif
3657 #ifdef INET
3658 	case AF_INET:
3659 		inp = in_pcblookup(&V_tcbinfo, fin->sin_addr, fin->sin_port,
3660 		    lin->sin_addr, lin->sin_port, INPLOOKUP_WLOCKPCB, NULL);
3661 		break;
3662 #endif
3663 	}
3664 	if (inp != NULL) {
3665 		if (inp->inp_flags & INP_TIMEWAIT) {
3666 			/*
3667 			 * XXXRW: There currently exists a state where an
3668 			 * inpcb is present, but its timewait state has been
3669 			 * discarded.  For now, don't allow dropping of this
3670 			 * type of inpcb.
3671 			 */
3672 			tw = intotw(inp);
3673 			if (tw != NULL)
3674 				tcp_twclose(tw, 0);
3675 			else
3676 				INP_WUNLOCK(inp);
3677 		} else if ((inp->inp_flags & INP_DROPPED) == 0 &&
3678 		    !SOLISTENING(inp->inp_socket)) {
3679 			tp = intotcpcb(inp);
3680 			tp = tcp_drop(tp, ECONNABORTED);
3681 			if (tp != NULL)
3682 				INP_WUNLOCK(inp);
3683 		} else
3684 			INP_WUNLOCK(inp);
3685 	} else
3686 		error = ESRCH;
3687 	NET_EPOCH_EXIT(et);
3688 	return (error);
3689 }
3690 
3691 SYSCTL_PROC(_net_inet_tcp, TCPCTL_DROP, drop,
3692     CTLFLAG_VNET | CTLTYPE_STRUCT | CTLFLAG_WR | CTLFLAG_SKIP |
3693     CTLFLAG_NEEDGIANT, NULL, 0, sysctl_drop, "",
3694     "Drop TCP connection");
3695 
3696 #ifdef KERN_TLS
3697 static int
sysctl_switch_tls(SYSCTL_HANDLER_ARGS)3698 sysctl_switch_tls(SYSCTL_HANDLER_ARGS)
3699 {
3700 	/* addrs[0] is a foreign socket, addrs[1] is a local one. */
3701 	struct sockaddr_storage addrs[2];
3702 	struct inpcb *inp;
3703 	struct sockaddr_in *fin, *lin;
3704 	struct epoch_tracker et;
3705 #ifdef INET6
3706 	struct sockaddr_in6 *fin6, *lin6;
3707 #endif
3708 	int error;
3709 
3710 	inp = NULL;
3711 	fin = lin = NULL;
3712 #ifdef INET6
3713 	fin6 = lin6 = NULL;
3714 #endif
3715 	error = 0;
3716 
3717 	if (req->oldptr != NULL || req->oldlen != 0)
3718 		return (EINVAL);
3719 	if (req->newptr == NULL)
3720 		return (EPERM);
3721 	if (req->newlen < sizeof(addrs))
3722 		return (ENOMEM);
3723 	error = SYSCTL_IN(req, &addrs, sizeof(addrs));
3724 	if (error)
3725 		return (error);
3726 
3727 	switch (addrs[0].ss_family) {
3728 #ifdef INET6
3729 	case AF_INET6:
3730 		fin6 = (struct sockaddr_in6 *)&addrs[0];
3731 		lin6 = (struct sockaddr_in6 *)&addrs[1];
3732 		if (fin6->sin6_len != sizeof(struct sockaddr_in6) ||
3733 		    lin6->sin6_len != sizeof(struct sockaddr_in6))
3734 			return (EINVAL);
3735 		if (IN6_IS_ADDR_V4MAPPED(&fin6->sin6_addr)) {
3736 			if (!IN6_IS_ADDR_V4MAPPED(&lin6->sin6_addr))
3737 				return (EINVAL);
3738 			in6_sin6_2_sin_in_sock((struct sockaddr *)&addrs[0]);
3739 			in6_sin6_2_sin_in_sock((struct sockaddr *)&addrs[1]);
3740 			fin = (struct sockaddr_in *)&addrs[0];
3741 			lin = (struct sockaddr_in *)&addrs[1];
3742 			break;
3743 		}
3744 		error = sa6_embedscope(fin6, V_ip6_use_defzone);
3745 		if (error)
3746 			return (error);
3747 		error = sa6_embedscope(lin6, V_ip6_use_defzone);
3748 		if (error)
3749 			return (error);
3750 		break;
3751 #endif
3752 #ifdef INET
3753 	case AF_INET:
3754 		fin = (struct sockaddr_in *)&addrs[0];
3755 		lin = (struct sockaddr_in *)&addrs[1];
3756 		if (fin->sin_len != sizeof(struct sockaddr_in) ||
3757 		    lin->sin_len != sizeof(struct sockaddr_in))
3758 			return (EINVAL);
3759 		break;
3760 #endif
3761 	default:
3762 		return (EINVAL);
3763 	}
3764 	NET_EPOCH_ENTER(et);
3765 	switch (addrs[0].ss_family) {
3766 #ifdef INET6
3767 	case AF_INET6:
3768 		inp = in6_pcblookup(&V_tcbinfo, &fin6->sin6_addr,
3769 		    fin6->sin6_port, &lin6->sin6_addr, lin6->sin6_port,
3770 		    INPLOOKUP_WLOCKPCB, NULL);
3771 		break;
3772 #endif
3773 #ifdef INET
3774 	case AF_INET:
3775 		inp = in_pcblookup(&V_tcbinfo, fin->sin_addr, fin->sin_port,
3776 		    lin->sin_addr, lin->sin_port, INPLOOKUP_WLOCKPCB, NULL);
3777 		break;
3778 #endif
3779 	}
3780 	NET_EPOCH_EXIT(et);
3781 	if (inp != NULL) {
3782 		if ((inp->inp_flags & (INP_TIMEWAIT | INP_DROPPED)) != 0 ||
3783 		    inp->inp_socket == NULL) {
3784 			error = ECONNRESET;
3785 			INP_WUNLOCK(inp);
3786 		} else {
3787 			struct socket *so;
3788 
3789 			so = inp->inp_socket;
3790 			soref(so);
3791 			error = ktls_set_tx_mode(so,
3792 			    arg2 == 0 ? TCP_TLS_MODE_SW : TCP_TLS_MODE_IFNET);
3793 			INP_WUNLOCK(inp);
3794 			SOCK_LOCK(so);
3795 			sorele(so);
3796 		}
3797 	} else
3798 		error = ESRCH;
3799 	return (error);
3800 }
3801 
3802 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, switch_to_sw_tls,
3803     CTLFLAG_VNET | CTLTYPE_STRUCT | CTLFLAG_WR | CTLFLAG_SKIP |
3804     CTLFLAG_NEEDGIANT, NULL, 0, sysctl_switch_tls, "",
3805     "Switch TCP connection to SW TLS");
3806 SYSCTL_PROC(_net_inet_tcp, OID_AUTO, switch_to_ifnet_tls,
3807     CTLFLAG_VNET | CTLTYPE_STRUCT | CTLFLAG_WR | CTLFLAG_SKIP |
3808     CTLFLAG_NEEDGIANT, NULL, 1, sysctl_switch_tls, "",
3809     "Switch TCP connection to ifnet TLS");
3810 #endif
3811 
3812 /*
3813  * Generate a standardized TCP log line for use throughout the
3814  * tcp subsystem.  Memory allocation is done with M_NOWAIT to
3815  * allow use in the interrupt context.
3816  *
3817  * NB: The caller MUST free(s, M_TCPLOG) the returned string.
3818  * NB: The function may return NULL if memory allocation failed.
3819  *
3820  * Due to header inclusion and ordering limitations the struct ip
3821  * and ip6_hdr pointers have to be passed as void pointers.
3822  */
3823 char *
tcp_log_vain(struct in_conninfo * inc,struct tcphdr * th,void * ip4hdr,const void * ip6hdr)3824 tcp_log_vain(struct in_conninfo *inc, struct tcphdr *th, void *ip4hdr,
3825     const void *ip6hdr)
3826 {
3827 
3828 	/* Is logging enabled? */
3829 	if (V_tcp_log_in_vain == 0)
3830 		return (NULL);
3831 
3832 	return (tcp_log_addr(inc, th, ip4hdr, ip6hdr));
3833 }
3834 
3835 char *
tcp_log_addrs(struct in_conninfo * inc,struct tcphdr * th,void * ip4hdr,const void * ip6hdr)3836 tcp_log_addrs(struct in_conninfo *inc, struct tcphdr *th, void *ip4hdr,
3837     const void *ip6hdr)
3838 {
3839 
3840 	/* Is logging enabled? */
3841 	if (tcp_log_debug == 0)
3842 		return (NULL);
3843 
3844 	return (tcp_log_addr(inc, th, ip4hdr, ip6hdr));
3845 }
3846 
3847 static char *
tcp_log_addr(struct in_conninfo * inc,struct tcphdr * th,void * ip4hdr,const void * ip6hdr)3848 tcp_log_addr(struct in_conninfo *inc, struct tcphdr *th, void *ip4hdr,
3849     const void *ip6hdr)
3850 {
3851 	char *s, *sp;
3852 	size_t size;
3853 	struct ip *ip;
3854 #ifdef INET6
3855 	const struct ip6_hdr *ip6;
3856 
3857 	ip6 = (const struct ip6_hdr *)ip6hdr;
3858 #endif /* INET6 */
3859 	ip = (struct ip *)ip4hdr;
3860 
3861 	/*
3862 	 * The log line looks like this:
3863 	 * "TCP: [1.2.3.4]:50332 to [1.2.3.4]:80 tcpflags 0x2<SYN>"
3864 	 */
3865 	size = sizeof("TCP: []:12345 to []:12345 tcpflags 0x2<>") +
3866 	    sizeof(PRINT_TH_FLAGS) + 1 +
3867 #ifdef INET6
3868 	    2 * INET6_ADDRSTRLEN;
3869 #else
3870 	    2 * INET_ADDRSTRLEN;
3871 #endif /* INET6 */
3872 
3873 	s = malloc(size, M_TCPLOG, M_ZERO|M_NOWAIT);
3874 	if (s == NULL)
3875 		return (NULL);
3876 
3877 	strcat(s, "TCP: [");
3878 	sp = s + strlen(s);
3879 
3880 	if (inc && ((inc->inc_flags & INC_ISIPV6) == 0)) {
3881 		inet_ntoa_r(inc->inc_faddr, sp);
3882 		sp = s + strlen(s);
3883 		sprintf(sp, "]:%i to [", ntohs(inc->inc_fport));
3884 		sp = s + strlen(s);
3885 		inet_ntoa_r(inc->inc_laddr, sp);
3886 		sp = s + strlen(s);
3887 		sprintf(sp, "]:%i", ntohs(inc->inc_lport));
3888 #ifdef INET6
3889 	} else if (inc) {
3890 		ip6_sprintf(sp, &inc->inc6_faddr);
3891 		sp = s + strlen(s);
3892 		sprintf(sp, "]:%i to [", ntohs(inc->inc_fport));
3893 		sp = s + strlen(s);
3894 		ip6_sprintf(sp, &inc->inc6_laddr);
3895 		sp = s + strlen(s);
3896 		sprintf(sp, "]:%i", ntohs(inc->inc_lport));
3897 	} else if (ip6 && th) {
3898 		ip6_sprintf(sp, &ip6->ip6_src);
3899 		sp = s + strlen(s);
3900 		sprintf(sp, "]:%i to [", ntohs(th->th_sport));
3901 		sp = s + strlen(s);
3902 		ip6_sprintf(sp, &ip6->ip6_dst);
3903 		sp = s + strlen(s);
3904 		sprintf(sp, "]:%i", ntohs(th->th_dport));
3905 #endif /* INET6 */
3906 #ifdef INET
3907 	} else if (ip && th) {
3908 		inet_ntoa_r(ip->ip_src, sp);
3909 		sp = s + strlen(s);
3910 		sprintf(sp, "]:%i to [", ntohs(th->th_sport));
3911 		sp = s + strlen(s);
3912 		inet_ntoa_r(ip->ip_dst, sp);
3913 		sp = s + strlen(s);
3914 		sprintf(sp, "]:%i", ntohs(th->th_dport));
3915 #endif /* INET */
3916 	} else {
3917 		free(s, M_TCPLOG);
3918 		return (NULL);
3919 	}
3920 	sp = s + strlen(s);
3921 	if (th)
3922 		sprintf(sp, " tcpflags 0x%b", th->th_flags, PRINT_TH_FLAGS);
3923 	if (*(s + size - 1) != '\0')
3924 		panic("%s: string too long", __func__);
3925 	return (s);
3926 }
3927 
3928 /*
3929  * A subroutine which makes it easy to track TCP state changes with DTrace.
3930  * This function shouldn't be called for t_state initializations that don't
3931  * correspond to actual TCP state transitions.
3932  */
3933 void
tcp_state_change(struct tcpcb * tp,int newstate)3934 tcp_state_change(struct tcpcb *tp, int newstate)
3935 {
3936 #if defined(KDTRACE_HOOKS)
3937 	int pstate = tp->t_state;
3938 #endif
3939 
3940 	TCPSTATES_DEC(tp->t_state);
3941 	TCPSTATES_INC(newstate);
3942 	tp->t_state = newstate;
3943 	TCP_PROBE6(state__change, NULL, tp, NULL, tp, NULL, pstate);
3944 }
3945 
3946 /*
3947  * Create an external-format (``xtcpcb'') structure using the information in
3948  * the kernel-format tcpcb structure pointed to by tp.  This is done to
3949  * reduce the spew of irrelevant information over this interface, to isolate
3950  * user code from changes in the kernel structure, and potentially to provide
3951  * information-hiding if we decide that some of this information should be
3952  * hidden from users.
3953  */
3954 void
tcp_inptoxtp(const struct inpcb * inp,struct xtcpcb * xt)3955 tcp_inptoxtp(const struct inpcb *inp, struct xtcpcb *xt)
3956 {
3957 	struct tcpcb *tp = intotcpcb(inp);
3958 	struct tcptw *tw = intotw(inp);
3959 	sbintime_t now;
3960 
3961 	bzero(xt, sizeof(*xt));
3962 	if (inp->inp_flags & INP_TIMEWAIT) {
3963 		xt->t_state = TCPS_TIME_WAIT;
3964 		xt->xt_encaps_port = tw->t_port;
3965 	} else {
3966 		xt->t_state = tp->t_state;
3967 		xt->t_logstate = tp->t_logstate;
3968 		xt->t_flags = tp->t_flags;
3969 		xt->t_sndzerowin = tp->t_sndzerowin;
3970 		xt->t_sndrexmitpack = tp->t_sndrexmitpack;
3971 		xt->t_rcvoopack = tp->t_rcvoopack;
3972 		xt->t_rcv_wnd = tp->rcv_wnd;
3973 		xt->t_snd_wnd = tp->snd_wnd;
3974 		xt->t_snd_cwnd = tp->snd_cwnd;
3975 		xt->t_snd_ssthresh = tp->snd_ssthresh;
3976 		xt->t_maxseg = tp->t_maxseg;
3977 		xt->xt_ecn = (tp->t_flags2 & TF2_ECN_PERMIT) ? 1 : 0 +
3978 			     (tp->t_flags2 & TF2_ACE_PERMIT) ? 2 : 0;
3979 
3980 		now = getsbinuptime();
3981 #define	COPYTIMER(ttt)	do {						\
3982 		if (callout_active(&tp->t_timers->ttt))			\
3983 			xt->ttt = (tp->t_timers->ttt.c_time - now) /	\
3984 			    SBT_1MS;					\
3985 		else							\
3986 			xt->ttt = 0;					\
3987 } while (0)
3988 		COPYTIMER(tt_delack);
3989 		COPYTIMER(tt_rexmt);
3990 		COPYTIMER(tt_persist);
3991 		COPYTIMER(tt_keep);
3992 		COPYTIMER(tt_2msl);
3993 #undef COPYTIMER
3994 		xt->t_rcvtime = 1000 * (ticks - tp->t_rcvtime) / hz;
3995 
3996 		xt->xt_encaps_port = tp->t_port;
3997 		bcopy(tp->t_fb->tfb_tcp_block_name, xt->xt_stack,
3998 		    TCP_FUNCTION_NAME_LEN_MAX);
3999 		bcopy(CC_ALGO(tp)->name, xt->xt_cc,
4000 		    TCP_CA_NAME_MAX);
4001 #ifdef TCP_BLACKBOX
4002 		(void)tcp_log_get_id(tp, xt->xt_logid);
4003 #endif
4004 	}
4005 
4006 	xt->xt_len = sizeof(struct xtcpcb);
4007 	in_pcbtoxinpcb(inp, &xt->xt_inp);
4008 	if (inp->inp_socket == NULL)
4009 		xt->xt_inp.xi_socket.xso_protocol = IPPROTO_TCP;
4010 }
4011 
4012 void
tcp_log_end_status(struct tcpcb * tp,uint8_t status)4013 tcp_log_end_status(struct tcpcb *tp, uint8_t status)
4014 {
4015 	uint32_t bit, i;
4016 
4017 	if ((tp == NULL) ||
4018 	    (status > TCP_EI_STATUS_MAX_VALUE) ||
4019 	    (status == 0)) {
4020 		/* Invalid */
4021 		return;
4022 	}
4023 	if (status > (sizeof(uint32_t) * 8)) {
4024 		/* Should this be a KASSERT? */
4025 		return;
4026 	}
4027 	bit = 1U << (status - 1);
4028 	if (bit & tp->t_end_info_status) {
4029 		/* already logged */
4030 		return;
4031 	}
4032 	for (i = 0; i < TCP_END_BYTE_INFO; i++) {
4033 		if (tp->t_end_info_bytes[i] == TCP_EI_EMPTY_SLOT) {
4034 			tp->t_end_info_bytes[i] = status;
4035 			tp->t_end_info_status |= bit;
4036 			break;
4037 		}
4038 	}
4039 }
4040 
4041 int
tcp_can_enable_pacing(void)4042 tcp_can_enable_pacing(void)
4043 {
4044 
4045 	if ((tcp_pacing_limit == -1) ||
4046 	    (tcp_pacing_limit > number_of_tcp_connections_pacing)) {
4047 		atomic_fetchadd_int(&number_of_tcp_connections_pacing, 1);
4048 		shadow_num_connections = number_of_tcp_connections_pacing;
4049 		return (1);
4050 	} else {
4051 		return (0);
4052 	}
4053 }
4054 
4055 static uint8_t tcp_pacing_warning = 0;
4056 
4057 void
tcp_decrement_paced_conn(void)4058 tcp_decrement_paced_conn(void)
4059 {
4060 	uint32_t ret;
4061 
4062 	ret = atomic_fetchadd_int(&number_of_tcp_connections_pacing, -1);
4063 	shadow_num_connections = number_of_tcp_connections_pacing;
4064 	KASSERT(ret != 0, ("tcp_paced_connection_exits -1 would cause wrap?"));
4065 	if (ret == 0) {
4066 		if (tcp_pacing_limit != -1) {
4067 			printf("Warning all pacing is now disabled, count decrements invalidly!\n");
4068 			tcp_pacing_limit = 0;
4069 		} else if (tcp_pacing_warning == 0) {
4070 			printf("Warning pacing count is invalid, invalid decrement\n");
4071 			tcp_pacing_warning = 1;
4072 		}
4073 	}
4074 }
4075