xref: /freebsd/sys/kern/subr_blist.c (revision 677b542ea243380af64822b30e4ef5f7a6d978ee)
17090df5aSMatthew Dillon /*
27090df5aSMatthew Dillon  * BLIST.C -	Bitmap allocator/deallocator, using a radix tree with hinting
37090df5aSMatthew Dillon  *
47090df5aSMatthew Dillon  *	(c)Copyright 1998, Matthew Dillon.  Terms for use and redistribution
57090df5aSMatthew Dillon  *	are covered by the BSD Copyright as found in /usr/src/COPYRIGHT.
67090df5aSMatthew Dillon  *
77090df5aSMatthew Dillon  *	This module implements a general bitmap allocator/deallocator.  The
87090df5aSMatthew Dillon  *	allocator eats around 2 bits per 'block'.  The module does not
97090df5aSMatthew Dillon  *	try to interpret the meaning of a 'block' other then to return
107090df5aSMatthew Dillon  *	SWAPBLK_NONE on an allocation failure.
117090df5aSMatthew Dillon  *
127090df5aSMatthew Dillon  *	A radix tree is used to maintain the bitmap.  Two radix constants are
137090df5aSMatthew Dillon  *	involved:  One for the bitmaps contained in the leaf nodes (typically
147090df5aSMatthew Dillon  *	32), and one for the meta nodes (typically 16).  Both meta and leaf
157090df5aSMatthew Dillon  *	nodes have a hint field.  This field gives us a hint as to the largest
167090df5aSMatthew Dillon  *	free contiguous range of blocks under the node.  It may contain a
177090df5aSMatthew Dillon  *	value that is too high, but will never contain a value that is too
187090df5aSMatthew Dillon  *	low.  When the radix tree is searched, allocation failures in subtrees
197090df5aSMatthew Dillon  *	update the hint.
207090df5aSMatthew Dillon  *
217090df5aSMatthew Dillon  *	The radix tree also implements two collapsed states for meta nodes:
227090df5aSMatthew Dillon  *	the ALL-ALLOCATED state and the ALL-FREE state.  If a meta node is
237090df5aSMatthew Dillon  *	in either of these two states, all information contained underneath
247090df5aSMatthew Dillon  *	the node is considered stale.  These states are used to optimize
257090df5aSMatthew Dillon  *	allocation and freeing operations.
267090df5aSMatthew Dillon  *
277090df5aSMatthew Dillon  * 	The hinting greatly increases code efficiency for allocations while
287090df5aSMatthew Dillon  *	the general radix structure optimizes both allocations and frees.  The
297090df5aSMatthew Dillon  *	radix tree should be able to operate well no matter how much
307090df5aSMatthew Dillon  *	fragmentation there is and no matter how large a bitmap is used.
317090df5aSMatthew Dillon  *
327090df5aSMatthew Dillon  *	Unlike the rlist code, the blist code wires all necessary memory at
337090df5aSMatthew Dillon  *	creation time.  Neither allocations nor frees require interaction with
347090df5aSMatthew Dillon  *	the memory subsystem.  In contrast, the rlist code may allocate memory
357090df5aSMatthew Dillon  *	on an rlist_free() call.  The non-blocking features of the blist code
367090df5aSMatthew Dillon  *	are used to great advantage in the swap code (vm/nswap_pager.c).  The
377090df5aSMatthew Dillon  *	rlist code uses a little less overall memory then the blist code (but
387090df5aSMatthew Dillon  *	due to swap interleaving not all that much less), but the blist code
397090df5aSMatthew Dillon  *	scales much, much better.
407090df5aSMatthew Dillon  *
417090df5aSMatthew Dillon  *	LAYOUT: The radix tree is layed out recursively using a
427090df5aSMatthew Dillon  *	linear array.  Each meta node is immediately followed (layed out
437090df5aSMatthew Dillon  *	sequentially in memory) by BLIST_META_RADIX lower level nodes.  This
447090df5aSMatthew Dillon  *	is a recursive structure but one that can be easily scanned through
457090df5aSMatthew Dillon  *	a very simple 'skip' calculation.  In order to support large radixes,
467090df5aSMatthew Dillon  *	portions of the tree may reside outside our memory allocation.  We
477090df5aSMatthew Dillon  *	handle this with an early-termination optimization (when bighint is
487090df5aSMatthew Dillon  *	set to -1) on the scan.  The memory allocation is only large enough
497090df5aSMatthew Dillon  *	to cover the number of blocks requested at creation time even if it
507090df5aSMatthew Dillon  *	must be encompassed in larger root-node radix.
517090df5aSMatthew Dillon  *
527090df5aSMatthew Dillon  *	NOTE: the allocator cannot currently allocate more then
537090df5aSMatthew Dillon  *	BLIST_BMAP_RADIX blocks per call.  It will panic with 'allocation too
547090df5aSMatthew Dillon  *	large' if you try.  This is an area that could use improvement.  The
557090df5aSMatthew Dillon  *	radix is large enough that this restriction does not effect the swap
567090df5aSMatthew Dillon  *	system, though.  Currently only the allocation code is effected by
577090df5aSMatthew Dillon  *	this algorithmic unfeature.  The freeing code can handle arbitrary
587090df5aSMatthew Dillon  *	ranges.
597090df5aSMatthew Dillon  *
607090df5aSMatthew Dillon  *	This code can be compiled stand-alone for debugging.
617090df5aSMatthew Dillon  */
627090df5aSMatthew Dillon 
63677b542eSDavid E. O'Brien #include <sys/cdefs.h>
64677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$");
65677b542eSDavid E. O'Brien 
66c4473420SPeter Wemm #ifdef _KERNEL
677090df5aSMatthew Dillon 
687090df5aSMatthew Dillon #include <sys/param.h>
697090df5aSMatthew Dillon #include <sys/systm.h>
707090df5aSMatthew Dillon #include <sys/lock.h>
717090df5aSMatthew Dillon #include <sys/kernel.h>
727090df5aSMatthew Dillon #include <sys/blist.h>
737090df5aSMatthew Dillon #include <sys/malloc.h>
740cddd8f0SMatthew Dillon #include <sys/proc.h>
7523955314SAlfred Perlstein #include <sys/mutex.h>
767090df5aSMatthew Dillon #include <vm/vm.h>
777090df5aSMatthew Dillon #include <vm/vm_object.h>
787090df5aSMatthew Dillon #include <vm/vm_kern.h>
797090df5aSMatthew Dillon #include <vm/vm_extern.h>
807090df5aSMatthew Dillon #include <vm/vm_page.h>
817090df5aSMatthew Dillon 
827090df5aSMatthew Dillon #else
837090df5aSMatthew Dillon 
847090df5aSMatthew Dillon #ifndef BLIST_NO_DEBUG
857090df5aSMatthew Dillon #define BLIST_DEBUG
867090df5aSMatthew Dillon #endif
877090df5aSMatthew Dillon 
887090df5aSMatthew Dillon #define SWAPBLK_NONE ((daddr_t)-1)
897090df5aSMatthew Dillon 
907090df5aSMatthew Dillon #include <sys/types.h>
917090df5aSMatthew Dillon #include <stdio.h>
927090df5aSMatthew Dillon #include <string.h>
937090df5aSMatthew Dillon #include <stdlib.h>
947090df5aSMatthew Dillon #include <stdarg.h>
957090df5aSMatthew Dillon 
9692da00bbSMatthew Dillon #define malloc(a,b,c)	calloc(a, 1)
977090df5aSMatthew Dillon #define free(a,b)	free(a)
987090df5aSMatthew Dillon 
997090df5aSMatthew Dillon typedef unsigned int u_daddr_t;
1007090df5aSMatthew Dillon 
1017090df5aSMatthew Dillon #include <sys/blist.h>
1027090df5aSMatthew Dillon 
1037090df5aSMatthew Dillon void panic(const char *ctl, ...);
1047090df5aSMatthew Dillon 
1057090df5aSMatthew Dillon #endif
1067090df5aSMatthew Dillon 
1077090df5aSMatthew Dillon /*
1087090df5aSMatthew Dillon  * static support functions
1097090df5aSMatthew Dillon  */
1107090df5aSMatthew Dillon 
1117090df5aSMatthew Dillon static daddr_t blst_leaf_alloc(blmeta_t *scan, daddr_t blk, int count);
1127090df5aSMatthew Dillon static daddr_t blst_meta_alloc(blmeta_t *scan, daddr_t blk,
1137090df5aSMatthew Dillon 				daddr_t count, daddr_t radix, int skip);
1147090df5aSMatthew Dillon static void blst_leaf_free(blmeta_t *scan, daddr_t relblk, int count);
1157090df5aSMatthew Dillon static void blst_meta_free(blmeta_t *scan, daddr_t freeBlk, daddr_t count,
1167090df5aSMatthew Dillon 					daddr_t radix, int skip, daddr_t blk);
1177090df5aSMatthew Dillon static void blst_copy(blmeta_t *scan, daddr_t blk, daddr_t radix,
1187090df5aSMatthew Dillon 				daddr_t skip, blist_t dest, daddr_t count);
11992da00bbSMatthew Dillon static int blst_leaf_fill(blmeta_t *scan, daddr_t blk, int count);
12092da00bbSMatthew Dillon static int blst_meta_fill(blmeta_t *scan, daddr_t allocBlk, daddr_t count,
12192da00bbSMatthew Dillon 				daddr_t radix, int skip, daddr_t blk);
1227090df5aSMatthew Dillon static daddr_t	blst_radix_init(blmeta_t *scan, daddr_t radix,
1237090df5aSMatthew Dillon 						int skip, daddr_t count);
124c4473420SPeter Wemm #ifndef _KERNEL
1257090df5aSMatthew Dillon static void	blst_radix_print(blmeta_t *scan, daddr_t blk,
1267090df5aSMatthew Dillon 					daddr_t radix, int skip, int tab);
1277090df5aSMatthew Dillon #endif
1287090df5aSMatthew Dillon 
129c4473420SPeter Wemm #ifdef _KERNEL
1307090df5aSMatthew Dillon static MALLOC_DEFINE(M_SWAP, "SWAP", "Swap space");
1317090df5aSMatthew Dillon #endif
1327090df5aSMatthew Dillon 
1337090df5aSMatthew Dillon /*
1347090df5aSMatthew Dillon  * blist_create() - create a blist capable of handling up to the specified
1357090df5aSMatthew Dillon  *		    number of blocks
1367090df5aSMatthew Dillon  *
1377090df5aSMatthew Dillon  *	blocks must be greater then 0
1387090df5aSMatthew Dillon  *
1397090df5aSMatthew Dillon  *	The smallest blist consists of a single leaf node capable of
1407090df5aSMatthew Dillon  *	managing BLIST_BMAP_RADIX blocks.
1417090df5aSMatthew Dillon  */
1427090df5aSMatthew Dillon 
1437090df5aSMatthew Dillon blist_t
1447090df5aSMatthew Dillon blist_create(daddr_t blocks)
1457090df5aSMatthew Dillon {
1467090df5aSMatthew Dillon 	blist_t bl;
1477090df5aSMatthew Dillon 	int radix;
1487090df5aSMatthew Dillon 	int skip = 0;
1497090df5aSMatthew Dillon 
1507090df5aSMatthew Dillon 	/*
1517090df5aSMatthew Dillon 	 * Calculate radix and skip field used for scanning.
1527090df5aSMatthew Dillon 	 */
1537090df5aSMatthew Dillon 	radix = BLIST_BMAP_RADIX;
1547090df5aSMatthew Dillon 
1557090df5aSMatthew Dillon 	while (radix < blocks) {
15657e6d29bSMatthew Dillon 		radix *= BLIST_META_RADIX;
15757e6d29bSMatthew Dillon 		skip = (skip + 1) * BLIST_META_RADIX;
1587090df5aSMatthew Dillon 	}
1597090df5aSMatthew Dillon 
160a163d034SWarner Losh 	bl = malloc(sizeof(struct blist), M_SWAP, M_WAITOK | M_ZERO);
1617090df5aSMatthew Dillon 
1627090df5aSMatthew Dillon 	bl->bl_blocks = blocks;
1637090df5aSMatthew Dillon 	bl->bl_radix = radix;
1647090df5aSMatthew Dillon 	bl->bl_skip = skip;
1657090df5aSMatthew Dillon 	bl->bl_rootblks = 1 +
1667090df5aSMatthew Dillon 	    blst_radix_init(NULL, bl->bl_radix, bl->bl_skip, blocks);
167a163d034SWarner Losh 	bl->bl_root = malloc(sizeof(blmeta_t) * bl->bl_rootblks, M_SWAP, M_WAITOK);
1687090df5aSMatthew Dillon 
1697090df5aSMatthew Dillon #if defined(BLIST_DEBUG)
1707090df5aSMatthew Dillon 	printf(
17192da00bbSMatthew Dillon 		"BLIST representing %lld blocks (%lld MB of swap)"
17292da00bbSMatthew Dillon 		", requiring %lldK of ram\n",
17392da00bbSMatthew Dillon 		(long long)bl->bl_blocks,
17492da00bbSMatthew Dillon 		(long long)bl->bl_blocks * 4 / 1024,
17592da00bbSMatthew Dillon 		(long long)(bl->bl_rootblks * sizeof(blmeta_t) + 1023) / 1024
1767090df5aSMatthew Dillon 	);
17792da00bbSMatthew Dillon 	printf("BLIST raw radix tree contains %lld records\n",
17892da00bbSMatthew Dillon 	    (long long)bl->bl_rootblks);
1797090df5aSMatthew Dillon #endif
1807090df5aSMatthew Dillon 	blst_radix_init(bl->bl_root, bl->bl_radix, bl->bl_skip, blocks);
1817090df5aSMatthew Dillon 
1827090df5aSMatthew Dillon 	return(bl);
1837090df5aSMatthew Dillon }
1847090df5aSMatthew Dillon 
1857090df5aSMatthew Dillon void
1867090df5aSMatthew Dillon blist_destroy(blist_t bl)
1877090df5aSMatthew Dillon {
1887090df5aSMatthew Dillon 	free(bl->bl_root, M_SWAP);
1897090df5aSMatthew Dillon 	free(bl, M_SWAP);
1907090df5aSMatthew Dillon }
1917090df5aSMatthew Dillon 
1927090df5aSMatthew Dillon /*
1937090df5aSMatthew Dillon  * blist_alloc() - reserve space in the block bitmap.  Return the base
1947090df5aSMatthew Dillon  *		     of a contiguous region or SWAPBLK_NONE if space could
1957090df5aSMatthew Dillon  *		     not be allocated.
1967090df5aSMatthew Dillon  */
1977090df5aSMatthew Dillon 
1987090df5aSMatthew Dillon daddr_t
1997090df5aSMatthew Dillon blist_alloc(blist_t bl, daddr_t count)
2007090df5aSMatthew Dillon {
2017090df5aSMatthew Dillon 	daddr_t blk = SWAPBLK_NONE;
2027090df5aSMatthew Dillon 
2037090df5aSMatthew Dillon 	if (bl) {
2047090df5aSMatthew Dillon 		if (bl->bl_radix == BLIST_BMAP_RADIX)
2057090df5aSMatthew Dillon 			blk = blst_leaf_alloc(bl->bl_root, 0, count);
2067090df5aSMatthew Dillon 		else
2077090df5aSMatthew Dillon 			blk = blst_meta_alloc(bl->bl_root, 0, count, bl->bl_radix, bl->bl_skip);
2087090df5aSMatthew Dillon 		if (blk != SWAPBLK_NONE)
2097090df5aSMatthew Dillon 			bl->bl_free -= count;
2107090df5aSMatthew Dillon 	}
2117090df5aSMatthew Dillon 	return(blk);
2127090df5aSMatthew Dillon }
2137090df5aSMatthew Dillon 
2147090df5aSMatthew Dillon /*
2157090df5aSMatthew Dillon  * blist_free() -	free up space in the block bitmap.  Return the base
2167090df5aSMatthew Dillon  *		     	of a contiguous region.  Panic if an inconsistancy is
2177090df5aSMatthew Dillon  *			found.
2187090df5aSMatthew Dillon  */
2197090df5aSMatthew Dillon 
2207090df5aSMatthew Dillon void
2217090df5aSMatthew Dillon blist_free(blist_t bl, daddr_t blkno, daddr_t count)
2227090df5aSMatthew Dillon {
2237090df5aSMatthew Dillon 	if (bl) {
2247090df5aSMatthew Dillon 		if (bl->bl_radix == BLIST_BMAP_RADIX)
2257090df5aSMatthew Dillon 			blst_leaf_free(bl->bl_root, blkno, count);
2267090df5aSMatthew Dillon 		else
2277090df5aSMatthew Dillon 			blst_meta_free(bl->bl_root, blkno, count, bl->bl_radix, bl->bl_skip, 0);
2287090df5aSMatthew Dillon 		bl->bl_free += count;
2297090df5aSMatthew Dillon 	}
2307090df5aSMatthew Dillon }
2317090df5aSMatthew Dillon 
2327090df5aSMatthew Dillon /*
23392da00bbSMatthew Dillon  * blist_fill() -	mark a region in the block bitmap as off-limits
23492da00bbSMatthew Dillon  *			to the allocator (i.e. allocate it), ignoring any
23592da00bbSMatthew Dillon  *			existing allocations.  Return the number of blocks
23692da00bbSMatthew Dillon  *			actually filled that were free before the call.
23792da00bbSMatthew Dillon  */
23892da00bbSMatthew Dillon 
23992da00bbSMatthew Dillon int
24092da00bbSMatthew Dillon blist_fill(blist_t bl, daddr_t blkno, daddr_t count)
24192da00bbSMatthew Dillon {
24292da00bbSMatthew Dillon 	int filled;
24392da00bbSMatthew Dillon 
24492da00bbSMatthew Dillon 	if (bl) {
24592da00bbSMatthew Dillon 		if (bl->bl_radix == BLIST_BMAP_RADIX)
24692da00bbSMatthew Dillon 			filled = blst_leaf_fill(bl->bl_root, blkno, count);
24792da00bbSMatthew Dillon 		else
24892da00bbSMatthew Dillon 			filled = blst_meta_fill(bl->bl_root, blkno, count,
24992da00bbSMatthew Dillon 			    bl->bl_radix, bl->bl_skip, 0);
25092da00bbSMatthew Dillon 		bl->bl_free -= filled;
25192da00bbSMatthew Dillon 		return filled;
25292da00bbSMatthew Dillon 	} else
25392da00bbSMatthew Dillon 		return 0;
25492da00bbSMatthew Dillon }
25592da00bbSMatthew Dillon 
25692da00bbSMatthew Dillon /*
2577090df5aSMatthew Dillon  * blist_resize() -	resize an existing radix tree to handle the
2587090df5aSMatthew Dillon  *			specified number of blocks.  This will reallocate
2597090df5aSMatthew Dillon  *			the tree and transfer the previous bitmap to the new
2607090df5aSMatthew Dillon  *			one.  When extending the tree you can specify whether
2617090df5aSMatthew Dillon  *			the new blocks are to left allocated or freed.
2627090df5aSMatthew Dillon  */
2637090df5aSMatthew Dillon 
2647090df5aSMatthew Dillon void
2657090df5aSMatthew Dillon blist_resize(blist_t *pbl, daddr_t count, int freenew)
2667090df5aSMatthew Dillon {
2677090df5aSMatthew Dillon     blist_t newbl = blist_create(count);
2687090df5aSMatthew Dillon     blist_t save = *pbl;
2697090df5aSMatthew Dillon 
2707090df5aSMatthew Dillon     *pbl = newbl;
2717090df5aSMatthew Dillon     if (count > save->bl_blocks)
2727090df5aSMatthew Dillon 	    count = save->bl_blocks;
2737090df5aSMatthew Dillon     blst_copy(save->bl_root, 0, save->bl_radix, save->bl_skip, newbl, count);
2747090df5aSMatthew Dillon 
2757090df5aSMatthew Dillon     /*
2767090df5aSMatthew Dillon      * If resizing upwards, should we free the new space or not?
2777090df5aSMatthew Dillon      */
2787090df5aSMatthew Dillon     if (freenew && count < newbl->bl_blocks) {
2797090df5aSMatthew Dillon 	    blist_free(newbl, count, newbl->bl_blocks - count);
2807090df5aSMatthew Dillon     }
2817090df5aSMatthew Dillon     blist_destroy(save);
2827090df5aSMatthew Dillon }
2837090df5aSMatthew Dillon 
2847090df5aSMatthew Dillon #ifdef BLIST_DEBUG
2857090df5aSMatthew Dillon 
2867090df5aSMatthew Dillon /*
2877090df5aSMatthew Dillon  * blist_print()    - dump radix tree
2887090df5aSMatthew Dillon  */
2897090df5aSMatthew Dillon 
2907090df5aSMatthew Dillon void
2917090df5aSMatthew Dillon blist_print(blist_t bl)
2927090df5aSMatthew Dillon {
2937090df5aSMatthew Dillon 	printf("BLIST {\n");
2947090df5aSMatthew Dillon 	blst_radix_print(bl->bl_root, 0, bl->bl_radix, bl->bl_skip, 4);
2957090df5aSMatthew Dillon 	printf("}\n");
2967090df5aSMatthew Dillon }
2977090df5aSMatthew Dillon 
2987090df5aSMatthew Dillon #endif
2997090df5aSMatthew Dillon 
3007090df5aSMatthew Dillon /************************************************************************
3017090df5aSMatthew Dillon  *			  ALLOCATION SUPPORT FUNCTIONS			*
3027090df5aSMatthew Dillon  ************************************************************************
3037090df5aSMatthew Dillon  *
3047090df5aSMatthew Dillon  *	These support functions do all the actual work.  They may seem
3057090df5aSMatthew Dillon  *	rather longish, but that's because I've commented them up.  The
3067090df5aSMatthew Dillon  *	actual code is straight forward.
3077090df5aSMatthew Dillon  *
3087090df5aSMatthew Dillon  */
3097090df5aSMatthew Dillon 
3107090df5aSMatthew Dillon /*
3117090df5aSMatthew Dillon  * blist_leaf_alloc() -	allocate at a leaf in the radix tree (a bitmap).
3127090df5aSMatthew Dillon  *
3137090df5aSMatthew Dillon  *	This is the core of the allocator and is optimized for the 1 block
3147090df5aSMatthew Dillon  *	and the BLIST_BMAP_RADIX block allocation cases.  Other cases are
3157090df5aSMatthew Dillon  *	somewhat slower.  The 1 block allocation case is log2 and extremely
3167090df5aSMatthew Dillon  *	quick.
3177090df5aSMatthew Dillon  */
3187090df5aSMatthew Dillon 
3197090df5aSMatthew Dillon static daddr_t
3207090df5aSMatthew Dillon blst_leaf_alloc(
3217090df5aSMatthew Dillon 	blmeta_t *scan,
3227090df5aSMatthew Dillon 	daddr_t blk,
3237090df5aSMatthew Dillon 	int count
3247090df5aSMatthew Dillon ) {
3257090df5aSMatthew Dillon 	u_daddr_t orig = scan->u.bmu_bitmap;
3267090df5aSMatthew Dillon 
3277090df5aSMatthew Dillon 	if (orig == 0) {
3287090df5aSMatthew Dillon 		/*
3297090df5aSMatthew Dillon 		 * Optimize bitmap all-allocated case.  Also, count = 1
3307090df5aSMatthew Dillon 		 * case assumes at least 1 bit is free in the bitmap, so
3317090df5aSMatthew Dillon 		 * we have to take care of this case here.
3327090df5aSMatthew Dillon 		 */
3337090df5aSMatthew Dillon 		scan->bm_bighint = 0;
3347090df5aSMatthew Dillon 		return(SWAPBLK_NONE);
3357090df5aSMatthew Dillon 	}
3367090df5aSMatthew Dillon 	if (count == 1) {
3377090df5aSMatthew Dillon 		/*
3387090df5aSMatthew Dillon 		 * Optimized code to allocate one bit out of the bitmap
3397090df5aSMatthew Dillon 		 */
3407090df5aSMatthew Dillon 		u_daddr_t mask;
3417090df5aSMatthew Dillon 		int j = BLIST_BMAP_RADIX/2;
3427090df5aSMatthew Dillon 		int r = 0;
3437090df5aSMatthew Dillon 
3447090df5aSMatthew Dillon 		mask = (u_daddr_t)-1 >> (BLIST_BMAP_RADIX/2);
3457090df5aSMatthew Dillon 
3467090df5aSMatthew Dillon 		while (j) {
3477090df5aSMatthew Dillon 			if ((orig & mask) == 0) {
3487090df5aSMatthew Dillon 			    r += j;
3497090df5aSMatthew Dillon 			    orig >>= j;
3507090df5aSMatthew Dillon 			}
3517090df5aSMatthew Dillon 			j >>= 1;
3527090df5aSMatthew Dillon 			mask >>= j;
3537090df5aSMatthew Dillon 		}
3547090df5aSMatthew Dillon 		scan->u.bmu_bitmap &= ~(1 << r);
3557090df5aSMatthew Dillon 		return(blk + r);
3567090df5aSMatthew Dillon 	}
3577090df5aSMatthew Dillon 	if (count <= BLIST_BMAP_RADIX) {
3587090df5aSMatthew Dillon 		/*
3597090df5aSMatthew Dillon 		 * non-optimized code to allocate N bits out of the bitmap.
3607090df5aSMatthew Dillon 		 * The more bits, the faster the code runs.  It will run
3617090df5aSMatthew Dillon 		 * the slowest allocating 2 bits, but since there aren't any
3627090df5aSMatthew Dillon 		 * memory ops in the core loop (or shouldn't be, anyway),
3637090df5aSMatthew Dillon 		 * you probably won't notice the difference.
3647090df5aSMatthew Dillon 		 */
3657090df5aSMatthew Dillon 		int j;
3667090df5aSMatthew Dillon 		int n = BLIST_BMAP_RADIX - count;
3677090df5aSMatthew Dillon 		u_daddr_t mask;
3687090df5aSMatthew Dillon 
3697090df5aSMatthew Dillon 		mask = (u_daddr_t)-1 >> n;
3707090df5aSMatthew Dillon 
3717090df5aSMatthew Dillon 		for (j = 0; j <= n; ++j) {
3727090df5aSMatthew Dillon 			if ((orig & mask) == mask) {
3737090df5aSMatthew Dillon 				scan->u.bmu_bitmap &= ~mask;
3747090df5aSMatthew Dillon 				return(blk + j);
3757090df5aSMatthew Dillon 			}
3767090df5aSMatthew Dillon 			mask = (mask << 1);
3777090df5aSMatthew Dillon 		}
3787090df5aSMatthew Dillon 	}
3797090df5aSMatthew Dillon 	/*
3807090df5aSMatthew Dillon 	 * We couldn't allocate count in this subtree, update bighint.
3817090df5aSMatthew Dillon 	 */
3827090df5aSMatthew Dillon 	scan->bm_bighint = count - 1;
3837090df5aSMatthew Dillon 	return(SWAPBLK_NONE);
3847090df5aSMatthew Dillon }
3857090df5aSMatthew Dillon 
3867090df5aSMatthew Dillon /*
3877090df5aSMatthew Dillon  * blist_meta_alloc() -	allocate at a meta in the radix tree.
3887090df5aSMatthew Dillon  *
3897090df5aSMatthew Dillon  *	Attempt to allocate at a meta node.  If we can't, we update
3907090df5aSMatthew Dillon  *	bighint and return a failure.  Updating bighint optimize future
3917090df5aSMatthew Dillon  *	calls that hit this node.  We have to check for our collapse cases
3927090df5aSMatthew Dillon  *	and we have a few optimizations strewn in as well.
3937090df5aSMatthew Dillon  */
3947090df5aSMatthew Dillon 
3957090df5aSMatthew Dillon static daddr_t
3967090df5aSMatthew Dillon blst_meta_alloc(
3977090df5aSMatthew Dillon 	blmeta_t *scan,
3987090df5aSMatthew Dillon 	daddr_t blk,
3997090df5aSMatthew Dillon 	daddr_t count,
4007090df5aSMatthew Dillon 	daddr_t radix,
4017090df5aSMatthew Dillon 	int skip
4027090df5aSMatthew Dillon ) {
4037090df5aSMatthew Dillon 	int i;
40457e6d29bSMatthew Dillon 	int next_skip = ((u_int)skip / BLIST_META_RADIX);
4057090df5aSMatthew Dillon 
4067090df5aSMatthew Dillon 	if (scan->u.bmu_avail == 0)  {
4077090df5aSMatthew Dillon 		/*
4087090df5aSMatthew Dillon 		 * ALL-ALLOCATED special case
4097090df5aSMatthew Dillon 		 */
4107090df5aSMatthew Dillon 		scan->bm_bighint = count;
4117090df5aSMatthew Dillon 		return(SWAPBLK_NONE);
4127090df5aSMatthew Dillon 	}
4137090df5aSMatthew Dillon 
4147090df5aSMatthew Dillon 	if (scan->u.bmu_avail == radix) {
41557e6d29bSMatthew Dillon 		radix /= BLIST_META_RADIX;
4167090df5aSMatthew Dillon 
4177090df5aSMatthew Dillon 		/*
4187090df5aSMatthew Dillon 		 * ALL-FREE special case, initialize uninitialize
4197090df5aSMatthew Dillon 		 * sublevel.
4207090df5aSMatthew Dillon 		 */
4217090df5aSMatthew Dillon 		for (i = 1; i <= skip; i += next_skip) {
4227090df5aSMatthew Dillon 			if (scan[i].bm_bighint == (daddr_t)-1)
4237090df5aSMatthew Dillon 				break;
4247090df5aSMatthew Dillon 			if (next_skip == 1) {
4257090df5aSMatthew Dillon 				scan[i].u.bmu_bitmap = (u_daddr_t)-1;
4267090df5aSMatthew Dillon 				scan[i].bm_bighint = BLIST_BMAP_RADIX;
4277090df5aSMatthew Dillon 			} else {
4287090df5aSMatthew Dillon 				scan[i].bm_bighint = radix;
4297090df5aSMatthew Dillon 				scan[i].u.bmu_avail = radix;
4307090df5aSMatthew Dillon 			}
4317090df5aSMatthew Dillon 		}
4327090df5aSMatthew Dillon 	} else {
43357e6d29bSMatthew Dillon 		radix /= BLIST_META_RADIX;
4347090df5aSMatthew Dillon 	}
4357090df5aSMatthew Dillon 
4367090df5aSMatthew Dillon 	for (i = 1; i <= skip; i += next_skip) {
4377090df5aSMatthew Dillon 		if (count <= scan[i].bm_bighint) {
4387090df5aSMatthew Dillon 			/*
4397090df5aSMatthew Dillon 			 * count fits in object
4407090df5aSMatthew Dillon 			 */
4417090df5aSMatthew Dillon 			daddr_t r;
4427090df5aSMatthew Dillon 			if (next_skip == 1) {
4437090df5aSMatthew Dillon 				r = blst_leaf_alloc(&scan[i], blk, count);
4447090df5aSMatthew Dillon 			} else {
4457090df5aSMatthew Dillon 				r = blst_meta_alloc(&scan[i], blk, count, radix, next_skip - 1);
4467090df5aSMatthew Dillon 			}
4477090df5aSMatthew Dillon 			if (r != SWAPBLK_NONE) {
4487090df5aSMatthew Dillon 				scan->u.bmu_avail -= count;
4497090df5aSMatthew Dillon 				if (scan->bm_bighint > scan->u.bmu_avail)
4507090df5aSMatthew Dillon 					scan->bm_bighint = scan->u.bmu_avail;
4517090df5aSMatthew Dillon 				return(r);
4527090df5aSMatthew Dillon 			}
4537090df5aSMatthew Dillon 		} else if (scan[i].bm_bighint == (daddr_t)-1) {
4547090df5aSMatthew Dillon 			/*
4557090df5aSMatthew Dillon 			 * Terminator
4567090df5aSMatthew Dillon 			 */
4577090df5aSMatthew Dillon 			break;
4587090df5aSMatthew Dillon 		} else if (count > radix) {
4597090df5aSMatthew Dillon 			/*
4607090df5aSMatthew Dillon 			 * count does not fit in object even if it were
4617090df5aSMatthew Dillon 			 * complete free.
4627090df5aSMatthew Dillon 			 */
4637090df5aSMatthew Dillon 			panic("blist_meta_alloc: allocation too large");
4647090df5aSMatthew Dillon 		}
4657090df5aSMatthew Dillon 		blk += radix;
4667090df5aSMatthew Dillon 	}
4677090df5aSMatthew Dillon 
4687090df5aSMatthew Dillon 	/*
4697090df5aSMatthew Dillon 	 * We couldn't allocate count in this subtree, update bighint.
4707090df5aSMatthew Dillon 	 */
4717090df5aSMatthew Dillon 	if (scan->bm_bighint >= count)
4727090df5aSMatthew Dillon 		scan->bm_bighint = count - 1;
4737090df5aSMatthew Dillon 	return(SWAPBLK_NONE);
4747090df5aSMatthew Dillon }
4757090df5aSMatthew Dillon 
4767090df5aSMatthew Dillon /*
4777090df5aSMatthew Dillon  * BLST_LEAF_FREE() -	free allocated block from leaf bitmap
4787090df5aSMatthew Dillon  *
4797090df5aSMatthew Dillon  */
4807090df5aSMatthew Dillon 
4817090df5aSMatthew Dillon static void
4827090df5aSMatthew Dillon blst_leaf_free(
4837090df5aSMatthew Dillon 	blmeta_t *scan,
4847090df5aSMatthew Dillon 	daddr_t blk,
4857090df5aSMatthew Dillon 	int count
4867090df5aSMatthew Dillon ) {
4877090df5aSMatthew Dillon 	/*
4887090df5aSMatthew Dillon 	 * free some data in this bitmap
4897090df5aSMatthew Dillon 	 *
4907090df5aSMatthew Dillon 	 * e.g.
4917090df5aSMatthew Dillon 	 *	0000111111111110000
4927090df5aSMatthew Dillon 	 *          \_________/\__/
4937090df5aSMatthew Dillon 	 *		v        n
4947090df5aSMatthew Dillon 	 */
4957090df5aSMatthew Dillon 	int n = blk & (BLIST_BMAP_RADIX - 1);
4967090df5aSMatthew Dillon 	u_daddr_t mask;
4977090df5aSMatthew Dillon 
4987090df5aSMatthew Dillon 	mask = ((u_daddr_t)-1 << n) &
4997090df5aSMatthew Dillon 	    ((u_daddr_t)-1 >> (BLIST_BMAP_RADIX - count - n));
5007090df5aSMatthew Dillon 
5017090df5aSMatthew Dillon 	if (scan->u.bmu_bitmap & mask)
5027090df5aSMatthew Dillon 		panic("blst_radix_free: freeing free block");
5037090df5aSMatthew Dillon 	scan->u.bmu_bitmap |= mask;
5047090df5aSMatthew Dillon 
5057090df5aSMatthew Dillon 	/*
5067090df5aSMatthew Dillon 	 * We could probably do a better job here.  We are required to make
5077090df5aSMatthew Dillon 	 * bighint at least as large as the biggest contiguous block of
5087090df5aSMatthew Dillon 	 * data.  If we just shoehorn it, a little extra overhead will
5097090df5aSMatthew Dillon 	 * be incured on the next allocation (but only that one typically).
5107090df5aSMatthew Dillon 	 */
5117090df5aSMatthew Dillon 	scan->bm_bighint = BLIST_BMAP_RADIX;
5127090df5aSMatthew Dillon }
5137090df5aSMatthew Dillon 
5147090df5aSMatthew Dillon /*
5157090df5aSMatthew Dillon  * BLST_META_FREE() - free allocated blocks from radix tree meta info
5167090df5aSMatthew Dillon  *
5177090df5aSMatthew Dillon  *	This support routine frees a range of blocks from the bitmap.
5187090df5aSMatthew Dillon  *	The range must be entirely enclosed by this radix node.  If a
5197090df5aSMatthew Dillon  *	meta node, we break the range down recursively to free blocks
5207090df5aSMatthew Dillon  *	in subnodes (which means that this code can free an arbitrary
5217090df5aSMatthew Dillon  *	range whereas the allocation code cannot allocate an arbitrary
5227090df5aSMatthew Dillon  *	range).
5237090df5aSMatthew Dillon  */
5247090df5aSMatthew Dillon 
5257090df5aSMatthew Dillon static void
5267090df5aSMatthew Dillon blst_meta_free(
5277090df5aSMatthew Dillon 	blmeta_t *scan,
5287090df5aSMatthew Dillon 	daddr_t freeBlk,
5297090df5aSMatthew Dillon 	daddr_t count,
5307090df5aSMatthew Dillon 	daddr_t radix,
5317090df5aSMatthew Dillon 	int skip,
5327090df5aSMatthew Dillon 	daddr_t blk
5337090df5aSMatthew Dillon ) {
5347090df5aSMatthew Dillon 	int i;
53557e6d29bSMatthew Dillon 	int next_skip = ((u_int)skip / BLIST_META_RADIX);
5367090df5aSMatthew Dillon 
5377090df5aSMatthew Dillon #if 0
53892da00bbSMatthew Dillon 	printf("FREE (%llx,%lld) FROM (%llx,%lld)\n",
53992da00bbSMatthew Dillon 	    (long long)freeBlk, (long long)count,
54092da00bbSMatthew Dillon 	    (long long)blk, (long long)radix
5417090df5aSMatthew Dillon 	);
5427090df5aSMatthew Dillon #endif
5437090df5aSMatthew Dillon 
5447090df5aSMatthew Dillon 	if (scan->u.bmu_avail == 0) {
5457090df5aSMatthew Dillon 		/*
5467090df5aSMatthew Dillon 		 * ALL-ALLOCATED special case, with possible
5477090df5aSMatthew Dillon 		 * shortcut to ALL-FREE special case.
5487090df5aSMatthew Dillon 		 */
5497090df5aSMatthew Dillon 		scan->u.bmu_avail = count;
5507090df5aSMatthew Dillon 		scan->bm_bighint = count;
5517090df5aSMatthew Dillon 
5527090df5aSMatthew Dillon 		if (count != radix)  {
5537090df5aSMatthew Dillon 			for (i = 1; i <= skip; i += next_skip) {
5547090df5aSMatthew Dillon 				if (scan[i].bm_bighint == (daddr_t)-1)
5557090df5aSMatthew Dillon 					break;
5567090df5aSMatthew Dillon 				scan[i].bm_bighint = 0;
5577090df5aSMatthew Dillon 				if (next_skip == 1) {
5587090df5aSMatthew Dillon 					scan[i].u.bmu_bitmap = 0;
5597090df5aSMatthew Dillon 				} else {
5607090df5aSMatthew Dillon 					scan[i].u.bmu_avail = 0;
5617090df5aSMatthew Dillon 				}
5627090df5aSMatthew Dillon 			}
5637090df5aSMatthew Dillon 			/* fall through */
5647090df5aSMatthew Dillon 		}
5657090df5aSMatthew Dillon 	} else {
5667090df5aSMatthew Dillon 		scan->u.bmu_avail += count;
5677090df5aSMatthew Dillon 		/* scan->bm_bighint = radix; */
5687090df5aSMatthew Dillon 	}
5697090df5aSMatthew Dillon 
5707090df5aSMatthew Dillon 	/*
5717090df5aSMatthew Dillon 	 * ALL-FREE special case.
5727090df5aSMatthew Dillon 	 */
5737090df5aSMatthew Dillon 
5747090df5aSMatthew Dillon 	if (scan->u.bmu_avail == radix)
5757090df5aSMatthew Dillon 		return;
5767090df5aSMatthew Dillon 	if (scan->u.bmu_avail > radix)
577bdc9a8d0SJohn Baldwin 		panic("blst_meta_free: freeing already free blocks (%lld) %lld/%lld",
578bdc9a8d0SJohn Baldwin 		    (long long)count, (long long)scan->u.bmu_avail,
579bdc9a8d0SJohn Baldwin 		    (long long)radix);
5807090df5aSMatthew Dillon 
5817090df5aSMatthew Dillon 	/*
5827090df5aSMatthew Dillon 	 * Break the free down into its components
5837090df5aSMatthew Dillon 	 */
5847090df5aSMatthew Dillon 
58557e6d29bSMatthew Dillon 	radix /= BLIST_META_RADIX;
5867090df5aSMatthew Dillon 
5877090df5aSMatthew Dillon 	i = (freeBlk - blk) / radix;
5887090df5aSMatthew Dillon 	blk += i * radix;
5897090df5aSMatthew Dillon 	i = i * next_skip + 1;
5907090df5aSMatthew Dillon 
5917090df5aSMatthew Dillon 	while (i <= skip && blk < freeBlk + count) {
5927090df5aSMatthew Dillon 		daddr_t v;
5937090df5aSMatthew Dillon 
5947090df5aSMatthew Dillon 		v = blk + radix - freeBlk;
5957090df5aSMatthew Dillon 		if (v > count)
5967090df5aSMatthew Dillon 			v = count;
5977090df5aSMatthew Dillon 
5987090df5aSMatthew Dillon 		if (scan->bm_bighint == (daddr_t)-1)
5997090df5aSMatthew Dillon 			panic("blst_meta_free: freeing unexpected range");
6007090df5aSMatthew Dillon 
6017090df5aSMatthew Dillon 		if (next_skip == 1) {
6027090df5aSMatthew Dillon 			blst_leaf_free(&scan[i], freeBlk, v);
6037090df5aSMatthew Dillon 		} else {
6047090df5aSMatthew Dillon 			blst_meta_free(&scan[i], freeBlk, v, radix, next_skip - 1, blk);
6057090df5aSMatthew Dillon 		}
6067090df5aSMatthew Dillon 		if (scan->bm_bighint < scan[i].bm_bighint)
6077090df5aSMatthew Dillon 		    scan->bm_bighint = scan[i].bm_bighint;
6087090df5aSMatthew Dillon 		count -= v;
6097090df5aSMatthew Dillon 		freeBlk += v;
6107090df5aSMatthew Dillon 		blk += radix;
6117090df5aSMatthew Dillon 		i += next_skip;
6127090df5aSMatthew Dillon 	}
6137090df5aSMatthew Dillon }
6147090df5aSMatthew Dillon 
6157090df5aSMatthew Dillon /*
6167090df5aSMatthew Dillon  * BLIST_RADIX_COPY() - copy one radix tree to another
6177090df5aSMatthew Dillon  *
6187090df5aSMatthew Dillon  *	Locates free space in the source tree and frees it in the destination
6197090df5aSMatthew Dillon  *	tree.  The space may not already be free in the destination.
6207090df5aSMatthew Dillon  */
6217090df5aSMatthew Dillon 
6227090df5aSMatthew Dillon static void blst_copy(
6237090df5aSMatthew Dillon 	blmeta_t *scan,
6247090df5aSMatthew Dillon 	daddr_t blk,
6257090df5aSMatthew Dillon 	daddr_t radix,
6267090df5aSMatthew Dillon 	daddr_t skip,
6277090df5aSMatthew Dillon 	blist_t dest,
6287090df5aSMatthew Dillon 	daddr_t count
6297090df5aSMatthew Dillon ) {
6307090df5aSMatthew Dillon 	int next_skip;
6317090df5aSMatthew Dillon 	int i;
6327090df5aSMatthew Dillon 
6337090df5aSMatthew Dillon 	/*
6347090df5aSMatthew Dillon 	 * Leaf node
6357090df5aSMatthew Dillon 	 */
6367090df5aSMatthew Dillon 
6377090df5aSMatthew Dillon 	if (radix == BLIST_BMAP_RADIX) {
6387090df5aSMatthew Dillon 		u_daddr_t v = scan->u.bmu_bitmap;
6397090df5aSMatthew Dillon 
6407090df5aSMatthew Dillon 		if (v == (u_daddr_t)-1) {
6417090df5aSMatthew Dillon 			blist_free(dest, blk, count);
6427090df5aSMatthew Dillon 		} else if (v != 0) {
6437090df5aSMatthew Dillon 			int i;
6447090df5aSMatthew Dillon 
6457090df5aSMatthew Dillon 			for (i = 0; i < BLIST_BMAP_RADIX && i < count; ++i) {
6467090df5aSMatthew Dillon 				if (v & (1 << i))
6477090df5aSMatthew Dillon 					blist_free(dest, blk + i, 1);
6487090df5aSMatthew Dillon 			}
6497090df5aSMatthew Dillon 		}
6507090df5aSMatthew Dillon 		return;
6517090df5aSMatthew Dillon 	}
6527090df5aSMatthew Dillon 
6537090df5aSMatthew Dillon 	/*
6547090df5aSMatthew Dillon 	 * Meta node
6557090df5aSMatthew Dillon 	 */
6567090df5aSMatthew Dillon 
6577090df5aSMatthew Dillon 	if (scan->u.bmu_avail == 0) {
6587090df5aSMatthew Dillon 		/*
6597090df5aSMatthew Dillon 		 * Source all allocated, leave dest allocated
6607090df5aSMatthew Dillon 		 */
6617090df5aSMatthew Dillon 		return;
6627090df5aSMatthew Dillon 	}
6637090df5aSMatthew Dillon 	if (scan->u.bmu_avail == radix) {
6647090df5aSMatthew Dillon 		/*
6657090df5aSMatthew Dillon 		 * Source all free, free entire dest
6667090df5aSMatthew Dillon 		 */
6677090df5aSMatthew Dillon 		if (count < radix)
6687090df5aSMatthew Dillon 			blist_free(dest, blk, count);
6697090df5aSMatthew Dillon 		else
6707090df5aSMatthew Dillon 			blist_free(dest, blk, radix);
6717090df5aSMatthew Dillon 		return;
6727090df5aSMatthew Dillon 	}
6737090df5aSMatthew Dillon 
6747090df5aSMatthew Dillon 
67557e6d29bSMatthew Dillon 	radix /= BLIST_META_RADIX;
67657e6d29bSMatthew Dillon 	next_skip = ((u_int)skip / BLIST_META_RADIX);
6777090df5aSMatthew Dillon 
6787090df5aSMatthew Dillon 	for (i = 1; count && i <= skip; i += next_skip) {
6797090df5aSMatthew Dillon 		if (scan[i].bm_bighint == (daddr_t)-1)
6807090df5aSMatthew Dillon 			break;
6817090df5aSMatthew Dillon 
6827090df5aSMatthew Dillon 		if (count >= radix) {
6837090df5aSMatthew Dillon 			blst_copy(
6847090df5aSMatthew Dillon 			    &scan[i],
6857090df5aSMatthew Dillon 			    blk,
6867090df5aSMatthew Dillon 			    radix,
6877090df5aSMatthew Dillon 			    next_skip - 1,
6887090df5aSMatthew Dillon 			    dest,
6897090df5aSMatthew Dillon 			    radix
6907090df5aSMatthew Dillon 			);
6917090df5aSMatthew Dillon 			count -= radix;
6927090df5aSMatthew Dillon 		} else {
6937090df5aSMatthew Dillon 			if (count) {
6947090df5aSMatthew Dillon 				blst_copy(
6957090df5aSMatthew Dillon 				    &scan[i],
6967090df5aSMatthew Dillon 				    blk,
6977090df5aSMatthew Dillon 				    radix,
6987090df5aSMatthew Dillon 				    next_skip - 1,
6997090df5aSMatthew Dillon 				    dest,
7007090df5aSMatthew Dillon 				    count
7017090df5aSMatthew Dillon 				);
7027090df5aSMatthew Dillon 			}
7037090df5aSMatthew Dillon 			count = 0;
7047090df5aSMatthew Dillon 		}
7057090df5aSMatthew Dillon 		blk += radix;
7067090df5aSMatthew Dillon 	}
7077090df5aSMatthew Dillon }
7087090df5aSMatthew Dillon 
7097090df5aSMatthew Dillon /*
71092da00bbSMatthew Dillon  * BLST_LEAF_FILL() -	allocate specific blocks in leaf bitmap
71192da00bbSMatthew Dillon  *
71292da00bbSMatthew Dillon  *	This routine allocates all blocks in the specified range
71392da00bbSMatthew Dillon  *	regardless of any existing allocations in that range.  Returns
71492da00bbSMatthew Dillon  *	the number of blocks allocated by the call.
71592da00bbSMatthew Dillon  */
71692da00bbSMatthew Dillon 
71792da00bbSMatthew Dillon static int
71892da00bbSMatthew Dillon blst_leaf_fill(blmeta_t *scan, daddr_t blk, int count)
71992da00bbSMatthew Dillon {
72092da00bbSMatthew Dillon 	int n = blk & (BLIST_BMAP_RADIX - 1);
72192da00bbSMatthew Dillon 	int nblks;
72292da00bbSMatthew Dillon 	u_daddr_t mask, bitmap;
72392da00bbSMatthew Dillon 
72492da00bbSMatthew Dillon 	mask = ((u_daddr_t)-1 << n) &
72592da00bbSMatthew Dillon 	    ((u_daddr_t)-1 >> (BLIST_BMAP_RADIX - count - n));
72692da00bbSMatthew Dillon 
72792da00bbSMatthew Dillon 	/* Count the number of blocks we're about to allocate */
72892da00bbSMatthew Dillon 	bitmap = scan->u.bmu_bitmap & mask;
72992da00bbSMatthew Dillon 	for (nblks = 0; bitmap != 0; nblks++)
73092da00bbSMatthew Dillon 		bitmap &= bitmap - 1;
73192da00bbSMatthew Dillon 
73292da00bbSMatthew Dillon 	scan->u.bmu_bitmap &= ~mask;
73392da00bbSMatthew Dillon 	return nblks;
73492da00bbSMatthew Dillon }
73592da00bbSMatthew Dillon 
73692da00bbSMatthew Dillon /*
73792da00bbSMatthew Dillon  * BLIST_META_FILL() -	allocate specific blocks at a meta node
73892da00bbSMatthew Dillon  *
73992da00bbSMatthew Dillon  *	This routine allocates the specified range of blocks,
74092da00bbSMatthew Dillon  *	regardless of any existing allocations in the range.  The
74192da00bbSMatthew Dillon  *	range must be within the extent of this node.  Returns the
74292da00bbSMatthew Dillon  *	number of blocks allocated by the call.
74392da00bbSMatthew Dillon  */
74492da00bbSMatthew Dillon static int
74592da00bbSMatthew Dillon blst_meta_fill(
74692da00bbSMatthew Dillon 	blmeta_t *scan,
74792da00bbSMatthew Dillon 	daddr_t allocBlk,
74892da00bbSMatthew Dillon 	daddr_t count,
74992da00bbSMatthew Dillon 	daddr_t radix,
75092da00bbSMatthew Dillon 	int skip,
75192da00bbSMatthew Dillon 	daddr_t blk
75292da00bbSMatthew Dillon ) {
75392da00bbSMatthew Dillon 	int i;
75457e6d29bSMatthew Dillon 	int next_skip = ((u_int)skip / BLIST_META_RADIX);
75592da00bbSMatthew Dillon 	int nblks = 0;
75692da00bbSMatthew Dillon 
75792da00bbSMatthew Dillon 	if (count == radix || scan->u.bmu_avail == 0)  {
75892da00bbSMatthew Dillon 		/*
75992da00bbSMatthew Dillon 		 * ALL-ALLOCATED special case
76092da00bbSMatthew Dillon 		 */
76192da00bbSMatthew Dillon 		nblks = scan->u.bmu_avail;
76292da00bbSMatthew Dillon 		scan->u.bmu_avail = 0;
76392da00bbSMatthew Dillon 		scan->bm_bighint = count;
76492da00bbSMatthew Dillon 		return nblks;
76592da00bbSMatthew Dillon 	}
76692da00bbSMatthew Dillon 
76792da00bbSMatthew Dillon 	if (scan->u.bmu_avail == radix) {
76857e6d29bSMatthew Dillon 		radix /= BLIST_META_RADIX;
76992da00bbSMatthew Dillon 
77092da00bbSMatthew Dillon 		/*
77192da00bbSMatthew Dillon 		 * ALL-FREE special case, initialize sublevel
77292da00bbSMatthew Dillon 		 */
77392da00bbSMatthew Dillon 		for (i = 1; i <= skip; i += next_skip) {
77492da00bbSMatthew Dillon 			if (scan[i].bm_bighint == (daddr_t)-1)
77592da00bbSMatthew Dillon 				break;
77692da00bbSMatthew Dillon 			if (next_skip == 1) {
77792da00bbSMatthew Dillon 				scan[i].u.bmu_bitmap = (u_daddr_t)-1;
77892da00bbSMatthew Dillon 				scan[i].bm_bighint = BLIST_BMAP_RADIX;
77992da00bbSMatthew Dillon 			} else {
78092da00bbSMatthew Dillon 				scan[i].bm_bighint = radix;
78192da00bbSMatthew Dillon 				scan[i].u.bmu_avail = radix;
78292da00bbSMatthew Dillon 			}
78392da00bbSMatthew Dillon 		}
78492da00bbSMatthew Dillon 	} else {
78557e6d29bSMatthew Dillon 		radix /= BLIST_META_RADIX;
78692da00bbSMatthew Dillon 	}
78792da00bbSMatthew Dillon 
78892da00bbSMatthew Dillon 	if (count > radix)
78992da00bbSMatthew Dillon 		panic("blist_meta_fill: allocation too large");
79092da00bbSMatthew Dillon 
79192da00bbSMatthew Dillon 	i = (allocBlk - blk) / radix;
79292da00bbSMatthew Dillon 	blk += i * radix;
79392da00bbSMatthew Dillon 	i = i * next_skip + 1;
79492da00bbSMatthew Dillon 
79592da00bbSMatthew Dillon 	while (i <= skip && blk < allocBlk + count) {
79692da00bbSMatthew Dillon 		daddr_t v;
79792da00bbSMatthew Dillon 
79892da00bbSMatthew Dillon 		v = blk + radix - allocBlk;
79992da00bbSMatthew Dillon 		if (v > count)
80092da00bbSMatthew Dillon 			v = count;
80192da00bbSMatthew Dillon 
80292da00bbSMatthew Dillon 		if (scan->bm_bighint == (daddr_t)-1)
80392da00bbSMatthew Dillon 			panic("blst_meta_fill: filling unexpected range");
80492da00bbSMatthew Dillon 
80592da00bbSMatthew Dillon 		if (next_skip == 1) {
80692da00bbSMatthew Dillon 			nblks += blst_leaf_fill(&scan[i], allocBlk, v);
80792da00bbSMatthew Dillon 		} else {
80892da00bbSMatthew Dillon 			nblks += blst_meta_fill(&scan[i], allocBlk, v,
80992da00bbSMatthew Dillon 			    radix, next_skip - 1, blk);
81092da00bbSMatthew Dillon 		}
81192da00bbSMatthew Dillon 		count -= v;
81292da00bbSMatthew Dillon 		allocBlk += v;
81392da00bbSMatthew Dillon 		blk += radix;
81492da00bbSMatthew Dillon 		i += next_skip;
81592da00bbSMatthew Dillon 	}
81692da00bbSMatthew Dillon 	scan->u.bmu_avail -= nblks;
81792da00bbSMatthew Dillon 	return nblks;
81892da00bbSMatthew Dillon }
81992da00bbSMatthew Dillon 
82092da00bbSMatthew Dillon /*
8217090df5aSMatthew Dillon  * BLST_RADIX_INIT() - initialize radix tree
8227090df5aSMatthew Dillon  *
8237090df5aSMatthew Dillon  *	Initialize our meta structures and bitmaps and calculate the exact
8247090df5aSMatthew Dillon  *	amount of space required to manage 'count' blocks - this space may
8257090df5aSMatthew Dillon  *	be considerably less then the calculated radix due to the large
8267090df5aSMatthew Dillon  *	RADIX values we use.
8277090df5aSMatthew Dillon  */
8287090df5aSMatthew Dillon 
8297090df5aSMatthew Dillon static daddr_t
8307090df5aSMatthew Dillon blst_radix_init(blmeta_t *scan, daddr_t radix, int skip, daddr_t count)
8317090df5aSMatthew Dillon {
8327090df5aSMatthew Dillon 	int i;
8337090df5aSMatthew Dillon 	int next_skip;
8347090df5aSMatthew Dillon 	daddr_t memindex = 0;
8357090df5aSMatthew Dillon 
8367090df5aSMatthew Dillon 	/*
8377090df5aSMatthew Dillon 	 * Leaf node
8387090df5aSMatthew Dillon 	 */
8397090df5aSMatthew Dillon 
8407090df5aSMatthew Dillon 	if (radix == BLIST_BMAP_RADIX) {
8417090df5aSMatthew Dillon 		if (scan) {
8427090df5aSMatthew Dillon 			scan->bm_bighint = 0;
8437090df5aSMatthew Dillon 			scan->u.bmu_bitmap = 0;
8447090df5aSMatthew Dillon 		}
8457090df5aSMatthew Dillon 		return(memindex);
8467090df5aSMatthew Dillon 	}
8477090df5aSMatthew Dillon 
8487090df5aSMatthew Dillon 	/*
8497090df5aSMatthew Dillon 	 * Meta node.  If allocating the entire object we can special
8507090df5aSMatthew Dillon 	 * case it.  However, we need to figure out how much memory
8517090df5aSMatthew Dillon 	 * is required to manage 'count' blocks, so we continue on anyway.
8527090df5aSMatthew Dillon 	 */
8537090df5aSMatthew Dillon 
8547090df5aSMatthew Dillon 	if (scan) {
8557090df5aSMatthew Dillon 		scan->bm_bighint = 0;
8567090df5aSMatthew Dillon 		scan->u.bmu_avail = 0;
8577090df5aSMatthew Dillon 	}
8587090df5aSMatthew Dillon 
85957e6d29bSMatthew Dillon 	radix /= BLIST_META_RADIX;
86057e6d29bSMatthew Dillon 	next_skip = ((u_int)skip / BLIST_META_RADIX);
8617090df5aSMatthew Dillon 
8627090df5aSMatthew Dillon 	for (i = 1; i <= skip; i += next_skip) {
8637090df5aSMatthew Dillon 		if (count >= radix) {
8647090df5aSMatthew Dillon 			/*
8657090df5aSMatthew Dillon 			 * Allocate the entire object
8667090df5aSMatthew Dillon 			 */
8677090df5aSMatthew Dillon 			memindex = i + blst_radix_init(
8687090df5aSMatthew Dillon 			    ((scan) ? &scan[i] : NULL),
8697090df5aSMatthew Dillon 			    radix,
8707090df5aSMatthew Dillon 			    next_skip - 1,
8717090df5aSMatthew Dillon 			    radix
8727090df5aSMatthew Dillon 			);
8737090df5aSMatthew Dillon 			count -= radix;
8747090df5aSMatthew Dillon 		} else if (count > 0) {
8757090df5aSMatthew Dillon 			/*
8767090df5aSMatthew Dillon 			 * Allocate a partial object
8777090df5aSMatthew Dillon 			 */
8787090df5aSMatthew Dillon 			memindex = i + blst_radix_init(
8797090df5aSMatthew Dillon 			    ((scan) ? &scan[i] : NULL),
8807090df5aSMatthew Dillon 			    radix,
8817090df5aSMatthew Dillon 			    next_skip - 1,
8827090df5aSMatthew Dillon 			    count
8837090df5aSMatthew Dillon 			);
8847090df5aSMatthew Dillon 			count = 0;
8857090df5aSMatthew Dillon 		} else {
8867090df5aSMatthew Dillon 			/*
8877090df5aSMatthew Dillon 			 * Add terminator and break out
8887090df5aSMatthew Dillon 			 */
8897090df5aSMatthew Dillon 			if (scan)
8907090df5aSMatthew Dillon 				scan[i].bm_bighint = (daddr_t)-1;
8917090df5aSMatthew Dillon 			break;
8927090df5aSMatthew Dillon 		}
8937090df5aSMatthew Dillon 	}
8947090df5aSMatthew Dillon 	if (memindex < i)
8957090df5aSMatthew Dillon 		memindex = i;
8967090df5aSMatthew Dillon 	return(memindex);
8977090df5aSMatthew Dillon }
8987090df5aSMatthew Dillon 
8997090df5aSMatthew Dillon #ifdef BLIST_DEBUG
9007090df5aSMatthew Dillon 
9017090df5aSMatthew Dillon static void
9027090df5aSMatthew Dillon blst_radix_print(blmeta_t *scan, daddr_t blk, daddr_t radix, int skip, int tab)
9037090df5aSMatthew Dillon {
9047090df5aSMatthew Dillon 	int i;
9057090df5aSMatthew Dillon 	int next_skip;
9067090df5aSMatthew Dillon 	int lastState = 0;
9077090df5aSMatthew Dillon 
9087090df5aSMatthew Dillon 	if (radix == BLIST_BMAP_RADIX) {
9097090df5aSMatthew Dillon 		printf(
91092da00bbSMatthew Dillon 		    "%*.*s(%08llx,%lld): bitmap %08llx big=%lld\n",
9117090df5aSMatthew Dillon 		    tab, tab, "",
91292da00bbSMatthew Dillon 		    (long long)blk, (long long)radix,
91392da00bbSMatthew Dillon 		    (long long)scan->u.bmu_bitmap,
91492da00bbSMatthew Dillon 		    (long long)scan->bm_bighint
9157090df5aSMatthew Dillon 		);
9167090df5aSMatthew Dillon 		return;
9177090df5aSMatthew Dillon 	}
9187090df5aSMatthew Dillon 
9197090df5aSMatthew Dillon 	if (scan->u.bmu_avail == 0) {
9207090df5aSMatthew Dillon 		printf(
92192da00bbSMatthew Dillon 		    "%*.*s(%08llx,%lld) ALL ALLOCATED\n",
9227090df5aSMatthew Dillon 		    tab, tab, "",
92392da00bbSMatthew Dillon 		    (long long)blk,
92492da00bbSMatthew Dillon 		    (long long)radix
9257090df5aSMatthew Dillon 		);
9267090df5aSMatthew Dillon 		return;
9277090df5aSMatthew Dillon 	}
9287090df5aSMatthew Dillon 	if (scan->u.bmu_avail == radix) {
9297090df5aSMatthew Dillon 		printf(
93092da00bbSMatthew Dillon 		    "%*.*s(%08llx,%lld) ALL FREE\n",
9317090df5aSMatthew Dillon 		    tab, tab, "",
93292da00bbSMatthew Dillon 		    (long long)blk,
93392da00bbSMatthew Dillon 		    (long long)radix
9347090df5aSMatthew Dillon 		);
9357090df5aSMatthew Dillon 		return;
9367090df5aSMatthew Dillon 	}
9377090df5aSMatthew Dillon 
9387090df5aSMatthew Dillon 	printf(
93992da00bbSMatthew Dillon 	    "%*.*s(%08llx,%lld): subtree (%lld/%lld) big=%lld {\n",
9407090df5aSMatthew Dillon 	    tab, tab, "",
94192da00bbSMatthew Dillon 	    (long long)blk, (long long)radix,
94292da00bbSMatthew Dillon 	    (long long)scan->u.bmu_avail,
94392da00bbSMatthew Dillon 	    (long long)radix,
94492da00bbSMatthew Dillon 	    (long long)scan->bm_bighint
9457090df5aSMatthew Dillon 	);
9467090df5aSMatthew Dillon 
94757e6d29bSMatthew Dillon 	radix /= BLIST_META_RADIX;
94857e6d29bSMatthew Dillon 	next_skip = ((u_int)skip / BLIST_META_RADIX);
9497090df5aSMatthew Dillon 	tab += 4;
9507090df5aSMatthew Dillon 
9517090df5aSMatthew Dillon 	for (i = 1; i <= skip; i += next_skip) {
9527090df5aSMatthew Dillon 		if (scan[i].bm_bighint == (daddr_t)-1) {
9537090df5aSMatthew Dillon 			printf(
95492da00bbSMatthew Dillon 			    "%*.*s(%08llx,%lld): Terminator\n",
9557090df5aSMatthew Dillon 			    tab, tab, "",
95692da00bbSMatthew Dillon 			    (long long)blk, (long long)radix
9577090df5aSMatthew Dillon 			);
9587090df5aSMatthew Dillon 			lastState = 0;
9597090df5aSMatthew Dillon 			break;
9607090df5aSMatthew Dillon 		}
9617090df5aSMatthew Dillon 		blst_radix_print(
9627090df5aSMatthew Dillon 		    &scan[i],
9637090df5aSMatthew Dillon 		    blk,
9647090df5aSMatthew Dillon 		    radix,
9657090df5aSMatthew Dillon 		    next_skip - 1,
9667090df5aSMatthew Dillon 		    tab
9677090df5aSMatthew Dillon 		);
9687090df5aSMatthew Dillon 		blk += radix;
9697090df5aSMatthew Dillon 	}
9707090df5aSMatthew Dillon 	tab -= 4;
9717090df5aSMatthew Dillon 
9727090df5aSMatthew Dillon 	printf(
9737090df5aSMatthew Dillon 	    "%*.*s}\n",
9747090df5aSMatthew Dillon 	    tab, tab, ""
9757090df5aSMatthew Dillon 	);
9767090df5aSMatthew Dillon }
9777090df5aSMatthew Dillon 
9787090df5aSMatthew Dillon #endif
9797090df5aSMatthew Dillon 
9807090df5aSMatthew Dillon #ifdef BLIST_DEBUG
9817090df5aSMatthew Dillon 
9827090df5aSMatthew Dillon int
9837090df5aSMatthew Dillon main(int ac, char **av)
9847090df5aSMatthew Dillon {
9857090df5aSMatthew Dillon 	int size = 1024;
9867090df5aSMatthew Dillon 	int i;
9877090df5aSMatthew Dillon 	blist_t bl;
9887090df5aSMatthew Dillon 
9897090df5aSMatthew Dillon 	for (i = 1; i < ac; ++i) {
9907090df5aSMatthew Dillon 		const char *ptr = av[i];
9917090df5aSMatthew Dillon 		if (*ptr != '-') {
9927090df5aSMatthew Dillon 			size = strtol(ptr, NULL, 0);
9937090df5aSMatthew Dillon 			continue;
9947090df5aSMatthew Dillon 		}
9957090df5aSMatthew Dillon 		ptr += 2;
9967090df5aSMatthew Dillon 		fprintf(stderr, "Bad option: %s\n", ptr - 2);
9977090df5aSMatthew Dillon 		exit(1);
9987090df5aSMatthew Dillon 	}
9997090df5aSMatthew Dillon 	bl = blist_create(size);
10007090df5aSMatthew Dillon 	blist_free(bl, 0, size);
10017090df5aSMatthew Dillon 
10027090df5aSMatthew Dillon 	for (;;) {
10037090df5aSMatthew Dillon 		char buf[1024];
10047090df5aSMatthew Dillon 		daddr_t da = 0;
10057090df5aSMatthew Dillon 		daddr_t count = 0;
10067090df5aSMatthew Dillon 
10077090df5aSMatthew Dillon 
100892da00bbSMatthew Dillon 		printf("%lld/%lld/%lld> ", (long long)bl->bl_free,
100992da00bbSMatthew Dillon 		    (long long)size, (long long)bl->bl_radix);
10107090df5aSMatthew Dillon 		fflush(stdout);
10117090df5aSMatthew Dillon 		if (fgets(buf, sizeof(buf), stdin) == NULL)
10127090df5aSMatthew Dillon 			break;
10137090df5aSMatthew Dillon 		switch(buf[0]) {
10147090df5aSMatthew Dillon 		case 'r':
101592da00bbSMatthew Dillon 			if (sscanf(buf + 1, "%lld", &count) == 1) {
10167090df5aSMatthew Dillon 				blist_resize(&bl, count, 1);
10177090df5aSMatthew Dillon 			} else {
10187090df5aSMatthew Dillon 				printf("?\n");
10197090df5aSMatthew Dillon 			}
10207090df5aSMatthew Dillon 		case 'p':
10217090df5aSMatthew Dillon 			blist_print(bl);
10227090df5aSMatthew Dillon 			break;
10237090df5aSMatthew Dillon 		case 'a':
102492da00bbSMatthew Dillon 			if (sscanf(buf + 1, "%lld", &count) == 1) {
10257090df5aSMatthew Dillon 				daddr_t blk = blist_alloc(bl, count);
102692da00bbSMatthew Dillon 				printf("    R=%08llx\n", (long long)blk);
10277090df5aSMatthew Dillon 			} else {
10287090df5aSMatthew Dillon 				printf("?\n");
10297090df5aSMatthew Dillon 			}
10307090df5aSMatthew Dillon 			break;
10317090df5aSMatthew Dillon 		case 'f':
103292da00bbSMatthew Dillon 			if (sscanf(buf + 1, "%llx %lld",
103392da00bbSMatthew Dillon 			    (long long *)&da, (long long *)&count) == 2) {
10347090df5aSMatthew Dillon 				blist_free(bl, da, count);
10357090df5aSMatthew Dillon 			} else {
10367090df5aSMatthew Dillon 				printf("?\n");
10377090df5aSMatthew Dillon 			}
10387090df5aSMatthew Dillon 			break;
103992da00bbSMatthew Dillon 		case 'l':
104092da00bbSMatthew Dillon 			if (sscanf(buf + 1, "%llx %lld",
104192da00bbSMatthew Dillon 			    (long long *)&da, (long long *)&count) == 2) {
104292da00bbSMatthew Dillon 				printf("    n=%d\n",
104392da00bbSMatthew Dillon 				    blist_fill(bl, da, count));
104492da00bbSMatthew Dillon 			} else {
104592da00bbSMatthew Dillon 				printf("?\n");
104692da00bbSMatthew Dillon 			}
104792da00bbSMatthew Dillon 			break;
10487090df5aSMatthew Dillon 		case '?':
10497090df5aSMatthew Dillon 		case 'h':
10507090df5aSMatthew Dillon 			puts(
10517090df5aSMatthew Dillon 			    "p          -print\n"
10527090df5aSMatthew Dillon 			    "a %d       -allocate\n"
10537090df5aSMatthew Dillon 			    "f %x %d    -free\n"
105492da00bbSMatthew Dillon 			    "l %x %d    -fill\n"
10557090df5aSMatthew Dillon 			    "r %d       -resize\n"
10567090df5aSMatthew Dillon 			    "h/?        -help"
10577090df5aSMatthew Dillon 			);
10587090df5aSMatthew Dillon 			break;
10597090df5aSMatthew Dillon 		default:
10607090df5aSMatthew Dillon 			printf("?\n");
10617090df5aSMatthew Dillon 			break;
10627090df5aSMatthew Dillon 		}
10637090df5aSMatthew Dillon 	}
10647090df5aSMatthew Dillon 	return(0);
10657090df5aSMatthew Dillon }
10667090df5aSMatthew Dillon 
10677090df5aSMatthew Dillon void
10687090df5aSMatthew Dillon panic(const char *ctl, ...)
10697090df5aSMatthew Dillon {
10707090df5aSMatthew Dillon 	va_list va;
10717090df5aSMatthew Dillon 
10727090df5aSMatthew Dillon 	va_start(va, ctl);
10737090df5aSMatthew Dillon 	vfprintf(stderr, ctl, va);
10747090df5aSMatthew Dillon 	fprintf(stderr, "\n");
10757090df5aSMatthew Dillon 	va_end(va);
10767090df5aSMatthew Dillon 	exit(1);
10777090df5aSMatthew Dillon }
10787090df5aSMatthew Dillon 
10797090df5aSMatthew Dillon #endif
10807090df5aSMatthew Dillon 
1081