xref: /freebsd/sys/vm/uma_core.c (revision 9c2cd7e5a9f071a6aecc18b7e70f815708a47012)
18355f576SJeff Roberson /*
28355f576SJeff Roberson  * Copyright (c) 2002, Jeffrey Roberson <jroberson@chesapeake.net>
38355f576SJeff Roberson  * All rights reserved.
48355f576SJeff Roberson  *
58355f576SJeff Roberson  * Redistribution and use in source and binary forms, with or without
68355f576SJeff Roberson  * modification, are permitted provided that the following conditions
78355f576SJeff Roberson  * are met:
88355f576SJeff Roberson  * 1. Redistributions of source code must retain the above copyright
98355f576SJeff Roberson  *    notice unmodified, this list of conditions, and the following
108355f576SJeff Roberson  *    disclaimer.
118355f576SJeff Roberson  * 2. Redistributions in binary form must reproduce the above copyright
128355f576SJeff Roberson  *    notice, this list of conditions and the following disclaimer in the
138355f576SJeff Roberson  *    documentation and/or other materials provided with the distribution.
148355f576SJeff Roberson  *
158355f576SJeff Roberson  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
168355f576SJeff Roberson  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
178355f576SJeff Roberson  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
188355f576SJeff Roberson  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
198355f576SJeff Roberson  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
208355f576SJeff Roberson  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
218355f576SJeff Roberson  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
228355f576SJeff Roberson  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
238355f576SJeff Roberson  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
248355f576SJeff Roberson  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
258355f576SJeff Roberson  *
268355f576SJeff Roberson  * $FreeBSD$
278355f576SJeff Roberson  *
288355f576SJeff Roberson  */
298355f576SJeff Roberson 
308355f576SJeff Roberson /*
318355f576SJeff Roberson  * uma_core.c  Implementation of the Universal Memory allocator
328355f576SJeff Roberson  *
338355f576SJeff Roberson  * This allocator is intended to replace the multitude of similar object caches
348355f576SJeff Roberson  * in the standard FreeBSD kernel.  The intent is to be flexible as well as
358355f576SJeff Roberson  * effecient.  A primary design goal is to return unused memory to the rest of
368355f576SJeff Roberson  * the system.  This will make the system as a whole more flexible due to the
378355f576SJeff Roberson  * ability to move memory to subsystems which most need it instead of leaving
388355f576SJeff Roberson  * pools of reserved memory unused.
398355f576SJeff Roberson  *
408355f576SJeff Roberson  * The basic ideas stem from similar slab/zone based allocators whose algorithms
418355f576SJeff Roberson  * are well known.
428355f576SJeff Roberson  *
438355f576SJeff Roberson  */
448355f576SJeff Roberson 
458355f576SJeff Roberson /*
468355f576SJeff Roberson  * TODO:
478355f576SJeff Roberson  *	- Improve memory usage for large allocations
488355f576SJeff Roberson  *	- Improve INVARIANTS (0xdeadc0de write out)
498355f576SJeff Roberson  *	- Investigate cache size adjustments
508355f576SJeff Roberson  */
518355f576SJeff Roberson 
528355f576SJeff Roberson /* I should really use ktr.. */
538355f576SJeff Roberson /*
548355f576SJeff Roberson #define UMA_DEBUG 1
558355f576SJeff Roberson #define UMA_DEBUG_ALLOC 1
568355f576SJeff Roberson #define UMA_DEBUG_ALLOC_1 1
578355f576SJeff Roberson */
588355f576SJeff Roberson 
598355f576SJeff Roberson 
608355f576SJeff Roberson #include "opt_param.h"
618355f576SJeff Roberson #include <sys/param.h>
628355f576SJeff Roberson #include <sys/systm.h>
638355f576SJeff Roberson #include <sys/kernel.h>
648355f576SJeff Roberson #include <sys/types.h>
658355f576SJeff Roberson #include <sys/queue.h>
668355f576SJeff Roberson #include <sys/malloc.h>
678355f576SJeff Roberson #include <sys/lock.h>
688355f576SJeff Roberson #include <sys/sysctl.h>
698355f576SJeff Roberson #include <machine/types.h>
708355f576SJeff Roberson #include <sys/mutex.h>
718355f576SJeff Roberson #include <sys/smp.h>
728355f576SJeff Roberson 
738355f576SJeff Roberson #include <vm/vm.h>
748355f576SJeff Roberson #include <vm/vm_object.h>
758355f576SJeff Roberson #include <vm/vm_page.h>
768355f576SJeff Roberson #include <vm/vm_param.h>
778355f576SJeff Roberson #include <vm/vm_map.h>
788355f576SJeff Roberson #include <vm/vm_kern.h>
798355f576SJeff Roberson #include <vm/vm_extern.h>
808355f576SJeff Roberson #include <vm/uma.h>
818355f576SJeff Roberson #include <vm/uma_int.h>
828355f576SJeff Roberson 
838355f576SJeff Roberson /*
848355f576SJeff Roberson  * This is the zone from which all zones are spawned.  The idea is that even
858355f576SJeff Roberson  * the zone heads are allocated from the allocator, so we use the bss section
868355f576SJeff Roberson  * to bootstrap us.
878355f576SJeff Roberson  */
888355f576SJeff Roberson static struct uma_zone master_zone;
898355f576SJeff Roberson static uma_zone_t zones = &master_zone;
908355f576SJeff Roberson 
918355f576SJeff Roberson /* This is the zone from which all of uma_slab_t's are allocated. */
928355f576SJeff Roberson static uma_zone_t slabzone;
938355f576SJeff Roberson 
948355f576SJeff Roberson /*
958355f576SJeff Roberson  * The initial hash tables come out of this zone so they can be allocated
968355f576SJeff Roberson  * prior to malloc coming up.
978355f576SJeff Roberson  */
988355f576SJeff Roberson static uma_zone_t hashzone;
998355f576SJeff Roberson 
1008355f576SJeff Roberson /*
1018355f576SJeff Roberson  * Zone that buckets come from.
1028355f576SJeff Roberson  */
1038355f576SJeff Roberson static uma_zone_t bucketzone;
1048355f576SJeff Roberson 
1058355f576SJeff Roberson /* Linked list of all zones in the system */
1068355f576SJeff Roberson static LIST_HEAD(,uma_zone) uma_zones = LIST_HEAD_INITIALIZER(&uma_zones);
1078355f576SJeff Roberson 
1088355f576SJeff Roberson /* This mutex protects the zone list */
1098355f576SJeff Roberson static struct mtx uma_mtx;
1108355f576SJeff Roberson 
1118355f576SJeff Roberson /* Linked list of boot time pages */
1128355f576SJeff Roberson static LIST_HEAD(,uma_slab) uma_boot_pages =
1138355f576SJeff Roberson     LIST_HEAD_INITIALIZER(&uma_boot_pages);
1148355f576SJeff Roberson 
1158355f576SJeff Roberson /* Count of free boottime pages */
1168355f576SJeff Roberson static int uma_boot_free = 0;
1178355f576SJeff Roberson 
1188355f576SJeff Roberson /* Is the VM done starting up? */
1198355f576SJeff Roberson static int booted = 0;
1208355f576SJeff Roberson 
1218355f576SJeff Roberson /* This is the handle used to schedule our working set calculator */
1228355f576SJeff Roberson static struct callout uma_callout;
1238355f576SJeff Roberson 
1248355f576SJeff Roberson /* This is mp_maxid + 1, for use while looping over each cpu */
1258355f576SJeff Roberson static int maxcpu;
1268355f576SJeff Roberson 
1278355f576SJeff Roberson /*
1288355f576SJeff Roberson  * This structure is passed as the zone ctor arg so that I don't have to create
1298355f576SJeff Roberson  * a special allocation function just for zones.
1308355f576SJeff Roberson  */
1318355f576SJeff Roberson struct uma_zctor_args {
1328355f576SJeff Roberson 	char *name;
1338355f576SJeff Roberson 	int size;
1348355f576SJeff Roberson 	uma_ctor ctor;
1358355f576SJeff Roberson 	uma_dtor dtor;
1368355f576SJeff Roberson 	uma_init uminit;
1378355f576SJeff Roberson 	uma_fini fini;
1388355f576SJeff Roberson 	int align;
1398355f576SJeff Roberson 	u_int16_t flags;
1408355f576SJeff Roberson };
1418355f576SJeff Roberson 
1428355f576SJeff Roberson /*
1438355f576SJeff Roberson  * This is the malloc hash table which is used to find the zone that a
1448355f576SJeff Roberson  * malloc allocation came from.  It is not currently resizeable.  The
1458355f576SJeff Roberson  * memory for the actual hash bucket is allocated in kmeminit.
1468355f576SJeff Roberson  */
1478355f576SJeff Roberson struct uma_hash mhash;
1488355f576SJeff Roberson struct uma_hash *mallochash = &mhash;
1498355f576SJeff Roberson 
1508355f576SJeff Roberson /* Prototypes.. */
1518355f576SJeff Roberson 
1528355f576SJeff Roberson static void *obj_alloc(uma_zone_t, int, u_int8_t *, int);
1538355f576SJeff Roberson static void *page_alloc(uma_zone_t, int, u_int8_t *, int);
1548355f576SJeff Roberson static void page_free(void *, int, u_int8_t);
1558355f576SJeff Roberson static uma_slab_t slab_zalloc(uma_zone_t, int);
1568355f576SJeff Roberson static void cache_drain(uma_zone_t);
1578355f576SJeff Roberson static void bucket_drain(uma_zone_t, uma_bucket_t);
1588355f576SJeff Roberson static void zone_drain(uma_zone_t);
1598355f576SJeff Roberson static void zone_ctor(void *, int, void *);
1609c2cd7e5SJeff Roberson static void zone_dtor(void *, int, void *);
1618355f576SJeff Roberson static void zero_init(void *, int);
1628355f576SJeff Roberson static void zone_small_init(uma_zone_t zone);
1638355f576SJeff Roberson static void zone_large_init(uma_zone_t zone);
1648355f576SJeff Roberson static void zone_foreach(void (*zfunc)(uma_zone_t));
1658355f576SJeff Roberson static void zone_timeout(uma_zone_t zone);
1668355f576SJeff Roberson static void hash_expand(struct uma_hash *);
1679c2cd7e5SJeff Roberson static void hash_free(struct uma_hash *hash);
1688355f576SJeff Roberson static void uma_timeout(void *);
1698355f576SJeff Roberson static void uma_startup3(void);
170a553d4b8SJeff Roberson static void *uma_zalloc_internal(uma_zone_t, void *, int, uma_bucket_t);
1718355f576SJeff Roberson static void uma_zfree_internal(uma_zone_t,
1728355f576SJeff Roberson     void *, void *, int);
1738355f576SJeff Roberson void uma_print_zone(uma_zone_t);
1748355f576SJeff Roberson void uma_print_stats(void);
1758355f576SJeff Roberson static int sysctl_vm_zone(SYSCTL_HANDLER_ARGS);
1768355f576SJeff Roberson 
1778355f576SJeff Roberson SYSCTL_OID(_vm, OID_AUTO, zone, CTLTYPE_STRING|CTLFLAG_RD,
1788355f576SJeff Roberson     NULL, 0, sysctl_vm_zone, "A", "Zone Info");
1798355f576SJeff Roberson SYSINIT(uma_startup3, SI_SUB_VM_CONF, SI_ORDER_SECOND, uma_startup3, NULL);
1808355f576SJeff Roberson 
1818355f576SJeff Roberson 
1828355f576SJeff Roberson /*
1838355f576SJeff Roberson  * Routine called by timeout which is used to fire off some time interval
1848355f576SJeff Roberson  * based calculations.  (working set, stats, etc.)
1858355f576SJeff Roberson  *
1868355f576SJeff Roberson  * Arguments:
1878355f576SJeff Roberson  *	arg   Unused
1888355f576SJeff Roberson  *
1898355f576SJeff Roberson  * Returns:
1908355f576SJeff Roberson  *	Nothing
1918355f576SJeff Roberson  */
1928355f576SJeff Roberson static void
1938355f576SJeff Roberson uma_timeout(void *unused)
1948355f576SJeff Roberson {
1958355f576SJeff Roberson 	zone_foreach(zone_timeout);
1968355f576SJeff Roberson 
1978355f576SJeff Roberson 	/* Reschedule this event */
1988355f576SJeff Roberson 	callout_reset(&uma_callout, UMA_WORKING_TIME * hz, uma_timeout, NULL);
1998355f576SJeff Roberson }
2008355f576SJeff Roberson 
2018355f576SJeff Roberson /*
2028355f576SJeff Roberson  * Routine to perform timeout driven calculations.  This does the working set
2038355f576SJeff Roberson  * as well as hash expanding, and per cpu statistics aggregation.
2048355f576SJeff Roberson  *
2058355f576SJeff Roberson  *  Arguments:
2068355f576SJeff Roberson  *	zone  The zone to operate on
2078355f576SJeff Roberson  *
2088355f576SJeff Roberson  *  Returns:
2098355f576SJeff Roberson  *	Nothing
2108355f576SJeff Roberson  */
2118355f576SJeff Roberson static void
2128355f576SJeff Roberson zone_timeout(uma_zone_t zone)
2138355f576SJeff Roberson {
2148355f576SJeff Roberson 	uma_cache_t cache;
2158355f576SJeff Roberson 	u_int64_t alloc;
2168355f576SJeff Roberson 	int free;
2178355f576SJeff Roberson 	int cpu;
2188355f576SJeff Roberson 
2198355f576SJeff Roberson 	alloc = 0;
2208355f576SJeff Roberson 	free = 0;
2218355f576SJeff Roberson 
2228355f576SJeff Roberson 	/*
2238355f576SJeff Roberson 	 * Aggregate per cpu cache statistics back to the zone.
2248355f576SJeff Roberson 	 *
2258355f576SJeff Roberson 	 * I may rewrite this to set a flag in the per cpu cache instead of
2268355f576SJeff Roberson 	 * locking.  If the flag is not cleared on the next round I will have
2278355f576SJeff Roberson 	 * to lock and do it here instead so that the statistics don't get too
2288355f576SJeff Roberson 	 * far out of sync.
2298355f576SJeff Roberson 	 */
2308355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_INTERNAL)) {
2318355f576SJeff Roberson 		for (cpu = 0; cpu < maxcpu; cpu++) {
2328355f576SJeff Roberson 			if (CPU_ABSENT(cpu))
2338355f576SJeff Roberson 				continue;
2348355f576SJeff Roberson 			CPU_LOCK(zone, cpu);
2358355f576SJeff Roberson 			cache = &zone->uz_cpu[cpu];
2368355f576SJeff Roberson 			/* Add them up, and reset */
2378355f576SJeff Roberson 			alloc += cache->uc_allocs;
2388355f576SJeff Roberson 			cache->uc_allocs = 0;
2398355f576SJeff Roberson 			if (cache->uc_allocbucket)
2408355f576SJeff Roberson 				free += cache->uc_allocbucket->ub_ptr + 1;
2418355f576SJeff Roberson 			if (cache->uc_freebucket)
2428355f576SJeff Roberson 				free += cache->uc_freebucket->ub_ptr + 1;
2438355f576SJeff Roberson 			CPU_UNLOCK(zone, cpu);
2448355f576SJeff Roberson 		}
2458355f576SJeff Roberson 	}
2468355f576SJeff Roberson 
2478355f576SJeff Roberson 	/* Now push these stats back into the zone.. */
2488355f576SJeff Roberson 	ZONE_LOCK(zone);
2498355f576SJeff Roberson 	zone->uz_allocs += alloc;
2508355f576SJeff Roberson 
2518355f576SJeff Roberson 	/*
2528355f576SJeff Roberson 	 * cachefree is an instantanious snapshot of what is in the per cpu
2538355f576SJeff Roberson 	 * caches, not an accurate counter
2548355f576SJeff Roberson 	 */
2558355f576SJeff Roberson 	zone->uz_cachefree = free;
2568355f576SJeff Roberson 
2578355f576SJeff Roberson 	/*
2588355f576SJeff Roberson 	 * Expand the zone hash table.
2598355f576SJeff Roberson 	 *
2608355f576SJeff Roberson 	 * This is done if the number of slabs is larger than the hash size.
2618355f576SJeff Roberson 	 * What I'm trying to do here is completely reduce collisions.  This
2628355f576SJeff Roberson 	 * may be a little aggressive.  Should I allow for two collisions max?
2638355f576SJeff Roberson 	 */
2648355f576SJeff Roberson 
2658355f576SJeff Roberson 	if ((zone->uz_flags & UMA_ZFLAG_OFFPAGE) &&
2668355f576SJeff Roberson 	    !(zone->uz_flags & UMA_ZFLAG_MALLOC)) {
2678355f576SJeff Roberson 		if (zone->uz_pages / zone->uz_ppera
2688355f576SJeff Roberson 		    >= zone->uz_hash.uh_hashsize)
2698355f576SJeff Roberson 			hash_expand(&zone->uz_hash);
2708355f576SJeff Roberson 	}
2718355f576SJeff Roberson 
2728355f576SJeff Roberson 	/*
2738355f576SJeff Roberson 	 * Here we compute the working set size as the total number of items
2748355f576SJeff Roberson 	 * left outstanding since the last time interval.  This is slightly
2758355f576SJeff Roberson 	 * suboptimal. What we really want is the highest number of outstanding
2768355f576SJeff Roberson 	 * items during the last time quantum.  This should be close enough.
2778355f576SJeff Roberson 	 *
2788355f576SJeff Roberson 	 * The working set size is used to throttle the zone_drain function.
2798355f576SJeff Roberson 	 * We don't want to return memory that we may need again immediately.
2808355f576SJeff Roberson 	 */
2818355f576SJeff Roberson 	alloc = zone->uz_allocs - zone->uz_oallocs;
2828355f576SJeff Roberson 	zone->uz_oallocs = zone->uz_allocs;
2838355f576SJeff Roberson 	zone->uz_wssize = alloc;
2848355f576SJeff Roberson 
2858355f576SJeff Roberson 	ZONE_UNLOCK(zone);
2868355f576SJeff Roberson }
2878355f576SJeff Roberson 
2888355f576SJeff Roberson /*
2898355f576SJeff Roberson  * Expands the hash table for OFFPAGE zones.  This is done from zone_timeout
2908355f576SJeff Roberson  * to reduce collisions.  This must not be done in the regular allocation path,
2918355f576SJeff Roberson  * otherwise, we can recurse on the vm while allocating pages.
2928355f576SJeff Roberson  *
2938355f576SJeff Roberson  * Arguments:
2948355f576SJeff Roberson  *	hash  The hash you want to expand by a factor of two.
2958355f576SJeff Roberson  *
2968355f576SJeff Roberson  * Returns:
2978355f576SJeff Roberson  * 	Nothing
2988355f576SJeff Roberson  *
2998355f576SJeff Roberson  * Discussion:
3008355f576SJeff Roberson  */
3018355f576SJeff Roberson static void
3028355f576SJeff Roberson hash_expand(struct uma_hash *hash)
3038355f576SJeff Roberson {
3048355f576SJeff Roberson 	struct slabhead *newhash;
3058355f576SJeff Roberson 	struct slabhead *oldhash;
3068355f576SJeff Roberson 	uma_slab_t slab;
3079c2cd7e5SJeff Roberson 	int oldsize;
3089c2cd7e5SJeff Roberson 	int newsize;
3098355f576SJeff Roberson 	int alloc;
3108355f576SJeff Roberson 	int hval;
3118355f576SJeff Roberson 	int i;
3128355f576SJeff Roberson 
3138355f576SJeff Roberson 
3148355f576SJeff Roberson 	/*
3158355f576SJeff Roberson 	 * Remember the old hash size and see if it has to go back to the
3168355f576SJeff Roberson  	 * hash zone, or malloc.  The hash zone is used for the initial hash
3178355f576SJeff Roberson 	 */
3188355f576SJeff Roberson 
3199c2cd7e5SJeff Roberson 	oldsize = hash->uh_hashsize;
3208355f576SJeff Roberson 	oldhash = hash->uh_slab_hash;
3218355f576SJeff Roberson 
3228355f576SJeff Roberson 	/* We're just going to go to a power of two greater */
3238355f576SJeff Roberson 	if (hash->uh_hashsize)  {
3249c2cd7e5SJeff Roberson 		newsize = oldsize * 2;
3259c2cd7e5SJeff Roberson 		alloc = sizeof(hash->uh_slab_hash[0]) * newsize;
3268355f576SJeff Roberson 		/* XXX Shouldn't be abusing DEVBUF here */
3278355f576SJeff Roberson 		newhash = (struct slabhead *)malloc(alloc, M_DEVBUF, M_NOWAIT);
3288355f576SJeff Roberson 		if (newhash == NULL) {
3298355f576SJeff Roberson 			return;
3308355f576SJeff Roberson 		}
3318355f576SJeff Roberson 	} else {
3328355f576SJeff Roberson 		alloc = sizeof(hash->uh_slab_hash[0]) * UMA_HASH_SIZE_INIT;
333a553d4b8SJeff Roberson 		newhash = uma_zalloc_internal(hashzone, NULL, M_WAITOK, NULL);
3349c2cd7e5SJeff Roberson 		newsize = UMA_HASH_SIZE_INIT;
3358355f576SJeff Roberson 	}
3368355f576SJeff Roberson 
3378355f576SJeff Roberson 	bzero(newhash, alloc);
3388355f576SJeff Roberson 
3399c2cd7e5SJeff Roberson 	hash->uh_hashmask = newsize - 1;
3408355f576SJeff Roberson 
3418355f576SJeff Roberson 	/*
3428355f576SJeff Roberson 	 * I need to investigate hash algorithms for resizing without a
3438355f576SJeff Roberson 	 * full rehash.
3448355f576SJeff Roberson 	 */
3458355f576SJeff Roberson 
3469c2cd7e5SJeff Roberson 	for (i = 0; i < oldsize; i++)
3478355f576SJeff Roberson 		while (!SLIST_EMPTY(&hash->uh_slab_hash[i])) {
3488355f576SJeff Roberson 			slab = SLIST_FIRST(&hash->uh_slab_hash[i]);
3498355f576SJeff Roberson 			SLIST_REMOVE_HEAD(&hash->uh_slab_hash[i], us_hlink);
3508355f576SJeff Roberson 			hval = UMA_HASH(hash, slab->us_data);
3518355f576SJeff Roberson 			SLIST_INSERT_HEAD(&newhash[hval], slab, us_hlink);
3528355f576SJeff Roberson 		}
3538355f576SJeff Roberson 
3549c2cd7e5SJeff Roberson 	if (oldhash)
3559c2cd7e5SJeff Roberson 		hash_free(hash);
3569c2cd7e5SJeff Roberson 
3579c2cd7e5SJeff Roberson 	hash->uh_slab_hash = newhash;
3589c2cd7e5SJeff Roberson 	hash->uh_hashsize = newsize;
3599c2cd7e5SJeff Roberson 
3609c2cd7e5SJeff Roberson 	return;
3619c2cd7e5SJeff Roberson }
3629c2cd7e5SJeff Roberson 
3639c2cd7e5SJeff Roberson static void
3649c2cd7e5SJeff Roberson hash_free(struct uma_hash *hash)
3659c2cd7e5SJeff Roberson {
3669c2cd7e5SJeff Roberson 	if (hash->uh_hashsize == UMA_HASH_SIZE_INIT)
3678355f576SJeff Roberson 		uma_zfree_internal(hashzone,
3688355f576SJeff Roberson 		    hash->uh_slab_hash, NULL, 0);
3698355f576SJeff Roberson 	else
3708355f576SJeff Roberson 		free(hash->uh_slab_hash, M_DEVBUF);
3718355f576SJeff Roberson 
3729c2cd7e5SJeff Roberson 	hash->uh_slab_hash = NULL;
3738355f576SJeff Roberson }
3748355f576SJeff Roberson 
3758355f576SJeff Roberson /*
3768355f576SJeff Roberson  * Frees all outstanding items in a bucket
3778355f576SJeff Roberson  *
3788355f576SJeff Roberson  * Arguments:
3798355f576SJeff Roberson  *	zone   The zone to free to, must be unlocked.
3808355f576SJeff Roberson  *	bucket The free/alloc bucket with items, cpu queue must be locked.
3818355f576SJeff Roberson  *
3828355f576SJeff Roberson  * Returns:
3838355f576SJeff Roberson  *	Nothing
3848355f576SJeff Roberson  */
3858355f576SJeff Roberson 
3868355f576SJeff Roberson static void
3878355f576SJeff Roberson bucket_drain(uma_zone_t zone, uma_bucket_t bucket)
3888355f576SJeff Roberson {
3898355f576SJeff Roberson 	uma_slab_t slab;
3908355f576SJeff Roberson 	int mzone;
3918355f576SJeff Roberson 	void *item;
3928355f576SJeff Roberson 
3938355f576SJeff Roberson 	if (bucket == NULL)
3948355f576SJeff Roberson 		return;
3958355f576SJeff Roberson 
3968355f576SJeff Roberson 	slab = NULL;
3978355f576SJeff Roberson 	mzone = 0;
3988355f576SJeff Roberson 
3998355f576SJeff Roberson 	/* We have to lookup the slab again for malloc.. */
4008355f576SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_MALLOC)
4018355f576SJeff Roberson 		mzone = 1;
4028355f576SJeff Roberson 
4038355f576SJeff Roberson 	while (bucket->ub_ptr > -1)  {
4048355f576SJeff Roberson 		item = bucket->ub_bucket[bucket->ub_ptr];
4058355f576SJeff Roberson #ifdef INVARIANTS
4068355f576SJeff Roberson 		bucket->ub_bucket[bucket->ub_ptr] = NULL;
4078355f576SJeff Roberson 		KASSERT(item != NULL,
4088355f576SJeff Roberson 		    ("bucket_drain: botched ptr, item is NULL"));
4098355f576SJeff Roberson #endif
4108355f576SJeff Roberson 		bucket->ub_ptr--;
4118355f576SJeff Roberson 		/*
4128355f576SJeff Roberson 		 * This is extremely inefficient.  The slab pointer was passed
4138355f576SJeff Roberson 		 * to uma_zfree_arg, but we lost it because the buckets don't
4148355f576SJeff Roberson 		 * hold them.  This will go away when free() gets a size passed
4158355f576SJeff Roberson 		 * to it.
4168355f576SJeff Roberson 		 */
4178355f576SJeff Roberson 		if (mzone)
4188355f576SJeff Roberson 			slab = hash_sfind(mallochash,
4198355f576SJeff Roberson 			    (u_int8_t *)((unsigned long)item &
4208355f576SJeff Roberson 			   (~UMA_SLAB_MASK)));
4218355f576SJeff Roberson 		uma_zfree_internal(zone, item, slab, 1);
4228355f576SJeff Roberson 	}
4238355f576SJeff Roberson }
4248355f576SJeff Roberson 
4258355f576SJeff Roberson /*
4268355f576SJeff Roberson  * Drains the per cpu caches for a zone.
4278355f576SJeff Roberson  *
4288355f576SJeff Roberson  * Arguments:
4298355f576SJeff Roberson  *	zone  The zone to drain, must be unlocked.
4308355f576SJeff Roberson  *
4318355f576SJeff Roberson  * Returns:
4328355f576SJeff Roberson  *	Nothing
4338355f576SJeff Roberson  *
4348355f576SJeff Roberson  * This function returns with the zone locked so that the per cpu queues can
4358355f576SJeff Roberson  * not be filled until zone_drain is finished.
4368355f576SJeff Roberson  *
4378355f576SJeff Roberson  */
4388355f576SJeff Roberson static void
4398355f576SJeff Roberson cache_drain(uma_zone_t zone)
4408355f576SJeff Roberson {
4418355f576SJeff Roberson 	uma_bucket_t bucket;
4428355f576SJeff Roberson 	uma_cache_t cache;
4438355f576SJeff Roberson 	int cpu;
4448355f576SJeff Roberson 
4458355f576SJeff Roberson 	/*
4468355f576SJeff Roberson 	 * Flush out the per cpu queues.
4478355f576SJeff Roberson 	 *
448157d7b35SAlfred Perlstein 	 * XXX This causes unnecessary thrashing due to immediately having
4498355f576SJeff Roberson 	 * empty per cpu queues.  I need to improve this.
4508355f576SJeff Roberson 	 */
4518355f576SJeff Roberson 
4528355f576SJeff Roberson 	/*
4538355f576SJeff Roberson 	 * We have to lock each cpu cache before locking the zone
4548355f576SJeff Roberson 	 */
4558355f576SJeff Roberson 	ZONE_UNLOCK(zone);
4568355f576SJeff Roberson 
4578355f576SJeff Roberson 	for (cpu = 0; cpu < maxcpu; cpu++) {
4588355f576SJeff Roberson 		if (CPU_ABSENT(cpu))
4598355f576SJeff Roberson 			continue;
4608355f576SJeff Roberson 		CPU_LOCK(zone, cpu);
4618355f576SJeff Roberson 		cache = &zone->uz_cpu[cpu];
4628355f576SJeff Roberson 		bucket_drain(zone, cache->uc_allocbucket);
4638355f576SJeff Roberson 		bucket_drain(zone, cache->uc_freebucket);
4648355f576SJeff Roberson 	}
4658355f576SJeff Roberson 
4668355f576SJeff Roberson 	/*
4678355f576SJeff Roberson 	 * Drain the bucket queues and free the buckets, we just keep two per
4688355f576SJeff Roberson 	 * cpu (alloc/free).
4698355f576SJeff Roberson 	 */
4708355f576SJeff Roberson 	ZONE_LOCK(zone);
4718355f576SJeff Roberson 	while ((bucket = LIST_FIRST(&zone->uz_full_bucket)) != NULL) {
4728355f576SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
4738355f576SJeff Roberson 		ZONE_UNLOCK(zone);
4748355f576SJeff Roberson 		bucket_drain(zone, bucket);
4758355f576SJeff Roberson 		uma_zfree_internal(bucketzone, bucket, NULL, 0);
4768355f576SJeff Roberson 		ZONE_LOCK(zone);
4778355f576SJeff Roberson 	}
4788355f576SJeff Roberson 
4798355f576SJeff Roberson 	/* Now we do the free queue.. */
4808355f576SJeff Roberson 	while ((bucket = LIST_FIRST(&zone->uz_free_bucket)) != NULL) {
4818355f576SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
4828355f576SJeff Roberson 		uma_zfree_internal(bucketzone, bucket, NULL, 0);
4838355f576SJeff Roberson 	}
4848355f576SJeff Roberson 
4858355f576SJeff Roberson 	/* We unlock here, but they will all block until the zone is unlocked */
4868355f576SJeff Roberson 	for (cpu = 0; cpu < maxcpu; cpu++) {
4878355f576SJeff Roberson 		if (CPU_ABSENT(cpu))
4888355f576SJeff Roberson 			continue;
4898355f576SJeff Roberson 		CPU_UNLOCK(zone, cpu);
4908355f576SJeff Roberson 	}
491f4af24d5SJeff Roberson 
492f4af24d5SJeff Roberson 	zone->uz_cachefree = 0;
4938355f576SJeff Roberson }
4948355f576SJeff Roberson 
4958355f576SJeff Roberson /*
4968355f576SJeff Roberson  * Frees pages from a zone back to the system.  This is done on demand from
4978355f576SJeff Roberson  * the pageout daemon.
4988355f576SJeff Roberson  *
4998355f576SJeff Roberson  * Arguments:
5008355f576SJeff Roberson  *	zone  The zone to free pages from
5019c2cd7e5SJeff Roberson  *	all   Should we drain all items?
5028355f576SJeff Roberson  *
5038355f576SJeff Roberson  * Returns:
5048355f576SJeff Roberson  *	Nothing.
5058355f576SJeff Roberson  */
5068355f576SJeff Roberson static void
5078355f576SJeff Roberson zone_drain(uma_zone_t zone)
5088355f576SJeff Roberson {
5098355f576SJeff Roberson 	uma_slab_t slab;
5108355f576SJeff Roberson 	uma_slab_t n;
5118355f576SJeff Roberson 	u_int64_t extra;
5128355f576SJeff Roberson 	u_int8_t flags;
5138355f576SJeff Roberson 	u_int8_t *mem;
5148355f576SJeff Roberson 	int i;
5158355f576SJeff Roberson 
5168355f576SJeff Roberson 	/*
5178355f576SJeff Roberson 	 * We don't want to take pages from staticly allocated zones at this
5188355f576SJeff Roberson 	 * time
5198355f576SJeff Roberson 	 */
5208355f576SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_NOFREE || zone->uz_freef == NULL)
5218355f576SJeff Roberson 		return;
5228355f576SJeff Roberson 
5238355f576SJeff Roberson 	ZONE_LOCK(zone);
5248355f576SJeff Roberson 
5258355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_INTERNAL))
5268355f576SJeff Roberson 		cache_drain(zone);
5278355f576SJeff Roberson 
5288355f576SJeff Roberson 	if (zone->uz_free < zone->uz_wssize)
5298355f576SJeff Roberson 		goto finished;
5308355f576SJeff Roberson #ifdef UMA_DEBUG
5318355f576SJeff Roberson 	printf("%s working set size: %llu free items: %u\n",
5328355f576SJeff Roberson 	    zone->uz_name, (unsigned long long)zone->uz_wssize, zone->uz_free);
5338355f576SJeff Roberson #endif
5349c2cd7e5SJeff Roberson 	extra = zone->uz_free - zone->uz_wssize;
5358355f576SJeff Roberson 	extra /= zone->uz_ipers;
5368355f576SJeff Roberson 
5378355f576SJeff Roberson 	/* extra is now the number of extra slabs that we can free */
5388355f576SJeff Roberson 
5398355f576SJeff Roberson 	if (extra == 0)
5408355f576SJeff Roberson 		goto finished;
5418355f576SJeff Roberson 
5428355f576SJeff Roberson 	slab = LIST_FIRST(&zone->uz_free_slab);
5438355f576SJeff Roberson 	while (slab && extra) {
5448355f576SJeff Roberson 		n = LIST_NEXT(slab, us_link);
5458355f576SJeff Roberson 
5468355f576SJeff Roberson 		/* We have no where to free these to */
5478355f576SJeff Roberson 		if (slab->us_flags & UMA_SLAB_BOOT) {
5488355f576SJeff Roberson 			slab = n;
5498355f576SJeff Roberson 			continue;
5508355f576SJeff Roberson 		}
5518355f576SJeff Roberson 
5528355f576SJeff Roberson 		LIST_REMOVE(slab, us_link);
5538355f576SJeff Roberson 		zone->uz_pages -= zone->uz_ppera;
5548355f576SJeff Roberson 		zone->uz_free -= zone->uz_ipers;
5558355f576SJeff Roberson 		if (zone->uz_fini)
5568355f576SJeff Roberson 			for (i = 0; i < zone->uz_ipers; i++)
5578355f576SJeff Roberson 				zone->uz_fini(
5588355f576SJeff Roberson 				    slab->us_data + (zone->uz_rsize * i),
5598355f576SJeff Roberson 				    zone->uz_size);
5608355f576SJeff Roberson 		flags = slab->us_flags;
5618355f576SJeff Roberson 		mem = slab->us_data;
5628355f576SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_OFFPAGE) {
5638355f576SJeff Roberson 			if (zone->uz_flags & UMA_ZFLAG_MALLOC) {
5648355f576SJeff Roberson 				UMA_HASH_REMOVE(mallochash,
5658355f576SJeff Roberson 				    slab, slab->us_data);
5668355f576SJeff Roberson 			} else {
5678355f576SJeff Roberson 				UMA_HASH_REMOVE(&zone->uz_hash,
5688355f576SJeff Roberson 				    slab, slab->us_data);
5698355f576SJeff Roberson 			}
5708355f576SJeff Roberson 			uma_zfree_internal(slabzone, slab, NULL, 0);
5718355f576SJeff Roberson 		} else if (zone->uz_flags & UMA_ZFLAG_MALLOC)
5728355f576SJeff Roberson 			UMA_HASH_REMOVE(mallochash, slab, slab->us_data);
5738355f576SJeff Roberson #ifdef UMA_DEBUG
5748355f576SJeff Roberson 		printf("%s: Returning %d bytes.\n",
5758355f576SJeff Roberson 		    zone->uz_name, UMA_SLAB_SIZE * zone->uz_ppera);
5768355f576SJeff Roberson #endif
5778355f576SJeff Roberson 		zone->uz_freef(mem, UMA_SLAB_SIZE * zone->uz_ppera, flags);
5788355f576SJeff Roberson 
5798355f576SJeff Roberson 		slab = n;
5808355f576SJeff Roberson 		extra--;
5818355f576SJeff Roberson 	}
5828355f576SJeff Roberson 
5838355f576SJeff Roberson finished:
5848355f576SJeff Roberson 	ZONE_UNLOCK(zone);
5858355f576SJeff Roberson }
5868355f576SJeff Roberson 
5878355f576SJeff Roberson /*
5888355f576SJeff Roberson  * Allocate a new slab for a zone.  This does not insert the slab onto a list.
5898355f576SJeff Roberson  *
5908355f576SJeff Roberson  * Arguments:
5918355f576SJeff Roberson  *	zone  The zone to allocate slabs for
5928355f576SJeff Roberson  *	wait  Shall we wait?
5938355f576SJeff Roberson  *
5948355f576SJeff Roberson  * Returns:
5958355f576SJeff Roberson  *	The slab that was allocated or NULL if there is no memory and the
5968355f576SJeff Roberson  *	caller specified M_NOWAIT.
5978355f576SJeff Roberson  *
5988355f576SJeff Roberson  */
5998355f576SJeff Roberson static uma_slab_t
6008355f576SJeff Roberson slab_zalloc(uma_zone_t zone, int wait)
6018355f576SJeff Roberson {
6028355f576SJeff Roberson 	uma_slab_t slab;	/* Starting slab */
6038355f576SJeff Roberson 	u_int8_t *mem;
6048355f576SJeff Roberson 	u_int8_t flags;
6058355f576SJeff Roberson 	int i;
6068355f576SJeff Roberson 
607a553d4b8SJeff Roberson 	slab = NULL;
608a553d4b8SJeff Roberson 
6098355f576SJeff Roberson #ifdef UMA_DEBUG
6108355f576SJeff Roberson 	printf("slab_zalloc:  Allocating a new slab for %s\n", zone->uz_name);
6118355f576SJeff Roberson #endif
612736ee590SJeff Roberson 	if (zone->uz_maxpages &&
613736ee590SJeff Roberson 	    zone->uz_pages + zone->uz_ppera > zone->uz_maxpages)
614736ee590SJeff Roberson 		return (NULL);
6158355f576SJeff Roberson 
6168355f576SJeff Roberson 	ZONE_UNLOCK(zone);
617a553d4b8SJeff Roberson 
618a553d4b8SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_OFFPAGE) {
619a553d4b8SJeff Roberson 		slab = uma_zalloc_internal(slabzone, NULL, wait, NULL);
620a553d4b8SJeff Roberson 		if (slab == NULL) {
621a553d4b8SJeff Roberson 			ZONE_LOCK(zone);
622a553d4b8SJeff Roberson 			return NULL;
623a553d4b8SJeff Roberson 		}
624a553d4b8SJeff Roberson 	}
625a553d4b8SJeff Roberson 
626a553d4b8SJeff Roberson 	if (booted || (zone->uz_flags & UMA_ZFLAG_PRIVALLOC)) {
6278355f576SJeff Roberson 		mtx_lock(&Giant);
628a553d4b8SJeff Roberson 		mem = zone->uz_allocf(zone,
6298355f576SJeff Roberson 		    zone->uz_ppera * UMA_SLAB_SIZE, &flags, wait);
6308355f576SJeff Roberson 		mtx_unlock(&Giant);
631a553d4b8SJeff Roberson 		if (mem == NULL) {
6328355f576SJeff Roberson 			ZONE_LOCK(zone);
6338355f576SJeff Roberson 			return (NULL);
634a553d4b8SJeff Roberson 		}
6358355f576SJeff Roberson 	} else {
636a553d4b8SJeff Roberson 		uma_slab_t tmps;
6378355f576SJeff Roberson 
6388355f576SJeff Roberson 		if (zone->uz_ppera > 1)
6398355f576SJeff Roberson 			panic("UMA: Attemping to allocate multiple pages before vm has started.\n");
6408355f576SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_MALLOC)
6418355f576SJeff Roberson 			panic("Mallocing before uma_startup2 has been called.\n");
6428355f576SJeff Roberson 		if (uma_boot_free == 0)
6438355f576SJeff Roberson 			panic("UMA: Ran out of pre init pages, increase UMA_BOOT_PAGES\n");
644a553d4b8SJeff Roberson 		tmps = LIST_FIRST(&uma_boot_pages);
645a553d4b8SJeff Roberson 		LIST_REMOVE(tmps, us_link);
6468355f576SJeff Roberson 		uma_boot_free--;
647a553d4b8SJeff Roberson 		mem = tmps->us_data;
6488355f576SJeff Roberson 	}
6498355f576SJeff Roberson 
650a553d4b8SJeff Roberson 	ZONE_LOCK(zone);
6518355f576SJeff Roberson 
6528355f576SJeff Roberson 	/* Alloc slab structure for offpage, otherwise adjust it's position */
6538355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_OFFPAGE)) {
6548355f576SJeff Roberson 		slab = (uma_slab_t )(mem + zone->uz_pgoff);
6558355f576SJeff Roberson 	} else  {
6568355f576SJeff Roberson 		if (!(zone->uz_flags & UMA_ZFLAG_MALLOC))
6578355f576SJeff Roberson 			UMA_HASH_INSERT(&zone->uz_hash, slab, mem);
6588355f576SJeff Roberson 	}
6598355f576SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_MALLOC) {
6608355f576SJeff Roberson #ifdef UMA_DEBUG
6618355f576SJeff Roberson 		printf("Inserting %p into malloc hash from slab %p\n",
6628355f576SJeff Roberson 		    mem, slab);
6638355f576SJeff Roberson #endif
664a553d4b8SJeff Roberson 		/* XXX Yikes! No lock on the malloc hash! */
6658355f576SJeff Roberson 		UMA_HASH_INSERT(mallochash, slab, mem);
6668355f576SJeff Roberson 	}
6678355f576SJeff Roberson 
6688355f576SJeff Roberson 	slab->us_zone = zone;
6698355f576SJeff Roberson 	slab->us_data = mem;
6708355f576SJeff Roberson 
6718355f576SJeff Roberson 	/*
6728355f576SJeff Roberson 	 * This is intended to spread data out across cache lines.
6738355f576SJeff Roberson 	 *
6748355f576SJeff Roberson 	 * This code doesn't seem to work properly on x86, and on alpha
6758355f576SJeff Roberson 	 * it makes absolutely no performance difference. I'm sure it could
6768355f576SJeff Roberson 	 * use some tuning, but sun makes outrageous claims about it's
6778355f576SJeff Roberson 	 * performance.
6788355f576SJeff Roberson 	 */
6798355f576SJeff Roberson #if 0
6808355f576SJeff Roberson 	if (zone->uz_cachemax) {
6818355f576SJeff Roberson 		slab->us_data += zone->uz_cacheoff;
6828355f576SJeff Roberson 		zone->uz_cacheoff += UMA_CACHE_INC;
6838355f576SJeff Roberson 		if (zone->uz_cacheoff > zone->uz_cachemax)
6848355f576SJeff Roberson 			zone->uz_cacheoff = 0;
6858355f576SJeff Roberson 	}
6868355f576SJeff Roberson #endif
6878355f576SJeff Roberson 
6888355f576SJeff Roberson 	slab->us_freecount = zone->uz_ipers;
6898355f576SJeff Roberson 	slab->us_firstfree = 0;
6908355f576SJeff Roberson 	slab->us_flags = flags;
6918355f576SJeff Roberson 	for (i = 0; i < zone->uz_ipers; i++)
6928355f576SJeff Roberson 		slab->us_freelist[i] = i+1;
6938355f576SJeff Roberson 
6948355f576SJeff Roberson 	if (zone->uz_init)
6958355f576SJeff Roberson 		for (i = 0; i < zone->uz_ipers; i++)
6968355f576SJeff Roberson 			zone->uz_init(slab->us_data + (zone->uz_rsize * i),
6978355f576SJeff Roberson 			    zone->uz_size);
6988355f576SJeff Roberson 
6998355f576SJeff Roberson 	zone->uz_pages += zone->uz_ppera;
7008355f576SJeff Roberson 	zone->uz_free += zone->uz_ipers;
7018355f576SJeff Roberson 
7028355f576SJeff Roberson 	return (slab);
7038355f576SJeff Roberson }
7048355f576SJeff Roberson 
7058355f576SJeff Roberson /*
7068355f576SJeff Roberson  * Allocates a number of pages from the system
7078355f576SJeff Roberson  *
7088355f576SJeff Roberson  * Arguments:
7098355f576SJeff Roberson  *	zone  Unused
7108355f576SJeff Roberson  *	bytes  The number of bytes requested
7118355f576SJeff Roberson  *	wait  Shall we wait?
7128355f576SJeff Roberson  *
7138355f576SJeff Roberson  * Returns:
7148355f576SJeff Roberson  *	A pointer to the alloced memory or possibly
7158355f576SJeff Roberson  *	NULL if M_NOWAIT is set.
7168355f576SJeff Roberson  */
7178355f576SJeff Roberson static void *
7188355f576SJeff Roberson page_alloc(uma_zone_t zone, int bytes, u_int8_t *pflag, int wait)
7198355f576SJeff Roberson {
7208355f576SJeff Roberson 	void *p;	/* Returned page */
7218355f576SJeff Roberson 
7228355f576SJeff Roberson 	/*
7238355f576SJeff Roberson 	 * XXX The original zone allocator did this, but I don't think it's
724157d7b35SAlfred Perlstein 	 * necessary in current.
7258355f576SJeff Roberson 	 */
7268355f576SJeff Roberson 
7278355f576SJeff Roberson 	if (lockstatus(&kernel_map->lock, NULL)) {
7288355f576SJeff Roberson 		*pflag = UMA_SLAB_KMEM;
7298355f576SJeff Roberson 		p = (void *) kmem_malloc(kmem_map, bytes, wait);
7308355f576SJeff Roberson 	} else {
7318355f576SJeff Roberson 		*pflag = UMA_SLAB_KMAP;
7328355f576SJeff Roberson 		p = (void *) kmem_alloc(kernel_map, bytes);
7338355f576SJeff Roberson 	}
7348355f576SJeff Roberson 
7358355f576SJeff Roberson 	return (p);
7368355f576SJeff Roberson }
7378355f576SJeff Roberson 
7388355f576SJeff Roberson /*
7398355f576SJeff Roberson  * Allocates a number of pages from within an object
7408355f576SJeff Roberson  *
7418355f576SJeff Roberson  * Arguments:
7428355f576SJeff Roberson  *	zone   Unused
7438355f576SJeff Roberson  *	bytes  The number of bytes requested
7448355f576SJeff Roberson  *	wait   Shall we wait?
7458355f576SJeff Roberson  *
7468355f576SJeff Roberson  * Returns:
7478355f576SJeff Roberson  *	A pointer to the alloced memory or possibly
7488355f576SJeff Roberson  *	NULL if M_NOWAIT is set.
7498355f576SJeff Roberson  */
7508355f576SJeff Roberson static void *
7518355f576SJeff Roberson obj_alloc(uma_zone_t zone, int bytes, u_int8_t *flags, int wait)
7528355f576SJeff Roberson {
7538355f576SJeff Roberson 	vm_offset_t zkva;
7548355f576SJeff Roberson 	vm_offset_t retkva;
7558355f576SJeff Roberson 	vm_page_t p;
7568355f576SJeff Roberson 	int pages;
7578355f576SJeff Roberson 
7588355f576SJeff Roberson 	retkva = NULL;
7598355f576SJeff Roberson 	pages = zone->uz_pages;
7608355f576SJeff Roberson 
7618355f576SJeff Roberson 	/*
7628355f576SJeff Roberson 	 * This looks a little weird since we're getting one page at a time
7638355f576SJeff Roberson 	 */
7648355f576SJeff Roberson 	while (bytes > 0) {
7658355f576SJeff Roberson 		p = vm_page_alloc(zone->uz_obj, pages,
7668355f576SJeff Roberson 		    VM_ALLOC_INTERRUPT);
7678355f576SJeff Roberson 		if (p == NULL)
7688355f576SJeff Roberson 			return (NULL);
7698355f576SJeff Roberson 
7708355f576SJeff Roberson 		zkva = zone->uz_kva + pages * PAGE_SIZE;
7718355f576SJeff Roberson 		if (retkva == NULL)
7728355f576SJeff Roberson 			retkva = zkva;
7738355f576SJeff Roberson 		pmap_qenter(zkva, &p, 1);
7748355f576SJeff Roberson 		bytes -= PAGE_SIZE;
7758355f576SJeff Roberson 		pages += 1;
7768355f576SJeff Roberson 	}
7778355f576SJeff Roberson 
7788355f576SJeff Roberson 	*flags = UMA_SLAB_PRIV;
7798355f576SJeff Roberson 
7808355f576SJeff Roberson 	return ((void *)retkva);
7818355f576SJeff Roberson }
7828355f576SJeff Roberson 
7838355f576SJeff Roberson /*
7848355f576SJeff Roberson  * Frees a number of pages to the system
7858355f576SJeff Roberson  *
7868355f576SJeff Roberson  * Arguments:
7878355f576SJeff Roberson  *	mem   A pointer to the memory to be freed
7888355f576SJeff Roberson  *	size  The size of the memory being freed
7898355f576SJeff Roberson  *	flags The original p->us_flags field
7908355f576SJeff Roberson  *
7918355f576SJeff Roberson  * Returns:
7928355f576SJeff Roberson  *	Nothing
7938355f576SJeff Roberson  *
7948355f576SJeff Roberson  */
7958355f576SJeff Roberson static void
7968355f576SJeff Roberson page_free(void *mem, int size, u_int8_t flags)
7978355f576SJeff Roberson {
7988355f576SJeff Roberson 	vm_map_t map;
7998355f576SJeff Roberson 	if (flags & UMA_SLAB_KMEM)
8008355f576SJeff Roberson 		map = kmem_map;
8018355f576SJeff Roberson 	else if (flags & UMA_SLAB_KMAP)
8028355f576SJeff Roberson 		map = kernel_map;
8038355f576SJeff Roberson 	else
8048355f576SJeff Roberson 		panic("UMA: page_free used with invalid flags %d\n", flags);
8058355f576SJeff Roberson 
8068355f576SJeff Roberson 	kmem_free(map, (vm_offset_t)mem, size);
8078355f576SJeff Roberson }
8088355f576SJeff Roberson 
8098355f576SJeff Roberson /*
8108355f576SJeff Roberson  * Zero fill initializer
8118355f576SJeff Roberson  *
8128355f576SJeff Roberson  * Arguments/Returns follow uma_init specifications
8138355f576SJeff Roberson  *
8148355f576SJeff Roberson  */
8158355f576SJeff Roberson static void
8168355f576SJeff Roberson zero_init(void *mem, int size)
8178355f576SJeff Roberson {
8188355f576SJeff Roberson 	bzero(mem, size);
8198355f576SJeff Roberson }
8208355f576SJeff Roberson 
8218355f576SJeff Roberson /*
8228355f576SJeff Roberson  * Finish creating a small uma zone.  This calculates ipers, and the zone size.
8238355f576SJeff Roberson  *
8248355f576SJeff Roberson  * Arguments
8258355f576SJeff Roberson  *	zone  The zone we should initialize
8268355f576SJeff Roberson  *
8278355f576SJeff Roberson  * Returns
8288355f576SJeff Roberson  *	Nothing
8298355f576SJeff Roberson  */
8308355f576SJeff Roberson static void
8318355f576SJeff Roberson zone_small_init(uma_zone_t zone)
8328355f576SJeff Roberson {
8338355f576SJeff Roberson 	int rsize;
8348355f576SJeff Roberson 	int memused;
8358355f576SJeff Roberson 	int ipers;
8368355f576SJeff Roberson 
8378355f576SJeff Roberson 	rsize = zone->uz_size;
8388355f576SJeff Roberson 
8398355f576SJeff Roberson 	if (rsize < UMA_SMALLEST_UNIT)
8408355f576SJeff Roberson 		rsize = UMA_SMALLEST_UNIT;
8418355f576SJeff Roberson 
8428355f576SJeff Roberson 	if (rsize & zone->uz_align)
8438355f576SJeff Roberson 		rsize = (rsize & ~zone->uz_align) + (zone->uz_align + 1);
8448355f576SJeff Roberson 
8458355f576SJeff Roberson 	zone->uz_rsize = rsize;
8468355f576SJeff Roberson 
8478355f576SJeff Roberson 	rsize += 1;	/* Account for the byte of linkage */
8488355f576SJeff Roberson 	zone->uz_ipers = (UMA_SLAB_SIZE - sizeof(struct uma_slab)) / rsize;
8498355f576SJeff Roberson 	zone->uz_ppera = 1;
8508355f576SJeff Roberson 
8518355f576SJeff Roberson 	memused = zone->uz_ipers * zone->uz_rsize;
8528355f576SJeff Roberson 
8538355f576SJeff Roberson 	/* Can we do any better? */
8548355f576SJeff Roberson 	if ((UMA_SLAB_SIZE - memused) >= UMA_MAX_WASTE) {
8558355f576SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_INTERNAL)
8568355f576SJeff Roberson 			return;
8578355f576SJeff Roberson 		ipers = UMA_SLAB_SIZE / zone->uz_rsize;
8588355f576SJeff Roberson 		if (ipers > zone->uz_ipers) {
8598355f576SJeff Roberson 			zone->uz_flags |= UMA_ZFLAG_OFFPAGE;
8608355f576SJeff Roberson 			zone->uz_ipers = ipers;
8618355f576SJeff Roberson 		}
8628355f576SJeff Roberson 	}
8638355f576SJeff Roberson 
8648355f576SJeff Roberson }
8658355f576SJeff Roberson 
8668355f576SJeff Roberson /*
8678355f576SJeff Roberson  * Finish creating a large (> UMA_SLAB_SIZE) uma zone.  Just give in and do
8688355f576SJeff Roberson  * OFFPAGE for now.  When I can allow for more dynamic slab sizes this will be
8698355f576SJeff Roberson  * more complicated.
8708355f576SJeff Roberson  *
8718355f576SJeff Roberson  * Arguments
8728355f576SJeff Roberson  *	zone  The zone we should initialize
8738355f576SJeff Roberson  *
8748355f576SJeff Roberson  * Returns
8758355f576SJeff Roberson  *	Nothing
8768355f576SJeff Roberson  */
8778355f576SJeff Roberson static void
8788355f576SJeff Roberson zone_large_init(uma_zone_t zone)
8798355f576SJeff Roberson {
8808355f576SJeff Roberson 	int pages;
8818355f576SJeff Roberson 
8828355f576SJeff Roberson 	pages = zone->uz_size / UMA_SLAB_SIZE;
8838355f576SJeff Roberson 
8848355f576SJeff Roberson 	/* Account for remainder */
8858355f576SJeff Roberson 	if ((pages * UMA_SLAB_SIZE) < zone->uz_size)
8868355f576SJeff Roberson 		pages++;
8878355f576SJeff Roberson 
8888355f576SJeff Roberson 	zone->uz_ppera = pages;
8898355f576SJeff Roberson 	zone->uz_ipers = 1;
8908355f576SJeff Roberson 
8918355f576SJeff Roberson 	zone->uz_flags |= UMA_ZFLAG_OFFPAGE;
8928355f576SJeff Roberson 	zone->uz_rsize = zone->uz_size;
8938355f576SJeff Roberson }
8948355f576SJeff Roberson 
8958355f576SJeff Roberson /*
8968355f576SJeff Roberson  * Zone header ctor.  This initializes all fields, locks, etc.  And inserts
8978355f576SJeff Roberson  * the zone onto the global zone list.
8988355f576SJeff Roberson  *
8998355f576SJeff Roberson  * Arguments/Returns follow uma_ctor specifications
9008355f576SJeff Roberson  *	udata  Actually uma_zcreat_args
9018355f576SJeff Roberson  *
9028355f576SJeff Roberson  */
9038355f576SJeff Roberson 
9048355f576SJeff Roberson static void
9058355f576SJeff Roberson zone_ctor(void *mem, int size, void *udata)
9068355f576SJeff Roberson {
9078355f576SJeff Roberson 	struct uma_zctor_args *arg = udata;
9088355f576SJeff Roberson 	uma_zone_t zone = mem;
9098355f576SJeff Roberson 	int cplen;
9108355f576SJeff Roberson 	int cpu;
9118355f576SJeff Roberson 
9128355f576SJeff Roberson 	bzero(zone, size);
9138355f576SJeff Roberson 	zone->uz_name = arg->name;
9148355f576SJeff Roberson 	zone->uz_size = arg->size;
9158355f576SJeff Roberson 	zone->uz_ctor = arg->ctor;
9168355f576SJeff Roberson 	zone->uz_dtor = arg->dtor;
9178355f576SJeff Roberson 	zone->uz_init = arg->uminit;
9188355f576SJeff Roberson 	zone->uz_align = arg->align;
9198355f576SJeff Roberson 	zone->uz_free = 0;
9208355f576SJeff Roberson 	zone->uz_pages = 0;
9218355f576SJeff Roberson 	zone->uz_flags = 0;
9228355f576SJeff Roberson 	zone->uz_allocf = page_alloc;
9238355f576SJeff Roberson 	zone->uz_freef = page_free;
9248355f576SJeff Roberson 
9258355f576SJeff Roberson 	if (arg->flags & UMA_ZONE_ZINIT)
9268355f576SJeff Roberson 		zone->uz_init = zero_init;
9278355f576SJeff Roberson 
9288355f576SJeff Roberson 	if (arg->flags & UMA_ZONE_INTERNAL)
9298355f576SJeff Roberson 		zone->uz_flags |= UMA_ZFLAG_INTERNAL;
9308355f576SJeff Roberson 
9318355f576SJeff Roberson 	if (arg->flags & UMA_ZONE_MALLOC)
9328355f576SJeff Roberson 		zone->uz_flags |= UMA_ZFLAG_MALLOC;
9338355f576SJeff Roberson 
9348355f576SJeff Roberson 	if (arg->flags & UMA_ZONE_NOFREE)
9358355f576SJeff Roberson 		zone->uz_flags |= UMA_ZFLAG_NOFREE;
9368355f576SJeff Roberson 
9378355f576SJeff Roberson 	if (zone->uz_size > UMA_SLAB_SIZE)
9388355f576SJeff Roberson 		zone_large_init(zone);
9398355f576SJeff Roberson 	else
9408355f576SJeff Roberson 		zone_small_init(zone);
9418355f576SJeff Roberson 
9428355f576SJeff Roberson 	/* We do this so that the per cpu lock name is unique for each zone */
9438355f576SJeff Roberson 	memcpy(zone->uz_lname, "PCPU ", 5);
9448355f576SJeff Roberson 	cplen = min(strlen(zone->uz_name) + 1, LOCKNAME_LEN - 6);
9458355f576SJeff Roberson 	memcpy(zone->uz_lname+5, zone->uz_name, cplen);
9468355f576SJeff Roberson 	zone->uz_lname[LOCKNAME_LEN - 1] = '\0';
9478355f576SJeff Roberson 
9488355f576SJeff Roberson 	/*
9498355f576SJeff Roberson 	 * If we're putting the slab header in the actual page we need to
9508355f576SJeff Roberson 	 * figure out where in each page it goes.  This calculates a right
9518355f576SJeff Roberson 	 * justified offset into the memory on a ALIGN_PTR boundary.
9528355f576SJeff Roberson 	 */
9538355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_OFFPAGE)) {
9548355f576SJeff Roberson 		int totsize;
9558355f576SJeff Roberson 		int waste;
9568355f576SJeff Roberson 
9578355f576SJeff Roberson 		/* Size of the slab struct and free list */
9588355f576SJeff Roberson 		totsize = sizeof(struct uma_slab) + zone->uz_ipers;
9598355f576SJeff Roberson 		if (totsize & UMA_ALIGN_PTR)
9608355f576SJeff Roberson 			totsize = (totsize & ~UMA_ALIGN_PTR) +
9618355f576SJeff Roberson 			    (UMA_ALIGN_PTR + 1);
9628355f576SJeff Roberson 		zone->uz_pgoff = UMA_SLAB_SIZE - totsize;
9638355f576SJeff Roberson 
9648355f576SJeff Roberson 		waste = zone->uz_pgoff;
9658355f576SJeff Roberson 		waste -= (zone->uz_ipers * zone->uz_rsize);
9668355f576SJeff Roberson 
9678355f576SJeff Roberson 		/*
9688355f576SJeff Roberson 		 * This calculates how much space we have for cache line size
9698355f576SJeff Roberson 		 * optimizations.  It works by offseting each slab slightly.
9708355f576SJeff Roberson 		 * Currently it breaks on x86, and so it is disabled.
9718355f576SJeff Roberson 		 */
9728355f576SJeff Roberson 
9738355f576SJeff Roberson 		if (zone->uz_align < UMA_CACHE_INC && waste > UMA_CACHE_INC) {
9748355f576SJeff Roberson 			zone->uz_cachemax = waste - UMA_CACHE_INC;
9758355f576SJeff Roberson 			zone->uz_cacheoff = 0;
9768355f576SJeff Roberson 		}
9778355f576SJeff Roberson 
9788355f576SJeff Roberson 		totsize = zone->uz_pgoff + sizeof(struct uma_slab)
9798355f576SJeff Roberson 		    + zone->uz_ipers;
9808355f576SJeff Roberson 		/* I don't think it's possible, but I'll make sure anyway */
9818355f576SJeff Roberson 		if (totsize > UMA_SLAB_SIZE) {
9828355f576SJeff Roberson 			printf("zone %s ipers %d rsize %d size %d\n",
9838355f576SJeff Roberson 			    zone->uz_name, zone->uz_ipers, zone->uz_rsize,
9848355f576SJeff Roberson 			    zone->uz_size);
9858355f576SJeff Roberson 			panic("UMA slab won't fit.\n");
9868355f576SJeff Roberson 		}
9878355f576SJeff Roberson 	} else {
9888355f576SJeff Roberson 		/* hash_expand here to allocate the initial hash table */
9898355f576SJeff Roberson 		hash_expand(&zone->uz_hash);
9908355f576SJeff Roberson 		zone->uz_pgoff = 0;
9918355f576SJeff Roberson 	}
9928355f576SJeff Roberson 
9938355f576SJeff Roberson #ifdef UMA_DEBUG
9948355f576SJeff Roberson 	printf("%s(%p) size = %d ipers = %d ppera = %d pgoff = %d\n",
9958355f576SJeff Roberson 	    zone->uz_name, zone,
9968355f576SJeff Roberson 	    zone->uz_size, zone->uz_ipers,
9978355f576SJeff Roberson 	    zone->uz_ppera, zone->uz_pgoff);
9988355f576SJeff Roberson #endif
9998355f576SJeff Roberson 	ZONE_LOCK_INIT(zone);
10008355f576SJeff Roberson 
10018355f576SJeff Roberson 	mtx_lock(&uma_mtx);
10028355f576SJeff Roberson 	LIST_INSERT_HEAD(&uma_zones, zone, uz_link);
10038355f576SJeff Roberson 	mtx_unlock(&uma_mtx);
10048355f576SJeff Roberson 
10058355f576SJeff Roberson 	/*
10068355f576SJeff Roberson 	 * Some internal zones don't have room allocated for the per cpu
10078355f576SJeff Roberson 	 * caches.  If we're internal, bail out here.
10088355f576SJeff Roberson 	 */
10098355f576SJeff Roberson 
10108355f576SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_INTERNAL)
10118355f576SJeff Roberson 		return;
10128355f576SJeff Roberson 
10138355f576SJeff Roberson 	if (zone->uz_ipers < UMA_BUCKET_SIZE)
1014a553d4b8SJeff Roberson 		zone->uz_count = zone->uz_ipers - 1;
10158355f576SJeff Roberson 	else
1016a553d4b8SJeff Roberson 		zone->uz_count = UMA_BUCKET_SIZE - 1;
1017a553d4b8SJeff Roberson 
1018a553d4b8SJeff Roberson 	for (cpu = 0; cpu < maxcpu; cpu++)
10198355f576SJeff Roberson 		CPU_LOCK_INIT(zone, cpu);
10208355f576SJeff Roberson }
10218355f576SJeff Roberson 
10228355f576SJeff Roberson /*
10239c2cd7e5SJeff Roberson  * Zone header dtor.  This frees all data, destroys locks, frees the hash table
10249c2cd7e5SJeff Roberson  * and removes the zone from the global list.
10259c2cd7e5SJeff Roberson  *
10269c2cd7e5SJeff Roberson  * Arguments/Returns follow uma_dtor specifications
10279c2cd7e5SJeff Roberson  *	udata  unused
10289c2cd7e5SJeff Roberson  */
10299c2cd7e5SJeff Roberson 
10309c2cd7e5SJeff Roberson static void
10319c2cd7e5SJeff Roberson zone_dtor(void *arg, int size, void *udata)
10329c2cd7e5SJeff Roberson {
10339c2cd7e5SJeff Roberson 	uma_zone_t zone;
10349c2cd7e5SJeff Roberson 	int cpu;
10359c2cd7e5SJeff Roberson 
10369c2cd7e5SJeff Roberson 	zone = (uma_zone_t)arg;
10379c2cd7e5SJeff Roberson 
10389c2cd7e5SJeff Roberson 	mtx_lock(&uma_mtx);
10399c2cd7e5SJeff Roberson 	LIST_REMOVE(zone, uz_link);
10409c2cd7e5SJeff Roberson 	mtx_unlock(&uma_mtx);
10419c2cd7e5SJeff Roberson 
10429c2cd7e5SJeff Roberson 	ZONE_LOCK(zone);
10439c2cd7e5SJeff Roberson 	zone->uz_wssize = 0;
10449c2cd7e5SJeff Roberson 	ZONE_UNLOCK(zone);
10459c2cd7e5SJeff Roberson 
10469c2cd7e5SJeff Roberson 	zone_drain(zone);
10479c2cd7e5SJeff Roberson 	ZONE_LOCK(zone);
10489c2cd7e5SJeff Roberson 	if (zone->uz_free != 0)
10499c2cd7e5SJeff Roberson 		printf("Zone %s was not empty.  Lost %d pages of memory.\n",
10509c2cd7e5SJeff Roberson 		    zone->uz_name, zone->uz_pages);
10519c2cd7e5SJeff Roberson 
10529c2cd7e5SJeff Roberson 	if ((zone->uz_flags & UMA_ZFLAG_INTERNAL) != 0)
10539c2cd7e5SJeff Roberson 		for (cpu = 0; cpu < maxcpu; cpu++)
10549c2cd7e5SJeff Roberson 			CPU_LOCK_FINI(zone, cpu);
10559c2cd7e5SJeff Roberson 
10569c2cd7e5SJeff Roberson 	if ((zone->uz_flags & UMA_ZFLAG_OFFPAGE) != 0)
10579c2cd7e5SJeff Roberson 		hash_free(&zone->uz_hash);
10589c2cd7e5SJeff Roberson 
10599c2cd7e5SJeff Roberson 	ZONE_UNLOCK(zone);
10609c2cd7e5SJeff Roberson 	ZONE_LOCK_FINI(zone);
10619c2cd7e5SJeff Roberson }
10629c2cd7e5SJeff Roberson /*
10638355f576SJeff Roberson  * Traverses every zone in the system and calls a callback
10648355f576SJeff Roberson  *
10658355f576SJeff Roberson  * Arguments:
10668355f576SJeff Roberson  *	zfunc  A pointer to a function which accepts a zone
10678355f576SJeff Roberson  *		as an argument.
10688355f576SJeff Roberson  *
10698355f576SJeff Roberson  * Returns:
10708355f576SJeff Roberson  *	Nothing
10718355f576SJeff Roberson  */
10728355f576SJeff Roberson static void
10738355f576SJeff Roberson zone_foreach(void (*zfunc)(uma_zone_t))
10748355f576SJeff Roberson {
10758355f576SJeff Roberson 	uma_zone_t zone;
10768355f576SJeff Roberson 
10778355f576SJeff Roberson 	mtx_lock(&uma_mtx);
10788355f576SJeff Roberson 	LIST_FOREACH(zone, &uma_zones, uz_link) {
10798355f576SJeff Roberson 		zfunc(zone);
10808355f576SJeff Roberson 	}
10818355f576SJeff Roberson 	mtx_unlock(&uma_mtx);
10828355f576SJeff Roberson }
10838355f576SJeff Roberson 
10848355f576SJeff Roberson /* Public functions */
10858355f576SJeff Roberson /* See uma.h */
10868355f576SJeff Roberson void
10878355f576SJeff Roberson uma_startup(void *bootmem)
10888355f576SJeff Roberson {
10898355f576SJeff Roberson 	struct uma_zctor_args args;
10908355f576SJeff Roberson 	uma_slab_t slab;
10918355f576SJeff Roberson 	int slabsize;
10928355f576SJeff Roberson 	int i;
10938355f576SJeff Roberson 
10948355f576SJeff Roberson #ifdef UMA_DEBUG
10958355f576SJeff Roberson 	printf("Creating uma zone headers zone.\n");
10968355f576SJeff Roberson #endif
10978355f576SJeff Roberson #ifdef SMP
10988355f576SJeff Roberson 	maxcpu = mp_maxid + 1;
10998355f576SJeff Roberson #else
11008355f576SJeff Roberson 	maxcpu = 1;
11018355f576SJeff Roberson #endif
11028355f576SJeff Roberson #ifdef UMA_DEBUG
11038355f576SJeff Roberson 	printf("Max cpu = %d, mp_maxid = %d\n", maxcpu, mp_maxid);
11048355f576SJeff Roberson 	Debugger("stop");
11058355f576SJeff Roberson #endif
11066008862bSJohn Baldwin 	mtx_init(&uma_mtx, "UMA lock", NULL, MTX_DEF);
11078355f576SJeff Roberson 	/* "manually" Create the initial zone */
11088355f576SJeff Roberson 	args.name = "UMA Zones";
11098355f576SJeff Roberson 	args.size = sizeof(struct uma_zone) +
11108355f576SJeff Roberson 	    (sizeof(struct uma_cache) * (maxcpu - 1));
11118355f576SJeff Roberson 	args.ctor = zone_ctor;
11129c2cd7e5SJeff Roberson 	args.dtor = zone_dtor;
11138355f576SJeff Roberson 	args.uminit = zero_init;
11148355f576SJeff Roberson 	args.fini = NULL;
11158355f576SJeff Roberson 	args.align = 32 - 1;
11168355f576SJeff Roberson 	args.flags = UMA_ZONE_INTERNAL;
11178355f576SJeff Roberson 	/* The initial zone has no Per cpu queues so it's smaller */
11188355f576SJeff Roberson 	zone_ctor(zones, sizeof(struct uma_zone), &args);
11198355f576SJeff Roberson 
11208355f576SJeff Roberson #ifdef UMA_DEBUG
11218355f576SJeff Roberson 	printf("Filling boot free list.\n");
11228355f576SJeff Roberson #endif
11238355f576SJeff Roberson 	for (i = 0; i < UMA_BOOT_PAGES; i++) {
11248355f576SJeff Roberson 		slab = (uma_slab_t)((u_int8_t *)bootmem + (i * UMA_SLAB_SIZE));
11258355f576SJeff Roberson 		slab->us_data = (u_int8_t *)slab;
11268355f576SJeff Roberson 		slab->us_flags = UMA_SLAB_BOOT;
11278355f576SJeff Roberson 		LIST_INSERT_HEAD(&uma_boot_pages, slab, us_link);
11288355f576SJeff Roberson 		uma_boot_free++;
11298355f576SJeff Roberson 	}
11308355f576SJeff Roberson 
11318355f576SJeff Roberson #ifdef UMA_DEBUG
11328355f576SJeff Roberson 	printf("Creating slab zone.\n");
11338355f576SJeff Roberson #endif
11348355f576SJeff Roberson 
11358355f576SJeff Roberson 	/*
11368355f576SJeff Roberson 	 * This is the max number of free list items we'll have with
11378355f576SJeff Roberson 	 * offpage slabs.
11388355f576SJeff Roberson 	 */
11398355f576SJeff Roberson 
11408355f576SJeff Roberson 	slabsize = UMA_SLAB_SIZE - sizeof(struct uma_slab);
11418355f576SJeff Roberson 	slabsize /= UMA_MAX_WASTE;
11428355f576SJeff Roberson 	slabsize++;			/* In case there it's rounded */
11438355f576SJeff Roberson 	slabsize += sizeof(struct uma_slab);
11448355f576SJeff Roberson 
11458355f576SJeff Roberson 	/* Now make a zone for slab headers */
11468355f576SJeff Roberson 	slabzone = uma_zcreate("UMA Slabs",
11478355f576SJeff Roberson 				slabsize,
11488355f576SJeff Roberson 				NULL, NULL, NULL, NULL,
11498355f576SJeff Roberson 				UMA_ALIGN_PTR, UMA_ZONE_INTERNAL);
11508355f576SJeff Roberson 
11518355f576SJeff Roberson 	hashzone = uma_zcreate("UMA Hash",
11528355f576SJeff Roberson 	    sizeof(struct slabhead *) * UMA_HASH_SIZE_INIT,
11538355f576SJeff Roberson 	    NULL, NULL, NULL, NULL,
11548355f576SJeff Roberson 	    UMA_ALIGN_PTR, UMA_ZONE_INTERNAL);
11558355f576SJeff Roberson 
11568355f576SJeff Roberson 	bucketzone = uma_zcreate("UMA Buckets", sizeof(struct uma_bucket),
11578355f576SJeff Roberson 	    NULL, NULL, NULL, NULL,
11588355f576SJeff Roberson 	    UMA_ALIGN_PTR, UMA_ZONE_INTERNAL);
11598355f576SJeff Roberson 
11608355f576SJeff Roberson 
11618355f576SJeff Roberson #ifdef UMA_DEBUG
11628355f576SJeff Roberson 	printf("UMA startup complete.\n");
11638355f576SJeff Roberson #endif
11648355f576SJeff Roberson }
11658355f576SJeff Roberson 
11668355f576SJeff Roberson /* see uma.h */
11678355f576SJeff Roberson void
11688355f576SJeff Roberson uma_startup2(void *hashmem, u_long elems)
11698355f576SJeff Roberson {
11708355f576SJeff Roberson 	bzero(hashmem, elems * sizeof(void *));
11718355f576SJeff Roberson 	mallochash->uh_slab_hash = hashmem;
11728355f576SJeff Roberson 	mallochash->uh_hashsize = elems;
11738355f576SJeff Roberson 	mallochash->uh_hashmask = elems - 1;
11748355f576SJeff Roberson 	booted = 1;
11758355f576SJeff Roberson #ifdef UMA_DEBUG
11768355f576SJeff Roberson 	printf("UMA startup2 complete.\n");
11778355f576SJeff Roberson #endif
11788355f576SJeff Roberson }
11798355f576SJeff Roberson 
11808355f576SJeff Roberson /*
11818355f576SJeff Roberson  * Initialize our callout handle
11828355f576SJeff Roberson  *
11838355f576SJeff Roberson  */
11848355f576SJeff Roberson 
11858355f576SJeff Roberson static void
11868355f576SJeff Roberson uma_startup3(void)
11878355f576SJeff Roberson {
11888355f576SJeff Roberson #ifdef UMA_DEBUG
11898355f576SJeff Roberson 	printf("Starting callout.\n");
11908355f576SJeff Roberson #endif
11918355f576SJeff Roberson 	/* We'll be mpsafe once the vm is locked. */
11928355f576SJeff Roberson 	callout_init(&uma_callout, 0);
11938355f576SJeff Roberson 	callout_reset(&uma_callout, UMA_WORKING_TIME * hz, uma_timeout, NULL);
11948355f576SJeff Roberson #ifdef UMA_DEBUG
11958355f576SJeff Roberson 	printf("UMA startup3 complete.\n");
11968355f576SJeff Roberson #endif
11978355f576SJeff Roberson }
11988355f576SJeff Roberson 
11998355f576SJeff Roberson /* See uma.h */
12008355f576SJeff Roberson uma_zone_t
12018355f576SJeff Roberson uma_zcreate(char *name, int size, uma_ctor ctor, uma_dtor dtor, uma_init uminit,
12028355f576SJeff Roberson 		     uma_fini fini, int align, u_int16_t flags)
12038355f576SJeff Roberson 
12048355f576SJeff Roberson {
12058355f576SJeff Roberson 	struct uma_zctor_args args;
12068355f576SJeff Roberson 
12078355f576SJeff Roberson 	/* This stuff is essential for the zone ctor */
12088355f576SJeff Roberson 	args.name = name;
12098355f576SJeff Roberson 	args.size = size;
12108355f576SJeff Roberson 	args.ctor = ctor;
12118355f576SJeff Roberson 	args.dtor = dtor;
12128355f576SJeff Roberson 	args.uminit = uminit;
12138355f576SJeff Roberson 	args.fini = fini;
12148355f576SJeff Roberson 	args.align = align;
12158355f576SJeff Roberson 	args.flags = flags;
12168355f576SJeff Roberson 
1217a553d4b8SJeff Roberson 	return (uma_zalloc_internal(zones, &args, M_WAITOK, NULL));
12188355f576SJeff Roberson }
12198355f576SJeff Roberson 
12208355f576SJeff Roberson /* See uma.h */
12219c2cd7e5SJeff Roberson void
12229c2cd7e5SJeff Roberson uma_zdestroy(uma_zone_t zone)
12239c2cd7e5SJeff Roberson {
12249c2cd7e5SJeff Roberson 	uma_zfree_internal(zones, zone, NULL, 0);
12259c2cd7e5SJeff Roberson }
12269c2cd7e5SJeff Roberson 
12279c2cd7e5SJeff Roberson /* See uma.h */
12288355f576SJeff Roberson void *
12298355f576SJeff Roberson uma_zalloc_arg(uma_zone_t zone, void *udata, int wait)
12308355f576SJeff Roberson {
12318355f576SJeff Roberson 	void *item;
12328355f576SJeff Roberson 	uma_cache_t cache;
12338355f576SJeff Roberson 	uma_bucket_t bucket;
12348355f576SJeff Roberson 	int cpu;
12358355f576SJeff Roberson 
12368355f576SJeff Roberson 	/* This is the fast path allocation */
12378355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC_1
12388355f576SJeff Roberson 	printf("Allocating one item from %s(%p)\n", zone->uz_name, zone);
12398355f576SJeff Roberson #endif
1240a553d4b8SJeff Roberson 
1241a553d4b8SJeff Roberson zalloc_restart:
12428355f576SJeff Roberson 	cpu = PCPU_GET(cpuid);
12438355f576SJeff Roberson 	CPU_LOCK(zone, cpu);
12448355f576SJeff Roberson 	cache = &zone->uz_cpu[cpu];
12458355f576SJeff Roberson 
12468355f576SJeff Roberson zalloc_start:
12478355f576SJeff Roberson 	bucket = cache->uc_allocbucket;
12488355f576SJeff Roberson 
12498355f576SJeff Roberson 	if (bucket) {
12508355f576SJeff Roberson 		if (bucket->ub_ptr > -1) {
12518355f576SJeff Roberson 			item = bucket->ub_bucket[bucket->ub_ptr];
12528355f576SJeff Roberson #ifdef INVARIANTS
12538355f576SJeff Roberson 			bucket->ub_bucket[bucket->ub_ptr] = NULL;
12548355f576SJeff Roberson #endif
12558355f576SJeff Roberson 			bucket->ub_ptr--;
12568355f576SJeff Roberson 			KASSERT(item != NULL,
12578355f576SJeff Roberson 			    ("uma_zalloc: Bucket pointer mangled."));
12588355f576SJeff Roberson 			cache->uc_allocs++;
12598355f576SJeff Roberson 			CPU_UNLOCK(zone, cpu);
12608355f576SJeff Roberson 			if (zone->uz_ctor)
12618355f576SJeff Roberson 				zone->uz_ctor(item, zone->uz_size, udata);
12628355f576SJeff Roberson 			return (item);
12638355f576SJeff Roberson 		} else if (cache->uc_freebucket) {
12648355f576SJeff Roberson 			/*
12658355f576SJeff Roberson 			 * We have run out of items in our allocbucket.
12668355f576SJeff Roberson 			 * See if we can switch with our free bucket.
12678355f576SJeff Roberson 			 */
12688355f576SJeff Roberson 			if (cache->uc_freebucket->ub_ptr > -1) {
12698355f576SJeff Roberson 				uma_bucket_t swap;
12708355f576SJeff Roberson 
12718355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
12728355f576SJeff Roberson 				printf("uma_zalloc: Swapping empty with alloc.\n");
12738355f576SJeff Roberson #endif
12748355f576SJeff Roberson 				swap = cache->uc_freebucket;
12758355f576SJeff Roberson 				cache->uc_freebucket = cache->uc_allocbucket;
12768355f576SJeff Roberson 				cache->uc_allocbucket = swap;
12778355f576SJeff Roberson 
12788355f576SJeff Roberson 				goto zalloc_start;
12798355f576SJeff Roberson 			}
12808355f576SJeff Roberson 		}
12818355f576SJeff Roberson 	}
1282a553d4b8SJeff Roberson 	ZONE_LOCK(zone);
1283a553d4b8SJeff Roberson 	/* Since we have locked the zone we may as well send back our stats */
1284a553d4b8SJeff Roberson 	zone->uz_allocs += cache->uc_allocs;
1285a553d4b8SJeff Roberson 	cache->uc_allocs = 0;
12868355f576SJeff Roberson 
1287a553d4b8SJeff Roberson 	/* Our old one is now a free bucket */
1288a553d4b8SJeff Roberson 	if (cache->uc_allocbucket) {
1289a553d4b8SJeff Roberson 		KASSERT(cache->uc_allocbucket->ub_ptr == -1,
1290a553d4b8SJeff Roberson 		    ("uma_zalloc_arg: Freeing a non free bucket."));
1291a553d4b8SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_free_bucket,
1292a553d4b8SJeff Roberson 		    cache->uc_allocbucket, ub_link);
1293a553d4b8SJeff Roberson 		cache->uc_allocbucket = NULL;
1294a553d4b8SJeff Roberson 	}
12958355f576SJeff Roberson 
1296a553d4b8SJeff Roberson 	/* Check the free list for a new alloc bucket */
1297a553d4b8SJeff Roberson 	if ((bucket = LIST_FIRST(&zone->uz_full_bucket)) != NULL) {
1298a553d4b8SJeff Roberson 		KASSERT(bucket->ub_ptr != -1,
1299a553d4b8SJeff Roberson 		    ("uma_zalloc_arg: Returning an empty bucket."));
13008355f576SJeff Roberson 
1301a553d4b8SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
1302a553d4b8SJeff Roberson 		cache->uc_allocbucket = bucket;
1303a553d4b8SJeff Roberson 		ZONE_UNLOCK(zone);
13048355f576SJeff Roberson 		goto zalloc_start;
1305a553d4b8SJeff Roberson 	}
1306a553d4b8SJeff Roberson 	/* Bump up our uz_count so we get here less */
1307a553d4b8SJeff Roberson 	if (zone->uz_count < UMA_BUCKET_SIZE - 1)
1308a553d4b8SJeff Roberson 		zone->uz_count++;
1309a553d4b8SJeff Roberson 
1310a553d4b8SJeff Roberson 	/* We are no longer associated with this cpu!!! */
1311a553d4b8SJeff Roberson 	CPU_UNLOCK(zone, cpu);
13128355f576SJeff Roberson 
13138355f576SJeff Roberson 	/*
1314a553d4b8SJeff Roberson 	 * Now lets just fill a bucket and put it on the free list.  If that
1315a553d4b8SJeff Roberson 	 * works we'll restart the allocation from the begining.
1316a553d4b8SJeff Roberson 	 *
1317a553d4b8SJeff Roberson 	 * Try this zone's free list first so we don't allocate extra buckets.
13188355f576SJeff Roberson 	 */
1319a553d4b8SJeff Roberson 
1320a553d4b8SJeff Roberson 	if ((bucket = LIST_FIRST(&zone->uz_free_bucket)) != NULL)
1321a553d4b8SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
1322a553d4b8SJeff Roberson 
1323a553d4b8SJeff Roberson 	/* Now we no longer need the zone lock. */
1324a553d4b8SJeff Roberson 	ZONE_UNLOCK(zone);
1325a553d4b8SJeff Roberson 
1326a553d4b8SJeff Roberson 	if (bucket == NULL)
1327a553d4b8SJeff Roberson 		bucket = uma_zalloc_internal(bucketzone,
1328a553d4b8SJeff Roberson 		    NULL, wait, NULL);
1329a553d4b8SJeff Roberson 
1330a553d4b8SJeff Roberson 	if (bucket != NULL) {
1331a553d4b8SJeff Roberson #ifdef INVARIANTS
1332a553d4b8SJeff Roberson 		bzero(bucket, bucketzone->uz_size);
1333a553d4b8SJeff Roberson #endif
1334a553d4b8SJeff Roberson 		bucket->ub_ptr = -1;
1335a553d4b8SJeff Roberson 
1336a553d4b8SJeff Roberson 		if (uma_zalloc_internal(zone, udata, wait, bucket))
1337a553d4b8SJeff Roberson 			goto zalloc_restart;
1338a553d4b8SJeff Roberson 		else
1339a553d4b8SJeff Roberson 			uma_zfree_internal(bucketzone, bucket, NULL, 0);
1340a553d4b8SJeff Roberson 	}
1341a553d4b8SJeff Roberson 	/*
1342a553d4b8SJeff Roberson 	 * We may not get a bucket if we recurse, so
1343a553d4b8SJeff Roberson 	 * return an actual item.
1344a553d4b8SJeff Roberson 	 */
1345a553d4b8SJeff Roberson #ifdef UMA_DEBUG
1346a553d4b8SJeff Roberson 	printf("uma_zalloc_arg: Bucketzone returned NULL\n");
1347a553d4b8SJeff Roberson #endif
1348a553d4b8SJeff Roberson 
1349a553d4b8SJeff Roberson 	return (uma_zalloc_internal(zone, udata, wait, NULL));
13508355f576SJeff Roberson }
13518355f576SJeff Roberson 
13528355f576SJeff Roberson /*
13538355f576SJeff Roberson  * Allocates an item for an internal zone OR fills a bucket
13548355f576SJeff Roberson  *
13558355f576SJeff Roberson  * Arguments
13568355f576SJeff Roberson  *	zone   The zone to alloc for.
13578355f576SJeff Roberson  *	udata  The data to be passed to the constructor.
13588355f576SJeff Roberson  *	wait   M_WAITOK or M_NOWAIT.
1359a553d4b8SJeff Roberson  *	bucket The bucket to fill or NULL
13608355f576SJeff Roberson  *
13618355f576SJeff Roberson  * Returns
13628355f576SJeff Roberson  *	NULL if there is no memory and M_NOWAIT is set
13638355f576SJeff Roberson  *	An item if called on an interal zone
13648355f576SJeff Roberson  *	Non NULL if called to fill a bucket and it was successful.
13658355f576SJeff Roberson  *
13668355f576SJeff Roberson  * Discussion:
13678355f576SJeff Roberson  *	This was much cleaner before it had to do per cpu caches.  It is
13688355f576SJeff Roberson  *	complicated now because it has to handle the simple internal case, and
1369a553d4b8SJeff Roberson  *	the more involved bucket filling and allocation.
13708355f576SJeff Roberson  */
13718355f576SJeff Roberson 
13728355f576SJeff Roberson static void *
1373a553d4b8SJeff Roberson uma_zalloc_internal(uma_zone_t zone, void *udata, int wait, uma_bucket_t bucket)
13748355f576SJeff Roberson {
13758355f576SJeff Roberson 	uma_slab_t slab;
13768355f576SJeff Roberson 	u_int8_t freei;
13778355f576SJeff Roberson 	void *item;
13788355f576SJeff Roberson 
13798355f576SJeff Roberson 	item = NULL;
13808355f576SJeff Roberson 
13818355f576SJeff Roberson 	/*
1382a553d4b8SJeff Roberson 	 * This is to stop us from allocating per cpu buckets while we're
1383a553d4b8SJeff Roberson 	 * running out of UMA_BOOT_PAGES.  Otherwise, we would exhaust the
1384a553d4b8SJeff Roberson 	 * boot pages.
13858355f576SJeff Roberson 	 */
13868355f576SJeff Roberson 
13878355f576SJeff Roberson 	if (!booted && zone == bucketzone)
13888355f576SJeff Roberson 		return (NULL);
13898355f576SJeff Roberson 
13908355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
13918355f576SJeff Roberson 	printf("INTERNAL: Allocating one item from %s(%p)\n", zone->uz_name, zone);
13928355f576SJeff Roberson #endif
13938355f576SJeff Roberson 	ZONE_LOCK(zone);
13948355f576SJeff Roberson 
13958355f576SJeff Roberson 	/*
1396a553d4b8SJeff Roberson 	 * This code is here to limit the number of simultaneous bucket fills
1397a553d4b8SJeff Roberson 	 * for any given zone to the number of per cpu caches in this zone. This
1398a553d4b8SJeff Roberson 	 * is done so that we don't allocate more memory than we really need.
13998355f576SJeff Roberson 	 */
1400a553d4b8SJeff Roberson 
1401a553d4b8SJeff Roberson 	if (bucket) {
1402a553d4b8SJeff Roberson #ifdef SMP
1403a553d4b8SJeff Roberson 		if (zone->uz_fills >= mp_ncpus)
1404a553d4b8SJeff Roberson #else
1405a553d4b8SJeff Roberson 		if (zone->uz_fills > 1)
14068355f576SJeff Roberson #endif
1407a553d4b8SJeff Roberson 			return (NULL);
1408a553d4b8SJeff Roberson 
1409a553d4b8SJeff Roberson 		zone->uz_fills++;
14108355f576SJeff Roberson 	}
14118355f576SJeff Roberson 
14128355f576SJeff Roberson new_slab:
14138355f576SJeff Roberson 
14148355f576SJeff Roberson 	/* Find a slab with some space */
14158355f576SJeff Roberson 	if (zone->uz_free) {
14168355f576SJeff Roberson 		if (!LIST_EMPTY(&zone->uz_part_slab)) {
14178355f576SJeff Roberson 			slab = LIST_FIRST(&zone->uz_part_slab);
14188355f576SJeff Roberson 		} else {
14198355f576SJeff Roberson 			slab = LIST_FIRST(&zone->uz_free_slab);
14208355f576SJeff Roberson 			LIST_REMOVE(slab, us_link);
14218355f576SJeff Roberson 			LIST_INSERT_HEAD(&zone->uz_part_slab, slab, us_link);
14228355f576SJeff Roberson 		}
14238355f576SJeff Roberson 	} else {
14248355f576SJeff Roberson 		/*
14258355f576SJeff Roberson 		 * This is to prevent us from recursively trying to allocate
14268355f576SJeff Roberson 		 * buckets.  The problem is that if an allocation forces us to
14278355f576SJeff Roberson 		 * grab a new bucket we will call page_alloc, which will go off
14288355f576SJeff Roberson 		 * and cause the vm to allocate vm_map_entries.  If we need new
14298355f576SJeff Roberson 		 * buckets there too we will recurse in kmem_alloc and bad
14308355f576SJeff Roberson 		 * things happen.  So instead we return a NULL bucket, and make
14318355f576SJeff Roberson 		 * the code that allocates buckets smart enough to deal with it			 */
14328355f576SJeff Roberson 		if (zone == bucketzone && zone->uz_recurse != 0) {
14338355f576SJeff Roberson 			ZONE_UNLOCK(zone);
14348355f576SJeff Roberson 			return (NULL);
14358355f576SJeff Roberson 		}
14368355f576SJeff Roberson 		zone->uz_recurse++;
14378355f576SJeff Roberson 		slab = slab_zalloc(zone, wait);
14388355f576SJeff Roberson 		zone->uz_recurse--;
14398355f576SJeff Roberson 		if (slab)  {
14408355f576SJeff Roberson 			LIST_INSERT_HEAD(&zone->uz_part_slab, slab, us_link);
14418355f576SJeff Roberson 		/*
14428355f576SJeff Roberson 		 * We might not have been able to get a page, but another cpu
14438355f576SJeff Roberson 		 * could have while we were unlocked.
14448355f576SJeff Roberson 		 */
14458355f576SJeff Roberson 		} else if (zone->uz_free == 0) {
14468355f576SJeff Roberson 			/* If we're filling a bucket return what we have */
1447a553d4b8SJeff Roberson 			if (bucket != NULL)
1448a553d4b8SJeff Roberson 				zone->uz_fills--;
1449a553d4b8SJeff Roberson 			ZONE_UNLOCK(zone);
1450a553d4b8SJeff Roberson 
1451a553d4b8SJeff Roberson 			if (bucket != NULL && bucket->ub_ptr != -1)
14528355f576SJeff Roberson 				return (bucket);
1453a553d4b8SJeff Roberson 			else
14548355f576SJeff Roberson 				return (NULL);
14558355f576SJeff Roberson 		} else {
14568355f576SJeff Roberson 			/* Another cpu must have succeeded */
14578355f576SJeff Roberson 			if ((slab = LIST_FIRST(&zone->uz_part_slab)) == NULL) {
14588355f576SJeff Roberson 				slab = LIST_FIRST(&zone->uz_free_slab);
14598355f576SJeff Roberson 				LIST_REMOVE(slab, us_link);
14608355f576SJeff Roberson 				LIST_INSERT_HEAD(&zone->uz_part_slab,
14618355f576SJeff Roberson 				    slab, us_link);
14628355f576SJeff Roberson 			}
14638355f576SJeff Roberson 		}
14648355f576SJeff Roberson 	}
1465a553d4b8SJeff Roberson 	/*
1466a553d4b8SJeff Roberson 	 * If this is our first time though put this guy on the list.
1467a553d4b8SJeff Roberson 	 */
1468a553d4b8SJeff Roberson 	if (bucket != NULL && bucket->ub_ptr == -1)
1469a553d4b8SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_full_bucket,
1470a553d4b8SJeff Roberson 		    bucket, ub_link);
1471a553d4b8SJeff Roberson 
14728355f576SJeff Roberson 
14738355f576SJeff Roberson 	while (slab->us_freecount) {
14748355f576SJeff Roberson 		freei = slab->us_firstfree;
14758355f576SJeff Roberson 		slab->us_firstfree = slab->us_freelist[freei];
14768355f576SJeff Roberson #ifdef INVARIANTS
14778355f576SJeff Roberson 		slab->us_freelist[freei] = 255;
14788355f576SJeff Roberson #endif
14798355f576SJeff Roberson 		slab->us_freecount--;
14808355f576SJeff Roberson 		zone->uz_free--;
14818355f576SJeff Roberson 		item = slab->us_data + (zone->uz_rsize * freei);
14828355f576SJeff Roberson 
1483a553d4b8SJeff Roberson 		if (bucket == NULL) {
14848355f576SJeff Roberson 			zone->uz_allocs++;
14858355f576SJeff Roberson 			break;
14868355f576SJeff Roberson 		}
14878355f576SJeff Roberson 		bucket->ub_bucket[++bucket->ub_ptr] = item;
14888355f576SJeff Roberson 
14898355f576SJeff Roberson 		/* Don't overfill the bucket! */
1490a553d4b8SJeff Roberson 		if (bucket->ub_ptr == zone->uz_count)
14918355f576SJeff Roberson 			break;
14928355f576SJeff Roberson 	}
14938355f576SJeff Roberson 
14948355f576SJeff Roberson 	/* Move this slab to the full list */
14958355f576SJeff Roberson 	if (slab->us_freecount == 0) {
14968355f576SJeff Roberson 		LIST_REMOVE(slab, us_link);
14978355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_full_slab, slab, us_link);
14988355f576SJeff Roberson 	}
14998355f576SJeff Roberson 
1500a553d4b8SJeff Roberson 	if (bucket != NULL) {
15018355f576SJeff Roberson 		/* Try to keep the buckets totally full, but don't block */
1502a553d4b8SJeff Roberson 		if (bucket->ub_ptr < zone->uz_count) {
15038355f576SJeff Roberson 			wait = M_NOWAIT;
15048355f576SJeff Roberson 			goto new_slab;
1505a553d4b8SJeff Roberson 		} else
1506a553d4b8SJeff Roberson 			zone->uz_fills--;
15078355f576SJeff Roberson 	}
15088355f576SJeff Roberson 
15098355f576SJeff Roberson 	ZONE_UNLOCK(zone);
15108355f576SJeff Roberson 
15118355f576SJeff Roberson 	/* Only construct at this time if we're not filling a bucket */
1512a553d4b8SJeff Roberson 	if (bucket == NULL && zone->uz_ctor != NULL)
15138355f576SJeff Roberson 		zone->uz_ctor(item, zone->uz_size, udata);
15148355f576SJeff Roberson 
15158355f576SJeff Roberson 	return (item);
15168355f576SJeff Roberson }
15178355f576SJeff Roberson 
15188355f576SJeff Roberson /* See uma.h */
15198355f576SJeff Roberson void
15208355f576SJeff Roberson uma_zfree_arg(uma_zone_t zone, void *item, void *udata)
15218355f576SJeff Roberson {
15228355f576SJeff Roberson 	uma_cache_t cache;
15238355f576SJeff Roberson 	uma_bucket_t bucket;
15248355f576SJeff Roberson 	int cpu;
15258355f576SJeff Roberson 
15268355f576SJeff Roberson 	/* This is the fast path free */
15278355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC_1
15288355f576SJeff Roberson 	printf("Freeing item %p to %s(%p)\n", item, zone->uz_name, zone);
15298355f576SJeff Roberson #endif
1530a553d4b8SJeff Roberson zfree_restart:
15318355f576SJeff Roberson 	cpu = PCPU_GET(cpuid);
15328355f576SJeff Roberson 	CPU_LOCK(zone, cpu);
15338355f576SJeff Roberson 	cache = &zone->uz_cpu[cpu];
15348355f576SJeff Roberson 
15358355f576SJeff Roberson zfree_start:
15368355f576SJeff Roberson 	bucket = cache->uc_freebucket;
15378355f576SJeff Roberson 
15388355f576SJeff Roberson 	if (bucket) {
1539a553d4b8SJeff Roberson 		/*
1540a553d4b8SJeff Roberson 		 * Do we have room in our bucket? It is OK for this uz count
1541a553d4b8SJeff Roberson 		 * check to be slightly out of sync.
1542a553d4b8SJeff Roberson 		 */
1543a553d4b8SJeff Roberson 
1544a553d4b8SJeff Roberson 		if (bucket->ub_ptr < zone->uz_count) {
15458355f576SJeff Roberson 			bucket->ub_ptr++;
15468355f576SJeff Roberson 			KASSERT(bucket->ub_bucket[bucket->ub_ptr] == NULL,
15478355f576SJeff Roberson 			    ("uma_zfree: Freeing to non free bucket index."));
15488355f576SJeff Roberson 			bucket->ub_bucket[bucket->ub_ptr] = item;
15498355f576SJeff Roberson 			CPU_UNLOCK(zone, cpu);
15508355f576SJeff Roberson 			if (zone->uz_dtor)
15518355f576SJeff Roberson 				zone->uz_dtor(item, zone->uz_size, udata);
15528355f576SJeff Roberson 			return;
15538355f576SJeff Roberson 		} else if (cache->uc_allocbucket) {
15548355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
15558355f576SJeff Roberson 			printf("uma_zfree: Swapping buckets.\n");
15568355f576SJeff Roberson #endif
15578355f576SJeff Roberson 			/*
15588355f576SJeff Roberson 			 * We have run out of space in our freebucket.
15598355f576SJeff Roberson 			 * See if we can switch with our alloc bucket.
15608355f576SJeff Roberson 			 */
15618355f576SJeff Roberson 			if (cache->uc_allocbucket->ub_ptr <
15628355f576SJeff Roberson 			    cache->uc_freebucket->ub_ptr) {
15638355f576SJeff Roberson 				uma_bucket_t swap;
15648355f576SJeff Roberson 
15658355f576SJeff Roberson 				swap = cache->uc_freebucket;
15668355f576SJeff Roberson 				cache->uc_freebucket = cache->uc_allocbucket;
15678355f576SJeff Roberson 				cache->uc_allocbucket = swap;
15688355f576SJeff Roberson 
15698355f576SJeff Roberson 				goto zfree_start;
15708355f576SJeff Roberson 			}
15718355f576SJeff Roberson 		}
15728355f576SJeff Roberson 	}
15738355f576SJeff Roberson 
15748355f576SJeff Roberson 	/*
1575a553d4b8SJeff Roberson 	 * We can get here for two reasons:
15768355f576SJeff Roberson 	 *
15778355f576SJeff Roberson 	 * 1) The buckets are NULL
1578a553d4b8SJeff Roberson 	 * 2) The alloc and free buckets are both somewhat full.
15798355f576SJeff Roberson 	 *
15808355f576SJeff Roberson 	 */
15818355f576SJeff Roberson 
15828355f576SJeff Roberson 	ZONE_LOCK(zone);
15838355f576SJeff Roberson 
15848355f576SJeff Roberson 	bucket = cache->uc_freebucket;
15858355f576SJeff Roberson 	cache->uc_freebucket = NULL;
15868355f576SJeff Roberson 
15878355f576SJeff Roberson 	/* Can we throw this on the zone full list? */
15888355f576SJeff Roberson 	if (bucket != NULL) {
15898355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
15908355f576SJeff Roberson 		printf("uma_zfree: Putting old bucket on the free list.\n");
15918355f576SJeff Roberson #endif
15928355f576SJeff Roberson 		/* ub_ptr is pointing to the last free item */
15938355f576SJeff Roberson 		KASSERT(bucket->ub_ptr != -1,
15948355f576SJeff Roberson 		    ("uma_zfree: Attempting to insert an empty bucket onto the full list.\n"));
15958355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_full_bucket,
15968355f576SJeff Roberson 		    bucket, ub_link);
15978355f576SJeff Roberson 	}
1598a553d4b8SJeff Roberson 	if ((bucket = LIST_FIRST(&zone->uz_free_bucket)) != NULL) {
1599a553d4b8SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
1600a553d4b8SJeff Roberson 		ZONE_UNLOCK(zone);
1601a553d4b8SJeff Roberson 		cache->uc_freebucket = bucket;
1602a553d4b8SJeff Roberson 		goto zfree_start;
1603a553d4b8SJeff Roberson 	}
1604a553d4b8SJeff Roberson 	/* We're done with this CPU now */
1605a553d4b8SJeff Roberson 	CPU_UNLOCK(zone, cpu);
1606a553d4b8SJeff Roberson 
1607a553d4b8SJeff Roberson 	/* And the zone.. */
1608a553d4b8SJeff Roberson 	ZONE_UNLOCK(zone);
1609a553d4b8SJeff Roberson 
16108355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
16118355f576SJeff Roberson 	printf("uma_zfree: Allocating new free bucket.\n");
16128355f576SJeff Roberson #endif
16138355f576SJeff Roberson 	bucket = uma_zalloc_internal(bucketzone,
1614a553d4b8SJeff Roberson 	    NULL, M_NOWAIT, NULL);
16158355f576SJeff Roberson 	if (bucket) {
16168355f576SJeff Roberson #ifdef INVARIANTS
16178355f576SJeff Roberson 		bzero(bucket, bucketzone->uz_size);
16188355f576SJeff Roberson #endif
16198355f576SJeff Roberson 		bucket->ub_ptr = -1;
1620a553d4b8SJeff Roberson 		ZONE_LOCK(zone);
1621a553d4b8SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_free_bucket,
1622a553d4b8SJeff Roberson 		    bucket, ub_link);
16238355f576SJeff Roberson 		ZONE_UNLOCK(zone);
1624a553d4b8SJeff Roberson 		goto zfree_restart;
16258355f576SJeff Roberson 	}
16268355f576SJeff Roberson 
1627a553d4b8SJeff Roberson 	/*
1628a553d4b8SJeff Roberson 	 * If nothing else caught this, we'll just do an internal free.
1629a553d4b8SJeff Roberson 	 */
16308355f576SJeff Roberson 
16318355f576SJeff Roberson 	uma_zfree_internal(zone, item, udata, 0);
16328355f576SJeff Roberson 
16338355f576SJeff Roberson 	return;
16348355f576SJeff Roberson 
16358355f576SJeff Roberson }
16368355f576SJeff Roberson 
16378355f576SJeff Roberson /*
16388355f576SJeff Roberson  * Frees an item to an INTERNAL zone or allocates a free bucket
16398355f576SJeff Roberson  *
16408355f576SJeff Roberson  * Arguments:
16418355f576SJeff Roberson  *	zone   The zone to free to
16428355f576SJeff Roberson  *	item   The item we're freeing
16438355f576SJeff Roberson  *	udata  User supplied data for the dtor
16448355f576SJeff Roberson  *	skip   Skip the dtor, it was done in uma_zfree_arg
16458355f576SJeff Roberson  */
16468355f576SJeff Roberson 
16478355f576SJeff Roberson static void
16488355f576SJeff Roberson uma_zfree_internal(uma_zone_t zone, void *item, void *udata, int skip)
16498355f576SJeff Roberson {
16508355f576SJeff Roberson 	uma_slab_t slab;
16518355f576SJeff Roberson 	u_int8_t *mem;
16528355f576SJeff Roberson 	u_int8_t freei;
16538355f576SJeff Roberson 
16548355f576SJeff Roberson 	ZONE_LOCK(zone);
16558355f576SJeff Roberson 
16568355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_MALLOC)) {
16578355f576SJeff Roberson 		mem = (u_int8_t *)((unsigned long)item & (~UMA_SLAB_MASK));
16588355f576SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_OFFPAGE)
16598355f576SJeff Roberson 			slab = hash_sfind(&zone->uz_hash, mem);
16608355f576SJeff Roberson 		else {
16618355f576SJeff Roberson 			mem += zone->uz_pgoff;
16628355f576SJeff Roberson 			slab = (uma_slab_t)mem;
16638355f576SJeff Roberson 		}
16648355f576SJeff Roberson 	} else {
16658355f576SJeff Roberson 		slab = (uma_slab_t)udata;
16668355f576SJeff Roberson 	}
16678355f576SJeff Roberson 
16688355f576SJeff Roberson 	/* Do we need to remove from any lists? */
16698355f576SJeff Roberson 	if (slab->us_freecount+1 == zone->uz_ipers) {
16708355f576SJeff Roberson 		LIST_REMOVE(slab, us_link);
16718355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_free_slab, slab, us_link);
16728355f576SJeff Roberson 	} else if (slab->us_freecount == 0) {
16738355f576SJeff Roberson 		LIST_REMOVE(slab, us_link);
16748355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_part_slab, slab, us_link);
16758355f576SJeff Roberson 	}
16768355f576SJeff Roberson 
16778355f576SJeff Roberson 	/* Slab management stuff */
16788355f576SJeff Roberson 	freei = ((unsigned long)item - (unsigned long)slab->us_data)
16798355f576SJeff Roberson 		/ zone->uz_rsize;
16808355f576SJeff Roberson #ifdef INVARIANTS
16818355f576SJeff Roberson 	if (((freei * zone->uz_rsize) + slab->us_data) != item)
16828355f576SJeff Roberson 		panic("zone: %s(%p) slab %p freed address %p unaligned.\n",
16838355f576SJeff Roberson 		    zone->uz_name, zone, slab, item);
16848355f576SJeff Roberson 	if (freei >= zone->uz_ipers)
16858355f576SJeff Roberson 		panic("zone: %s(%p) slab %p freelist %i out of range 0-%d\n",
16868355f576SJeff Roberson 		    zone->uz_name, zone, slab, freei, zone->uz_ipers-1);
16878355f576SJeff Roberson 
16888355f576SJeff Roberson 	if (slab->us_freelist[freei] != 255) {
16898355f576SJeff Roberson 		printf("Slab at %p, freei %d = %d.\n",
16908355f576SJeff Roberson 		    slab, freei, slab->us_freelist[freei]);
16918355f576SJeff Roberson 		panic("Duplicate free of item %p from zone %p(%s)\n",
16928355f576SJeff Roberson 		    item, zone, zone->uz_name);
16938355f576SJeff Roberson 	}
16948355f576SJeff Roberson #endif
16958355f576SJeff Roberson 	slab->us_freelist[freei] = slab->us_firstfree;
16968355f576SJeff Roberson 	slab->us_firstfree = freei;
16978355f576SJeff Roberson 	slab->us_freecount++;
16988355f576SJeff Roberson 
16998355f576SJeff Roberson 	/* Zone statistics */
17008355f576SJeff Roberson 	zone->uz_free++;
17018355f576SJeff Roberson 
17028355f576SJeff Roberson 	ZONE_UNLOCK(zone);
17038355f576SJeff Roberson 
17048355f576SJeff Roberson 	if (!skip && zone->uz_dtor)
17058355f576SJeff Roberson 		zone->uz_dtor(item, zone->uz_size, udata);
17068355f576SJeff Roberson }
17078355f576SJeff Roberson 
17088355f576SJeff Roberson /* See uma.h */
17098355f576SJeff Roberson void
1710736ee590SJeff Roberson uma_zone_set_max(uma_zone_t zone, int nitems)
1711736ee590SJeff Roberson {
1712736ee590SJeff Roberson 	ZONE_LOCK(zone);
1713736ee590SJeff Roberson 	if (zone->uz_ppera > 1)
1714736ee590SJeff Roberson 		zone->uz_maxpages = nitems / zone->uz_ppera;
1715736ee590SJeff Roberson 	else
1716736ee590SJeff Roberson 		zone->uz_maxpages = nitems / zone->uz_ipers;
1717736ee590SJeff Roberson 	ZONE_UNLOCK(zone);
1718736ee590SJeff Roberson }
1719736ee590SJeff Roberson 
1720736ee590SJeff Roberson /* See uma.h */
1721736ee590SJeff Roberson void
17228355f576SJeff Roberson uma_zone_set_freef(uma_zone_t zone, uma_free freef)
17238355f576SJeff Roberson {
17248355f576SJeff Roberson 	ZONE_LOCK(zone);
17258355f576SJeff Roberson 
17268355f576SJeff Roberson 	zone->uz_freef = freef;
17278355f576SJeff Roberson 
17288355f576SJeff Roberson 	ZONE_UNLOCK(zone);
17298355f576SJeff Roberson }
17308355f576SJeff Roberson 
17318355f576SJeff Roberson /* See uma.h */
17328355f576SJeff Roberson void
17338355f576SJeff Roberson uma_zone_set_allocf(uma_zone_t zone, uma_alloc allocf)
17348355f576SJeff Roberson {
17358355f576SJeff Roberson 	ZONE_LOCK(zone);
17368355f576SJeff Roberson 
17378355f576SJeff Roberson 	zone->uz_flags |= UMA_ZFLAG_PRIVALLOC;
17388355f576SJeff Roberson 	zone->uz_allocf = allocf;
17398355f576SJeff Roberson 
17408355f576SJeff Roberson 	ZONE_UNLOCK(zone);
17418355f576SJeff Roberson }
17428355f576SJeff Roberson 
17438355f576SJeff Roberson /* See uma.h */
17448355f576SJeff Roberson int
17458355f576SJeff Roberson uma_zone_set_obj(uma_zone_t zone, struct vm_object *obj, int count)
17468355f576SJeff Roberson {
17478355f576SJeff Roberson 	int pages;
17488355f576SJeff Roberson 	vm_offset_t kva;
17498355f576SJeff Roberson 
17508355f576SJeff Roberson 	mtx_lock(&Giant);
17518355f576SJeff Roberson 
17528355f576SJeff Roberson 	pages = count / zone->uz_ipers;
17538355f576SJeff Roberson 
17548355f576SJeff Roberson 	if (pages * zone->uz_ipers < count)
17558355f576SJeff Roberson 		pages++;
1756a553d4b8SJeff Roberson 
17578355f576SJeff Roberson 	kva = kmem_alloc_pageable(kernel_map, pages * UMA_SLAB_SIZE);
17588355f576SJeff Roberson 
1759a553d4b8SJeff Roberson 	if (kva == 0) {
1760a553d4b8SJeff Roberson 		mtx_unlock(&Giant);
17618355f576SJeff Roberson 		return (0);
17628355f576SJeff Roberson 	}
17638355f576SJeff Roberson 
17648355f576SJeff Roberson 
1765a553d4b8SJeff Roberson 	if (obj == NULL)
1766a553d4b8SJeff Roberson 		obj = vm_object_allocate(OBJT_DEFAULT,
17678355f576SJeff Roberson 		    zone->uz_maxpages);
17688355f576SJeff Roberson 	else
17698355f576SJeff Roberson 		_vm_object_allocate(OBJT_DEFAULT,
1770a553d4b8SJeff Roberson 		    zone->uz_maxpages, obj);
1771a553d4b8SJeff Roberson 
1772a553d4b8SJeff Roberson 	ZONE_LOCK(zone);
1773a553d4b8SJeff Roberson 	zone->uz_kva = kva;
1774a553d4b8SJeff Roberson 	zone->uz_obj = obj;
1775a553d4b8SJeff Roberson 	zone->uz_maxpages = pages;
17768355f576SJeff Roberson 
17778355f576SJeff Roberson 	zone->uz_allocf = obj_alloc;
17788355f576SJeff Roberson 	zone->uz_flags |= UMA_ZFLAG_NOFREE | UMA_ZFLAG_PRIVALLOC;
17798355f576SJeff Roberson 
17808355f576SJeff Roberson 	ZONE_UNLOCK(zone);
1781a553d4b8SJeff Roberson 	mtx_unlock(&Giant);
17828355f576SJeff Roberson 
17838355f576SJeff Roberson 	return (1);
17848355f576SJeff Roberson }
17858355f576SJeff Roberson 
17868355f576SJeff Roberson /* See uma.h */
17878355f576SJeff Roberson void
17888355f576SJeff Roberson uma_prealloc(uma_zone_t zone, int items)
17898355f576SJeff Roberson {
17908355f576SJeff Roberson 	int slabs;
17918355f576SJeff Roberson 	uma_slab_t slab;
17928355f576SJeff Roberson 
17938355f576SJeff Roberson 	ZONE_LOCK(zone);
17948355f576SJeff Roberson 	slabs = items / zone->uz_ipers;
17958355f576SJeff Roberson 	if (slabs * zone->uz_ipers < items)
17968355f576SJeff Roberson 		slabs++;
17978355f576SJeff Roberson 
17988355f576SJeff Roberson 	while (slabs > 0) {
17998355f576SJeff Roberson 		slab = slab_zalloc(zone, M_WAITOK);
18008355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_free_slab, slab, us_link);
18018355f576SJeff Roberson 		slabs--;
18028355f576SJeff Roberson 	}
18038355f576SJeff Roberson 	ZONE_UNLOCK(zone);
18048355f576SJeff Roberson }
18058355f576SJeff Roberson 
18068355f576SJeff Roberson /* See uma.h */
18078355f576SJeff Roberson void
18088355f576SJeff Roberson uma_reclaim(void)
18098355f576SJeff Roberson {
18108355f576SJeff Roberson 	/*
18118355f576SJeff Roberson 	 * You might think that the delay below would improve performance since
18128355f576SJeff Roberson 	 * the allocator will give away memory that it may ask for immediately.
18138355f576SJeff Roberson 	 * Really, it makes things worse, since cpu cycles are so much cheaper
18148355f576SJeff Roberson 	 * than disk activity.
18158355f576SJeff Roberson 	 */
18168355f576SJeff Roberson #if 0
18178355f576SJeff Roberson 	static struct timeval tv = {0};
18188355f576SJeff Roberson 	struct timeval now;
18198355f576SJeff Roberson 	getmicrouptime(&now);
18208355f576SJeff Roberson 	if (now.tv_sec > tv.tv_sec + 30)
18218355f576SJeff Roberson 		tv = now;
18228355f576SJeff Roberson 	else
18238355f576SJeff Roberson 		return;
18248355f576SJeff Roberson #endif
18258355f576SJeff Roberson #ifdef UMA_DEBUG
18268355f576SJeff Roberson 	printf("UMA: vm asked us to release pages!\n");
18278355f576SJeff Roberson #endif
18288355f576SJeff Roberson 	zone_foreach(zone_drain);
18298355f576SJeff Roberson 
18308355f576SJeff Roberson 	/*
18318355f576SJeff Roberson 	 * Some slabs may have been freed but this zone will be visited early
18328355f576SJeff Roberson 	 * we visit again so that we can free pages that are empty once other
18338355f576SJeff Roberson 	 * zones are drained.  We have to do the same for buckets.
18348355f576SJeff Roberson 	 */
18358355f576SJeff Roberson 	zone_drain(slabzone);
18368355f576SJeff Roberson 	zone_drain(bucketzone);
18378355f576SJeff Roberson }
18388355f576SJeff Roberson 
18398355f576SJeff Roberson void *
18408355f576SJeff Roberson uma_large_malloc(int size, int wait)
18418355f576SJeff Roberson {
18428355f576SJeff Roberson 	void *mem;
18438355f576SJeff Roberson 	uma_slab_t slab;
18448355f576SJeff Roberson 	u_int8_t flags;
18458355f576SJeff Roberson 
1846a553d4b8SJeff Roberson 	slab = uma_zalloc_internal(slabzone, NULL, wait, NULL);
18478355f576SJeff Roberson 	if (slab == NULL)
18488355f576SJeff Roberson 		return (NULL);
18498355f576SJeff Roberson 
18508355f576SJeff Roberson 	mem = page_alloc(NULL, size, &flags, wait);
18518355f576SJeff Roberson 	if (mem) {
18528355f576SJeff Roberson 		slab->us_data = mem;
18538355f576SJeff Roberson 		slab->us_flags = flags | UMA_SLAB_MALLOC;
18548355f576SJeff Roberson 		slab->us_size = size;
18558355f576SJeff Roberson 		UMA_HASH_INSERT(mallochash, slab, mem);
18568355f576SJeff Roberson 	} else {
18578355f576SJeff Roberson 		uma_zfree_internal(slabzone, slab, NULL, 0);
18588355f576SJeff Roberson 	}
18598355f576SJeff Roberson 
18608355f576SJeff Roberson 
18618355f576SJeff Roberson 	return (mem);
18628355f576SJeff Roberson }
18638355f576SJeff Roberson 
18648355f576SJeff Roberson void
18658355f576SJeff Roberson uma_large_free(uma_slab_t slab)
18668355f576SJeff Roberson {
18678355f576SJeff Roberson 	UMA_HASH_REMOVE(mallochash, slab, slab->us_data);
18688355f576SJeff Roberson 	page_free(slab->us_data, slab->us_size, slab->us_flags);
18698355f576SJeff Roberson 	uma_zfree_internal(slabzone, slab, NULL, 0);
18708355f576SJeff Roberson }
18718355f576SJeff Roberson 
18728355f576SJeff Roberson void
18738355f576SJeff Roberson uma_print_stats(void)
18748355f576SJeff Roberson {
18758355f576SJeff Roberson 	zone_foreach(uma_print_zone);
18768355f576SJeff Roberson }
18778355f576SJeff Roberson 
18788355f576SJeff Roberson void
18798355f576SJeff Roberson uma_print_zone(uma_zone_t zone)
18808355f576SJeff Roberson {
18818355f576SJeff Roberson 	printf("%s(%p) size %d(%d) flags %d ipers %d ppera %d out %d free %d\n",
18828355f576SJeff Roberson 	    zone->uz_name, zone, zone->uz_size, zone->uz_rsize, zone->uz_flags,
18838355f576SJeff Roberson 	    zone->uz_ipers, zone->uz_ppera,
18848355f576SJeff Roberson 	    (zone->uz_ipers * zone->uz_pages) - zone->uz_free, zone->uz_free);
18858355f576SJeff Roberson }
18868355f576SJeff Roberson 
18878355f576SJeff Roberson /*
18888355f576SJeff Roberson  * Sysctl handler for vm.zone
18898355f576SJeff Roberson  *
18908355f576SJeff Roberson  * stolen from vm_zone.c
18918355f576SJeff Roberson  */
18928355f576SJeff Roberson static int
18938355f576SJeff Roberson sysctl_vm_zone(SYSCTL_HANDLER_ARGS)
18948355f576SJeff Roberson {
18958355f576SJeff Roberson 	int error, len, cnt;
18968355f576SJeff Roberson 	const int linesize = 128;	/* conservative */
18978355f576SJeff Roberson 	int totalfree;
18988355f576SJeff Roberson 	char *tmpbuf, *offset;
18998355f576SJeff Roberson 	uma_zone_t z;
19008355f576SJeff Roberson 	char *p;
19018355f576SJeff Roberson 
19028355f576SJeff Roberson 	cnt = 0;
19038355f576SJeff Roberson 	LIST_FOREACH(z, &uma_zones, uz_link)
19048355f576SJeff Roberson 		cnt++;
19058355f576SJeff Roberson 	MALLOC(tmpbuf, char *, (cnt == 0 ? 1 : cnt) * linesize,
19068355f576SJeff Roberson 			M_TEMP, M_WAITOK);
19078355f576SJeff Roberson 	len = snprintf(tmpbuf, linesize,
19088355f576SJeff Roberson 	    "\nITEM            SIZE     LIMIT     USED    FREE  REQUESTS\n\n");
19098355f576SJeff Roberson 	if (cnt == 0)
19108355f576SJeff Roberson 		tmpbuf[len - 1] = '\0';
19118355f576SJeff Roberson 	error = SYSCTL_OUT(req, tmpbuf, cnt == 0 ? len-1 : len);
19128355f576SJeff Roberson 	if (error || cnt == 0)
19138355f576SJeff Roberson 		goto out;
19148355f576SJeff Roberson 	offset = tmpbuf;
1915f4af24d5SJeff Roberson 	mtx_lock(&uma_mtx);
19168355f576SJeff Roberson 	LIST_FOREACH(z, &uma_zones, uz_link) {
19178355f576SJeff Roberson 		if (cnt == 0)	/* list may have changed size */
19188355f576SJeff Roberson 			break;
19198355f576SJeff Roberson 		ZONE_LOCK(z);
19208355f576SJeff Roberson 		totalfree = z->uz_free + z->uz_cachefree;
19218355f576SJeff Roberson 		len = snprintf(offset, linesize,
19228355f576SJeff Roberson 		    "%-12.12s  %6.6u, %8.8u, %6.6u, %6.6u, %8.8llu\n",
19238355f576SJeff Roberson 		    z->uz_name, z->uz_size,
19248355f576SJeff Roberson 		    z->uz_maxpages * z->uz_ipers,
19258355f576SJeff Roberson 		    (z->uz_ipers * (z->uz_pages / z->uz_ppera)) - totalfree,
19268355f576SJeff Roberson 		    totalfree,
19278355f576SJeff Roberson 		    (unsigned long long)z->uz_allocs);
19288355f576SJeff Roberson 		ZONE_UNLOCK(z);
19298355f576SJeff Roberson 		for (p = offset + 12; p > offset && *p == ' '; --p)
19308355f576SJeff Roberson 			/* nothing */ ;
19318355f576SJeff Roberson 		p[1] = ':';
19328355f576SJeff Roberson 		cnt--;
19338355f576SJeff Roberson 		offset += len;
19348355f576SJeff Roberson 	}
1935f4af24d5SJeff Roberson 	mtx_unlock(&uma_mtx);
19368355f576SJeff Roberson 	*offset++ = '\0';
19378355f576SJeff Roberson 	error = SYSCTL_OUT(req, tmpbuf, offset - tmpbuf);
19388355f576SJeff Roberson out:
19398355f576SJeff Roberson 	FREE(tmpbuf, M_TEMP);
19408355f576SJeff Roberson 	return (error);
19418355f576SJeff Roberson }
1942