xref: /onnv-gate/usr/src/uts/common/fs/zfs/zap_leaf.c (revision 1578:1fd8ab638fc0)
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 /*
221491Sahrens  * Copyright 2006 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 
41789Sahrens #define	CHAIN_END 0xffff /* end of the chunk chain */
42789Sahrens 
431491Sahrens /* half the (current) minimum block size */
44789Sahrens #define	MAX_ARRAY_BYTES (8<<10)
45789Sahrens 
46789Sahrens #define	LEAF_HASH(l, h) \
471491Sahrens 	((ZAP_LEAF_HASH_NUMENTRIES(l)-1) & \
48*1578Sahrens 	((h) >> (64 - ZAP_LEAF_HASH_SHIFT(l)-(l)->l_phys->l_hdr.lh_prefix_len)))
49789Sahrens 
50789Sahrens #define	LEAF_HASH_ENTPTR(l, h) (&(l)->l_phys->l_hash[LEAF_HASH(l, h)])
51789Sahrens 
52789Sahrens 
53789Sahrens static void
54789Sahrens zap_memset(void *a, int c, size_t n)
55789Sahrens {
56789Sahrens 	char *cp = a;
57789Sahrens 	char *cpend = cp + n;
58789Sahrens 
59789Sahrens 	while (cp < cpend)
60789Sahrens 		*cp++ = c;
61789Sahrens }
62789Sahrens 
63789Sahrens static void
64789Sahrens stv(int len, void *addr, uint64_t value)
65789Sahrens {
66789Sahrens 	switch (len) {
67789Sahrens 	case 1:
68789Sahrens 		*(uint8_t *)addr = value;
69789Sahrens 		return;
70789Sahrens 	case 2:
71789Sahrens 		*(uint16_t *)addr = value;
72789Sahrens 		return;
73789Sahrens 	case 4:
74789Sahrens 		*(uint32_t *)addr = value;
75789Sahrens 		return;
76789Sahrens 	case 8:
77789Sahrens 		*(uint64_t *)addr = value;
78789Sahrens 		return;
79789Sahrens 	}
80789Sahrens 	ASSERT(!"bad int len");
81789Sahrens }
82789Sahrens 
83789Sahrens static uint64_t
84789Sahrens ldv(int len, const void *addr)
85789Sahrens {
86789Sahrens 	switch (len) {
87789Sahrens 	case 1:
88789Sahrens 		return (*(uint8_t *)addr);
89789Sahrens 	case 2:
90789Sahrens 		return (*(uint16_t *)addr);
91789Sahrens 	case 4:
92789Sahrens 		return (*(uint32_t *)addr);
93789Sahrens 	case 8:
94789Sahrens 		return (*(uint64_t *)addr);
95789Sahrens 	}
96789Sahrens 	ASSERT(!"bad int len");
97789Sahrens 	return (0xFEEDFACEDEADBEEF);
98789Sahrens }
99789Sahrens 
100789Sahrens void
1011491Sahrens zap_leaf_byteswap(zap_leaf_phys_t *buf, int size)
102789Sahrens {
103789Sahrens 	int i;
1041491Sahrens 	zap_leaf_t l;
1051491Sahrens 	l.l_bs = highbit(size)-1;
1061491Sahrens 	l.l_phys = buf;
107789Sahrens 
108*1578Sahrens 	buf->l_hdr.lh_block_type = 	BSWAP_64(buf->l_hdr.lh_block_type);
109*1578Sahrens 	buf->l_hdr.lh_prefix = 		BSWAP_64(buf->l_hdr.lh_prefix);
110*1578Sahrens 	buf->l_hdr.lh_magic = 		BSWAP_32(buf->l_hdr.lh_magic);
111*1578Sahrens 	buf->l_hdr.lh_nfree = 		BSWAP_16(buf->l_hdr.lh_nfree);
112*1578Sahrens 	buf->l_hdr.lh_nentries = 	BSWAP_16(buf->l_hdr.lh_nentries);
113*1578Sahrens 	buf->l_hdr.lh_prefix_len = 	BSWAP_16(buf->l_hdr.lh_prefix_len);
114789Sahrens 	buf->l_hdr.lh_freelist = 	BSWAP_16(buf->l_hdr.lh_freelist);
115789Sahrens 
1161491Sahrens 	for (i = 0; i < ZAP_LEAF_HASH_NUMENTRIES(&l); i++)
117789Sahrens 		buf->l_hash[i] = BSWAP_16(buf->l_hash[i]);
118789Sahrens 
1191491Sahrens 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(&l); i++) {
1201491Sahrens 		zap_leaf_chunk_t *lc = &ZAP_LEAF_CHUNK(&l, i);
121789Sahrens 		struct zap_leaf_entry *le;
122789Sahrens 
1231491Sahrens 		switch (lc->l_free.lf_type) {
1241491Sahrens 		case ZAP_CHUNK_ENTRY:
1251491Sahrens 			le = &lc->l_entry;
126789Sahrens 
127*1578Sahrens 			le->le_type =		BSWAP_8(le->le_type);
128*1578Sahrens 			le->le_int_size =	BSWAP_8(le->le_int_size);
129*1578Sahrens 			le->le_next =		BSWAP_16(le->le_next);
130*1578Sahrens 			le->le_name_chunk =	BSWAP_16(le->le_name_chunk);
131*1578Sahrens 			le->le_name_length =	BSWAP_16(le->le_name_length);
132*1578Sahrens 			le->le_value_chunk =	BSWAP_16(le->le_value_chunk);
133*1578Sahrens 			le->le_value_length =	BSWAP_16(le->le_value_length);
134*1578Sahrens 			le->le_cd =		BSWAP_32(le->le_cd);
135*1578Sahrens 			le->le_hash =		BSWAP_64(le->le_hash);
136789Sahrens 			break;
1371491Sahrens 		case ZAP_CHUNK_FREE:
138*1578Sahrens 			lc->l_free.lf_type =	BSWAP_8(lc->l_free.lf_type);
139*1578Sahrens 			lc->l_free.lf_next =	BSWAP_16(lc->l_free.lf_next);
140789Sahrens 			break;
1411491Sahrens 		case ZAP_CHUNK_ARRAY:
142*1578Sahrens 			lc->l_array.la_type =	BSWAP_8(lc->l_array.la_type);
143*1578Sahrens 			lc->l_array.la_next =	BSWAP_16(lc->l_array.la_next);
144789Sahrens 			/* la_array doesn't need swapping */
145789Sahrens 			break;
146789Sahrens 		default:
147789Sahrens 			ASSERT(!"bad leaf type");
148789Sahrens 		}
149789Sahrens 	}
150789Sahrens }
151789Sahrens 
152789Sahrens void
153789Sahrens zap_leaf_init(zap_leaf_t *l)
154789Sahrens {
155789Sahrens 	int i;
156789Sahrens 
1571491Sahrens 	l->l_bs = highbit(l->l_dbuf->db_size)-1;
158789Sahrens 	zap_memset(&l->l_phys->l_hdr, 0, sizeof (struct zap_leaf_header));
1591491Sahrens 	zap_memset(l->l_phys->l_hash, CHAIN_END, 2*ZAP_LEAF_HASH_NUMENTRIES(l));
1601491Sahrens 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(l); i++) {
1611491Sahrens 		ZAP_LEAF_CHUNK(l, i).l_free.lf_type = ZAP_CHUNK_FREE;
1621491Sahrens 		ZAP_LEAF_CHUNK(l, i).l_free.lf_next = i+1;
163789Sahrens 	}
1641491Sahrens 	ZAP_LEAF_CHUNK(l, ZAP_LEAF_NUMCHUNKS(l)-1).l_free.lf_next = CHAIN_END;
165*1578Sahrens 	l->l_phys->l_hdr.lh_block_type = ZBT_LEAF;
166*1578Sahrens 	l->l_phys->l_hdr.lh_magic = ZAP_LEAF_MAGIC;
167*1578Sahrens 	l->l_phys->l_hdr.lh_nfree = ZAP_LEAF_NUMCHUNKS(l);
168789Sahrens }
169789Sahrens 
170789Sahrens /*
171789Sahrens  * Routines which manipulate leaf chunks (l_chunk[]).
172789Sahrens  */
173789Sahrens 
174789Sahrens static uint16_t
175789Sahrens zap_leaf_chunk_alloc(zap_leaf_t *l)
176789Sahrens {
177789Sahrens 	int chunk;
178789Sahrens 
179*1578Sahrens 	ASSERT(l->l_phys->l_hdr.lh_nfree > 0);
180789Sahrens 
181789Sahrens 	chunk = l->l_phys->l_hdr.lh_freelist;
1821491Sahrens 	ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
1831491Sahrens 	ASSERT3U(ZAP_LEAF_CHUNK(l, chunk).l_free.lf_type, ==, ZAP_CHUNK_FREE);
184789Sahrens 
1851491Sahrens 	l->l_phys->l_hdr.lh_freelist = ZAP_LEAF_CHUNK(l, chunk).l_free.lf_next;
186789Sahrens 
187*1578Sahrens 	l->l_phys->l_hdr.lh_nfree--;
188789Sahrens 
189789Sahrens 	return (chunk);
190789Sahrens }
191789Sahrens 
192789Sahrens static void
193789Sahrens zap_leaf_chunk_free(zap_leaf_t *l, uint16_t chunk)
194789Sahrens {
1951491Sahrens 	struct zap_leaf_free *zlf = &ZAP_LEAF_CHUNK(l, chunk).l_free;
196*1578Sahrens 	ASSERT3U(l->l_phys->l_hdr.lh_nfree, <, ZAP_LEAF_NUMCHUNKS(l));
1971491Sahrens 	ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
1981491Sahrens 	ASSERT(zlf->lf_type != ZAP_CHUNK_FREE);
199789Sahrens 
2001491Sahrens 	zlf->lf_type = ZAP_CHUNK_FREE;
201789Sahrens 	zlf->lf_next = l->l_phys->l_hdr.lh_freelist;
202789Sahrens 	bzero(zlf->lf_pad, sizeof (zlf->lf_pad)); /* help it to compress */
203789Sahrens 	l->l_phys->l_hdr.lh_freelist = chunk;
204789Sahrens 
205*1578Sahrens 	l->l_phys->l_hdr.lh_nfree++;
206789Sahrens }
207789Sahrens 
208789Sahrens /*
209789Sahrens  * Routines which manipulate leaf arrays (zap_leaf_array type chunks).
210789Sahrens  */
211789Sahrens 
212789Sahrens static uint16_t
213*1578Sahrens zap_leaf_array_create(zap_leaf_t *l, const char *buf,
214789Sahrens 	int integer_size, int num_integers)
215789Sahrens {
216789Sahrens 	uint16_t chunk_head;
217789Sahrens 	uint16_t *chunkp = &chunk_head;
218789Sahrens 	int byten = 0;
219789Sahrens 	uint64_t value;
220789Sahrens 	int shift = (integer_size-1)*8;
221789Sahrens 	int len = num_integers;
222789Sahrens 
223789Sahrens 	ASSERT3U(num_integers * integer_size, <, MAX_ARRAY_BYTES);
224789Sahrens 
225789Sahrens 	while (len > 0) {
226789Sahrens 		uint16_t chunk = zap_leaf_chunk_alloc(l);
2271491Sahrens 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
228789Sahrens 		int i;
229789Sahrens 
2301491Sahrens 		la->la_type = ZAP_CHUNK_ARRAY;
231789Sahrens 		for (i = 0; i < ZAP_LEAF_ARRAY_BYTES; i++) {
232789Sahrens 			if (byten == 0)
233789Sahrens 				value = ldv(integer_size, buf);
234789Sahrens 			la->la_array[i] = (value & (0xff << shift)) >> shift;
235789Sahrens 			value <<= 8;
236789Sahrens 			if (++byten == integer_size) {
237789Sahrens 				byten = 0;
238789Sahrens 				buf += integer_size;
239789Sahrens 				if (--len == 0)
240789Sahrens 					break;
241789Sahrens 			}
242789Sahrens 		}
243789Sahrens 
244789Sahrens 		*chunkp = chunk;
245789Sahrens 		chunkp = &la->la_next;
246789Sahrens 	}
247789Sahrens 	*chunkp = CHAIN_END;
248789Sahrens 
249789Sahrens 	return (chunk_head);
250789Sahrens }
251789Sahrens 
252789Sahrens static void
253*1578Sahrens zap_leaf_array_free(zap_leaf_t *l, uint16_t *chunkp)
254789Sahrens {
255789Sahrens 	uint16_t chunk = *chunkp;
256789Sahrens 
257789Sahrens 	*chunkp = CHAIN_END;
258789Sahrens 
259789Sahrens 	while (chunk != CHAIN_END) {
2601491Sahrens 		int nextchunk = ZAP_LEAF_CHUNK(l, chunk).l_array.la_next;
2611491Sahrens 		ASSERT3U(ZAP_LEAF_CHUNK(l, chunk).l_array.la_type, ==,
2621491Sahrens 		    ZAP_CHUNK_ARRAY);
263789Sahrens 		zap_leaf_chunk_free(l, chunk);
264789Sahrens 		chunk = nextchunk;
265789Sahrens 	}
266789Sahrens }
267789Sahrens 
268789Sahrens /* array_len and buf_len are in integers, not bytes */
269789Sahrens static void
270*1578Sahrens zap_leaf_array_read(zap_leaf_t *l, uint16_t chunk,
271789Sahrens     int array_int_len, int array_len, int buf_int_len, uint64_t buf_len,
272789Sahrens     char *buf)
273789Sahrens {
274789Sahrens 	int len = MIN(array_len, buf_len);
275789Sahrens 	int byten = 0;
276789Sahrens 	uint64_t value = 0;
277789Sahrens 
278789Sahrens 	ASSERT3U(array_int_len, <=, buf_int_len);
279789Sahrens 
280885Sahrens 	/* Fast path for one 8-byte integer */
281885Sahrens 	if (array_int_len == 8 && buf_int_len == 8 && len == 1) {
2821491Sahrens 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
283899Sbonwick 		uint8_t *ip = la->la_array;
284885Sahrens 		uint64_t *buf64 = (uint64_t *)buf;
285899Sbonwick 
286899Sbonwick 		*buf64 = (uint64_t)ip[0] << 56 | (uint64_t)ip[1] << 48 |
287899Sbonwick 		    (uint64_t)ip[2] << 40 | (uint64_t)ip[3] << 32 |
288899Sbonwick 		    (uint64_t)ip[4] << 24 | (uint64_t)ip[5] << 16 |
289899Sbonwick 		    (uint64_t)ip[6] << 8 | (uint64_t)ip[7];
290885Sahrens 		return;
291885Sahrens 	}
292885Sahrens 
293885Sahrens 	/* Fast path for an array of 1-byte integers (eg. the entry name) */
294885Sahrens 	if (array_int_len == 1 && buf_int_len == 1 &&
295885Sahrens 	    buf_len > array_len + ZAP_LEAF_ARRAY_BYTES) {
296885Sahrens 		while (chunk != CHAIN_END) {
297885Sahrens 			struct zap_leaf_array *la =
2981491Sahrens 			    &ZAP_LEAF_CHUNK(l, chunk).l_array;
299885Sahrens 			bcopy(la->la_array, buf, ZAP_LEAF_ARRAY_BYTES);
300885Sahrens 			buf += ZAP_LEAF_ARRAY_BYTES;
301885Sahrens 			chunk = la->la_next;
302885Sahrens 		}
303885Sahrens 		return;
304885Sahrens 	}
305885Sahrens 
306789Sahrens 	while (len > 0) {
3071491Sahrens 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
308789Sahrens 		int i;
309789Sahrens 
3101491Sahrens 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
311789Sahrens 		for (i = 0; i < ZAP_LEAF_ARRAY_BYTES && len > 0; i++) {
312789Sahrens 			value = (value << 8) | la->la_array[i];
313789Sahrens 			byten++;
314789Sahrens 			if (byten == array_int_len) {
315789Sahrens 				stv(buf_int_len, buf, value);
316789Sahrens 				byten = 0;
317789Sahrens 				len--;
318789Sahrens 				if (len == 0)
319789Sahrens 					return;
320789Sahrens 				buf += buf_int_len;
321789Sahrens 			}
322789Sahrens 		}
323789Sahrens 		chunk = la->la_next;
324789Sahrens 	}
325789Sahrens }
326789Sahrens 
327789Sahrens /*
328789Sahrens  * Only to be used on 8-bit arrays.
329789Sahrens  * array_len is actual len in bytes (not encoded le_value_length).
330789Sahrens  * buf is null-terminated.
331789Sahrens  */
332789Sahrens static int
333*1578Sahrens zap_leaf_array_equal(zap_leaf_t *l, int chunk,
334789Sahrens     int array_len, const char *buf)
335789Sahrens {
336789Sahrens 	int bseen = 0;
337789Sahrens 
338789Sahrens 	while (bseen < array_len) {
3391491Sahrens 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
340789Sahrens 		int toread = MIN(array_len - bseen, ZAP_LEAF_ARRAY_BYTES);
3411491Sahrens 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
342789Sahrens 		if (bcmp(la->la_array, buf + bseen, toread))
343789Sahrens 			break;
344789Sahrens 		chunk = la->la_next;
345789Sahrens 		bseen += toread;
346789Sahrens 	}
347789Sahrens 	return (bseen == array_len);
348789Sahrens }
349789Sahrens 
350789Sahrens /*
351789Sahrens  * Routines which manipulate leaf entries.
352789Sahrens  */
353789Sahrens 
354789Sahrens int
355789Sahrens zap_leaf_lookup(zap_leaf_t *l,
356789Sahrens     const char *name, uint64_t h, zap_entry_handle_t *zeh)
357789Sahrens {
358789Sahrens 	uint16_t *chunkp;
359789Sahrens 	struct zap_leaf_entry *le;
360789Sahrens 
361*1578Sahrens 	ASSERT3U(l->l_phys->l_hdr.lh_magic, ==, ZAP_LEAF_MAGIC);
362789Sahrens 
363789Sahrens 	for (chunkp = LEAF_HASH_ENTPTR(l, h);
364789Sahrens 	    *chunkp != CHAIN_END; chunkp = &le->le_next) {
365789Sahrens 		uint16_t chunk = *chunkp;
3661491Sahrens 		le = ZAP_LEAF_ENTRY(l, chunk);
367789Sahrens 
3681491Sahrens 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
3691491Sahrens 		ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
370789Sahrens 
371789Sahrens 		if (le->le_hash != h)
372789Sahrens 			continue;
373789Sahrens 
374*1578Sahrens 		if (zap_leaf_array_equal(l, le->le_name_chunk,
375789Sahrens 		    le->le_name_length, name)) {
376789Sahrens 			zeh->zeh_num_integers = le->le_value_length;
377789Sahrens 			zeh->zeh_integer_size = le->le_int_size;
378789Sahrens 			zeh->zeh_cd = le->le_cd;
379789Sahrens 			zeh->zeh_hash = le->le_hash;
380789Sahrens 			zeh->zeh_chunkp = chunkp;
381*1578Sahrens 			zeh->zeh_leaf = l;
382789Sahrens 			return (0);
383789Sahrens 		}
384789Sahrens 	}
385789Sahrens 
386789Sahrens 	return (ENOENT);
387789Sahrens }
388789Sahrens 
389789Sahrens /* Return (h1,cd1 >= h2,cd2) */
390885Sahrens #define	HCD_GTEQ(h1, cd1, h2, cd2) \
391885Sahrens 	((h1 > h2) ? TRUE : ((h1 == h2 && cd1 >= cd2) ? TRUE : FALSE))
392789Sahrens 
393789Sahrens int
394789Sahrens zap_leaf_lookup_closest(zap_leaf_t *l,
395789Sahrens     uint64_t h, uint32_t cd, zap_entry_handle_t *zeh)
396789Sahrens {
397789Sahrens 	uint16_t chunk;
398789Sahrens 	uint64_t besth = -1ULL;
399789Sahrens 	uint32_t bestcd = ZAP_MAXCD;
4001491Sahrens 	uint16_t bestlh = ZAP_LEAF_HASH_NUMENTRIES(l)-1;
401789Sahrens 	uint16_t lh;
402789Sahrens 	struct zap_leaf_entry *le;
403789Sahrens 
404*1578Sahrens 	ASSERT3U(l->l_phys->l_hdr.lh_magic, ==, ZAP_LEAF_MAGIC);
405789Sahrens 
406789Sahrens 	for (lh = LEAF_HASH(l, h); lh <= bestlh; lh++) {
407789Sahrens 		for (chunk = l->l_phys->l_hash[lh];
408789Sahrens 		    chunk != CHAIN_END; chunk = le->le_next) {
4091491Sahrens 			le = ZAP_LEAF_ENTRY(l, chunk);
410789Sahrens 
4111491Sahrens 			ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
4121491Sahrens 			ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
413789Sahrens 
414885Sahrens 			if (HCD_GTEQ(le->le_hash, le->le_cd, h, cd) &&
415885Sahrens 			    HCD_GTEQ(besth, bestcd, le->le_hash, le->le_cd)) {
416789Sahrens 				ASSERT3U(bestlh, >=, lh);
417789Sahrens 				bestlh = lh;
418789Sahrens 				besth = le->le_hash;
419789Sahrens 				bestcd = le->le_cd;
420789Sahrens 
421789Sahrens 				zeh->zeh_num_integers = le->le_value_length;
422789Sahrens 				zeh->zeh_integer_size = le->le_int_size;
423789Sahrens 				zeh->zeh_cd = le->le_cd;
424789Sahrens 				zeh->zeh_hash = le->le_hash;
425789Sahrens 				zeh->zeh_fakechunk = chunk;
426789Sahrens 				zeh->zeh_chunkp = &zeh->zeh_fakechunk;
427*1578Sahrens 				zeh->zeh_leaf = l;
428789Sahrens 			}
429789Sahrens 		}
430789Sahrens 	}
431789Sahrens 
432789Sahrens 	return (bestcd == ZAP_MAXCD ? ENOENT : 0);
433789Sahrens }
434789Sahrens 
435789Sahrens int
436789Sahrens zap_entry_read(const zap_entry_handle_t *zeh,
437789Sahrens     uint8_t integer_size, uint64_t num_integers, void *buf)
438789Sahrens {
4391491Sahrens 	struct zap_leaf_entry *le =
440*1578Sahrens 	    ZAP_LEAF_ENTRY(zeh->zeh_leaf, *zeh->zeh_chunkp);
4411491Sahrens 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
442789Sahrens 
443789Sahrens 	if (le->le_int_size > integer_size)
444789Sahrens 		return (EINVAL);
445789Sahrens 
446*1578Sahrens 	zap_leaf_array_read(zeh->zeh_leaf, le->le_value_chunk, le->le_int_size,
447789Sahrens 	    le->le_value_length, integer_size, num_integers, buf);
448789Sahrens 
449789Sahrens 	if (zeh->zeh_num_integers > num_integers)
450789Sahrens 		return (EOVERFLOW);
451789Sahrens 	return (0);
452789Sahrens 
453789Sahrens }
454789Sahrens 
455789Sahrens int
456789Sahrens zap_entry_read_name(const zap_entry_handle_t *zeh, uint16_t buflen, char *buf)
457789Sahrens {
4581491Sahrens 	struct zap_leaf_entry *le =
459*1578Sahrens 	    ZAP_LEAF_ENTRY(zeh->zeh_leaf, *zeh->zeh_chunkp);
4601491Sahrens 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
461789Sahrens 
462*1578Sahrens 	zap_leaf_array_read(zeh->zeh_leaf, le->le_name_chunk, 1,
463789Sahrens 	    le->le_name_length, 1, buflen, buf);
464789Sahrens 	if (le->le_name_length > buflen)
465789Sahrens 		return (EOVERFLOW);
466789Sahrens 	return (0);
467789Sahrens }
468789Sahrens 
469789Sahrens int
470789Sahrens zap_entry_update(zap_entry_handle_t *zeh,
471789Sahrens 	uint8_t integer_size, uint64_t num_integers, const void *buf)
472789Sahrens {
473789Sahrens 	int delta_chunks;
474*1578Sahrens 	zap_leaf_t *l = zeh->zeh_leaf;
475*1578Sahrens 	struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, *zeh->zeh_chunkp);
476789Sahrens 
477*1578Sahrens 	delta_chunks = ZAP_LEAF_ARRAY_NCHUNKS(num_integers * integer_size) -
478*1578Sahrens 	    ZAP_LEAF_ARRAY_NCHUNKS(le->le_value_length * le->le_int_size);
479789Sahrens 
480*1578Sahrens 	if ((int)l->l_phys->l_hdr.lh_nfree < delta_chunks)
481789Sahrens 		return (EAGAIN);
482789Sahrens 
483789Sahrens 	/*
484789Sahrens 	 * We should search other chained leaves (via
485789Sahrens 	 * zap_entry_remove,create?) otherwise returning EAGAIN will
486789Sahrens 	 * just send us into an infinite loop if we have to chain
487789Sahrens 	 * another leaf block, rather than being able to split this
488789Sahrens 	 * block.
489789Sahrens 	 */
490789Sahrens 
491*1578Sahrens 	zap_leaf_array_free(l, &le->le_value_chunk);
492789Sahrens 	le->le_value_chunk =
493*1578Sahrens 	    zap_leaf_array_create(l, buf, integer_size, num_integers);
4941491Sahrens 	le->le_value_length = num_integers;
495789Sahrens 	le->le_int_size = integer_size;
496789Sahrens 	return (0);
497789Sahrens }
498789Sahrens 
499789Sahrens void
500789Sahrens zap_entry_remove(zap_entry_handle_t *zeh)
501789Sahrens {
502789Sahrens 	uint16_t entry_chunk;
503789Sahrens 	struct zap_leaf_entry *le;
504*1578Sahrens 	zap_leaf_t *l = zeh->zeh_leaf;
505789Sahrens 
506789Sahrens 	ASSERT3P(zeh->zeh_chunkp, !=, &zeh->zeh_fakechunk);
507789Sahrens 
508789Sahrens 	entry_chunk = *zeh->zeh_chunkp;
5091491Sahrens 	le = ZAP_LEAF_ENTRY(l, entry_chunk);
5101491Sahrens 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
511789Sahrens 
512*1578Sahrens 	zap_leaf_array_free(l, &le->le_name_chunk);
513*1578Sahrens 	zap_leaf_array_free(l, &le->le_value_chunk);
514789Sahrens 
515789Sahrens 	*zeh->zeh_chunkp = le->le_next;
516789Sahrens 	zap_leaf_chunk_free(l, entry_chunk);
517789Sahrens 
518*1578Sahrens 	l->l_phys->l_hdr.lh_nentries--;
519789Sahrens }
520789Sahrens 
521789Sahrens int
522789Sahrens zap_entry_create(zap_leaf_t *l, const char *name, uint64_t h, uint32_t cd,
523789Sahrens     uint8_t integer_size, uint64_t num_integers, const void *buf,
524789Sahrens     zap_entry_handle_t *zeh)
525789Sahrens {
526789Sahrens 	uint16_t chunk;
527789Sahrens 	uint16_t *chunkp;
528789Sahrens 	struct zap_leaf_entry *le;
529789Sahrens 	uint64_t namelen, valuelen;
530789Sahrens 	int numchunks;
531789Sahrens 
532789Sahrens 	valuelen = integer_size * num_integers;
533789Sahrens 	namelen = strlen(name) + 1;
534789Sahrens 	ASSERT(namelen >= 2);
535789Sahrens 
536*1578Sahrens 	numchunks = 1 + ZAP_LEAF_ARRAY_NCHUNKS(namelen) +
537*1578Sahrens 	    ZAP_LEAF_ARRAY_NCHUNKS(valuelen);
5381491Sahrens 	if (numchunks > ZAP_LEAF_NUMCHUNKS(l))
539789Sahrens 		return (E2BIG);
540789Sahrens 
541789Sahrens 	if (cd == ZAP_MAXCD) {
542789Sahrens 		for (cd = 0; cd < ZAP_MAXCD; cd++) {
543*1578Sahrens 			for (chunk = *LEAF_HASH_ENTPTR(l, h);
544*1578Sahrens 			    chunk != CHAIN_END; chunk = le->le_next) {
545*1578Sahrens 				le = ZAP_LEAF_ENTRY(l, chunk);
546*1578Sahrens 				if (le->le_hash == h &&
547*1578Sahrens 				    le->le_cd == cd) {
548*1578Sahrens 					break;
549789Sahrens 				}
550789Sahrens 			}
551789Sahrens 			/* If this cd is not in use, we are good. */
552789Sahrens 			if (chunk == CHAIN_END)
553789Sahrens 				break;
554789Sahrens 		}
555789Sahrens 		/* If we tried all the cd's, we lose. */
556789Sahrens 		if (cd == ZAP_MAXCD)
557789Sahrens 			return (ENOSPC);
558789Sahrens 	}
559789Sahrens 
560*1578Sahrens 	if (l->l_phys->l_hdr.lh_nfree < numchunks)
561789Sahrens 		return (EAGAIN);
562789Sahrens 
563789Sahrens 	/* make the entry */
564789Sahrens 	chunk = zap_leaf_chunk_alloc(l);
5651491Sahrens 	le = ZAP_LEAF_ENTRY(l, chunk);
5661491Sahrens 	le->le_type = ZAP_CHUNK_ENTRY;
567*1578Sahrens 	le->le_name_chunk = zap_leaf_array_create(l, name, 1, namelen);
568789Sahrens 	le->le_name_length = namelen;
569789Sahrens 	le->le_value_chunk =
570*1578Sahrens 	    zap_leaf_array_create(l, buf, integer_size, num_integers);
5711491Sahrens 	le->le_value_length = num_integers;
572789Sahrens 	le->le_int_size = integer_size;
573789Sahrens 	le->le_hash = h;
574789Sahrens 	le->le_cd = cd;
575789Sahrens 
576789Sahrens 	/* link it into the hash chain */
577789Sahrens 	chunkp = LEAF_HASH_ENTPTR(l, h);
578789Sahrens 	le->le_next = *chunkp;
579789Sahrens 	*chunkp = chunk;
580789Sahrens 
581*1578Sahrens 	l->l_phys->l_hdr.lh_nentries++;
582789Sahrens 
583*1578Sahrens 	zeh->zeh_leaf = l;
584789Sahrens 	zeh->zeh_num_integers = num_integers;
585789Sahrens 	zeh->zeh_integer_size = le->le_int_size;
586789Sahrens 	zeh->zeh_cd = le->le_cd;
587789Sahrens 	zeh->zeh_hash = le->le_hash;
588789Sahrens 	zeh->zeh_chunkp = chunkp;
589789Sahrens 
590789Sahrens 	return (0);
591789Sahrens }
592789Sahrens 
593789Sahrens /*
594789Sahrens  * Routines for transferring entries between leafs.
595789Sahrens  */
596789Sahrens 
597789Sahrens static void
598789Sahrens zap_leaf_rehash_entry(zap_leaf_t *l, uint16_t entry)
599789Sahrens {
6001491Sahrens 	struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, entry);
601789Sahrens 	uint16_t *ptr = LEAF_HASH_ENTPTR(l, le->le_hash);
602789Sahrens 	le->le_next = *ptr;
603789Sahrens 	*ptr = entry;
604789Sahrens }
605789Sahrens 
606789Sahrens static uint16_t
607789Sahrens zap_leaf_transfer_array(zap_leaf_t *l, uint16_t chunk, zap_leaf_t *nl)
608789Sahrens {
609789Sahrens 	uint16_t new_chunk;
610789Sahrens 	uint16_t *nchunkp = &new_chunk;
611789Sahrens 
612789Sahrens 	while (chunk != CHAIN_END) {
613789Sahrens 		uint16_t nchunk = zap_leaf_chunk_alloc(nl);
614789Sahrens 		struct zap_leaf_array *nla =
6151491Sahrens 		    &ZAP_LEAF_CHUNK(nl, nchunk).l_array;
616789Sahrens 		struct zap_leaf_array *la =
6171491Sahrens 		    &ZAP_LEAF_CHUNK(l, chunk).l_array;
618789Sahrens 		int nextchunk = la->la_next;
619789Sahrens 
6201491Sahrens 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
6211491Sahrens 		ASSERT3U(nchunk, <, ZAP_LEAF_NUMCHUNKS(l));
622789Sahrens 
623*1578Sahrens 		*nla = *la; /* structure assignment */
624789Sahrens 
625789Sahrens 		zap_leaf_chunk_free(l, chunk);
626789Sahrens 		chunk = nextchunk;
627789Sahrens 		*nchunkp = nchunk;
628789Sahrens 		nchunkp = &nla->la_next;
629789Sahrens 	}
630789Sahrens 	*nchunkp = CHAIN_END;
631789Sahrens 	return (new_chunk);
632789Sahrens }
633789Sahrens 
634789Sahrens static void
635*1578Sahrens zap_leaf_transfer_entry(zap_leaf_t *l, int entry, zap_leaf_t *nl)
636789Sahrens {
637789Sahrens 	struct zap_leaf_entry *le, *nle;
638*1578Sahrens 	uint16_t chunk;
639789Sahrens 
6401491Sahrens 	le = ZAP_LEAF_ENTRY(l, entry);
6411491Sahrens 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
642789Sahrens 
643789Sahrens 	chunk = zap_leaf_chunk_alloc(nl);
6441491Sahrens 	nle = ZAP_LEAF_ENTRY(nl, chunk);
645*1578Sahrens 	*nle = *le; /* structure assignment */
646789Sahrens 
647789Sahrens 	zap_leaf_rehash_entry(nl, chunk);
648789Sahrens 
649789Sahrens 	nle->le_name_chunk = zap_leaf_transfer_array(l, le->le_name_chunk, nl);
650789Sahrens 	nle->le_value_chunk =
651789Sahrens 	    zap_leaf_transfer_array(l, le->le_value_chunk, nl);
652789Sahrens 
653789Sahrens 	zap_leaf_chunk_free(l, entry);
654789Sahrens 
655*1578Sahrens 	l->l_phys->l_hdr.lh_nentries--;
656*1578Sahrens 	nl->l_phys->l_hdr.lh_nentries++;
657789Sahrens }
658789Sahrens 
659789Sahrens /*
660*1578Sahrens  * Transfer the entries whose hash prefix ends in 1 to the new leaf.
661789Sahrens  */
662*1578Sahrens void
663*1578Sahrens zap_leaf_split(zap_leaf_t *l, zap_leaf_t *nl)
664789Sahrens {
665789Sahrens 	int i;
666*1578Sahrens 	int bit = 64 - 1 - l->l_phys->l_hdr.lh_prefix_len;
667789Sahrens 
668*1578Sahrens 	/* set new prefix and prefix_len */
669*1578Sahrens 	l->l_phys->l_hdr.lh_prefix <<= 1;
670*1578Sahrens 	l->l_phys->l_hdr.lh_prefix_len++;
671*1578Sahrens 	nl->l_phys->l_hdr.lh_prefix = l->l_phys->l_hdr.lh_prefix | 1;
672*1578Sahrens 	nl->l_phys->l_hdr.lh_prefix_len = l->l_phys->l_hdr.lh_prefix_len;
673*1578Sahrens 
674789Sahrens 	/* break existing hash chains */
6751491Sahrens 	zap_memset(l->l_phys->l_hash, CHAIN_END, 2*ZAP_LEAF_HASH_NUMENTRIES(l));
676789Sahrens 
677*1578Sahrens 	/*
678*1578Sahrens 	 * Transfer entries whose hash bit 'bit' is set to nl; rehash
679*1578Sahrens 	 * the remaining entries
680*1578Sahrens 	 *
681*1578Sahrens 	 * NB: We could find entries via the hashtable instead. That
682*1578Sahrens 	 * would be O(hashents+numents) rather than O(numblks+numents),
683*1578Sahrens 	 * but this accesses memory more sequentially, and when we're
684*1578Sahrens 	 * called, the block is usually pretty full.
685*1578Sahrens 	 */
6861491Sahrens 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(l); i++) {
6871491Sahrens 		struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, i);
6881491Sahrens 		if (le->le_type != ZAP_CHUNK_ENTRY)
689789Sahrens 			continue;
690789Sahrens 
691*1578Sahrens 		if (le->le_hash & (1ULL << bit))
692*1578Sahrens 			zap_leaf_transfer_entry(l, i, nl);
693*1578Sahrens 		else
694*1578Sahrens 			zap_leaf_rehash_entry(l, i);
695789Sahrens 	}
696789Sahrens }
697789Sahrens 
698789Sahrens void
699*1578Sahrens zap_leaf_stats(zap_t *zap, zap_leaf_t *l, zap_stats_t *zs)
700789Sahrens {
701*1578Sahrens 	int i, n;
702789Sahrens 
703*1578Sahrens 	n = zap->zap_f.zap_phys->zap_ptrtbl.zt_shift -
704*1578Sahrens 	    l->l_phys->l_hdr.lh_prefix_len;
705789Sahrens 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
706789Sahrens 	zs->zs_leafs_with_2n_pointers[n]++;
707789Sahrens 
708789Sahrens 
709*1578Sahrens 	n = l->l_phys->l_hdr.lh_nentries/5;
710*1578Sahrens 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
711*1578Sahrens 	zs->zs_blocks_with_n5_entries[n]++;
712789Sahrens 
713*1578Sahrens 	n = ((1<<FZAP_BLOCK_SHIFT(zap)) -
714*1578Sahrens 	    l->l_phys->l_hdr.lh_nfree * (ZAP_LEAF_ARRAY_BYTES+1))*10 /
715*1578Sahrens 	    (1<<FZAP_BLOCK_SHIFT(zap));
716*1578Sahrens 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
717*1578Sahrens 	zs->zs_blocks_n_tenths_full[n]++;
718789Sahrens 
719*1578Sahrens 	for (i = 0; i < ZAP_LEAF_HASH_NUMENTRIES(l); i++) {
720*1578Sahrens 		int nentries = 0;
721*1578Sahrens 		int chunk = l->l_phys->l_hash[i];
722*1578Sahrens 
723*1578Sahrens 		while (chunk != CHAIN_END) {
724*1578Sahrens 			struct zap_leaf_entry *le =
725*1578Sahrens 			    ZAP_LEAF_ENTRY(l, chunk);
726789Sahrens 
727*1578Sahrens 			n = 1 + ZAP_LEAF_ARRAY_NCHUNKS(le->le_name_length) +
728*1578Sahrens 			    ZAP_LEAF_ARRAY_NCHUNKS(le->le_value_length *
729*1578Sahrens 				le->le_int_size);
730*1578Sahrens 			n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
731*1578Sahrens 			zs->zs_entries_using_n_chunks[n]++;
732789Sahrens 
733*1578Sahrens 			chunk = le->le_next;
734*1578Sahrens 			nentries++;
735789Sahrens 		}
736789Sahrens 
737*1578Sahrens 		n = nentries;
738*1578Sahrens 		n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
739*1578Sahrens 		zs->zs_buckets_with_n_entries[n]++;
740*1578Sahrens 	}
741789Sahrens }
742