xref: /onnv-gate/usr/src/uts/common/fs/zfs/zap_leaf.c (revision 5331:3047ad28a67b)
1789Sahrens /*
2789Sahrens  * CDDL HEADER START
3789Sahrens  *
4789Sahrens  * The contents of this file are subject to the terms of the
51491Sahrens  * Common Development and Distribution License (the "License").
61491Sahrens  * You may not use this file except in compliance with the License.
7789Sahrens  *
8789Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9789Sahrens  * or http://www.opensolaris.org/os/licensing.
10789Sahrens  * See the License for the specific language governing permissions
11789Sahrens  * and limitations under the License.
12789Sahrens  *
13789Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
14789Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15789Sahrens  * If applicable, add the following below this CDDL HEADER, with the
16789Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
17789Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
18789Sahrens  *
19789Sahrens  * CDDL HEADER END
20789Sahrens  */
21789Sahrens /*
22*5331Samw  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
23789Sahrens  * Use is subject to license terms.
24789Sahrens  */
25789Sahrens 
26789Sahrens #pragma ident	"%Z%%M%	%I%	%E% SMI"
27789Sahrens 
28789Sahrens /*
29789Sahrens  * The 512-byte leaf is broken into 32 16-byte chunks.
30789Sahrens  * chunk number n means l_chunk[n], even though the header precedes it.
31789Sahrens  * the names are stored null-terminated.
32789Sahrens  */
33789Sahrens 
34789Sahrens #include <sys/zfs_context.h>
35789Sahrens #include <sys/zap.h>
36789Sahrens #include <sys/zap_impl.h>
37789Sahrens #include <sys/zap_leaf.h>
38789Sahrens #include <sys/spa.h>
39789Sahrens #include <sys/dmu.h>
40789Sahrens 
41*5331Samw static uint16_t *zap_leaf_rehash_entry(zap_leaf_t *l, uint16_t entry);
42*5331Samw 
43789Sahrens #define	CHAIN_END 0xffff /* end of the chunk chain */
44789Sahrens 
451491Sahrens /* half the (current) minimum block size */
46789Sahrens #define	MAX_ARRAY_BYTES (8<<10)
47789Sahrens 
48789Sahrens #define	LEAF_HASH(l, h) \
491491Sahrens 	((ZAP_LEAF_HASH_NUMENTRIES(l)-1) & \
501578Sahrens 	((h) >> (64 - ZAP_LEAF_HASH_SHIFT(l)-(l)->l_phys->l_hdr.lh_prefix_len)))
51789Sahrens 
52789Sahrens #define	LEAF_HASH_ENTPTR(l, h) (&(l)->l_phys->l_hash[LEAF_HASH(l, h)])
53789Sahrens 
54789Sahrens 
55789Sahrens static void
56789Sahrens zap_memset(void *a, int c, size_t n)
57789Sahrens {
58789Sahrens 	char *cp = a;
59789Sahrens 	char *cpend = cp + n;
60789Sahrens 
61789Sahrens 	while (cp < cpend)
62789Sahrens 		*cp++ = c;
63789Sahrens }
64789Sahrens 
65789Sahrens static void
66789Sahrens stv(int len, void *addr, uint64_t value)
67789Sahrens {
68789Sahrens 	switch (len) {
69789Sahrens 	case 1:
70789Sahrens 		*(uint8_t *)addr = value;
71789Sahrens 		return;
72789Sahrens 	case 2:
73789Sahrens 		*(uint16_t *)addr = value;
74789Sahrens 		return;
75789Sahrens 	case 4:
76789Sahrens 		*(uint32_t *)addr = value;
77789Sahrens 		return;
78789Sahrens 	case 8:
79789Sahrens 		*(uint64_t *)addr = value;
80789Sahrens 		return;
81789Sahrens 	}
82789Sahrens 	ASSERT(!"bad int len");
83789Sahrens }
84789Sahrens 
85789Sahrens static uint64_t
86789Sahrens ldv(int len, const void *addr)
87789Sahrens {
88789Sahrens 	switch (len) {
89789Sahrens 	case 1:
90789Sahrens 		return (*(uint8_t *)addr);
91789Sahrens 	case 2:
92789Sahrens 		return (*(uint16_t *)addr);
93789Sahrens 	case 4:
94789Sahrens 		return (*(uint32_t *)addr);
95789Sahrens 	case 8:
96789Sahrens 		return (*(uint64_t *)addr);
97789Sahrens 	}
98789Sahrens 	ASSERT(!"bad int len");
992856Snd150628 	return (0xFEEDFACEDEADBEEFULL);
100789Sahrens }
101789Sahrens 
102789Sahrens void
1031491Sahrens zap_leaf_byteswap(zap_leaf_phys_t *buf, int size)
104789Sahrens {
105789Sahrens 	int i;
1061491Sahrens 	zap_leaf_t l;
1071491Sahrens 	l.l_bs = highbit(size)-1;
1081491Sahrens 	l.l_phys = buf;
109789Sahrens 
1101578Sahrens 	buf->l_hdr.lh_block_type = 	BSWAP_64(buf->l_hdr.lh_block_type);
1111578Sahrens 	buf->l_hdr.lh_prefix = 		BSWAP_64(buf->l_hdr.lh_prefix);
1121578Sahrens 	buf->l_hdr.lh_magic = 		BSWAP_32(buf->l_hdr.lh_magic);
1131578Sahrens 	buf->l_hdr.lh_nfree = 		BSWAP_16(buf->l_hdr.lh_nfree);
1141578Sahrens 	buf->l_hdr.lh_nentries = 	BSWAP_16(buf->l_hdr.lh_nentries);
1151578Sahrens 	buf->l_hdr.lh_prefix_len = 	BSWAP_16(buf->l_hdr.lh_prefix_len);
116789Sahrens 	buf->l_hdr.lh_freelist = 	BSWAP_16(buf->l_hdr.lh_freelist);
117789Sahrens 
1181491Sahrens 	for (i = 0; i < ZAP_LEAF_HASH_NUMENTRIES(&l); i++)
119789Sahrens 		buf->l_hash[i] = BSWAP_16(buf->l_hash[i]);
120789Sahrens 
1211491Sahrens 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(&l); i++) {
1221491Sahrens 		zap_leaf_chunk_t *lc = &ZAP_LEAF_CHUNK(&l, i);
123789Sahrens 		struct zap_leaf_entry *le;
124789Sahrens 
1251491Sahrens 		switch (lc->l_free.lf_type) {
1261491Sahrens 		case ZAP_CHUNK_ENTRY:
1271491Sahrens 			le = &lc->l_entry;
128789Sahrens 
1291578Sahrens 			le->le_type =		BSWAP_8(le->le_type);
1301578Sahrens 			le->le_int_size =	BSWAP_8(le->le_int_size);
1311578Sahrens 			le->le_next =		BSWAP_16(le->le_next);
1321578Sahrens 			le->le_name_chunk =	BSWAP_16(le->le_name_chunk);
1331578Sahrens 			le->le_name_length =	BSWAP_16(le->le_name_length);
1341578Sahrens 			le->le_value_chunk =	BSWAP_16(le->le_value_chunk);
1351578Sahrens 			le->le_value_length =	BSWAP_16(le->le_value_length);
1361578Sahrens 			le->le_cd =		BSWAP_32(le->le_cd);
1371578Sahrens 			le->le_hash =		BSWAP_64(le->le_hash);
138789Sahrens 			break;
1391491Sahrens 		case ZAP_CHUNK_FREE:
1401578Sahrens 			lc->l_free.lf_type =	BSWAP_8(lc->l_free.lf_type);
1411578Sahrens 			lc->l_free.lf_next =	BSWAP_16(lc->l_free.lf_next);
142789Sahrens 			break;
1431491Sahrens 		case ZAP_CHUNK_ARRAY:
1441578Sahrens 			lc->l_array.la_type =	BSWAP_8(lc->l_array.la_type);
1451578Sahrens 			lc->l_array.la_next =	BSWAP_16(lc->l_array.la_next);
146789Sahrens 			/* la_array doesn't need swapping */
147789Sahrens 			break;
148789Sahrens 		default:
149789Sahrens 			ASSERT(!"bad leaf type");
150789Sahrens 		}
151789Sahrens 	}
152789Sahrens }
153789Sahrens 
154789Sahrens void
155*5331Samw zap_leaf_init(zap_leaf_t *l, int version)
156789Sahrens {
157789Sahrens 	int i;
158789Sahrens 
1591491Sahrens 	l->l_bs = highbit(l->l_dbuf->db_size)-1;
160789Sahrens 	zap_memset(&l->l_phys->l_hdr, 0, sizeof (struct zap_leaf_header));
1611491Sahrens 	zap_memset(l->l_phys->l_hash, CHAIN_END, 2*ZAP_LEAF_HASH_NUMENTRIES(l));
1621491Sahrens 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(l); i++) {
1631491Sahrens 		ZAP_LEAF_CHUNK(l, i).l_free.lf_type = ZAP_CHUNK_FREE;
1641491Sahrens 		ZAP_LEAF_CHUNK(l, i).l_free.lf_next = i+1;
165789Sahrens 	}
1661491Sahrens 	ZAP_LEAF_CHUNK(l, ZAP_LEAF_NUMCHUNKS(l)-1).l_free.lf_next = CHAIN_END;
1671578Sahrens 	l->l_phys->l_hdr.lh_block_type = ZBT_LEAF;
1681578Sahrens 	l->l_phys->l_hdr.lh_magic = ZAP_LEAF_MAGIC;
1691578Sahrens 	l->l_phys->l_hdr.lh_nfree = ZAP_LEAF_NUMCHUNKS(l);
170*5331Samw 	if (version >= SPA_VERSION_NORMALIZATION)
171*5331Samw 		l->l_phys->l_hdr.lh_flags |= ZLF_ENTRIES_CDSORTED;
172789Sahrens }
173789Sahrens 
174789Sahrens /*
175789Sahrens  * Routines which manipulate leaf chunks (l_chunk[]).
176789Sahrens  */
177789Sahrens 
178789Sahrens static uint16_t
179789Sahrens zap_leaf_chunk_alloc(zap_leaf_t *l)
180789Sahrens {
181789Sahrens 	int chunk;
182789Sahrens 
1831578Sahrens 	ASSERT(l->l_phys->l_hdr.lh_nfree > 0);
184789Sahrens 
185789Sahrens 	chunk = l->l_phys->l_hdr.lh_freelist;
1861491Sahrens 	ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
1871491Sahrens 	ASSERT3U(ZAP_LEAF_CHUNK(l, chunk).l_free.lf_type, ==, ZAP_CHUNK_FREE);
188789Sahrens 
1891491Sahrens 	l->l_phys->l_hdr.lh_freelist = ZAP_LEAF_CHUNK(l, chunk).l_free.lf_next;
190789Sahrens 
1911578Sahrens 	l->l_phys->l_hdr.lh_nfree--;
192789Sahrens 
193789Sahrens 	return (chunk);
194789Sahrens }
195789Sahrens 
196789Sahrens static void
197789Sahrens zap_leaf_chunk_free(zap_leaf_t *l, uint16_t chunk)
198789Sahrens {
1991491Sahrens 	struct zap_leaf_free *zlf = &ZAP_LEAF_CHUNK(l, chunk).l_free;
2001578Sahrens 	ASSERT3U(l->l_phys->l_hdr.lh_nfree, <, ZAP_LEAF_NUMCHUNKS(l));
2011491Sahrens 	ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
2021491Sahrens 	ASSERT(zlf->lf_type != ZAP_CHUNK_FREE);
203789Sahrens 
2041491Sahrens 	zlf->lf_type = ZAP_CHUNK_FREE;
205789Sahrens 	zlf->lf_next = l->l_phys->l_hdr.lh_freelist;
206789Sahrens 	bzero(zlf->lf_pad, sizeof (zlf->lf_pad)); /* help it to compress */
207789Sahrens 	l->l_phys->l_hdr.lh_freelist = chunk;
208789Sahrens 
2091578Sahrens 	l->l_phys->l_hdr.lh_nfree++;
210789Sahrens }
211789Sahrens 
212789Sahrens /*
213789Sahrens  * Routines which manipulate leaf arrays (zap_leaf_array type chunks).
214789Sahrens  */
215789Sahrens 
216789Sahrens static uint16_t
2171578Sahrens zap_leaf_array_create(zap_leaf_t *l, const char *buf,
218789Sahrens 	int integer_size, int num_integers)
219789Sahrens {
220789Sahrens 	uint16_t chunk_head;
221789Sahrens 	uint16_t *chunkp = &chunk_head;
222789Sahrens 	int byten = 0;
223789Sahrens 	uint64_t value;
224789Sahrens 	int shift = (integer_size-1)*8;
225789Sahrens 	int len = num_integers;
226789Sahrens 
227789Sahrens 	ASSERT3U(num_integers * integer_size, <, MAX_ARRAY_BYTES);
228789Sahrens 
229789Sahrens 	while (len > 0) {
230789Sahrens 		uint16_t chunk = zap_leaf_chunk_alloc(l);
2311491Sahrens 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
232789Sahrens 		int i;
233789Sahrens 
2341491Sahrens 		la->la_type = ZAP_CHUNK_ARRAY;
235789Sahrens 		for (i = 0; i < ZAP_LEAF_ARRAY_BYTES; i++) {
236789Sahrens 			if (byten == 0)
237789Sahrens 				value = ldv(integer_size, buf);
2383052Sahrens 			la->la_array[i] = value >> shift;
239789Sahrens 			value <<= 8;
240789Sahrens 			if (++byten == integer_size) {
241789Sahrens 				byten = 0;
242789Sahrens 				buf += integer_size;
243789Sahrens 				if (--len == 0)
244789Sahrens 					break;
245789Sahrens 			}
246789Sahrens 		}
247789Sahrens 
248789Sahrens 		*chunkp = chunk;
249789Sahrens 		chunkp = &la->la_next;
250789Sahrens 	}
251789Sahrens 	*chunkp = CHAIN_END;
252789Sahrens 
253789Sahrens 	return (chunk_head);
254789Sahrens }
255789Sahrens 
256789Sahrens static void
2571578Sahrens zap_leaf_array_free(zap_leaf_t *l, uint16_t *chunkp)
258789Sahrens {
259789Sahrens 	uint16_t chunk = *chunkp;
260789Sahrens 
261789Sahrens 	*chunkp = CHAIN_END;
262789Sahrens 
263789Sahrens 	while (chunk != CHAIN_END) {
2641491Sahrens 		int nextchunk = ZAP_LEAF_CHUNK(l, chunk).l_array.la_next;
2651491Sahrens 		ASSERT3U(ZAP_LEAF_CHUNK(l, chunk).l_array.la_type, ==,
2661491Sahrens 		    ZAP_CHUNK_ARRAY);
267789Sahrens 		zap_leaf_chunk_free(l, chunk);
268789Sahrens 		chunk = nextchunk;
269789Sahrens 	}
270789Sahrens }
271789Sahrens 
272789Sahrens /* array_len and buf_len are in integers, not bytes */
273789Sahrens static void
2741578Sahrens zap_leaf_array_read(zap_leaf_t *l, uint16_t chunk,
275789Sahrens     int array_int_len, int array_len, int buf_int_len, uint64_t buf_len,
276789Sahrens     char *buf)
277789Sahrens {
278789Sahrens 	int len = MIN(array_len, buf_len);
279789Sahrens 	int byten = 0;
280789Sahrens 	uint64_t value = 0;
281789Sahrens 
282789Sahrens 	ASSERT3U(array_int_len, <=, buf_int_len);
283789Sahrens 
284885Sahrens 	/* Fast path for one 8-byte integer */
285885Sahrens 	if (array_int_len == 8 && buf_int_len == 8 && len == 1) {
2861491Sahrens 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
287899Sbonwick 		uint8_t *ip = la->la_array;
288885Sahrens 		uint64_t *buf64 = (uint64_t *)buf;
289899Sbonwick 
290899Sbonwick 		*buf64 = (uint64_t)ip[0] << 56 | (uint64_t)ip[1] << 48 |
291899Sbonwick 		    (uint64_t)ip[2] << 40 | (uint64_t)ip[3] << 32 |
292899Sbonwick 		    (uint64_t)ip[4] << 24 | (uint64_t)ip[5] << 16 |
293899Sbonwick 		    (uint64_t)ip[6] << 8 | (uint64_t)ip[7];
294885Sahrens 		return;
295885Sahrens 	}
296885Sahrens 
297885Sahrens 	/* Fast path for an array of 1-byte integers (eg. the entry name) */
298885Sahrens 	if (array_int_len == 1 && buf_int_len == 1 &&
299885Sahrens 	    buf_len > array_len + ZAP_LEAF_ARRAY_BYTES) {
300885Sahrens 		while (chunk != CHAIN_END) {
301885Sahrens 			struct zap_leaf_array *la =
3021491Sahrens 			    &ZAP_LEAF_CHUNK(l, chunk).l_array;
303885Sahrens 			bcopy(la->la_array, buf, ZAP_LEAF_ARRAY_BYTES);
304885Sahrens 			buf += ZAP_LEAF_ARRAY_BYTES;
305885Sahrens 			chunk = la->la_next;
306885Sahrens 		}
307885Sahrens 		return;
308885Sahrens 	}
309885Sahrens 
310789Sahrens 	while (len > 0) {
3111491Sahrens 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
312789Sahrens 		int i;
313789Sahrens 
3141491Sahrens 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
315789Sahrens 		for (i = 0; i < ZAP_LEAF_ARRAY_BYTES && len > 0; i++) {
316789Sahrens 			value = (value << 8) | la->la_array[i];
317789Sahrens 			byten++;
318789Sahrens 			if (byten == array_int_len) {
319789Sahrens 				stv(buf_int_len, buf, value);
320789Sahrens 				byten = 0;
321789Sahrens 				len--;
322789Sahrens 				if (len == 0)
323789Sahrens 					return;
324789Sahrens 				buf += buf_int_len;
325789Sahrens 			}
326789Sahrens 		}
327789Sahrens 		chunk = la->la_next;
328789Sahrens 	}
329789Sahrens }
330789Sahrens 
331789Sahrens /*
332789Sahrens  * Only to be used on 8-bit arrays.
333789Sahrens  * array_len is actual len in bytes (not encoded le_value_length).
334*5331Samw  * namenorm is null-terminated.
335789Sahrens  */
336*5331Samw static boolean_t
337*5331Samw zap_leaf_array_match(zap_leaf_t *l, zap_name_t *zn, int chunk, int array_len)
338789Sahrens {
339789Sahrens 	int bseen = 0;
340789Sahrens 
341*5331Samw 	if (zn->zn_matchtype == MT_FIRST) {
342*5331Samw 		char *thisname = kmem_alloc(array_len, KM_SLEEP);
343*5331Samw 		boolean_t match;
344*5331Samw 
345*5331Samw 		zap_leaf_array_read(l, chunk, 1, array_len, 1,
346*5331Samw 		    array_len, thisname);
347*5331Samw 		match = zap_match(zn, thisname);
348*5331Samw 		kmem_free(thisname, array_len);
349*5331Samw 		return (match);
350*5331Samw 	}
351*5331Samw 
352*5331Samw 	/* Fast path for exact matching */
353789Sahrens 	while (bseen < array_len) {
3541491Sahrens 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
355789Sahrens 		int toread = MIN(array_len - bseen, ZAP_LEAF_ARRAY_BYTES);
3561491Sahrens 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
357*5331Samw 		if (bcmp(la->la_array, zn->zn_name_orij + bseen, toread))
358789Sahrens 			break;
359789Sahrens 		chunk = la->la_next;
360789Sahrens 		bseen += toread;
361789Sahrens 	}
362789Sahrens 	return (bseen == array_len);
363789Sahrens }
364789Sahrens 
365789Sahrens /*
366789Sahrens  * Routines which manipulate leaf entries.
367789Sahrens  */
368789Sahrens 
369789Sahrens int
370*5331Samw zap_leaf_lookup(zap_leaf_t *l, zap_name_t *zn, zap_entry_handle_t *zeh)
371789Sahrens {
372789Sahrens 	uint16_t *chunkp;
373789Sahrens 	struct zap_leaf_entry *le;
374789Sahrens 
3751578Sahrens 	ASSERT3U(l->l_phys->l_hdr.lh_magic, ==, ZAP_LEAF_MAGIC);
376789Sahrens 
377*5331Samw again:
378*5331Samw 	for (chunkp = LEAF_HASH_ENTPTR(l, zn->zn_hash);
379789Sahrens 	    *chunkp != CHAIN_END; chunkp = &le->le_next) {
380789Sahrens 		uint16_t chunk = *chunkp;
3811491Sahrens 		le = ZAP_LEAF_ENTRY(l, chunk);
382789Sahrens 
3831491Sahrens 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
3841491Sahrens 		ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
385789Sahrens 
386*5331Samw 		if (le->le_hash != zn->zn_hash)
387789Sahrens 			continue;
388789Sahrens 
389*5331Samw 		/*
390*5331Samw 		 * NB: the entry chain is always sorted by cd on
391*5331Samw 		 * normalized zap objects, so this will find the
392*5331Samw 		 * lowest-cd match for MT_FIRST.
393*5331Samw 		 */
394*5331Samw 		ASSERT(zn->zn_matchtype == MT_EXACT ||
395*5331Samw 		    (l->l_phys->l_hdr.lh_flags & ZLF_ENTRIES_CDSORTED));
396*5331Samw 		if (zap_leaf_array_match(l, zn, le->le_name_chunk,
397*5331Samw 		    le->le_name_length)) {
398789Sahrens 			zeh->zeh_num_integers = le->le_value_length;
399789Sahrens 			zeh->zeh_integer_size = le->le_int_size;
400789Sahrens 			zeh->zeh_cd = le->le_cd;
401789Sahrens 			zeh->zeh_hash = le->le_hash;
402789Sahrens 			zeh->zeh_chunkp = chunkp;
4031578Sahrens 			zeh->zeh_leaf = l;
404789Sahrens 			return (0);
405789Sahrens 		}
406789Sahrens 	}
407789Sahrens 
408*5331Samw 	/*
409*5331Samw 	 * NB: we could of course do this in one pass, but that would be
410*5331Samw 	 * a pain.  We'll see if MT_BEST is even used much.
411*5331Samw 	 */
412*5331Samw 	if (zn->zn_matchtype == MT_BEST) {
413*5331Samw 		zn->zn_matchtype = MT_FIRST;
414*5331Samw 		goto again;
415*5331Samw 	}
416*5331Samw 
417789Sahrens 	return (ENOENT);
418789Sahrens }
419789Sahrens 
420789Sahrens /* Return (h1,cd1 >= h2,cd2) */
421885Sahrens #define	HCD_GTEQ(h1, cd1, h2, cd2) \
422885Sahrens 	((h1 > h2) ? TRUE : ((h1 == h2 && cd1 >= cd2) ? TRUE : FALSE))
423789Sahrens 
424789Sahrens int
425789Sahrens zap_leaf_lookup_closest(zap_leaf_t *l,
426789Sahrens     uint64_t h, uint32_t cd, zap_entry_handle_t *zeh)
427789Sahrens {
428789Sahrens 	uint16_t chunk;
429789Sahrens 	uint64_t besth = -1ULL;
430789Sahrens 	uint32_t bestcd = ZAP_MAXCD;
4311491Sahrens 	uint16_t bestlh = ZAP_LEAF_HASH_NUMENTRIES(l)-1;
432789Sahrens 	uint16_t lh;
433789Sahrens 	struct zap_leaf_entry *le;
434789Sahrens 
4351578Sahrens 	ASSERT3U(l->l_phys->l_hdr.lh_magic, ==, ZAP_LEAF_MAGIC);
436789Sahrens 
437789Sahrens 	for (lh = LEAF_HASH(l, h); lh <= bestlh; lh++) {
438789Sahrens 		for (chunk = l->l_phys->l_hash[lh];
439789Sahrens 		    chunk != CHAIN_END; chunk = le->le_next) {
4401491Sahrens 			le = ZAP_LEAF_ENTRY(l, chunk);
441789Sahrens 
4421491Sahrens 			ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
4431491Sahrens 			ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
444789Sahrens 
445885Sahrens 			if (HCD_GTEQ(le->le_hash, le->le_cd, h, cd) &&
446885Sahrens 			    HCD_GTEQ(besth, bestcd, le->le_hash, le->le_cd)) {
447789Sahrens 				ASSERT3U(bestlh, >=, lh);
448789Sahrens 				bestlh = lh;
449789Sahrens 				besth = le->le_hash;
450789Sahrens 				bestcd = le->le_cd;
451789Sahrens 
452789Sahrens 				zeh->zeh_num_integers = le->le_value_length;
453789Sahrens 				zeh->zeh_integer_size = le->le_int_size;
454789Sahrens 				zeh->zeh_cd = le->le_cd;
455789Sahrens 				zeh->zeh_hash = le->le_hash;
456789Sahrens 				zeh->zeh_fakechunk = chunk;
457789Sahrens 				zeh->zeh_chunkp = &zeh->zeh_fakechunk;
4581578Sahrens 				zeh->zeh_leaf = l;
459789Sahrens 			}
460789Sahrens 		}
461789Sahrens 	}
462789Sahrens 
463789Sahrens 	return (bestcd == ZAP_MAXCD ? ENOENT : 0);
464789Sahrens }
465789Sahrens 
466789Sahrens int
467789Sahrens zap_entry_read(const zap_entry_handle_t *zeh,
468789Sahrens     uint8_t integer_size, uint64_t num_integers, void *buf)
469789Sahrens {
4701491Sahrens 	struct zap_leaf_entry *le =
4711578Sahrens 	    ZAP_LEAF_ENTRY(zeh->zeh_leaf, *zeh->zeh_chunkp);
4721491Sahrens 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
473789Sahrens 
474789Sahrens 	if (le->le_int_size > integer_size)
475789Sahrens 		return (EINVAL);
476789Sahrens 
4771578Sahrens 	zap_leaf_array_read(zeh->zeh_leaf, le->le_value_chunk, le->le_int_size,
478789Sahrens 	    le->le_value_length, integer_size, num_integers, buf);
479789Sahrens 
480789Sahrens 	if (zeh->zeh_num_integers > num_integers)
481789Sahrens 		return (EOVERFLOW);
482789Sahrens 	return (0);
483789Sahrens 
484789Sahrens }
485789Sahrens 
486789Sahrens int
487789Sahrens zap_entry_read_name(const zap_entry_handle_t *zeh, uint16_t buflen, char *buf)
488789Sahrens {
4891491Sahrens 	struct zap_leaf_entry *le =
4901578Sahrens 	    ZAP_LEAF_ENTRY(zeh->zeh_leaf, *zeh->zeh_chunkp);
4911491Sahrens 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
492789Sahrens 
4931578Sahrens 	zap_leaf_array_read(zeh->zeh_leaf, le->le_name_chunk, 1,
494789Sahrens 	    le->le_name_length, 1, buflen, buf);
495789Sahrens 	if (le->le_name_length > buflen)
496789Sahrens 		return (EOVERFLOW);
497789Sahrens 	return (0);
498789Sahrens }
499789Sahrens 
500789Sahrens int
501789Sahrens zap_entry_update(zap_entry_handle_t *zeh,
502789Sahrens 	uint8_t integer_size, uint64_t num_integers, const void *buf)
503789Sahrens {
504789Sahrens 	int delta_chunks;
5051578Sahrens 	zap_leaf_t *l = zeh->zeh_leaf;
5061578Sahrens 	struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, *zeh->zeh_chunkp);
507789Sahrens 
5081578Sahrens 	delta_chunks = ZAP_LEAF_ARRAY_NCHUNKS(num_integers * integer_size) -
5091578Sahrens 	    ZAP_LEAF_ARRAY_NCHUNKS(le->le_value_length * le->le_int_size);
510789Sahrens 
5111578Sahrens 	if ((int)l->l_phys->l_hdr.lh_nfree < delta_chunks)
512789Sahrens 		return (EAGAIN);
513789Sahrens 
514789Sahrens 	/*
515789Sahrens 	 * We should search other chained leaves (via
516789Sahrens 	 * zap_entry_remove,create?) otherwise returning EAGAIN will
517789Sahrens 	 * just send us into an infinite loop if we have to chain
518789Sahrens 	 * another leaf block, rather than being able to split this
519789Sahrens 	 * block.
520789Sahrens 	 */
521789Sahrens 
5221578Sahrens 	zap_leaf_array_free(l, &le->le_value_chunk);
523789Sahrens 	le->le_value_chunk =
5241578Sahrens 	    zap_leaf_array_create(l, buf, integer_size, num_integers);
5251491Sahrens 	le->le_value_length = num_integers;
526789Sahrens 	le->le_int_size = integer_size;
527789Sahrens 	return (0);
528789Sahrens }
529789Sahrens 
530789Sahrens void
531789Sahrens zap_entry_remove(zap_entry_handle_t *zeh)
532789Sahrens {
533789Sahrens 	uint16_t entry_chunk;
534789Sahrens 	struct zap_leaf_entry *le;
5351578Sahrens 	zap_leaf_t *l = zeh->zeh_leaf;
536789Sahrens 
537789Sahrens 	ASSERT3P(zeh->zeh_chunkp, !=, &zeh->zeh_fakechunk);
538789Sahrens 
539789Sahrens 	entry_chunk = *zeh->zeh_chunkp;
5401491Sahrens 	le = ZAP_LEAF_ENTRY(l, entry_chunk);
5411491Sahrens 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
542789Sahrens 
5431578Sahrens 	zap_leaf_array_free(l, &le->le_name_chunk);
5441578Sahrens 	zap_leaf_array_free(l, &le->le_value_chunk);
545789Sahrens 
546789Sahrens 	*zeh->zeh_chunkp = le->le_next;
547789Sahrens 	zap_leaf_chunk_free(l, entry_chunk);
548789Sahrens 
5491578Sahrens 	l->l_phys->l_hdr.lh_nentries--;
550789Sahrens }
551789Sahrens 
552789Sahrens int
553789Sahrens zap_entry_create(zap_leaf_t *l, const char *name, uint64_t h, uint32_t cd,
554789Sahrens     uint8_t integer_size, uint64_t num_integers, const void *buf,
555789Sahrens     zap_entry_handle_t *zeh)
556789Sahrens {
557789Sahrens 	uint16_t chunk;
558789Sahrens 	uint16_t *chunkp;
559789Sahrens 	struct zap_leaf_entry *le;
560789Sahrens 	uint64_t namelen, valuelen;
561789Sahrens 	int numchunks;
562789Sahrens 
563789Sahrens 	valuelen = integer_size * num_integers;
564789Sahrens 	namelen = strlen(name) + 1;
565789Sahrens 	ASSERT(namelen >= 2);
566789Sahrens 
5671578Sahrens 	numchunks = 1 + ZAP_LEAF_ARRAY_NCHUNKS(namelen) +
5681578Sahrens 	    ZAP_LEAF_ARRAY_NCHUNKS(valuelen);
5691491Sahrens 	if (numchunks > ZAP_LEAF_NUMCHUNKS(l))
570789Sahrens 		return (E2BIG);
571789Sahrens 
572789Sahrens 	if (cd == ZAP_MAXCD) {
573*5331Samw 		/* find the lowest unused cd */
574*5331Samw 		if (l->l_phys->l_hdr.lh_flags & ZLF_ENTRIES_CDSORTED) {
575*5331Samw 			cd = 0;
576*5331Samw 
5771578Sahrens 			for (chunk = *LEAF_HASH_ENTPTR(l, h);
5781578Sahrens 			    chunk != CHAIN_END; chunk = le->le_next) {
5791578Sahrens 				le = ZAP_LEAF_ENTRY(l, chunk);
580*5331Samw 				if (le->le_cd > cd)
5811578Sahrens 					break;
582*5331Samw 				if (le->le_hash == h) {
583*5331Samw 					ASSERT3U(cd, ==, le->le_cd);
584*5331Samw 					cd++;
585789Sahrens 				}
586789Sahrens 			}
587*5331Samw 		} else {
588*5331Samw 			/* old unsorted format; do it the O(n^2) way */
589*5331Samw 			for (cd = 0; cd < ZAP_MAXCD; cd++) {
590*5331Samw 				for (chunk = *LEAF_HASH_ENTPTR(l, h);
591*5331Samw 				    chunk != CHAIN_END; chunk = le->le_next) {
592*5331Samw 					le = ZAP_LEAF_ENTRY(l, chunk);
593*5331Samw 					if (le->le_hash == h &&
594*5331Samw 					    le->le_cd == cd) {
595*5331Samw 						break;
596*5331Samw 					}
597*5331Samw 				}
598*5331Samw 				/* If this cd is not in use, we are good. */
599*5331Samw 				if (chunk == CHAIN_END)
600*5331Samw 					break;
601*5331Samw 			}
602789Sahrens 		}
603*5331Samw 		/*
604*5331Samw 		 * we would run out of space in a block before we could
605*5331Samw 		 * have ZAP_MAXCD entries
606*5331Samw 		 */
607*5331Samw 		ASSERT3U(cd, <, ZAP_MAXCD);
608789Sahrens 	}
609789Sahrens 
6101578Sahrens 	if (l->l_phys->l_hdr.lh_nfree < numchunks)
611789Sahrens 		return (EAGAIN);
612789Sahrens 
613789Sahrens 	/* make the entry */
614789Sahrens 	chunk = zap_leaf_chunk_alloc(l);
6151491Sahrens 	le = ZAP_LEAF_ENTRY(l, chunk);
6161491Sahrens 	le->le_type = ZAP_CHUNK_ENTRY;
6171578Sahrens 	le->le_name_chunk = zap_leaf_array_create(l, name, 1, namelen);
618789Sahrens 	le->le_name_length = namelen;
619789Sahrens 	le->le_value_chunk =
6201578Sahrens 	    zap_leaf_array_create(l, buf, integer_size, num_integers);
6211491Sahrens 	le->le_value_length = num_integers;
622789Sahrens 	le->le_int_size = integer_size;
623789Sahrens 	le->le_hash = h;
624789Sahrens 	le->le_cd = cd;
625789Sahrens 
626789Sahrens 	/* link it into the hash chain */
627*5331Samw 	/* XXX if we did the search above, we could just use that */
628*5331Samw 	chunkp = zap_leaf_rehash_entry(l, chunk);
629789Sahrens 
6301578Sahrens 	l->l_phys->l_hdr.lh_nentries++;
631789Sahrens 
6321578Sahrens 	zeh->zeh_leaf = l;
633789Sahrens 	zeh->zeh_num_integers = num_integers;
634789Sahrens 	zeh->zeh_integer_size = le->le_int_size;
635789Sahrens 	zeh->zeh_cd = le->le_cd;
636789Sahrens 	zeh->zeh_hash = le->le_hash;
637789Sahrens 	zeh->zeh_chunkp = chunkp;
638789Sahrens 
639789Sahrens 	return (0);
640789Sahrens }
641789Sahrens 
642789Sahrens /*
643*5331Samw  * Determine if there is another entry with the same normalized form.
644*5331Samw  * For performance purposes, either zn or name must be provided (the
645*5331Samw  * other can be NULL).  Note, there usually won't be any hash
646*5331Samw  * conflicts, in which case we don't need the concatenated/normalized
647*5331Samw  * form of the name.  But all callers have one of these on hand anyway,
648*5331Samw  * so might as well take advantage.  A cleaner but slower interface
649*5331Samw  * would accept neither argument, and compute the normalized name as
650*5331Samw  * needed (using zap_name_alloc(zap_entry_read_name(zeh))).
651*5331Samw  */
652*5331Samw boolean_t
653*5331Samw zap_entry_normalization_conflict(zap_entry_handle_t *zeh, zap_name_t *zn,
654*5331Samw     const char *name, zap_t *zap)
655*5331Samw {
656*5331Samw 	uint64_t chunk;
657*5331Samw 	struct zap_leaf_entry *le;
658*5331Samw 	boolean_t allocdzn = B_FALSE;
659*5331Samw 
660*5331Samw 	if (zap->zap_normflags == 0)
661*5331Samw 		return (B_FALSE);
662*5331Samw 
663*5331Samw 	for (chunk = *LEAF_HASH_ENTPTR(zeh->zeh_leaf, zeh->zeh_hash);
664*5331Samw 	    chunk != CHAIN_END; chunk = le->le_next) {
665*5331Samw 		le = ZAP_LEAF_ENTRY(zeh->zeh_leaf, chunk);
666*5331Samw 		if (le->le_hash != zeh->zeh_hash)
667*5331Samw 			continue;
668*5331Samw 		if (le->le_cd == zeh->zeh_cd)
669*5331Samw 			continue;
670*5331Samw 
671*5331Samw 		if (zn == NULL) {
672*5331Samw 			zn = zap_name_alloc(zap, name, MT_FIRST);
673*5331Samw 			allocdzn = B_TRUE;
674*5331Samw 		}
675*5331Samw 		if (zap_leaf_array_match(zeh->zeh_leaf, zn,
676*5331Samw 		    le->le_name_chunk, le->le_name_length)) {
677*5331Samw 			if (allocdzn)
678*5331Samw 				zap_name_free(zn);
679*5331Samw 			return (B_TRUE);
680*5331Samw 		}
681*5331Samw 	}
682*5331Samw 	if (allocdzn)
683*5331Samw 		zap_name_free(zn);
684*5331Samw 	return (B_FALSE);
685*5331Samw }
686*5331Samw 
687*5331Samw /*
688789Sahrens  * Routines for transferring entries between leafs.
689789Sahrens  */
690789Sahrens 
691*5331Samw static uint16_t *
692789Sahrens zap_leaf_rehash_entry(zap_leaf_t *l, uint16_t entry)
693789Sahrens {
6941491Sahrens 	struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, entry);
695*5331Samw 	struct zap_leaf_entry *le2;
696*5331Samw 	uint16_t *chunkp;
697*5331Samw 
698*5331Samw 	/*
699*5331Samw 	 * keep the entry chain sorted by cd
700*5331Samw 	 * NB: this will not cause problems for unsorted leafs, though
701*5331Samw 	 * it is unnecessary there.
702*5331Samw 	 */
703*5331Samw 	for (chunkp = LEAF_HASH_ENTPTR(l, le->le_hash);
704*5331Samw 	    *chunkp != CHAIN_END; chunkp = &le2->le_next) {
705*5331Samw 		le2 = ZAP_LEAF_ENTRY(l, *chunkp);
706*5331Samw 		if (le2->le_cd > le->le_cd)
707*5331Samw 			break;
708*5331Samw 	}
709*5331Samw 
710*5331Samw 	le->le_next = *chunkp;
711*5331Samw 	*chunkp = entry;
712*5331Samw 	return (chunkp);
713789Sahrens }
714789Sahrens 
715789Sahrens static uint16_t
716789Sahrens zap_leaf_transfer_array(zap_leaf_t *l, uint16_t chunk, zap_leaf_t *nl)
717789Sahrens {
718789Sahrens 	uint16_t new_chunk;
719789Sahrens 	uint16_t *nchunkp = &new_chunk;
720789Sahrens 
721789Sahrens 	while (chunk != CHAIN_END) {
722789Sahrens 		uint16_t nchunk = zap_leaf_chunk_alloc(nl);
723789Sahrens 		struct zap_leaf_array *nla =
7241491Sahrens 		    &ZAP_LEAF_CHUNK(nl, nchunk).l_array;
725789Sahrens 		struct zap_leaf_array *la =
7261491Sahrens 		    &ZAP_LEAF_CHUNK(l, chunk).l_array;
727789Sahrens 		int nextchunk = la->la_next;
728789Sahrens 
7291491Sahrens 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
7301491Sahrens 		ASSERT3U(nchunk, <, ZAP_LEAF_NUMCHUNKS(l));
731789Sahrens 
7321578Sahrens 		*nla = *la; /* structure assignment */
733789Sahrens 
734789Sahrens 		zap_leaf_chunk_free(l, chunk);
735789Sahrens 		chunk = nextchunk;
736789Sahrens 		*nchunkp = nchunk;
737789Sahrens 		nchunkp = &nla->la_next;
738789Sahrens 	}
739789Sahrens 	*nchunkp = CHAIN_END;
740789Sahrens 	return (new_chunk);
741789Sahrens }
742789Sahrens 
743789Sahrens static void
7441578Sahrens zap_leaf_transfer_entry(zap_leaf_t *l, int entry, zap_leaf_t *nl)
745789Sahrens {
746789Sahrens 	struct zap_leaf_entry *le, *nle;
7471578Sahrens 	uint16_t chunk;
748789Sahrens 
7491491Sahrens 	le = ZAP_LEAF_ENTRY(l, entry);
7501491Sahrens 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
751789Sahrens 
752789Sahrens 	chunk = zap_leaf_chunk_alloc(nl);
7531491Sahrens 	nle = ZAP_LEAF_ENTRY(nl, chunk);
7541578Sahrens 	*nle = *le; /* structure assignment */
755789Sahrens 
756*5331Samw 	(void) zap_leaf_rehash_entry(nl, chunk);
757789Sahrens 
758789Sahrens 	nle->le_name_chunk = zap_leaf_transfer_array(l, le->le_name_chunk, nl);
759789Sahrens 	nle->le_value_chunk =
760789Sahrens 	    zap_leaf_transfer_array(l, le->le_value_chunk, nl);
761789Sahrens 
762789Sahrens 	zap_leaf_chunk_free(l, entry);
763789Sahrens 
7641578Sahrens 	l->l_phys->l_hdr.lh_nentries--;
7651578Sahrens 	nl->l_phys->l_hdr.lh_nentries++;
766789Sahrens }
767789Sahrens 
768789Sahrens /*
7691578Sahrens  * Transfer the entries whose hash prefix ends in 1 to the new leaf.
770789Sahrens  */
7711578Sahrens void
772*5331Samw zap_leaf_split(zap_leaf_t *l, zap_leaf_t *nl, int version)
773789Sahrens {
774789Sahrens 	int i;
7751578Sahrens 	int bit = 64 - 1 - l->l_phys->l_hdr.lh_prefix_len;
776789Sahrens 
7771578Sahrens 	/* set new prefix and prefix_len */
7781578Sahrens 	l->l_phys->l_hdr.lh_prefix <<= 1;
7791578Sahrens 	l->l_phys->l_hdr.lh_prefix_len++;
7801578Sahrens 	nl->l_phys->l_hdr.lh_prefix = l->l_phys->l_hdr.lh_prefix | 1;
7811578Sahrens 	nl->l_phys->l_hdr.lh_prefix_len = l->l_phys->l_hdr.lh_prefix_len;
7821578Sahrens 
783789Sahrens 	/* break existing hash chains */
7841491Sahrens 	zap_memset(l->l_phys->l_hash, CHAIN_END, 2*ZAP_LEAF_HASH_NUMENTRIES(l));
785789Sahrens 
786*5331Samw 	if (version >= SPA_VERSION_NORMALIZATION)
787*5331Samw 		l->l_phys->l_hdr.lh_flags |= ZLF_ENTRIES_CDSORTED;
788*5331Samw 
7891578Sahrens 	/*
7901578Sahrens 	 * Transfer entries whose hash bit 'bit' is set to nl; rehash
7911578Sahrens 	 * the remaining entries
7921578Sahrens 	 *
7931578Sahrens 	 * NB: We could find entries via the hashtable instead. That
7941578Sahrens 	 * would be O(hashents+numents) rather than O(numblks+numents),
7951578Sahrens 	 * but this accesses memory more sequentially, and when we're
7961578Sahrens 	 * called, the block is usually pretty full.
7971578Sahrens 	 */
7981491Sahrens 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(l); i++) {
7991491Sahrens 		struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, i);
8001491Sahrens 		if (le->le_type != ZAP_CHUNK_ENTRY)
801789Sahrens 			continue;
802789Sahrens 
8031578Sahrens 		if (le->le_hash & (1ULL << bit))
8041578Sahrens 			zap_leaf_transfer_entry(l, i, nl);
8051578Sahrens 		else
806*5331Samw 			(void) zap_leaf_rehash_entry(l, i);
807789Sahrens 	}
808789Sahrens }
809789Sahrens 
810789Sahrens void
8111578Sahrens zap_leaf_stats(zap_t *zap, zap_leaf_t *l, zap_stats_t *zs)
812789Sahrens {
8131578Sahrens 	int i, n;
814789Sahrens 
8151578Sahrens 	n = zap->zap_f.zap_phys->zap_ptrtbl.zt_shift -
8161578Sahrens 	    l->l_phys->l_hdr.lh_prefix_len;
817789Sahrens 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
818789Sahrens 	zs->zs_leafs_with_2n_pointers[n]++;
819789Sahrens 
820789Sahrens 
8211578Sahrens 	n = l->l_phys->l_hdr.lh_nentries/5;
8221578Sahrens 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
8231578Sahrens 	zs->zs_blocks_with_n5_entries[n]++;
824789Sahrens 
8251578Sahrens 	n = ((1<<FZAP_BLOCK_SHIFT(zap)) -
8261578Sahrens 	    l->l_phys->l_hdr.lh_nfree * (ZAP_LEAF_ARRAY_BYTES+1))*10 /
8271578Sahrens 	    (1<<FZAP_BLOCK_SHIFT(zap));
8281578Sahrens 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
8291578Sahrens 	zs->zs_blocks_n_tenths_full[n]++;
830789Sahrens 
8311578Sahrens 	for (i = 0; i < ZAP_LEAF_HASH_NUMENTRIES(l); i++) {
8321578Sahrens 		int nentries = 0;
8331578Sahrens 		int chunk = l->l_phys->l_hash[i];
8341578Sahrens 
8351578Sahrens 		while (chunk != CHAIN_END) {
8361578Sahrens 			struct zap_leaf_entry *le =
8371578Sahrens 			    ZAP_LEAF_ENTRY(l, chunk);
838789Sahrens 
8391578Sahrens 			n = 1 + ZAP_LEAF_ARRAY_NCHUNKS(le->le_name_length) +
8401578Sahrens 			    ZAP_LEAF_ARRAY_NCHUNKS(le->le_value_length *
8411578Sahrens 				le->le_int_size);
8421578Sahrens 			n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
8431578Sahrens 			zs->zs_entries_using_n_chunks[n]++;
844789Sahrens 
8451578Sahrens 			chunk = le->le_next;
8461578Sahrens 			nentries++;
847789Sahrens 		}
848789Sahrens 
8491578Sahrens 		n = nentries;
8501578Sahrens 		n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
8511578Sahrens 		zs->zs_buckets_with_n_entries[n]++;
8521578Sahrens 	}
853789Sahrens }
854