xref: /freebsd/sys/vm/uma_core.c (revision 4c1cc01cd834034963247faeea45682458da9924)
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  *	- Investigate cache size adjustments
498355f576SJeff Roberson  */
508355f576SJeff Roberson 
518355f576SJeff Roberson /* I should really use ktr.. */
528355f576SJeff Roberson /*
538355f576SJeff Roberson #define UMA_DEBUG 1
548355f576SJeff Roberson #define UMA_DEBUG_ALLOC 1
558355f576SJeff Roberson #define UMA_DEBUG_ALLOC_1 1
568355f576SJeff Roberson */
578355f576SJeff Roberson 
588355f576SJeff Roberson 
598355f576SJeff Roberson #include "opt_param.h"
608355f576SJeff Roberson #include <sys/param.h>
618355f576SJeff Roberson #include <sys/systm.h>
628355f576SJeff Roberson #include <sys/kernel.h>
638355f576SJeff Roberson #include <sys/types.h>
648355f576SJeff Roberson #include <sys/queue.h>
658355f576SJeff Roberson #include <sys/malloc.h>
668355f576SJeff Roberson #include <sys/lock.h>
678355f576SJeff Roberson #include <sys/sysctl.h>
688355f576SJeff Roberson #include <sys/mutex.h>
694c1cc01cSJohn Baldwin #include <sys/proc.h>
708355f576SJeff Roberson #include <sys/smp.h>
7186bbae32SJeff Roberson #include <sys/vmmeter.h>
7286bbae32SJeff Roberson 
7386bbae32SJeff Roberson #include <machine/types.h>
748355f576SJeff Roberson 
758355f576SJeff Roberson #include <vm/vm.h>
768355f576SJeff Roberson #include <vm/vm_object.h>
778355f576SJeff Roberson #include <vm/vm_page.h>
788355f576SJeff Roberson #include <vm/vm_param.h>
798355f576SJeff Roberson #include <vm/vm_map.h>
808355f576SJeff Roberson #include <vm/vm_kern.h>
818355f576SJeff Roberson #include <vm/vm_extern.h>
828355f576SJeff Roberson #include <vm/uma.h>
838355f576SJeff Roberson #include <vm/uma_int.h>
84639c9550SJeff Roberson #include <vm/uma_dbg.h>
858355f576SJeff Roberson 
868355f576SJeff Roberson /*
878355f576SJeff Roberson  * This is the zone from which all zones are spawned.  The idea is that even
888355f576SJeff Roberson  * the zone heads are allocated from the allocator, so we use the bss section
898355f576SJeff Roberson  * to bootstrap us.
908355f576SJeff Roberson  */
9186bbae32SJeff Roberson static struct uma_zone masterzone;
9286bbae32SJeff Roberson static uma_zone_t zones = &masterzone;
938355f576SJeff Roberson 
948355f576SJeff Roberson /* This is the zone from which all of uma_slab_t's are allocated. */
958355f576SJeff Roberson static uma_zone_t slabzone;
968355f576SJeff Roberson 
978355f576SJeff Roberson /*
988355f576SJeff Roberson  * The initial hash tables come out of this zone so they can be allocated
998355f576SJeff Roberson  * prior to malloc coming up.
1008355f576SJeff Roberson  */
1018355f576SJeff Roberson static uma_zone_t hashzone;
1028355f576SJeff Roberson 
1038355f576SJeff Roberson /*
1048355f576SJeff Roberson  * Zone that buckets come from.
1058355f576SJeff Roberson  */
1068355f576SJeff Roberson static uma_zone_t bucketzone;
1078355f576SJeff Roberson 
10886bbae32SJeff Roberson /*
10986bbae32SJeff Roberson  * Are we allowed to allocate buckets?
11086bbae32SJeff Roberson  */
11186bbae32SJeff Roberson static int bucketdisable = 1;
11286bbae32SJeff Roberson 
1138355f576SJeff Roberson /* Linked list of all zones in the system */
1148355f576SJeff Roberson static LIST_HEAD(,uma_zone) uma_zones = LIST_HEAD_INITIALIZER(&uma_zones);
1158355f576SJeff Roberson 
1168355f576SJeff Roberson /* This mutex protects the zone list */
1178355f576SJeff Roberson static struct mtx uma_mtx;
1188355f576SJeff Roberson 
1198355f576SJeff Roberson /* Linked list of boot time pages */
1208355f576SJeff Roberson static LIST_HEAD(,uma_slab) uma_boot_pages =
1218355f576SJeff Roberson     LIST_HEAD_INITIALIZER(&uma_boot_pages);
1228355f576SJeff Roberson 
1238355f576SJeff Roberson /* Count of free boottime pages */
1248355f576SJeff Roberson static int uma_boot_free = 0;
1258355f576SJeff Roberson 
1268355f576SJeff Roberson /* Is the VM done starting up? */
1278355f576SJeff Roberson static int booted = 0;
1288355f576SJeff Roberson 
1298355f576SJeff Roberson /* This is the handle used to schedule our working set calculator */
1308355f576SJeff Roberson static struct callout uma_callout;
1318355f576SJeff Roberson 
1328355f576SJeff Roberson /* This is mp_maxid + 1, for use while looping over each cpu */
1338355f576SJeff Roberson static int maxcpu;
1348355f576SJeff Roberson 
1358355f576SJeff Roberson /*
1368355f576SJeff Roberson  * This structure is passed as the zone ctor arg so that I don't have to create
1378355f576SJeff Roberson  * a special allocation function just for zones.
1388355f576SJeff Roberson  */
1398355f576SJeff Roberson struct uma_zctor_args {
1408355f576SJeff Roberson 	char *name;
141c3bdc05fSAndrew R. Reiter 	size_t size;
1428355f576SJeff Roberson 	uma_ctor ctor;
1438355f576SJeff Roberson 	uma_dtor dtor;
1448355f576SJeff Roberson 	uma_init uminit;
1458355f576SJeff Roberson 	uma_fini fini;
1468355f576SJeff Roberson 	int align;
1478355f576SJeff Roberson 	u_int16_t flags;
1488355f576SJeff Roberson };
1498355f576SJeff Roberson 
1508355f576SJeff Roberson /*
1518355f576SJeff Roberson  * This is the malloc hash table which is used to find the zone that a
1528355f576SJeff Roberson  * malloc allocation came from.  It is not currently resizeable.  The
1538355f576SJeff Roberson  * memory for the actual hash bucket is allocated in kmeminit.
1548355f576SJeff Roberson  */
1558355f576SJeff Roberson struct uma_hash mhash;
1568355f576SJeff Roberson struct uma_hash *mallochash = &mhash;
1578355f576SJeff Roberson 
1588355f576SJeff Roberson /* Prototypes.. */
1598355f576SJeff Roberson 
1608355f576SJeff Roberson static void *obj_alloc(uma_zone_t, int, u_int8_t *, int);
1618355f576SJeff Roberson static void *page_alloc(uma_zone_t, int, u_int8_t *, int);
1628355f576SJeff Roberson static void page_free(void *, int, u_int8_t);
1638355f576SJeff Roberson static uma_slab_t slab_zalloc(uma_zone_t, int);
1648355f576SJeff Roberson static void cache_drain(uma_zone_t);
1658355f576SJeff Roberson static void bucket_drain(uma_zone_t, uma_bucket_t);
1668355f576SJeff Roberson static void zone_drain(uma_zone_t);
1678355f576SJeff Roberson static void zone_ctor(void *, int, void *);
1689c2cd7e5SJeff Roberson static void zone_dtor(void *, int, void *);
1698355f576SJeff Roberson static void zero_init(void *, int);
1708355f576SJeff Roberson static void zone_small_init(uma_zone_t zone);
1718355f576SJeff Roberson static void zone_large_init(uma_zone_t zone);
1728355f576SJeff Roberson static void zone_foreach(void (*zfunc)(uma_zone_t));
1738355f576SJeff Roberson static void zone_timeout(uma_zone_t zone);
1740aef6126SJeff Roberson static int hash_alloc(struct uma_hash *);
1750aef6126SJeff Roberson static int hash_expand(struct uma_hash *, struct uma_hash *);
1760aef6126SJeff Roberson static void hash_free(struct uma_hash *hash);
1778355f576SJeff Roberson static void uma_timeout(void *);
1788355f576SJeff Roberson static void uma_startup3(void);
179a553d4b8SJeff Roberson static void *uma_zalloc_internal(uma_zone_t, void *, int, uma_bucket_t);
18086bbae32SJeff Roberson static void uma_zfree_internal(uma_zone_t, void *, void *, int);
18186bbae32SJeff Roberson static void bucket_enable(void);
1828355f576SJeff Roberson void uma_print_zone(uma_zone_t);
1838355f576SJeff Roberson void uma_print_stats(void);
1848355f576SJeff Roberson static int sysctl_vm_zone(SYSCTL_HANDLER_ARGS);
1858355f576SJeff Roberson 
1868355f576SJeff Roberson SYSCTL_OID(_vm, OID_AUTO, zone, CTLTYPE_STRING|CTLFLAG_RD,
1878355f576SJeff Roberson     NULL, 0, sysctl_vm_zone, "A", "Zone Info");
1888355f576SJeff Roberson SYSINIT(uma_startup3, SI_SUB_VM_CONF, SI_ORDER_SECOND, uma_startup3, NULL);
1898355f576SJeff Roberson 
19086bbae32SJeff Roberson /*
19186bbae32SJeff Roberson  * This routine checks to see whether or not it's safe to enable buckets.
19286bbae32SJeff Roberson  */
19386bbae32SJeff Roberson 
19486bbae32SJeff Roberson static void
19586bbae32SJeff Roberson bucket_enable(void)
19686bbae32SJeff Roberson {
19786bbae32SJeff Roberson 	if (cnt.v_free_count < cnt.v_free_min)
19886bbae32SJeff Roberson 		bucketdisable = 1;
19986bbae32SJeff Roberson 	else
20086bbae32SJeff Roberson 		bucketdisable = 0;
20186bbae32SJeff Roberson }
20286bbae32SJeff Roberson 
2038355f576SJeff Roberson 
2048355f576SJeff Roberson /*
2058355f576SJeff Roberson  * Routine called by timeout which is used to fire off some time interval
2068355f576SJeff Roberson  * based calculations.  (working set, stats, etc.)
2078355f576SJeff Roberson  *
2088355f576SJeff Roberson  * Arguments:
2098355f576SJeff Roberson  *	arg   Unused
2108355f576SJeff Roberson  *
2118355f576SJeff Roberson  * Returns:
2128355f576SJeff Roberson  *	Nothing
2138355f576SJeff Roberson  */
2148355f576SJeff Roberson static void
2158355f576SJeff Roberson uma_timeout(void *unused)
2168355f576SJeff Roberson {
21786bbae32SJeff Roberson 	bucket_enable();
2188355f576SJeff Roberson 	zone_foreach(zone_timeout);
2198355f576SJeff Roberson 
2208355f576SJeff Roberson 	/* Reschedule this event */
2218355f576SJeff Roberson 	callout_reset(&uma_callout, UMA_WORKING_TIME * hz, uma_timeout, NULL);
2228355f576SJeff Roberson }
2238355f576SJeff Roberson 
2248355f576SJeff Roberson /*
2258355f576SJeff Roberson  * Routine to perform timeout driven calculations.  This does the working set
2268355f576SJeff Roberson  * as well as hash expanding, and per cpu statistics aggregation.
2278355f576SJeff Roberson  *
2288355f576SJeff Roberson  *  Arguments:
2298355f576SJeff Roberson  *	zone  The zone to operate on
2308355f576SJeff Roberson  *
2318355f576SJeff Roberson  *  Returns:
2328355f576SJeff Roberson  *	Nothing
2338355f576SJeff Roberson  */
2348355f576SJeff Roberson static void
2358355f576SJeff Roberson zone_timeout(uma_zone_t zone)
2368355f576SJeff Roberson {
2378355f576SJeff Roberson 	uma_cache_t cache;
2388355f576SJeff Roberson 	u_int64_t alloc;
2398355f576SJeff Roberson 	int free;
2408355f576SJeff Roberson 	int cpu;
2418355f576SJeff Roberson 
2428355f576SJeff Roberson 	alloc = 0;
2438355f576SJeff Roberson 	free = 0;
2448355f576SJeff Roberson 
2458355f576SJeff Roberson 	/*
2468355f576SJeff Roberson 	 * Aggregate per cpu cache statistics back to the zone.
2478355f576SJeff Roberson 	 *
2488355f576SJeff Roberson 	 * I may rewrite this to set a flag in the per cpu cache instead of
2498355f576SJeff Roberson 	 * locking.  If the flag is not cleared on the next round I will have
2508355f576SJeff Roberson 	 * to lock and do it here instead so that the statistics don't get too
2518355f576SJeff Roberson 	 * far out of sync.
2528355f576SJeff Roberson 	 */
2538355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_INTERNAL)) {
2548355f576SJeff Roberson 		for (cpu = 0; cpu < maxcpu; cpu++) {
2558355f576SJeff Roberson 			if (CPU_ABSENT(cpu))
2568355f576SJeff Roberson 				continue;
2578355f576SJeff Roberson 			CPU_LOCK(zone, cpu);
2588355f576SJeff Roberson 			cache = &zone->uz_cpu[cpu];
2598355f576SJeff Roberson 			/* Add them up, and reset */
2608355f576SJeff Roberson 			alloc += cache->uc_allocs;
2618355f576SJeff Roberson 			cache->uc_allocs = 0;
2628355f576SJeff Roberson 			if (cache->uc_allocbucket)
2638355f576SJeff Roberson 				free += cache->uc_allocbucket->ub_ptr + 1;
2648355f576SJeff Roberson 			if (cache->uc_freebucket)
2658355f576SJeff Roberson 				free += cache->uc_freebucket->ub_ptr + 1;
2668355f576SJeff Roberson 			CPU_UNLOCK(zone, cpu);
2678355f576SJeff Roberson 		}
2688355f576SJeff Roberson 	}
2698355f576SJeff Roberson 
2708355f576SJeff Roberson 	/* Now push these stats back into the zone.. */
2718355f576SJeff Roberson 	ZONE_LOCK(zone);
2728355f576SJeff Roberson 	zone->uz_allocs += alloc;
2738355f576SJeff Roberson 
2748355f576SJeff Roberson 	/*
2758355f576SJeff Roberson 	 * cachefree is an instantanious snapshot of what is in the per cpu
2768355f576SJeff Roberson 	 * caches, not an accurate counter
2778355f576SJeff Roberson 	 */
2788355f576SJeff Roberson 	zone->uz_cachefree = free;
2798355f576SJeff Roberson 
2808355f576SJeff Roberson 	/*
2818355f576SJeff Roberson 	 * Expand the zone hash table.
2828355f576SJeff Roberson 	 *
2838355f576SJeff Roberson 	 * This is done if the number of slabs is larger than the hash size.
2848355f576SJeff Roberson 	 * What I'm trying to do here is completely reduce collisions.  This
2858355f576SJeff Roberson 	 * may be a little aggressive.  Should I allow for two collisions max?
2868355f576SJeff Roberson 	 */
2878355f576SJeff Roberson 
2888355f576SJeff Roberson 	if ((zone->uz_flags & UMA_ZFLAG_OFFPAGE) &&
2898355f576SJeff Roberson 	    !(zone->uz_flags & UMA_ZFLAG_MALLOC)) {
2908355f576SJeff Roberson 		if (zone->uz_pages / zone->uz_ppera
2915300d9ddSJeff Roberson 		    >= zone->uz_hash.uh_hashsize) {
2920aef6126SJeff Roberson 			struct uma_hash newhash;
2930aef6126SJeff Roberson 			struct uma_hash oldhash;
2940aef6126SJeff Roberson 			int ret;
2955300d9ddSJeff Roberson 
2960aef6126SJeff Roberson 			/*
2970aef6126SJeff Roberson 			 * This is so involved because allocating and freeing
2980aef6126SJeff Roberson 			 * while the zone lock is held will lead to deadlock.
2990aef6126SJeff Roberson 			 * I have to do everything in stages and check for
3000aef6126SJeff Roberson 			 * races.
3010aef6126SJeff Roberson 			 */
3020aef6126SJeff Roberson 			newhash = zone->uz_hash;
3035300d9ddSJeff Roberson 			ZONE_UNLOCK(zone);
3040aef6126SJeff Roberson 			ret = hash_alloc(&newhash);
3055300d9ddSJeff Roberson 			ZONE_LOCK(zone);
3060aef6126SJeff Roberson 			if (ret) {
3070aef6126SJeff Roberson 				if (hash_expand(&zone->uz_hash, &newhash)) {
3080aef6126SJeff Roberson 					oldhash = zone->uz_hash;
3090aef6126SJeff Roberson 					zone->uz_hash = newhash;
3100aef6126SJeff Roberson 				} else
3110aef6126SJeff Roberson 					oldhash = newhash;
3120aef6126SJeff Roberson 
3130aef6126SJeff Roberson 				ZONE_UNLOCK(zone);
3140aef6126SJeff Roberson 				hash_free(&oldhash);
3150aef6126SJeff Roberson 				ZONE_LOCK(zone);
3160aef6126SJeff Roberson 			}
3175300d9ddSJeff Roberson 		}
3188355f576SJeff Roberson 	}
3198355f576SJeff Roberson 
3208355f576SJeff Roberson 	/*
3218355f576SJeff Roberson 	 * Here we compute the working set size as the total number of items
3228355f576SJeff Roberson 	 * left outstanding since the last time interval.  This is slightly
3238355f576SJeff Roberson 	 * suboptimal. What we really want is the highest number of outstanding
3248355f576SJeff Roberson 	 * items during the last time quantum.  This should be close enough.
3258355f576SJeff Roberson 	 *
3268355f576SJeff Roberson 	 * The working set size is used to throttle the zone_drain function.
3278355f576SJeff Roberson 	 * We don't want to return memory that we may need again immediately.
3288355f576SJeff Roberson 	 */
3298355f576SJeff Roberson 	alloc = zone->uz_allocs - zone->uz_oallocs;
3308355f576SJeff Roberson 	zone->uz_oallocs = zone->uz_allocs;
3318355f576SJeff Roberson 	zone->uz_wssize = alloc;
3328355f576SJeff Roberson 
3338355f576SJeff Roberson 	ZONE_UNLOCK(zone);
3348355f576SJeff Roberson }
3358355f576SJeff Roberson 
3368355f576SJeff Roberson /*
3375300d9ddSJeff Roberson  * Allocate and zero fill the next sized hash table from the appropriate
3385300d9ddSJeff Roberson  * backing store.
3395300d9ddSJeff Roberson  *
3405300d9ddSJeff Roberson  * Arguments:
3410aef6126SJeff Roberson  *	hash  A new hash structure with the old hash size in uh_hashsize
3425300d9ddSJeff Roberson  *
3435300d9ddSJeff Roberson  * Returns:
3440aef6126SJeff Roberson  *	1 on sucess and 0 on failure.
3455300d9ddSJeff Roberson  */
3460aef6126SJeff Roberson int
3470aef6126SJeff Roberson hash_alloc(struct uma_hash *hash)
3485300d9ddSJeff Roberson {
3490aef6126SJeff Roberson 	int oldsize;
3505300d9ddSJeff Roberson 	int alloc;
3515300d9ddSJeff Roberson 
3520aef6126SJeff Roberson 	oldsize = hash->uh_hashsize;
3530aef6126SJeff Roberson 
3545300d9ddSJeff Roberson 	/* We're just going to go to a power of two greater */
3550aef6126SJeff Roberson 	if (oldsize)  {
3560aef6126SJeff Roberson 		hash->uh_hashsize = oldsize * 2;
3570aef6126SJeff Roberson 		alloc = sizeof(hash->uh_slab_hash[0]) * hash->uh_hashsize;
3585300d9ddSJeff Roberson 		/* XXX Shouldn't be abusing DEVBUF here */
3590aef6126SJeff Roberson 		hash->uh_slab_hash = (struct slabhead *)malloc(alloc,
3600aef6126SJeff Roberson 		    M_DEVBUF, M_NOWAIT);
3615300d9ddSJeff Roberson 	} else {
3620aef6126SJeff Roberson 		alloc = sizeof(hash->uh_slab_hash[0]) * UMA_HASH_SIZE_INIT;
3630aef6126SJeff Roberson 		hash->uh_slab_hash = uma_zalloc_internal(hashzone, NULL,
3640aef6126SJeff Roberson 		    M_WAITOK, NULL);
3650aef6126SJeff Roberson 		hash->uh_hashsize = UMA_HASH_SIZE_INIT;
3665300d9ddSJeff Roberson 	}
3670aef6126SJeff Roberson 	if (hash->uh_slab_hash) {
3680aef6126SJeff Roberson 		bzero(hash->uh_slab_hash, alloc);
3690aef6126SJeff Roberson 		hash->uh_hashmask = hash->uh_hashsize - 1;
3700aef6126SJeff Roberson 		return (1);
3710aef6126SJeff Roberson 	}
3725300d9ddSJeff Roberson 
3730aef6126SJeff Roberson 	return (0);
3745300d9ddSJeff Roberson }
3755300d9ddSJeff Roberson 
3765300d9ddSJeff Roberson /*
3778355f576SJeff Roberson  * Expands the hash table for OFFPAGE zones.  This is done from zone_timeout
3788355f576SJeff Roberson  * to reduce collisions.  This must not be done in the regular allocation path,
3798355f576SJeff Roberson  * otherwise, we can recurse on the vm while allocating pages.
3808355f576SJeff Roberson  *
3818355f576SJeff Roberson  * Arguments:
3820aef6126SJeff Roberson  *	oldhash  The hash you want to expand
3830aef6126SJeff Roberson  *	newhash  The hash structure for the new table
3848355f576SJeff Roberson  *
3858355f576SJeff Roberson  * Returns:
3868355f576SJeff Roberson  * 	Nothing
3878355f576SJeff Roberson  *
3888355f576SJeff Roberson  * Discussion:
3898355f576SJeff Roberson  */
3900aef6126SJeff Roberson static int
3910aef6126SJeff Roberson hash_expand(struct uma_hash *oldhash, struct uma_hash *newhash)
3928355f576SJeff Roberson {
3938355f576SJeff Roberson 	uma_slab_t slab;
3948355f576SJeff Roberson 	int hval;
3958355f576SJeff Roberson 	int i;
3968355f576SJeff Roberson 
3970aef6126SJeff Roberson 	if (!newhash->uh_slab_hash)
3980aef6126SJeff Roberson 		return (0);
3998355f576SJeff Roberson 
4000aef6126SJeff Roberson 	if (oldhash->uh_hashsize >= newhash->uh_hashsize)
4010aef6126SJeff Roberson 		return (0);
4028355f576SJeff Roberson 
4038355f576SJeff Roberson 	/*
4048355f576SJeff Roberson 	 * I need to investigate hash algorithms for resizing without a
4058355f576SJeff Roberson 	 * full rehash.
4068355f576SJeff Roberson 	 */
4078355f576SJeff Roberson 
4080aef6126SJeff Roberson 	for (i = 0; i < oldhash->uh_hashsize; i++)
4090aef6126SJeff Roberson 		while (!SLIST_EMPTY(&oldhash->uh_slab_hash[i])) {
4100aef6126SJeff Roberson 			slab = SLIST_FIRST(&oldhash->uh_slab_hash[i]);
4110aef6126SJeff Roberson 			SLIST_REMOVE_HEAD(&oldhash->uh_slab_hash[i], us_hlink);
4120aef6126SJeff Roberson 			hval = UMA_HASH(newhash, slab->us_data);
4130aef6126SJeff Roberson 			SLIST_INSERT_HEAD(&newhash->uh_slab_hash[hval],
4140aef6126SJeff Roberson 			    slab, us_hlink);
4158355f576SJeff Roberson 		}
4168355f576SJeff Roberson 
4170aef6126SJeff Roberson 	return (1);
4189c2cd7e5SJeff Roberson }
4199c2cd7e5SJeff Roberson 
4205300d9ddSJeff Roberson /*
4215300d9ddSJeff Roberson  * Free the hash bucket to the appropriate backing store.
4225300d9ddSJeff Roberson  *
4235300d9ddSJeff Roberson  * Arguments:
4245300d9ddSJeff Roberson  *	slab_hash  The hash bucket we're freeing
4255300d9ddSJeff Roberson  *	hashsize   The number of entries in that hash bucket
4265300d9ddSJeff Roberson  *
4275300d9ddSJeff Roberson  * Returns:
4285300d9ddSJeff Roberson  *	Nothing
4295300d9ddSJeff Roberson  */
4309c2cd7e5SJeff Roberson static void
4310aef6126SJeff Roberson hash_free(struct uma_hash *hash)
4329c2cd7e5SJeff Roberson {
4330aef6126SJeff Roberson 	if (hash->uh_slab_hash == NULL)
4340aef6126SJeff Roberson 		return;
4350aef6126SJeff Roberson 	if (hash->uh_hashsize == UMA_HASH_SIZE_INIT)
4368355f576SJeff Roberson 		uma_zfree_internal(hashzone,
4370aef6126SJeff Roberson 		    hash->uh_slab_hash, NULL, 0);
4388355f576SJeff Roberson 	else
4390aef6126SJeff Roberson 		free(hash->uh_slab_hash, M_DEVBUF);
4408355f576SJeff Roberson }
4418355f576SJeff Roberson 
4428355f576SJeff Roberson /*
4438355f576SJeff Roberson  * Frees all outstanding items in a bucket
4448355f576SJeff Roberson  *
4458355f576SJeff Roberson  * Arguments:
4468355f576SJeff Roberson  *	zone   The zone to free to, must be unlocked.
4478355f576SJeff Roberson  *	bucket The free/alloc bucket with items, cpu queue must be locked.
4488355f576SJeff Roberson  *
4498355f576SJeff Roberson  * Returns:
4508355f576SJeff Roberson  *	Nothing
4518355f576SJeff Roberson  */
4528355f576SJeff Roberson 
4538355f576SJeff Roberson static void
4548355f576SJeff Roberson bucket_drain(uma_zone_t zone, uma_bucket_t bucket)
4558355f576SJeff Roberson {
4568355f576SJeff Roberson 	uma_slab_t slab;
4578355f576SJeff Roberson 	int mzone;
4588355f576SJeff Roberson 	void *item;
4598355f576SJeff Roberson 
4608355f576SJeff Roberson 	if (bucket == NULL)
4618355f576SJeff Roberson 		return;
4628355f576SJeff Roberson 
4638355f576SJeff Roberson 	slab = NULL;
4648355f576SJeff Roberson 	mzone = 0;
4658355f576SJeff Roberson 
4668355f576SJeff Roberson 	/* We have to lookup the slab again for malloc.. */
4678355f576SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_MALLOC)
4688355f576SJeff Roberson 		mzone = 1;
4698355f576SJeff Roberson 
4708355f576SJeff Roberson 	while (bucket->ub_ptr > -1)  {
4718355f576SJeff Roberson 		item = bucket->ub_bucket[bucket->ub_ptr];
4728355f576SJeff Roberson #ifdef INVARIANTS
4738355f576SJeff Roberson 		bucket->ub_bucket[bucket->ub_ptr] = NULL;
4748355f576SJeff Roberson 		KASSERT(item != NULL,
4758355f576SJeff Roberson 		    ("bucket_drain: botched ptr, item is NULL"));
4768355f576SJeff Roberson #endif
4778355f576SJeff Roberson 		bucket->ub_ptr--;
4788355f576SJeff Roberson 		/*
4798355f576SJeff Roberson 		 * This is extremely inefficient.  The slab pointer was passed
4808355f576SJeff Roberson 		 * to uma_zfree_arg, but we lost it because the buckets don't
4818355f576SJeff Roberson 		 * hold them.  This will go away when free() gets a size passed
4828355f576SJeff Roberson 		 * to it.
4838355f576SJeff Roberson 		 */
4845a34a9f0SJeff Roberson 		if (mzone) {
4855a34a9f0SJeff Roberson 			mtx_lock(&malloc_mtx);
4868355f576SJeff Roberson 			slab = hash_sfind(mallochash,
4878355f576SJeff Roberson 			    (u_int8_t *)((unsigned long)item &
4888355f576SJeff Roberson 			   (~UMA_SLAB_MASK)));
4895a34a9f0SJeff Roberson 			mtx_unlock(&malloc_mtx);
4905a34a9f0SJeff Roberson 		}
4918355f576SJeff Roberson 		uma_zfree_internal(zone, item, slab, 1);
4928355f576SJeff Roberson 	}
4938355f576SJeff Roberson }
4948355f576SJeff Roberson 
4958355f576SJeff Roberson /*
4968355f576SJeff Roberson  * Drains the per cpu caches for a zone.
4978355f576SJeff Roberson  *
4988355f576SJeff Roberson  * Arguments:
4998355f576SJeff Roberson  *	zone  The zone to drain, must be unlocked.
5008355f576SJeff Roberson  *
5018355f576SJeff Roberson  * Returns:
5028355f576SJeff Roberson  *	Nothing
5038355f576SJeff Roberson  *
5048355f576SJeff Roberson  * This function returns with the zone locked so that the per cpu queues can
5058355f576SJeff Roberson  * not be filled until zone_drain is finished.
5068355f576SJeff Roberson  *
5078355f576SJeff Roberson  */
5088355f576SJeff Roberson static void
5098355f576SJeff Roberson cache_drain(uma_zone_t zone)
5108355f576SJeff Roberson {
5118355f576SJeff Roberson 	uma_bucket_t bucket;
5128355f576SJeff Roberson 	uma_cache_t cache;
5138355f576SJeff Roberson 	int cpu;
5148355f576SJeff Roberson 
5158355f576SJeff Roberson 	/*
5168355f576SJeff Roberson 	 * Flush out the per cpu queues.
5178355f576SJeff Roberson 	 *
518157d7b35SAlfred Perlstein 	 * XXX This causes unnecessary thrashing due to immediately having
5198355f576SJeff Roberson 	 * empty per cpu queues.  I need to improve this.
5208355f576SJeff Roberson 	 */
5218355f576SJeff Roberson 
5228355f576SJeff Roberson 	/*
5238355f576SJeff Roberson 	 * We have to lock each cpu cache before locking the zone
5248355f576SJeff Roberson 	 */
5258355f576SJeff Roberson 	ZONE_UNLOCK(zone);
5268355f576SJeff Roberson 
5278355f576SJeff Roberson 	for (cpu = 0; cpu < maxcpu; cpu++) {
5288355f576SJeff Roberson 		if (CPU_ABSENT(cpu))
5298355f576SJeff Roberson 			continue;
5308355f576SJeff Roberson 		CPU_LOCK(zone, cpu);
5318355f576SJeff Roberson 		cache = &zone->uz_cpu[cpu];
5328355f576SJeff Roberson 		bucket_drain(zone, cache->uc_allocbucket);
5338355f576SJeff Roberson 		bucket_drain(zone, cache->uc_freebucket);
5348355f576SJeff Roberson 	}
5358355f576SJeff Roberson 
5368355f576SJeff Roberson 	/*
5378355f576SJeff Roberson 	 * Drain the bucket queues and free the buckets, we just keep two per
5388355f576SJeff Roberson 	 * cpu (alloc/free).
5398355f576SJeff Roberson 	 */
5408355f576SJeff Roberson 	ZONE_LOCK(zone);
5418355f576SJeff Roberson 	while ((bucket = LIST_FIRST(&zone->uz_full_bucket)) != NULL) {
5428355f576SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
5438355f576SJeff Roberson 		ZONE_UNLOCK(zone);
5448355f576SJeff Roberson 		bucket_drain(zone, bucket);
5458355f576SJeff Roberson 		uma_zfree_internal(bucketzone, bucket, NULL, 0);
5468355f576SJeff Roberson 		ZONE_LOCK(zone);
5478355f576SJeff Roberson 	}
5488355f576SJeff Roberson 
5498355f576SJeff Roberson 	/* Now we do the free queue.. */
5508355f576SJeff Roberson 	while ((bucket = LIST_FIRST(&zone->uz_free_bucket)) != NULL) {
5518355f576SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
5528355f576SJeff Roberson 		uma_zfree_internal(bucketzone, bucket, NULL, 0);
5538355f576SJeff Roberson 	}
5548355f576SJeff Roberson 
5558355f576SJeff Roberson 	/* We unlock here, but they will all block until the zone is unlocked */
5568355f576SJeff Roberson 	for (cpu = 0; cpu < maxcpu; cpu++) {
5578355f576SJeff Roberson 		if (CPU_ABSENT(cpu))
5588355f576SJeff Roberson 			continue;
5598355f576SJeff Roberson 		CPU_UNLOCK(zone, cpu);
5608355f576SJeff Roberson 	}
561f4af24d5SJeff Roberson 
562f4af24d5SJeff Roberson 	zone->uz_cachefree = 0;
5638355f576SJeff Roberson }
5648355f576SJeff Roberson 
5658355f576SJeff Roberson /*
5668355f576SJeff Roberson  * Frees pages from a zone back to the system.  This is done on demand from
5678355f576SJeff Roberson  * the pageout daemon.
5688355f576SJeff Roberson  *
5698355f576SJeff Roberson  * Arguments:
5708355f576SJeff Roberson  *	zone  The zone to free pages from
5719c2cd7e5SJeff Roberson  *	all   Should we drain all items?
5728355f576SJeff Roberson  *
5738355f576SJeff Roberson  * Returns:
5748355f576SJeff Roberson  *	Nothing.
5758355f576SJeff Roberson  */
5768355f576SJeff Roberson static void
5778355f576SJeff Roberson zone_drain(uma_zone_t zone)
5788355f576SJeff Roberson {
579713deb36SJeff Roberson 	struct slabhead freeslabs = {};
5808355f576SJeff Roberson 	uma_slab_t slab;
5818355f576SJeff Roberson 	uma_slab_t n;
5828355f576SJeff Roberson 	u_int64_t extra;
5838355f576SJeff Roberson 	u_int8_t flags;
5848355f576SJeff Roberson 	u_int8_t *mem;
5858355f576SJeff Roberson 	int i;
5868355f576SJeff Roberson 
5878355f576SJeff Roberson 	/*
5888355f576SJeff Roberson 	 * We don't want to take pages from staticly allocated zones at this
5898355f576SJeff Roberson 	 * time
5908355f576SJeff Roberson 	 */
5918355f576SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_NOFREE || zone->uz_freef == NULL)
5928355f576SJeff Roberson 		return;
5938355f576SJeff Roberson 
5948355f576SJeff Roberson 	ZONE_LOCK(zone);
5958355f576SJeff Roberson 
5968355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_INTERNAL))
5978355f576SJeff Roberson 		cache_drain(zone);
5988355f576SJeff Roberson 
5998355f576SJeff Roberson 	if (zone->uz_free < zone->uz_wssize)
6008355f576SJeff Roberson 		goto finished;
6018355f576SJeff Roberson #ifdef UMA_DEBUG
6028355f576SJeff Roberson 	printf("%s working set size: %llu free items: %u\n",
6038355f576SJeff Roberson 	    zone->uz_name, (unsigned long long)zone->uz_wssize, zone->uz_free);
6048355f576SJeff Roberson #endif
6059c2cd7e5SJeff Roberson 	extra = zone->uz_free - zone->uz_wssize;
6068355f576SJeff Roberson 	extra /= zone->uz_ipers;
6078355f576SJeff Roberson 
6088355f576SJeff Roberson 	/* extra is now the number of extra slabs that we can free */
6098355f576SJeff Roberson 
6108355f576SJeff Roberson 	if (extra == 0)
6118355f576SJeff Roberson 		goto finished;
6128355f576SJeff Roberson 
6138355f576SJeff Roberson 	slab = LIST_FIRST(&zone->uz_free_slab);
6148355f576SJeff Roberson 	while (slab && extra) {
6158355f576SJeff Roberson 		n = LIST_NEXT(slab, us_link);
6168355f576SJeff Roberson 
6178355f576SJeff Roberson 		/* We have no where to free these to */
6188355f576SJeff Roberson 		if (slab->us_flags & UMA_SLAB_BOOT) {
6198355f576SJeff Roberson 			slab = n;
6208355f576SJeff Roberson 			continue;
6218355f576SJeff Roberson 		}
6228355f576SJeff Roberson 
6238355f576SJeff Roberson 		LIST_REMOVE(slab, us_link);
6248355f576SJeff Roberson 		zone->uz_pages -= zone->uz_ppera;
6258355f576SJeff Roberson 		zone->uz_free -= zone->uz_ipers;
626713deb36SJeff Roberson 
627713deb36SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_MALLOC) {
628713deb36SJeff Roberson 			mtx_lock(&malloc_mtx);
629713deb36SJeff Roberson 			UMA_HASH_REMOVE(mallochash, slab, slab->us_data);
630713deb36SJeff Roberson 			mtx_unlock(&malloc_mtx);
631713deb36SJeff Roberson 		}
632713deb36SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_OFFPAGE &&
633713deb36SJeff Roberson 		    !(zone->uz_flags & UMA_ZFLAG_MALLOC))
634713deb36SJeff Roberson 			UMA_HASH_REMOVE(&zone->uz_hash, slab, slab->us_data);
635713deb36SJeff Roberson 
636713deb36SJeff Roberson 		SLIST_INSERT_HEAD(&freeslabs, slab, us_hlink);
637713deb36SJeff Roberson 
638713deb36SJeff Roberson 		slab = n;
639713deb36SJeff Roberson 		extra--;
640713deb36SJeff Roberson 	}
641713deb36SJeff Roberson finished:
642713deb36SJeff Roberson 	ZONE_UNLOCK(zone);
643713deb36SJeff Roberson 
644713deb36SJeff Roberson 	while ((slab = SLIST_FIRST(&freeslabs)) != NULL) {
645713deb36SJeff Roberson 		SLIST_REMOVE(&freeslabs, slab, uma_slab, us_hlink);
6468355f576SJeff Roberson 		if (zone->uz_fini)
6478355f576SJeff Roberson 			for (i = 0; i < zone->uz_ipers; i++)
6488355f576SJeff Roberson 				zone->uz_fini(
6498355f576SJeff Roberson 				    slab->us_data + (zone->uz_rsize * i),
6508355f576SJeff Roberson 				    zone->uz_size);
6518355f576SJeff Roberson 		flags = slab->us_flags;
6528355f576SJeff Roberson 		mem = slab->us_data;
6535a34a9f0SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_OFFPAGE) {
6548355f576SJeff Roberson 			uma_zfree_internal(slabzone, slab, NULL, 0);
6555a34a9f0SJeff Roberson 		}
6568355f576SJeff Roberson #ifdef UMA_DEBUG
6578355f576SJeff Roberson 		printf("%s: Returning %d bytes.\n",
6588355f576SJeff Roberson 		    zone->uz_name, UMA_SLAB_SIZE * zone->uz_ppera);
6598355f576SJeff Roberson #endif
6608355f576SJeff Roberson 		zone->uz_freef(mem, UMA_SLAB_SIZE * zone->uz_ppera, flags);
6618355f576SJeff Roberson 	}
6628355f576SJeff Roberson 
6638355f576SJeff Roberson }
6648355f576SJeff Roberson 
6658355f576SJeff Roberson /*
6668355f576SJeff Roberson  * Allocate a new slab for a zone.  This does not insert the slab onto a list.
6678355f576SJeff Roberson  *
6688355f576SJeff Roberson  * Arguments:
6698355f576SJeff Roberson  *	zone  The zone to allocate slabs for
6708355f576SJeff Roberson  *	wait  Shall we wait?
6718355f576SJeff Roberson  *
6728355f576SJeff Roberson  * Returns:
6738355f576SJeff Roberson  *	The slab that was allocated or NULL if there is no memory and the
6748355f576SJeff Roberson  *	caller specified M_NOWAIT.
6758355f576SJeff Roberson  *
6768355f576SJeff Roberson  */
6778355f576SJeff Roberson static uma_slab_t
6788355f576SJeff Roberson slab_zalloc(uma_zone_t zone, int wait)
6798355f576SJeff Roberson {
6808355f576SJeff Roberson 	uma_slab_t slab;	/* Starting slab */
6818355f576SJeff Roberson 	u_int8_t *mem;
6828355f576SJeff Roberson 	u_int8_t flags;
6838355f576SJeff Roberson 	int i;
6848355f576SJeff Roberson 
685a553d4b8SJeff Roberson 	slab = NULL;
686a553d4b8SJeff Roberson 
6878355f576SJeff Roberson #ifdef UMA_DEBUG
6888355f576SJeff Roberson 	printf("slab_zalloc:  Allocating a new slab for %s\n", zone->uz_name);
6898355f576SJeff Roberson #endif
6908355f576SJeff Roberson 	ZONE_UNLOCK(zone);
691a553d4b8SJeff Roberson 
692a553d4b8SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_OFFPAGE) {
693a553d4b8SJeff Roberson 		slab = uma_zalloc_internal(slabzone, NULL, wait, NULL);
694a553d4b8SJeff Roberson 		if (slab == NULL) {
695a553d4b8SJeff Roberson 			ZONE_LOCK(zone);
696a553d4b8SJeff Roberson 			return NULL;
697a553d4b8SJeff Roberson 		}
698a553d4b8SJeff Roberson 	}
699a553d4b8SJeff Roberson 
700a553d4b8SJeff Roberson 	if (booted || (zone->uz_flags & UMA_ZFLAG_PRIVALLOC)) {
7018355f576SJeff Roberson 		mtx_lock(&Giant);
702a553d4b8SJeff Roberson 		mem = zone->uz_allocf(zone,
7038355f576SJeff Roberson 		    zone->uz_ppera * UMA_SLAB_SIZE, &flags, wait);
7048355f576SJeff Roberson 		mtx_unlock(&Giant);
705a553d4b8SJeff Roberson 		if (mem == NULL) {
7068355f576SJeff Roberson 			ZONE_LOCK(zone);
7078355f576SJeff Roberson 			return (NULL);
708a553d4b8SJeff Roberson 		}
7098355f576SJeff Roberson 	} else {
710a553d4b8SJeff Roberson 		uma_slab_t tmps;
7118355f576SJeff Roberson 
7128355f576SJeff Roberson 		if (zone->uz_ppera > 1)
7138355f576SJeff Roberson 			panic("UMA: Attemping to allocate multiple pages before vm has started.\n");
7148355f576SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_MALLOC)
7158355f576SJeff Roberson 			panic("Mallocing before uma_startup2 has been called.\n");
7168355f576SJeff Roberson 		if (uma_boot_free == 0)
7178355f576SJeff Roberson 			panic("UMA: Ran out of pre init pages, increase UMA_BOOT_PAGES\n");
718a553d4b8SJeff Roberson 		tmps = LIST_FIRST(&uma_boot_pages);
719a553d4b8SJeff Roberson 		LIST_REMOVE(tmps, us_link);
7208355f576SJeff Roberson 		uma_boot_free--;
721a553d4b8SJeff Roberson 		mem = tmps->us_data;
7228355f576SJeff Roberson 	}
7238355f576SJeff Roberson 
724a553d4b8SJeff Roberson 	ZONE_LOCK(zone);
7258355f576SJeff Roberson 
7268355f576SJeff Roberson 	/* Alloc slab structure for offpage, otherwise adjust it's position */
7278355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_OFFPAGE)) {
7288355f576SJeff Roberson 		slab = (uma_slab_t )(mem + zone->uz_pgoff);
7298355f576SJeff Roberson 	} else  {
7308355f576SJeff Roberson 		if (!(zone->uz_flags & UMA_ZFLAG_MALLOC))
7318355f576SJeff Roberson 			UMA_HASH_INSERT(&zone->uz_hash, slab, mem);
7328355f576SJeff Roberson 	}
7338355f576SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_MALLOC) {
7348355f576SJeff Roberson #ifdef UMA_DEBUG
7358355f576SJeff Roberson 		printf("Inserting %p into malloc hash from slab %p\n",
7368355f576SJeff Roberson 		    mem, slab);
7378355f576SJeff Roberson #endif
7385a34a9f0SJeff Roberson 		mtx_lock(&malloc_mtx);
7398355f576SJeff Roberson 		UMA_HASH_INSERT(mallochash, slab, mem);
7405a34a9f0SJeff Roberson 		mtx_unlock(&malloc_mtx);
7418355f576SJeff Roberson 	}
7428355f576SJeff Roberson 
7438355f576SJeff Roberson 	slab->us_zone = zone;
7448355f576SJeff Roberson 	slab->us_data = mem;
7458355f576SJeff Roberson 
7468355f576SJeff Roberson 	/*
7478355f576SJeff Roberson 	 * This is intended to spread data out across cache lines.
7488355f576SJeff Roberson 	 *
7498355f576SJeff Roberson 	 * This code doesn't seem to work properly on x86, and on alpha
7508355f576SJeff Roberson 	 * it makes absolutely no performance difference. I'm sure it could
7518355f576SJeff Roberson 	 * use some tuning, but sun makes outrageous claims about it's
7528355f576SJeff Roberson 	 * performance.
7538355f576SJeff Roberson 	 */
7548355f576SJeff Roberson #if 0
7558355f576SJeff Roberson 	if (zone->uz_cachemax) {
7568355f576SJeff Roberson 		slab->us_data += zone->uz_cacheoff;
7578355f576SJeff Roberson 		zone->uz_cacheoff += UMA_CACHE_INC;
7588355f576SJeff Roberson 		if (zone->uz_cacheoff > zone->uz_cachemax)
7598355f576SJeff Roberson 			zone->uz_cacheoff = 0;
7608355f576SJeff Roberson 	}
7618355f576SJeff Roberson #endif
7628355f576SJeff Roberson 
7638355f576SJeff Roberson 	slab->us_freecount = zone->uz_ipers;
7648355f576SJeff Roberson 	slab->us_firstfree = 0;
7658355f576SJeff Roberson 	slab->us_flags = flags;
7668355f576SJeff Roberson 	for (i = 0; i < zone->uz_ipers; i++)
7678355f576SJeff Roberson 		slab->us_freelist[i] = i+1;
7688355f576SJeff Roberson 
7698355f576SJeff Roberson 	if (zone->uz_init)
7708355f576SJeff Roberson 		for (i = 0; i < zone->uz_ipers; i++)
7718355f576SJeff Roberson 			zone->uz_init(slab->us_data + (zone->uz_rsize * i),
7728355f576SJeff Roberson 			    zone->uz_size);
7738355f576SJeff Roberson 
7748355f576SJeff Roberson 	zone->uz_pages += zone->uz_ppera;
7758355f576SJeff Roberson 	zone->uz_free += zone->uz_ipers;
7768355f576SJeff Roberson 
7778355f576SJeff Roberson 	return (slab);
7788355f576SJeff Roberson }
7798355f576SJeff Roberson 
7808355f576SJeff Roberson /*
7818355f576SJeff Roberson  * Allocates a number of pages from the system
7828355f576SJeff Roberson  *
7838355f576SJeff Roberson  * Arguments:
7848355f576SJeff Roberson  *	zone  Unused
7858355f576SJeff Roberson  *	bytes  The number of bytes requested
7868355f576SJeff Roberson  *	wait  Shall we wait?
7878355f576SJeff Roberson  *
7888355f576SJeff Roberson  * Returns:
7898355f576SJeff Roberson  *	A pointer to the alloced memory or possibly
7908355f576SJeff Roberson  *	NULL if M_NOWAIT is set.
7918355f576SJeff Roberson  */
7928355f576SJeff Roberson static void *
7938355f576SJeff Roberson page_alloc(uma_zone_t zone, int bytes, u_int8_t *pflag, int wait)
7948355f576SJeff Roberson {
7958355f576SJeff Roberson 	void *p;	/* Returned page */
7968355f576SJeff Roberson 
7978355f576SJeff Roberson 	/*
7988355f576SJeff Roberson 	 * XXX The original zone allocator did this, but I don't think it's
799157d7b35SAlfred Perlstein 	 * necessary in current.
8008355f576SJeff Roberson 	 */
8018355f576SJeff Roberson 
8028355f576SJeff Roberson 	if (lockstatus(&kernel_map->lock, NULL)) {
8038355f576SJeff Roberson 		*pflag = UMA_SLAB_KMEM;
8048355f576SJeff Roberson 		p = (void *) kmem_malloc(kmem_map, bytes, wait);
8058355f576SJeff Roberson 	} else {
8068355f576SJeff Roberson 		*pflag = UMA_SLAB_KMAP;
8078355f576SJeff Roberson 		p = (void *) kmem_alloc(kernel_map, bytes);
8088355f576SJeff Roberson 	}
8098355f576SJeff Roberson 
8108355f576SJeff Roberson 	return (p);
8118355f576SJeff Roberson }
8128355f576SJeff Roberson 
8138355f576SJeff Roberson /*
8148355f576SJeff Roberson  * Allocates a number of pages from within an object
8158355f576SJeff Roberson  *
8168355f576SJeff Roberson  * Arguments:
8178355f576SJeff Roberson  *	zone   Unused
8188355f576SJeff Roberson  *	bytes  The number of bytes requested
8198355f576SJeff Roberson  *	wait   Shall we wait?
8208355f576SJeff Roberson  *
8218355f576SJeff Roberson  * Returns:
8228355f576SJeff Roberson  *	A pointer to the alloced memory or possibly
8238355f576SJeff Roberson  *	NULL if M_NOWAIT is set.
8248355f576SJeff Roberson  */
8258355f576SJeff Roberson static void *
8268355f576SJeff Roberson obj_alloc(uma_zone_t zone, int bytes, u_int8_t *flags, int wait)
8278355f576SJeff Roberson {
8288355f576SJeff Roberson 	vm_offset_t zkva;
8298355f576SJeff Roberson 	vm_offset_t retkva;
8308355f576SJeff Roberson 	vm_page_t p;
8318355f576SJeff Roberson 	int pages;
8328355f576SJeff Roberson 
8338355f576SJeff Roberson 	retkva = NULL;
8348355f576SJeff Roberson 	pages = zone->uz_pages;
8358355f576SJeff Roberson 
8368355f576SJeff Roberson 	/*
8378355f576SJeff Roberson 	 * This looks a little weird since we're getting one page at a time
8388355f576SJeff Roberson 	 */
8398355f576SJeff Roberson 	while (bytes > 0) {
8408355f576SJeff Roberson 		p = vm_page_alloc(zone->uz_obj, pages,
8418355f576SJeff Roberson 		    VM_ALLOC_INTERRUPT);
8428355f576SJeff Roberson 		if (p == NULL)
8438355f576SJeff Roberson 			return (NULL);
8448355f576SJeff Roberson 
8458355f576SJeff Roberson 		zkva = zone->uz_kva + pages * PAGE_SIZE;
8468355f576SJeff Roberson 		if (retkva == NULL)
8478355f576SJeff Roberson 			retkva = zkva;
8488355f576SJeff Roberson 		pmap_qenter(zkva, &p, 1);
8498355f576SJeff Roberson 		bytes -= PAGE_SIZE;
8508355f576SJeff Roberson 		pages += 1;
8518355f576SJeff Roberson 	}
8528355f576SJeff Roberson 
8538355f576SJeff Roberson 	*flags = UMA_SLAB_PRIV;
8548355f576SJeff Roberson 
8558355f576SJeff Roberson 	return ((void *)retkva);
8568355f576SJeff Roberson }
8578355f576SJeff Roberson 
8588355f576SJeff Roberson /*
8598355f576SJeff Roberson  * Frees a number of pages to the system
8608355f576SJeff Roberson  *
8618355f576SJeff Roberson  * Arguments:
8628355f576SJeff Roberson  *	mem   A pointer to the memory to be freed
8638355f576SJeff Roberson  *	size  The size of the memory being freed
8648355f576SJeff Roberson  *	flags The original p->us_flags field
8658355f576SJeff Roberson  *
8668355f576SJeff Roberson  * Returns:
8678355f576SJeff Roberson  *	Nothing
8688355f576SJeff Roberson  *
8698355f576SJeff Roberson  */
8708355f576SJeff Roberson static void
8718355f576SJeff Roberson page_free(void *mem, int size, u_int8_t flags)
8728355f576SJeff Roberson {
8738355f576SJeff Roberson 	vm_map_t map;
8748355f576SJeff Roberson 	if (flags & UMA_SLAB_KMEM)
8758355f576SJeff Roberson 		map = kmem_map;
8768355f576SJeff Roberson 	else if (flags & UMA_SLAB_KMAP)
8778355f576SJeff Roberson 		map = kernel_map;
8788355f576SJeff Roberson 	else
8798355f576SJeff Roberson 		panic("UMA: page_free used with invalid flags %d\n", flags);
8808355f576SJeff Roberson 
8818355f576SJeff Roberson 	kmem_free(map, (vm_offset_t)mem, size);
8828355f576SJeff Roberson }
8838355f576SJeff Roberson 
8848355f576SJeff Roberson /*
8858355f576SJeff Roberson  * Zero fill initializer
8868355f576SJeff Roberson  *
8878355f576SJeff Roberson  * Arguments/Returns follow uma_init specifications
8888355f576SJeff Roberson  *
8898355f576SJeff Roberson  */
8908355f576SJeff Roberson static void
8918355f576SJeff Roberson zero_init(void *mem, int size)
8928355f576SJeff Roberson {
8938355f576SJeff Roberson 	bzero(mem, size);
8948355f576SJeff Roberson }
8958355f576SJeff Roberson 
8968355f576SJeff Roberson /*
8978355f576SJeff Roberson  * Finish creating a small uma zone.  This calculates ipers, and the zone size.
8988355f576SJeff Roberson  *
8998355f576SJeff Roberson  * Arguments
9008355f576SJeff Roberson  *	zone  The zone we should initialize
9018355f576SJeff Roberson  *
9028355f576SJeff Roberson  * Returns
9038355f576SJeff Roberson  *	Nothing
9048355f576SJeff Roberson  */
9058355f576SJeff Roberson static void
9068355f576SJeff Roberson zone_small_init(uma_zone_t zone)
9078355f576SJeff Roberson {
9088355f576SJeff Roberson 	int rsize;
9098355f576SJeff Roberson 	int memused;
9108355f576SJeff Roberson 	int ipers;
9118355f576SJeff Roberson 
9128355f576SJeff Roberson 	rsize = zone->uz_size;
9138355f576SJeff Roberson 
9148355f576SJeff Roberson 	if (rsize < UMA_SMALLEST_UNIT)
9158355f576SJeff Roberson 		rsize = UMA_SMALLEST_UNIT;
9168355f576SJeff Roberson 
9178355f576SJeff Roberson 	if (rsize & zone->uz_align)
9188355f576SJeff Roberson 		rsize = (rsize & ~zone->uz_align) + (zone->uz_align + 1);
9198355f576SJeff Roberson 
9208355f576SJeff Roberson 	zone->uz_rsize = rsize;
9218355f576SJeff Roberson 
9228355f576SJeff Roberson 	rsize += 1;	/* Account for the byte of linkage */
9238355f576SJeff Roberson 	zone->uz_ipers = (UMA_SLAB_SIZE - sizeof(struct uma_slab)) / rsize;
9248355f576SJeff Roberson 	zone->uz_ppera = 1;
9258355f576SJeff Roberson 
9268355f576SJeff Roberson 	memused = zone->uz_ipers * zone->uz_rsize;
9278355f576SJeff Roberson 
9288355f576SJeff Roberson 	/* Can we do any better? */
9298355f576SJeff Roberson 	if ((UMA_SLAB_SIZE - memused) >= UMA_MAX_WASTE) {
9308355f576SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_INTERNAL)
9318355f576SJeff Roberson 			return;
9328355f576SJeff Roberson 		ipers = UMA_SLAB_SIZE / zone->uz_rsize;
9338355f576SJeff Roberson 		if (ipers > zone->uz_ipers) {
9348355f576SJeff Roberson 			zone->uz_flags |= UMA_ZFLAG_OFFPAGE;
9358355f576SJeff Roberson 			zone->uz_ipers = ipers;
9368355f576SJeff Roberson 		}
9378355f576SJeff Roberson 	}
9388355f576SJeff Roberson 
9398355f576SJeff Roberson }
9408355f576SJeff Roberson 
9418355f576SJeff Roberson /*
9428355f576SJeff Roberson  * Finish creating a large (> UMA_SLAB_SIZE) uma zone.  Just give in and do
9438355f576SJeff Roberson  * OFFPAGE for now.  When I can allow for more dynamic slab sizes this will be
9448355f576SJeff Roberson  * more complicated.
9458355f576SJeff Roberson  *
9468355f576SJeff Roberson  * Arguments
9478355f576SJeff Roberson  *	zone  The zone we should initialize
9488355f576SJeff Roberson  *
9498355f576SJeff Roberson  * Returns
9508355f576SJeff Roberson  *	Nothing
9518355f576SJeff Roberson  */
9528355f576SJeff Roberson static void
9538355f576SJeff Roberson zone_large_init(uma_zone_t zone)
9548355f576SJeff Roberson {
9558355f576SJeff Roberson 	int pages;
9568355f576SJeff Roberson 
9578355f576SJeff Roberson 	pages = zone->uz_size / UMA_SLAB_SIZE;
9588355f576SJeff Roberson 
9598355f576SJeff Roberson 	/* Account for remainder */
9608355f576SJeff Roberson 	if ((pages * UMA_SLAB_SIZE) < zone->uz_size)
9618355f576SJeff Roberson 		pages++;
9628355f576SJeff Roberson 
9638355f576SJeff Roberson 	zone->uz_ppera = pages;
9648355f576SJeff Roberson 	zone->uz_ipers = 1;
9658355f576SJeff Roberson 
9668355f576SJeff Roberson 	zone->uz_flags |= UMA_ZFLAG_OFFPAGE;
9678355f576SJeff Roberson 	zone->uz_rsize = zone->uz_size;
9688355f576SJeff Roberson }
9698355f576SJeff Roberson 
9708355f576SJeff Roberson /*
9718355f576SJeff Roberson  * Zone header ctor.  This initializes all fields, locks, etc.  And inserts
9728355f576SJeff Roberson  * the zone onto the global zone list.
9738355f576SJeff Roberson  *
9748355f576SJeff Roberson  * Arguments/Returns follow uma_ctor specifications
9758355f576SJeff Roberson  *	udata  Actually uma_zcreat_args
9768355f576SJeff Roberson  *
9778355f576SJeff Roberson  */
9788355f576SJeff Roberson 
9798355f576SJeff Roberson static void
9808355f576SJeff Roberson zone_ctor(void *mem, int size, void *udata)
9818355f576SJeff Roberson {
9828355f576SJeff Roberson 	struct uma_zctor_args *arg = udata;
9838355f576SJeff Roberson 	uma_zone_t zone = mem;
98428bc4419SJeff Roberson 	int privlc;
9858355f576SJeff Roberson 	int cplen;
9868355f576SJeff Roberson 	int cpu;
9878355f576SJeff Roberson 
9888355f576SJeff Roberson 	bzero(zone, size);
9898355f576SJeff Roberson 	zone->uz_name = arg->name;
9908355f576SJeff Roberson 	zone->uz_size = arg->size;
9918355f576SJeff Roberson 	zone->uz_ctor = arg->ctor;
9928355f576SJeff Roberson 	zone->uz_dtor = arg->dtor;
9938355f576SJeff Roberson 	zone->uz_init = arg->uminit;
9948355f576SJeff Roberson 	zone->uz_align = arg->align;
9958355f576SJeff Roberson 	zone->uz_free = 0;
9968355f576SJeff Roberson 	zone->uz_pages = 0;
9978355f576SJeff Roberson 	zone->uz_flags = 0;
9988355f576SJeff Roberson 	zone->uz_allocf = page_alloc;
9998355f576SJeff Roberson 	zone->uz_freef = page_free;
10008355f576SJeff Roberson 
10018355f576SJeff Roberson 	if (arg->flags & UMA_ZONE_ZINIT)
10028355f576SJeff Roberson 		zone->uz_init = zero_init;
10038355f576SJeff Roberson 
10048355f576SJeff Roberson 	if (arg->flags & UMA_ZONE_INTERNAL)
10058355f576SJeff Roberson 		zone->uz_flags |= UMA_ZFLAG_INTERNAL;
10068355f576SJeff Roberson 
10078355f576SJeff Roberson 	if (arg->flags & UMA_ZONE_MALLOC)
10088355f576SJeff Roberson 		zone->uz_flags |= UMA_ZFLAG_MALLOC;
10098355f576SJeff Roberson 
10108355f576SJeff Roberson 	if (arg->flags & UMA_ZONE_NOFREE)
10118355f576SJeff Roberson 		zone->uz_flags |= UMA_ZFLAG_NOFREE;
10128355f576SJeff Roberson 
10138355f576SJeff Roberson 	if (zone->uz_size > UMA_SLAB_SIZE)
10148355f576SJeff Roberson 		zone_large_init(zone);
10158355f576SJeff Roberson 	else
10168355f576SJeff Roberson 		zone_small_init(zone);
10178355f576SJeff Roberson 
101828bc4419SJeff Roberson 	if (arg->flags & UMA_ZONE_MTXCLASS)
101928bc4419SJeff Roberson 		privlc = 1;
102028bc4419SJeff Roberson 	else
102128bc4419SJeff Roberson 		privlc = 0;
102228bc4419SJeff Roberson 
10238355f576SJeff Roberson 	/* We do this so that the per cpu lock name is unique for each zone */
10248355f576SJeff Roberson 	memcpy(zone->uz_lname, "PCPU ", 5);
10258355f576SJeff Roberson 	cplen = min(strlen(zone->uz_name) + 1, LOCKNAME_LEN - 6);
10268355f576SJeff Roberson 	memcpy(zone->uz_lname+5, zone->uz_name, cplen);
10278355f576SJeff Roberson 	zone->uz_lname[LOCKNAME_LEN - 1] = '\0';
10288355f576SJeff Roberson 
10298355f576SJeff Roberson 	/*
10308355f576SJeff Roberson 	 * If we're putting the slab header in the actual page we need to
10318355f576SJeff Roberson 	 * figure out where in each page it goes.  This calculates a right
10328355f576SJeff Roberson 	 * justified offset into the memory on a ALIGN_PTR boundary.
10338355f576SJeff Roberson 	 */
10348355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_OFFPAGE)) {
10358355f576SJeff Roberson 		int totsize;
10368355f576SJeff Roberson 		int waste;
10378355f576SJeff Roberson 
10388355f576SJeff Roberson 		/* Size of the slab struct and free list */
10398355f576SJeff Roberson 		totsize = sizeof(struct uma_slab) + zone->uz_ipers;
10408355f576SJeff Roberson 		if (totsize & UMA_ALIGN_PTR)
10418355f576SJeff Roberson 			totsize = (totsize & ~UMA_ALIGN_PTR) +
10428355f576SJeff Roberson 			    (UMA_ALIGN_PTR + 1);
10438355f576SJeff Roberson 		zone->uz_pgoff = UMA_SLAB_SIZE - totsize;
10448355f576SJeff Roberson 
10458355f576SJeff Roberson 		waste = zone->uz_pgoff;
10468355f576SJeff Roberson 		waste -= (zone->uz_ipers * zone->uz_rsize);
10478355f576SJeff Roberson 
10488355f576SJeff Roberson 		/*
10498355f576SJeff Roberson 		 * This calculates how much space we have for cache line size
10508355f576SJeff Roberson 		 * optimizations.  It works by offseting each slab slightly.
10518355f576SJeff Roberson 		 * Currently it breaks on x86, and so it is disabled.
10528355f576SJeff Roberson 		 */
10538355f576SJeff Roberson 
10548355f576SJeff Roberson 		if (zone->uz_align < UMA_CACHE_INC && waste > UMA_CACHE_INC) {
10558355f576SJeff Roberson 			zone->uz_cachemax = waste - UMA_CACHE_INC;
10568355f576SJeff Roberson 			zone->uz_cacheoff = 0;
10578355f576SJeff Roberson 		}
10588355f576SJeff Roberson 
10598355f576SJeff Roberson 		totsize = zone->uz_pgoff + sizeof(struct uma_slab)
10608355f576SJeff Roberson 		    + zone->uz_ipers;
10618355f576SJeff Roberson 		/* I don't think it's possible, but I'll make sure anyway */
10628355f576SJeff Roberson 		if (totsize > UMA_SLAB_SIZE) {
10638355f576SJeff Roberson 			printf("zone %s ipers %d rsize %d size %d\n",
10648355f576SJeff Roberson 			    zone->uz_name, zone->uz_ipers, zone->uz_rsize,
10658355f576SJeff Roberson 			    zone->uz_size);
10668355f576SJeff Roberson 			panic("UMA slab won't fit.\n");
10678355f576SJeff Roberson 		}
10688355f576SJeff Roberson 	} else {
10690aef6126SJeff Roberson 		hash_alloc(&zone->uz_hash);
10708355f576SJeff Roberson 		zone->uz_pgoff = 0;
10718355f576SJeff Roberson 	}
10728355f576SJeff Roberson 
10738355f576SJeff Roberson #ifdef UMA_DEBUG
10748355f576SJeff Roberson 	printf("%s(%p) size = %d ipers = %d ppera = %d pgoff = %d\n",
10758355f576SJeff Roberson 	    zone->uz_name, zone,
10768355f576SJeff Roberson 	    zone->uz_size, zone->uz_ipers,
10778355f576SJeff Roberson 	    zone->uz_ppera, zone->uz_pgoff);
10788355f576SJeff Roberson #endif
107928bc4419SJeff Roberson 	ZONE_LOCK_INIT(zone, privlc);
10808355f576SJeff Roberson 
10818355f576SJeff Roberson 	mtx_lock(&uma_mtx);
10828355f576SJeff Roberson 	LIST_INSERT_HEAD(&uma_zones, zone, uz_link);
10838355f576SJeff Roberson 	mtx_unlock(&uma_mtx);
10848355f576SJeff Roberson 
10858355f576SJeff Roberson 	/*
10868355f576SJeff Roberson 	 * Some internal zones don't have room allocated for the per cpu
10878355f576SJeff Roberson 	 * caches.  If we're internal, bail out here.
10888355f576SJeff Roberson 	 */
10898355f576SJeff Roberson 
10908355f576SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_INTERNAL)
10918355f576SJeff Roberson 		return;
10928355f576SJeff Roberson 
10938355f576SJeff Roberson 	if (zone->uz_ipers < UMA_BUCKET_SIZE)
1094a553d4b8SJeff Roberson 		zone->uz_count = zone->uz_ipers - 1;
10958355f576SJeff Roberson 	else
1096a553d4b8SJeff Roberson 		zone->uz_count = UMA_BUCKET_SIZE - 1;
1097a553d4b8SJeff Roberson 
1098a553d4b8SJeff Roberson 	for (cpu = 0; cpu < maxcpu; cpu++)
109928bc4419SJeff Roberson 		CPU_LOCK_INIT(zone, cpu, privlc);
11008355f576SJeff Roberson }
11018355f576SJeff Roberson 
11028355f576SJeff Roberson /*
11039c2cd7e5SJeff Roberson  * Zone header dtor.  This frees all data, destroys locks, frees the hash table
11049c2cd7e5SJeff Roberson  * and removes the zone from the global list.
11059c2cd7e5SJeff Roberson  *
11069c2cd7e5SJeff Roberson  * Arguments/Returns follow uma_dtor specifications
11079c2cd7e5SJeff Roberson  *	udata  unused
11089c2cd7e5SJeff Roberson  */
11099c2cd7e5SJeff Roberson 
11109c2cd7e5SJeff Roberson static void
11119c2cd7e5SJeff Roberson zone_dtor(void *arg, int size, void *udata)
11129c2cd7e5SJeff Roberson {
11139c2cd7e5SJeff Roberson 	uma_zone_t zone;
11149c2cd7e5SJeff Roberson 	int cpu;
11159c2cd7e5SJeff Roberson 
11169c2cd7e5SJeff Roberson 	zone = (uma_zone_t)arg;
11179c2cd7e5SJeff Roberson 
11189c2cd7e5SJeff Roberson 	mtx_lock(&uma_mtx);
11199c2cd7e5SJeff Roberson 	LIST_REMOVE(zone, uz_link);
11209c2cd7e5SJeff Roberson 	mtx_unlock(&uma_mtx);
11219c2cd7e5SJeff Roberson 
11229c2cd7e5SJeff Roberson 	ZONE_LOCK(zone);
11239c2cd7e5SJeff Roberson 	zone->uz_wssize = 0;
11249c2cd7e5SJeff Roberson 	ZONE_UNLOCK(zone);
11259c2cd7e5SJeff Roberson 
11269c2cd7e5SJeff Roberson 	zone_drain(zone);
11279c2cd7e5SJeff Roberson 	ZONE_LOCK(zone);
11289c2cd7e5SJeff Roberson 	if (zone->uz_free != 0)
11299c2cd7e5SJeff Roberson 		printf("Zone %s was not empty.  Lost %d pages of memory.\n",
11309c2cd7e5SJeff Roberson 		    zone->uz_name, zone->uz_pages);
11319c2cd7e5SJeff Roberson 
11329c2cd7e5SJeff Roberson 	if ((zone->uz_flags & UMA_ZFLAG_INTERNAL) != 0)
11339c2cd7e5SJeff Roberson 		for (cpu = 0; cpu < maxcpu; cpu++)
11349c2cd7e5SJeff Roberson 			CPU_LOCK_FINI(zone, cpu);
11359c2cd7e5SJeff Roberson 
11369c2cd7e5SJeff Roberson 	ZONE_UNLOCK(zone);
11370aef6126SJeff Roberson 	if ((zone->uz_flags & UMA_ZFLAG_OFFPAGE) != 0)
11380aef6126SJeff Roberson 		hash_free(&zone->uz_hash);
11390aef6126SJeff Roberson 
11409c2cd7e5SJeff Roberson 	ZONE_LOCK_FINI(zone);
11419c2cd7e5SJeff Roberson }
11429c2cd7e5SJeff Roberson /*
11438355f576SJeff Roberson  * Traverses every zone in the system and calls a callback
11448355f576SJeff Roberson  *
11458355f576SJeff Roberson  * Arguments:
11468355f576SJeff Roberson  *	zfunc  A pointer to a function which accepts a zone
11478355f576SJeff Roberson  *		as an argument.
11488355f576SJeff Roberson  *
11498355f576SJeff Roberson  * Returns:
11508355f576SJeff Roberson  *	Nothing
11518355f576SJeff Roberson  */
11528355f576SJeff Roberson static void
11538355f576SJeff Roberson zone_foreach(void (*zfunc)(uma_zone_t))
11548355f576SJeff Roberson {
11558355f576SJeff Roberson 	uma_zone_t zone;
11568355f576SJeff Roberson 
11578355f576SJeff Roberson 	mtx_lock(&uma_mtx);
11588355f576SJeff Roberson 	LIST_FOREACH(zone, &uma_zones, uz_link) {
11598355f576SJeff Roberson 		zfunc(zone);
11608355f576SJeff Roberson 	}
11618355f576SJeff Roberson 	mtx_unlock(&uma_mtx);
11628355f576SJeff Roberson }
11638355f576SJeff Roberson 
11648355f576SJeff Roberson /* Public functions */
11658355f576SJeff Roberson /* See uma.h */
11668355f576SJeff Roberson void
11678355f576SJeff Roberson uma_startup(void *bootmem)
11688355f576SJeff Roberson {
11698355f576SJeff Roberson 	struct uma_zctor_args args;
11708355f576SJeff Roberson 	uma_slab_t slab;
11718355f576SJeff Roberson 	int slabsize;
11728355f576SJeff Roberson 	int i;
11738355f576SJeff Roberson 
11748355f576SJeff Roberson #ifdef UMA_DEBUG
11758355f576SJeff Roberson 	printf("Creating uma zone headers zone.\n");
11768355f576SJeff Roberson #endif
11778355f576SJeff Roberson #ifdef SMP
11788355f576SJeff Roberson 	maxcpu = mp_maxid + 1;
11798355f576SJeff Roberson #else
11808355f576SJeff Roberson 	maxcpu = 1;
11818355f576SJeff Roberson #endif
11828355f576SJeff Roberson #ifdef UMA_DEBUG
11838355f576SJeff Roberson 	printf("Max cpu = %d, mp_maxid = %d\n", maxcpu, mp_maxid);
11848355f576SJeff Roberson 	Debugger("stop");
11858355f576SJeff Roberson #endif
11866008862bSJohn Baldwin 	mtx_init(&uma_mtx, "UMA lock", NULL, MTX_DEF);
11878355f576SJeff Roberson 	/* "manually" Create the initial zone */
11888355f576SJeff Roberson 	args.name = "UMA Zones";
11898355f576SJeff Roberson 	args.size = sizeof(struct uma_zone) +
11908355f576SJeff Roberson 	    (sizeof(struct uma_cache) * (maxcpu - 1));
11918355f576SJeff Roberson 	args.ctor = zone_ctor;
11929c2cd7e5SJeff Roberson 	args.dtor = zone_dtor;
11938355f576SJeff Roberson 	args.uminit = zero_init;
11948355f576SJeff Roberson 	args.fini = NULL;
11958355f576SJeff Roberson 	args.align = 32 - 1;
11968355f576SJeff Roberson 	args.flags = UMA_ZONE_INTERNAL;
11978355f576SJeff Roberson 	/* The initial zone has no Per cpu queues so it's smaller */
11988355f576SJeff Roberson 	zone_ctor(zones, sizeof(struct uma_zone), &args);
11998355f576SJeff Roberson 
12008355f576SJeff Roberson #ifdef UMA_DEBUG
12018355f576SJeff Roberson 	printf("Filling boot free list.\n");
12028355f576SJeff Roberson #endif
12038355f576SJeff Roberson 	for (i = 0; i < UMA_BOOT_PAGES; i++) {
12048355f576SJeff Roberson 		slab = (uma_slab_t)((u_int8_t *)bootmem + (i * UMA_SLAB_SIZE));
12058355f576SJeff Roberson 		slab->us_data = (u_int8_t *)slab;
12068355f576SJeff Roberson 		slab->us_flags = UMA_SLAB_BOOT;
12078355f576SJeff Roberson 		LIST_INSERT_HEAD(&uma_boot_pages, slab, us_link);
12088355f576SJeff Roberson 		uma_boot_free++;
12098355f576SJeff Roberson 	}
12108355f576SJeff Roberson 
12118355f576SJeff Roberson #ifdef UMA_DEBUG
12128355f576SJeff Roberson 	printf("Creating slab zone.\n");
12138355f576SJeff Roberson #endif
12148355f576SJeff Roberson 
12158355f576SJeff Roberson 	/*
12168355f576SJeff Roberson 	 * This is the max number of free list items we'll have with
12178355f576SJeff Roberson 	 * offpage slabs.
12188355f576SJeff Roberson 	 */
12198355f576SJeff Roberson 
12208355f576SJeff Roberson 	slabsize = UMA_SLAB_SIZE - sizeof(struct uma_slab);
12218355f576SJeff Roberson 	slabsize /= UMA_MAX_WASTE;
12228355f576SJeff Roberson 	slabsize++;			/* In case there it's rounded */
12238355f576SJeff Roberson 	slabsize += sizeof(struct uma_slab);
12248355f576SJeff Roberson 
12258355f576SJeff Roberson 	/* Now make a zone for slab headers */
12268355f576SJeff Roberson 	slabzone = uma_zcreate("UMA Slabs",
12278355f576SJeff Roberson 				slabsize,
12288355f576SJeff Roberson 				NULL, NULL, NULL, NULL,
12298355f576SJeff Roberson 				UMA_ALIGN_PTR, UMA_ZONE_INTERNAL);
12308355f576SJeff Roberson 
12318355f576SJeff Roberson 	hashzone = uma_zcreate("UMA Hash",
12328355f576SJeff Roberson 	    sizeof(struct slabhead *) * UMA_HASH_SIZE_INIT,
12338355f576SJeff Roberson 	    NULL, NULL, NULL, NULL,
12348355f576SJeff Roberson 	    UMA_ALIGN_PTR, UMA_ZONE_INTERNAL);
12358355f576SJeff Roberson 
12368355f576SJeff Roberson 	bucketzone = uma_zcreate("UMA Buckets", sizeof(struct uma_bucket),
12378355f576SJeff Roberson 	    NULL, NULL, NULL, NULL,
12388355f576SJeff Roberson 	    UMA_ALIGN_PTR, UMA_ZONE_INTERNAL);
12398355f576SJeff Roberson 
12408355f576SJeff Roberson 
12418355f576SJeff Roberson #ifdef UMA_DEBUG
12428355f576SJeff Roberson 	printf("UMA startup complete.\n");
12438355f576SJeff Roberson #endif
12448355f576SJeff Roberson }
12458355f576SJeff Roberson 
12468355f576SJeff Roberson /* see uma.h */
12478355f576SJeff Roberson void
12488355f576SJeff Roberson uma_startup2(void *hashmem, u_long elems)
12498355f576SJeff Roberson {
12508355f576SJeff Roberson 	bzero(hashmem, elems * sizeof(void *));
12518355f576SJeff Roberson 	mallochash->uh_slab_hash = hashmem;
12528355f576SJeff Roberson 	mallochash->uh_hashsize = elems;
12538355f576SJeff Roberson 	mallochash->uh_hashmask = elems - 1;
12548355f576SJeff Roberson 	booted = 1;
125586bbae32SJeff Roberson 	bucket_enable();
12568355f576SJeff Roberson #ifdef UMA_DEBUG
12578355f576SJeff Roberson 	printf("UMA startup2 complete.\n");
12588355f576SJeff Roberson #endif
12598355f576SJeff Roberson }
12608355f576SJeff Roberson 
12618355f576SJeff Roberson /*
12628355f576SJeff Roberson  * Initialize our callout handle
12638355f576SJeff Roberson  *
12648355f576SJeff Roberson  */
12658355f576SJeff Roberson 
12668355f576SJeff Roberson static void
12678355f576SJeff Roberson uma_startup3(void)
12688355f576SJeff Roberson {
12698355f576SJeff Roberson #ifdef UMA_DEBUG
12708355f576SJeff Roberson 	printf("Starting callout.\n");
12718355f576SJeff Roberson #endif
12728355f576SJeff Roberson 	callout_init(&uma_callout, 0);
12738355f576SJeff Roberson 	callout_reset(&uma_callout, UMA_WORKING_TIME * hz, uma_timeout, NULL);
12748355f576SJeff Roberson #ifdef UMA_DEBUG
12758355f576SJeff Roberson 	printf("UMA startup3 complete.\n");
12768355f576SJeff Roberson #endif
12778355f576SJeff Roberson }
12788355f576SJeff Roberson 
12798355f576SJeff Roberson /* See uma.h */
12808355f576SJeff Roberson uma_zone_t
1281c3bdc05fSAndrew R. Reiter uma_zcreate(char *name, size_t size, uma_ctor ctor, uma_dtor dtor,
1282c3bdc05fSAndrew R. Reiter 		uma_init uminit, uma_fini fini, int align, u_int16_t flags)
12838355f576SJeff Roberson 
12848355f576SJeff Roberson {
12858355f576SJeff Roberson 	struct uma_zctor_args args;
12868355f576SJeff Roberson 
12878355f576SJeff Roberson 	/* This stuff is essential for the zone ctor */
12888355f576SJeff Roberson 	args.name = name;
12898355f576SJeff Roberson 	args.size = size;
12908355f576SJeff Roberson 	args.ctor = ctor;
12918355f576SJeff Roberson 	args.dtor = dtor;
12928355f576SJeff Roberson 	args.uminit = uminit;
12938355f576SJeff Roberson 	args.fini = fini;
12948355f576SJeff Roberson 	args.align = align;
12958355f576SJeff Roberson 	args.flags = flags;
12968355f576SJeff Roberson 
1297a553d4b8SJeff Roberson 	return (uma_zalloc_internal(zones, &args, M_WAITOK, NULL));
12988355f576SJeff Roberson }
12998355f576SJeff Roberson 
13008355f576SJeff Roberson /* See uma.h */
13019c2cd7e5SJeff Roberson void
13029c2cd7e5SJeff Roberson uma_zdestroy(uma_zone_t zone)
13039c2cd7e5SJeff Roberson {
13049c2cd7e5SJeff Roberson 	uma_zfree_internal(zones, zone, NULL, 0);
13059c2cd7e5SJeff Roberson }
13069c2cd7e5SJeff Roberson 
13079c2cd7e5SJeff Roberson /* See uma.h */
13088355f576SJeff Roberson void *
13092cc35ff9SJeff Roberson uma_zalloc_arg(uma_zone_t zone, void *udata, int flags)
13108355f576SJeff Roberson {
13118355f576SJeff Roberson 	void *item;
13128355f576SJeff Roberson 	uma_cache_t cache;
13138355f576SJeff Roberson 	uma_bucket_t bucket;
13148355f576SJeff Roberson 	int cpu;
13158355f576SJeff Roberson 
13168355f576SJeff Roberson 	/* This is the fast path allocation */
13178355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC_1
13188355f576SJeff Roberson 	printf("Allocating one item from %s(%p)\n", zone->uz_name, zone);
13198355f576SJeff Roberson #endif
1320a553d4b8SJeff Roberson 
13214c1cc01cSJohn Baldwin 	if (!(flags & M_NOWAIT)) {
13224c1cc01cSJohn Baldwin 		KASSERT(curthread->td_intr_nesting_level == 0,
13234c1cc01cSJohn Baldwin 		   ("malloc(M_WAITOK) in interrupt context"));
13244c1cc01cSJohn Baldwin 		WITNESS_SLEEP(1, NULL);
13254c1cc01cSJohn Baldwin 	}
13264c1cc01cSJohn Baldwin 
1327a553d4b8SJeff Roberson zalloc_restart:
13288355f576SJeff Roberson 	cpu = PCPU_GET(cpuid);
13298355f576SJeff Roberson 	CPU_LOCK(zone, cpu);
13308355f576SJeff Roberson 	cache = &zone->uz_cpu[cpu];
13318355f576SJeff Roberson 
13328355f576SJeff Roberson zalloc_start:
13338355f576SJeff Roberson 	bucket = cache->uc_allocbucket;
13348355f576SJeff Roberson 
13358355f576SJeff Roberson 	if (bucket) {
13368355f576SJeff Roberson 		if (bucket->ub_ptr > -1) {
13378355f576SJeff Roberson 			item = bucket->ub_bucket[bucket->ub_ptr];
13388355f576SJeff Roberson #ifdef INVARIANTS
13398355f576SJeff Roberson 			bucket->ub_bucket[bucket->ub_ptr] = NULL;
13408355f576SJeff Roberson #endif
13418355f576SJeff Roberson 			bucket->ub_ptr--;
13428355f576SJeff Roberson 			KASSERT(item != NULL,
13438355f576SJeff Roberson 			    ("uma_zalloc: Bucket pointer mangled."));
13448355f576SJeff Roberson 			cache->uc_allocs++;
1345639c9550SJeff Roberson #ifdef INVARIANTS
1346639c9550SJeff Roberson 			uma_dbg_alloc(zone, NULL, item);
1347639c9550SJeff Roberson #endif
1348b9ba8931SJeff Roberson 			CPU_UNLOCK(zone, cpu);
13498355f576SJeff Roberson 			if (zone->uz_ctor)
13508355f576SJeff Roberson 				zone->uz_ctor(item, zone->uz_size, udata);
13512cc35ff9SJeff Roberson 			if (flags & M_ZERO)
13522cc35ff9SJeff Roberson 				bzero(item, zone->uz_size);
13538355f576SJeff Roberson 			return (item);
13548355f576SJeff Roberson 		} else if (cache->uc_freebucket) {
13558355f576SJeff Roberson 			/*
13568355f576SJeff Roberson 			 * We have run out of items in our allocbucket.
13578355f576SJeff Roberson 			 * See if we can switch with our free bucket.
13588355f576SJeff Roberson 			 */
13598355f576SJeff Roberson 			if (cache->uc_freebucket->ub_ptr > -1) {
13608355f576SJeff Roberson 				uma_bucket_t swap;
13618355f576SJeff Roberson 
13628355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
13638355f576SJeff Roberson 				printf("uma_zalloc: Swapping empty with alloc.\n");
13648355f576SJeff Roberson #endif
13658355f576SJeff Roberson 				swap = cache->uc_freebucket;
13668355f576SJeff Roberson 				cache->uc_freebucket = cache->uc_allocbucket;
13678355f576SJeff Roberson 				cache->uc_allocbucket = swap;
13688355f576SJeff Roberson 
13698355f576SJeff Roberson 				goto zalloc_start;
13708355f576SJeff Roberson 			}
13718355f576SJeff Roberson 		}
13728355f576SJeff Roberson 	}
1373a553d4b8SJeff Roberson 	ZONE_LOCK(zone);
1374a553d4b8SJeff Roberson 	/* Since we have locked the zone we may as well send back our stats */
1375a553d4b8SJeff Roberson 	zone->uz_allocs += cache->uc_allocs;
1376a553d4b8SJeff Roberson 	cache->uc_allocs = 0;
13778355f576SJeff Roberson 
1378a553d4b8SJeff Roberson 	/* Our old one is now a free bucket */
1379a553d4b8SJeff Roberson 	if (cache->uc_allocbucket) {
1380a553d4b8SJeff Roberson 		KASSERT(cache->uc_allocbucket->ub_ptr == -1,
1381a553d4b8SJeff Roberson 		    ("uma_zalloc_arg: Freeing a non free bucket."));
1382a553d4b8SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_free_bucket,
1383a553d4b8SJeff Roberson 		    cache->uc_allocbucket, ub_link);
1384a553d4b8SJeff Roberson 		cache->uc_allocbucket = NULL;
1385a553d4b8SJeff Roberson 	}
13868355f576SJeff Roberson 
1387a553d4b8SJeff Roberson 	/* Check the free list for a new alloc bucket */
1388a553d4b8SJeff Roberson 	if ((bucket = LIST_FIRST(&zone->uz_full_bucket)) != NULL) {
1389a553d4b8SJeff Roberson 		KASSERT(bucket->ub_ptr != -1,
1390a553d4b8SJeff Roberson 		    ("uma_zalloc_arg: Returning an empty bucket."));
13918355f576SJeff Roberson 
1392a553d4b8SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
1393a553d4b8SJeff Roberson 		cache->uc_allocbucket = bucket;
1394a553d4b8SJeff Roberson 		ZONE_UNLOCK(zone);
13958355f576SJeff Roberson 		goto zalloc_start;
1396a553d4b8SJeff Roberson 	}
1397a553d4b8SJeff Roberson 	/* Bump up our uz_count so we get here less */
1398a553d4b8SJeff Roberson 	if (zone->uz_count < UMA_BUCKET_SIZE - 1)
1399a553d4b8SJeff Roberson 		zone->uz_count++;
1400a553d4b8SJeff Roberson 
1401a553d4b8SJeff Roberson 	/* We are no longer associated with this cpu!!! */
1402a553d4b8SJeff Roberson 	CPU_UNLOCK(zone, cpu);
14038355f576SJeff Roberson 
14048355f576SJeff Roberson 	/*
1405a553d4b8SJeff Roberson 	 * Now lets just fill a bucket and put it on the free list.  If that
1406a553d4b8SJeff Roberson 	 * works we'll restart the allocation from the begining.
1407a553d4b8SJeff Roberson 	 *
1408a553d4b8SJeff Roberson 	 * Try this zone's free list first so we don't allocate extra buckets.
14098355f576SJeff Roberson 	 */
1410a553d4b8SJeff Roberson 
1411a553d4b8SJeff Roberson 	if ((bucket = LIST_FIRST(&zone->uz_free_bucket)) != NULL)
1412a553d4b8SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
1413a553d4b8SJeff Roberson 
1414a553d4b8SJeff Roberson 	/* Now we no longer need the zone lock. */
1415a553d4b8SJeff Roberson 	ZONE_UNLOCK(zone);
1416a553d4b8SJeff Roberson 
1417a553d4b8SJeff Roberson 	if (bucket == NULL)
1418a553d4b8SJeff Roberson 		bucket = uma_zalloc_internal(bucketzone,
14192cc35ff9SJeff Roberson 		    NULL, flags, NULL);
1420a553d4b8SJeff Roberson 
1421a553d4b8SJeff Roberson 	if (bucket != NULL) {
1422a553d4b8SJeff Roberson #ifdef INVARIANTS
1423a553d4b8SJeff Roberson 		bzero(bucket, bucketzone->uz_size);
1424a553d4b8SJeff Roberson #endif
1425a553d4b8SJeff Roberson 		bucket->ub_ptr = -1;
1426a553d4b8SJeff Roberson 
14272cc35ff9SJeff Roberson 		if (uma_zalloc_internal(zone, udata, flags, bucket))
1428a553d4b8SJeff Roberson 			goto zalloc_restart;
1429a553d4b8SJeff Roberson 		else
1430a553d4b8SJeff Roberson 			uma_zfree_internal(bucketzone, bucket, NULL, 0);
1431a553d4b8SJeff Roberson 	}
1432a553d4b8SJeff Roberson 	/*
1433a553d4b8SJeff Roberson 	 * We may not get a bucket if we recurse, so
1434a553d4b8SJeff Roberson 	 * return an actual item.
1435a553d4b8SJeff Roberson 	 */
1436a553d4b8SJeff Roberson #ifdef UMA_DEBUG
1437a553d4b8SJeff Roberson 	printf("uma_zalloc_arg: Bucketzone returned NULL\n");
1438a553d4b8SJeff Roberson #endif
1439a553d4b8SJeff Roberson 
14402cc35ff9SJeff Roberson 	return (uma_zalloc_internal(zone, udata, flags, NULL));
14418355f576SJeff Roberson }
14428355f576SJeff Roberson 
14438355f576SJeff Roberson /*
14448355f576SJeff Roberson  * Allocates an item for an internal zone OR fills a bucket
14458355f576SJeff Roberson  *
14468355f576SJeff Roberson  * Arguments
14478355f576SJeff Roberson  *	zone   The zone to alloc for.
14488355f576SJeff Roberson  *	udata  The data to be passed to the constructor.
14492cc35ff9SJeff Roberson  *	flags  M_WAITOK, M_NOWAIT, M_ZERO.
1450a553d4b8SJeff Roberson  *	bucket The bucket to fill or NULL
14518355f576SJeff Roberson  *
14528355f576SJeff Roberson  * Returns
14538355f576SJeff Roberson  *	NULL if there is no memory and M_NOWAIT is set
14548355f576SJeff Roberson  *	An item if called on an interal zone
14558355f576SJeff Roberson  *	Non NULL if called to fill a bucket and it was successful.
14568355f576SJeff Roberson  *
14578355f576SJeff Roberson  * Discussion:
14588355f576SJeff Roberson  *	This was much cleaner before it had to do per cpu caches.  It is
14598355f576SJeff Roberson  *	complicated now because it has to handle the simple internal case, and
1460a553d4b8SJeff Roberson  *	the more involved bucket filling and allocation.
14618355f576SJeff Roberson  */
14628355f576SJeff Roberson 
14638355f576SJeff Roberson static void *
14642cc35ff9SJeff Roberson uma_zalloc_internal(uma_zone_t zone, void *udata, int flags, uma_bucket_t bucket)
14658355f576SJeff Roberson {
14668355f576SJeff Roberson 	uma_slab_t slab;
14678355f576SJeff Roberson 	u_int8_t freei;
14688355f576SJeff Roberson 	void *item;
14698355f576SJeff Roberson 
14708355f576SJeff Roberson 	item = NULL;
14718355f576SJeff Roberson 
14728355f576SJeff Roberson 	/*
1473a553d4b8SJeff Roberson 	 * This is to stop us from allocating per cpu buckets while we're
1474a553d4b8SJeff Roberson 	 * running out of UMA_BOOT_PAGES.  Otherwise, we would exhaust the
1475a553d4b8SJeff Roberson 	 * boot pages.
14768355f576SJeff Roberson 	 */
14778355f576SJeff Roberson 
147886bbae32SJeff Roberson 	if (bucketdisable && zone == bucketzone)
14798355f576SJeff Roberson 		return (NULL);
14808355f576SJeff Roberson 
14818355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
14828355f576SJeff Roberson 	printf("INTERNAL: Allocating one item from %s(%p)\n", zone->uz_name, zone);
14838355f576SJeff Roberson #endif
14848355f576SJeff Roberson 	ZONE_LOCK(zone);
14858355f576SJeff Roberson 
14868355f576SJeff Roberson 	/*
1487a553d4b8SJeff Roberson 	 * This code is here to limit the number of simultaneous bucket fills
1488a553d4b8SJeff Roberson 	 * for any given zone to the number of per cpu caches in this zone. This
1489a553d4b8SJeff Roberson 	 * is done so that we don't allocate more memory than we really need.
14908355f576SJeff Roberson 	 */
1491a553d4b8SJeff Roberson 
1492a553d4b8SJeff Roberson 	if (bucket) {
1493a553d4b8SJeff Roberson #ifdef SMP
1494bce97791SJeff Roberson 		if (zone->uz_fills >= mp_ncpus) {
1495a553d4b8SJeff Roberson #else
1496bce97791SJeff Roberson 		if (zone->uz_fills > 1) {
14978355f576SJeff Roberson #endif
1498bce97791SJeff Roberson 			ZONE_UNLOCK(zone);
1499a553d4b8SJeff Roberson 			return (NULL);
1500bce97791SJeff Roberson 		}
1501a553d4b8SJeff Roberson 
1502a553d4b8SJeff Roberson 		zone->uz_fills++;
15038355f576SJeff Roberson 	}
15048355f576SJeff Roberson 
15058355f576SJeff Roberson new_slab:
15068355f576SJeff Roberson 
15078355f576SJeff Roberson 	/* Find a slab with some space */
15088355f576SJeff Roberson 	if (zone->uz_free) {
15098355f576SJeff Roberson 		if (!LIST_EMPTY(&zone->uz_part_slab)) {
15108355f576SJeff Roberson 			slab = LIST_FIRST(&zone->uz_part_slab);
15118355f576SJeff Roberson 		} else {
15128355f576SJeff Roberson 			slab = LIST_FIRST(&zone->uz_free_slab);
15138355f576SJeff Roberson 			LIST_REMOVE(slab, us_link);
15148355f576SJeff Roberson 			LIST_INSERT_HEAD(&zone->uz_part_slab, slab, us_link);
15158355f576SJeff Roberson 		}
15168355f576SJeff Roberson 	} else {
15178355f576SJeff Roberson 		/*
15188355f576SJeff Roberson 		 * This is to prevent us from recursively trying to allocate
15198355f576SJeff Roberson 		 * buckets.  The problem is that if an allocation forces us to
15208355f576SJeff Roberson 		 * grab a new bucket we will call page_alloc, which will go off
15218355f576SJeff Roberson 		 * and cause the vm to allocate vm_map_entries.  If we need new
15228355f576SJeff Roberson 		 * buckets there too we will recurse in kmem_alloc and bad
15238355f576SJeff Roberson 		 * things happen.  So instead we return a NULL bucket, and make
15248355f576SJeff Roberson 		 * the code that allocates buckets smart enough to deal with it			 */
15258355f576SJeff Roberson 		if (zone == bucketzone && zone->uz_recurse != 0) {
15268355f576SJeff Roberson 			ZONE_UNLOCK(zone);
15278355f576SJeff Roberson 			return (NULL);
15288355f576SJeff Roberson 		}
1529af7f9b97SJeff Roberson 		while (zone->uz_maxpages &&
1530af7f9b97SJeff Roberson 		    zone->uz_pages >= zone->uz_maxpages) {
1531af7f9b97SJeff Roberson 			zone->uz_flags |= UMA_ZFLAG_FULL;
1532af7f9b97SJeff Roberson 
15332cc35ff9SJeff Roberson 			if (flags & M_WAITOK)
1534af7f9b97SJeff Roberson 				msleep(zone, &zone->uz_lock, PVM, "zonelimit", 0);
1535af7f9b97SJeff Roberson 			else
1536af7f9b97SJeff Roberson 				goto alloc_fail;
1537af7f9b97SJeff Roberson 
1538af7f9b97SJeff Roberson 			goto new_slab;
1539af7f9b97SJeff Roberson 		}
1540af7f9b97SJeff Roberson 
15418355f576SJeff Roberson 		zone->uz_recurse++;
15422cc35ff9SJeff Roberson 		slab = slab_zalloc(zone, flags);
15438355f576SJeff Roberson 		zone->uz_recurse--;
15448355f576SJeff Roberson 		/*
1545af7f9b97SJeff Roberson 		 * We might not have been able to get a slab but another cpu
1546af7f9b97SJeff Roberson 		 * could have while we were unlocked.  If we did get a slab put
1547af7f9b97SJeff Roberson 		 * it on the partially used slab list.  If not check the free
1548af7f9b97SJeff Roberson 		 * count and restart or fail accordingly.
15498355f576SJeff Roberson 		 */
1550af7f9b97SJeff Roberson 		if (slab)
1551af7f9b97SJeff Roberson 			LIST_INSERT_HEAD(&zone->uz_part_slab, slab, us_link);
1552af7f9b97SJeff Roberson 		else if (zone->uz_free == 0)
1553af7f9b97SJeff Roberson 			goto alloc_fail;
1554a553d4b8SJeff Roberson 		else
1555af7f9b97SJeff Roberson 			goto new_slab;
15568355f576SJeff Roberson 	}
1557a553d4b8SJeff Roberson 	/*
1558a553d4b8SJeff Roberson 	 * If this is our first time though put this guy on the list.
1559a553d4b8SJeff Roberson 	 */
1560a553d4b8SJeff Roberson 	if (bucket != NULL && bucket->ub_ptr == -1)
1561a553d4b8SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_full_bucket,
1562a553d4b8SJeff Roberson 		    bucket, ub_link);
1563a553d4b8SJeff Roberson 
15648355f576SJeff Roberson 
15658355f576SJeff Roberson 	while (slab->us_freecount) {
15668355f576SJeff Roberson 		freei = slab->us_firstfree;
15678355f576SJeff Roberson 		slab->us_firstfree = slab->us_freelist[freei];
1568639c9550SJeff Roberson 
15698355f576SJeff Roberson 		item = slab->us_data + (zone->uz_rsize * freei);
15708355f576SJeff Roberson 
1571639c9550SJeff Roberson 		slab->us_freecount--;
1572639c9550SJeff Roberson 		zone->uz_free--;
1573639c9550SJeff Roberson #ifdef INVARIANTS
1574639c9550SJeff Roberson 		uma_dbg_alloc(zone, slab, item);
1575639c9550SJeff Roberson #endif
1576a553d4b8SJeff Roberson 		if (bucket == NULL) {
15778355f576SJeff Roberson 			zone->uz_allocs++;
15788355f576SJeff Roberson 			break;
15798355f576SJeff Roberson 		}
15808355f576SJeff Roberson 		bucket->ub_bucket[++bucket->ub_ptr] = item;
15818355f576SJeff Roberson 
15828355f576SJeff Roberson 		/* Don't overfill the bucket! */
1583a553d4b8SJeff Roberson 		if (bucket->ub_ptr == zone->uz_count)
15848355f576SJeff Roberson 			break;
15858355f576SJeff Roberson 	}
15868355f576SJeff Roberson 
15878355f576SJeff Roberson 	/* Move this slab to the full list */
15888355f576SJeff Roberson 	if (slab->us_freecount == 0) {
15898355f576SJeff Roberson 		LIST_REMOVE(slab, us_link);
15908355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_full_slab, slab, us_link);
15918355f576SJeff Roberson 	}
15928355f576SJeff Roberson 
1593a553d4b8SJeff Roberson 	if (bucket != NULL) {
15948355f576SJeff Roberson 		/* Try to keep the buckets totally full, but don't block */
1595a553d4b8SJeff Roberson 		if (bucket->ub_ptr < zone->uz_count) {
15962cc35ff9SJeff Roberson 			flags |= M_NOWAIT;
15972cc35ff9SJeff Roberson 			flags &= ~M_WAITOK;
15988355f576SJeff Roberson 			goto new_slab;
1599a553d4b8SJeff Roberson 		} else
1600a553d4b8SJeff Roberson 			zone->uz_fills--;
16018355f576SJeff Roberson 	}
16028355f576SJeff Roberson 
16038355f576SJeff Roberson 	ZONE_UNLOCK(zone);
16048355f576SJeff Roberson 
16058355f576SJeff Roberson 	/* Only construct at this time if we're not filling a bucket */
16062cc35ff9SJeff Roberson 	if (bucket == NULL && zone->uz_ctor != NULL)  {
16078355f576SJeff Roberson 		zone->uz_ctor(item, zone->uz_size, udata);
16082cc35ff9SJeff Roberson 		if (flags & M_ZERO)
16092cc35ff9SJeff Roberson 			bzero(item, zone->uz_size);
16102cc35ff9SJeff Roberson 	}
16118355f576SJeff Roberson 
16128355f576SJeff Roberson 	return (item);
1613af7f9b97SJeff Roberson 
1614af7f9b97SJeff Roberson alloc_fail:
1615af7f9b97SJeff Roberson 	if (bucket != NULL)
1616af7f9b97SJeff Roberson 		zone->uz_fills--;
1617af7f9b97SJeff Roberson 	ZONE_UNLOCK(zone);
1618af7f9b97SJeff Roberson 
1619af7f9b97SJeff Roberson 	if (bucket != NULL && bucket->ub_ptr != -1)
1620af7f9b97SJeff Roberson 		return (bucket);
1621af7f9b97SJeff Roberson 
1622af7f9b97SJeff Roberson 	return (NULL);
16238355f576SJeff Roberson }
16248355f576SJeff Roberson 
16258355f576SJeff Roberson /* See uma.h */
16268355f576SJeff Roberson void
16278355f576SJeff Roberson uma_zfree_arg(uma_zone_t zone, void *item, void *udata)
16288355f576SJeff Roberson {
16298355f576SJeff Roberson 	uma_cache_t cache;
16308355f576SJeff Roberson 	uma_bucket_t bucket;
16318355f576SJeff Roberson 	int cpu;
16328355f576SJeff Roberson 
16338355f576SJeff Roberson 	/* This is the fast path free */
16348355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC_1
16358355f576SJeff Roberson 	printf("Freeing item %p to %s(%p)\n", item, zone->uz_name, zone);
16368355f576SJeff Roberson #endif
1637af7f9b97SJeff Roberson 	/*
1638af7f9b97SJeff Roberson 	 * The race here is acceptable.  If we miss it we'll just have to wait
1639af7f9b97SJeff Roberson 	 * a little longer for the limits to be reset.
1640af7f9b97SJeff Roberson 	 */
1641af7f9b97SJeff Roberson 
1642af7f9b97SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_FULL)
1643af7f9b97SJeff Roberson 		goto zfree_internal;
1644af7f9b97SJeff Roberson 
1645a553d4b8SJeff Roberson zfree_restart:
16468355f576SJeff Roberson 	cpu = PCPU_GET(cpuid);
16478355f576SJeff Roberson 	CPU_LOCK(zone, cpu);
16488355f576SJeff Roberson 	cache = &zone->uz_cpu[cpu];
16498355f576SJeff Roberson 
16508355f576SJeff Roberson zfree_start:
16518355f576SJeff Roberson 	bucket = cache->uc_freebucket;
16528355f576SJeff Roberson 
16538355f576SJeff Roberson 	if (bucket) {
1654a553d4b8SJeff Roberson 		/*
1655a553d4b8SJeff Roberson 		 * Do we have room in our bucket? It is OK for this uz count
1656a553d4b8SJeff Roberson 		 * check to be slightly out of sync.
1657a553d4b8SJeff Roberson 		 */
1658a553d4b8SJeff Roberson 
1659a553d4b8SJeff Roberson 		if (bucket->ub_ptr < zone->uz_count) {
16608355f576SJeff Roberson 			bucket->ub_ptr++;
16618355f576SJeff Roberson 			KASSERT(bucket->ub_bucket[bucket->ub_ptr] == NULL,
16628355f576SJeff Roberson 			    ("uma_zfree: Freeing to non free bucket index."));
16638355f576SJeff Roberson 			bucket->ub_bucket[bucket->ub_ptr] = item;
16648355f576SJeff Roberson 			if (zone->uz_dtor)
16658355f576SJeff Roberson 				zone->uz_dtor(item, zone->uz_size, udata);
1666b9ba8931SJeff Roberson #ifdef INVARIANTS
1667b9ba8931SJeff Roberson 			if (zone->uz_flags & UMA_ZFLAG_MALLOC)
1668b9ba8931SJeff Roberson 				uma_dbg_free(zone, udata, item);
1669b9ba8931SJeff Roberson 			else
1670b9ba8931SJeff Roberson 				uma_dbg_free(zone, NULL, item);
1671b9ba8931SJeff Roberson #endif
1672605cbd6aSJeff Roberson 			CPU_UNLOCK(zone, cpu);
16738355f576SJeff Roberson 			return;
16748355f576SJeff Roberson 		} else if (cache->uc_allocbucket) {
16758355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
16768355f576SJeff Roberson 			printf("uma_zfree: Swapping buckets.\n");
16778355f576SJeff Roberson #endif
16788355f576SJeff Roberson 			/*
16798355f576SJeff Roberson 			 * We have run out of space in our freebucket.
16808355f576SJeff Roberson 			 * See if we can switch with our alloc bucket.
16818355f576SJeff Roberson 			 */
16828355f576SJeff Roberson 			if (cache->uc_allocbucket->ub_ptr <
16838355f576SJeff Roberson 			    cache->uc_freebucket->ub_ptr) {
16848355f576SJeff Roberson 				uma_bucket_t swap;
16858355f576SJeff Roberson 
16868355f576SJeff Roberson 				swap = cache->uc_freebucket;
16878355f576SJeff Roberson 				cache->uc_freebucket = cache->uc_allocbucket;
16888355f576SJeff Roberson 				cache->uc_allocbucket = swap;
16898355f576SJeff Roberson 
16908355f576SJeff Roberson 				goto zfree_start;
16918355f576SJeff Roberson 			}
16928355f576SJeff Roberson 		}
16938355f576SJeff Roberson 	}
16948355f576SJeff Roberson 
16958355f576SJeff Roberson 	/*
1696a553d4b8SJeff Roberson 	 * We can get here for two reasons:
16978355f576SJeff Roberson 	 *
16988355f576SJeff Roberson 	 * 1) The buckets are NULL
1699a553d4b8SJeff Roberson 	 * 2) The alloc and free buckets are both somewhat full.
17008355f576SJeff Roberson 	 *
17018355f576SJeff Roberson 	 */
17028355f576SJeff Roberson 
17038355f576SJeff Roberson 	ZONE_LOCK(zone);
17048355f576SJeff Roberson 
17058355f576SJeff Roberson 	bucket = cache->uc_freebucket;
17068355f576SJeff Roberson 	cache->uc_freebucket = NULL;
17078355f576SJeff Roberson 
17088355f576SJeff Roberson 	/* Can we throw this on the zone full list? */
17098355f576SJeff Roberson 	if (bucket != NULL) {
17108355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
17118355f576SJeff Roberson 		printf("uma_zfree: Putting old bucket on the free list.\n");
17128355f576SJeff Roberson #endif
17138355f576SJeff Roberson 		/* ub_ptr is pointing to the last free item */
17148355f576SJeff Roberson 		KASSERT(bucket->ub_ptr != -1,
17158355f576SJeff Roberson 		    ("uma_zfree: Attempting to insert an empty bucket onto the full list.\n"));
17168355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_full_bucket,
17178355f576SJeff Roberson 		    bucket, ub_link);
17188355f576SJeff Roberson 	}
1719a553d4b8SJeff Roberson 	if ((bucket = LIST_FIRST(&zone->uz_free_bucket)) != NULL) {
1720a553d4b8SJeff Roberson 		LIST_REMOVE(bucket, ub_link);
1721a553d4b8SJeff Roberson 		ZONE_UNLOCK(zone);
1722a553d4b8SJeff Roberson 		cache->uc_freebucket = bucket;
1723a553d4b8SJeff Roberson 		goto zfree_start;
1724a553d4b8SJeff Roberson 	}
1725a553d4b8SJeff Roberson 	/* We're done with this CPU now */
1726a553d4b8SJeff Roberson 	CPU_UNLOCK(zone, cpu);
1727a553d4b8SJeff Roberson 
1728a553d4b8SJeff Roberson 	/* And the zone.. */
1729a553d4b8SJeff Roberson 	ZONE_UNLOCK(zone);
1730a553d4b8SJeff Roberson 
17318355f576SJeff Roberson #ifdef UMA_DEBUG_ALLOC
17328355f576SJeff Roberson 	printf("uma_zfree: Allocating new free bucket.\n");
17338355f576SJeff Roberson #endif
17348355f576SJeff Roberson 	bucket = uma_zalloc_internal(bucketzone,
1735a553d4b8SJeff Roberson 	    NULL, M_NOWAIT, NULL);
17368355f576SJeff Roberson 	if (bucket) {
17378355f576SJeff Roberson #ifdef INVARIANTS
17388355f576SJeff Roberson 		bzero(bucket, bucketzone->uz_size);
17398355f576SJeff Roberson #endif
17408355f576SJeff Roberson 		bucket->ub_ptr = -1;
1741a553d4b8SJeff Roberson 		ZONE_LOCK(zone);
1742a553d4b8SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_free_bucket,
1743a553d4b8SJeff Roberson 		    bucket, ub_link);
17448355f576SJeff Roberson 		ZONE_UNLOCK(zone);
1745a553d4b8SJeff Roberson 		goto zfree_restart;
17468355f576SJeff Roberson 	}
17478355f576SJeff Roberson 
1748a553d4b8SJeff Roberson 	/*
1749a553d4b8SJeff Roberson 	 * If nothing else caught this, we'll just do an internal free.
1750a553d4b8SJeff Roberson 	 */
17518355f576SJeff Roberson 
1752af7f9b97SJeff Roberson zfree_internal:
1753af7f9b97SJeff Roberson 
17548355f576SJeff Roberson 	uma_zfree_internal(zone, item, udata, 0);
17558355f576SJeff Roberson 
17568355f576SJeff Roberson 	return;
17578355f576SJeff Roberson 
17588355f576SJeff Roberson }
17598355f576SJeff Roberson 
17608355f576SJeff Roberson /*
17618355f576SJeff Roberson  * Frees an item to an INTERNAL zone or allocates a free bucket
17628355f576SJeff Roberson  *
17638355f576SJeff Roberson  * Arguments:
17648355f576SJeff Roberson  *	zone   The zone to free to
17658355f576SJeff Roberson  *	item   The item we're freeing
17668355f576SJeff Roberson  *	udata  User supplied data for the dtor
17678355f576SJeff Roberson  *	skip   Skip the dtor, it was done in uma_zfree_arg
17688355f576SJeff Roberson  */
17698355f576SJeff Roberson 
17708355f576SJeff Roberson static void
17718355f576SJeff Roberson uma_zfree_internal(uma_zone_t zone, void *item, void *udata, int skip)
17728355f576SJeff Roberson {
17738355f576SJeff Roberson 	uma_slab_t slab;
17748355f576SJeff Roberson 	u_int8_t *mem;
17758355f576SJeff Roberson 	u_int8_t freei;
17768355f576SJeff Roberson 
17778355f576SJeff Roberson 	ZONE_LOCK(zone);
17788355f576SJeff Roberson 
17798355f576SJeff Roberson 	if (!(zone->uz_flags & UMA_ZFLAG_MALLOC)) {
17808355f576SJeff Roberson 		mem = (u_int8_t *)((unsigned long)item & (~UMA_SLAB_MASK));
17818355f576SJeff Roberson 		if (zone->uz_flags & UMA_ZFLAG_OFFPAGE)
17828355f576SJeff Roberson 			slab = hash_sfind(&zone->uz_hash, mem);
17838355f576SJeff Roberson 		else {
17848355f576SJeff Roberson 			mem += zone->uz_pgoff;
17858355f576SJeff Roberson 			slab = (uma_slab_t)mem;
17868355f576SJeff Roberson 		}
17878355f576SJeff Roberson 	} else {
17888355f576SJeff Roberson 		slab = (uma_slab_t)udata;
17898355f576SJeff Roberson 	}
17908355f576SJeff Roberson 
17918355f576SJeff Roberson 	/* Do we need to remove from any lists? */
17928355f576SJeff Roberson 	if (slab->us_freecount+1 == zone->uz_ipers) {
17938355f576SJeff Roberson 		LIST_REMOVE(slab, us_link);
17948355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_free_slab, slab, us_link);
17958355f576SJeff Roberson 	} else if (slab->us_freecount == 0) {
17968355f576SJeff Roberson 		LIST_REMOVE(slab, us_link);
17978355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_part_slab, slab, us_link);
17988355f576SJeff Roberson 	}
17998355f576SJeff Roberson 
18008355f576SJeff Roberson 	/* Slab management stuff */
18018355f576SJeff Roberson 	freei = ((unsigned long)item - (unsigned long)slab->us_data)
18028355f576SJeff Roberson 		/ zone->uz_rsize;
18038355f576SJeff Roberson 
1804639c9550SJeff Roberson #ifdef INVARIANTS
1805639c9550SJeff Roberson 	if (!skip)
1806639c9550SJeff Roberson 		uma_dbg_free(zone, slab, item);
18078355f576SJeff Roberson #endif
1808639c9550SJeff Roberson 
18098355f576SJeff Roberson 	slab->us_freelist[freei] = slab->us_firstfree;
18108355f576SJeff Roberson 	slab->us_firstfree = freei;
18118355f576SJeff Roberson 	slab->us_freecount++;
18128355f576SJeff Roberson 
18138355f576SJeff Roberson 	/* Zone statistics */
18148355f576SJeff Roberson 	zone->uz_free++;
18158355f576SJeff Roberson 
18168355f576SJeff Roberson 	if (!skip && zone->uz_dtor)
18178355f576SJeff Roberson 		zone->uz_dtor(item, zone->uz_size, udata);
1818605cbd6aSJeff Roberson 
1819af7f9b97SJeff Roberson 	if (zone->uz_flags & UMA_ZFLAG_FULL) {
1820af7f9b97SJeff Roberson 		if (zone->uz_pages < zone->uz_maxpages)
1821af7f9b97SJeff Roberson 			zone->uz_flags &= ~UMA_ZFLAG_FULL;
1822af7f9b97SJeff Roberson 
1823af7f9b97SJeff Roberson 		/* We can handle one more allocation */
1824af7f9b97SJeff Roberson 		wakeup_one(&zone);
1825af7f9b97SJeff Roberson 	}
1826af7f9b97SJeff Roberson 
1827605cbd6aSJeff Roberson 	ZONE_UNLOCK(zone);
18288355f576SJeff Roberson }
18298355f576SJeff Roberson 
18308355f576SJeff Roberson /* See uma.h */
18318355f576SJeff Roberson void
1832736ee590SJeff Roberson uma_zone_set_max(uma_zone_t zone, int nitems)
1833736ee590SJeff Roberson {
1834736ee590SJeff Roberson 	ZONE_LOCK(zone);
1835736ee590SJeff Roberson 	if (zone->uz_ppera > 1)
1836af7f9b97SJeff Roberson 		zone->uz_maxpages = nitems * zone->uz_ppera;
1837736ee590SJeff Roberson 	else
1838736ee590SJeff Roberson 		zone->uz_maxpages = nitems / zone->uz_ipers;
183928bc4419SJeff Roberson 
1840d4d6aee5SAndrew R. Reiter 	if (zone->uz_maxpages * zone->uz_ipers < nitems)
1841d4d6aee5SAndrew R. Reiter 		zone->uz_maxpages++;
184228bc4419SJeff Roberson 
1843736ee590SJeff Roberson 	ZONE_UNLOCK(zone);
1844736ee590SJeff Roberson }
1845736ee590SJeff Roberson 
1846736ee590SJeff Roberson /* See uma.h */
1847736ee590SJeff Roberson void
18488355f576SJeff Roberson uma_zone_set_freef(uma_zone_t zone, uma_free freef)
18498355f576SJeff Roberson {
18508355f576SJeff Roberson 	ZONE_LOCK(zone);
18518355f576SJeff Roberson 
18528355f576SJeff Roberson 	zone->uz_freef = freef;
18538355f576SJeff Roberson 
18548355f576SJeff Roberson 	ZONE_UNLOCK(zone);
18558355f576SJeff Roberson }
18568355f576SJeff Roberson 
18578355f576SJeff Roberson /* See uma.h */
18588355f576SJeff Roberson void
18598355f576SJeff Roberson uma_zone_set_allocf(uma_zone_t zone, uma_alloc allocf)
18608355f576SJeff Roberson {
18618355f576SJeff Roberson 	ZONE_LOCK(zone);
18628355f576SJeff Roberson 
18638355f576SJeff Roberson 	zone->uz_flags |= UMA_ZFLAG_PRIVALLOC;
18648355f576SJeff Roberson 	zone->uz_allocf = allocf;
18658355f576SJeff Roberson 
18668355f576SJeff Roberson 	ZONE_UNLOCK(zone);
18678355f576SJeff Roberson }
18688355f576SJeff Roberson 
18698355f576SJeff Roberson /* See uma.h */
18708355f576SJeff Roberson int
18718355f576SJeff Roberson uma_zone_set_obj(uma_zone_t zone, struct vm_object *obj, int count)
18728355f576SJeff Roberson {
18738355f576SJeff Roberson 	int pages;
18748355f576SJeff Roberson 	vm_offset_t kva;
18758355f576SJeff Roberson 
18768355f576SJeff Roberson 	mtx_lock(&Giant);
18778355f576SJeff Roberson 
18788355f576SJeff Roberson 	pages = count / zone->uz_ipers;
18798355f576SJeff Roberson 
18808355f576SJeff Roberson 	if (pages * zone->uz_ipers < count)
18818355f576SJeff Roberson 		pages++;
1882a553d4b8SJeff Roberson 
18838355f576SJeff Roberson 	kva = kmem_alloc_pageable(kernel_map, pages * UMA_SLAB_SIZE);
18848355f576SJeff Roberson 
1885a553d4b8SJeff Roberson 	if (kva == 0) {
1886a553d4b8SJeff Roberson 		mtx_unlock(&Giant);
18878355f576SJeff Roberson 		return (0);
18888355f576SJeff Roberson 	}
18898355f576SJeff Roberson 
18908355f576SJeff Roberson 
1891a553d4b8SJeff Roberson 	if (obj == NULL)
1892a553d4b8SJeff Roberson 		obj = vm_object_allocate(OBJT_DEFAULT,
1893c7173f58SJeff Roberson 		    pages);
18948355f576SJeff Roberson 	else
18958355f576SJeff Roberson 		_vm_object_allocate(OBJT_DEFAULT,
1896c7173f58SJeff Roberson 		    pages, obj);
1897a553d4b8SJeff Roberson 
1898a553d4b8SJeff Roberson 	ZONE_LOCK(zone);
1899a553d4b8SJeff Roberson 	zone->uz_kva = kva;
1900a553d4b8SJeff Roberson 	zone->uz_obj = obj;
1901a553d4b8SJeff Roberson 	zone->uz_maxpages = pages;
19028355f576SJeff Roberson 
19038355f576SJeff Roberson 	zone->uz_allocf = obj_alloc;
19048355f576SJeff Roberson 	zone->uz_flags |= UMA_ZFLAG_NOFREE | UMA_ZFLAG_PRIVALLOC;
19058355f576SJeff Roberson 
19068355f576SJeff Roberson 	ZONE_UNLOCK(zone);
1907a553d4b8SJeff Roberson 	mtx_unlock(&Giant);
19088355f576SJeff Roberson 
19098355f576SJeff Roberson 	return (1);
19108355f576SJeff Roberson }
19118355f576SJeff Roberson 
19128355f576SJeff Roberson /* See uma.h */
19138355f576SJeff Roberson void
19148355f576SJeff Roberson uma_prealloc(uma_zone_t zone, int items)
19158355f576SJeff Roberson {
19168355f576SJeff Roberson 	int slabs;
19178355f576SJeff Roberson 	uma_slab_t slab;
19188355f576SJeff Roberson 
19198355f576SJeff Roberson 	ZONE_LOCK(zone);
19208355f576SJeff Roberson 	slabs = items / zone->uz_ipers;
19218355f576SJeff Roberson 	if (slabs * zone->uz_ipers < items)
19228355f576SJeff Roberson 		slabs++;
19238355f576SJeff Roberson 
19248355f576SJeff Roberson 	while (slabs > 0) {
19258355f576SJeff Roberson 		slab = slab_zalloc(zone, M_WAITOK);
19268355f576SJeff Roberson 		LIST_INSERT_HEAD(&zone->uz_free_slab, slab, us_link);
19278355f576SJeff Roberson 		slabs--;
19288355f576SJeff Roberson 	}
19298355f576SJeff Roberson 	ZONE_UNLOCK(zone);
19308355f576SJeff Roberson }
19318355f576SJeff Roberson 
19328355f576SJeff Roberson /* See uma.h */
19338355f576SJeff Roberson void
19348355f576SJeff Roberson uma_reclaim(void)
19358355f576SJeff Roberson {
19368355f576SJeff Roberson 	/*
19378355f576SJeff Roberson 	 * You might think that the delay below would improve performance since
19388355f576SJeff Roberson 	 * the allocator will give away memory that it may ask for immediately.
19398355f576SJeff Roberson 	 * Really, it makes things worse, since cpu cycles are so much cheaper
19408355f576SJeff Roberson 	 * than disk activity.
19418355f576SJeff Roberson 	 */
19428355f576SJeff Roberson #if 0
19438355f576SJeff Roberson 	static struct timeval tv = {0};
19448355f576SJeff Roberson 	struct timeval now;
19458355f576SJeff Roberson 	getmicrouptime(&now);
19468355f576SJeff Roberson 	if (now.tv_sec > tv.tv_sec + 30)
19478355f576SJeff Roberson 		tv = now;
19488355f576SJeff Roberson 	else
19498355f576SJeff Roberson 		return;
19508355f576SJeff Roberson #endif
19518355f576SJeff Roberson #ifdef UMA_DEBUG
19528355f576SJeff Roberson 	printf("UMA: vm asked us to release pages!\n");
19538355f576SJeff Roberson #endif
195486bbae32SJeff Roberson 	bucket_enable();
19558355f576SJeff Roberson 	zone_foreach(zone_drain);
19568355f576SJeff Roberson 
19578355f576SJeff Roberson 	/*
19588355f576SJeff Roberson 	 * Some slabs may have been freed but this zone will be visited early
19598355f576SJeff Roberson 	 * we visit again so that we can free pages that are empty once other
19608355f576SJeff Roberson 	 * zones are drained.  We have to do the same for buckets.
19618355f576SJeff Roberson 	 */
19628355f576SJeff Roberson 	zone_drain(slabzone);
19638355f576SJeff Roberson 	zone_drain(bucketzone);
19648355f576SJeff Roberson }
19658355f576SJeff Roberson 
19668355f576SJeff Roberson void *
19678355f576SJeff Roberson uma_large_malloc(int size, int wait)
19688355f576SJeff Roberson {
19698355f576SJeff Roberson 	void *mem;
19708355f576SJeff Roberson 	uma_slab_t slab;
19718355f576SJeff Roberson 	u_int8_t flags;
19728355f576SJeff Roberson 
1973a553d4b8SJeff Roberson 	slab = uma_zalloc_internal(slabzone, NULL, wait, NULL);
19748355f576SJeff Roberson 	if (slab == NULL)
19758355f576SJeff Roberson 		return (NULL);
19768355f576SJeff Roberson 
19778355f576SJeff Roberson 	mem = page_alloc(NULL, size, &flags, wait);
19788355f576SJeff Roberson 	if (mem) {
19798355f576SJeff Roberson 		slab->us_data = mem;
19808355f576SJeff Roberson 		slab->us_flags = flags | UMA_SLAB_MALLOC;
19818355f576SJeff Roberson 		slab->us_size = size;
19825a34a9f0SJeff Roberson 		mtx_lock(&malloc_mtx);
19838355f576SJeff Roberson 		UMA_HASH_INSERT(mallochash, slab, mem);
19845a34a9f0SJeff Roberson 		mtx_unlock(&malloc_mtx);
19858355f576SJeff Roberson 	} else {
19868355f576SJeff Roberson 		uma_zfree_internal(slabzone, slab, NULL, 0);
19878355f576SJeff Roberson 	}
19888355f576SJeff Roberson 
19898355f576SJeff Roberson 
19908355f576SJeff Roberson 	return (mem);
19918355f576SJeff Roberson }
19928355f576SJeff Roberson 
19938355f576SJeff Roberson void
19948355f576SJeff Roberson uma_large_free(uma_slab_t slab)
19958355f576SJeff Roberson {
19965a34a9f0SJeff Roberson 	mtx_lock(&malloc_mtx);
19978355f576SJeff Roberson 	UMA_HASH_REMOVE(mallochash, slab, slab->us_data);
19985a34a9f0SJeff Roberson 	mtx_unlock(&malloc_mtx);
19998355f576SJeff Roberson 	page_free(slab->us_data, slab->us_size, slab->us_flags);
20008355f576SJeff Roberson 	uma_zfree_internal(slabzone, slab, NULL, 0);
20018355f576SJeff Roberson }
20028355f576SJeff Roberson 
20038355f576SJeff Roberson void
20048355f576SJeff Roberson uma_print_stats(void)
20058355f576SJeff Roberson {
20068355f576SJeff Roberson 	zone_foreach(uma_print_zone);
20078355f576SJeff Roberson }
20088355f576SJeff Roberson 
20098355f576SJeff Roberson void
20108355f576SJeff Roberson uma_print_zone(uma_zone_t zone)
20118355f576SJeff Roberson {
20128355f576SJeff Roberson 	printf("%s(%p) size %d(%d) flags %d ipers %d ppera %d out %d free %d\n",
20138355f576SJeff Roberson 	    zone->uz_name, zone, zone->uz_size, zone->uz_rsize, zone->uz_flags,
20148355f576SJeff Roberson 	    zone->uz_ipers, zone->uz_ppera,
20158355f576SJeff Roberson 	    (zone->uz_ipers * zone->uz_pages) - zone->uz_free, zone->uz_free);
20168355f576SJeff Roberson }
20178355f576SJeff Roberson 
20188355f576SJeff Roberson /*
20198355f576SJeff Roberson  * Sysctl handler for vm.zone
20208355f576SJeff Roberson  *
20218355f576SJeff Roberson  * stolen from vm_zone.c
20228355f576SJeff Roberson  */
20238355f576SJeff Roberson static int
20248355f576SJeff Roberson sysctl_vm_zone(SYSCTL_HANDLER_ARGS)
20258355f576SJeff Roberson {
20268355f576SJeff Roberson 	int error, len, cnt;
20278355f576SJeff Roberson 	const int linesize = 128;	/* conservative */
20288355f576SJeff Roberson 	int totalfree;
20298355f576SJeff Roberson 	char *tmpbuf, *offset;
20308355f576SJeff Roberson 	uma_zone_t z;
20318355f576SJeff Roberson 	char *p;
20328355f576SJeff Roberson 
20338355f576SJeff Roberson 	cnt = 0;
20340da47b2fSJeff Roberson 	mtx_lock(&uma_mtx);
20358355f576SJeff Roberson 	LIST_FOREACH(z, &uma_zones, uz_link)
20368355f576SJeff Roberson 		cnt++;
20370da47b2fSJeff Roberson 	mtx_unlock(&uma_mtx);
20388355f576SJeff Roberson 	MALLOC(tmpbuf, char *, (cnt == 0 ? 1 : cnt) * linesize,
20398355f576SJeff Roberson 			M_TEMP, M_WAITOK);
20408355f576SJeff Roberson 	len = snprintf(tmpbuf, linesize,
20418355f576SJeff Roberson 	    "\nITEM            SIZE     LIMIT     USED    FREE  REQUESTS\n\n");
20428355f576SJeff Roberson 	if (cnt == 0)
20438355f576SJeff Roberson 		tmpbuf[len - 1] = '\0';
20448355f576SJeff Roberson 	error = SYSCTL_OUT(req, tmpbuf, cnt == 0 ? len-1 : len);
20458355f576SJeff Roberson 	if (error || cnt == 0)
20468355f576SJeff Roberson 		goto out;
20478355f576SJeff Roberson 	offset = tmpbuf;
2048f4af24d5SJeff Roberson 	mtx_lock(&uma_mtx);
20498355f576SJeff Roberson 	LIST_FOREACH(z, &uma_zones, uz_link) {
20508355f576SJeff Roberson 		if (cnt == 0)	/* list may have changed size */
20518355f576SJeff Roberson 			break;
20528355f576SJeff Roberson 		ZONE_LOCK(z);
20538355f576SJeff Roberson 		totalfree = z->uz_free + z->uz_cachefree;
20548355f576SJeff Roberson 		len = snprintf(offset, linesize,
20558355f576SJeff Roberson 		    "%-12.12s  %6.6u, %8.8u, %6.6u, %6.6u, %8.8llu\n",
20568355f576SJeff Roberson 		    z->uz_name, z->uz_size,
20578355f576SJeff Roberson 		    z->uz_maxpages * z->uz_ipers,
20588355f576SJeff Roberson 		    (z->uz_ipers * (z->uz_pages / z->uz_ppera)) - totalfree,
20598355f576SJeff Roberson 		    totalfree,
20608355f576SJeff Roberson 		    (unsigned long long)z->uz_allocs);
20618355f576SJeff Roberson 		ZONE_UNLOCK(z);
20628355f576SJeff Roberson 		for (p = offset + 12; p > offset && *p == ' '; --p)
20638355f576SJeff Roberson 			/* nothing */ ;
20648355f576SJeff Roberson 		p[1] = ':';
20658355f576SJeff Roberson 		cnt--;
20668355f576SJeff Roberson 		offset += len;
20678355f576SJeff Roberson 	}
2068f4af24d5SJeff Roberson 	mtx_unlock(&uma_mtx);
20698355f576SJeff Roberson 	*offset++ = '\0';
20708355f576SJeff Roberson 	error = SYSCTL_OUT(req, tmpbuf, offset - tmpbuf);
20718355f576SJeff Roberson out:
20728355f576SJeff Roberson 	FREE(tmpbuf, M_TEMP);
20738355f576SJeff Roberson 	return (error);
20748355f576SJeff Roberson }
2075