mirror of
https://git.FreeBSD.org/src.git
synced 2025-01-05 12:56:08 +00:00
bectl(8): Split list functionality out into its own file as well
This commit is contained in:
Kyle Evans
parent
d694059f1e
commit
526ad58e80
Notes:
svn2git
2020-12-20 02:59:44 +00:00
svn path=/projects/bectl/; revision=337369
@ -3,7 +3,7 @@
|
||||
PROG= bectl
|
||||
MAN= bectl.8
|
||||
|
||||
SRCS= bectl.c bectl_jail.c
|
||||
SRCS= bectl.c bectl_jail.c bectl_list.c
|
||||
|
||||
LIBADD+= be
|
||||
LIBADD+= jail
|
||||
|
||||
@ -43,40 +43,12 @@
|
||||
|
||||
#include "bectl.h"
|
||||
|
||||
#define HEADER_BE "BE"
|
||||
#define HEADER_BEPLUS "BE/Dataset/Snapshot"
|
||||
#define HEADER_ACTIVE "Active"
|
||||
#define HEADER_MOUNT "Mountpoint"
|
||||
#define HEADER_SPACE "Space"
|
||||
#define HEADER_CREATED "Created"
|
||||
|
||||
/* Spaces */
|
||||
#define INDENT_INCREMENT 2
|
||||
|
||||
struct printc {
|
||||
int active_colsz_def;
|
||||
int be_colsz;
|
||||
int current_indent;
|
||||
int mount_colsz;
|
||||
int space_colsz;
|
||||
bool script_fmt;
|
||||
bool show_all_datasets;
|
||||
bool show_snaps;
|
||||
bool show_space;
|
||||
};
|
||||
|
||||
static int bectl_cmd_activate(int argc, char *argv[]);
|
||||
static int bectl_cmd_create(int argc, char *argv[]);
|
||||
static int bectl_cmd_destroy(int argc, char *argv[]);
|
||||
static int bectl_cmd_export(int argc, char *argv[]);
|
||||
static int bectl_cmd_import(int argc, char *argv[]);
|
||||
static int bectl_cmd_add(int argc, char *argv[]);
|
||||
static const char *get_origin_props(nvlist_t *dsprops, nvlist_t **originprops);
|
||||
static void print_padding(const char *fval, int colsz, struct printc *pc);
|
||||
static int print_snapshots(const char *dsname, struct printc *pc);
|
||||
static void print_info(const char *name, nvlist_t *dsprops, struct printc *pc);
|
||||
static void print_headers(nvlist_t *props, struct printc *pc);
|
||||
static int bectl_cmd_list(int argc, char *argv[]);
|
||||
static int bectl_cmd_mount(int argc, char *argv[]);
|
||||
static int bectl_cmd_rename(int argc, char *argv[]);
|
||||
static int bectl_cmd_unmount(int argc, char *argv[]);
|
||||
@ -375,357 +347,6 @@ bectl_cmd_destroy(int argc, char *argv[])
|
||||
return (err);
|
||||
}
|
||||
|
||||
/*
|
||||
* Given a set of dataset properties (for a BE dataset), populate originprops
|
||||
* with the origin's properties.
|
||||
*/
|
||||
static const char *
|
||||
get_origin_props(nvlist_t *dsprops, nvlist_t **originprops)
|
||||
{
|
||||
char *propstr;
|
||||
|
||||
if (nvlist_lookup_string(dsprops, "origin", &propstr) == 0) {
|
||||
if (be_prop_list_alloc(originprops) != 0) {
|
||||
fprintf(stderr,
|
||||
"bectl list: failed to allocate origin prop nvlist\n");
|
||||
return (NULL);
|
||||
}
|
||||
if (be_get_dataset_props(be, propstr, *originprops) != 0) {
|
||||
/* XXX TODO: Real errors */
|
||||
fprintf(stderr,
|
||||
"bectl list: failed to fetch origin properties\n");
|
||||
return (NULL);
|
||||
}
|
||||
|
||||
return (propstr);
|
||||
}
|
||||
return (NULL);
|
||||
}
|
||||
|
||||
static void
|
||||
print_padding(const char *fval, int colsz, struct printc *pc)
|
||||
{
|
||||
|
||||
/* -H flag handling; all delimiters/padding are a single tab */
|
||||
if (pc->script_fmt) {
|
||||
printf("\t");
|
||||
return;
|
||||
}
|
||||
|
||||
if (fval != NULL)
|
||||
colsz -= strlen(fval);
|
||||
printf("%*s ", colsz, "");
|
||||
}
|
||||
|
||||
static unsigned long long
|
||||
dataset_space(const char *oname)
|
||||
{
|
||||
unsigned long long space;
|
||||
char *dsname, *propstr, *sep;
|
||||
nvlist_t *dsprops;
|
||||
|
||||
space = 0;
|
||||
dsname = strdup(oname);
|
||||
if (dsname == NULL)
|
||||
return (0);
|
||||
|
||||
/* Truncate snapshot to dataset name, as needed */
|
||||
if ((sep = strchr(dsname, '@')) != NULL)
|
||||
*sep = '\0';
|
||||
|
||||
if (be_prop_list_alloc(&dsprops) != 0) {
|
||||
free(dsname);
|
||||
return (0);
|
||||
}
|
||||
|
||||
if (be_get_dataset_props(be, dsname, dsprops) != 0) {
|
||||
nvlist_free(dsprops);
|
||||
free(dsname);
|
||||
return (0);
|
||||
}
|
||||
|
||||
if (nvlist_lookup_string(dsprops, "used", &propstr) == 0)
|
||||
space = strtoull(propstr, NULL, 10);
|
||||
|
||||
nvlist_free(dsprops);
|
||||
free(dsname);
|
||||
return (space);
|
||||
}
|
||||
|
||||
static int
|
||||
print_snapshots(const char *dsname, struct printc *pc)
|
||||
{
|
||||
nvpair_t *cur;
|
||||
nvlist_t *props, *sprops;
|
||||
|
||||
if (be_prop_list_alloc(&props) != 0) {
|
||||
fprintf(stderr, "bectl list: failed to allocate snapshot nvlist\n");
|
||||
return (1);
|
||||
}
|
||||
if (be_get_dataset_snapshots(be, dsname, props) != 0) {
|
||||
fprintf(stderr, "bectl list: failed to fetch boot ds snapshots\n");
|
||||
return (1);
|
||||
}
|
||||
for (cur = nvlist_next_nvpair(props, NULL); cur != NULL;
|
||||
cur = nvlist_next_nvpair(props, cur)) {
|
||||
nvpair_value_nvlist(cur, &sprops);
|
||||
print_info(nvpair_name(cur), sprops, pc);
|
||||
}
|
||||
return (0);
|
||||
}
|
||||
|
||||
static void
|
||||
print_info(const char *name, nvlist_t *dsprops, struct printc *pc)
|
||||
{
|
||||
#define BUFSZ 64
|
||||
char buf[BUFSZ];
|
||||
unsigned long long ctimenum, space;
|
||||
nvlist_t *originprops;
|
||||
const char *oname;
|
||||
char *dsname, *propstr;
|
||||
int active_colsz;
|
||||
boolean_t active_now, active_reboot;
|
||||
|
||||
dsname = NULL;
|
||||
originprops = NULL;
|
||||
printf("%*s%s", pc->current_indent, "", name);
|
||||
nvlist_lookup_string(dsprops, "dataset", &dsname);
|
||||
|
||||
/* Recurse at the base level if we're breaking info down */
|
||||
if (pc->current_indent == 0 && (pc->show_all_datasets ||
|
||||
pc->show_snaps)) {
|
||||
printf("\n");
|
||||
if (dsname == NULL)
|
||||
/* XXX TODO: Error? */
|
||||
return;
|
||||
/*
|
||||
* Whether we're dealing with -a or -s, we'll always print the
|
||||
* dataset name/information followed by its origin. For -s, we
|
||||
* additionally iterate through all snapshots of this boot
|
||||
* environment and also print their information.
|
||||
*/
|
||||
pc->current_indent += INDENT_INCREMENT;
|
||||
print_info(dsname, dsprops, pc);
|
||||
pc->current_indent += INDENT_INCREMENT;
|
||||
if ((oname = get_origin_props(dsprops, &originprops)) != NULL) {
|
||||
print_info(oname, originprops, pc);
|
||||
nvlist_free(originprops);
|
||||
}
|
||||
|
||||
/* Back up a level; snapshots at the same level as dataset */
|
||||
pc->current_indent -= INDENT_INCREMENT;
|
||||
if (pc->show_snaps)
|
||||
print_snapshots(dsname, pc);
|
||||
pc->current_indent = 0;
|
||||
return;
|
||||
} else
|
||||
print_padding(name, pc->be_colsz - pc->current_indent, pc);
|
||||
|
||||
active_colsz = pc->active_colsz_def;
|
||||
if (nvlist_lookup_boolean_value(dsprops, "active",
|
||||
&active_now) == 0 && active_now) {
|
||||
printf("N");
|
||||
active_colsz--;
|
||||
}
|
||||
if (nvlist_lookup_boolean_value(dsprops, "nextboot",
|
||||
&active_reboot) == 0 && active_reboot) {
|
||||
printf("R");
|
||||
active_colsz--;
|
||||
}
|
||||
if (active_colsz == pc->active_colsz_def) {
|
||||
printf("-");
|
||||
active_colsz--;
|
||||
}
|
||||
print_padding(NULL, active_colsz, pc);
|
||||
if (nvlist_lookup_string(dsprops, "mountpoint", &propstr) == 0){
|
||||
printf("%s", propstr);
|
||||
print_padding(propstr, pc->mount_colsz, pc);
|
||||
} else {
|
||||
printf("%s", "-");
|
||||
print_padding("-", pc->mount_colsz, pc);
|
||||
}
|
||||
|
||||
oname = get_origin_props(dsprops, &originprops);
|
||||
if (nvlist_lookup_string(dsprops, "used", &propstr) == 0) {
|
||||
/*
|
||||
* The space used column is some composition of:
|
||||
* - The "used" property of the dataset
|
||||
* - The "used" property of the origin snapshot (not -a or -s)
|
||||
* - The "used" property of the origin dataset (-D flag only)
|
||||
*
|
||||
* The -D flag is ignored if -a or -s are specified.
|
||||
*/
|
||||
space = strtoull(propstr, NULL, 10);
|
||||
|
||||
if (!pc->show_all_datasets && !pc->show_snaps &&
|
||||
originprops != NULL &&
|
||||
nvlist_lookup_string(originprops, "used", &propstr) == 0)
|
||||
space += strtoull(propstr, NULL, 10);
|
||||
|
||||
if (pc->show_space && oname != NULL)
|
||||
space += dataset_space(oname);
|
||||
|
||||
/* Alas, there's more to it,. */
|
||||
be_nicenum(space, buf, 6);
|
||||
printf("%s", buf);
|
||||
print_padding(buf, pc->space_colsz, pc);
|
||||
} else {
|
||||
printf("-");
|
||||
print_padding("-", pc->space_colsz, pc);
|
||||
}
|
||||
|
||||
if (nvlist_lookup_string(dsprops, "creation", &propstr) == 0) {
|
||||
ctimenum = strtoull(propstr, NULL, 10);
|
||||
strftime(buf, BUFSZ, "%Y-%m-%d %H:%M",
|
||||
localtime((time_t *)&ctimenum));
|
||||
printf("%s", buf);
|
||||
}
|
||||
|
||||
printf("\n");
|
||||
if (originprops != NULL)
|
||||
be_prop_list_free(originprops);
|
||||
#undef BUFSZ
|
||||
}
|
||||
|
||||
static void
|
||||
print_headers(nvlist_t *props, struct printc *pc)
|
||||
{
|
||||
const char *chosen_be_header;
|
||||
nvpair_t *cur;
|
||||
nvlist_t *dsprops;
|
||||
char *propstr;
|
||||
size_t be_maxcol;
|
||||
|
||||
if (pc->show_all_datasets || pc->show_snaps)
|
||||
chosen_be_header = HEADER_BEPLUS;
|
||||
else
|
||||
chosen_be_header = HEADER_BE;
|
||||
be_maxcol = strlen(chosen_be_header);
|
||||
for (cur = nvlist_next_nvpair(props, NULL); cur != NULL;
|
||||
cur = nvlist_next_nvpair(props, cur)) {
|
||||
be_maxcol = MAX(be_maxcol, strlen(nvpair_name(cur)));
|
||||
if (!pc->show_all_datasets && !pc->show_snaps)
|
||||
continue;
|
||||
nvpair_value_nvlist(cur, &dsprops);
|
||||
if (nvlist_lookup_string(dsprops, "dataset", &propstr) != 0)
|
||||
continue;
|
||||
be_maxcol = MAX(be_maxcol, strlen(propstr) + INDENT_INCREMENT);
|
||||
if (nvlist_lookup_string(dsprops, "origin", &propstr) != 0)
|
||||
continue;
|
||||
be_maxcol = MAX(be_maxcol,
|
||||
strlen(propstr) + INDENT_INCREMENT * 2);
|
||||
}
|
||||
|
||||
pc->be_colsz = be_maxcol;
|
||||
pc->active_colsz_def = strlen(HEADER_ACTIVE);
|
||||
pc->mount_colsz = strlen(HEADER_MOUNT);
|
||||
pc->space_colsz = strlen(HEADER_SPACE);
|
||||
printf("%*s %s %s %s %s\n", -pc->be_colsz, chosen_be_header,
|
||||
HEADER_ACTIVE, HEADER_MOUNT, HEADER_SPACE, HEADER_CREATED);
|
||||
|
||||
/*
|
||||
* All other invocations in which we aren't using the default header
|
||||
* will produce quite a bit of input. Throw an extra blank line after
|
||||
* the header to make it look nicer.
|
||||
*/
|
||||
if (chosen_be_header != HEADER_BE)
|
||||
printf("\n");
|
||||
}
|
||||
|
||||
static int
|
||||
bectl_cmd_list(int argc, char *argv[])
|
||||
{
|
||||
struct printc pc;
|
||||
nvpair_t *cur;
|
||||
nvlist_t *dsprops, *props;
|
||||
int opt, printed;
|
||||
boolean_t active_now, active_reboot;
|
||||
|
||||
props = NULL;
|
||||
printed = 0;
|
||||
bzero(&pc, sizeof(pc));
|
||||
while ((opt = getopt(argc, argv, "aDHs")) != -1) {
|
||||
switch (opt) {
|
||||
case 'a':
|
||||
pc.show_all_datasets = true;
|
||||
break;
|
||||
case 'D':
|
||||
pc.show_space = true;
|
||||
break;
|
||||
case 'H':
|
||||
pc.script_fmt = true;
|
||||
break;
|
||||
case 's':
|
||||
pc.show_snaps = true;
|
||||
break;
|
||||
default:
|
||||
fprintf(stderr, "bectl list: unknown option '-%c'\n",
|
||||
optopt);
|
||||
return (usage(false));
|
||||
}
|
||||
}
|
||||
|
||||
argc -= optind;
|
||||
|
||||
if (argc != 0) {
|
||||
fprintf(stderr, "bectl list: extra argument provided\n");
|
||||
return (usage(false));
|
||||
}
|
||||
|
||||
if (be_prop_list_alloc(&props) != 0) {
|
||||
fprintf(stderr, "bectl list: failed to allocate prop nvlist\n");
|
||||
return (1);
|
||||
}
|
||||
if (be_get_bootenv_props(be, props) != 0) {
|
||||
/* XXX TODO: Real errors */
|
||||
fprintf(stderr, "bectl list: failed to fetch boot environments\n");
|
||||
return (1);
|
||||
}
|
||||
|
||||
/* Force -D off if either -a or -s are specified */
|
||||
if (pc.show_all_datasets || pc.show_snaps)
|
||||
pc.show_space = false;
|
||||
if (!pc.script_fmt)
|
||||
print_headers(props, &pc);
|
||||
/* Do a first pass to print active and next active first */
|
||||
for (cur = nvlist_next_nvpair(props, NULL); cur != NULL;
|
||||
cur = nvlist_next_nvpair(props, cur)) {
|
||||
nvpair_value_nvlist(cur, &dsprops);
|
||||
active_now = active_reboot = false;
|
||||
|
||||
nvlist_lookup_boolean_value(dsprops, "active", &active_now);
|
||||
nvlist_lookup_boolean_value(dsprops, "nextboot",
|
||||
&active_reboot);
|
||||
if (!active_now && !active_reboot)
|
||||
continue;
|
||||
if (printed > 0 && (pc.show_all_datasets || pc.show_snaps))
|
||||
printf("\n");
|
||||
print_info(nvpair_name(cur), dsprops, &pc);
|
||||
printed++;
|
||||
}
|
||||
|
||||
/* Now pull everything else */
|
||||
for (cur = nvlist_next_nvpair(props, NULL); cur != NULL;
|
||||
cur = nvlist_next_nvpair(props, cur)) {
|
||||
nvpair_value_nvlist(cur, &dsprops);
|
||||
active_now = active_reboot = false;
|
||||
|
||||
nvlist_lookup_boolean_value(dsprops, "active", &active_now);
|
||||
nvlist_lookup_boolean_value(dsprops, "nextboot",
|
||||
&active_reboot);
|
||||
if (active_now || active_reboot)
|
||||
continue;
|
||||
if (printed > 0 && (pc.show_all_datasets || pc.show_snaps))
|
||||
printf("\n");
|
||||
print_info(nvpair_name(cur), dsprops, &pc);
|
||||
printed++;
|
||||
}
|
||||
be_prop_list_free(props);
|
||||
|
||||
return (0);
|
||||
}
|
||||
|
||||
|
||||
static int
|
||||
bectl_cmd_mount(int argc, char *argv[])
|
||||
{
|
||||
|
||||
@ -32,4 +32,6 @@ int usage(bool explicit);
|
||||
int bectl_cmd_jail(int argc, char *argv[]);
|
||||
int bectl_cmd_unjail(int argc, char *argv[]);
|
||||
|
||||
int bectl_cmd_list(int argc, char *argv[]);
|
||||
|
||||
extern libbe_handle_t *be;
|
||||
|
||||
418
sbin/bectl/bectl_list.c
Normal file
418
sbin/bectl/bectl_list.c
Normal file
@ -0,0 +1,418 @@
|
||||
/*-
|
||||
* SPDX-License-Identifier: BSD-2-Clause-FreeBSD
|
||||
*
|
||||
* Copyright (c) 2018 Kyle Evans <kevans@FreeBSD.org>
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions
|
||||
* are met:
|
||||
* 1. Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright
|
||||
* notice, this list of conditions and the following disclaimer in the
|
||||
* documentation and/or other materials provided with the distribution.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
||||
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
||||
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
||||
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
||||
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
||||
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
||||
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
||||
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
||||
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
||||
* SUCH DAMAGE.
|
||||
*
|
||||
* $FreeBSD$
|
||||
*/
|
||||
|
||||
#include <sys/param.h>
|
||||
#include <stdbool.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include <be.h>
|
||||
|
||||
#include "bectl.h"
|
||||
|
||||
struct printc {
|
||||
int active_colsz_def;
|
||||
int be_colsz;
|
||||
int current_indent;
|
||||
int mount_colsz;
|
||||
int space_colsz;
|
||||
bool script_fmt;
|
||||
bool show_all_datasets;
|
||||
bool show_snaps;
|
||||
bool show_space;
|
||||
};
|
||||
|
||||
static const char *get_origin_props(nvlist_t *dsprops, nvlist_t **originprops);
|
||||
static void print_padding(const char *fval, int colsz, struct printc *pc);
|
||||
static int print_snapshots(const char *dsname, struct printc *pc);
|
||||
static void print_info(const char *name, nvlist_t *dsprops, struct printc *pc);
|
||||
static void print_headers(nvlist_t *props, struct printc *pc);
|
||||
static unsigned long long dataset_space(const char *oname);
|
||||
|
||||
#define HEADER_BE "BE"
|
||||
#define HEADER_BEPLUS "BE/Dataset/Snapshot"
|
||||
#define HEADER_ACTIVE "Active"
|
||||
#define HEADER_MOUNT "Mountpoint"
|
||||
#define HEADER_SPACE "Space"
|
||||
#define HEADER_CREATED "Created"
|
||||
|
||||
/* Spaces */
|
||||
#define INDENT_INCREMENT 2
|
||||
|
||||
/*
|
||||
* Given a set of dataset properties (for a BE dataset), populate originprops
|
||||
* with the origin's properties.
|
||||
*/
|
||||
static const char *
|
||||
get_origin_props(nvlist_t *dsprops, nvlist_t **originprops)
|
||||
{
|
||||
char *propstr;
|
||||
|
||||
if (nvlist_lookup_string(dsprops, "origin", &propstr) == 0) {
|
||||
if (be_prop_list_alloc(originprops) != 0) {
|
||||
fprintf(stderr,
|
||||
"bectl list: failed to allocate origin prop nvlist\n");
|
||||
return (NULL);
|
||||
}
|
||||
if (be_get_dataset_props(be, propstr, *originprops) != 0) {
|
||||
/* XXX TODO: Real errors */
|
||||
fprintf(stderr,
|
||||
"bectl list: failed to fetch origin properties\n");
|
||||
return (NULL);
|
||||
}
|
||||
|
||||
return (propstr);
|
||||
}
|
||||
return (NULL);
|
||||
}
|
||||
|
||||
static void
|
||||
print_padding(const char *fval, int colsz, struct printc *pc)
|
||||
{
|
||||
|
||||
/* -H flag handling; all delimiters/padding are a single tab */
|
||||
if (pc->script_fmt) {
|
||||
printf("\t");
|
||||
return;
|
||||
}
|
||||
|
||||
if (fval != NULL)
|
||||
colsz -= strlen(fval);
|
||||
printf("%*s ", colsz, "");
|
||||
}
|
||||
|
||||
static unsigned long long
|
||||
dataset_space(const char *oname)
|
||||
{
|
||||
unsigned long long space;
|
||||
char *dsname, *propstr, *sep;
|
||||
nvlist_t *dsprops;
|
||||
|
||||
space = 0;
|
||||
dsname = strdup(oname);
|
||||
if (dsname == NULL)
|
||||
return (0);
|
||||
|
||||
/* Truncate snapshot to dataset name, as needed */
|
||||
if ((sep = strchr(dsname, '@')) != NULL)
|
||||
*sep = '\0';
|
||||
|
||||
if (be_prop_list_alloc(&dsprops) != 0) {
|
||||
free(dsname);
|
||||
return (0);
|
||||
}
|
||||
|
||||
if (be_get_dataset_props(be, dsname, dsprops) != 0) {
|
||||
nvlist_free(dsprops);
|
||||
free(dsname);
|
||||
return (0);
|
||||
}
|
||||
|
||||
if (nvlist_lookup_string(dsprops, "used", &propstr) == 0)
|
||||
space = strtoull(propstr, NULL, 10);
|
||||
|
||||
nvlist_free(dsprops);
|
||||
free(dsname);
|
||||
return (space);
|
||||
}
|
||||
|
||||
static int
|
||||
print_snapshots(const char *dsname, struct printc *pc)
|
||||
{
|
||||
nvpair_t *cur;
|
||||
nvlist_t *props, *sprops;
|
||||
|
||||
if (be_prop_list_alloc(&props) != 0) {
|
||||
fprintf(stderr, "bectl list: failed to allocate snapshot nvlist\n");
|
||||
return (1);
|
||||
}
|
||||
if (be_get_dataset_snapshots(be, dsname, props) != 0) {
|
||||
fprintf(stderr, "bectl list: failed to fetch boot ds snapshots\n");
|
||||
return (1);
|
||||
}
|
||||
for (cur = nvlist_next_nvpair(props, NULL); cur != NULL;
|
||||
cur = nvlist_next_nvpair(props, cur)) {
|
||||
nvpair_value_nvlist(cur, &sprops);
|
||||
print_info(nvpair_name(cur), sprops, pc);
|
||||
}
|
||||
return (0);
|
||||
}
|
||||
|
||||
static void
|
||||
print_info(const char *name, nvlist_t *dsprops, struct printc *pc)
|
||||
{
|
||||
#define BUFSZ 64
|
||||
char buf[BUFSZ];
|
||||
unsigned long long ctimenum, space;
|
||||
nvlist_t *originprops;
|
||||
const char *oname;
|
||||
char *dsname, *propstr;
|
||||
int active_colsz;
|
||||
boolean_t active_now, active_reboot;
|
||||
|
||||
dsname = NULL;
|
||||
originprops = NULL;
|
||||
printf("%*s%s", pc->current_indent, "", name);
|
||||
nvlist_lookup_string(dsprops, "dataset", &dsname);
|
||||
|
||||
/* Recurse at the base level if we're breaking info down */
|
||||
if (pc->current_indent == 0 && (pc->show_all_datasets ||
|
||||
pc->show_snaps)) {
|
||||
printf("\n");
|
||||
if (dsname == NULL)
|
||||
/* XXX TODO: Error? */
|
||||
return;
|
||||
/*
|
||||
* Whether we're dealing with -a or -s, we'll always print the
|
||||
* dataset name/information followed by its origin. For -s, we
|
||||
* additionally iterate through all snapshots of this boot
|
||||
* environment and also print their information.
|
||||
*/
|
||||
pc->current_indent += INDENT_INCREMENT;
|
||||
print_info(dsname, dsprops, pc);
|
||||
pc->current_indent += INDENT_INCREMENT;
|
||||
if ((oname = get_origin_props(dsprops, &originprops)) != NULL) {
|
||||
print_info(oname, originprops, pc);
|
||||
nvlist_free(originprops);
|
||||
}
|
||||
|
||||
/* Back up a level; snapshots at the same level as dataset */
|
||||
pc->current_indent -= INDENT_INCREMENT;
|
||||
if (pc->show_snaps)
|
||||
print_snapshots(dsname, pc);
|
||||
pc->current_indent = 0;
|
||||
return;
|
||||
} else
|
||||
print_padding(name, pc->be_colsz - pc->current_indent, pc);
|
||||
|
||||
active_colsz = pc->active_colsz_def;
|
||||
if (nvlist_lookup_boolean_value(dsprops, "active",
|
||||
&active_now) == 0 && active_now) {
|
||||
printf("N");
|
||||
active_colsz--;
|
||||
}
|
||||
if (nvlist_lookup_boolean_value(dsprops, "nextboot",
|
||||
&active_reboot) == 0 && active_reboot) {
|
||||
printf("R");
|
||||
active_colsz--;
|
||||
}
|
||||
if (active_colsz == pc->active_colsz_def) {
|
||||
printf("-");
|
||||
active_colsz--;
|
||||
}
|
||||
print_padding(NULL, active_colsz, pc);
|
||||
if (nvlist_lookup_string(dsprops, "mountpoint", &propstr) == 0){
|
||||
printf("%s", propstr);
|
||||
print_padding(propstr, pc->mount_colsz, pc);
|
||||
} else {
|
||||
printf("%s", "-");
|
||||
print_padding("-", pc->mount_colsz, pc);
|
||||
}
|
||||
|
||||
oname = get_origin_props(dsprops, &originprops);
|
||||
if (nvlist_lookup_string(dsprops, "used", &propstr) == 0) {
|
||||
/*
|
||||
* The space used column is some composition of:
|
||||
* - The "used" property of the dataset
|
||||
* - The "used" property of the origin snapshot (not -a or -s)
|
||||
* - The "used" property of the origin dataset (-D flag only)
|
||||
*
|
||||
* The -D flag is ignored if -a or -s are specified.
|
||||
*/
|
||||
space = strtoull(propstr, NULL, 10);
|
||||
|
||||
if (!pc->show_all_datasets && !pc->show_snaps &&
|
||||
originprops != NULL &&
|
||||
nvlist_lookup_string(originprops, "used", &propstr) == 0)
|
||||
space += strtoull(propstr, NULL, 10);
|
||||
|
||||
if (pc->show_space && oname != NULL)
|
||||
space += dataset_space(oname);
|
||||
|
||||
/* Alas, there's more to it,. */
|
||||
be_nicenum(space, buf, 6);
|
||||
printf("%s", buf);
|
||||
print_padding(buf, pc->space_colsz, pc);
|
||||
} else {
|
||||
printf("-");
|
||||
print_padding("-", pc->space_colsz, pc);
|
||||
}
|
||||
|
||||
if (nvlist_lookup_string(dsprops, "creation", &propstr) == 0) {
|
||||
ctimenum = strtoull(propstr, NULL, 10);
|
||||
strftime(buf, BUFSZ, "%Y-%m-%d %H:%M",
|
||||
localtime((time_t *)&ctimenum));
|
||||
printf("%s", buf);
|
||||
}
|
||||
|
||||
printf("\n");
|
||||
if (originprops != NULL)
|
||||
be_prop_list_free(originprops);
|
||||
#undef BUFSZ
|
||||
}
|
||||
|
||||
static void
|
||||
print_headers(nvlist_t *props, struct printc *pc)
|
||||
{
|
||||
const char *chosen_be_header;
|
||||
nvpair_t *cur;
|
||||
nvlist_t *dsprops;
|
||||
char *propstr;
|
||||
size_t be_maxcol;
|
||||
|
||||
if (pc->show_all_datasets || pc->show_snaps)
|
||||
chosen_be_header = HEADER_BEPLUS;
|
||||
else
|
||||
chosen_be_header = HEADER_BE;
|
||||
be_maxcol = strlen(chosen_be_header);
|
||||
for (cur = nvlist_next_nvpair(props, NULL); cur != NULL;
|
||||
cur = nvlist_next_nvpair(props, cur)) {
|
||||
be_maxcol = MAX(be_maxcol, strlen(nvpair_name(cur)));
|
||||
if (!pc->show_all_datasets && !pc->show_snaps)
|
||||
continue;
|
||||
nvpair_value_nvlist(cur, &dsprops);
|
||||
if (nvlist_lookup_string(dsprops, "dataset", &propstr) != 0)
|
||||
continue;
|
||||
be_maxcol = MAX(be_maxcol, strlen(propstr) + INDENT_INCREMENT);
|
||||
if (nvlist_lookup_string(dsprops, "origin", &propstr) != 0)
|
||||
continue;
|
||||
be_maxcol = MAX(be_maxcol,
|
||||
strlen(propstr) + INDENT_INCREMENT * 2);
|
||||
}
|
||||
|
||||
pc->be_colsz = be_maxcol;
|
||||
pc->active_colsz_def = strlen(HEADER_ACTIVE);
|
||||
pc->mount_colsz = strlen(HEADER_MOUNT);
|
||||
pc->space_colsz = strlen(HEADER_SPACE);
|
||||
printf("%*s %s %s %s %s\n", -pc->be_colsz, chosen_be_header,
|
||||
HEADER_ACTIVE, HEADER_MOUNT, HEADER_SPACE, HEADER_CREATED);
|
||||
|
||||
/*
|
||||
* All other invocations in which we aren't using the default header
|
||||
* will produce quite a bit of input. Throw an extra blank line after
|
||||
* the header to make it look nicer.
|
||||
*/
|
||||
if (chosen_be_header != HEADER_BE)
|
||||
printf("\n");
|
||||
}
|
||||
|
||||
int
|
||||
bectl_cmd_list(int argc, char *argv[])
|
||||
{
|
||||
struct printc pc;
|
||||
nvpair_t *cur;
|
||||
nvlist_t *dsprops, *props;
|
||||
int opt, printed;
|
||||
boolean_t active_now, active_reboot;
|
||||
|
||||
props = NULL;
|
||||
printed = 0;
|
||||
bzero(&pc, sizeof(pc));
|
||||
while ((opt = getopt(argc, argv, "aDHs")) != -1) {
|
||||
switch (opt) {
|
||||
case 'a':
|
||||
pc.show_all_datasets = true;
|
||||
break;
|
||||
case 'D':
|
||||
pc.show_space = true;
|
||||
break;
|
||||
case 'H':
|
||||
pc.script_fmt = true;
|
||||
break;
|
||||
case 's':
|
||||
pc.show_snaps = true;
|
||||
break;
|
||||
default:
|
||||
fprintf(stderr, "bectl list: unknown option '-%c'\n",
|
||||
optopt);
|
||||
return (usage(false));
|
||||
}
|
||||
}
|
||||
|
||||
argc -= optind;
|
||||
|
||||
if (argc != 0) {
|
||||
fprintf(stderr, "bectl list: extra argument provided\n");
|
||||
return (usage(false));
|
||||
}
|
||||
|
||||
if (be_prop_list_alloc(&props) != 0) {
|
||||
fprintf(stderr, "bectl list: failed to allocate prop nvlist\n");
|
||||
return (1);
|
||||
}
|
||||
if (be_get_bootenv_props(be, props) != 0) {
|
||||
/* XXX TODO: Real errors */
|
||||
fprintf(stderr, "bectl list: failed to fetch boot environments\n");
|
||||
return (1);
|
||||
}
|
||||
|
||||
/* Force -D off if either -a or -s are specified */
|
||||
if (pc.show_all_datasets || pc.show_snaps)
|
||||
pc.show_space = false;
|
||||
if (!pc.script_fmt)
|
||||
print_headers(props, &pc);
|
||||
/* Do a first pass to print active and next active first */
|
||||
for (cur = nvlist_next_nvpair(props, NULL); cur != NULL;
|
||||
cur = nvlist_next_nvpair(props, cur)) {
|
||||
nvpair_value_nvlist(cur, &dsprops);
|
||||
active_now = active_reboot = false;
|
||||
|
||||
nvlist_lookup_boolean_value(dsprops, "active", &active_now);
|
||||
nvlist_lookup_boolean_value(dsprops, "nextboot",
|
||||
&active_reboot);
|
||||
if (!active_now && !active_reboot)
|
||||
continue;
|
||||
if (printed > 0 && (pc.show_all_datasets || pc.show_snaps))
|
||||
printf("\n");
|
||||
print_info(nvpair_name(cur), dsprops, &pc);
|
||||
printed++;
|
||||
}
|
||||
|
||||
/* Now pull everything else */
|
||||
for (cur = nvlist_next_nvpair(props, NULL); cur != NULL;
|
||||
cur = nvlist_next_nvpair(props, cur)) {
|
||||
nvpair_value_nvlist(cur, &dsprops);
|
||||
active_now = active_reboot = false;
|
||||
|
||||
nvlist_lookup_boolean_value(dsprops, "active", &active_now);
|
||||
nvlist_lookup_boolean_value(dsprops, "nextboot",
|
||||
&active_reboot);
|
||||
if (active_now || active_reboot)
|
||||
continue;
|
||||
if (printed > 0 && (pc.show_all_datasets || pc.show_snaps))
|
||||
printf("\n");
|
||||
print_info(nvpair_name(cur), dsprops, &pc);
|
||||
printed++;
|
||||
}
|
||||
be_prop_list_free(props);
|
||||
|
||||
return (0);
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user