zfs_cmd.c revision 344377 
1/*-
2 * Copyright (c) 2018 Warner Losh <imp@freebd.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD: stable/11/stand/common/zfs_cmd.c 344377 2019-02-20 19:13:09Z kevans $");
29
30/*
31 * MD bootstrap main() and assorted miscellaneous
32 * commands.
33 */
34
35#include <stand.h>
36#include <stddef.h>
37#include <sys/disk.h>
38#include <sys/reboot.h>
39
40#include "bootstrap.h"
41
42#ifdef LOADER_ZFS_SUPPORT
43#include "libzfs.h"
44#endif
45
46COMMAND_SET(lszfs, "lszfs", "list child datasets of a zfs dataset",
47	    command_lszfs);
48
49static int
50command_lszfs(int argc, char *argv[])
51{
52	int err;
53
54	if (argc != 2) {
55		command_errmsg = "a single dataset must be supplied";
56		return (CMD_ERROR);
57	}
58
59	err = zfs_list(argv[1]);
60	if (err != 0) {
61		command_errmsg = strerror(err);
62		return (CMD_ERROR);
63	}
64	return (CMD_OK);
65}
66
67COMMAND_SET(reloadbe, "reloadbe", "refresh the list of ZFS Boot Environments",
68	    command_reloadbe);
69
70static int
71command_reloadbe(int argc, char *argv[])
72{
73	int err;
74	char *root;
75
76	if (argc > 2) {
77		command_errmsg = "wrong number of arguments";
78		return (CMD_ERROR);
79	}
80
81	if (argc == 2) {
82		err = zfs_bootenv(argv[1]);
83	} else {
84		root = getenv("zfs_be_root");
85		if (root == NULL) {
86			/* There does not appear to be a ZFS pool here, exit without error */
87			return (CMD_OK);
88		}
89		err = zfs_bootenv(root);
90	}
91
92	if (err != 0) {
93		command_errmsg = strerror(err);
94		return (CMD_ERROR);
95	}
96
97	return (CMD_OK);
98}
99
100uint64_t
101ldi_get_size(void *priv)
102{
103	int fd = (uintptr_t) priv;
104	uint64_t size;
105
106	ioctl(fd, DIOCGMEDIASIZE, &size);
107	return (size);
108}
109