xref: /onnv-gate/usr/src/cmd/zfs/zfs_main.c (revision 7538:18c2451107fd)
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  */
213126Sahl 
22789Sahrens /*
235904Stimh  * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
24789Sahrens  * Use is subject to license terms.
25789Sahrens  */
26789Sahrens 
27789Sahrens #include <assert.h>
282676Seschrock #include <ctype.h>
29789Sahrens #include <errno.h>
30789Sahrens #include <libgen.h>
31789Sahrens #include <libintl.h>
32789Sahrens #include <libuutil.h>
334543Smarks #include <libnvpair.h>
34789Sahrens #include <locale.h>
35789Sahrens #include <stddef.h>
36789Sahrens #include <stdio.h>
37789Sahrens #include <stdlib.h>
38789Sahrens #include <strings.h>
39789Sahrens #include <unistd.h>
40789Sahrens #include <fcntl.h>
41789Sahrens #include <zone.h>
42789Sahrens #include <sys/mkdev.h>
43789Sahrens #include <sys/mntent.h>
44789Sahrens #include <sys/mnttab.h>
45789Sahrens #include <sys/mount.h>
46789Sahrens #include <sys/stat.h>
474543Smarks #include <sys/avl.h>
48789Sahrens 
49789Sahrens #include <libzfs.h>
504543Smarks #include <libuutil.h>
51789Sahrens 
52789Sahrens #include "zfs_iter.h"
532082Seschrock #include "zfs_util.h"
542082Seschrock 
552082Seschrock libzfs_handle_t *g_zfs;
56789Sahrens 
57789Sahrens static FILE *mnttab_file;
584988Sek110237 static char history_str[HIS_MAX_RECORD_LEN];
59789Sahrens 
60789Sahrens static int zfs_do_clone(int argc, char **argv);
61789Sahrens static int zfs_do_create(int argc, char **argv);
62789Sahrens static int zfs_do_destroy(int argc, char **argv);
63789Sahrens static int zfs_do_get(int argc, char **argv);
64789Sahrens static int zfs_do_inherit(int argc, char **argv);
65789Sahrens static int zfs_do_list(int argc, char **argv);
66789Sahrens static int zfs_do_mount(int argc, char **argv);
67789Sahrens static int zfs_do_rename(int argc, char **argv);
68789Sahrens static int zfs_do_rollback(int argc, char **argv);
69789Sahrens static int zfs_do_set(int argc, char **argv);
704577Sahrens static int zfs_do_upgrade(int argc, char **argv);
71789Sahrens static int zfs_do_snapshot(int argc, char **argv);
72789Sahrens static int zfs_do_unmount(int argc, char **argv);
73789Sahrens static int zfs_do_share(int argc, char **argv);
74789Sahrens static int zfs_do_unshare(int argc, char **argv);
751749Sahrens static int zfs_do_send(int argc, char **argv);
761749Sahrens static int zfs_do_receive(int argc, char **argv);
772082Seschrock static int zfs_do_promote(int argc, char **argv);
784543Smarks static int zfs_do_allow(int argc, char **argv);
794543Smarks static int zfs_do_unallow(int argc, char **argv);
80789Sahrens 
81789Sahrens /*
826865Srm160521  * Enable a reasonable set of defaults for libumem debugging on DEBUG builds.
83789Sahrens  */
846865Srm160521 
856865Srm160521 #ifdef DEBUG
86789Sahrens const char *
873126Sahl _umem_debug_init(void)
88789Sahrens {
89789Sahrens 	return ("default,verbose"); /* $UMEM_DEBUG setting */
90789Sahrens }
91789Sahrens 
92789Sahrens const char *
93789Sahrens _umem_logging_init(void)
94789Sahrens {
95789Sahrens 	return ("fail,contents"); /* $UMEM_LOGGING setting */
96789Sahrens }
976865Srm160521 #endif
98789Sahrens 
991387Seschrock typedef enum {
1001387Seschrock 	HELP_CLONE,
1011387Seschrock 	HELP_CREATE,
1021387Seschrock 	HELP_DESTROY,
1031387Seschrock 	HELP_GET,
1041387Seschrock 	HELP_INHERIT,
1054577Sahrens 	HELP_UPGRADE,
1061387Seschrock 	HELP_LIST,
1071387Seschrock 	HELP_MOUNT,
1082082Seschrock 	HELP_PROMOTE,
1091749Sahrens 	HELP_RECEIVE,
1101387Seschrock 	HELP_RENAME,
1111387Seschrock 	HELP_ROLLBACK,
1121749Sahrens 	HELP_SEND,
1131387Seschrock 	HELP_SET,
1141387Seschrock 	HELP_SHARE,
1151387Seschrock 	HELP_SNAPSHOT,
1161387Seschrock 	HELP_UNMOUNT,
1174543Smarks 	HELP_UNSHARE,
1184543Smarks 	HELP_ALLOW,
1194543Smarks 	HELP_UNALLOW
1201387Seschrock } zfs_help_t;
1211387Seschrock 
122789Sahrens typedef struct zfs_command {
123789Sahrens 	const char	*name;
124789Sahrens 	int		(*func)(int argc, char **argv);
1251387Seschrock 	zfs_help_t	usage;
126789Sahrens } zfs_command_t;
127789Sahrens 
128789Sahrens /*
129789Sahrens  * Master command table.  Each ZFS command has a name, associated function, and
1301544Seschrock  * usage message.  The usage messages need to be internationalized, so we have
1311544Seschrock  * to have a function to return the usage message based on a command index.
1321387Seschrock  *
1331387Seschrock  * These commands are organized according to how they are displayed in the usage
1341387Seschrock  * message.  An empty command (one with a NULL name) indicates an empty line in
1351387Seschrock  * the generic usage message.
136789Sahrens  */
137789Sahrens static zfs_command_t command_table[] = {
1381387Seschrock 	{ "create",	zfs_do_create,		HELP_CREATE		},
1391387Seschrock 	{ "destroy",	zfs_do_destroy,		HELP_DESTROY		},
140789Sahrens 	{ NULL },
1411387Seschrock 	{ "snapshot",	zfs_do_snapshot,	HELP_SNAPSHOT		},
1421387Seschrock 	{ "rollback",	zfs_do_rollback,	HELP_ROLLBACK		},
1431387Seschrock 	{ "clone",	zfs_do_clone,		HELP_CLONE		},
1442082Seschrock 	{ "promote",	zfs_do_promote,		HELP_PROMOTE		},
1451387Seschrock 	{ "rename",	zfs_do_rename,		HELP_RENAME		},
146789Sahrens 	{ NULL },
1471387Seschrock 	{ "list",	zfs_do_list,		HELP_LIST		},
148789Sahrens 	{ NULL },
1491387Seschrock 	{ "set",	zfs_do_set,		HELP_SET		},
1501387Seschrock 	{ "get", 	zfs_do_get,		HELP_GET		},
1511387Seschrock 	{ "inherit",	zfs_do_inherit,		HELP_INHERIT		},
1524577Sahrens 	{ "upgrade",	zfs_do_upgrade,		HELP_UPGRADE		},
153789Sahrens 	{ NULL },
1541387Seschrock 	{ "mount",	zfs_do_mount,		HELP_MOUNT		},
1551387Seschrock 	{ "unmount",	zfs_do_unmount,		HELP_UNMOUNT		},
1561387Seschrock 	{ "share",	zfs_do_share,		HELP_SHARE		},
1571387Seschrock 	{ "unshare",	zfs_do_unshare,		HELP_UNSHARE		},
158789Sahrens 	{ NULL },
1591749Sahrens 	{ "send",	zfs_do_send,		HELP_SEND		},
1601749Sahrens 	{ "receive",	zfs_do_receive,		HELP_RECEIVE		},
1614543Smarks 	{ NULL },
1624543Smarks 	{ "allow",	zfs_do_allow,		HELP_ALLOW		},
1634543Smarks 	{ NULL },
1644543Smarks 	{ "unallow",	zfs_do_unallow,		HELP_UNALLOW		},
165789Sahrens };
166789Sahrens 
167789Sahrens #define	NCOMMAND	(sizeof (command_table) / sizeof (command_table[0]))
168789Sahrens 
169789Sahrens zfs_command_t *current_command;
170789Sahrens 
1711387Seschrock static const char *
1721387Seschrock get_usage(zfs_help_t idx)
1731387Seschrock {
1741387Seschrock 	switch (idx) {
1751387Seschrock 	case HELP_CLONE:
1767265Sahrens 		return (gettext("\tclone [-p] [-o property=value] ... "
1777265Sahrens 		    "<snapshot> <filesystem|volume>\n"));
1781387Seschrock 	case HELP_CREATE:
1794849Sahrens 		return (gettext("\tcreate [-p] [-o property=value] ... "
1802676Seschrock 		    "<filesystem>\n"
1814849Sahrens 		    "\tcreate [-ps] [-b blocksize] [-o property=value] ... "
1824849Sahrens 		    "-V <size> <volume>\n"));
1831387Seschrock 	case HELP_DESTROY:
1841387Seschrock 		return (gettext("\tdestroy [-rRf] "
1851387Seschrock 		    "<filesystem|volume|snapshot>\n"));
1861387Seschrock 	case HELP_GET:
1874849Sahrens 		return (gettext("\tget [-rHp] [-o field[,...]] "
1884849Sahrens 		    "[-s source[,...]]\n"
1894849Sahrens 		    "\t    <\"all\" | property[,...]> "
1902676Seschrock 		    "[filesystem|volume|snapshot] ...\n"));
1911387Seschrock 	case HELP_INHERIT:
1921387Seschrock 		return (gettext("\tinherit [-r] <property> "
1937265Sahrens 		    "<filesystem|volume|snapshot> ...\n"));
1944577Sahrens 	case HELP_UPGRADE:
1954577Sahrens 		return (gettext("\tupgrade [-v]\n"
1964577Sahrens 		    "\tupgrade [-r] [-V version] <-a | filesystem ...>\n"));
1971387Seschrock 	case HELP_LIST:
1984849Sahrens 		return (gettext("\tlist [-rH] [-o property[,...]] "
1994849Sahrens 		    "[-t type[,...]] [-s property] ...\n"
2004849Sahrens 		    "\t    [-S property] ... "
2014849Sahrens 		    "[filesystem|volume|snapshot] ...\n"));
2021387Seschrock 	case HELP_MOUNT:
2031387Seschrock 		return (gettext("\tmount\n"
2044849Sahrens 		    "\tmount [-vO] [-o opts] <-a | filesystem>\n"));
2052082Seschrock 	case HELP_PROMOTE:
2064849Sahrens 		return (gettext("\tpromote <clone-filesystem>\n"));
2071749Sahrens 	case HELP_RECEIVE:
2082665Snd150628 		return (gettext("\treceive [-vnF] <filesystem|volume|"
2092665Snd150628 		"snapshot>\n"
2102665Snd150628 		"\treceive [-vnF] -d <filesystem>\n"));
2111387Seschrock 	case HELP_RENAME:
2121387Seschrock 		return (gettext("\trename <filesystem|volume|snapshot> "
2134007Smmusante 		    "<filesystem|volume|snapshot>\n"
2144490Svb160487 		    "\trename -p <filesystem|volume> <filesystem|volume>\n"
2154007Smmusante 		    "\trename -r <snapshot> <snapshot>"));
2161387Seschrock 	case HELP_ROLLBACK:
2175568Sahrens 		return (gettext("\trollback [-rRf] <snapshot>\n"));
2181749Sahrens 	case HELP_SEND:
2195367Sahrens 		return (gettext("\tsend [-R] [-[iI] snapshot] <snapshot>\n"));
2201387Seschrock 	case HELP_SET:
2211387Seschrock 		return (gettext("\tset <property=value> "
2227265Sahrens 		    "<filesystem|volume|snapshot> ...\n"));
2231387Seschrock 	case HELP_SHARE:
2244849Sahrens 		return (gettext("\tshare <-a | filesystem>\n"));
2251387Seschrock 	case HELP_SNAPSHOT:
2267265Sahrens 		return (gettext("\tsnapshot [-r] [-o property=value] ... "
2274849Sahrens 		    "<filesystem@snapname|volume@snapname>\n"));
2281387Seschrock 	case HELP_UNMOUNT:
2294849Sahrens 		return (gettext("\tunmount [-f] "
2304849Sahrens 		    "<-a | filesystem|mountpoint>\n"));
2311387Seschrock 	case HELP_UNSHARE:
2324849Sahrens 		return (gettext("\tunshare [-f] "
2334849Sahrens 		    "<-a | filesystem|mountpoint>\n"));
2344543Smarks 	case HELP_ALLOW:
2354849Sahrens 		return (gettext("\tallow [-ldug] "
2364849Sahrens 		    "<\"everyone\"|user|group>[,...] <perm|@setname>[,...]\n"
2374849Sahrens 		    "\t    <filesystem|volume>\n"
2384849Sahrens 		    "\tallow [-ld] -e <perm|@setname>[,...] "
2394849Sahrens 		    "<filesystem|volume>\n"
2404849Sahrens 		    "\tallow -c <perm|@setname>[,...] <filesystem|volume>\n"
2414849Sahrens 		    "\tallow -s @setname <perm|@setname>[,...] "
2424849Sahrens 		    "<filesystem|volume>\n"));
2434543Smarks 	case HELP_UNALLOW:
2444849Sahrens 		return (gettext("\tunallow [-rldug] "
2454849Sahrens 		    "<\"everyone\"|user|group>[,...]\n"
2464849Sahrens 		    "\t    [<perm|@setname>[,...]] <filesystem|volume>\n"
2474849Sahrens 		    "\tunallow [-rld] -e [<perm|@setname>[,...]] "
2484849Sahrens 		    "<filesystem|volume>\n"
2494849Sahrens 		    "\tunallow [-r] -c [<perm|@setname>[,...]] "
2504849Sahrens 		    "<filesystem|volume>\n"
2514849Sahrens 		    "\tunallow [-r] -s @setname [<perm|@setname>[,...]] "
2524849Sahrens 		    "<filesystem|volume>\n"));
2531387Seschrock 	}
2541387Seschrock 
2551387Seschrock 	abort();
2561387Seschrock 	/* NOTREACHED */
2571387Seschrock }
2581387Seschrock 
259789Sahrens /*
260789Sahrens  * Utility function to guarantee malloc() success.
261789Sahrens  */
262789Sahrens void *
263789Sahrens safe_malloc(size_t size)
264789Sahrens {
265789Sahrens 	void *data;
266789Sahrens 
267789Sahrens 	if ((data = calloc(1, size)) == NULL) {
268789Sahrens 		(void) fprintf(stderr, "internal error: out of memory\n");
269789Sahrens 		exit(1);
270789Sahrens 	}
271789Sahrens 
272789Sahrens 	return (data);
273789Sahrens }
274789Sahrens 
275789Sahrens /*
2765094Slling  * Callback routine that will print out information for each of
2773654Sgw25295  * the properties.
2783654Sgw25295  */
2795094Slling static int
2805094Slling usage_prop_cb(int prop, void *cb)
2813654Sgw25295 {
2823654Sgw25295 	FILE *fp = cb;
2833654Sgw25295 
2846643Seschrock 	(void) fprintf(fp, "\t%-15s ", zfs_prop_to_name(prop));
2856643Seschrock 
2866643Seschrock 	if (zfs_prop_readonly(prop))
2876643Seschrock 		(void) fprintf(fp, " NO    ");
2883654Sgw25295 	else
2896643Seschrock 		(void) fprintf(fp, "YES    ");
2903654Sgw25295 
2913654Sgw25295 	if (zfs_prop_inheritable(prop))
2923654Sgw25295 		(void) fprintf(fp, "  YES   ");
2933654Sgw25295 	else
2943654Sgw25295 		(void) fprintf(fp, "   NO   ");
2953654Sgw25295 
2963654Sgw25295 	if (zfs_prop_values(prop) == NULL)
2973654Sgw25295 		(void) fprintf(fp, "-\n");
2983654Sgw25295 	else
2993654Sgw25295 		(void) fprintf(fp, "%s\n", zfs_prop_values(prop));
3003654Sgw25295 
3015094Slling 	return (ZPROP_CONT);
3023654Sgw25295 }
3033654Sgw25295 
3043654Sgw25295 /*
305789Sahrens  * Display usage message.  If we're inside a command, display only the usage for
306789Sahrens  * that command.  Otherwise, iterate over the entire command table and display
307789Sahrens  * a complete usage message.
308789Sahrens  */
309789Sahrens static void
3102082Seschrock usage(boolean_t requested)
311789Sahrens {
312789Sahrens 	int i;
3132082Seschrock 	boolean_t show_properties = B_FALSE;
3145993Smarks 	boolean_t show_permissions = B_FALSE;
315789Sahrens 	FILE *fp = requested ? stdout : stderr;
316789Sahrens 
317789Sahrens 	if (current_command == NULL) {
318789Sahrens 
319789Sahrens 		(void) fprintf(fp, gettext("usage: zfs command args ...\n"));
320789Sahrens 		(void) fprintf(fp,
321789Sahrens 		    gettext("where 'command' is one of the following:\n\n"));
322789Sahrens 
323789Sahrens 		for (i = 0; i < NCOMMAND; i++) {
324789Sahrens 			if (command_table[i].name == NULL)
325789Sahrens 				(void) fprintf(fp, "\n");
326789Sahrens 			else
327789Sahrens 				(void) fprintf(fp, "%s",
3281387Seschrock 				    get_usage(command_table[i].usage));
329789Sahrens 		}
330789Sahrens 
331789Sahrens 		(void) fprintf(fp, gettext("\nEach dataset is of the form: "
332789Sahrens 		    "pool/[dataset/]*dataset[@name]\n"));
333789Sahrens 	} else {
334789Sahrens 		(void) fprintf(fp, gettext("usage:\n"));
3351387Seschrock 		(void) fprintf(fp, "%s", get_usage(current_command->usage));
336789Sahrens 	}
337789Sahrens 
3382190Sdarrenm 	if (current_command != NULL &&
3392190Sdarrenm 	    (strcmp(current_command->name, "set") == 0 ||
340789Sahrens 	    strcmp(current_command->name, "get") == 0 ||
341789Sahrens 	    strcmp(current_command->name, "inherit") == 0 ||
3422190Sdarrenm 	    strcmp(current_command->name, "list") == 0))
3432082Seschrock 		show_properties = B_TRUE;
344789Sahrens 
3455993Smarks 	if (current_command != NULL &&
3465993Smarks 	    (strcmp(current_command->name, "allow") == 0 ||
3475993Smarks 	    strcmp(current_command->name, "unallow") == 0))
3485993Smarks 		show_permissions = B_TRUE;
3495993Smarks 
350789Sahrens 	if (show_properties) {
351789Sahrens 
352789Sahrens 		(void) fprintf(fp,
353789Sahrens 		    gettext("\nThe following properties are supported:\n"));
354789Sahrens 
3555378Sck153898 		(void) fprintf(fp, "\n\t%-14s %s  %s   %s\n\n",
356789Sahrens 		    "PROPERTY", "EDIT", "INHERIT", "VALUES");
357789Sahrens 
3583654Sgw25295 		/* Iterate over all properties */
3595094Slling 		(void) zprop_iter(usage_prop_cb, fp, B_FALSE, B_TRUE,
3605094Slling 		    ZFS_TYPE_DATASET);
3613654Sgw25295 
362789Sahrens 		(void) fprintf(fp, gettext("\nSizes are specified in bytes "
363789Sahrens 		    "with standard units such as K, M, G, etc.\n"));
3647390SMatthew.Ahrens@Sun.COM 		(void) fprintf(fp, gettext("\nUser-defined properties can "
3652676Seschrock 		    "be specified by using a name containing a colon (:).\n"));
3665993Smarks 
3675993Smarks 	} else if (show_permissions) {
3685993Smarks 		(void) fprintf(fp,
3695993Smarks 		    gettext("\nThe following permissions are supported:\n"));
3705993Smarks 
3715993Smarks 		zfs_deleg_permissions();
3722190Sdarrenm 	} else {
3732190Sdarrenm 		/*
3742190Sdarrenm 		 * TRANSLATION NOTE:
3752190Sdarrenm 		 * "zfs set|get" must not be localised this is the
3762190Sdarrenm 		 * command name and arguments.
3772190Sdarrenm 		 */
3785993Smarks 
3792190Sdarrenm 		(void) fprintf(fp,
3802190Sdarrenm 		    gettext("\nFor the property list, run: zfs set|get\n"));
3815993Smarks 
3825993Smarks 		(void) fprintf(fp,
3835993Smarks 		    gettext("\nFor the delegated permission list, run:"
3845993Smarks 		    " zfs allow|unallow\n"));
385789Sahrens 	}
386789Sahrens 
3872676Seschrock 	/*
3882676Seschrock 	 * See comments at end of main().
3892676Seschrock 	 */
3902676Seschrock 	if (getenv("ZFS_ABORT") != NULL) {
3912676Seschrock 		(void) printf("dumping core by request\n");
3922676Seschrock 		abort();
3932676Seschrock 	}
3942676Seschrock 
395789Sahrens 	exit(requested ? 0 : 2);
396789Sahrens }
397789Sahrens 
3987265Sahrens static int
3997265Sahrens parseprop(nvlist_t *props)
4007265Sahrens {
4017265Sahrens 	char *propname = optarg;
4027265Sahrens 	char *propval, *strval;
4037265Sahrens 
4047265Sahrens 	if ((propval = strchr(propname, '=')) == NULL) {
4057265Sahrens 		(void) fprintf(stderr, gettext("missing "
4067265Sahrens 		    "'=' for -o option\n"));
4077265Sahrens 		return (-1);
4087265Sahrens 	}
4097265Sahrens 	*propval = '\0';
4107265Sahrens 	propval++;
4117265Sahrens 	if (nvlist_lookup_string(props, propname, &strval) == 0) {
4127265Sahrens 		(void) fprintf(stderr, gettext("property '%s' "
4137265Sahrens 		    "specified multiple times\n"), propname);
4147265Sahrens 		return (-1);
4157265Sahrens 	}
4167265Sahrens 	if (nvlist_add_string(props, propname, propval) != 0) {
4177265Sahrens 		(void) fprintf(stderr, gettext("internal "
4187265Sahrens 		    "error: out of memory\n"));
4197265Sahrens 		return (-1);
4207265Sahrens 	}
4217265Sahrens 	return (0);
4227265Sahrens 
4237265Sahrens }
4247265Sahrens 
425789Sahrens /*
4267265Sahrens  * zfs clone [-p] [-o prop=value] ... <snap> <fs | vol>
427789Sahrens  *
428789Sahrens  * Given an existing dataset, create a writable copy whose initial contents
429789Sahrens  * are the same as the source.  The newly created dataset maintains a
430789Sahrens  * dependency on the original; the original cannot be destroyed so long as
431789Sahrens  * the clone exists.
4324490Svb160487  *
4334490Svb160487  * The '-p' flag creates all the non-existing ancestors of the target first.
434789Sahrens  */
435789Sahrens static int
436789Sahrens zfs_do_clone(int argc, char **argv)
437789Sahrens {
4387265Sahrens 	zfs_handle_t *zhp = NULL;
4394490Svb160487 	boolean_t parents = B_FALSE;
4407265Sahrens 	nvlist_t *props;
441789Sahrens 	int ret;
4424490Svb160487 	int c;
443789Sahrens 
4447265Sahrens 	if (nvlist_alloc(&props, NV_UNIQUE_NAME, 0) != 0) {
4457265Sahrens 		(void) fprintf(stderr, gettext("internal error: "
4467265Sahrens 		    "out of memory\n"));
4477265Sahrens 		return (1);
4487265Sahrens 	}
4497265Sahrens 
450789Sahrens 	/* check options */
4517265Sahrens 	while ((c = getopt(argc, argv, "o:p")) != -1) {
4524490Svb160487 		switch (c) {
4537265Sahrens 		case 'o':
4547265Sahrens 			if (parseprop(props))
4557265Sahrens 				return (1);
4567265Sahrens 			break;
4574490Svb160487 		case 'p':
4584490Svb160487 			parents = B_TRUE;
4594490Svb160487 			break;
4604490Svb160487 		case '?':
4614490Svb160487 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
4624490Svb160487 			    optopt);
4637265Sahrens 			goto usage;
4644490Svb160487 		}
465789Sahrens 	}
466789Sahrens 
4674490Svb160487 	argc -= optind;
4684490Svb160487 	argv += optind;
4694490Svb160487 
470789Sahrens 	/* check number of arguments */
4714490Svb160487 	if (argc < 1) {
472789Sahrens 		(void) fprintf(stderr, gettext("missing source dataset "
473789Sahrens 		    "argument\n"));
4747265Sahrens 		goto usage;
475789Sahrens 	}
4764490Svb160487 	if (argc < 2) {
477789Sahrens 		(void) fprintf(stderr, gettext("missing target dataset "
478789Sahrens 		    "argument\n"));
4797265Sahrens 		goto usage;
480789Sahrens 	}
4814490Svb160487 	if (argc > 2) {
482789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
4837265Sahrens 		goto usage;
484789Sahrens 	}
485789Sahrens 
486789Sahrens 	/* open the source dataset */
4874490Svb160487 	if ((zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_SNAPSHOT)) == NULL)
488789Sahrens 		return (1);
489789Sahrens 
4904490Svb160487 	if (parents && zfs_name_valid(argv[1], ZFS_TYPE_FILESYSTEM |
4914490Svb160487 	    ZFS_TYPE_VOLUME)) {
4924490Svb160487 		/*
4934490Svb160487 		 * Now create the ancestors of the target dataset.  If the
4944490Svb160487 		 * target already exists and '-p' option was used we should not
4954490Svb160487 		 * complain.
4964490Svb160487 		 */
4974490Svb160487 		if (zfs_dataset_exists(g_zfs, argv[1], ZFS_TYPE_FILESYSTEM |
4984490Svb160487 		    ZFS_TYPE_VOLUME))
4994490Svb160487 			return (0);
5004490Svb160487 		if (zfs_create_ancestors(g_zfs, argv[1]) != 0)
5014490Svb160487 			return (1);
5024490Svb160487 	}
5034490Svb160487 
504789Sahrens 	/* pass to libzfs */
5057265Sahrens 	ret = zfs_clone(zhp, argv[1], props);
506789Sahrens 
507789Sahrens 	/* create the mountpoint if necessary */
508789Sahrens 	if (ret == 0) {
5095094Slling 		zfs_handle_t *clone;
5105094Slling 
5115094Slling 		clone = zfs_open(g_zfs, argv[1], ZFS_TYPE_DATASET);
512789Sahrens 		if (clone != NULL) {
513789Sahrens 			if ((ret = zfs_mount(clone, NULL, 0)) == 0)
514789Sahrens 				ret = zfs_share(clone);
515789Sahrens 			zfs_close(clone);
516789Sahrens 		}
517789Sahrens 	}
518789Sahrens 
519789Sahrens 	zfs_close(zhp);
5207265Sahrens 	nvlist_free(props);
5217265Sahrens 
5227265Sahrens 	return (!!ret);
5237265Sahrens 
5247265Sahrens usage:
5257265Sahrens 	if (zhp)
5267265Sahrens 		zfs_close(zhp);
5277265Sahrens 	nvlist_free(props);
5287265Sahrens 	usage(B_FALSE);
5297265Sahrens 	return (-1);
530789Sahrens }
531789Sahrens 
532789Sahrens /*
5334490Svb160487  * zfs create [-p] [-o prop=value] ... fs
5344490Svb160487  * zfs create [-ps] [-b blocksize] [-o prop=value] ... -V vol size
535789Sahrens  *
536789Sahrens  * Create a new dataset.  This command can be used to create filesystems
537789Sahrens  * and volumes.  Snapshot creation is handled by 'zfs snapshot'.
538789Sahrens  * For volumes, the user must specify a size to be used.
539789Sahrens  *
540789Sahrens  * The '-s' flag applies only to volumes, and indicates that we should not try
541789Sahrens  * to set the reservation for this volume.  By default we set a reservation
5425481Sck153898  * equal to the size for any volume.  For pools with SPA_VERSION >=
5435481Sck153898  * SPA_VERSION_REFRESERVATION, we set a refreservation instead.
5444490Svb160487  *
5454490Svb160487  * The '-p' flag creates all the non-existing ancestors of the target first.
546789Sahrens  */
547789Sahrens static int
548789Sahrens zfs_do_create(int argc, char **argv)
549789Sahrens {
550789Sahrens 	zfs_type_t type = ZFS_TYPE_FILESYSTEM;
5512676Seschrock 	zfs_handle_t *zhp = NULL;
5522676Seschrock 	uint64_t volsize;
553789Sahrens 	int c;
5542082Seschrock 	boolean_t noreserve = B_FALSE;
5555367Sahrens 	boolean_t bflag = B_FALSE;
5564490Svb160487 	boolean_t parents = B_FALSE;
5572676Seschrock 	int ret = 1;
5587265Sahrens 	nvlist_t *props;
5592676Seschrock 	uint64_t intval;
5606168Shs24103 	int canmount;
5612676Seschrock 
5622676Seschrock 	if (nvlist_alloc(&props, NV_UNIQUE_NAME, 0) != 0) {
5632676Seschrock 		(void) fprintf(stderr, gettext("internal error: "
5642676Seschrock 		    "out of memory\n"));
5652676Seschrock 		return (1);
5662676Seschrock 	}
567789Sahrens 
568789Sahrens 	/* check options */
5694490Svb160487 	while ((c = getopt(argc, argv, ":V:b:so:p")) != -1) {
570789Sahrens 		switch (c) {
571789Sahrens 		case 'V':
572789Sahrens 			type = ZFS_TYPE_VOLUME;
5732676Seschrock 			if (zfs_nicestrtonum(g_zfs, optarg, &intval) != 0) {
5742676Seschrock 				(void) fprintf(stderr, gettext("bad volume "
5752676Seschrock 				    "size '%s': %s\n"), optarg,
5762676Seschrock 				    libzfs_error_description(g_zfs));
5772676Seschrock 				goto error;
5782676Seschrock 			}
5792676Seschrock 
5802676Seschrock 			if (nvlist_add_uint64(props,
5812676Seschrock 			    zfs_prop_to_name(ZFS_PROP_VOLSIZE),
5822676Seschrock 			    intval) != 0) {
5832676Seschrock 				(void) fprintf(stderr, gettext("internal "
5842676Seschrock 				    "error: out of memory\n"));
5852676Seschrock 				goto error;
5862676Seschrock 			}
5872676Seschrock 			volsize = intval;
588789Sahrens 			break;
5894490Svb160487 		case 'p':
5904490Svb160487 			parents = B_TRUE;
5914490Svb160487 			break;
592789Sahrens 		case 'b':
5935367Sahrens 			bflag = B_TRUE;
5942676Seschrock 			if (zfs_nicestrtonum(g_zfs, optarg, &intval) != 0) {
5952676Seschrock 				(void) fprintf(stderr, gettext("bad volume "
5962676Seschrock 				    "block size '%s': %s\n"), optarg,
5972676Seschrock 				    libzfs_error_description(g_zfs));
5982676Seschrock 				goto error;
5992676Seschrock 			}
6002676Seschrock 
6012676Seschrock 			if (nvlist_add_uint64(props,
6022676Seschrock 			    zfs_prop_to_name(ZFS_PROP_VOLBLOCKSIZE),
6032676Seschrock 			    intval) != 0) {
6042676Seschrock 				(void) fprintf(stderr, gettext("internal "
6052676Seschrock 				    "error: out of memory\n"));
6062676Seschrock 				goto error;
6072676Seschrock 			}
6082676Seschrock 			break;
6092676Seschrock 		case 'o':
6107265Sahrens 			if (parseprop(props))
6112676Seschrock 				goto error;
612789Sahrens 			break;
613789Sahrens 		case 's':
6142082Seschrock 			noreserve = B_TRUE;
615789Sahrens 			break;
616789Sahrens 		case ':':
617789Sahrens 			(void) fprintf(stderr, gettext("missing size "
618789Sahrens 			    "argument\n"));
6192676Seschrock 			goto badusage;
620789Sahrens 			break;
621789Sahrens 		case '?':
622789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
623789Sahrens 			    optopt);
6242676Seschrock 			goto badusage;
625789Sahrens 		}
626789Sahrens 	}
627789Sahrens 
6285367Sahrens 	if ((bflag || noreserve) && type != ZFS_TYPE_VOLUME) {
6295367Sahrens 		(void) fprintf(stderr, gettext("'-s' and '-b' can only be "
6305367Sahrens 		    "used when creating a volume\n"));
6312676Seschrock 		goto badusage;
632789Sahrens 	}
633789Sahrens 
634789Sahrens 	argc -= optind;
635789Sahrens 	argv += optind;
636789Sahrens 
637789Sahrens 	/* check number of arguments */
638789Sahrens 	if (argc == 0) {
639789Sahrens 		(void) fprintf(stderr, gettext("missing %s argument\n"),
640789Sahrens 		    zfs_type_to_name(type));
6412676Seschrock 		goto badusage;
642789Sahrens 	}
643789Sahrens 	if (argc > 1) {
644789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
6452676Seschrock 		goto badusage;
6462676Seschrock 	}
6472676Seschrock 
6485481Sck153898 	if (type == ZFS_TYPE_VOLUME && !noreserve) {
6495481Sck153898 		zpool_handle_t *zpool_handle;
6505481Sck153898 		uint64_t spa_version;
6515481Sck153898 		char *p;
6525481Sck153898 		zfs_prop_t resv_prop;
6537265Sahrens 		char *strval;
6545481Sck153898 
6555481Sck153898 		if (p = strchr(argv[0], '/'))
6565481Sck153898 			*p = '\0';
6575481Sck153898 		zpool_handle = zpool_open(g_zfs, argv[0]);
6585481Sck153898 		if (p != NULL)
6595481Sck153898 			*p = '/';
6605481Sck153898 		if (zpool_handle == NULL)
6615481Sck153898 			goto error;
6625481Sck153898 		spa_version = zpool_get_prop_int(zpool_handle,
6635481Sck153898 		    ZPOOL_PROP_VERSION, NULL);
6645481Sck153898 		zpool_close(zpool_handle);
6655481Sck153898 		if (spa_version >= SPA_VERSION_REFRESERVATION)
6665481Sck153898 			resv_prop = ZFS_PROP_REFRESERVATION;
6675481Sck153898 		else
6685481Sck153898 			resv_prop = ZFS_PROP_RESERVATION;
6695481Sck153898 
6705481Sck153898 		if (nvlist_lookup_string(props, zfs_prop_to_name(resv_prop),
6715481Sck153898 		    &strval) != 0) {
6725481Sck153898 			if (nvlist_add_uint64(props,
6735481Sck153898 			    zfs_prop_to_name(resv_prop), volsize) != 0) {
6745481Sck153898 				(void) fprintf(stderr, gettext("internal "
6755481Sck153898 				    "error: out of memory\n"));
6765481Sck153898 				nvlist_free(props);
6775481Sck153898 				return (1);
6785481Sck153898 			}
6792676Seschrock 		}
680789Sahrens 	}
681789Sahrens 
6824490Svb160487 	if (parents && zfs_name_valid(argv[0], type)) {
6834490Svb160487 		/*
6844490Svb160487 		 * Now create the ancestors of target dataset.  If the target
6854490Svb160487 		 * already exists and '-p' option was used we should not
6864490Svb160487 		 * complain.
6874490Svb160487 		 */
6884490Svb160487 		if (zfs_dataset_exists(g_zfs, argv[0], type)) {
6894490Svb160487 			ret = 0;
6904490Svb160487 			goto error;
6914490Svb160487 		}
6924490Svb160487 		if (zfs_create_ancestors(g_zfs, argv[0]) != 0)
6934490Svb160487 			goto error;
6944490Svb160487 	}
6954490Svb160487 
696789Sahrens 	/* pass to libzfs */
6972676Seschrock 	if (zfs_create(g_zfs, argv[0], type, props) != 0)
6982676Seschrock 		goto error;
699789Sahrens 
7005094Slling 	if ((zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_DATASET)) == NULL)
7012676Seschrock 		goto error;
7026168Shs24103 	/*
7036168Shs24103 	 * if the user doesn't want the dataset automatically mounted,
7046168Shs24103 	 * then skip the mount/share step
7056168Shs24103 	 */
7066168Shs24103 
7076168Shs24103 	canmount = zfs_prop_get_int(zhp, ZFS_PROP_CANMOUNT);
708789Sahrens 
709789Sahrens 	/*
710789Sahrens 	 * Mount and/or share the new filesystem as appropriate.  We provide a
711789Sahrens 	 * verbose error message to let the user know that their filesystem was
712789Sahrens 	 * in fact created, even if we failed to mount or share it.
713789Sahrens 	 */
7146168Shs24103 	ret = 0;
7156168Shs24103 	if (canmount == ZFS_CANMOUNT_ON) {
7166168Shs24103 		if (zfs_mount(zhp, NULL, 0) != 0) {
7176168Shs24103 			(void) fprintf(stderr, gettext("filesystem "
7186168Shs24103 			    "successfully created, but not mounted\n"));
7196168Shs24103 			ret = 1;
7206168Shs24103 		} else if (zfs_share(zhp) != 0) {
7216168Shs24103 			(void) fprintf(stderr, gettext("filesystem "
7226168Shs24103 			    "successfully created, but not shared\n"));
7236168Shs24103 			ret = 1;
7246168Shs24103 		}
725789Sahrens 	}
726789Sahrens 
7272676Seschrock error:
7282676Seschrock 	if (zhp)
7292676Seschrock 		zfs_close(zhp);
7302676Seschrock 	nvlist_free(props);
731789Sahrens 	return (ret);
7322676Seschrock badusage:
7332676Seschrock 	nvlist_free(props);
7342676Seschrock 	usage(B_FALSE);
7352676Seschrock 	return (2);
736789Sahrens }
737789Sahrens 
738789Sahrens /*
739789Sahrens  * zfs destroy [-rf] <fs, snap, vol>
740789Sahrens  *
741789Sahrens  * 	-r	Recursively destroy all children
742789Sahrens  * 	-R	Recursively destroy all dependents, including clones
743789Sahrens  * 	-f	Force unmounting of any dependents
744789Sahrens  *
745789Sahrens  * Destroys the given dataset.  By default, it will unmount any filesystems,
746789Sahrens  * and refuse to destroy a dataset that has any dependents.  A dependent can
747789Sahrens  * either be a child, or a clone of a child.
748789Sahrens  */
749789Sahrens typedef struct destroy_cbdata {
7502082Seschrock 	boolean_t	cb_first;
751789Sahrens 	int		cb_force;
752789Sahrens 	int		cb_recurse;
753789Sahrens 	int		cb_error;
754789Sahrens 	int		cb_needforce;
755789Sahrens 	int		cb_doclones;
7563265Sahrens 	boolean_t	cb_closezhp;
757789Sahrens 	zfs_handle_t	*cb_target;
7582199Sahrens 	char		*cb_snapname;
759789Sahrens } destroy_cbdata_t;
760789Sahrens 
761789Sahrens /*
762789Sahrens  * Check for any dependents based on the '-r' or '-R' flags.
763789Sahrens  */
764789Sahrens static int
765789Sahrens destroy_check_dependent(zfs_handle_t *zhp, void *data)
766789Sahrens {
767789Sahrens 	destroy_cbdata_t *cbp = data;
768789Sahrens 	const char *tname = zfs_get_name(cbp->cb_target);
769789Sahrens 	const char *name = zfs_get_name(zhp);
770789Sahrens 
771789Sahrens 	if (strncmp(tname, name, strlen(tname)) == 0 &&
772789Sahrens 	    (name[strlen(tname)] == '/' || name[strlen(tname)] == '@')) {
773789Sahrens 		/*
774789Sahrens 		 * This is a direct descendant, not a clone somewhere else in
775789Sahrens 		 * the hierarchy.
776789Sahrens 		 */
777789Sahrens 		if (cbp->cb_recurse)
778789Sahrens 			goto out;
779789Sahrens 
780789Sahrens 		if (cbp->cb_first) {
781789Sahrens 			(void) fprintf(stderr, gettext("cannot destroy '%s': "
782789Sahrens 			    "%s has children\n"),
783789Sahrens 			    zfs_get_name(cbp->cb_target),
784789Sahrens 			    zfs_type_to_name(zfs_get_type(cbp->cb_target)));
785789Sahrens 			(void) fprintf(stderr, gettext("use '-r' to destroy "
786789Sahrens 			    "the following datasets:\n"));
7872082Seschrock 			cbp->cb_first = B_FALSE;
788789Sahrens 			cbp->cb_error = 1;
789789Sahrens 		}
790789Sahrens 
791789Sahrens 		(void) fprintf(stderr, "%s\n", zfs_get_name(zhp));
792789Sahrens 	} else {
793789Sahrens 		/*
794789Sahrens 		 * This is a clone.  We only want to report this if the '-r'
795789Sahrens 		 * wasn't specified, or the target is a snapshot.
796789Sahrens 		 */
797789Sahrens 		if (!cbp->cb_recurse &&
798789Sahrens 		    zfs_get_type(cbp->cb_target) != ZFS_TYPE_SNAPSHOT)
799789Sahrens 			goto out;
800789Sahrens 
801789Sahrens 		if (cbp->cb_first) {
802789Sahrens 			(void) fprintf(stderr, gettext("cannot destroy '%s': "
803789Sahrens 			    "%s has dependent clones\n"),
804789Sahrens 			    zfs_get_name(cbp->cb_target),
805789Sahrens 			    zfs_type_to_name(zfs_get_type(cbp->cb_target)));
806789Sahrens 			(void) fprintf(stderr, gettext("use '-R' to destroy "
807789Sahrens 			    "the following datasets:\n"));
8082082Seschrock 			cbp->cb_first = B_FALSE;
809789Sahrens 			cbp->cb_error = 1;
810789Sahrens 		}
811789Sahrens 
812789Sahrens 		(void) fprintf(stderr, "%s\n", zfs_get_name(zhp));
813789Sahrens 	}
814789Sahrens 
815789Sahrens out:
816789Sahrens 	zfs_close(zhp);
817789Sahrens 	return (0);
818789Sahrens }
819789Sahrens 
820789Sahrens static int
821789Sahrens destroy_callback(zfs_handle_t *zhp, void *data)
822789Sahrens {
823789Sahrens 	destroy_cbdata_t *cbp = data;
824789Sahrens 
825789Sahrens 	/*
826789Sahrens 	 * Ignore pools (which we've already flagged as an error before getting
827789Sahrens 	 * here.
828789Sahrens 	 */
829789Sahrens 	if (strchr(zfs_get_name(zhp), '/') == NULL &&
830789Sahrens 	    zfs_get_type(zhp) == ZFS_TYPE_FILESYSTEM) {
831789Sahrens 		zfs_close(zhp);
832789Sahrens 		return (0);
833789Sahrens 	}
834789Sahrens 
835789Sahrens 	/*
836789Sahrens 	 * Bail out on the first error.
837789Sahrens 	 */
838789Sahrens 	if (zfs_unmount(zhp, NULL, cbp->cb_force ? MS_FORCE : 0) != 0 ||
839789Sahrens 	    zfs_destroy(zhp) != 0) {
840789Sahrens 		zfs_close(zhp);
841789Sahrens 		return (-1);
842789Sahrens 	}
843789Sahrens 
844789Sahrens 	zfs_close(zhp);
845789Sahrens 	return (0);
846789Sahrens }
847789Sahrens 
8482199Sahrens static int
8492199Sahrens destroy_snap_clones(zfs_handle_t *zhp, void *arg)
8502199Sahrens {
8512199Sahrens 	destroy_cbdata_t *cbp = arg;
8522199Sahrens 	char thissnap[MAXPATHLEN];
8532199Sahrens 	zfs_handle_t *szhp;
8543265Sahrens 	boolean_t closezhp = cbp->cb_closezhp;
8553265Sahrens 	int rv;
8562199Sahrens 
8572199Sahrens 	(void) snprintf(thissnap, sizeof (thissnap),
8582199Sahrens 	    "%s@%s", zfs_get_name(zhp), cbp->cb_snapname);
8592199Sahrens 
8602199Sahrens 	libzfs_print_on_error(g_zfs, B_FALSE);
8612199Sahrens 	szhp = zfs_open(g_zfs, thissnap, ZFS_TYPE_SNAPSHOT);
8622199Sahrens 	libzfs_print_on_error(g_zfs, B_TRUE);
8632199Sahrens 	if (szhp) {
8642199Sahrens 		/*
8652199Sahrens 		 * Destroy any clones of this snapshot
8662199Sahrens 		 */
8672474Seschrock 		if (zfs_iter_dependents(szhp, B_FALSE, destroy_callback,
8682474Seschrock 		    cbp) != 0) {
8692474Seschrock 			zfs_close(szhp);
8703265Sahrens 			if (closezhp)
8713265Sahrens 				zfs_close(zhp);
8722474Seschrock 			return (-1);
8732474Seschrock 		}
8742199Sahrens 		zfs_close(szhp);
8752199Sahrens 	}
8762199Sahrens 
8773265Sahrens 	cbp->cb_closezhp = B_TRUE;
8783265Sahrens 	rv = zfs_iter_filesystems(zhp, destroy_snap_clones, arg);
8793265Sahrens 	if (closezhp)
8803265Sahrens 		zfs_close(zhp);
8813265Sahrens 	return (rv);
8822199Sahrens }
883789Sahrens 
884789Sahrens static int
885789Sahrens zfs_do_destroy(int argc, char **argv)
886789Sahrens {
887789Sahrens 	destroy_cbdata_t cb = { 0 };
888789Sahrens 	int c;
889789Sahrens 	zfs_handle_t *zhp;
8902199Sahrens 	char *cp;
891789Sahrens 
892789Sahrens 	/* check options */
893789Sahrens 	while ((c = getopt(argc, argv, "frR")) != -1) {
894789Sahrens 		switch (c) {
895789Sahrens 		case 'f':
896789Sahrens 			cb.cb_force = 1;
897789Sahrens 			break;
898789Sahrens 		case 'r':
899789Sahrens 			cb.cb_recurse = 1;
900789Sahrens 			break;
901789Sahrens 		case 'R':
902789Sahrens 			cb.cb_recurse = 1;
903789Sahrens 			cb.cb_doclones = 1;
904789Sahrens 			break;
905789Sahrens 		case '?':
906789Sahrens 		default:
907789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
908789Sahrens 			    optopt);
9092082Seschrock 			usage(B_FALSE);
910789Sahrens 		}
911789Sahrens 	}
912789Sahrens 
913789Sahrens 	argc -= optind;
914789Sahrens 	argv += optind;
915789Sahrens 
916789Sahrens 	/* check number of arguments */
917789Sahrens 	if (argc == 0) {
918789Sahrens 		(void) fprintf(stderr, gettext("missing path argument\n"));
9192082Seschrock 		usage(B_FALSE);
920789Sahrens 	}
921789Sahrens 	if (argc > 1) {
922789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
9232082Seschrock 		usage(B_FALSE);
924789Sahrens 	}
925789Sahrens 
9262199Sahrens 	/*
9272199Sahrens 	 * If we are doing recursive destroy of a snapshot, then the
9282199Sahrens 	 * named snapshot may not exist.  Go straight to libzfs.
9292199Sahrens 	 */
9302199Sahrens 	if (cb.cb_recurse && (cp = strchr(argv[0], '@'))) {
9312199Sahrens 		int ret;
9322199Sahrens 
9332199Sahrens 		*cp = '\0';
9345094Slling 		if ((zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_DATASET)) == NULL)
9352199Sahrens 			return (1);
9362199Sahrens 		*cp = '@';
9372199Sahrens 		cp++;
9382199Sahrens 
9392199Sahrens 		if (cb.cb_doclones) {
9402199Sahrens 			cb.cb_snapname = cp;
9412474Seschrock 			if (destroy_snap_clones(zhp, &cb) != 0) {
9422474Seschrock 				zfs_close(zhp);
9432474Seschrock 				return (1);
9442474Seschrock 			}
9452199Sahrens 		}
9462199Sahrens 
9472199Sahrens 		ret = zfs_destroy_snaps(zhp, cp);
9482199Sahrens 		zfs_close(zhp);
9492199Sahrens 		if (ret) {
9502199Sahrens 			(void) fprintf(stderr,
9512199Sahrens 			    gettext("no snapshots destroyed\n"));
9522199Sahrens 		}
9532199Sahrens 		return (ret != 0);
9542199Sahrens 	}
9552199Sahrens 
9562199Sahrens 
957789Sahrens 	/* Open the given dataset */
9585094Slling 	if ((zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_DATASET)) == NULL)
959789Sahrens 		return (1);
960789Sahrens 
961789Sahrens 	cb.cb_target = zhp;
962789Sahrens 
963789Sahrens 	/*
964789Sahrens 	 * Perform an explicit check for pools before going any further.
965789Sahrens 	 */
966789Sahrens 	if (!cb.cb_recurse && strchr(zfs_get_name(zhp), '/') == NULL &&
967789Sahrens 	    zfs_get_type(zhp) == ZFS_TYPE_FILESYSTEM) {
968789Sahrens 		(void) fprintf(stderr, gettext("cannot destroy '%s': "
969789Sahrens 		    "operation does not apply to pools\n"),
970789Sahrens 		    zfs_get_name(zhp));
971789Sahrens 		(void) fprintf(stderr, gettext("use 'zfs destroy -r "
972789Sahrens 		    "%s' to destroy all datasets in the pool\n"),
973789Sahrens 		    zfs_get_name(zhp));
974789Sahrens 		(void) fprintf(stderr, gettext("use 'zpool destroy %s' "
975789Sahrens 		    "to destroy the pool itself\n"), zfs_get_name(zhp));
976789Sahrens 		zfs_close(zhp);
977789Sahrens 		return (1);
978789Sahrens 	}
979789Sahrens 
980789Sahrens 	/*
981789Sahrens 	 * Check for any dependents and/or clones.
982789Sahrens 	 */
9832082Seschrock 	cb.cb_first = B_TRUE;
9842474Seschrock 	if (!cb.cb_doclones &&
9852474Seschrock 	    zfs_iter_dependents(zhp, B_TRUE, destroy_check_dependent,
9862474Seschrock 	    &cb) != 0) {
9872474Seschrock 		zfs_close(zhp);
9882474Seschrock 		return (1);
9892474Seschrock 	}
9902474Seschrock 
9912474Seschrock 	if (cb.cb_error ||
9922474Seschrock 	    zfs_iter_dependents(zhp, B_FALSE, destroy_callback, &cb) != 0) {
993789Sahrens 		zfs_close(zhp);
994789Sahrens 		return (1);
995789Sahrens 	}
996789Sahrens 
997789Sahrens 	/*
9982474Seschrock 	 * Do the real thing.  The callback will close the handle regardless of
9992474Seschrock 	 * whether it succeeds or not.
1000789Sahrens 	 */
10014543Smarks 
10022474Seschrock 	if (destroy_callback(zhp, &cb) != 0)
10032474Seschrock 		return (1);
10042474Seschrock 
10052926Sek110237 
10062474Seschrock 	return (0);
1007789Sahrens }
1008789Sahrens 
1009789Sahrens /*
1010866Seschrock  * zfs get [-rHp] [-o field[,field]...] [-s source[,source]...]
1011866Seschrock  * 	< all | property[,property]... > < fs | snap | vol > ...
1012789Sahrens  *
1013789Sahrens  *	-r	recurse over any child datasets
1014789Sahrens  *	-H	scripted mode.  Headers are stripped, and fields are separated
1015789Sahrens  *		by tabs instead of spaces.
1016789Sahrens  *	-o	Set of fields to display.  One of "name,property,value,source".
1017789Sahrens  *		Default is all four.
1018789Sahrens  *	-s	Set of sources to allow.  One of
1019789Sahrens  *		"local,default,inherited,temporary,none".  Default is all
1020789Sahrens  *		five.
1021789Sahrens  *	-p	Display values in parsable (literal) format.
1022789Sahrens  *
1023789Sahrens  *  Prints properties for the given datasets.  The user can control which
1024789Sahrens  *  columns to display as well as which property types to allow.
1025789Sahrens  */
1026789Sahrens 
1027789Sahrens /*
1028789Sahrens  * Invoked to display the properties for a single dataset.
1029789Sahrens  */
1030789Sahrens static int
1031789Sahrens get_callback(zfs_handle_t *zhp, void *data)
1032789Sahrens {
1033789Sahrens 	char buf[ZFS_MAXPROPLEN];
10345094Slling 	zprop_source_t sourcetype;
1035789Sahrens 	char source[ZFS_MAXNAMELEN];
10365094Slling 	zprop_get_cbdata_t *cbp = data;
10372676Seschrock 	nvlist_t *userprop = zfs_get_user_props(zhp);
10385094Slling 	zprop_list_t *pl = cbp->cb_proplist;
10392676Seschrock 	nvlist_t *propval;
10402676Seschrock 	char *strval;
10412676Seschrock 	char *sourceval;
10422676Seschrock 
10432676Seschrock 	for (; pl != NULL; pl = pl->pl_next) {
10442676Seschrock 		/*
10452676Seschrock 		 * Skip the special fake placeholder.  This will also skip over
10462676Seschrock 		 * the name property when 'all' is specified.
10472676Seschrock 		 */
10482676Seschrock 		if (pl->pl_prop == ZFS_PROP_NAME &&
10492676Seschrock 		    pl == cbp->cb_proplist)
10502676Seschrock 			continue;
10512676Seschrock 
10525094Slling 		if (pl->pl_prop != ZPROP_INVAL) {
10532676Seschrock 			if (zfs_prop_get(zhp, pl->pl_prop, buf,
10542676Seschrock 			    sizeof (buf), &sourcetype, source,
10552676Seschrock 			    sizeof (source),
10562676Seschrock 			    cbp->cb_literal) != 0) {
10572676Seschrock 				if (pl->pl_all)
10582676Seschrock 					continue;
10593912Slling 				if (!zfs_prop_valid_for_type(pl->pl_prop,
10605094Slling 				    ZFS_TYPE_DATASET)) {
10613912Slling 					(void) fprintf(stderr,
10623912Slling 					    gettext("No such property '%s'\n"),
10633912Slling 					    zfs_prop_to_name(pl->pl_prop));
10643912Slling 					continue;
10653912Slling 				}
10665094Slling 				sourcetype = ZPROP_SRC_NONE;
10672676Seschrock 				(void) strlcpy(buf, "-", sizeof (buf));
10682676Seschrock 			}
10692676Seschrock 
10705094Slling 			zprop_print_one_property(zfs_get_name(zhp), cbp,
10712676Seschrock 			    zfs_prop_to_name(pl->pl_prop),
10722676Seschrock 			    buf, sourcetype, source);
10732676Seschrock 		} else {
10742676Seschrock 			if (nvlist_lookup_nvlist(userprop,
10752676Seschrock 			    pl->pl_user_prop, &propval) != 0) {
10762676Seschrock 				if (pl->pl_all)
10772676Seschrock 					continue;
10785094Slling 				sourcetype = ZPROP_SRC_NONE;
10792676Seschrock 				strval = "-";
10802676Seschrock 			} else {
10812676Seschrock 				verify(nvlist_lookup_string(propval,
10825094Slling 				    ZPROP_VALUE, &strval) == 0);
10832676Seschrock 				verify(nvlist_lookup_string(propval,
10845094Slling 				    ZPROP_SOURCE, &sourceval) == 0);
10852676Seschrock 
10862676Seschrock 				if (strcmp(sourceval,
10872676Seschrock 				    zfs_get_name(zhp)) == 0) {
10885094Slling 					sourcetype = ZPROP_SRC_LOCAL;
10892676Seschrock 				} else {
10905094Slling 					sourcetype = ZPROP_SRC_INHERITED;
10912676Seschrock 					(void) strlcpy(source,
10922676Seschrock 					    sourceval, sizeof (source));
10932676Seschrock 				}
10942676Seschrock 			}
10952676Seschrock 
10965094Slling 			zprop_print_one_property(zfs_get_name(zhp), cbp,
10972676Seschrock 			    pl->pl_user_prop, strval, sourcetype,
10982676Seschrock 			    source);
1099866Seschrock 		}
1100789Sahrens 	}
1101789Sahrens 
1102789Sahrens 	return (0);
1103789Sahrens }
1104789Sahrens 
1105789Sahrens static int
1106789Sahrens zfs_do_get(int argc, char **argv)
1107789Sahrens {
11085094Slling 	zprop_get_cbdata_t cb = { 0 };
1109*7538SRichard.Morris@Sun.COM 	int i, c, flags = 0;
11102676Seschrock 	char *value, *fields;
1111866Seschrock 	int ret;
11125094Slling 	zprop_list_t fake_name = { 0 };
1113789Sahrens 
1114789Sahrens 	/*
1115789Sahrens 	 * Set up default columns and sources.
1116789Sahrens 	 */
11175094Slling 	cb.cb_sources = ZPROP_SRC_ALL;
1118789Sahrens 	cb.cb_columns[0] = GET_COL_NAME;
1119789Sahrens 	cb.cb_columns[1] = GET_COL_PROPERTY;
1120789Sahrens 	cb.cb_columns[2] = GET_COL_VALUE;
1121789Sahrens 	cb.cb_columns[3] = GET_COL_SOURCE;
11225094Slling 	cb.cb_type = ZFS_TYPE_DATASET;
1123789Sahrens 
1124789Sahrens 	/* check options */
1125789Sahrens 	while ((c = getopt(argc, argv, ":o:s:rHp")) != -1) {
1126789Sahrens 		switch (c) {
1127789Sahrens 		case 'p':
11282082Seschrock 			cb.cb_literal = B_TRUE;
1129789Sahrens 			break;
1130789Sahrens 		case 'r':
1131*7538SRichard.Morris@Sun.COM 			flags |= ZFS_ITER_RECURSE;
1132789Sahrens 			break;
1133789Sahrens 		case 'H':
11342082Seschrock 			cb.cb_scripted = B_TRUE;
1135789Sahrens 			break;
1136789Sahrens 		case ':':
1137789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
1138789Sahrens 			    "'%c' option\n"), optopt);
11392082Seschrock 			usage(B_FALSE);
1140789Sahrens 			break;
1141789Sahrens 		case 'o':
1142789Sahrens 			/*
1143789Sahrens 			 * Process the set of columns to display.  We zero out
1144789Sahrens 			 * the structure to give us a blank slate.
1145789Sahrens 			 */
1146789Sahrens 			bzero(&cb.cb_columns, sizeof (cb.cb_columns));
1147789Sahrens 			i = 0;
1148789Sahrens 			while (*optarg != '\0') {
1149789Sahrens 				static char *col_subopts[] =
1150789Sahrens 				    { "name", "property", "value", "source",
1151789Sahrens 				    NULL };
1152789Sahrens 
1153789Sahrens 				if (i == 4) {
1154789Sahrens 					(void) fprintf(stderr, gettext("too "
1155789Sahrens 					    "many fields given to -o "
1156789Sahrens 					    "option\n"));
11572082Seschrock 					usage(B_FALSE);
1158789Sahrens 				}
1159789Sahrens 
1160789Sahrens 				switch (getsubopt(&optarg, col_subopts,
1161789Sahrens 				    &value)) {
1162789Sahrens 				case 0:
1163789Sahrens 					cb.cb_columns[i++] = GET_COL_NAME;
1164789Sahrens 					break;
1165789Sahrens 				case 1:
1166789Sahrens 					cb.cb_columns[i++] = GET_COL_PROPERTY;
1167789Sahrens 					break;
1168789Sahrens 				case 2:
1169789Sahrens 					cb.cb_columns[i++] = GET_COL_VALUE;
1170789Sahrens 					break;
1171789Sahrens 				case 3:
1172789Sahrens 					cb.cb_columns[i++] = GET_COL_SOURCE;
1173789Sahrens 					break;
1174789Sahrens 				default:
1175789Sahrens 					(void) fprintf(stderr,
1176789Sahrens 					    gettext("invalid column name "
1177789Sahrens 					    "'%s'\n"), value);
11783912Slling 					usage(B_FALSE);
1179789Sahrens 				}
1180789Sahrens 			}
1181789Sahrens 			break;
1182789Sahrens 
1183789Sahrens 		case 's':
1184789Sahrens 			cb.cb_sources = 0;
1185789Sahrens 			while (*optarg != '\0') {
1186789Sahrens 				static char *source_subopts[] = {
1187789Sahrens 					"local", "default", "inherited",
1188789Sahrens 					"temporary", "none", NULL };
1189789Sahrens 
1190789Sahrens 				switch (getsubopt(&optarg, source_subopts,
1191789Sahrens 				    &value)) {
1192789Sahrens 				case 0:
11935094Slling 					cb.cb_sources |= ZPROP_SRC_LOCAL;
1194789Sahrens 					break;
1195789Sahrens 				case 1:
11965094Slling 					cb.cb_sources |= ZPROP_SRC_DEFAULT;
1197789Sahrens 					break;
1198789Sahrens 				case 2:
11995094Slling 					cb.cb_sources |= ZPROP_SRC_INHERITED;
1200789Sahrens 					break;
1201789Sahrens 				case 3:
12025094Slling 					cb.cb_sources |= ZPROP_SRC_TEMPORARY;
1203789Sahrens 					break;
1204789Sahrens 				case 4:
12055094Slling 					cb.cb_sources |= ZPROP_SRC_NONE;
1206789Sahrens 					break;
1207789Sahrens 				default:
1208789Sahrens 					(void) fprintf(stderr,
1209789Sahrens 					    gettext("invalid source "
1210789Sahrens 					    "'%s'\n"), value);
12113912Slling 					usage(B_FALSE);
1212789Sahrens 				}
1213789Sahrens 			}
1214789Sahrens 			break;
1215789Sahrens 
1216789Sahrens 		case '?':
1217789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
1218789Sahrens 			    optopt);
12192082Seschrock 			usage(B_FALSE);
1220789Sahrens 		}
1221789Sahrens 	}
1222789Sahrens 
1223789Sahrens 	argc -= optind;
1224789Sahrens 	argv += optind;
1225789Sahrens 
1226789Sahrens 	if (argc < 1) {
1227789Sahrens 		(void) fprintf(stderr, gettext("missing property "
1228789Sahrens 		    "argument\n"));
12292082Seschrock 		usage(B_FALSE);
1230789Sahrens 	}
1231789Sahrens 
1232789Sahrens 	fields = argv[0];
1233789Sahrens 
12345094Slling 	if (zprop_get_list(g_zfs, fields, &cb.cb_proplist, ZFS_TYPE_DATASET)
12355094Slling 	    != 0)
12362082Seschrock 		usage(B_FALSE);
1237789Sahrens 
1238789Sahrens 	argc--;
1239789Sahrens 	argv++;
1240789Sahrens 
1241789Sahrens 	/*
12422676Seschrock 	 * As part of zfs_expand_proplist(), we keep track of the maximum column
12432676Seschrock 	 * width for each property.  For the 'NAME' (and 'SOURCE') columns, we
12442676Seschrock 	 * need to know the maximum name length.  However, the user likely did
12452676Seschrock 	 * not specify 'name' as one of the properties to fetch, so we need to
12462676Seschrock 	 * make sure we always include at least this property for
12472676Seschrock 	 * print_get_headers() to work properly.
1248789Sahrens 	 */
12492676Seschrock 	if (cb.cb_proplist != NULL) {
12502676Seschrock 		fake_name.pl_prop = ZFS_PROP_NAME;
12512676Seschrock 		fake_name.pl_width = strlen(gettext("NAME"));
12522676Seschrock 		fake_name.pl_next = cb.cb_proplist;
12532676Seschrock 		cb.cb_proplist = &fake_name;
1254789Sahrens 	}
1255789Sahrens 
12562676Seschrock 	cb.cb_first = B_TRUE;
12572676Seschrock 
1258789Sahrens 	/* run for each object */
1259*7538SRichard.Morris@Sun.COM 	ret = zfs_for_each(argc, argv, flags, ZFS_TYPE_DATASET, NULL,
1260*7538SRichard.Morris@Sun.COM 	    &cb.cb_proplist, get_callback, &cb);
12612676Seschrock 
12622676Seschrock 	if (cb.cb_proplist == &fake_name)
12635094Slling 		zprop_free_list(fake_name.pl_next);
12642676Seschrock 	else
12655094Slling 		zprop_free_list(cb.cb_proplist);
12662676Seschrock 
12672676Seschrock 	return (ret);
1268789Sahrens }
1269789Sahrens 
1270789Sahrens /*
1271789Sahrens  * inherit [-r] <property> <fs|vol> ...
1272789Sahrens  *
1273789Sahrens  * 	-r	Recurse over all children
1274789Sahrens  *
1275789Sahrens  * For each dataset specified on the command line, inherit the given property
1276789Sahrens  * from its parent.  Inheriting a property at the pool level will cause it to
1277789Sahrens  * use the default value.  The '-r' flag will recurse over all children, and is
1278789Sahrens  * useful for setting a property on a hierarchy-wide basis, regardless of any
1279789Sahrens  * local modifications for each dataset.
1280789Sahrens  */
12812926Sek110237 
1282789Sahrens static int
12837265Sahrens inherit_recurse_cb(zfs_handle_t *zhp, void *data)
1284789Sahrens {
12854543Smarks 	char *propname = data;
12867265Sahrens 	zfs_prop_t prop = zfs_name_to_prop(propname);
12877265Sahrens 
12887265Sahrens 	/*
12897265Sahrens 	 * If we're doing it recursively, then ignore properties that
12907265Sahrens 	 * are not valid for this type of dataset.
12917265Sahrens 	 */
12927265Sahrens 	if (prop != ZPROP_INVAL &&
12937265Sahrens 	    !zfs_prop_valid_for_type(prop, zfs_get_type(zhp)))
12947265Sahrens 		return (0);
12957265Sahrens 
12967265Sahrens 	return (zfs_prop_inherit(zhp, propname) != 0);
12977265Sahrens }
12987265Sahrens 
12997265Sahrens static int
13007265Sahrens inherit_cb(zfs_handle_t *zhp, void *data)
13017265Sahrens {
13027265Sahrens 	char *propname = data;
13037265Sahrens 
13047265Sahrens 	return (zfs_prop_inherit(zhp, propname) != 0);
1305789Sahrens }
1306789Sahrens 
1307789Sahrens static int
1308789Sahrens zfs_do_inherit(int argc, char **argv)
1309789Sahrens {
1310789Sahrens 	int c;
1311789Sahrens 	zfs_prop_t prop;
13124543Smarks 	char *propname;
13132926Sek110237 	int ret;
1314*7538SRichard.Morris@Sun.COM 	int flags = 0;
1315789Sahrens 
1316789Sahrens 	/* check options */
1317789Sahrens 	while ((c = getopt(argc, argv, "r")) != -1) {
1318789Sahrens 		switch (c) {
1319789Sahrens 		case 'r':
1320*7538SRichard.Morris@Sun.COM 			flags |= ZFS_ITER_RECURSE;
1321789Sahrens 			break;
1322789Sahrens 		case '?':
1323789Sahrens 		default:
1324789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
1325789Sahrens 			    optopt);
13262082Seschrock 			usage(B_FALSE);
1327789Sahrens 		}
1328789Sahrens 	}
1329789Sahrens 
1330789Sahrens 	argc -= optind;
1331789Sahrens 	argv += optind;
1332789Sahrens 
1333789Sahrens 	/* check number of arguments */
1334789Sahrens 	if (argc < 1) {
1335789Sahrens 		(void) fprintf(stderr, gettext("missing property argument\n"));
13362082Seschrock 		usage(B_FALSE);
1337789Sahrens 	}
1338789Sahrens 	if (argc < 2) {
1339789Sahrens 		(void) fprintf(stderr, gettext("missing dataset argument\n"));
13402082Seschrock 		usage(B_FALSE);
1341789Sahrens 	}
1342789Sahrens 
13434543Smarks 	propname = argv[0];
13442676Seschrock 	argc--;
13452676Seschrock 	argv++;
13462676Seschrock 
13475094Slling 	if ((prop = zfs_name_to_prop(propname)) != ZPROP_INVAL) {
13482676Seschrock 		if (zfs_prop_readonly(prop)) {
13492676Seschrock 			(void) fprintf(stderr, gettext(
13502676Seschrock 			    "%s property is read-only\n"),
13514543Smarks 			    propname);
13522676Seschrock 			return (1);
13532676Seschrock 		}
13542676Seschrock 		if (!zfs_prop_inheritable(prop)) {
13552676Seschrock 			(void) fprintf(stderr, gettext("'%s' property cannot "
13564543Smarks 			    "be inherited\n"), propname);
13572676Seschrock 			if (prop == ZFS_PROP_QUOTA ||
13585378Sck153898 			    prop == ZFS_PROP_RESERVATION ||
13595378Sck153898 			    prop == ZFS_PROP_REFQUOTA ||
13605378Sck153898 			    prop == ZFS_PROP_REFRESERVATION)
13612676Seschrock 				(void) fprintf(stderr, gettext("use 'zfs set "
13624543Smarks 				    "%s=none' to clear\n"), propname);
13632676Seschrock 			return (1);
13642676Seschrock 		}
13654543Smarks 	} else if (!zfs_prop_user(propname)) {
13664543Smarks 		(void) fprintf(stderr, gettext("invalid property '%s'\n"),
13674543Smarks 		    propname);
13682082Seschrock 		usage(B_FALSE);
1369789Sahrens 	}
13702676Seschrock 
1371*7538SRichard.Morris@Sun.COM 	if (flags & ZFS_ITER_RECURSE) {
1372*7538SRichard.Morris@Sun.COM 		ret = zfs_for_each(argc, argv, flags, ZFS_TYPE_DATASET,
1373*7538SRichard.Morris@Sun.COM 		    NULL, NULL, inherit_recurse_cb, propname);
13747265Sahrens 	} else {
1375*7538SRichard.Morris@Sun.COM 		ret = zfs_for_each(argc, argv, flags, ZFS_TYPE_DATASET,
1376*7538SRichard.Morris@Sun.COM 		    NULL, NULL, inherit_cb, propname);
13777265Sahrens 	}
13782926Sek110237 
13792926Sek110237 	return (ret);
1380789Sahrens }
1381789Sahrens 
13824577Sahrens typedef struct upgrade_cbdata {
13834577Sahrens 	uint64_t cb_numupgraded;
13844577Sahrens 	uint64_t cb_numsamegraded;
13854849Sahrens 	uint64_t cb_numfailed;
13864577Sahrens 	uint64_t cb_version;
13874577Sahrens 	boolean_t cb_newer;
13884577Sahrens 	boolean_t cb_foundone;
13894577Sahrens 	char cb_lastfs[ZFS_MAXNAMELEN];
13904577Sahrens } upgrade_cbdata_t;
13914577Sahrens 
13924577Sahrens static int
13934577Sahrens same_pool(zfs_handle_t *zhp, const char *name)
13944577Sahrens {
13954577Sahrens 	int len1 = strcspn(name, "/@");
13964577Sahrens 	const char *zhname = zfs_get_name(zhp);
13974577Sahrens 	int len2 = strcspn(zhname, "/@");
13984577Sahrens 
13994577Sahrens 	if (len1 != len2)
14004577Sahrens 		return (B_FALSE);
14014988Sek110237 	return (strncmp(name, zhname, len1) == 0);
14024577Sahrens }
14034577Sahrens 
14044577Sahrens static int
14054577Sahrens upgrade_list_callback(zfs_handle_t *zhp, void *data)
14064577Sahrens {
14074577Sahrens 	upgrade_cbdata_t *cb = data;
14084577Sahrens 	int version = zfs_prop_get_int(zhp, ZFS_PROP_VERSION);
14094577Sahrens 
14104577Sahrens 	/* list if it's old/new */
14114577Sahrens 	if ((!cb->cb_newer && version < ZPL_VERSION) ||
14125367Sahrens 	    (cb->cb_newer && version > ZPL_VERSION)) {
14134577Sahrens 		char *str;
14144577Sahrens 		if (cb->cb_newer) {
14154577Sahrens 			str = gettext("The following filesystems are "
14164577Sahrens 			    "formatted using a newer software version and\n"
14174577Sahrens 			    "cannot be accessed on the current system.\n\n");
14184577Sahrens 		} else {
14194577Sahrens 			str = gettext("The following filesystems are "
14204577Sahrens 			    "out of date, and can be upgraded.  After being\n"
14214577Sahrens 			    "upgraded, these filesystems (and any 'zfs send' "
14224577Sahrens 			    "streams generated from\n"
14234577Sahrens 			    "subsequent snapshots) will no longer be "
14244577Sahrens 			    "accessible by older software versions.\n\n");
14254577Sahrens 		}
14264577Sahrens 
14274577Sahrens 		if (!cb->cb_foundone) {
14284577Sahrens 			(void) puts(str);
14294577Sahrens 			(void) printf(gettext("VER  FILESYSTEM\n"));
14304577Sahrens 			(void) printf(gettext("---  ------------\n"));
14314577Sahrens 			cb->cb_foundone = B_TRUE;
14324577Sahrens 		}
14334577Sahrens 
14344577Sahrens 		(void) printf("%2u   %s\n", version, zfs_get_name(zhp));
14354577Sahrens 	}
14364577Sahrens 
14374577Sahrens 	return (0);
14384577Sahrens }
14394577Sahrens 
14404577Sahrens static int
14414577Sahrens upgrade_set_callback(zfs_handle_t *zhp, void *data)
14424577Sahrens {
14434577Sahrens 	upgrade_cbdata_t *cb = data;
14444577Sahrens 	int version = zfs_prop_get_int(zhp, ZFS_PROP_VERSION);
14454577Sahrens 
14465331Samw 	if (cb->cb_version >= ZPL_VERSION_FUID) {
14475331Samw 		int spa_version;
14485713Srm160521 
14495713Srm160521 		if (zfs_spa_version(zhp, &spa_version) < 0)
14505331Samw 			return (-1);
14515331Samw 
14525331Samw 		if (spa_version < SPA_VERSION_FUID) {
14535331Samw 			/* can't upgrade */
14545331Samw 			(void) printf(gettext("%s: can not be upgraded; "
14555331Samw 			    "the pool version needs to first be upgraded\nto "
14565331Samw 			    "version %d\n\n"),
14575331Samw 			    zfs_get_name(zhp), SPA_VERSION_FUID);
14585331Samw 			cb->cb_numfailed++;
14595331Samw 			return (0);
14605331Samw 		}
14615331Samw 	}
14625331Samw 
14634577Sahrens 	/* upgrade */
14644577Sahrens 	if (version < cb->cb_version) {
14654577Sahrens 		char verstr[16];
14664849Sahrens 		(void) snprintf(verstr, sizeof (verstr),
14674849Sahrens 		    "%llu", cb->cb_version);
14684577Sahrens 		if (cb->cb_lastfs[0] && !same_pool(zhp, cb->cb_lastfs)) {
14694577Sahrens 			/*
14704577Sahrens 			 * If they did "zfs upgrade -a", then we could
14714577Sahrens 			 * be doing ioctls to different pools.  We need
14724577Sahrens 			 * to log this history once to each pool.
14734577Sahrens 			 */
14744988Sek110237 			verify(zpool_stage_history(g_zfs, history_str) == 0);
14754577Sahrens 		}
14764577Sahrens 		if (zfs_prop_set(zhp, "version", verstr) == 0)
14774577Sahrens 			cb->cb_numupgraded++;
14784849Sahrens 		else
14794849Sahrens 			cb->cb_numfailed++;
14804577Sahrens 		(void) strcpy(cb->cb_lastfs, zfs_get_name(zhp));
14814577Sahrens 	} else if (version > cb->cb_version) {
14824577Sahrens 		/* can't downgrade */
14834577Sahrens 		(void) printf(gettext("%s: can not be downgraded; "
14844577Sahrens 		    "it is already at version %u\n"),
14854577Sahrens 		    zfs_get_name(zhp), version);
14864849Sahrens 		cb->cb_numfailed++;
14874577Sahrens 	} else {
14884577Sahrens 		cb->cb_numsamegraded++;
14894577Sahrens 	}
14904577Sahrens 	return (0);
14914577Sahrens }
14924577Sahrens 
14934577Sahrens /*
14944577Sahrens  * zfs upgrade
14954577Sahrens  * zfs upgrade -v
14964577Sahrens  * zfs upgrade [-r] [-V <version>] <-a | filesystem>
14974577Sahrens  */
14984577Sahrens static int
14994577Sahrens zfs_do_upgrade(int argc, char **argv)
15004577Sahrens {
15014577Sahrens 	boolean_t all = B_FALSE;
15024577Sahrens 	boolean_t showversions = B_FALSE;
15034577Sahrens 	int ret;
15044577Sahrens 	upgrade_cbdata_t cb = { 0 };
15054577Sahrens 	char c;
1506*7538SRichard.Morris@Sun.COM 	int flags = ZFS_ITER_ARGS_CAN_BE_PATHS;
15074577Sahrens 
15084577Sahrens 	/* check options */
15094577Sahrens 	while ((c = getopt(argc, argv, "rvV:a")) != -1) {
15104577Sahrens 		switch (c) {
15114577Sahrens 		case 'r':
1512*7538SRichard.Morris@Sun.COM 			flags |= ZFS_ITER_RECURSE;
15134577Sahrens 			break;
15144577Sahrens 		case 'v':
15154577Sahrens 			showversions = B_TRUE;
15164577Sahrens 			break;
15174577Sahrens 		case 'V':
15184577Sahrens 			if (zfs_prop_string_to_index(ZFS_PROP_VERSION,
15194577Sahrens 			    optarg, &cb.cb_version) != 0) {
15204577Sahrens 				(void) fprintf(stderr,
15214577Sahrens 				    gettext("invalid version %s\n"), optarg);
15224577Sahrens 				usage(B_FALSE);
15234577Sahrens 			}
15244577Sahrens 			break;
15254577Sahrens 		case 'a':
15264577Sahrens 			all = B_TRUE;
15274577Sahrens 			break;
15284577Sahrens 		case '?':
15294577Sahrens 		default:
15304577Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
15314577Sahrens 			    optopt);
15324577Sahrens 			usage(B_FALSE);
15334577Sahrens 		}
15344577Sahrens 	}
15354577Sahrens 
15364577Sahrens 	argc -= optind;
15374577Sahrens 	argv += optind;
15384577Sahrens 
1539*7538SRichard.Morris@Sun.COM 	if ((!all && !argc) && ((flags & ZFS_ITER_RECURSE) | cb.cb_version))
15404577Sahrens 		usage(B_FALSE);
1541*7538SRichard.Morris@Sun.COM 	if (showversions && (flags & ZFS_ITER_RECURSE || all ||
1542*7538SRichard.Morris@Sun.COM 	    cb.cb_version || argc))
15434577Sahrens 		usage(B_FALSE);
15444577Sahrens 	if ((all || argc) && (showversions))
15454577Sahrens 		usage(B_FALSE);
15464577Sahrens 	if (all && argc)
15474577Sahrens 		usage(B_FALSE);
15484577Sahrens 
15494577Sahrens 	if (showversions) {
15504577Sahrens 		/* Show info on available versions. */
15514577Sahrens 		(void) printf(gettext("The following filesystem versions are "
15524577Sahrens 		    "supported:\n\n"));
15534577Sahrens 		(void) printf(gettext("VER  DESCRIPTION\n"));
15544577Sahrens 		(void) printf("---  -----------------------------------------"
15554577Sahrens 		    "---------------\n");
15564577Sahrens 		(void) printf(gettext(" 1   Initial ZFS filesystem version\n"));
15574577Sahrens 		(void) printf(gettext(" 2   Enhanced directory entries\n"));
15585331Samw 		(void) printf(gettext(" 3   Case insensitive and File system "
15595331Samw 		    "unique identifer (FUID)\n"));
15604577Sahrens 		(void) printf(gettext("\nFor more information on a particular "
15614577Sahrens 		    "version, including supported releases, see:\n\n"));
15624577Sahrens 		(void) printf("http://www.opensolaris.org/os/community/zfs/"
15634577Sahrens 		    "version/zpl/N\n\n");
15644577Sahrens 		(void) printf(gettext("Where 'N' is the version number.\n"));
15654577Sahrens 		ret = 0;
15664577Sahrens 	} else if (argc || all) {
15674577Sahrens 		/* Upgrade filesystems */
15684577Sahrens 		if (cb.cb_version == 0)
15694577Sahrens 			cb.cb_version = ZPL_VERSION;
1570*7538SRichard.Morris@Sun.COM 		ret = zfs_for_each(argc, argv, flags, ZFS_TYPE_FILESYSTEM,
1571*7538SRichard.Morris@Sun.COM 		    NULL, NULL, upgrade_set_callback, &cb);
15724577Sahrens 		(void) printf(gettext("%llu filesystems upgraded\n"),
15734577Sahrens 		    cb.cb_numupgraded);
15744577Sahrens 		if (cb.cb_numsamegraded) {
15754577Sahrens 			(void) printf(gettext("%llu filesystems already at "
15764577Sahrens 			    "this version\n"),
15774577Sahrens 			    cb.cb_numsamegraded);
15784577Sahrens 		}
15794849Sahrens 		if (cb.cb_numfailed != 0)
15804577Sahrens 			ret = 1;
15814577Sahrens 	} else {
15824577Sahrens 		/* List old-version filesytems */
15834577Sahrens 		boolean_t found;
15844577Sahrens 		(void) printf(gettext("This system is currently running "
15854577Sahrens 		    "ZFS filesystem version %llu.\n\n"), ZPL_VERSION);
15864577Sahrens 
1587*7538SRichard.Morris@Sun.COM 		flags |= ZFS_ITER_RECURSE;
1588*7538SRichard.Morris@Sun.COM 		ret = zfs_for_each(0, NULL, flags, ZFS_TYPE_FILESYSTEM,
1589*7538SRichard.Morris@Sun.COM 		    NULL, NULL, upgrade_list_callback, &cb);
15904577Sahrens 
15914577Sahrens 		found = cb.cb_foundone;
15924577Sahrens 		cb.cb_foundone = B_FALSE;
15934577Sahrens 		cb.cb_newer = B_TRUE;
15944577Sahrens 
1595*7538SRichard.Morris@Sun.COM 		ret = zfs_for_each(0, NULL, flags, ZFS_TYPE_FILESYSTEM,
1596*7538SRichard.Morris@Sun.COM 		    NULL, NULL, upgrade_list_callback, &cb);
15974577Sahrens 
15984577Sahrens 		if (!cb.cb_foundone && !found) {
15994577Sahrens 			(void) printf(gettext("All filesystems are "
16004577Sahrens 			    "formatted with the current version.\n"));
16014577Sahrens 		}
16024577Sahrens 	}
16034577Sahrens 
16044577Sahrens 	return (ret);
16054577Sahrens }
16064577Sahrens 
1607789Sahrens /*
16082379Ssjelinek  * list [-rH] [-o property[,property]...] [-t type[,type]...]
16092379Ssjelinek  *      [-s property [-s property]...] [-S property [-S property]...]
16102379Ssjelinek  *      <dataset> ...
1611789Sahrens  *
1612789Sahrens  * 	-r	Recurse over all children
16135331Samw  * 	-H	Scripted mode; elide headers and separate columns by tabs
1614789Sahrens  * 	-o	Control which fields to display.
1615866Seschrock  * 	-t	Control which object types to display.
16162379Ssjelinek  *	-s	Specify sort columns, descending order.
16172379Ssjelinek  *	-S	Specify sort columns, ascending order.
1618789Sahrens  *
1619789Sahrens  * When given no arguments, lists all filesystems in the system.
1620789Sahrens  * Otherwise, list the specified datasets, optionally recursing down them if
1621789Sahrens  * '-r' is specified.
1622789Sahrens  */
1623789Sahrens typedef struct list_cbdata {
16242082Seschrock 	boolean_t	cb_first;
16252082Seschrock 	boolean_t	cb_scripted;
16265094Slling 	zprop_list_t	*cb_proplist;
1627789Sahrens } list_cbdata_t;
1628789Sahrens 
1629789Sahrens /*
1630789Sahrens  * Given a list of columns to display, output appropriate headers for each one.
1631789Sahrens  */
1632789Sahrens static void
16335094Slling print_header(zprop_list_t *pl)
1634789Sahrens {
16352676Seschrock 	char headerbuf[ZFS_MAXPROPLEN];
16362676Seschrock 	const char *header;
1637789Sahrens 	int i;
16382676Seschrock 	boolean_t first = B_TRUE;
16392676Seschrock 	boolean_t right_justify;
16402676Seschrock 
16412676Seschrock 	for (; pl != NULL; pl = pl->pl_next) {
16422676Seschrock 		if (!first) {
1643789Sahrens 			(void) printf("  ");
16442676Seschrock 		} else {
16452676Seschrock 			first = B_FALSE;
16462676Seschrock 		}
16472676Seschrock 
16482676Seschrock 		right_justify = B_FALSE;
16495094Slling 		if (pl->pl_prop != ZPROP_INVAL) {
16502676Seschrock 			header = zfs_prop_column_name(pl->pl_prop);
16512676Seschrock 			right_justify = zfs_prop_align_right(pl->pl_prop);
16522676Seschrock 		} else {
16532676Seschrock 			for (i = 0; pl->pl_user_prop[i] != '\0'; i++)
16542676Seschrock 				headerbuf[i] = toupper(pl->pl_user_prop[i]);
16552676Seschrock 			headerbuf[i] = '\0';
16562676Seschrock 			header = headerbuf;
16572676Seschrock 		}
16582676Seschrock 
16592676Seschrock 		if (pl->pl_next == NULL && !right_justify)
16602676Seschrock 			(void) printf("%s", header);
16612676Seschrock 		else if (right_justify)
16622676Seschrock 			(void) printf("%*s", pl->pl_width, header);
16632676Seschrock 		else
16642676Seschrock 			(void) printf("%-*s", pl->pl_width, header);
1665789Sahrens 	}
1666789Sahrens 
1667789Sahrens 	(void) printf("\n");
1668789Sahrens }
1669789Sahrens 
1670789Sahrens /*
1671789Sahrens  * Given a dataset and a list of fields, print out all the properties according
1672789Sahrens  * to the described layout.
1673789Sahrens  */
1674789Sahrens static void
16755094Slling print_dataset(zfs_handle_t *zhp, zprop_list_t *pl, boolean_t scripted)
1676789Sahrens {
16772676Seschrock 	boolean_t first = B_TRUE;
1678789Sahrens 	char property[ZFS_MAXPROPLEN];
16792676Seschrock 	nvlist_t *userprops = zfs_get_user_props(zhp);
16802676Seschrock 	nvlist_t *propval;
16812676Seschrock 	char *propstr;
16822676Seschrock 	boolean_t right_justify;
16832676Seschrock 	int width;
16842676Seschrock 
16852676Seschrock 	for (; pl != NULL; pl = pl->pl_next) {
16862676Seschrock 		if (!first) {
1687789Sahrens 			if (scripted)
1688789Sahrens 				(void) printf("\t");
1689789Sahrens 			else
1690789Sahrens 				(void) printf("  ");
16912676Seschrock 		} else {
16922676Seschrock 			first = B_FALSE;
1693789Sahrens 		}
1694789Sahrens 
16952676Seschrock 		right_justify = B_FALSE;
16965094Slling 		if (pl->pl_prop != ZPROP_INVAL) {
16972676Seschrock 			if (zfs_prop_get(zhp, pl->pl_prop, property,
16982676Seschrock 			    sizeof (property), NULL, NULL, 0, B_FALSE) != 0)
16992676Seschrock 				propstr = "-";
17002676Seschrock 			else
17012676Seschrock 				propstr = property;
17022676Seschrock 
17032676Seschrock 			right_justify = zfs_prop_align_right(pl->pl_prop);
17042676Seschrock 		} else {
17052676Seschrock 			if (nvlist_lookup_nvlist(userprops,
17062676Seschrock 			    pl->pl_user_prop, &propval) != 0)
17072676Seschrock 				propstr = "-";
17082676Seschrock 			else
17092676Seschrock 				verify(nvlist_lookup_string(propval,
17105094Slling 				    ZPROP_VALUE, &propstr) == 0);
17112676Seschrock 		}
17122676Seschrock 
17132676Seschrock 		width = pl->pl_width;
1714789Sahrens 
1715866Seschrock 		/*
1716866Seschrock 		 * If this is being called in scripted mode, or if this is the
1717866Seschrock 		 * last column and it is left-justified, don't include a width
1718866Seschrock 		 * format specifier.
1719866Seschrock 		 */
17202676Seschrock 		if (scripted || (pl->pl_next == NULL && !right_justify))
17212676Seschrock 			(void) printf("%s", propstr);
17222676Seschrock 		else if (right_justify)
17232676Seschrock 			(void) printf("%*s", width, propstr);
17242676Seschrock 		else
17252676Seschrock 			(void) printf("%-*s", width, propstr);
1726789Sahrens 	}
1727789Sahrens 
1728789Sahrens 	(void) printf("\n");
1729789Sahrens }
1730789Sahrens 
1731789Sahrens /*
1732789Sahrens  * Generic callback function to list a dataset or snapshot.
1733789Sahrens  */
1734789Sahrens static int
1735789Sahrens list_callback(zfs_handle_t *zhp, void *data)
1736789Sahrens {
1737789Sahrens 	list_cbdata_t *cbp = data;
1738789Sahrens 
1739789Sahrens 	if (cbp->cb_first) {
1740789Sahrens 		if (!cbp->cb_scripted)
17412676Seschrock 			print_header(cbp->cb_proplist);
17422082Seschrock 		cbp->cb_first = B_FALSE;
1743789Sahrens 	}
1744789Sahrens 
17452676Seschrock 	print_dataset(zhp, cbp->cb_proplist, cbp->cb_scripted);
1746789Sahrens 
1747789Sahrens 	return (0);
1748789Sahrens }
1749789Sahrens 
1750789Sahrens static int
1751789Sahrens zfs_do_list(int argc, char **argv)
1752789Sahrens {
1753789Sahrens 	int c;
17542082Seschrock 	boolean_t scripted = B_FALSE;
1755789Sahrens 	static char default_fields[] =
1756789Sahrens 	    "name,used,available,referenced,mountpoint";
1757*7538SRichard.Morris@Sun.COM 	int types = ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME;
1758*7538SRichard.Morris@Sun.COM 	boolean_t types_specified = B_FALSE;
1759789Sahrens 	char *fields = NULL;
1760789Sahrens 	list_cbdata_t cb = { 0 };
1761789Sahrens 	char *value;
1762789Sahrens 	int ret;
17632379Ssjelinek 	zfs_sort_column_t *sortcol = NULL;
1764*7538SRichard.Morris@Sun.COM 	int flags = ZFS_ITER_PROP_LISTSNAPS | ZFS_ITER_ARGS_CAN_BE_PATHS;
1765789Sahrens 
1766789Sahrens 	/* check options */
17672379Ssjelinek 	while ((c = getopt(argc, argv, ":o:rt:Hs:S:")) != -1) {
1768789Sahrens 		switch (c) {
1769789Sahrens 		case 'o':
1770789Sahrens 			fields = optarg;
1771789Sahrens 			break;
1772789Sahrens 		case 'r':
1773*7538SRichard.Morris@Sun.COM 			flags |= ZFS_ITER_RECURSE;
1774789Sahrens 			break;
1775789Sahrens 		case 'H':
17762082Seschrock 			scripted = B_TRUE;
1777789Sahrens 			break;
17782379Ssjelinek 		case 's':
17792676Seschrock 			if (zfs_add_sort_column(&sortcol, optarg,
17802676Seschrock 			    B_FALSE) != 0) {
17812379Ssjelinek 				(void) fprintf(stderr,
17822379Ssjelinek 				    gettext("invalid property '%s'\n"), optarg);
17832379Ssjelinek 				usage(B_FALSE);
17842379Ssjelinek 			}
17852379Ssjelinek 			break;
17862379Ssjelinek 		case 'S':
17872676Seschrock 			if (zfs_add_sort_column(&sortcol, optarg,
17882676Seschrock 			    B_TRUE) != 0) {
17892379Ssjelinek 				(void) fprintf(stderr,
17902379Ssjelinek 				    gettext("invalid property '%s'\n"), optarg);
17912379Ssjelinek 				usage(B_FALSE);
17922379Ssjelinek 			}
17932379Ssjelinek 			break;
1794789Sahrens 		case 't':
1795789Sahrens 			types = 0;
1796*7538SRichard.Morris@Sun.COM 			types_specified = B_TRUE;
1797*7538SRichard.Morris@Sun.COM 			flags &= ~ZFS_ITER_PROP_LISTSNAPS;
1798789Sahrens 			while (*optarg != '\0') {
1799*7538SRichard.Morris@Sun.COM 				static char *type_subopts[] = { "filesystem",
1800*7538SRichard.Morris@Sun.COM 				    "volume", "snapshot", "all", NULL };
1801*7538SRichard.Morris@Sun.COM 
1802789Sahrens 				switch (getsubopt(&optarg, type_subopts,
1803789Sahrens 				    &value)) {
1804789Sahrens 				case 0:
1805789Sahrens 					types |= ZFS_TYPE_FILESYSTEM;
1806789Sahrens 					break;
1807789Sahrens 				case 1:
1808789Sahrens 					types |= ZFS_TYPE_VOLUME;
1809789Sahrens 					break;
1810789Sahrens 				case 2:
1811789Sahrens 					types |= ZFS_TYPE_SNAPSHOT;
1812789Sahrens 					break;
1813*7538SRichard.Morris@Sun.COM 				case 3:
1814*7538SRichard.Morris@Sun.COM 					types = ZFS_TYPE_DATASET;
1815*7538SRichard.Morris@Sun.COM 					break;
1816*7538SRichard.Morris@Sun.COM 
1817789Sahrens 				default:
1818789Sahrens 					(void) fprintf(stderr,
1819789Sahrens 					    gettext("invalid type '%s'\n"),
1820789Sahrens 					    value);
18212082Seschrock 					usage(B_FALSE);
1822789Sahrens 				}
1823789Sahrens 			}
1824789Sahrens 			break;
1825789Sahrens 		case ':':
1826789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
1827789Sahrens 			    "'%c' option\n"), optopt);
18282082Seschrock 			usage(B_FALSE);
1829789Sahrens 			break;
1830789Sahrens 		case '?':
1831789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
1832789Sahrens 			    optopt);
18332082Seschrock 			usage(B_FALSE);
1834789Sahrens 		}
1835789Sahrens 	}
1836789Sahrens 
1837789Sahrens 	argc -= optind;
1838789Sahrens 	argv += optind;
1839789Sahrens 
1840789Sahrens 	if (fields == NULL)
18417390SMatthew.Ahrens@Sun.COM 		fields = default_fields;
18427390SMatthew.Ahrens@Sun.COM 
18437390SMatthew.Ahrens@Sun.COM 	/*
1844*7538SRichard.Morris@Sun.COM 	 * If "-o space" and no types were specified, don't display snapshots.
18457390SMatthew.Ahrens@Sun.COM 	 */
1846*7538SRichard.Morris@Sun.COM 	if (strcmp(fields, "space") == 0 && types_specified == B_FALSE)
18477390SMatthew.Ahrens@Sun.COM 		types &= ~ZFS_TYPE_SNAPSHOT;
1848789Sahrens 
1849866Seschrock 	/*
18505094Slling 	 * If the user specifies '-o all', the zprop_get_list() doesn't
1851866Seschrock 	 * normally include the name of the dataset.  For 'zfs list', we always
1852866Seschrock 	 * want this property to be first.
1853866Seschrock 	 */
18545094Slling 	if (zprop_get_list(g_zfs, fields, &cb.cb_proplist, ZFS_TYPE_DATASET)
18555094Slling 	    != 0)
18562082Seschrock 		usage(B_FALSE);
18572676Seschrock 
1858789Sahrens 	cb.cb_scripted = scripted;
18592082Seschrock 	cb.cb_first = B_TRUE;
1860789Sahrens 
1861*7538SRichard.Morris@Sun.COM 	ret = zfs_for_each(argc, argv, flags, types, sortcol, &cb.cb_proplist,
1862*7538SRichard.Morris@Sun.COM 	    list_callback, &cb);
18632379Ssjelinek 
18645094Slling 	zprop_free_list(cb.cb_proplist);
18652379Ssjelinek 	zfs_free_sort_columns(sortcol);
1866789Sahrens 
18674221Smmusante 	if (ret == 0 && cb.cb_first && !cb.cb_scripted)
1868789Sahrens 		(void) printf(gettext("no datasets available\n"));
1869789Sahrens 
1870789Sahrens 	return (ret);
1871789Sahrens }
1872789Sahrens 
1873789Sahrens /*
18744490Svb160487  * zfs rename <fs | snap | vol> <fs | snap | vol>
18754490Svb160487  * zfs rename -p <fs | vol> <fs | vol>
18764490Svb160487  * zfs rename -r <snap> <snap>
1877789Sahrens  *
1878789Sahrens  * Renames the given dataset to another of the same type.
18794490Svb160487  *
18804490Svb160487  * The '-p' flag creates all the non-existing ancestors of the target first.
1881789Sahrens  */
1882789Sahrens /* ARGSUSED */
1883789Sahrens static int
1884789Sahrens zfs_do_rename(int argc, char **argv)
1885789Sahrens {
1886789Sahrens 	zfs_handle_t *zhp;
18874007Smmusante 	int c;
18882082Seschrock 	int ret;
18894490Svb160487 	boolean_t recurse = B_FALSE;
18904490Svb160487 	boolean_t parents = B_FALSE;
1891789Sahrens 
1892789Sahrens 	/* check options */
18934490Svb160487 	while ((c = getopt(argc, argv, "pr")) != -1) {
18944007Smmusante 		switch (c) {
18954490Svb160487 		case 'p':
18964490Svb160487 			parents = B_TRUE;
18974490Svb160487 			break;
18984007Smmusante 		case 'r':
18994490Svb160487 			recurse = B_TRUE;
19004007Smmusante 			break;
19014007Smmusante 		case '?':
19024007Smmusante 		default:
19034007Smmusante 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
19044007Smmusante 			    optopt);
19054007Smmusante 			usage(B_FALSE);
19064007Smmusante 		}
1907789Sahrens 	}
1908789Sahrens 
19094007Smmusante 	argc -= optind;
19104007Smmusante 	argv += optind;
19114007Smmusante 
1912789Sahrens 	/* check number of arguments */
19134007Smmusante 	if (argc < 1) {
1914789Sahrens 		(void) fprintf(stderr, gettext("missing source dataset "
1915789Sahrens 		    "argument\n"));
19162082Seschrock 		usage(B_FALSE);
1917789Sahrens 	}
19184007Smmusante 	if (argc < 2) {
1919789Sahrens 		(void) fprintf(stderr, gettext("missing target dataset "
1920789Sahrens 		    "argument\n"));
19212082Seschrock 		usage(B_FALSE);
1922789Sahrens 	}
19234007Smmusante 	if (argc > 2) {
1924789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
19252082Seschrock 		usage(B_FALSE);
1926789Sahrens 	}
1927789Sahrens 
19284490Svb160487 	if (recurse && parents) {
19294490Svb160487 		(void) fprintf(stderr, gettext("-p and -r options are mutually "
19304490Svb160487 		    "exclusive\n"));
19314490Svb160487 		usage(B_FALSE);
19324490Svb160487 	}
19334490Svb160487 
19344007Smmusante 	if (recurse && strchr(argv[0], '@') == 0) {
19354007Smmusante 		(void) fprintf(stderr, gettext("source dataset for recursive "
19364007Smmusante 		    "rename must be a snapshot\n"));
19374007Smmusante 		usage(B_FALSE);
19384007Smmusante 	}
19394007Smmusante 
19404490Svb160487 	if ((zhp = zfs_open(g_zfs, argv[0], parents ? ZFS_TYPE_FILESYSTEM |
19415094Slling 	    ZFS_TYPE_VOLUME : ZFS_TYPE_DATASET)) == NULL)
1942789Sahrens 		return (1);
1943789Sahrens 
19444490Svb160487 	/* If we were asked and the name looks good, try to create ancestors. */
19454490Svb160487 	if (parents && zfs_name_valid(argv[1], zfs_get_type(zhp)) &&
19464490Svb160487 	    zfs_create_ancestors(g_zfs, argv[1]) != 0) {
19474490Svb160487 		zfs_close(zhp);
19484490Svb160487 		return (1);
19494490Svb160487 	}
19504490Svb160487 
19514007Smmusante 	ret = (zfs_rename(zhp, argv[1], recurse) != 0);
19522082Seschrock 
19532082Seschrock 	zfs_close(zhp);
19542082Seschrock 	return (ret);
19552082Seschrock }
19562082Seschrock 
19572082Seschrock /*
19582082Seschrock  * zfs promote <fs>
19592082Seschrock  *
19602082Seschrock  * Promotes the given clone fs to be the parent
19612082Seschrock  */
19622082Seschrock /* ARGSUSED */
19632082Seschrock static int
19642082Seschrock zfs_do_promote(int argc, char **argv)
19652082Seschrock {
19662082Seschrock 	zfs_handle_t *zhp;
19672082Seschrock 	int ret;
19682082Seschrock 
19692082Seschrock 	/* check options */
19702082Seschrock 	if (argc > 1 && argv[1][0] == '-') {
19712082Seschrock 		(void) fprintf(stderr, gettext("invalid option '%c'\n"),
19722082Seschrock 		    argv[1][1]);
19732082Seschrock 		usage(B_FALSE);
19742082Seschrock 	}
19752082Seschrock 
19762082Seschrock 	/* check number of arguments */
19772082Seschrock 	if (argc < 2) {
19782082Seschrock 		(void) fprintf(stderr, gettext("missing clone filesystem"
19792597Snd150628 		    " argument\n"));
19802082Seschrock 		usage(B_FALSE);
19812082Seschrock 	}
19822082Seschrock 	if (argc > 2) {
19832082Seschrock 		(void) fprintf(stderr, gettext("too many arguments\n"));
19842082Seschrock 		usage(B_FALSE);
19852082Seschrock 	}
19862082Seschrock 
19872082Seschrock 	zhp = zfs_open(g_zfs, argv[1], ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
19882082Seschrock 	if (zhp == NULL)
19892082Seschrock 		return (1);
19902082Seschrock 
19912082Seschrock 	ret = (zfs_promote(zhp) != 0);
19922082Seschrock 
19932926Sek110237 
1994789Sahrens 	zfs_close(zhp);
1995789Sahrens 	return (ret);
1996789Sahrens }
1997789Sahrens 
1998789Sahrens /*
19995568Sahrens  * zfs rollback [-rRf] <snapshot>
2000789Sahrens  *
2001789Sahrens  * 	-r	Delete any intervening snapshots before doing rollback
2002789Sahrens  * 	-R	Delete any snapshots and their clones
20035568Sahrens  * 	-f	ignored for backwards compatability
2004789Sahrens  *
2005789Sahrens  * Given a filesystem, rollback to a specific snapshot, discarding any changes
2006789Sahrens  * since then and making it the active dataset.  If more recent snapshots exist,
2007789Sahrens  * the command will complain unless the '-r' flag is given.
2008789Sahrens  */
2009789Sahrens typedef struct rollback_cbdata {
2010789Sahrens 	uint64_t	cb_create;
20112082Seschrock 	boolean_t	cb_first;
2012789Sahrens 	int		cb_doclones;
2013789Sahrens 	char		*cb_target;
2014789Sahrens 	int		cb_error;
20152082Seschrock 	boolean_t	cb_recurse;
20162082Seschrock 	boolean_t	cb_dependent;
2017789Sahrens } rollback_cbdata_t;
2018789Sahrens 
2019789Sahrens /*
2020789Sahrens  * Report any snapshots more recent than the one specified.  Used when '-r' is
2021789Sahrens  * not specified.  We reuse this same callback for the snapshot dependents - if
2022789Sahrens  * 'cb_dependent' is set, then this is a dependent and we should report it
2023789Sahrens  * without checking the transaction group.
2024789Sahrens  */
2025789Sahrens static int
2026789Sahrens rollback_check(zfs_handle_t *zhp, void *data)
2027789Sahrens {
2028789Sahrens 	rollback_cbdata_t *cbp = data;
2029789Sahrens 
20302082Seschrock 	if (cbp->cb_doclones) {
20312082Seschrock 		zfs_close(zhp);
2032789Sahrens 		return (0);
20332082Seschrock 	}
2034789Sahrens 
2035789Sahrens 	if (!cbp->cb_dependent) {
2036789Sahrens 		if (strcmp(zfs_get_name(zhp), cbp->cb_target) != 0 &&
20371294Slling 		    zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT &&
2038789Sahrens 		    zfs_prop_get_int(zhp, ZFS_PROP_CREATETXG) >
2039789Sahrens 		    cbp->cb_create) {
2040789Sahrens 
2041789Sahrens 			if (cbp->cb_first && !cbp->cb_recurse) {
2042789Sahrens 				(void) fprintf(stderr, gettext("cannot "
2043789Sahrens 				    "rollback to '%s': more recent snapshots "
2044789Sahrens 				    "exist\n"),
2045789Sahrens 				    cbp->cb_target);
2046789Sahrens 				(void) fprintf(stderr, gettext("use '-r' to "
2047789Sahrens 				    "force deletion of the following "
2048789Sahrens 				    "snapshots:\n"));
2049789Sahrens 				cbp->cb_first = 0;
2050789Sahrens 				cbp->cb_error = 1;
2051789Sahrens 			}
2052789Sahrens 
2053789Sahrens 			if (cbp->cb_recurse) {
20542082Seschrock 				cbp->cb_dependent = B_TRUE;
20552474Seschrock 				if (zfs_iter_dependents(zhp, B_TRUE,
20562474Seschrock 				    rollback_check, cbp) != 0) {
20572474Seschrock 					zfs_close(zhp);
20582474Seschrock 					return (-1);
20592474Seschrock 				}
20602082Seschrock 				cbp->cb_dependent = B_FALSE;
2061789Sahrens 			} else {
2062789Sahrens 				(void) fprintf(stderr, "%s\n",
2063789Sahrens 				    zfs_get_name(zhp));
2064789Sahrens 			}
2065789Sahrens 		}
2066789Sahrens 	} else {
2067789Sahrens 		if (cbp->cb_first && cbp->cb_recurse) {
2068789Sahrens 			(void) fprintf(stderr, gettext("cannot rollback to "
2069789Sahrens 			    "'%s': clones of previous snapshots exist\n"),
2070789Sahrens 			    cbp->cb_target);
2071789Sahrens 			(void) fprintf(stderr, gettext("use '-R' to "
2072789Sahrens 			    "force deletion of the following clones and "
2073789Sahrens 			    "dependents:\n"));
2074789Sahrens 			cbp->cb_first = 0;
2075789Sahrens 			cbp->cb_error = 1;
2076789Sahrens 		}
2077789Sahrens 
2078789Sahrens 		(void) fprintf(stderr, "%s\n", zfs_get_name(zhp));
2079789Sahrens 	}
2080789Sahrens 
2081789Sahrens 	zfs_close(zhp);
2082789Sahrens 	return (0);
2083789Sahrens }
2084789Sahrens 
2085789Sahrens static int
2086789Sahrens zfs_do_rollback(int argc, char **argv)
2087789Sahrens {
2088789Sahrens 	int ret;
2089789Sahrens 	int c;
20905749Sahrens 	boolean_t force = B_FALSE;
2091789Sahrens 	rollback_cbdata_t cb = { 0 };
2092789Sahrens 	zfs_handle_t *zhp, *snap;
2093789Sahrens 	char parentname[ZFS_MAXNAMELEN];
2094789Sahrens 	char *delim;
2095789Sahrens 
2096789Sahrens 	/* check options */
20975568Sahrens 	while ((c = getopt(argc, argv, "rRf")) != -1) {
2098789Sahrens 		switch (c) {
2099789Sahrens 		case 'r':
2100789Sahrens 			cb.cb_recurse = 1;
2101789Sahrens 			break;
2102789Sahrens 		case 'R':
2103789Sahrens 			cb.cb_recurse = 1;
2104789Sahrens 			cb.cb_doclones = 1;
2105789Sahrens 			break;
21065568Sahrens 		case 'f':
21075749Sahrens 			force = B_TRUE;
21085568Sahrens 			break;
2109789Sahrens 		case '?':
2110789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
2111789Sahrens 			    optopt);
21122082Seschrock 			usage(B_FALSE);
2113789Sahrens 		}
2114789Sahrens 	}
2115789Sahrens 
2116789Sahrens 	argc -= optind;
2117789Sahrens 	argv += optind;
2118789Sahrens 
2119789Sahrens 	/* check number of arguments */
2120789Sahrens 	if (argc < 1) {
2121789Sahrens 		(void) fprintf(stderr, gettext("missing dataset argument\n"));
21222082Seschrock 		usage(B_FALSE);
2123789Sahrens 	}
2124789Sahrens 	if (argc > 1) {
2125789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
21262082Seschrock 		usage(B_FALSE);
2127789Sahrens 	}
2128789Sahrens 
2129789Sahrens 	/* open the snapshot */
21302082Seschrock 	if ((snap = zfs_open(g_zfs, argv[0], ZFS_TYPE_SNAPSHOT)) == NULL)
2131789Sahrens 		return (1);
2132789Sahrens 
21331294Slling 	/* open the parent dataset */
21341294Slling 	(void) strlcpy(parentname, argv[0], sizeof (parentname));
2135789Sahrens 	verify((delim = strrchr(parentname, '@')) != NULL);
2136789Sahrens 	*delim = '\0';
21375094Slling 	if ((zhp = zfs_open(g_zfs, parentname, ZFS_TYPE_DATASET)) == NULL) {
2138789Sahrens 		zfs_close(snap);
2139789Sahrens 		return (1);
2140789Sahrens 	}
2141789Sahrens 
2142789Sahrens 	/*
2143789Sahrens 	 * Check for more recent snapshots and/or clones based on the presence
2144789Sahrens 	 * of '-r' and '-R'.
2145789Sahrens 	 */
21461294Slling 	cb.cb_target = argv[0];
21471294Slling 	cb.cb_create = zfs_prop_get_int(snap, ZFS_PROP_CREATETXG);
21482082Seschrock 	cb.cb_first = B_TRUE;
2149789Sahrens 	cb.cb_error = 0;
21502474Seschrock 	if ((ret = zfs_iter_children(zhp, rollback_check, &cb)) != 0)
21512474Seschrock 		goto out;
2152789Sahrens 
2153789Sahrens 	if ((ret = cb.cb_error) != 0)
2154789Sahrens 		goto out;
2155789Sahrens 
2156789Sahrens 	/*
21571294Slling 	 * Rollback parent to the given snapshot.
2158789Sahrens 	 */
21595749Sahrens 	ret = zfs_rollback(zhp, snap, force);
2160789Sahrens 
2161789Sahrens out:
2162789Sahrens 	zfs_close(snap);
2163789Sahrens 	zfs_close(zhp);
2164789Sahrens 
2165789Sahrens 	if (ret == 0)
2166789Sahrens 		return (0);
2167789Sahrens 	else
2168789Sahrens 		return (1);
2169789Sahrens }
2170789Sahrens 
2171789Sahrens /*
2172789Sahrens  * zfs set property=value { fs | snap | vol } ...
2173789Sahrens  *
2174789Sahrens  * Sets the given property for all datasets specified on the command line.
2175789Sahrens  */
2176789Sahrens typedef struct set_cbdata {
2177789Sahrens 	char		*cb_propname;
2178789Sahrens 	char		*cb_value;
2179789Sahrens } set_cbdata_t;
2180789Sahrens 
2181789Sahrens static int
2182789Sahrens set_callback(zfs_handle_t *zhp, void *data)
2183789Sahrens {
2184789Sahrens 	set_cbdata_t *cbp = data;
2185789Sahrens 
21862676Seschrock 	if (zfs_prop_set(zhp, cbp->cb_propname, cbp->cb_value) != 0) {
21872169Snd150628 		switch (libzfs_errno(g_zfs)) {
21882169Snd150628 		case EZFS_MOUNTFAILED:
21892169Snd150628 			(void) fprintf(stderr, gettext("property may be set "
21902169Snd150628 			    "but unable to remount filesystem\n"));
21912169Snd150628 			break;
21923126Sahl 		case EZFS_SHARENFSFAILED:
21932169Snd150628 			(void) fprintf(stderr, gettext("property may be set "
21942169Snd150628 			    "but unable to reshare filesystem\n"));
21952169Snd150628 			break;
21962169Snd150628 		}
2197789Sahrens 		return (1);
21982169Snd150628 	}
21992856Snd150628 	return (0);
2200789Sahrens }
2201789Sahrens 
2202789Sahrens static int
2203789Sahrens zfs_do_set(int argc, char **argv)
2204789Sahrens {
2205789Sahrens 	set_cbdata_t cb;
22062926Sek110237 	int ret;
2207789Sahrens 
2208789Sahrens 	/* check for options */
2209789Sahrens 	if (argc > 1 && argv[1][0] == '-') {
2210789Sahrens 		(void) fprintf(stderr, gettext("invalid option '%c'\n"),
2211789Sahrens 		    argv[1][1]);
22122082Seschrock 		usage(B_FALSE);
2213789Sahrens 	}
2214789Sahrens 
2215789Sahrens 	/* check number of arguments */
2216789Sahrens 	if (argc < 2) {
2217789Sahrens 		(void) fprintf(stderr, gettext("missing property=value "
2218789Sahrens 		    "argument\n"));
22192082Seschrock 		usage(B_FALSE);
2220789Sahrens 	}
2221789Sahrens 	if (argc < 3) {
2222789Sahrens 		(void) fprintf(stderr, gettext("missing dataset name\n"));
22232082Seschrock 		usage(B_FALSE);
2224789Sahrens 	}
2225789Sahrens 
2226789Sahrens 	/* validate property=value argument */
2227789Sahrens 	cb.cb_propname = argv[1];
22286993Ssg201626 	if (((cb.cb_value = strchr(cb.cb_propname, '=')) == NULL) ||
22296993Ssg201626 	    (cb.cb_value[1] == '\0')) {
2230789Sahrens 		(void) fprintf(stderr, gettext("missing value in "
2231789Sahrens 		    "property=value argument\n"));
22322082Seschrock 		usage(B_FALSE);
2233789Sahrens 	}
2234789Sahrens 
2235789Sahrens 	*cb.cb_value = '\0';
2236789Sahrens 	cb.cb_value++;
2237789Sahrens 
2238789Sahrens 	if (*cb.cb_propname == '\0') {
2239789Sahrens 		(void) fprintf(stderr,
2240789Sahrens 		    gettext("missing property in property=value argument\n"));
22412082Seschrock 		usage(B_FALSE);
2242789Sahrens 	}
2243789Sahrens 
2244*7538SRichard.Morris@Sun.COM 	ret = zfs_for_each(argc - 2, argv + 2, NULL,
2245*7538SRichard.Morris@Sun.COM 	    ZFS_TYPE_DATASET, NULL, NULL, set_callback, &cb);
22462926Sek110237 
22472926Sek110237 	return (ret);
2248789Sahrens }
2249789Sahrens 
2250789Sahrens /*
22517265Sahrens  * zfs snapshot [-r] [-o prop=value] ... <fs@snap>
2252789Sahrens  *
2253789Sahrens  * Creates a snapshot with the given name.  While functionally equivalent to
22545331Samw  * 'zfs create', it is a separate command to differentiate intent.
2255789Sahrens  */
2256789Sahrens static int
2257789Sahrens zfs_do_snapshot(int argc, char **argv)
2258789Sahrens {
22594490Svb160487 	boolean_t recursive = B_FALSE;
22602199Sahrens 	int ret;
22612199Sahrens 	char c;
22627265Sahrens 	nvlist_t *props;
22637265Sahrens 
22647265Sahrens 	if (nvlist_alloc(&props, NV_UNIQUE_NAME, 0) != 0) {
22657265Sahrens 		(void) fprintf(stderr, gettext("internal error: "
22667265Sahrens 		    "out of memory\n"));
22677265Sahrens 		return (1);
22687265Sahrens 	}
22692199Sahrens 
2270789Sahrens 	/* check options */
22717265Sahrens 	while ((c = getopt(argc, argv, "ro:")) != -1) {
22722199Sahrens 		switch (c) {
22737265Sahrens 		case 'o':
22747265Sahrens 			if (parseprop(props))
22757265Sahrens 				return (1);
22767265Sahrens 			break;
22772199Sahrens 		case 'r':
22782199Sahrens 			recursive = B_TRUE;
22792199Sahrens 			break;
22802199Sahrens 		case '?':
22812199Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
22822199Sahrens 			    optopt);
22837265Sahrens 			goto usage;
22842199Sahrens 		}
2285789Sahrens 	}
2286789Sahrens 
22872199Sahrens 	argc -= optind;
22882199Sahrens 	argv += optind;
22892199Sahrens 
2290789Sahrens 	/* check number of arguments */
22912199Sahrens 	if (argc < 1) {
2292789Sahrens 		(void) fprintf(stderr, gettext("missing snapshot argument\n"));
22937265Sahrens 		goto usage;
2294789Sahrens 	}
22952199Sahrens 	if (argc > 1) {
2296789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
22977265Sahrens 		goto usage;
2298789Sahrens 	}
2299789Sahrens 
23007265Sahrens 	ret = zfs_snapshot(g_zfs, argv[0], recursive, props);
23017265Sahrens 	nvlist_free(props);
23022199Sahrens 	if (ret && recursive)
23032199Sahrens 		(void) fprintf(stderr, gettext("no snapshots were created\n"));
23042199Sahrens 	return (ret != 0);
23057265Sahrens 
23067265Sahrens usage:
23077265Sahrens 	nvlist_free(props);
23087265Sahrens 	usage(B_FALSE);
23097265Sahrens 	return (-1);
2310789Sahrens }
2311789Sahrens 
2312789Sahrens /*
23135367Sahrens  * zfs send [-v] -R [-i|-I <@snap>] <fs@snap>
23145367Sahrens  * zfs send [-v] [-i|-I <@snap>] <fs@snap>
2315789Sahrens  *
2316789Sahrens  * Send a backup stream to stdout.
2317789Sahrens  */
2318789Sahrens static int
23191749Sahrens zfs_do_send(int argc, char **argv)
2320789Sahrens {
2321789Sahrens 	char *fromname = NULL;
23225367Sahrens 	char *toname = NULL;
23232885Sahrens 	char *cp;
23242885Sahrens 	zfs_handle_t *zhp;
23255367Sahrens 	boolean_t doall = B_FALSE;
23265367Sahrens 	boolean_t replicate = B_FALSE;
23275367Sahrens 	boolean_t fromorigin = B_FALSE;
23285367Sahrens 	boolean_t verbose = B_FALSE;
2329789Sahrens 	int c, err;
2330789Sahrens 
2331789Sahrens 	/* check options */
23325367Sahrens 	while ((c = getopt(argc, argv, ":i:I:Rv")) != -1) {
2333789Sahrens 		switch (c) {
2334789Sahrens 		case 'i':
23352885Sahrens 			if (fromname)
23362885Sahrens 				usage(B_FALSE);
2337789Sahrens 			fromname = optarg;
2338789Sahrens 			break;
23395367Sahrens 		case 'I':
23405367Sahrens 			if (fromname)
23415367Sahrens 				usage(B_FALSE);
23425367Sahrens 			fromname = optarg;
23435367Sahrens 			doall = B_TRUE;
23445367Sahrens 			break;
23455367Sahrens 		case 'R':
23465367Sahrens 			replicate = B_TRUE;
23475367Sahrens 			break;
23485367Sahrens 		case 'v':
23495367Sahrens 			verbose = B_TRUE;
23505367Sahrens 			break;
2351789Sahrens 		case ':':
2352789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
2353789Sahrens 			    "'%c' option\n"), optopt);
23542082Seschrock 			usage(B_FALSE);
2355789Sahrens 			break;
2356789Sahrens 		case '?':
2357789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
2358789Sahrens 			    optopt);
23592082Seschrock 			usage(B_FALSE);
2360789Sahrens 		}
2361789Sahrens 	}
2362789Sahrens 
2363789Sahrens 	argc -= optind;
2364789Sahrens 	argv += optind;
2365789Sahrens 
2366789Sahrens 	/* check number of arguments */
2367789Sahrens 	if (argc < 1) {
2368789Sahrens 		(void) fprintf(stderr, gettext("missing snapshot argument\n"));
23692082Seschrock 		usage(B_FALSE);
2370789Sahrens 	}
2371789Sahrens 	if (argc > 1) {
2372789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
23732082Seschrock 		usage(B_FALSE);
2374789Sahrens 	}
2375789Sahrens 
2376789Sahrens 	if (isatty(STDOUT_FILENO)) {
2377789Sahrens 		(void) fprintf(stderr,
23782885Sahrens 		    gettext("Error: Stream can not be written to a terminal.\n"
23793912Slling 		    "You must redirect standard output.\n"));
2380789Sahrens 		return (1);
2381789Sahrens 	}
2382789Sahrens 
23835367Sahrens 	cp = strchr(argv[0], '@');
23845367Sahrens 	if (cp == NULL) {
23855367Sahrens 		(void) fprintf(stderr,
23865367Sahrens 		    gettext("argument must be a snapshot\n"));
23875367Sahrens 		usage(B_FALSE);
23885367Sahrens 	}
23895367Sahrens 	*cp = '\0';
23905367Sahrens 	toname = cp + 1;
23915367Sahrens 	zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
23925367Sahrens 	if (zhp == NULL)
23932665Snd150628 		return (1);
23942665Snd150628 
23952885Sahrens 	/*
23962885Sahrens 	 * If they specified the full path to the snapshot, chop off
23975367Sahrens 	 * everything except the short name of the snapshot, but special
23985367Sahrens 	 * case if they specify the origin.
23992885Sahrens 	 */
24002885Sahrens 	if (fromname && (cp = strchr(fromname, '@')) != NULL) {
24015367Sahrens 		char origin[ZFS_MAXNAMELEN];
24025367Sahrens 		zprop_source_t src;
24035367Sahrens 
24045367Sahrens 		(void) zfs_prop_get(zhp, ZFS_PROP_ORIGIN,
24055367Sahrens 		    origin, sizeof (origin), &src, NULL, 0, B_FALSE);
24065367Sahrens 
24075367Sahrens 		if (strcmp(origin, fromname) == 0) {
24085367Sahrens 			fromname = NULL;
24095367Sahrens 			fromorigin = B_TRUE;
24105367Sahrens 		} else {
24115367Sahrens 			*cp = '\0';
24125367Sahrens 			if (cp != fromname && strcmp(argv[0], fromname)) {
24135367Sahrens 				(void) fprintf(stderr,
24145367Sahrens 				    gettext("incremental source must be "
24155367Sahrens 				    "in same filesystem\n"));
24165367Sahrens 				usage(B_FALSE);
24175367Sahrens 			}
24185367Sahrens 			fromname = cp + 1;
24195367Sahrens 			if (strchr(fromname, '@') || strchr(fromname, '/')) {
24205367Sahrens 				(void) fprintf(stderr,
24215367Sahrens 				    gettext("invalid incremental source\n"));
24225367Sahrens 				usage(B_FALSE);
24235367Sahrens 			}
24242885Sahrens 		}
2425789Sahrens 	}
2426789Sahrens 
24275367Sahrens 	if (replicate && fromname == NULL)
24285367Sahrens 		doall = B_TRUE;
24295367Sahrens 
24305367Sahrens 	err = zfs_send(zhp, fromname, toname, replicate, doall, fromorigin,
24315367Sahrens 	    verbose, STDOUT_FILENO);
24322885Sahrens 	zfs_close(zhp);
2433789Sahrens 
2434789Sahrens 	return (err != 0);
2435789Sahrens }
2436789Sahrens 
2437789Sahrens /*
24385367Sahrens  * zfs receive [-dnvF] <fs@snap>
2439789Sahrens  *
2440789Sahrens  * Restore a backup stream from stdin.
2441789Sahrens  */
2442789Sahrens static int
24431749Sahrens zfs_do_receive(int argc, char **argv)
2444789Sahrens {
2445789Sahrens 	int c, err;
24465367Sahrens 	recvflags_t flags;
24475367Sahrens 
24485367Sahrens 	bzero(&flags, sizeof (recvflags_t));
2449789Sahrens 	/* check options */
24502665Snd150628 	while ((c = getopt(argc, argv, ":dnvF")) != -1) {
2451789Sahrens 		switch (c) {
2452789Sahrens 		case 'd':
24535367Sahrens 			flags.isprefix = B_TRUE;
2454789Sahrens 			break;
2455789Sahrens 		case 'n':
24565367Sahrens 			flags.dryrun = B_TRUE;
2457789Sahrens 			break;
2458789Sahrens 		case 'v':
24595367Sahrens 			flags.verbose = B_TRUE;
2460789Sahrens 			break;
24612665Snd150628 		case 'F':
24625367Sahrens 			flags.force = B_TRUE;
24632665Snd150628 			break;
2464789Sahrens 		case ':':
2465789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
2466789Sahrens 			    "'%c' option\n"), optopt);
24672082Seschrock 			usage(B_FALSE);
2468789Sahrens 			break;
2469789Sahrens 		case '?':
2470789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
2471789Sahrens 			    optopt);
24722082Seschrock 			usage(B_FALSE);
2473789Sahrens 		}
2474789Sahrens 	}
2475789Sahrens 
2476789Sahrens 	argc -= optind;
2477789Sahrens 	argv += optind;
2478789Sahrens 
2479789Sahrens 	/* check number of arguments */
2480789Sahrens 	if (argc < 1) {
2481789Sahrens 		(void) fprintf(stderr, gettext("missing snapshot argument\n"));
24822082Seschrock 		usage(B_FALSE);
2483789Sahrens 	}
2484789Sahrens 	if (argc > 1) {
2485789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
24862082Seschrock 		usage(B_FALSE);
2487789Sahrens 	}
2488789Sahrens 
2489789Sahrens 	if (isatty(STDIN_FILENO)) {
2490789Sahrens 		(void) fprintf(stderr,
2491789Sahrens 		    gettext("Error: Backup stream can not be read "
24923912Slling 		    "from a terminal.\n"
24933912Slling 		    "You must redirect standard input.\n"));
2494789Sahrens 		return (1);
2495789Sahrens 	}
2496789Sahrens 
24975367Sahrens 	err = zfs_receive(g_zfs, argv[0], flags, STDIN_FILENO, NULL);
24982926Sek110237 
24994543Smarks 	return (err != 0);
25004543Smarks }
25014543Smarks 
25024543Smarks typedef struct allow_cb {
25034543Smarks 	int  a_permcnt;
25044543Smarks 	size_t a_treeoffset;
25054543Smarks } allow_cb_t;
25064543Smarks 
25074543Smarks static void
25084543Smarks zfs_print_perms(avl_tree_t *tree)
25094543Smarks {
25104543Smarks 	zfs_perm_node_t *permnode;
25114543Smarks 
25124543Smarks 	permnode = avl_first(tree);
25134543Smarks 	while (permnode != NULL) {
25144543Smarks 		(void) printf("%s", permnode->z_pname);
25154543Smarks 		permnode = AVL_NEXT(tree, permnode);
25164543Smarks 		if (permnode)
25174543Smarks 			(void) printf(",");
25184543Smarks 		else
25194543Smarks 			(void) printf("\n");
25204543Smarks 	}
25214543Smarks }
25224543Smarks 
25234543Smarks /*
25244543Smarks  * Iterate over user/groups/everyone/... and the call perm_iter
25254543Smarks  * function to print actual permission when tree has >0 nodes.
25264543Smarks  */
25274543Smarks static void
25284543Smarks zfs_iter_perms(avl_tree_t *tree, const char *banner, allow_cb_t *cb)
25294543Smarks {
25304543Smarks 	zfs_allow_node_t *item;
25314543Smarks 	avl_tree_t *ptree;
25324543Smarks 
25334543Smarks 	item = avl_first(tree);
25344543Smarks 	while (item) {
25354543Smarks 		ptree = (void *)((char *)item + cb->a_treeoffset);
25364543Smarks 		if (avl_numnodes(ptree)) {
25374543Smarks 			if (cb->a_permcnt++ == 0)
25384543Smarks 				(void) printf("%s\n", banner);
25394543Smarks 			(void) printf("\t%s", item->z_key);
25404543Smarks 			/*
25414543Smarks 			 * Avoid an extra space being printed
25424543Smarks 			 * for "everyone" which is keyed with a null
25434543Smarks 			 * string
25444543Smarks 			 */
25454543Smarks 			if (item->z_key[0] != '\0')
25464543Smarks 				(void) printf(" ");
25474543Smarks 			zfs_print_perms(ptree);
25484543Smarks 		}
25494543Smarks 		item = AVL_NEXT(tree, item);
25504543Smarks 	}
25514543Smarks }
25524543Smarks 
25534543Smarks #define	LINES "-------------------------------------------------------------\n"
25544543Smarks static int
25554543Smarks zfs_print_allows(char *ds)
25564543Smarks {
25574543Smarks 	zfs_allow_t *curperms, *perms;
25584543Smarks 	zfs_handle_t *zhp;
25594543Smarks 	allow_cb_t allowcb = { 0 };
25604543Smarks 	char banner[MAXPATHLEN];
25614543Smarks 
25624543Smarks 	if (ds[0] == '-')
25634543Smarks 		usage(B_FALSE);
25644543Smarks 
25654543Smarks 	if (strrchr(ds, '@')) {
25664543Smarks 		(void) fprintf(stderr, gettext("Snapshots don't have 'allow'"
25674543Smarks 		    " permissions\n"));
25684543Smarks 		return (1);
25694543Smarks 	}
25705094Slling 	if ((zhp = zfs_open(g_zfs, ds, ZFS_TYPE_DATASET)) == NULL)
25714543Smarks 		return (1);
25724543Smarks 
25734543Smarks 	if (zfs_perm_get(zhp, &perms)) {
25744543Smarks 		(void) fprintf(stderr,
25754543Smarks 		    gettext("Failed to retrieve 'allows' on %s\n"), ds);
25764543Smarks 		zfs_close(zhp);
25774543Smarks 		return (1);
25784543Smarks 	}
25794543Smarks 
25804543Smarks 	zfs_close(zhp);
25814543Smarks 
25824543Smarks 	if (perms != NULL)
25834543Smarks 		(void) printf("%s", LINES);
25844543Smarks 	for (curperms = perms; curperms; curperms = curperms->z_next) {
25854543Smarks 
25864543Smarks 		(void) snprintf(banner, sizeof (banner),
25874543Smarks 		    "Permission sets on (%s)", curperms->z_setpoint);
25884543Smarks 		allowcb.a_treeoffset =
25894543Smarks 		    offsetof(zfs_allow_node_t, z_localdescend);
25904543Smarks 		allowcb.a_permcnt = 0;
25914543Smarks 		zfs_iter_perms(&curperms->z_sets, banner, &allowcb);
25924543Smarks 
25934543Smarks 		(void) snprintf(banner, sizeof (banner),
25944543Smarks 		    "Create time permissions on (%s)", curperms->z_setpoint);
25954543Smarks 		allowcb.a_treeoffset =
25964543Smarks 		    offsetof(zfs_allow_node_t, z_localdescend);
25974543Smarks 		allowcb.a_permcnt = 0;
25984543Smarks 		zfs_iter_perms(&curperms->z_crperms, banner, &allowcb);
25994543Smarks 
26004543Smarks 
26014543Smarks 		(void) snprintf(banner, sizeof (banner),
26024543Smarks 		    "Local permissions on (%s)", curperms->z_setpoint);
26034543Smarks 		allowcb.a_treeoffset = offsetof(zfs_allow_node_t, z_local);
26044543Smarks 		allowcb.a_permcnt = 0;
26054543Smarks 		zfs_iter_perms(&curperms->z_user, banner, &allowcb);
26064543Smarks 		zfs_iter_perms(&curperms->z_group, banner, &allowcb);
26074543Smarks 		zfs_iter_perms(&curperms->z_everyone, banner, &allowcb);
26084543Smarks 
26094543Smarks 		(void) snprintf(banner, sizeof (banner),
26104543Smarks 		    "Descendent permissions on (%s)", curperms->z_setpoint);
26114543Smarks 		allowcb.a_treeoffset = offsetof(zfs_allow_node_t, z_descend);
26124543Smarks 		allowcb.a_permcnt = 0;
26134543Smarks 		zfs_iter_perms(&curperms->z_user, banner, &allowcb);
26144543Smarks 		zfs_iter_perms(&curperms->z_group, banner, &allowcb);
26154543Smarks 		zfs_iter_perms(&curperms->z_everyone, banner, &allowcb);
26164543Smarks 
26174543Smarks 		(void) snprintf(banner, sizeof (banner),
26184543Smarks 		    "Local+Descendent permissions on (%s)",
26194543Smarks 		    curperms->z_setpoint);
26204543Smarks 		allowcb.a_treeoffset =
26214543Smarks 		    offsetof(zfs_allow_node_t, z_localdescend);
26224543Smarks 		allowcb.a_permcnt = 0;
26234543Smarks 		zfs_iter_perms(&curperms->z_user, banner, &allowcb);
26244543Smarks 		zfs_iter_perms(&curperms->z_group, banner, &allowcb);
26254543Smarks 		zfs_iter_perms(&curperms->z_everyone, banner, &allowcb);
26264543Smarks 
26274543Smarks 		(void) printf("%s", LINES);
26282926Sek110237 	}
26294543Smarks 	zfs_free_allows(perms);
26304543Smarks 	return (0);
26314543Smarks }
26324543Smarks 
26334543Smarks #define	ALLOWOPTIONS "ldcsu:g:e"
26344543Smarks #define	UNALLOWOPTIONS "ldcsu:g:er"
26354543Smarks 
26364543Smarks /*
26374543Smarks  * Validate options, and build necessary datastructure to display/remove/add
26384543Smarks  * permissions.
26394543Smarks  * Returns 0 - If permissions should be added/removed
26404543Smarks  * Returns 1 - If permissions should be displayed.
26414543Smarks  * Returns -1 - on failure
26424543Smarks  */
26434543Smarks int
26444543Smarks parse_allow_args(int *argc, char **argv[], boolean_t unallow,
26454543Smarks     char **ds, int *recurse, nvlist_t **zperms)
26464543Smarks {
26474543Smarks 	int c;
26484543Smarks 	char *options = unallow ? UNALLOWOPTIONS : ALLOWOPTIONS;
26494543Smarks 	zfs_deleg_inherit_t deleg_type = ZFS_DELEG_NONE;
26504543Smarks 	zfs_deleg_who_type_t who_type = ZFS_DELEG_WHO_UNKNOWN;
26514860Smarks 	char *who = NULL;
26524543Smarks 	char *perms = NULL;
26534543Smarks 	zfs_handle_t *zhp;
26544543Smarks 
26554543Smarks 	while ((c = getopt(*argc, *argv, options)) != -1) {
26564543Smarks 		switch (c) {
26574543Smarks 		case 'l':
26584543Smarks 			if (who_type == ZFS_DELEG_CREATE ||
26594543Smarks 			    who_type == ZFS_DELEG_NAMED_SET)
26604543Smarks 				usage(B_FALSE);
26614543Smarks 
26624543Smarks 			deleg_type |= ZFS_DELEG_PERM_LOCAL;
26634543Smarks 			break;
26644543Smarks 		case 'd':
26654543Smarks 			if (who_type == ZFS_DELEG_CREATE ||
26664543Smarks 			    who_type == ZFS_DELEG_NAMED_SET)
26674543Smarks 				usage(B_FALSE);
26684543Smarks 
26694543Smarks 			deleg_type |= ZFS_DELEG_PERM_DESCENDENT;
26704543Smarks 			break;
26714543Smarks 		case 'r':
26724543Smarks 			*recurse = B_TRUE;
26734543Smarks 			break;
26744543Smarks 		case 'c':
26754543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26764543Smarks 				usage(B_FALSE);
26774543Smarks 			if (deleg_type)
26784543Smarks 				usage(B_FALSE);
26794543Smarks 			who_type = ZFS_DELEG_CREATE;
26804543Smarks 			break;
26814543Smarks 		case 's':
26824543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26834543Smarks 				usage(B_FALSE);
26844543Smarks 			if (deleg_type)
26854543Smarks 				usage(B_FALSE);
26864543Smarks 			who_type = ZFS_DELEG_NAMED_SET;
26874543Smarks 			break;
26884543Smarks 		case 'u':
26894543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26904543Smarks 				usage(B_FALSE);
26914543Smarks 			who_type = ZFS_DELEG_USER;
26924543Smarks 			who = optarg;
26934543Smarks 			break;
26944543Smarks 		case 'g':
26954543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26964543Smarks 				usage(B_FALSE);
26974543Smarks 			who_type = ZFS_DELEG_GROUP;
26984543Smarks 			who = optarg;
26994543Smarks 			break;
27004543Smarks 		case 'e':
27014543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
27024543Smarks 				usage(B_FALSE);
27034543Smarks 			who_type = ZFS_DELEG_EVERYONE;
27044543Smarks 			break;
27054543Smarks 		default:
27064543Smarks 			usage(B_FALSE);
27074543Smarks 			break;
27084543Smarks 		}
27094543Smarks 	}
27104543Smarks 
27114543Smarks 	if (deleg_type == 0)
27124543Smarks 		deleg_type = ZFS_DELEG_PERM_LOCALDESCENDENT;
27134543Smarks 
27144543Smarks 	*argc -= optind;
27154543Smarks 	*argv += optind;
27164543Smarks 
27174543Smarks 	if (unallow == B_FALSE && *argc == 1) {
27184543Smarks 		/*
27194543Smarks 		 * Only print permissions if no options were processed
27204543Smarks 		 */
27214543Smarks 		if (optind == 1)
27224543Smarks 			return (1);
27234543Smarks 		else
27244543Smarks 			usage(B_FALSE);
27254543Smarks 	}
27264543Smarks 
27274543Smarks 	/*
27284543Smarks 	 * initialize variables for zfs_build_perms based on number
27294543Smarks 	 * of arguments.
27304543Smarks 	 * 3 arguments ==>	zfs [un]allow joe perm,perm,perm <dataset> or
27314543Smarks 	 *			zfs [un]allow -s @set1 perm,perm <dataset>
27324543Smarks 	 * 2 arguments ==>	zfs [un]allow -c perm,perm <dataset> or
27334543Smarks 	 *			zfs [un]allow -u|-g <name> perm <dataset> or
27344543Smarks 	 *			zfs [un]allow -e perm,perm <dataset>
27354543Smarks 	 *			zfs unallow joe <dataset>
27364543Smarks 	 *			zfs unallow -s @set1 <dataset>
27374543Smarks 	 * 1 argument  ==>	zfs [un]allow -e <dataset> or
27384543Smarks 	 *			zfs [un]allow -c <dataset>
27394543Smarks 	 */
27404543Smarks 
27414543Smarks 	switch (*argc) {
27424543Smarks 	case 3:
27434543Smarks 		perms = (*argv)[1];
27444543Smarks 		who = (*argv)[0];
27454543Smarks 		*ds = (*argv)[2];
27464543Smarks 
27474543Smarks 		/*
27484543Smarks 		 * advance argc/argv for do_allow cases.
27494543Smarks 		 * for do_allow case make sure who have a know who type
27504543Smarks 		 * and its not a permission set.
27514543Smarks 		 */
27524543Smarks 		if (unallow == B_TRUE) {
27534543Smarks 			*argc -= 2;
27544543Smarks 			*argv += 2;
27554543Smarks 		} else if (who_type != ZFS_DELEG_WHO_UNKNOWN &&
27564543Smarks 		    who_type != ZFS_DELEG_NAMED_SET)
27574543Smarks 			usage(B_FALSE);
27584543Smarks 		break;
27594543Smarks 
27604543Smarks 	case 2:
27614543Smarks 		if (unallow == B_TRUE && (who_type == ZFS_DELEG_EVERYONE ||
27624543Smarks 		    who_type == ZFS_DELEG_CREATE || who != NULL)) {
27634543Smarks 			perms = (*argv)[0];
27644543Smarks 			*ds = (*argv)[1];
27654543Smarks 		} else {
27664543Smarks 			if (unallow == B_FALSE &&
27674543Smarks 			    (who_type == ZFS_DELEG_WHO_UNKNOWN ||
27684543Smarks 			    who_type == ZFS_DELEG_NAMED_SET))
27694543Smarks 				usage(B_FALSE);
27704543Smarks 			else if (who_type == ZFS_DELEG_WHO_UNKNOWN ||
27714543Smarks 			    who_type == ZFS_DELEG_NAMED_SET)
27724543Smarks 				who = (*argv)[0];
27734543Smarks 			else if (who_type != ZFS_DELEG_NAMED_SET)
27744543Smarks 				perms = (*argv)[0];
27754543Smarks 			*ds = (*argv)[1];
27764543Smarks 		}
27774543Smarks 		if (unallow == B_TRUE) {
27784543Smarks 			(*argc)--;
27794543Smarks 			(*argv)++;
27804543Smarks 		}
27814543Smarks 		break;
27824543Smarks 
27834543Smarks 	case 1:
27844543Smarks 		if (unallow == B_FALSE)
27854543Smarks 			usage(B_FALSE);
27864543Smarks 		if (who == NULL && who_type != ZFS_DELEG_CREATE &&
27874543Smarks 		    who_type != ZFS_DELEG_EVERYONE)
27884543Smarks 			usage(B_FALSE);
27894543Smarks 		*ds = (*argv)[0];
27904543Smarks 		break;
27914543Smarks 
27924543Smarks 	default:
27934543Smarks 		usage(B_FALSE);
27944543Smarks 	}
27954543Smarks 
27964543Smarks 	if (strrchr(*ds, '@')) {
27974543Smarks 		(void) fprintf(stderr,
27984543Smarks 		    gettext("Can't set or remove 'allow' permissions "
27994543Smarks 		    "on snapshots.\n"));
28004543Smarks 			return (-1);
28014543Smarks 	}
28024543Smarks 
28035094Slling 	if ((zhp = zfs_open(g_zfs, *ds, ZFS_TYPE_DATASET)) == NULL)
28044543Smarks 		return (-1);
28054543Smarks 
28064543Smarks 	if ((zfs_build_perms(zhp, who, perms,
28074543Smarks 	    who_type, deleg_type, zperms)) != 0) {
28084543Smarks 		zfs_close(zhp);
28094543Smarks 		return (-1);
28104543Smarks 	}
28114543Smarks 	zfs_close(zhp);
28124543Smarks 	return (0);
28134543Smarks }
28144543Smarks 
28154543Smarks static int
28164543Smarks zfs_do_allow(int argc, char **argv)
28174543Smarks {
28184543Smarks 	char *ds;
28194543Smarks 	nvlist_t *zperms = NULL;
28204543Smarks 	zfs_handle_t *zhp;
28214543Smarks 	int unused;
28224543Smarks 	int ret;
28234543Smarks 
28244543Smarks 	if ((ret = parse_allow_args(&argc, &argv, B_FALSE, &ds,
28254543Smarks 	    &unused, &zperms)) == -1)
28264543Smarks 		return (1);
28274543Smarks 
28284543Smarks 	if (ret == 1)
28294543Smarks 		return (zfs_print_allows(argv[0]));
28304543Smarks 
28315094Slling 	if ((zhp = zfs_open(g_zfs, ds, ZFS_TYPE_DATASET)) == NULL)
28324543Smarks 		return (1);
28334543Smarks 
28344543Smarks 	if (zfs_perm_set(zhp, zperms)) {
28354543Smarks 		zfs_close(zhp);
28364543Smarks 		nvlist_free(zperms);
28374543Smarks 		return (1);
28384543Smarks 	}
28394543Smarks 	nvlist_free(zperms);
28404543Smarks 	zfs_close(zhp);
28414543Smarks 
28424543Smarks 	return (0);
28434543Smarks }
28444543Smarks 
28454543Smarks static int
28464543Smarks unallow_callback(zfs_handle_t *zhp, void *data)
28474543Smarks {
28484543Smarks 	nvlist_t *nvp = (nvlist_t *)data;
28494543Smarks 	int error;
28504543Smarks 
28514543Smarks 	error = zfs_perm_remove(zhp, nvp);
28524543Smarks 	if (error) {
28534543Smarks 		(void) fprintf(stderr, gettext("Failed to remove permissions "
28544543Smarks 		    "on %s\n"), zfs_get_name(zhp));
28554543Smarks 	}
28564543Smarks 	return (error);
28574543Smarks }
28584543Smarks 
28594543Smarks static int
28604543Smarks zfs_do_unallow(int argc, char **argv)
28614543Smarks {
28624543Smarks 	int recurse = B_FALSE;
28634543Smarks 	char *ds;
28644543Smarks 	int error;
28654543Smarks 	nvlist_t *zperms = NULL;
2866*7538SRichard.Morris@Sun.COM 	int flags = 0;
28674543Smarks 
28684543Smarks 	if (parse_allow_args(&argc, &argv, B_TRUE,
28694543Smarks 	    &ds, &recurse, &zperms) == -1)
28704543Smarks 		return (1);
28714543Smarks 
2872*7538SRichard.Morris@Sun.COM 	if (recurse)
2873*7538SRichard.Morris@Sun.COM 		flags |= ZFS_ITER_RECURSE;
2874*7538SRichard.Morris@Sun.COM 	error = zfs_for_each(argc, argv, flags,
28754543Smarks 	    ZFS_TYPE_FILESYSTEM|ZFS_TYPE_VOLUME, NULL,
2876*7538SRichard.Morris@Sun.COM 	    NULL, unallow_callback, (void *)zperms);
28774543Smarks 
28784543Smarks 	if (zperms)
28794543Smarks 		nvlist_free(zperms);
28804543Smarks 
28814543Smarks 	return (error);
2882789Sahrens }
2883789Sahrens 
28841356Seschrock typedef struct get_all_cbdata {
28851356Seschrock 	zfs_handle_t	**cb_handles;
28861356Seschrock 	size_t		cb_alloc;
28871356Seschrock 	size_t		cb_used;
28883126Sahl 	uint_t		cb_types;
28894737Smmusante 	boolean_t	cb_verbose;
28901356Seschrock } get_all_cbdata_t;
28911356Seschrock 
28924737Smmusante #define	CHECK_SPINNER 30
28934737Smmusante #define	SPINNER_TIME 3		/* seconds */
28944737Smmusante #define	MOUNT_TIME 5		/* seconds */
28954737Smmusante 
28961356Seschrock static int
28973126Sahl get_one_dataset(zfs_handle_t *zhp, void *data)
28981356Seschrock {
28994737Smmusante 	static char spin[] = { '-', '\\', '|', '/' };
29004737Smmusante 	static int spinval = 0;
29014737Smmusante 	static int spincheck = 0;
29024737Smmusante 	static time_t last_spin_time = (time_t)0;
29031356Seschrock 	get_all_cbdata_t *cbp = data;
29043126Sahl 	zfs_type_t type = zfs_get_type(zhp);
29051356Seschrock 
29064737Smmusante 	if (cbp->cb_verbose) {
29074737Smmusante 		if (--spincheck < 0) {
29084737Smmusante 			time_t now = time(NULL);
29094737Smmusante 			if (last_spin_time + SPINNER_TIME < now) {
29104737Smmusante 				(void) printf("\b%c", spin[spinval++ % 4]);
29114737Smmusante 				(void) fflush(stdout);
29124737Smmusante 				last_spin_time = now;
29134737Smmusante 			}
29144737Smmusante 			spincheck = CHECK_SPINNER;
29154737Smmusante 		}
29164737Smmusante 	}
29174737Smmusante 
29181356Seschrock 	/*
29193126Sahl 	 * Interate over any nested datasets.
29201356Seschrock 	 */
29213126Sahl 	if (type == ZFS_TYPE_FILESYSTEM &&
29223126Sahl 	    zfs_iter_filesystems(zhp, get_one_dataset, data) != 0) {
29233265Sahrens 		zfs_close(zhp);
29243126Sahl 		return (1);
29253126Sahl 	}
29263126Sahl 
29273126Sahl 	/*
29283126Sahl 	 * Skip any datasets whose type does not match.
29293126Sahl 	 */
29303126Sahl 	if ((type & cbp->cb_types) == 0) {
29311356Seschrock 		zfs_close(zhp);
29321356Seschrock 		return (0);
29331356Seschrock 	}
29341356Seschrock 
29351356Seschrock 	if (cbp->cb_alloc == cbp->cb_used) {
29361356Seschrock 		zfs_handle_t **handles;
29371356Seschrock 
29381356Seschrock 		if (cbp->cb_alloc == 0)
29391356Seschrock 			cbp->cb_alloc = 64;
29401356Seschrock 		else
29411356Seschrock 			cbp->cb_alloc *= 2;
29421356Seschrock 
29431356Seschrock 		handles = safe_malloc(cbp->cb_alloc * sizeof (void *));
29441356Seschrock 
29451356Seschrock 		if (cbp->cb_handles) {
29461356Seschrock 			bcopy(cbp->cb_handles, handles,
29471356Seschrock 			    cbp->cb_used * sizeof (void *));
29481356Seschrock 			free(cbp->cb_handles);
29491356Seschrock 		}
29501356Seschrock 
29511356Seschrock 		cbp->cb_handles = handles;
29521356Seschrock 	}
29531356Seschrock 
29541356Seschrock 	cbp->cb_handles[cbp->cb_used++] = zhp;
29551356Seschrock 
29563126Sahl 	return (0);
29571356Seschrock }
29581356Seschrock 
29591356Seschrock static void
29604737Smmusante get_all_datasets(uint_t types, zfs_handle_t ***dslist, size_t *count,
29614737Smmusante     boolean_t verbose)
29621356Seschrock {
29631356Seschrock 	get_all_cbdata_t cb = { 0 };
29643126Sahl 	cb.cb_types = types;
29654737Smmusante 	cb.cb_verbose = verbose;
29664737Smmusante 
29674737Smmusante 	if (verbose) {
29684737Smmusante 		(void) printf("%s: *", gettext("Reading ZFS config"));
29694737Smmusante 		(void) fflush(stdout);
29704737Smmusante 	}
29713126Sahl 
29723126Sahl 	(void) zfs_iter_root(g_zfs, get_one_dataset, &cb);
29733126Sahl 
29743126Sahl 	*dslist = cb.cb_handles;
29751356Seschrock 	*count = cb.cb_used;
29764737Smmusante 
29774737Smmusante 	if (verbose) {
29784737Smmusante 		(void) printf("\b%s\n", gettext("done."));
29794737Smmusante 	}
29801356Seschrock }
29811356Seschrock 
29821356Seschrock static int
29833126Sahl dataset_cmp(const void *a, const void *b)
29841356Seschrock {
29851356Seschrock 	zfs_handle_t **za = (zfs_handle_t **)a;
29861356Seschrock 	zfs_handle_t **zb = (zfs_handle_t **)b;
29871356Seschrock 	char mounta[MAXPATHLEN];
29881356Seschrock 	char mountb[MAXPATHLEN];
29893126Sahl 	boolean_t gota, gotb;
29903126Sahl 
29913126Sahl 	if ((gota = (zfs_get_type(*za) == ZFS_TYPE_FILESYSTEM)) != 0)
29923126Sahl 		verify(zfs_prop_get(*za, ZFS_PROP_MOUNTPOINT, mounta,
29933126Sahl 		    sizeof (mounta), NULL, NULL, 0, B_FALSE) == 0);
29943126Sahl 	if ((gotb = (zfs_get_type(*zb) == ZFS_TYPE_FILESYSTEM)) != 0)
29953126Sahl 		verify(zfs_prop_get(*zb, ZFS_PROP_MOUNTPOINT, mountb,
29963126Sahl 		    sizeof (mountb), NULL, NULL, 0, B_FALSE) == 0);
29973126Sahl 
29983126Sahl 	if (gota && gotb)
29993126Sahl 		return (strcmp(mounta, mountb));
30003126Sahl 
30013126Sahl 	if (gota)
30023126Sahl 		return (-1);
30033126Sahl 	if (gotb)
30043126Sahl 		return (1);
30053126Sahl 
30063126Sahl 	return (strcmp(zfs_get_name(a), zfs_get_name(b)));
30071356Seschrock }
3008789Sahrens 
3009789Sahrens /*
3010789Sahrens  * Generic callback for sharing or mounting filesystems.  Because the code is so
3011789Sahrens  * similar, we have a common function with an extra parameter to determine which
3012789Sahrens  * mode we are using.
3013789Sahrens  */
3014789Sahrens #define	OP_SHARE	0x1
3015789Sahrens #define	OP_MOUNT	0x2
3016789Sahrens 
3017789Sahrens /*
30183126Sahl  * Share or mount a dataset.
3019789Sahrens  */
3020789Sahrens static int
30215331Samw share_mount_one(zfs_handle_t *zhp, int op, int flags, char *protocol,
30225331Samw     boolean_t explicit, const char *options)
3023789Sahrens {
3024789Sahrens 	char mountpoint[ZFS_MAXPROPLEN];
3025789Sahrens 	char shareopts[ZFS_MAXPROPLEN];
30265331Samw 	char smbshareopts[ZFS_MAXPROPLEN];
30273126Sahl 	const char *cmdname = op == OP_SHARE ? "share" : "mount";
3028789Sahrens 	struct mnttab mnt;
30292676Seschrock 	uint64_t zoned, canmount;
30303126Sahl 	zfs_type_t type = zfs_get_type(zhp);
30315331Samw 	boolean_t shared_nfs, shared_smb;
30323126Sahl 
30333126Sahl 	assert(type & (ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME));
30343126Sahl 
30353126Sahl 	if (type == ZFS_TYPE_FILESYSTEM) {
30363126Sahl 		/*
30373126Sahl 		 * Check to make sure we can mount/share this dataset.  If we
30383126Sahl 		 * are in the global zone and the filesystem is exported to a
30393126Sahl 		 * local zone, or if we are in a local zone and the
30403126Sahl 		 * filesystem is not exported, then it is an error.
30413126Sahl 		 */
30423126Sahl 		zoned = zfs_prop_get_int(zhp, ZFS_PROP_ZONED);
30433126Sahl 
30443126Sahl 		if (zoned && getzoneid() == GLOBAL_ZONEID) {
30453126Sahl 			if (!explicit)
30463126Sahl 				return (0);
30473126Sahl 
30483126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': "
30493126Sahl 			    "dataset is exported to a local zone\n"), cmdname,
30503126Sahl 			    zfs_get_name(zhp));
30513126Sahl 			return (1);
30523126Sahl 
30533126Sahl 		} else if (!zoned && getzoneid() != GLOBAL_ZONEID) {
30543126Sahl 			if (!explicit)
30553126Sahl 				return (0);
30563126Sahl 
30573126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': "
30583126Sahl 			    "permission denied\n"), cmdname,
30593126Sahl 			    zfs_get_name(zhp));
30603126Sahl 			return (1);
30613126Sahl 		}
30623126Sahl 
30633126Sahl 		/*
30643126Sahl 		 * Ignore any filesystems which don't apply to us. This
30653126Sahl 		 * includes those with a legacy mountpoint, or those with
30663126Sahl 		 * legacy share options.
30673126Sahl 		 */
30683126Sahl 		verify(zfs_prop_get(zhp, ZFS_PROP_MOUNTPOINT, mountpoint,
30693126Sahl 		    sizeof (mountpoint), NULL, NULL, 0, B_FALSE) == 0);
30703126Sahl 		verify(zfs_prop_get(zhp, ZFS_PROP_SHARENFS, shareopts,
30713126Sahl 		    sizeof (shareopts), NULL, NULL, 0, B_FALSE) == 0);
30725331Samw 		verify(zfs_prop_get(zhp, ZFS_PROP_SHARESMB, smbshareopts,
30735331Samw 		    sizeof (smbshareopts), NULL, NULL, 0, B_FALSE) == 0);
30743126Sahl 		canmount = zfs_prop_get_int(zhp, ZFS_PROP_CANMOUNT);
30753126Sahl 
30765331Samw 		if (op == OP_SHARE && strcmp(shareopts, "off") == 0 &&
30775331Samw 		    strcmp(smbshareopts, "off") == 0) {
30783126Sahl 			if (!explicit)
3079789Sahrens 				return (0);
3080789Sahrens 
3081789Sahrens 			(void) fprintf(stderr, gettext("cannot share '%s': "
3082789Sahrens 			    "legacy share\n"), zfs_get_name(zhp));
3083789Sahrens 			(void) fprintf(stderr, gettext("use share(1M) to "
3084789Sahrens 			    "share this filesystem\n"));
3085789Sahrens 			return (1);
3086789Sahrens 		}
30873126Sahl 
30883126Sahl 		/*
30893126Sahl 		 * We cannot share or mount legacy filesystems. If the
30903126Sahl 		 * shareopts is non-legacy but the mountpoint is legacy, we
30913126Sahl 		 * treat it as a legacy share.
30923126Sahl 		 */
30933126Sahl 		if (strcmp(mountpoint, "legacy") == 0) {
30943126Sahl 			if (!explicit)
30953126Sahl 				return (0);
30963126Sahl 
30973126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': "
30983126Sahl 			    "legacy mountpoint\n"), cmdname, zfs_get_name(zhp));
30995367Sahrens 			(void) fprintf(stderr, gettext("use %s(1M) to "
31005367Sahrens 			    "%s this filesystem\n"), cmdname, cmdname);
31013126Sahl 			return (1);
31023126Sahl 		}
31033126Sahl 
31043126Sahl 		if (strcmp(mountpoint, "none") == 0) {
31053126Sahl 			if (!explicit)
31063126Sahl 				return (0);
31073126Sahl 
31083126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': no "
31093126Sahl 			    "mountpoint set\n"), cmdname, zfs_get_name(zhp));
31103126Sahl 			return (1);
31113126Sahl 		}
31123126Sahl 
31136168Shs24103 		/*
31146168Shs24103 		 * canmount	explicit	outcome
31156168Shs24103 		 * on		no		pass through
31166168Shs24103 		 * on		yes		pass through
31176168Shs24103 		 * off		no		return 0
31186168Shs24103 		 * off		yes		display error, return 1
31196168Shs24103 		 * noauto	no		return 0
31206168Shs24103 		 * noauto	yes		pass through
31216168Shs24103 		 */
31226168Shs24103 		if (canmount == ZFS_CANMOUNT_OFF) {
31233126Sahl 			if (!explicit)
31243126Sahl 				return (0);
31253126Sahl 
31263126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': "
31273126Sahl 			    "'canmount' property is set to 'off'\n"), cmdname,
31283126Sahl 			    zfs_get_name(zhp));
31293126Sahl 			return (1);
31306168Shs24103 		} else if (canmount == ZFS_CANMOUNT_NOAUTO && !explicit) {
31316168Shs24103 			return (0);
31323126Sahl 		}
31333126Sahl 
31343126Sahl 		/*
31353126Sahl 		 * At this point, we have verified that the mountpoint and/or
31363126Sahl 		 * shareopts are appropriate for auto management. If the
31373126Sahl 		 * filesystem is already mounted or shared, return (failing
31383126Sahl 		 * for explicit requests); otherwise mount or share the
31393126Sahl 		 * filesystem.
31403126Sahl 		 */
31413126Sahl 		switch (op) {
31423126Sahl 		case OP_SHARE:
31435331Samw 
31445331Samw 			shared_nfs = zfs_is_shared_nfs(zhp, NULL);
31455331Samw 			shared_smb = zfs_is_shared_smb(zhp, NULL);
31465331Samw 
31475331Samw 			if (shared_nfs && shared_smb ||
31485331Samw 			    (shared_nfs && strcmp(shareopts, "on") == 0 &&
31495331Samw 			    strcmp(smbshareopts, "off") == 0) ||
31505331Samw 			    (shared_smb && strcmp(smbshareopts, "on") == 0 &&
31515331Samw 			    strcmp(shareopts, "off") == 0)) {
31523126Sahl 				if (!explicit)
31533126Sahl 					return (0);
31543126Sahl 
3155789Sahrens 				(void) fprintf(stderr, gettext("cannot share "
3156789Sahrens 				    "'%s': filesystem already shared\n"),
3157789Sahrens 				    zfs_get_name(zhp));
3158789Sahrens 				return (1);
3159789Sahrens 			}
31603126Sahl 
31613126Sahl 			if (!zfs_is_mounted(zhp, NULL) &&
31623126Sahl 			    zfs_mount(zhp, NULL, 0) != 0)
31633126Sahl 				return (1);
31643126Sahl 
31655331Samw 			if (protocol == NULL) {
31665331Samw 				if (zfs_shareall(zhp) != 0)
31675331Samw 					return (1);
31685331Samw 			} else if (strcmp(protocol, "nfs") == 0) {
31695331Samw 				if (zfs_share_nfs(zhp))
31705331Samw 					return (1);
31715331Samw 			} else if (strcmp(protocol, "smb") == 0) {
31725331Samw 				if (zfs_share_smb(zhp))
31735331Samw 					return (1);
31745331Samw 			} else {
31755331Samw 				(void) fprintf(stderr, gettext("cannot share "
31765331Samw 				    "'%s': invalid share type '%s' "
31775331Samw 				    "specified\n"),
31785331Samw 				    zfs_get_name(zhp), protocol);
31793126Sahl 				return (1);
31805331Samw 			}
31815331Samw 
31823126Sahl 			break;
31833126Sahl 
31843126Sahl 		case OP_MOUNT:
31853126Sahl 			if (options == NULL)
31863126Sahl 				mnt.mnt_mntopts = "";
31873126Sahl 			else
31883126Sahl 				mnt.mnt_mntopts = (char *)options;
31893126Sahl 
31903126Sahl 			if (!hasmntopt(&mnt, MNTOPT_REMOUNT) &&
31913126Sahl 			    zfs_is_mounted(zhp, NULL)) {
31923126Sahl 				if (!explicit)
31933126Sahl 					return (0);
31943126Sahl 
3195789Sahrens 				(void) fprintf(stderr, gettext("cannot mount "
3196789Sahrens 				    "'%s': filesystem already mounted\n"),
3197789Sahrens 				    zfs_get_name(zhp));
3198789Sahrens 				return (1);
3199789Sahrens 			}
32003126Sahl 
32013126Sahl 			if (zfs_mount(zhp, options, flags) != 0)
32023126Sahl 				return (1);
32033126Sahl 			break;
3204789Sahrens 		}
32053126Sahl 	} else {
32063126Sahl 		assert(op == OP_SHARE);
32073126Sahl 
32083126Sahl 		/*
32093126Sahl 		 * Ignore any volumes that aren't shared.
32103126Sahl 		 */
32113126Sahl 		verify(zfs_prop_get(zhp, ZFS_PROP_SHAREISCSI, shareopts,
32123126Sahl 		    sizeof (shareopts), NULL, NULL, 0, B_FALSE) == 0);
32133126Sahl 
32143126Sahl 		if (strcmp(shareopts, "off") == 0) {
32153126Sahl 			if (!explicit)
32163126Sahl 				return (0);
32173126Sahl 
32183126Sahl 			(void) fprintf(stderr, gettext("cannot share '%s': "
32193126Sahl 			    "'shareiscsi' property not set\n"),
32203126Sahl 			    zfs_get_name(zhp));
32213126Sahl 			(void) fprintf(stderr, gettext("set 'shareiscsi' "
32223126Sahl 			    "property or use iscsitadm(1M) to share this "
32233126Sahl 			    "volume\n"));
32243126Sahl 			return (1);
3225789Sahrens 		}
32263126Sahl 
32273126Sahl 		if (zfs_is_shared_iscsi(zhp)) {
32283126Sahl 			if (!explicit)
32293126Sahl 				return (0);
32303126Sahl 
32313126Sahl 			(void) fprintf(stderr, gettext("cannot share "
32323126Sahl 			    "'%s': volume already shared\n"),
32333126Sahl 			    zfs_get_name(zhp));
3234789Sahrens 			return (1);
32353126Sahl 		}
32363126Sahl 
32373126Sahl 		if (zfs_share_iscsi(zhp) != 0)
32383126Sahl 			return (1);
3239789Sahrens 	}
3240789Sahrens 
3241789Sahrens 	return (0);
3242789Sahrens }
3243789Sahrens 
32444737Smmusante /*
32454737Smmusante  * Reports progress in the form "(current/total)".  Not thread-safe.
32464737Smmusante  */
32474737Smmusante static void
32484737Smmusante report_mount_progress(int current, int total)
32494737Smmusante {
32504737Smmusante 	static int len;
32514737Smmusante 	static char *reverse = "\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b"
32524737Smmusante 	    "\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b";
32534737Smmusante 	static time_t last_progress_time;
32544737Smmusante 	time_t now = time(NULL);
32554737Smmusante 
32564737Smmusante 	/* report 1..n instead of 0..n-1 */
32574737Smmusante 	++current;
32584737Smmusante 
32594737Smmusante 	/* display header if we're here for the first time */
32604737Smmusante 	if (current == 1) {
32614737Smmusante 		(void) printf(gettext("Mounting ZFS filesystems: "));
32624737Smmusante 		len = 0;
32635367Sahrens 	} else if (current != total && last_progress_time + MOUNT_TIME >= now) {
32645367Sahrens 		/* too soon to report again */
32655367Sahrens 		return;
32665367Sahrens 	}
32674737Smmusante 
32684737Smmusante 	last_progress_time = now;
32694737Smmusante 
32704737Smmusante 	/* back up to prepare for overwriting */
32714737Smmusante 	if (len)
32724737Smmusante 		(void) printf("%*.*s", len, len, reverse);
32734737Smmusante 
32744737Smmusante 	/* We put a newline at the end if this is the last one.  */
32754737Smmusante 	len = printf("(%d/%d)%s", current, total, current == total ? "\n" : "");
32764737Smmusante 	(void) fflush(stdout);
32774737Smmusante }
32784737Smmusante 
32796289Smmusante static void
32806289Smmusante append_options(char *mntopts, char *newopts)
32816289Smmusante {
32826289Smmusante 	int len = strlen(mntopts);
32836289Smmusante 
32846289Smmusante 	/* original length plus new string to append plus 1 for the comma */
32856289Smmusante 	if (len + 1 + strlen(newopts) >= MNT_LINE_MAX) {
32866289Smmusante 		(void) fprintf(stderr, gettext("the opts argument for "
32876289Smmusante 		    "'%c' option is too long (more than %d chars)\n"),
32886289Smmusante 		    "-o", MNT_LINE_MAX);
32896289Smmusante 		usage(B_FALSE);
32906289Smmusante 	}
32916289Smmusante 
32926289Smmusante 	if (*mntopts)
32936289Smmusante 		mntopts[len++] = ',';
32946289Smmusante 
32956289Smmusante 	(void) strcpy(&mntopts[len], newopts);
32966289Smmusante }
32976289Smmusante 
3298789Sahrens static int
32993126Sahl share_mount(int op, int argc, char **argv)
3300789Sahrens {
3301789Sahrens 	int do_all = 0;
33024737Smmusante 	boolean_t verbose = B_FALSE;
33032372Slling 	int c, ret = 0;
33046289Smmusante 	char *options = NULL;
33053126Sahl 	int types, flags = 0;
3306789Sahrens 
3307789Sahrens 	/* check options */
33084737Smmusante 	while ((c = getopt(argc, argv, op == OP_MOUNT ? ":avo:O" : "a"))
3309789Sahrens 	    != -1) {
3310789Sahrens 		switch (c) {
3311789Sahrens 		case 'a':
3312789Sahrens 			do_all = 1;
3313789Sahrens 			break;
33144737Smmusante 		case 'v':
33154737Smmusante 			verbose = B_TRUE;
33164737Smmusante 			break;
3317789Sahrens 		case 'o':
33186289Smmusante 			if (*optarg == '\0') {
33196289Smmusante 				(void) fprintf(stderr, gettext("empty mount "
33206289Smmusante 				    "options (-o) specified\n"));
33216289Smmusante 				usage(B_FALSE);
33224717Srm160521 			}
33236289Smmusante 
33246289Smmusante 			if (options == NULL)
33256289Smmusante 				options = safe_malloc(MNT_LINE_MAX + 1);
33266289Smmusante 
33276289Smmusante 			/* option validation is done later */
33286289Smmusante 			append_options(options, optarg);
3329789Sahrens 			break;
33304717Srm160521 
3331789Sahrens 		case 'O':
33323126Sahl 			flags |= MS_OVERLAY;
3333789Sahrens 			break;
3334789Sahrens 		case ':':
3335789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
3336789Sahrens 			    "'%c' option\n"), optopt);
33372082Seschrock 			usage(B_FALSE);
3338789Sahrens 			break;
3339789Sahrens 		case '?':
3340789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
3341789Sahrens 			    optopt);
33422082Seschrock 			usage(B_FALSE);
3343789Sahrens 		}
3344789Sahrens 	}
3345789Sahrens 
3346789Sahrens 	argc -= optind;
3347789Sahrens 	argv += optind;
3348789Sahrens 
3349789Sahrens 	/* check number of arguments */
3350789Sahrens 	if (do_all) {
33513126Sahl 		zfs_handle_t **dslist = NULL;
33521356Seschrock 		size_t i, count = 0;
33535331Samw 		char *protocol = NULL;
33541356Seschrock 
33553126Sahl 		if (op == OP_MOUNT) {
33563126Sahl 			types = ZFS_TYPE_FILESYSTEM;
33573126Sahl 		} else if (argc > 0) {
33585331Samw 			if (strcmp(argv[0], "nfs") == 0 ||
33595331Samw 			    strcmp(argv[0], "smb") == 0) {
33603126Sahl 				types = ZFS_TYPE_FILESYSTEM;
33613126Sahl 			} else if (strcmp(argv[0], "iscsi") == 0) {
33623126Sahl 				types = ZFS_TYPE_VOLUME;
33633126Sahl 			} else {
33643126Sahl 				(void) fprintf(stderr, gettext("share type "
33655331Samw 				    "must be 'nfs', 'smb' or 'iscsi'\n"));
33663126Sahl 				usage(B_FALSE);
33673126Sahl 			}
33685331Samw 			protocol = argv[0];
33693126Sahl 			argc--;
33703126Sahl 			argv++;
33713126Sahl 		} else {
33723126Sahl 			types = ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME;
33733126Sahl 		}
33743126Sahl 
3375789Sahrens 		if (argc != 0) {
3376789Sahrens 			(void) fprintf(stderr, gettext("too many arguments\n"));
33772082Seschrock 			usage(B_FALSE);
3378789Sahrens 		}
3379789Sahrens 
33804737Smmusante 		get_all_datasets(types, &dslist, &count, verbose);
33811356Seschrock 
33821356Seschrock 		if (count == 0)
33831356Seschrock 			return (0);
33841356Seschrock 
33853126Sahl 		qsort(dslist, count, sizeof (void *), dataset_cmp);
33861356Seschrock 
33871356Seschrock 		for (i = 0; i < count; i++) {
33884737Smmusante 			if (verbose)
33894737Smmusante 				report_mount_progress(i, count);
33904737Smmusante 
33915331Samw 			if (share_mount_one(dslist[i], op, flags, protocol,
33925331Samw 			    B_FALSE, options) != 0)
33932369Slling 				ret = 1;
33943126Sahl 			zfs_close(dslist[i]);
33951356Seschrock 		}
33961356Seschrock 
33973126Sahl 		free(dslist);
3398789Sahrens 	} else if (argc == 0) {
3399789Sahrens 		struct mnttab entry;
3400789Sahrens 
34014717Srm160521 		if ((op == OP_SHARE) || (options != NULL)) {
3402789Sahrens 			(void) fprintf(stderr, gettext("missing filesystem "
34034717Srm160521 			    "argument (specify -a for all)\n"));
34042082Seschrock 			usage(B_FALSE);
3405789Sahrens 		}
3406789Sahrens 
3407789Sahrens 		/*
3408789Sahrens 		 * When mount is given no arguments, go through /etc/mnttab and
3409789Sahrens 		 * display any active ZFS mounts.  We hide any snapshots, since
3410789Sahrens 		 * they are controlled automatically.
3411789Sahrens 		 */
3412789Sahrens 		rewind(mnttab_file);
3413789Sahrens 		while (getmntent(mnttab_file, &entry) == 0) {
3414789Sahrens 			if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0 ||
3415789Sahrens 			    strchr(entry.mnt_special, '@') != NULL)
3416789Sahrens 				continue;
3417789Sahrens 
3418789Sahrens 			(void) printf("%-30s  %s\n", entry.mnt_special,
3419789Sahrens 			    entry.mnt_mountp);
3420789Sahrens 		}
3421789Sahrens 
3422789Sahrens 	} else {
3423789Sahrens 		zfs_handle_t *zhp;
3424789Sahrens 
34253126Sahl 		types = ZFS_TYPE_FILESYSTEM;
34263126Sahl 		if (op == OP_SHARE)
34273126Sahl 			types |= ZFS_TYPE_VOLUME;
34283126Sahl 
3429789Sahrens 		if (argc > 1) {
3430789Sahrens 			(void) fprintf(stderr,
3431789Sahrens 			    gettext("too many arguments\n"));
34322082Seschrock 			usage(B_FALSE);
3433789Sahrens 		}
3434789Sahrens 
34353126Sahl 		if ((zhp = zfs_open(g_zfs, argv[0], types)) == NULL) {
3436789Sahrens 			ret = 1;
34373126Sahl 		} else {
34385331Samw 			ret = share_mount_one(zhp, op, flags, NULL, B_TRUE,
34393126Sahl 			    options);
3440789Sahrens 			zfs_close(zhp);
3441789Sahrens 		}
3442789Sahrens 	}
3443789Sahrens 
3444789Sahrens 	return (ret);
3445789Sahrens }
3446789Sahrens 
3447789Sahrens /*
34483126Sahl  * zfs mount -a [nfs | iscsi]
3449789Sahrens  * zfs mount filesystem
3450789Sahrens  *
3451789Sahrens  * Mount all filesystems, or mount the given filesystem.
3452789Sahrens  */
3453789Sahrens static int
3454789Sahrens zfs_do_mount(int argc, char **argv)
3455789Sahrens {
34563126Sahl 	return (share_mount(OP_MOUNT, argc, argv));
3457789Sahrens }
3458789Sahrens 
3459789Sahrens /*
34605331Samw  * zfs share -a [nfs | iscsi | smb]
3461789Sahrens  * zfs share filesystem
3462789Sahrens  *
3463789Sahrens  * Share all filesystems, or share the given filesystem.
3464789Sahrens  */
3465789Sahrens static int
3466789Sahrens zfs_do_share(int argc, char **argv)
3467789Sahrens {
34683126Sahl 	return (share_mount(OP_SHARE, argc, argv));
3469789Sahrens }
3470789Sahrens 
3471789Sahrens typedef struct unshare_unmount_node {
3472789Sahrens 	zfs_handle_t	*un_zhp;
3473789Sahrens 	char		*un_mountp;
3474789Sahrens 	uu_avl_node_t	un_avlnode;
3475789Sahrens } unshare_unmount_node_t;
3476789Sahrens 
3477789Sahrens /* ARGSUSED */
3478789Sahrens static int
3479789Sahrens unshare_unmount_compare(const void *larg, const void *rarg, void *unused)
3480789Sahrens {
3481789Sahrens 	const unshare_unmount_node_t *l = larg;
3482789Sahrens 	const unshare_unmount_node_t *r = rarg;
3483789Sahrens 
3484789Sahrens 	return (strcmp(l->un_mountp, r->un_mountp));
3485789Sahrens }
3486789Sahrens 
3487789Sahrens /*
3488789Sahrens  * Convenience routine used by zfs_do_umount() and manual_unmount().  Given an
3489789Sahrens  * absolute path, find the entry /etc/mnttab, verify that its a ZFS filesystem,
3490789Sahrens  * and unmount it appropriately.
3491789Sahrens  */
3492789Sahrens static int
34933126Sahl unshare_unmount_path(int op, char *path, int flags, boolean_t is_manual)
3494789Sahrens {
3495789Sahrens 	zfs_handle_t *zhp;
3496789Sahrens 	int ret;
3497789Sahrens 	struct stat64 statbuf;
3498789Sahrens 	struct extmnttab entry;
34993126Sahl 	const char *cmdname = (op == OP_SHARE) ? "unshare" : "unmount";
35006289Smmusante 	ino_t path_inode;
3501789Sahrens 
3502789Sahrens 	/*
3503789Sahrens 	 * Search for the path in /etc/mnttab.  Rather than looking for the
3504789Sahrens 	 * specific path, which can be fooled by non-standard paths (i.e. ".."
3505789Sahrens 	 * or "//"), we stat() the path and search for the corresponding
3506789Sahrens 	 * (major,minor) device pair.
3507789Sahrens 	 */
3508789Sahrens 	if (stat64(path, &statbuf) != 0) {
3509789Sahrens 		(void) fprintf(stderr, gettext("cannot %s '%s': %s\n"),
3510789Sahrens 		    cmdname, path, strerror(errno));
3511789Sahrens 		return (1);
3512789Sahrens 	}
35136289Smmusante 	path_inode = statbuf.st_ino;
3514789Sahrens 
3515789Sahrens 	/*
3516789Sahrens 	 * Search for the given (major,minor) pair in the mount table.
3517789Sahrens 	 */
3518789Sahrens 	rewind(mnttab_file);
3519789Sahrens 	while ((ret = getextmntent(mnttab_file, &entry, 0)) == 0) {
3520789Sahrens 		if (entry.mnt_major == major(statbuf.st_dev) &&
3521789Sahrens 		    entry.mnt_minor == minor(statbuf.st_dev))
3522789Sahrens 			break;
3523789Sahrens 	}
3524789Sahrens 	if (ret != 0) {
35255904Stimh 		if (op == OP_SHARE) {
35265904Stimh 			(void) fprintf(stderr, gettext("cannot %s '%s': not "
35275904Stimh 			    "currently mounted\n"), cmdname, path);
35285904Stimh 			return (1);
35295904Stimh 		}
35305904Stimh 		(void) fprintf(stderr, gettext("warning: %s not in mnttab\n"),
35315904Stimh 		    path);
35325904Stimh 		if ((ret = umount2(path, flags)) != 0)
35335904Stimh 			(void) fprintf(stderr, gettext("%s: %s\n"), path,
35345904Stimh 			    strerror(errno));
35355904Stimh 		return (ret != 0);
3536789Sahrens 	}
3537789Sahrens 
3538789Sahrens 	if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0) {
3539789Sahrens 		(void) fprintf(stderr, gettext("cannot %s '%s': not a ZFS "
3540789Sahrens 		    "filesystem\n"), cmdname, path);
3541789Sahrens 		return (1);
3542789Sahrens 	}
3543789Sahrens 
35442082Seschrock 	if ((zhp = zfs_open(g_zfs, entry.mnt_special,
35452082Seschrock 	    ZFS_TYPE_FILESYSTEM)) == NULL)
3546789Sahrens 		return (1);
3547789Sahrens 
35486289Smmusante 	ret = 1;
35497386SSachin.Gaikwad@Sun.COM 	if (stat64(entry.mnt_mountp, &statbuf) != 0) {
35507386SSachin.Gaikwad@Sun.COM 		(void) fprintf(stderr, gettext("cannot %s '%s': %s\n"),
35517386SSachin.Gaikwad@Sun.COM 		    cmdname, path, strerror(errno));
35527386SSachin.Gaikwad@Sun.COM 		goto out;
35537386SSachin.Gaikwad@Sun.COM 	} else if (statbuf.st_ino != path_inode) {
35547386SSachin.Gaikwad@Sun.COM 		(void) fprintf(stderr, gettext("cannot "
35557386SSachin.Gaikwad@Sun.COM 		    "%s '%s': not a mountpoint\n"), cmdname, path);
35567386SSachin.Gaikwad@Sun.COM 		goto out;
35577386SSachin.Gaikwad@Sun.COM 	}
35587386SSachin.Gaikwad@Sun.COM 
35593126Sahl 	if (op == OP_SHARE) {
35606289Smmusante 		char nfs_mnt_prop[ZFS_MAXPROPLEN];
35616289Smmusante 		char smbshare_prop[ZFS_MAXPROPLEN];
35626289Smmusante 
35636289Smmusante 		verify(zfs_prop_get(zhp, ZFS_PROP_SHARENFS, nfs_mnt_prop,
35646289Smmusante 		    sizeof (nfs_mnt_prop), NULL, NULL, 0, B_FALSE) == 0);
35656289Smmusante 		verify(zfs_prop_get(zhp, ZFS_PROP_SHARESMB, smbshare_prop,
35666289Smmusante 		    sizeof (smbshare_prop), NULL, NULL, 0, B_FALSE) == 0);
35676289Smmusante 
35685331Samw 		if (strcmp(nfs_mnt_prop, "off") == 0 &&
35695331Samw 		    strcmp(smbshare_prop, "off") == 0) {
3570789Sahrens 			(void) fprintf(stderr, gettext("cannot unshare "
3571789Sahrens 			    "'%s': legacy share\n"), path);
3572789Sahrens 			(void) fprintf(stderr, gettext("use "
3573789Sahrens 			    "unshare(1M) to unshare this filesystem\n"));
35745331Samw 		} else if (!zfs_is_shared(zhp)) {
3575789Sahrens 			(void) fprintf(stderr, gettext("cannot unshare '%s': "
3576789Sahrens 			    "not currently shared\n"), path);
3577789Sahrens 		} else {
35785331Samw 			ret = zfs_unshareall_bypath(zhp, path);
3579789Sahrens 		}
3580789Sahrens 	} else {
35816289Smmusante 		char mtpt_prop[ZFS_MAXPROPLEN];
35826289Smmusante 
35836289Smmusante 		verify(zfs_prop_get(zhp, ZFS_PROP_MOUNTPOINT, mtpt_prop,
35846289Smmusante 		    sizeof (mtpt_prop), NULL, NULL, 0, B_FALSE) == 0);
35856289Smmusante 
35867386SSachin.Gaikwad@Sun.COM 		if (is_manual) {
35871264Slling 			ret = zfs_unmount(zhp, NULL, flags);
35886289Smmusante 		} else if (strcmp(mtpt_prop, "legacy") == 0) {
35891264Slling 			(void) fprintf(stderr, gettext("cannot unmount "
35901264Slling 			    "'%s': legacy mountpoint\n"),
35911264Slling 			    zfs_get_name(zhp));
35921264Slling 			(void) fprintf(stderr, gettext("use umount(1M) "
35931264Slling 			    "to unmount this filesystem\n"));
3594789Sahrens 		} else {
3595789Sahrens 			ret = zfs_unmountall(zhp, flags);
3596789Sahrens 		}
3597789Sahrens 	}
3598789Sahrens 
35997386SSachin.Gaikwad@Sun.COM out:
3600789Sahrens 	zfs_close(zhp);
3601789Sahrens 
3602789Sahrens 	return (ret != 0);
3603789Sahrens }
3604789Sahrens 
3605789Sahrens /*
3606789Sahrens  * Generic callback for unsharing or unmounting a filesystem.
3607789Sahrens  */
3608789Sahrens static int
36093126Sahl unshare_unmount(int op, int argc, char **argv)
3610789Sahrens {
3611789Sahrens 	int do_all = 0;
3612789Sahrens 	int flags = 0;
3613789Sahrens 	int ret = 0;
36143126Sahl 	int types, c;
3615789Sahrens 	zfs_handle_t *zhp;
36165331Samw 	char nfsiscsi_mnt_prop[ZFS_MAXPROPLEN];
36175331Samw 	char sharesmb[ZFS_MAXPROPLEN];
3618789Sahrens 
3619789Sahrens 	/* check options */
36203126Sahl 	while ((c = getopt(argc, argv, op == OP_SHARE ? "a" : "af")) != -1) {
3621789Sahrens 		switch (c) {
3622789Sahrens 		case 'a':
3623789Sahrens 			do_all = 1;
3624789Sahrens 			break;
3625789Sahrens 		case 'f':
3626789Sahrens 			flags = MS_FORCE;
3627789Sahrens 			break;
3628789Sahrens 		case '?':
3629789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
3630789Sahrens 			    optopt);
36312082Seschrock 			usage(B_FALSE);
3632789Sahrens 		}
3633789Sahrens 	}
3634789Sahrens 
3635789Sahrens 	argc -= optind;
3636789Sahrens 	argv += optind;
3637789Sahrens 
3638789Sahrens 	if (do_all) {
3639789Sahrens 		/*
3640789Sahrens 		 * We could make use of zfs_for_each() to walk all datasets in
3641789Sahrens 		 * the system, but this would be very inefficient, especially
3642789Sahrens 		 * since we would have to linearly search /etc/mnttab for each
3643789Sahrens 		 * one.  Instead, do one pass through /etc/mnttab looking for
3644789Sahrens 		 * zfs entries and call zfs_unmount() for each one.
3645789Sahrens 		 *
3646789Sahrens 		 * Things get a little tricky if the administrator has created
3647789Sahrens 		 * mountpoints beneath other ZFS filesystems.  In this case, we
3648789Sahrens 		 * have to unmount the deepest filesystems first.  To accomplish
3649789Sahrens 		 * this, we place all the mountpoints in an AVL tree sorted by
3650789Sahrens 		 * the special type (dataset name), and walk the result in
3651789Sahrens 		 * reverse to make sure to get any snapshots first.
3652789Sahrens 		 */
3653789Sahrens 		struct mnttab entry;
3654789Sahrens 		uu_avl_pool_t *pool;
3655789Sahrens 		uu_avl_t *tree;
3656789Sahrens 		unshare_unmount_node_t *node;
3657789Sahrens 		uu_avl_index_t idx;
3658789Sahrens 		uu_avl_walk_t *walk;
3659789Sahrens 
36603126Sahl 		if (argc != 0) {
36613126Sahl 			(void) fprintf(stderr, gettext("too many arguments\n"));
36623126Sahl 			usage(B_FALSE);
36633126Sahl 		}
36643126Sahl 
3665789Sahrens 		if ((pool = uu_avl_pool_create("unmount_pool",
3666789Sahrens 		    sizeof (unshare_unmount_node_t),
3667789Sahrens 		    offsetof(unshare_unmount_node_t, un_avlnode),
3668789Sahrens 		    unshare_unmount_compare,
3669789Sahrens 		    UU_DEFAULT)) == NULL) {
3670789Sahrens 			(void) fprintf(stderr, gettext("internal error: "
3671789Sahrens 			    "out of memory\n"));
3672789Sahrens 			exit(1);
3673789Sahrens 		}
3674789Sahrens 
3675789Sahrens 		if ((tree = uu_avl_create(pool, NULL, UU_DEFAULT)) == NULL) {
3676789Sahrens 			(void) fprintf(stderr, gettext("internal error: "
3677789Sahrens 			    "out of memory\n"));
3678789Sahrens 			exit(1);
3679789Sahrens 		}
3680789Sahrens 
3681789Sahrens 		rewind(mnttab_file);
3682789Sahrens 		while (getmntent(mnttab_file, &entry) == 0) {
3683789Sahrens 
3684789Sahrens 			/* ignore non-ZFS entries */
3685789Sahrens 			if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0)
3686789Sahrens 				continue;
3687789Sahrens 
3688789Sahrens 			/* ignore snapshots */
3689789Sahrens 			if (strchr(entry.mnt_special, '@') != NULL)
3690789Sahrens 				continue;
3691789Sahrens 
36922082Seschrock 			if ((zhp = zfs_open(g_zfs, entry.mnt_special,
3693789Sahrens 			    ZFS_TYPE_FILESYSTEM)) == NULL) {
3694789Sahrens 				ret = 1;
3695789Sahrens 				continue;
3696789Sahrens 			}
3697789Sahrens 
36985331Samw 			switch (op) {
36995331Samw 			case OP_SHARE:
37005331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_SHARENFS,
37015331Samw 				    nfsiscsi_mnt_prop,
37025331Samw 				    sizeof (nfsiscsi_mnt_prop),
37035331Samw 				    NULL, NULL, 0, B_FALSE) == 0);
37045331Samw 				if (strcmp(nfsiscsi_mnt_prop, "off") != 0)
37055331Samw 					break;
37065331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_SHARESMB,
37075331Samw 				    nfsiscsi_mnt_prop,
37085331Samw 				    sizeof (nfsiscsi_mnt_prop),
37095331Samw 				    NULL, NULL, 0, B_FALSE) == 0);
37105331Samw 				if (strcmp(nfsiscsi_mnt_prop, "off") == 0)
37115331Samw 					continue;
37125331Samw 				break;
37135331Samw 			case OP_MOUNT:
37145331Samw 				/* Ignore legacy mounts */
37155331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_MOUNTPOINT,
37165331Samw 				    nfsiscsi_mnt_prop,
37175331Samw 				    sizeof (nfsiscsi_mnt_prop),
37185331Samw 				    NULL, NULL, 0, B_FALSE) == 0);
37195331Samw 				if (strcmp(nfsiscsi_mnt_prop, "legacy") == 0)
37205331Samw 					continue;
37216168Shs24103 				/* Ignore canmount=noauto mounts */
37226168Shs24103 				if (zfs_prop_get_int(zhp, ZFS_PROP_CANMOUNT) ==
37236168Shs24103 				    ZFS_CANMOUNT_NOAUTO)
37246168Shs24103 					continue;
37255331Samw 			default:
37265331Samw 				break;
3727789Sahrens 			}
3728789Sahrens 
3729789Sahrens 			node = safe_malloc(sizeof (unshare_unmount_node_t));
3730789Sahrens 			node->un_zhp = zhp;
3731789Sahrens 
3732789Sahrens 			if ((node->un_mountp = strdup(entry.mnt_mountp)) ==
3733789Sahrens 			    NULL) {
3734789Sahrens 				(void) fprintf(stderr, gettext("internal error:"
3735789Sahrens 				    " out of memory\n"));
3736789Sahrens 				exit(1);
3737789Sahrens 			}
3738789Sahrens 
3739789Sahrens 			uu_avl_node_init(node, &node->un_avlnode, pool);
3740789Sahrens 
3741789Sahrens 			if (uu_avl_find(tree, node, NULL, &idx) == NULL) {
3742789Sahrens 				uu_avl_insert(tree, node, idx);
3743789Sahrens 			} else {
3744789Sahrens 				zfs_close(node->un_zhp);
3745789Sahrens 				free(node->un_mountp);
3746789Sahrens 				free(node);
3747789Sahrens 			}
3748789Sahrens 		}
3749789Sahrens 
3750789Sahrens 		/*
3751789Sahrens 		 * Walk the AVL tree in reverse, unmounting each filesystem and
3752789Sahrens 		 * removing it from the AVL tree in the process.
3753789Sahrens 		 */
3754789Sahrens 		if ((walk = uu_avl_walk_start(tree,
3755789Sahrens 		    UU_WALK_REVERSE | UU_WALK_ROBUST)) == NULL) {
3756789Sahrens 			(void) fprintf(stderr,
3757789Sahrens 			    gettext("internal error: out of memory"));
3758789Sahrens 			exit(1);
3759789Sahrens 		}
3760789Sahrens 
3761789Sahrens 		while ((node = uu_avl_walk_next(walk)) != NULL) {
3762789Sahrens 			uu_avl_remove(tree, node);
3763789Sahrens 
37643126Sahl 			switch (op) {
3765789Sahrens 			case OP_SHARE:
37665331Samw 				if (zfs_unshareall_bypath(node->un_zhp,
3767789Sahrens 				    node->un_mountp) != 0)
3768789Sahrens 					ret = 1;
3769789Sahrens 				break;
3770789Sahrens 
3771789Sahrens 			case OP_MOUNT:
3772789Sahrens 				if (zfs_unmount(node->un_zhp,
3773789Sahrens 				    node->un_mountp, flags) != 0)
3774789Sahrens 					ret = 1;
3775789Sahrens 				break;
3776789Sahrens 			}
3777789Sahrens 
3778789Sahrens 			zfs_close(node->un_zhp);
3779789Sahrens 			free(node->un_mountp);
3780789Sahrens 			free(node);
3781789Sahrens 		}
3782789Sahrens 
3783789Sahrens 		uu_avl_walk_end(walk);
3784789Sahrens 		uu_avl_destroy(tree);
3785789Sahrens 		uu_avl_pool_destroy(pool);
37863126Sahl 
37873126Sahl 		if (op == OP_SHARE) {
37883126Sahl 			/*
37893126Sahl 			 * Finally, unshare any volumes shared via iSCSI.
37903126Sahl 			 */
37913126Sahl 			zfs_handle_t **dslist = NULL;
37923126Sahl 			size_t i, count = 0;
37933126Sahl 
37944737Smmusante 			get_all_datasets(ZFS_TYPE_VOLUME, &dslist, &count,
37954737Smmusante 			    B_FALSE);
37963126Sahl 
37973126Sahl 			if (count != 0) {
37983126Sahl 				qsort(dslist, count, sizeof (void *),
37993126Sahl 				    dataset_cmp);
38003126Sahl 
38013126Sahl 				for (i = 0; i < count; i++) {
38023126Sahl 					if (zfs_unshare_iscsi(dslist[i]) != 0)
38033126Sahl 						ret = 1;
38043126Sahl 					zfs_close(dslist[i]);
38053126Sahl 				}
38063126Sahl 
38073126Sahl 				free(dslist);
38083126Sahl 			}
38093126Sahl 		}
3810789Sahrens 	} else {
38113126Sahl 		if (argc != 1) {
38123126Sahl 			if (argc == 0)
38133126Sahl 				(void) fprintf(stderr,
38143126Sahl 				    gettext("missing filesystem argument\n"));
38153126Sahl 			else
38163126Sahl 				(void) fprintf(stderr,
38173126Sahl 				    gettext("too many arguments\n"));
38183126Sahl 			usage(B_FALSE);
38193126Sahl 		}
38203126Sahl 
3821789Sahrens 		/*
3822789Sahrens 		 * We have an argument, but it may be a full path or a ZFS
3823789Sahrens 		 * filesystem.  Pass full paths off to unmount_path() (shared by
3824789Sahrens 		 * manual_unmount), otherwise open the filesystem and pass to
3825789Sahrens 		 * zfs_unmount().
3826789Sahrens 		 */
3827789Sahrens 		if (argv[0][0] == '/')
38283126Sahl 			return (unshare_unmount_path(op, argv[0],
38293912Slling 			    flags, B_FALSE));
38302082Seschrock 
38313126Sahl 		types = ZFS_TYPE_FILESYSTEM;
38323126Sahl 		if (op == OP_SHARE)
38333126Sahl 			types |= ZFS_TYPE_VOLUME;
38343126Sahl 
38353126Sahl 		if ((zhp = zfs_open(g_zfs, argv[0], types)) == NULL)
3836789Sahrens 			return (1);
3837789Sahrens 
38383126Sahl 		if (zfs_get_type(zhp) == ZFS_TYPE_FILESYSTEM) {
38393126Sahl 			verify(zfs_prop_get(zhp, op == OP_SHARE ?
38405331Samw 			    ZFS_PROP_SHARENFS : ZFS_PROP_MOUNTPOINT,
38415331Samw 			    nfsiscsi_mnt_prop, sizeof (nfsiscsi_mnt_prop), NULL,
38425331Samw 			    NULL, 0, B_FALSE) == 0);
38433126Sahl 
38443126Sahl 			switch (op) {
38453126Sahl 			case OP_SHARE:
38465331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_SHARENFS,
38475331Samw 				    nfsiscsi_mnt_prop,
38485331Samw 				    sizeof (nfsiscsi_mnt_prop),
38495331Samw 				    NULL, NULL, 0, B_FALSE) == 0);
38505331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_SHARESMB,
38515331Samw 				    sharesmb, sizeof (sharesmb), NULL, NULL,
38525331Samw 				    0, B_FALSE) == 0);
38535331Samw 
38545331Samw 				if (strcmp(nfsiscsi_mnt_prop, "off") == 0 &&
38555331Samw 				    strcmp(sharesmb, "off") == 0) {
38563126Sahl 					(void) fprintf(stderr, gettext("cannot "
38573126Sahl 					    "unshare '%s': legacy share\n"),
38583126Sahl 					    zfs_get_name(zhp));
38593126Sahl 					(void) fprintf(stderr, gettext("use "
38603126Sahl 					    "unshare(1M) to unshare this "
38613126Sahl 					    "filesystem\n"));
38623126Sahl 					ret = 1;
38635331Samw 				} else if (!zfs_is_shared(zhp)) {
38643126Sahl 					(void) fprintf(stderr, gettext("cannot "
38653126Sahl 					    "unshare '%s': not currently "
38663126Sahl 					    "shared\n"), zfs_get_name(zhp));
38673126Sahl 					ret = 1;
38685331Samw 				} else if (zfs_unshareall(zhp) != 0) {
38693126Sahl 					ret = 1;
38703126Sahl 				}
38713126Sahl 				break;
38723126Sahl 
38733126Sahl 			case OP_MOUNT:
38745331Samw 				if (strcmp(nfsiscsi_mnt_prop, "legacy") == 0) {
38753126Sahl 					(void) fprintf(stderr, gettext("cannot "
38763126Sahl 					    "unmount '%s': legacy "
38773126Sahl 					    "mountpoint\n"), zfs_get_name(zhp));
38783126Sahl 					(void) fprintf(stderr, gettext("use "
38793126Sahl 					    "umount(1M) to unmount this "
38803126Sahl 					    "filesystem\n"));
38813126Sahl 					ret = 1;
38823126Sahl 				} else if (!zfs_is_mounted(zhp, NULL)) {
38833126Sahl 					(void) fprintf(stderr, gettext("cannot "
38843126Sahl 					    "unmount '%s': not currently "
38853126Sahl 					    "mounted\n"),
38863126Sahl 					    zfs_get_name(zhp));
38873126Sahl 					ret = 1;
38883126Sahl 				} else if (zfs_unmountall(zhp, flags) != 0) {
38893126Sahl 					ret = 1;
38903126Sahl 				}
38913126Sahl 				break;
38923126Sahl 			}
38933126Sahl 		} else {
38943126Sahl 			assert(op == OP_SHARE);
38953126Sahl 
38965331Samw 			verify(zfs_prop_get(zhp, ZFS_PROP_SHAREISCSI,
38975331Samw 			    nfsiscsi_mnt_prop, sizeof (nfsiscsi_mnt_prop),
38985331Samw 			    NULL, NULL, 0, B_FALSE) == 0);
38995331Samw 
39005331Samw 			if (strcmp(nfsiscsi_mnt_prop, "off") == 0) {
3901789Sahrens 				(void) fprintf(stderr, gettext("cannot unshare "
39023126Sahl 				    "'%s': 'shareiscsi' property not set\n"),
39033126Sahl 				    zfs_get_name(zhp));
39043126Sahl 				(void) fprintf(stderr, gettext("set "
39053126Sahl 				    "'shareiscsi' property or use "
39063126Sahl 				    "iscsitadm(1M) to share this volume\n"));
3907789Sahrens 				ret = 1;
39083126Sahl 			} else if (!zfs_is_shared_iscsi(zhp)) {
39093126Sahl 				(void) fprintf(stderr, gettext("cannot "
39103126Sahl 				    "unshare '%s': not currently shared\n"),
3911789Sahrens 				    zfs_get_name(zhp));
3912789Sahrens 				ret = 1;
39133126Sahl 			} else if (zfs_unshare_iscsi(zhp) != 0) {
3914789Sahrens 				ret = 1;
3915789Sahrens 			}
3916789Sahrens 		}
3917789Sahrens 
3918789Sahrens 		zfs_close(zhp);
3919789Sahrens 	}
3920789Sahrens 
3921789Sahrens 	return (ret);
3922789Sahrens }
3923789Sahrens 
3924789Sahrens /*
3925789Sahrens  * zfs unmount -a
3926789Sahrens  * zfs unmount filesystem
3927789Sahrens  *
3928789Sahrens  * Unmount all filesystems, or a specific ZFS filesystem.
3929789Sahrens  */
3930789Sahrens static int
3931789Sahrens zfs_do_unmount(int argc, char **argv)
3932789Sahrens {
3933789Sahrens 	return (unshare_unmount(OP_MOUNT, argc, argv));
3934789Sahrens }
3935789Sahrens 
3936789Sahrens /*
3937789Sahrens  * zfs unshare -a
3938789Sahrens  * zfs unshare filesystem
3939789Sahrens  *
3940789Sahrens  * Unshare all filesystems, or a specific ZFS filesystem.
3941789Sahrens  */
3942789Sahrens static int
3943789Sahrens zfs_do_unshare(int argc, char **argv)
3944789Sahrens {
3945789Sahrens 	return (unshare_unmount(OP_SHARE, argc, argv));
3946789Sahrens }
3947789Sahrens 
3948789Sahrens /*
3949789Sahrens  * Called when invoked as /etc/fs/zfs/mount.  Do the mount if the mountpoint is
3950789Sahrens  * 'legacy'.  Otherwise, complain that use should be using 'zfs mount'.
3951789Sahrens  */
3952789Sahrens static int
3953789Sahrens manual_mount(int argc, char **argv)
3954789Sahrens {
3955789Sahrens 	zfs_handle_t *zhp;
3956789Sahrens 	char mountpoint[ZFS_MAXPROPLEN];
3957789Sahrens 	char mntopts[MNT_LINE_MAX] = { '\0' };
3958789Sahrens 	int ret;
3959789Sahrens 	int c;
3960789Sahrens 	int flags = 0;
3961789Sahrens 	char *dataset, *path;
3962789Sahrens 
3963789Sahrens 	/* check options */
39641544Seschrock 	while ((c = getopt(argc, argv, ":mo:O")) != -1) {
3965789Sahrens 		switch (c) {
3966789Sahrens 		case 'o':
3967789Sahrens 			(void) strlcpy(mntopts, optarg, sizeof (mntopts));
3968789Sahrens 			break;
3969789Sahrens 		case 'O':
3970789Sahrens 			flags |= MS_OVERLAY;
3971789Sahrens 			break;
39721544Seschrock 		case 'm':
39731544Seschrock 			flags |= MS_NOMNTTAB;
39741544Seschrock 			break;
3975789Sahrens 		case ':':
3976789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
3977789Sahrens 			    "'%c' option\n"), optopt);
39782082Seschrock 			usage(B_FALSE);
3979789Sahrens 			break;
3980789Sahrens 		case '?':
3981789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
3982789Sahrens 			    optopt);
3983789Sahrens 			(void) fprintf(stderr, gettext("usage: mount [-o opts] "
3984789Sahrens 			    "<path>\n"));
3985789Sahrens 			return (2);
3986789Sahrens 		}
3987789Sahrens 	}
3988789Sahrens 
3989789Sahrens 	argc -= optind;
3990789Sahrens 	argv += optind;
3991789Sahrens 
3992789Sahrens 	/* check that we only have two arguments */
3993789Sahrens 	if (argc != 2) {
3994789Sahrens 		if (argc == 0)
3995789Sahrens 			(void) fprintf(stderr, gettext("missing dataset "
3996789Sahrens 			    "argument\n"));
3997789Sahrens 		else if (argc == 1)
3998789Sahrens 			(void) fprintf(stderr,
3999789Sahrens 			    gettext("missing mountpoint argument\n"));
4000789Sahrens 		else
4001789Sahrens 			(void) fprintf(stderr, gettext("too many arguments\n"));
4002789Sahrens 		(void) fprintf(stderr, "usage: mount <dataset> <mountpoint>\n");
4003789Sahrens 		return (2);
4004789Sahrens 	}
4005789Sahrens 
4006789Sahrens 	dataset = argv[0];
4007789Sahrens 	path = argv[1];
4008789Sahrens 
4009789Sahrens 	/* try to open the dataset */
40102082Seschrock 	if ((zhp = zfs_open(g_zfs, dataset, ZFS_TYPE_FILESYSTEM)) == NULL)
4011789Sahrens 		return (1);
4012789Sahrens 
4013789Sahrens 	(void) zfs_prop_get(zhp, ZFS_PROP_MOUNTPOINT, mountpoint,
40142082Seschrock 	    sizeof (mountpoint), NULL, NULL, 0, B_FALSE);
4015789Sahrens 
4016789Sahrens 	/* check for legacy mountpoint and complain appropriately */
4017789Sahrens 	ret = 0;
4018789Sahrens 	if (strcmp(mountpoint, ZFS_MOUNTPOINT_LEGACY) == 0) {
4019789Sahrens 		if (mount(dataset, path, MS_OPTIONSTR | flags, MNTTYPE_ZFS,
4020789Sahrens 		    NULL, 0, mntopts, sizeof (mntopts)) != 0) {
4021789Sahrens 			(void) fprintf(stderr, gettext("mount failed: %s\n"),
4022789Sahrens 			    strerror(errno));
4023789Sahrens 			ret = 1;
4024789Sahrens 		}
4025789Sahrens 	} else {
4026789Sahrens 		(void) fprintf(stderr, gettext("filesystem '%s' cannot be "
4027789Sahrens 		    "mounted using 'mount -F zfs'\n"), dataset);
4028789Sahrens 		(void) fprintf(stderr, gettext("Use 'zfs set mountpoint=%s' "
4029789Sahrens 		    "instead.\n"), path);
4030789Sahrens 		(void) fprintf(stderr, gettext("If you must use 'mount -F zfs' "
4031789Sahrens 		    "or /etc/vfstab, use 'zfs set mountpoint=legacy'.\n"));
4032789Sahrens 		(void) fprintf(stderr, gettext("See zfs(1M) for more "
4033789Sahrens 		    "information.\n"));
4034789Sahrens 		ret = 1;
4035789Sahrens 	}
4036789Sahrens 
4037789Sahrens 	return (ret);
4038789Sahrens }
4039789Sahrens 
4040789Sahrens /*
4041789Sahrens  * Called when invoked as /etc/fs/zfs/umount.  Unlike a manual mount, we allow
4042789Sahrens  * unmounts of non-legacy filesystems, as this is the dominant administrative
4043789Sahrens  * interface.
4044789Sahrens  */
4045789Sahrens static int
4046789Sahrens manual_unmount(int argc, char **argv)
4047789Sahrens {
4048789Sahrens 	int flags = 0;
4049789Sahrens 	int c;
4050789Sahrens 
4051789Sahrens 	/* check options */
4052789Sahrens 	while ((c = getopt(argc, argv, "f")) != -1) {
4053789Sahrens 		switch (c) {
4054789Sahrens 		case 'f':
4055789Sahrens 			flags = MS_FORCE;
4056789Sahrens 			break;
4057789Sahrens 		case '?':
4058789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
4059789Sahrens 			    optopt);
4060789Sahrens 			(void) fprintf(stderr, gettext("usage: unmount [-f] "
4061789Sahrens 			    "<path>\n"));
4062789Sahrens 			return (2);
4063789Sahrens 		}
4064789Sahrens 	}
4065789Sahrens 
4066789Sahrens 	argc -= optind;
4067789Sahrens 	argv += optind;
4068789Sahrens 
4069789Sahrens 	/* check arguments */
4070789Sahrens 	if (argc != 1) {
4071789Sahrens 		if (argc == 0)
4072789Sahrens 			(void) fprintf(stderr, gettext("missing path "
4073789Sahrens 			    "argument\n"));
4074789Sahrens 		else
4075789Sahrens 			(void) fprintf(stderr, gettext("too many arguments\n"));
4076789Sahrens 		(void) fprintf(stderr, gettext("usage: unmount [-f] <path>\n"));
4077789Sahrens 		return (2);
4078789Sahrens 	}
4079789Sahrens 
40802082Seschrock 	return (unshare_unmount_path(OP_MOUNT, argv[0], flags, B_TRUE));
4081789Sahrens }
4082789Sahrens 
4083789Sahrens static int
4084789Sahrens volcheck(zpool_handle_t *zhp, void *data)
4085789Sahrens {
40862856Snd150628 	boolean_t isinit = *((boolean_t *)data);
4087789Sahrens 
4088789Sahrens 	if (isinit)
4089789Sahrens 		return (zpool_create_zvol_links(zhp));
4090789Sahrens 	else
4091789Sahrens 		return (zpool_remove_zvol_links(zhp));
4092789Sahrens }
4093789Sahrens 
4094789Sahrens /*
4095789Sahrens  * Iterate over all pools in the system and either create or destroy /dev/zvol
4096789Sahrens  * links, depending on the value of 'isinit'.
4097789Sahrens  */
4098789Sahrens static int
40992082Seschrock do_volcheck(boolean_t isinit)
4100789Sahrens {
41012856Snd150628 	return (zpool_iter(g_zfs, volcheck, &isinit) ? 1 : 0);
4102789Sahrens }
4103789Sahrens 
41044543Smarks static int
41054543Smarks find_command_idx(char *command, int *idx)
41064543Smarks {
41074543Smarks 	int i;
41084543Smarks 
41094543Smarks 	for (i = 0; i < NCOMMAND; i++) {
41104543Smarks 		if (command_table[i].name == NULL)
41114543Smarks 			continue;
41124543Smarks 
41134543Smarks 		if (strcmp(command, command_table[i].name) == 0) {
41144543Smarks 			*idx = i;
41154543Smarks 			return (0);
41164543Smarks 		}
41174543Smarks 	}
41184543Smarks 	return (1);
41194543Smarks }
41204543Smarks 
4121789Sahrens int
4122789Sahrens main(int argc, char **argv)
4123789Sahrens {
4124789Sahrens 	int ret;
4125789Sahrens 	int i;
4126789Sahrens 	char *progname;
4127789Sahrens 	char *cmdname;
4128789Sahrens 
4129789Sahrens 	(void) setlocale(LC_ALL, "");
4130789Sahrens 	(void) textdomain(TEXT_DOMAIN);
4131789Sahrens 
4132789Sahrens 	opterr = 0;
4133789Sahrens 
41342082Seschrock 	if ((g_zfs = libzfs_init()) == NULL) {
41352082Seschrock 		(void) fprintf(stderr, gettext("internal error: failed to "
41362082Seschrock 		    "initialize ZFS library\n"));
41372082Seschrock 		return (1);
41382082Seschrock 	}
41392082Seschrock 
41404988Sek110237 	zpool_set_history_str("zfs", argc, argv, history_str);
41414988Sek110237 	verify(zpool_stage_history(g_zfs, history_str) == 0);
41424543Smarks 
41432082Seschrock 	libzfs_print_on_error(g_zfs, B_TRUE);
41442082Seschrock 
4145789Sahrens 	if ((mnttab_file = fopen(MNTTAB, "r")) == NULL) {
4146789Sahrens 		(void) fprintf(stderr, gettext("internal error: unable to "
4147789Sahrens 		    "open %s\n"), MNTTAB);
4148789Sahrens 		return (1);
4149789Sahrens 	}
4150789Sahrens 
4151789Sahrens 	/*
4152789Sahrens 	 * This command also doubles as the /etc/fs mount and unmount program.
4153789Sahrens 	 * Determine if we should take this behavior based on argv[0].
4154789Sahrens 	 */
4155789Sahrens 	progname = basename(argv[0]);
4156789Sahrens 	if (strcmp(progname, "mount") == 0) {
4157789Sahrens 		ret = manual_mount(argc, argv);
4158789Sahrens 	} else if (strcmp(progname, "umount") == 0) {
4159789Sahrens 		ret = manual_unmount(argc, argv);
4160789Sahrens 	} else {
4161789Sahrens 		/*
4162789Sahrens 		 * Make sure the user has specified some command.
4163789Sahrens 		 */
4164789Sahrens 		if (argc < 2) {
4165789Sahrens 			(void) fprintf(stderr, gettext("missing command\n"));
41662082Seschrock 			usage(B_FALSE);
4167789Sahrens 		}
4168789Sahrens 
4169789Sahrens 		cmdname = argv[1];
4170789Sahrens 
4171789Sahrens 		/*
4172789Sahrens 		 * The 'umount' command is an alias for 'unmount'
4173789Sahrens 		 */
4174789Sahrens 		if (strcmp(cmdname, "umount") == 0)
4175789Sahrens 			cmdname = "unmount";
4176789Sahrens 
4177789Sahrens 		/*
41781749Sahrens 		 * The 'recv' command is an alias for 'receive'
41791749Sahrens 		 */
41801749Sahrens 		if (strcmp(cmdname, "recv") == 0)
41811749Sahrens 			cmdname = "receive";
41821749Sahrens 
41831749Sahrens 		/*
4184789Sahrens 		 * Special case '-?'
4185789Sahrens 		 */
4186789Sahrens 		if (strcmp(cmdname, "-?") == 0)
41872082Seschrock 			usage(B_TRUE);
4188789Sahrens 
4189789Sahrens 		/*
4190789Sahrens 		 * 'volinit' and 'volfini' do not appear in the usage message,
4191789Sahrens 		 * so we have to special case them here.
4192789Sahrens 		 */
4193789Sahrens 		if (strcmp(cmdname, "volinit") == 0)
41942082Seschrock 			return (do_volcheck(B_TRUE));
4195789Sahrens 		else if (strcmp(cmdname, "volfini") == 0)
41962082Seschrock 			return (do_volcheck(B_FALSE));
4197789Sahrens 
4198789Sahrens 		/*
4199789Sahrens 		 * Run the appropriate command.
4200789Sahrens 		 */
42014543Smarks 		if (find_command_idx(cmdname, &i) == 0) {
42024543Smarks 			current_command = &command_table[i];
42034543Smarks 			ret = command_table[i].func(argc - 1, argv + 1);
42044787Sahrens 		} else if (strchr(cmdname, '=') != NULL) {
42054787Sahrens 			verify(find_command_idx("set", &i) == 0);
42064787Sahrens 			current_command = &command_table[i];
42074787Sahrens 			ret = command_table[i].func(argc, argv);
42084787Sahrens 		} else {
4209789Sahrens 			(void) fprintf(stderr, gettext("unrecognized "
4210789Sahrens 			    "command '%s'\n"), cmdname);
42112082Seschrock 			usage(B_FALSE);
4212789Sahrens 		}
4213789Sahrens 	}
4214789Sahrens 
4215789Sahrens 	(void) fclose(mnttab_file);
4216789Sahrens 
42172082Seschrock 	libzfs_fini(g_zfs);
42182082Seschrock 
4219789Sahrens 	/*
4220789Sahrens 	 * The 'ZFS_ABORT' environment variable causes us to dump core on exit
4221789Sahrens 	 * for the purposes of running ::findleaks.
4222789Sahrens 	 */
4223789Sahrens 	if (getenv("ZFS_ABORT") != NULL) {
4224789Sahrens 		(void) printf("dumping core by request\n");
4225789Sahrens 		abort();
4226789Sahrens 	}
4227789Sahrens 
4228789Sahrens 	return (ret);
4229789Sahrens }
4230