1139823Simp/*-
2122922Sandre * Copyright (c) 2002 Andre Oppermann, Internet Business Solutions AG
3122922Sandre * All rights reserved.
4122922Sandre *
5122922Sandre * Redistribution and use in source and binary forms, with or without
6122922Sandre * modification, are permitted provided that the following conditions
7122922Sandre * are met:
8122922Sandre * 1. Redistributions of source code must retain the above copyright
9122922Sandre *    notice, this list of conditions and the following disclaimer.
10122922Sandre * 2. Redistributions in binary form must reproduce the above copyright
11122922Sandre *    notice, this list of conditions and the following disclaimer in the
12122922Sandre *    documentation and/or other materials provided with the distribution.
13122922Sandre * 3. The name of the author may not be used to endorse or promote
14122922Sandre *    products derived from this software without specific prior written
15122922Sandre *    permission.
16122922Sandre *
17122922Sandre * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18122922Sandre * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19122922Sandre * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20122922Sandre * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21122922Sandre * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22122922Sandre * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23122922Sandre * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24122922Sandre * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25122922Sandre * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26122922Sandre * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27122922Sandre * SUCH DAMAGE.
28122922Sandre */
29122922Sandre
30122922Sandre/*
31170030Srwatson * The tcp_hostcache moves the tcp-specific cached metrics from the routing
32170030Srwatson * table to a dedicated structure indexed by the remote IP address.  It keeps
33170030Srwatson * information on the measured TCP parameters of past TCP sessions to allow
34170030Srwatson * better initial start values to be used with later connections to/from the
35170030Srwatson * same source.  Depending on the network parameters (delay, bandwidth, max
36170030Srwatson * MTU, congestion window) between local and remote sites, this can lead to
37170030Srwatson * significant speed-ups for new TCP connections after the first one.
38122922Sandre *
39170030Srwatson * Due to the tcp_hostcache, all TCP-specific metrics information in the
40182411Srpaulo * routing table have been removed.  The inpcb no longer keeps a pointer to
41170030Srwatson * the routing entry, and protocol-initiated route cloning has been removed
42170030Srwatson * as well.  With these changes, the routing table has gone back to being
43170030Srwatson * more lightwight and only carries information related to packet forwarding.
44122922Sandre *
45170030Srwatson * tcp_hostcache is designed for multiple concurrent access in SMP
46170030Srwatson * environments and high contention.  All bucket rows have their own lock and
47170030Srwatson * thus multiple lookups and modifies can be done at the same time as long as
48170030Srwatson * they are in different bucket rows.  If a request for insertion of a new
49170030Srwatson * record can't be satisfied, it simply returns an empty structure.  Nobody
50170030Srwatson * and nothing outside of tcp_hostcache.c will ever point directly to any
51170030Srwatson * entry in the tcp_hostcache.  All communication is done in an
52170030Srwatson * object-oriented way and only functions of tcp_hostcache will manipulate
53170030Srwatson * hostcache entries.  Otherwise, we are unable to achieve good behaviour in
54170030Srwatson * concurrent access situations.  Since tcp_hostcache is only caching
55170030Srwatson * information, there are no fatal consequences if we either can't satisfy
56170030Srwatson * any particular request or have to drop/overwrite an existing entry because
57170030Srwatson * of bucket limit memory constrains.
58122922Sandre */
59122922Sandre
60122922Sandre/*
61122922Sandre * Many thanks to jlemon for basic structure of tcp_syncache which is being
62122922Sandre * followed here.
63122922Sandre */
64122922Sandre
65172467Ssilby#include <sys/cdefs.h>
66172467Ssilby__FBSDID("$FreeBSD$");
67172467Ssilby
68122922Sandre#include "opt_inet6.h"
69122922Sandre
70122922Sandre#include <sys/param.h>
71122922Sandre#include <sys/systm.h>
72122922Sandre#include <sys/kernel.h>
73122922Sandre#include <sys/lock.h>
74122922Sandre#include <sys/mutex.h>
75122922Sandre#include <sys/malloc.h>
76122922Sandre#include <sys/socket.h>
77122922Sandre#include <sys/socketvar.h>
78122922Sandre#include <sys/sysctl.h>
79122922Sandre
80122922Sandre#include <net/if.h>
81194739Sbz#include <net/route.h>
82196019Srwatson#include <net/vnet.h>
83122922Sandre
84122922Sandre#include <netinet/in.h>
85122922Sandre#include <netinet/in_systm.h>
86122922Sandre#include <netinet/ip.h>
87122922Sandre#include <netinet/in_var.h>
88122922Sandre#include <netinet/in_pcb.h>
89122922Sandre#include <netinet/ip_var.h>
90122922Sandre#ifdef INET6
91122922Sandre#include <netinet/ip6.h>
92122922Sandre#include <netinet6/ip6_var.h>
93122922Sandre#endif
94122922Sandre#include <netinet/tcp.h>
95122922Sandre#include <netinet/tcp_var.h>
96185571Sbz#include <netinet/tcp_hostcache.h>
97122922Sandre#ifdef INET6
98122922Sandre#include <netinet6/tcp6_var.h>
99122922Sandre#endif
100122922Sandre
101122922Sandre#include <vm/uma.h>
102122922Sandre
103122922Sandre/* Arbitrary values */
104122922Sandre#define TCP_HOSTCACHE_HASHSIZE		512
105122922Sandre#define TCP_HOSTCACHE_BUCKETLIMIT	30
106122922Sandre#define TCP_HOSTCACHE_EXPIRE		60*60	/* one hour */
107122922Sandre#define TCP_HOSTCACHE_PRUNE		5*60	/* every 5 minutes */
108122922Sandre
109215701Sdimstatic VNET_DEFINE(struct tcp_hostcache, tcp_hostcache);
110207369Sbz#define	V_tcp_hostcache		VNET(tcp_hostcache)
111207369Sbz
112215701Sdimstatic VNET_DEFINE(struct callout, tcp_hc_callout);
113195727Srwatson#define	V_tcp_hc_callout	VNET(tcp_hc_callout)
114195699Srwatson
115122922Sandrestatic struct hc_metrics *tcp_hc_lookup(struct in_conninfo *);
116122922Sandrestatic struct hc_metrics *tcp_hc_insert(struct in_conninfo *);
117122922Sandrestatic int sysctl_tcp_hc_list(SYSCTL_HANDLER_ARGS);
118203724Sbzstatic void tcp_hc_purge_internal(int);
119122922Sandrestatic void tcp_hc_purge(void *);
120122922Sandre
121227309Sedstatic SYSCTL_NODE(_net_inet_tcp, OID_AUTO, hostcache, CTLFLAG_RW, 0,
122182633Sbrooks    "TCP Host cache");
123122922Sandre
124217322SmdfSYSCTL_VNET_UINT(_net_inet_tcp_hostcache, OID_AUTO, cachelimit, CTLFLAG_RDTUN,
125195699Srwatson    &VNET_NAME(tcp_hostcache.cache_limit), 0,
126183550Szec    "Overall entry limit for hostcache");
127122922Sandre
128217322SmdfSYSCTL_VNET_UINT(_net_inet_tcp_hostcache, OID_AUTO, hashsize, CTLFLAG_RDTUN,
129195699Srwatson    &VNET_NAME(tcp_hostcache.hashsize), 0,
130183550Szec    "Size of TCP hostcache hashtable");
131122922Sandre
132217322SmdfSYSCTL_VNET_UINT(_net_inet_tcp_hostcache, OID_AUTO, bucketlimit,
133195699Srwatson    CTLFLAG_RDTUN, &VNET_NAME(tcp_hostcache.bucket_limit), 0,
134183550Szec    "Per-bucket hash limit for hostcache");
135122922Sandre
136217322SmdfSYSCTL_VNET_UINT(_net_inet_tcp_hostcache, OID_AUTO, count, CTLFLAG_RD,
137195699Srwatson     &VNET_NAME(tcp_hostcache.cache_count), 0,
138183550Szec    "Current number of entries in hostcache");
139122922Sandre
140195699SrwatsonSYSCTL_VNET_INT(_net_inet_tcp_hostcache, OID_AUTO, expire, CTLFLAG_RW,
141195699Srwatson    &VNET_NAME(tcp_hostcache.expire), 0,
142183550Szec    "Expire time of TCP hostcache entries");
143122922Sandre
144195699SrwatsonSYSCTL_VNET_INT(_net_inet_tcp_hostcache, OID_AUTO, prune, CTLFLAG_RW,
145195699Srwatson    &VNET_NAME(tcp_hostcache.prune), 0,
146195699Srwatson    "Time between purge runs");
147170434Syar
148195699SrwatsonSYSCTL_VNET_INT(_net_inet_tcp_hostcache, OID_AUTO, purge, CTLFLAG_RW,
149195699Srwatson    &VNET_NAME(tcp_hostcache.purgeall), 0,
150183550Szec    "Expire all entires on next purge run");
151122922Sandre
152122922SandreSYSCTL_PROC(_net_inet_tcp_hostcache, OID_AUTO, list,
153167784Sandre    CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_SKIP, 0, 0,
154167784Sandre    sysctl_tcp_hc_list, "A", "List of all hostcache entries");
155122922Sandre
156122922Sandre
157122922Sandrestatic MALLOC_DEFINE(M_HOSTCACHE, "hostcache", "TCP hostcache");
158122922Sandre
159122922Sandre#define HOSTCACHE_HASH(ip) \
160133874Srwatson	(((ip)->s_addr ^ ((ip)->s_addr >> 7) ^ ((ip)->s_addr >> 17)) &	\
161181803Sbz	  V_tcp_hostcache.hashmask)
162122922Sandre
163122922Sandre/* XXX: What is the recommended hash to get good entropy for IPv6 addresses? */
164133874Srwatson#define HOSTCACHE_HASH6(ip6)				\
165122922Sandre	(((ip6)->s6_addr32[0] ^				\
166122922Sandre	  (ip6)->s6_addr32[1] ^				\
167122922Sandre	  (ip6)->s6_addr32[2] ^				\
168122922Sandre	  (ip6)->s6_addr32[3]) &			\
169181803Sbz	 V_tcp_hostcache.hashmask)
170122922Sandre
171122922Sandre#define THC_LOCK(lp)		mtx_lock(lp)
172122922Sandre#define THC_UNLOCK(lp)		mtx_unlock(lp)
173122922Sandre
174122922Sandrevoid
175122922Sandretcp_hc_init(void)
176122922Sandre{
177241735Szont	u_int cache_limit;
178122922Sandre	int i;
179122922Sandre
180122922Sandre	/*
181170030Srwatson	 * Initialize hostcache structures.
182122922Sandre	 */
183181803Sbz	V_tcp_hostcache.cache_count = 0;
184181803Sbz	V_tcp_hostcache.hashsize = TCP_HOSTCACHE_HASHSIZE;
185181803Sbz	V_tcp_hostcache.bucket_limit = TCP_HOSTCACHE_BUCKETLIMIT;
186181803Sbz	V_tcp_hostcache.expire = TCP_HOSTCACHE_EXPIRE;
187181803Sbz	V_tcp_hostcache.prune = TCP_HOSTCACHE_PRUNE;
188122922Sandre
189133874Srwatson	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.hashsize",
190181803Sbz	    &V_tcp_hostcache.hashsize);
191181803Sbz	if (!powerof2(V_tcp_hostcache.hashsize)) {
192133874Srwatson		printf("WARNING: hostcache hash size is not a power of 2.\n");
193181803Sbz		V_tcp_hostcache.hashsize = TCP_HOSTCACHE_HASHSIZE; /* default */
194133874Srwatson	}
195181803Sbz	V_tcp_hostcache.hashmask = V_tcp_hostcache.hashsize - 1;
196122922Sandre
197241735Szont	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.bucketlimit",
198241735Szont	    &V_tcp_hostcache.bucket_limit);
199241735Szont
200241735Szont	cache_limit = V_tcp_hostcache.hashsize * V_tcp_hostcache.bucket_limit;
201241735Szont	V_tcp_hostcache.cache_limit = cache_limit;
202241735Szont	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.cachelimit",
203241735Szont	    &V_tcp_hostcache.cache_limit);
204241735Szont	if (V_tcp_hostcache.cache_limit > cache_limit)
205241735Szont		V_tcp_hostcache.cache_limit = cache_limit;
206241735Szont
207122922Sandre	/*
208170030Srwatson	 * Allocate the hash table.
209122922Sandre	 */
210181803Sbz	V_tcp_hostcache.hashbase = (struct hc_head *)
211181803Sbz	    malloc(V_tcp_hostcache.hashsize * sizeof(struct hc_head),
212122922Sandre		   M_HOSTCACHE, M_WAITOK | M_ZERO);
213122922Sandre
214122922Sandre	/*
215170030Srwatson	 * Initialize the hash buckets.
216122922Sandre	 */
217181803Sbz	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
218181803Sbz		TAILQ_INIT(&V_tcp_hostcache.hashbase[i].hch_bucket);
219181803Sbz		V_tcp_hostcache.hashbase[i].hch_length = 0;
220181803Sbz		mtx_init(&V_tcp_hostcache.hashbase[i].hch_mtx, "tcp_hc_entry",
221122922Sandre			  NULL, MTX_DEF);
222122922Sandre	}
223122922Sandre
224122922Sandre	/*
225122922Sandre	 * Allocate the hostcache entries.
226122922Sandre	 */
227181887Sjulian	V_tcp_hostcache.zone =
228181888Sjulian	    uma_zcreate("hostcache", sizeof(struct hc_metrics),
229181888Sjulian	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
230181803Sbz	uma_zone_set_max(V_tcp_hostcache.zone, V_tcp_hostcache.cache_limit);
231122922Sandre
232122922Sandre	/*
233122922Sandre	 * Set up periodic cache cleanup.
234122922Sandre	 */
235181803Sbz	callout_init(&V_tcp_hc_callout, CALLOUT_MPSAFE);
236181887Sjulian	callout_reset(&V_tcp_hc_callout, V_tcp_hostcache.prune * hz,
237191816Szec	    tcp_hc_purge, curvnet);
238122922Sandre}
239122922Sandre
240193731Szec#ifdef VIMAGE
241193731Szecvoid
242193731Szectcp_hc_destroy(void)
243193731Szec{
244203724Sbz	int i;
245193731Szec
246203724Sbz	callout_drain(&V_tcp_hc_callout);
247193731Szec
248203724Sbz	/* Purge all hc entries. */
249203724Sbz	tcp_hc_purge_internal(1);
250203724Sbz
251203724Sbz	/* Free the uma zone and the allocated hash table. */
252203724Sbz	uma_zdestroy(V_tcp_hostcache.zone);
253203724Sbz
254203724Sbz	for (i = 0; i < V_tcp_hostcache.hashsize; i++)
255203724Sbz		mtx_destroy(&V_tcp_hostcache.hashbase[i].hch_mtx);
256203724Sbz	free(V_tcp_hostcache.hashbase, M_HOSTCACHE);
257193731Szec}
258193731Szec#endif
259193731Szec
260122922Sandre/*
261170030Srwatson * Internal function: look up an entry in the hostcache or return NULL.
262122922Sandre *
263122922Sandre * If an entry has been returned, the caller becomes responsible for
264122922Sandre * unlocking the bucket row after he is done reading/modifying the entry.
265122922Sandre */
266122922Sandrestatic struct hc_metrics *
267122922Sandretcp_hc_lookup(struct in_conninfo *inc)
268122922Sandre{
269122922Sandre	int hash;
270122922Sandre	struct hc_head *hc_head;
271122922Sandre	struct hc_metrics *hc_entry;
272122922Sandre
273122922Sandre	KASSERT(inc != NULL, ("tcp_hc_lookup with NULL in_conninfo pointer"));
274122922Sandre
275122922Sandre	/*
276122922Sandre	 * Hash the foreign ip address.
277122922Sandre	 */
278186222Sbz	if (inc->inc_flags & INC_ISIPV6)
279122922Sandre		hash = HOSTCACHE_HASH6(&inc->inc6_faddr);
280122922Sandre	else
281122922Sandre		hash = HOSTCACHE_HASH(&inc->inc_faddr);
282122922Sandre
283181803Sbz	hc_head = &V_tcp_hostcache.hashbase[hash];
284122922Sandre
285122922Sandre	/*
286170030Srwatson	 * Acquire lock for this bucket row; we release the lock if we don't
287170030Srwatson	 * find an entry, otherwise the caller has to unlock after he is
288170030Srwatson	 * done.
289122922Sandre	 */
290122922Sandre	THC_LOCK(&hc_head->hch_mtx);
291122922Sandre
292122922Sandre	/*
293170030Srwatson	 * Iterate through entries in bucket row looking for a match.
294122922Sandre	 */
295122922Sandre	TAILQ_FOREACH(hc_entry, &hc_head->hch_bucket, rmx_q) {
296186222Sbz		if (inc->inc_flags & INC_ISIPV6) {
297122922Sandre			if (memcmp(&inc->inc6_faddr, &hc_entry->ip6,
298122922Sandre			    sizeof(inc->inc6_faddr)) == 0)
299122922Sandre				return hc_entry;
300122922Sandre		} else {
301122922Sandre			if (memcmp(&inc->inc_faddr, &hc_entry->ip4,
302122922Sandre			    sizeof(inc->inc_faddr)) == 0)
303122922Sandre				return hc_entry;
304122922Sandre		}
305122922Sandre	}
306122922Sandre
307122922Sandre	/*
308170030Srwatson	 * We were unsuccessful and didn't find anything.
309122922Sandre	 */
310122922Sandre	THC_UNLOCK(&hc_head->hch_mtx);
311122922Sandre	return NULL;
312122922Sandre}
313122922Sandre
314122922Sandre/*
315170030Srwatson * Internal function: insert an entry into the hostcache or return NULL if
316170030Srwatson * unable to allocate a new one.
317133874Srwatson *
318122922Sandre * If an entry has been returned, the caller becomes responsible for
319122922Sandre * unlocking the bucket row after he is done reading/modifying the entry.
320122922Sandre */
321122922Sandrestatic struct hc_metrics *
322122922Sandretcp_hc_insert(struct in_conninfo *inc)
323122922Sandre{
324122922Sandre	int hash;
325122922Sandre	struct hc_head *hc_head;
326122922Sandre	struct hc_metrics *hc_entry;
327122922Sandre
328122922Sandre	KASSERT(inc != NULL, ("tcp_hc_insert with NULL in_conninfo pointer"));
329122922Sandre
330122922Sandre	/*
331170030Srwatson	 * Hash the foreign ip address.
332122922Sandre	 */
333186222Sbz	if (inc->inc_flags & INC_ISIPV6)
334122922Sandre		hash = HOSTCACHE_HASH6(&inc->inc6_faddr);
335122922Sandre	else
336122922Sandre		hash = HOSTCACHE_HASH(&inc->inc_faddr);
337122922Sandre
338181803Sbz	hc_head = &V_tcp_hostcache.hashbase[hash];
339122922Sandre
340122922Sandre	/*
341170030Srwatson	 * Acquire lock for this bucket row; we release the lock if we don't
342170030Srwatson	 * find an entry, otherwise the caller has to unlock after he is
343170030Srwatson	 * done.
344122922Sandre	 */
345122922Sandre	THC_LOCK(&hc_head->hch_mtx);
346122922Sandre
347122922Sandre	/*
348170030Srwatson	 * If the bucket limit is reached, reuse the least-used element.
349122922Sandre	 */
350181803Sbz	if (hc_head->hch_length >= V_tcp_hostcache.bucket_limit ||
351181803Sbz	    V_tcp_hostcache.cache_count >= V_tcp_hostcache.cache_limit) {
352122922Sandre		hc_entry = TAILQ_LAST(&hc_head->hch_bucket, hc_qhead);
353122922Sandre		/*
354122922Sandre		 * At first we were dropping the last element, just to
355170030Srwatson		 * reacquire it in the next two lines again, which isn't very
356170030Srwatson		 * efficient.  Instead just reuse the least used element.
357170030Srwatson		 * We may drop something that is still "in-use" but we can be
358170030Srwatson		 * "lossy".
359170405Sandre		 * Just give up if this bucket row is empty and we don't have
360170405Sandre		 * anything to replace.
361122922Sandre		 */
362170405Sandre		if (hc_entry == NULL) {
363170405Sandre			THC_UNLOCK(&hc_head->hch_mtx);
364170405Sandre			return NULL;
365170405Sandre		}
366122922Sandre		TAILQ_REMOVE(&hc_head->hch_bucket, hc_entry, rmx_q);
367181803Sbz		V_tcp_hostcache.hashbase[hash].hch_length--;
368181803Sbz		V_tcp_hostcache.cache_count--;
369190948Srwatson		TCPSTAT_INC(tcps_hc_bucketoverflow);
370123028Sandre#if 0
371181803Sbz		uma_zfree(V_tcp_hostcache.zone, hc_entry);
372122922Sandre#endif
373122922Sandre	} else {
374122922Sandre		/*
375170030Srwatson		 * Allocate a new entry, or balk if not possible.
376122922Sandre		 */
377181803Sbz		hc_entry = uma_zalloc(V_tcp_hostcache.zone, M_NOWAIT);
378122922Sandre		if (hc_entry == NULL) {
379122922Sandre			THC_UNLOCK(&hc_head->hch_mtx);
380122922Sandre			return NULL;
381122922Sandre		}
382122922Sandre	}
383122922Sandre
384122922Sandre	/*
385170030Srwatson	 * Initialize basic information of hostcache entry.
386122922Sandre	 */
387122922Sandre	bzero(hc_entry, sizeof(*hc_entry));
388186222Sbz	if (inc->inc_flags & INC_ISIPV6)
389123113Sandre		bcopy(&inc->inc6_faddr, &hc_entry->ip6, sizeof(hc_entry->ip6));
390122922Sandre	else
391122922Sandre		hc_entry->ip4 = inc->inc_faddr;
392122922Sandre	hc_entry->rmx_head = hc_head;
393181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire;
394122922Sandre
395122922Sandre	/*
396170030Srwatson	 * Put it upfront.
397122922Sandre	 */
398122922Sandre	TAILQ_INSERT_HEAD(&hc_head->hch_bucket, hc_entry, rmx_q);
399181803Sbz	V_tcp_hostcache.hashbase[hash].hch_length++;
400181803Sbz	V_tcp_hostcache.cache_count++;
401190948Srwatson	TCPSTAT_INC(tcps_hc_added);
402122922Sandre
403122922Sandre	return hc_entry;
404122922Sandre}
405122922Sandre
406122922Sandre/*
407170030Srwatson * External function: look up an entry in the hostcache and fill out the
408170030Srwatson * supplied TCP metrics structure.  Fills in NULL when no entry was found or
409170030Srwatson * a value is not set.
410122922Sandre */
411122922Sandrevoid
412122922Sandretcp_hc_get(struct in_conninfo *inc, struct hc_metrics_lite *hc_metrics_lite)
413122922Sandre{
414122922Sandre	struct hc_metrics *hc_entry;
415122922Sandre
416122922Sandre	/*
417170030Srwatson	 * Find the right bucket.
418122922Sandre	 */
419122922Sandre	hc_entry = tcp_hc_lookup(inc);
420122922Sandre
421122922Sandre	/*
422170030Srwatson	 * If we don't have an existing object.
423122922Sandre	 */
424122922Sandre	if (hc_entry == NULL) {
425122922Sandre		bzero(hc_metrics_lite, sizeof(*hc_metrics_lite));
426122922Sandre		return;
427122922Sandre	}
428122922Sandre	hc_entry->rmx_hits++;
429181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
430122922Sandre
431122922Sandre	hc_metrics_lite->rmx_mtu = hc_entry->rmx_mtu;
432122922Sandre	hc_metrics_lite->rmx_ssthresh = hc_entry->rmx_ssthresh;
433122922Sandre	hc_metrics_lite->rmx_rtt = hc_entry->rmx_rtt;
434122922Sandre	hc_metrics_lite->rmx_rttvar = hc_entry->rmx_rttvar;
435122922Sandre	hc_metrics_lite->rmx_bandwidth = hc_entry->rmx_bandwidth;
436122922Sandre	hc_metrics_lite->rmx_cwnd = hc_entry->rmx_cwnd;
437122922Sandre	hc_metrics_lite->rmx_sendpipe = hc_entry->rmx_sendpipe;
438122922Sandre	hc_metrics_lite->rmx_recvpipe = hc_entry->rmx_recvpipe;
439122922Sandre
440122922Sandre	/*
441170030Srwatson	 * Unlock bucket row.
442122922Sandre	 */
443122922Sandre	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
444122922Sandre}
445122922Sandre
446122922Sandre/*
447170030Srwatson * External function: look up an entry in the hostcache and return the
448170030Srwatson * discovered path MTU.  Returns NULL if no entry is found or value is not
449138409Srwatson * set.
450122922Sandre */
451122922Sandreu_long
452122922Sandretcp_hc_getmtu(struct in_conninfo *inc)
453122922Sandre{
454122922Sandre	struct hc_metrics *hc_entry;
455122922Sandre	u_long mtu;
456122922Sandre
457122922Sandre	hc_entry = tcp_hc_lookup(inc);
458122922Sandre	if (hc_entry == NULL) {
459122922Sandre		return 0;
460122922Sandre	}
461122922Sandre	hc_entry->rmx_hits++;
462181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
463122922Sandre
464122922Sandre	mtu = hc_entry->rmx_mtu;
465122922Sandre	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
466122922Sandre	return mtu;
467122922Sandre}
468122922Sandre
469122922Sandre/*
470170030Srwatson * External function: update the MTU value of an entry in the hostcache.
471122922Sandre * Creates a new entry if none was found.
472122922Sandre */
473122922Sandrevoid
474122922Sandretcp_hc_updatemtu(struct in_conninfo *inc, u_long mtu)
475122922Sandre{
476122922Sandre	struct hc_metrics *hc_entry;
477122922Sandre
478122922Sandre	/*
479170030Srwatson	 * Find the right bucket.
480122922Sandre	 */
481122922Sandre	hc_entry = tcp_hc_lookup(inc);
482122922Sandre
483122922Sandre	/*
484170030Srwatson	 * If we don't have an existing object, try to insert a new one.
485122922Sandre	 */
486122922Sandre	if (hc_entry == NULL) {
487122922Sandre		hc_entry = tcp_hc_insert(inc);
488122922Sandre		if (hc_entry == NULL)
489122922Sandre			return;
490122922Sandre	}
491122922Sandre	hc_entry->rmx_updates++;
492181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
493122922Sandre
494122922Sandre	hc_entry->rmx_mtu = mtu;
495122922Sandre
496122922Sandre	/*
497170030Srwatson	 * Put it upfront so we find it faster next time.
498122922Sandre	 */
499122922Sandre	TAILQ_REMOVE(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
500122922Sandre	TAILQ_INSERT_HEAD(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
501122922Sandre
502122922Sandre	/*
503170030Srwatson	 * Unlock bucket row.
504122922Sandre	 */
505122922Sandre	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
506122922Sandre}
507122922Sandre
508122922Sandre/*
509170030Srwatson * External function: update the TCP metrics of an entry in the hostcache.
510122922Sandre * Creates a new entry if none was found.
511122922Sandre */
512122922Sandrevoid
513122922Sandretcp_hc_update(struct in_conninfo *inc, struct hc_metrics_lite *hcml)
514122922Sandre{
515122922Sandre	struct hc_metrics *hc_entry;
516122922Sandre
517122922Sandre	hc_entry = tcp_hc_lookup(inc);
518122922Sandre	if (hc_entry == NULL) {
519122922Sandre		hc_entry = tcp_hc_insert(inc);
520122922Sandre		if (hc_entry == NULL)
521122922Sandre			return;
522122922Sandre	}
523122922Sandre	hc_entry->rmx_updates++;
524181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
525122922Sandre
526122922Sandre	if (hcml->rmx_rtt != 0) {
527122922Sandre		if (hc_entry->rmx_rtt == 0)
528122922Sandre			hc_entry->rmx_rtt = hcml->rmx_rtt;
529122922Sandre		else
530122922Sandre			hc_entry->rmx_rtt =
531122922Sandre			    (hc_entry->rmx_rtt + hcml->rmx_rtt) / 2;
532190948Srwatson		TCPSTAT_INC(tcps_cachedrtt);
533122922Sandre	}
534122922Sandre	if (hcml->rmx_rttvar != 0) {
535122922Sandre	        if (hc_entry->rmx_rttvar == 0)
536133874Srwatson			hc_entry->rmx_rttvar = hcml->rmx_rttvar;
537122922Sandre		else
538122922Sandre			hc_entry->rmx_rttvar =
539122922Sandre			    (hc_entry->rmx_rttvar + hcml->rmx_rttvar) / 2;
540190948Srwatson		TCPSTAT_INC(tcps_cachedrttvar);
541122922Sandre	}
542122922Sandre	if (hcml->rmx_ssthresh != 0) {
543122922Sandre		if (hc_entry->rmx_ssthresh == 0)
544122922Sandre			hc_entry->rmx_ssthresh = hcml->rmx_ssthresh;
545122922Sandre		else
546122922Sandre			hc_entry->rmx_ssthresh =
547122922Sandre			    (hc_entry->rmx_ssthresh + hcml->rmx_ssthresh) / 2;
548190948Srwatson		TCPSTAT_INC(tcps_cachedssthresh);
549122922Sandre	}
550122922Sandre	if (hcml->rmx_bandwidth != 0) {
551122922Sandre		if (hc_entry->rmx_bandwidth == 0)
552122922Sandre			hc_entry->rmx_bandwidth = hcml->rmx_bandwidth;
553122922Sandre		else
554122922Sandre			hc_entry->rmx_bandwidth =
555122922Sandre			    (hc_entry->rmx_bandwidth + hcml->rmx_bandwidth) / 2;
556190948Srwatson		/* TCPSTAT_INC(tcps_cachedbandwidth); */
557122922Sandre	}
558122922Sandre	if (hcml->rmx_cwnd != 0) {
559122922Sandre		if (hc_entry->rmx_cwnd == 0)
560122922Sandre			hc_entry->rmx_cwnd = hcml->rmx_cwnd;
561122922Sandre		else
562122922Sandre			hc_entry->rmx_cwnd =
563122922Sandre			    (hc_entry->rmx_cwnd + hcml->rmx_cwnd) / 2;
564190948Srwatson		/* TCPSTAT_INC(tcps_cachedcwnd); */
565122922Sandre	}
566122922Sandre	if (hcml->rmx_sendpipe != 0) {
567122922Sandre		if (hc_entry->rmx_sendpipe == 0)
568122922Sandre			hc_entry->rmx_sendpipe = hcml->rmx_sendpipe;
569122922Sandre		else
570122922Sandre			hc_entry->rmx_sendpipe =
571122922Sandre			    (hc_entry->rmx_sendpipe + hcml->rmx_sendpipe) /2;
572190948Srwatson		/* TCPSTAT_INC(tcps_cachedsendpipe); */
573133874Srwatson	}
574122922Sandre	if (hcml->rmx_recvpipe != 0) {
575122922Sandre		if (hc_entry->rmx_recvpipe == 0)
576122922Sandre			hc_entry->rmx_recvpipe = hcml->rmx_recvpipe;
577122922Sandre		else
578122922Sandre			hc_entry->rmx_recvpipe =
579122922Sandre			    (hc_entry->rmx_recvpipe + hcml->rmx_recvpipe) /2;
580190948Srwatson		/* TCPSTAT_INC(tcps_cachedrecvpipe); */
581122922Sandre	}
582122922Sandre
583122922Sandre	TAILQ_REMOVE(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
584122922Sandre	TAILQ_INSERT_HEAD(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
585122922Sandre	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
586122922Sandre}
587122922Sandre
588122922Sandre/*
589122922Sandre * Sysctl function: prints the list and values of all hostcache entries in
590122922Sandre * unsorted order.
591122922Sandre */
592122922Sandrestatic int
593122922Sandresysctl_tcp_hc_list(SYSCTL_HANDLER_ARGS)
594122922Sandre{
595122922Sandre	int bufsize;
596122922Sandre	int linesize = 128;
597122922Sandre	char *p, *buf;
598122922Sandre	int len, i, error;
599122922Sandre	struct hc_metrics *hc_entry;
600165118Sbz#ifdef INET6
601165118Sbz	char ip6buf[INET6_ADDRSTRLEN];
602165118Sbz#endif
603122922Sandre
604181803Sbz	bufsize = linesize * (V_tcp_hostcache.cache_count + 1);
605122922Sandre
606122922Sandre	p = buf = (char *)malloc(bufsize, M_TEMP, M_WAITOK|M_ZERO);
607122922Sandre
608122922Sandre	len = snprintf(p, linesize,
609122922Sandre		"\nIP address        MTU  SSTRESH      RTT   RTTVAR BANDWIDTH "
610122922Sandre		"    CWND SENDPIPE RECVPIPE HITS  UPD  EXP\n");
611122922Sandre	p += len;
612122922Sandre
613122922Sandre#define msec(u) (((u) + 500) / 1000)
614181803Sbz	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
615181803Sbz		THC_LOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
616181803Sbz		TAILQ_FOREACH(hc_entry, &V_tcp_hostcache.hashbase[i].hch_bucket,
617122922Sandre			      rmx_q) {
618122922Sandre			len = snprintf(p, linesize,
619122922Sandre			    "%-15s %5lu %8lu %6lums %6lums %9lu %8lu %8lu %8lu "
620122922Sandre			    "%4lu %4lu %4i\n",
621122922Sandre			    hc_entry->ip4.s_addr ? inet_ntoa(hc_entry->ip4) :
622122922Sandre#ifdef INET6
623165118Sbz				ip6_sprintf(ip6buf, &hc_entry->ip6),
624122922Sandre#else
625122922Sandre				"IPv6?",
626122922Sandre#endif
627122922Sandre			    hc_entry->rmx_mtu,
628122922Sandre			    hc_entry->rmx_ssthresh,
629122922Sandre			    msec(hc_entry->rmx_rtt *
630122922Sandre				(RTM_RTTUNIT / (hz * TCP_RTT_SCALE))),
631122922Sandre			    msec(hc_entry->rmx_rttvar *
632238083Strociny				(RTM_RTTUNIT / (hz * TCP_RTTVAR_SCALE))),
633133477Sandre			    hc_entry->rmx_bandwidth * 8,
634122922Sandre			    hc_entry->rmx_cwnd,
635122922Sandre			    hc_entry->rmx_sendpipe,
636122922Sandre			    hc_entry->rmx_recvpipe,
637122922Sandre			    hc_entry->rmx_hits,
638122922Sandre			    hc_entry->rmx_updates,
639122922Sandre			    hc_entry->rmx_expire);
640122922Sandre			p += len;
641122922Sandre		}
642181803Sbz		THC_UNLOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
643122922Sandre	}
644122922Sandre#undef msec
645122922Sandre	error = SYSCTL_OUT(req, buf, p - buf);
646122922Sandre	free(buf, M_TEMP);
647122922Sandre	return(error);
648122922Sandre}
649122922Sandre
650122922Sandre/*
651203724Sbz * Caller has to make sure the curvnet is set properly.
652122922Sandre */
653122922Sandrestatic void
654203724Sbztcp_hc_purge_internal(int all)
655122922Sandre{
656128574Sandre	struct hc_metrics *hc_entry, *hc_next;
657122922Sandre	int i;
658122922Sandre
659181803Sbz	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
660181803Sbz		THC_LOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
661181887Sjulian		TAILQ_FOREACH_SAFE(hc_entry,
662181888Sjulian		    &V_tcp_hostcache.hashbase[i].hch_bucket, rmx_q, hc_next) {
663122922Sandre			if (all || hc_entry->rmx_expire <= 0) {
664181803Sbz				TAILQ_REMOVE(&V_tcp_hostcache.hashbase[i].hch_bucket,
665122922Sandre					      hc_entry, rmx_q);
666181803Sbz				uma_zfree(V_tcp_hostcache.zone, hc_entry);
667181803Sbz				V_tcp_hostcache.hashbase[i].hch_length--;
668181803Sbz				V_tcp_hostcache.cache_count--;
669122922Sandre			} else
670181803Sbz				hc_entry->rmx_expire -= V_tcp_hostcache.prune;
671122922Sandre		}
672181803Sbz		THC_UNLOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
673122922Sandre	}
674203724Sbz}
675181887Sjulian
676203724Sbz/*
677203724Sbz * Expire and purge (old|all) entries in the tcp_hostcache.  Runs
678203724Sbz * periodically from the callout.
679203724Sbz */
680203724Sbzstatic void
681203724Sbztcp_hc_purge(void *arg)
682203724Sbz{
683203724Sbz	CURVNET_SET((struct vnet *) arg);
684203724Sbz	int all = 0;
685203724Sbz
686203724Sbz	if (V_tcp_hostcache.purgeall) {
687203724Sbz		all = 1;
688203724Sbz		V_tcp_hostcache.purgeall = 0;
689203724Sbz	}
690203724Sbz
691203724Sbz	tcp_hc_purge_internal(all);
692203724Sbz
693181887Sjulian	callout_reset(&V_tcp_hc_callout, V_tcp_hostcache.prune * hz,
694181887Sjulian	    tcp_hc_purge, arg);
695191816Szec	CURVNET_RESTORE();
696122922Sandre}
697