xref: /titanic_51/usr/src/uts/common/fs/zfs/zap_leaf.c (revision a31148363f598def767ac48c5d82e1572e44b935)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*
22  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 /*
27  * The 512-byte leaf is broken into 32 16-byte chunks.
28  * chunk number n means l_chunk[n], even though the header precedes it.
29  * the names are stored null-terminated.
30  */
31 
32 #include <sys/zio.h>
33 #include <sys/spa.h>
34 #include <sys/dmu.h>
35 #include <sys/zfs_context.h>
36 #include <sys/fs/zfs.h>
37 #include <sys/zap.h>
38 #include <sys/zap_impl.h>
39 #include <sys/zap_leaf.h>
40 
41 static uint16_t *zap_leaf_rehash_entry(zap_leaf_t *l, uint16_t entry);
42 
43 #define	CHAIN_END 0xffff /* end of the chunk chain */
44 
45 /* half the (current) minimum block size */
46 #define	MAX_ARRAY_BYTES (8<<10)
47 
48 #define	LEAF_HASH(l, h) \
49 	((ZAP_LEAF_HASH_NUMENTRIES(l)-1) & \
50 	((h) >> (64 - ZAP_LEAF_HASH_SHIFT(l)-(l)->l_phys->l_hdr.lh_prefix_len)))
51 
52 #define	LEAF_HASH_ENTPTR(l, h) (&(l)->l_phys->l_hash[LEAF_HASH(l, h)])
53 
54 
55 static void
56 zap_memset(void *a, int c, size_t n)
57 {
58 	char *cp = a;
59 	char *cpend = cp + n;
60 
61 	while (cp < cpend)
62 		*cp++ = c;
63 }
64 
65 static void
66 stv(int len, void *addr, uint64_t value)
67 {
68 	switch (len) {
69 	case 1:
70 		*(uint8_t *)addr = value;
71 		return;
72 	case 2:
73 		*(uint16_t *)addr = value;
74 		return;
75 	case 4:
76 		*(uint32_t *)addr = value;
77 		return;
78 	case 8:
79 		*(uint64_t *)addr = value;
80 		return;
81 	}
82 	ASSERT(!"bad int len");
83 }
84 
85 static uint64_t
86 ldv(int len, const void *addr)
87 {
88 	switch (len) {
89 	case 1:
90 		return (*(uint8_t *)addr);
91 	case 2:
92 		return (*(uint16_t *)addr);
93 	case 4:
94 		return (*(uint32_t *)addr);
95 	case 8:
96 		return (*(uint64_t *)addr);
97 	}
98 	ASSERT(!"bad int len");
99 	return (0xFEEDFACEDEADBEEFULL);
100 }
101 
102 void
103 zap_leaf_byteswap(zap_leaf_phys_t *buf, int size)
104 {
105 	int i;
106 	zap_leaf_t l;
107 	l.l_bs = highbit(size)-1;
108 	l.l_phys = buf;
109 
110 	buf->l_hdr.lh_block_type = 	BSWAP_64(buf->l_hdr.lh_block_type);
111 	buf->l_hdr.lh_prefix = 		BSWAP_64(buf->l_hdr.lh_prefix);
112 	buf->l_hdr.lh_magic = 		BSWAP_32(buf->l_hdr.lh_magic);
113 	buf->l_hdr.lh_nfree = 		BSWAP_16(buf->l_hdr.lh_nfree);
114 	buf->l_hdr.lh_nentries = 	BSWAP_16(buf->l_hdr.lh_nentries);
115 	buf->l_hdr.lh_prefix_len = 	BSWAP_16(buf->l_hdr.lh_prefix_len);
116 	buf->l_hdr.lh_freelist = 	BSWAP_16(buf->l_hdr.lh_freelist);
117 
118 	for (i = 0; i < ZAP_LEAF_HASH_NUMENTRIES(&l); i++)
119 		buf->l_hash[i] = BSWAP_16(buf->l_hash[i]);
120 
121 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(&l); i++) {
122 		zap_leaf_chunk_t *lc = &ZAP_LEAF_CHUNK(&l, i);
123 		struct zap_leaf_entry *le;
124 
125 		switch (lc->l_free.lf_type) {
126 		case ZAP_CHUNK_ENTRY:
127 			le = &lc->l_entry;
128 
129 			le->le_type =		BSWAP_8(le->le_type);
130 			le->le_value_intlen =	BSWAP_8(le->le_value_intlen);
131 			le->le_next =		BSWAP_16(le->le_next);
132 			le->le_name_chunk =	BSWAP_16(le->le_name_chunk);
133 			le->le_name_numints =	BSWAP_16(le->le_name_numints);
134 			le->le_value_chunk =	BSWAP_16(le->le_value_chunk);
135 			le->le_value_numints =	BSWAP_16(le->le_value_numints);
136 			le->le_cd =		BSWAP_32(le->le_cd);
137 			le->le_hash =		BSWAP_64(le->le_hash);
138 			break;
139 		case ZAP_CHUNK_FREE:
140 			lc->l_free.lf_type =	BSWAP_8(lc->l_free.lf_type);
141 			lc->l_free.lf_next =	BSWAP_16(lc->l_free.lf_next);
142 			break;
143 		case ZAP_CHUNK_ARRAY:
144 			lc->l_array.la_type =	BSWAP_8(lc->l_array.la_type);
145 			lc->l_array.la_next =	BSWAP_16(lc->l_array.la_next);
146 			/* la_array doesn't need swapping */
147 			break;
148 		default:
149 			ASSERT(!"bad leaf type");
150 		}
151 	}
152 }
153 
154 void
155 zap_leaf_init(zap_leaf_t *l, boolean_t sort)
156 {
157 	int i;
158 
159 	l->l_bs = highbit(l->l_dbuf->db_size)-1;
160 	zap_memset(&l->l_phys->l_hdr, 0, sizeof (struct zap_leaf_header));
161 	zap_memset(l->l_phys->l_hash, CHAIN_END, 2*ZAP_LEAF_HASH_NUMENTRIES(l));
162 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(l); i++) {
163 		ZAP_LEAF_CHUNK(l, i).l_free.lf_type = ZAP_CHUNK_FREE;
164 		ZAP_LEAF_CHUNK(l, i).l_free.lf_next = i+1;
165 	}
166 	ZAP_LEAF_CHUNK(l, ZAP_LEAF_NUMCHUNKS(l)-1).l_free.lf_next = CHAIN_END;
167 	l->l_phys->l_hdr.lh_block_type = ZBT_LEAF;
168 	l->l_phys->l_hdr.lh_magic = ZAP_LEAF_MAGIC;
169 	l->l_phys->l_hdr.lh_nfree = ZAP_LEAF_NUMCHUNKS(l);
170 	if (sort)
171 		l->l_phys->l_hdr.lh_flags |= ZLF_ENTRIES_CDSORTED;
172 }
173 
174 /*
175  * Routines which manipulate leaf chunks (l_chunk[]).
176  */
177 
178 static uint16_t
179 zap_leaf_chunk_alloc(zap_leaf_t *l)
180 {
181 	int chunk;
182 
183 	ASSERT(l->l_phys->l_hdr.lh_nfree > 0);
184 
185 	chunk = l->l_phys->l_hdr.lh_freelist;
186 	ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
187 	ASSERT3U(ZAP_LEAF_CHUNK(l, chunk).l_free.lf_type, ==, ZAP_CHUNK_FREE);
188 
189 	l->l_phys->l_hdr.lh_freelist = ZAP_LEAF_CHUNK(l, chunk).l_free.lf_next;
190 
191 	l->l_phys->l_hdr.lh_nfree--;
192 
193 	return (chunk);
194 }
195 
196 static void
197 zap_leaf_chunk_free(zap_leaf_t *l, uint16_t chunk)
198 {
199 	struct zap_leaf_free *zlf = &ZAP_LEAF_CHUNK(l, chunk).l_free;
200 	ASSERT3U(l->l_phys->l_hdr.lh_nfree, <, ZAP_LEAF_NUMCHUNKS(l));
201 	ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
202 	ASSERT(zlf->lf_type != ZAP_CHUNK_FREE);
203 
204 	zlf->lf_type = ZAP_CHUNK_FREE;
205 	zlf->lf_next = l->l_phys->l_hdr.lh_freelist;
206 	bzero(zlf->lf_pad, sizeof (zlf->lf_pad)); /* help it to compress */
207 	l->l_phys->l_hdr.lh_freelist = chunk;
208 
209 	l->l_phys->l_hdr.lh_nfree++;
210 }
211 
212 /*
213  * Routines which manipulate leaf arrays (zap_leaf_array type chunks).
214  */
215 
216 static uint16_t
217 zap_leaf_array_create(zap_leaf_t *l, const char *buf,
218     int integer_size, int num_integers)
219 {
220 	uint16_t chunk_head;
221 	uint16_t *chunkp = &chunk_head;
222 	int byten = 0;
223 	uint64_t value;
224 	int shift = (integer_size-1)*8;
225 	int len = num_integers;
226 
227 	ASSERT3U(num_integers * integer_size, <, MAX_ARRAY_BYTES);
228 
229 	while (len > 0) {
230 		uint16_t chunk = zap_leaf_chunk_alloc(l);
231 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
232 		int i;
233 
234 		la->la_type = ZAP_CHUNK_ARRAY;
235 		for (i = 0; i < ZAP_LEAF_ARRAY_BYTES; i++) {
236 			if (byten == 0)
237 				value = ldv(integer_size, buf);
238 			la->la_array[i] = value >> shift;
239 			value <<= 8;
240 			if (++byten == integer_size) {
241 				byten = 0;
242 				buf += integer_size;
243 				if (--len == 0)
244 					break;
245 			}
246 		}
247 
248 		*chunkp = chunk;
249 		chunkp = &la->la_next;
250 	}
251 	*chunkp = CHAIN_END;
252 
253 	return (chunk_head);
254 }
255 
256 static void
257 zap_leaf_array_free(zap_leaf_t *l, uint16_t *chunkp)
258 {
259 	uint16_t chunk = *chunkp;
260 
261 	*chunkp = CHAIN_END;
262 
263 	while (chunk != CHAIN_END) {
264 		int nextchunk = ZAP_LEAF_CHUNK(l, chunk).l_array.la_next;
265 		ASSERT3U(ZAP_LEAF_CHUNK(l, chunk).l_array.la_type, ==,
266 		    ZAP_CHUNK_ARRAY);
267 		zap_leaf_chunk_free(l, chunk);
268 		chunk = nextchunk;
269 	}
270 }
271 
272 /* array_len and buf_len are in integers, not bytes */
273 static void
274 zap_leaf_array_read(zap_leaf_t *l, uint16_t chunk,
275     int array_int_len, int array_len, int buf_int_len, uint64_t buf_len,
276     void *buf)
277 {
278 	int len = MIN(array_len, buf_len);
279 	int byten = 0;
280 	uint64_t value = 0;
281 	char *p = buf;
282 
283 	ASSERT3U(array_int_len, <=, buf_int_len);
284 
285 	/* Fast path for one 8-byte integer */
286 	if (array_int_len == 8 && buf_int_len == 8 && len == 1) {
287 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
288 		uint8_t *ip = la->la_array;
289 		uint64_t *buf64 = buf;
290 
291 		*buf64 = (uint64_t)ip[0] << 56 | (uint64_t)ip[1] << 48 |
292 		    (uint64_t)ip[2] << 40 | (uint64_t)ip[3] << 32 |
293 		    (uint64_t)ip[4] << 24 | (uint64_t)ip[5] << 16 |
294 		    (uint64_t)ip[6] << 8 | (uint64_t)ip[7];
295 		return;
296 	}
297 
298 	/* Fast path for an array of 1-byte integers (eg. the entry name) */
299 	if (array_int_len == 1 && buf_int_len == 1 &&
300 	    buf_len > array_len + ZAP_LEAF_ARRAY_BYTES) {
301 		while (chunk != CHAIN_END) {
302 			struct zap_leaf_array *la =
303 			    &ZAP_LEAF_CHUNK(l, chunk).l_array;
304 			bcopy(la->la_array, p, ZAP_LEAF_ARRAY_BYTES);
305 			p += ZAP_LEAF_ARRAY_BYTES;
306 			chunk = la->la_next;
307 		}
308 		return;
309 	}
310 
311 	while (len > 0) {
312 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
313 		int i;
314 
315 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
316 		for (i = 0; i < ZAP_LEAF_ARRAY_BYTES && len > 0; i++) {
317 			value = (value << 8) | la->la_array[i];
318 			byten++;
319 			if (byten == array_int_len) {
320 				stv(buf_int_len, p, value);
321 				byten = 0;
322 				len--;
323 				if (len == 0)
324 					return;
325 				p += buf_int_len;
326 			}
327 		}
328 		chunk = la->la_next;
329 	}
330 }
331 
332 static boolean_t
333 zap_leaf_array_match(zap_leaf_t *l, zap_name_t *zn,
334     int chunk, int array_numints)
335 {
336 	int bseen = 0;
337 
338 	if (zap_getflags(zn->zn_zap) & ZAP_FLAG_UINT64_KEY) {
339 		uint64_t *thiskey;
340 		boolean_t match;
341 
342 		ASSERT(zn->zn_key_intlen == sizeof (*thiskey));
343 		thiskey = kmem_alloc(array_numints * sizeof (*thiskey),
344 		    KM_SLEEP);
345 
346 		zap_leaf_array_read(l, chunk, sizeof (*thiskey), array_numints,
347 		    sizeof (*thiskey), array_numints, thiskey);
348 		match = bcmp(thiskey, zn->zn_key_orig,
349 		    array_numints * sizeof (*thiskey)) == 0;
350 		kmem_free(thiskey, array_numints * sizeof (*thiskey));
351 		return (match);
352 	}
353 
354 	ASSERT(zn->zn_key_intlen == 1);
355 	if (zn->zn_matchtype == MT_FIRST) {
356 		char *thisname = kmem_alloc(array_numints, KM_SLEEP);
357 		boolean_t match;
358 
359 		zap_leaf_array_read(l, chunk, sizeof (char), array_numints,
360 		    sizeof (char), array_numints, thisname);
361 		match = zap_match(zn, thisname);
362 		kmem_free(thisname, array_numints);
363 		return (match);
364 	}
365 
366 	/*
367 	 * Fast path for exact matching.
368 	 * First check that the lengths match, so that we don't read
369 	 * past the end of the zn_key_orig array.
370 	 */
371 	if (array_numints != zn->zn_key_orig_numints)
372 		return (B_FALSE);
373 	while (bseen < array_numints) {
374 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(l, chunk).l_array;
375 		int toread = MIN(array_numints - bseen, ZAP_LEAF_ARRAY_BYTES);
376 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
377 		if (bcmp(la->la_array, (char *)zn->zn_key_orig + bseen, toread))
378 			break;
379 		chunk = la->la_next;
380 		bseen += toread;
381 	}
382 	return (bseen == array_numints);
383 }
384 
385 /*
386  * Routines which manipulate leaf entries.
387  */
388 
389 int
390 zap_leaf_lookup(zap_leaf_t *l, zap_name_t *zn, zap_entry_handle_t *zeh)
391 {
392 	uint16_t *chunkp;
393 	struct zap_leaf_entry *le;
394 
395 	ASSERT3U(l->l_phys->l_hdr.lh_magic, ==, ZAP_LEAF_MAGIC);
396 
397 again:
398 	for (chunkp = LEAF_HASH_ENTPTR(l, zn->zn_hash);
399 	    *chunkp != CHAIN_END; chunkp = &le->le_next) {
400 		uint16_t chunk = *chunkp;
401 		le = ZAP_LEAF_ENTRY(l, chunk);
402 
403 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
404 		ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
405 
406 		if (le->le_hash != zn->zn_hash)
407 			continue;
408 
409 		/*
410 		 * NB: the entry chain is always sorted by cd on
411 		 * normalized zap objects, so this will find the
412 		 * lowest-cd match for MT_FIRST.
413 		 */
414 		ASSERT(zn->zn_matchtype == MT_EXACT ||
415 		    (l->l_phys->l_hdr.lh_flags & ZLF_ENTRIES_CDSORTED));
416 		if (zap_leaf_array_match(l, zn, le->le_name_chunk,
417 		    le->le_name_numints)) {
418 			zeh->zeh_num_integers = le->le_value_numints;
419 			zeh->zeh_integer_size = le->le_value_intlen;
420 			zeh->zeh_cd = le->le_cd;
421 			zeh->zeh_hash = le->le_hash;
422 			zeh->zeh_chunkp = chunkp;
423 			zeh->zeh_leaf = l;
424 			return (0);
425 		}
426 	}
427 
428 	/*
429 	 * NB: we could of course do this in one pass, but that would be
430 	 * a pain.  We'll see if MT_BEST is even used much.
431 	 */
432 	if (zn->zn_matchtype == MT_BEST) {
433 		zn->zn_matchtype = MT_FIRST;
434 		goto again;
435 	}
436 
437 	return (ENOENT);
438 }
439 
440 /* Return (h1,cd1 >= h2,cd2) */
441 #define	HCD_GTEQ(h1, cd1, h2, cd2) \
442 	((h1 > h2) ? TRUE : ((h1 == h2 && cd1 >= cd2) ? TRUE : FALSE))
443 
444 int
445 zap_leaf_lookup_closest(zap_leaf_t *l,
446     uint64_t h, uint32_t cd, zap_entry_handle_t *zeh)
447 {
448 	uint16_t chunk;
449 	uint64_t besth = -1ULL;
450 	uint32_t bestcd = -1U;
451 	uint16_t bestlh = ZAP_LEAF_HASH_NUMENTRIES(l)-1;
452 	uint16_t lh;
453 	struct zap_leaf_entry *le;
454 
455 	ASSERT3U(l->l_phys->l_hdr.lh_magic, ==, ZAP_LEAF_MAGIC);
456 
457 	for (lh = LEAF_HASH(l, h); lh <= bestlh; lh++) {
458 		for (chunk = l->l_phys->l_hash[lh];
459 		    chunk != CHAIN_END; chunk = le->le_next) {
460 			le = ZAP_LEAF_ENTRY(l, chunk);
461 
462 			ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
463 			ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
464 
465 			if (HCD_GTEQ(le->le_hash, le->le_cd, h, cd) &&
466 			    HCD_GTEQ(besth, bestcd, le->le_hash, le->le_cd)) {
467 				ASSERT3U(bestlh, >=, lh);
468 				bestlh = lh;
469 				besth = le->le_hash;
470 				bestcd = le->le_cd;
471 
472 				zeh->zeh_num_integers = le->le_value_numints;
473 				zeh->zeh_integer_size = le->le_value_intlen;
474 				zeh->zeh_cd = le->le_cd;
475 				zeh->zeh_hash = le->le_hash;
476 				zeh->zeh_fakechunk = chunk;
477 				zeh->zeh_chunkp = &zeh->zeh_fakechunk;
478 				zeh->zeh_leaf = l;
479 			}
480 		}
481 	}
482 
483 	return (bestcd == -1U ? ENOENT : 0);
484 }
485 
486 int
487 zap_entry_read(const zap_entry_handle_t *zeh,
488     uint8_t integer_size, uint64_t num_integers, void *buf)
489 {
490 	struct zap_leaf_entry *le =
491 	    ZAP_LEAF_ENTRY(zeh->zeh_leaf, *zeh->zeh_chunkp);
492 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
493 
494 	if (le->le_value_intlen > integer_size)
495 		return (EINVAL);
496 
497 	zap_leaf_array_read(zeh->zeh_leaf, le->le_value_chunk,
498 	    le->le_value_intlen, le->le_value_numints,
499 	    integer_size, num_integers, buf);
500 
501 	if (zeh->zeh_num_integers > num_integers)
502 		return (EOVERFLOW);
503 	return (0);
504 
505 }
506 
507 int
508 zap_entry_read_name(zap_t *zap, const zap_entry_handle_t *zeh, uint16_t buflen,
509     char *buf)
510 {
511 	struct zap_leaf_entry *le =
512 	    ZAP_LEAF_ENTRY(zeh->zeh_leaf, *zeh->zeh_chunkp);
513 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
514 
515 	if (zap_getflags(zap) & ZAP_FLAG_UINT64_KEY) {
516 		zap_leaf_array_read(zeh->zeh_leaf, le->le_name_chunk, 8,
517 		    le->le_name_numints, 8, buflen / 8, buf);
518 	} else {
519 		zap_leaf_array_read(zeh->zeh_leaf, le->le_name_chunk, 1,
520 		    le->le_name_numints, 1, buflen, buf);
521 	}
522 	if (le->le_name_numints > buflen)
523 		return (EOVERFLOW);
524 	return (0);
525 }
526 
527 int
528 zap_entry_update(zap_entry_handle_t *zeh,
529 	uint8_t integer_size, uint64_t num_integers, const void *buf)
530 {
531 	int delta_chunks;
532 	zap_leaf_t *l = zeh->zeh_leaf;
533 	struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, *zeh->zeh_chunkp);
534 
535 	delta_chunks = ZAP_LEAF_ARRAY_NCHUNKS(num_integers * integer_size) -
536 	    ZAP_LEAF_ARRAY_NCHUNKS(le->le_value_numints * le->le_value_intlen);
537 
538 	if ((int)l->l_phys->l_hdr.lh_nfree < delta_chunks)
539 		return (EAGAIN);
540 
541 	/*
542 	 * We should search other chained leaves (via
543 	 * zap_entry_remove,create?) otherwise returning EAGAIN will
544 	 * just send us into an infinite loop if we have to chain
545 	 * another leaf block, rather than being able to split this
546 	 * block.
547 	 */
548 
549 	zap_leaf_array_free(l, &le->le_value_chunk);
550 	le->le_value_chunk =
551 	    zap_leaf_array_create(l, buf, integer_size, num_integers);
552 	le->le_value_numints = num_integers;
553 	le->le_value_intlen = integer_size;
554 	return (0);
555 }
556 
557 void
558 zap_entry_remove(zap_entry_handle_t *zeh)
559 {
560 	uint16_t entry_chunk;
561 	struct zap_leaf_entry *le;
562 	zap_leaf_t *l = zeh->zeh_leaf;
563 
564 	ASSERT3P(zeh->zeh_chunkp, !=, &zeh->zeh_fakechunk);
565 
566 	entry_chunk = *zeh->zeh_chunkp;
567 	le = ZAP_LEAF_ENTRY(l, entry_chunk);
568 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
569 
570 	zap_leaf_array_free(l, &le->le_name_chunk);
571 	zap_leaf_array_free(l, &le->le_value_chunk);
572 
573 	*zeh->zeh_chunkp = le->le_next;
574 	zap_leaf_chunk_free(l, entry_chunk);
575 
576 	l->l_phys->l_hdr.lh_nentries--;
577 }
578 
579 int
580 zap_entry_create(zap_leaf_t *l, zap_name_t *zn, uint32_t cd,
581     uint8_t integer_size, uint64_t num_integers, const void *buf,
582     zap_entry_handle_t *zeh)
583 {
584 	uint16_t chunk;
585 	uint16_t *chunkp;
586 	struct zap_leaf_entry *le;
587 	uint64_t valuelen;
588 	int numchunks;
589 	uint64_t h = zn->zn_hash;
590 
591 	valuelen = integer_size * num_integers;
592 
593 	numchunks = 1 + ZAP_LEAF_ARRAY_NCHUNKS(zn->zn_key_orig_numints *
594 	    zn->zn_key_intlen) + ZAP_LEAF_ARRAY_NCHUNKS(valuelen);
595 	if (numchunks > ZAP_LEAF_NUMCHUNKS(l))
596 		return (E2BIG);
597 
598 	if (cd == ZAP_NEED_CD) {
599 		/* find the lowest unused cd */
600 		if (l->l_phys->l_hdr.lh_flags & ZLF_ENTRIES_CDSORTED) {
601 			cd = 0;
602 
603 			for (chunk = *LEAF_HASH_ENTPTR(l, h);
604 			    chunk != CHAIN_END; chunk = le->le_next) {
605 				le = ZAP_LEAF_ENTRY(l, chunk);
606 				if (le->le_cd > cd)
607 					break;
608 				if (le->le_hash == h) {
609 					ASSERT3U(cd, ==, le->le_cd);
610 					cd++;
611 				}
612 			}
613 		} else {
614 			/* old unsorted format; do it the O(n^2) way */
615 			for (cd = 0; ; cd++) {
616 				for (chunk = *LEAF_HASH_ENTPTR(l, h);
617 				    chunk != CHAIN_END; chunk = le->le_next) {
618 					le = ZAP_LEAF_ENTRY(l, chunk);
619 					if (le->le_hash == h &&
620 					    le->le_cd == cd) {
621 						break;
622 					}
623 				}
624 				/* If this cd is not in use, we are good. */
625 				if (chunk == CHAIN_END)
626 					break;
627 			}
628 		}
629 		/*
630 		 * We would run out of space in a block before we could
631 		 * store enough entries to run out of CD values.
632 		 */
633 		ASSERT3U(cd, <, zap_maxcd(zn->zn_zap));
634 	}
635 
636 	if (l->l_phys->l_hdr.lh_nfree < numchunks)
637 		return (EAGAIN);
638 
639 	/* make the entry */
640 	chunk = zap_leaf_chunk_alloc(l);
641 	le = ZAP_LEAF_ENTRY(l, chunk);
642 	le->le_type = ZAP_CHUNK_ENTRY;
643 	le->le_name_chunk = zap_leaf_array_create(l, zn->zn_key_orig,
644 	    zn->zn_key_intlen, zn->zn_key_orig_numints);
645 	le->le_name_numints = zn->zn_key_orig_numints;
646 	le->le_value_chunk =
647 	    zap_leaf_array_create(l, buf, integer_size, num_integers);
648 	le->le_value_numints = num_integers;
649 	le->le_value_intlen = integer_size;
650 	le->le_hash = h;
651 	le->le_cd = cd;
652 
653 	/* link it into the hash chain */
654 	/* XXX if we did the search above, we could just use that */
655 	chunkp = zap_leaf_rehash_entry(l, chunk);
656 
657 	l->l_phys->l_hdr.lh_nentries++;
658 
659 	zeh->zeh_leaf = l;
660 	zeh->zeh_num_integers = num_integers;
661 	zeh->zeh_integer_size = le->le_value_intlen;
662 	zeh->zeh_cd = le->le_cd;
663 	zeh->zeh_hash = le->le_hash;
664 	zeh->zeh_chunkp = chunkp;
665 
666 	return (0);
667 }
668 
669 /*
670  * Determine if there is another entry with the same normalized form.
671  * For performance purposes, either zn or name must be provided (the
672  * other can be NULL).  Note, there usually won't be any hash
673  * conflicts, in which case we don't need the concatenated/normalized
674  * form of the name.  But all callers have one of these on hand anyway,
675  * so might as well take advantage.  A cleaner but slower interface
676  * would accept neither argument, and compute the normalized name as
677  * needed (using zap_name_alloc(zap_entry_read_name(zeh))).
678  */
679 boolean_t
680 zap_entry_normalization_conflict(zap_entry_handle_t *zeh, zap_name_t *zn,
681     const char *name, zap_t *zap)
682 {
683 	uint64_t chunk;
684 	struct zap_leaf_entry *le;
685 	boolean_t allocdzn = B_FALSE;
686 
687 	if (zap->zap_normflags == 0)
688 		return (B_FALSE);
689 
690 	for (chunk = *LEAF_HASH_ENTPTR(zeh->zeh_leaf, zeh->zeh_hash);
691 	    chunk != CHAIN_END; chunk = le->le_next) {
692 		le = ZAP_LEAF_ENTRY(zeh->zeh_leaf, chunk);
693 		if (le->le_hash != zeh->zeh_hash)
694 			continue;
695 		if (le->le_cd == zeh->zeh_cd)
696 			continue;
697 
698 		if (zn == NULL) {
699 			zn = zap_name_alloc(zap, name, MT_FIRST);
700 			allocdzn = B_TRUE;
701 		}
702 		if (zap_leaf_array_match(zeh->zeh_leaf, zn,
703 		    le->le_name_chunk, le->le_name_numints)) {
704 			if (allocdzn)
705 				zap_name_free(zn);
706 			return (B_TRUE);
707 		}
708 	}
709 	if (allocdzn)
710 		zap_name_free(zn);
711 	return (B_FALSE);
712 }
713 
714 /*
715  * Routines for transferring entries between leafs.
716  */
717 
718 static uint16_t *
719 zap_leaf_rehash_entry(zap_leaf_t *l, uint16_t entry)
720 {
721 	struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, entry);
722 	struct zap_leaf_entry *le2;
723 	uint16_t *chunkp;
724 
725 	/*
726 	 * keep the entry chain sorted by cd
727 	 * NB: this will not cause problems for unsorted leafs, though
728 	 * it is unnecessary there.
729 	 */
730 	for (chunkp = LEAF_HASH_ENTPTR(l, le->le_hash);
731 	    *chunkp != CHAIN_END; chunkp = &le2->le_next) {
732 		le2 = ZAP_LEAF_ENTRY(l, *chunkp);
733 		if (le2->le_cd > le->le_cd)
734 			break;
735 	}
736 
737 	le->le_next = *chunkp;
738 	*chunkp = entry;
739 	return (chunkp);
740 }
741 
742 static uint16_t
743 zap_leaf_transfer_array(zap_leaf_t *l, uint16_t chunk, zap_leaf_t *nl)
744 {
745 	uint16_t new_chunk;
746 	uint16_t *nchunkp = &new_chunk;
747 
748 	while (chunk != CHAIN_END) {
749 		uint16_t nchunk = zap_leaf_chunk_alloc(nl);
750 		struct zap_leaf_array *nla =
751 		    &ZAP_LEAF_CHUNK(nl, nchunk).l_array;
752 		struct zap_leaf_array *la =
753 		    &ZAP_LEAF_CHUNK(l, chunk).l_array;
754 		int nextchunk = la->la_next;
755 
756 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(l));
757 		ASSERT3U(nchunk, <, ZAP_LEAF_NUMCHUNKS(l));
758 
759 		*nla = *la; /* structure assignment */
760 
761 		zap_leaf_chunk_free(l, chunk);
762 		chunk = nextchunk;
763 		*nchunkp = nchunk;
764 		nchunkp = &nla->la_next;
765 	}
766 	*nchunkp = CHAIN_END;
767 	return (new_chunk);
768 }
769 
770 static void
771 zap_leaf_transfer_entry(zap_leaf_t *l, int entry, zap_leaf_t *nl)
772 {
773 	struct zap_leaf_entry *le, *nle;
774 	uint16_t chunk;
775 
776 	le = ZAP_LEAF_ENTRY(l, entry);
777 	ASSERT3U(le->le_type, ==, ZAP_CHUNK_ENTRY);
778 
779 	chunk = zap_leaf_chunk_alloc(nl);
780 	nle = ZAP_LEAF_ENTRY(nl, chunk);
781 	*nle = *le; /* structure assignment */
782 
783 	(void) zap_leaf_rehash_entry(nl, chunk);
784 
785 	nle->le_name_chunk = zap_leaf_transfer_array(l, le->le_name_chunk, nl);
786 	nle->le_value_chunk =
787 	    zap_leaf_transfer_array(l, le->le_value_chunk, nl);
788 
789 	zap_leaf_chunk_free(l, entry);
790 
791 	l->l_phys->l_hdr.lh_nentries--;
792 	nl->l_phys->l_hdr.lh_nentries++;
793 }
794 
795 /*
796  * Transfer the entries whose hash prefix ends in 1 to the new leaf.
797  */
798 void
799 zap_leaf_split(zap_leaf_t *l, zap_leaf_t *nl, boolean_t sort)
800 {
801 	int i;
802 	int bit = 64 - 1 - l->l_phys->l_hdr.lh_prefix_len;
803 
804 	/* set new prefix and prefix_len */
805 	l->l_phys->l_hdr.lh_prefix <<= 1;
806 	l->l_phys->l_hdr.lh_prefix_len++;
807 	nl->l_phys->l_hdr.lh_prefix = l->l_phys->l_hdr.lh_prefix | 1;
808 	nl->l_phys->l_hdr.lh_prefix_len = l->l_phys->l_hdr.lh_prefix_len;
809 
810 	/* break existing hash chains */
811 	zap_memset(l->l_phys->l_hash, CHAIN_END, 2*ZAP_LEAF_HASH_NUMENTRIES(l));
812 
813 	if (sort)
814 		l->l_phys->l_hdr.lh_flags |= ZLF_ENTRIES_CDSORTED;
815 
816 	/*
817 	 * Transfer entries whose hash bit 'bit' is set to nl; rehash
818 	 * the remaining entries
819 	 *
820 	 * NB: We could find entries via the hashtable instead. That
821 	 * would be O(hashents+numents) rather than O(numblks+numents),
822 	 * but this accesses memory more sequentially, and when we're
823 	 * called, the block is usually pretty full.
824 	 */
825 	for (i = 0; i < ZAP_LEAF_NUMCHUNKS(l); i++) {
826 		struct zap_leaf_entry *le = ZAP_LEAF_ENTRY(l, i);
827 		if (le->le_type != ZAP_CHUNK_ENTRY)
828 			continue;
829 
830 		if (le->le_hash & (1ULL << bit))
831 			zap_leaf_transfer_entry(l, i, nl);
832 		else
833 			(void) zap_leaf_rehash_entry(l, i);
834 	}
835 }
836 
837 void
838 zap_leaf_stats(zap_t *zap, zap_leaf_t *l, zap_stats_t *zs)
839 {
840 	int i, n;
841 
842 	n = zap->zap_f.zap_phys->zap_ptrtbl.zt_shift -
843 	    l->l_phys->l_hdr.lh_prefix_len;
844 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
845 	zs->zs_leafs_with_2n_pointers[n]++;
846 
847 
848 	n = l->l_phys->l_hdr.lh_nentries/5;
849 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
850 	zs->zs_blocks_with_n5_entries[n]++;
851 
852 	n = ((1<<FZAP_BLOCK_SHIFT(zap)) -
853 	    l->l_phys->l_hdr.lh_nfree * (ZAP_LEAF_ARRAY_BYTES+1))*10 /
854 	    (1<<FZAP_BLOCK_SHIFT(zap));
855 	n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
856 	zs->zs_blocks_n_tenths_full[n]++;
857 
858 	for (i = 0; i < ZAP_LEAF_HASH_NUMENTRIES(l); i++) {
859 		int nentries = 0;
860 		int chunk = l->l_phys->l_hash[i];
861 
862 		while (chunk != CHAIN_END) {
863 			struct zap_leaf_entry *le =
864 			    ZAP_LEAF_ENTRY(l, chunk);
865 
866 			n = 1 + ZAP_LEAF_ARRAY_NCHUNKS(le->le_name_numints) +
867 			    ZAP_LEAF_ARRAY_NCHUNKS(le->le_value_numints *
868 			    le->le_value_intlen);
869 			n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
870 			zs->zs_entries_using_n_chunks[n]++;
871 
872 			chunk = le->le_next;
873 			nentries++;
874 		}
875 
876 		n = nentries;
877 		n = MIN(n, ZAP_HISTOGRAM_SIZE-1);
878 		zs->zs_buckets_with_n_entries[n]++;
879 	}
880 }
881