tcp_hostcache.c revision 182633
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: head/sys/netinet/tcp_hostcache.c 182633 2008-09-01 19:25:27Z brooks $");
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>
79181803Sbz#include <sys/vimage.h>
80122922Sandre
81122922Sandre#include <net/if.h>
82122922Sandre
83122922Sandre#include <netinet/in.h>
84122922Sandre#include <netinet/in_systm.h>
85122922Sandre#include <netinet/ip.h>
86122922Sandre#include <netinet/in_var.h>
87122922Sandre#include <netinet/in_pcb.h>
88122922Sandre#include <netinet/ip_var.h>
89122922Sandre#ifdef INET6
90122922Sandre#include <netinet/ip6.h>
91122922Sandre#include <netinet6/ip6_var.h>
92122922Sandre#endif
93122922Sandre#include <netinet/tcp.h>
94122922Sandre#include <netinet/tcp_var.h>
95122922Sandre#ifdef INET6
96122922Sandre#include <netinet6/tcp6_var.h>
97122922Sandre#endif
98122922Sandre
99122922Sandre#include <vm/uma.h>
100122922Sandre
101122922Sandre
102122922SandreTAILQ_HEAD(hc_qhead, hc_metrics);
103122922Sandre
104122922Sandrestruct hc_head {
105122922Sandre	struct hc_qhead	hch_bucket;
106122922Sandre	u_int		hch_length;
107122922Sandre	struct mtx	hch_mtx;
108122922Sandre};
109122922Sandre
110122922Sandrestruct hc_metrics {
111122922Sandre	/* housekeeping */
112122922Sandre	TAILQ_ENTRY(hc_metrics) rmx_q;
113122922Sandre	struct	hc_head *rmx_head; /* head of bucket tail queue */
114122922Sandre	struct	in_addr ip4;	/* IP address */
115122922Sandre	struct	in6_addr ip6;	/* IP6 address */
116170030Srwatson	/* endpoint specific values for TCP */
117122922Sandre	u_long	rmx_mtu;	/* MTU for this path */
118122922Sandre	u_long	rmx_ssthresh;	/* outbound gateway buffer limit */
119122922Sandre	u_long	rmx_rtt;	/* estimated round trip time */
120122922Sandre	u_long	rmx_rttvar;	/* estimated rtt variance */
121122922Sandre	u_long	rmx_bandwidth;	/* estimated bandwidth */
122122922Sandre	u_long	rmx_cwnd;	/* congestion window */
123122922Sandre	u_long	rmx_sendpipe;	/* outbound delay-bandwidth product */
124122922Sandre	u_long	rmx_recvpipe;	/* inbound delay-bandwidth product */
125170030Srwatson	/* TCP hostcache internal data */
126122922Sandre	int	rmx_expire;	/* lifetime for object */
127122922Sandre	u_long	rmx_hits;	/* number of hits */
128122922Sandre	u_long	rmx_updates;	/* number of updates */
129122922Sandre};
130122922Sandre
131122922Sandre/* Arbitrary values */
132122922Sandre#define TCP_HOSTCACHE_HASHSIZE		512
133122922Sandre#define TCP_HOSTCACHE_BUCKETLIMIT	30
134122922Sandre#define TCP_HOSTCACHE_EXPIRE		60*60	/* one hour */
135122922Sandre#define TCP_HOSTCACHE_PRUNE		5*60	/* every 5 minutes */
136122922Sandre
137122922Sandrestruct tcp_hostcache {
138122922Sandre	struct	hc_head *hashbase;
139122922Sandre	uma_zone_t zone;
140122922Sandre	u_int	hashsize;
141122922Sandre	u_int	hashmask;
142122922Sandre	u_int	bucket_limit;
143122922Sandre	u_int	cache_count;
144122922Sandre	u_int	cache_limit;
145122922Sandre	int	expire;
146170434Syar	int	prune;
147122922Sandre	int	purgeall;
148122922Sandre};
149122922Sandrestatic struct tcp_hostcache tcp_hostcache;
150122922Sandre
151122922Sandrestatic struct callout tcp_hc_callout;
152122922Sandre
153122922Sandrestatic struct hc_metrics *tcp_hc_lookup(struct in_conninfo *);
154122922Sandrestatic struct hc_metrics *tcp_hc_insert(struct in_conninfo *);
155122922Sandrestatic int sysctl_tcp_hc_list(SYSCTL_HANDLER_ARGS);
156122922Sandrestatic void tcp_hc_purge(void *);
157122922Sandre
158182633SbrooksSYSCTL_NODE(_net_inet_tcp, OID_AUTO, hostcache, CTLFLAG_RW, 0,
159182633Sbrooks    "TCP Host cache");
160122922Sandre
161122922SandreSYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, cachelimit, CTLFLAG_RDTUN,
162167784Sandre    &tcp_hostcache.cache_limit, 0, "Overall entry limit for hostcache");
163122922Sandre
164122922SandreSYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, hashsize, CTLFLAG_RDTUN,
165167784Sandre    &tcp_hostcache.hashsize, 0, "Size of TCP hostcache hashtable");
166122922Sandre
167122922SandreSYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, bucketlimit, CTLFLAG_RDTUN,
168167784Sandre    &tcp_hostcache.bucket_limit, 0, "Per-bucket hash limit for hostcache");
169122922Sandre
170122922SandreSYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, count, CTLFLAG_RD,
171167784Sandre    &tcp_hostcache.cache_count, 0, "Current number of entries in hostcache");
172122922Sandre
173122922SandreSYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, expire, CTLFLAG_RW,
174167784Sandre    &tcp_hostcache.expire, 0, "Expire time of TCP hostcache entries");
175122922Sandre
176170434SyarSYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, prune, CTLFLAG_RW,
177170434Syar     &tcp_hostcache.prune, 0, "Time between purge runs");
178170434Syar
179122922SandreSYSCTL_INT(_net_inet_tcp_hostcache, OID_AUTO, purge, CTLFLAG_RW,
180167784Sandre    &tcp_hostcache.purgeall, 0, "Expire all entires on next purge run");
181122922Sandre
182122922SandreSYSCTL_PROC(_net_inet_tcp_hostcache, OID_AUTO, list,
183167784Sandre    CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_SKIP, 0, 0,
184167784Sandre    sysctl_tcp_hc_list, "A", "List of all hostcache entries");
185122922Sandre
186122922Sandre
187122922Sandrestatic MALLOC_DEFINE(M_HOSTCACHE, "hostcache", "TCP hostcache");
188122922Sandre
189122922Sandre#define HOSTCACHE_HASH(ip) \
190133874Srwatson	(((ip)->s_addr ^ ((ip)->s_addr >> 7) ^ ((ip)->s_addr >> 17)) &	\
191181803Sbz	  V_tcp_hostcache.hashmask)
192122922Sandre
193122922Sandre/* XXX: What is the recommended hash to get good entropy for IPv6 addresses? */
194133874Srwatson#define HOSTCACHE_HASH6(ip6)				\
195122922Sandre	(((ip6)->s6_addr32[0] ^				\
196122922Sandre	  (ip6)->s6_addr32[1] ^				\
197122922Sandre	  (ip6)->s6_addr32[2] ^				\
198122922Sandre	  (ip6)->s6_addr32[3]) &			\
199181803Sbz	 V_tcp_hostcache.hashmask)
200122922Sandre
201122922Sandre#define THC_LOCK(lp)		mtx_lock(lp)
202122922Sandre#define THC_UNLOCK(lp)		mtx_unlock(lp)
203122922Sandre
204122922Sandrevoid
205122922Sandretcp_hc_init(void)
206122922Sandre{
207122922Sandre	int i;
208122922Sandre
209122922Sandre	/*
210170030Srwatson	 * Initialize hostcache structures.
211122922Sandre	 */
212181803Sbz	V_tcp_hostcache.cache_count = 0;
213181803Sbz	V_tcp_hostcache.hashsize = TCP_HOSTCACHE_HASHSIZE;
214181803Sbz	V_tcp_hostcache.bucket_limit = TCP_HOSTCACHE_BUCKETLIMIT;
215181803Sbz	V_tcp_hostcache.cache_limit =
216181803Sbz	    V_tcp_hostcache.hashsize * V_tcp_hostcache.bucket_limit;
217181803Sbz	V_tcp_hostcache.expire = TCP_HOSTCACHE_EXPIRE;
218181803Sbz	V_tcp_hostcache.prune = TCP_HOSTCACHE_PRUNE;
219122922Sandre
220133874Srwatson	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.hashsize",
221181803Sbz	    &V_tcp_hostcache.hashsize);
222133874Srwatson	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.cachelimit",
223181803Sbz	    &V_tcp_hostcache.cache_limit);
224133874Srwatson	TUNABLE_INT_FETCH("net.inet.tcp.hostcache.bucketlimit",
225181803Sbz	    &V_tcp_hostcache.bucket_limit);
226181803Sbz	if (!powerof2(V_tcp_hostcache.hashsize)) {
227133874Srwatson		printf("WARNING: hostcache hash size is not a power of 2.\n");
228181803Sbz		V_tcp_hostcache.hashsize = TCP_HOSTCACHE_HASHSIZE; /* default */
229133874Srwatson	}
230181803Sbz	V_tcp_hostcache.hashmask = V_tcp_hostcache.hashsize - 1;
231122922Sandre
232122922Sandre	/*
233170030Srwatson	 * Allocate the hash table.
234122922Sandre	 */
235181803Sbz	V_tcp_hostcache.hashbase = (struct hc_head *)
236181803Sbz	    malloc(V_tcp_hostcache.hashsize * sizeof(struct hc_head),
237122922Sandre		   M_HOSTCACHE, M_WAITOK | M_ZERO);
238122922Sandre
239122922Sandre	/*
240170030Srwatson	 * Initialize the hash buckets.
241122922Sandre	 */
242181803Sbz	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
243181803Sbz		TAILQ_INIT(&V_tcp_hostcache.hashbase[i].hch_bucket);
244181803Sbz		V_tcp_hostcache.hashbase[i].hch_length = 0;
245181803Sbz		mtx_init(&V_tcp_hostcache.hashbase[i].hch_mtx, "tcp_hc_entry",
246122922Sandre			  NULL, MTX_DEF);
247122922Sandre	}
248122922Sandre
249122922Sandre	/*
250122922Sandre	 * Allocate the hostcache entries.
251122922Sandre	 */
252181887Sjulian	V_tcp_hostcache.zone =
253181888Sjulian	    uma_zcreate("hostcache", sizeof(struct hc_metrics),
254181888Sjulian	    NULL, NULL, NULL, NULL, UMA_ALIGN_PTR, 0);
255181803Sbz	uma_zone_set_max(V_tcp_hostcache.zone, V_tcp_hostcache.cache_limit);
256122922Sandre
257122922Sandre	/*
258122922Sandre	 * Set up periodic cache cleanup.
259122922Sandre	 */
260181803Sbz	callout_init(&V_tcp_hc_callout, CALLOUT_MPSAFE);
261181887Sjulian	callout_reset(&V_tcp_hc_callout, V_tcp_hostcache.prune * hz,
262181887Sjulian	    tcp_hc_purge, 0);
263122922Sandre}
264122922Sandre
265122922Sandre/*
266170030Srwatson * Internal function: look up an entry in the hostcache or return NULL.
267122922Sandre *
268122922Sandre * If an entry has been returned, the caller becomes responsible for
269122922Sandre * unlocking the bucket row after he is done reading/modifying the entry.
270122922Sandre */
271122922Sandrestatic struct hc_metrics *
272122922Sandretcp_hc_lookup(struct in_conninfo *inc)
273122922Sandre{
274122922Sandre	int hash;
275122922Sandre	struct hc_head *hc_head;
276122922Sandre	struct hc_metrics *hc_entry;
277122922Sandre
278122922Sandre	KASSERT(inc != NULL, ("tcp_hc_lookup with NULL in_conninfo pointer"));
279122922Sandre
280122922Sandre	/*
281122922Sandre	 * Hash the foreign ip address.
282122922Sandre	 */
283122922Sandre	if (inc->inc_isipv6)
284122922Sandre		hash = HOSTCACHE_HASH6(&inc->inc6_faddr);
285122922Sandre	else
286122922Sandre		hash = HOSTCACHE_HASH(&inc->inc_faddr);
287122922Sandre
288181803Sbz	hc_head = &V_tcp_hostcache.hashbase[hash];
289122922Sandre
290122922Sandre	/*
291170030Srwatson	 * Acquire lock for this bucket row; we release the lock if we don't
292170030Srwatson	 * find an entry, otherwise the caller has to unlock after he is
293170030Srwatson	 * done.
294122922Sandre	 */
295122922Sandre	THC_LOCK(&hc_head->hch_mtx);
296122922Sandre
297122922Sandre	/*
298170030Srwatson	 * Iterate through entries in bucket row looking for a match.
299122922Sandre	 */
300122922Sandre	TAILQ_FOREACH(hc_entry, &hc_head->hch_bucket, rmx_q) {
301122922Sandre		if (inc->inc_isipv6) {
302122922Sandre			if (memcmp(&inc->inc6_faddr, &hc_entry->ip6,
303122922Sandre			    sizeof(inc->inc6_faddr)) == 0)
304122922Sandre				return hc_entry;
305122922Sandre		} else {
306122922Sandre			if (memcmp(&inc->inc_faddr, &hc_entry->ip4,
307122922Sandre			    sizeof(inc->inc_faddr)) == 0)
308122922Sandre				return hc_entry;
309122922Sandre		}
310122922Sandre	}
311122922Sandre
312122922Sandre	/*
313170030Srwatson	 * We were unsuccessful and didn't find anything.
314122922Sandre	 */
315122922Sandre	THC_UNLOCK(&hc_head->hch_mtx);
316122922Sandre	return NULL;
317122922Sandre}
318122922Sandre
319122922Sandre/*
320170030Srwatson * Internal function: insert an entry into the hostcache or return NULL if
321170030Srwatson * unable to allocate a new one.
322133874Srwatson *
323122922Sandre * If an entry has been returned, the caller becomes responsible for
324122922Sandre * unlocking the bucket row after he is done reading/modifying the entry.
325122922Sandre */
326122922Sandrestatic struct hc_metrics *
327122922Sandretcp_hc_insert(struct in_conninfo *inc)
328122922Sandre{
329122922Sandre	int hash;
330122922Sandre	struct hc_head *hc_head;
331122922Sandre	struct hc_metrics *hc_entry;
332122922Sandre
333122922Sandre	KASSERT(inc != NULL, ("tcp_hc_insert with NULL in_conninfo pointer"));
334122922Sandre
335122922Sandre	/*
336170030Srwatson	 * Hash the foreign ip address.
337122922Sandre	 */
338122922Sandre	if (inc->inc_isipv6)
339122922Sandre		hash = HOSTCACHE_HASH6(&inc->inc6_faddr);
340122922Sandre	else
341122922Sandre		hash = HOSTCACHE_HASH(&inc->inc_faddr);
342122922Sandre
343181803Sbz	hc_head = &V_tcp_hostcache.hashbase[hash];
344122922Sandre
345122922Sandre	/*
346170030Srwatson	 * Acquire lock for this bucket row; we release the lock if we don't
347170030Srwatson	 * find an entry, otherwise the caller has to unlock after he is
348170030Srwatson	 * done.
349122922Sandre	 */
350122922Sandre	THC_LOCK(&hc_head->hch_mtx);
351122922Sandre
352122922Sandre	/*
353170030Srwatson	 * If the bucket limit is reached, reuse the least-used element.
354122922Sandre	 */
355181803Sbz	if (hc_head->hch_length >= V_tcp_hostcache.bucket_limit ||
356181803Sbz	    V_tcp_hostcache.cache_count >= V_tcp_hostcache.cache_limit) {
357122922Sandre		hc_entry = TAILQ_LAST(&hc_head->hch_bucket, hc_qhead);
358122922Sandre		/*
359122922Sandre		 * At first we were dropping the last element, just to
360170030Srwatson		 * reacquire it in the next two lines again, which isn't very
361170030Srwatson		 * efficient.  Instead just reuse the least used element.
362170030Srwatson		 * We may drop something that is still "in-use" but we can be
363170030Srwatson		 * "lossy".
364170405Sandre		 * Just give up if this bucket row is empty and we don't have
365170405Sandre		 * anything to replace.
366122922Sandre		 */
367170405Sandre		if (hc_entry == NULL) {
368170405Sandre			THC_UNLOCK(&hc_head->hch_mtx);
369170405Sandre			return NULL;
370170405Sandre		}
371122922Sandre		TAILQ_REMOVE(&hc_head->hch_bucket, hc_entry, rmx_q);
372181803Sbz		V_tcp_hostcache.hashbase[hash].hch_length--;
373181803Sbz		V_tcp_hostcache.cache_count--;
374181803Sbz		V_tcpstat.tcps_hc_bucketoverflow++;
375123028Sandre#if 0
376181803Sbz		uma_zfree(V_tcp_hostcache.zone, hc_entry);
377122922Sandre#endif
378122922Sandre	} else {
379122922Sandre		/*
380170030Srwatson		 * Allocate a new entry, or balk if not possible.
381122922Sandre		 */
382181803Sbz		hc_entry = uma_zalloc(V_tcp_hostcache.zone, M_NOWAIT);
383122922Sandre		if (hc_entry == NULL) {
384122922Sandre			THC_UNLOCK(&hc_head->hch_mtx);
385122922Sandre			return NULL;
386122922Sandre		}
387122922Sandre	}
388122922Sandre
389122922Sandre	/*
390170030Srwatson	 * Initialize basic information of hostcache entry.
391122922Sandre	 */
392122922Sandre	bzero(hc_entry, sizeof(*hc_entry));
393122922Sandre	if (inc->inc_isipv6)
394123113Sandre		bcopy(&inc->inc6_faddr, &hc_entry->ip6, sizeof(hc_entry->ip6));
395122922Sandre	else
396122922Sandre		hc_entry->ip4 = inc->inc_faddr;
397122922Sandre	hc_entry->rmx_head = hc_head;
398181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire;
399122922Sandre
400122922Sandre	/*
401170030Srwatson	 * Put it upfront.
402122922Sandre	 */
403122922Sandre	TAILQ_INSERT_HEAD(&hc_head->hch_bucket, hc_entry, rmx_q);
404181803Sbz	V_tcp_hostcache.hashbase[hash].hch_length++;
405181803Sbz	V_tcp_hostcache.cache_count++;
406181803Sbz	V_tcpstat.tcps_hc_added++;
407122922Sandre
408122922Sandre	return hc_entry;
409122922Sandre}
410122922Sandre
411122922Sandre/*
412170030Srwatson * External function: look up an entry in the hostcache and fill out the
413170030Srwatson * supplied TCP metrics structure.  Fills in NULL when no entry was found or
414170030Srwatson * a value is not set.
415122922Sandre */
416122922Sandrevoid
417122922Sandretcp_hc_get(struct in_conninfo *inc, struct hc_metrics_lite *hc_metrics_lite)
418122922Sandre{
419122922Sandre	struct hc_metrics *hc_entry;
420122922Sandre
421122922Sandre	/*
422170030Srwatson	 * Find the right bucket.
423122922Sandre	 */
424122922Sandre	hc_entry = tcp_hc_lookup(inc);
425122922Sandre
426122922Sandre	/*
427170030Srwatson	 * If we don't have an existing object.
428122922Sandre	 */
429122922Sandre	if (hc_entry == NULL) {
430122922Sandre		bzero(hc_metrics_lite, sizeof(*hc_metrics_lite));
431122922Sandre		return;
432122922Sandre	}
433122922Sandre	hc_entry->rmx_hits++;
434181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
435122922Sandre
436122922Sandre	hc_metrics_lite->rmx_mtu = hc_entry->rmx_mtu;
437122922Sandre	hc_metrics_lite->rmx_ssthresh = hc_entry->rmx_ssthresh;
438122922Sandre	hc_metrics_lite->rmx_rtt = hc_entry->rmx_rtt;
439122922Sandre	hc_metrics_lite->rmx_rttvar = hc_entry->rmx_rttvar;
440122922Sandre	hc_metrics_lite->rmx_bandwidth = hc_entry->rmx_bandwidth;
441122922Sandre	hc_metrics_lite->rmx_cwnd = hc_entry->rmx_cwnd;
442122922Sandre	hc_metrics_lite->rmx_sendpipe = hc_entry->rmx_sendpipe;
443122922Sandre	hc_metrics_lite->rmx_recvpipe = hc_entry->rmx_recvpipe;
444122922Sandre
445122922Sandre	/*
446170030Srwatson	 * Unlock bucket row.
447122922Sandre	 */
448122922Sandre	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
449122922Sandre}
450122922Sandre
451122922Sandre/*
452170030Srwatson * External function: look up an entry in the hostcache and return the
453170030Srwatson * discovered path MTU.  Returns NULL if no entry is found or value is not
454138409Srwatson * set.
455122922Sandre */
456122922Sandreu_long
457122922Sandretcp_hc_getmtu(struct in_conninfo *inc)
458122922Sandre{
459122922Sandre	struct hc_metrics *hc_entry;
460122922Sandre	u_long mtu;
461122922Sandre
462122922Sandre	hc_entry = tcp_hc_lookup(inc);
463122922Sandre	if (hc_entry == NULL) {
464122922Sandre		return 0;
465122922Sandre	}
466122922Sandre	hc_entry->rmx_hits++;
467181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
468122922Sandre
469122922Sandre	mtu = hc_entry->rmx_mtu;
470122922Sandre	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
471122922Sandre	return mtu;
472122922Sandre}
473122922Sandre
474122922Sandre/*
475170030Srwatson * External function: update the MTU value of an entry in the hostcache.
476122922Sandre * Creates a new entry if none was found.
477122922Sandre */
478122922Sandrevoid
479122922Sandretcp_hc_updatemtu(struct in_conninfo *inc, u_long mtu)
480122922Sandre{
481122922Sandre	struct hc_metrics *hc_entry;
482122922Sandre
483122922Sandre	/*
484170030Srwatson	 * Find the right bucket.
485122922Sandre	 */
486122922Sandre	hc_entry = tcp_hc_lookup(inc);
487122922Sandre
488122922Sandre	/*
489170030Srwatson	 * If we don't have an existing object, try to insert a new one.
490122922Sandre	 */
491122922Sandre	if (hc_entry == NULL) {
492122922Sandre		hc_entry = tcp_hc_insert(inc);
493122922Sandre		if (hc_entry == NULL)
494122922Sandre			return;
495122922Sandre	}
496122922Sandre	hc_entry->rmx_updates++;
497181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
498122922Sandre
499122922Sandre	hc_entry->rmx_mtu = mtu;
500122922Sandre
501122922Sandre	/*
502170030Srwatson	 * Put it upfront so we find it faster next time.
503122922Sandre	 */
504122922Sandre	TAILQ_REMOVE(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
505122922Sandre	TAILQ_INSERT_HEAD(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
506122922Sandre
507122922Sandre	/*
508170030Srwatson	 * Unlock bucket row.
509122922Sandre	 */
510122922Sandre	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
511122922Sandre}
512122922Sandre
513122922Sandre/*
514170030Srwatson * External function: update the TCP metrics of an entry in the hostcache.
515122922Sandre * Creates a new entry if none was found.
516122922Sandre */
517122922Sandrevoid
518122922Sandretcp_hc_update(struct in_conninfo *inc, struct hc_metrics_lite *hcml)
519122922Sandre{
520122922Sandre	struct hc_metrics *hc_entry;
521122922Sandre
522122922Sandre	hc_entry = tcp_hc_lookup(inc);
523122922Sandre	if (hc_entry == NULL) {
524122922Sandre		hc_entry = tcp_hc_insert(inc);
525122922Sandre		if (hc_entry == NULL)
526122922Sandre			return;
527122922Sandre	}
528122922Sandre	hc_entry->rmx_updates++;
529181803Sbz	hc_entry->rmx_expire = V_tcp_hostcache.expire; /* start over again */
530122922Sandre
531122922Sandre	if (hcml->rmx_rtt != 0) {
532122922Sandre		if (hc_entry->rmx_rtt == 0)
533122922Sandre			hc_entry->rmx_rtt = hcml->rmx_rtt;
534122922Sandre		else
535122922Sandre			hc_entry->rmx_rtt =
536122922Sandre			    (hc_entry->rmx_rtt + hcml->rmx_rtt) / 2;
537181803Sbz		V_tcpstat.tcps_cachedrtt++;
538122922Sandre	}
539122922Sandre	if (hcml->rmx_rttvar != 0) {
540122922Sandre	        if (hc_entry->rmx_rttvar == 0)
541133874Srwatson			hc_entry->rmx_rttvar = hcml->rmx_rttvar;
542122922Sandre		else
543122922Sandre			hc_entry->rmx_rttvar =
544122922Sandre			    (hc_entry->rmx_rttvar + hcml->rmx_rttvar) / 2;
545181803Sbz		V_tcpstat.tcps_cachedrttvar++;
546122922Sandre	}
547122922Sandre	if (hcml->rmx_ssthresh != 0) {
548122922Sandre		if (hc_entry->rmx_ssthresh == 0)
549122922Sandre			hc_entry->rmx_ssthresh = hcml->rmx_ssthresh;
550122922Sandre		else
551122922Sandre			hc_entry->rmx_ssthresh =
552122922Sandre			    (hc_entry->rmx_ssthresh + hcml->rmx_ssthresh) / 2;
553181803Sbz		V_tcpstat.tcps_cachedssthresh++;
554122922Sandre	}
555122922Sandre	if (hcml->rmx_bandwidth != 0) {
556122922Sandre		if (hc_entry->rmx_bandwidth == 0)
557122922Sandre			hc_entry->rmx_bandwidth = hcml->rmx_bandwidth;
558122922Sandre		else
559122922Sandre			hc_entry->rmx_bandwidth =
560122922Sandre			    (hc_entry->rmx_bandwidth + hcml->rmx_bandwidth) / 2;
561181803Sbz		/* V_tcpstat.tcps_cachedbandwidth++; */
562122922Sandre	}
563122922Sandre	if (hcml->rmx_cwnd != 0) {
564122922Sandre		if (hc_entry->rmx_cwnd == 0)
565122922Sandre			hc_entry->rmx_cwnd = hcml->rmx_cwnd;
566122922Sandre		else
567122922Sandre			hc_entry->rmx_cwnd =
568122922Sandre			    (hc_entry->rmx_cwnd + hcml->rmx_cwnd) / 2;
569181803Sbz		/* V_tcpstat.tcps_cachedcwnd++; */
570122922Sandre	}
571122922Sandre	if (hcml->rmx_sendpipe != 0) {
572122922Sandre		if (hc_entry->rmx_sendpipe == 0)
573122922Sandre			hc_entry->rmx_sendpipe = hcml->rmx_sendpipe;
574122922Sandre		else
575122922Sandre			hc_entry->rmx_sendpipe =
576122922Sandre			    (hc_entry->rmx_sendpipe + hcml->rmx_sendpipe) /2;
577181803Sbz		/* V_tcpstat.tcps_cachedsendpipe++; */
578133874Srwatson	}
579122922Sandre	if (hcml->rmx_recvpipe != 0) {
580122922Sandre		if (hc_entry->rmx_recvpipe == 0)
581122922Sandre			hc_entry->rmx_recvpipe = hcml->rmx_recvpipe;
582122922Sandre		else
583122922Sandre			hc_entry->rmx_recvpipe =
584122922Sandre			    (hc_entry->rmx_recvpipe + hcml->rmx_recvpipe) /2;
585181803Sbz		/* V_tcpstat.tcps_cachedrecvpipe++; */
586122922Sandre	}
587122922Sandre
588122922Sandre	TAILQ_REMOVE(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
589122922Sandre	TAILQ_INSERT_HEAD(&hc_entry->rmx_head->hch_bucket, hc_entry, rmx_q);
590122922Sandre	THC_UNLOCK(&hc_entry->rmx_head->hch_mtx);
591122922Sandre}
592122922Sandre
593122922Sandre/*
594122922Sandre * Sysctl function: prints the list and values of all hostcache entries in
595122922Sandre * unsorted order.
596122922Sandre */
597122922Sandrestatic int
598122922Sandresysctl_tcp_hc_list(SYSCTL_HANDLER_ARGS)
599122922Sandre{
600122922Sandre	int bufsize;
601122922Sandre	int linesize = 128;
602122922Sandre	char *p, *buf;
603122922Sandre	int len, i, error;
604122922Sandre	struct hc_metrics *hc_entry;
605165118Sbz#ifdef INET6
606165118Sbz	char ip6buf[INET6_ADDRSTRLEN];
607165118Sbz#endif
608122922Sandre
609181803Sbz	bufsize = linesize * (V_tcp_hostcache.cache_count + 1);
610122922Sandre
611122922Sandre	p = buf = (char *)malloc(bufsize, M_TEMP, M_WAITOK|M_ZERO);
612122922Sandre
613122922Sandre	len = snprintf(p, linesize,
614122922Sandre		"\nIP address        MTU  SSTRESH      RTT   RTTVAR BANDWIDTH "
615122922Sandre		"    CWND SENDPIPE RECVPIPE HITS  UPD  EXP\n");
616122922Sandre	p += len;
617122922Sandre
618122922Sandre#define msec(u) (((u) + 500) / 1000)
619181803Sbz	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
620181803Sbz		THC_LOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
621181803Sbz		TAILQ_FOREACH(hc_entry, &V_tcp_hostcache.hashbase[i].hch_bucket,
622122922Sandre			      rmx_q) {
623122922Sandre			len = snprintf(p, linesize,
624122922Sandre			    "%-15s %5lu %8lu %6lums %6lums %9lu %8lu %8lu %8lu "
625122922Sandre			    "%4lu %4lu %4i\n",
626122922Sandre			    hc_entry->ip4.s_addr ? inet_ntoa(hc_entry->ip4) :
627122922Sandre#ifdef INET6
628165118Sbz				ip6_sprintf(ip6buf, &hc_entry->ip6),
629122922Sandre#else
630122922Sandre				"IPv6?",
631122922Sandre#endif
632122922Sandre			    hc_entry->rmx_mtu,
633122922Sandre			    hc_entry->rmx_ssthresh,
634122922Sandre			    msec(hc_entry->rmx_rtt *
635122922Sandre				(RTM_RTTUNIT / (hz * TCP_RTT_SCALE))),
636122922Sandre			    msec(hc_entry->rmx_rttvar *
637122922Sandre				(RTM_RTTUNIT / (hz * TCP_RTT_SCALE))),
638133477Sandre			    hc_entry->rmx_bandwidth * 8,
639122922Sandre			    hc_entry->rmx_cwnd,
640122922Sandre			    hc_entry->rmx_sendpipe,
641122922Sandre			    hc_entry->rmx_recvpipe,
642122922Sandre			    hc_entry->rmx_hits,
643122922Sandre			    hc_entry->rmx_updates,
644122922Sandre			    hc_entry->rmx_expire);
645122922Sandre			p += len;
646122922Sandre		}
647181803Sbz		THC_UNLOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
648122922Sandre	}
649122922Sandre#undef msec
650122922Sandre	error = SYSCTL_OUT(req, buf, p - buf);
651122922Sandre	free(buf, M_TEMP);
652122922Sandre	return(error);
653122922Sandre}
654122922Sandre
655122922Sandre/*
656170030Srwatson * Expire and purge (old|all) entries in the tcp_hostcache.  Runs
657170030Srwatson * periodically from the callout.
658122922Sandre */
659122922Sandrestatic void
660122922Sandretcp_hc_purge(void *arg)
661122922Sandre{
662128574Sandre	struct hc_metrics *hc_entry, *hc_next;
663122922Sandre	int all = (intptr_t)arg;
664122922Sandre	int i;
665122922Sandre
666181803Sbz	if (V_tcp_hostcache.purgeall) {
667122922Sandre		all = 1;
668181803Sbz		V_tcp_hostcache.purgeall = 0;
669122922Sandre	}
670122922Sandre
671181803Sbz	for (i = 0; i < V_tcp_hostcache.hashsize; i++) {
672181803Sbz		THC_LOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
673181887Sjulian		TAILQ_FOREACH_SAFE(hc_entry,
674181888Sjulian		    &V_tcp_hostcache.hashbase[i].hch_bucket, rmx_q, hc_next) {
675122922Sandre			if (all || hc_entry->rmx_expire <= 0) {
676181803Sbz				TAILQ_REMOVE(&V_tcp_hostcache.hashbase[i].hch_bucket,
677122922Sandre					      hc_entry, rmx_q);
678181803Sbz				uma_zfree(V_tcp_hostcache.zone, hc_entry);
679181803Sbz				V_tcp_hostcache.hashbase[i].hch_length--;
680181803Sbz				V_tcp_hostcache.cache_count--;
681122922Sandre			} else
682181803Sbz				hc_entry->rmx_expire -= V_tcp_hostcache.prune;
683122922Sandre		}
684181803Sbz		THC_UNLOCK(&V_tcp_hostcache.hashbase[i].hch_mtx);
685122922Sandre	}
686181887Sjulian
687181887Sjulian	callout_reset(&V_tcp_hc_callout, V_tcp_hostcache.prune * hz,
688181887Sjulian	    tcp_hc_purge, arg);
689122922Sandre}
690