xref: /onnv-gate/usr/src/uts/common/fs/zfs/dmu_objset.c (revision 1596:2e2377ccbf85)
1789Sahrens /*
2789Sahrens  * CDDL HEADER START
3789Sahrens  *
4789Sahrens  * The contents of this file are subject to the terms of the
51544Seschrock  * Common Development and Distribution License (the "License").
61544Seschrock  * You may not use this file except in compliance with the License.
7789Sahrens  *
8789Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9789Sahrens  * or http://www.opensolaris.org/os/licensing.
10789Sahrens  * See the License for the specific language governing permissions
11789Sahrens  * and limitations under the License.
12789Sahrens  *
13789Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
14789Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15789Sahrens  * If applicable, add the following below this CDDL HEADER, with the
16789Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
17789Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
18789Sahrens  *
19789Sahrens  * CDDL HEADER END
20789Sahrens  */
21789Sahrens /*
221544Seschrock  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
23789Sahrens  * Use is subject to license terms.
24789Sahrens  */
25789Sahrens 
26789Sahrens #pragma ident	"%Z%%M%	%I%	%E% SMI"
27789Sahrens 
28789Sahrens #include <sys/zfs_context.h>
29789Sahrens #include <sys/dmu_objset.h>
30789Sahrens #include <sys/dsl_dir.h>
31789Sahrens #include <sys/dsl_dataset.h>
32789Sahrens #include <sys/dsl_prop.h>
33789Sahrens #include <sys/dsl_pool.h>
34789Sahrens #include <sys/dnode.h>
35789Sahrens #include <sys/dbuf.h>
36789Sahrens #include <sys/dmu_tx.h>
37789Sahrens #include <sys/zio_checksum.h>
38789Sahrens #include <sys/zap.h>
39789Sahrens #include <sys/zil.h>
40789Sahrens #include <sys/dmu_impl.h>
41789Sahrens 
42789Sahrens 
43789Sahrens spa_t *
44789Sahrens dmu_objset_spa(objset_t *os)
45789Sahrens {
46789Sahrens 	return (os->os->os_spa);
47789Sahrens }
48789Sahrens 
49789Sahrens zilog_t *
50789Sahrens dmu_objset_zil(objset_t *os)
51789Sahrens {
52789Sahrens 	return (os->os->os_zil);
53789Sahrens }
54789Sahrens 
55789Sahrens dsl_pool_t *
56789Sahrens dmu_objset_pool(objset_t *os)
57789Sahrens {
58789Sahrens 	dsl_dataset_t *ds;
59789Sahrens 
60789Sahrens 	if ((ds = os->os->os_dsl_dataset) != NULL && ds->ds_dir)
61789Sahrens 		return (ds->ds_dir->dd_pool);
62789Sahrens 	else
63789Sahrens 		return (spa_get_dsl(os->os->os_spa));
64789Sahrens }
65789Sahrens 
66789Sahrens dsl_dataset_t *
67789Sahrens dmu_objset_ds(objset_t *os)
68789Sahrens {
69789Sahrens 	return (os->os->os_dsl_dataset);
70789Sahrens }
71789Sahrens 
72789Sahrens dmu_objset_type_t
73789Sahrens dmu_objset_type(objset_t *os)
74789Sahrens {
75789Sahrens 	return (os->os->os_phys->os_type);
76789Sahrens }
77789Sahrens 
78789Sahrens void
79789Sahrens dmu_objset_name(objset_t *os, char *buf)
80789Sahrens {
81789Sahrens 	dsl_dataset_name(os->os->os_dsl_dataset, buf);
82789Sahrens }
83789Sahrens 
84789Sahrens uint64_t
85789Sahrens dmu_objset_id(objset_t *os)
86789Sahrens {
87789Sahrens 	dsl_dataset_t *ds = os->os->os_dsl_dataset;
88789Sahrens 
89789Sahrens 	return (ds ? ds->ds_object : 0);
90789Sahrens }
91789Sahrens 
92789Sahrens static void
93789Sahrens checksum_changed_cb(void *arg, uint64_t newval)
94789Sahrens {
95789Sahrens 	objset_impl_t *osi = arg;
96789Sahrens 
97789Sahrens 	/*
98789Sahrens 	 * Inheritance should have been done by now.
99789Sahrens 	 */
100789Sahrens 	ASSERT(newval != ZIO_CHECKSUM_INHERIT);
101789Sahrens 
102789Sahrens 	osi->os_checksum = zio_checksum_select(newval, ZIO_CHECKSUM_ON_VALUE);
103789Sahrens }
104789Sahrens 
105789Sahrens static void
106789Sahrens compression_changed_cb(void *arg, uint64_t newval)
107789Sahrens {
108789Sahrens 	objset_impl_t *osi = arg;
109789Sahrens 
110789Sahrens 	/*
111789Sahrens 	 * Inheritance and range checking should have been done by now.
112789Sahrens 	 */
113789Sahrens 	ASSERT(newval != ZIO_COMPRESS_INHERIT);
114789Sahrens 
115789Sahrens 	osi->os_compress = zio_compress_select(newval, ZIO_COMPRESS_ON_VALUE);
116789Sahrens }
117789Sahrens 
118789Sahrens void
119789Sahrens dmu_objset_byteswap(void *buf, size_t size)
120789Sahrens {
121789Sahrens 	objset_phys_t *osp = buf;
122789Sahrens 
123789Sahrens 	ASSERT(size == sizeof (objset_phys_t));
124789Sahrens 	dnode_byteswap(&osp->os_meta_dnode);
125789Sahrens 	byteswap_uint64_array(&osp->os_zil_header, sizeof (zil_header_t));
126789Sahrens 	osp->os_type = BSWAP_64(osp->os_type);
127789Sahrens }
128789Sahrens 
1291544Seschrock int
1301544Seschrock dmu_objset_open_impl(spa_t *spa, dsl_dataset_t *ds, blkptr_t *bp,
1311544Seschrock     objset_impl_t **osip)
132789Sahrens {
133789Sahrens 	objset_impl_t *winner, *osi;
134789Sahrens 	int i, err, checksum;
135789Sahrens 
136789Sahrens 	osi = kmem_zalloc(sizeof (objset_impl_t), KM_SLEEP);
137789Sahrens 	osi->os.os = osi;
138789Sahrens 	osi->os_dsl_dataset = ds;
139789Sahrens 	osi->os_spa = spa;
140789Sahrens 	if (bp)
141789Sahrens 		osi->os_rootbp = *bp;
142789Sahrens 	osi->os_phys = zio_buf_alloc(sizeof (objset_phys_t));
143789Sahrens 	if (!BP_IS_HOLE(&osi->os_rootbp)) {
1441544Seschrock 		zbookmark_t zb;
1451544Seschrock 		zb.zb_objset = ds ? ds->ds_object : 0;
1461544Seschrock 		zb.zb_object = 0;
1471544Seschrock 		zb.zb_level = -1;
1481544Seschrock 		zb.zb_blkid = 0;
1491544Seschrock 
150789Sahrens 		dprintf_bp(&osi->os_rootbp, "reading %s", "");
1511544Seschrock 		err = arc_read(NULL, spa, &osi->os_rootbp,
152789Sahrens 		    dmu_ot[DMU_OT_OBJSET].ot_byteswap,
153789Sahrens 		    arc_bcopy_func, osi->os_phys,
1541544Seschrock 		    ZIO_PRIORITY_SYNC_READ, ZIO_FLAG_CANFAIL, ARC_WAIT, &zb);
1551544Seschrock 		if (err) {
1561544Seschrock 			zio_buf_free(osi->os_phys, sizeof (objset_phys_t));
1571544Seschrock 			kmem_free(osi, sizeof (objset_impl_t));
1581544Seschrock 			return (err);
1591544Seschrock 		}
160789Sahrens 	} else {
161789Sahrens 		bzero(osi->os_phys, sizeof (objset_phys_t));
162789Sahrens 	}
163789Sahrens 
164789Sahrens 	/*
165789Sahrens 	 * Note: the changed_cb will be called once before the register
166789Sahrens 	 * func returns, thus changing the checksum/compression from the
167789Sahrens 	 * default (fletcher2/off).
168789Sahrens 	 */
169789Sahrens 	if (ds) {
170789Sahrens 		err = dsl_prop_register(ds, "checksum",
171789Sahrens 		    checksum_changed_cb, osi);
1721544Seschrock 		if (err == 0)
1731544Seschrock 			err = dsl_prop_register(ds, "compression",
1741544Seschrock 			    compression_changed_cb, osi);
1751544Seschrock 		if (err) {
1761544Seschrock 			zio_buf_free(osi->os_phys, sizeof (objset_phys_t));
1771544Seschrock 			kmem_free(osi, sizeof (objset_impl_t));
1781544Seschrock 			return (err);
1791544Seschrock 		}
180789Sahrens 	} else {
181789Sahrens 		/* It's the meta-objset. */
182789Sahrens 		osi->os_checksum = ZIO_CHECKSUM_FLETCHER_4;
1831544Seschrock 		osi->os_compress = ZIO_COMPRESS_LZJB;
184789Sahrens 	}
185789Sahrens 
1861544Seschrock 	osi->os_zil = zil_alloc(&osi->os, &osi->os_phys->os_zil_header);
1871544Seschrock 
188789Sahrens 	/*
189789Sahrens 	 * Metadata always gets compressed and checksummed.
190789Sahrens 	 * If the data checksum is multi-bit correctable, and it's not
191789Sahrens 	 * a ZBT-style checksum, then it's suitable for metadata as well.
192789Sahrens 	 * Otherwise, the metadata checksum defaults to fletcher4.
193789Sahrens 	 */
194789Sahrens 	checksum = osi->os_checksum;
195789Sahrens 
196789Sahrens 	if (zio_checksum_table[checksum].ci_correctable &&
197789Sahrens 	    !zio_checksum_table[checksum].ci_zbt)
198789Sahrens 		osi->os_md_checksum = checksum;
199789Sahrens 	else
200789Sahrens 		osi->os_md_checksum = ZIO_CHECKSUM_FLETCHER_4;
2011544Seschrock 	osi->os_md_compress = ZIO_COMPRESS_LZJB;
202789Sahrens 
203789Sahrens 	for (i = 0; i < TXG_SIZE; i++) {
204789Sahrens 		list_create(&osi->os_dirty_dnodes[i], sizeof (dnode_t),
205789Sahrens 		    offsetof(dnode_t, dn_dirty_link[i]));
206789Sahrens 		list_create(&osi->os_free_dnodes[i], sizeof (dnode_t),
207789Sahrens 		    offsetof(dnode_t, dn_dirty_link[i]));
208789Sahrens 	}
209789Sahrens 	list_create(&osi->os_dnodes, sizeof (dnode_t),
210789Sahrens 	    offsetof(dnode_t, dn_link));
211789Sahrens 	list_create(&osi->os_downgraded_dbufs, sizeof (dmu_buf_impl_t),
212789Sahrens 	    offsetof(dmu_buf_impl_t, db_link));
213789Sahrens 
214789Sahrens 	osi->os_meta_dnode = dnode_special_open(osi,
215789Sahrens 	    &osi->os_phys->os_meta_dnode, DMU_META_DNODE_OBJECT);
216789Sahrens 
217789Sahrens 	if (ds != NULL) {
218789Sahrens 		winner = dsl_dataset_set_user_ptr(ds, osi, dmu_objset_evict);
219789Sahrens 		if (winner) {
220789Sahrens 			dmu_objset_evict(ds, osi);
221789Sahrens 			osi = winner;
222789Sahrens 		}
223789Sahrens 	}
224789Sahrens 
2251544Seschrock 	*osip = osi;
2261544Seschrock 	return (0);
227789Sahrens }
228789Sahrens 
229789Sahrens /* called from zpl */
230789Sahrens int
231789Sahrens dmu_objset_open(const char *name, dmu_objset_type_t type, int mode,
232789Sahrens     objset_t **osp)
233789Sahrens {
234789Sahrens 	dsl_dataset_t *ds;
235789Sahrens 	int err;
236789Sahrens 	objset_t *os;
237789Sahrens 	objset_impl_t *osi;
238789Sahrens 
239789Sahrens 	os = kmem_alloc(sizeof (objset_t), KM_SLEEP);
240789Sahrens 	err = dsl_dataset_open(name, mode, os, &ds);
241789Sahrens 	if (err) {
242789Sahrens 		kmem_free(os, sizeof (objset_t));
243789Sahrens 		return (err);
244789Sahrens 	}
245789Sahrens 
246789Sahrens 	osi = dsl_dataset_get_user_ptr(ds);
247789Sahrens 	if (osi == NULL) {
248789Sahrens 		blkptr_t bp;
249789Sahrens 
250789Sahrens 		dsl_dataset_get_blkptr(ds, &bp);
2511544Seschrock 		err = dmu_objset_open_impl(dsl_dataset_get_spa(ds),
2521544Seschrock 		    ds, &bp, &osi);
2531544Seschrock 		if (err) {
2541544Seschrock 			dsl_dataset_close(ds, mode, os);
2551544Seschrock 			kmem_free(os, sizeof (objset_t));
2561544Seschrock 			return (err);
2571544Seschrock 		}
258789Sahrens 	}
259789Sahrens 
260789Sahrens 	os->os = osi;
261789Sahrens 	os->os_mode = mode;
262789Sahrens 
263789Sahrens 	if (type != DMU_OST_ANY && type != os->os->os_phys->os_type) {
264789Sahrens 		dmu_objset_close(os);
265789Sahrens 		return (EINVAL);
266789Sahrens 	}
267789Sahrens 	*osp = os;
268789Sahrens 	return (0);
269789Sahrens }
270789Sahrens 
271789Sahrens void
272789Sahrens dmu_objset_close(objset_t *os)
273789Sahrens {
274789Sahrens 	dsl_dataset_close(os->os->os_dsl_dataset, os->os_mode, os);
275789Sahrens 	kmem_free(os, sizeof (objset_t));
276789Sahrens }
277789Sahrens 
278789Sahrens void
2791544Seschrock dmu_objset_evict_dbufs(objset_t *os)
2801544Seschrock {
2811544Seschrock 	objset_impl_t *osi = os->os;
2821544Seschrock 	dnode_t *dn;
283*1596Sahrens 
284*1596Sahrens 	mutex_enter(&osi->os_lock);
285*1596Sahrens 
286*1596Sahrens 	/* process the mdn last, since the other dnodes have holds on it */
287*1596Sahrens 	list_remove(&osi->os_dnodes, osi->os_meta_dnode);
288*1596Sahrens 	list_insert_tail(&osi->os_dnodes, osi->os_meta_dnode);
2891544Seschrock 
2901544Seschrock 	/*
291*1596Sahrens 	 * Find the first dnode with holds.  We have to do this dance
292*1596Sahrens 	 * because dnode_add_ref() only works if you already have a
293*1596Sahrens 	 * hold.  If there are no holds then it has no dbufs so OK to
294*1596Sahrens 	 * skip.
2951544Seschrock 	 */
296*1596Sahrens 	for (dn = list_head(&osi->os_dnodes);
297*1596Sahrens 	    dn && refcount_is_zero(&dn->dn_holds);
298*1596Sahrens 	    dn = list_next(&osi->os_dnodes, dn))
299*1596Sahrens 		continue;
300*1596Sahrens 	if (dn)
301*1596Sahrens 		dnode_add_ref(dn, FTAG);
302*1596Sahrens 
303*1596Sahrens 	while (dn) {
304*1596Sahrens 		dnode_t *next_dn = dn;
305*1596Sahrens 
306*1596Sahrens 		do {
307*1596Sahrens 			next_dn = list_next(&osi->os_dnodes, next_dn);
308*1596Sahrens 		} while (next_dn && refcount_is_zero(&next_dn->dn_holds));
309*1596Sahrens 		if (next_dn)
310*1596Sahrens 			dnode_add_ref(next_dn, FTAG);
311*1596Sahrens 
312*1596Sahrens 		mutex_exit(&osi->os_lock);
313*1596Sahrens 		dnode_evict_dbufs(dn);
314*1596Sahrens 		dnode_rele(dn, FTAG);
315*1596Sahrens 		mutex_enter(&osi->os_lock);
316*1596Sahrens 		dn = next_dn;
3171544Seschrock 	}
3181544Seschrock 	mutex_exit(&osi->os_lock);
3191544Seschrock }
3201544Seschrock 
3211544Seschrock void
322789Sahrens dmu_objset_evict(dsl_dataset_t *ds, void *arg)
323789Sahrens {
324789Sahrens 	objset_impl_t *osi = arg;
3251544Seschrock 	objset_t os;
326789Sahrens 	int err, i;
327789Sahrens 
328789Sahrens 	for (i = 0; i < TXG_SIZE; i++) {
329789Sahrens 		ASSERT(list_head(&osi->os_dirty_dnodes[i]) == NULL);
330789Sahrens 		ASSERT(list_head(&osi->os_free_dnodes[i]) == NULL);
331789Sahrens 	}
332789Sahrens 
333789Sahrens 	if (ds) {
334789Sahrens 		err = dsl_prop_unregister(ds, "checksum",
335789Sahrens 		    checksum_changed_cb, osi);
336789Sahrens 		ASSERT(err == 0);
337789Sahrens 
338789Sahrens 		err = dsl_prop_unregister(ds, "compression",
339789Sahrens 		    compression_changed_cb, osi);
340789Sahrens 		ASSERT(err == 0);
341789Sahrens 	}
342789Sahrens 
3431544Seschrock 	/*
3441544Seschrock 	 * We should need only a single pass over the dnode list, since
3451544Seschrock 	 * nothing can be added to the list at this point.
3461544Seschrock 	 */
3471544Seschrock 	os.os = osi;
3481544Seschrock 	dmu_objset_evict_dbufs(&os);
3491544Seschrock 
350789Sahrens 	ASSERT3P(list_head(&osi->os_dnodes), ==, osi->os_meta_dnode);
351789Sahrens 	ASSERT3P(list_tail(&osi->os_dnodes), ==, osi->os_meta_dnode);
352789Sahrens 	ASSERT3P(list_head(&osi->os_meta_dnode->dn_dbufs), ==, NULL);
353789Sahrens 
354789Sahrens 	dnode_special_close(osi->os_meta_dnode);
355789Sahrens 	zil_free(osi->os_zil);
356789Sahrens 
357789Sahrens 	zio_buf_free(osi->os_phys, sizeof (objset_phys_t));
358789Sahrens 	kmem_free(osi, sizeof (objset_impl_t));
359789Sahrens }
360789Sahrens 
361789Sahrens /* called from dsl for meta-objset */
362789Sahrens objset_impl_t *
363789Sahrens dmu_objset_create_impl(spa_t *spa, dsl_dataset_t *ds, dmu_objset_type_t type,
364789Sahrens     dmu_tx_t *tx)
365789Sahrens {
366789Sahrens 	objset_impl_t *osi;
367789Sahrens 	dnode_t *mdn;
368789Sahrens 
369789Sahrens 	ASSERT(dmu_tx_is_syncing(tx));
3701544Seschrock 	VERIFY(0 == dmu_objset_open_impl(spa, ds, NULL, &osi));
371789Sahrens 	mdn = osi->os_meta_dnode;
372789Sahrens 
373789Sahrens 	dnode_allocate(mdn, DMU_OT_DNODE, 1 << DNODE_BLOCK_SHIFT,
374789Sahrens 	    DN_MAX_INDBLKSHIFT, DMU_OT_NONE, 0, tx);
375789Sahrens 
376789Sahrens 	/*
377789Sahrens 	 * We don't want to have to increase the meta-dnode's nlevels
378789Sahrens 	 * later, because then we could do it in quescing context while
379789Sahrens 	 * we are also accessing it in open context.
380789Sahrens 	 *
381789Sahrens 	 * This precaution is not necessary for the MOS (ds == NULL),
382789Sahrens 	 * because the MOS is only updated in syncing context.
383789Sahrens 	 * This is most fortunate: the MOS is the only objset that
384789Sahrens 	 * needs to be synced multiple times as spa_sync() iterates
385789Sahrens 	 * to convergence, so minimizing its dn_nlevels matters.
386789Sahrens 	 */
3871544Seschrock 	if (ds != NULL) {
3881544Seschrock 		int levels = 1;
3891544Seschrock 
3901544Seschrock 		/*
3911544Seschrock 		 * Determine the number of levels necessary for the meta-dnode
3921544Seschrock 		 * to contain DN_MAX_OBJECT dnodes.
3931544Seschrock 		 */
3941544Seschrock 		while ((uint64_t)mdn->dn_nblkptr << (mdn->dn_datablkshift +
3951544Seschrock 		    (levels - 1) * (mdn->dn_indblkshift - SPA_BLKPTRSHIFT)) <
3961544Seschrock 		    DN_MAX_OBJECT * sizeof (dnode_phys_t))
3971544Seschrock 			levels++;
3981544Seschrock 
399789Sahrens 		mdn->dn_next_nlevels[tx->tx_txg & TXG_MASK] =
4001544Seschrock 		    mdn->dn_nlevels = levels;
4011544Seschrock 	}
402789Sahrens 
403789Sahrens 	ASSERT(type != DMU_OST_NONE);
404789Sahrens 	ASSERT(type != DMU_OST_ANY);
405789Sahrens 	ASSERT(type < DMU_OST_NUMTYPES);
406789Sahrens 	osi->os_phys->os_type = type;
407789Sahrens 
408789Sahrens 	dsl_dataset_dirty(ds, tx);
409789Sahrens 
410789Sahrens 	return (osi);
411789Sahrens }
412789Sahrens 
413789Sahrens struct oscarg {
414789Sahrens 	void (*userfunc)(objset_t *os, void *arg, dmu_tx_t *tx);
415789Sahrens 	void *userarg;
416789Sahrens 	dsl_dataset_t *clone_parent;
417789Sahrens 	const char *fullname;
418789Sahrens 	const char *lastname;
419789Sahrens 	dmu_objset_type_t type;
420789Sahrens };
421789Sahrens 
422789Sahrens static int
423789Sahrens dmu_objset_create_sync(dsl_dir_t *dd, void *arg, dmu_tx_t *tx)
424789Sahrens {
425789Sahrens 	struct oscarg *oa = arg;
426789Sahrens 	dsl_dataset_t *ds;
427789Sahrens 	int err;
428789Sahrens 	blkptr_t bp;
429789Sahrens 
430789Sahrens 	ASSERT(dmu_tx_is_syncing(tx));
431789Sahrens 
432789Sahrens 	err = dsl_dataset_create_sync(dd, oa->fullname, oa->lastname,
433789Sahrens 	    oa->clone_parent, tx);
434789Sahrens 	dprintf_dd(dd, "fn=%s ln=%s err=%d\n",
435789Sahrens 	    oa->fullname, oa->lastname, err);
436789Sahrens 	if (err)
437789Sahrens 		return (err);
438789Sahrens 
4391544Seschrock 	VERIFY(0 == dsl_dataset_open_spa(dd->dd_pool->dp_spa, oa->fullname,
4401544Seschrock 	    DS_MODE_STANDARD | DS_MODE_READONLY, FTAG, &ds));
441789Sahrens 	dsl_dataset_get_blkptr(ds, &bp);
442789Sahrens 	if (BP_IS_HOLE(&bp)) {
443789Sahrens 		objset_impl_t *osi;
444789Sahrens 
445789Sahrens 		/* This is an empty dmu_objset; not a clone. */
446789Sahrens 		osi = dmu_objset_create_impl(dsl_dataset_get_spa(ds),
447789Sahrens 		    ds, oa->type, tx);
448789Sahrens 
449789Sahrens 		if (oa->userfunc)
450789Sahrens 			oa->userfunc(&osi->os, oa->userarg, tx);
451789Sahrens 	}
452789Sahrens 	dsl_dataset_close(ds, DS_MODE_STANDARD | DS_MODE_READONLY, FTAG);
453789Sahrens 
454789Sahrens 	return (0);
455789Sahrens }
456789Sahrens 
457789Sahrens int
458789Sahrens dmu_objset_create(const char *name, dmu_objset_type_t type,
459789Sahrens     objset_t *clone_parent,
460789Sahrens     void (*func)(objset_t *os, void *arg, dmu_tx_t *tx), void *arg)
461789Sahrens {
462789Sahrens 	dsl_dir_t *pds;
463789Sahrens 	const char *tail;
464789Sahrens 	int err = 0;
465789Sahrens 
4661544Seschrock 	err = dsl_dir_open(name, FTAG, &pds, &tail);
4671544Seschrock 	if (err)
4681544Seschrock 		return (err);
469789Sahrens 	if (tail == NULL) {
470789Sahrens 		dsl_dir_close(pds, FTAG);
471789Sahrens 		return (EEXIST);
472789Sahrens 	}
473789Sahrens 
474789Sahrens 	dprintf("name=%s\n", name);
475789Sahrens 
476789Sahrens 	if (tail[0] == '@') {
477789Sahrens 		/*
478789Sahrens 		 * If we're creating a snapshot, make sure everything
479789Sahrens 		 * they might want is on disk.  XXX Sketchy to know
480789Sahrens 		 * about snapshots here, better to put in DSL.
481789Sahrens 		 */
482789Sahrens 		objset_t *os;
483789Sahrens 		size_t plen = strchr(name, '@') - name + 1;
484789Sahrens 		char *pbuf = kmem_alloc(plen, KM_SLEEP);
485789Sahrens 		bcopy(name, pbuf, plen - 1);
486789Sahrens 		pbuf[plen - 1] = '\0';
487789Sahrens 
488789Sahrens 		err = dmu_objset_open(pbuf, DMU_OST_ANY, DS_MODE_STANDARD, &os);
489789Sahrens 		if (err == 0) {
490789Sahrens 			err = zil_suspend(dmu_objset_zil(os));
491789Sahrens 			if (err == 0) {
492789Sahrens 				err = dsl_dir_sync_task(pds,
493789Sahrens 				    dsl_dataset_snapshot_sync,
494789Sahrens 				    (void*)(tail+1), 16*1024);
495789Sahrens 				zil_resume(dmu_objset_zil(os));
496789Sahrens 			}
497789Sahrens 			dmu_objset_close(os);
498789Sahrens 		}
499789Sahrens 		kmem_free(pbuf, plen);
500789Sahrens 	} else {
501789Sahrens 		struct oscarg oa = { 0 };
502789Sahrens 		oa.userfunc = func;
503789Sahrens 		oa.userarg = arg;
504789Sahrens 		oa.fullname = name;
505789Sahrens 		oa.lastname = tail;
506789Sahrens 		oa.type = type;
507789Sahrens 		if (clone_parent != NULL) {
508789Sahrens 			/*
509789Sahrens 			 * You can't clone to a different type.
510789Sahrens 			 */
511789Sahrens 			if (clone_parent->os->os_phys->os_type != type) {
512789Sahrens 				dsl_dir_close(pds, FTAG);
513789Sahrens 				return (EINVAL);
514789Sahrens 			}
515789Sahrens 			oa.clone_parent = clone_parent->os->os_dsl_dataset;
516789Sahrens 		}
517789Sahrens 		err = dsl_dir_sync_task(pds, dmu_objset_create_sync, &oa,
518789Sahrens 		    256*1024);
519789Sahrens 	}
520789Sahrens 	dsl_dir_close(pds, FTAG);
521789Sahrens 	return (err);
522789Sahrens }
523789Sahrens 
524789Sahrens int
525789Sahrens dmu_objset_destroy(const char *name)
526789Sahrens {
527789Sahrens 	objset_t *os;
528789Sahrens 	int error;
529789Sahrens 
530789Sahrens 	/*
531789Sahrens 	 * If it looks like we'll be able to destroy it, and there's
532789Sahrens 	 * an unplayed replay log sitting around, destroy the log.
533789Sahrens 	 * It would be nicer to do this in dsl_dataset_destroy_sync(),
534789Sahrens 	 * but the replay log objset is modified in open context.
535789Sahrens 	 */
536789Sahrens 	error = dmu_objset_open(name, DMU_OST_ANY, DS_MODE_EXCLUSIVE, &os);
537789Sahrens 	if (error == 0) {
538789Sahrens 		zil_destroy(dmu_objset_zil(os));
539789Sahrens 		dmu_objset_close(os);
540789Sahrens 	}
541789Sahrens 
542789Sahrens 	/* XXX uncache everything? */
543789Sahrens 	return (dsl_dataset_destroy(name));
544789Sahrens }
545789Sahrens 
546789Sahrens int
547789Sahrens dmu_objset_rollback(const char *name)
548789Sahrens {
549789Sahrens 	int err;
550789Sahrens 	objset_t *os;
551789Sahrens 
552789Sahrens 	err = dmu_objset_open(name, DMU_OST_ANY, DS_MODE_EXCLUSIVE, &os);
553789Sahrens 	if (err == 0) {
554789Sahrens 		err = zil_suspend(dmu_objset_zil(os));
555789Sahrens 		if (err == 0)
556789Sahrens 			zil_resume(dmu_objset_zil(os));
557789Sahrens 		dmu_objset_close(os);
558789Sahrens 		if (err == 0) {
559789Sahrens 			/* XXX uncache everything? */
560789Sahrens 			err = dsl_dataset_rollback(name);
561789Sahrens 		}
562789Sahrens 	}
563789Sahrens 	return (err);
564789Sahrens }
565789Sahrens 
566789Sahrens static void
567789Sahrens dmu_objset_sync_dnodes(objset_impl_t *os, list_t *list, dmu_tx_t *tx)
568789Sahrens {
569789Sahrens 	dnode_t *dn = list_head(list);
570789Sahrens 	int level, err;
571789Sahrens 
572789Sahrens 	for (level = 0; dn = list_head(list); level++) {
573789Sahrens 		zio_t *zio;
574789Sahrens 		zio = zio_root(os->os_spa, NULL, NULL, ZIO_FLAG_MUSTSUCCEED);
575789Sahrens 
576789Sahrens 		ASSERT3U(level, <=, DN_MAX_LEVELS);
577789Sahrens 
578789Sahrens 		while (dn) {
579789Sahrens 			dnode_t *next = list_next(list, dn);
580789Sahrens 
581789Sahrens 			list_remove(list, dn);
582789Sahrens 			if (dnode_sync(dn, level, zio, tx) == 0) {
583789Sahrens 				/*
584789Sahrens 				 * This dnode requires syncing at higher
585789Sahrens 				 * levels; put it back onto the list.
586789Sahrens 				 */
587789Sahrens 				if (next)
588789Sahrens 					list_insert_before(list, next, dn);
589789Sahrens 				else
590789Sahrens 					list_insert_tail(list, dn);
591789Sahrens 			}
592789Sahrens 			dn = next;
593789Sahrens 		}
594789Sahrens 		err = zio_wait(zio);
595789Sahrens 		ASSERT(err == 0);
596789Sahrens 	}
597789Sahrens }
598789Sahrens 
599789Sahrens /* ARGSUSED */
600789Sahrens static void
601789Sahrens killer(zio_t *zio, arc_buf_t *abuf, void *arg)
602789Sahrens {
603789Sahrens 	objset_impl_t *os = arg;
604789Sahrens 	objset_phys_t *osphys = zio->io_data;
605789Sahrens 	dnode_phys_t *dnp = &osphys->os_meta_dnode;
606789Sahrens 	int i;
607789Sahrens 
608789Sahrens 	ASSERT3U(zio->io_error, ==, 0);
609789Sahrens 
610789Sahrens 	/*
611789Sahrens 	 * Update rootbp fill count.
612789Sahrens 	 */
613789Sahrens 	os->os_rootbp.blk_fill = 1;	/* count the meta-dnode */
614789Sahrens 	for (i = 0; i < dnp->dn_nblkptr; i++)
615789Sahrens 		os->os_rootbp.blk_fill += dnp->dn_blkptr[i].blk_fill;
616789Sahrens 
617789Sahrens 	BP_SET_TYPE(zio->io_bp, DMU_OT_OBJSET);
618789Sahrens 	BP_SET_LEVEL(zio->io_bp, 0);
619789Sahrens 
620789Sahrens 	if (!DVA_EQUAL(BP_IDENTITY(zio->io_bp),
621789Sahrens 	    BP_IDENTITY(&zio->io_bp_orig))) {
622789Sahrens 		dsl_dataset_block_kill(os->os_dsl_dataset, &zio->io_bp_orig,
623789Sahrens 		    os->os_synctx);
624789Sahrens 		dsl_dataset_block_born(os->os_dsl_dataset, zio->io_bp,
625789Sahrens 		    os->os_synctx);
626789Sahrens 	}
627789Sahrens }
628789Sahrens 
629789Sahrens 
630789Sahrens /* called from dsl */
631789Sahrens void
632789Sahrens dmu_objset_sync(objset_impl_t *os, dmu_tx_t *tx)
633789Sahrens {
634789Sahrens 	extern taskq_t *dbuf_tq;
635789Sahrens 	int txgoff;
636789Sahrens 	list_t *dirty_list;
637789Sahrens 	int err;
6381544Seschrock 	zbookmark_t zb;
639789Sahrens 	arc_buf_t *abuf =
640789Sahrens 	    arc_buf_alloc(os->os_spa, sizeof (objset_phys_t), FTAG);
641789Sahrens 
642789Sahrens 	ASSERT(dmu_tx_is_syncing(tx));
643789Sahrens 	ASSERT(os->os_synctx == NULL);
644789Sahrens 	/* XXX the write_done callback should really give us the tx... */
645789Sahrens 	os->os_synctx = tx;
646789Sahrens 
647789Sahrens 	dprintf_ds(os->os_dsl_dataset, "txg=%llu\n", tx->tx_txg);
648789Sahrens 
649789Sahrens 	txgoff = tx->tx_txg & TXG_MASK;
650789Sahrens 
651789Sahrens 	dmu_objset_sync_dnodes(os, &os->os_free_dnodes[txgoff], tx);
652789Sahrens 	dmu_objset_sync_dnodes(os, &os->os_dirty_dnodes[txgoff], tx);
653789Sahrens 
654789Sahrens 	/*
655789Sahrens 	 * Free intent log blocks up to this tx.
656789Sahrens 	 */
657789Sahrens 	zil_sync(os->os_zil, tx);
658789Sahrens 
659789Sahrens 	/*
660789Sahrens 	 * Sync meta-dnode
661789Sahrens 	 */
662789Sahrens 	dirty_list = &os->os_dirty_dnodes[txgoff];
663789Sahrens 	ASSERT(list_head(dirty_list) == NULL);
664789Sahrens 	list_insert_tail(dirty_list, os->os_meta_dnode);
665789Sahrens 	dmu_objset_sync_dnodes(os, dirty_list, tx);
666789Sahrens 
667789Sahrens 	/*
668789Sahrens 	 * Sync the root block.
669789Sahrens 	 */
670789Sahrens 	bcopy(os->os_phys, abuf->b_data, sizeof (objset_phys_t));
6711544Seschrock 	zb.zb_objset = os->os_dsl_dataset ? os->os_dsl_dataset->ds_object : 0;
6721544Seschrock 	zb.zb_object = 0;
6731544Seschrock 	zb.zb_level = -1;
6741544Seschrock 	zb.zb_blkid = 0;
675789Sahrens 	err = arc_write(NULL, os->os_spa, os->os_md_checksum,
676789Sahrens 	    os->os_md_compress, tx->tx_txg, &os->os_rootbp, abuf, killer, os,
6771544Seschrock 	    ZIO_PRIORITY_ASYNC_WRITE, ZIO_FLAG_MUSTSUCCEED, ARC_WAIT, &zb);
678789Sahrens 	ASSERT(err == 0);
6791544Seschrock 	VERIFY(arc_buf_remove_ref(abuf, FTAG) == 1);
680789Sahrens 
681789Sahrens 	dsl_dataset_set_blkptr(os->os_dsl_dataset, &os->os_rootbp, tx);
682789Sahrens 
683789Sahrens 	ASSERT3P(os->os_synctx, ==, tx);
684789Sahrens 	taskq_wait(dbuf_tq);
685789Sahrens 	os->os_synctx = NULL;
686789Sahrens }
687789Sahrens 
688789Sahrens void
689789Sahrens dmu_objset_stats(objset_t *os, dmu_objset_stats_t *dds)
690789Sahrens {
691789Sahrens 	if (os->os->os_dsl_dataset != NULL) {
692789Sahrens 		dsl_dataset_stats(os->os->os_dsl_dataset, dds);
693789Sahrens 	} else {
694789Sahrens 		ASSERT(os->os->os_phys->os_type == DMU_OST_META);
695789Sahrens 		bzero(dds, sizeof (*dds));
696789Sahrens 	}
697789Sahrens 	dds->dds_type = os->os->os_phys->os_type;
698789Sahrens }
699789Sahrens 
700789Sahrens int
701789Sahrens dmu_objset_is_snapshot(objset_t *os)
702789Sahrens {
703789Sahrens 	if (os->os->os_dsl_dataset != NULL)
704789Sahrens 		return (dsl_dataset_is_snapshot(os->os->os_dsl_dataset));
705789Sahrens 	else
706789Sahrens 		return (B_FALSE);
707789Sahrens }
708789Sahrens 
709789Sahrens int
710789Sahrens dmu_snapshot_list_next(objset_t *os, int namelen, char *name,
711885Sahrens     uint64_t *idp, uint64_t *offp)
712789Sahrens {
713789Sahrens 	dsl_dataset_t *ds = os->os->os_dsl_dataset;
714789Sahrens 	zap_cursor_t cursor;
715789Sahrens 	zap_attribute_t attr;
716789Sahrens 
717789Sahrens 	if (ds->ds_phys->ds_snapnames_zapobj == 0)
718789Sahrens 		return (ENOENT);
719789Sahrens 
720789Sahrens 	zap_cursor_init_serialized(&cursor,
721789Sahrens 	    ds->ds_dir->dd_pool->dp_meta_objset,
722789Sahrens 	    ds->ds_phys->ds_snapnames_zapobj, *offp);
723789Sahrens 
724885Sahrens 	if (zap_cursor_retrieve(&cursor, &attr) != 0) {
725885Sahrens 		zap_cursor_fini(&cursor);
726885Sahrens 		return (ENOENT);
727885Sahrens 	}
728885Sahrens 
729885Sahrens 	if (strlen(attr.za_name) + 1 > namelen) {
730885Sahrens 		zap_cursor_fini(&cursor);
731885Sahrens 		return (ENAMETOOLONG);
732885Sahrens 	}
733885Sahrens 
734885Sahrens 	(void) strcpy(name, attr.za_name);
735885Sahrens 	if (idp)
736885Sahrens 		*idp = attr.za_first_integer;
737885Sahrens 	zap_cursor_advance(&cursor);
738885Sahrens 	*offp = zap_cursor_serialize(&cursor);
739885Sahrens 	zap_cursor_fini(&cursor);
740885Sahrens 
741885Sahrens 	return (0);
742885Sahrens }
743885Sahrens 
744885Sahrens int
745885Sahrens dmu_dir_list_next(objset_t *os, int namelen, char *name,
746885Sahrens     uint64_t *idp, uint64_t *offp)
747885Sahrens {
748885Sahrens 	dsl_dir_t *dd = os->os->os_dsl_dataset->ds_dir;
749885Sahrens 	zap_cursor_t cursor;
750885Sahrens 	zap_attribute_t attr;
751885Sahrens 
752885Sahrens 	if (dd->dd_phys->dd_child_dir_zapobj == 0)
753789Sahrens 		return (ENOENT);
754789Sahrens 
755885Sahrens 	/* there is no next dir on a snapshot! */
756885Sahrens 	if (os->os->os_dsl_dataset->ds_object !=
757885Sahrens 	    dd->dd_phys->dd_head_dataset_obj)
758885Sahrens 		return (ENOENT);
759885Sahrens 
760885Sahrens 	zap_cursor_init_serialized(&cursor,
761885Sahrens 	    dd->dd_pool->dp_meta_objset,
762885Sahrens 	    dd->dd_phys->dd_child_dir_zapobj, *offp);
763885Sahrens 
764885Sahrens 	if (zap_cursor_retrieve(&cursor, &attr) != 0) {
765885Sahrens 		zap_cursor_fini(&cursor);
766885Sahrens 		return (ENOENT);
767885Sahrens 	}
768885Sahrens 
769885Sahrens 	if (strlen(attr.za_name) + 1 > namelen) {
770885Sahrens 		zap_cursor_fini(&cursor);
771789Sahrens 		return (ENAMETOOLONG);
772885Sahrens 	}
773789Sahrens 
774789Sahrens 	(void) strcpy(name, attr.za_name);
775885Sahrens 	if (idp)
776885Sahrens 		*idp = attr.za_first_integer;
777789Sahrens 	zap_cursor_advance(&cursor);
778789Sahrens 	*offp = zap_cursor_serialize(&cursor);
779885Sahrens 	zap_cursor_fini(&cursor);
780789Sahrens 
781789Sahrens 	return (0);
782789Sahrens }
783789Sahrens 
784789Sahrens /*
785789Sahrens  * Find all objsets under name, and for each, call 'func(child_name, arg)'.
786789Sahrens  */
787789Sahrens void
788789Sahrens dmu_objset_find(char *name, void func(char *, void *), void *arg, int flags)
789789Sahrens {
790789Sahrens 	dsl_dir_t *dd;
791789Sahrens 	objset_t *os;
792789Sahrens 	uint64_t snapobj;
793789Sahrens 	zap_cursor_t zc;
794789Sahrens 	zap_attribute_t attr;
795789Sahrens 	char *child;
7961544Seschrock 	int do_self, err;
797789Sahrens 
7981544Seschrock 	err = dsl_dir_open(name, FTAG, &dd, NULL);
7991544Seschrock 	if (err)
800789Sahrens 		return;
801789Sahrens 
802789Sahrens 	do_self = (dd->dd_phys->dd_head_dataset_obj != 0);
803789Sahrens 
804789Sahrens 	/*
805789Sahrens 	 * Iterate over all children.
806789Sahrens 	 */
807789Sahrens 	if (dd->dd_phys->dd_child_dir_zapobj != 0) {
808789Sahrens 		for (zap_cursor_init(&zc, dd->dd_pool->dp_meta_objset,
809789Sahrens 		    dd->dd_phys->dd_child_dir_zapobj);
810789Sahrens 		    zap_cursor_retrieve(&zc, &attr) == 0;
811789Sahrens 		    (void) zap_cursor_advance(&zc)) {
812789Sahrens 			ASSERT(attr.za_integer_length == sizeof (uint64_t));
813789Sahrens 			ASSERT(attr.za_num_integers == 1);
814789Sahrens 
815789Sahrens 			/*
816789Sahrens 			 * No separating '/' because parent's name ends in /.
817789Sahrens 			 */
818789Sahrens 			child = kmem_alloc(MAXPATHLEN, KM_SLEEP);
819789Sahrens 			/* XXX could probably just use name here */
820789Sahrens 			dsl_dir_name(dd, child);
821789Sahrens 			(void) strcat(child, "/");
822789Sahrens 			(void) strcat(child, attr.za_name);
823789Sahrens 			dmu_objset_find(child, func, arg, flags);
824789Sahrens 			kmem_free(child, MAXPATHLEN);
825789Sahrens 		}
826885Sahrens 		zap_cursor_fini(&zc);
827789Sahrens 	}
828789Sahrens 
829789Sahrens 	/*
830789Sahrens 	 * Iterate over all snapshots.
831789Sahrens 	 */
832789Sahrens 	if ((flags & DS_FIND_SNAPSHOTS) &&
833789Sahrens 	    dmu_objset_open(name, DMU_OST_ANY,
834789Sahrens 	    DS_MODE_STANDARD | DS_MODE_READONLY, &os) == 0) {
835789Sahrens 
836789Sahrens 		snapobj = os->os->os_dsl_dataset->ds_phys->ds_snapnames_zapobj;
837789Sahrens 		dmu_objset_close(os);
838789Sahrens 
839789Sahrens 		for (zap_cursor_init(&zc, dd->dd_pool->dp_meta_objset, snapobj);
840789Sahrens 		    zap_cursor_retrieve(&zc, &attr) == 0;
841789Sahrens 		    (void) zap_cursor_advance(&zc)) {
842789Sahrens 			ASSERT(attr.za_integer_length == sizeof (uint64_t));
843789Sahrens 			ASSERT(attr.za_num_integers == 1);
844789Sahrens 
845789Sahrens 			child = kmem_alloc(MAXPATHLEN, KM_SLEEP);
846789Sahrens 			/* XXX could probably just use name here */
847789Sahrens 			dsl_dir_name(dd, child);
848789Sahrens 			(void) strcat(child, "@");
849789Sahrens 			(void) strcat(child, attr.za_name);
850789Sahrens 			func(child, arg);
851789Sahrens 			kmem_free(child, MAXPATHLEN);
852789Sahrens 		}
853885Sahrens 		zap_cursor_fini(&zc);
854789Sahrens 	}
855789Sahrens 
856789Sahrens 	dsl_dir_close(dd, FTAG);
857789Sahrens 
858789Sahrens 	/*
859789Sahrens 	 * Apply to self if appropriate.
860789Sahrens 	 */
861789Sahrens 	if (do_self)
862789Sahrens 		func(name, arg);
863789Sahrens }
864