xref: /freebsd/sys/contrib/openzfs/module/zfs/zap_micro.c (revision da5137abdf463bb5fee85061958a14dd12bc043e)
1eda14cbcSMatt Macy /*
2eda14cbcSMatt Macy  * CDDL HEADER START
3eda14cbcSMatt Macy  *
4eda14cbcSMatt Macy  * The contents of this file are subject to the terms of the
5eda14cbcSMatt Macy  * Common Development and Distribution License (the "License").
6eda14cbcSMatt Macy  * You may not use this file except in compliance with the License.
7eda14cbcSMatt Macy  *
8eda14cbcSMatt Macy  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9eda14cbcSMatt Macy  * or http://www.opensolaris.org/os/licensing.
10eda14cbcSMatt Macy  * See the License for the specific language governing permissions
11eda14cbcSMatt Macy  * and limitations under the License.
12eda14cbcSMatt Macy  *
13eda14cbcSMatt Macy  * When distributing Covered Code, include this CDDL HEADER in each
14eda14cbcSMatt Macy  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15eda14cbcSMatt Macy  * If applicable, add the following below this CDDL HEADER, with the
16eda14cbcSMatt Macy  * fields enclosed by brackets "[]" replaced with your own identifying
17eda14cbcSMatt Macy  * information: Portions Copyright [yyyy] [name of copyright owner]
18eda14cbcSMatt Macy  *
19eda14cbcSMatt Macy  * CDDL HEADER END
20eda14cbcSMatt Macy  */
21eda14cbcSMatt Macy 
22eda14cbcSMatt Macy /*
23eda14cbcSMatt Macy  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
24eda14cbcSMatt Macy  * Copyright (c) 2011, 2018 by Delphix. All rights reserved.
25eda14cbcSMatt Macy  * Copyright (c) 2014 Spectra Logic Corporation, All rights reserved.
26eda14cbcSMatt Macy  * Copyright 2017 Nexenta Systems, Inc.
27eda14cbcSMatt Macy  */
28eda14cbcSMatt Macy 
29eda14cbcSMatt Macy #include <sys/zio.h>
30eda14cbcSMatt Macy #include <sys/spa.h>
31eda14cbcSMatt Macy #include <sys/dmu.h>
32eda14cbcSMatt Macy #include <sys/zfs_context.h>
33eda14cbcSMatt Macy #include <sys/zap.h>
34eda14cbcSMatt Macy #include <sys/zap_impl.h>
35eda14cbcSMatt Macy #include <sys/zap_leaf.h>
36eda14cbcSMatt Macy #include <sys/avl.h>
37eda14cbcSMatt Macy #include <sys/arc.h>
38eda14cbcSMatt Macy #include <sys/dmu_objset.h>
39eda14cbcSMatt Macy 
40eda14cbcSMatt Macy #ifdef _KERNEL
41eda14cbcSMatt Macy #include <sys/sunddi.h>
42eda14cbcSMatt Macy #endif
43eda14cbcSMatt Macy 
44eda14cbcSMatt Macy static int mzap_upgrade(zap_t **zapp,
45eda14cbcSMatt Macy     void *tag, dmu_tx_t *tx, zap_flags_t flags);
46eda14cbcSMatt Macy 
47eda14cbcSMatt Macy uint64_t
48eda14cbcSMatt Macy zap_getflags(zap_t *zap)
49eda14cbcSMatt Macy {
50eda14cbcSMatt Macy 	if (zap->zap_ismicro)
51eda14cbcSMatt Macy 		return (0);
52eda14cbcSMatt Macy 	return (zap_f_phys(zap)->zap_flags);
53eda14cbcSMatt Macy }
54eda14cbcSMatt Macy 
55eda14cbcSMatt Macy int
56eda14cbcSMatt Macy zap_hashbits(zap_t *zap)
57eda14cbcSMatt Macy {
58eda14cbcSMatt Macy 	if (zap_getflags(zap) & ZAP_FLAG_HASH64)
59eda14cbcSMatt Macy 		return (48);
60eda14cbcSMatt Macy 	else
61eda14cbcSMatt Macy 		return (28);
62eda14cbcSMatt Macy }
63eda14cbcSMatt Macy 
64eda14cbcSMatt Macy uint32_t
65eda14cbcSMatt Macy zap_maxcd(zap_t *zap)
66eda14cbcSMatt Macy {
67eda14cbcSMatt Macy 	if (zap_getflags(zap) & ZAP_FLAG_HASH64)
68eda14cbcSMatt Macy 		return ((1<<16)-1);
69eda14cbcSMatt Macy 	else
70eda14cbcSMatt Macy 		return (-1U);
71eda14cbcSMatt Macy }
72eda14cbcSMatt Macy 
73eda14cbcSMatt Macy static uint64_t
74eda14cbcSMatt Macy zap_hash(zap_name_t *zn)
75eda14cbcSMatt Macy {
76eda14cbcSMatt Macy 	zap_t *zap = zn->zn_zap;
77eda14cbcSMatt Macy 	uint64_t h = 0;
78eda14cbcSMatt Macy 
79eda14cbcSMatt Macy 	if (zap_getflags(zap) & ZAP_FLAG_PRE_HASHED_KEY) {
80eda14cbcSMatt Macy 		ASSERT(zap_getflags(zap) & ZAP_FLAG_UINT64_KEY);
81eda14cbcSMatt Macy 		h = *(uint64_t *)zn->zn_key_orig;
82eda14cbcSMatt Macy 	} else {
83eda14cbcSMatt Macy 		h = zap->zap_salt;
84eda14cbcSMatt Macy 		ASSERT(h != 0);
85eda14cbcSMatt Macy 		ASSERT(zfs_crc64_table[128] == ZFS_CRC64_POLY);
86eda14cbcSMatt Macy 
87eda14cbcSMatt Macy 		if (zap_getflags(zap) & ZAP_FLAG_UINT64_KEY) {
88eda14cbcSMatt Macy 			const uint64_t *wp = zn->zn_key_norm;
89eda14cbcSMatt Macy 
90eda14cbcSMatt Macy 			ASSERT(zn->zn_key_intlen == 8);
91eda14cbcSMatt Macy 			for (int i = 0; i < zn->zn_key_norm_numints;
92eda14cbcSMatt Macy 			    wp++, i++) {
93eda14cbcSMatt Macy 				uint64_t word = *wp;
94eda14cbcSMatt Macy 
95eda14cbcSMatt Macy 				for (int j = 0; j < zn->zn_key_intlen; j++) {
96eda14cbcSMatt Macy 					h = (h >> 8) ^
97eda14cbcSMatt Macy 					    zfs_crc64_table[(h ^ word) & 0xFF];
98eda14cbcSMatt Macy 					word >>= NBBY;
99eda14cbcSMatt Macy 				}
100eda14cbcSMatt Macy 			}
101eda14cbcSMatt Macy 		} else {
102eda14cbcSMatt Macy 			const uint8_t *cp = zn->zn_key_norm;
103eda14cbcSMatt Macy 
104eda14cbcSMatt Macy 			/*
105eda14cbcSMatt Macy 			 * We previously stored the terminating null on
106eda14cbcSMatt Macy 			 * disk, but didn't hash it, so we need to
107eda14cbcSMatt Macy 			 * continue to not hash it.  (The
108eda14cbcSMatt Macy 			 * zn_key_*_numints includes the terminating
109eda14cbcSMatt Macy 			 * null for non-binary keys.)
110eda14cbcSMatt Macy 			 */
111eda14cbcSMatt Macy 			int len = zn->zn_key_norm_numints - 1;
112eda14cbcSMatt Macy 
113eda14cbcSMatt Macy 			ASSERT(zn->zn_key_intlen == 1);
114eda14cbcSMatt Macy 			for (int i = 0; i < len; cp++, i++) {
115eda14cbcSMatt Macy 				h = (h >> 8) ^
116eda14cbcSMatt Macy 				    zfs_crc64_table[(h ^ *cp) & 0xFF];
117eda14cbcSMatt Macy 			}
118eda14cbcSMatt Macy 		}
119eda14cbcSMatt Macy 	}
120eda14cbcSMatt Macy 	/*
121eda14cbcSMatt Macy 	 * Don't use all 64 bits, since we need some in the cookie for
122eda14cbcSMatt Macy 	 * the collision differentiator.  We MUST use the high bits,
123eda14cbcSMatt Macy 	 * since those are the ones that we first pay attention to when
124eda14cbcSMatt Macy 	 * choosing the bucket.
125eda14cbcSMatt Macy 	 */
126eda14cbcSMatt Macy 	h &= ~((1ULL << (64 - zap_hashbits(zap))) - 1);
127eda14cbcSMatt Macy 
128eda14cbcSMatt Macy 	return (h);
129eda14cbcSMatt Macy }
130eda14cbcSMatt Macy 
131eda14cbcSMatt Macy static int
132eda14cbcSMatt Macy zap_normalize(zap_t *zap, const char *name, char *namenorm, int normflags)
133eda14cbcSMatt Macy {
134eda14cbcSMatt Macy 	ASSERT(!(zap_getflags(zap) & ZAP_FLAG_UINT64_KEY));
135eda14cbcSMatt Macy 
136eda14cbcSMatt Macy 	size_t inlen = strlen(name) + 1;
137eda14cbcSMatt Macy 	size_t outlen = ZAP_MAXNAMELEN;
138eda14cbcSMatt Macy 
139eda14cbcSMatt Macy 	int err = 0;
140eda14cbcSMatt Macy 	(void) u8_textprep_str((char *)name, &inlen, namenorm, &outlen,
141eda14cbcSMatt Macy 	    normflags | U8_TEXTPREP_IGNORE_NULL | U8_TEXTPREP_IGNORE_INVALID,
142eda14cbcSMatt Macy 	    U8_UNICODE_LATEST, &err);
143eda14cbcSMatt Macy 
144eda14cbcSMatt Macy 	return (err);
145eda14cbcSMatt Macy }
146eda14cbcSMatt Macy 
147eda14cbcSMatt Macy boolean_t
148eda14cbcSMatt Macy zap_match(zap_name_t *zn, const char *matchname)
149eda14cbcSMatt Macy {
150eda14cbcSMatt Macy 	ASSERT(!(zap_getflags(zn->zn_zap) & ZAP_FLAG_UINT64_KEY));
151eda14cbcSMatt Macy 
152eda14cbcSMatt Macy 	if (zn->zn_matchtype & MT_NORMALIZE) {
153eda14cbcSMatt Macy 		char norm[ZAP_MAXNAMELEN];
154eda14cbcSMatt Macy 
155eda14cbcSMatt Macy 		if (zap_normalize(zn->zn_zap, matchname, norm,
156eda14cbcSMatt Macy 		    zn->zn_normflags) != 0)
157eda14cbcSMatt Macy 			return (B_FALSE);
158eda14cbcSMatt Macy 
159eda14cbcSMatt Macy 		return (strcmp(zn->zn_key_norm, norm) == 0);
160eda14cbcSMatt Macy 	} else {
161eda14cbcSMatt Macy 		return (strcmp(zn->zn_key_orig, matchname) == 0);
162eda14cbcSMatt Macy 	}
163eda14cbcSMatt Macy }
164eda14cbcSMatt Macy 
165eda14cbcSMatt Macy void
166eda14cbcSMatt Macy zap_name_free(zap_name_t *zn)
167eda14cbcSMatt Macy {
168eda14cbcSMatt Macy 	kmem_free(zn, sizeof (zap_name_t));
169eda14cbcSMatt Macy }
170eda14cbcSMatt Macy 
171eda14cbcSMatt Macy zap_name_t *
172eda14cbcSMatt Macy zap_name_alloc(zap_t *zap, const char *key, matchtype_t mt)
173eda14cbcSMatt Macy {
174eda14cbcSMatt Macy 	zap_name_t *zn = kmem_alloc(sizeof (zap_name_t), KM_SLEEP);
175eda14cbcSMatt Macy 
176eda14cbcSMatt Macy 	zn->zn_zap = zap;
177eda14cbcSMatt Macy 	zn->zn_key_intlen = sizeof (*key);
178eda14cbcSMatt Macy 	zn->zn_key_orig = key;
179eda14cbcSMatt Macy 	zn->zn_key_orig_numints = strlen(zn->zn_key_orig) + 1;
180eda14cbcSMatt Macy 	zn->zn_matchtype = mt;
181eda14cbcSMatt Macy 	zn->zn_normflags = zap->zap_normflags;
182eda14cbcSMatt Macy 
183eda14cbcSMatt Macy 	/*
184eda14cbcSMatt Macy 	 * If we're dealing with a case sensitive lookup on a mixed or
185eda14cbcSMatt Macy 	 * insensitive fs, remove U8_TEXTPREP_TOUPPER or the lookup
186eda14cbcSMatt Macy 	 * will fold case to all caps overriding the lookup request.
187eda14cbcSMatt Macy 	 */
188eda14cbcSMatt Macy 	if (mt & MT_MATCH_CASE)
189eda14cbcSMatt Macy 		zn->zn_normflags &= ~U8_TEXTPREP_TOUPPER;
190eda14cbcSMatt Macy 
191eda14cbcSMatt Macy 	if (zap->zap_normflags) {
192eda14cbcSMatt Macy 		/*
193eda14cbcSMatt Macy 		 * We *must* use zap_normflags because this normalization is
194eda14cbcSMatt Macy 		 * what the hash is computed from.
195eda14cbcSMatt Macy 		 */
196eda14cbcSMatt Macy 		if (zap_normalize(zap, key, zn->zn_normbuf,
197eda14cbcSMatt Macy 		    zap->zap_normflags) != 0) {
198eda14cbcSMatt Macy 			zap_name_free(zn);
199eda14cbcSMatt Macy 			return (NULL);
200eda14cbcSMatt Macy 		}
201eda14cbcSMatt Macy 		zn->zn_key_norm = zn->zn_normbuf;
202eda14cbcSMatt Macy 		zn->zn_key_norm_numints = strlen(zn->zn_key_norm) + 1;
203eda14cbcSMatt Macy 	} else {
204eda14cbcSMatt Macy 		if (mt != 0) {
205eda14cbcSMatt Macy 			zap_name_free(zn);
206eda14cbcSMatt Macy 			return (NULL);
207eda14cbcSMatt Macy 		}
208eda14cbcSMatt Macy 		zn->zn_key_norm = zn->zn_key_orig;
209eda14cbcSMatt Macy 		zn->zn_key_norm_numints = zn->zn_key_orig_numints;
210eda14cbcSMatt Macy 	}
211eda14cbcSMatt Macy 
212eda14cbcSMatt Macy 	zn->zn_hash = zap_hash(zn);
213eda14cbcSMatt Macy 
214eda14cbcSMatt Macy 	if (zap->zap_normflags != zn->zn_normflags) {
215eda14cbcSMatt Macy 		/*
216eda14cbcSMatt Macy 		 * We *must* use zn_normflags because this normalization is
217eda14cbcSMatt Macy 		 * what the matching is based on.  (Not the hash!)
218eda14cbcSMatt Macy 		 */
219eda14cbcSMatt Macy 		if (zap_normalize(zap, key, zn->zn_normbuf,
220eda14cbcSMatt Macy 		    zn->zn_normflags) != 0) {
221eda14cbcSMatt Macy 			zap_name_free(zn);
222eda14cbcSMatt Macy 			return (NULL);
223eda14cbcSMatt Macy 		}
224eda14cbcSMatt Macy 		zn->zn_key_norm_numints = strlen(zn->zn_key_norm) + 1;
225eda14cbcSMatt Macy 	}
226eda14cbcSMatt Macy 
227eda14cbcSMatt Macy 	return (zn);
228eda14cbcSMatt Macy }
229eda14cbcSMatt Macy 
230eda14cbcSMatt Macy static zap_name_t *
231eda14cbcSMatt Macy zap_name_alloc_uint64(zap_t *zap, const uint64_t *key, int numints)
232eda14cbcSMatt Macy {
233eda14cbcSMatt Macy 	zap_name_t *zn = kmem_alloc(sizeof (zap_name_t), KM_SLEEP);
234eda14cbcSMatt Macy 
235eda14cbcSMatt Macy 	ASSERT(zap->zap_normflags == 0);
236eda14cbcSMatt Macy 	zn->zn_zap = zap;
237eda14cbcSMatt Macy 	zn->zn_key_intlen = sizeof (*key);
238eda14cbcSMatt Macy 	zn->zn_key_orig = zn->zn_key_norm = key;
239eda14cbcSMatt Macy 	zn->zn_key_orig_numints = zn->zn_key_norm_numints = numints;
240eda14cbcSMatt Macy 	zn->zn_matchtype = 0;
241eda14cbcSMatt Macy 
242eda14cbcSMatt Macy 	zn->zn_hash = zap_hash(zn);
243eda14cbcSMatt Macy 	return (zn);
244eda14cbcSMatt Macy }
245eda14cbcSMatt Macy 
246eda14cbcSMatt Macy static void
247eda14cbcSMatt Macy mzap_byteswap(mzap_phys_t *buf, size_t size)
248eda14cbcSMatt Macy {
249eda14cbcSMatt Macy 	buf->mz_block_type = BSWAP_64(buf->mz_block_type);
250eda14cbcSMatt Macy 	buf->mz_salt = BSWAP_64(buf->mz_salt);
251eda14cbcSMatt Macy 	buf->mz_normflags = BSWAP_64(buf->mz_normflags);
252eda14cbcSMatt Macy 	int max = (size / MZAP_ENT_LEN) - 1;
253eda14cbcSMatt Macy 	for (int i = 0; i < max; i++) {
254eda14cbcSMatt Macy 		buf->mz_chunk[i].mze_value =
255eda14cbcSMatt Macy 		    BSWAP_64(buf->mz_chunk[i].mze_value);
256eda14cbcSMatt Macy 		buf->mz_chunk[i].mze_cd =
257eda14cbcSMatt Macy 		    BSWAP_32(buf->mz_chunk[i].mze_cd);
258eda14cbcSMatt Macy 	}
259eda14cbcSMatt Macy }
260eda14cbcSMatt Macy 
261eda14cbcSMatt Macy void
262eda14cbcSMatt Macy zap_byteswap(void *buf, size_t size)
263eda14cbcSMatt Macy {
264eda14cbcSMatt Macy 	uint64_t block_type = *(uint64_t *)buf;
265eda14cbcSMatt Macy 
266eda14cbcSMatt Macy 	if (block_type == ZBT_MICRO || block_type == BSWAP_64(ZBT_MICRO)) {
267eda14cbcSMatt Macy 		/* ASSERT(magic == ZAP_LEAF_MAGIC); */
268eda14cbcSMatt Macy 		mzap_byteswap(buf, size);
269eda14cbcSMatt Macy 	} else {
270eda14cbcSMatt Macy 		fzap_byteswap(buf, size);
271eda14cbcSMatt Macy 	}
272eda14cbcSMatt Macy }
273eda14cbcSMatt Macy 
274eda14cbcSMatt Macy static int
275eda14cbcSMatt Macy mze_compare(const void *arg1, const void *arg2)
276eda14cbcSMatt Macy {
277eda14cbcSMatt Macy 	const mzap_ent_t *mze1 = arg1;
278eda14cbcSMatt Macy 	const mzap_ent_t *mze2 = arg2;
279eda14cbcSMatt Macy 
280eda14cbcSMatt Macy 	int cmp = TREE_CMP(mze1->mze_hash, mze2->mze_hash);
281eda14cbcSMatt Macy 	if (likely(cmp))
282eda14cbcSMatt Macy 		return (cmp);
283eda14cbcSMatt Macy 
284eda14cbcSMatt Macy 	return (TREE_CMP(mze1->mze_cd, mze2->mze_cd));
285eda14cbcSMatt Macy }
286eda14cbcSMatt Macy 
287eda14cbcSMatt Macy static void
288eda14cbcSMatt Macy mze_insert(zap_t *zap, int chunkid, uint64_t hash)
289eda14cbcSMatt Macy {
290eda14cbcSMatt Macy 	ASSERT(zap->zap_ismicro);
291eda14cbcSMatt Macy 	ASSERT(RW_WRITE_HELD(&zap->zap_rwlock));
292eda14cbcSMatt Macy 
293eda14cbcSMatt Macy 	mzap_ent_t *mze = kmem_alloc(sizeof (mzap_ent_t), KM_SLEEP);
294eda14cbcSMatt Macy 	mze->mze_chunkid = chunkid;
295eda14cbcSMatt Macy 	mze->mze_hash = hash;
296eda14cbcSMatt Macy 	mze->mze_cd = MZE_PHYS(zap, mze)->mze_cd;
297eda14cbcSMatt Macy 	ASSERT(MZE_PHYS(zap, mze)->mze_name[0] != 0);
298eda14cbcSMatt Macy 	avl_add(&zap->zap_m.zap_avl, mze);
299eda14cbcSMatt Macy }
300eda14cbcSMatt Macy 
301eda14cbcSMatt Macy static mzap_ent_t *
302eda14cbcSMatt Macy mze_find(zap_name_t *zn)
303eda14cbcSMatt Macy {
304eda14cbcSMatt Macy 	mzap_ent_t mze_tofind;
305eda14cbcSMatt Macy 	mzap_ent_t *mze;
306eda14cbcSMatt Macy 	avl_index_t idx;
307eda14cbcSMatt Macy 	avl_tree_t *avl = &zn->zn_zap->zap_m.zap_avl;
308eda14cbcSMatt Macy 
309eda14cbcSMatt Macy 	ASSERT(zn->zn_zap->zap_ismicro);
310eda14cbcSMatt Macy 	ASSERT(RW_LOCK_HELD(&zn->zn_zap->zap_rwlock));
311eda14cbcSMatt Macy 
312eda14cbcSMatt Macy 	mze_tofind.mze_hash = zn->zn_hash;
313eda14cbcSMatt Macy 	mze_tofind.mze_cd = 0;
314eda14cbcSMatt Macy 
315eda14cbcSMatt Macy 	mze = avl_find(avl, &mze_tofind, &idx);
316eda14cbcSMatt Macy 	if (mze == NULL)
317eda14cbcSMatt Macy 		mze = avl_nearest(avl, idx, AVL_AFTER);
318eda14cbcSMatt Macy 	for (; mze && mze->mze_hash == zn->zn_hash; mze = AVL_NEXT(avl, mze)) {
319eda14cbcSMatt Macy 		ASSERT3U(mze->mze_cd, ==, MZE_PHYS(zn->zn_zap, mze)->mze_cd);
320eda14cbcSMatt Macy 		if (zap_match(zn, MZE_PHYS(zn->zn_zap, mze)->mze_name))
321eda14cbcSMatt Macy 			return (mze);
322eda14cbcSMatt Macy 	}
323eda14cbcSMatt Macy 
324eda14cbcSMatt Macy 	return (NULL);
325eda14cbcSMatt Macy }
326eda14cbcSMatt Macy 
327eda14cbcSMatt Macy static uint32_t
328eda14cbcSMatt Macy mze_find_unused_cd(zap_t *zap, uint64_t hash)
329eda14cbcSMatt Macy {
330eda14cbcSMatt Macy 	mzap_ent_t mze_tofind;
331eda14cbcSMatt Macy 	avl_index_t idx;
332eda14cbcSMatt Macy 	avl_tree_t *avl = &zap->zap_m.zap_avl;
333eda14cbcSMatt Macy 
334eda14cbcSMatt Macy 	ASSERT(zap->zap_ismicro);
335eda14cbcSMatt Macy 	ASSERT(RW_LOCK_HELD(&zap->zap_rwlock));
336eda14cbcSMatt Macy 
337eda14cbcSMatt Macy 	mze_tofind.mze_hash = hash;
338eda14cbcSMatt Macy 	mze_tofind.mze_cd = 0;
339eda14cbcSMatt Macy 
340eda14cbcSMatt Macy 	uint32_t cd = 0;
341eda14cbcSMatt Macy 	for (mzap_ent_t *mze = avl_find(avl, &mze_tofind, &idx);
342eda14cbcSMatt Macy 	    mze && mze->mze_hash == hash; mze = AVL_NEXT(avl, mze)) {
343eda14cbcSMatt Macy 		if (mze->mze_cd != cd)
344eda14cbcSMatt Macy 			break;
345eda14cbcSMatt Macy 		cd++;
346eda14cbcSMatt Macy 	}
347eda14cbcSMatt Macy 
348eda14cbcSMatt Macy 	return (cd);
349eda14cbcSMatt Macy }
350eda14cbcSMatt Macy 
351eda14cbcSMatt Macy /*
352eda14cbcSMatt Macy  * Each mzap entry requires at max : 4 chunks
353eda14cbcSMatt Macy  * 3 chunks for names + 1 chunk for value.
354eda14cbcSMatt Macy  */
355eda14cbcSMatt Macy #define	MZAP_ENT_CHUNKS	(1 + ZAP_LEAF_ARRAY_NCHUNKS(MZAP_NAME_LEN) + \
356eda14cbcSMatt Macy 	ZAP_LEAF_ARRAY_NCHUNKS(sizeof (uint64_t)))
357eda14cbcSMatt Macy 
358eda14cbcSMatt Macy /*
359eda14cbcSMatt Macy  * Check if the current entry keeps the colliding entries under the fatzap leaf
360eda14cbcSMatt Macy  * size.
361eda14cbcSMatt Macy  */
362eda14cbcSMatt Macy static boolean_t
363eda14cbcSMatt Macy mze_canfit_fzap_leaf(zap_name_t *zn, uint64_t hash)
364eda14cbcSMatt Macy {
365eda14cbcSMatt Macy 	zap_t *zap = zn->zn_zap;
366eda14cbcSMatt Macy 	mzap_ent_t mze_tofind;
367eda14cbcSMatt Macy 	mzap_ent_t *mze;
368eda14cbcSMatt Macy 	avl_index_t idx;
369eda14cbcSMatt Macy 	avl_tree_t *avl = &zap->zap_m.zap_avl;
370eda14cbcSMatt Macy 	uint32_t mzap_ents = 0;
371eda14cbcSMatt Macy 
372eda14cbcSMatt Macy 	mze_tofind.mze_hash = hash;
373eda14cbcSMatt Macy 	mze_tofind.mze_cd = 0;
374eda14cbcSMatt Macy 
375eda14cbcSMatt Macy 	for (mze = avl_find(avl, &mze_tofind, &idx);
376eda14cbcSMatt Macy 	    mze && mze->mze_hash == hash; mze = AVL_NEXT(avl, mze)) {
377eda14cbcSMatt Macy 		mzap_ents++;
378eda14cbcSMatt Macy 	}
379eda14cbcSMatt Macy 
380eda14cbcSMatt Macy 	/* Include the new entry being added */
381eda14cbcSMatt Macy 	mzap_ents++;
382eda14cbcSMatt Macy 
383eda14cbcSMatt Macy 	return (ZAP_LEAF_NUMCHUNKS_DEF > (mzap_ents * MZAP_ENT_CHUNKS));
384eda14cbcSMatt Macy }
385eda14cbcSMatt Macy 
386eda14cbcSMatt Macy static void
387eda14cbcSMatt Macy mze_remove(zap_t *zap, mzap_ent_t *mze)
388eda14cbcSMatt Macy {
389eda14cbcSMatt Macy 	ASSERT(zap->zap_ismicro);
390eda14cbcSMatt Macy 	ASSERT(RW_WRITE_HELD(&zap->zap_rwlock));
391eda14cbcSMatt Macy 
392eda14cbcSMatt Macy 	avl_remove(&zap->zap_m.zap_avl, mze);
393eda14cbcSMatt Macy 	kmem_free(mze, sizeof (mzap_ent_t));
394eda14cbcSMatt Macy }
395eda14cbcSMatt Macy 
396eda14cbcSMatt Macy static void
397eda14cbcSMatt Macy mze_destroy(zap_t *zap)
398eda14cbcSMatt Macy {
399eda14cbcSMatt Macy 	mzap_ent_t *mze;
400eda14cbcSMatt Macy 	void *avlcookie = NULL;
401eda14cbcSMatt Macy 
402eda14cbcSMatt Macy 	while ((mze = avl_destroy_nodes(&zap->zap_m.zap_avl, &avlcookie)))
403eda14cbcSMatt Macy 		kmem_free(mze, sizeof (mzap_ent_t));
404eda14cbcSMatt Macy 	avl_destroy(&zap->zap_m.zap_avl);
405eda14cbcSMatt Macy }
406eda14cbcSMatt Macy 
407eda14cbcSMatt Macy static zap_t *
408eda14cbcSMatt Macy mzap_open(objset_t *os, uint64_t obj, dmu_buf_t *db)
409eda14cbcSMatt Macy {
410eda14cbcSMatt Macy 	zap_t *winner;
411eda14cbcSMatt Macy 	uint64_t *zap_hdr = (uint64_t *)db->db_data;
412eda14cbcSMatt Macy 	uint64_t zap_block_type = zap_hdr[0];
413eda14cbcSMatt Macy 	uint64_t zap_magic = zap_hdr[1];
414eda14cbcSMatt Macy 
415eda14cbcSMatt Macy 	ASSERT3U(MZAP_ENT_LEN, ==, sizeof (mzap_ent_phys_t));
416eda14cbcSMatt Macy 
417eda14cbcSMatt Macy 	zap_t *zap = kmem_zalloc(sizeof (zap_t), KM_SLEEP);
418eda14cbcSMatt Macy 	rw_init(&zap->zap_rwlock, NULL, RW_DEFAULT, NULL);
419eda14cbcSMatt Macy 	rw_enter(&zap->zap_rwlock, RW_WRITER);
420eda14cbcSMatt Macy 	zap->zap_objset = os;
421eda14cbcSMatt Macy 	zap->zap_object = obj;
422eda14cbcSMatt Macy 	zap->zap_dbuf = db;
423eda14cbcSMatt Macy 
424eda14cbcSMatt Macy 	if (zap_block_type != ZBT_MICRO) {
425eda14cbcSMatt Macy 		mutex_init(&zap->zap_f.zap_num_entries_mtx, 0, MUTEX_DEFAULT,
426eda14cbcSMatt Macy 		    0);
427eda14cbcSMatt Macy 		zap->zap_f.zap_block_shift = highbit64(db->db_size) - 1;
428eda14cbcSMatt Macy 		if (zap_block_type != ZBT_HEADER || zap_magic != ZAP_MAGIC) {
429eda14cbcSMatt Macy 			winner = NULL;	/* No actual winner here... */
430eda14cbcSMatt Macy 			goto handle_winner;
431eda14cbcSMatt Macy 		}
432eda14cbcSMatt Macy 	} else {
433eda14cbcSMatt Macy 		zap->zap_ismicro = TRUE;
434eda14cbcSMatt Macy 	}
435eda14cbcSMatt Macy 
436eda14cbcSMatt Macy 	/*
437eda14cbcSMatt Macy 	 * Make sure that zap_ismicro is set before we let others see
438eda14cbcSMatt Macy 	 * it, because zap_lockdir() checks zap_ismicro without the lock
439eda14cbcSMatt Macy 	 * held.
440eda14cbcSMatt Macy 	 */
441eda14cbcSMatt Macy 	dmu_buf_init_user(&zap->zap_dbu, zap_evict_sync, NULL, &zap->zap_dbuf);
442eda14cbcSMatt Macy 	winner = dmu_buf_set_user(db, &zap->zap_dbu);
443eda14cbcSMatt Macy 
444eda14cbcSMatt Macy 	if (winner != NULL)
445eda14cbcSMatt Macy 		goto handle_winner;
446eda14cbcSMatt Macy 
447eda14cbcSMatt Macy 	if (zap->zap_ismicro) {
448eda14cbcSMatt Macy 		zap->zap_salt = zap_m_phys(zap)->mz_salt;
449eda14cbcSMatt Macy 		zap->zap_normflags = zap_m_phys(zap)->mz_normflags;
450eda14cbcSMatt Macy 		zap->zap_m.zap_num_chunks = db->db_size / MZAP_ENT_LEN - 1;
451eda14cbcSMatt Macy 		avl_create(&zap->zap_m.zap_avl, mze_compare,
452eda14cbcSMatt Macy 		    sizeof (mzap_ent_t), offsetof(mzap_ent_t, mze_node));
453eda14cbcSMatt Macy 
454eda14cbcSMatt Macy 		for (int i = 0; i < zap->zap_m.zap_num_chunks; i++) {
455eda14cbcSMatt Macy 			mzap_ent_phys_t *mze =
456eda14cbcSMatt Macy 			    &zap_m_phys(zap)->mz_chunk[i];
457eda14cbcSMatt Macy 			if (mze->mze_name[0]) {
458eda14cbcSMatt Macy 				zap_name_t *zn;
459eda14cbcSMatt Macy 
460eda14cbcSMatt Macy 				zap->zap_m.zap_num_entries++;
461eda14cbcSMatt Macy 				zn = zap_name_alloc(zap, mze->mze_name, 0);
462eda14cbcSMatt Macy 				mze_insert(zap, i, zn->zn_hash);
463eda14cbcSMatt Macy 				zap_name_free(zn);
464eda14cbcSMatt Macy 			}
465eda14cbcSMatt Macy 		}
466eda14cbcSMatt Macy 	} else {
467eda14cbcSMatt Macy 		zap->zap_salt = zap_f_phys(zap)->zap_salt;
468eda14cbcSMatt Macy 		zap->zap_normflags = zap_f_phys(zap)->zap_normflags;
469eda14cbcSMatt Macy 
470eda14cbcSMatt Macy 		ASSERT3U(sizeof (struct zap_leaf_header), ==,
471eda14cbcSMatt Macy 		    2*ZAP_LEAF_CHUNKSIZE);
472eda14cbcSMatt Macy 
473eda14cbcSMatt Macy 		/*
474eda14cbcSMatt Macy 		 * The embedded pointer table should not overlap the
475eda14cbcSMatt Macy 		 * other members.
476eda14cbcSMatt Macy 		 */
477eda14cbcSMatt Macy 		ASSERT3P(&ZAP_EMBEDDED_PTRTBL_ENT(zap, 0), >,
478eda14cbcSMatt Macy 		    &zap_f_phys(zap)->zap_salt);
479eda14cbcSMatt Macy 
480eda14cbcSMatt Macy 		/*
481eda14cbcSMatt Macy 		 * The embedded pointer table should end at the end of
482eda14cbcSMatt Macy 		 * the block
483eda14cbcSMatt Macy 		 */
484eda14cbcSMatt Macy 		ASSERT3U((uintptr_t)&ZAP_EMBEDDED_PTRTBL_ENT(zap,
485eda14cbcSMatt Macy 		    1<<ZAP_EMBEDDED_PTRTBL_SHIFT(zap)) -
486eda14cbcSMatt Macy 		    (uintptr_t)zap_f_phys(zap), ==,
487eda14cbcSMatt Macy 		    zap->zap_dbuf->db_size);
488eda14cbcSMatt Macy 	}
489eda14cbcSMatt Macy 	rw_exit(&zap->zap_rwlock);
490eda14cbcSMatt Macy 	return (zap);
491eda14cbcSMatt Macy 
492eda14cbcSMatt Macy handle_winner:
493eda14cbcSMatt Macy 	rw_exit(&zap->zap_rwlock);
494eda14cbcSMatt Macy 	rw_destroy(&zap->zap_rwlock);
495eda14cbcSMatt Macy 	if (!zap->zap_ismicro)
496eda14cbcSMatt Macy 		mutex_destroy(&zap->zap_f.zap_num_entries_mtx);
497eda14cbcSMatt Macy 	kmem_free(zap, sizeof (zap_t));
498eda14cbcSMatt Macy 	return (winner);
499eda14cbcSMatt Macy }
500eda14cbcSMatt Macy 
501eda14cbcSMatt Macy /*
502eda14cbcSMatt Macy  * This routine "consumes" the caller's hold on the dbuf, which must
503eda14cbcSMatt Macy  * have the specified tag.
504eda14cbcSMatt Macy  */
505eda14cbcSMatt Macy static int
506eda14cbcSMatt Macy zap_lockdir_impl(dmu_buf_t *db, void *tag, dmu_tx_t *tx,
507eda14cbcSMatt Macy     krw_t lti, boolean_t fatreader, boolean_t adding, zap_t **zapp)
508eda14cbcSMatt Macy {
509eda14cbcSMatt Macy 	ASSERT0(db->db_offset);
510eda14cbcSMatt Macy 	objset_t *os = dmu_buf_get_objset(db);
511eda14cbcSMatt Macy 	uint64_t obj = db->db_object;
512eda14cbcSMatt Macy 	dmu_object_info_t doi;
513eda14cbcSMatt Macy 
514eda14cbcSMatt Macy 	*zapp = NULL;
515eda14cbcSMatt Macy 
516eda14cbcSMatt Macy 	dmu_object_info_from_db(db, &doi);
517eda14cbcSMatt Macy 	if (DMU_OT_BYTESWAP(doi.doi_type) != DMU_BSWAP_ZAP)
518eda14cbcSMatt Macy 		return (SET_ERROR(EINVAL));
519eda14cbcSMatt Macy 
520eda14cbcSMatt Macy 	zap_t *zap = dmu_buf_get_user(db);
521eda14cbcSMatt Macy 	if (zap == NULL) {
522eda14cbcSMatt Macy 		zap = mzap_open(os, obj, db);
523eda14cbcSMatt Macy 		if (zap == NULL) {
524eda14cbcSMatt Macy 			/*
525eda14cbcSMatt Macy 			 * mzap_open() didn't like what it saw on-disk.
526eda14cbcSMatt Macy 			 * Check for corruption!
527eda14cbcSMatt Macy 			 */
528eda14cbcSMatt Macy 			return (SET_ERROR(EIO));
529eda14cbcSMatt Macy 		}
530eda14cbcSMatt Macy 	}
531eda14cbcSMatt Macy 
532eda14cbcSMatt Macy 	/*
533eda14cbcSMatt Macy 	 * We're checking zap_ismicro without the lock held, in order to
534eda14cbcSMatt Macy 	 * tell what type of lock we want.  Once we have some sort of
535eda14cbcSMatt Macy 	 * lock, see if it really is the right type.  In practice this
536eda14cbcSMatt Macy 	 * can only be different if it was upgraded from micro to fat,
537eda14cbcSMatt Macy 	 * and micro wanted WRITER but fat only needs READER.
538eda14cbcSMatt Macy 	 */
539eda14cbcSMatt Macy 	krw_t lt = (!zap->zap_ismicro && fatreader) ? RW_READER : lti;
540eda14cbcSMatt Macy 	rw_enter(&zap->zap_rwlock, lt);
541eda14cbcSMatt Macy 	if (lt != ((!zap->zap_ismicro && fatreader) ? RW_READER : lti)) {
542eda14cbcSMatt Macy 		/* it was upgraded, now we only need reader */
543eda14cbcSMatt Macy 		ASSERT(lt == RW_WRITER);
544eda14cbcSMatt Macy 		ASSERT(RW_READER ==
545eda14cbcSMatt Macy 		    ((!zap->zap_ismicro && fatreader) ? RW_READER : lti));
546eda14cbcSMatt Macy 		rw_downgrade(&zap->zap_rwlock);
547eda14cbcSMatt Macy 		lt = RW_READER;
548eda14cbcSMatt Macy 	}
549eda14cbcSMatt Macy 
550eda14cbcSMatt Macy 	zap->zap_objset = os;
551eda14cbcSMatt Macy 
552eda14cbcSMatt Macy 	if (lt == RW_WRITER)
553eda14cbcSMatt Macy 		dmu_buf_will_dirty(db, tx);
554eda14cbcSMatt Macy 
555eda14cbcSMatt Macy 	ASSERT3P(zap->zap_dbuf, ==, db);
556eda14cbcSMatt Macy 
557eda14cbcSMatt Macy 	ASSERT(!zap->zap_ismicro ||
558eda14cbcSMatt Macy 	    zap->zap_m.zap_num_entries <= zap->zap_m.zap_num_chunks);
559eda14cbcSMatt Macy 	if (zap->zap_ismicro && tx && adding &&
560eda14cbcSMatt Macy 	    zap->zap_m.zap_num_entries == zap->zap_m.zap_num_chunks) {
561eda14cbcSMatt Macy 		uint64_t newsz = db->db_size + SPA_MINBLOCKSIZE;
562eda14cbcSMatt Macy 		if (newsz > MZAP_MAX_BLKSZ) {
563eda14cbcSMatt Macy 			dprintf("upgrading obj %llu: num_entries=%u\n",
56433b8c039SMartin Matuska 			    (u_longlong_t)obj, zap->zap_m.zap_num_entries);
565eda14cbcSMatt Macy 			*zapp = zap;
566eda14cbcSMatt Macy 			int err = mzap_upgrade(zapp, tag, tx, 0);
567eda14cbcSMatt Macy 			if (err != 0)
568eda14cbcSMatt Macy 				rw_exit(&zap->zap_rwlock);
569eda14cbcSMatt Macy 			return (err);
570eda14cbcSMatt Macy 		}
571eda14cbcSMatt Macy 		VERIFY0(dmu_object_set_blocksize(os, obj, newsz, 0, tx));
572eda14cbcSMatt Macy 		zap->zap_m.zap_num_chunks =
573eda14cbcSMatt Macy 		    db->db_size / MZAP_ENT_LEN - 1;
574eda14cbcSMatt Macy 	}
575eda14cbcSMatt Macy 
576eda14cbcSMatt Macy 	*zapp = zap;
577eda14cbcSMatt Macy 	return (0);
578eda14cbcSMatt Macy }
579eda14cbcSMatt Macy 
580eda14cbcSMatt Macy static int
581eda14cbcSMatt Macy zap_lockdir_by_dnode(dnode_t *dn, dmu_tx_t *tx,
582eda14cbcSMatt Macy     krw_t lti, boolean_t fatreader, boolean_t adding, void *tag, zap_t **zapp)
583eda14cbcSMatt Macy {
584eda14cbcSMatt Macy 	dmu_buf_t *db;
585eda14cbcSMatt Macy 
586eda14cbcSMatt Macy 	int err = dmu_buf_hold_by_dnode(dn, 0, tag, &db, DMU_READ_NO_PREFETCH);
587eda14cbcSMatt Macy 	if (err != 0) {
588eda14cbcSMatt Macy 		return (err);
589eda14cbcSMatt Macy 	}
590eda14cbcSMatt Macy #ifdef ZFS_DEBUG
591eda14cbcSMatt Macy 	{
592eda14cbcSMatt Macy 		dmu_object_info_t doi;
593eda14cbcSMatt Macy 		dmu_object_info_from_db(db, &doi);
594eda14cbcSMatt Macy 		ASSERT3U(DMU_OT_BYTESWAP(doi.doi_type), ==, DMU_BSWAP_ZAP);
595eda14cbcSMatt Macy 	}
596eda14cbcSMatt Macy #endif
597eda14cbcSMatt Macy 
598eda14cbcSMatt Macy 	err = zap_lockdir_impl(db, tag, tx, lti, fatreader, adding, zapp);
599eda14cbcSMatt Macy 	if (err != 0) {
600eda14cbcSMatt Macy 		dmu_buf_rele(db, tag);
601eda14cbcSMatt Macy 	}
602eda14cbcSMatt Macy 	return (err);
603eda14cbcSMatt Macy }
604eda14cbcSMatt Macy 
605eda14cbcSMatt Macy int
606eda14cbcSMatt Macy zap_lockdir(objset_t *os, uint64_t obj, dmu_tx_t *tx,
607eda14cbcSMatt Macy     krw_t lti, boolean_t fatreader, boolean_t adding, void *tag, zap_t **zapp)
608eda14cbcSMatt Macy {
609eda14cbcSMatt Macy 	dmu_buf_t *db;
610eda14cbcSMatt Macy 
611eda14cbcSMatt Macy 	int err = dmu_buf_hold(os, obj, 0, tag, &db, DMU_READ_NO_PREFETCH);
612eda14cbcSMatt Macy 	if (err != 0)
613eda14cbcSMatt Macy 		return (err);
614eda14cbcSMatt Macy #ifdef ZFS_DEBUG
615eda14cbcSMatt Macy 	{
616eda14cbcSMatt Macy 		dmu_object_info_t doi;
617eda14cbcSMatt Macy 		dmu_object_info_from_db(db, &doi);
618eda14cbcSMatt Macy 		ASSERT3U(DMU_OT_BYTESWAP(doi.doi_type), ==, DMU_BSWAP_ZAP);
619eda14cbcSMatt Macy 	}
620eda14cbcSMatt Macy #endif
621eda14cbcSMatt Macy 	err = zap_lockdir_impl(db, tag, tx, lti, fatreader, adding, zapp);
622eda14cbcSMatt Macy 	if (err != 0)
623eda14cbcSMatt Macy 		dmu_buf_rele(db, tag);
624eda14cbcSMatt Macy 	return (err);
625eda14cbcSMatt Macy }
626eda14cbcSMatt Macy 
627eda14cbcSMatt Macy void
628eda14cbcSMatt Macy zap_unlockdir(zap_t *zap, void *tag)
629eda14cbcSMatt Macy {
630eda14cbcSMatt Macy 	rw_exit(&zap->zap_rwlock);
631eda14cbcSMatt Macy 	dmu_buf_rele(zap->zap_dbuf, tag);
632eda14cbcSMatt Macy }
633eda14cbcSMatt Macy 
634eda14cbcSMatt Macy static int
635eda14cbcSMatt Macy mzap_upgrade(zap_t **zapp, void *tag, dmu_tx_t *tx, zap_flags_t flags)
636eda14cbcSMatt Macy {
637eda14cbcSMatt Macy 	int err = 0;
638eda14cbcSMatt Macy 	zap_t *zap = *zapp;
639eda14cbcSMatt Macy 
640eda14cbcSMatt Macy 	ASSERT(RW_WRITE_HELD(&zap->zap_rwlock));
641eda14cbcSMatt Macy 
642eda14cbcSMatt Macy 	int sz = zap->zap_dbuf->db_size;
643eda14cbcSMatt Macy 	mzap_phys_t *mzp = vmem_alloc(sz, KM_SLEEP);
644*da5137abSMartin Matuska 	memcpy(mzp, zap->zap_dbuf->db_data, sz);
645eda14cbcSMatt Macy 	int nchunks = zap->zap_m.zap_num_chunks;
646eda14cbcSMatt Macy 
647eda14cbcSMatt Macy 	if (!flags) {
648eda14cbcSMatt Macy 		err = dmu_object_set_blocksize(zap->zap_objset, zap->zap_object,
649eda14cbcSMatt Macy 		    1ULL << fzap_default_block_shift, 0, tx);
650eda14cbcSMatt Macy 		if (err != 0) {
651eda14cbcSMatt Macy 			vmem_free(mzp, sz);
652eda14cbcSMatt Macy 			return (err);
653eda14cbcSMatt Macy 		}
654eda14cbcSMatt Macy 	}
655eda14cbcSMatt Macy 
656eda14cbcSMatt Macy 	dprintf("upgrading obj=%llu with %u chunks\n",
65733b8c039SMartin Matuska 	    (u_longlong_t)zap->zap_object, nchunks);
658eda14cbcSMatt Macy 	/* XXX destroy the avl later, so we can use the stored hash value */
659eda14cbcSMatt Macy 	mze_destroy(zap);
660eda14cbcSMatt Macy 
661eda14cbcSMatt Macy 	fzap_upgrade(zap, tx, flags);
662eda14cbcSMatt Macy 
663eda14cbcSMatt Macy 	for (int i = 0; i < nchunks; i++) {
664eda14cbcSMatt Macy 		mzap_ent_phys_t *mze = &mzp->mz_chunk[i];
665eda14cbcSMatt Macy 		if (mze->mze_name[0] == 0)
666eda14cbcSMatt Macy 			continue;
667eda14cbcSMatt Macy 		dprintf("adding %s=%llu\n",
66833b8c039SMartin Matuska 		    mze->mze_name, (u_longlong_t)mze->mze_value);
669eda14cbcSMatt Macy 		zap_name_t *zn = zap_name_alloc(zap, mze->mze_name, 0);
670eda14cbcSMatt Macy 		/* If we fail here, we would end up losing entries */
671eda14cbcSMatt Macy 		VERIFY0(fzap_add_cd(zn, 8, 1, &mze->mze_value, mze->mze_cd,
672eda14cbcSMatt Macy 		    tag, tx));
673eda14cbcSMatt Macy 		zap = zn->zn_zap;	/* fzap_add_cd() may change zap */
674eda14cbcSMatt Macy 		zap_name_free(zn);
675eda14cbcSMatt Macy 	}
676eda14cbcSMatt Macy 	vmem_free(mzp, sz);
677eda14cbcSMatt Macy 	*zapp = zap;
678eda14cbcSMatt Macy 	return (0);
679eda14cbcSMatt Macy }
680eda14cbcSMatt Macy 
681eda14cbcSMatt Macy /*
682eda14cbcSMatt Macy  * The "normflags" determine the behavior of the matchtype_t which is
683eda14cbcSMatt Macy  * passed to zap_lookup_norm().  Names which have the same normalized
684eda14cbcSMatt Macy  * version will be stored with the same hash value, and therefore we can
685eda14cbcSMatt Macy  * perform normalization-insensitive lookups.  We can be Unicode form-
686eda14cbcSMatt Macy  * insensitive and/or case-insensitive.  The following flags are valid for
687eda14cbcSMatt Macy  * "normflags":
688eda14cbcSMatt Macy  *
689eda14cbcSMatt Macy  * U8_TEXTPREP_NFC
690eda14cbcSMatt Macy  * U8_TEXTPREP_NFD
691eda14cbcSMatt Macy  * U8_TEXTPREP_NFKC
692eda14cbcSMatt Macy  * U8_TEXTPREP_NFKD
693eda14cbcSMatt Macy  * U8_TEXTPREP_TOUPPER
694eda14cbcSMatt Macy  *
695eda14cbcSMatt Macy  * The *_NF* (Normalization Form) flags are mutually exclusive; at most one
696eda14cbcSMatt Macy  * of them may be supplied.
697eda14cbcSMatt Macy  */
698eda14cbcSMatt Macy void
699eda14cbcSMatt Macy mzap_create_impl(dnode_t *dn, int normflags, zap_flags_t flags, dmu_tx_t *tx)
700eda14cbcSMatt Macy {
701eda14cbcSMatt Macy 	dmu_buf_t *db;
702eda14cbcSMatt Macy 
703eda14cbcSMatt Macy 	VERIFY0(dmu_buf_hold_by_dnode(dn, 0, FTAG, &db, DMU_READ_NO_PREFETCH));
704eda14cbcSMatt Macy 
705eda14cbcSMatt Macy 	dmu_buf_will_dirty(db, tx);
706eda14cbcSMatt Macy 	mzap_phys_t *zp = db->db_data;
707eda14cbcSMatt Macy 	zp->mz_block_type = ZBT_MICRO;
708eda14cbcSMatt Macy 	zp->mz_salt =
709eda14cbcSMatt Macy 	    ((uintptr_t)db ^ (uintptr_t)tx ^ (dn->dn_object << 1)) | 1ULL;
710eda14cbcSMatt Macy 	zp->mz_normflags = normflags;
711eda14cbcSMatt Macy 
712eda14cbcSMatt Macy 	if (flags != 0) {
713eda14cbcSMatt Macy 		zap_t *zap;
714eda14cbcSMatt Macy 		/* Only fat zap supports flags; upgrade immediately. */
715eda14cbcSMatt Macy 		VERIFY0(zap_lockdir_impl(db, FTAG, tx, RW_WRITER,
716eda14cbcSMatt Macy 		    B_FALSE, B_FALSE, &zap));
717eda14cbcSMatt Macy 		VERIFY0(mzap_upgrade(&zap, FTAG, tx, flags));
718eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
719eda14cbcSMatt Macy 	} else {
720eda14cbcSMatt Macy 		dmu_buf_rele(db, FTAG);
721eda14cbcSMatt Macy 	}
722eda14cbcSMatt Macy }
723eda14cbcSMatt Macy 
724eda14cbcSMatt Macy static uint64_t
725eda14cbcSMatt Macy zap_create_impl(objset_t *os, int normflags, zap_flags_t flags,
726eda14cbcSMatt Macy     dmu_object_type_t ot, int leaf_blockshift, int indirect_blockshift,
727eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, int dnodesize,
728eda14cbcSMatt Macy     dnode_t **allocated_dnode, void *tag, dmu_tx_t *tx)
729eda14cbcSMatt Macy {
730eda14cbcSMatt Macy 	uint64_t obj;
731eda14cbcSMatt Macy 
732eda14cbcSMatt Macy 	ASSERT3U(DMU_OT_BYTESWAP(ot), ==, DMU_BSWAP_ZAP);
733eda14cbcSMatt Macy 
734eda14cbcSMatt Macy 	if (allocated_dnode == NULL) {
735eda14cbcSMatt Macy 		dnode_t *dn;
736eda14cbcSMatt Macy 		obj = dmu_object_alloc_hold(os, ot, 1ULL << leaf_blockshift,
737eda14cbcSMatt Macy 		    indirect_blockshift, bonustype, bonuslen, dnodesize,
738eda14cbcSMatt Macy 		    &dn, FTAG, tx);
739eda14cbcSMatt Macy 		mzap_create_impl(dn, normflags, flags, tx);
740eda14cbcSMatt Macy 		dnode_rele(dn, FTAG);
741eda14cbcSMatt Macy 	} else {
742eda14cbcSMatt Macy 		obj = dmu_object_alloc_hold(os, ot, 1ULL << leaf_blockshift,
743eda14cbcSMatt Macy 		    indirect_blockshift, bonustype, bonuslen, dnodesize,
744eda14cbcSMatt Macy 		    allocated_dnode, tag, tx);
745eda14cbcSMatt Macy 		mzap_create_impl(*allocated_dnode, normflags, flags, tx);
746eda14cbcSMatt Macy 	}
747eda14cbcSMatt Macy 
748eda14cbcSMatt Macy 	return (obj);
749eda14cbcSMatt Macy }
750eda14cbcSMatt Macy 
751eda14cbcSMatt Macy int
752eda14cbcSMatt Macy zap_create_claim(objset_t *os, uint64_t obj, dmu_object_type_t ot,
753eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, dmu_tx_t *tx)
754eda14cbcSMatt Macy {
755eda14cbcSMatt Macy 	return (zap_create_claim_dnsize(os, obj, ot, bonustype, bonuslen,
756eda14cbcSMatt Macy 	    0, tx));
757eda14cbcSMatt Macy }
758eda14cbcSMatt Macy 
759eda14cbcSMatt Macy int
760eda14cbcSMatt Macy zap_create_claim_dnsize(objset_t *os, uint64_t obj, dmu_object_type_t ot,
761eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, int dnodesize, dmu_tx_t *tx)
762eda14cbcSMatt Macy {
763eda14cbcSMatt Macy 	return (zap_create_claim_norm_dnsize(os, obj,
764eda14cbcSMatt Macy 	    0, ot, bonustype, bonuslen, dnodesize, tx));
765eda14cbcSMatt Macy }
766eda14cbcSMatt Macy 
767eda14cbcSMatt Macy int
768eda14cbcSMatt Macy zap_create_claim_norm(objset_t *os, uint64_t obj, int normflags,
769eda14cbcSMatt Macy     dmu_object_type_t ot,
770eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, dmu_tx_t *tx)
771eda14cbcSMatt Macy {
772eda14cbcSMatt Macy 	return (zap_create_claim_norm_dnsize(os, obj, normflags, ot, bonustype,
773eda14cbcSMatt Macy 	    bonuslen, 0, tx));
774eda14cbcSMatt Macy }
775eda14cbcSMatt Macy 
776eda14cbcSMatt Macy int
777eda14cbcSMatt Macy zap_create_claim_norm_dnsize(objset_t *os, uint64_t obj, int normflags,
778eda14cbcSMatt Macy     dmu_object_type_t ot, dmu_object_type_t bonustype, int bonuslen,
779eda14cbcSMatt Macy     int dnodesize, dmu_tx_t *tx)
780eda14cbcSMatt Macy {
781eda14cbcSMatt Macy 	dnode_t *dn;
782eda14cbcSMatt Macy 	int error;
783eda14cbcSMatt Macy 
784eda14cbcSMatt Macy 	ASSERT3U(DMU_OT_BYTESWAP(ot), ==, DMU_BSWAP_ZAP);
785eda14cbcSMatt Macy 	error = dmu_object_claim_dnsize(os, obj, ot, 0, bonustype, bonuslen,
786eda14cbcSMatt Macy 	    dnodesize, tx);
787eda14cbcSMatt Macy 	if (error != 0)
788eda14cbcSMatt Macy 		return (error);
789eda14cbcSMatt Macy 
790eda14cbcSMatt Macy 	error = dnode_hold(os, obj, FTAG, &dn);
791eda14cbcSMatt Macy 	if (error != 0)
792eda14cbcSMatt Macy 		return (error);
793eda14cbcSMatt Macy 
794eda14cbcSMatt Macy 	mzap_create_impl(dn, normflags, 0, tx);
795eda14cbcSMatt Macy 
796eda14cbcSMatt Macy 	dnode_rele(dn, FTAG);
797eda14cbcSMatt Macy 
798eda14cbcSMatt Macy 	return (0);
799eda14cbcSMatt Macy }
800eda14cbcSMatt Macy 
801eda14cbcSMatt Macy uint64_t
802eda14cbcSMatt Macy zap_create(objset_t *os, dmu_object_type_t ot,
803eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, dmu_tx_t *tx)
804eda14cbcSMatt Macy {
805eda14cbcSMatt Macy 	return (zap_create_norm(os, 0, ot, bonustype, bonuslen, tx));
806eda14cbcSMatt Macy }
807eda14cbcSMatt Macy 
808eda14cbcSMatt Macy uint64_t
809eda14cbcSMatt Macy zap_create_dnsize(objset_t *os, dmu_object_type_t ot,
810eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, int dnodesize, dmu_tx_t *tx)
811eda14cbcSMatt Macy {
812eda14cbcSMatt Macy 	return (zap_create_norm_dnsize(os, 0, ot, bonustype, bonuslen,
813eda14cbcSMatt Macy 	    dnodesize, tx));
814eda14cbcSMatt Macy }
815eda14cbcSMatt Macy 
816eda14cbcSMatt Macy uint64_t
817eda14cbcSMatt Macy zap_create_norm(objset_t *os, int normflags, dmu_object_type_t ot,
818eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, dmu_tx_t *tx)
819eda14cbcSMatt Macy {
820eda14cbcSMatt Macy 	return (zap_create_norm_dnsize(os, normflags, ot, bonustype, bonuslen,
821eda14cbcSMatt Macy 	    0, tx));
822eda14cbcSMatt Macy }
823eda14cbcSMatt Macy 
824eda14cbcSMatt Macy uint64_t
825eda14cbcSMatt Macy zap_create_norm_dnsize(objset_t *os, int normflags, dmu_object_type_t ot,
826eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, int dnodesize, dmu_tx_t *tx)
827eda14cbcSMatt Macy {
828eda14cbcSMatt Macy 	return (zap_create_impl(os, normflags, 0, ot, 0, 0,
829eda14cbcSMatt Macy 	    bonustype, bonuslen, dnodesize, NULL, NULL, tx));
830eda14cbcSMatt Macy }
831eda14cbcSMatt Macy 
832eda14cbcSMatt Macy uint64_t
833eda14cbcSMatt Macy zap_create_flags(objset_t *os, int normflags, zap_flags_t flags,
834eda14cbcSMatt Macy     dmu_object_type_t ot, int leaf_blockshift, int indirect_blockshift,
835eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, dmu_tx_t *tx)
836eda14cbcSMatt Macy {
837eda14cbcSMatt Macy 	return (zap_create_flags_dnsize(os, normflags, flags, ot,
838eda14cbcSMatt Macy 	    leaf_blockshift, indirect_blockshift, bonustype, bonuslen, 0, tx));
839eda14cbcSMatt Macy }
840eda14cbcSMatt Macy 
841eda14cbcSMatt Macy uint64_t
842eda14cbcSMatt Macy zap_create_flags_dnsize(objset_t *os, int normflags, zap_flags_t flags,
843eda14cbcSMatt Macy     dmu_object_type_t ot, int leaf_blockshift, int indirect_blockshift,
844eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, int dnodesize, dmu_tx_t *tx)
845eda14cbcSMatt Macy {
846eda14cbcSMatt Macy 	return (zap_create_impl(os, normflags, flags, ot, leaf_blockshift,
847eda14cbcSMatt Macy 	    indirect_blockshift, bonustype, bonuslen, dnodesize, NULL, NULL,
848eda14cbcSMatt Macy 	    tx));
849eda14cbcSMatt Macy }
850eda14cbcSMatt Macy 
851eda14cbcSMatt Macy /*
852eda14cbcSMatt Macy  * Create a zap object and return a pointer to the newly allocated dnode via
853eda14cbcSMatt Macy  * the allocated_dnode argument.  The returned dnode will be held and the
854eda14cbcSMatt Macy  * caller is responsible for releasing the hold by calling dnode_rele().
855eda14cbcSMatt Macy  */
856eda14cbcSMatt Macy uint64_t
857eda14cbcSMatt Macy zap_create_hold(objset_t *os, int normflags, zap_flags_t flags,
858eda14cbcSMatt Macy     dmu_object_type_t ot, int leaf_blockshift, int indirect_blockshift,
859eda14cbcSMatt Macy     dmu_object_type_t bonustype, int bonuslen, int dnodesize,
860eda14cbcSMatt Macy     dnode_t **allocated_dnode, void *tag, dmu_tx_t *tx)
861eda14cbcSMatt Macy {
862eda14cbcSMatt Macy 	return (zap_create_impl(os, normflags, flags, ot, leaf_blockshift,
863eda14cbcSMatt Macy 	    indirect_blockshift, bonustype, bonuslen, dnodesize,
864eda14cbcSMatt Macy 	    allocated_dnode, tag, tx));
865eda14cbcSMatt Macy }
866eda14cbcSMatt Macy 
867eda14cbcSMatt Macy int
868eda14cbcSMatt Macy zap_destroy(objset_t *os, uint64_t zapobj, dmu_tx_t *tx)
869eda14cbcSMatt Macy {
870eda14cbcSMatt Macy 	/*
871eda14cbcSMatt Macy 	 * dmu_object_free will free the object number and free the
872eda14cbcSMatt Macy 	 * data.  Freeing the data will cause our pageout function to be
873eda14cbcSMatt Macy 	 * called, which will destroy our data (zap_leaf_t's and zap_t).
874eda14cbcSMatt Macy 	 */
875eda14cbcSMatt Macy 
876eda14cbcSMatt Macy 	return (dmu_object_free(os, zapobj, tx));
877eda14cbcSMatt Macy }
878eda14cbcSMatt Macy 
879eda14cbcSMatt Macy void
880eda14cbcSMatt Macy zap_evict_sync(void *dbu)
881eda14cbcSMatt Macy {
882eda14cbcSMatt Macy 	zap_t *zap = dbu;
883eda14cbcSMatt Macy 
884eda14cbcSMatt Macy 	rw_destroy(&zap->zap_rwlock);
885eda14cbcSMatt Macy 
886eda14cbcSMatt Macy 	if (zap->zap_ismicro)
887eda14cbcSMatt Macy 		mze_destroy(zap);
888eda14cbcSMatt Macy 	else
889eda14cbcSMatt Macy 		mutex_destroy(&zap->zap_f.zap_num_entries_mtx);
890eda14cbcSMatt Macy 
891eda14cbcSMatt Macy 	kmem_free(zap, sizeof (zap_t));
892eda14cbcSMatt Macy }
893eda14cbcSMatt Macy 
894eda14cbcSMatt Macy int
895eda14cbcSMatt Macy zap_count(objset_t *os, uint64_t zapobj, uint64_t *count)
896eda14cbcSMatt Macy {
897eda14cbcSMatt Macy 	zap_t *zap;
898eda14cbcSMatt Macy 
899eda14cbcSMatt Macy 	int err =
900eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, NULL, RW_READER, TRUE, FALSE, FTAG, &zap);
901eda14cbcSMatt Macy 	if (err != 0)
902eda14cbcSMatt Macy 		return (err);
903eda14cbcSMatt Macy 	if (!zap->zap_ismicro) {
904eda14cbcSMatt Macy 		err = fzap_count(zap, count);
905eda14cbcSMatt Macy 	} else {
906eda14cbcSMatt Macy 		*count = zap->zap_m.zap_num_entries;
907eda14cbcSMatt Macy 	}
908eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
909eda14cbcSMatt Macy 	return (err);
910eda14cbcSMatt Macy }
911eda14cbcSMatt Macy 
912eda14cbcSMatt Macy /*
913eda14cbcSMatt Macy  * zn may be NULL; if not specified, it will be computed if needed.
914eda14cbcSMatt Macy  * See also the comment above zap_entry_normalization_conflict().
915eda14cbcSMatt Macy  */
916eda14cbcSMatt Macy static boolean_t
917eda14cbcSMatt Macy mzap_normalization_conflict(zap_t *zap, zap_name_t *zn, mzap_ent_t *mze)
918eda14cbcSMatt Macy {
919eda14cbcSMatt Macy 	int direction = AVL_BEFORE;
920eda14cbcSMatt Macy 	boolean_t allocdzn = B_FALSE;
921eda14cbcSMatt Macy 
922eda14cbcSMatt Macy 	if (zap->zap_normflags == 0)
923eda14cbcSMatt Macy 		return (B_FALSE);
924eda14cbcSMatt Macy 
925eda14cbcSMatt Macy again:
926eda14cbcSMatt Macy 	for (mzap_ent_t *other = avl_walk(&zap->zap_m.zap_avl, mze, direction);
927eda14cbcSMatt Macy 	    other && other->mze_hash == mze->mze_hash;
928eda14cbcSMatt Macy 	    other = avl_walk(&zap->zap_m.zap_avl, other, direction)) {
929eda14cbcSMatt Macy 
930eda14cbcSMatt Macy 		if (zn == NULL) {
931eda14cbcSMatt Macy 			zn = zap_name_alloc(zap, MZE_PHYS(zap, mze)->mze_name,
932eda14cbcSMatt Macy 			    MT_NORMALIZE);
933eda14cbcSMatt Macy 			allocdzn = B_TRUE;
934eda14cbcSMatt Macy 		}
935eda14cbcSMatt Macy 		if (zap_match(zn, MZE_PHYS(zap, other)->mze_name)) {
936eda14cbcSMatt Macy 			if (allocdzn)
937eda14cbcSMatt Macy 				zap_name_free(zn);
938eda14cbcSMatt Macy 			return (B_TRUE);
939eda14cbcSMatt Macy 		}
940eda14cbcSMatt Macy 	}
941eda14cbcSMatt Macy 
942eda14cbcSMatt Macy 	if (direction == AVL_BEFORE) {
943eda14cbcSMatt Macy 		direction = AVL_AFTER;
944eda14cbcSMatt Macy 		goto again;
945eda14cbcSMatt Macy 	}
946eda14cbcSMatt Macy 
947eda14cbcSMatt Macy 	if (allocdzn)
948eda14cbcSMatt Macy 		zap_name_free(zn);
949eda14cbcSMatt Macy 	return (B_FALSE);
950eda14cbcSMatt Macy }
951eda14cbcSMatt Macy 
952eda14cbcSMatt Macy /*
953eda14cbcSMatt Macy  * Routines for manipulating attributes.
954eda14cbcSMatt Macy  */
955eda14cbcSMatt Macy 
956eda14cbcSMatt Macy int
957eda14cbcSMatt Macy zap_lookup(objset_t *os, uint64_t zapobj, const char *name,
958eda14cbcSMatt Macy     uint64_t integer_size, uint64_t num_integers, void *buf)
959eda14cbcSMatt Macy {
960eda14cbcSMatt Macy 	return (zap_lookup_norm(os, zapobj, name, integer_size,
961eda14cbcSMatt Macy 	    num_integers, buf, 0, NULL, 0, NULL));
962eda14cbcSMatt Macy }
963eda14cbcSMatt Macy 
964eda14cbcSMatt Macy static int
965eda14cbcSMatt Macy zap_lookup_impl(zap_t *zap, const char *name,
966eda14cbcSMatt Macy     uint64_t integer_size, uint64_t num_integers, void *buf,
967eda14cbcSMatt Macy     matchtype_t mt, char *realname, int rn_len,
968eda14cbcSMatt Macy     boolean_t *ncp)
969eda14cbcSMatt Macy {
970eda14cbcSMatt Macy 	int err = 0;
971eda14cbcSMatt Macy 
972eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc(zap, name, mt);
973eda14cbcSMatt Macy 	if (zn == NULL)
974eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
975eda14cbcSMatt Macy 
976eda14cbcSMatt Macy 	if (!zap->zap_ismicro) {
977eda14cbcSMatt Macy 		err = fzap_lookup(zn, integer_size, num_integers, buf,
978eda14cbcSMatt Macy 		    realname, rn_len, ncp);
979eda14cbcSMatt Macy 	} else {
980eda14cbcSMatt Macy 		mzap_ent_t *mze = mze_find(zn);
981eda14cbcSMatt Macy 		if (mze == NULL) {
982eda14cbcSMatt Macy 			err = SET_ERROR(ENOENT);
983eda14cbcSMatt Macy 		} else {
984eda14cbcSMatt Macy 			if (num_integers < 1) {
985eda14cbcSMatt Macy 				err = SET_ERROR(EOVERFLOW);
986eda14cbcSMatt Macy 			} else if (integer_size != 8) {
987eda14cbcSMatt Macy 				err = SET_ERROR(EINVAL);
988eda14cbcSMatt Macy 			} else {
989eda14cbcSMatt Macy 				*(uint64_t *)buf =
990eda14cbcSMatt Macy 				    MZE_PHYS(zap, mze)->mze_value;
991eda14cbcSMatt Macy 				(void) strlcpy(realname,
992eda14cbcSMatt Macy 				    MZE_PHYS(zap, mze)->mze_name, rn_len);
993eda14cbcSMatt Macy 				if (ncp) {
994eda14cbcSMatt Macy 					*ncp = mzap_normalization_conflict(zap,
995eda14cbcSMatt Macy 					    zn, mze);
996eda14cbcSMatt Macy 				}
997eda14cbcSMatt Macy 			}
998eda14cbcSMatt Macy 		}
999eda14cbcSMatt Macy 	}
1000eda14cbcSMatt Macy 	zap_name_free(zn);
1001eda14cbcSMatt Macy 	return (err);
1002eda14cbcSMatt Macy }
1003eda14cbcSMatt Macy 
1004eda14cbcSMatt Macy int
1005eda14cbcSMatt Macy zap_lookup_norm(objset_t *os, uint64_t zapobj, const char *name,
1006eda14cbcSMatt Macy     uint64_t integer_size, uint64_t num_integers, void *buf,
1007eda14cbcSMatt Macy     matchtype_t mt, char *realname, int rn_len,
1008eda14cbcSMatt Macy     boolean_t *ncp)
1009eda14cbcSMatt Macy {
1010eda14cbcSMatt Macy 	zap_t *zap;
1011eda14cbcSMatt Macy 
1012eda14cbcSMatt Macy 	int err =
1013eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, NULL, RW_READER, TRUE, FALSE, FTAG, &zap);
1014eda14cbcSMatt Macy 	if (err != 0)
1015eda14cbcSMatt Macy 		return (err);
1016eda14cbcSMatt Macy 	err = zap_lookup_impl(zap, name, integer_size,
1017eda14cbcSMatt Macy 	    num_integers, buf, mt, realname, rn_len, ncp);
1018eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1019eda14cbcSMatt Macy 	return (err);
1020eda14cbcSMatt Macy }
1021eda14cbcSMatt Macy 
1022eda14cbcSMatt Macy int
1023eda14cbcSMatt Macy zap_prefetch(objset_t *os, uint64_t zapobj, const char *name)
1024eda14cbcSMatt Macy {
1025eda14cbcSMatt Macy 	zap_t *zap;
1026eda14cbcSMatt Macy 	int err;
1027eda14cbcSMatt Macy 	zap_name_t *zn;
1028eda14cbcSMatt Macy 
1029eda14cbcSMatt Macy 	err = zap_lockdir(os, zapobj, NULL, RW_READER, TRUE, FALSE, FTAG, &zap);
1030eda14cbcSMatt Macy 	if (err)
1031eda14cbcSMatt Macy 		return (err);
1032eda14cbcSMatt Macy 	zn = zap_name_alloc(zap, name, 0);
1033eda14cbcSMatt Macy 	if (zn == NULL) {
1034eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1035eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1036eda14cbcSMatt Macy 	}
1037eda14cbcSMatt Macy 
1038eda14cbcSMatt Macy 	fzap_prefetch(zn);
1039eda14cbcSMatt Macy 	zap_name_free(zn);
1040eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1041eda14cbcSMatt Macy 	return (err);
1042eda14cbcSMatt Macy }
1043eda14cbcSMatt Macy 
1044eda14cbcSMatt Macy int
1045eda14cbcSMatt Macy zap_lookup_by_dnode(dnode_t *dn, const char *name,
1046eda14cbcSMatt Macy     uint64_t integer_size, uint64_t num_integers, void *buf)
1047eda14cbcSMatt Macy {
1048eda14cbcSMatt Macy 	return (zap_lookup_norm_by_dnode(dn, name, integer_size,
1049eda14cbcSMatt Macy 	    num_integers, buf, 0, NULL, 0, NULL));
1050eda14cbcSMatt Macy }
1051eda14cbcSMatt Macy 
1052eda14cbcSMatt Macy int
1053eda14cbcSMatt Macy zap_lookup_norm_by_dnode(dnode_t *dn, const char *name,
1054eda14cbcSMatt Macy     uint64_t integer_size, uint64_t num_integers, void *buf,
1055eda14cbcSMatt Macy     matchtype_t mt, char *realname, int rn_len,
1056eda14cbcSMatt Macy     boolean_t *ncp)
1057eda14cbcSMatt Macy {
1058eda14cbcSMatt Macy 	zap_t *zap;
1059eda14cbcSMatt Macy 
1060eda14cbcSMatt Macy 	int err = zap_lockdir_by_dnode(dn, NULL, RW_READER, TRUE, FALSE,
1061eda14cbcSMatt Macy 	    FTAG, &zap);
1062eda14cbcSMatt Macy 	if (err != 0)
1063eda14cbcSMatt Macy 		return (err);
1064eda14cbcSMatt Macy 	err = zap_lookup_impl(zap, name, integer_size,
1065eda14cbcSMatt Macy 	    num_integers, buf, mt, realname, rn_len, ncp);
1066eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1067eda14cbcSMatt Macy 	return (err);
1068eda14cbcSMatt Macy }
1069eda14cbcSMatt Macy 
1070eda14cbcSMatt Macy int
1071eda14cbcSMatt Macy zap_prefetch_uint64(objset_t *os, uint64_t zapobj, const uint64_t *key,
1072eda14cbcSMatt Macy     int key_numints)
1073eda14cbcSMatt Macy {
1074eda14cbcSMatt Macy 	zap_t *zap;
1075eda14cbcSMatt Macy 
1076eda14cbcSMatt Macy 	int err =
1077eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, NULL, RW_READER, TRUE, FALSE, FTAG, &zap);
1078eda14cbcSMatt Macy 	if (err != 0)
1079eda14cbcSMatt Macy 		return (err);
1080eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc_uint64(zap, key, key_numints);
1081eda14cbcSMatt Macy 	if (zn == NULL) {
1082eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1083eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1084eda14cbcSMatt Macy 	}
1085eda14cbcSMatt Macy 
1086eda14cbcSMatt Macy 	fzap_prefetch(zn);
1087eda14cbcSMatt Macy 	zap_name_free(zn);
1088eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1089eda14cbcSMatt Macy 	return (err);
1090eda14cbcSMatt Macy }
1091eda14cbcSMatt Macy 
1092eda14cbcSMatt Macy int
1093eda14cbcSMatt Macy zap_lookup_uint64(objset_t *os, uint64_t zapobj, const uint64_t *key,
1094eda14cbcSMatt Macy     int key_numints, uint64_t integer_size, uint64_t num_integers, void *buf)
1095eda14cbcSMatt Macy {
1096eda14cbcSMatt Macy 	zap_t *zap;
1097eda14cbcSMatt Macy 
1098eda14cbcSMatt Macy 	int err =
1099eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, NULL, RW_READER, TRUE, FALSE, FTAG, &zap);
1100eda14cbcSMatt Macy 	if (err != 0)
1101eda14cbcSMatt Macy 		return (err);
1102eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc_uint64(zap, key, key_numints);
1103eda14cbcSMatt Macy 	if (zn == NULL) {
1104eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1105eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1106eda14cbcSMatt Macy 	}
1107eda14cbcSMatt Macy 
1108eda14cbcSMatt Macy 	err = fzap_lookup(zn, integer_size, num_integers, buf,
1109eda14cbcSMatt Macy 	    NULL, 0, NULL);
1110eda14cbcSMatt Macy 	zap_name_free(zn);
1111eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1112eda14cbcSMatt Macy 	return (err);
1113eda14cbcSMatt Macy }
1114eda14cbcSMatt Macy 
1115eda14cbcSMatt Macy int
1116eda14cbcSMatt Macy zap_contains(objset_t *os, uint64_t zapobj, const char *name)
1117eda14cbcSMatt Macy {
1118eda14cbcSMatt Macy 	int err = zap_lookup_norm(os, zapobj, name, 0,
1119eda14cbcSMatt Macy 	    0, NULL, 0, NULL, 0, NULL);
1120eda14cbcSMatt Macy 	if (err == EOVERFLOW || err == EINVAL)
1121eda14cbcSMatt Macy 		err = 0; /* found, but skipped reading the value */
1122eda14cbcSMatt Macy 	return (err);
1123eda14cbcSMatt Macy }
1124eda14cbcSMatt Macy 
1125eda14cbcSMatt Macy int
1126eda14cbcSMatt Macy zap_length(objset_t *os, uint64_t zapobj, const char *name,
1127eda14cbcSMatt Macy     uint64_t *integer_size, uint64_t *num_integers)
1128eda14cbcSMatt Macy {
1129eda14cbcSMatt Macy 	zap_t *zap;
1130eda14cbcSMatt Macy 
1131eda14cbcSMatt Macy 	int err =
1132eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, NULL, RW_READER, TRUE, FALSE, FTAG, &zap);
1133eda14cbcSMatt Macy 	if (err != 0)
1134eda14cbcSMatt Macy 		return (err);
1135eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc(zap, name, 0);
1136eda14cbcSMatt Macy 	if (zn == NULL) {
1137eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1138eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1139eda14cbcSMatt Macy 	}
1140eda14cbcSMatt Macy 	if (!zap->zap_ismicro) {
1141eda14cbcSMatt Macy 		err = fzap_length(zn, integer_size, num_integers);
1142eda14cbcSMatt Macy 	} else {
1143eda14cbcSMatt Macy 		mzap_ent_t *mze = mze_find(zn);
1144eda14cbcSMatt Macy 		if (mze == NULL) {
1145eda14cbcSMatt Macy 			err = SET_ERROR(ENOENT);
1146eda14cbcSMatt Macy 		} else {
1147eda14cbcSMatt Macy 			if (integer_size)
1148eda14cbcSMatt Macy 				*integer_size = 8;
1149eda14cbcSMatt Macy 			if (num_integers)
1150eda14cbcSMatt Macy 				*num_integers = 1;
1151eda14cbcSMatt Macy 		}
1152eda14cbcSMatt Macy 	}
1153eda14cbcSMatt Macy 	zap_name_free(zn);
1154eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1155eda14cbcSMatt Macy 	return (err);
1156eda14cbcSMatt Macy }
1157eda14cbcSMatt Macy 
1158eda14cbcSMatt Macy int
1159eda14cbcSMatt Macy zap_length_uint64(objset_t *os, uint64_t zapobj, const uint64_t *key,
1160eda14cbcSMatt Macy     int key_numints, uint64_t *integer_size, uint64_t *num_integers)
1161eda14cbcSMatt Macy {
1162eda14cbcSMatt Macy 	zap_t *zap;
1163eda14cbcSMatt Macy 
1164eda14cbcSMatt Macy 	int err =
1165eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, NULL, RW_READER, TRUE, FALSE, FTAG, &zap);
1166eda14cbcSMatt Macy 	if (err != 0)
1167eda14cbcSMatt Macy 		return (err);
1168eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc_uint64(zap, key, key_numints);
1169eda14cbcSMatt Macy 	if (zn == NULL) {
1170eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1171eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1172eda14cbcSMatt Macy 	}
1173eda14cbcSMatt Macy 	err = fzap_length(zn, integer_size, num_integers);
1174eda14cbcSMatt Macy 	zap_name_free(zn);
1175eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1176eda14cbcSMatt Macy 	return (err);
1177eda14cbcSMatt Macy }
1178eda14cbcSMatt Macy 
1179eda14cbcSMatt Macy static void
1180eda14cbcSMatt Macy mzap_addent(zap_name_t *zn, uint64_t value)
1181eda14cbcSMatt Macy {
1182eda14cbcSMatt Macy 	zap_t *zap = zn->zn_zap;
1183eda14cbcSMatt Macy 	int start = zap->zap_m.zap_alloc_next;
1184eda14cbcSMatt Macy 
1185eda14cbcSMatt Macy 	ASSERT(RW_WRITE_HELD(&zap->zap_rwlock));
1186eda14cbcSMatt Macy 
1187eda14cbcSMatt Macy #ifdef ZFS_DEBUG
1188eda14cbcSMatt Macy 	for (int i = 0; i < zap->zap_m.zap_num_chunks; i++) {
1189eda14cbcSMatt Macy 		mzap_ent_phys_t *mze = &zap_m_phys(zap)->mz_chunk[i];
1190eda14cbcSMatt Macy 		ASSERT(strcmp(zn->zn_key_orig, mze->mze_name) != 0);
1191eda14cbcSMatt Macy 	}
1192eda14cbcSMatt Macy #endif
1193eda14cbcSMatt Macy 
1194eda14cbcSMatt Macy 	uint32_t cd = mze_find_unused_cd(zap, zn->zn_hash);
1195eda14cbcSMatt Macy 	/* given the limited size of the microzap, this can't happen */
1196eda14cbcSMatt Macy 	ASSERT(cd < zap_maxcd(zap));
1197eda14cbcSMatt Macy 
1198eda14cbcSMatt Macy again:
1199eda14cbcSMatt Macy 	for (int i = start; i < zap->zap_m.zap_num_chunks; i++) {
1200eda14cbcSMatt Macy 		mzap_ent_phys_t *mze = &zap_m_phys(zap)->mz_chunk[i];
1201eda14cbcSMatt Macy 		if (mze->mze_name[0] == 0) {
1202eda14cbcSMatt Macy 			mze->mze_value = value;
1203eda14cbcSMatt Macy 			mze->mze_cd = cd;
1204eda14cbcSMatt Macy 			(void) strlcpy(mze->mze_name, zn->zn_key_orig,
1205eda14cbcSMatt Macy 			    sizeof (mze->mze_name));
1206eda14cbcSMatt Macy 			zap->zap_m.zap_num_entries++;
1207eda14cbcSMatt Macy 			zap->zap_m.zap_alloc_next = i+1;
1208eda14cbcSMatt Macy 			if (zap->zap_m.zap_alloc_next ==
1209eda14cbcSMatt Macy 			    zap->zap_m.zap_num_chunks)
1210eda14cbcSMatt Macy 				zap->zap_m.zap_alloc_next = 0;
1211eda14cbcSMatt Macy 			mze_insert(zap, i, zn->zn_hash);
1212eda14cbcSMatt Macy 			return;
1213eda14cbcSMatt Macy 		}
1214eda14cbcSMatt Macy 	}
1215eda14cbcSMatt Macy 	if (start != 0) {
1216eda14cbcSMatt Macy 		start = 0;
1217eda14cbcSMatt Macy 		goto again;
1218eda14cbcSMatt Macy 	}
1219eda14cbcSMatt Macy 	cmn_err(CE_PANIC, "out of entries!");
1220eda14cbcSMatt Macy }
1221eda14cbcSMatt Macy 
1222eda14cbcSMatt Macy static int
1223eda14cbcSMatt Macy zap_add_impl(zap_t *zap, const char *key,
1224eda14cbcSMatt Macy     int integer_size, uint64_t num_integers,
1225eda14cbcSMatt Macy     const void *val, dmu_tx_t *tx, void *tag)
1226eda14cbcSMatt Macy {
1227eda14cbcSMatt Macy 	const uint64_t *intval = val;
1228eda14cbcSMatt Macy 	int err = 0;
1229eda14cbcSMatt Macy 
1230eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc(zap, key, 0);
1231eda14cbcSMatt Macy 	if (zn == NULL) {
1232eda14cbcSMatt Macy 		zap_unlockdir(zap, tag);
1233eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1234eda14cbcSMatt Macy 	}
1235eda14cbcSMatt Macy 	if (!zap->zap_ismicro) {
1236eda14cbcSMatt Macy 		err = fzap_add(zn, integer_size, num_integers, val, tag, tx);
1237eda14cbcSMatt Macy 		zap = zn->zn_zap;	/* fzap_add() may change zap */
1238eda14cbcSMatt Macy 	} else if (integer_size != 8 || num_integers != 1 ||
1239eda14cbcSMatt Macy 	    strlen(key) >= MZAP_NAME_LEN ||
1240eda14cbcSMatt Macy 	    !mze_canfit_fzap_leaf(zn, zn->zn_hash)) {
1241eda14cbcSMatt Macy 		err = mzap_upgrade(&zn->zn_zap, tag, tx, 0);
1242eda14cbcSMatt Macy 		if (err == 0) {
1243eda14cbcSMatt Macy 			err = fzap_add(zn, integer_size, num_integers, val,
1244eda14cbcSMatt Macy 			    tag, tx);
1245eda14cbcSMatt Macy 		}
1246eda14cbcSMatt Macy 		zap = zn->zn_zap;	/* fzap_add() may change zap */
1247eda14cbcSMatt Macy 	} else {
1248eda14cbcSMatt Macy 		if (mze_find(zn) != NULL) {
1249eda14cbcSMatt Macy 			err = SET_ERROR(EEXIST);
1250eda14cbcSMatt Macy 		} else {
1251eda14cbcSMatt Macy 			mzap_addent(zn, *intval);
1252eda14cbcSMatt Macy 		}
1253eda14cbcSMatt Macy 	}
1254eda14cbcSMatt Macy 	ASSERT(zap == zn->zn_zap);
1255eda14cbcSMatt Macy 	zap_name_free(zn);
1256eda14cbcSMatt Macy 	if (zap != NULL)	/* may be NULL if fzap_add() failed */
1257eda14cbcSMatt Macy 		zap_unlockdir(zap, tag);
1258eda14cbcSMatt Macy 	return (err);
1259eda14cbcSMatt Macy }
1260eda14cbcSMatt Macy 
1261eda14cbcSMatt Macy int
1262eda14cbcSMatt Macy zap_add(objset_t *os, uint64_t zapobj, const char *key,
1263eda14cbcSMatt Macy     int integer_size, uint64_t num_integers,
1264eda14cbcSMatt Macy     const void *val, dmu_tx_t *tx)
1265eda14cbcSMatt Macy {
1266eda14cbcSMatt Macy 	zap_t *zap;
1267eda14cbcSMatt Macy 	int err;
1268eda14cbcSMatt Macy 
1269eda14cbcSMatt Macy 	err = zap_lockdir(os, zapobj, tx, RW_WRITER, TRUE, TRUE, FTAG, &zap);
1270eda14cbcSMatt Macy 	if (err != 0)
1271eda14cbcSMatt Macy 		return (err);
1272eda14cbcSMatt Macy 	err = zap_add_impl(zap, key, integer_size, num_integers, val, tx, FTAG);
1273eda14cbcSMatt Macy 	/* zap_add_impl() calls zap_unlockdir() */
1274eda14cbcSMatt Macy 	return (err);
1275eda14cbcSMatt Macy }
1276eda14cbcSMatt Macy 
1277eda14cbcSMatt Macy int
1278eda14cbcSMatt Macy zap_add_by_dnode(dnode_t *dn, const char *key,
1279eda14cbcSMatt Macy     int integer_size, uint64_t num_integers,
1280eda14cbcSMatt Macy     const void *val, dmu_tx_t *tx)
1281eda14cbcSMatt Macy {
1282eda14cbcSMatt Macy 	zap_t *zap;
1283eda14cbcSMatt Macy 	int err;
1284eda14cbcSMatt Macy 
1285eda14cbcSMatt Macy 	err = zap_lockdir_by_dnode(dn, tx, RW_WRITER, TRUE, TRUE, FTAG, &zap);
1286eda14cbcSMatt Macy 	if (err != 0)
1287eda14cbcSMatt Macy 		return (err);
1288eda14cbcSMatt Macy 	err = zap_add_impl(zap, key, integer_size, num_integers, val, tx, FTAG);
1289eda14cbcSMatt Macy 	/* zap_add_impl() calls zap_unlockdir() */
1290eda14cbcSMatt Macy 	return (err);
1291eda14cbcSMatt Macy }
1292eda14cbcSMatt Macy 
1293eda14cbcSMatt Macy int
1294eda14cbcSMatt Macy zap_add_uint64(objset_t *os, uint64_t zapobj, const uint64_t *key,
1295eda14cbcSMatt Macy     int key_numints, int integer_size, uint64_t num_integers,
1296eda14cbcSMatt Macy     const void *val, dmu_tx_t *tx)
1297eda14cbcSMatt Macy {
1298eda14cbcSMatt Macy 	zap_t *zap;
1299eda14cbcSMatt Macy 
1300eda14cbcSMatt Macy 	int err =
1301eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, tx, RW_WRITER, TRUE, TRUE, FTAG, &zap);
1302eda14cbcSMatt Macy 	if (err != 0)
1303eda14cbcSMatt Macy 		return (err);
1304eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc_uint64(zap, key, key_numints);
1305eda14cbcSMatt Macy 	if (zn == NULL) {
1306eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1307eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1308eda14cbcSMatt Macy 	}
1309eda14cbcSMatt Macy 	err = fzap_add(zn, integer_size, num_integers, val, FTAG, tx);
1310eda14cbcSMatt Macy 	zap = zn->zn_zap;	/* fzap_add() may change zap */
1311eda14cbcSMatt Macy 	zap_name_free(zn);
1312eda14cbcSMatt Macy 	if (zap != NULL)	/* may be NULL if fzap_add() failed */
1313eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1314eda14cbcSMatt Macy 	return (err);
1315eda14cbcSMatt Macy }
1316eda14cbcSMatt Macy 
1317eda14cbcSMatt Macy int
1318eda14cbcSMatt Macy zap_update(objset_t *os, uint64_t zapobj, const char *name,
1319eda14cbcSMatt Macy     int integer_size, uint64_t num_integers, const void *val, dmu_tx_t *tx)
1320eda14cbcSMatt Macy {
1321eda14cbcSMatt Macy 	zap_t *zap;
1322eda14cbcSMatt Macy 	const uint64_t *intval = val;
1323eda14cbcSMatt Macy 
1324eda14cbcSMatt Macy 	int err =
1325eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, tx, RW_WRITER, TRUE, TRUE, FTAG, &zap);
1326eda14cbcSMatt Macy 	if (err != 0)
1327eda14cbcSMatt Macy 		return (err);
1328eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc(zap, name, 0);
1329eda14cbcSMatt Macy 	if (zn == NULL) {
1330eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1331eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1332eda14cbcSMatt Macy 	}
1333eda14cbcSMatt Macy 	if (!zap->zap_ismicro) {
1334eda14cbcSMatt Macy 		err = fzap_update(zn, integer_size, num_integers, val,
1335eda14cbcSMatt Macy 		    FTAG, tx);
1336eda14cbcSMatt Macy 		zap = zn->zn_zap;	/* fzap_update() may change zap */
1337eda14cbcSMatt Macy 	} else if (integer_size != 8 || num_integers != 1 ||
1338eda14cbcSMatt Macy 	    strlen(name) >= MZAP_NAME_LEN) {
1339eda14cbcSMatt Macy 		dprintf("upgrading obj %llu: intsz=%u numint=%llu name=%s\n",
134033b8c039SMartin Matuska 		    (u_longlong_t)zapobj, integer_size,
134133b8c039SMartin Matuska 		    (u_longlong_t)num_integers, name);
1342eda14cbcSMatt Macy 		err = mzap_upgrade(&zn->zn_zap, FTAG, tx, 0);
1343eda14cbcSMatt Macy 		if (err == 0) {
1344eda14cbcSMatt Macy 			err = fzap_update(zn, integer_size, num_integers,
1345eda14cbcSMatt Macy 			    val, FTAG, tx);
1346eda14cbcSMatt Macy 		}
1347eda14cbcSMatt Macy 		zap = zn->zn_zap;	/* fzap_update() may change zap */
1348eda14cbcSMatt Macy 	} else {
1349eda14cbcSMatt Macy 		mzap_ent_t *mze = mze_find(zn);
1350eda14cbcSMatt Macy 		if (mze != NULL) {
1351eda14cbcSMatt Macy 			MZE_PHYS(zap, mze)->mze_value = *intval;
1352eda14cbcSMatt Macy 		} else {
1353eda14cbcSMatt Macy 			mzap_addent(zn, *intval);
1354eda14cbcSMatt Macy 		}
1355eda14cbcSMatt Macy 	}
1356eda14cbcSMatt Macy 	ASSERT(zap == zn->zn_zap);
1357eda14cbcSMatt Macy 	zap_name_free(zn);
1358eda14cbcSMatt Macy 	if (zap != NULL)	/* may be NULL if fzap_upgrade() failed */
1359eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1360eda14cbcSMatt Macy 	return (err);
1361eda14cbcSMatt Macy }
1362eda14cbcSMatt Macy 
1363eda14cbcSMatt Macy int
1364eda14cbcSMatt Macy zap_update_uint64(objset_t *os, uint64_t zapobj, const uint64_t *key,
1365eda14cbcSMatt Macy     int key_numints,
1366eda14cbcSMatt Macy     int integer_size, uint64_t num_integers, const void *val, dmu_tx_t *tx)
1367eda14cbcSMatt Macy {
1368eda14cbcSMatt Macy 	zap_t *zap;
1369eda14cbcSMatt Macy 
1370eda14cbcSMatt Macy 	int err =
1371eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, tx, RW_WRITER, TRUE, TRUE, FTAG, &zap);
1372eda14cbcSMatt Macy 	if (err != 0)
1373eda14cbcSMatt Macy 		return (err);
1374eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc_uint64(zap, key, key_numints);
1375eda14cbcSMatt Macy 	if (zn == NULL) {
1376eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1377eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1378eda14cbcSMatt Macy 	}
1379eda14cbcSMatt Macy 	err = fzap_update(zn, integer_size, num_integers, val, FTAG, tx);
1380eda14cbcSMatt Macy 	zap = zn->zn_zap;	/* fzap_update() may change zap */
1381eda14cbcSMatt Macy 	zap_name_free(zn);
1382eda14cbcSMatt Macy 	if (zap != NULL)	/* may be NULL if fzap_upgrade() failed */
1383eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1384eda14cbcSMatt Macy 	return (err);
1385eda14cbcSMatt Macy }
1386eda14cbcSMatt Macy 
1387eda14cbcSMatt Macy int
1388eda14cbcSMatt Macy zap_remove(objset_t *os, uint64_t zapobj, const char *name, dmu_tx_t *tx)
1389eda14cbcSMatt Macy {
1390eda14cbcSMatt Macy 	return (zap_remove_norm(os, zapobj, name, 0, tx));
1391eda14cbcSMatt Macy }
1392eda14cbcSMatt Macy 
1393eda14cbcSMatt Macy static int
1394eda14cbcSMatt Macy zap_remove_impl(zap_t *zap, const char *name,
1395eda14cbcSMatt Macy     matchtype_t mt, dmu_tx_t *tx)
1396eda14cbcSMatt Macy {
1397eda14cbcSMatt Macy 	int err = 0;
1398eda14cbcSMatt Macy 
1399eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc(zap, name, mt);
1400eda14cbcSMatt Macy 	if (zn == NULL)
1401eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1402eda14cbcSMatt Macy 	if (!zap->zap_ismicro) {
1403eda14cbcSMatt Macy 		err = fzap_remove(zn, tx);
1404eda14cbcSMatt Macy 	} else {
1405eda14cbcSMatt Macy 		mzap_ent_t *mze = mze_find(zn);
1406eda14cbcSMatt Macy 		if (mze == NULL) {
1407eda14cbcSMatt Macy 			err = SET_ERROR(ENOENT);
1408eda14cbcSMatt Macy 		} else {
1409eda14cbcSMatt Macy 			zap->zap_m.zap_num_entries--;
1410*da5137abSMartin Matuska 			memset(&zap_m_phys(zap)->mz_chunk[mze->mze_chunkid], 0,
1411eda14cbcSMatt Macy 			    sizeof (mzap_ent_phys_t));
1412eda14cbcSMatt Macy 			mze_remove(zap, mze);
1413eda14cbcSMatt Macy 		}
1414eda14cbcSMatt Macy 	}
1415eda14cbcSMatt Macy 	zap_name_free(zn);
1416eda14cbcSMatt Macy 	return (err);
1417eda14cbcSMatt Macy }
1418eda14cbcSMatt Macy 
1419eda14cbcSMatt Macy int
1420eda14cbcSMatt Macy zap_remove_norm(objset_t *os, uint64_t zapobj, const char *name,
1421eda14cbcSMatt Macy     matchtype_t mt, dmu_tx_t *tx)
1422eda14cbcSMatt Macy {
1423eda14cbcSMatt Macy 	zap_t *zap;
1424eda14cbcSMatt Macy 	int err;
1425eda14cbcSMatt Macy 
1426eda14cbcSMatt Macy 	err = zap_lockdir(os, zapobj, tx, RW_WRITER, TRUE, FALSE, FTAG, &zap);
1427eda14cbcSMatt Macy 	if (err)
1428eda14cbcSMatt Macy 		return (err);
1429eda14cbcSMatt Macy 	err = zap_remove_impl(zap, name, mt, tx);
1430eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1431eda14cbcSMatt Macy 	return (err);
1432eda14cbcSMatt Macy }
1433eda14cbcSMatt Macy 
1434eda14cbcSMatt Macy int
1435eda14cbcSMatt Macy zap_remove_by_dnode(dnode_t *dn, const char *name, dmu_tx_t *tx)
1436eda14cbcSMatt Macy {
1437eda14cbcSMatt Macy 	zap_t *zap;
1438eda14cbcSMatt Macy 	int err;
1439eda14cbcSMatt Macy 
1440eda14cbcSMatt Macy 	err = zap_lockdir_by_dnode(dn, tx, RW_WRITER, TRUE, FALSE, FTAG, &zap);
1441eda14cbcSMatt Macy 	if (err)
1442eda14cbcSMatt Macy 		return (err);
1443eda14cbcSMatt Macy 	err = zap_remove_impl(zap, name, 0, tx);
1444eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1445eda14cbcSMatt Macy 	return (err);
1446eda14cbcSMatt Macy }
1447eda14cbcSMatt Macy 
1448eda14cbcSMatt Macy int
1449eda14cbcSMatt Macy zap_remove_uint64(objset_t *os, uint64_t zapobj, const uint64_t *key,
1450eda14cbcSMatt Macy     int key_numints, dmu_tx_t *tx)
1451eda14cbcSMatt Macy {
1452eda14cbcSMatt Macy 	zap_t *zap;
1453eda14cbcSMatt Macy 
1454eda14cbcSMatt Macy 	int err =
1455eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, tx, RW_WRITER, TRUE, FALSE, FTAG, &zap);
1456eda14cbcSMatt Macy 	if (err != 0)
1457eda14cbcSMatt Macy 		return (err);
1458eda14cbcSMatt Macy 	zap_name_t *zn = zap_name_alloc_uint64(zap, key, key_numints);
1459eda14cbcSMatt Macy 	if (zn == NULL) {
1460eda14cbcSMatt Macy 		zap_unlockdir(zap, FTAG);
1461eda14cbcSMatt Macy 		return (SET_ERROR(ENOTSUP));
1462eda14cbcSMatt Macy 	}
1463eda14cbcSMatt Macy 	err = fzap_remove(zn, tx);
1464eda14cbcSMatt Macy 	zap_name_free(zn);
1465eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1466eda14cbcSMatt Macy 	return (err);
1467eda14cbcSMatt Macy }
1468eda14cbcSMatt Macy 
1469eda14cbcSMatt Macy /*
1470eda14cbcSMatt Macy  * Routines for iterating over the attributes.
1471eda14cbcSMatt Macy  */
1472eda14cbcSMatt Macy 
1473eda14cbcSMatt Macy static void
1474eda14cbcSMatt Macy zap_cursor_init_impl(zap_cursor_t *zc, objset_t *os, uint64_t zapobj,
1475eda14cbcSMatt Macy     uint64_t serialized, boolean_t prefetch)
1476eda14cbcSMatt Macy {
1477eda14cbcSMatt Macy 	zc->zc_objset = os;
1478eda14cbcSMatt Macy 	zc->zc_zap = NULL;
1479eda14cbcSMatt Macy 	zc->zc_leaf = NULL;
1480eda14cbcSMatt Macy 	zc->zc_zapobj = zapobj;
1481eda14cbcSMatt Macy 	zc->zc_serialized = serialized;
1482eda14cbcSMatt Macy 	zc->zc_hash = 0;
1483eda14cbcSMatt Macy 	zc->zc_cd = 0;
1484eda14cbcSMatt Macy 	zc->zc_prefetch = prefetch;
1485eda14cbcSMatt Macy }
1486eda14cbcSMatt Macy void
1487eda14cbcSMatt Macy zap_cursor_init_serialized(zap_cursor_t *zc, objset_t *os, uint64_t zapobj,
1488eda14cbcSMatt Macy     uint64_t serialized)
1489eda14cbcSMatt Macy {
1490eda14cbcSMatt Macy 	zap_cursor_init_impl(zc, os, zapobj, serialized, B_TRUE);
1491eda14cbcSMatt Macy }
1492eda14cbcSMatt Macy 
1493eda14cbcSMatt Macy /*
1494eda14cbcSMatt Macy  * Initialize a cursor at the beginning of the ZAP object.  The entire
1495eda14cbcSMatt Macy  * ZAP object will be prefetched.
1496eda14cbcSMatt Macy  */
1497eda14cbcSMatt Macy void
1498eda14cbcSMatt Macy zap_cursor_init(zap_cursor_t *zc, objset_t *os, uint64_t zapobj)
1499eda14cbcSMatt Macy {
1500eda14cbcSMatt Macy 	zap_cursor_init_impl(zc, os, zapobj, 0, B_TRUE);
1501eda14cbcSMatt Macy }
1502eda14cbcSMatt Macy 
1503eda14cbcSMatt Macy /*
1504eda14cbcSMatt Macy  * Initialize a cursor at the beginning, but request that we not prefetch
1505eda14cbcSMatt Macy  * the entire ZAP object.
1506eda14cbcSMatt Macy  */
1507eda14cbcSMatt Macy void
1508eda14cbcSMatt Macy zap_cursor_init_noprefetch(zap_cursor_t *zc, objset_t *os, uint64_t zapobj)
1509eda14cbcSMatt Macy {
1510eda14cbcSMatt Macy 	zap_cursor_init_impl(zc, os, zapobj, 0, B_FALSE);
1511eda14cbcSMatt Macy }
1512eda14cbcSMatt Macy 
1513eda14cbcSMatt Macy void
1514eda14cbcSMatt Macy zap_cursor_fini(zap_cursor_t *zc)
1515eda14cbcSMatt Macy {
1516eda14cbcSMatt Macy 	if (zc->zc_zap) {
1517eda14cbcSMatt Macy 		rw_enter(&zc->zc_zap->zap_rwlock, RW_READER);
1518eda14cbcSMatt Macy 		zap_unlockdir(zc->zc_zap, NULL);
1519eda14cbcSMatt Macy 		zc->zc_zap = NULL;
1520eda14cbcSMatt Macy 	}
1521eda14cbcSMatt Macy 	if (zc->zc_leaf) {
1522eda14cbcSMatt Macy 		rw_enter(&zc->zc_leaf->l_rwlock, RW_READER);
1523eda14cbcSMatt Macy 		zap_put_leaf(zc->zc_leaf);
1524eda14cbcSMatt Macy 		zc->zc_leaf = NULL;
1525eda14cbcSMatt Macy 	}
1526eda14cbcSMatt Macy 	zc->zc_objset = NULL;
1527eda14cbcSMatt Macy }
1528eda14cbcSMatt Macy 
1529eda14cbcSMatt Macy uint64_t
1530eda14cbcSMatt Macy zap_cursor_serialize(zap_cursor_t *zc)
1531eda14cbcSMatt Macy {
1532eda14cbcSMatt Macy 	if (zc->zc_hash == -1ULL)
1533eda14cbcSMatt Macy 		return (-1ULL);
1534eda14cbcSMatt Macy 	if (zc->zc_zap == NULL)
1535eda14cbcSMatt Macy 		return (zc->zc_serialized);
1536eda14cbcSMatt Macy 	ASSERT((zc->zc_hash & zap_maxcd(zc->zc_zap)) == 0);
1537eda14cbcSMatt Macy 	ASSERT(zc->zc_cd < zap_maxcd(zc->zc_zap));
1538eda14cbcSMatt Macy 
1539eda14cbcSMatt Macy 	/*
1540eda14cbcSMatt Macy 	 * We want to keep the high 32 bits of the cursor zero if we can, so
1541eda14cbcSMatt Macy 	 * that 32-bit programs can access this.  So usually use a small
1542eda14cbcSMatt Macy 	 * (28-bit) hash value so we can fit 4 bits of cd into the low 32-bits
1543eda14cbcSMatt Macy 	 * of the cursor.
1544eda14cbcSMatt Macy 	 *
1545eda14cbcSMatt Macy 	 * [ collision differentiator | zap_hashbits()-bit hash value ]
1546eda14cbcSMatt Macy 	 */
1547eda14cbcSMatt Macy 	return ((zc->zc_hash >> (64 - zap_hashbits(zc->zc_zap))) |
1548eda14cbcSMatt Macy 	    ((uint64_t)zc->zc_cd << zap_hashbits(zc->zc_zap)));
1549eda14cbcSMatt Macy }
1550eda14cbcSMatt Macy 
1551eda14cbcSMatt Macy int
1552eda14cbcSMatt Macy zap_cursor_retrieve(zap_cursor_t *zc, zap_attribute_t *za)
1553eda14cbcSMatt Macy {
1554eda14cbcSMatt Macy 	int err;
1555eda14cbcSMatt Macy 
1556eda14cbcSMatt Macy 	if (zc->zc_hash == -1ULL)
1557eda14cbcSMatt Macy 		return (SET_ERROR(ENOENT));
1558eda14cbcSMatt Macy 
1559eda14cbcSMatt Macy 	if (zc->zc_zap == NULL) {
1560eda14cbcSMatt Macy 		int hb;
1561eda14cbcSMatt Macy 		err = zap_lockdir(zc->zc_objset, zc->zc_zapobj, NULL,
1562eda14cbcSMatt Macy 		    RW_READER, TRUE, FALSE, NULL, &zc->zc_zap);
1563eda14cbcSMatt Macy 		if (err != 0)
1564eda14cbcSMatt Macy 			return (err);
1565eda14cbcSMatt Macy 
1566eda14cbcSMatt Macy 		/*
1567eda14cbcSMatt Macy 		 * To support zap_cursor_init_serialized, advance, retrieve,
1568eda14cbcSMatt Macy 		 * we must add to the existing zc_cd, which may already
1569eda14cbcSMatt Macy 		 * be 1 due to the zap_cursor_advance.
1570eda14cbcSMatt Macy 		 */
1571eda14cbcSMatt Macy 		ASSERT(zc->zc_hash == 0);
1572eda14cbcSMatt Macy 		hb = zap_hashbits(zc->zc_zap);
1573eda14cbcSMatt Macy 		zc->zc_hash = zc->zc_serialized << (64 - hb);
1574eda14cbcSMatt Macy 		zc->zc_cd += zc->zc_serialized >> hb;
1575eda14cbcSMatt Macy 		if (zc->zc_cd >= zap_maxcd(zc->zc_zap)) /* corrupt serialized */
1576eda14cbcSMatt Macy 			zc->zc_cd = 0;
1577eda14cbcSMatt Macy 	} else {
1578eda14cbcSMatt Macy 		rw_enter(&zc->zc_zap->zap_rwlock, RW_READER);
1579eda14cbcSMatt Macy 	}
1580eda14cbcSMatt Macy 	if (!zc->zc_zap->zap_ismicro) {
1581eda14cbcSMatt Macy 		err = fzap_cursor_retrieve(zc->zc_zap, zc, za);
1582eda14cbcSMatt Macy 	} else {
1583eda14cbcSMatt Macy 		avl_index_t idx;
1584eda14cbcSMatt Macy 		mzap_ent_t mze_tofind;
1585eda14cbcSMatt Macy 
1586eda14cbcSMatt Macy 		mze_tofind.mze_hash = zc->zc_hash;
1587eda14cbcSMatt Macy 		mze_tofind.mze_cd = zc->zc_cd;
1588eda14cbcSMatt Macy 
1589eda14cbcSMatt Macy 		mzap_ent_t *mze =
1590eda14cbcSMatt Macy 		    avl_find(&zc->zc_zap->zap_m.zap_avl, &mze_tofind, &idx);
1591eda14cbcSMatt Macy 		if (mze == NULL) {
1592eda14cbcSMatt Macy 			mze = avl_nearest(&zc->zc_zap->zap_m.zap_avl,
1593eda14cbcSMatt Macy 			    idx, AVL_AFTER);
1594eda14cbcSMatt Macy 		}
1595eda14cbcSMatt Macy 		if (mze) {
1596eda14cbcSMatt Macy 			mzap_ent_phys_t *mzep = MZE_PHYS(zc->zc_zap, mze);
1597eda14cbcSMatt Macy 			ASSERT3U(mze->mze_cd, ==, mzep->mze_cd);
1598eda14cbcSMatt Macy 			za->za_normalization_conflict =
1599eda14cbcSMatt Macy 			    mzap_normalization_conflict(zc->zc_zap, NULL, mze);
1600eda14cbcSMatt Macy 			za->za_integer_length = 8;
1601eda14cbcSMatt Macy 			za->za_num_integers = 1;
1602eda14cbcSMatt Macy 			za->za_first_integer = mzep->mze_value;
1603eda14cbcSMatt Macy 			(void) strlcpy(za->za_name, mzep->mze_name,
1604eda14cbcSMatt Macy 			    sizeof (za->za_name));
1605eda14cbcSMatt Macy 			zc->zc_hash = mze->mze_hash;
1606eda14cbcSMatt Macy 			zc->zc_cd = mze->mze_cd;
1607eda14cbcSMatt Macy 			err = 0;
1608eda14cbcSMatt Macy 		} else {
1609eda14cbcSMatt Macy 			zc->zc_hash = -1ULL;
1610eda14cbcSMatt Macy 			err = SET_ERROR(ENOENT);
1611eda14cbcSMatt Macy 		}
1612eda14cbcSMatt Macy 	}
1613eda14cbcSMatt Macy 	rw_exit(&zc->zc_zap->zap_rwlock);
1614eda14cbcSMatt Macy 	return (err);
1615eda14cbcSMatt Macy }
1616eda14cbcSMatt Macy 
1617eda14cbcSMatt Macy void
1618eda14cbcSMatt Macy zap_cursor_advance(zap_cursor_t *zc)
1619eda14cbcSMatt Macy {
1620eda14cbcSMatt Macy 	if (zc->zc_hash == -1ULL)
1621eda14cbcSMatt Macy 		return;
1622eda14cbcSMatt Macy 	zc->zc_cd++;
1623eda14cbcSMatt Macy }
1624eda14cbcSMatt Macy 
1625eda14cbcSMatt Macy int
1626eda14cbcSMatt Macy zap_get_stats(objset_t *os, uint64_t zapobj, zap_stats_t *zs)
1627eda14cbcSMatt Macy {
1628eda14cbcSMatt Macy 	zap_t *zap;
1629eda14cbcSMatt Macy 
1630eda14cbcSMatt Macy 	int err =
1631eda14cbcSMatt Macy 	    zap_lockdir(os, zapobj, NULL, RW_READER, TRUE, FALSE, FTAG, &zap);
1632eda14cbcSMatt Macy 	if (err != 0)
1633eda14cbcSMatt Macy 		return (err);
1634eda14cbcSMatt Macy 
1635*da5137abSMartin Matuska 	memset(zs, 0, sizeof (zap_stats_t));
1636eda14cbcSMatt Macy 
1637eda14cbcSMatt Macy 	if (zap->zap_ismicro) {
1638eda14cbcSMatt Macy 		zs->zs_blocksize = zap->zap_dbuf->db_size;
1639eda14cbcSMatt Macy 		zs->zs_num_entries = zap->zap_m.zap_num_entries;
1640eda14cbcSMatt Macy 		zs->zs_num_blocks = 1;
1641eda14cbcSMatt Macy 	} else {
1642eda14cbcSMatt Macy 		fzap_get_stats(zap, zs);
1643eda14cbcSMatt Macy 	}
1644eda14cbcSMatt Macy 	zap_unlockdir(zap, FTAG);
1645eda14cbcSMatt Macy 	return (0);
1646eda14cbcSMatt Macy }
1647eda14cbcSMatt Macy 
1648eda14cbcSMatt Macy #if defined(_KERNEL)
1649eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create);
1650eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_dnsize);
1651eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_norm);
1652eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_norm_dnsize);
1653eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_flags);
1654eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_flags_dnsize);
1655eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_claim);
1656eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_claim_norm);
1657eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_claim_norm_dnsize);
1658eda14cbcSMatt Macy EXPORT_SYMBOL(zap_create_hold);
1659eda14cbcSMatt Macy EXPORT_SYMBOL(zap_destroy);
1660eda14cbcSMatt Macy EXPORT_SYMBOL(zap_lookup);
1661eda14cbcSMatt Macy EXPORT_SYMBOL(zap_lookup_by_dnode);
1662eda14cbcSMatt Macy EXPORT_SYMBOL(zap_lookup_norm);
1663eda14cbcSMatt Macy EXPORT_SYMBOL(zap_lookup_uint64);
1664eda14cbcSMatt Macy EXPORT_SYMBOL(zap_contains);
1665eda14cbcSMatt Macy EXPORT_SYMBOL(zap_prefetch);
1666eda14cbcSMatt Macy EXPORT_SYMBOL(zap_prefetch_uint64);
1667eda14cbcSMatt Macy EXPORT_SYMBOL(zap_add);
1668eda14cbcSMatt Macy EXPORT_SYMBOL(zap_add_by_dnode);
1669eda14cbcSMatt Macy EXPORT_SYMBOL(zap_add_uint64);
1670eda14cbcSMatt Macy EXPORT_SYMBOL(zap_update);
1671eda14cbcSMatt Macy EXPORT_SYMBOL(zap_update_uint64);
1672eda14cbcSMatt Macy EXPORT_SYMBOL(zap_length);
1673eda14cbcSMatt Macy EXPORT_SYMBOL(zap_length_uint64);
1674eda14cbcSMatt Macy EXPORT_SYMBOL(zap_remove);
1675eda14cbcSMatt Macy EXPORT_SYMBOL(zap_remove_by_dnode);
1676eda14cbcSMatt Macy EXPORT_SYMBOL(zap_remove_norm);
1677eda14cbcSMatt Macy EXPORT_SYMBOL(zap_remove_uint64);
1678eda14cbcSMatt Macy EXPORT_SYMBOL(zap_count);
1679eda14cbcSMatt Macy EXPORT_SYMBOL(zap_value_search);
1680eda14cbcSMatt Macy EXPORT_SYMBOL(zap_join);
1681eda14cbcSMatt Macy EXPORT_SYMBOL(zap_join_increment);
1682eda14cbcSMatt Macy EXPORT_SYMBOL(zap_add_int);
1683eda14cbcSMatt Macy EXPORT_SYMBOL(zap_remove_int);
1684eda14cbcSMatt Macy EXPORT_SYMBOL(zap_lookup_int);
1685eda14cbcSMatt Macy EXPORT_SYMBOL(zap_increment_int);
1686eda14cbcSMatt Macy EXPORT_SYMBOL(zap_add_int_key);
1687eda14cbcSMatt Macy EXPORT_SYMBOL(zap_lookup_int_key);
1688eda14cbcSMatt Macy EXPORT_SYMBOL(zap_increment);
1689eda14cbcSMatt Macy EXPORT_SYMBOL(zap_cursor_init);
1690eda14cbcSMatt Macy EXPORT_SYMBOL(zap_cursor_fini);
1691eda14cbcSMatt Macy EXPORT_SYMBOL(zap_cursor_retrieve);
1692eda14cbcSMatt Macy EXPORT_SYMBOL(zap_cursor_advance);
1693eda14cbcSMatt Macy EXPORT_SYMBOL(zap_cursor_serialize);
1694eda14cbcSMatt Macy EXPORT_SYMBOL(zap_cursor_init_serialized);
1695eda14cbcSMatt Macy EXPORT_SYMBOL(zap_get_stats);
1696eda14cbcSMatt Macy #endif
1697