2930 beadm should sort by creation date
[unleashed.git] / usr / src / cmd / beadm / beadm.c
blobe290b745cd275484eb51b5a7136f27618af812ec
1 /*
2 * CDDL HEADER START
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.
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.
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]
19 * CDDL HEADER END
23 * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
27 * Copyright 2013 Nexenta Systems, Inc. All rights reserved.
28 * Copyright 2015 Toomas Soome <tsoome@me.com>
32 * System includes
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>
45 #include "libbe.h"
47 #ifndef lint
48 #define _(x) gettext(x)
49 #else
50 #define _(x) (x)
51 #endif
53 #ifndef TEXT_DOMAIN
54 #define TEXT_DOMAIN "SYS_TEST"
55 #endif
57 #define DT_BUF_LEN (128)
58 #define NUM_COLS (6)
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);
71 * single column name/width output format description
73 struct col_info {
74 const char *col_name;
75 size_t width;
79 * all columns output format
81 struct hdr_info {
82 struct col_info cols[NUM_COLS];
86 * type of possible output formats
88 enum be_fmt {
89 BE_FMT_DEFAULT,
90 BE_FMT_DATASET,
91 BE_FMT_SNAPSHOT,
92 BE_FMT_ALL
96 * command handler description
98 typedef struct be_command {
99 const char *name;
100 int (*func)(int argc, char **argv);
101 } be_command_t;
104 * sorted list of be commands
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 },
119 static void
120 usage(void)
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"));
145 static int
146 run_be_cmd(const char *cmdname, int argc, char **argv)
148 const be_command_t *command;
150 for (command = &be_command_tbl[0]; command->name != NULL; command++)
151 if (strcmp(command->name, cmdname) == 0)
152 return (command->func(argc, argv));
154 (void) fprintf(stderr, _("Invalid command: %s\n"), cmdname);
155 usage();
156 return (1);
160 main(int argc, char **argv)
162 const char *cmdname;
164 (void) setlocale(LC_ALL, "");
165 (void) textdomain(TEXT_DOMAIN);
167 if (argc < 2) {
168 usage();
169 return (1);
172 cmdname = argv[1];
174 /* Turn error printing off */
175 libbe_print_errors(B_FALSE);
177 return (run_be_cmd(cmdname, --argc, ++argv));
180 static void
181 print_hdr(struct hdr_info *hdr_info)
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;
192 if (first) {
193 (void) printf("%-*s", width, name);
194 first = B_FALSE;
195 } else
196 (void) printf(" %-*s", width, name);
198 (void) putchar('\n');
201 static void
202 init_hdr_cols(enum be_fmt be_fmt, struct hdr_info *hdr)
204 struct col_info *col = hdr->cols;
205 size_t i;
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;
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");
231 for (i = 0; i < NUM_COLS; i++) {
232 const char *name = col[i].col_name;
233 col[i].width = 0;
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);
252 static void
253 nicenum(uint64_t num, char *buf, size_t buflen)
255 uint64_t n = num;
256 int index = 0;
257 char u;
259 while (n >= 1024) {
260 n /= 1024;
261 index++;
264 u = " KMGTPE"[index];
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;
278 static void
279 count_widths(enum be_fmt be_fmt, struct hdr_info *hdr, be_node_list_t *be_nodes)
281 size_t len[NUM_COLS];
282 char buf[DT_BUF_LEN];
283 int i;
284 be_node_list_t *cur_be;
286 for (i = 0; i < NUM_COLS; i++)
287 len[i] = hdr->cols[i].width;
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;
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);
307 (void) strlcpy(name, root_ds, sizeof (name));
308 pos = strstr(name, be_name);
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;
318 if (mntpt_len > len[2])
319 len[2] = mntpt_len;
320 if (policy_len > len[4])
321 len[4] = policy_len;
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);
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;
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;
352 nicenum(used, buf, sizeof (buf));
355 for (i = 0; i < NUM_COLS; i++)
356 hdr->cols[i].width = len[i];
359 static void
360 print_be_nodes(const char *be_name, boolean_t parsable, struct hdr_info *hdr,
361 be_node_list_t *nodes)
363 char buf[64];
364 char datetime[DT_BUF_LEN];
365 be_node_list_t *cur_be;
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;
378 if (be_name != NULL && strcmp(be_name, name) != 0)
379 continue;
381 if (parsable)
382 active[0] = '\0';
384 tm = localtime(&creation);
385 (void) strftime(datetime, DT_BUF_LEN, datetime_fmt, tm);
387 for (snap = cur_be->be_node_snapshots; snap != NULL;
388 snap = snap->be_next_snapshot)
389 used += snap->be_snapshot_space_used;
391 if (!cur_be->be_global_active)
392 active[ai++] = 'x';
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';
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);
425 static void
426 print_be_snapshots(be_node_list_t *be, struct hdr_info *hdr, boolean_t parsable)
428 char buf[64];
429 char datetime[DT_BUF_LEN];
430 be_snapshot_list_t *snap = NULL;
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);
444 (void) strncpy(name, root_ds, sizeof (name));
445 pos = strstr(name, be_name);
446 (void) strcpy(pos, snap_name);
448 (void) strftime(datetime, DT_BUF_LEN, datetime_fmt, tm);
449 nicenum(used, buf, sizeof (buf));
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,
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);
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)
491 char buf[64];
492 char datetime[DT_BUF_LEN];
493 be_node_list_t *cur_be;
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;
505 if (be_name != NULL && strcmp(be_name, name) != 0)
506 continue;
508 if (!parsable)
509 (void) printf("%-s\n", name);
510 else
511 active[0] = '\0';
513 tm = localtime(&creation);
514 (void) strftime(datetime, DT_BUF_LEN, datetime_fmt, tm);
516 if (cur_be->be_active)
517 active[ai++] = 'N';
518 if (cur_be->be_active_on_boot)
519 active[ai] = 'R';
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);
543 if (be_fmt & BE_FMT_SNAPSHOT)
544 print_be_snapshots(cur_be, hdr, parsable);
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)
552 struct hdr_info hdr;
553 enum be_fmt be_fmt = BE_FMT_DEFAULT;
555 if (dsets)
556 be_fmt |= BE_FMT_DATASET;
557 if (snaps)
558 be_fmt |= BE_FMT_SNAPSHOT;
560 if (!parsable) {
561 init_hdr_cols(be_fmt, &hdr);
562 count_widths(be_fmt, &hdr, be_nodes);
563 print_hdr(&hdr);
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);
572 static boolean_t
573 confirm_destroy(const char *name)
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;
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);
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);
596 (void) printf(_("Are you sure you want to destroy %s?\n"
597 "This action cannot be undone (y/[n]): "), name);
599 answer = fgets(buf, sizeof (buf), stdin);
600 if (answer == NULL || *answer == '\0' || *answer == 10)
601 goto out;
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"));
610 out:
611 regfree(&yes_re);
612 regfree(&no_re);
613 return (res);
616 static int
617 be_nvl_alloc(nvlist_t **nvlp)
619 assert(nvlp != NULL);
621 if (nvlist_alloc(nvlp, NV_UNIQUE_NAME, 0) != 0) {
622 (void) perror(_("nvlist_alloc failed.\n"));
623 return (1);
626 return (0);
629 static int
630 be_nvl_add_string(nvlist_t *nvl, const char *name, const char *val)
632 assert(nvl != NULL);
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);
640 return (0);
643 static int
644 be_nvl_add_nvlist(nvlist_t *nvl, const char *name, nvlist_t *val)
646 assert(nvl != NULL);
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);
654 return (0);
657 static int
658 be_nvl_add_uint16(nvlist_t *nvl, const char *name, uint16_t val)
660 assert(nvl != NULL);
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);
668 return (0);
671 static int
672 be_do_activate(int argc, char **argv)
674 nvlist_t *be_attrs;
675 int err = 1;
676 int c;
677 char *obe_name;
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);
690 argc -= optind;
691 argv += optind;
693 if (argc != 1) {
694 usage();
695 return (1);
698 obe_name = argv[0];
700 if (be_nvl_alloc(&be_attrs) != 0)
701 return (1);
703 if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
704 goto out;
706 err = be_activate(be_attrs);
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));
729 out:
730 nvlist_free(be_attrs);
731 return (err);
734 static int
735 be_do_create(int argc, char **argv)
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;
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);
767 propname = optarg;
768 if ((propval = strchr(propname, '=')) == NULL) {
769 (void) fprintf(stderr, _("missing "
770 "'=' for -o option\n"));
771 goto out2;
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;
782 if (be_nvl_add_string(zfs_props, propname, propval)
783 != 0)
784 goto out2;
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;
799 argc -= optind;
800 argv += optind;
802 if (argc != 1) {
803 usage();
804 goto out2;
807 nbe_name = argv[0];
809 if ((snap_name = strrchr(nbe_name, '@')) != NULL) {
810 if (snap_name[1] == '\0') {
811 usage();
812 goto out2;
815 snap_name[0] = '\0';
816 snap_name++;
817 is_snap = B_TRUE;
820 if (obe_name) {
821 if (is_snap) {
822 usage();
823 goto out2;
827 * Check if obe_name is really a snapshot name.
828 * If so, split it out.
830 if ((snap_name = strrchr(obe_name, '@')) != NULL) {
831 if (snap_name[1] == '\0') {
832 usage();
833 goto out2;
836 snap_name[0] = '\0';
837 snap_name++;
839 } else if (is_snap) {
840 obe_name = nbe_name;
841 nbe_name = NULL;
844 if (be_nvl_alloc(&be_attrs) != 0)
845 goto out2;
848 if (zfs_props != NULL && be_nvl_add_nvlist(be_attrs,
849 BE_ATTR_ORIG_BE_NAME, zfs_props) != 0)
850 goto out;
852 if (obe_name != NULL && be_nvl_add_string(be_attrs,
853 BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
854 goto out;
856 if (snap_name != NULL && be_nvl_add_string(be_attrs,
857 BE_ATTR_SNAP_NAME, snap_name) != 0)
858 goto out;
860 if (nbe_zpool != NULL && be_nvl_add_string(be_attrs,
861 BE_ATTR_NEW_BE_POOL, nbe_zpool) != 0)
862 goto out;
864 if (nbe_name != NULL && be_nvl_add_string(be_attrs,
865 BE_ATTR_NEW_BE_NAME, nbe_name) != 0)
866 goto out;
868 if (nbe_desc != NULL && be_nvl_add_string(be_attrs,
869 BE_ATTR_NEW_BE_DESC, nbe_desc) != 0)
870 goto out;
872 if (is_snap)
873 err = be_create_snapshot(be_attrs);
874 else
875 err = be_copy(be_attrs);
877 switch (err) {
878 case BE_SUCCESS:
879 if (!is_snap && !nbe_name) {
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.
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);
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);
904 if (!is_snap && activate) {
905 char *args[] = { "activate", "", NULL };
906 args[1] = nbe_name;
907 optind = 1;
909 err = be_do_activate(2, args);
910 goto out;
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));
945 out:
946 nvlist_free(be_attrs);
947 out2:
948 if (zfs_props != NULL)
949 nvlist_free(zfs_props);
951 return (err);
954 static int
955 be_do_destroy(int argc, char **argv)
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;
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);
986 argc -= optind;
987 argv += optind;
989 if (argc != 1) {
990 usage();
991 return (1);
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);
1000 if ((snap_name = strrchr(be_name, '@')) != NULL) {
1001 if (snap_name[1] == '\0') {
1002 usage();
1003 return (1);
1006 is_snap = B_TRUE;
1007 *snap_name = '\0';
1008 snap_name++;
1011 if (be_nvl_alloc(&be_attrs) != 0)
1012 return (1);
1015 if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, be_name) != 0)
1016 goto out;
1018 if (is_snap) {
1019 if (be_nvl_add_string(be_attrs, BE_ATTR_SNAP_NAME,
1020 snap_name) != 0)
1021 goto out;
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;
1029 err = be_destroy(be_attrs);
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));
1074 out:
1075 nvlist_free(be_attrs);
1076 return (err);
1079 static int
1080 be_do_list(int argc, char **argv)
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;
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);
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;
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;
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;
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);
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);
1155 if (snaps) {
1156 (void) fprintf(stderr, _("Invalid options: -a and %s "
1157 "are mutually exclusive.\n"), "-s");
1158 usage();
1159 return (1);
1162 dsets = B_TRUE;
1163 snaps = B_TRUE;
1166 argc -= optind;
1167 argv += optind;
1170 if (argc == 1)
1171 be_name = argv[0];
1173 err = be_list(be_name, &be_nodes);
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;
1181 if (order != BE_SORT_DATE)
1182 be_sort(&be_nodes, order);
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);
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));
1202 if (be_nodes != NULL)
1203 be_free_list(be_nodes);
1204 return (err);
1207 static int
1208 be_do_mount(int argc, char **argv)
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;
1219 while ((c = getopt(argc, argv, "s:v")) != -1) {
1220 switch (c) {
1221 case 's':
1222 shared_fs = B_TRUE;
1224 mount_flags |= BE_MOUNT_FLAG_SHARED_FS;
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);
1235 break;
1236 case 'v':
1237 libbe_print_errors(B_TRUE);
1238 break;
1239 default:
1240 usage();
1241 return (1);
1245 argc -= optind;
1246 argv += optind;
1248 if (argc < 1 || argc > 2) {
1249 usage();
1250 return (1);
1253 obe_name = argv[0];
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);
1262 } else {
1263 const char *tmpdir = getenv("TMPDIR");
1264 const char *tmpname = "tmp.XXXXXX";
1265 int sz;
1267 if (tmpdir == NULL)
1268 tmpdir = "/tmp";
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);
1277 mountpoint = mkdtemp(tmp_mp);
1280 if (be_nvl_alloc(&be_attrs) != 0)
1281 return (1);
1283 if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
1284 goto out;
1286 if (be_nvl_add_string(be_attrs, BE_ATTR_MOUNTPOINT, mountpoint) != 0)
1287 goto out;
1289 if (shared_fs && be_nvl_add_uint16(be_attrs, BE_ATTR_MOUNT_FLAGS,
1290 mount_flags) != 0)
1291 goto out;
1293 err = be_mount(be_attrs);
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));
1324 out:
1325 if (tmp_mp != NULL)
1326 free(tmp_mp);
1327 nvlist_free(be_attrs);
1328 return (err);
1331 static int
1332 be_do_unmount(int argc, char **argv)
1334 nvlist_t *be_attrs;
1335 char *obe_name;
1336 int err = 1;
1337 int c;
1338 int unmount_flags = 0;
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);
1354 argc -= optind;
1355 argv += optind;
1357 if (argc != 1) {
1358 usage();
1359 return (1);
1362 obe_name = argv[0];
1364 if (be_nvl_alloc(&be_attrs) != 0)
1365 return (1);
1368 if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
1369 goto out;
1371 if (be_nvl_add_uint16(be_attrs, BE_ATTR_UNMOUNT_FLAGS,
1372 unmount_flags) != 0)
1373 goto out;
1375 err = be_unmount(be_attrs);
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));
1406 out:
1407 nvlist_free(be_attrs);
1408 return (err);
1411 static int
1412 be_do_rename(int argc, char **argv)
1414 nvlist_t *be_attrs;
1415 char *obe_name;
1416 char *nbe_name;
1417 int err = 1;
1418 int c;
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);
1431 argc -= optind;
1432 argv += optind;
1434 if (argc != 2) {
1435 usage();
1436 return (1);
1439 obe_name = argv[0];
1440 nbe_name = argv[1];
1442 if (be_nvl_alloc(&be_attrs) != 0)
1443 return (1);
1445 if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
1446 goto out;
1448 if (be_nvl_add_string(be_attrs, BE_ATTR_NEW_BE_NAME, nbe_name) != 0)
1449 goto out;
1451 err = be_rename(be_attrs);
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));
1475 out:
1476 nvlist_free(be_attrs);
1477 return (err);
1480 static int
1481 be_do_rollback(int argc, char **argv)
1483 nvlist_t *be_attrs;
1484 char *obe_name;
1485 char *snap_name;
1486 int err = 1;
1487 int c;
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);
1500 argc -= optind;
1501 argv += optind;
1503 if (argc < 1 || argc > 2) {
1504 usage();
1505 return (1);
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);
1518 snap_name[0] = '\0';
1519 snap_name++;
1520 } else {
1521 usage();
1522 return (1);
1526 if (be_nvl_alloc(&be_attrs) != 0)
1527 return (1);
1529 if (be_nvl_add_string(be_attrs, BE_ATTR_ORIG_BE_NAME, obe_name) != 0)
1530 goto out;
1532 if (be_nvl_add_string(be_attrs, BE_ATTR_SNAP_NAME, snap_name) != 0)
1533 goto out;
1535 err = be_rollback(be_attrs);
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));
1564 out:
1565 nvlist_free(be_attrs);
1566 return (err);