1 files changed, 148 insertions, 0 deletions
diff --git a/usr/src/cmd/zpool/zpool_dataset.c b/usr/src/cmd/zpool/zpool_dataset.c
new file mode 100644
index 0000000000..0432f53652
--- /dev/null
+++ b/usr/src/cmd/zpool/zpool_dataset.c
@@ -0,0 +1,148 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License, Version 1.0 only
+ * (the "License").  You may not use this file except in compliance
+ * with the License.
+ *
+ * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
+ * or http://www.opensolaris.org/os/licensing.
+ * See the License for the specific language governing permissions
+ * and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+/*
+ * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
+ * Use is subject to license terms.
+ */
+
+#pragma ident	"%Z%%M%	%I%	%E% SMI"
+
+#include <libintl.h>
+#include <libzfs.h>
+#include <sys/mount.h>
+
+#include "zpool_util.h"
+
+/*
+ * For export and destroy, we have to support iterating over all datasets and
+ * unmounting and/or destroying them.  This file contains the routines to
+ * support this.
+ */
+typedef struct cbdata {
+	int	cb_force;
+	int	cb_failed;
+	const char *cb_mntopts;
+} cbdata_t;
+
+/*
+ * Unmount a single ZFS dataset.
+ */
+int
+do_unmount(zfs_handle_t *zfsp, void *data)
+{
+	cbdata_t *cbp = data;
+
+	if (zfs_unmount(zfsp, NULL, cbp->cb_force ? MS_FORCE : 0) != 0)
+		cbp->cb_failed = 1;
+
+	return (0);
+}
+
+/*
+ * Unmount all datasets within the given pool.
+ *
+ * XXZFS it would be much more efficient, and correct, to iterate over
+ * mountpoints based on /etc/mnttab.
+ */
+int
+unmount_datasets(zpool_handle_t *zhp, int force)
+{
+	cbdata_t cb = { 0 };
+	zfs_handle_t *zfsp;
+
+	/* For unavailable pools, we don't do anything */
+	if (zpool_get_state(zhp) == POOL_STATE_UNAVAIL)
+		return (0);
+
+	if ((zfsp = zfs_open(zpool_get_name(zhp), ZFS_TYPE_FILESYSTEM)) == NULL)
+		return (-1);
+
+	cb.cb_force = force;
+
+	if (zfs_iter_dependents(zfsp, do_unmount, &cb) != 0 ||
+	    cb.cb_failed != 0) {
+		zfs_close(zfsp);
+		return (-1);
+	}
+
+	if (do_unmount(zfsp, &cb) != 0 || cb.cb_failed != 0) {
+		zfs_close(zfsp);
+		return (-1);
+	}
+
+	zfs_close(zfsp);
+
+	return (0);
+}
+
+/*
+ * Mount a single dataset
+ */
+static int
+do_mount(zfs_handle_t *zfsp, void *data)
+{
+	cbdata_t *cbp = data;
+	int ret;
+
+	if (zfs_get_type(zfsp) != ZFS_TYPE_FILESYSTEM)
+		return (0);
+
+	if (zfs_mount(zfsp, cbp->cb_mntopts, 0) != 0)
+		cbp->cb_failed = 1;
+
+	ret = zfs_iter_children(zfsp, do_mount, data);
+
+	return (ret);
+}
+
+
+/*
+ * Go through and mount all datasets within a pool.  We need to mount all
+ * datasets in order, so that we mount parents before any children.  A complete
+ * fix would gather all mountpoints, sort them, and mount them in lexical order.
+ * There are many more problems if you start to have nested filesystems - we
+ * just want to get inherited filesystems right.
+ */
+int
+mount_datasets(zpool_handle_t *zhp, const char *options)
+{
+	cbdata_t cb = { 0 };
+	zfs_handle_t *zfsp;
+
+	cb.cb_mntopts = options;
+
+	/* For unavailable pools, we don't do anything */
+	if (zpool_get_state(zhp) == POOL_STATE_UNAVAIL)
+		return (0);
+
+	if ((zfsp = zfs_open(zpool_get_name(zhp), ZFS_TYPE_FILESYSTEM)) == NULL)
+		return (-1);
+
+	if (do_mount(zfsp, &cb) != 0 || cb.cb_failed != 0) {
+		zfs_close(zfsp);
+		return (-1);
+	}
+
+	zfs_close(zfsp);
+
+	return (0);
+}