xref: /titanic_41/usr/src/cmd/beadm/beadm.c (revision f6911211edbabe97da98b0229c5e476991bbc73b)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
24  */
25 
26 /*
27  * Copyright 2013 Nexenta Systems, Inc. All rights reserved.
28  * Copyright 2015 Toomas Soome <tsoome@me.com>
29  */
30 
31 /*
32  * System includes
33  */
34 
35 #include <assert.h>
36 #include <stdio.h>
37 #include <strings.h>
38 #include <libzfs.h>
39 #include <locale.h>
40 #include <langinfo.h>
41 #include <stdlib.h>
42 #include <wchar.h>
43 #include <sys/types.h>
44 
45 #include "libbe.h"
46 
47 #ifndef lint
48 #define	_(x) gettext(x)
49 #else
50 #define	_(x) (x)
51 #endif
52 
53 #ifndef TEXT_DOMAIN
54 #define	TEXT_DOMAIN "SYS_TEST"
55 #endif
56 
57 #define	DT_BUF_LEN (128)
58 #define	NUM_COLS (6)
59 
60 static int be_do_activate(int argc, char **argv);
61 static int be_do_create(int argc, char **argv);
62 static int be_do_destroy(int argc, char **argv);
63 static int be_do_list(int argc, char **argv);
64 static int be_do_mount(int argc, char **argv);
65 static int be_do_unmount(int argc, char **argv);
66 static int be_do_rename(int argc, char **argv);
67 static int be_do_rollback(int argc, char **argv);
68 static void usage(void);
69 
70 /*
71  * single column name/width output format description
72  */
73 struct col_info {
74 	const char *col_name;
75 	size_t width;
76 };
77 
78 /*
79  * all columns output format
80  */
81 struct hdr_info {
82 	struct col_info cols[NUM_COLS];
83 };
84 
85 /*
86  * type of possible output formats
87  */
88 enum be_fmt {
89 	BE_FMT_DEFAULT,
90 	BE_FMT_DATASET,
91 	BE_FMT_SNAPSHOT,
92 	BE_FMT_ALL
93 };
94 
95 /*
96  * command handler description
97  */
98 typedef struct be_command {
99 	const char	*name;
100 	int		(*func)(int argc, char **argv);
101 } be_command_t;
102 
103 /*
104  * sorted list of be commands
105  */
106 static const be_command_t be_command_tbl[] = {
107 	{ "activate",		be_do_activate },
108 	{ "create",		be_do_create },
109 	{ "destroy",		be_do_destroy },
110 	{ "list",		be_do_list },
111 	{ "mount",		be_do_mount },
112 	{ "unmount",		be_do_unmount },
113 	{ "umount",		be_do_unmount }, /* unmount alias */
114 	{ "rename",		be_do_rename },
115 	{ "rollback",		be_do_rollback },
116 	{ NULL,			NULL },
117 };
118 
119 static void
120 usage(void)
121 {
122 	(void) fprintf(stderr, _("usage:\n"
123 	    "\tbeadm subcommand cmd_options\n"
124 	    "\n"
125 	    "\tsubcommands:\n"
126 	    "\n"
127 	    "\tbeadm activate [-v] beName\n"
128 	    "\tbeadm create [-a] [-d BE_desc]\n"
129 	    "\t\t[-o property=value] ... [-p zpool] \n"
130 	    "\t\t[-e nonActiveBe | beName@snapshot] [-v] beName\n"
131 	    "\tbeadm create [-d BE_desc]\n"
132 	    "\t\t[-o property=value] ... [-p zpool] [-v] beName@snapshot\n"
133 	    "\tbeadm destroy [-Ffsv] beName \n"
134 	    "\tbeadm destroy [-Fv] beName@snapshot \n"
135 	    "\tbeadm list [[-a] | [-d] [-s]] [-H]\n"
136 	    "\t\t[-k|-K date | name | space] [-v] [beName]\n"
137 	    "\tbeadm mount [-s ro|rw] [-v] beName [mountpoint]\n"
138 	    "\tbeadm unmount [-fv] beName | mountpoint\n"
139 	    "\tbeadm umount [-fv] beName | mountpoint\n"
140 	    "\tbeadm rename [-v] origBeName newBeName\n"
141 	    "\tbeadm rollback [-v] beName snapshot\n"
142 	    "\tbeadm rollback [-v] beName@snapshot\n"));
143 }
144 
145 static int
146 run_be_cmd(const char *cmdname, int argc, char **argv)
147 {
148 	const be_command_t *command;
149 
150 	for (command = &be_command_tbl[0]; command->name != NULL; command++)
151 		if (strcmp(command->name, cmdname) == 0)
152 			return (command->func(argc, argv));
153 
154 	(void) fprintf(stderr, _("Invalid command: %s\n"), cmdname);
155 	usage();
156 	return (1);
157 }
158 
159 int
160 main(int argc, char **argv)
161 {
162 	const char *cmdname;
163 
164 	(void) setlocale(LC_ALL, "");
165 	(void) textdomain(TEXT_DOMAIN);
166 
167 	if (argc < 2) {
168 		usage();
169 		return (1);
170 	}
171 
172 	cmdname = argv[1];
173 
174 	/* Turn error printing off */
175 	libbe_print_errors(B_FALSE);
176 
177 	return (run_be_cmd(cmdname, --argc, ++argv));
178 }
179 
180 static void
181 print_hdr(struct hdr_info *hdr_info)
182 {
183 	boolean_t first = B_TRUE;
184 	size_t i;
185 	for (i = 0; i < NUM_COLS; i++) {
186 		struct col_info *col_info = &hdr_info->cols[i];
187 		const char *name = col_info->col_name;
188 		size_t width = col_info->width;
189 		if (name == NULL)
190 			continue;
191 
192 		if (first) {
193 			(void) printf("%-*s", width, name);
194 			first = B_FALSE;
195 		} else
196 			(void) printf(" %-*s", width, name);
197 	}
198 	(void) putchar('\n');
199 }
200 
201 static void
202 init_hdr_cols(enum be_fmt be_fmt, struct hdr_info *hdr)
203 {
204 	struct col_info *col = hdr->cols;
205 	size_t i;
206 
207 	col[1].col_name = _("Active");
208 	col[2].col_name = _("Mountpoint");
209 	col[3].col_name = _("Space");
210 	col[4].col_name = _("Policy");
211 	col[5].col_name = _("Created");
212 	col[6].col_name = NULL;
213 
214 	switch (be_fmt) {
215 	case BE_FMT_ALL:
216 		col[0].col_name = _("BE/Dataset/Snapshot");
217 		break;
218 	case BE_FMT_DATASET:
219 		col[0].col_name = _("BE/Dataset");
220 		break;
221 	case BE_FMT_SNAPSHOT:
222 		col[0].col_name = _("BE/Snapshot");
223 		col[1].col_name = NULL;
224 		col[2].col_name = NULL;
225 		break;
226 	case BE_FMT_DEFAULT:
227 	default:
228 		col[0].col_name = _("BE");
229 	}
230 
231 	for (i = 0; i < NUM_COLS; i++) {
232 		const char *name = col[i].col_name;
233 		col[i].width = 0;
234 
235 		if (name != NULL) {
236 			wchar_t wname[128];
237 			size_t sz = mbstowcs(wname, name, sizeof (wname) /
238 			    sizeof (wchar_t));
239 			if (sz > 0) {
240 				int wcsw = wcswidth(wname, sz);
241 				if (wcsw > 0)
242 					col[i].width = wcsw;
243 				else
244 					col[i].width = sz;
245 			} else {
246 				col[i].width = strlen(name);
247 			}
248 		}
249 	}
250 }
251 
252 static void
253 nicenum(uint64_t num, char *buf, size_t buflen)
254 {
255 	uint64_t n = num;
256 	int index = 0;
257 	char u;
258 
259 	while (n >= 1024) {
260 		n /= 1024;
261 		index++;
262 	}
263 
264 	u = " KMGTPE"[index];
265 
266 	if (index == 0) {
267 		(void) snprintf(buf, buflen, "%llu", n);
268 	} else {
269 		int i;
270 		for (i = 2; i >= 0; i--) {
271 			if (snprintf(buf, buflen, "%.*f%c", i,
272 			    (double)num / (1ULL << 10 * index), u) <= 5)
273 				break;
274 		}
275 	}
276 }
277 
278 static void
279 count_widths(enum be_fmt be_fmt, struct hdr_info *hdr, be_node_list_t *be_nodes)
280 {
281 	size_t len[NUM_COLS];
282 	char buf[DT_BUF_LEN];
283 	int i;
284 	be_node_list_t *cur_be;
285 
286 	for (i = 0; i < NUM_COLS; i++)
287 		len[i] = hdr->cols[i].width;
288 
289 	for (cur_be = be_nodes; cur_be != NULL; cur_be = cur_be->be_next_node) {
290 		char name[ZFS_MAXNAMELEN+1];
291 		const char *be_name = cur_be->be_node_name;
292 		const char *root_ds = cur_be->be_root_ds;
293 		char *pos;
294 		size_t node_name_len = strlen(cur_be->be_node_name);
295 		size_t root_ds_len = strlen(cur_be->be_root_ds);
296 		size_t mntpt_len = 0;
297 		size_t policy_len = 0;
298 		size_t used_len;
299 		uint64_t used = cur_be->be_space_used;
300 		be_snapshot_list_t *snap = NULL;
301 
302 		if (cur_be->be_mntpt != NULL)
303 			mntpt_len = strlen(cur_be->be_mntpt);
304 		if (cur_be->be_policy_type != NULL)
305 			policy_len = strlen(cur_be->be_policy_type);
306 
307 		(void) strlcpy(name, root_ds, sizeof (name));
308 		pos = strstr(name, be_name);
309 
310 		if (be_fmt == BE_FMT_DEFAULT) {
311 			if (node_name_len > len[0])
312 				len[0] = node_name_len;
313 		} else {
314 			if (root_ds_len + 3 > len[0])
315 				len[0] = root_ds_len + 3;
316 		}
317 
318 		if (mntpt_len > len[2])
319 			len[2] = mntpt_len;
320 		if (policy_len > len[4])
321 			len[4] = policy_len;
322 
323 		for (snap = cur_be->be_node_snapshots; snap != NULL;
324 		    snap = snap->be_next_snapshot) {
325 			uint64_t snap_used = snap->be_snapshot_space_used;
326 			const char *snap_name = snap->be_snapshot_name;
327 			(void) strcpy(pos, snap_name);
328 
329 			if (be_fmt == BE_FMT_DEFAULT)
330 				used += snap_used;
331 			else if (be_fmt & BE_FMT_SNAPSHOT) {
332 				int snap_len = strlen(name) + 3;
333 				if (be_fmt == BE_FMT_SNAPSHOT)
334 					snap_len -= pos - name;
335 				if (snap_len > len[0])
336 					len[0] = snap_len;
337 				nicenum(snap_used, buf, sizeof (buf));
338 				used_len = strlen(buf);
339 				if (used_len > len[3])
340 					len[3] = used_len;
341 			}
342 		}
343 
344 		if (be_fmt == BE_FMT_DEFAULT) {
345 			int used_len;
346 			nicenum(used, buf, sizeof (buf));
347 			used_len = strlen(buf);
348 			if (used_len > len[3])
349 				len[3] = used_len;
350 		}
351 
352 		nicenum(used, buf, sizeof (buf));
353 	}
354 
355 	for (i = 0; i < NUM_COLS; i++)
356 		hdr->cols[i].width = len[i];
357 }
358 
359 static void
360 print_be_nodes(const char *be_name, boolean_t parsable, struct hdr_info *hdr,
361     be_node_list_t *nodes)
362 {
363 	char buf[64];
364 	char datetime[DT_BUF_LEN];
365 	be_node_list_t	*cur_be;
366 
367 	for (cur_be = nodes; cur_be != NULL; cur_be = cur_be->be_next_node) {
368 		char active[3] = "-\0";
369 		int ai = 0;
370 		const char *datetime_fmt = "%F %R";
371 		const char *name = cur_be->be_node_name;
372 		const char *mntpt = cur_be->be_mntpt;
373 		be_snapshot_list_t *snap = NULL;
374 		uint64_t used = cur_be->be_space_used;
375 		time_t creation = cur_be->be_node_creation;
376 		struct tm *tm;
377 
378 		if (be_name != NULL && strcmp(be_name, name) != 0)
379 			continue;
380 
381 		if (parsable)
382 			active[0] = '\0';
383 
384 		tm = localtime(&creation);
385 		(void) strftime(datetime, DT_BUF_LEN, datetime_fmt, tm);
386 
387 		for (snap = cur_be->be_node_snapshots; snap != NULL;
388 		    snap = snap->be_next_snapshot)
389 			used += snap->be_snapshot_space_used;
390 
391 		if (!cur_be->be_global_active)
392 			active[ai++] = 'x';
393 
394 		if (cur_be->be_active)
395 			active[ai++] = 'N';
396 		if (cur_be->be_active_on_boot) {
397 			if (!cur_be->be_global_active)
398 				active[ai] = 'b';
399 			else
400 				active[ai] = 'R';
401 		}
402 
403 		nicenum(used, buf, sizeof (buf));
404 		if (parsable)
405 			(void) printf("%s;%s;%s;%s;%llu;%s;%ld\n",
406 			    name,
407 			    cur_be->be_uuid_str,
408 			    active,
409 			    (cur_be->be_mounted ? mntpt: ""),
410 			    used,
411 			    cur_be->be_policy_type,
412 			    creation);
413 		else
414 			(void) printf("%-*s %-*s %-*s %-*s %-*s %-*s\n",
415 			    hdr->cols[0].width, name,
416 			    hdr->cols[1].width, active,
417 			    hdr->cols[2].width, (cur_be->be_mounted ? mntpt:
418 			    "-"),
419 			    hdr->cols[3].width, buf,
420 			    hdr->cols[4].width, cur_be->be_policy_type,
421 			    hdr->cols[5].width, datetime);
422 	}
423 }
424 
425 static void
426 print_be_snapshots(be_node_list_t *be, struct hdr_info *hdr, boolean_t parsable)
427 {
428 	char buf[64];
429 	char datetime[DT_BUF_LEN];
430 	be_snapshot_list_t *snap = NULL;
431 
432 	for (snap = be->be_node_snapshots; snap != NULL;
433 	    snap = snap->be_next_snapshot) {
434 		char name[ZFS_MAXNAMELEN+1];
435 		const char *datetime_fmt = "%F %R";
436 		const char *be_name = be->be_node_name;
437 		const char *root_ds = be->be_root_ds;
438 		const char *snap_name = snap->be_snapshot_name;
439 		char *pos;
440 		uint64_t used = snap->be_snapshot_space_used;
441 		time_t creation = snap->be_snapshot_creation;
442 		struct tm *tm = localtime(&creation);
443 
444 		(void) strncpy(name, root_ds, sizeof (name));
445 		pos = strstr(name, be_name);
446 		(void) strcpy(pos, snap_name);
447 
448 		(void) strftime(datetime, DT_BUF_LEN, datetime_fmt, tm);
449 		nicenum(used, buf, sizeof (buf));
450 
451 		if (parsable)
452 			if (hdr->cols[1].width != 0)
453 				(void) printf("%s;%s;%s;%s;%llu;%s;%ld\n",
454 				    be_name,
455 				    snap_name,
456 				    "",
457 				    "",
458 				    used,
459 				    be->be_policy_type,
460 				    creation);
461 			else
462 				(void) printf("%s;%s;%llu;%s;%ld\n",
463 				    be_name,
464 				    snap_name,
465 				    used,
466 				    be->be_policy_type,
467 				    creation);
468 		else
469 			if (hdr->cols[1].width != 0)
470 				(void) printf("   %-*s %-*s %-*s %-*s %-*s "
471 				    "%-*s\n",
472 				    hdr->cols[0].width-3, name,
473 				    hdr->cols[1].width, "-",
474 				    hdr->cols[2].width, "-",
475 				    hdr->cols[3].width, buf,
476 				    hdr->cols[4].width, be->be_policy_type,
477 				    hdr->cols[5].width, datetime);
478 			else
479 				(void) printf("   %-*s %-*s %-*s %-*s\n",
480 				    hdr->cols[0].width-3, snap_name,
481 				    hdr->cols[3].width, buf,
482 				    hdr->cols[4].width, be->be_policy_type,
483 				    hdr->cols[5].width, datetime);
484 	}
485 }
486 
487 static void
488 print_fmt_nodes(const char *be_name, enum be_fmt be_fmt, boolean_t parsable,
489     struct hdr_info *hdr, be_node_list_t *nodes)
490 {
491 	char buf[64];
492 	char datetime[DT_BUF_LEN];
493 	be_node_list_t	*cur_be;
494 
495 	for (cur_be = nodes; cur_be != NULL; cur_be = cur_be->be_next_node) {
496 		char active[3] = "-\0";
497 		int ai = 0;
498 		const char *datetime_fmt = "%F %R";
499 		const char *name = cur_be->be_node_name;
500 		const char *mntpt = cur_be->be_mntpt;
501 		uint64_t used = cur_be->be_space_used;
502 		time_t creation = cur_be->be_node_creation;
503 		struct tm *tm;
504 
505 		if (be_name != NULL && strcmp(be_name, name) != 0)
506 			continue;
507 
508 		if (!parsable)
509 			(void) printf("%-s\n", name);
510 		else
511 			active[0] = '\0';
512 
513 		tm = localtime(&creation);
514 		(void) strftime(datetime, DT_BUF_LEN, datetime_fmt, tm);
515 
516 		if (cur_be->be_active)
517 			active[ai++] = 'N';
518 		if (cur_be->be_active_on_boot)
519 			active[ai] = 'R';
520 
521 		nicenum(used, buf, sizeof (buf));
522 		if (be_fmt & BE_FMT_DATASET)
523 			if (parsable)
524 				(void) printf("%s;%s;%s;%s;%llu;%s;%ld\n",
525 				    cur_be->be_node_name,
526 				    cur_be->be_root_ds,
527 				    active,
528 				    (cur_be->be_mounted ? mntpt: ""),
529 				    used,
530 				    cur_be->be_policy_type,
531 				    creation);
532 			else
533 				(void) printf("   %-*s %-*s %-*s %-*s %-*s "
534 				    "%-*s\n",
535 				    hdr->cols[0].width-3, cur_be->be_root_ds,
536 				    hdr->cols[1].width, active,
537 				    hdr->cols[2].width, (cur_be->be_mounted ?
538 				    mntpt: "-"),
539 				    hdr->cols[3].width, buf,
540 				    hdr->cols[4].width, cur_be->be_policy_type,
541 				    hdr->cols[5].width, datetime);
542 
543 		if (be_fmt & BE_FMT_SNAPSHOT)
544 			print_be_snapshots(cur_be, hdr, parsable);
545 	}
546 }
547 
548 static void
549 print_nodes(const char *be_name, boolean_t dsets, boolean_t snaps,
550     boolean_t parsable, be_node_list_t *be_nodes)
551 {
552 	struct hdr_info hdr;
553 	enum be_fmt be_fmt  = BE_FMT_DEFAULT;
554 
555 	if (dsets)
556 		be_fmt |= BE_FMT_DATASET;
557 	if (snaps)
558 		be_fmt |= BE_FMT_SNAPSHOT;
559 
560 	if (!parsable) {
561 		init_hdr_cols(be_fmt, &hdr);
562 		count_widths(be_fmt, &hdr, be_nodes);
563 		print_hdr(&hdr);
564 	}
565 
566 	if (be_fmt == BE_FMT_DEFAULT)
567 		print_be_nodes(be_name, parsable, &hdr, be_nodes);
568 	else
569 		print_fmt_nodes(be_name, be_fmt, parsable, &hdr, be_nodes);
570 }
571 
572 static boolean_t
573 confirm_destroy(const char *name)
574 {
575 	boolean_t res = B_FALSE;
576 	const char *yesre = nl_langinfo(YESEXPR);
577 	const char *nore = nl_langinfo(NOEXPR);
578 	regex_t yes_re;
579 	regex_t no_re;
580 	char buf[128];
581 	char *answer;
582 	int cflags = REG_EXTENDED;
583 
584 	if (regcomp(&yes_re, yesre, cflags) != 0) {
585 		/* should not happen */
586 		(void) fprintf(stderr, _("Failed to compile 'yes' regexp\n"));
587 		return (res);
588 	}
589 	if (regcomp(&no_re, nore, cflags) != 0) {
590 		/* should not happen */
591 		(void) fprintf(stderr, _("Failed to compile 'no' regexp\n"));
592 		regfree(&yes_re);
593 		return (res);
594 	}
595 
596 	(void) printf(_("Are you sure you want to destroy %s?\n"
597 	    "This action cannot be undone (y/[n]): "), name);
598 
599 	answer = fgets(buf, sizeof (buf), stdin);
600 	if (answer == NULL || *answer == '\0' || *answer == 10)
601 		goto out;
602 
603 	if (regexec(&yes_re, answer, 0, NULL, 0) == 0) {
604 		res = B_TRUE;
605 	} else if (regexec(&no_re, answer, 0, NULL, 0) != 0) {
606 		(void) fprintf(stderr, _("Invalid response. "
607 		    "Please enter 'y' or 'n'.\n"));
608 	}
609 
610 out:
611 	regfree(&yes_re);
612 	regfree(&no_re);
613 	return (res);
614 }
615 
616 static int
617 be_nvl_alloc(nvlist_t **nvlp)
618 {
619 	assert(nvlp != NULL);
620 
621 	if (nvlist_alloc(nvlp, NV_UNIQUE_NAME, 0) != 0) {
622 		(void) perror(_("nvlist_alloc failed.\n"));
623 		return (1);
624 	}
625 
626 	return (0);
627 }
628 
629 static int
630 be_nvl_add_string(nvlist_t *nvl, const char *name, const char *val)
631 {
632 	assert(nvl != NULL);
633 
634 	if (nvlist_add_string(nvl, name, val) != 0) {
635 		(void) fprintf(stderr, _("nvlist_add_string failed for "
636 		    "%s (%s).\n"), name, val);
637 		return (1);
638 	}
639 
640 	return (0);
641 }
642 
643 static int
644 be_nvl_add_nvlist(nvlist_t *nvl, const char *name, nvlist_t *val)
645 {
646 	assert(nvl != NULL);
647 
648 	if (nvlist_add_nvlist(nvl, name, val) != 0) {
649 		(void) fprintf(stderr, _("nvlist_add_nvlist failed for %s.\n"),
650 		    name);
651 		return (1);
652 	}
653 
654 	return (0);
655 }
656 
657 static int
658 be_nvl_add_uint16(nvlist_t *nvl, const char *name, uint16_t val)
659 {
660 	assert(nvl != NULL);
661 
662 	if (nvlist_add_uint16(nvl, name, val) != 0) {
663 		(void) fprintf(stderr, _("nvlist_add_uint16 failed for "
664 		    "%s (%hu).\n"), name, val);
665 		return (1);
666 	}
667 
668 	return (0);
669 }
670 
671 static int
672 be_do_activate(int argc, char **argv)
673 {
674 	nvlist_t	*be_attrs;
675 	int		err = 1;
676 	int		c;
677 	char		*obe_name;
678 
679 	while ((c = getopt(argc, argv, "v")) != -1) {
680 		switch (c) {
681 		case 'v':
682 			libbe_print_errors(B_TRUE);
683 			break;
684 		default:
685 			usage();
686 			return (1);
687 		}
688 	}
689 
690 	argc -= optind;
691 	argv += optind;
692 
693 	if (argc != 1) {
694 		usage();
695 		return (1);
696 	}
697 
698 	obe_name = argv[0];
699 
700 	if (be_nvl_alloc(&be_attrs) != 0)
701 		return (1);
702 
703 	if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
704 		goto out;
705 
706 	err = be_activate(be_attrs);
707 
708 	switch (err) {
709 	case BE_SUCCESS:
710 		(void) printf(_("Activated successfully\n"));
711 		break;
712 	case BE_ERR_BE_NOENT:
713 		(void) fprintf(stderr, _("%s does not exist or appear "
714 		    "to be a valid BE.\nPlease check that the name of "
715 		    "the BE provided is correct.\n"), obe_name);
716 		break;
717 	case BE_ERR_PERM:
718 	case BE_ERR_ACCESS:
719 		(void) fprintf(stderr, _("Unable to activate %s.\n"), obe_name);
720 		(void) fprintf(stderr, _("You have insufficient privileges to "
721 		    "execute this command.\n"));
722 		break;
723 	case BE_ERR_ACTIVATE_CURR:
724 	default:
725 		(void) fprintf(stderr, _("Unable to activate %s.\n"), obe_name);
726 		(void) fprintf(stderr, "%s\n", be_err_to_str(err));
727 	}
728 
729 out:
730 	nvlist_free(be_attrs);
731 	return (err);
732 }
733 
734 static int
735 be_do_create(int argc, char **argv)
736 {
737 	nvlist_t	*be_attrs;
738 	nvlist_t	*zfs_props = NULL;
739 	boolean_t	activate = B_FALSE;
740 	boolean_t	is_snap = B_FALSE;
741 	int		c;
742 	int		err = 1;
743 	char		*obe_name = NULL;
744 	char		*snap_name = NULL;
745 	char		*nbe_zpool = NULL;
746 	char		*nbe_name = NULL;
747 	char		*nbe_desc = NULL;
748 	char		*propname = NULL;
749 	char		*propval = NULL;
750 	char		*strval = NULL;
751 
752 	while ((c = getopt(argc, argv, "ad:e:io:p:v")) != -1) {
753 		switch (c) {
754 		case 'a':
755 			activate = B_TRUE;
756 			break;
757 		case 'd':
758 			nbe_desc = optarg;
759 			break;
760 		case 'e':
761 			obe_name = optarg;
762 			break;
763 		case 'o':
764 			if (zfs_props == NULL && be_nvl_alloc(&zfs_props) != 0)
765 				return (1);
766 
767 			propname = optarg;
768 			if ((propval = strchr(propname, '=')) == NULL) {
769 				(void) fprintf(stderr, _("missing "
770 				    "'=' for -o option\n"));
771 				goto out2;
772 			}
773 			*propval = '\0';
774 			propval++;
775 			if (nvlist_lookup_string(zfs_props, propname,
776 			    &strval) == 0) {
777 				(void) fprintf(stderr, _("property '%s' "
778 				    "specified multiple times\n"), propname);
779 				goto out2;
780 
781 			}
782 			if (be_nvl_add_string(zfs_props, propname, propval)
783 			    != 0)
784 				goto out2;
785 
786 			break;
787 		case 'p':
788 			nbe_zpool = optarg;
789 			break;
790 		case 'v':
791 			libbe_print_errors(B_TRUE);
792 			break;
793 		default:
794 			usage();
795 			goto out2;
796 		}
797 	}
798 
799 	argc -= optind;
800 	argv += optind;
801 
802 	if (argc != 1) {
803 		usage();
804 		goto out2;
805 	}
806 
807 	nbe_name = argv[0];
808 
809 	if ((snap_name = strrchr(nbe_name, '@')) != NULL) {
810 		if (snap_name[1] == '\0') {
811 			usage();
812 			goto out2;
813 		}
814 
815 		snap_name[0] = '\0';
816 		snap_name++;
817 		is_snap = B_TRUE;
818 	}
819 
820 	if (obe_name) {
821 		if (is_snap) {
822 			usage();
823 			goto out2;
824 		}
825 
826 		/*
827 		 * Check if obe_name is really a snapshot name.
828 		 * If so, split it out.
829 		 */
830 		if ((snap_name = strrchr(obe_name, '@')) != NULL) {
831 			if (snap_name[1] == '\0') {
832 				usage();
833 				goto out2;
834 			}
835 
836 			snap_name[0] = '\0';
837 			snap_name++;
838 		}
839 	} else if (is_snap) {
840 		obe_name = nbe_name;
841 		nbe_name = NULL;
842 	}
843 
844 	if (be_nvl_alloc(&be_attrs) != 0)
845 		goto out2;
846 
847 
848 	if (zfs_props != NULL && be_nvl_add_nvlist(be_attrs,
849 	    BE_ATTR_ORIG_BE_NAME, zfs_props) != 0)
850 		goto out;
851 
852 	if (obe_name != NULL && be_nvl_add_string(be_attrs,
853 	    BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
854 		goto out;
855 
856 	if (snap_name != NULL && be_nvl_add_string(be_attrs,
857 	    BE_ATTR_SNAP_NAME, snap_name) != 0)
858 		goto out;
859 
860 	if (nbe_zpool != NULL && be_nvl_add_string(be_attrs,
861 	    BE_ATTR_NEW_BE_POOL, nbe_zpool) != 0)
862 		goto out;
863 
864 	if (nbe_name != NULL && be_nvl_add_string(be_attrs,
865 	    BE_ATTR_NEW_BE_NAME, nbe_name) != 0)
866 		goto out;
867 
868 	if (nbe_desc != NULL && be_nvl_add_string(be_attrs,
869 	    BE_ATTR_NEW_BE_DESC, nbe_desc) != 0)
870 		goto out;
871 
872 	if (is_snap)
873 		err = be_create_snapshot(be_attrs);
874 	else
875 		err = be_copy(be_attrs);
876 
877 	switch (err) {
878 	case BE_SUCCESS:
879 		if (!is_snap && !nbe_name) {
880 			/*
881 			 * We requested an auto named BE; find out the
882 			 * name of the BE that was created for us and
883 			 * the auto snapshot created from the original BE.
884 			 */
885 			if (nvlist_lookup_string(be_attrs, BE_ATTR_NEW_BE_NAME,
886 			    &nbe_name) != 0) {
887 				(void) fprintf(stderr, _("failed to get %s "
888 				    "attribute\n"), BE_ATTR_NEW_BE_NAME);
889 				break;
890 			} else
891 				(void) printf(_("Auto named BE: %s\n"),
892 				    nbe_name);
893 
894 			if (nvlist_lookup_string(be_attrs, BE_ATTR_SNAP_NAME,
895 			    &snap_name) != 0) {
896 				(void) fprintf(stderr, _("failed to get %s "
897 				    "attribute\n"), BE_ATTR_SNAP_NAME);
898 				break;
899 			} else
900 				(void) printf(_("Auto named snapshot: %s\n"),
901 				    snap_name);
902 		}
903 
904 		if (!is_snap && activate) {
905 			char *args[] = { "activate", "", NULL };
906 			args[1] = nbe_name;
907 			optind = 1;
908 
909 			err = be_do_activate(2, args);
910 			goto out;
911 		}
912 
913 		(void) printf(_("Created successfully\n"));
914 		break;
915 	case BE_ERR_BE_EXISTS:
916 		(void) fprintf(stderr, _("BE %s already exists\n."
917 		    "Please choose a different BE name.\n"), nbe_name);
918 		break;
919 	case BE_ERR_SS_EXISTS:
920 		(void) fprintf(stderr, _("BE %s snapshot %s already exists.\n"
921 		    "Please choose a different snapshot name.\n"), obe_name,
922 		    snap_name);
923 		break;
924 	case BE_ERR_PERM:
925 	case BE_ERR_ACCESS:
926 		if (is_snap)
927 			(void) fprintf(stderr, _("Unable to create snapshot "
928 			    "%s.\n"), snap_name);
929 		else
930 			(void) fprintf(stderr, _("Unable to create %s.\n"),
931 			    nbe_name);
932 		(void) fprintf(stderr, _("You have insufficient privileges to "
933 		    "execute this command.\n"));
934 		break;
935 	default:
936 		if (is_snap)
937 			(void) fprintf(stderr, _("Unable to create snapshot "
938 			    "%s.\n"), snap_name);
939 		else
940 			(void) fprintf(stderr, _("Unable to create %s.\n"),
941 			    nbe_name);
942 		(void) fprintf(stderr, "%s\n", be_err_to_str(err));
943 	}
944 
945 out:
946 	nvlist_free(be_attrs);
947 out2:
948 	if (zfs_props != NULL)
949 		nvlist_free(zfs_props);
950 
951 	return (err);
952 }
953 
954 static int
955 be_do_destroy(int argc, char **argv)
956 {
957 	nvlist_t	*be_attrs;
958 	boolean_t	is_snap = B_FALSE;
959 	boolean_t	suppress_prompt = B_FALSE;
960 	int		err = 1;
961 	int		c;
962 	int		destroy_flags = 0;
963 	char		*snap_name;
964 	char		*be_name;
965 
966 	while ((c = getopt(argc, argv, "fFsv")) != -1) {
967 		switch (c) {
968 		case 'f':
969 			destroy_flags |= BE_DESTROY_FLAG_FORCE_UNMOUNT;
970 			break;
971 		case 's':
972 			destroy_flags |= BE_DESTROY_FLAG_SNAPSHOTS;
973 			break;
974 		case 'v':
975 			libbe_print_errors(B_TRUE);
976 			break;
977 		case 'F':
978 			suppress_prompt = B_TRUE;
979 			break;
980 		default:
981 			usage();
982 			return (1);
983 		}
984 	}
985 
986 	argc -= optind;
987 	argv += optind;
988 
989 	if (argc != 1) {
990 		usage();
991 		return (1);
992 	}
993 
994 	be_name = argv[0];
995 	if (!suppress_prompt && !confirm_destroy(be_name)) {
996 		(void) printf(_("%s has not been destroyed.\n"), be_name);
997 		return (0);
998 	}
999 
1000 	if ((snap_name = strrchr(be_name, '@')) != NULL) {
1001 		if (snap_name[1] == '\0') {
1002 			usage();
1003 			return (1);
1004 		}
1005 
1006 		is_snap = B_TRUE;
1007 		*snap_name = '\0';
1008 		snap_name++;
1009 	}
1010 
1011 	if (be_nvl_alloc(&be_attrs) != 0)
1012 		return (1);
1013 
1014 
1015 	if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, be_name) != 0)
1016 		goto out;
1017 
1018 	if (is_snap) {
1019 		if (be_nvl_add_string(be_attrs, BE_ATTR_SNAP_NAME,
1020 		    snap_name) != 0)
1021 			goto out;
1022 
1023 		err = be_destroy_snapshot(be_attrs);
1024 	} else {
1025 		if (be_nvl_add_uint16(be_attrs, BE_ATTR_DESTROY_FLAGS,
1026 		    destroy_flags) != 0)
1027 			goto out;
1028 
1029 		err = be_destroy(be_attrs);
1030 	}
1031 
1032 	switch (err) {
1033 	case BE_SUCCESS:
1034 		(void) printf(_("Destroyed successfully\n"));
1035 		break;
1036 	case BE_ERR_MOUNTED:
1037 		(void) fprintf(stderr, _("Unable to destroy %s.\n"), be_name);
1038 		(void) fprintf(stderr, _("It is currently mounted and must be "
1039 		    "unmounted before it can be destroyed.\n" "Use 'beadm "
1040 		    "unmount %s' to unmount the BE before destroying\nit or "
1041 		    "'beadm destroy -f %s'.\n"), be_name, be_name);
1042 		break;
1043 	case BE_ERR_DESTROY_CURR_BE:
1044 		(void) fprintf(stderr, _("%s is the currently active BE and "
1045 		    "cannot be destroyed.\nYou must boot from another BE in "
1046 		    "order to destroy %s.\n"), be_name, be_name);
1047 		break;
1048 	case BE_ERR_ZONES_UNMOUNT:
1049 		(void) fprintf(stderr, _("Unable to destroy one of " "%s's "
1050 		    "zone BE's.\nUse 'beadm destroy -f %s' or "
1051 		    "'zfs -f destroy <dataset>'.\n"), be_name, be_name);
1052 		break;
1053 	case BE_ERR_SS_NOENT:
1054 		(void) fprintf(stderr, _("%s does not exist or appear "
1055 		    "to be a valid snapshot.\nPlease check that the name of "
1056 		    "the snapshot provided is correct.\n"), snap_name);
1057 		break;
1058 	case BE_ERR_PERM:
1059 	case BE_ERR_ACCESS:
1060 		(void) fprintf(stderr, _("Unable to destroy %s.\n"), be_name);
1061 		(void) fprintf(stderr, _("You have insufficient privileges to "
1062 		    "execute this command.\n"));
1063 		break;
1064 	case BE_ERR_SS_EXISTS:
1065 		(void) fprintf(stderr, _("Unable to destroy %s: "
1066 		    "BE has snapshots.\nUse 'beadm destroy -s %s' or "
1067 		    "'zfs -r destroy <dataset>'.\n"), be_name, be_name);
1068 		break;
1069 	default:
1070 		(void) fprintf(stderr, _("Unable to destroy %s.\n"), be_name);
1071 		(void) fprintf(stderr, "%s\n", be_err_to_str(err));
1072 	}
1073 
1074 out:
1075 	nvlist_free(be_attrs);
1076 	return (err);
1077 }
1078 
1079 static int
1080 be_do_list(int argc, char **argv)
1081 {
1082 	be_node_list_t	*be_nodes = NULL;
1083 	boolean_t	all = B_FALSE;
1084 	boolean_t	dsets = B_FALSE;
1085 	boolean_t	snaps = B_FALSE;
1086 	boolean_t	parsable = B_FALSE;
1087 	int		err = 1;
1088 	int		c = 0;
1089 	char		*be_name = NULL;
1090 	be_sort_t	order = BE_SORT_UNSPECIFIED;
1091 
1092 	while ((c = getopt(argc, argv, "adk:svHK:")) != -1) {
1093 		switch (c) {
1094 		case 'a':
1095 			all = B_TRUE;
1096 			break;
1097 		case 'd':
1098 			dsets = B_TRUE;
1099 			break;
1100 		case 'k':
1101 		case 'K':
1102 			if (order != BE_SORT_UNSPECIFIED) {
1103 				(void) fprintf(stderr, _("Sort key can be "
1104 				    "specified only once.\n"));
1105 				usage();
1106 				return (1);
1107 			}
1108 			if (strcmp(optarg, "date") == 0) {
1109 				if (c == 'k')
1110 					order = BE_SORT_DATE;
1111 				else
1112 					order = BE_SORT_DATE_REV;
1113 				break;
1114 			}
1115 			if (strcmp(optarg, "name") == 0) {
1116 				if (c == 'k')
1117 					order = BE_SORT_NAME;
1118 				else
1119 					order = BE_SORT_NAME_REV;
1120 				break;
1121 			}
1122 			if (strcmp(optarg, "space") == 0) {
1123 				if (c == 'k')
1124 					order = BE_SORT_SPACE;
1125 				else
1126 					order = BE_SORT_SPACE_REV;
1127 				break;
1128 			}
1129 			(void) fprintf(stderr, _("Unknown sort key: %s\n"),
1130 			    optarg);
1131 			usage();
1132 			return (1);
1133 		case 's':
1134 			snaps = B_TRUE;
1135 			break;
1136 		case 'v':
1137 			libbe_print_errors(B_TRUE);
1138 			break;
1139 		case 'H':
1140 			parsable = B_TRUE;
1141 			break;
1142 		default:
1143 			usage();
1144 			return (1);
1145 		}
1146 	}
1147 
1148 	if (all) {
1149 		if (dsets) {
1150 			(void) fprintf(stderr, _("Invalid options: -a and %s "
1151 			    "are mutually exclusive.\n"), "-d");
1152 			usage();
1153 			return (1);
1154 		}
1155 		if (snaps) {
1156 			(void) fprintf(stderr, _("Invalid options: -a and %s "
1157 			    "are mutually exclusive.\n"), "-s");
1158 			usage();
1159 			return (1);
1160 		}
1161 
1162 		dsets = B_TRUE;
1163 		snaps = B_TRUE;
1164 	}
1165 
1166 	argc -= optind;
1167 	argv += optind;
1168 
1169 
1170 	if (argc == 1)
1171 		be_name = argv[0];
1172 
1173 	err = be_list(be_name, &be_nodes);
1174 
1175 	switch (err) {
1176 	case BE_SUCCESS:
1177 		/* the default sort is ascending date, no need to sort twice */
1178 		if (order == BE_SORT_UNSPECIFIED)
1179 			order = BE_SORT_DATE;
1180 
1181 		if (order != BE_SORT_DATE)
1182 			be_sort(&be_nodes, order);
1183 
1184 		print_nodes(be_name, dsets, snaps, parsable, be_nodes);
1185 		break;
1186 	case BE_ERR_BE_NOENT:
1187 		if (be_name == NULL)
1188 			(void) fprintf(stderr, _("No boot environments found "
1189 			    "on this system.\n"));
1190 		else {
1191 			(void) fprintf(stderr, _("%s does not exist or appear "
1192 			    "to be a valid BE.\nPlease check that the name of "
1193 			    "the BE provided is correct.\n"), be_name);
1194 		}
1195 		break;
1196 	default:
1197 		(void) fprintf(stderr, _("Unable to display Boot "
1198 		    "Environment\n"));
1199 		(void) fprintf(stderr, "%s\n", be_err_to_str(err));
1200 	}
1201 
1202 	if (be_nodes != NULL)
1203 		be_free_list(be_nodes);
1204 	return (err);
1205 }
1206 
1207 static int
1208 be_do_mount(int argc, char **argv)
1209 {
1210 	nvlist_t	*be_attrs;
1211 	boolean_t	shared_fs = B_FALSE;
1212 	int		err = 1;
1213 	int		c;
1214 	int		mount_flags = 0;
1215 	char		*obe_name;
1216 	char		*mountpoint;
1217 	char		*tmp_mp = NULL;
1218 
1219 	while ((c = getopt(argc, argv, "s:v")) != -1) {
1220 		switch (c) {
1221 		case 's':
1222 			shared_fs = B_TRUE;
1223 
1224 			mount_flags |= BE_MOUNT_FLAG_SHARED_FS;
1225 
1226 			if (strcmp(optarg, "rw") == 0) {
1227 				mount_flags |= BE_MOUNT_FLAG_SHARED_RW;
1228 			} else if (strcmp(optarg, "ro") != 0) {
1229 				(void) fprintf(stderr, _("The -s flag "
1230 				    "requires an argument [ rw | ro ]\n"));
1231 				usage();
1232 				return (1);
1233 			}
1234 
1235 			break;
1236 		case 'v':
1237 			libbe_print_errors(B_TRUE);
1238 			break;
1239 		default:
1240 			usage();
1241 			return (1);
1242 		}
1243 	}
1244 
1245 	argc -= optind;
1246 	argv += optind;
1247 
1248 	if (argc < 1 || argc > 2) {
1249 		usage();
1250 		return (1);
1251 	}
1252 
1253 	obe_name = argv[0];
1254 
1255 	if (argc == 2) {
1256 		mountpoint = argv[1];
1257 		if (mountpoint[0] != '/') {
1258 			(void) fprintf(stderr, _("Invalid mount point %s. "
1259 			    "Mount point must start with a /.\n"), mountpoint);
1260 			return (1);
1261 		}
1262 	} else {
1263 		const char *tmpdir = getenv("TMPDIR");
1264 		const char *tmpname = "tmp.XXXXXX";
1265 		int sz;
1266 
1267 		if (tmpdir == NULL)
1268 			tmpdir = "/tmp";
1269 
1270 		sz = asprintf(&tmp_mp, "%s/%s", tmpdir, tmpname);
1271 		if (sz < 0) {
1272 			(void) fprintf(stderr, _("internal error: "
1273 			    "out of memory\n"));
1274 			return (1);
1275 		}
1276 
1277 		mountpoint = mkdtemp(tmp_mp);
1278 	}
1279 
1280 	if (be_nvl_alloc(&be_attrs) != 0)
1281 		return (1);
1282 
1283 	if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
1284 		goto out;
1285 
1286 	if (be_nvl_add_string(be_attrs, BE_ATTR_MOUNTPOINT, mountpoint) != 0)
1287 		goto out;
1288 
1289 	if (shared_fs && be_nvl_add_uint16(be_attrs, BE_ATTR_MOUNT_FLAGS,
1290 	    mount_flags) != 0)
1291 		goto out;
1292 
1293 	err = be_mount(be_attrs);
1294 
1295 	switch (err) {
1296 	case BE_SUCCESS:
1297 		(void) printf(_("Mounted successfully on: '%s'\n"), mountpoint);
1298 		break;
1299 	case BE_ERR_BE_NOENT:
1300 		(void) fprintf(stderr, _("%s does not exist or appear "
1301 		    "to be a valid BE.\nPlease check that the name of "
1302 		    "the BE provided is correct.\n"), obe_name);
1303 		break;
1304 	case BE_ERR_MOUNTED:
1305 		(void) fprintf(stderr, _("%s is already mounted.\n"
1306 		    "Please unmount the BE before mounting it again.\n"),
1307 		    obe_name);
1308 		break;
1309 	case BE_ERR_PERM:
1310 	case BE_ERR_ACCESS:
1311 		(void) fprintf(stderr, _("Unable to mount %s.\n"), obe_name);
1312 		(void) fprintf(stderr, _("You have insufficient privileges to "
1313 		    "execute this command.\n"));
1314 		break;
1315 	case BE_ERR_NO_MOUNTED_ZONE:
1316 		(void) fprintf(stderr, _("Mounted on '%s'.\nUnable to mount "
1317 		    "one of %s's zone BE's.\n"), mountpoint, obe_name);
1318 		break;
1319 	default:
1320 		(void) fprintf(stderr, _("Unable to mount %s.\n"), obe_name);
1321 		(void) fprintf(stderr, "%s\n", be_err_to_str(err));
1322 	}
1323 
1324 out:
1325 	if (tmp_mp != NULL)
1326 		free(tmp_mp);
1327 	nvlist_free(be_attrs);
1328 	return (err);
1329 }
1330 
1331 static int
1332 be_do_unmount(int argc, char **argv)
1333 {
1334 	nvlist_t	*be_attrs;
1335 	char		*obe_name;
1336 	int		err = 1;
1337 	int		c;
1338 	int		unmount_flags = 0;
1339 
1340 	while ((c = getopt(argc, argv, "fv")) != -1) {
1341 		switch (c) {
1342 		case 'f':
1343 			unmount_flags |= BE_UNMOUNT_FLAG_FORCE;
1344 			break;
1345 		case 'v':
1346 			libbe_print_errors(B_TRUE);
1347 			break;
1348 		default:
1349 			usage();
1350 			return (1);
1351 		}
1352 	}
1353 
1354 	argc -= optind;
1355 	argv += optind;
1356 
1357 	if (argc != 1) {
1358 		usage();
1359 		return (1);
1360 	}
1361 
1362 	obe_name = argv[0];
1363 
1364 	if (be_nvl_alloc(&be_attrs) != 0)
1365 		return (1);
1366 
1367 
1368 	if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
1369 		goto out;
1370 
1371 	if (be_nvl_add_uint16(be_attrs, BE_ATTR_UNMOUNT_FLAGS,
1372 	    unmount_flags) != 0)
1373 		goto out;
1374 
1375 	err = be_unmount(be_attrs);
1376 
1377 	switch (err) {
1378 	case BE_SUCCESS:
1379 		(void) printf(_("Unmounted successfully\n"));
1380 		break;
1381 	case BE_ERR_BE_NOENT:
1382 		(void) fprintf(stderr, _("%s does not exist or appear "
1383 		    "to be a valid BE.\nPlease check that the name of "
1384 		    "the BE provided is correct.\n"), obe_name);
1385 		break;
1386 	case BE_ERR_UMOUNT_CURR_BE:
1387 		(void) fprintf(stderr, _("%s is the currently active BE.\n"
1388 		    "It cannot be unmounted unless another BE is the "
1389 		    "currently active BE.\n"), obe_name);
1390 		break;
1391 	case BE_ERR_UMOUNT_SHARED:
1392 		(void) fprintf(stderr, _("%s is a shared file system and it "
1393 		    "cannot be unmounted.\n"), obe_name);
1394 		break;
1395 	case BE_ERR_PERM:
1396 	case BE_ERR_ACCESS:
1397 		(void) fprintf(stderr, _("Unable to unmount %s.\n"), obe_name);
1398 		(void) fprintf(stderr, _("You have insufficient privileges to "
1399 		    "execute this command.\n"));
1400 		break;
1401 	default:
1402 		(void) fprintf(stderr, _("Unable to unmount %s.\n"), obe_name);
1403 		(void) fprintf(stderr, "%s\n", be_err_to_str(err));
1404 	}
1405 
1406 out:
1407 	nvlist_free(be_attrs);
1408 	return (err);
1409 }
1410 
1411 static int
1412 be_do_rename(int argc, char **argv)
1413 {
1414 	nvlist_t	*be_attrs;
1415 	char		*obe_name;
1416 	char		*nbe_name;
1417 	int err = 1;
1418 	int c;
1419 
1420 	while ((c = getopt(argc, argv, "v")) != -1) {
1421 		switch (c) {
1422 		case 'v':
1423 			libbe_print_errors(B_TRUE);
1424 			break;
1425 		default:
1426 			usage();
1427 			return (1);
1428 		}
1429 	}
1430 
1431 	argc -= optind;
1432 	argv += optind;
1433 
1434 	if (argc != 2) {
1435 		usage();
1436 		return (1);
1437 	}
1438 
1439 	obe_name = argv[0];
1440 	nbe_name = argv[1];
1441 
1442 	if (be_nvl_alloc(&be_attrs) != 0)
1443 		return (1);
1444 
1445 	if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
1446 		goto out;
1447 
1448 	if (be_nvl_add_string(be_attrs, BE_ATTR_NEW_BE_NAME, nbe_name) != 0)
1449 		goto out;
1450 
1451 	err = be_rename(be_attrs);
1452 
1453 	switch (err) {
1454 	case BE_SUCCESS:
1455 		(void) printf(_("Renamed successfully\n"));
1456 		break;
1457 	case BE_ERR_BE_NOENT:
1458 		(void) fprintf(stderr, _("%s does not exist or appear "
1459 		    "to be a valid BE.\nPlease check that the name of "
1460 		    "the BE provided is correct.\n"), obe_name);
1461 		break;
1462 	case BE_ERR_PERM:
1463 	case BE_ERR_ACCESS:
1464 		(void) fprintf(stderr, _("Rename of BE %s failed.\n"),
1465 		    obe_name);
1466 		(void) fprintf(stderr, _("You have insufficient privileges to "
1467 		    "execute this command.\n"));
1468 		break;
1469 	default:
1470 		(void) fprintf(stderr, _("Rename of BE %s failed.\n"),
1471 		    obe_name);
1472 		(void) fprintf(stderr, "%s\n", be_err_to_str(err));
1473 	}
1474 
1475 out:
1476 	nvlist_free(be_attrs);
1477 	return (err);
1478 }
1479 
1480 static int
1481 be_do_rollback(int argc, char **argv)
1482 {
1483 	nvlist_t	*be_attrs;
1484 	char		*obe_name;
1485 	char		*snap_name;
1486 	int		err = 1;
1487 	int		c;
1488 
1489 	while ((c = getopt(argc, argv, "v")) != -1) {
1490 		switch (c) {
1491 		case 'v':
1492 			libbe_print_errors(B_TRUE);
1493 			break;
1494 		default:
1495 			usage();
1496 			return (1);
1497 		}
1498 	}
1499 
1500 	argc -= optind;
1501 	argv += optind;
1502 
1503 	if (argc < 1 || argc > 2) {
1504 		usage();
1505 		return (1);
1506 	}
1507 
1508 	obe_name = argv[0];
1509 	if (argc == 2)
1510 		snap_name = argv[1];
1511 	else { /* argc == 1 */
1512 		if ((snap_name = strrchr(obe_name, '@')) != NULL) {
1513 			if (snap_name[1] == '\0') {
1514 				usage();
1515 				return (1);
1516 			}
1517 
1518 			snap_name[0] = '\0';
1519 			snap_name++;
1520 		} else {
1521 			usage();
1522 			return (1);
1523 		}
1524 	}
1525 
1526 	if (be_nvl_alloc(&be_attrs) != 0)
1527 		return (1);
1528 
1529 	if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
1530 		goto out;
1531 
1532 	if (be_nvl_add_string(be_attrs, BE_ATTR_SNAP_NAME, snap_name) != 0)
1533 		goto out;
1534 
1535 	err = be_rollback(be_attrs);
1536 
1537 	switch (err) {
1538 	case BE_SUCCESS:
1539 		(void) printf(_("Rolled back successfully\n"));
1540 		break;
1541 	case BE_ERR_BE_NOENT:
1542 		(void) fprintf(stderr, _("%s does not exist or appear "
1543 		    "to be a valid BE.\nPlease check that the name of "
1544 		    "the BE provided is correct.\n"), obe_name);
1545 		break;
1546 	case BE_ERR_SS_NOENT:
1547 		(void) fprintf(stderr, _("%s does not exist or appear "
1548 		    "to be a valid snapshot.\nPlease check that the name of "
1549 		    "the snapshot provided is correct.\n"), snap_name);
1550 		break;
1551 	case BE_ERR_PERM:
1552 	case BE_ERR_ACCESS:
1553 		(void) fprintf(stderr, _("Rollback of BE %s snapshot %s "
1554 		    "failed.\n"), obe_name, snap_name);
1555 		(void) fprintf(stderr, _("You have insufficient privileges to "
1556 		    "execute this command.\n"));
1557 		break;
1558 	default:
1559 		(void) fprintf(stderr, _("Rollback of BE %s snapshot %s "
1560 		    "failed.\n"), obe_name, snap_name);
1561 		(void) fprintf(stderr, "%s\n", be_err_to_str(err));
1562 	}
1563 
1564 out:
1565 	nvlist_free(be_attrs);
1566 	return (err);
1567 }
1568