xref: /f-stack/freebsd/netinet/tcp_hostcache.c (revision 22ce4aff)
1a9643ea8Slogwang /*-
2*22ce4affSfengbojiang  * SPDX-License-Identifier: BSD-3-Clause
3*22ce4affSfengbojiang  *
4a9643ea8Slogwang  * Copyright (c) 2002 Andre Oppermann, Internet Business Solutions AG
5a9643ea8Slogwang  * All rights reserved.
6a9643ea8Slogwang  *
7a9643ea8Slogwang  * Redistribution and use in source and binary forms, with or without
8a9643ea8Slogwang  * modification, are permitted provided that the following conditions
9a9643ea8Slogwang  * are met:
10a9643ea8Slogwang  * 1. Redistributions of source code must retain the above copyright
11a9643ea8Slogwang  *    notice, this list of conditions and the following disclaimer.
12a9643ea8Slogwang  * 2. Redistributions in binary form must reproduce the above copyright
13a9643ea8Slogwang  *    notice, this list of conditions and the following disclaimer in the
14a9643ea8Slogwang  *    documentation and/or other materials provided with the distribution.
15a9643ea8Slogwang  * 3. The name of the author may not be used to endorse or promote
16a9643ea8Slogwang  *    products derived from this software without specific prior written
17a9643ea8Slogwang  *    permission.
18a9643ea8Slogwang  *
19a9643ea8Slogwang  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20a9643ea8Slogwang  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21a9643ea8Slogwang  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22a9643ea8Slogwang  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23a9643ea8Slogwang  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24a9643ea8Slogwang  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25a9643ea8Slogwang  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26a9643ea8Slogwang  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27a9643ea8Slogwang  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28a9643ea8Slogwang  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29a9643ea8Slogwang  * SUCH DAMAGE.
30a9643ea8Slogwang  */
31a9643ea8Slogwang 
32a9643ea8Slogwang /*
33a9643ea8Slogwang  * The tcp_hostcache moves the tcp-specific cached metrics from the routing
34a9643ea8Slogwang  * table to a dedicated structure indexed by the remote IP address.  It keeps
35a9643ea8Slogwang  * information on the measured TCP parameters of past TCP sessions to allow
36a9643ea8Slogwang  * better initial start values to be used with later connections to/from the
37a9643ea8Slogwang  * same source.  Depending on the network parameters (delay, max MTU,
38a9643ea8Slogwang  * congestion window) between local and remote sites, this can lead to
39a9643ea8Slogwang  * significant speed-ups for new TCP connections after the first one.
40a9643ea8Slogwang  *
41a9643ea8Slogwang  * Due to the tcp_hostcache, all TCP-specific metrics information in the
42a9643ea8Slogwang  * routing table have been removed.  The inpcb no longer keeps a pointer to
43a9643ea8Slogwang  * the routing entry, and protocol-initiated route cloning has been removed
44a9643ea8Slogwang  * as well.  With these changes, the routing table has gone back to being
45a9643ea8Slogwang  * more lightwight and only carries information related to packet forwarding.
46a9643ea8Slogwang  *
47a9643ea8Slogwang  * tcp_hostcache is designed for multiple concurrent access in SMP
48a9643ea8Slogwang  * environments and high contention.  All bucket rows have their own lock and
49a9643ea8Slogwang  * thus multiple lookups and modifies can be done at the same time as long as
50a9643ea8Slogwang  * they are in different bucket rows.  If a request for insertion of a new
51a9643ea8Slogwang  * record can't be satisfied, it simply returns an empty structure.  Nobody
52a9643ea8Slogwang  * and nothing outside of tcp_hostcache.c will ever point directly to any
53a9643ea8Slogwang  * entry in the tcp_hostcache.  All communication is done in an
54a9643ea8Slogwang  * object-oriented way and only functions of tcp_hostcache will manipulate
55a9643ea8Slogwang  * hostcache entries.  Otherwise, we are unable to achieve good behaviour in
56a9643ea8Slogwang  * concurrent access situations.  Since tcp_hostcache is only caching
57a9643ea8Slogwang  * information, there are no fatal consequences if we either can't satisfy
58a9643ea8Slogwang  * any particular request or have to drop/overwrite an existing entry because
59a9643ea8Slogwang  * of bucket limit memory constrains.
60a9643ea8Slogwang  */
61a9643ea8Slogwang 
62a9643ea8Slogwang /*
63a9643ea8Slogwang  * Many thanks to jlemon for basic structure of tcp_syncache which is being
64a9643ea8Slogwang  * followed here.
65a9643ea8Slogwang  */
66a9643ea8Slogwang 
67a9643ea8Slogwang #include <sys/cdefs.h>
68a9643ea8Slogwang __FBSDID("$FreeBSD$");
69a9643ea8Slogwang 
70a9643ea8Slogwang #include "opt_inet6.h"
71a9643ea8Slogwang 
72a9643ea8Slogwang #include <sys/param.h>
73a9643ea8Slogwang #include <sys/systm.h>
74*22ce4affSfengbojiang #include <sys/jail.h>
75a9643ea8Slogwang #include <sys/kernel.h>
76a9643ea8Slogwang #include <sys/lock.h>
77a9643ea8Slogwang #include <sys/mutex.h>
78a9643ea8Slogwang #include <sys/malloc.h>
79*22ce4affSfengbojiang #include <sys/proc.h>
80a9643ea8Slogwang #include <sys/sbuf.h>
81a9643ea8Slogwang #include <sys/socket.h>
82a9643ea8Slogwang #include <sys/socketvar.h>
83a9643ea8Slogwang #include <sys/sysctl.h>
84a9643ea8Slogwang 
85a9643ea8Slogwang #include <net/if.h>
86a9643ea8Slogwang #include <net/if_var.h>
87a9643ea8Slogwang #include <net/route.h>
88a9643ea8Slogwang #include <net/vnet.h>
89a9643ea8Slogwang 
90a9643ea8Slogwang #include <netinet/in.h>
91a9643ea8Slogwang #include <netinet/in_systm.h>
92a9643ea8Slogwang #include <netinet/ip.h>
93a9643ea8Slogwang #include <netinet/in_var.h>
94a9643ea8Slogwang #include <netinet/in_pcb.h>
95a9643ea8Slogwang #include <netinet/ip_var.h>
96a9643ea8Slogwang #ifdef INET6
97a9643ea8Slogwang #include <netinet/ip6.h>
98a9643ea8Slogwang #include <netinet6/ip6_var.h>
99a9643ea8Slogwang #endif
100a9643ea8Slogwang #include <netinet/tcp.h>
101a9643ea8Slogwang #include <netinet/tcp_var.h>
102a9643ea8Slogwang #include <netinet/tcp_hostcache.h>
103a9643ea8Slogwang #ifdef INET6
104a9643ea8Slogwang #include <netinet6/tcp6_var.h>
105a9643ea8Slogwang #endif
106a9643ea8Slogwang 
107a9643ea8Slogwang #include <vm/uma.h>
108a9643ea8Slogwang 
109a9643ea8Slogwang /* Arbitrary values */
110a9643ea8Slogwang #define TCP_HOSTCACHE_HASHSIZE		512
111a9643ea8Slogwang #define TCP_HOSTCACHE_BUCKETLIMIT	30
112a9643ea8Slogwang #define TCP_HOSTCACHE_EXPIRE		60*60	/* one hour */
113a9643ea8Slogwang #define TCP_HOSTCACHE_PRUNE		5*60	/* every 5 minutes */
114a9643ea8Slogwang 
115*22ce4affSfengbojiang VNET_DEFINE_STATIC(struct tcp_hostcache, tcp_hostcache);
116a9643ea8Slogwang #define	V_tcp_hostcache		VNET(tcp_hostcache)
117a9643ea8Slogwang 
118*22ce4affSfengbojiang VNET_DEFINE_STATIC(struct callout, tcp_hc_callout);
119a9643ea8Slogwang #define	V_tcp_hc_callout	VNET(tcp_hc_callout)
120a9643ea8Slogwang 
121a9643ea8Slogwang static struct hc_metrics *tcp_hc_lookup(struct in_conninfo *);
122a9643ea8Slogwang static struct hc_metrics *tcp_hc_insert(struct in_conninfo *);
123a9643ea8Slogwang static int sysctl_tcp_hc_list(SYSCTL_HANDLER_ARGS);
124a9643ea8Slogwang static int sysctl_tcp_hc_purgenow(SYSCTL_HANDLER_ARGS);
125a9643ea8Slogwang static void tcp_hc_purge_internal(int);
126a9643ea8Slogwang static void tcp_hc_purge(void *);
127a9643ea8Slogwang 
128*22ce4affSfengbojiang static SYSCTL_NODE(_net_inet_tcp, OID_AUTO, hostcache,
129*22ce4affSfengbojiang     CTLFLAG_RW | CTLFLAG_MPSAFE, 0,
130a9643ea8Slogwang     "TCP Host cache");
131a9643ea8Slogwang 
132*22ce4affSfengbojiang VNET_DEFINE(int, tcp_use_hostcache) = 1;
133*22ce4affSfengbojiang #define V_tcp_use_hostcache  VNET(tcp_use_hostcache)
134*22ce4affSfengbojiang SYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, enable, CTLFLAG_VNET | CTLFLAG_RW,
135*22ce4affSfengbojiang     &VNET_NAME(tcp_use_hostcache), 0,
136*22ce4affSfengbojiang     "Enable the TCP hostcache");
137*22ce4affSfengbojiang 
138a9643ea8Slogwang SYSCTL_UINT(_net_inet_tcp_hostcache, OID_AUTO, cachelimit, CTLFLAG_VNET | CTLFLAG_RDTUN,
139a9643ea8Slogwang     &VNET_NAME(tcp_hostcache.cache_limit), 0,
140a9643ea8Slogwang     "Overall entry limit for hostcache");
141a9643ea8Slogwang 
142a9643ea8Slogwang SYSCTL_UINT(_net_inet_tcp_hostcache, OID_AUTO, hashsize, CTLFLAG_VNET | CTLFLAG_RDTUN,
143a9643ea8Slogwang     &VNET_NAME(tcp_hostcache.hashsize), 0,
144a9643ea8Slogwang     "Size of TCP hostcache hashtable");
145a9643ea8Slogwang 
146a9643ea8Slogwang SYSCTL_UINT(_net_inet_tcp_hostcache, OID_AUTO, bucketlimit,
147a9643ea8Slogwang     CTLFLAG_VNET | CTLFLAG_RDTUN, &VNET_NAME(tcp_hostcache.bucket_limit), 0,
148a9643ea8Slogwang     "Per-bucket hash limit for hostcache");
149a9643ea8Slogwang 
150a9643ea8Slogwang SYSCTL_UINT(_net_inet_tcp_hostcache, OID_AUTO, count, CTLFLAG_VNET | CTLFLAG_RD,
151a9643ea8Slogwang      &VNET_NAME(tcp_hostcache.cache_count), 0,
152a9643ea8Slogwang     "Current number of entries in hostcache");
153a9643ea8Slogwang 
154a9643ea8Slogwang SYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, expire, CTLFLAG_VNET | CTLFLAG_RW,
155a9643ea8Slogwang     &VNET_NAME(tcp_hostcache.expire), 0,
156a9643ea8Slogwang     "Expire time of TCP hostcache entries");
157a9643ea8Slogwang 
158a9643ea8Slogwang SYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, prune, CTLFLAG_VNET | CTLFLAG_RW,
159a9643ea8Slogwang     &VNET_NAME(tcp_hostcache.prune), 0,
160a9643ea8Slogwang     "Time between purge runs");
161a9643ea8Slogwang 
162a9643ea8Slogwang SYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, purge, CTLFLAG_VNET | CTLFLAG_RW,
163a9643ea8Slogwang     &VNET_NAME(tcp_hostcache.purgeall), 0,
164a9643ea8Slogwang     "Expire all entires on next purge run");
165a9643ea8Slogwang 
166a9643ea8Slogwang SYSCTL_PROC(_net_inet_tcp_hostcache, OID_AUTO, list,
167*22ce4affSfengbojiang     CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_SKIP | CTLFLAG_MPSAFE,
168*22ce4affSfengbojiang     0, 0, sysctl_tcp_hc_list, "A",
169*22ce4affSfengbojiang     "List of all hostcache entries");
170a9643ea8Slogwang 
171a9643ea8Slogwang SYSCTL_PROC(_net_inet_tcp_hostcache, OID_AUTO, purgenow,
172*22ce4affSfengbojiang     CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_MPSAFE,
173*22ce4affSfengbojiang     NULL, 0, sysctl_tcp_hc_purgenow, "I",
174*22ce4affSfengbojiang     "Immediately purge all entries");
175a9643ea8Slogwang 
176a9643ea8Slogwang static MALLOC_DEFINE(M_HOSTCACHE, "hostcache", "TCP hostcache");
177a9643ea8Slogwang 
178a9643ea8Slogwang #define HOSTCACHE_HASH(ip) \
179a9643ea8Slogwang 	(((ip)->s_addr ^ ((ip)->s_addr >> 7) ^ ((ip)->s_addr >> 17)) &	\
180a9643ea8Slogwang 	  V_tcp_hostcache.hashmask)
181a9643ea8Slogwang 
182a9643ea8Slogwang /* XXX: What is the recommended hash to get good entropy for IPv6 addresses? */
183a9643ea8Slogwang #define HOSTCACHE_HASH6(ip6)				\
184a9643ea8Slogwang 	(((ip6)->s6_addr32[0] ^				\
185a9643ea8Slogwang 	  (ip6)->s6_addr32[1] ^				\
186a9643ea8Slogwang 	  (ip6)->s6_addr32[2] ^				\
187a9643ea8Slogwang 	  (ip6)->s6_addr32[3]) &			\
188a9643ea8Slogwang 	 V_tcp_hostcache.hashmask)
189a9643ea8Slogwang 
190a9643ea8Slogwang #define THC_LOCK(lp)		mtx_lock(lp)
191a9643ea8Slogwang #define THC_UNLOCK(lp)		mtx_unlock(lp)
192a9643ea8Slogwang 
193a9643ea8Slogwang void
tcp_hc_init(void)194a9643ea8Slogwang tcp_hc_init(void)
195a9643ea8Slogwang {
196a9643ea8Slogwang 	u_int cache_limit;
197a9643ea8Slogwang 	int i;
198a9643ea8Slogwang 
199a9643ea8Slogwang 	/*
200a9643ea8Slogwang 	 * Initialize hostcache structures.
201a9643ea8Slogwang 	 */
202a9643ea8Slogwang 	V_tcp_hostcache.cache_count = 0;
203a9643ea8Slogwang 	V_tcp_hostcache.hashsize = TCP_HOSTCACHE_HASHSIZE;
204a9643ea8Slogwang 	V_tcp_hostcache.bucket_limit = TCP_HOSTCACHE_BUCKETLIMIT;
205a9643ea8Slogwang 	V_tcp_hostcache.expire = TCP_HOSTCACHE_EXPIRE;
206a9643ea8Slogwang 	V_tcp_hostcache.prune = TCP_HOSTCACHE_PRUNE;
207a9643ea8Slogwang 
208a9643ea8Slogwang 	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.hashsize",
209a9643ea8Slogwang 	    &V_tcp_hostcache.hashsize);
210a9643ea8Slogwang 	if (!powerof2(V_tcp_hostcache.hashsize)) {
211a9643ea8Slogwang 		printf("WARNING: hostcache hash size is not a power of 2.\n");
212a9643ea8Slogwang 		V_tcp_hostcache.hashsize = TCP_HOSTCACHE_HASHSIZE; /* default */
213a9643ea8Slogwang 	}
214a9643ea8Slogwang 	V_tcp_hostcache.hashmask = V_tcp_hostcache.hashsize - 1;
215a9643ea8Slogwang 
216a9643ea8Slogwang 	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.bucketlimit",
217a9643ea8Slogwang 	    &V_tcp_hostcache.bucket_limit);
218a9643ea8Slogwang 
219a9643ea8Slogwang 	cache_limit = V_tcp_hostcache.hashsize * V_tcp_hostcache.bucket_limit;
220a9643ea8Slogwang 	V_tcp_hostcache.cache_limit = cache_limit;
221a9643ea8Slogwang 	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.cachelimit",
222a9643ea8Slogwang 	    &V_tcp_hostcache.cache_limit);
223a9643ea8Slogwang 	if (V_tcp_hostcache.cache_limit > cache_limit)
224a9643ea8Slogwang 		V_tcp_hostcache.cache_limit = cache_limit;
225a9643ea8Slogwang 
226a9643ea8Slogwang 	/*
227a9643ea8Slogwang 	 * Allocate the hash table.
228a9643ea8Slogwang 	 */
229a9643ea8Slogwang 	V_tcp_hostcache.hashbase = (struct hc_head *)
230a9643ea8Slogwang 	    malloc(V_tcp_hostcache.hashsize * sizeof(struct hc_head),
231a9643ea8Slogwang 		   M_HOSTCACHE, M_WAITOK | M_ZERO);
232a9643ea8Slogwang 
233a9643ea8Slogwang 	/*
234a9643ea8Slogwang 	 * Initialize the hash buckets.
235a9643ea8Slogwang 	 */
236a9643ea8Slogwang 	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
237a9643ea8Slogwang 		TAILQ_INIT(&V_tcp_hostcache.hashbase[i].hch_bucket);
238a9643ea8Slogwang 		V_tcp_hostcache.hashbase[i].hch_length = 0;
239a9643ea8Slogwang 		mtx_init(&V_tcp_hostcache.hashbase[i].hch_mtx, "tcp_hc_entry",
240a9643ea8Slogwang 			  NULL, MTX_DEF);
241a9643ea8Slogwang 	}
242a9643ea8Slogwang 
243a9643ea8Slogwang 	/*
244a9643ea8Slogwang 	 * Allocate the hostcache entries.
245a9643ea8Slogwang 	 */
246a9643ea8Slogwang 	V_tcp_hostcache.zone =
247a9643ea8Slogwang 	    uma_zcreate("hostcache", sizeof(struct hc_metrics),
248a9643ea8Slogwang 	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
249a9643ea8Slogwang 	uma_zone_set_max(V_tcp_hostcache.zone, V_tcp_hostcache.cache_limit);
250a9643ea8Slogwang 
251a9643ea8Slogwang 	/*
252a9643ea8Slogwang 	 * Set up periodic cache cleanup.
253a9643ea8Slogwang 	 */
254a9643ea8Slogwang 	callout_init(&V_tcp_hc_callout, 1);
255a9643ea8Slogwang 	callout_reset(&V_tcp_hc_callout, V_tcp_hostcache.prune * hz,
256a9643ea8Slogwang 	    tcp_hc_purge, curvnet);
257a9643ea8Slogwang }
258a9643ea8Slogwang 
259a9643ea8Slogwang #ifdef VIMAGE
260a9643ea8Slogwang void
tcp_hc_destroy(void)261a9643ea8Slogwang tcp_hc_destroy(void)
262a9643ea8Slogwang {
263a9643ea8Slogwang 	int i;
264a9643ea8Slogwang 
265a9643ea8Slogwang 	callout_drain(&V_tcp_hc_callout);
266a9643ea8Slogwang 
267a9643ea8Slogwang 	/* Purge all hc entries. */
268a9643ea8Slogwang 	tcp_hc_purge_internal(1);
269a9643ea8Slogwang 
270a9643ea8Slogwang 	/* Free the uma zone and the allocated hash table. */
271a9643ea8Slogwang 	uma_zdestroy(V_tcp_hostcache.zone);
272a9643ea8Slogwang 
273a9643ea8Slogwang 	for (i = 0; i < V_tcp_hostcache.hashsize; i++)
274a9643ea8Slogwang 		mtx_destroy(&V_tcp_hostcache.hashbase[i].hch_mtx);
275a9643ea8Slogwang 	free(V_tcp_hostcache.hashbase, M_HOSTCACHE);
276a9643ea8Slogwang }
277a9643ea8Slogwang #endif
278a9643ea8Slogwang 
279a9643ea8Slogwang /*
280a9643ea8Slogwang  * Internal function: look up an entry in the hostcache or return NULL.
281a9643ea8Slogwang  *
282a9643ea8Slogwang  * If an entry has been returned, the caller becomes responsible for
283a9643ea8Slogwang  * unlocking the bucket row after he is done reading/modifying the entry.
284a9643ea8Slogwang  */
285a9643ea8Slogwang static struct hc_metrics *
tcp_hc_lookup(struct in_conninfo * inc)286a9643ea8Slogwang tcp_hc_lookup(struct in_conninfo *inc)
287a9643ea8Slogwang {
288a9643ea8Slogwang 	int hash;
289a9643ea8Slogwang 	struct hc_head *hc_head;
290a9643ea8Slogwang 	struct hc_metrics *hc_entry;
291a9643ea8Slogwang 
292*22ce4affSfengbojiang 	if (!V_tcp_use_hostcache)
293*22ce4affSfengbojiang 		return NULL;
294*22ce4affSfengbojiang 
295a9643ea8Slogwang 	KASSERT(inc != NULL, ("tcp_hc_lookup with NULL in_conninfo pointer"));
296a9643ea8Slogwang 
297a9643ea8Slogwang 	/*
298a9643ea8Slogwang 	 * Hash the foreign ip address.
299a9643ea8Slogwang 	 */
300a9643ea8Slogwang 	if (inc->inc_flags & INC_ISIPV6)
301a9643ea8Slogwang 		hash = HOSTCACHE_HASH6(&inc->inc6_faddr);
302a9643ea8Slogwang 	else
303a9643ea8Slogwang 		hash = HOSTCACHE_HASH(&inc->inc_faddr);
304a9643ea8Slogwang 
305a9643ea8Slogwang 	hc_head = &V_tcp_hostcache.hashbase[hash];
306a9643ea8Slogwang 
307a9643ea8Slogwang 	/*
308a9643ea8Slogwang 	 * Acquire lock for this bucket row; we release the lock if we don't
309a9643ea8Slogwang 	 * find an entry, otherwise the caller has to unlock after he is
310a9643ea8Slogwang 	 * done.
311a9643ea8Slogwang 	 */
312a9643ea8Slogwang 	THC_LOCK(&hc_head->hch_mtx);
313a9643ea8Slogwang 
314a9643ea8Slogwang 	/*
315a9643ea8Slogwang 	 * Iterate through entries in bucket row looking for a match.
316a9643ea8Slogwang 	 */
317a9643ea8Slogwang 	TAILQ_FOREACH(hc_entry, &hc_head->hch_bucket, rmx_q) {
318a9643ea8Slogwang 		if (inc->inc_flags & INC_ISIPV6) {
319a9643ea8Slogwang 			/* XXX: check ip6_zoneid */
320a9643ea8Slogwang 			if (memcmp(&inc->inc6_faddr, &hc_entry->ip6,
321a9643ea8Slogwang 			    sizeof(inc->inc6_faddr)) == 0)
322a9643ea8Slogwang 				return hc_entry;
323a9643ea8Slogwang 		} else {
324a9643ea8Slogwang 			if (memcmp(&inc->inc_faddr, &hc_entry->ip4,
325a9643ea8Slogwang 			    sizeof(inc->inc_faddr)) == 0)
326a9643ea8Slogwang 				return hc_entry;
327a9643ea8Slogwang 		}
328a9643ea8Slogwang 	}
329a9643ea8Slogwang 
330a9643ea8Slogwang 	/*
331a9643ea8Slogwang 	 * We were unsuccessful and didn't find anything.
332a9643ea8Slogwang 	 */
333a9643ea8Slogwang 	THC_UNLOCK(&hc_head->hch_mtx);
334a9643ea8Slogwang 	return NULL;
335a9643ea8Slogwang }
336a9643ea8Slogwang 
337a9643ea8Slogwang /*
338a9643ea8Slogwang  * Internal function: insert an entry into the hostcache or return NULL if
339a9643ea8Slogwang  * unable to allocate a new one.
340a9643ea8Slogwang  *
341a9643ea8Slogwang  * If an entry has been returned, the caller becomes responsible for
342a9643ea8Slogwang  * unlocking the bucket row after he is done reading/modifying the entry.
343a9643ea8Slogwang  */
344a9643ea8Slogwang static struct hc_metrics *
tcp_hc_insert(struct in_conninfo * inc)345a9643ea8Slogwang tcp_hc_insert(struct in_conninfo *inc)
346a9643ea8Slogwang {
347a9643ea8Slogwang 	int hash;
348a9643ea8Slogwang 	struct hc_head *hc_head;
349a9643ea8Slogwang 	struct hc_metrics *hc_entry;
350a9643ea8Slogwang 
351*22ce4affSfengbojiang 	if (!V_tcp_use_hostcache)
352*22ce4affSfengbojiang 		return NULL;
353*22ce4affSfengbojiang 
354a9643ea8Slogwang 	KASSERT(inc != NULL, ("tcp_hc_insert with NULL in_conninfo pointer"));
355a9643ea8Slogwang 
356a9643ea8Slogwang 	/*
357a9643ea8Slogwang 	 * Hash the foreign ip address.
358a9643ea8Slogwang 	 */
359a9643ea8Slogwang 	if (inc->inc_flags & INC_ISIPV6)
360a9643ea8Slogwang 		hash = HOSTCACHE_HASH6(&inc->inc6_faddr);
361a9643ea8Slogwang 	else
362a9643ea8Slogwang 		hash = HOSTCACHE_HASH(&inc->inc_faddr);
363a9643ea8Slogwang 
364a9643ea8Slogwang 	hc_head = &V_tcp_hostcache.hashbase[hash];
365a9643ea8Slogwang 
366a9643ea8Slogwang 	/*
367a9643ea8Slogwang 	 * Acquire lock for this bucket row; we release the lock if we don't
368a9643ea8Slogwang 	 * find an entry, otherwise the caller has to unlock after he is
369a9643ea8Slogwang 	 * done.
370a9643ea8Slogwang 	 */
371a9643ea8Slogwang 	THC_LOCK(&hc_head->hch_mtx);
372a9643ea8Slogwang 
373a9643ea8Slogwang 	/*
374a9643ea8Slogwang 	 * If the bucket limit is reached, reuse the least-used element.
375a9643ea8Slogwang 	 */
376a9643ea8Slogwang 	if (hc_head->hch_length >= V_tcp_hostcache.bucket_limit ||
377a9643ea8Slogwang 	    V_tcp_hostcache.cache_count >= V_tcp_hostcache.cache_limit) {
378a9643ea8Slogwang 		hc_entry = TAILQ_LAST(&hc_head->hch_bucket, hc_qhead);
379a9643ea8Slogwang 		/*
380a9643ea8Slogwang 		 * At first we were dropping the last element, just to
381a9643ea8Slogwang 		 * reacquire it in the next two lines again, which isn't very
382a9643ea8Slogwang 		 * efficient.  Instead just reuse the least used element.
383a9643ea8Slogwang 		 * We may drop something that is still "in-use" but we can be
384a9643ea8Slogwang 		 * "lossy".
385a9643ea8Slogwang 		 * Just give up if this bucket row is empty and we don't have
386a9643ea8Slogwang 		 * anything to replace.
387a9643ea8Slogwang 		 */
388a9643ea8Slogwang 		if (hc_entry == NULL) {
389a9643ea8Slogwang 			THC_UNLOCK(&hc_head->hch_mtx);
390a9643ea8Slogwang 			return NULL;
391a9643ea8Slogwang 		}
392a9643ea8Slogwang 		TAILQ_REMOVE(&hc_head->hch_bucket, hc_entry, rmx_q);
393a9643ea8Slogwang 		V_tcp_hostcache.hashbase[hash].hch_length--;
394a9643ea8Slogwang 		V_tcp_hostcache.cache_count--;
395a9643ea8Slogwang 		TCPSTAT_INC(tcps_hc_bucketoverflow);
396a9643ea8Slogwang #if 0
397a9643ea8Slogwang 		uma_zfree(V_tcp_hostcache.zone, hc_entry);
398a9643ea8Slogwang #endif
399a9643ea8Slogwang 	} else {
400a9643ea8Slogwang 		/*
401a9643ea8Slogwang 		 * Allocate a new entry, or balk if not possible.
402a9643ea8Slogwang 		 */
403a9643ea8Slogwang 		hc_entry = uma_zalloc(V_tcp_hostcache.zone, M_NOWAIT);
404a9643ea8Slogwang 		if (hc_entry == NULL) {
405a9643ea8Slogwang 			THC_UNLOCK(&hc_head->hch_mtx);
406a9643ea8Slogwang 			return NULL;
407a9643ea8Slogwang 		}
408a9643ea8Slogwang 	}
409a9643ea8Slogwang 
410a9643ea8Slogwang 	/*
411a9643ea8Slogwang 	 * Initialize basic information of hostcache entry.
412a9643ea8Slogwang 	 */
413a9643ea8Slogwang 	bzero(hc_entry, sizeof(*hc_entry));
414a9643ea8Slogwang 	if (inc->inc_flags & INC_ISIPV6) {
415a9643ea8Slogwang 		hc_entry->ip6 = inc->inc6_faddr;
416a9643ea8Slogwang 		hc_entry->ip6_zoneid = inc->inc6_zoneid;
417a9643ea8Slogwang 	} else
418a9643ea8Slogwang 		hc_entry->ip4 = inc->inc_faddr;
419a9643ea8Slogwang 	hc_entry->rmx_head = hc_head;
420a9643ea8Slogwang 	hc_entry->rmx_expire = V_tcp_hostcache.expire;
421a9643ea8Slogwang 
422a9643ea8Slogwang 	/*
423a9643ea8Slogwang 	 * Put it upfront.
424a9643ea8Slogwang 	 */
425a9643ea8Slogwang 	TAILQ_INSERT_HEAD(&hc_head->hch_bucket, hc_entry, rmx_q);
426a9643ea8Slogwang 	V_tcp_hostcache.hashbase[hash].hch_length++;
427a9643ea8Slogwang 	V_tcp_hostcache.cache_count++;
428a9643ea8Slogwang 	TCPSTAT_INC(tcps_hc_added);
429a9643ea8Slogwang 
430a9643ea8Slogwang 	return hc_entry;
431a9643ea8Slogwang }
432a9643ea8Slogwang 
433a9643ea8Slogwang /*
434a9643ea8Slogwang  * External function: look up an entry in the hostcache and fill out the
435a9643ea8Slogwang  * supplied TCP metrics structure.  Fills in NULL when no entry was found or
436a9643ea8Slogwang  * a value is not set.
437a9643ea8Slogwang  */
438a9643ea8Slogwang void
tcp_hc_get(struct in_conninfo * inc,struct hc_metrics_lite * hc_metrics_lite)439a9643ea8Slogwang tcp_hc_get(struct in_conninfo *inc, struct hc_metrics_lite *hc_metrics_lite)
440a9643ea8Slogwang {
441a9643ea8Slogwang 	struct hc_metrics *hc_entry;
442a9643ea8Slogwang 
443*22ce4affSfengbojiang 	if (!V_tcp_use_hostcache) {
444*22ce4affSfengbojiang 		bzero(hc_metrics_lite, sizeof(*hc_metrics_lite));
445*22ce4affSfengbojiang 		return;
446*22ce4affSfengbojiang 	}
447*22ce4affSfengbojiang 
448a9643ea8Slogwang 	/*
449a9643ea8Slogwang 	 * Find the right bucket.
450a9643ea8Slogwang 	 */
451a9643ea8Slogwang 	hc_entry = tcp_hc_lookup(inc);
452a9643ea8Slogwang 
453a9643ea8Slogwang 	/*
454a9643ea8Slogwang 	 * If we don't have an existing object.
455a9643ea8Slogwang 	 */
456a9643ea8Slogwang 	if (hc_entry == NULL) {
457a9643ea8Slogwang 		bzero(hc_metrics_lite, sizeof(*hc_metrics_lite));
458a9643ea8Slogwang 		return;
459a9643ea8Slogwang 	}
460a9643ea8Slogwang 	hc_entry->rmx_hits++;
461a9643ea8Slogwang 	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
462a9643ea8Slogwang 
463a9643ea8Slogwang 	hc_metrics_lite->rmx_mtu = hc_entry->rmx_mtu;
464a9643ea8Slogwang 	hc_metrics_lite->rmx_ssthresh = hc_entry->rmx_ssthresh;
465a9643ea8Slogwang 	hc_metrics_lite->rmx_rtt = hc_entry->rmx_rtt;
466a9643ea8Slogwang 	hc_metrics_lite->rmx_rttvar = hc_entry->rmx_rttvar;
467a9643ea8Slogwang 	hc_metrics_lite->rmx_cwnd = hc_entry->rmx_cwnd;
468a9643ea8Slogwang 	hc_metrics_lite->rmx_sendpipe = hc_entry->rmx_sendpipe;
469a9643ea8Slogwang 	hc_metrics_lite->rmx_recvpipe = hc_entry->rmx_recvpipe;
470a9643ea8Slogwang 
471a9643ea8Slogwang 	/*
472a9643ea8Slogwang 	 * Unlock bucket row.
473a9643ea8Slogwang 	 */
474a9643ea8Slogwang 	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
475a9643ea8Slogwang }
476a9643ea8Slogwang 
477a9643ea8Slogwang /*
478a9643ea8Slogwang  * External function: look up an entry in the hostcache and return the
479*22ce4affSfengbojiang  * discovered path MTU.  Returns 0 if no entry is found or value is not
480a9643ea8Slogwang  * set.
481a9643ea8Slogwang  */
482*22ce4affSfengbojiang uint32_t
tcp_hc_getmtu(struct in_conninfo * inc)483a9643ea8Slogwang tcp_hc_getmtu(struct in_conninfo *inc)
484a9643ea8Slogwang {
485a9643ea8Slogwang 	struct hc_metrics *hc_entry;
486*22ce4affSfengbojiang 	uint32_t mtu;
487*22ce4affSfengbojiang 
488*22ce4affSfengbojiang 	if (!V_tcp_use_hostcache)
489*22ce4affSfengbojiang 		return 0;
490a9643ea8Slogwang 
491a9643ea8Slogwang 	hc_entry = tcp_hc_lookup(inc);
492a9643ea8Slogwang 	if (hc_entry == NULL) {
493a9643ea8Slogwang 		return 0;
494a9643ea8Slogwang 	}
495a9643ea8Slogwang 	hc_entry->rmx_hits++;
496a9643ea8Slogwang 	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
497a9643ea8Slogwang 
498a9643ea8Slogwang 	mtu = hc_entry->rmx_mtu;
499a9643ea8Slogwang 	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
500a9643ea8Slogwang 	return mtu;
501a9643ea8Slogwang }
502a9643ea8Slogwang 
503a9643ea8Slogwang /*
504a9643ea8Slogwang  * External function: update the MTU value of an entry in the hostcache.
505a9643ea8Slogwang  * Creates a new entry if none was found.
506a9643ea8Slogwang  */
507a9643ea8Slogwang void
tcp_hc_updatemtu(struct in_conninfo * inc,uint32_t mtu)508*22ce4affSfengbojiang tcp_hc_updatemtu(struct in_conninfo *inc, uint32_t mtu)
509a9643ea8Slogwang {
510a9643ea8Slogwang 	struct hc_metrics *hc_entry;
511a9643ea8Slogwang 
512*22ce4affSfengbojiang 	if (!V_tcp_use_hostcache)
513*22ce4affSfengbojiang 		return;
514*22ce4affSfengbojiang 
515a9643ea8Slogwang 	/*
516a9643ea8Slogwang 	 * Find the right bucket.
517a9643ea8Slogwang 	 */
518a9643ea8Slogwang 	hc_entry = tcp_hc_lookup(inc);
519a9643ea8Slogwang 
520a9643ea8Slogwang 	/*
521a9643ea8Slogwang 	 * If we don't have an existing object, try to insert a new one.
522a9643ea8Slogwang 	 */
523a9643ea8Slogwang 	if (hc_entry == NULL) {
524a9643ea8Slogwang 		hc_entry = tcp_hc_insert(inc);
525a9643ea8Slogwang 		if (hc_entry == NULL)
526a9643ea8Slogwang 			return;
527a9643ea8Slogwang 	}
528a9643ea8Slogwang 	hc_entry->rmx_updates++;
529a9643ea8Slogwang 	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
530a9643ea8Slogwang 
531a9643ea8Slogwang 	hc_entry->rmx_mtu = mtu;
532a9643ea8Slogwang 
533a9643ea8Slogwang 	/*
534a9643ea8Slogwang 	 * Put it upfront so we find it faster next time.
535a9643ea8Slogwang 	 */
536a9643ea8Slogwang 	TAILQ_REMOVE(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
537a9643ea8Slogwang 	TAILQ_INSERT_HEAD(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
538a9643ea8Slogwang 
539a9643ea8Slogwang 	/*
540a9643ea8Slogwang 	 * Unlock bucket row.
541a9643ea8Slogwang 	 */
542a9643ea8Slogwang 	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
543a9643ea8Slogwang }
544a9643ea8Slogwang 
545a9643ea8Slogwang /*
546a9643ea8Slogwang  * External function: update the TCP metrics of an entry in the hostcache.
547a9643ea8Slogwang  * Creates a new entry if none was found.
548a9643ea8Slogwang  */
549a9643ea8Slogwang void
tcp_hc_update(struct in_conninfo * inc,struct hc_metrics_lite * hcml)550a9643ea8Slogwang tcp_hc_update(struct in_conninfo *inc, struct hc_metrics_lite *hcml)
551a9643ea8Slogwang {
552a9643ea8Slogwang 	struct hc_metrics *hc_entry;
553a9643ea8Slogwang 
554*22ce4affSfengbojiang 	if (!V_tcp_use_hostcache)
555*22ce4affSfengbojiang 		return;
556*22ce4affSfengbojiang 
557a9643ea8Slogwang 	hc_entry = tcp_hc_lookup(inc);
558a9643ea8Slogwang 	if (hc_entry == NULL) {
559a9643ea8Slogwang 		hc_entry = tcp_hc_insert(inc);
560a9643ea8Slogwang 		if (hc_entry == NULL)
561a9643ea8Slogwang 			return;
562a9643ea8Slogwang 	}
563a9643ea8Slogwang 	hc_entry->rmx_updates++;
564a9643ea8Slogwang 	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
565a9643ea8Slogwang 
566a9643ea8Slogwang 	if (hcml->rmx_rtt != 0) {
567a9643ea8Slogwang 		if (hc_entry->rmx_rtt == 0)
568a9643ea8Slogwang 			hc_entry->rmx_rtt = hcml->rmx_rtt;
569a9643ea8Slogwang 		else
570*22ce4affSfengbojiang 			hc_entry->rmx_rtt = ((uint64_t)hc_entry->rmx_rtt +
571*22ce4affSfengbojiang 			    (uint64_t)hcml->rmx_rtt) / 2;
572a9643ea8Slogwang 		TCPSTAT_INC(tcps_cachedrtt);
573a9643ea8Slogwang 	}
574a9643ea8Slogwang 	if (hcml->rmx_rttvar != 0) {
575a9643ea8Slogwang 	        if (hc_entry->rmx_rttvar == 0)
576a9643ea8Slogwang 			hc_entry->rmx_rttvar = hcml->rmx_rttvar;
577a9643ea8Slogwang 		else
578*22ce4affSfengbojiang 			hc_entry->rmx_rttvar = ((uint64_t)hc_entry->rmx_rttvar +
579*22ce4affSfengbojiang 			    (uint64_t)hcml->rmx_rttvar) / 2;
580a9643ea8Slogwang 		TCPSTAT_INC(tcps_cachedrttvar);
581a9643ea8Slogwang 	}
582a9643ea8Slogwang 	if (hcml->rmx_ssthresh != 0) {
583a9643ea8Slogwang 		if (hc_entry->rmx_ssthresh == 0)
584a9643ea8Slogwang 			hc_entry->rmx_ssthresh = hcml->rmx_ssthresh;
585a9643ea8Slogwang 		else
586a9643ea8Slogwang 			hc_entry->rmx_ssthresh =
587a9643ea8Slogwang 			    (hc_entry->rmx_ssthresh + hcml->rmx_ssthresh) / 2;
588a9643ea8Slogwang 		TCPSTAT_INC(tcps_cachedssthresh);
589a9643ea8Slogwang 	}
590a9643ea8Slogwang 	if (hcml->rmx_cwnd != 0) {
591a9643ea8Slogwang 		if (hc_entry->rmx_cwnd == 0)
592a9643ea8Slogwang 			hc_entry->rmx_cwnd = hcml->rmx_cwnd;
593a9643ea8Slogwang 		else
594*22ce4affSfengbojiang 			hc_entry->rmx_cwnd = ((uint64_t)hc_entry->rmx_cwnd +
595*22ce4affSfengbojiang 			    (uint64_t)hcml->rmx_cwnd) / 2;
596a9643ea8Slogwang 		/* TCPSTAT_INC(tcps_cachedcwnd); */
597a9643ea8Slogwang 	}
598a9643ea8Slogwang 	if (hcml->rmx_sendpipe != 0) {
599a9643ea8Slogwang 		if (hc_entry->rmx_sendpipe == 0)
600a9643ea8Slogwang 			hc_entry->rmx_sendpipe = hcml->rmx_sendpipe;
601a9643ea8Slogwang 		else
602a9643ea8Slogwang 			hc_entry->rmx_sendpipe =
603*22ce4affSfengbojiang 			    ((uint64_t)hc_entry->rmx_sendpipe +
604*22ce4affSfengbojiang 			    (uint64_t)hcml->rmx_sendpipe) /2;
605a9643ea8Slogwang 		/* TCPSTAT_INC(tcps_cachedsendpipe); */
606a9643ea8Slogwang 	}
607a9643ea8Slogwang 	if (hcml->rmx_recvpipe != 0) {
608a9643ea8Slogwang 		if (hc_entry->rmx_recvpipe == 0)
609a9643ea8Slogwang 			hc_entry->rmx_recvpipe = hcml->rmx_recvpipe;
610a9643ea8Slogwang 		else
611a9643ea8Slogwang 			hc_entry->rmx_recvpipe =
612*22ce4affSfengbojiang 			    ((uint64_t)hc_entry->rmx_recvpipe +
613*22ce4affSfengbojiang 			    (uint64_t)hcml->rmx_recvpipe) /2;
614a9643ea8Slogwang 		/* TCPSTAT_INC(tcps_cachedrecvpipe); */
615a9643ea8Slogwang 	}
616a9643ea8Slogwang 
617a9643ea8Slogwang 	TAILQ_REMOVE(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
618a9643ea8Slogwang 	TAILQ_INSERT_HEAD(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
619a9643ea8Slogwang 	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
620a9643ea8Slogwang }
621a9643ea8Slogwang 
622a9643ea8Slogwang /*
623a9643ea8Slogwang  * Sysctl function: prints the list and values of all hostcache entries in
624a9643ea8Slogwang  * unsorted order.
625a9643ea8Slogwang  */
626a9643ea8Slogwang static int
sysctl_tcp_hc_list(SYSCTL_HANDLER_ARGS)627a9643ea8Slogwang sysctl_tcp_hc_list(SYSCTL_HANDLER_ARGS)
628a9643ea8Slogwang {
629a9643ea8Slogwang 	const int linesize = 128;
630a9643ea8Slogwang 	struct sbuf sb;
631a9643ea8Slogwang 	int i, error;
632a9643ea8Slogwang 	struct hc_metrics *hc_entry;
633*22ce4affSfengbojiang 	char ip4buf[INET_ADDRSTRLEN];
634a9643ea8Slogwang #ifdef INET6
635a9643ea8Slogwang 	char ip6buf[INET6_ADDRSTRLEN];
636a9643ea8Slogwang #endif
637a9643ea8Slogwang 
638*22ce4affSfengbojiang 	if (jailed_without_vnet(curthread->td_ucred) != 0)
639*22ce4affSfengbojiang 		return (EPERM);
640*22ce4affSfengbojiang 
641a9643ea8Slogwang 	sbuf_new(&sb, NULL, linesize * (V_tcp_hostcache.cache_count + 1),
642a9643ea8Slogwang 		SBUF_INCLUDENUL);
643a9643ea8Slogwang 
644a9643ea8Slogwang 	sbuf_printf(&sb,
645a9643ea8Slogwang 	        "\nIP address        MTU  SSTRESH      RTT   RTTVAR "
646a9643ea8Slogwang 		"    CWND SENDPIPE RECVPIPE HITS  UPD  EXP\n");
647a9643ea8Slogwang 
648a9643ea8Slogwang #define msec(u) (((u) + 500) / 1000)
649a9643ea8Slogwang 	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
650a9643ea8Slogwang 		THC_LOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
651a9643ea8Slogwang 		TAILQ_FOREACH(hc_entry, &V_tcp_hostcache.hashbase[i].hch_bucket,
652a9643ea8Slogwang 			      rmx_q) {
653a9643ea8Slogwang 			sbuf_printf(&sb,
654*22ce4affSfengbojiang 			    "%-15s %5u %8u %6lums %6lums %8u %8u %8u %4lu "
655a9643ea8Slogwang 			    "%4lu %4i\n",
656*22ce4affSfengbojiang 			    hc_entry->ip4.s_addr ?
657*22ce4affSfengbojiang 			        inet_ntoa_r(hc_entry->ip4, ip4buf) :
658a9643ea8Slogwang #ifdef INET6
659a9643ea8Slogwang 				ip6_sprintf(ip6buf, &hc_entry->ip6),
660a9643ea8Slogwang #else
661a9643ea8Slogwang 				"IPv6?",
662a9643ea8Slogwang #endif
663a9643ea8Slogwang 			    hc_entry->rmx_mtu,
664a9643ea8Slogwang 			    hc_entry->rmx_ssthresh,
665*22ce4affSfengbojiang 			    msec((u_long)hc_entry->rmx_rtt *
666a9643ea8Slogwang 				(RTM_RTTUNIT / (hz * TCP_RTT_SCALE))),
667*22ce4affSfengbojiang 			    msec((u_long)hc_entry->rmx_rttvar *
668a9643ea8Slogwang 				(RTM_RTTUNIT / (hz * TCP_RTTVAR_SCALE))),
669a9643ea8Slogwang 			    hc_entry->rmx_cwnd,
670a9643ea8Slogwang 			    hc_entry->rmx_sendpipe,
671a9643ea8Slogwang 			    hc_entry->rmx_recvpipe,
672a9643ea8Slogwang 			    hc_entry->rmx_hits,
673a9643ea8Slogwang 			    hc_entry->rmx_updates,
674a9643ea8Slogwang 			    hc_entry->rmx_expire);
675a9643ea8Slogwang 		}
676a9643ea8Slogwang 		THC_UNLOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
677a9643ea8Slogwang 	}
678a9643ea8Slogwang #undef msec
679a9643ea8Slogwang 	error = sbuf_finish(&sb);
680a9643ea8Slogwang 	if (error == 0)
681a9643ea8Slogwang 		error = SYSCTL_OUT(req, sbuf_data(&sb), sbuf_len(&sb));
682a9643ea8Slogwang 	sbuf_delete(&sb);
683a9643ea8Slogwang 	return(error);
684a9643ea8Slogwang }
685a9643ea8Slogwang 
686a9643ea8Slogwang /*
687a9643ea8Slogwang  * Caller has to make sure the curvnet is set properly.
688a9643ea8Slogwang  */
689a9643ea8Slogwang static void
tcp_hc_purge_internal(int all)690a9643ea8Slogwang tcp_hc_purge_internal(int all)
691a9643ea8Slogwang {
692a9643ea8Slogwang 	struct hc_metrics *hc_entry, *hc_next;
693a9643ea8Slogwang 	int i;
694a9643ea8Slogwang 
695a9643ea8Slogwang 	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
696a9643ea8Slogwang 		THC_LOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
697a9643ea8Slogwang 		TAILQ_FOREACH_SAFE(hc_entry,
698a9643ea8Slogwang 		    &V_tcp_hostcache.hashbase[i].hch_bucket, rmx_q, hc_next) {
699a9643ea8Slogwang 			if (all || hc_entry->rmx_expire <= 0) {
700a9643ea8Slogwang 				TAILQ_REMOVE(&V_tcp_hostcache.hashbase[i].hch_bucket,
701a9643ea8Slogwang 					      hc_entry, rmx_q);
702a9643ea8Slogwang 				uma_zfree(V_tcp_hostcache.zone, hc_entry);
703a9643ea8Slogwang 				V_tcp_hostcache.hashbase[i].hch_length--;
704a9643ea8Slogwang 				V_tcp_hostcache.cache_count--;
705a9643ea8Slogwang 			} else
706a9643ea8Slogwang 				hc_entry->rmx_expire -= V_tcp_hostcache.prune;
707a9643ea8Slogwang 		}
708a9643ea8Slogwang 		THC_UNLOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
709a9643ea8Slogwang 	}
710a9643ea8Slogwang }
711a9643ea8Slogwang 
712a9643ea8Slogwang /*
713a9643ea8Slogwang  * Expire and purge (old|all) entries in the tcp_hostcache.  Runs
714a9643ea8Slogwang  * periodically from the callout.
715a9643ea8Slogwang  */
716a9643ea8Slogwang static void
tcp_hc_purge(void * arg)717a9643ea8Slogwang tcp_hc_purge(void *arg)
718a9643ea8Slogwang {
719a9643ea8Slogwang 	CURVNET_SET((struct vnet *) arg);
720a9643ea8Slogwang 	int all = 0;
721a9643ea8Slogwang 
722a9643ea8Slogwang 	if (V_tcp_hostcache.purgeall) {
723a9643ea8Slogwang 		all = 1;
724a9643ea8Slogwang 		V_tcp_hostcache.purgeall = 0;
725a9643ea8Slogwang 	}
726a9643ea8Slogwang 
727a9643ea8Slogwang 	tcp_hc_purge_internal(all);
728a9643ea8Slogwang 
729a9643ea8Slogwang 	callout_reset(&V_tcp_hc_callout, V_tcp_hostcache.prune * hz,
730a9643ea8Slogwang 	    tcp_hc_purge, arg);
731a9643ea8Slogwang 	CURVNET_RESTORE();
732a9643ea8Slogwang }
733a9643ea8Slogwang 
734a9643ea8Slogwang /*
735a9643ea8Slogwang  * Expire and purge all entries in hostcache immediately.
736a9643ea8Slogwang  */
737a9643ea8Slogwang static int
sysctl_tcp_hc_purgenow(SYSCTL_HANDLER_ARGS)738a9643ea8Slogwang sysctl_tcp_hc_purgenow(SYSCTL_HANDLER_ARGS)
739a9643ea8Slogwang {
740a9643ea8Slogwang 	int error, val;
741a9643ea8Slogwang 
742a9643ea8Slogwang 	val = 0;
743a9643ea8Slogwang 	error = sysctl_handle_int(oidp, &val, 0, req);
744a9643ea8Slogwang 	if (error || !req->newptr)
745a9643ea8Slogwang 		return (error);
746a9643ea8Slogwang 
747a9643ea8Slogwang 	tcp_hc_purge_internal(1);
748a9643ea8Slogwang 
749a9643ea8Slogwang 	callout_reset(&V_tcp_hc_callout, V_tcp_hostcache.prune * hz,
750a9643ea8Slogwang 	    tcp_hc_purge, curvnet);
751a9643ea8Slogwang 
752a9643ea8Slogwang 	return (0);
753a9643ea8Slogwang }
754