xref: /onnv-gate/usr/src/cmd/zfs/zfs_main.c (revision 6865:176d18c623cc)
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 #pragma ident	"%Z%%M%	%I%	%E% SMI"
28789Sahrens 
29789Sahrens #include <assert.h>
302676Seschrock #include <ctype.h>
31789Sahrens #include <errno.h>
32789Sahrens #include <libgen.h>
33789Sahrens #include <libintl.h>
34789Sahrens #include <libuutil.h>
354543Smarks #include <libnvpair.h>
36789Sahrens #include <locale.h>
37789Sahrens #include <stddef.h>
38789Sahrens #include <stdio.h>
39789Sahrens #include <stdlib.h>
40789Sahrens #include <strings.h>
41789Sahrens #include <unistd.h>
42789Sahrens #include <fcntl.h>
43789Sahrens #include <zone.h>
44789Sahrens #include <sys/mkdev.h>
45789Sahrens #include <sys/mntent.h>
46789Sahrens #include <sys/mnttab.h>
47789Sahrens #include <sys/mount.h>
48789Sahrens #include <sys/stat.h>
494543Smarks #include <sys/avl.h>
50789Sahrens 
51789Sahrens #include <libzfs.h>
524543Smarks #include <libuutil.h>
53789Sahrens 
54789Sahrens #include "zfs_iter.h"
552082Seschrock #include "zfs_util.h"
562082Seschrock 
572082Seschrock libzfs_handle_t *g_zfs;
58789Sahrens 
59789Sahrens static FILE *mnttab_file;
604988Sek110237 static char history_str[HIS_MAX_RECORD_LEN];
61789Sahrens 
62789Sahrens static int zfs_do_clone(int argc, char **argv);
63789Sahrens static int zfs_do_create(int argc, char **argv);
64789Sahrens static int zfs_do_destroy(int argc, char **argv);
65789Sahrens static int zfs_do_get(int argc, char **argv);
66789Sahrens static int zfs_do_inherit(int argc, char **argv);
67789Sahrens static int zfs_do_list(int argc, char **argv);
68789Sahrens static int zfs_do_mount(int argc, char **argv);
69789Sahrens static int zfs_do_rename(int argc, char **argv);
70789Sahrens static int zfs_do_rollback(int argc, char **argv);
71789Sahrens static int zfs_do_set(int argc, char **argv);
724577Sahrens static int zfs_do_upgrade(int argc, char **argv);
73789Sahrens static int zfs_do_snapshot(int argc, char **argv);
74789Sahrens static int zfs_do_unmount(int argc, char **argv);
75789Sahrens static int zfs_do_share(int argc, char **argv);
76789Sahrens static int zfs_do_unshare(int argc, char **argv);
771749Sahrens static int zfs_do_send(int argc, char **argv);
781749Sahrens static int zfs_do_receive(int argc, char **argv);
792082Seschrock static int zfs_do_promote(int argc, char **argv);
804543Smarks static int zfs_do_allow(int argc, char **argv);
814543Smarks static int zfs_do_unallow(int argc, char **argv);
82789Sahrens 
83789Sahrens /*
84*6865Srm160521  * Enable a reasonable set of defaults for libumem debugging on DEBUG builds.
85789Sahrens  */
86*6865Srm160521 
87*6865Srm160521 #ifdef DEBUG
88789Sahrens const char *
893126Sahl _umem_debug_init(void)
90789Sahrens {
91789Sahrens 	return ("default,verbose"); /* $UMEM_DEBUG setting */
92789Sahrens }
93789Sahrens 
94789Sahrens const char *
95789Sahrens _umem_logging_init(void)
96789Sahrens {
97789Sahrens 	return ("fail,contents"); /* $UMEM_LOGGING setting */
98789Sahrens }
99*6865Srm160521 #endif
100789Sahrens 
1011387Seschrock typedef enum {
1021387Seschrock 	HELP_CLONE,
1031387Seschrock 	HELP_CREATE,
1041387Seschrock 	HELP_DESTROY,
1051387Seschrock 	HELP_GET,
1061387Seschrock 	HELP_INHERIT,
1074577Sahrens 	HELP_UPGRADE,
1081387Seschrock 	HELP_LIST,
1091387Seschrock 	HELP_MOUNT,
1102082Seschrock 	HELP_PROMOTE,
1111749Sahrens 	HELP_RECEIVE,
1121387Seschrock 	HELP_RENAME,
1131387Seschrock 	HELP_ROLLBACK,
1141749Sahrens 	HELP_SEND,
1151387Seschrock 	HELP_SET,
1161387Seschrock 	HELP_SHARE,
1171387Seschrock 	HELP_SNAPSHOT,
1181387Seschrock 	HELP_UNMOUNT,
1194543Smarks 	HELP_UNSHARE,
1204543Smarks 	HELP_ALLOW,
1214543Smarks 	HELP_UNALLOW
1221387Seschrock } zfs_help_t;
1231387Seschrock 
124789Sahrens typedef struct zfs_command {
125789Sahrens 	const char	*name;
126789Sahrens 	int		(*func)(int argc, char **argv);
1271387Seschrock 	zfs_help_t	usage;
128789Sahrens } zfs_command_t;
129789Sahrens 
130789Sahrens /*
131789Sahrens  * Master command table.  Each ZFS command has a name, associated function, and
1321544Seschrock  * usage message.  The usage messages need to be internationalized, so we have
1331544Seschrock  * to have a function to return the usage message based on a command index.
1341387Seschrock  *
1351387Seschrock  * These commands are organized according to how they are displayed in the usage
1361387Seschrock  * message.  An empty command (one with a NULL name) indicates an empty line in
1371387Seschrock  * the generic usage message.
138789Sahrens  */
139789Sahrens static zfs_command_t command_table[] = {
1401387Seschrock 	{ "create",	zfs_do_create,		HELP_CREATE		},
1411387Seschrock 	{ "destroy",	zfs_do_destroy,		HELP_DESTROY		},
142789Sahrens 	{ NULL },
1431387Seschrock 	{ "snapshot",	zfs_do_snapshot,	HELP_SNAPSHOT		},
1441387Seschrock 	{ "rollback",	zfs_do_rollback,	HELP_ROLLBACK		},
1451387Seschrock 	{ "clone",	zfs_do_clone,		HELP_CLONE		},
1462082Seschrock 	{ "promote",	zfs_do_promote,		HELP_PROMOTE		},
1471387Seschrock 	{ "rename",	zfs_do_rename,		HELP_RENAME		},
148789Sahrens 	{ NULL },
1491387Seschrock 	{ "list",	zfs_do_list,		HELP_LIST		},
150789Sahrens 	{ NULL },
1511387Seschrock 	{ "set",	zfs_do_set,		HELP_SET		},
1521387Seschrock 	{ "get", 	zfs_do_get,		HELP_GET		},
1531387Seschrock 	{ "inherit",	zfs_do_inherit,		HELP_INHERIT		},
1544577Sahrens 	{ "upgrade",	zfs_do_upgrade,		HELP_UPGRADE		},
155789Sahrens 	{ NULL },
1561387Seschrock 	{ "mount",	zfs_do_mount,		HELP_MOUNT		},
1571387Seschrock 	{ "unmount",	zfs_do_unmount,		HELP_UNMOUNT		},
1581387Seschrock 	{ "share",	zfs_do_share,		HELP_SHARE		},
1591387Seschrock 	{ "unshare",	zfs_do_unshare,		HELP_UNSHARE		},
160789Sahrens 	{ NULL },
1611749Sahrens 	{ "send",	zfs_do_send,		HELP_SEND		},
1621749Sahrens 	{ "receive",	zfs_do_receive,		HELP_RECEIVE		},
1634543Smarks 	{ NULL },
1644543Smarks 	{ "allow",	zfs_do_allow,		HELP_ALLOW		},
1654543Smarks 	{ NULL },
1664543Smarks 	{ "unallow",	zfs_do_unallow,		HELP_UNALLOW		},
167789Sahrens };
168789Sahrens 
169789Sahrens #define	NCOMMAND	(sizeof (command_table) / sizeof (command_table[0]))
170789Sahrens 
171789Sahrens zfs_command_t *current_command;
172789Sahrens 
1731387Seschrock static const char *
1741387Seschrock get_usage(zfs_help_t idx)
1751387Seschrock {
1761387Seschrock 	switch (idx) {
1771387Seschrock 	case HELP_CLONE:
1784490Svb160487 		return (gettext("\tclone [-p] <snapshot> "
1794490Svb160487 		    "<filesystem|volume>\n"));
1801387Seschrock 	case HELP_CREATE:
1814849Sahrens 		return (gettext("\tcreate [-p] [-o property=value] ... "
1822676Seschrock 		    "<filesystem>\n"
1834849Sahrens 		    "\tcreate [-ps] [-b blocksize] [-o property=value] ... "
1844849Sahrens 		    "-V <size> <volume>\n"));
1851387Seschrock 	case HELP_DESTROY:
1861387Seschrock 		return (gettext("\tdestroy [-rRf] "
1871387Seschrock 		    "<filesystem|volume|snapshot>\n"));
1881387Seschrock 	case HELP_GET:
1894849Sahrens 		return (gettext("\tget [-rHp] [-o field[,...]] "
1904849Sahrens 		    "[-s source[,...]]\n"
1914849Sahrens 		    "\t    <\"all\" | property[,...]> "
1922676Seschrock 		    "[filesystem|volume|snapshot] ...\n"));
1931387Seschrock 	case HELP_INHERIT:
1941387Seschrock 		return (gettext("\tinherit [-r] <property> "
1951387Seschrock 		    "<filesystem|volume> ...\n"));
1964577Sahrens 	case HELP_UPGRADE:
1974577Sahrens 		return (gettext("\tupgrade [-v]\n"
1984577Sahrens 		    "\tupgrade [-r] [-V version] <-a | filesystem ...>\n"));
1991387Seschrock 	case HELP_LIST:
2004849Sahrens 		return (gettext("\tlist [-rH] [-o property[,...]] "
2014849Sahrens 		    "[-t type[,...]] [-s property] ...\n"
2024849Sahrens 		    "\t    [-S property] ... "
2034849Sahrens 		    "[filesystem|volume|snapshot] ...\n"));
2041387Seschrock 	case HELP_MOUNT:
2051387Seschrock 		return (gettext("\tmount\n"
2064849Sahrens 		    "\tmount [-vO] [-o opts] <-a | filesystem>\n"));
2072082Seschrock 	case HELP_PROMOTE:
2084849Sahrens 		return (gettext("\tpromote <clone-filesystem>\n"));
2091749Sahrens 	case HELP_RECEIVE:
2102665Snd150628 		return (gettext("\treceive [-vnF] <filesystem|volume|"
2112665Snd150628 		"snapshot>\n"
2122665Snd150628 		"\treceive [-vnF] -d <filesystem>\n"));
2131387Seschrock 	case HELP_RENAME:
2141387Seschrock 		return (gettext("\trename <filesystem|volume|snapshot> "
2154007Smmusante 		    "<filesystem|volume|snapshot>\n"
2164490Svb160487 		    "\trename -p <filesystem|volume> <filesystem|volume>\n"
2174007Smmusante 		    "\trename -r <snapshot> <snapshot>"));
2181387Seschrock 	case HELP_ROLLBACK:
2195568Sahrens 		return (gettext("\trollback [-rRf] <snapshot>\n"));
2201749Sahrens 	case HELP_SEND:
2215367Sahrens 		return (gettext("\tsend [-R] [-[iI] snapshot] <snapshot>\n"));
2221387Seschrock 	case HELP_SET:
2231387Seschrock 		return (gettext("\tset <property=value> "
2241387Seschrock 		    "<filesystem|volume> ...\n"));
2251387Seschrock 	case HELP_SHARE:
2264849Sahrens 		return (gettext("\tshare <-a | filesystem>\n"));
2271387Seschrock 	case HELP_SNAPSHOT:
2282199Sahrens 		return (gettext("\tsnapshot [-r] "
2294849Sahrens 		    "<filesystem@snapname|volume@snapname>\n"));
2301387Seschrock 	case HELP_UNMOUNT:
2314849Sahrens 		return (gettext("\tunmount [-f] "
2324849Sahrens 		    "<-a | filesystem|mountpoint>\n"));
2331387Seschrock 	case HELP_UNSHARE:
2344849Sahrens 		return (gettext("\tunshare [-f] "
2354849Sahrens 		    "<-a | filesystem|mountpoint>\n"));
2364543Smarks 	case HELP_ALLOW:
2374849Sahrens 		return (gettext("\tallow [-ldug] "
2384849Sahrens 		    "<\"everyone\"|user|group>[,...] <perm|@setname>[,...]\n"
2394849Sahrens 		    "\t    <filesystem|volume>\n"
2404849Sahrens 		    "\tallow [-ld] -e <perm|@setname>[,...] "
2414849Sahrens 		    "<filesystem|volume>\n"
2424849Sahrens 		    "\tallow -c <perm|@setname>[,...] <filesystem|volume>\n"
2434849Sahrens 		    "\tallow -s @setname <perm|@setname>[,...] "
2444849Sahrens 		    "<filesystem|volume>\n"));
2454543Smarks 	case HELP_UNALLOW:
2464849Sahrens 		return (gettext("\tunallow [-rldug] "
2474849Sahrens 		    "<\"everyone\"|user|group>[,...]\n"
2484849Sahrens 		    "\t    [<perm|@setname>[,...]] <filesystem|volume>\n"
2494849Sahrens 		    "\tunallow [-rld] -e [<perm|@setname>[,...]] "
2504849Sahrens 		    "<filesystem|volume>\n"
2514849Sahrens 		    "\tunallow [-r] -c [<perm|@setname>[,...]] "
2524849Sahrens 		    "<filesystem|volume>\n"
2534849Sahrens 		    "\tunallow [-r] -s @setname [<perm|@setname>[,...]] "
2544849Sahrens 		    "<filesystem|volume>\n"));
2551387Seschrock 	}
2561387Seschrock 
2571387Seschrock 	abort();
2581387Seschrock 	/* NOTREACHED */
2591387Seschrock }
2601387Seschrock 
261789Sahrens /*
262789Sahrens  * Utility function to guarantee malloc() success.
263789Sahrens  */
264789Sahrens void *
265789Sahrens safe_malloc(size_t size)
266789Sahrens {
267789Sahrens 	void *data;
268789Sahrens 
269789Sahrens 	if ((data = calloc(1, size)) == NULL) {
270789Sahrens 		(void) fprintf(stderr, "internal error: out of memory\n");
271789Sahrens 		exit(1);
272789Sahrens 	}
273789Sahrens 
274789Sahrens 	return (data);
275789Sahrens }
276789Sahrens 
277789Sahrens /*
2785094Slling  * Callback routine that will print out information for each of
2793654Sgw25295  * the properties.
2803654Sgw25295  */
2815094Slling static int
2825094Slling usage_prop_cb(int prop, void *cb)
2833654Sgw25295 {
2843654Sgw25295 	FILE *fp = cb;
2853654Sgw25295 
2866643Seschrock 	(void) fprintf(fp, "\t%-15s ", zfs_prop_to_name(prop));
2876643Seschrock 
2886643Seschrock 	if (zfs_prop_readonly(prop))
2896643Seschrock 		(void) fprintf(fp, " NO    ");
2903654Sgw25295 	else
2916643Seschrock 		(void) fprintf(fp, "YES    ");
2923654Sgw25295 
2933654Sgw25295 	if (zfs_prop_inheritable(prop))
2943654Sgw25295 		(void) fprintf(fp, "  YES   ");
2953654Sgw25295 	else
2963654Sgw25295 		(void) fprintf(fp, "   NO   ");
2973654Sgw25295 
2983654Sgw25295 	if (zfs_prop_values(prop) == NULL)
2993654Sgw25295 		(void) fprintf(fp, "-\n");
3003654Sgw25295 	else
3013654Sgw25295 		(void) fprintf(fp, "%s\n", zfs_prop_values(prop));
3023654Sgw25295 
3035094Slling 	return (ZPROP_CONT);
3043654Sgw25295 }
3053654Sgw25295 
3063654Sgw25295 /*
307789Sahrens  * Display usage message.  If we're inside a command, display only the usage for
308789Sahrens  * that command.  Otherwise, iterate over the entire command table and display
309789Sahrens  * a complete usage message.
310789Sahrens  */
311789Sahrens static void
3122082Seschrock usage(boolean_t requested)
313789Sahrens {
314789Sahrens 	int i;
3152082Seschrock 	boolean_t show_properties = B_FALSE;
3165993Smarks 	boolean_t show_permissions = B_FALSE;
317789Sahrens 	FILE *fp = requested ? stdout : stderr;
318789Sahrens 
319789Sahrens 	if (current_command == NULL) {
320789Sahrens 
321789Sahrens 		(void) fprintf(fp, gettext("usage: zfs command args ...\n"));
322789Sahrens 		(void) fprintf(fp,
323789Sahrens 		    gettext("where 'command' is one of the following:\n\n"));
324789Sahrens 
325789Sahrens 		for (i = 0; i < NCOMMAND; i++) {
326789Sahrens 			if (command_table[i].name == NULL)
327789Sahrens 				(void) fprintf(fp, "\n");
328789Sahrens 			else
329789Sahrens 				(void) fprintf(fp, "%s",
3301387Seschrock 				    get_usage(command_table[i].usage));
331789Sahrens 		}
332789Sahrens 
333789Sahrens 		(void) fprintf(fp, gettext("\nEach dataset is of the form: "
334789Sahrens 		    "pool/[dataset/]*dataset[@name]\n"));
335789Sahrens 	} else {
336789Sahrens 		(void) fprintf(fp, gettext("usage:\n"));
3371387Seschrock 		(void) fprintf(fp, "%s", get_usage(current_command->usage));
338789Sahrens 	}
339789Sahrens 
3402190Sdarrenm 	if (current_command != NULL &&
3412190Sdarrenm 	    (strcmp(current_command->name, "set") == 0 ||
342789Sahrens 	    strcmp(current_command->name, "get") == 0 ||
343789Sahrens 	    strcmp(current_command->name, "inherit") == 0 ||
3442190Sdarrenm 	    strcmp(current_command->name, "list") == 0))
3452082Seschrock 		show_properties = B_TRUE;
346789Sahrens 
3475993Smarks 	if (current_command != NULL &&
3485993Smarks 	    (strcmp(current_command->name, "allow") == 0 ||
3495993Smarks 	    strcmp(current_command->name, "unallow") == 0))
3505993Smarks 		show_permissions = B_TRUE;
3515993Smarks 
352789Sahrens 	if (show_properties) {
353789Sahrens 
354789Sahrens 		(void) fprintf(fp,
355789Sahrens 		    gettext("\nThe following properties are supported:\n"));
356789Sahrens 
3575378Sck153898 		(void) fprintf(fp, "\n\t%-14s %s  %s   %s\n\n",
358789Sahrens 		    "PROPERTY", "EDIT", "INHERIT", "VALUES");
359789Sahrens 
3603654Sgw25295 		/* Iterate over all properties */
3615094Slling 		(void) zprop_iter(usage_prop_cb, fp, B_FALSE, B_TRUE,
3625094Slling 		    ZFS_TYPE_DATASET);
3633654Sgw25295 
364789Sahrens 		(void) fprintf(fp, gettext("\nSizes are specified in bytes "
365789Sahrens 		    "with standard units such as K, M, G, etc.\n"));
3662676Seschrock 		(void) fprintf(fp, gettext("\n\nUser-defined properties can "
3672676Seschrock 		    "be specified by using a name containing a colon (:).\n"));
3685993Smarks 
3695993Smarks 	} else if (show_permissions) {
3705993Smarks 		(void) fprintf(fp,
3715993Smarks 		    gettext("\nThe following permissions are supported:\n"));
3725993Smarks 
3735993Smarks 		zfs_deleg_permissions();
3742190Sdarrenm 	} else {
3752190Sdarrenm 		/*
3762190Sdarrenm 		 * TRANSLATION NOTE:
3772190Sdarrenm 		 * "zfs set|get" must not be localised this is the
3782190Sdarrenm 		 * command name and arguments.
3792190Sdarrenm 		 */
3805993Smarks 
3812190Sdarrenm 		(void) fprintf(fp,
3822190Sdarrenm 		    gettext("\nFor the property list, run: zfs set|get\n"));
3835993Smarks 
3845993Smarks 		(void) fprintf(fp,
3855993Smarks 		    gettext("\nFor the delegated permission list, run:"
3865993Smarks 		    " zfs allow|unallow\n"));
387789Sahrens 	}
388789Sahrens 
3892676Seschrock 	/*
3902676Seschrock 	 * See comments at end of main().
3912676Seschrock 	 */
3922676Seschrock 	if (getenv("ZFS_ABORT") != NULL) {
3932676Seschrock 		(void) printf("dumping core by request\n");
3942676Seschrock 		abort();
3952676Seschrock 	}
3962676Seschrock 
397789Sahrens 	exit(requested ? 0 : 2);
398789Sahrens }
399789Sahrens 
400789Sahrens /*
4014490Svb160487  * zfs clone [-p] <snap> <fs | vol>
402789Sahrens  *
403789Sahrens  * Given an existing dataset, create a writable copy whose initial contents
404789Sahrens  * are the same as the source.  The newly created dataset maintains a
405789Sahrens  * dependency on the original; the original cannot be destroyed so long as
406789Sahrens  * the clone exists.
4074490Svb160487  *
4084490Svb160487  * The '-p' flag creates all the non-existing ancestors of the target first.
409789Sahrens  */
410789Sahrens static int
411789Sahrens zfs_do_clone(int argc, char **argv)
412789Sahrens {
413789Sahrens 	zfs_handle_t *zhp;
4144490Svb160487 	boolean_t parents = B_FALSE;
415789Sahrens 	int ret;
4164490Svb160487 	int c;
417789Sahrens 
418789Sahrens 	/* check options */
4194490Svb160487 	while ((c = getopt(argc, argv, "p")) != -1) {
4204490Svb160487 		switch (c) {
4214490Svb160487 		case 'p':
4224490Svb160487 			parents = B_TRUE;
4234490Svb160487 			break;
4244490Svb160487 		case '?':
4254490Svb160487 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
4264490Svb160487 			    optopt);
4274490Svb160487 			usage(B_FALSE);
4284490Svb160487 		}
429789Sahrens 	}
430789Sahrens 
4314490Svb160487 	argc -= optind;
4324490Svb160487 	argv += optind;
4334490Svb160487 
434789Sahrens 	/* check number of arguments */
4354490Svb160487 	if (argc < 1) {
436789Sahrens 		(void) fprintf(stderr, gettext("missing source dataset "
437789Sahrens 		    "argument\n"));
4382082Seschrock 		usage(B_FALSE);
439789Sahrens 	}
4404490Svb160487 	if (argc < 2) {
441789Sahrens 		(void) fprintf(stderr, gettext("missing target dataset "
442789Sahrens 		    "argument\n"));
4432082Seschrock 		usage(B_FALSE);
444789Sahrens 	}
4454490Svb160487 	if (argc > 2) {
446789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
4472082Seschrock 		usage(B_FALSE);
448789Sahrens 	}
449789Sahrens 
450789Sahrens 	/* open the source dataset */
4514490Svb160487 	if ((zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_SNAPSHOT)) == NULL)
452789Sahrens 		return (1);
453789Sahrens 
4544490Svb160487 	if (parents && zfs_name_valid(argv[1], ZFS_TYPE_FILESYSTEM |
4554490Svb160487 	    ZFS_TYPE_VOLUME)) {
4564490Svb160487 		/*
4574490Svb160487 		 * Now create the ancestors of the target dataset.  If the
4584490Svb160487 		 * target already exists and '-p' option was used we should not
4594490Svb160487 		 * complain.
4604490Svb160487 		 */
4614490Svb160487 		if (zfs_dataset_exists(g_zfs, argv[1], ZFS_TYPE_FILESYSTEM |
4624490Svb160487 		    ZFS_TYPE_VOLUME))
4634490Svb160487 			return (0);
4644490Svb160487 		if (zfs_create_ancestors(g_zfs, argv[1]) != 0)
4654490Svb160487 			return (1);
4664490Svb160487 	}
4674490Svb160487 
468789Sahrens 	/* pass to libzfs */
4694490Svb160487 	ret = zfs_clone(zhp, argv[1], NULL);
470789Sahrens 
471789Sahrens 	/* create the mountpoint if necessary */
472789Sahrens 	if (ret == 0) {
4735094Slling 		zfs_handle_t *clone;
4745094Slling 
4755094Slling 		clone = zfs_open(g_zfs, argv[1], ZFS_TYPE_DATASET);
476789Sahrens 		if (clone != NULL) {
477789Sahrens 			if ((ret = zfs_mount(clone, NULL, 0)) == 0)
478789Sahrens 				ret = zfs_share(clone);
479789Sahrens 			zfs_close(clone);
480789Sahrens 		}
481789Sahrens 	}
482789Sahrens 
483789Sahrens 	zfs_close(zhp);
484789Sahrens 
485789Sahrens 	return (ret == 0 ? 0 : 1);
486789Sahrens }
487789Sahrens 
488789Sahrens /*
4894490Svb160487  * zfs create [-p] [-o prop=value] ... fs
4904490Svb160487  * zfs create [-ps] [-b blocksize] [-o prop=value] ... -V vol size
491789Sahrens  *
492789Sahrens  * Create a new dataset.  This command can be used to create filesystems
493789Sahrens  * and volumes.  Snapshot creation is handled by 'zfs snapshot'.
494789Sahrens  * For volumes, the user must specify a size to be used.
495789Sahrens  *
496789Sahrens  * The '-s' flag applies only to volumes, and indicates that we should not try
497789Sahrens  * to set the reservation for this volume.  By default we set a reservation
4985481Sck153898  * equal to the size for any volume.  For pools with SPA_VERSION >=
4995481Sck153898  * SPA_VERSION_REFRESERVATION, we set a refreservation instead.
5004490Svb160487  *
5014490Svb160487  * The '-p' flag creates all the non-existing ancestors of the target first.
502789Sahrens  */
503789Sahrens static int
504789Sahrens zfs_do_create(int argc, char **argv)
505789Sahrens {
506789Sahrens 	zfs_type_t type = ZFS_TYPE_FILESYSTEM;
5072676Seschrock 	zfs_handle_t *zhp = NULL;
5082676Seschrock 	uint64_t volsize;
509789Sahrens 	int c;
5102082Seschrock 	boolean_t noreserve = B_FALSE;
5115367Sahrens 	boolean_t bflag = B_FALSE;
5124490Svb160487 	boolean_t parents = B_FALSE;
5132676Seschrock 	int ret = 1;
5142676Seschrock 	nvlist_t *props = NULL;
5152676Seschrock 	uint64_t intval;
5162676Seschrock 	char *propname;
5172926Sek110237 	char *propval = NULL;
5182926Sek110237 	char *strval;
5196168Shs24103 	int canmount;
5202676Seschrock 
5212676Seschrock 	if (nvlist_alloc(&props, NV_UNIQUE_NAME, 0) != 0) {
5222676Seschrock 		(void) fprintf(stderr, gettext("internal error: "
5232676Seschrock 		    "out of memory\n"));
5242676Seschrock 		return (1);
5252676Seschrock 	}
526789Sahrens 
527789Sahrens 	/* check options */
5284490Svb160487 	while ((c = getopt(argc, argv, ":V:b:so:p")) != -1) {
529789Sahrens 		switch (c) {
530789Sahrens 		case 'V':
531789Sahrens 			type = ZFS_TYPE_VOLUME;
5322676Seschrock 			if (zfs_nicestrtonum(g_zfs, optarg, &intval) != 0) {
5332676Seschrock 				(void) fprintf(stderr, gettext("bad volume "
5342676Seschrock 				    "size '%s': %s\n"), optarg,
5352676Seschrock 				    libzfs_error_description(g_zfs));
5362676Seschrock 				goto error;
5372676Seschrock 			}
5382676Seschrock 
5392676Seschrock 			if (nvlist_add_uint64(props,
5402676Seschrock 			    zfs_prop_to_name(ZFS_PROP_VOLSIZE),
5412676Seschrock 			    intval) != 0) {
5422676Seschrock 				(void) fprintf(stderr, gettext("internal "
5432676Seschrock 				    "error: out of memory\n"));
5442676Seschrock 				goto error;
5452676Seschrock 			}
5462676Seschrock 			volsize = intval;
547789Sahrens 			break;
5484490Svb160487 		case 'p':
5494490Svb160487 			parents = B_TRUE;
5504490Svb160487 			break;
551789Sahrens 		case 'b':
5525367Sahrens 			bflag = B_TRUE;
5532676Seschrock 			if (zfs_nicestrtonum(g_zfs, optarg, &intval) != 0) {
5542676Seschrock 				(void) fprintf(stderr, gettext("bad volume "
5552676Seschrock 				    "block size '%s': %s\n"), optarg,
5562676Seschrock 				    libzfs_error_description(g_zfs));
5572676Seschrock 				goto error;
5582676Seschrock 			}
5592676Seschrock 
5602676Seschrock 			if (nvlist_add_uint64(props,
5612676Seschrock 			    zfs_prop_to_name(ZFS_PROP_VOLBLOCKSIZE),
5622676Seschrock 			    intval) != 0) {
5632676Seschrock 				(void) fprintf(stderr, gettext("internal "
5642676Seschrock 				    "error: out of memory\n"));
5652676Seschrock 				goto error;
5662676Seschrock 			}
5672676Seschrock 			break;
5682676Seschrock 		case 'o':
5692676Seschrock 			propname = optarg;
5702676Seschrock 			if ((propval = strchr(propname, '=')) == NULL) {
5712676Seschrock 				(void) fprintf(stderr, gettext("missing "
5722676Seschrock 				    "'=' for -o option\n"));
5732676Seschrock 				goto error;
5742676Seschrock 			}
5752676Seschrock 			*propval = '\0';
5762676Seschrock 			propval++;
5772676Seschrock 			if (nvlist_lookup_string(props, propname,
5782676Seschrock 			    &strval) == 0) {
5792676Seschrock 				(void) fprintf(stderr, gettext("property '%s' "
5802676Seschrock 				    "specified multiple times\n"), propname);
5812676Seschrock 				goto error;
5822676Seschrock 			}
5832676Seschrock 			if (nvlist_add_string(props, propname, propval) != 0) {
5842676Seschrock 				(void) fprintf(stderr, gettext("internal "
5852676Seschrock 				    "error: out of memory\n"));
5862676Seschrock 				goto error;
5872676Seschrock 			}
588789Sahrens 			break;
589789Sahrens 		case 's':
5902082Seschrock 			noreserve = B_TRUE;
591789Sahrens 			break;
592789Sahrens 		case ':':
593789Sahrens 			(void) fprintf(stderr, gettext("missing size "
594789Sahrens 			    "argument\n"));
5952676Seschrock 			goto badusage;
596789Sahrens 			break;
597789Sahrens 		case '?':
598789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
599789Sahrens 			    optopt);
6002676Seschrock 			goto badusage;
601789Sahrens 		}
602789Sahrens 	}
603789Sahrens 
6045367Sahrens 	if ((bflag || noreserve) && type != ZFS_TYPE_VOLUME) {
6055367Sahrens 		(void) fprintf(stderr, gettext("'-s' and '-b' can only be "
6065367Sahrens 		    "used when creating a volume\n"));
6072676Seschrock 		goto badusage;
608789Sahrens 	}
609789Sahrens 
610789Sahrens 	argc -= optind;
611789Sahrens 	argv += optind;
612789Sahrens 
613789Sahrens 	/* check number of arguments */
614789Sahrens 	if (argc == 0) {
615789Sahrens 		(void) fprintf(stderr, gettext("missing %s argument\n"),
616789Sahrens 		    zfs_type_to_name(type));
6172676Seschrock 		goto badusage;
618789Sahrens 	}
619789Sahrens 	if (argc > 1) {
620789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
6212676Seschrock 		goto badusage;
6222676Seschrock 	}
6232676Seschrock 
6245481Sck153898 	if (type == ZFS_TYPE_VOLUME && !noreserve) {
6255481Sck153898 		zpool_handle_t *zpool_handle;
6265481Sck153898 		uint64_t spa_version;
6275481Sck153898 		char *p;
6285481Sck153898 		zfs_prop_t resv_prop;
6295481Sck153898 
6305481Sck153898 		if (p = strchr(argv[0], '/'))
6315481Sck153898 			*p = '\0';
6325481Sck153898 		zpool_handle = zpool_open(g_zfs, argv[0]);
6335481Sck153898 		if (p != NULL)
6345481Sck153898 			*p = '/';
6355481Sck153898 		if (zpool_handle == NULL)
6365481Sck153898 			goto error;
6375481Sck153898 		spa_version = zpool_get_prop_int(zpool_handle,
6385481Sck153898 		    ZPOOL_PROP_VERSION, NULL);
6395481Sck153898 		zpool_close(zpool_handle);
6405481Sck153898 		if (spa_version >= SPA_VERSION_REFRESERVATION)
6415481Sck153898 			resv_prop = ZFS_PROP_REFRESERVATION;
6425481Sck153898 		else
6435481Sck153898 			resv_prop = ZFS_PROP_RESERVATION;
6445481Sck153898 
6455481Sck153898 		if (nvlist_lookup_string(props, zfs_prop_to_name(resv_prop),
6465481Sck153898 		    &strval) != 0) {
6475481Sck153898 			if (nvlist_add_uint64(props,
6485481Sck153898 			    zfs_prop_to_name(resv_prop), volsize) != 0) {
6495481Sck153898 				(void) fprintf(stderr, gettext("internal "
6505481Sck153898 				    "error: out of memory\n"));
6515481Sck153898 				nvlist_free(props);
6525481Sck153898 				return (1);
6535481Sck153898 			}
6542676Seschrock 		}
655789Sahrens 	}
656789Sahrens 
6574490Svb160487 	if (parents && zfs_name_valid(argv[0], type)) {
6584490Svb160487 		/*
6594490Svb160487 		 * Now create the ancestors of target dataset.  If the target
6604490Svb160487 		 * already exists and '-p' option was used we should not
6614490Svb160487 		 * complain.
6624490Svb160487 		 */
6634490Svb160487 		if (zfs_dataset_exists(g_zfs, argv[0], type)) {
6644490Svb160487 			ret = 0;
6654490Svb160487 			goto error;
6664490Svb160487 		}
6674490Svb160487 		if (zfs_create_ancestors(g_zfs, argv[0]) != 0)
6684490Svb160487 			goto error;
6694490Svb160487 	}
6704490Svb160487 
671789Sahrens 	/* pass to libzfs */
6722676Seschrock 	if (zfs_create(g_zfs, argv[0], type, props) != 0)
6732676Seschrock 		goto error;
674789Sahrens 
6755094Slling 	if ((zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_DATASET)) == NULL)
6762676Seschrock 		goto error;
6776168Shs24103 	/*
6786168Shs24103 	 * if the user doesn't want the dataset automatically mounted,
6796168Shs24103 	 * then skip the mount/share step
6806168Shs24103 	 */
6816168Shs24103 
6826168Shs24103 	canmount = zfs_prop_get_int(zhp, ZFS_PROP_CANMOUNT);
683789Sahrens 
684789Sahrens 	/*
685789Sahrens 	 * Mount and/or share the new filesystem as appropriate.  We provide a
686789Sahrens 	 * verbose error message to let the user know that their filesystem was
687789Sahrens 	 * in fact created, even if we failed to mount or share it.
688789Sahrens 	 */
6896168Shs24103 	ret = 0;
6906168Shs24103 	if (canmount == ZFS_CANMOUNT_ON) {
6916168Shs24103 		if (zfs_mount(zhp, NULL, 0) != 0) {
6926168Shs24103 			(void) fprintf(stderr, gettext("filesystem "
6936168Shs24103 			    "successfully created, but not mounted\n"));
6946168Shs24103 			ret = 1;
6956168Shs24103 		} else if (zfs_share(zhp) != 0) {
6966168Shs24103 			(void) fprintf(stderr, gettext("filesystem "
6976168Shs24103 			    "successfully created, but not shared\n"));
6986168Shs24103 			ret = 1;
6996168Shs24103 		}
700789Sahrens 	}
701789Sahrens 
7022676Seschrock error:
7032676Seschrock 	if (zhp)
7042676Seschrock 		zfs_close(zhp);
7052676Seschrock 	nvlist_free(props);
706789Sahrens 	return (ret);
7072676Seschrock badusage:
7082676Seschrock 	nvlist_free(props);
7092676Seschrock 	usage(B_FALSE);
7102676Seschrock 	return (2);
711789Sahrens }
712789Sahrens 
713789Sahrens /*
714789Sahrens  * zfs destroy [-rf] <fs, snap, vol>
715789Sahrens  *
716789Sahrens  * 	-r	Recursively destroy all children
717789Sahrens  * 	-R	Recursively destroy all dependents, including clones
718789Sahrens  * 	-f	Force unmounting of any dependents
719789Sahrens  *
720789Sahrens  * Destroys the given dataset.  By default, it will unmount any filesystems,
721789Sahrens  * and refuse to destroy a dataset that has any dependents.  A dependent can
722789Sahrens  * either be a child, or a clone of a child.
723789Sahrens  */
724789Sahrens typedef struct destroy_cbdata {
7252082Seschrock 	boolean_t	cb_first;
726789Sahrens 	int		cb_force;
727789Sahrens 	int		cb_recurse;
728789Sahrens 	int		cb_error;
729789Sahrens 	int		cb_needforce;
730789Sahrens 	int		cb_doclones;
7313265Sahrens 	boolean_t	cb_closezhp;
732789Sahrens 	zfs_handle_t	*cb_target;
7332199Sahrens 	char		*cb_snapname;
734789Sahrens } destroy_cbdata_t;
735789Sahrens 
736789Sahrens /*
737789Sahrens  * Check for any dependents based on the '-r' or '-R' flags.
738789Sahrens  */
739789Sahrens static int
740789Sahrens destroy_check_dependent(zfs_handle_t *zhp, void *data)
741789Sahrens {
742789Sahrens 	destroy_cbdata_t *cbp = data;
743789Sahrens 	const char *tname = zfs_get_name(cbp->cb_target);
744789Sahrens 	const char *name = zfs_get_name(zhp);
745789Sahrens 
746789Sahrens 	if (strncmp(tname, name, strlen(tname)) == 0 &&
747789Sahrens 	    (name[strlen(tname)] == '/' || name[strlen(tname)] == '@')) {
748789Sahrens 		/*
749789Sahrens 		 * This is a direct descendant, not a clone somewhere else in
750789Sahrens 		 * the hierarchy.
751789Sahrens 		 */
752789Sahrens 		if (cbp->cb_recurse)
753789Sahrens 			goto out;
754789Sahrens 
755789Sahrens 		if (cbp->cb_first) {
756789Sahrens 			(void) fprintf(stderr, gettext("cannot destroy '%s': "
757789Sahrens 			    "%s has children\n"),
758789Sahrens 			    zfs_get_name(cbp->cb_target),
759789Sahrens 			    zfs_type_to_name(zfs_get_type(cbp->cb_target)));
760789Sahrens 			(void) fprintf(stderr, gettext("use '-r' to destroy "
761789Sahrens 			    "the following datasets:\n"));
7622082Seschrock 			cbp->cb_first = B_FALSE;
763789Sahrens 			cbp->cb_error = 1;
764789Sahrens 		}
765789Sahrens 
766789Sahrens 		(void) fprintf(stderr, "%s\n", zfs_get_name(zhp));
767789Sahrens 	} else {
768789Sahrens 		/*
769789Sahrens 		 * This is a clone.  We only want to report this if the '-r'
770789Sahrens 		 * wasn't specified, or the target is a snapshot.
771789Sahrens 		 */
772789Sahrens 		if (!cbp->cb_recurse &&
773789Sahrens 		    zfs_get_type(cbp->cb_target) != ZFS_TYPE_SNAPSHOT)
774789Sahrens 			goto out;
775789Sahrens 
776789Sahrens 		if (cbp->cb_first) {
777789Sahrens 			(void) fprintf(stderr, gettext("cannot destroy '%s': "
778789Sahrens 			    "%s has dependent clones\n"),
779789Sahrens 			    zfs_get_name(cbp->cb_target),
780789Sahrens 			    zfs_type_to_name(zfs_get_type(cbp->cb_target)));
781789Sahrens 			(void) fprintf(stderr, gettext("use '-R' to destroy "
782789Sahrens 			    "the following datasets:\n"));
7832082Seschrock 			cbp->cb_first = B_FALSE;
784789Sahrens 			cbp->cb_error = 1;
785789Sahrens 		}
786789Sahrens 
787789Sahrens 		(void) fprintf(stderr, "%s\n", zfs_get_name(zhp));
788789Sahrens 	}
789789Sahrens 
790789Sahrens out:
791789Sahrens 	zfs_close(zhp);
792789Sahrens 	return (0);
793789Sahrens }
794789Sahrens 
795789Sahrens static int
796789Sahrens destroy_callback(zfs_handle_t *zhp, void *data)
797789Sahrens {
798789Sahrens 	destroy_cbdata_t *cbp = data;
799789Sahrens 
800789Sahrens 	/*
801789Sahrens 	 * Ignore pools (which we've already flagged as an error before getting
802789Sahrens 	 * here.
803789Sahrens 	 */
804789Sahrens 	if (strchr(zfs_get_name(zhp), '/') == NULL &&
805789Sahrens 	    zfs_get_type(zhp) == ZFS_TYPE_FILESYSTEM) {
806789Sahrens 		zfs_close(zhp);
807789Sahrens 		return (0);
808789Sahrens 	}
809789Sahrens 
810789Sahrens 	/*
811789Sahrens 	 * Bail out on the first error.
812789Sahrens 	 */
813789Sahrens 	if (zfs_unmount(zhp, NULL, cbp->cb_force ? MS_FORCE : 0) != 0 ||
814789Sahrens 	    zfs_destroy(zhp) != 0) {
815789Sahrens 		zfs_close(zhp);
816789Sahrens 		return (-1);
817789Sahrens 	}
818789Sahrens 
819789Sahrens 	zfs_close(zhp);
820789Sahrens 	return (0);
821789Sahrens }
822789Sahrens 
8232199Sahrens static int
8242199Sahrens destroy_snap_clones(zfs_handle_t *zhp, void *arg)
8252199Sahrens {
8262199Sahrens 	destroy_cbdata_t *cbp = arg;
8272199Sahrens 	char thissnap[MAXPATHLEN];
8282199Sahrens 	zfs_handle_t *szhp;
8293265Sahrens 	boolean_t closezhp = cbp->cb_closezhp;
8303265Sahrens 	int rv;
8312199Sahrens 
8322199Sahrens 	(void) snprintf(thissnap, sizeof (thissnap),
8332199Sahrens 	    "%s@%s", zfs_get_name(zhp), cbp->cb_snapname);
8342199Sahrens 
8352199Sahrens 	libzfs_print_on_error(g_zfs, B_FALSE);
8362199Sahrens 	szhp = zfs_open(g_zfs, thissnap, ZFS_TYPE_SNAPSHOT);
8372199Sahrens 	libzfs_print_on_error(g_zfs, B_TRUE);
8382199Sahrens 	if (szhp) {
8392199Sahrens 		/*
8402199Sahrens 		 * Destroy any clones of this snapshot
8412199Sahrens 		 */
8422474Seschrock 		if (zfs_iter_dependents(szhp, B_FALSE, destroy_callback,
8432474Seschrock 		    cbp) != 0) {
8442474Seschrock 			zfs_close(szhp);
8453265Sahrens 			if (closezhp)
8463265Sahrens 				zfs_close(zhp);
8472474Seschrock 			return (-1);
8482474Seschrock 		}
8492199Sahrens 		zfs_close(szhp);
8502199Sahrens 	}
8512199Sahrens 
8523265Sahrens 	cbp->cb_closezhp = B_TRUE;
8533265Sahrens 	rv = zfs_iter_filesystems(zhp, destroy_snap_clones, arg);
8543265Sahrens 	if (closezhp)
8553265Sahrens 		zfs_close(zhp);
8563265Sahrens 	return (rv);
8572199Sahrens }
858789Sahrens 
859789Sahrens static int
860789Sahrens zfs_do_destroy(int argc, char **argv)
861789Sahrens {
862789Sahrens 	destroy_cbdata_t cb = { 0 };
863789Sahrens 	int c;
864789Sahrens 	zfs_handle_t *zhp;
8652199Sahrens 	char *cp;
866789Sahrens 
867789Sahrens 	/* check options */
868789Sahrens 	while ((c = getopt(argc, argv, "frR")) != -1) {
869789Sahrens 		switch (c) {
870789Sahrens 		case 'f':
871789Sahrens 			cb.cb_force = 1;
872789Sahrens 			break;
873789Sahrens 		case 'r':
874789Sahrens 			cb.cb_recurse = 1;
875789Sahrens 			break;
876789Sahrens 		case 'R':
877789Sahrens 			cb.cb_recurse = 1;
878789Sahrens 			cb.cb_doclones = 1;
879789Sahrens 			break;
880789Sahrens 		case '?':
881789Sahrens 		default:
882789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
883789Sahrens 			    optopt);
8842082Seschrock 			usage(B_FALSE);
885789Sahrens 		}
886789Sahrens 	}
887789Sahrens 
888789Sahrens 	argc -= optind;
889789Sahrens 	argv += optind;
890789Sahrens 
891789Sahrens 	/* check number of arguments */
892789Sahrens 	if (argc == 0) {
893789Sahrens 		(void) fprintf(stderr, gettext("missing path argument\n"));
8942082Seschrock 		usage(B_FALSE);
895789Sahrens 	}
896789Sahrens 	if (argc > 1) {
897789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
8982082Seschrock 		usage(B_FALSE);
899789Sahrens 	}
900789Sahrens 
9012199Sahrens 	/*
9022199Sahrens 	 * If we are doing recursive destroy of a snapshot, then the
9032199Sahrens 	 * named snapshot may not exist.  Go straight to libzfs.
9042199Sahrens 	 */
9052199Sahrens 	if (cb.cb_recurse && (cp = strchr(argv[0], '@'))) {
9062199Sahrens 		int ret;
9072199Sahrens 
9082199Sahrens 		*cp = '\0';
9095094Slling 		if ((zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_DATASET)) == NULL)
9102199Sahrens 			return (1);
9112199Sahrens 		*cp = '@';
9122199Sahrens 		cp++;
9132199Sahrens 
9142199Sahrens 		if (cb.cb_doclones) {
9152199Sahrens 			cb.cb_snapname = cp;
9162474Seschrock 			if (destroy_snap_clones(zhp, &cb) != 0) {
9172474Seschrock 				zfs_close(zhp);
9182474Seschrock 				return (1);
9192474Seschrock 			}
9202199Sahrens 		}
9212199Sahrens 
9222199Sahrens 		ret = zfs_destroy_snaps(zhp, cp);
9232199Sahrens 		zfs_close(zhp);
9242199Sahrens 		if (ret) {
9252199Sahrens 			(void) fprintf(stderr,
9262199Sahrens 			    gettext("no snapshots destroyed\n"));
9272199Sahrens 		}
9282199Sahrens 		return (ret != 0);
9292199Sahrens 	}
9302199Sahrens 
9312199Sahrens 
932789Sahrens 	/* Open the given dataset */
9335094Slling 	if ((zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_DATASET)) == NULL)
934789Sahrens 		return (1);
935789Sahrens 
936789Sahrens 	cb.cb_target = zhp;
937789Sahrens 
938789Sahrens 	/*
939789Sahrens 	 * Perform an explicit check for pools before going any further.
940789Sahrens 	 */
941789Sahrens 	if (!cb.cb_recurse && strchr(zfs_get_name(zhp), '/') == NULL &&
942789Sahrens 	    zfs_get_type(zhp) == ZFS_TYPE_FILESYSTEM) {
943789Sahrens 		(void) fprintf(stderr, gettext("cannot destroy '%s': "
944789Sahrens 		    "operation does not apply to pools\n"),
945789Sahrens 		    zfs_get_name(zhp));
946789Sahrens 		(void) fprintf(stderr, gettext("use 'zfs destroy -r "
947789Sahrens 		    "%s' to destroy all datasets in the pool\n"),
948789Sahrens 		    zfs_get_name(zhp));
949789Sahrens 		(void) fprintf(stderr, gettext("use 'zpool destroy %s' "
950789Sahrens 		    "to destroy the pool itself\n"), zfs_get_name(zhp));
951789Sahrens 		zfs_close(zhp);
952789Sahrens 		return (1);
953789Sahrens 	}
954789Sahrens 
955789Sahrens 	/*
956789Sahrens 	 * Check for any dependents and/or clones.
957789Sahrens 	 */
9582082Seschrock 	cb.cb_first = B_TRUE;
9592474Seschrock 	if (!cb.cb_doclones &&
9602474Seschrock 	    zfs_iter_dependents(zhp, B_TRUE, destroy_check_dependent,
9612474Seschrock 	    &cb) != 0) {
9622474Seschrock 		zfs_close(zhp);
9632474Seschrock 		return (1);
9642474Seschrock 	}
9652474Seschrock 
9662474Seschrock 	if (cb.cb_error ||
9672474Seschrock 	    zfs_iter_dependents(zhp, B_FALSE, destroy_callback, &cb) != 0) {
968789Sahrens 		zfs_close(zhp);
969789Sahrens 		return (1);
970789Sahrens 	}
971789Sahrens 
972789Sahrens 	/*
9732474Seschrock 	 * Do the real thing.  The callback will close the handle regardless of
9742474Seschrock 	 * whether it succeeds or not.
975789Sahrens 	 */
9764543Smarks 
9772474Seschrock 	if (destroy_callback(zhp, &cb) != 0)
9782474Seschrock 		return (1);
9792474Seschrock 
9802926Sek110237 
9812474Seschrock 	return (0);
982789Sahrens }
983789Sahrens 
984789Sahrens /*
985866Seschrock  * zfs get [-rHp] [-o field[,field]...] [-s source[,source]...]
986866Seschrock  * 	< all | property[,property]... > < fs | snap | vol > ...
987789Sahrens  *
988789Sahrens  *	-r	recurse over any child datasets
989789Sahrens  *	-H	scripted mode.  Headers are stripped, and fields are separated
990789Sahrens  *		by tabs instead of spaces.
991789Sahrens  *	-o	Set of fields to display.  One of "name,property,value,source".
992789Sahrens  *		Default is all four.
993789Sahrens  *	-s	Set of sources to allow.  One of
994789Sahrens  *		"local,default,inherited,temporary,none".  Default is all
995789Sahrens  *		five.
996789Sahrens  *	-p	Display values in parsable (literal) format.
997789Sahrens  *
998789Sahrens  *  Prints properties for the given datasets.  The user can control which
999789Sahrens  *  columns to display as well as which property types to allow.
1000789Sahrens  */
1001789Sahrens 
1002789Sahrens /*
1003789Sahrens  * Invoked to display the properties for a single dataset.
1004789Sahrens  */
1005789Sahrens static int
1006789Sahrens get_callback(zfs_handle_t *zhp, void *data)
1007789Sahrens {
1008789Sahrens 	char buf[ZFS_MAXPROPLEN];
10095094Slling 	zprop_source_t sourcetype;
1010789Sahrens 	char source[ZFS_MAXNAMELEN];
10115094Slling 	zprop_get_cbdata_t *cbp = data;
10122676Seschrock 	nvlist_t *userprop = zfs_get_user_props(zhp);
10135094Slling 	zprop_list_t *pl = cbp->cb_proplist;
10142676Seschrock 	nvlist_t *propval;
10152676Seschrock 	char *strval;
10162676Seschrock 	char *sourceval;
10172676Seschrock 
10182676Seschrock 	for (; pl != NULL; pl = pl->pl_next) {
10192676Seschrock 		/*
10202676Seschrock 		 * Skip the special fake placeholder.  This will also skip over
10212676Seschrock 		 * the name property when 'all' is specified.
10222676Seschrock 		 */
10232676Seschrock 		if (pl->pl_prop == ZFS_PROP_NAME &&
10242676Seschrock 		    pl == cbp->cb_proplist)
10252676Seschrock 			continue;
10262676Seschrock 
10275094Slling 		if (pl->pl_prop != ZPROP_INVAL) {
10282676Seschrock 			if (zfs_prop_get(zhp, pl->pl_prop, buf,
10292676Seschrock 			    sizeof (buf), &sourcetype, source,
10302676Seschrock 			    sizeof (source),
10312676Seschrock 			    cbp->cb_literal) != 0) {
10322676Seschrock 				if (pl->pl_all)
10332676Seschrock 					continue;
10343912Slling 				if (!zfs_prop_valid_for_type(pl->pl_prop,
10355094Slling 				    ZFS_TYPE_DATASET)) {
10363912Slling 					(void) fprintf(stderr,
10373912Slling 					    gettext("No such property '%s'\n"),
10383912Slling 					    zfs_prop_to_name(pl->pl_prop));
10393912Slling 					continue;
10403912Slling 				}
10415094Slling 				sourcetype = ZPROP_SRC_NONE;
10422676Seschrock 				(void) strlcpy(buf, "-", sizeof (buf));
10432676Seschrock 			}
10442676Seschrock 
10455094Slling 			zprop_print_one_property(zfs_get_name(zhp), cbp,
10462676Seschrock 			    zfs_prop_to_name(pl->pl_prop),
10472676Seschrock 			    buf, sourcetype, source);
10482676Seschrock 		} else {
10492676Seschrock 			if (nvlist_lookup_nvlist(userprop,
10502676Seschrock 			    pl->pl_user_prop, &propval) != 0) {
10512676Seschrock 				if (pl->pl_all)
10522676Seschrock 					continue;
10535094Slling 				sourcetype = ZPROP_SRC_NONE;
10542676Seschrock 				strval = "-";
10552676Seschrock 			} else {
10562676Seschrock 				verify(nvlist_lookup_string(propval,
10575094Slling 				    ZPROP_VALUE, &strval) == 0);
10582676Seschrock 				verify(nvlist_lookup_string(propval,
10595094Slling 				    ZPROP_SOURCE, &sourceval) == 0);
10602676Seschrock 
10612676Seschrock 				if (strcmp(sourceval,
10622676Seschrock 				    zfs_get_name(zhp)) == 0) {
10635094Slling 					sourcetype = ZPROP_SRC_LOCAL;
10642676Seschrock 				} else {
10655094Slling 					sourcetype = ZPROP_SRC_INHERITED;
10662676Seschrock 					(void) strlcpy(source,
10672676Seschrock 					    sourceval, sizeof (source));
10682676Seschrock 				}
10692676Seschrock 			}
10702676Seschrock 
10715094Slling 			zprop_print_one_property(zfs_get_name(zhp), cbp,
10722676Seschrock 			    pl->pl_user_prop, strval, sourcetype,
10732676Seschrock 			    source);
1074866Seschrock 		}
1075789Sahrens 	}
1076789Sahrens 
1077789Sahrens 	return (0);
1078789Sahrens }
1079789Sahrens 
1080789Sahrens static int
1081789Sahrens zfs_do_get(int argc, char **argv)
1082789Sahrens {
10835094Slling 	zprop_get_cbdata_t cb = { 0 };
10842082Seschrock 	boolean_t recurse = B_FALSE;
10852676Seschrock 	int i, c;
10862676Seschrock 	char *value, *fields;
1087866Seschrock 	int ret;
10885094Slling 	zprop_list_t fake_name = { 0 };
1089789Sahrens 
1090789Sahrens 	/*
1091789Sahrens 	 * Set up default columns and sources.
1092789Sahrens 	 */
10935094Slling 	cb.cb_sources = ZPROP_SRC_ALL;
1094789Sahrens 	cb.cb_columns[0] = GET_COL_NAME;
1095789Sahrens 	cb.cb_columns[1] = GET_COL_PROPERTY;
1096789Sahrens 	cb.cb_columns[2] = GET_COL_VALUE;
1097789Sahrens 	cb.cb_columns[3] = GET_COL_SOURCE;
10985094Slling 	cb.cb_type = ZFS_TYPE_DATASET;
1099789Sahrens 
1100789Sahrens 	/* check options */
1101789Sahrens 	while ((c = getopt(argc, argv, ":o:s:rHp")) != -1) {
1102789Sahrens 		switch (c) {
1103789Sahrens 		case 'p':
11042082Seschrock 			cb.cb_literal = B_TRUE;
1105789Sahrens 			break;
1106789Sahrens 		case 'r':
11072082Seschrock 			recurse = B_TRUE;
1108789Sahrens 			break;
1109789Sahrens 		case 'H':
11102082Seschrock 			cb.cb_scripted = B_TRUE;
1111789Sahrens 			break;
1112789Sahrens 		case ':':
1113789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
1114789Sahrens 			    "'%c' option\n"), optopt);
11152082Seschrock 			usage(B_FALSE);
1116789Sahrens 			break;
1117789Sahrens 		case 'o':
1118789Sahrens 			/*
1119789Sahrens 			 * Process the set of columns to display.  We zero out
1120789Sahrens 			 * the structure to give us a blank slate.
1121789Sahrens 			 */
1122789Sahrens 			bzero(&cb.cb_columns, sizeof (cb.cb_columns));
1123789Sahrens 			i = 0;
1124789Sahrens 			while (*optarg != '\0') {
1125789Sahrens 				static char *col_subopts[] =
1126789Sahrens 				    { "name", "property", "value", "source",
1127789Sahrens 				    NULL };
1128789Sahrens 
1129789Sahrens 				if (i == 4) {
1130789Sahrens 					(void) fprintf(stderr, gettext("too "
1131789Sahrens 					    "many fields given to -o "
1132789Sahrens 					    "option\n"));
11332082Seschrock 					usage(B_FALSE);
1134789Sahrens 				}
1135789Sahrens 
1136789Sahrens 				switch (getsubopt(&optarg, col_subopts,
1137789Sahrens 				    &value)) {
1138789Sahrens 				case 0:
1139789Sahrens 					cb.cb_columns[i++] = GET_COL_NAME;
1140789Sahrens 					break;
1141789Sahrens 				case 1:
1142789Sahrens 					cb.cb_columns[i++] = GET_COL_PROPERTY;
1143789Sahrens 					break;
1144789Sahrens 				case 2:
1145789Sahrens 					cb.cb_columns[i++] = GET_COL_VALUE;
1146789Sahrens 					break;
1147789Sahrens 				case 3:
1148789Sahrens 					cb.cb_columns[i++] = GET_COL_SOURCE;
1149789Sahrens 					break;
1150789Sahrens 				default:
1151789Sahrens 					(void) fprintf(stderr,
1152789Sahrens 					    gettext("invalid column name "
1153789Sahrens 					    "'%s'\n"), value);
11543912Slling 					usage(B_FALSE);
1155789Sahrens 				}
1156789Sahrens 			}
1157789Sahrens 			break;
1158789Sahrens 
1159789Sahrens 		case 's':
1160789Sahrens 			cb.cb_sources = 0;
1161789Sahrens 			while (*optarg != '\0') {
1162789Sahrens 				static char *source_subopts[] = {
1163789Sahrens 					"local", "default", "inherited",
1164789Sahrens 					"temporary", "none", NULL };
1165789Sahrens 
1166789Sahrens 				switch (getsubopt(&optarg, source_subopts,
1167789Sahrens 				    &value)) {
1168789Sahrens 				case 0:
11695094Slling 					cb.cb_sources |= ZPROP_SRC_LOCAL;
1170789Sahrens 					break;
1171789Sahrens 				case 1:
11725094Slling 					cb.cb_sources |= ZPROP_SRC_DEFAULT;
1173789Sahrens 					break;
1174789Sahrens 				case 2:
11755094Slling 					cb.cb_sources |= ZPROP_SRC_INHERITED;
1176789Sahrens 					break;
1177789Sahrens 				case 3:
11785094Slling 					cb.cb_sources |= ZPROP_SRC_TEMPORARY;
1179789Sahrens 					break;
1180789Sahrens 				case 4:
11815094Slling 					cb.cb_sources |= ZPROP_SRC_NONE;
1182789Sahrens 					break;
1183789Sahrens 				default:
1184789Sahrens 					(void) fprintf(stderr,
1185789Sahrens 					    gettext("invalid source "
1186789Sahrens 					    "'%s'\n"), value);
11873912Slling 					usage(B_FALSE);
1188789Sahrens 				}
1189789Sahrens 			}
1190789Sahrens 			break;
1191789Sahrens 
1192789Sahrens 		case '?':
1193789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
1194789Sahrens 			    optopt);
11952082Seschrock 			usage(B_FALSE);
1196789Sahrens 		}
1197789Sahrens 	}
1198789Sahrens 
1199789Sahrens 	argc -= optind;
1200789Sahrens 	argv += optind;
1201789Sahrens 
1202789Sahrens 	if (argc < 1) {
1203789Sahrens 		(void) fprintf(stderr, gettext("missing property "
1204789Sahrens 		    "argument\n"));
12052082Seschrock 		usage(B_FALSE);
1206789Sahrens 	}
1207789Sahrens 
1208789Sahrens 	fields = argv[0];
1209789Sahrens 
12105094Slling 	if (zprop_get_list(g_zfs, fields, &cb.cb_proplist, ZFS_TYPE_DATASET)
12115094Slling 	    != 0)
12122082Seschrock 		usage(B_FALSE);
1213789Sahrens 
1214789Sahrens 	argc--;
1215789Sahrens 	argv++;
1216789Sahrens 
1217789Sahrens 	/*
12182676Seschrock 	 * As part of zfs_expand_proplist(), we keep track of the maximum column
12192676Seschrock 	 * width for each property.  For the 'NAME' (and 'SOURCE') columns, we
12202676Seschrock 	 * need to know the maximum name length.  However, the user likely did
12212676Seschrock 	 * not specify 'name' as one of the properties to fetch, so we need to
12222676Seschrock 	 * make sure we always include at least this property for
12232676Seschrock 	 * print_get_headers() to work properly.
1224789Sahrens 	 */
12252676Seschrock 	if (cb.cb_proplist != NULL) {
12262676Seschrock 		fake_name.pl_prop = ZFS_PROP_NAME;
12272676Seschrock 		fake_name.pl_width = strlen(gettext("NAME"));
12282676Seschrock 		fake_name.pl_next = cb.cb_proplist;
12292676Seschrock 		cb.cb_proplist = &fake_name;
1230789Sahrens 	}
1231789Sahrens 
12322676Seschrock 	cb.cb_first = B_TRUE;
12332676Seschrock 
1234789Sahrens 	/* run for each object */
12355094Slling 	ret = zfs_for_each(argc, argv, recurse, ZFS_TYPE_DATASET, NULL,
12363635Sck153898 	    &cb.cb_proplist, get_callback, &cb, B_FALSE);
12372676Seschrock 
12382676Seschrock 	if (cb.cb_proplist == &fake_name)
12395094Slling 		zprop_free_list(fake_name.pl_next);
12402676Seschrock 	else
12415094Slling 		zprop_free_list(cb.cb_proplist);
12422676Seschrock 
12432676Seschrock 	return (ret);
1244789Sahrens }
1245789Sahrens 
1246789Sahrens /*
1247789Sahrens  * inherit [-r] <property> <fs|vol> ...
1248789Sahrens  *
1249789Sahrens  * 	-r	Recurse over all children
1250789Sahrens  *
1251789Sahrens  * For each dataset specified on the command line, inherit the given property
1252789Sahrens  * from its parent.  Inheriting a property at the pool level will cause it to
1253789Sahrens  * use the default value.  The '-r' flag will recurse over all children, and is
1254789Sahrens  * useful for setting a property on a hierarchy-wide basis, regardless of any
1255789Sahrens  * local modifications for each dataset.
1256789Sahrens  */
12572926Sek110237 
1258789Sahrens static int
1259789Sahrens inherit_callback(zfs_handle_t *zhp, void *data)
1260789Sahrens {
12614543Smarks 	char *propname = data;
12622926Sek110237 	int ret;
12632926Sek110237 
12644543Smarks 	ret = zfs_prop_inherit(zhp, propname);
12652926Sek110237 	return (ret != 0);
1266789Sahrens }
1267789Sahrens 
1268789Sahrens static int
1269789Sahrens zfs_do_inherit(int argc, char **argv)
1270789Sahrens {
12712082Seschrock 	boolean_t recurse = B_FALSE;
1272789Sahrens 	int c;
1273789Sahrens 	zfs_prop_t prop;
12744543Smarks 	char *propname;
12752926Sek110237 	int ret;
1276789Sahrens 
1277789Sahrens 	/* check options */
1278789Sahrens 	while ((c = getopt(argc, argv, "r")) != -1) {
1279789Sahrens 		switch (c) {
1280789Sahrens 		case 'r':
12812082Seschrock 			recurse = B_TRUE;
1282789Sahrens 			break;
1283789Sahrens 		case '?':
1284789Sahrens 		default:
1285789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
1286789Sahrens 			    optopt);
12872082Seschrock 			usage(B_FALSE);
1288789Sahrens 		}
1289789Sahrens 	}
1290789Sahrens 
1291789Sahrens 	argc -= optind;
1292789Sahrens 	argv += optind;
1293789Sahrens 
1294789Sahrens 	/* check number of arguments */
1295789Sahrens 	if (argc < 1) {
1296789Sahrens 		(void) fprintf(stderr, gettext("missing property argument\n"));
12972082Seschrock 		usage(B_FALSE);
1298789Sahrens 	}
1299789Sahrens 	if (argc < 2) {
1300789Sahrens 		(void) fprintf(stderr, gettext("missing dataset argument\n"));
13012082Seschrock 		usage(B_FALSE);
1302789Sahrens 	}
1303789Sahrens 
13044543Smarks 	propname = argv[0];
13052676Seschrock 	argc--;
13062676Seschrock 	argv++;
13072676Seschrock 
13085094Slling 	if ((prop = zfs_name_to_prop(propname)) != ZPROP_INVAL) {
13092676Seschrock 		if (zfs_prop_readonly(prop)) {
13102676Seschrock 			(void) fprintf(stderr, gettext(
13112676Seschrock 			    "%s property is read-only\n"),
13124543Smarks 			    propname);
13132676Seschrock 			return (1);
13142676Seschrock 		}
13152676Seschrock 		if (!zfs_prop_inheritable(prop)) {
13162676Seschrock 			(void) fprintf(stderr, gettext("'%s' property cannot "
13174543Smarks 			    "be inherited\n"), propname);
13182676Seschrock 			if (prop == ZFS_PROP_QUOTA ||
13195378Sck153898 			    prop == ZFS_PROP_RESERVATION ||
13205378Sck153898 			    prop == ZFS_PROP_REFQUOTA ||
13215378Sck153898 			    prop == ZFS_PROP_REFRESERVATION)
13222676Seschrock 				(void) fprintf(stderr, gettext("use 'zfs set "
13234543Smarks 				    "%s=none' to clear\n"), propname);
13242676Seschrock 			return (1);
13252676Seschrock 		}
13264543Smarks 	} else if (!zfs_prop_user(propname)) {
13274543Smarks 		(void) fprintf(stderr, gettext("invalid property '%s'\n"),
13284543Smarks 		    propname);
13292082Seschrock 		usage(B_FALSE);
1330789Sahrens 	}
13312676Seschrock 
13322926Sek110237 	ret = zfs_for_each(argc, argv, recurse,
13332676Seschrock 	    ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME, NULL, NULL,
13344543Smarks 	    inherit_callback, propname, B_FALSE);
13352926Sek110237 
13362926Sek110237 	return (ret);
1337789Sahrens }
1338789Sahrens 
13394577Sahrens typedef struct upgrade_cbdata {
13404577Sahrens 	uint64_t cb_numupgraded;
13414577Sahrens 	uint64_t cb_numsamegraded;
13424849Sahrens 	uint64_t cb_numfailed;
13434577Sahrens 	uint64_t cb_version;
13444577Sahrens 	boolean_t cb_newer;
13454577Sahrens 	boolean_t cb_foundone;
13464577Sahrens 	char cb_lastfs[ZFS_MAXNAMELEN];
13474577Sahrens } upgrade_cbdata_t;
13484577Sahrens 
13494577Sahrens static int
13504577Sahrens same_pool(zfs_handle_t *zhp, const char *name)
13514577Sahrens {
13524577Sahrens 	int len1 = strcspn(name, "/@");
13534577Sahrens 	const char *zhname = zfs_get_name(zhp);
13544577Sahrens 	int len2 = strcspn(zhname, "/@");
13554577Sahrens 
13564577Sahrens 	if (len1 != len2)
13574577Sahrens 		return (B_FALSE);
13584988Sek110237 	return (strncmp(name, zhname, len1) == 0);
13594577Sahrens }
13604577Sahrens 
13614577Sahrens static int
13624577Sahrens upgrade_list_callback(zfs_handle_t *zhp, void *data)
13634577Sahrens {
13644577Sahrens 	upgrade_cbdata_t *cb = data;
13654577Sahrens 	int version = zfs_prop_get_int(zhp, ZFS_PROP_VERSION);
13664577Sahrens 
13674577Sahrens 	/* list if it's old/new */
13684577Sahrens 	if ((!cb->cb_newer && version < ZPL_VERSION) ||
13695367Sahrens 	    (cb->cb_newer && version > ZPL_VERSION)) {
13704577Sahrens 		char *str;
13714577Sahrens 		if (cb->cb_newer) {
13724577Sahrens 			str = gettext("The following filesystems are "
13734577Sahrens 			    "formatted using a newer software version and\n"
13744577Sahrens 			    "cannot be accessed on the current system.\n\n");
13754577Sahrens 		} else {
13764577Sahrens 			str = gettext("The following filesystems are "
13774577Sahrens 			    "out of date, and can be upgraded.  After being\n"
13784577Sahrens 			    "upgraded, these filesystems (and any 'zfs send' "
13794577Sahrens 			    "streams generated from\n"
13804577Sahrens 			    "subsequent snapshots) will no longer be "
13814577Sahrens 			    "accessible by older software versions.\n\n");
13824577Sahrens 		}
13834577Sahrens 
13844577Sahrens 		if (!cb->cb_foundone) {
13854577Sahrens 			(void) puts(str);
13864577Sahrens 			(void) printf(gettext("VER  FILESYSTEM\n"));
13874577Sahrens 			(void) printf(gettext("---  ------------\n"));
13884577Sahrens 			cb->cb_foundone = B_TRUE;
13894577Sahrens 		}
13904577Sahrens 
13914577Sahrens 		(void) printf("%2u   %s\n", version, zfs_get_name(zhp));
13924577Sahrens 	}
13934577Sahrens 
13944577Sahrens 	return (0);
13954577Sahrens }
13964577Sahrens 
13974577Sahrens static int
13984577Sahrens upgrade_set_callback(zfs_handle_t *zhp, void *data)
13994577Sahrens {
14004577Sahrens 	upgrade_cbdata_t *cb = data;
14014577Sahrens 	int version = zfs_prop_get_int(zhp, ZFS_PROP_VERSION);
14024577Sahrens 
14035331Samw 	if (cb->cb_version >= ZPL_VERSION_FUID) {
14045331Samw 		int spa_version;
14055713Srm160521 
14065713Srm160521 		if (zfs_spa_version(zhp, &spa_version) < 0)
14075331Samw 			return (-1);
14085331Samw 
14095331Samw 		if (spa_version < SPA_VERSION_FUID) {
14105331Samw 			/* can't upgrade */
14115331Samw 			(void) printf(gettext("%s: can not be upgraded; "
14125331Samw 			    "the pool version needs to first be upgraded\nto "
14135331Samw 			    "version %d\n\n"),
14145331Samw 			    zfs_get_name(zhp), SPA_VERSION_FUID);
14155331Samw 			cb->cb_numfailed++;
14165331Samw 			return (0);
14175331Samw 		}
14185331Samw 	}
14195331Samw 
14204577Sahrens 	/* upgrade */
14214577Sahrens 	if (version < cb->cb_version) {
14224577Sahrens 		char verstr[16];
14234849Sahrens 		(void) snprintf(verstr, sizeof (verstr),
14244849Sahrens 		    "%llu", cb->cb_version);
14254577Sahrens 		if (cb->cb_lastfs[0] && !same_pool(zhp, cb->cb_lastfs)) {
14264577Sahrens 			/*
14274577Sahrens 			 * If they did "zfs upgrade -a", then we could
14284577Sahrens 			 * be doing ioctls to different pools.  We need
14294577Sahrens 			 * to log this history once to each pool.
14304577Sahrens 			 */
14314988Sek110237 			verify(zpool_stage_history(g_zfs, history_str) == 0);
14324577Sahrens 		}
14334577Sahrens 		if (zfs_prop_set(zhp, "version", verstr) == 0)
14344577Sahrens 			cb->cb_numupgraded++;
14354849Sahrens 		else
14364849Sahrens 			cb->cb_numfailed++;
14374577Sahrens 		(void) strcpy(cb->cb_lastfs, zfs_get_name(zhp));
14384577Sahrens 	} else if (version > cb->cb_version) {
14394577Sahrens 		/* can't downgrade */
14404577Sahrens 		(void) printf(gettext("%s: can not be downgraded; "
14414577Sahrens 		    "it is already at version %u\n"),
14424577Sahrens 		    zfs_get_name(zhp), version);
14434849Sahrens 		cb->cb_numfailed++;
14444577Sahrens 	} else {
14454577Sahrens 		cb->cb_numsamegraded++;
14464577Sahrens 	}
14474577Sahrens 	return (0);
14484577Sahrens }
14494577Sahrens 
14504577Sahrens /*
14514577Sahrens  * zfs upgrade
14524577Sahrens  * zfs upgrade -v
14534577Sahrens  * zfs upgrade [-r] [-V <version>] <-a | filesystem>
14544577Sahrens  */
14554577Sahrens static int
14564577Sahrens zfs_do_upgrade(int argc, char **argv)
14574577Sahrens {
14584577Sahrens 	boolean_t recurse = B_FALSE;
14594577Sahrens 	boolean_t all = B_FALSE;
14604577Sahrens 	boolean_t showversions = B_FALSE;
14614577Sahrens 	int ret;
14624577Sahrens 	upgrade_cbdata_t cb = { 0 };
14634577Sahrens 	char c;
14644577Sahrens 
14654577Sahrens 	/* check options */
14664577Sahrens 	while ((c = getopt(argc, argv, "rvV:a")) != -1) {
14674577Sahrens 		switch (c) {
14684577Sahrens 		case 'r':
14694577Sahrens 			recurse = B_TRUE;
14704577Sahrens 			break;
14714577Sahrens 		case 'v':
14724577Sahrens 			showversions = B_TRUE;
14734577Sahrens 			break;
14744577Sahrens 		case 'V':
14754577Sahrens 			if (zfs_prop_string_to_index(ZFS_PROP_VERSION,
14764577Sahrens 			    optarg, &cb.cb_version) != 0) {
14774577Sahrens 				(void) fprintf(stderr,
14784577Sahrens 				    gettext("invalid version %s\n"), optarg);
14794577Sahrens 				usage(B_FALSE);
14804577Sahrens 			}
14814577Sahrens 			break;
14824577Sahrens 		case 'a':
14834577Sahrens 			all = B_TRUE;
14844577Sahrens 			break;
14854577Sahrens 		case '?':
14864577Sahrens 		default:
14874577Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
14884577Sahrens 			    optopt);
14894577Sahrens 			usage(B_FALSE);
14904577Sahrens 		}
14914577Sahrens 	}
14924577Sahrens 
14934577Sahrens 	argc -= optind;
14944577Sahrens 	argv += optind;
14954577Sahrens 
14964577Sahrens 	if ((!all && !argc) && (recurse | cb.cb_version))
14974577Sahrens 		usage(B_FALSE);
14984577Sahrens 	if (showversions && (recurse || all || cb.cb_version || argc))
14994577Sahrens 		usage(B_FALSE);
15004577Sahrens 	if ((all || argc) && (showversions))
15014577Sahrens 		usage(B_FALSE);
15024577Sahrens 	if (all && argc)
15034577Sahrens 		usage(B_FALSE);
15044577Sahrens 
15054577Sahrens 	if (showversions) {
15064577Sahrens 		/* Show info on available versions. */
15074577Sahrens 		(void) printf(gettext("The following filesystem versions are "
15084577Sahrens 		    "supported:\n\n"));
15094577Sahrens 		(void) printf(gettext("VER  DESCRIPTION\n"));
15104577Sahrens 		(void) printf("---  -----------------------------------------"
15114577Sahrens 		    "---------------\n");
15124577Sahrens 		(void) printf(gettext(" 1   Initial ZFS filesystem version\n"));
15134577Sahrens 		(void) printf(gettext(" 2   Enhanced directory entries\n"));
15145331Samw 		(void) printf(gettext(" 3   Case insensitive and File system "
15155331Samw 		    "unique identifer (FUID)\n"));
15164577Sahrens 		(void) printf(gettext("\nFor more information on a particular "
15174577Sahrens 		    "version, including supported releases, see:\n\n"));
15184577Sahrens 		(void) printf("http://www.opensolaris.org/os/community/zfs/"
15194577Sahrens 		    "version/zpl/N\n\n");
15204577Sahrens 		(void) printf(gettext("Where 'N' is the version number.\n"));
15214577Sahrens 		ret = 0;
15224577Sahrens 	} else if (argc || all) {
15234577Sahrens 		/* Upgrade filesystems */
15244577Sahrens 		if (cb.cb_version == 0)
15254577Sahrens 			cb.cb_version = ZPL_VERSION;
15264577Sahrens 		ret = zfs_for_each(argc, argv, recurse, ZFS_TYPE_FILESYSTEM,
15274577Sahrens 		    NULL, NULL, upgrade_set_callback, &cb, B_TRUE);
15284577Sahrens 		(void) printf(gettext("%llu filesystems upgraded\n"),
15294577Sahrens 		    cb.cb_numupgraded);
15304577Sahrens 		if (cb.cb_numsamegraded) {
15314577Sahrens 			(void) printf(gettext("%llu filesystems already at "
15324577Sahrens 			    "this version\n"),
15334577Sahrens 			    cb.cb_numsamegraded);
15344577Sahrens 		}
15354849Sahrens 		if (cb.cb_numfailed != 0)
15364577Sahrens 			ret = 1;
15374577Sahrens 	} else {
15384577Sahrens 		/* List old-version filesytems */
15394577Sahrens 		boolean_t found;
15404577Sahrens 		(void) printf(gettext("This system is currently running "
15414577Sahrens 		    "ZFS filesystem version %llu.\n\n"), ZPL_VERSION);
15424577Sahrens 
15434577Sahrens 		ret = zfs_for_each(0, NULL, B_TRUE, ZFS_TYPE_FILESYSTEM,
15444577Sahrens 		    NULL, NULL, upgrade_list_callback, &cb, B_TRUE);
15454577Sahrens 
15464577Sahrens 		found = cb.cb_foundone;
15474577Sahrens 		cb.cb_foundone = B_FALSE;
15484577Sahrens 		cb.cb_newer = B_TRUE;
15494577Sahrens 
15504577Sahrens 		ret = zfs_for_each(0, NULL, B_TRUE, ZFS_TYPE_FILESYSTEM,
15514577Sahrens 		    NULL, NULL, upgrade_list_callback, &cb, B_TRUE);
15524577Sahrens 
15534577Sahrens 		if (!cb.cb_foundone && !found) {
15544577Sahrens 			(void) printf(gettext("All filesystems are "
15554577Sahrens 			    "formatted with the current version.\n"));
15564577Sahrens 		}
15574577Sahrens 	}
15584577Sahrens 
15594577Sahrens 	return (ret);
15604577Sahrens }
15614577Sahrens 
1562789Sahrens /*
15632379Ssjelinek  * list [-rH] [-o property[,property]...] [-t type[,type]...]
15642379Ssjelinek  *      [-s property [-s property]...] [-S property [-S property]...]
15652379Ssjelinek  *      <dataset> ...
1566789Sahrens  *
1567789Sahrens  * 	-r	Recurse over all children
15685331Samw  * 	-H	Scripted mode; elide headers and separate columns by tabs
1569789Sahrens  * 	-o	Control which fields to display.
1570866Seschrock  * 	-t	Control which object types to display.
15712379Ssjelinek  *	-s	Specify sort columns, descending order.
15722379Ssjelinek  *	-S	Specify sort columns, ascending order.
1573789Sahrens  *
1574789Sahrens  * When given no arguments, lists all filesystems in the system.
1575789Sahrens  * Otherwise, list the specified datasets, optionally recursing down them if
1576789Sahrens  * '-r' is specified.
1577789Sahrens  */
1578789Sahrens typedef struct list_cbdata {
15792082Seschrock 	boolean_t	cb_first;
15802082Seschrock 	boolean_t	cb_scripted;
15815094Slling 	zprop_list_t	*cb_proplist;
1582789Sahrens } list_cbdata_t;
1583789Sahrens 
1584789Sahrens /*
1585789Sahrens  * Given a list of columns to display, output appropriate headers for each one.
1586789Sahrens  */
1587789Sahrens static void
15885094Slling print_header(zprop_list_t *pl)
1589789Sahrens {
15902676Seschrock 	char headerbuf[ZFS_MAXPROPLEN];
15912676Seschrock 	const char *header;
1592789Sahrens 	int i;
15932676Seschrock 	boolean_t first = B_TRUE;
15942676Seschrock 	boolean_t right_justify;
15952676Seschrock 
15962676Seschrock 	for (; pl != NULL; pl = pl->pl_next) {
15972676Seschrock 		if (!first) {
1598789Sahrens 			(void) printf("  ");
15992676Seschrock 		} else {
16002676Seschrock 			first = B_FALSE;
16012676Seschrock 		}
16022676Seschrock 
16032676Seschrock 		right_justify = B_FALSE;
16045094Slling 		if (pl->pl_prop != ZPROP_INVAL) {
16052676Seschrock 			header = zfs_prop_column_name(pl->pl_prop);
16062676Seschrock 			right_justify = zfs_prop_align_right(pl->pl_prop);
16072676Seschrock 		} else {
16082676Seschrock 			for (i = 0; pl->pl_user_prop[i] != '\0'; i++)
16092676Seschrock 				headerbuf[i] = toupper(pl->pl_user_prop[i]);
16102676Seschrock 			headerbuf[i] = '\0';
16112676Seschrock 			header = headerbuf;
16122676Seschrock 		}
16132676Seschrock 
16142676Seschrock 		if (pl->pl_next == NULL && !right_justify)
16152676Seschrock 			(void) printf("%s", header);
16162676Seschrock 		else if (right_justify)
16172676Seschrock 			(void) printf("%*s", pl->pl_width, header);
16182676Seschrock 		else
16192676Seschrock 			(void) printf("%-*s", pl->pl_width, header);
1620789Sahrens 	}
1621789Sahrens 
1622789Sahrens 	(void) printf("\n");
1623789Sahrens }
1624789Sahrens 
1625789Sahrens /*
1626789Sahrens  * Given a dataset and a list of fields, print out all the properties according
1627789Sahrens  * to the described layout.
1628789Sahrens  */
1629789Sahrens static void
16305094Slling print_dataset(zfs_handle_t *zhp, zprop_list_t *pl, boolean_t scripted)
1631789Sahrens {
16322676Seschrock 	boolean_t first = B_TRUE;
1633789Sahrens 	char property[ZFS_MAXPROPLEN];
16342676Seschrock 	nvlist_t *userprops = zfs_get_user_props(zhp);
16352676Seschrock 	nvlist_t *propval;
16362676Seschrock 	char *propstr;
16372676Seschrock 	boolean_t right_justify;
16382676Seschrock 	int width;
16392676Seschrock 
16402676Seschrock 	for (; pl != NULL; pl = pl->pl_next) {
16412676Seschrock 		if (!first) {
1642789Sahrens 			if (scripted)
1643789Sahrens 				(void) printf("\t");
1644789Sahrens 			else
1645789Sahrens 				(void) printf("  ");
16462676Seschrock 		} else {
16472676Seschrock 			first = B_FALSE;
1648789Sahrens 		}
1649789Sahrens 
16502676Seschrock 		right_justify = B_FALSE;
16515094Slling 		if (pl->pl_prop != ZPROP_INVAL) {
16522676Seschrock 			if (zfs_prop_get(zhp, pl->pl_prop, property,
16532676Seschrock 			    sizeof (property), NULL, NULL, 0, B_FALSE) != 0)
16542676Seschrock 				propstr = "-";
16552676Seschrock 			else
16562676Seschrock 				propstr = property;
16572676Seschrock 
16582676Seschrock 			right_justify = zfs_prop_align_right(pl->pl_prop);
16592676Seschrock 		} else {
16602676Seschrock 			if (nvlist_lookup_nvlist(userprops,
16612676Seschrock 			    pl->pl_user_prop, &propval) != 0)
16622676Seschrock 				propstr = "-";
16632676Seschrock 			else
16642676Seschrock 				verify(nvlist_lookup_string(propval,
16655094Slling 				    ZPROP_VALUE, &propstr) == 0);
16662676Seschrock 		}
16672676Seschrock 
16682676Seschrock 		width = pl->pl_width;
1669789Sahrens 
1670866Seschrock 		/*
1671866Seschrock 		 * If this is being called in scripted mode, or if this is the
1672866Seschrock 		 * last column and it is left-justified, don't include a width
1673866Seschrock 		 * format specifier.
1674866Seschrock 		 */
16752676Seschrock 		if (scripted || (pl->pl_next == NULL && !right_justify))
16762676Seschrock 			(void) printf("%s", propstr);
16772676Seschrock 		else if (right_justify)
16782676Seschrock 			(void) printf("%*s", width, propstr);
16792676Seschrock 		else
16802676Seschrock 			(void) printf("%-*s", width, propstr);
1681789Sahrens 	}
1682789Sahrens 
1683789Sahrens 	(void) printf("\n");
1684789Sahrens }
1685789Sahrens 
1686789Sahrens /*
1687789Sahrens  * Generic callback function to list a dataset or snapshot.
1688789Sahrens  */
1689789Sahrens static int
1690789Sahrens list_callback(zfs_handle_t *zhp, void *data)
1691789Sahrens {
1692789Sahrens 	list_cbdata_t *cbp = data;
1693789Sahrens 
1694789Sahrens 	if (cbp->cb_first) {
1695789Sahrens 		if (!cbp->cb_scripted)
16962676Seschrock 			print_header(cbp->cb_proplist);
16972082Seschrock 		cbp->cb_first = B_FALSE;
1698789Sahrens 	}
1699789Sahrens 
17002676Seschrock 	print_dataset(zhp, cbp->cb_proplist, cbp->cb_scripted);
1701789Sahrens 
1702789Sahrens 	return (0);
1703789Sahrens }
1704789Sahrens 
1705789Sahrens static int
1706789Sahrens zfs_do_list(int argc, char **argv)
1707789Sahrens {
1708789Sahrens 	int c;
17092082Seschrock 	boolean_t recurse = B_FALSE;
17102082Seschrock 	boolean_t scripted = B_FALSE;
1711789Sahrens 	static char default_fields[] =
1712789Sahrens 	    "name,used,available,referenced,mountpoint";
17135094Slling 	int types = ZFS_TYPE_DATASET;
1714789Sahrens 	char *fields = NULL;
1715789Sahrens 	char *basic_fields = default_fields;
1716789Sahrens 	list_cbdata_t cb = { 0 };
1717789Sahrens 	char *value;
1718789Sahrens 	int ret;
1719789Sahrens 	char *type_subopts[] = { "filesystem", "volume", "snapshot", NULL };
17202379Ssjelinek 	zfs_sort_column_t *sortcol = NULL;
1721789Sahrens 
1722789Sahrens 	/* check options */
17232379Ssjelinek 	while ((c = getopt(argc, argv, ":o:rt:Hs:S:")) != -1) {
1724789Sahrens 		switch (c) {
1725789Sahrens 		case 'o':
1726789Sahrens 			fields = optarg;
1727789Sahrens 			break;
1728789Sahrens 		case 'r':
17292082Seschrock 			recurse = B_TRUE;
1730789Sahrens 			break;
1731789Sahrens 		case 'H':
17322082Seschrock 			scripted = B_TRUE;
1733789Sahrens 			break;
17342379Ssjelinek 		case 's':
17352676Seschrock 			if (zfs_add_sort_column(&sortcol, optarg,
17362676Seschrock 			    B_FALSE) != 0) {
17372379Ssjelinek 				(void) fprintf(stderr,
17382379Ssjelinek 				    gettext("invalid property '%s'\n"), optarg);
17392379Ssjelinek 				usage(B_FALSE);
17402379Ssjelinek 			}
17412379Ssjelinek 			break;
17422379Ssjelinek 		case 'S':
17432676Seschrock 			if (zfs_add_sort_column(&sortcol, optarg,
17442676Seschrock 			    B_TRUE) != 0) {
17452379Ssjelinek 				(void) fprintf(stderr,
17462379Ssjelinek 				    gettext("invalid property '%s'\n"), optarg);
17472379Ssjelinek 				usage(B_FALSE);
17482379Ssjelinek 			}
17492379Ssjelinek 			break;
1750789Sahrens 		case 't':
1751789Sahrens 			types = 0;
1752789Sahrens 			while (*optarg != '\0') {
1753789Sahrens 				switch (getsubopt(&optarg, type_subopts,
1754789Sahrens 				    &value)) {
1755789Sahrens 				case 0:
1756789Sahrens 					types |= ZFS_TYPE_FILESYSTEM;
1757789Sahrens 					break;
1758789Sahrens 				case 1:
1759789Sahrens 					types |= ZFS_TYPE_VOLUME;
1760789Sahrens 					break;
1761789Sahrens 				case 2:
1762789Sahrens 					types |= ZFS_TYPE_SNAPSHOT;
1763789Sahrens 					break;
1764789Sahrens 				default:
1765789Sahrens 					(void) fprintf(stderr,
1766789Sahrens 					    gettext("invalid type '%s'\n"),
1767789Sahrens 					    value);
17682082Seschrock 					usage(B_FALSE);
1769789Sahrens 				}
1770789Sahrens 			}
1771789Sahrens 			break;
1772789Sahrens 		case ':':
1773789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
1774789Sahrens 			    "'%c' option\n"), optopt);
17752082Seschrock 			usage(B_FALSE);
1776789Sahrens 			break;
1777789Sahrens 		case '?':
1778789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
1779789Sahrens 			    optopt);
17802082Seschrock 			usage(B_FALSE);
1781789Sahrens 		}
1782789Sahrens 	}
1783789Sahrens 
1784789Sahrens 	argc -= optind;
1785789Sahrens 	argv += optind;
1786789Sahrens 
1787789Sahrens 	if (fields == NULL)
1788789Sahrens 		fields = basic_fields;
1789789Sahrens 
1790866Seschrock 	/*
17915094Slling 	 * If the user specifies '-o all', the zprop_get_list() doesn't
1792866Seschrock 	 * normally include the name of the dataset.  For 'zfs list', we always
1793866Seschrock 	 * want this property to be first.
1794866Seschrock 	 */
17955094Slling 	if (zprop_get_list(g_zfs, fields, &cb.cb_proplist, ZFS_TYPE_DATASET)
17965094Slling 	    != 0)
17972082Seschrock 		usage(B_FALSE);
17982676Seschrock 
1799789Sahrens 	cb.cb_scripted = scripted;
18002082Seschrock 	cb.cb_first = B_TRUE;
1801789Sahrens 
18022676Seschrock 	ret = zfs_for_each(argc, argv, recurse, types, sortcol, &cb.cb_proplist,
18033635Sck153898 	    list_callback, &cb, B_TRUE);
18042379Ssjelinek 
18055094Slling 	zprop_free_list(cb.cb_proplist);
18062379Ssjelinek 	zfs_free_sort_columns(sortcol);
1807789Sahrens 
18084221Smmusante 	if (ret == 0 && cb.cb_first && !cb.cb_scripted)
1809789Sahrens 		(void) printf(gettext("no datasets available\n"));
1810789Sahrens 
1811789Sahrens 	return (ret);
1812789Sahrens }
1813789Sahrens 
1814789Sahrens /*
18154490Svb160487  * zfs rename <fs | snap | vol> <fs | snap | vol>
18164490Svb160487  * zfs rename -p <fs | vol> <fs | vol>
18174490Svb160487  * zfs rename -r <snap> <snap>
1818789Sahrens  *
1819789Sahrens  * Renames the given dataset to another of the same type.
18204490Svb160487  *
18214490Svb160487  * The '-p' flag creates all the non-existing ancestors of the target first.
1822789Sahrens  */
1823789Sahrens /* ARGSUSED */
1824789Sahrens static int
1825789Sahrens zfs_do_rename(int argc, char **argv)
1826789Sahrens {
1827789Sahrens 	zfs_handle_t *zhp;
18284007Smmusante 	int c;
18292082Seschrock 	int ret;
18304490Svb160487 	boolean_t recurse = B_FALSE;
18314490Svb160487 	boolean_t parents = B_FALSE;
1832789Sahrens 
1833789Sahrens 	/* check options */
18344490Svb160487 	while ((c = getopt(argc, argv, "pr")) != -1) {
18354007Smmusante 		switch (c) {
18364490Svb160487 		case 'p':
18374490Svb160487 			parents = B_TRUE;
18384490Svb160487 			break;
18394007Smmusante 		case 'r':
18404490Svb160487 			recurse = B_TRUE;
18414007Smmusante 			break;
18424007Smmusante 		case '?':
18434007Smmusante 		default:
18444007Smmusante 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
18454007Smmusante 			    optopt);
18464007Smmusante 			usage(B_FALSE);
18474007Smmusante 		}
1848789Sahrens 	}
1849789Sahrens 
18504007Smmusante 	argc -= optind;
18514007Smmusante 	argv += optind;
18524007Smmusante 
1853789Sahrens 	/* check number of arguments */
18544007Smmusante 	if (argc < 1) {
1855789Sahrens 		(void) fprintf(stderr, gettext("missing source dataset "
1856789Sahrens 		    "argument\n"));
18572082Seschrock 		usage(B_FALSE);
1858789Sahrens 	}
18594007Smmusante 	if (argc < 2) {
1860789Sahrens 		(void) fprintf(stderr, gettext("missing target dataset "
1861789Sahrens 		    "argument\n"));
18622082Seschrock 		usage(B_FALSE);
1863789Sahrens 	}
18644007Smmusante 	if (argc > 2) {
1865789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
18662082Seschrock 		usage(B_FALSE);
1867789Sahrens 	}
1868789Sahrens 
18694490Svb160487 	if (recurse && parents) {
18704490Svb160487 		(void) fprintf(stderr, gettext("-p and -r options are mutually "
18714490Svb160487 		    "exclusive\n"));
18724490Svb160487 		usage(B_FALSE);
18734490Svb160487 	}
18744490Svb160487 
18754007Smmusante 	if (recurse && strchr(argv[0], '@') == 0) {
18764007Smmusante 		(void) fprintf(stderr, gettext("source dataset for recursive "
18774007Smmusante 		    "rename must be a snapshot\n"));
18784007Smmusante 		usage(B_FALSE);
18794007Smmusante 	}
18804007Smmusante 
18814490Svb160487 	if ((zhp = zfs_open(g_zfs, argv[0], parents ? ZFS_TYPE_FILESYSTEM |
18825094Slling 	    ZFS_TYPE_VOLUME : ZFS_TYPE_DATASET)) == NULL)
1883789Sahrens 		return (1);
1884789Sahrens 
18854490Svb160487 	/* If we were asked and the name looks good, try to create ancestors. */
18864490Svb160487 	if (parents && zfs_name_valid(argv[1], zfs_get_type(zhp)) &&
18874490Svb160487 	    zfs_create_ancestors(g_zfs, argv[1]) != 0) {
18884490Svb160487 		zfs_close(zhp);
18894490Svb160487 		return (1);
18904490Svb160487 	}
18914490Svb160487 
18924007Smmusante 	ret = (zfs_rename(zhp, argv[1], recurse) != 0);
18932082Seschrock 
18942082Seschrock 	zfs_close(zhp);
18952082Seschrock 	return (ret);
18962082Seschrock }
18972082Seschrock 
18982082Seschrock /*
18992082Seschrock  * zfs promote <fs>
19002082Seschrock  *
19012082Seschrock  * Promotes the given clone fs to be the parent
19022082Seschrock  */
19032082Seschrock /* ARGSUSED */
19042082Seschrock static int
19052082Seschrock zfs_do_promote(int argc, char **argv)
19062082Seschrock {
19072082Seschrock 	zfs_handle_t *zhp;
19082082Seschrock 	int ret;
19092082Seschrock 
19102082Seschrock 	/* check options */
19112082Seschrock 	if (argc > 1 && argv[1][0] == '-') {
19122082Seschrock 		(void) fprintf(stderr, gettext("invalid option '%c'\n"),
19132082Seschrock 		    argv[1][1]);
19142082Seschrock 		usage(B_FALSE);
19152082Seschrock 	}
19162082Seschrock 
19172082Seschrock 	/* check number of arguments */
19182082Seschrock 	if (argc < 2) {
19192082Seschrock 		(void) fprintf(stderr, gettext("missing clone filesystem"
19202597Snd150628 		    " argument\n"));
19212082Seschrock 		usage(B_FALSE);
19222082Seschrock 	}
19232082Seschrock 	if (argc > 2) {
19242082Seschrock 		(void) fprintf(stderr, gettext("too many arguments\n"));
19252082Seschrock 		usage(B_FALSE);
19262082Seschrock 	}
19272082Seschrock 
19282082Seschrock 	zhp = zfs_open(g_zfs, argv[1], ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
19292082Seschrock 	if (zhp == NULL)
19302082Seschrock 		return (1);
19312082Seschrock 
19322082Seschrock 	ret = (zfs_promote(zhp) != 0);
19332082Seschrock 
19342926Sek110237 
1935789Sahrens 	zfs_close(zhp);
1936789Sahrens 	return (ret);
1937789Sahrens }
1938789Sahrens 
1939789Sahrens /*
19405568Sahrens  * zfs rollback [-rRf] <snapshot>
1941789Sahrens  *
1942789Sahrens  * 	-r	Delete any intervening snapshots before doing rollback
1943789Sahrens  * 	-R	Delete any snapshots and their clones
19445568Sahrens  * 	-f	ignored for backwards compatability
1945789Sahrens  *
1946789Sahrens  * Given a filesystem, rollback to a specific snapshot, discarding any changes
1947789Sahrens  * since then and making it the active dataset.  If more recent snapshots exist,
1948789Sahrens  * the command will complain unless the '-r' flag is given.
1949789Sahrens  */
1950789Sahrens typedef struct rollback_cbdata {
1951789Sahrens 	uint64_t	cb_create;
19522082Seschrock 	boolean_t	cb_first;
1953789Sahrens 	int		cb_doclones;
1954789Sahrens 	char		*cb_target;
1955789Sahrens 	int		cb_error;
19562082Seschrock 	boolean_t	cb_recurse;
19572082Seschrock 	boolean_t	cb_dependent;
1958789Sahrens } rollback_cbdata_t;
1959789Sahrens 
1960789Sahrens /*
1961789Sahrens  * Report any snapshots more recent than the one specified.  Used when '-r' is
1962789Sahrens  * not specified.  We reuse this same callback for the snapshot dependents - if
1963789Sahrens  * 'cb_dependent' is set, then this is a dependent and we should report it
1964789Sahrens  * without checking the transaction group.
1965789Sahrens  */
1966789Sahrens static int
1967789Sahrens rollback_check(zfs_handle_t *zhp, void *data)
1968789Sahrens {
1969789Sahrens 	rollback_cbdata_t *cbp = data;
1970789Sahrens 
19712082Seschrock 	if (cbp->cb_doclones) {
19722082Seschrock 		zfs_close(zhp);
1973789Sahrens 		return (0);
19742082Seschrock 	}
1975789Sahrens 
1976789Sahrens 	if (!cbp->cb_dependent) {
1977789Sahrens 		if (strcmp(zfs_get_name(zhp), cbp->cb_target) != 0 &&
19781294Slling 		    zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT &&
1979789Sahrens 		    zfs_prop_get_int(zhp, ZFS_PROP_CREATETXG) >
1980789Sahrens 		    cbp->cb_create) {
1981789Sahrens 
1982789Sahrens 			if (cbp->cb_first && !cbp->cb_recurse) {
1983789Sahrens 				(void) fprintf(stderr, gettext("cannot "
1984789Sahrens 				    "rollback to '%s': more recent snapshots "
1985789Sahrens 				    "exist\n"),
1986789Sahrens 				    cbp->cb_target);
1987789Sahrens 				(void) fprintf(stderr, gettext("use '-r' to "
1988789Sahrens 				    "force deletion of the following "
1989789Sahrens 				    "snapshots:\n"));
1990789Sahrens 				cbp->cb_first = 0;
1991789Sahrens 				cbp->cb_error = 1;
1992789Sahrens 			}
1993789Sahrens 
1994789Sahrens 			if (cbp->cb_recurse) {
19952082Seschrock 				cbp->cb_dependent = B_TRUE;
19962474Seschrock 				if (zfs_iter_dependents(zhp, B_TRUE,
19972474Seschrock 				    rollback_check, cbp) != 0) {
19982474Seschrock 					zfs_close(zhp);
19992474Seschrock 					return (-1);
20002474Seschrock 				}
20012082Seschrock 				cbp->cb_dependent = B_FALSE;
2002789Sahrens 			} else {
2003789Sahrens 				(void) fprintf(stderr, "%s\n",
2004789Sahrens 				    zfs_get_name(zhp));
2005789Sahrens 			}
2006789Sahrens 		}
2007789Sahrens 	} else {
2008789Sahrens 		if (cbp->cb_first && cbp->cb_recurse) {
2009789Sahrens 			(void) fprintf(stderr, gettext("cannot rollback to "
2010789Sahrens 			    "'%s': clones of previous snapshots exist\n"),
2011789Sahrens 			    cbp->cb_target);
2012789Sahrens 			(void) fprintf(stderr, gettext("use '-R' to "
2013789Sahrens 			    "force deletion of the following clones and "
2014789Sahrens 			    "dependents:\n"));
2015789Sahrens 			cbp->cb_first = 0;
2016789Sahrens 			cbp->cb_error = 1;
2017789Sahrens 		}
2018789Sahrens 
2019789Sahrens 		(void) fprintf(stderr, "%s\n", zfs_get_name(zhp));
2020789Sahrens 	}
2021789Sahrens 
2022789Sahrens 	zfs_close(zhp);
2023789Sahrens 	return (0);
2024789Sahrens }
2025789Sahrens 
2026789Sahrens static int
2027789Sahrens zfs_do_rollback(int argc, char **argv)
2028789Sahrens {
2029789Sahrens 	int ret;
2030789Sahrens 	int c;
20315749Sahrens 	boolean_t force = B_FALSE;
2032789Sahrens 	rollback_cbdata_t cb = { 0 };
2033789Sahrens 	zfs_handle_t *zhp, *snap;
2034789Sahrens 	char parentname[ZFS_MAXNAMELEN];
2035789Sahrens 	char *delim;
2036789Sahrens 
2037789Sahrens 	/* check options */
20385568Sahrens 	while ((c = getopt(argc, argv, "rRf")) != -1) {
2039789Sahrens 		switch (c) {
2040789Sahrens 		case 'r':
2041789Sahrens 			cb.cb_recurse = 1;
2042789Sahrens 			break;
2043789Sahrens 		case 'R':
2044789Sahrens 			cb.cb_recurse = 1;
2045789Sahrens 			cb.cb_doclones = 1;
2046789Sahrens 			break;
20475568Sahrens 		case 'f':
20485749Sahrens 			force = B_TRUE;
20495568Sahrens 			break;
2050789Sahrens 		case '?':
2051789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
2052789Sahrens 			    optopt);
20532082Seschrock 			usage(B_FALSE);
2054789Sahrens 		}
2055789Sahrens 	}
2056789Sahrens 
2057789Sahrens 	argc -= optind;
2058789Sahrens 	argv += optind;
2059789Sahrens 
2060789Sahrens 	/* check number of arguments */
2061789Sahrens 	if (argc < 1) {
2062789Sahrens 		(void) fprintf(stderr, gettext("missing dataset argument\n"));
20632082Seschrock 		usage(B_FALSE);
2064789Sahrens 	}
2065789Sahrens 	if (argc > 1) {
2066789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
20672082Seschrock 		usage(B_FALSE);
2068789Sahrens 	}
2069789Sahrens 
2070789Sahrens 	/* open the snapshot */
20712082Seschrock 	if ((snap = zfs_open(g_zfs, argv[0], ZFS_TYPE_SNAPSHOT)) == NULL)
2072789Sahrens 		return (1);
2073789Sahrens 
20741294Slling 	/* open the parent dataset */
20751294Slling 	(void) strlcpy(parentname, argv[0], sizeof (parentname));
2076789Sahrens 	verify((delim = strrchr(parentname, '@')) != NULL);
2077789Sahrens 	*delim = '\0';
20785094Slling 	if ((zhp = zfs_open(g_zfs, parentname, ZFS_TYPE_DATASET)) == NULL) {
2079789Sahrens 		zfs_close(snap);
2080789Sahrens 		return (1);
2081789Sahrens 	}
2082789Sahrens 
2083789Sahrens 	/*
2084789Sahrens 	 * Check for more recent snapshots and/or clones based on the presence
2085789Sahrens 	 * of '-r' and '-R'.
2086789Sahrens 	 */
20871294Slling 	cb.cb_target = argv[0];
20881294Slling 	cb.cb_create = zfs_prop_get_int(snap, ZFS_PROP_CREATETXG);
20892082Seschrock 	cb.cb_first = B_TRUE;
2090789Sahrens 	cb.cb_error = 0;
20912474Seschrock 	if ((ret = zfs_iter_children(zhp, rollback_check, &cb)) != 0)
20922474Seschrock 		goto out;
2093789Sahrens 
2094789Sahrens 	if ((ret = cb.cb_error) != 0)
2095789Sahrens 		goto out;
2096789Sahrens 
2097789Sahrens 	/*
20981294Slling 	 * Rollback parent to the given snapshot.
2099789Sahrens 	 */
21005749Sahrens 	ret = zfs_rollback(zhp, snap, force);
2101789Sahrens 
2102789Sahrens out:
2103789Sahrens 	zfs_close(snap);
2104789Sahrens 	zfs_close(zhp);
2105789Sahrens 
2106789Sahrens 	if (ret == 0)
2107789Sahrens 		return (0);
2108789Sahrens 	else
2109789Sahrens 		return (1);
2110789Sahrens }
2111789Sahrens 
2112789Sahrens /*
2113789Sahrens  * zfs set property=value { fs | snap | vol } ...
2114789Sahrens  *
2115789Sahrens  * Sets the given property for all datasets specified on the command line.
2116789Sahrens  */
2117789Sahrens typedef struct set_cbdata {
2118789Sahrens 	char		*cb_propname;
2119789Sahrens 	char		*cb_value;
2120789Sahrens } set_cbdata_t;
2121789Sahrens 
2122789Sahrens static int
2123789Sahrens set_callback(zfs_handle_t *zhp, void *data)
2124789Sahrens {
2125789Sahrens 	set_cbdata_t *cbp = data;
2126789Sahrens 
21272676Seschrock 	if (zfs_prop_set(zhp, cbp->cb_propname, cbp->cb_value) != 0) {
21282169Snd150628 		switch (libzfs_errno(g_zfs)) {
21292169Snd150628 		case EZFS_MOUNTFAILED:
21302169Snd150628 			(void) fprintf(stderr, gettext("property may be set "
21312169Snd150628 			    "but unable to remount filesystem\n"));
21322169Snd150628 			break;
21333126Sahl 		case EZFS_SHARENFSFAILED:
21342169Snd150628 			(void) fprintf(stderr, gettext("property may be set "
21352169Snd150628 			    "but unable to reshare filesystem\n"));
21362169Snd150628 			break;
21372169Snd150628 		}
2138789Sahrens 		return (1);
21392169Snd150628 	}
21402856Snd150628 	return (0);
2141789Sahrens }
2142789Sahrens 
2143789Sahrens static int
2144789Sahrens zfs_do_set(int argc, char **argv)
2145789Sahrens {
2146789Sahrens 	set_cbdata_t cb;
21472926Sek110237 	int ret;
2148789Sahrens 
2149789Sahrens 	/* check for options */
2150789Sahrens 	if (argc > 1 && argv[1][0] == '-') {
2151789Sahrens 		(void) fprintf(stderr, gettext("invalid option '%c'\n"),
2152789Sahrens 		    argv[1][1]);
21532082Seschrock 		usage(B_FALSE);
2154789Sahrens 	}
2155789Sahrens 
2156789Sahrens 	/* check number of arguments */
2157789Sahrens 	if (argc < 2) {
2158789Sahrens 		(void) fprintf(stderr, gettext("missing property=value "
2159789Sahrens 		    "argument\n"));
21602082Seschrock 		usage(B_FALSE);
2161789Sahrens 	}
2162789Sahrens 	if (argc < 3) {
2163789Sahrens 		(void) fprintf(stderr, gettext("missing dataset name\n"));
21642082Seschrock 		usage(B_FALSE);
2165789Sahrens 	}
2166789Sahrens 
2167789Sahrens 	/* validate property=value argument */
2168789Sahrens 	cb.cb_propname = argv[1];
2169789Sahrens 	if ((cb.cb_value = strchr(cb.cb_propname, '=')) == NULL) {
2170789Sahrens 		(void) fprintf(stderr, gettext("missing value in "
2171789Sahrens 		    "property=value argument\n"));
21722082Seschrock 		usage(B_FALSE);
2173789Sahrens 	}
2174789Sahrens 
2175789Sahrens 	*cb.cb_value = '\0';
2176789Sahrens 	cb.cb_value++;
2177789Sahrens 
2178789Sahrens 	if (*cb.cb_propname == '\0') {
2179789Sahrens 		(void) fprintf(stderr,
2180789Sahrens 		    gettext("missing property in property=value argument\n"));
21812082Seschrock 		usage(B_FALSE);
2182789Sahrens 	}
2183789Sahrens 
21844543Smarks 
21852926Sek110237 	ret = zfs_for_each(argc - 2, argv + 2, B_FALSE,
21865094Slling 	    ZFS_TYPE_DATASET, NULL, NULL, set_callback, &cb, B_FALSE);
21872926Sek110237 
21882926Sek110237 	return (ret);
2189789Sahrens }
2190789Sahrens 
2191789Sahrens /*
21922199Sahrens  * zfs snapshot [-r] <fs@snap>
2193789Sahrens  *
2194789Sahrens  * Creates a snapshot with the given name.  While functionally equivalent to
21955331Samw  * 'zfs create', it is a separate command to differentiate intent.
2196789Sahrens  */
2197789Sahrens static int
2198789Sahrens zfs_do_snapshot(int argc, char **argv)
2199789Sahrens {
22004490Svb160487 	boolean_t recursive = B_FALSE;
22012199Sahrens 	int ret;
22022199Sahrens 	char c;
22032199Sahrens 
2204789Sahrens 	/* check options */
22052199Sahrens 	while ((c = getopt(argc, argv, ":r")) != -1) {
22062199Sahrens 		switch (c) {
22072199Sahrens 		case 'r':
22082199Sahrens 			recursive = B_TRUE;
22092199Sahrens 			break;
22102199Sahrens 		case '?':
22112199Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
22122199Sahrens 			    optopt);
22132199Sahrens 			usage(B_FALSE);
22142199Sahrens 		}
2215789Sahrens 	}
2216789Sahrens 
22172199Sahrens 	argc -= optind;
22182199Sahrens 	argv += optind;
22192199Sahrens 
2220789Sahrens 	/* check number of arguments */
22212199Sahrens 	if (argc < 1) {
2222789Sahrens 		(void) fprintf(stderr, gettext("missing snapshot argument\n"));
22232082Seschrock 		usage(B_FALSE);
2224789Sahrens 	}
22252199Sahrens 	if (argc > 1) {
2226789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
22272082Seschrock 		usage(B_FALSE);
2228789Sahrens 	}
2229789Sahrens 
22302199Sahrens 	ret = zfs_snapshot(g_zfs, argv[0], recursive);
22312199Sahrens 	if (ret && recursive)
22322199Sahrens 		(void) fprintf(stderr, gettext("no snapshots were created\n"));
22332199Sahrens 	return (ret != 0);
2234789Sahrens }
2235789Sahrens 
2236789Sahrens /*
22375367Sahrens  * zfs send [-v] -R [-i|-I <@snap>] <fs@snap>
22385367Sahrens  * zfs send [-v] [-i|-I <@snap>] <fs@snap>
2239789Sahrens  *
2240789Sahrens  * Send a backup stream to stdout.
2241789Sahrens  */
2242789Sahrens static int
22431749Sahrens zfs_do_send(int argc, char **argv)
2244789Sahrens {
2245789Sahrens 	char *fromname = NULL;
22465367Sahrens 	char *toname = NULL;
22472885Sahrens 	char *cp;
22482885Sahrens 	zfs_handle_t *zhp;
22495367Sahrens 	boolean_t doall = B_FALSE;
22505367Sahrens 	boolean_t replicate = B_FALSE;
22515367Sahrens 	boolean_t fromorigin = B_FALSE;
22525367Sahrens 	boolean_t verbose = B_FALSE;
2253789Sahrens 	int c, err;
2254789Sahrens 
2255789Sahrens 	/* check options */
22565367Sahrens 	while ((c = getopt(argc, argv, ":i:I:Rv")) != -1) {
2257789Sahrens 		switch (c) {
2258789Sahrens 		case 'i':
22592885Sahrens 			if (fromname)
22602885Sahrens 				usage(B_FALSE);
2261789Sahrens 			fromname = optarg;
2262789Sahrens 			break;
22635367Sahrens 		case 'I':
22645367Sahrens 			if (fromname)
22655367Sahrens 				usage(B_FALSE);
22665367Sahrens 			fromname = optarg;
22675367Sahrens 			doall = B_TRUE;
22685367Sahrens 			break;
22695367Sahrens 		case 'R':
22705367Sahrens 			replicate = B_TRUE;
22715367Sahrens 			break;
22725367Sahrens 		case 'v':
22735367Sahrens 			verbose = B_TRUE;
22745367Sahrens 			break;
2275789Sahrens 		case ':':
2276789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
2277789Sahrens 			    "'%c' option\n"), optopt);
22782082Seschrock 			usage(B_FALSE);
2279789Sahrens 			break;
2280789Sahrens 		case '?':
2281789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
2282789Sahrens 			    optopt);
22832082Seschrock 			usage(B_FALSE);
2284789Sahrens 		}
2285789Sahrens 	}
2286789Sahrens 
2287789Sahrens 	argc -= optind;
2288789Sahrens 	argv += optind;
2289789Sahrens 
2290789Sahrens 	/* check number of arguments */
2291789Sahrens 	if (argc < 1) {
2292789Sahrens 		(void) fprintf(stderr, gettext("missing snapshot argument\n"));
22932082Seschrock 		usage(B_FALSE);
2294789Sahrens 	}
2295789Sahrens 	if (argc > 1) {
2296789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
22972082Seschrock 		usage(B_FALSE);
2298789Sahrens 	}
2299789Sahrens 
2300789Sahrens 	if (isatty(STDOUT_FILENO)) {
2301789Sahrens 		(void) fprintf(stderr,
23022885Sahrens 		    gettext("Error: Stream can not be written to a terminal.\n"
23033912Slling 		    "You must redirect standard output.\n"));
2304789Sahrens 		return (1);
2305789Sahrens 	}
2306789Sahrens 
23075367Sahrens 	cp = strchr(argv[0], '@');
23085367Sahrens 	if (cp == NULL) {
23095367Sahrens 		(void) fprintf(stderr,
23105367Sahrens 		    gettext("argument must be a snapshot\n"));
23115367Sahrens 		usage(B_FALSE);
23125367Sahrens 	}
23135367Sahrens 	*cp = '\0';
23145367Sahrens 	toname = cp + 1;
23155367Sahrens 	zhp = zfs_open(g_zfs, argv[0], ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
23165367Sahrens 	if (zhp == NULL)
23172665Snd150628 		return (1);
23182665Snd150628 
23192885Sahrens 	/*
23202885Sahrens 	 * If they specified the full path to the snapshot, chop off
23215367Sahrens 	 * everything except the short name of the snapshot, but special
23225367Sahrens 	 * case if they specify the origin.
23232885Sahrens 	 */
23242885Sahrens 	if (fromname && (cp = strchr(fromname, '@')) != NULL) {
23255367Sahrens 		char origin[ZFS_MAXNAMELEN];
23265367Sahrens 		zprop_source_t src;
23275367Sahrens 
23285367Sahrens 		(void) zfs_prop_get(zhp, ZFS_PROP_ORIGIN,
23295367Sahrens 		    origin, sizeof (origin), &src, NULL, 0, B_FALSE);
23305367Sahrens 
23315367Sahrens 		if (strcmp(origin, fromname) == 0) {
23325367Sahrens 			fromname = NULL;
23335367Sahrens 			fromorigin = B_TRUE;
23345367Sahrens 		} else {
23355367Sahrens 			*cp = '\0';
23365367Sahrens 			if (cp != fromname && strcmp(argv[0], fromname)) {
23375367Sahrens 				(void) fprintf(stderr,
23385367Sahrens 				    gettext("incremental source must be "
23395367Sahrens 				    "in same filesystem\n"));
23405367Sahrens 				usage(B_FALSE);
23415367Sahrens 			}
23425367Sahrens 			fromname = cp + 1;
23435367Sahrens 			if (strchr(fromname, '@') || strchr(fromname, '/')) {
23445367Sahrens 				(void) fprintf(stderr,
23455367Sahrens 				    gettext("invalid incremental source\n"));
23465367Sahrens 				usage(B_FALSE);
23475367Sahrens 			}
23482885Sahrens 		}
2349789Sahrens 	}
2350789Sahrens 
23515367Sahrens 	if (replicate && fromname == NULL)
23525367Sahrens 		doall = B_TRUE;
23535367Sahrens 
23545367Sahrens 	err = zfs_send(zhp, fromname, toname, replicate, doall, fromorigin,
23555367Sahrens 	    verbose, STDOUT_FILENO);
23562885Sahrens 	zfs_close(zhp);
2357789Sahrens 
2358789Sahrens 	return (err != 0);
2359789Sahrens }
2360789Sahrens 
2361789Sahrens /*
23625367Sahrens  * zfs receive [-dnvF] <fs@snap>
2363789Sahrens  *
2364789Sahrens  * Restore a backup stream from stdin.
2365789Sahrens  */
2366789Sahrens static int
23671749Sahrens zfs_do_receive(int argc, char **argv)
2368789Sahrens {
2369789Sahrens 	int c, err;
23705367Sahrens 	recvflags_t flags;
23715367Sahrens 
23725367Sahrens 	bzero(&flags, sizeof (recvflags_t));
2373789Sahrens 	/* check options */
23742665Snd150628 	while ((c = getopt(argc, argv, ":dnvF")) != -1) {
2375789Sahrens 		switch (c) {
2376789Sahrens 		case 'd':
23775367Sahrens 			flags.isprefix = B_TRUE;
2378789Sahrens 			break;
2379789Sahrens 		case 'n':
23805367Sahrens 			flags.dryrun = B_TRUE;
2381789Sahrens 			break;
2382789Sahrens 		case 'v':
23835367Sahrens 			flags.verbose = B_TRUE;
2384789Sahrens 			break;
23852665Snd150628 		case 'F':
23865367Sahrens 			flags.force = B_TRUE;
23872665Snd150628 			break;
2388789Sahrens 		case ':':
2389789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
2390789Sahrens 			    "'%c' option\n"), optopt);
23912082Seschrock 			usage(B_FALSE);
2392789Sahrens 			break;
2393789Sahrens 		case '?':
2394789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
2395789Sahrens 			    optopt);
23962082Seschrock 			usage(B_FALSE);
2397789Sahrens 		}
2398789Sahrens 	}
2399789Sahrens 
2400789Sahrens 	argc -= optind;
2401789Sahrens 	argv += optind;
2402789Sahrens 
2403789Sahrens 	/* check number of arguments */
2404789Sahrens 	if (argc < 1) {
2405789Sahrens 		(void) fprintf(stderr, gettext("missing snapshot argument\n"));
24062082Seschrock 		usage(B_FALSE);
2407789Sahrens 	}
2408789Sahrens 	if (argc > 1) {
2409789Sahrens 		(void) fprintf(stderr, gettext("too many arguments\n"));
24102082Seschrock 		usage(B_FALSE);
2411789Sahrens 	}
2412789Sahrens 
2413789Sahrens 	if (isatty(STDIN_FILENO)) {
2414789Sahrens 		(void) fprintf(stderr,
2415789Sahrens 		    gettext("Error: Backup stream can not be read "
24163912Slling 		    "from a terminal.\n"
24173912Slling 		    "You must redirect standard input.\n"));
2418789Sahrens 		return (1);
2419789Sahrens 	}
2420789Sahrens 
24215367Sahrens 	err = zfs_receive(g_zfs, argv[0], flags, STDIN_FILENO, NULL);
24222926Sek110237 
24234543Smarks 	return (err != 0);
24244543Smarks }
24254543Smarks 
24264543Smarks typedef struct allow_cb {
24274543Smarks 	int  a_permcnt;
24284543Smarks 	size_t a_treeoffset;
24294543Smarks } allow_cb_t;
24304543Smarks 
24314543Smarks static void
24324543Smarks zfs_print_perms(avl_tree_t *tree)
24334543Smarks {
24344543Smarks 	zfs_perm_node_t *permnode;
24354543Smarks 
24364543Smarks 	permnode = avl_first(tree);
24374543Smarks 	while (permnode != NULL) {
24384543Smarks 		(void) printf("%s", permnode->z_pname);
24394543Smarks 		permnode = AVL_NEXT(tree, permnode);
24404543Smarks 		if (permnode)
24414543Smarks 			(void) printf(",");
24424543Smarks 		else
24434543Smarks 			(void) printf("\n");
24444543Smarks 	}
24454543Smarks }
24464543Smarks 
24474543Smarks /*
24484543Smarks  * Iterate over user/groups/everyone/... and the call perm_iter
24494543Smarks  * function to print actual permission when tree has >0 nodes.
24504543Smarks  */
24514543Smarks static void
24524543Smarks zfs_iter_perms(avl_tree_t *tree, const char *banner, allow_cb_t *cb)
24534543Smarks {
24544543Smarks 	zfs_allow_node_t *item;
24554543Smarks 	avl_tree_t *ptree;
24564543Smarks 
24574543Smarks 	item = avl_first(tree);
24584543Smarks 	while (item) {
24594543Smarks 		ptree = (void *)((char *)item + cb->a_treeoffset);
24604543Smarks 		if (avl_numnodes(ptree)) {
24614543Smarks 			if (cb->a_permcnt++ == 0)
24624543Smarks 				(void) printf("%s\n", banner);
24634543Smarks 			(void) printf("\t%s", item->z_key);
24644543Smarks 			/*
24654543Smarks 			 * Avoid an extra space being printed
24664543Smarks 			 * for "everyone" which is keyed with a null
24674543Smarks 			 * string
24684543Smarks 			 */
24694543Smarks 			if (item->z_key[0] != '\0')
24704543Smarks 				(void) printf(" ");
24714543Smarks 			zfs_print_perms(ptree);
24724543Smarks 		}
24734543Smarks 		item = AVL_NEXT(tree, item);
24744543Smarks 	}
24754543Smarks }
24764543Smarks 
24774543Smarks #define	LINES "-------------------------------------------------------------\n"
24784543Smarks static int
24794543Smarks zfs_print_allows(char *ds)
24804543Smarks {
24814543Smarks 	zfs_allow_t *curperms, *perms;
24824543Smarks 	zfs_handle_t *zhp;
24834543Smarks 	allow_cb_t allowcb = { 0 };
24844543Smarks 	char banner[MAXPATHLEN];
24854543Smarks 
24864543Smarks 	if (ds[0] == '-')
24874543Smarks 		usage(B_FALSE);
24884543Smarks 
24894543Smarks 	if (strrchr(ds, '@')) {
24904543Smarks 		(void) fprintf(stderr, gettext("Snapshots don't have 'allow'"
24914543Smarks 		    " permissions\n"));
24924543Smarks 		return (1);
24934543Smarks 	}
24945094Slling 	if ((zhp = zfs_open(g_zfs, ds, ZFS_TYPE_DATASET)) == NULL)
24954543Smarks 		return (1);
24964543Smarks 
24974543Smarks 	if (zfs_perm_get(zhp, &perms)) {
24984543Smarks 		(void) fprintf(stderr,
24994543Smarks 		    gettext("Failed to retrieve 'allows' on %s\n"), ds);
25004543Smarks 		zfs_close(zhp);
25014543Smarks 		return (1);
25024543Smarks 	}
25034543Smarks 
25044543Smarks 	zfs_close(zhp);
25054543Smarks 
25064543Smarks 	if (perms != NULL)
25074543Smarks 		(void) printf("%s", LINES);
25084543Smarks 	for (curperms = perms; curperms; curperms = curperms->z_next) {
25094543Smarks 
25104543Smarks 		(void) snprintf(banner, sizeof (banner),
25114543Smarks 		    "Permission sets on (%s)", curperms->z_setpoint);
25124543Smarks 		allowcb.a_treeoffset =
25134543Smarks 		    offsetof(zfs_allow_node_t, z_localdescend);
25144543Smarks 		allowcb.a_permcnt = 0;
25154543Smarks 		zfs_iter_perms(&curperms->z_sets, banner, &allowcb);
25164543Smarks 
25174543Smarks 		(void) snprintf(banner, sizeof (banner),
25184543Smarks 		    "Create time permissions on (%s)", curperms->z_setpoint);
25194543Smarks 		allowcb.a_treeoffset =
25204543Smarks 		    offsetof(zfs_allow_node_t, z_localdescend);
25214543Smarks 		allowcb.a_permcnt = 0;
25224543Smarks 		zfs_iter_perms(&curperms->z_crperms, banner, &allowcb);
25234543Smarks 
25244543Smarks 
25254543Smarks 		(void) snprintf(banner, sizeof (banner),
25264543Smarks 		    "Local permissions on (%s)", curperms->z_setpoint);
25274543Smarks 		allowcb.a_treeoffset = offsetof(zfs_allow_node_t, z_local);
25284543Smarks 		allowcb.a_permcnt = 0;
25294543Smarks 		zfs_iter_perms(&curperms->z_user, banner, &allowcb);
25304543Smarks 		zfs_iter_perms(&curperms->z_group, banner, &allowcb);
25314543Smarks 		zfs_iter_perms(&curperms->z_everyone, banner, &allowcb);
25324543Smarks 
25334543Smarks 		(void) snprintf(banner, sizeof (banner),
25344543Smarks 		    "Descendent permissions on (%s)", curperms->z_setpoint);
25354543Smarks 		allowcb.a_treeoffset = offsetof(zfs_allow_node_t, z_descend);
25364543Smarks 		allowcb.a_permcnt = 0;
25374543Smarks 		zfs_iter_perms(&curperms->z_user, banner, &allowcb);
25384543Smarks 		zfs_iter_perms(&curperms->z_group, banner, &allowcb);
25394543Smarks 		zfs_iter_perms(&curperms->z_everyone, banner, &allowcb);
25404543Smarks 
25414543Smarks 		(void) snprintf(banner, sizeof (banner),
25424543Smarks 		    "Local+Descendent permissions on (%s)",
25434543Smarks 		    curperms->z_setpoint);
25444543Smarks 		allowcb.a_treeoffset =
25454543Smarks 		    offsetof(zfs_allow_node_t, z_localdescend);
25464543Smarks 		allowcb.a_permcnt = 0;
25474543Smarks 		zfs_iter_perms(&curperms->z_user, banner, &allowcb);
25484543Smarks 		zfs_iter_perms(&curperms->z_group, banner, &allowcb);
25494543Smarks 		zfs_iter_perms(&curperms->z_everyone, banner, &allowcb);
25504543Smarks 
25514543Smarks 		(void) printf("%s", LINES);
25522926Sek110237 	}
25534543Smarks 	zfs_free_allows(perms);
25544543Smarks 	return (0);
25554543Smarks }
25564543Smarks 
25574543Smarks #define	ALLOWOPTIONS "ldcsu:g:e"
25584543Smarks #define	UNALLOWOPTIONS "ldcsu:g:er"
25594543Smarks 
25604543Smarks /*
25614543Smarks  * Validate options, and build necessary datastructure to display/remove/add
25624543Smarks  * permissions.
25634543Smarks  * Returns 0 - If permissions should be added/removed
25644543Smarks  * Returns 1 - If permissions should be displayed.
25654543Smarks  * Returns -1 - on failure
25664543Smarks  */
25674543Smarks int
25684543Smarks parse_allow_args(int *argc, char **argv[], boolean_t unallow,
25694543Smarks     char **ds, int *recurse, nvlist_t **zperms)
25704543Smarks {
25714543Smarks 	int c;
25724543Smarks 	char *options = unallow ? UNALLOWOPTIONS : ALLOWOPTIONS;
25734543Smarks 	zfs_deleg_inherit_t deleg_type = ZFS_DELEG_NONE;
25744543Smarks 	zfs_deleg_who_type_t who_type = ZFS_DELEG_WHO_UNKNOWN;
25754860Smarks 	char *who = NULL;
25764543Smarks 	char *perms = NULL;
25774543Smarks 	zfs_handle_t *zhp;
25784543Smarks 
25794543Smarks 	while ((c = getopt(*argc, *argv, options)) != -1) {
25804543Smarks 		switch (c) {
25814543Smarks 		case 'l':
25824543Smarks 			if (who_type == ZFS_DELEG_CREATE ||
25834543Smarks 			    who_type == ZFS_DELEG_NAMED_SET)
25844543Smarks 				usage(B_FALSE);
25854543Smarks 
25864543Smarks 			deleg_type |= ZFS_DELEG_PERM_LOCAL;
25874543Smarks 			break;
25884543Smarks 		case 'd':
25894543Smarks 			if (who_type == ZFS_DELEG_CREATE ||
25904543Smarks 			    who_type == ZFS_DELEG_NAMED_SET)
25914543Smarks 				usage(B_FALSE);
25924543Smarks 
25934543Smarks 			deleg_type |= ZFS_DELEG_PERM_DESCENDENT;
25944543Smarks 			break;
25954543Smarks 		case 'r':
25964543Smarks 			*recurse = B_TRUE;
25974543Smarks 			break;
25984543Smarks 		case 'c':
25994543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26004543Smarks 				usage(B_FALSE);
26014543Smarks 			if (deleg_type)
26024543Smarks 				usage(B_FALSE);
26034543Smarks 			who_type = ZFS_DELEG_CREATE;
26044543Smarks 			break;
26054543Smarks 		case 's':
26064543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26074543Smarks 				usage(B_FALSE);
26084543Smarks 			if (deleg_type)
26094543Smarks 				usage(B_FALSE);
26104543Smarks 			who_type = ZFS_DELEG_NAMED_SET;
26114543Smarks 			break;
26124543Smarks 		case 'u':
26134543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26144543Smarks 				usage(B_FALSE);
26154543Smarks 			who_type = ZFS_DELEG_USER;
26164543Smarks 			who = optarg;
26174543Smarks 			break;
26184543Smarks 		case 'g':
26194543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26204543Smarks 				usage(B_FALSE);
26214543Smarks 			who_type = ZFS_DELEG_GROUP;
26224543Smarks 			who = optarg;
26234543Smarks 			break;
26244543Smarks 		case 'e':
26254543Smarks 			if (who_type != ZFS_DELEG_WHO_UNKNOWN)
26264543Smarks 				usage(B_FALSE);
26274543Smarks 			who_type = ZFS_DELEG_EVERYONE;
26284543Smarks 			break;
26294543Smarks 		default:
26304543Smarks 			usage(B_FALSE);
26314543Smarks 			break;
26324543Smarks 		}
26334543Smarks 	}
26344543Smarks 
26354543Smarks 	if (deleg_type == 0)
26364543Smarks 		deleg_type = ZFS_DELEG_PERM_LOCALDESCENDENT;
26374543Smarks 
26384543Smarks 	*argc -= optind;
26394543Smarks 	*argv += optind;
26404543Smarks 
26414543Smarks 	if (unallow == B_FALSE && *argc == 1) {
26424543Smarks 		/*
26434543Smarks 		 * Only print permissions if no options were processed
26444543Smarks 		 */
26454543Smarks 		if (optind == 1)
26464543Smarks 			return (1);
26474543Smarks 		else
26484543Smarks 			usage(B_FALSE);
26494543Smarks 	}
26504543Smarks 
26514543Smarks 	/*
26524543Smarks 	 * initialize variables for zfs_build_perms based on number
26534543Smarks 	 * of arguments.
26544543Smarks 	 * 3 arguments ==>	zfs [un]allow joe perm,perm,perm <dataset> or
26554543Smarks 	 *			zfs [un]allow -s @set1 perm,perm <dataset>
26564543Smarks 	 * 2 arguments ==>	zfs [un]allow -c perm,perm <dataset> or
26574543Smarks 	 *			zfs [un]allow -u|-g <name> perm <dataset> or
26584543Smarks 	 *			zfs [un]allow -e perm,perm <dataset>
26594543Smarks 	 *			zfs unallow joe <dataset>
26604543Smarks 	 *			zfs unallow -s @set1 <dataset>
26614543Smarks 	 * 1 argument  ==>	zfs [un]allow -e <dataset> or
26624543Smarks 	 *			zfs [un]allow -c <dataset>
26634543Smarks 	 */
26644543Smarks 
26654543Smarks 	switch (*argc) {
26664543Smarks 	case 3:
26674543Smarks 		perms = (*argv)[1];
26684543Smarks 		who = (*argv)[0];
26694543Smarks 		*ds = (*argv)[2];
26704543Smarks 
26714543Smarks 		/*
26724543Smarks 		 * advance argc/argv for do_allow cases.
26734543Smarks 		 * for do_allow case make sure who have a know who type
26744543Smarks 		 * and its not a permission set.
26754543Smarks 		 */
26764543Smarks 		if (unallow == B_TRUE) {
26774543Smarks 			*argc -= 2;
26784543Smarks 			*argv += 2;
26794543Smarks 		} else if (who_type != ZFS_DELEG_WHO_UNKNOWN &&
26804543Smarks 		    who_type != ZFS_DELEG_NAMED_SET)
26814543Smarks 			usage(B_FALSE);
26824543Smarks 		break;
26834543Smarks 
26844543Smarks 	case 2:
26854543Smarks 		if (unallow == B_TRUE && (who_type == ZFS_DELEG_EVERYONE ||
26864543Smarks 		    who_type == ZFS_DELEG_CREATE || who != NULL)) {
26874543Smarks 			perms = (*argv)[0];
26884543Smarks 			*ds = (*argv)[1];
26894543Smarks 		} else {
26904543Smarks 			if (unallow == B_FALSE &&
26914543Smarks 			    (who_type == ZFS_DELEG_WHO_UNKNOWN ||
26924543Smarks 			    who_type == ZFS_DELEG_NAMED_SET))
26934543Smarks 				usage(B_FALSE);
26944543Smarks 			else if (who_type == ZFS_DELEG_WHO_UNKNOWN ||
26954543Smarks 			    who_type == ZFS_DELEG_NAMED_SET)
26964543Smarks 				who = (*argv)[0];
26974543Smarks 			else if (who_type != ZFS_DELEG_NAMED_SET)
26984543Smarks 				perms = (*argv)[0];
26994543Smarks 			*ds = (*argv)[1];
27004543Smarks 		}
27014543Smarks 		if (unallow == B_TRUE) {
27024543Smarks 			(*argc)--;
27034543Smarks 			(*argv)++;
27044543Smarks 		}
27054543Smarks 		break;
27064543Smarks 
27074543Smarks 	case 1:
27084543Smarks 		if (unallow == B_FALSE)
27094543Smarks 			usage(B_FALSE);
27104543Smarks 		if (who == NULL && who_type != ZFS_DELEG_CREATE &&
27114543Smarks 		    who_type != ZFS_DELEG_EVERYONE)
27124543Smarks 			usage(B_FALSE);
27134543Smarks 		*ds = (*argv)[0];
27144543Smarks 		break;
27154543Smarks 
27164543Smarks 	default:
27174543Smarks 		usage(B_FALSE);
27184543Smarks 	}
27194543Smarks 
27204543Smarks 	if (strrchr(*ds, '@')) {
27214543Smarks 		(void) fprintf(stderr,
27224543Smarks 		    gettext("Can't set or remove 'allow' permissions "
27234543Smarks 		    "on snapshots.\n"));
27244543Smarks 			return (-1);
27254543Smarks 	}
27264543Smarks 
27275094Slling 	if ((zhp = zfs_open(g_zfs, *ds, ZFS_TYPE_DATASET)) == NULL)
27284543Smarks 		return (-1);
27294543Smarks 
27304543Smarks 	if ((zfs_build_perms(zhp, who, perms,
27314543Smarks 	    who_type, deleg_type, zperms)) != 0) {
27324543Smarks 		zfs_close(zhp);
27334543Smarks 		return (-1);
27344543Smarks 	}
27354543Smarks 	zfs_close(zhp);
27364543Smarks 	return (0);
27374543Smarks }
27384543Smarks 
27394543Smarks static int
27404543Smarks zfs_do_allow(int argc, char **argv)
27414543Smarks {
27424543Smarks 	char *ds;
27434543Smarks 	nvlist_t *zperms = NULL;
27444543Smarks 	zfs_handle_t *zhp;
27454543Smarks 	int unused;
27464543Smarks 	int ret;
27474543Smarks 
27484543Smarks 	if ((ret = parse_allow_args(&argc, &argv, B_FALSE, &ds,
27494543Smarks 	    &unused, &zperms)) == -1)
27504543Smarks 		return (1);
27514543Smarks 
27524543Smarks 	if (ret == 1)
27534543Smarks 		return (zfs_print_allows(argv[0]));
27544543Smarks 
27555094Slling 	if ((zhp = zfs_open(g_zfs, ds, ZFS_TYPE_DATASET)) == NULL)
27564543Smarks 		return (1);
27574543Smarks 
27584543Smarks 	if (zfs_perm_set(zhp, zperms)) {
27594543Smarks 		zfs_close(zhp);
27604543Smarks 		nvlist_free(zperms);
27614543Smarks 		return (1);
27624543Smarks 	}
27634543Smarks 	nvlist_free(zperms);
27644543Smarks 	zfs_close(zhp);
27654543Smarks 
27664543Smarks 	return (0);
27674543Smarks }
27684543Smarks 
27694543Smarks static int
27704543Smarks unallow_callback(zfs_handle_t *zhp, void *data)
27714543Smarks {
27724543Smarks 	nvlist_t *nvp = (nvlist_t *)data;
27734543Smarks 	int error;
27744543Smarks 
27754543Smarks 	error = zfs_perm_remove(zhp, nvp);
27764543Smarks 	if (error) {
27774543Smarks 		(void) fprintf(stderr, gettext("Failed to remove permissions "
27784543Smarks 		    "on %s\n"), zfs_get_name(zhp));
27794543Smarks 	}
27804543Smarks 	return (error);
27814543Smarks }
27824543Smarks 
27834543Smarks static int
27844543Smarks zfs_do_unallow(int argc, char **argv)
27854543Smarks {
27864543Smarks 	int recurse = B_FALSE;
27874543Smarks 	char *ds;
27884543Smarks 	int error;
27894543Smarks 	nvlist_t *zperms = NULL;
27904543Smarks 
27914543Smarks 	if (parse_allow_args(&argc, &argv, B_TRUE,
27924543Smarks 	    &ds, &recurse, &zperms) == -1)
27934543Smarks 		return (1);
27944543Smarks 
27954543Smarks 	error = zfs_for_each(argc, argv, recurse,
27964543Smarks 	    ZFS_TYPE_FILESYSTEM|ZFS_TYPE_VOLUME, NULL,
27974543Smarks 	    NULL, unallow_callback, (void *)zperms, B_FALSE);
27984543Smarks 
27994543Smarks 	if (zperms)
28004543Smarks 		nvlist_free(zperms);
28014543Smarks 
28024543Smarks 	return (error);
2803789Sahrens }
2804789Sahrens 
28051356Seschrock typedef struct get_all_cbdata {
28061356Seschrock 	zfs_handle_t	**cb_handles;
28071356Seschrock 	size_t		cb_alloc;
28081356Seschrock 	size_t		cb_used;
28093126Sahl 	uint_t		cb_types;
28104737Smmusante 	boolean_t	cb_verbose;
28111356Seschrock } get_all_cbdata_t;
28121356Seschrock 
28134737Smmusante #define	CHECK_SPINNER 30
28144737Smmusante #define	SPINNER_TIME 3		/* seconds */
28154737Smmusante #define	MOUNT_TIME 5		/* seconds */
28164737Smmusante 
28171356Seschrock static int
28183126Sahl get_one_dataset(zfs_handle_t *zhp, void *data)
28191356Seschrock {
28204737Smmusante 	static char spin[] = { '-', '\\', '|', '/' };
28214737Smmusante 	static int spinval = 0;
28224737Smmusante 	static int spincheck = 0;
28234737Smmusante 	static time_t last_spin_time = (time_t)0;
28241356Seschrock 	get_all_cbdata_t *cbp = data;
28253126Sahl 	zfs_type_t type = zfs_get_type(zhp);
28261356Seschrock 
28274737Smmusante 	if (cbp->cb_verbose) {
28284737Smmusante 		if (--spincheck < 0) {
28294737Smmusante 			time_t now = time(NULL);
28304737Smmusante 			if (last_spin_time + SPINNER_TIME < now) {
28314737Smmusante 				(void) printf("\b%c", spin[spinval++ % 4]);
28324737Smmusante 				(void) fflush(stdout);
28334737Smmusante 				last_spin_time = now;
28344737Smmusante 			}
28354737Smmusante 			spincheck = CHECK_SPINNER;
28364737Smmusante 		}
28374737Smmusante 	}
28384737Smmusante 
28391356Seschrock 	/*
28403126Sahl 	 * Interate over any nested datasets.
28411356Seschrock 	 */
28423126Sahl 	if (type == ZFS_TYPE_FILESYSTEM &&
28433126Sahl 	    zfs_iter_filesystems(zhp, get_one_dataset, data) != 0) {
28443265Sahrens 		zfs_close(zhp);
28453126Sahl 		return (1);
28463126Sahl 	}
28473126Sahl 
28483126Sahl 	/*
28493126Sahl 	 * Skip any datasets whose type does not match.
28503126Sahl 	 */
28513126Sahl 	if ((type & cbp->cb_types) == 0) {
28521356Seschrock 		zfs_close(zhp);
28531356Seschrock 		return (0);
28541356Seschrock 	}
28551356Seschrock 
28561356Seschrock 	if (cbp->cb_alloc == cbp->cb_used) {
28571356Seschrock 		zfs_handle_t **handles;
28581356Seschrock 
28591356Seschrock 		if (cbp->cb_alloc == 0)
28601356Seschrock 			cbp->cb_alloc = 64;
28611356Seschrock 		else
28621356Seschrock 			cbp->cb_alloc *= 2;
28631356Seschrock 
28641356Seschrock 		handles = safe_malloc(cbp->cb_alloc * sizeof (void *));
28651356Seschrock 
28661356Seschrock 		if (cbp->cb_handles) {
28671356Seschrock 			bcopy(cbp->cb_handles, handles,
28681356Seschrock 			    cbp->cb_used * sizeof (void *));
28691356Seschrock 			free(cbp->cb_handles);
28701356Seschrock 		}
28711356Seschrock 
28721356Seschrock 		cbp->cb_handles = handles;
28731356Seschrock 	}
28741356Seschrock 
28751356Seschrock 	cbp->cb_handles[cbp->cb_used++] = zhp;
28761356Seschrock 
28773126Sahl 	return (0);
28781356Seschrock }
28791356Seschrock 
28801356Seschrock static void
28814737Smmusante get_all_datasets(uint_t types, zfs_handle_t ***dslist, size_t *count,
28824737Smmusante     boolean_t verbose)
28831356Seschrock {
28841356Seschrock 	get_all_cbdata_t cb = { 0 };
28853126Sahl 	cb.cb_types = types;
28864737Smmusante 	cb.cb_verbose = verbose;
28874737Smmusante 
28884737Smmusante 	if (verbose) {
28894737Smmusante 		(void) printf("%s: *", gettext("Reading ZFS config"));
28904737Smmusante 		(void) fflush(stdout);
28914737Smmusante 	}
28923126Sahl 
28933126Sahl 	(void) zfs_iter_root(g_zfs, get_one_dataset, &cb);
28943126Sahl 
28953126Sahl 	*dslist = cb.cb_handles;
28961356Seschrock 	*count = cb.cb_used;
28974737Smmusante 
28984737Smmusante 	if (verbose) {
28994737Smmusante 		(void) printf("\b%s\n", gettext("done."));
29004737Smmusante 	}
29011356Seschrock }
29021356Seschrock 
29031356Seschrock static int
29043126Sahl dataset_cmp(const void *a, const void *b)
29051356Seschrock {
29061356Seschrock 	zfs_handle_t **za = (zfs_handle_t **)a;
29071356Seschrock 	zfs_handle_t **zb = (zfs_handle_t **)b;
29081356Seschrock 	char mounta[MAXPATHLEN];
29091356Seschrock 	char mountb[MAXPATHLEN];
29103126Sahl 	boolean_t gota, gotb;
29113126Sahl 
29123126Sahl 	if ((gota = (zfs_get_type(*za) == ZFS_TYPE_FILESYSTEM)) != 0)
29133126Sahl 		verify(zfs_prop_get(*za, ZFS_PROP_MOUNTPOINT, mounta,
29143126Sahl 		    sizeof (mounta), NULL, NULL, 0, B_FALSE) == 0);
29153126Sahl 	if ((gotb = (zfs_get_type(*zb) == ZFS_TYPE_FILESYSTEM)) != 0)
29163126Sahl 		verify(zfs_prop_get(*zb, ZFS_PROP_MOUNTPOINT, mountb,
29173126Sahl 		    sizeof (mountb), NULL, NULL, 0, B_FALSE) == 0);
29183126Sahl 
29193126Sahl 	if (gota && gotb)
29203126Sahl 		return (strcmp(mounta, mountb));
29213126Sahl 
29223126Sahl 	if (gota)
29233126Sahl 		return (-1);
29243126Sahl 	if (gotb)
29253126Sahl 		return (1);
29263126Sahl 
29273126Sahl 	return (strcmp(zfs_get_name(a), zfs_get_name(b)));
29281356Seschrock }
2929789Sahrens 
2930789Sahrens /*
2931789Sahrens  * Generic callback for sharing or mounting filesystems.  Because the code is so
2932789Sahrens  * similar, we have a common function with an extra parameter to determine which
2933789Sahrens  * mode we are using.
2934789Sahrens  */
2935789Sahrens #define	OP_SHARE	0x1
2936789Sahrens #define	OP_MOUNT	0x2
2937789Sahrens 
2938789Sahrens /*
29393126Sahl  * Share or mount a dataset.
2940789Sahrens  */
2941789Sahrens static int
29425331Samw share_mount_one(zfs_handle_t *zhp, int op, int flags, char *protocol,
29435331Samw     boolean_t explicit, const char *options)
2944789Sahrens {
2945789Sahrens 	char mountpoint[ZFS_MAXPROPLEN];
2946789Sahrens 	char shareopts[ZFS_MAXPROPLEN];
29475331Samw 	char smbshareopts[ZFS_MAXPROPLEN];
29483126Sahl 	const char *cmdname = op == OP_SHARE ? "share" : "mount";
2949789Sahrens 	struct mnttab mnt;
29502676Seschrock 	uint64_t zoned, canmount;
29513126Sahl 	zfs_type_t type = zfs_get_type(zhp);
29525331Samw 	boolean_t shared_nfs, shared_smb;
29533126Sahl 
29543126Sahl 	assert(type & (ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME));
29553126Sahl 
29563126Sahl 	if (type == ZFS_TYPE_FILESYSTEM) {
29573126Sahl 		/*
29583126Sahl 		 * Check to make sure we can mount/share this dataset.  If we
29593126Sahl 		 * are in the global zone and the filesystem is exported to a
29603126Sahl 		 * local zone, or if we are in a local zone and the
29613126Sahl 		 * filesystem is not exported, then it is an error.
29623126Sahl 		 */
29633126Sahl 		zoned = zfs_prop_get_int(zhp, ZFS_PROP_ZONED);
29643126Sahl 
29653126Sahl 		if (zoned && getzoneid() == GLOBAL_ZONEID) {
29663126Sahl 			if (!explicit)
29673126Sahl 				return (0);
29683126Sahl 
29693126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': "
29703126Sahl 			    "dataset is exported to a local zone\n"), cmdname,
29713126Sahl 			    zfs_get_name(zhp));
29723126Sahl 			return (1);
29733126Sahl 
29743126Sahl 		} else if (!zoned && getzoneid() != GLOBAL_ZONEID) {
29753126Sahl 			if (!explicit)
29763126Sahl 				return (0);
29773126Sahl 
29783126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': "
29793126Sahl 			    "permission denied\n"), cmdname,
29803126Sahl 			    zfs_get_name(zhp));
29813126Sahl 			return (1);
29823126Sahl 		}
29833126Sahl 
29843126Sahl 		/*
29853126Sahl 		 * Ignore any filesystems which don't apply to us. This
29863126Sahl 		 * includes those with a legacy mountpoint, or those with
29873126Sahl 		 * legacy share options.
29883126Sahl 		 */
29893126Sahl 		verify(zfs_prop_get(zhp, ZFS_PROP_MOUNTPOINT, mountpoint,
29903126Sahl 		    sizeof (mountpoint), NULL, NULL, 0, B_FALSE) == 0);
29913126Sahl 		verify(zfs_prop_get(zhp, ZFS_PROP_SHARENFS, shareopts,
29923126Sahl 		    sizeof (shareopts), NULL, NULL, 0, B_FALSE) == 0);
29935331Samw 		verify(zfs_prop_get(zhp, ZFS_PROP_SHARESMB, smbshareopts,
29945331Samw 		    sizeof (smbshareopts), NULL, NULL, 0, B_FALSE) == 0);
29953126Sahl 		canmount = zfs_prop_get_int(zhp, ZFS_PROP_CANMOUNT);
29963126Sahl 
29975331Samw 		if (op == OP_SHARE && strcmp(shareopts, "off") == 0 &&
29985331Samw 		    strcmp(smbshareopts, "off") == 0) {
29993126Sahl 			if (!explicit)
3000789Sahrens 				return (0);
3001789Sahrens 
3002789Sahrens 			(void) fprintf(stderr, gettext("cannot share '%s': "
3003789Sahrens 			    "legacy share\n"), zfs_get_name(zhp));
3004789Sahrens 			(void) fprintf(stderr, gettext("use share(1M) to "
3005789Sahrens 			    "share this filesystem\n"));
3006789Sahrens 			return (1);
3007789Sahrens 		}
30083126Sahl 
30093126Sahl 		/*
30103126Sahl 		 * We cannot share or mount legacy filesystems. If the
30113126Sahl 		 * shareopts is non-legacy but the mountpoint is legacy, we
30123126Sahl 		 * treat it as a legacy share.
30133126Sahl 		 */
30143126Sahl 		if (strcmp(mountpoint, "legacy") == 0) {
30153126Sahl 			if (!explicit)
30163126Sahl 				return (0);
30173126Sahl 
30183126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': "
30193126Sahl 			    "legacy mountpoint\n"), cmdname, zfs_get_name(zhp));
30205367Sahrens 			(void) fprintf(stderr, gettext("use %s(1M) to "
30215367Sahrens 			    "%s this filesystem\n"), cmdname, cmdname);
30223126Sahl 			return (1);
30233126Sahl 		}
30243126Sahl 
30253126Sahl 		if (strcmp(mountpoint, "none") == 0) {
30263126Sahl 			if (!explicit)
30273126Sahl 				return (0);
30283126Sahl 
30293126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': no "
30303126Sahl 			    "mountpoint set\n"), cmdname, zfs_get_name(zhp));
30313126Sahl 			return (1);
30323126Sahl 		}
30333126Sahl 
30346168Shs24103 		/*
30356168Shs24103 		 * canmount	explicit	outcome
30366168Shs24103 		 * on		no		pass through
30376168Shs24103 		 * on		yes		pass through
30386168Shs24103 		 * off		no		return 0
30396168Shs24103 		 * off		yes		display error, return 1
30406168Shs24103 		 * noauto	no		return 0
30416168Shs24103 		 * noauto	yes		pass through
30426168Shs24103 		 */
30436168Shs24103 		if (canmount == ZFS_CANMOUNT_OFF) {
30443126Sahl 			if (!explicit)
30453126Sahl 				return (0);
30463126Sahl 
30473126Sahl 			(void) fprintf(stderr, gettext("cannot %s '%s': "
30483126Sahl 			    "'canmount' property is set to 'off'\n"), cmdname,
30493126Sahl 			    zfs_get_name(zhp));
30503126Sahl 			return (1);
30516168Shs24103 		} else if (canmount == ZFS_CANMOUNT_NOAUTO && !explicit) {
30526168Shs24103 			return (0);
30533126Sahl 		}
30543126Sahl 
30553126Sahl 		/*
30563126Sahl 		 * At this point, we have verified that the mountpoint and/or
30573126Sahl 		 * shareopts are appropriate for auto management. If the
30583126Sahl 		 * filesystem is already mounted or shared, return (failing
30593126Sahl 		 * for explicit requests); otherwise mount or share the
30603126Sahl 		 * filesystem.
30613126Sahl 		 */
30623126Sahl 		switch (op) {
30633126Sahl 		case OP_SHARE:
30645331Samw 
30655331Samw 			shared_nfs = zfs_is_shared_nfs(zhp, NULL);
30665331Samw 			shared_smb = zfs_is_shared_smb(zhp, NULL);
30675331Samw 
30685331Samw 			if (shared_nfs && shared_smb ||
30695331Samw 			    (shared_nfs && strcmp(shareopts, "on") == 0 &&
30705331Samw 			    strcmp(smbshareopts, "off") == 0) ||
30715331Samw 			    (shared_smb && strcmp(smbshareopts, "on") == 0 &&
30725331Samw 			    strcmp(shareopts, "off") == 0)) {
30733126Sahl 				if (!explicit)
30743126Sahl 					return (0);
30753126Sahl 
3076789Sahrens 				(void) fprintf(stderr, gettext("cannot share "
3077789Sahrens 				    "'%s': filesystem already shared\n"),
3078789Sahrens 				    zfs_get_name(zhp));
3079789Sahrens 				return (1);
3080789Sahrens 			}
30813126Sahl 
30823126Sahl 			if (!zfs_is_mounted(zhp, NULL) &&
30833126Sahl 			    zfs_mount(zhp, NULL, 0) != 0)
30843126Sahl 				return (1);
30853126Sahl 
30865331Samw 			if (protocol == NULL) {
30875331Samw 				if (zfs_shareall(zhp) != 0)
30885331Samw 					return (1);
30895331Samw 			} else if (strcmp(protocol, "nfs") == 0) {
30905331Samw 				if (zfs_share_nfs(zhp))
30915331Samw 					return (1);
30925331Samw 			} else if (strcmp(protocol, "smb") == 0) {
30935331Samw 				if (zfs_share_smb(zhp))
30945331Samw 					return (1);
30955331Samw 			} else {
30965331Samw 				(void) fprintf(stderr, gettext("cannot share "
30975331Samw 				    "'%s': invalid share type '%s' "
30985331Samw 				    "specified\n"),
30995331Samw 				    zfs_get_name(zhp), protocol);
31003126Sahl 				return (1);
31015331Samw 			}
31025331Samw 
31033126Sahl 			break;
31043126Sahl 
31053126Sahl 		case OP_MOUNT:
31063126Sahl 			if (options == NULL)
31073126Sahl 				mnt.mnt_mntopts = "";
31083126Sahl 			else
31093126Sahl 				mnt.mnt_mntopts = (char *)options;
31103126Sahl 
31113126Sahl 			if (!hasmntopt(&mnt, MNTOPT_REMOUNT) &&
31123126Sahl 			    zfs_is_mounted(zhp, NULL)) {
31133126Sahl 				if (!explicit)
31143126Sahl 					return (0);
31153126Sahl 
3116789Sahrens 				(void) fprintf(stderr, gettext("cannot mount "
3117789Sahrens 				    "'%s': filesystem already mounted\n"),
3118789Sahrens 				    zfs_get_name(zhp));
3119789Sahrens 				return (1);
3120789Sahrens 			}
31213126Sahl 
31223126Sahl 			if (zfs_mount(zhp, options, flags) != 0)
31233126Sahl 				return (1);
31243126Sahl 			break;
3125789Sahrens 		}
31263126Sahl 	} else {
31273126Sahl 		assert(op == OP_SHARE);
31283126Sahl 
31293126Sahl 		/*
31303126Sahl 		 * Ignore any volumes that aren't shared.
31313126Sahl 		 */
31323126Sahl 		verify(zfs_prop_get(zhp, ZFS_PROP_SHAREISCSI, shareopts,
31333126Sahl 		    sizeof (shareopts), NULL, NULL, 0, B_FALSE) == 0);
31343126Sahl 
31353126Sahl 		if (strcmp(shareopts, "off") == 0) {
31363126Sahl 			if (!explicit)
31373126Sahl 				return (0);
31383126Sahl 
31393126Sahl 			(void) fprintf(stderr, gettext("cannot share '%s': "
31403126Sahl 			    "'shareiscsi' property not set\n"),
31413126Sahl 			    zfs_get_name(zhp));
31423126Sahl 			(void) fprintf(stderr, gettext("set 'shareiscsi' "
31433126Sahl 			    "property or use iscsitadm(1M) to share this "
31443126Sahl 			    "volume\n"));
31453126Sahl 			return (1);
3146789Sahrens 		}
31473126Sahl 
31483126Sahl 		if (zfs_is_shared_iscsi(zhp)) {
31493126Sahl 			if (!explicit)
31503126Sahl 				return (0);
31513126Sahl 
31523126Sahl 			(void) fprintf(stderr, gettext("cannot share "
31533126Sahl 			    "'%s': volume already shared\n"),
31543126Sahl 			    zfs_get_name(zhp));
3155789Sahrens 			return (1);
31563126Sahl 		}
31573126Sahl 
31583126Sahl 		if (zfs_share_iscsi(zhp) != 0)
31593126Sahl 			return (1);
3160789Sahrens 	}
3161789Sahrens 
3162789Sahrens 	return (0);
3163789Sahrens }
3164789Sahrens 
31654737Smmusante /*
31664737Smmusante  * Reports progress in the form "(current/total)".  Not thread-safe.
31674737Smmusante  */
31684737Smmusante static void
31694737Smmusante report_mount_progress(int current, int total)
31704737Smmusante {
31714737Smmusante 	static int len;
31724737Smmusante 	static char *reverse = "\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b"
31734737Smmusante 	    "\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b\b";
31744737Smmusante 	static time_t last_progress_time;
31754737Smmusante 	time_t now = time(NULL);
31764737Smmusante 
31774737Smmusante 	/* report 1..n instead of 0..n-1 */
31784737Smmusante 	++current;
31794737Smmusante 
31804737Smmusante 	/* display header if we're here for the first time */
31814737Smmusante 	if (current == 1) {
31824737Smmusante 		(void) printf(gettext("Mounting ZFS filesystems: "));
31834737Smmusante 		len = 0;
31845367Sahrens 	} else if (current != total && last_progress_time + MOUNT_TIME >= now) {
31855367Sahrens 		/* too soon to report again */
31865367Sahrens 		return;
31875367Sahrens 	}
31884737Smmusante 
31894737Smmusante 	last_progress_time = now;
31904737Smmusante 
31914737Smmusante 	/* back up to prepare for overwriting */
31924737Smmusante 	if (len)
31934737Smmusante 		(void) printf("%*.*s", len, len, reverse);
31944737Smmusante 
31954737Smmusante 	/* We put a newline at the end if this is the last one.  */
31964737Smmusante 	len = printf("(%d/%d)%s", current, total, current == total ? "\n" : "");
31974737Smmusante 	(void) fflush(stdout);
31984737Smmusante }
31994737Smmusante 
32006289Smmusante static void
32016289Smmusante append_options(char *mntopts, char *newopts)
32026289Smmusante {
32036289Smmusante 	int len = strlen(mntopts);
32046289Smmusante 
32056289Smmusante 	/* original length plus new string to append plus 1 for the comma */
32066289Smmusante 	if (len + 1 + strlen(newopts) >= MNT_LINE_MAX) {
32076289Smmusante 		(void) fprintf(stderr, gettext("the opts argument for "
32086289Smmusante 		    "'%c' option is too long (more than %d chars)\n"),
32096289Smmusante 		    "-o", MNT_LINE_MAX);
32106289Smmusante 		usage(B_FALSE);
32116289Smmusante 	}
32126289Smmusante 
32136289Smmusante 	if (*mntopts)
32146289Smmusante 		mntopts[len++] = ',';
32156289Smmusante 
32166289Smmusante 	(void) strcpy(&mntopts[len], newopts);
32176289Smmusante }
32186289Smmusante 
3219789Sahrens static int
32203126Sahl share_mount(int op, int argc, char **argv)
3221789Sahrens {
3222789Sahrens 	int do_all = 0;
32234737Smmusante 	boolean_t verbose = B_FALSE;
32242372Slling 	int c, ret = 0;
32256289Smmusante 	char *options = NULL;
32263126Sahl 	int types, flags = 0;
3227789Sahrens 
3228789Sahrens 	/* check options */
32294737Smmusante 	while ((c = getopt(argc, argv, op == OP_MOUNT ? ":avo:O" : "a"))
3230789Sahrens 	    != -1) {
3231789Sahrens 		switch (c) {
3232789Sahrens 		case 'a':
3233789Sahrens 			do_all = 1;
3234789Sahrens 			break;
32354737Smmusante 		case 'v':
32364737Smmusante 			verbose = B_TRUE;
32374737Smmusante 			break;
3238789Sahrens 		case 'o':
32396289Smmusante 			if (*optarg == '\0') {
32406289Smmusante 				(void) fprintf(stderr, gettext("empty mount "
32416289Smmusante 				    "options (-o) specified\n"));
32426289Smmusante 				usage(B_FALSE);
32434717Srm160521 			}
32446289Smmusante 
32456289Smmusante 			if (options == NULL)
32466289Smmusante 				options = safe_malloc(MNT_LINE_MAX + 1);
32476289Smmusante 
32486289Smmusante 			/* option validation is done later */
32496289Smmusante 			append_options(options, optarg);
3250789Sahrens 			break;
32514717Srm160521 
3252789Sahrens 		case 'O':
32533126Sahl 			flags |= MS_OVERLAY;
3254789Sahrens 			break;
3255789Sahrens 		case ':':
3256789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
3257789Sahrens 			    "'%c' option\n"), optopt);
32582082Seschrock 			usage(B_FALSE);
3259789Sahrens 			break;
3260789Sahrens 		case '?':
3261789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
3262789Sahrens 			    optopt);
32632082Seschrock 			usage(B_FALSE);
3264789Sahrens 		}
3265789Sahrens 	}
3266789Sahrens 
3267789Sahrens 	argc -= optind;
3268789Sahrens 	argv += optind;
3269789Sahrens 
3270789Sahrens 	/* check number of arguments */
3271789Sahrens 	if (do_all) {
32723126Sahl 		zfs_handle_t **dslist = NULL;
32731356Seschrock 		size_t i, count = 0;
32745331Samw 		char *protocol = NULL;
32751356Seschrock 
32763126Sahl 		if (op == OP_MOUNT) {
32773126Sahl 			types = ZFS_TYPE_FILESYSTEM;
32783126Sahl 		} else if (argc > 0) {
32795331Samw 			if (strcmp(argv[0], "nfs") == 0 ||
32805331Samw 			    strcmp(argv[0], "smb") == 0) {
32813126Sahl 				types = ZFS_TYPE_FILESYSTEM;
32823126Sahl 			} else if (strcmp(argv[0], "iscsi") == 0) {
32833126Sahl 				types = ZFS_TYPE_VOLUME;
32843126Sahl 			} else {
32853126Sahl 				(void) fprintf(stderr, gettext("share type "
32865331Samw 				    "must be 'nfs', 'smb' or 'iscsi'\n"));
32873126Sahl 				usage(B_FALSE);
32883126Sahl 			}
32895331Samw 			protocol = argv[0];
32903126Sahl 			argc--;
32913126Sahl 			argv++;
32923126Sahl 		} else {
32933126Sahl 			types = ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME;
32943126Sahl 		}
32953126Sahl 
3296789Sahrens 		if (argc != 0) {
3297789Sahrens 			(void) fprintf(stderr, gettext("too many arguments\n"));
32982082Seschrock 			usage(B_FALSE);
3299789Sahrens 		}
3300789Sahrens 
33014737Smmusante 		get_all_datasets(types, &dslist, &count, verbose);
33021356Seschrock 
33031356Seschrock 		if (count == 0)
33041356Seschrock 			return (0);
33051356Seschrock 
33063126Sahl 		qsort(dslist, count, sizeof (void *), dataset_cmp);
33071356Seschrock 
33081356Seschrock 		for (i = 0; i < count; i++) {
33094737Smmusante 			if (verbose)
33104737Smmusante 				report_mount_progress(i, count);
33114737Smmusante 
33125331Samw 			if (share_mount_one(dslist[i], op, flags, protocol,
33135331Samw 			    B_FALSE, options) != 0)
33142369Slling 				ret = 1;
33153126Sahl 			zfs_close(dslist[i]);
33161356Seschrock 		}
33171356Seschrock 
33183126Sahl 		free(dslist);
3319789Sahrens 	} else if (argc == 0) {
3320789Sahrens 		struct mnttab entry;
3321789Sahrens 
33224717Srm160521 		if ((op == OP_SHARE) || (options != NULL)) {
3323789Sahrens 			(void) fprintf(stderr, gettext("missing filesystem "
33244717Srm160521 			    "argument (specify -a for all)\n"));
33252082Seschrock 			usage(B_FALSE);
3326789Sahrens 		}
3327789Sahrens 
3328789Sahrens 		/*
3329789Sahrens 		 * When mount is given no arguments, go through /etc/mnttab and
3330789Sahrens 		 * display any active ZFS mounts.  We hide any snapshots, since
3331789Sahrens 		 * they are controlled automatically.
3332789Sahrens 		 */
3333789Sahrens 		rewind(mnttab_file);
3334789Sahrens 		while (getmntent(mnttab_file, &entry) == 0) {
3335789Sahrens 			if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0 ||
3336789Sahrens 			    strchr(entry.mnt_special, '@') != NULL)
3337789Sahrens 				continue;
3338789Sahrens 
3339789Sahrens 			(void) printf("%-30s  %s\n", entry.mnt_special,
3340789Sahrens 			    entry.mnt_mountp);
3341789Sahrens 		}
3342789Sahrens 
3343789Sahrens 	} else {
3344789Sahrens 		zfs_handle_t *zhp;
3345789Sahrens 
33463126Sahl 		types = ZFS_TYPE_FILESYSTEM;
33473126Sahl 		if (op == OP_SHARE)
33483126Sahl 			types |= ZFS_TYPE_VOLUME;
33493126Sahl 
3350789Sahrens 		if (argc > 1) {
3351789Sahrens 			(void) fprintf(stderr,
3352789Sahrens 			    gettext("too many arguments\n"));
33532082Seschrock 			usage(B_FALSE);
3354789Sahrens 		}
3355789Sahrens 
33563126Sahl 		if ((zhp = zfs_open(g_zfs, argv[0], types)) == NULL) {
3357789Sahrens 			ret = 1;
33583126Sahl 		} else {
33595331Samw 			ret = share_mount_one(zhp, op, flags, NULL, B_TRUE,
33603126Sahl 			    options);
3361789Sahrens 			zfs_close(zhp);
3362789Sahrens 		}
3363789Sahrens 	}
3364789Sahrens 
3365789Sahrens 	return (ret);
3366789Sahrens }
3367789Sahrens 
3368789Sahrens /*
33693126Sahl  * zfs mount -a [nfs | iscsi]
3370789Sahrens  * zfs mount filesystem
3371789Sahrens  *
3372789Sahrens  * Mount all filesystems, or mount the given filesystem.
3373789Sahrens  */
3374789Sahrens static int
3375789Sahrens zfs_do_mount(int argc, char **argv)
3376789Sahrens {
33773126Sahl 	return (share_mount(OP_MOUNT, argc, argv));
3378789Sahrens }
3379789Sahrens 
3380789Sahrens /*
33815331Samw  * zfs share -a [nfs | iscsi | smb]
3382789Sahrens  * zfs share filesystem
3383789Sahrens  *
3384789Sahrens  * Share all filesystems, or share the given filesystem.
3385789Sahrens  */
3386789Sahrens static int
3387789Sahrens zfs_do_share(int argc, char **argv)
3388789Sahrens {
33893126Sahl 	return (share_mount(OP_SHARE, argc, argv));
3390789Sahrens }
3391789Sahrens 
3392789Sahrens typedef struct unshare_unmount_node {
3393789Sahrens 	zfs_handle_t	*un_zhp;
3394789Sahrens 	char		*un_mountp;
3395789Sahrens 	uu_avl_node_t	un_avlnode;
3396789Sahrens } unshare_unmount_node_t;
3397789Sahrens 
3398789Sahrens /* ARGSUSED */
3399789Sahrens static int
3400789Sahrens unshare_unmount_compare(const void *larg, const void *rarg, void *unused)
3401789Sahrens {
3402789Sahrens 	const unshare_unmount_node_t *l = larg;
3403789Sahrens 	const unshare_unmount_node_t *r = rarg;
3404789Sahrens 
3405789Sahrens 	return (strcmp(l->un_mountp, r->un_mountp));
3406789Sahrens }
3407789Sahrens 
3408789Sahrens /*
3409789Sahrens  * Convenience routine used by zfs_do_umount() and manual_unmount().  Given an
3410789Sahrens  * absolute path, find the entry /etc/mnttab, verify that its a ZFS filesystem,
3411789Sahrens  * and unmount it appropriately.
3412789Sahrens  */
3413789Sahrens static int
34143126Sahl unshare_unmount_path(int op, char *path, int flags, boolean_t is_manual)
3415789Sahrens {
3416789Sahrens 	zfs_handle_t *zhp;
3417789Sahrens 	int ret;
3418789Sahrens 	struct stat64 statbuf;
3419789Sahrens 	struct extmnttab entry;
34203126Sahl 	const char *cmdname = (op == OP_SHARE) ? "unshare" : "unmount";
34216289Smmusante 	ino_t path_inode;
3422789Sahrens 
3423789Sahrens 	/*
3424789Sahrens 	 * Search for the path in /etc/mnttab.  Rather than looking for the
3425789Sahrens 	 * specific path, which can be fooled by non-standard paths (i.e. ".."
3426789Sahrens 	 * or "//"), we stat() the path and search for the corresponding
3427789Sahrens 	 * (major,minor) device pair.
3428789Sahrens 	 */
3429789Sahrens 	if (stat64(path, &statbuf) != 0) {
3430789Sahrens 		(void) fprintf(stderr, gettext("cannot %s '%s': %s\n"),
3431789Sahrens 		    cmdname, path, strerror(errno));
3432789Sahrens 		return (1);
3433789Sahrens 	}
34346289Smmusante 	path_inode = statbuf.st_ino;
3435789Sahrens 
3436789Sahrens 	/*
3437789Sahrens 	 * Search for the given (major,minor) pair in the mount table.
3438789Sahrens 	 */
3439789Sahrens 	rewind(mnttab_file);
3440789Sahrens 	while ((ret = getextmntent(mnttab_file, &entry, 0)) == 0) {
3441789Sahrens 		if (entry.mnt_major == major(statbuf.st_dev) &&
3442789Sahrens 		    entry.mnt_minor == minor(statbuf.st_dev))
3443789Sahrens 			break;
3444789Sahrens 	}
3445789Sahrens 	if (ret != 0) {
34465904Stimh 		if (op == OP_SHARE) {
34475904Stimh 			(void) fprintf(stderr, gettext("cannot %s '%s': not "
34485904Stimh 			    "currently mounted\n"), cmdname, path);
34495904Stimh 			return (1);
34505904Stimh 		}
34515904Stimh 		(void) fprintf(stderr, gettext("warning: %s not in mnttab\n"),
34525904Stimh 		    path);
34535904Stimh 		if ((ret = umount2(path, flags)) != 0)
34545904Stimh 			(void) fprintf(stderr, gettext("%s: %s\n"), path,
34555904Stimh 			    strerror(errno));
34565904Stimh 		return (ret != 0);
3457789Sahrens 	}
3458789Sahrens 
3459789Sahrens 	if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0) {
3460789Sahrens 		(void) fprintf(stderr, gettext("cannot %s '%s': not a ZFS "
3461789Sahrens 		    "filesystem\n"), cmdname, path);
3462789Sahrens 		return (1);
3463789Sahrens 	}
3464789Sahrens 
34652082Seschrock 	if ((zhp = zfs_open(g_zfs, entry.mnt_special,
34662082Seschrock 	    ZFS_TYPE_FILESYSTEM)) == NULL)
3467789Sahrens 		return (1);
3468789Sahrens 
34696289Smmusante 
34706289Smmusante 	ret = 1;
34713126Sahl 	if (op == OP_SHARE) {
34726289Smmusante 		char nfs_mnt_prop[ZFS_MAXPROPLEN];
34736289Smmusante 		char smbshare_prop[ZFS_MAXPROPLEN];
34746289Smmusante 
34756289Smmusante 		verify(zfs_prop_get(zhp, ZFS_PROP_SHARENFS, nfs_mnt_prop,
34766289Smmusante 		    sizeof (nfs_mnt_prop), NULL, NULL, 0, B_FALSE) == 0);
34776289Smmusante 		verify(zfs_prop_get(zhp, ZFS_PROP_SHARESMB, smbshare_prop,
34786289Smmusante 		    sizeof (smbshare_prop), NULL, NULL, 0, B_FALSE) == 0);
34796289Smmusante 
34805331Samw 		if (strcmp(nfs_mnt_prop, "off") == 0 &&
34815331Samw 		    strcmp(smbshare_prop, "off") == 0) {
3482789Sahrens 			(void) fprintf(stderr, gettext("cannot unshare "
3483789Sahrens 			    "'%s': legacy share\n"), path);
3484789Sahrens 			(void) fprintf(stderr, gettext("use "
3485789Sahrens 			    "unshare(1M) to unshare this filesystem\n"));
34865331Samw 		} else if (!zfs_is_shared(zhp)) {
3487789Sahrens 			(void) fprintf(stderr, gettext("cannot unshare '%s': "
3488789Sahrens 			    "not currently shared\n"), path);
3489789Sahrens 		} else {
34905331Samw 			ret = zfs_unshareall_bypath(zhp, path);
3491789Sahrens 		}
3492789Sahrens 	} else {
34936289Smmusante 		char mtpt_prop[ZFS_MAXPROPLEN];
34946289Smmusante 
34956289Smmusante 		verify(zfs_prop_get(zhp, ZFS_PROP_MOUNTPOINT, mtpt_prop,
34966289Smmusante 		    sizeof (mtpt_prop), NULL, NULL, 0, B_FALSE) == 0);
34976289Smmusante 
34986289Smmusante 		if (stat64(entry.mnt_mountp, &statbuf) != 0) {
34996289Smmusante 			(void) fprintf(stderr, gettext("cannot %s '%s': %s\n"),
35006289Smmusante 			    cmdname, path, strerror(errno));
35016289Smmusante 		} else if (statbuf.st_ino != path_inode) {
35026289Smmusante 			(void) fprintf(stderr, gettext("cannot "
35036289Smmusante 			    "unmount '%s': not a mountpoint\n"), path);
35046289Smmusante 		} else if (is_manual) {
35051264Slling 			ret = zfs_unmount(zhp, NULL, flags);
35066289Smmusante 		} else if (strcmp(mtpt_prop, "legacy") == 0) {
35071264Slling 			(void) fprintf(stderr, gettext("cannot unmount "
35081264Slling 			    "'%s': legacy mountpoint\n"),
35091264Slling 			    zfs_get_name(zhp));
35101264Slling 			(void) fprintf(stderr, gettext("use umount(1M) "
35111264Slling 			    "to unmount this filesystem\n"));
3512789Sahrens 		} else {
3513789Sahrens 			ret = zfs_unmountall(zhp, flags);
3514789Sahrens 		}
3515789Sahrens 	}
3516789Sahrens 
3517789Sahrens 	zfs_close(zhp);
3518789Sahrens 
3519789Sahrens 	return (ret != 0);
3520789Sahrens }
3521789Sahrens 
3522789Sahrens /*
3523789Sahrens  * Generic callback for unsharing or unmounting a filesystem.
3524789Sahrens  */
3525789Sahrens static int
35263126Sahl unshare_unmount(int op, int argc, char **argv)
3527789Sahrens {
3528789Sahrens 	int do_all = 0;
3529789Sahrens 	int flags = 0;
3530789Sahrens 	int ret = 0;
35313126Sahl 	int types, c;
3532789Sahrens 	zfs_handle_t *zhp;
35335331Samw 	char nfsiscsi_mnt_prop[ZFS_MAXPROPLEN];
35345331Samw 	char sharesmb[ZFS_MAXPROPLEN];
3535789Sahrens 
3536789Sahrens 	/* check options */
35373126Sahl 	while ((c = getopt(argc, argv, op == OP_SHARE ? "a" : "af")) != -1) {
3538789Sahrens 		switch (c) {
3539789Sahrens 		case 'a':
3540789Sahrens 			do_all = 1;
3541789Sahrens 			break;
3542789Sahrens 		case 'f':
3543789Sahrens 			flags = MS_FORCE;
3544789Sahrens 			break;
3545789Sahrens 		case '?':
3546789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
3547789Sahrens 			    optopt);
35482082Seschrock 			usage(B_FALSE);
3549789Sahrens 		}
3550789Sahrens 	}
3551789Sahrens 
3552789Sahrens 	argc -= optind;
3553789Sahrens 	argv += optind;
3554789Sahrens 
3555789Sahrens 	if (do_all) {
3556789Sahrens 		/*
3557789Sahrens 		 * We could make use of zfs_for_each() to walk all datasets in
3558789Sahrens 		 * the system, but this would be very inefficient, especially
3559789Sahrens 		 * since we would have to linearly search /etc/mnttab for each
3560789Sahrens 		 * one.  Instead, do one pass through /etc/mnttab looking for
3561789Sahrens 		 * zfs entries and call zfs_unmount() for each one.
3562789Sahrens 		 *
3563789Sahrens 		 * Things get a little tricky if the administrator has created
3564789Sahrens 		 * mountpoints beneath other ZFS filesystems.  In this case, we
3565789Sahrens 		 * have to unmount the deepest filesystems first.  To accomplish
3566789Sahrens 		 * this, we place all the mountpoints in an AVL tree sorted by
3567789Sahrens 		 * the special type (dataset name), and walk the result in
3568789Sahrens 		 * reverse to make sure to get any snapshots first.
3569789Sahrens 		 */
3570789Sahrens 		struct mnttab entry;
3571789Sahrens 		uu_avl_pool_t *pool;
3572789Sahrens 		uu_avl_t *tree;
3573789Sahrens 		unshare_unmount_node_t *node;
3574789Sahrens 		uu_avl_index_t idx;
3575789Sahrens 		uu_avl_walk_t *walk;
3576789Sahrens 
35773126Sahl 		if (argc != 0) {
35783126Sahl 			(void) fprintf(stderr, gettext("too many arguments\n"));
35793126Sahl 			usage(B_FALSE);
35803126Sahl 		}
35813126Sahl 
3582789Sahrens 		if ((pool = uu_avl_pool_create("unmount_pool",
3583789Sahrens 		    sizeof (unshare_unmount_node_t),
3584789Sahrens 		    offsetof(unshare_unmount_node_t, un_avlnode),
3585789Sahrens 		    unshare_unmount_compare,
3586789Sahrens 		    UU_DEFAULT)) == NULL) {
3587789Sahrens 			(void) fprintf(stderr, gettext("internal error: "
3588789Sahrens 			    "out of memory\n"));
3589789Sahrens 			exit(1);
3590789Sahrens 		}
3591789Sahrens 
3592789Sahrens 		if ((tree = uu_avl_create(pool, NULL, UU_DEFAULT)) == NULL) {
3593789Sahrens 			(void) fprintf(stderr, gettext("internal error: "
3594789Sahrens 			    "out of memory\n"));
3595789Sahrens 			exit(1);
3596789Sahrens 		}
3597789Sahrens 
3598789Sahrens 		rewind(mnttab_file);
3599789Sahrens 		while (getmntent(mnttab_file, &entry) == 0) {
3600789Sahrens 
3601789Sahrens 			/* ignore non-ZFS entries */
3602789Sahrens 			if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0)
3603789Sahrens 				continue;
3604789Sahrens 
3605789Sahrens 			/* ignore snapshots */
3606789Sahrens 			if (strchr(entry.mnt_special, '@') != NULL)
3607789Sahrens 				continue;
3608789Sahrens 
36092082Seschrock 			if ((zhp = zfs_open(g_zfs, entry.mnt_special,
3610789Sahrens 			    ZFS_TYPE_FILESYSTEM)) == NULL) {
3611789Sahrens 				ret = 1;
3612789Sahrens 				continue;
3613789Sahrens 			}
3614789Sahrens 
36155331Samw 			switch (op) {
36165331Samw 			case OP_SHARE:
36175331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_SHARENFS,
36185331Samw 				    nfsiscsi_mnt_prop,
36195331Samw 				    sizeof (nfsiscsi_mnt_prop),
36205331Samw 				    NULL, NULL, 0, B_FALSE) == 0);
36215331Samw 				if (strcmp(nfsiscsi_mnt_prop, "off") != 0)
36225331Samw 					break;
36235331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_SHARESMB,
36245331Samw 				    nfsiscsi_mnt_prop,
36255331Samw 				    sizeof (nfsiscsi_mnt_prop),
36265331Samw 				    NULL, NULL, 0, B_FALSE) == 0);
36275331Samw 				if (strcmp(nfsiscsi_mnt_prop, "off") == 0)
36285331Samw 					continue;
36295331Samw 				break;
36305331Samw 			case OP_MOUNT:
36315331Samw 				/* Ignore legacy mounts */
36325331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_MOUNTPOINT,
36335331Samw 				    nfsiscsi_mnt_prop,
36345331Samw 				    sizeof (nfsiscsi_mnt_prop),
36355331Samw 				    NULL, NULL, 0, B_FALSE) == 0);
36365331Samw 				if (strcmp(nfsiscsi_mnt_prop, "legacy") == 0)
36375331Samw 					continue;
36386168Shs24103 				/* Ignore canmount=noauto mounts */
36396168Shs24103 				if (zfs_prop_get_int(zhp, ZFS_PROP_CANMOUNT) ==
36406168Shs24103 				    ZFS_CANMOUNT_NOAUTO)
36416168Shs24103 					continue;
36425331Samw 			default:
36435331Samw 				break;
3644789Sahrens 			}
3645789Sahrens 
3646789Sahrens 			node = safe_malloc(sizeof (unshare_unmount_node_t));
3647789Sahrens 			node->un_zhp = zhp;
3648789Sahrens 
3649789Sahrens 			if ((node->un_mountp = strdup(entry.mnt_mountp)) ==
3650789Sahrens 			    NULL) {
3651789Sahrens 				(void) fprintf(stderr, gettext("internal error:"
3652789Sahrens 				    " out of memory\n"));
3653789Sahrens 				exit(1);
3654789Sahrens 			}
3655789Sahrens 
3656789Sahrens 			uu_avl_node_init(node, &node->un_avlnode, pool);
3657789Sahrens 
3658789Sahrens 			if (uu_avl_find(tree, node, NULL, &idx) == NULL) {
3659789Sahrens 				uu_avl_insert(tree, node, idx);
3660789Sahrens 			} else {
3661789Sahrens 				zfs_close(node->un_zhp);
3662789Sahrens 				free(node->un_mountp);
3663789Sahrens 				free(node);
3664789Sahrens 			}
3665789Sahrens 		}
3666789Sahrens 
3667789Sahrens 		/*
3668789Sahrens 		 * Walk the AVL tree in reverse, unmounting each filesystem and
3669789Sahrens 		 * removing it from the AVL tree in the process.
3670789Sahrens 		 */
3671789Sahrens 		if ((walk = uu_avl_walk_start(tree,
3672789Sahrens 		    UU_WALK_REVERSE | UU_WALK_ROBUST)) == NULL) {
3673789Sahrens 			(void) fprintf(stderr,
3674789Sahrens 			    gettext("internal error: out of memory"));
3675789Sahrens 			exit(1);
3676789Sahrens 		}
3677789Sahrens 
3678789Sahrens 		while ((node = uu_avl_walk_next(walk)) != NULL) {
3679789Sahrens 			uu_avl_remove(tree, node);
3680789Sahrens 
36813126Sahl 			switch (op) {
3682789Sahrens 			case OP_SHARE:
36835331Samw 				if (zfs_unshareall_bypath(node->un_zhp,
3684789Sahrens 				    node->un_mountp) != 0)
3685789Sahrens 					ret = 1;
3686789Sahrens 				break;
3687789Sahrens 
3688789Sahrens 			case OP_MOUNT:
3689789Sahrens 				if (zfs_unmount(node->un_zhp,
3690789Sahrens 				    node->un_mountp, flags) != 0)
3691789Sahrens 					ret = 1;
3692789Sahrens 				break;
3693789Sahrens 			}
3694789Sahrens 
3695789Sahrens 			zfs_close(node->un_zhp);
3696789Sahrens 			free(node->un_mountp);
3697789Sahrens 			free(node);
3698789Sahrens 		}
3699789Sahrens 
3700789Sahrens 		uu_avl_walk_end(walk);
3701789Sahrens 		uu_avl_destroy(tree);
3702789Sahrens 		uu_avl_pool_destroy(pool);
37033126Sahl 
37043126Sahl 		if (op == OP_SHARE) {
37053126Sahl 			/*
37063126Sahl 			 * Finally, unshare any volumes shared via iSCSI.
37073126Sahl 			 */
37083126Sahl 			zfs_handle_t **dslist = NULL;
37093126Sahl 			size_t i, count = 0;
37103126Sahl 
37114737Smmusante 			get_all_datasets(ZFS_TYPE_VOLUME, &dslist, &count,
37124737Smmusante 			    B_FALSE);
37133126Sahl 
37143126Sahl 			if (count != 0) {
37153126Sahl 				qsort(dslist, count, sizeof (void *),
37163126Sahl 				    dataset_cmp);
37173126Sahl 
37183126Sahl 				for (i = 0; i < count; i++) {
37193126Sahl 					if (zfs_unshare_iscsi(dslist[i]) != 0)
37203126Sahl 						ret = 1;
37213126Sahl 					zfs_close(dslist[i]);
37223126Sahl 				}
37233126Sahl 
37243126Sahl 				free(dslist);
37253126Sahl 			}
37263126Sahl 		}
3727789Sahrens 	} else {
37283126Sahl 		if (argc != 1) {
37293126Sahl 			if (argc == 0)
37303126Sahl 				(void) fprintf(stderr,
37313126Sahl 				    gettext("missing filesystem argument\n"));
37323126Sahl 			else
37333126Sahl 				(void) fprintf(stderr,
37343126Sahl 				    gettext("too many arguments\n"));
37353126Sahl 			usage(B_FALSE);
37363126Sahl 		}
37373126Sahl 
3738789Sahrens 		/*
3739789Sahrens 		 * We have an argument, but it may be a full path or a ZFS
3740789Sahrens 		 * filesystem.  Pass full paths off to unmount_path() (shared by
3741789Sahrens 		 * manual_unmount), otherwise open the filesystem and pass to
3742789Sahrens 		 * zfs_unmount().
3743789Sahrens 		 */
3744789Sahrens 		if (argv[0][0] == '/')
37453126Sahl 			return (unshare_unmount_path(op, argv[0],
37463912Slling 			    flags, B_FALSE));
37472082Seschrock 
37483126Sahl 		types = ZFS_TYPE_FILESYSTEM;
37493126Sahl 		if (op == OP_SHARE)
37503126Sahl 			types |= ZFS_TYPE_VOLUME;
37513126Sahl 
37523126Sahl 		if ((zhp = zfs_open(g_zfs, argv[0], types)) == NULL)
3753789Sahrens 			return (1);
3754789Sahrens 
37553126Sahl 		if (zfs_get_type(zhp) == ZFS_TYPE_FILESYSTEM) {
37563126Sahl 			verify(zfs_prop_get(zhp, op == OP_SHARE ?
37575331Samw 			    ZFS_PROP_SHARENFS : ZFS_PROP_MOUNTPOINT,
37585331Samw 			    nfsiscsi_mnt_prop, sizeof (nfsiscsi_mnt_prop), NULL,
37595331Samw 			    NULL, 0, B_FALSE) == 0);
37603126Sahl 
37613126Sahl 			switch (op) {
37623126Sahl 			case OP_SHARE:
37635331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_SHARENFS,
37645331Samw 				    nfsiscsi_mnt_prop,
37655331Samw 				    sizeof (nfsiscsi_mnt_prop),
37665331Samw 				    NULL, NULL, 0, B_FALSE) == 0);
37675331Samw 				verify(zfs_prop_get(zhp, ZFS_PROP_SHARESMB,
37685331Samw 				    sharesmb, sizeof (sharesmb), NULL, NULL,
37695331Samw 				    0, B_FALSE) == 0);
37705331Samw 
37715331Samw 				if (strcmp(nfsiscsi_mnt_prop, "off") == 0 &&
37725331Samw 				    strcmp(sharesmb, "off") == 0) {
37733126Sahl 					(void) fprintf(stderr, gettext("cannot "
37743126Sahl 					    "unshare '%s': legacy share\n"),
37753126Sahl 					    zfs_get_name(zhp));
37763126Sahl 					(void) fprintf(stderr, gettext("use "
37773126Sahl 					    "unshare(1M) to unshare this "
37783126Sahl 					    "filesystem\n"));
37793126Sahl 					ret = 1;
37805331Samw 				} else if (!zfs_is_shared(zhp)) {
37813126Sahl 					(void) fprintf(stderr, gettext("cannot "
37823126Sahl 					    "unshare '%s': not currently "
37833126Sahl 					    "shared\n"), zfs_get_name(zhp));
37843126Sahl 					ret = 1;
37855331Samw 				} else if (zfs_unshareall(zhp) != 0) {
37863126Sahl 					ret = 1;
37873126Sahl 				}
37883126Sahl 				break;
37893126Sahl 
37903126Sahl 			case OP_MOUNT:
37915331Samw 				if (strcmp(nfsiscsi_mnt_prop, "legacy") == 0) {
37923126Sahl 					(void) fprintf(stderr, gettext("cannot "
37933126Sahl 					    "unmount '%s': legacy "
37943126Sahl 					    "mountpoint\n"), zfs_get_name(zhp));
37953126Sahl 					(void) fprintf(stderr, gettext("use "
37963126Sahl 					    "umount(1M) to unmount this "
37973126Sahl 					    "filesystem\n"));
37983126Sahl 					ret = 1;
37993126Sahl 				} else if (!zfs_is_mounted(zhp, NULL)) {
38003126Sahl 					(void) fprintf(stderr, gettext("cannot "
38013126Sahl 					    "unmount '%s': not currently "
38023126Sahl 					    "mounted\n"),
38033126Sahl 					    zfs_get_name(zhp));
38043126Sahl 					ret = 1;
38053126Sahl 				} else if (zfs_unmountall(zhp, flags) != 0) {
38063126Sahl 					ret = 1;
38073126Sahl 				}
38083126Sahl 				break;
38093126Sahl 			}
38103126Sahl 		} else {
38113126Sahl 			assert(op == OP_SHARE);
38123126Sahl 
38135331Samw 			verify(zfs_prop_get(zhp, ZFS_PROP_SHAREISCSI,
38145331Samw 			    nfsiscsi_mnt_prop, sizeof (nfsiscsi_mnt_prop),
38155331Samw 			    NULL, NULL, 0, B_FALSE) == 0);
38165331Samw 
38175331Samw 			if (strcmp(nfsiscsi_mnt_prop, "off") == 0) {
3818789Sahrens 				(void) fprintf(stderr, gettext("cannot unshare "
38193126Sahl 				    "'%s': 'shareiscsi' property not set\n"),
38203126Sahl 				    zfs_get_name(zhp));
38213126Sahl 				(void) fprintf(stderr, gettext("set "
38223126Sahl 				    "'shareiscsi' property or use "
38233126Sahl 				    "iscsitadm(1M) to share this volume\n"));
3824789Sahrens 				ret = 1;
38253126Sahl 			} else if (!zfs_is_shared_iscsi(zhp)) {
38263126Sahl 				(void) fprintf(stderr, gettext("cannot "
38273126Sahl 				    "unshare '%s': not currently shared\n"),
3828789Sahrens 				    zfs_get_name(zhp));
3829789Sahrens 				ret = 1;
38303126Sahl 			} else if (zfs_unshare_iscsi(zhp) != 0) {
3831789Sahrens 				ret = 1;
3832789Sahrens 			}
3833789Sahrens 		}
3834789Sahrens 
3835789Sahrens 		zfs_close(zhp);
3836789Sahrens 	}
3837789Sahrens 
3838789Sahrens 	return (ret);
3839789Sahrens }
3840789Sahrens 
3841789Sahrens /*
3842789Sahrens  * zfs unmount -a
3843789Sahrens  * zfs unmount filesystem
3844789Sahrens  *
3845789Sahrens  * Unmount all filesystems, or a specific ZFS filesystem.
3846789Sahrens  */
3847789Sahrens static int
3848789Sahrens zfs_do_unmount(int argc, char **argv)
3849789Sahrens {
3850789Sahrens 	return (unshare_unmount(OP_MOUNT, argc, argv));
3851789Sahrens }
3852789Sahrens 
3853789Sahrens /*
3854789Sahrens  * zfs unshare -a
3855789Sahrens  * zfs unshare filesystem
3856789Sahrens  *
3857789Sahrens  * Unshare all filesystems, or a specific ZFS filesystem.
3858789Sahrens  */
3859789Sahrens static int
3860789Sahrens zfs_do_unshare(int argc, char **argv)
3861789Sahrens {
3862789Sahrens 	return (unshare_unmount(OP_SHARE, argc, argv));
3863789Sahrens }
3864789Sahrens 
3865789Sahrens /*
3866789Sahrens  * Called when invoked as /etc/fs/zfs/mount.  Do the mount if the mountpoint is
3867789Sahrens  * 'legacy'.  Otherwise, complain that use should be using 'zfs mount'.
3868789Sahrens  */
3869789Sahrens static int
3870789Sahrens manual_mount(int argc, char **argv)
3871789Sahrens {
3872789Sahrens 	zfs_handle_t *zhp;
3873789Sahrens 	char mountpoint[ZFS_MAXPROPLEN];
3874789Sahrens 	char mntopts[MNT_LINE_MAX] = { '\0' };
3875789Sahrens 	int ret;
3876789Sahrens 	int c;
3877789Sahrens 	int flags = 0;
3878789Sahrens 	char *dataset, *path;
3879789Sahrens 
3880789Sahrens 	/* check options */
38811544Seschrock 	while ((c = getopt(argc, argv, ":mo:O")) != -1) {
3882789Sahrens 		switch (c) {
3883789Sahrens 		case 'o':
3884789Sahrens 			(void) strlcpy(mntopts, optarg, sizeof (mntopts));
3885789Sahrens 			break;
3886789Sahrens 		case 'O':
3887789Sahrens 			flags |= MS_OVERLAY;
3888789Sahrens 			break;
38891544Seschrock 		case 'm':
38901544Seschrock 			flags |= MS_NOMNTTAB;
38911544Seschrock 			break;
3892789Sahrens 		case ':':
3893789Sahrens 			(void) fprintf(stderr, gettext("missing argument for "
3894789Sahrens 			    "'%c' option\n"), optopt);
38952082Seschrock 			usage(B_FALSE);
3896789Sahrens 			break;
3897789Sahrens 		case '?':
3898789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
3899789Sahrens 			    optopt);
3900789Sahrens 			(void) fprintf(stderr, gettext("usage: mount [-o opts] "
3901789Sahrens 			    "<path>\n"));
3902789Sahrens 			return (2);
3903789Sahrens 		}
3904789Sahrens 	}
3905789Sahrens 
3906789Sahrens 	argc -= optind;
3907789Sahrens 	argv += optind;
3908789Sahrens 
3909789Sahrens 	/* check that we only have two arguments */
3910789Sahrens 	if (argc != 2) {
3911789Sahrens 		if (argc == 0)
3912789Sahrens 			(void) fprintf(stderr, gettext("missing dataset "
3913789Sahrens 			    "argument\n"));
3914789Sahrens 		else if (argc == 1)
3915789Sahrens 			(void) fprintf(stderr,
3916789Sahrens 			    gettext("missing mountpoint argument\n"));
3917789Sahrens 		else
3918789Sahrens 			(void) fprintf(stderr, gettext("too many arguments\n"));
3919789Sahrens 		(void) fprintf(stderr, "usage: mount <dataset> <mountpoint>\n");
3920789Sahrens 		return (2);
3921789Sahrens 	}
3922789Sahrens 
3923789Sahrens 	dataset = argv[0];
3924789Sahrens 	path = argv[1];
3925789Sahrens 
3926789Sahrens 	/* try to open the dataset */
39272082Seschrock 	if ((zhp = zfs_open(g_zfs, dataset, ZFS_TYPE_FILESYSTEM)) == NULL)
3928789Sahrens 		return (1);
3929789Sahrens 
3930789Sahrens 	(void) zfs_prop_get(zhp, ZFS_PROP_MOUNTPOINT, mountpoint,
39312082Seschrock 	    sizeof (mountpoint), NULL, NULL, 0, B_FALSE);
3932789Sahrens 
3933789Sahrens 	/* check for legacy mountpoint and complain appropriately */
3934789Sahrens 	ret = 0;
3935789Sahrens 	if (strcmp(mountpoint, ZFS_MOUNTPOINT_LEGACY) == 0) {
3936789Sahrens 		if (mount(dataset, path, MS_OPTIONSTR | flags, MNTTYPE_ZFS,
3937789Sahrens 		    NULL, 0, mntopts, sizeof (mntopts)) != 0) {
3938789Sahrens 			(void) fprintf(stderr, gettext("mount failed: %s\n"),
3939789Sahrens 			    strerror(errno));
3940789Sahrens 			ret = 1;
3941789Sahrens 		}
3942789Sahrens 	} else {
3943789Sahrens 		(void) fprintf(stderr, gettext("filesystem '%s' cannot be "
3944789Sahrens 		    "mounted using 'mount -F zfs'\n"), dataset);
3945789Sahrens 		(void) fprintf(stderr, gettext("Use 'zfs set mountpoint=%s' "
3946789Sahrens 		    "instead.\n"), path);
3947789Sahrens 		(void) fprintf(stderr, gettext("If you must use 'mount -F zfs' "
3948789Sahrens 		    "or /etc/vfstab, use 'zfs set mountpoint=legacy'.\n"));
3949789Sahrens 		(void) fprintf(stderr, gettext("See zfs(1M) for more "
3950789Sahrens 		    "information.\n"));
3951789Sahrens 		ret = 1;
3952789Sahrens 	}
3953789Sahrens 
3954789Sahrens 	return (ret);
3955789Sahrens }
3956789Sahrens 
3957789Sahrens /*
3958789Sahrens  * Called when invoked as /etc/fs/zfs/umount.  Unlike a manual mount, we allow
3959789Sahrens  * unmounts of non-legacy filesystems, as this is the dominant administrative
3960789Sahrens  * interface.
3961789Sahrens  */
3962789Sahrens static int
3963789Sahrens manual_unmount(int argc, char **argv)
3964789Sahrens {
3965789Sahrens 	int flags = 0;
3966789Sahrens 	int c;
3967789Sahrens 
3968789Sahrens 	/* check options */
3969789Sahrens 	while ((c = getopt(argc, argv, "f")) != -1) {
3970789Sahrens 		switch (c) {
3971789Sahrens 		case 'f':
3972789Sahrens 			flags = MS_FORCE;
3973789Sahrens 			break;
3974789Sahrens 		case '?':
3975789Sahrens 			(void) fprintf(stderr, gettext("invalid option '%c'\n"),
3976789Sahrens 			    optopt);
3977789Sahrens 			(void) fprintf(stderr, gettext("usage: unmount [-f] "
3978789Sahrens 			    "<path>\n"));
3979789Sahrens 			return (2);
3980789Sahrens 		}
3981789Sahrens 	}
3982789Sahrens 
3983789Sahrens 	argc -= optind;
3984789Sahrens 	argv += optind;
3985789Sahrens 
3986789Sahrens 	/* check arguments */
3987789Sahrens 	if (argc != 1) {
3988789Sahrens 		if (argc == 0)
3989789Sahrens 			(void) fprintf(stderr, gettext("missing path "
3990789Sahrens 			    "argument\n"));
3991789Sahrens 		else
3992789Sahrens 			(void) fprintf(stderr, gettext("too many arguments\n"));
3993789Sahrens 		(void) fprintf(stderr, gettext("usage: unmount [-f] <path>\n"));
3994789Sahrens 		return (2);
3995789Sahrens 	}
3996789Sahrens 
39972082Seschrock 	return (unshare_unmount_path(OP_MOUNT, argv[0], flags, B_TRUE));
3998789Sahrens }
3999789Sahrens 
4000789Sahrens static int
4001789Sahrens volcheck(zpool_handle_t *zhp, void *data)
4002789Sahrens {
40032856Snd150628 	boolean_t isinit = *((boolean_t *)data);
4004789Sahrens 
4005789Sahrens 	if (isinit)
4006789Sahrens 		return (zpool_create_zvol_links(zhp));
4007789Sahrens 	else
4008789Sahrens 		return (zpool_remove_zvol_links(zhp));
4009789Sahrens }
4010789Sahrens 
4011789Sahrens /*
4012789Sahrens  * Iterate over all pools in the system and either create or destroy /dev/zvol
4013789Sahrens  * links, depending on the value of 'isinit'.
4014789Sahrens  */
4015789Sahrens static int
40162082Seschrock do_volcheck(boolean_t isinit)
4017789Sahrens {
40182856Snd150628 	return (zpool_iter(g_zfs, volcheck, &isinit) ? 1 : 0);
4019789Sahrens }
4020789Sahrens 
40214543Smarks static int
40224543Smarks find_command_idx(char *command, int *idx)
40234543Smarks {
40244543Smarks 	int i;
40254543Smarks 
40264543Smarks 	for (i = 0; i < NCOMMAND; i++) {
40274543Smarks 		if (command_table[i].name == NULL)
40284543Smarks 			continue;
40294543Smarks 
40304543Smarks 		if (strcmp(command, command_table[i].name) == 0) {
40314543Smarks 			*idx = i;
40324543Smarks 			return (0);
40334543Smarks 		}
40344543Smarks 	}
40354543Smarks 	return (1);
40364543Smarks }
40374543Smarks 
4038789Sahrens int
4039789Sahrens main(int argc, char **argv)
4040789Sahrens {
4041789Sahrens 	int ret;
4042789Sahrens 	int i;
4043789Sahrens 	char *progname;
4044789Sahrens 	char *cmdname;
4045789Sahrens 
4046789Sahrens 	(void) setlocale(LC_ALL, "");
4047789Sahrens 	(void) textdomain(TEXT_DOMAIN);
4048789Sahrens 
4049789Sahrens 	opterr = 0;
4050789Sahrens 
40512082Seschrock 	if ((g_zfs = libzfs_init()) == NULL) {
40522082Seschrock 		(void) fprintf(stderr, gettext("internal error: failed to "
40532082Seschrock 		    "initialize ZFS library\n"));
40542082Seschrock 		return (1);
40552082Seschrock 	}
40562082Seschrock 
40574988Sek110237 	zpool_set_history_str("zfs", argc, argv, history_str);
40584988Sek110237 	verify(zpool_stage_history(g_zfs, history_str) == 0);
40594543Smarks 
40602082Seschrock 	libzfs_print_on_error(g_zfs, B_TRUE);
40612082Seschrock 
4062789Sahrens 	if ((mnttab_file = fopen(MNTTAB, "r")) == NULL) {
4063789Sahrens 		(void) fprintf(stderr, gettext("internal error: unable to "
4064789Sahrens 		    "open %s\n"), MNTTAB);
4065789Sahrens 		return (1);
4066789Sahrens 	}
4067789Sahrens 
4068789Sahrens 	/*
4069789Sahrens 	 * This command also doubles as the /etc/fs mount and unmount program.
4070789Sahrens 	 * Determine if we should take this behavior based on argv[0].
4071789Sahrens 	 */
4072789Sahrens 	progname = basename(argv[0]);
4073789Sahrens 	if (strcmp(progname, "mount") == 0) {
4074789Sahrens 		ret = manual_mount(argc, argv);
4075789Sahrens 	} else if (strcmp(progname, "umount") == 0) {
4076789Sahrens 		ret = manual_unmount(argc, argv);
4077789Sahrens 	} else {
4078789Sahrens 		/*
4079789Sahrens 		 * Make sure the user has specified some command.
4080789Sahrens 		 */
4081789Sahrens 		if (argc < 2) {
4082789Sahrens 			(void) fprintf(stderr, gettext("missing command\n"));
40832082Seschrock 			usage(B_FALSE);
4084789Sahrens 		}
4085789Sahrens 
4086789Sahrens 		cmdname = argv[1];
4087789Sahrens 
4088789Sahrens 		/*
4089789Sahrens 		 * The 'umount' command is an alias for 'unmount'
4090789Sahrens 		 */
4091789Sahrens 		if (strcmp(cmdname, "umount") == 0)
4092789Sahrens 			cmdname = "unmount";
4093789Sahrens 
4094789Sahrens 		/*
40951749Sahrens 		 * The 'recv' command is an alias for 'receive'
40961749Sahrens 		 */
40971749Sahrens 		if (strcmp(cmdname, "recv") == 0)
40981749Sahrens 			cmdname = "receive";
40991749Sahrens 
41001749Sahrens 		/*
4101789Sahrens 		 * Special case '-?'
4102789Sahrens 		 */
4103789Sahrens 		if (strcmp(cmdname, "-?") == 0)
41042082Seschrock 			usage(B_TRUE);
4105789Sahrens 
4106789Sahrens 		/*
4107789Sahrens 		 * 'volinit' and 'volfini' do not appear in the usage message,
4108789Sahrens 		 * so we have to special case them here.
4109789Sahrens 		 */
4110789Sahrens 		if (strcmp(cmdname, "volinit") == 0)
41112082Seschrock 			return (do_volcheck(B_TRUE));
4112789Sahrens 		else if (strcmp(cmdname, "volfini") == 0)
41132082Seschrock 			return (do_volcheck(B_FALSE));
4114789Sahrens 
4115789Sahrens 		/*
4116789Sahrens 		 * Run the appropriate command.
4117789Sahrens 		 */
41184543Smarks 		if (find_command_idx(cmdname, &i) == 0) {
41194543Smarks 			current_command = &command_table[i];
41204543Smarks 			ret = command_table[i].func(argc - 1, argv + 1);
41214787Sahrens 		} else if (strchr(cmdname, '=') != NULL) {
41224787Sahrens 			verify(find_command_idx("set", &i) == 0);
41234787Sahrens 			current_command = &command_table[i];
41244787Sahrens 			ret = command_table[i].func(argc, argv);
41254787Sahrens 		} else {
4126789Sahrens 			(void) fprintf(stderr, gettext("unrecognized "
4127789Sahrens 			    "command '%s'\n"), cmdname);
41282082Seschrock 			usage(B_FALSE);
4129789Sahrens 		}
4130789Sahrens 	}
4131789Sahrens 
4132789Sahrens 	(void) fclose(mnttab_file);
4133789Sahrens 
41342082Seschrock 	libzfs_fini(g_zfs);
41352082Seschrock 
4136789Sahrens 	/*
4137789Sahrens 	 * The 'ZFS_ABORT' environment variable causes us to dump core on exit
4138789Sahrens 	 * for the purposes of running ::findleaks.
4139789Sahrens 	 */
4140789Sahrens 	if (getenv("ZFS_ABORT") != NULL) {
4141789Sahrens 		(void) printf("dumping core by request\n");
4142789Sahrens 		abort();
4143789Sahrens 	}
4144789Sahrens 
4145789Sahrens 	return (ret);
4146789Sahrens }
4147