xref: /freebsd-src/sys/contrib/openzfs/lib/libzfs/libzfs_util.c (revision eda14cbc264d6969b02f2b1994cef11148e914f1)
1*eda14cbcSMatt Macy /*
2*eda14cbcSMatt Macy  * CDDL HEADER START
3*eda14cbcSMatt Macy  *
4*eda14cbcSMatt Macy  * The contents of this file are subject to the terms of the
5*eda14cbcSMatt Macy  * Common Development and Distribution License (the "License").
6*eda14cbcSMatt Macy  * You may not use this file except in compliance with the License.
7*eda14cbcSMatt Macy  *
8*eda14cbcSMatt Macy  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9*eda14cbcSMatt Macy  * or http://www.opensolaris.org/os/licensing.
10*eda14cbcSMatt Macy  * See the License for the specific language governing permissions
11*eda14cbcSMatt Macy  * and limitations under the License.
12*eda14cbcSMatt Macy  *
13*eda14cbcSMatt Macy  * When distributing Covered Code, include this CDDL HEADER in each
14*eda14cbcSMatt Macy  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15*eda14cbcSMatt Macy  * If applicable, add the following below this CDDL HEADER, with the
16*eda14cbcSMatt Macy  * fields enclosed by brackets "[]" replaced with your own identifying
17*eda14cbcSMatt Macy  * information: Portions Copyright [yyyy] [name of copyright owner]
18*eda14cbcSMatt Macy  *
19*eda14cbcSMatt Macy  * CDDL HEADER END
20*eda14cbcSMatt Macy  */
21*eda14cbcSMatt Macy 
22*eda14cbcSMatt Macy /*
23*eda14cbcSMatt Macy  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
24*eda14cbcSMatt Macy  * Copyright 2020 Joyent, Inc. All rights reserved.
25*eda14cbcSMatt Macy  * Copyright (c) 2011, 2020 by Delphix. All rights reserved.
26*eda14cbcSMatt Macy  * Copyright 2016 Igor Kozhukhov <ikozhukhov@gmail.com>
27*eda14cbcSMatt Macy  * Copyright (c) 2017 Datto Inc.
28*eda14cbcSMatt Macy  * Copyright (c) 2020 The FreeBSD Foundation
29*eda14cbcSMatt Macy  *
30*eda14cbcSMatt Macy  * Portions of this software were developed by Allan Jude
31*eda14cbcSMatt Macy  * under sponsorship from the FreeBSD Foundation.
32*eda14cbcSMatt Macy  */
33*eda14cbcSMatt Macy 
34*eda14cbcSMatt Macy /*
35*eda14cbcSMatt Macy  * Internal utility routines for the ZFS library.
36*eda14cbcSMatt Macy  */
37*eda14cbcSMatt Macy 
38*eda14cbcSMatt Macy #include <errno.h>
39*eda14cbcSMatt Macy #include <fcntl.h>
40*eda14cbcSMatt Macy #include <libintl.h>
41*eda14cbcSMatt Macy #include <stdarg.h>
42*eda14cbcSMatt Macy #include <stdio.h>
43*eda14cbcSMatt Macy #include <stdlib.h>
44*eda14cbcSMatt Macy #include <strings.h>
45*eda14cbcSMatt Macy #include <unistd.h>
46*eda14cbcSMatt Macy #include <math.h>
47*eda14cbcSMatt Macy #include <sys/stat.h>
48*eda14cbcSMatt Macy #include <sys/mnttab.h>
49*eda14cbcSMatt Macy #include <sys/mntent.h>
50*eda14cbcSMatt Macy #include <sys/types.h>
51*eda14cbcSMatt Macy #include <sys/wait.h>
52*eda14cbcSMatt Macy 
53*eda14cbcSMatt Macy #include <libzfs.h>
54*eda14cbcSMatt Macy #include <libzfs_core.h>
55*eda14cbcSMatt Macy 
56*eda14cbcSMatt Macy #include "libzfs_impl.h"
57*eda14cbcSMatt Macy #include "zfs_prop.h"
58*eda14cbcSMatt Macy #include "zfeature_common.h"
59*eda14cbcSMatt Macy #include <zfs_fletcher.h>
60*eda14cbcSMatt Macy #include <libzutil.h>
61*eda14cbcSMatt Macy 
62*eda14cbcSMatt Macy /*
63*eda14cbcSMatt Macy  * We only care about the scheme in order to match the scheme
64*eda14cbcSMatt Macy  * with the handler. Each handler should validate the full URI
65*eda14cbcSMatt Macy  * as necessary.
66*eda14cbcSMatt Macy  */
67*eda14cbcSMatt Macy #define	URI_REGEX	"^\\([A-Za-z][A-Za-z0-9+.\\-]*\\):"
68*eda14cbcSMatt Macy 
69*eda14cbcSMatt Macy int
70*eda14cbcSMatt Macy libzfs_errno(libzfs_handle_t *hdl)
71*eda14cbcSMatt Macy {
72*eda14cbcSMatt Macy 	return (hdl->libzfs_error);
73*eda14cbcSMatt Macy }
74*eda14cbcSMatt Macy 
75*eda14cbcSMatt Macy const char *
76*eda14cbcSMatt Macy libzfs_error_action(libzfs_handle_t *hdl)
77*eda14cbcSMatt Macy {
78*eda14cbcSMatt Macy 	return (hdl->libzfs_action);
79*eda14cbcSMatt Macy }
80*eda14cbcSMatt Macy 
81*eda14cbcSMatt Macy const char *
82*eda14cbcSMatt Macy libzfs_error_description(libzfs_handle_t *hdl)
83*eda14cbcSMatt Macy {
84*eda14cbcSMatt Macy 	if (hdl->libzfs_desc[0] != '\0')
85*eda14cbcSMatt Macy 		return (hdl->libzfs_desc);
86*eda14cbcSMatt Macy 
87*eda14cbcSMatt Macy 	switch (hdl->libzfs_error) {
88*eda14cbcSMatt Macy 	case EZFS_NOMEM:
89*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "out of memory"));
90*eda14cbcSMatt Macy 	case EZFS_BADPROP:
91*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid property value"));
92*eda14cbcSMatt Macy 	case EZFS_PROPREADONLY:
93*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "read-only property"));
94*eda14cbcSMatt Macy 	case EZFS_PROPTYPE:
95*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "property doesn't apply to "
96*eda14cbcSMatt Macy 		    "datasets of this type"));
97*eda14cbcSMatt Macy 	case EZFS_PROPNONINHERIT:
98*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "property cannot be inherited"));
99*eda14cbcSMatt Macy 	case EZFS_PROPSPACE:
100*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid quota or reservation"));
101*eda14cbcSMatt Macy 	case EZFS_BADTYPE:
102*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "operation not applicable to "
103*eda14cbcSMatt Macy 		    "datasets of this type"));
104*eda14cbcSMatt Macy 	case EZFS_BUSY:
105*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "pool or dataset is busy"));
106*eda14cbcSMatt Macy 	case EZFS_EXISTS:
107*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "pool or dataset exists"));
108*eda14cbcSMatt Macy 	case EZFS_NOENT:
109*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "no such pool or dataset"));
110*eda14cbcSMatt Macy 	case EZFS_BADSTREAM:
111*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid backup stream"));
112*eda14cbcSMatt Macy 	case EZFS_DSREADONLY:
113*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "dataset is read-only"));
114*eda14cbcSMatt Macy 	case EZFS_VOLTOOBIG:
115*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "volume size exceeds limit for "
116*eda14cbcSMatt Macy 		    "this system"));
117*eda14cbcSMatt Macy 	case EZFS_INVALIDNAME:
118*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid name"));
119*eda14cbcSMatt Macy 	case EZFS_BADRESTORE:
120*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "unable to restore to "
121*eda14cbcSMatt Macy 		    "destination"));
122*eda14cbcSMatt Macy 	case EZFS_BADBACKUP:
123*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "backup failed"));
124*eda14cbcSMatt Macy 	case EZFS_BADTARGET:
125*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid target vdev"));
126*eda14cbcSMatt Macy 	case EZFS_NODEVICE:
127*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "no such device in pool"));
128*eda14cbcSMatt Macy 	case EZFS_BADDEV:
129*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid device"));
130*eda14cbcSMatt Macy 	case EZFS_NOREPLICAS:
131*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "no valid replicas"));
132*eda14cbcSMatt Macy 	case EZFS_RESILVERING:
133*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "currently resilvering"));
134*eda14cbcSMatt Macy 	case EZFS_BADVERSION:
135*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "unsupported version or "
136*eda14cbcSMatt Macy 		    "feature"));
137*eda14cbcSMatt Macy 	case EZFS_POOLUNAVAIL:
138*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "pool is unavailable"));
139*eda14cbcSMatt Macy 	case EZFS_DEVOVERFLOW:
140*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "too many devices in one vdev"));
141*eda14cbcSMatt Macy 	case EZFS_BADPATH:
142*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "must be an absolute path"));
143*eda14cbcSMatt Macy 	case EZFS_CROSSTARGET:
144*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "operation crosses datasets or "
145*eda14cbcSMatt Macy 		    "pools"));
146*eda14cbcSMatt Macy 	case EZFS_ZONED:
147*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "dataset in use by local zone"));
148*eda14cbcSMatt Macy 	case EZFS_MOUNTFAILED:
149*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "mount failed"));
150*eda14cbcSMatt Macy 	case EZFS_UMOUNTFAILED:
151*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "umount failed"));
152*eda14cbcSMatt Macy 	case EZFS_UNSHARENFSFAILED:
153*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "unshare(1M) failed"));
154*eda14cbcSMatt Macy 	case EZFS_SHARENFSFAILED:
155*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "share(1M) failed"));
156*eda14cbcSMatt Macy 	case EZFS_UNSHARESMBFAILED:
157*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "smb remove share failed"));
158*eda14cbcSMatt Macy 	case EZFS_SHARESMBFAILED:
159*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "smb add share failed"));
160*eda14cbcSMatt Macy 	case EZFS_PERM:
161*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "permission denied"));
162*eda14cbcSMatt Macy 	case EZFS_NOSPC:
163*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "out of space"));
164*eda14cbcSMatt Macy 	case EZFS_FAULT:
165*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "bad address"));
166*eda14cbcSMatt Macy 	case EZFS_IO:
167*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "I/O error"));
168*eda14cbcSMatt Macy 	case EZFS_INTR:
169*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "signal received"));
170*eda14cbcSMatt Macy 	case EZFS_ISSPARE:
171*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "device is reserved as a hot "
172*eda14cbcSMatt Macy 		    "spare"));
173*eda14cbcSMatt Macy 	case EZFS_INVALCONFIG:
174*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid vdev configuration"));
175*eda14cbcSMatt Macy 	case EZFS_RECURSIVE:
176*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "recursive dataset dependency"));
177*eda14cbcSMatt Macy 	case EZFS_NOHISTORY:
178*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "no history available"));
179*eda14cbcSMatt Macy 	case EZFS_POOLPROPS:
180*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "failed to retrieve "
181*eda14cbcSMatt Macy 		    "pool properties"));
182*eda14cbcSMatt Macy 	case EZFS_POOL_NOTSUP:
183*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "operation not supported "
184*eda14cbcSMatt Macy 		    "on this type of pool"));
185*eda14cbcSMatt Macy 	case EZFS_POOL_INVALARG:
186*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid argument for "
187*eda14cbcSMatt Macy 		    "this pool operation"));
188*eda14cbcSMatt Macy 	case EZFS_NAMETOOLONG:
189*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "dataset name is too long"));
190*eda14cbcSMatt Macy 	case EZFS_OPENFAILED:
191*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "open failed"));
192*eda14cbcSMatt Macy 	case EZFS_NOCAP:
193*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN,
194*eda14cbcSMatt Macy 		    "disk capacity information could not be retrieved"));
195*eda14cbcSMatt Macy 	case EZFS_LABELFAILED:
196*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "write of label failed"));
197*eda14cbcSMatt Macy 	case EZFS_BADWHO:
198*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid user/group"));
199*eda14cbcSMatt Macy 	case EZFS_BADPERM:
200*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid permission"));
201*eda14cbcSMatt Macy 	case EZFS_BADPERMSET:
202*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid permission set name"));
203*eda14cbcSMatt Macy 	case EZFS_NODELEGATION:
204*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "delegated administration is "
205*eda14cbcSMatt Macy 		    "disabled on pool"));
206*eda14cbcSMatt Macy 	case EZFS_BADCACHE:
207*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid or missing cache file"));
208*eda14cbcSMatt Macy 	case EZFS_ISL2CACHE:
209*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "device is in use as a cache"));
210*eda14cbcSMatt Macy 	case EZFS_VDEVNOTSUP:
211*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "vdev specification is not "
212*eda14cbcSMatt Macy 		    "supported"));
213*eda14cbcSMatt Macy 	case EZFS_NOTSUP:
214*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "operation not supported "
215*eda14cbcSMatt Macy 		    "on this dataset"));
216*eda14cbcSMatt Macy 	case EZFS_IOC_NOTSUPPORTED:
217*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "operation not supported by "
218*eda14cbcSMatt Macy 		    "zfs kernel module"));
219*eda14cbcSMatt Macy 	case EZFS_ACTIVE_SPARE:
220*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "pool has active shared spare "
221*eda14cbcSMatt Macy 		    "device"));
222*eda14cbcSMatt Macy 	case EZFS_UNPLAYED_LOGS:
223*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "log device has unplayed intent "
224*eda14cbcSMatt Macy 		    "logs"));
225*eda14cbcSMatt Macy 	case EZFS_REFTAG_RELE:
226*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "no such tag on this dataset"));
227*eda14cbcSMatt Macy 	case EZFS_REFTAG_HOLD:
228*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "tag already exists on this "
229*eda14cbcSMatt Macy 		    "dataset"));
230*eda14cbcSMatt Macy 	case EZFS_TAGTOOLONG:
231*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "tag too long"));
232*eda14cbcSMatt Macy 	case EZFS_PIPEFAILED:
233*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "pipe create failed"));
234*eda14cbcSMatt Macy 	case EZFS_THREADCREATEFAILED:
235*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "thread create failed"));
236*eda14cbcSMatt Macy 	case EZFS_POSTSPLIT_ONLINE:
237*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "disk was split from this pool "
238*eda14cbcSMatt Macy 		    "into a new one"));
239*eda14cbcSMatt Macy 	case EZFS_SCRUB_PAUSED:
240*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "scrub is paused; "
241*eda14cbcSMatt Macy 		    "use 'zpool scrub' to resume"));
242*eda14cbcSMatt Macy 	case EZFS_SCRUBBING:
243*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "currently scrubbing; "
244*eda14cbcSMatt Macy 		    "use 'zpool scrub -s' to cancel current scrub"));
245*eda14cbcSMatt Macy 	case EZFS_NO_SCRUB:
246*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "there is no active scrub"));
247*eda14cbcSMatt Macy 	case EZFS_DIFF:
248*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "unable to generate diffs"));
249*eda14cbcSMatt Macy 	case EZFS_DIFFDATA:
250*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid diff data"));
251*eda14cbcSMatt Macy 	case EZFS_POOLREADONLY:
252*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "pool is read-only"));
253*eda14cbcSMatt Macy 	case EZFS_NO_PENDING:
254*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "operation is not "
255*eda14cbcSMatt Macy 		    "in progress"));
256*eda14cbcSMatt Macy 	case EZFS_CHECKPOINT_EXISTS:
257*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "checkpoint exists"));
258*eda14cbcSMatt Macy 	case EZFS_DISCARDING_CHECKPOINT:
259*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "currently discarding "
260*eda14cbcSMatt Macy 		    "checkpoint"));
261*eda14cbcSMatt Macy 	case EZFS_NO_CHECKPOINT:
262*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "checkpoint does not exist"));
263*eda14cbcSMatt Macy 	case EZFS_DEVRM_IN_PROGRESS:
264*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "device removal in progress"));
265*eda14cbcSMatt Macy 	case EZFS_VDEV_TOO_BIG:
266*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "device exceeds supported size"));
267*eda14cbcSMatt Macy 	case EZFS_ACTIVE_POOL:
268*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "pool is imported on a "
269*eda14cbcSMatt Macy 		    "different host"));
270*eda14cbcSMatt Macy 	case EZFS_CRYPTOFAILED:
271*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "encryption failure"));
272*eda14cbcSMatt Macy 	case EZFS_TOOMANY:
273*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "argument list too long"));
274*eda14cbcSMatt Macy 	case EZFS_INITIALIZING:
275*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "currently initializing"));
276*eda14cbcSMatt Macy 	case EZFS_NO_INITIALIZE:
277*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "there is no active "
278*eda14cbcSMatt Macy 		    "initialization"));
279*eda14cbcSMatt Macy 	case EZFS_WRONG_PARENT:
280*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "invalid parent dataset"));
281*eda14cbcSMatt Macy 	case EZFS_TRIMMING:
282*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "currently trimming"));
283*eda14cbcSMatt Macy 	case EZFS_NO_TRIM:
284*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "there is no active trim"));
285*eda14cbcSMatt Macy 	case EZFS_TRIM_NOTSUP:
286*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "trim operations are not "
287*eda14cbcSMatt Macy 		    "supported by this device"));
288*eda14cbcSMatt Macy 	case EZFS_NO_RESILVER_DEFER:
289*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "this action requires the "
290*eda14cbcSMatt Macy 		    "resilver_defer feature"));
291*eda14cbcSMatt Macy 	case EZFS_EXPORT_IN_PROGRESS:
292*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "pool export in progress"));
293*eda14cbcSMatt Macy 	case EZFS_REBUILDING:
294*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "currently sequentially "
295*eda14cbcSMatt Macy 		    "resilvering"));
296*eda14cbcSMatt Macy 	case EZFS_UNKNOWN:
297*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "unknown error"));
298*eda14cbcSMatt Macy 	default:
299*eda14cbcSMatt Macy 		assert(hdl->libzfs_error == 0);
300*eda14cbcSMatt Macy 		return (dgettext(TEXT_DOMAIN, "no error"));
301*eda14cbcSMatt Macy 	}
302*eda14cbcSMatt Macy }
303*eda14cbcSMatt Macy 
304*eda14cbcSMatt Macy /*PRINTFLIKE2*/
305*eda14cbcSMatt Macy void
306*eda14cbcSMatt Macy zfs_error_aux(libzfs_handle_t *hdl, const char *fmt, ...)
307*eda14cbcSMatt Macy {
308*eda14cbcSMatt Macy 	va_list ap;
309*eda14cbcSMatt Macy 
310*eda14cbcSMatt Macy 	va_start(ap, fmt);
311*eda14cbcSMatt Macy 
312*eda14cbcSMatt Macy 	(void) vsnprintf(hdl->libzfs_desc, sizeof (hdl->libzfs_desc),
313*eda14cbcSMatt Macy 	    fmt, ap);
314*eda14cbcSMatt Macy 	hdl->libzfs_desc_active = 1;
315*eda14cbcSMatt Macy 
316*eda14cbcSMatt Macy 	va_end(ap);
317*eda14cbcSMatt Macy }
318*eda14cbcSMatt Macy 
319*eda14cbcSMatt Macy static void
320*eda14cbcSMatt Macy zfs_verror(libzfs_handle_t *hdl, int error, const char *fmt, va_list ap)
321*eda14cbcSMatt Macy {
322*eda14cbcSMatt Macy 	(void) vsnprintf(hdl->libzfs_action, sizeof (hdl->libzfs_action),
323*eda14cbcSMatt Macy 	    fmt, ap);
324*eda14cbcSMatt Macy 	hdl->libzfs_error = error;
325*eda14cbcSMatt Macy 
326*eda14cbcSMatt Macy 	if (hdl->libzfs_desc_active)
327*eda14cbcSMatt Macy 		hdl->libzfs_desc_active = 0;
328*eda14cbcSMatt Macy 	else
329*eda14cbcSMatt Macy 		hdl->libzfs_desc[0] = '\0';
330*eda14cbcSMatt Macy 
331*eda14cbcSMatt Macy 	if (hdl->libzfs_printerr) {
332*eda14cbcSMatt Macy 		if (error == EZFS_UNKNOWN) {
333*eda14cbcSMatt Macy 			(void) fprintf(stderr, dgettext(TEXT_DOMAIN, "internal "
334*eda14cbcSMatt Macy 			    "error: %s: %s\n"), hdl->libzfs_action,
335*eda14cbcSMatt Macy 			    libzfs_error_description(hdl));
336*eda14cbcSMatt Macy 			abort();
337*eda14cbcSMatt Macy 		}
338*eda14cbcSMatt Macy 
339*eda14cbcSMatt Macy 		(void) fprintf(stderr, "%s: %s\n", hdl->libzfs_action,
340*eda14cbcSMatt Macy 		    libzfs_error_description(hdl));
341*eda14cbcSMatt Macy 		if (error == EZFS_NOMEM)
342*eda14cbcSMatt Macy 			exit(1);
343*eda14cbcSMatt Macy 	}
344*eda14cbcSMatt Macy }
345*eda14cbcSMatt Macy 
346*eda14cbcSMatt Macy int
347*eda14cbcSMatt Macy zfs_error(libzfs_handle_t *hdl, int error, const char *msg)
348*eda14cbcSMatt Macy {
349*eda14cbcSMatt Macy 	return (zfs_error_fmt(hdl, error, "%s", msg));
350*eda14cbcSMatt Macy }
351*eda14cbcSMatt Macy 
352*eda14cbcSMatt Macy /*PRINTFLIKE3*/
353*eda14cbcSMatt Macy int
354*eda14cbcSMatt Macy zfs_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
355*eda14cbcSMatt Macy {
356*eda14cbcSMatt Macy 	va_list ap;
357*eda14cbcSMatt Macy 
358*eda14cbcSMatt Macy 	va_start(ap, fmt);
359*eda14cbcSMatt Macy 
360*eda14cbcSMatt Macy 	zfs_verror(hdl, error, fmt, ap);
361*eda14cbcSMatt Macy 
362*eda14cbcSMatt Macy 	va_end(ap);
363*eda14cbcSMatt Macy 
364*eda14cbcSMatt Macy 	return (-1);
365*eda14cbcSMatt Macy }
366*eda14cbcSMatt Macy 
367*eda14cbcSMatt Macy static int
368*eda14cbcSMatt Macy zfs_common_error(libzfs_handle_t *hdl, int error, const char *fmt,
369*eda14cbcSMatt Macy     va_list ap)
370*eda14cbcSMatt Macy {
371*eda14cbcSMatt Macy 	switch (error) {
372*eda14cbcSMatt Macy 	case EPERM:
373*eda14cbcSMatt Macy 	case EACCES:
374*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_PERM, fmt, ap);
375*eda14cbcSMatt Macy 		return (-1);
376*eda14cbcSMatt Macy 
377*eda14cbcSMatt Macy 	case ECANCELED:
378*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NODELEGATION, fmt, ap);
379*eda14cbcSMatt Macy 		return (-1);
380*eda14cbcSMatt Macy 
381*eda14cbcSMatt Macy 	case EIO:
382*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_IO, fmt, ap);
383*eda14cbcSMatt Macy 		return (-1);
384*eda14cbcSMatt Macy 
385*eda14cbcSMatt Macy 	case EFAULT:
386*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_FAULT, fmt, ap);
387*eda14cbcSMatt Macy 		return (-1);
388*eda14cbcSMatt Macy 
389*eda14cbcSMatt Macy 	case EINTR:
390*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_INTR, fmt, ap);
391*eda14cbcSMatt Macy 		return (-1);
392*eda14cbcSMatt Macy 	}
393*eda14cbcSMatt Macy 
394*eda14cbcSMatt Macy 	return (0);
395*eda14cbcSMatt Macy }
396*eda14cbcSMatt Macy 
397*eda14cbcSMatt Macy int
398*eda14cbcSMatt Macy zfs_standard_error(libzfs_handle_t *hdl, int error, const char *msg)
399*eda14cbcSMatt Macy {
400*eda14cbcSMatt Macy 	return (zfs_standard_error_fmt(hdl, error, "%s", msg));
401*eda14cbcSMatt Macy }
402*eda14cbcSMatt Macy 
403*eda14cbcSMatt Macy /*PRINTFLIKE3*/
404*eda14cbcSMatt Macy int
405*eda14cbcSMatt Macy zfs_standard_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
406*eda14cbcSMatt Macy {
407*eda14cbcSMatt Macy 	va_list ap;
408*eda14cbcSMatt Macy 
409*eda14cbcSMatt Macy 	va_start(ap, fmt);
410*eda14cbcSMatt Macy 
411*eda14cbcSMatt Macy 	if (zfs_common_error(hdl, error, fmt, ap) != 0) {
412*eda14cbcSMatt Macy 		va_end(ap);
413*eda14cbcSMatt Macy 		return (-1);
414*eda14cbcSMatt Macy 	}
415*eda14cbcSMatt Macy 
416*eda14cbcSMatt Macy 	switch (error) {
417*eda14cbcSMatt Macy 	case ENXIO:
418*eda14cbcSMatt Macy 	case ENODEV:
419*eda14cbcSMatt Macy 	case EPIPE:
420*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_IO, fmt, ap);
421*eda14cbcSMatt Macy 		break;
422*eda14cbcSMatt Macy 
423*eda14cbcSMatt Macy 	case ENOENT:
424*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
425*eda14cbcSMatt Macy 		    "dataset does not exist"));
426*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NOENT, fmt, ap);
427*eda14cbcSMatt Macy 		break;
428*eda14cbcSMatt Macy 
429*eda14cbcSMatt Macy 	case ENOSPC:
430*eda14cbcSMatt Macy 	case EDQUOT:
431*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NOSPC, fmt, ap);
432*eda14cbcSMatt Macy 		break;
433*eda14cbcSMatt Macy 
434*eda14cbcSMatt Macy 	case EEXIST:
435*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
436*eda14cbcSMatt Macy 		    "dataset already exists"));
437*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_EXISTS, fmt, ap);
438*eda14cbcSMatt Macy 		break;
439*eda14cbcSMatt Macy 
440*eda14cbcSMatt Macy 	case EBUSY:
441*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
442*eda14cbcSMatt Macy 		    "dataset is busy"));
443*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_BUSY, fmt, ap);
444*eda14cbcSMatt Macy 		break;
445*eda14cbcSMatt Macy 	case EROFS:
446*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_POOLREADONLY, fmt, ap);
447*eda14cbcSMatt Macy 		break;
448*eda14cbcSMatt Macy 	case ENAMETOOLONG:
449*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NAMETOOLONG, fmt, ap);
450*eda14cbcSMatt Macy 		break;
451*eda14cbcSMatt Macy 	case ENOTSUP:
452*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_BADVERSION, fmt, ap);
453*eda14cbcSMatt Macy 		break;
454*eda14cbcSMatt Macy 	case EAGAIN:
455*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
456*eda14cbcSMatt Macy 		    "pool I/O is currently suspended"));
457*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_POOLUNAVAIL, fmt, ap);
458*eda14cbcSMatt Macy 		break;
459*eda14cbcSMatt Macy 	case EREMOTEIO:
460*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_ACTIVE_POOL, fmt, ap);
461*eda14cbcSMatt Macy 		break;
462*eda14cbcSMatt Macy 	case ZFS_ERR_UNKNOWN_SEND_STREAM_FEATURE:
463*eda14cbcSMatt Macy 	case ZFS_ERR_IOC_CMD_UNAVAIL:
464*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "the loaded zfs "
465*eda14cbcSMatt Macy 		    "module does not support this operation. A reboot may "
466*eda14cbcSMatt Macy 		    "be required to enable this operation."));
467*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_IOC_NOTSUPPORTED, fmt, ap);
468*eda14cbcSMatt Macy 		break;
469*eda14cbcSMatt Macy 	case ZFS_ERR_IOC_ARG_UNAVAIL:
470*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "the loaded zfs "
471*eda14cbcSMatt Macy 		    "module does not support an option for this operation. "
472*eda14cbcSMatt Macy 		    "A reboot may be required to enable this option."));
473*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_IOC_NOTSUPPORTED, fmt, ap);
474*eda14cbcSMatt Macy 		break;
475*eda14cbcSMatt Macy 	case ZFS_ERR_IOC_ARG_REQUIRED:
476*eda14cbcSMatt Macy 	case ZFS_ERR_IOC_ARG_BADTYPE:
477*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_IOC_NOTSUPPORTED, fmt, ap);
478*eda14cbcSMatt Macy 		break;
479*eda14cbcSMatt Macy 	case ZFS_ERR_WRONG_PARENT:
480*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_WRONG_PARENT, fmt, ap);
481*eda14cbcSMatt Macy 		break;
482*eda14cbcSMatt Macy 	case ZFS_ERR_BADPROP:
483*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_BADPROP, fmt, ap);
484*eda14cbcSMatt Macy 		break;
485*eda14cbcSMatt Macy 	default:
486*eda14cbcSMatt Macy 		zfs_error_aux(hdl, strerror(error));
487*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_UNKNOWN, fmt, ap);
488*eda14cbcSMatt Macy 		break;
489*eda14cbcSMatt Macy 	}
490*eda14cbcSMatt Macy 
491*eda14cbcSMatt Macy 	va_end(ap);
492*eda14cbcSMatt Macy 	return (-1);
493*eda14cbcSMatt Macy }
494*eda14cbcSMatt Macy 
495*eda14cbcSMatt Macy void
496*eda14cbcSMatt Macy zfs_setprop_error(libzfs_handle_t *hdl, zfs_prop_t prop, int err,
497*eda14cbcSMatt Macy     char *errbuf)
498*eda14cbcSMatt Macy {
499*eda14cbcSMatt Macy 	switch (err) {
500*eda14cbcSMatt Macy 
501*eda14cbcSMatt Macy 	case ENOSPC:
502*eda14cbcSMatt Macy 		/*
503*eda14cbcSMatt Macy 		 * For quotas and reservations, ENOSPC indicates
504*eda14cbcSMatt Macy 		 * something different; setting a quota or reservation
505*eda14cbcSMatt Macy 		 * doesn't use any disk space.
506*eda14cbcSMatt Macy 		 */
507*eda14cbcSMatt Macy 		switch (prop) {
508*eda14cbcSMatt Macy 		case ZFS_PROP_QUOTA:
509*eda14cbcSMatt Macy 		case ZFS_PROP_REFQUOTA:
510*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
511*eda14cbcSMatt Macy 			    "size is less than current used or "
512*eda14cbcSMatt Macy 			    "reserved space"));
513*eda14cbcSMatt Macy 			(void) zfs_error(hdl, EZFS_PROPSPACE, errbuf);
514*eda14cbcSMatt Macy 			break;
515*eda14cbcSMatt Macy 
516*eda14cbcSMatt Macy 		case ZFS_PROP_RESERVATION:
517*eda14cbcSMatt Macy 		case ZFS_PROP_REFRESERVATION:
518*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
519*eda14cbcSMatt Macy 			    "size is greater than available space"));
520*eda14cbcSMatt Macy 			(void) zfs_error(hdl, EZFS_PROPSPACE, errbuf);
521*eda14cbcSMatt Macy 			break;
522*eda14cbcSMatt Macy 
523*eda14cbcSMatt Macy 		default:
524*eda14cbcSMatt Macy 			(void) zfs_standard_error(hdl, err, errbuf);
525*eda14cbcSMatt Macy 			break;
526*eda14cbcSMatt Macy 		}
527*eda14cbcSMatt Macy 		break;
528*eda14cbcSMatt Macy 
529*eda14cbcSMatt Macy 	case EBUSY:
530*eda14cbcSMatt Macy 		(void) zfs_standard_error(hdl, EBUSY, errbuf);
531*eda14cbcSMatt Macy 		break;
532*eda14cbcSMatt Macy 
533*eda14cbcSMatt Macy 	case EROFS:
534*eda14cbcSMatt Macy 		(void) zfs_error(hdl, EZFS_DSREADONLY, errbuf);
535*eda14cbcSMatt Macy 		break;
536*eda14cbcSMatt Macy 
537*eda14cbcSMatt Macy 	case E2BIG:
538*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
539*eda14cbcSMatt Macy 		    "property value too long"));
540*eda14cbcSMatt Macy 		(void) zfs_error(hdl, EZFS_BADPROP, errbuf);
541*eda14cbcSMatt Macy 		break;
542*eda14cbcSMatt Macy 
543*eda14cbcSMatt Macy 	case ENOTSUP:
544*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
545*eda14cbcSMatt Macy 		    "pool and or dataset must be upgraded to set this "
546*eda14cbcSMatt Macy 		    "property or value"));
547*eda14cbcSMatt Macy 		(void) zfs_error(hdl, EZFS_BADVERSION, errbuf);
548*eda14cbcSMatt Macy 		break;
549*eda14cbcSMatt Macy 
550*eda14cbcSMatt Macy 	case ERANGE:
551*eda14cbcSMatt Macy 		if (prop == ZFS_PROP_COMPRESSION ||
552*eda14cbcSMatt Macy 		    prop == ZFS_PROP_DNODESIZE ||
553*eda14cbcSMatt Macy 		    prop == ZFS_PROP_RECORDSIZE) {
554*eda14cbcSMatt Macy 			(void) zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
555*eda14cbcSMatt Macy 			    "property setting is not allowed on "
556*eda14cbcSMatt Macy 			    "bootable datasets"));
557*eda14cbcSMatt Macy 			(void) zfs_error(hdl, EZFS_NOTSUP, errbuf);
558*eda14cbcSMatt Macy 		} else if (prop == ZFS_PROP_CHECKSUM ||
559*eda14cbcSMatt Macy 		    prop == ZFS_PROP_DEDUP) {
560*eda14cbcSMatt Macy 			(void) zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
561*eda14cbcSMatt Macy 			    "property setting is not allowed on "
562*eda14cbcSMatt Macy 			    "root pools"));
563*eda14cbcSMatt Macy 			(void) zfs_error(hdl, EZFS_NOTSUP, errbuf);
564*eda14cbcSMatt Macy 		} else {
565*eda14cbcSMatt Macy 			(void) zfs_standard_error(hdl, err, errbuf);
566*eda14cbcSMatt Macy 		}
567*eda14cbcSMatt Macy 		break;
568*eda14cbcSMatt Macy 
569*eda14cbcSMatt Macy 	case EINVAL:
570*eda14cbcSMatt Macy 		if (prop == ZPROP_INVAL) {
571*eda14cbcSMatt Macy 			(void) zfs_error(hdl, EZFS_BADPROP, errbuf);
572*eda14cbcSMatt Macy 		} else {
573*eda14cbcSMatt Macy 			(void) zfs_standard_error(hdl, err, errbuf);
574*eda14cbcSMatt Macy 		}
575*eda14cbcSMatt Macy 		break;
576*eda14cbcSMatt Macy 
577*eda14cbcSMatt Macy 	case ZFS_ERR_BADPROP:
578*eda14cbcSMatt Macy 		(void) zfs_error(hdl, EZFS_BADPROP, errbuf);
579*eda14cbcSMatt Macy 		break;
580*eda14cbcSMatt Macy 
581*eda14cbcSMatt Macy 	case EACCES:
582*eda14cbcSMatt Macy 		if (prop == ZFS_PROP_KEYLOCATION) {
583*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
584*eda14cbcSMatt Macy 			    "keylocation may only be set on encryption roots"));
585*eda14cbcSMatt Macy 			(void) zfs_error(hdl, EZFS_BADPROP, errbuf);
586*eda14cbcSMatt Macy 		} else {
587*eda14cbcSMatt Macy 			(void) zfs_standard_error(hdl, err, errbuf);
588*eda14cbcSMatt Macy 		}
589*eda14cbcSMatt Macy 		break;
590*eda14cbcSMatt Macy 
591*eda14cbcSMatt Macy 	case EOVERFLOW:
592*eda14cbcSMatt Macy 		/*
593*eda14cbcSMatt Macy 		 * This platform can't address a volume this big.
594*eda14cbcSMatt Macy 		 */
595*eda14cbcSMatt Macy #ifdef _ILP32
596*eda14cbcSMatt Macy 		if (prop == ZFS_PROP_VOLSIZE) {
597*eda14cbcSMatt Macy 			(void) zfs_error(hdl, EZFS_VOLTOOBIG, errbuf);
598*eda14cbcSMatt Macy 			break;
599*eda14cbcSMatt Macy 		}
600*eda14cbcSMatt Macy #endif
601*eda14cbcSMatt Macy 		/* FALLTHROUGH */
602*eda14cbcSMatt Macy 	default:
603*eda14cbcSMatt Macy 		(void) zfs_standard_error(hdl, err, errbuf);
604*eda14cbcSMatt Macy 	}
605*eda14cbcSMatt Macy }
606*eda14cbcSMatt Macy 
607*eda14cbcSMatt Macy int
608*eda14cbcSMatt Macy zpool_standard_error(libzfs_handle_t *hdl, int error, const char *msg)
609*eda14cbcSMatt Macy {
610*eda14cbcSMatt Macy 	return (zpool_standard_error_fmt(hdl, error, "%s", msg));
611*eda14cbcSMatt Macy }
612*eda14cbcSMatt Macy 
613*eda14cbcSMatt Macy /*PRINTFLIKE3*/
614*eda14cbcSMatt Macy int
615*eda14cbcSMatt Macy zpool_standard_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
616*eda14cbcSMatt Macy {
617*eda14cbcSMatt Macy 	va_list ap;
618*eda14cbcSMatt Macy 
619*eda14cbcSMatt Macy 	va_start(ap, fmt);
620*eda14cbcSMatt Macy 
621*eda14cbcSMatt Macy 	if (zfs_common_error(hdl, error, fmt, ap) != 0) {
622*eda14cbcSMatt Macy 		va_end(ap);
623*eda14cbcSMatt Macy 		return (-1);
624*eda14cbcSMatt Macy 	}
625*eda14cbcSMatt Macy 
626*eda14cbcSMatt Macy 	switch (error) {
627*eda14cbcSMatt Macy 	case ENODEV:
628*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NODEVICE, fmt, ap);
629*eda14cbcSMatt Macy 		break;
630*eda14cbcSMatt Macy 
631*eda14cbcSMatt Macy 	case ENOENT:
632*eda14cbcSMatt Macy 		zfs_error_aux(hdl,
633*eda14cbcSMatt Macy 		    dgettext(TEXT_DOMAIN, "no such pool or dataset"));
634*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NOENT, fmt, ap);
635*eda14cbcSMatt Macy 		break;
636*eda14cbcSMatt Macy 
637*eda14cbcSMatt Macy 	case EEXIST:
638*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
639*eda14cbcSMatt Macy 		    "pool already exists"));
640*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_EXISTS, fmt, ap);
641*eda14cbcSMatt Macy 		break;
642*eda14cbcSMatt Macy 
643*eda14cbcSMatt Macy 	case EBUSY:
644*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "pool is busy"));
645*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_BUSY, fmt, ap);
646*eda14cbcSMatt Macy 		break;
647*eda14cbcSMatt Macy 
648*eda14cbcSMatt Macy 	/* There is no pending operation to cancel */
649*eda14cbcSMatt Macy 	case ENOTACTIVE:
650*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NO_PENDING, fmt, ap);
651*eda14cbcSMatt Macy 		break;
652*eda14cbcSMatt Macy 
653*eda14cbcSMatt Macy 	case ENXIO:
654*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
655*eda14cbcSMatt Macy 		    "one or more devices is currently unavailable"));
656*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_BADDEV, fmt, ap);
657*eda14cbcSMatt Macy 		break;
658*eda14cbcSMatt Macy 
659*eda14cbcSMatt Macy 	case ENAMETOOLONG:
660*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_DEVOVERFLOW, fmt, ap);
661*eda14cbcSMatt Macy 		break;
662*eda14cbcSMatt Macy 
663*eda14cbcSMatt Macy 	case ENOTSUP:
664*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_POOL_NOTSUP, fmt, ap);
665*eda14cbcSMatt Macy 		break;
666*eda14cbcSMatt Macy 
667*eda14cbcSMatt Macy 	case EINVAL:
668*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_POOL_INVALARG, fmt, ap);
669*eda14cbcSMatt Macy 		break;
670*eda14cbcSMatt Macy 
671*eda14cbcSMatt Macy 	case ENOSPC:
672*eda14cbcSMatt Macy 	case EDQUOT:
673*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NOSPC, fmt, ap);
674*eda14cbcSMatt Macy 		return (-1);
675*eda14cbcSMatt Macy 
676*eda14cbcSMatt Macy 	case EAGAIN:
677*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
678*eda14cbcSMatt Macy 		    "pool I/O is currently suspended"));
679*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_POOLUNAVAIL, fmt, ap);
680*eda14cbcSMatt Macy 		break;
681*eda14cbcSMatt Macy 
682*eda14cbcSMatt Macy 	case EROFS:
683*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_POOLREADONLY, fmt, ap);
684*eda14cbcSMatt Macy 		break;
685*eda14cbcSMatt Macy 	case EDOM:
686*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
687*eda14cbcSMatt Macy 		    "block size out of range or does not match"));
688*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_BADPROP, fmt, ap);
689*eda14cbcSMatt Macy 		break;
690*eda14cbcSMatt Macy 	case EREMOTEIO:
691*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_ACTIVE_POOL, fmt, ap);
692*eda14cbcSMatt Macy 		break;
693*eda14cbcSMatt Macy 	case ZFS_ERR_CHECKPOINT_EXISTS:
694*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_CHECKPOINT_EXISTS, fmt, ap);
695*eda14cbcSMatt Macy 		break;
696*eda14cbcSMatt Macy 	case ZFS_ERR_DISCARDING_CHECKPOINT:
697*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_DISCARDING_CHECKPOINT, fmt, ap);
698*eda14cbcSMatt Macy 		break;
699*eda14cbcSMatt Macy 	case ZFS_ERR_NO_CHECKPOINT:
700*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_NO_CHECKPOINT, fmt, ap);
701*eda14cbcSMatt Macy 		break;
702*eda14cbcSMatt Macy 	case ZFS_ERR_DEVRM_IN_PROGRESS:
703*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_DEVRM_IN_PROGRESS, fmt, ap);
704*eda14cbcSMatt Macy 		break;
705*eda14cbcSMatt Macy 	case ZFS_ERR_VDEV_TOO_BIG:
706*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_VDEV_TOO_BIG, fmt, ap);
707*eda14cbcSMatt Macy 		break;
708*eda14cbcSMatt Macy 	case ZFS_ERR_EXPORT_IN_PROGRESS:
709*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_EXPORT_IN_PROGRESS, fmt, ap);
710*eda14cbcSMatt Macy 		break;
711*eda14cbcSMatt Macy 	case ZFS_ERR_RESILVER_IN_PROGRESS:
712*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_RESILVERING, fmt, ap);
713*eda14cbcSMatt Macy 		break;
714*eda14cbcSMatt Macy 	case ZFS_ERR_REBUILD_IN_PROGRESS:
715*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_REBUILDING, fmt, ap);
716*eda14cbcSMatt Macy 		break;
717*eda14cbcSMatt Macy 	case ZFS_ERR_BADPROP:
718*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_BADPROP, fmt, ap);
719*eda14cbcSMatt Macy 		break;
720*eda14cbcSMatt Macy 	case ZFS_ERR_IOC_CMD_UNAVAIL:
721*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "the loaded zfs "
722*eda14cbcSMatt Macy 		    "module does not support this operation. A reboot may "
723*eda14cbcSMatt Macy 		    "be required to enable this operation."));
724*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_IOC_NOTSUPPORTED, fmt, ap);
725*eda14cbcSMatt Macy 		break;
726*eda14cbcSMatt Macy 	case ZFS_ERR_IOC_ARG_UNAVAIL:
727*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "the loaded zfs "
728*eda14cbcSMatt Macy 		    "module does not support an option for this operation. "
729*eda14cbcSMatt Macy 		    "A reboot may be required to enable this option."));
730*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_IOC_NOTSUPPORTED, fmt, ap);
731*eda14cbcSMatt Macy 		break;
732*eda14cbcSMatt Macy 	case ZFS_ERR_IOC_ARG_REQUIRED:
733*eda14cbcSMatt Macy 	case ZFS_ERR_IOC_ARG_BADTYPE:
734*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_IOC_NOTSUPPORTED, fmt, ap);
735*eda14cbcSMatt Macy 		break;
736*eda14cbcSMatt Macy 	default:
737*eda14cbcSMatt Macy 		zfs_error_aux(hdl, strerror(error));
738*eda14cbcSMatt Macy 		zfs_verror(hdl, EZFS_UNKNOWN, fmt, ap);
739*eda14cbcSMatt Macy 	}
740*eda14cbcSMatt Macy 
741*eda14cbcSMatt Macy 	va_end(ap);
742*eda14cbcSMatt Macy 	return (-1);
743*eda14cbcSMatt Macy }
744*eda14cbcSMatt Macy 
745*eda14cbcSMatt Macy /*
746*eda14cbcSMatt Macy  * Display an out of memory error message and abort the current program.
747*eda14cbcSMatt Macy  */
748*eda14cbcSMatt Macy int
749*eda14cbcSMatt Macy no_memory(libzfs_handle_t *hdl)
750*eda14cbcSMatt Macy {
751*eda14cbcSMatt Macy 	return (zfs_error(hdl, EZFS_NOMEM, "internal error"));
752*eda14cbcSMatt Macy }
753*eda14cbcSMatt Macy 
754*eda14cbcSMatt Macy /*
755*eda14cbcSMatt Macy  * A safe form of malloc() which will die if the allocation fails.
756*eda14cbcSMatt Macy  */
757*eda14cbcSMatt Macy void *
758*eda14cbcSMatt Macy zfs_alloc(libzfs_handle_t *hdl, size_t size)
759*eda14cbcSMatt Macy {
760*eda14cbcSMatt Macy 	void *data;
761*eda14cbcSMatt Macy 
762*eda14cbcSMatt Macy 	if ((data = calloc(1, size)) == NULL)
763*eda14cbcSMatt Macy 		(void) no_memory(hdl);
764*eda14cbcSMatt Macy 
765*eda14cbcSMatt Macy 	return (data);
766*eda14cbcSMatt Macy }
767*eda14cbcSMatt Macy 
768*eda14cbcSMatt Macy /*
769*eda14cbcSMatt Macy  * A safe form of asprintf() which will die if the allocation fails.
770*eda14cbcSMatt Macy  */
771*eda14cbcSMatt Macy /*PRINTFLIKE2*/
772*eda14cbcSMatt Macy char *
773*eda14cbcSMatt Macy zfs_asprintf(libzfs_handle_t *hdl, const char *fmt, ...)
774*eda14cbcSMatt Macy {
775*eda14cbcSMatt Macy 	va_list ap;
776*eda14cbcSMatt Macy 	char *ret;
777*eda14cbcSMatt Macy 	int err;
778*eda14cbcSMatt Macy 
779*eda14cbcSMatt Macy 	va_start(ap, fmt);
780*eda14cbcSMatt Macy 
781*eda14cbcSMatt Macy 	err = vasprintf(&ret, fmt, ap);
782*eda14cbcSMatt Macy 
783*eda14cbcSMatt Macy 	va_end(ap);
784*eda14cbcSMatt Macy 
785*eda14cbcSMatt Macy 	if (err < 0)
786*eda14cbcSMatt Macy 		(void) no_memory(hdl);
787*eda14cbcSMatt Macy 
788*eda14cbcSMatt Macy 	return (ret);
789*eda14cbcSMatt Macy }
790*eda14cbcSMatt Macy 
791*eda14cbcSMatt Macy /*
792*eda14cbcSMatt Macy  * A safe form of realloc(), which also zeroes newly allocated space.
793*eda14cbcSMatt Macy  */
794*eda14cbcSMatt Macy void *
795*eda14cbcSMatt Macy zfs_realloc(libzfs_handle_t *hdl, void *ptr, size_t oldsize, size_t newsize)
796*eda14cbcSMatt Macy {
797*eda14cbcSMatt Macy 	void *ret;
798*eda14cbcSMatt Macy 
799*eda14cbcSMatt Macy 	if ((ret = realloc(ptr, newsize)) == NULL) {
800*eda14cbcSMatt Macy 		(void) no_memory(hdl);
801*eda14cbcSMatt Macy 		return (NULL);
802*eda14cbcSMatt Macy 	}
803*eda14cbcSMatt Macy 
804*eda14cbcSMatt Macy 	bzero((char *)ret + oldsize, (newsize - oldsize));
805*eda14cbcSMatt Macy 	return (ret);
806*eda14cbcSMatt Macy }
807*eda14cbcSMatt Macy 
808*eda14cbcSMatt Macy /*
809*eda14cbcSMatt Macy  * A safe form of strdup() which will die if the allocation fails.
810*eda14cbcSMatt Macy  */
811*eda14cbcSMatt Macy char *
812*eda14cbcSMatt Macy zfs_strdup(libzfs_handle_t *hdl, const char *str)
813*eda14cbcSMatt Macy {
814*eda14cbcSMatt Macy 	char *ret;
815*eda14cbcSMatt Macy 
816*eda14cbcSMatt Macy 	if ((ret = strdup(str)) == NULL)
817*eda14cbcSMatt Macy 		(void) no_memory(hdl);
818*eda14cbcSMatt Macy 
819*eda14cbcSMatt Macy 	return (ret);
820*eda14cbcSMatt Macy }
821*eda14cbcSMatt Macy 
822*eda14cbcSMatt Macy void
823*eda14cbcSMatt Macy libzfs_print_on_error(libzfs_handle_t *hdl, boolean_t printerr)
824*eda14cbcSMatt Macy {
825*eda14cbcSMatt Macy 	hdl->libzfs_printerr = printerr;
826*eda14cbcSMatt Macy }
827*eda14cbcSMatt Macy 
828*eda14cbcSMatt Macy /*
829*eda14cbcSMatt Macy  * Read lines from an open file descriptor and store them in an array of
830*eda14cbcSMatt Macy  * strings until EOF.  lines[] will be allocated and populated with all the
831*eda14cbcSMatt Macy  * lines read.  All newlines are replaced with NULL terminators for
832*eda14cbcSMatt Macy  * convenience.  lines[] must be freed after use with libzfs_free_str_array().
833*eda14cbcSMatt Macy  *
834*eda14cbcSMatt Macy  * Returns the number of lines read.
835*eda14cbcSMatt Macy  */
836*eda14cbcSMatt Macy static int
837*eda14cbcSMatt Macy libzfs_read_stdout_from_fd(int fd, char **lines[])
838*eda14cbcSMatt Macy {
839*eda14cbcSMatt Macy 
840*eda14cbcSMatt Macy 	FILE *fp;
841*eda14cbcSMatt Macy 	int lines_cnt = 0;
842*eda14cbcSMatt Macy 	size_t len = 0;
843*eda14cbcSMatt Macy 	char *line = NULL;
844*eda14cbcSMatt Macy 	char **tmp_lines = NULL, **tmp;
845*eda14cbcSMatt Macy 	char *nl = NULL;
846*eda14cbcSMatt Macy 	int rc;
847*eda14cbcSMatt Macy 
848*eda14cbcSMatt Macy 	fp = fdopen(fd, "r");
849*eda14cbcSMatt Macy 	if (fp == NULL)
850*eda14cbcSMatt Macy 		return (0);
851*eda14cbcSMatt Macy 	while (1) {
852*eda14cbcSMatt Macy 		rc = getline(&line, &len, fp);
853*eda14cbcSMatt Macy 		if (rc == -1)
854*eda14cbcSMatt Macy 			break;
855*eda14cbcSMatt Macy 
856*eda14cbcSMatt Macy 		tmp = realloc(tmp_lines, sizeof (*tmp_lines) * (lines_cnt + 1));
857*eda14cbcSMatt Macy 		if (tmp == NULL) {
858*eda14cbcSMatt Macy 			/* Return the lines we were able to process */
859*eda14cbcSMatt Macy 			break;
860*eda14cbcSMatt Macy 		}
861*eda14cbcSMatt Macy 		tmp_lines = tmp;
862*eda14cbcSMatt Macy 
863*eda14cbcSMatt Macy 		/* Terminate newlines */
864*eda14cbcSMatt Macy 		if ((nl = strchr(line, '\n')) != NULL)
865*eda14cbcSMatt Macy 			*nl = '\0';
866*eda14cbcSMatt Macy 		tmp_lines[lines_cnt] = line;
867*eda14cbcSMatt Macy 		lines_cnt++;
868*eda14cbcSMatt Macy 		line = NULL;
869*eda14cbcSMatt Macy 	}
870*eda14cbcSMatt Macy 	fclose(fp);
871*eda14cbcSMatt Macy 	*lines = tmp_lines;
872*eda14cbcSMatt Macy 	return (lines_cnt);
873*eda14cbcSMatt Macy }
874*eda14cbcSMatt Macy 
875*eda14cbcSMatt Macy static int
876*eda14cbcSMatt Macy libzfs_run_process_impl(const char *path, char *argv[], char *env[], int flags,
877*eda14cbcSMatt Macy     char **lines[], int *lines_cnt)
878*eda14cbcSMatt Macy {
879*eda14cbcSMatt Macy 	pid_t pid;
880*eda14cbcSMatt Macy 	int error, devnull_fd;
881*eda14cbcSMatt Macy 	int link[2];
882*eda14cbcSMatt Macy 
883*eda14cbcSMatt Macy 	/*
884*eda14cbcSMatt Macy 	 * Setup a pipe between our child and parent process if we're
885*eda14cbcSMatt Macy 	 * reading stdout.
886*eda14cbcSMatt Macy 	 */
887*eda14cbcSMatt Macy 	if ((lines != NULL) && pipe(link) == -1)
888*eda14cbcSMatt Macy 		return (-EPIPE);
889*eda14cbcSMatt Macy 
890*eda14cbcSMatt Macy 	pid = vfork();
891*eda14cbcSMatt Macy 	if (pid == 0) {
892*eda14cbcSMatt Macy 		/* Child process */
893*eda14cbcSMatt Macy 		devnull_fd = open("/dev/null", O_WRONLY);
894*eda14cbcSMatt Macy 
895*eda14cbcSMatt Macy 		if (devnull_fd < 0)
896*eda14cbcSMatt Macy 			_exit(-1);
897*eda14cbcSMatt Macy 
898*eda14cbcSMatt Macy 		if (!(flags & STDOUT_VERBOSE) && (lines == NULL))
899*eda14cbcSMatt Macy 			(void) dup2(devnull_fd, STDOUT_FILENO);
900*eda14cbcSMatt Macy 		else if (lines != NULL) {
901*eda14cbcSMatt Macy 			/* Save the output to lines[] */
902*eda14cbcSMatt Macy 			dup2(link[1], STDOUT_FILENO);
903*eda14cbcSMatt Macy 			close(link[0]);
904*eda14cbcSMatt Macy 			close(link[1]);
905*eda14cbcSMatt Macy 		}
906*eda14cbcSMatt Macy 
907*eda14cbcSMatt Macy 		if (!(flags & STDERR_VERBOSE))
908*eda14cbcSMatt Macy 			(void) dup2(devnull_fd, STDERR_FILENO);
909*eda14cbcSMatt Macy 
910*eda14cbcSMatt Macy 		close(devnull_fd);
911*eda14cbcSMatt Macy 
912*eda14cbcSMatt Macy 		if (flags & NO_DEFAULT_PATH) {
913*eda14cbcSMatt Macy 			if (env == NULL)
914*eda14cbcSMatt Macy 				execv(path, argv);
915*eda14cbcSMatt Macy 			else
916*eda14cbcSMatt Macy 				execve(path, argv, env);
917*eda14cbcSMatt Macy 		} else {
918*eda14cbcSMatt Macy 			if (env == NULL)
919*eda14cbcSMatt Macy 				execvp(path, argv);
920*eda14cbcSMatt Macy 			else
921*eda14cbcSMatt Macy 				execvpe(path, argv, env);
922*eda14cbcSMatt Macy 		}
923*eda14cbcSMatt Macy 
924*eda14cbcSMatt Macy 		_exit(-1);
925*eda14cbcSMatt Macy 	} else if (pid > 0) {
926*eda14cbcSMatt Macy 		/* Parent process */
927*eda14cbcSMatt Macy 		int status;
928*eda14cbcSMatt Macy 
929*eda14cbcSMatt Macy 		while ((error = waitpid(pid, &status, 0)) == -1 &&
930*eda14cbcSMatt Macy 		    errno == EINTR) { }
931*eda14cbcSMatt Macy 		if (error < 0 || !WIFEXITED(status))
932*eda14cbcSMatt Macy 			return (-1);
933*eda14cbcSMatt Macy 
934*eda14cbcSMatt Macy 		if (lines != NULL) {
935*eda14cbcSMatt Macy 			close(link[1]);
936*eda14cbcSMatt Macy 			*lines_cnt = libzfs_read_stdout_from_fd(link[0], lines);
937*eda14cbcSMatt Macy 		}
938*eda14cbcSMatt Macy 		return (WEXITSTATUS(status));
939*eda14cbcSMatt Macy 	}
940*eda14cbcSMatt Macy 
941*eda14cbcSMatt Macy 	return (-1);
942*eda14cbcSMatt Macy }
943*eda14cbcSMatt Macy 
944*eda14cbcSMatt Macy int
945*eda14cbcSMatt Macy libzfs_run_process(const char *path, char *argv[], int flags)
946*eda14cbcSMatt Macy {
947*eda14cbcSMatt Macy 	return (libzfs_run_process_impl(path, argv, NULL, flags, NULL, NULL));
948*eda14cbcSMatt Macy }
949*eda14cbcSMatt Macy 
950*eda14cbcSMatt Macy /*
951*eda14cbcSMatt Macy  * Run a command and store its stdout lines in an array of strings (lines[]).
952*eda14cbcSMatt Macy  * lines[] is allocated and populated for you, and the number of lines is set in
953*eda14cbcSMatt Macy  * lines_cnt.  lines[] must be freed after use with libzfs_free_str_array().
954*eda14cbcSMatt Macy  * All newlines (\n) in lines[] are terminated for convenience.
955*eda14cbcSMatt Macy  */
956*eda14cbcSMatt Macy int
957*eda14cbcSMatt Macy libzfs_run_process_get_stdout(const char *path, char *argv[], char *env[],
958*eda14cbcSMatt Macy     char **lines[], int *lines_cnt)
959*eda14cbcSMatt Macy {
960*eda14cbcSMatt Macy 	return (libzfs_run_process_impl(path, argv, env, 0, lines, lines_cnt));
961*eda14cbcSMatt Macy }
962*eda14cbcSMatt Macy 
963*eda14cbcSMatt Macy /*
964*eda14cbcSMatt Macy  * Same as libzfs_run_process_get_stdout(), but run without $PATH set.  This
965*eda14cbcSMatt Macy  * means that *path needs to be the full path to the executable.
966*eda14cbcSMatt Macy  */
967*eda14cbcSMatt Macy int
968*eda14cbcSMatt Macy libzfs_run_process_get_stdout_nopath(const char *path, char *argv[],
969*eda14cbcSMatt Macy     char *env[], char **lines[], int *lines_cnt)
970*eda14cbcSMatt Macy {
971*eda14cbcSMatt Macy 	return (libzfs_run_process_impl(path, argv, env, NO_DEFAULT_PATH,
972*eda14cbcSMatt Macy 	    lines, lines_cnt));
973*eda14cbcSMatt Macy }
974*eda14cbcSMatt Macy 
975*eda14cbcSMatt Macy /*
976*eda14cbcSMatt Macy  * Free an array of strings.  Free both the strings contained in the array and
977*eda14cbcSMatt Macy  * the array itself.
978*eda14cbcSMatt Macy  */
979*eda14cbcSMatt Macy void
980*eda14cbcSMatt Macy libzfs_free_str_array(char **strs, int count)
981*eda14cbcSMatt Macy {
982*eda14cbcSMatt Macy 	while (--count >= 0)
983*eda14cbcSMatt Macy 		free(strs[count]);
984*eda14cbcSMatt Macy 
985*eda14cbcSMatt Macy 	free(strs);
986*eda14cbcSMatt Macy }
987*eda14cbcSMatt Macy 
988*eda14cbcSMatt Macy /*
989*eda14cbcSMatt Macy  * Returns 1 if environment variable is set to "YES", "yes", "ON", "on", or
990*eda14cbcSMatt Macy  * a non-zero number.
991*eda14cbcSMatt Macy  *
992*eda14cbcSMatt Macy  * Returns 0 otherwise.
993*eda14cbcSMatt Macy  */
994*eda14cbcSMatt Macy int
995*eda14cbcSMatt Macy libzfs_envvar_is_set(char *envvar)
996*eda14cbcSMatt Macy {
997*eda14cbcSMatt Macy 	char *env = getenv(envvar);
998*eda14cbcSMatt Macy 	if (env && (strtoul(env, NULL, 0) > 0 ||
999*eda14cbcSMatt Macy 	    (!strncasecmp(env, "YES", 3) && strnlen(env, 4) == 3) ||
1000*eda14cbcSMatt Macy 	    (!strncasecmp(env, "ON", 2) && strnlen(env, 3) == 2)))
1001*eda14cbcSMatt Macy 		return (1);
1002*eda14cbcSMatt Macy 
1003*eda14cbcSMatt Macy 	return (0);
1004*eda14cbcSMatt Macy }
1005*eda14cbcSMatt Macy 
1006*eda14cbcSMatt Macy libzfs_handle_t *
1007*eda14cbcSMatt Macy libzfs_init(void)
1008*eda14cbcSMatt Macy {
1009*eda14cbcSMatt Macy 	libzfs_handle_t *hdl;
1010*eda14cbcSMatt Macy 	int error;
1011*eda14cbcSMatt Macy 
1012*eda14cbcSMatt Macy 	error = libzfs_load_module();
1013*eda14cbcSMatt Macy 	if (error) {
1014*eda14cbcSMatt Macy 		errno = error;
1015*eda14cbcSMatt Macy 		return (NULL);
1016*eda14cbcSMatt Macy 	}
1017*eda14cbcSMatt Macy 
1018*eda14cbcSMatt Macy 	if ((hdl = calloc(1, sizeof (libzfs_handle_t))) == NULL) {
1019*eda14cbcSMatt Macy 		return (NULL);
1020*eda14cbcSMatt Macy 	}
1021*eda14cbcSMatt Macy 
1022*eda14cbcSMatt Macy 	if (regcomp(&hdl->libzfs_urire, URI_REGEX, 0) != 0) {
1023*eda14cbcSMatt Macy 		free(hdl);
1024*eda14cbcSMatt Macy 		return (NULL);
1025*eda14cbcSMatt Macy 	}
1026*eda14cbcSMatt Macy 
1027*eda14cbcSMatt Macy 	if ((hdl->libzfs_fd = open(ZFS_DEV, O_RDWR|O_EXCL)) < 0) {
1028*eda14cbcSMatt Macy 		free(hdl);
1029*eda14cbcSMatt Macy 		return (NULL);
1030*eda14cbcSMatt Macy 	}
1031*eda14cbcSMatt Macy 
1032*eda14cbcSMatt Macy #ifdef HAVE_SETMNTENT
1033*eda14cbcSMatt Macy 	if ((hdl->libzfs_mnttab = setmntent(MNTTAB, "r")) == NULL) {
1034*eda14cbcSMatt Macy #else
1035*eda14cbcSMatt Macy 	if ((hdl->libzfs_mnttab = fopen(MNTTAB, "r")) == NULL) {
1036*eda14cbcSMatt Macy #endif
1037*eda14cbcSMatt Macy 		(void) close(hdl->libzfs_fd);
1038*eda14cbcSMatt Macy 		free(hdl);
1039*eda14cbcSMatt Macy 		return (NULL);
1040*eda14cbcSMatt Macy 	}
1041*eda14cbcSMatt Macy 
1042*eda14cbcSMatt Macy 	if (libzfs_core_init() != 0) {
1043*eda14cbcSMatt Macy 		(void) close(hdl->libzfs_fd);
1044*eda14cbcSMatt Macy 		(void) fclose(hdl->libzfs_mnttab);
1045*eda14cbcSMatt Macy 		free(hdl);
1046*eda14cbcSMatt Macy 		return (NULL);
1047*eda14cbcSMatt Macy 	}
1048*eda14cbcSMatt Macy 
1049*eda14cbcSMatt Macy 	zfs_prop_init();
1050*eda14cbcSMatt Macy 	zpool_prop_init();
1051*eda14cbcSMatt Macy 	zpool_feature_init();
1052*eda14cbcSMatt Macy 	libzfs_mnttab_init(hdl);
1053*eda14cbcSMatt Macy 	fletcher_4_init();
1054*eda14cbcSMatt Macy 
1055*eda14cbcSMatt Macy 	if (getenv("ZFS_PROP_DEBUG") != NULL) {
1056*eda14cbcSMatt Macy 		hdl->libzfs_prop_debug = B_TRUE;
1057*eda14cbcSMatt Macy 	}
1058*eda14cbcSMatt Macy 
1059*eda14cbcSMatt Macy 	/*
1060*eda14cbcSMatt Macy 	 * For testing, remove some settable properties and features
1061*eda14cbcSMatt Macy 	 */
1062*eda14cbcSMatt Macy 	if (libzfs_envvar_is_set("ZFS_SYSFS_PROP_SUPPORT_TEST")) {
1063*eda14cbcSMatt Macy 		zprop_desc_t *proptbl;
1064*eda14cbcSMatt Macy 
1065*eda14cbcSMatt Macy 		proptbl = zpool_prop_get_table();
1066*eda14cbcSMatt Macy 		proptbl[ZPOOL_PROP_COMMENT].pd_zfs_mod_supported = B_FALSE;
1067*eda14cbcSMatt Macy 
1068*eda14cbcSMatt Macy 		proptbl = zfs_prop_get_table();
1069*eda14cbcSMatt Macy 		proptbl[ZFS_PROP_DNODESIZE].pd_zfs_mod_supported = B_FALSE;
1070*eda14cbcSMatt Macy 
1071*eda14cbcSMatt Macy 		zfeature_info_t *ftbl = spa_feature_table;
1072*eda14cbcSMatt Macy 		ftbl[SPA_FEATURE_LARGE_BLOCKS].fi_zfs_mod_supported = B_FALSE;
1073*eda14cbcSMatt Macy 	}
1074*eda14cbcSMatt Macy 
1075*eda14cbcSMatt Macy 	return (hdl);
1076*eda14cbcSMatt Macy }
1077*eda14cbcSMatt Macy 
1078*eda14cbcSMatt Macy void
1079*eda14cbcSMatt Macy libzfs_fini(libzfs_handle_t *hdl)
1080*eda14cbcSMatt Macy {
1081*eda14cbcSMatt Macy 	(void) close(hdl->libzfs_fd);
1082*eda14cbcSMatt Macy 	if (hdl->libzfs_mnttab)
1083*eda14cbcSMatt Macy #ifdef HAVE_SETMNTENT
1084*eda14cbcSMatt Macy 		(void) endmntent(hdl->libzfs_mnttab);
1085*eda14cbcSMatt Macy #else
1086*eda14cbcSMatt Macy 		(void) fclose(hdl->libzfs_mnttab);
1087*eda14cbcSMatt Macy #endif
1088*eda14cbcSMatt Macy 	zpool_free_handles(hdl);
1089*eda14cbcSMatt Macy 	namespace_clear(hdl);
1090*eda14cbcSMatt Macy 	libzfs_mnttab_fini(hdl);
1091*eda14cbcSMatt Macy 	libzfs_core_fini();
1092*eda14cbcSMatt Macy 	regfree(&hdl->libzfs_urire);
1093*eda14cbcSMatt Macy 	fletcher_4_fini();
1094*eda14cbcSMatt Macy 	free(hdl);
1095*eda14cbcSMatt Macy }
1096*eda14cbcSMatt Macy 
1097*eda14cbcSMatt Macy libzfs_handle_t *
1098*eda14cbcSMatt Macy zpool_get_handle(zpool_handle_t *zhp)
1099*eda14cbcSMatt Macy {
1100*eda14cbcSMatt Macy 	return (zhp->zpool_hdl);
1101*eda14cbcSMatt Macy }
1102*eda14cbcSMatt Macy 
1103*eda14cbcSMatt Macy libzfs_handle_t *
1104*eda14cbcSMatt Macy zfs_get_handle(zfs_handle_t *zhp)
1105*eda14cbcSMatt Macy {
1106*eda14cbcSMatt Macy 	return (zhp->zfs_hdl);
1107*eda14cbcSMatt Macy }
1108*eda14cbcSMatt Macy 
1109*eda14cbcSMatt Macy zpool_handle_t *
1110*eda14cbcSMatt Macy zfs_get_pool_handle(const zfs_handle_t *zhp)
1111*eda14cbcSMatt Macy {
1112*eda14cbcSMatt Macy 	return (zhp->zpool_hdl);
1113*eda14cbcSMatt Macy }
1114*eda14cbcSMatt Macy 
1115*eda14cbcSMatt Macy /*
1116*eda14cbcSMatt Macy  * Given a name, determine whether or not it's a valid path
1117*eda14cbcSMatt Macy  * (starts with '/' or "./").  If so, walk the mnttab trying
1118*eda14cbcSMatt Macy  * to match the device number.  If not, treat the path as an
1119*eda14cbcSMatt Macy  * fs/vol/snap/bkmark name.
1120*eda14cbcSMatt Macy  */
1121*eda14cbcSMatt Macy zfs_handle_t *
1122*eda14cbcSMatt Macy zfs_path_to_zhandle(libzfs_handle_t *hdl, const char *path, zfs_type_t argtype)
1123*eda14cbcSMatt Macy {
1124*eda14cbcSMatt Macy 	struct stat64 statbuf;
1125*eda14cbcSMatt Macy 	struct extmnttab entry;
1126*eda14cbcSMatt Macy 
1127*eda14cbcSMatt Macy 	if (path[0] != '/' && strncmp(path, "./", strlen("./")) != 0) {
1128*eda14cbcSMatt Macy 		/*
1129*eda14cbcSMatt Macy 		 * It's not a valid path, assume it's a name of type 'argtype'.
1130*eda14cbcSMatt Macy 		 */
1131*eda14cbcSMatt Macy 		return (zfs_open(hdl, path, argtype));
1132*eda14cbcSMatt Macy 	}
1133*eda14cbcSMatt Macy 
1134*eda14cbcSMatt Macy 	/* Reopen MNTTAB to prevent reading stale data from open file */
1135*eda14cbcSMatt Macy 	if (freopen(MNTTAB, "r", hdl->libzfs_mnttab) == NULL)
1136*eda14cbcSMatt Macy 		return (NULL);
1137*eda14cbcSMatt Macy 
1138*eda14cbcSMatt Macy 	if (getextmntent(path, &entry, &statbuf) != 0)
1139*eda14cbcSMatt Macy 		return (NULL);
1140*eda14cbcSMatt Macy 
1141*eda14cbcSMatt Macy 	if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0) {
1142*eda14cbcSMatt Macy 		(void) fprintf(stderr, gettext("'%s': not a ZFS filesystem\n"),
1143*eda14cbcSMatt Macy 		    path);
1144*eda14cbcSMatt Macy 		return (NULL);
1145*eda14cbcSMatt Macy 	}
1146*eda14cbcSMatt Macy 
1147*eda14cbcSMatt Macy 	return (zfs_open(hdl, entry.mnt_special, ZFS_TYPE_FILESYSTEM));
1148*eda14cbcSMatt Macy }
1149*eda14cbcSMatt Macy 
1150*eda14cbcSMatt Macy /*
1151*eda14cbcSMatt Macy  * Initialize the zc_nvlist_dst member to prepare for receiving an nvlist from
1152*eda14cbcSMatt Macy  * an ioctl().
1153*eda14cbcSMatt Macy  */
1154*eda14cbcSMatt Macy int
1155*eda14cbcSMatt Macy zcmd_alloc_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, size_t len)
1156*eda14cbcSMatt Macy {
1157*eda14cbcSMatt Macy 	if (len == 0)
1158*eda14cbcSMatt Macy 		len = 256 * 1024;
1159*eda14cbcSMatt Macy 	zc->zc_nvlist_dst_size = len;
1160*eda14cbcSMatt Macy 	zc->zc_nvlist_dst =
1161*eda14cbcSMatt Macy 	    (uint64_t)(uintptr_t)zfs_alloc(hdl, zc->zc_nvlist_dst_size);
1162*eda14cbcSMatt Macy 	if (zc->zc_nvlist_dst == 0)
1163*eda14cbcSMatt Macy 		return (-1);
1164*eda14cbcSMatt Macy 
1165*eda14cbcSMatt Macy 	return (0);
1166*eda14cbcSMatt Macy }
1167*eda14cbcSMatt Macy 
1168*eda14cbcSMatt Macy /*
1169*eda14cbcSMatt Macy  * Called when an ioctl() which returns an nvlist fails with ENOMEM.  This will
1170*eda14cbcSMatt Macy  * expand the nvlist to the size specified in 'zc_nvlist_dst_size', which was
1171*eda14cbcSMatt Macy  * filled in by the kernel to indicate the actual required size.
1172*eda14cbcSMatt Macy  */
1173*eda14cbcSMatt Macy int
1174*eda14cbcSMatt Macy zcmd_expand_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc)
1175*eda14cbcSMatt Macy {
1176*eda14cbcSMatt Macy 	free((void *)(uintptr_t)zc->zc_nvlist_dst);
1177*eda14cbcSMatt Macy 	zc->zc_nvlist_dst =
1178*eda14cbcSMatt Macy 	    (uint64_t)(uintptr_t)zfs_alloc(hdl, zc->zc_nvlist_dst_size);
1179*eda14cbcSMatt Macy 	if (zc->zc_nvlist_dst == 0)
1180*eda14cbcSMatt Macy 		return (-1);
1181*eda14cbcSMatt Macy 
1182*eda14cbcSMatt Macy 	return (0);
1183*eda14cbcSMatt Macy }
1184*eda14cbcSMatt Macy 
1185*eda14cbcSMatt Macy /*
1186*eda14cbcSMatt Macy  * Called to free the src and dst nvlists stored in the command structure.
1187*eda14cbcSMatt Macy  */
1188*eda14cbcSMatt Macy void
1189*eda14cbcSMatt Macy zcmd_free_nvlists(zfs_cmd_t *zc)
1190*eda14cbcSMatt Macy {
1191*eda14cbcSMatt Macy 	free((void *)(uintptr_t)zc->zc_nvlist_conf);
1192*eda14cbcSMatt Macy 	free((void *)(uintptr_t)zc->zc_nvlist_src);
1193*eda14cbcSMatt Macy 	free((void *)(uintptr_t)zc->zc_nvlist_dst);
1194*eda14cbcSMatt Macy 	zc->zc_nvlist_conf = 0;
1195*eda14cbcSMatt Macy 	zc->zc_nvlist_src = 0;
1196*eda14cbcSMatt Macy 	zc->zc_nvlist_dst = 0;
1197*eda14cbcSMatt Macy }
1198*eda14cbcSMatt Macy 
1199*eda14cbcSMatt Macy static int
1200*eda14cbcSMatt Macy zcmd_write_nvlist_com(libzfs_handle_t *hdl, uint64_t *outnv, uint64_t *outlen,
1201*eda14cbcSMatt Macy     nvlist_t *nvl)
1202*eda14cbcSMatt Macy {
1203*eda14cbcSMatt Macy 	char *packed;
1204*eda14cbcSMatt Macy 	size_t len;
1205*eda14cbcSMatt Macy 
1206*eda14cbcSMatt Macy 	verify(nvlist_size(nvl, &len, NV_ENCODE_NATIVE) == 0);
1207*eda14cbcSMatt Macy 
1208*eda14cbcSMatt Macy 	if ((packed = zfs_alloc(hdl, len)) == NULL)
1209*eda14cbcSMatt Macy 		return (-1);
1210*eda14cbcSMatt Macy 
1211*eda14cbcSMatt Macy 	verify(nvlist_pack(nvl, &packed, &len, NV_ENCODE_NATIVE, 0) == 0);
1212*eda14cbcSMatt Macy 
1213*eda14cbcSMatt Macy 	*outnv = (uint64_t)(uintptr_t)packed;
1214*eda14cbcSMatt Macy 	*outlen = len;
1215*eda14cbcSMatt Macy 
1216*eda14cbcSMatt Macy 	return (0);
1217*eda14cbcSMatt Macy }
1218*eda14cbcSMatt Macy 
1219*eda14cbcSMatt Macy int
1220*eda14cbcSMatt Macy zcmd_write_conf_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, nvlist_t *nvl)
1221*eda14cbcSMatt Macy {
1222*eda14cbcSMatt Macy 	return (zcmd_write_nvlist_com(hdl, &zc->zc_nvlist_conf,
1223*eda14cbcSMatt Macy 	    &zc->zc_nvlist_conf_size, nvl));
1224*eda14cbcSMatt Macy }
1225*eda14cbcSMatt Macy 
1226*eda14cbcSMatt Macy int
1227*eda14cbcSMatt Macy zcmd_write_src_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, nvlist_t *nvl)
1228*eda14cbcSMatt Macy {
1229*eda14cbcSMatt Macy 	return (zcmd_write_nvlist_com(hdl, &zc->zc_nvlist_src,
1230*eda14cbcSMatt Macy 	    &zc->zc_nvlist_src_size, nvl));
1231*eda14cbcSMatt Macy }
1232*eda14cbcSMatt Macy 
1233*eda14cbcSMatt Macy /*
1234*eda14cbcSMatt Macy  * Unpacks an nvlist from the ZFS ioctl command structure.
1235*eda14cbcSMatt Macy  */
1236*eda14cbcSMatt Macy int
1237*eda14cbcSMatt Macy zcmd_read_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, nvlist_t **nvlp)
1238*eda14cbcSMatt Macy {
1239*eda14cbcSMatt Macy 	if (nvlist_unpack((void *)(uintptr_t)zc->zc_nvlist_dst,
1240*eda14cbcSMatt Macy 	    zc->zc_nvlist_dst_size, nvlp, 0) != 0)
1241*eda14cbcSMatt Macy 		return (no_memory(hdl));
1242*eda14cbcSMatt Macy 
1243*eda14cbcSMatt Macy 	return (0);
1244*eda14cbcSMatt Macy }
1245*eda14cbcSMatt Macy 
1246*eda14cbcSMatt Macy /*
1247*eda14cbcSMatt Macy  * ================================================================
1248*eda14cbcSMatt Macy  * API shared by zfs and zpool property management
1249*eda14cbcSMatt Macy  * ================================================================
1250*eda14cbcSMatt Macy  */
1251*eda14cbcSMatt Macy 
1252*eda14cbcSMatt Macy static void
1253*eda14cbcSMatt Macy zprop_print_headers(zprop_get_cbdata_t *cbp, zfs_type_t type)
1254*eda14cbcSMatt Macy {
1255*eda14cbcSMatt Macy 	zprop_list_t *pl = cbp->cb_proplist;
1256*eda14cbcSMatt Macy 	int i;
1257*eda14cbcSMatt Macy 	char *title;
1258*eda14cbcSMatt Macy 	size_t len;
1259*eda14cbcSMatt Macy 
1260*eda14cbcSMatt Macy 	cbp->cb_first = B_FALSE;
1261*eda14cbcSMatt Macy 	if (cbp->cb_scripted)
1262*eda14cbcSMatt Macy 		return;
1263*eda14cbcSMatt Macy 
1264*eda14cbcSMatt Macy 	/*
1265*eda14cbcSMatt Macy 	 * Start with the length of the column headers.
1266*eda14cbcSMatt Macy 	 */
1267*eda14cbcSMatt Macy 	cbp->cb_colwidths[GET_COL_NAME] = strlen(dgettext(TEXT_DOMAIN, "NAME"));
1268*eda14cbcSMatt Macy 	cbp->cb_colwidths[GET_COL_PROPERTY] = strlen(dgettext(TEXT_DOMAIN,
1269*eda14cbcSMatt Macy 	    "PROPERTY"));
1270*eda14cbcSMatt Macy 	cbp->cb_colwidths[GET_COL_VALUE] = strlen(dgettext(TEXT_DOMAIN,
1271*eda14cbcSMatt Macy 	    "VALUE"));
1272*eda14cbcSMatt Macy 	cbp->cb_colwidths[GET_COL_RECVD] = strlen(dgettext(TEXT_DOMAIN,
1273*eda14cbcSMatt Macy 	    "RECEIVED"));
1274*eda14cbcSMatt Macy 	cbp->cb_colwidths[GET_COL_SOURCE] = strlen(dgettext(TEXT_DOMAIN,
1275*eda14cbcSMatt Macy 	    "SOURCE"));
1276*eda14cbcSMatt Macy 
1277*eda14cbcSMatt Macy 	/* first property is always NAME */
1278*eda14cbcSMatt Macy 	assert(cbp->cb_proplist->pl_prop ==
1279*eda14cbcSMatt Macy 	    ((type == ZFS_TYPE_POOL) ?  ZPOOL_PROP_NAME : ZFS_PROP_NAME));
1280*eda14cbcSMatt Macy 
1281*eda14cbcSMatt Macy 	/*
1282*eda14cbcSMatt Macy 	 * Go through and calculate the widths for each column.  For the
1283*eda14cbcSMatt Macy 	 * 'source' column, we kludge it up by taking the worst-case scenario of
1284*eda14cbcSMatt Macy 	 * inheriting from the longest name.  This is acceptable because in the
1285*eda14cbcSMatt Macy 	 * majority of cases 'SOURCE' is the last column displayed, and we don't
1286*eda14cbcSMatt Macy 	 * use the width anyway.  Note that the 'VALUE' column can be oversized,
1287*eda14cbcSMatt Macy 	 * if the name of the property is much longer than any values we find.
1288*eda14cbcSMatt Macy 	 */
1289*eda14cbcSMatt Macy 	for (pl = cbp->cb_proplist; pl != NULL; pl = pl->pl_next) {
1290*eda14cbcSMatt Macy 		/*
1291*eda14cbcSMatt Macy 		 * 'PROPERTY' column
1292*eda14cbcSMatt Macy 		 */
1293*eda14cbcSMatt Macy 		if (pl->pl_prop != ZPROP_INVAL) {
1294*eda14cbcSMatt Macy 			const char *propname = (type == ZFS_TYPE_POOL) ?
1295*eda14cbcSMatt Macy 			    zpool_prop_to_name(pl->pl_prop) :
1296*eda14cbcSMatt Macy 			    zfs_prop_to_name(pl->pl_prop);
1297*eda14cbcSMatt Macy 
1298*eda14cbcSMatt Macy 			len = strlen(propname);
1299*eda14cbcSMatt Macy 			if (len > cbp->cb_colwidths[GET_COL_PROPERTY])
1300*eda14cbcSMatt Macy 				cbp->cb_colwidths[GET_COL_PROPERTY] = len;
1301*eda14cbcSMatt Macy 		} else {
1302*eda14cbcSMatt Macy 			len = strlen(pl->pl_user_prop);
1303*eda14cbcSMatt Macy 			if (len > cbp->cb_colwidths[GET_COL_PROPERTY])
1304*eda14cbcSMatt Macy 				cbp->cb_colwidths[GET_COL_PROPERTY] = len;
1305*eda14cbcSMatt Macy 		}
1306*eda14cbcSMatt Macy 
1307*eda14cbcSMatt Macy 		/*
1308*eda14cbcSMatt Macy 		 * 'VALUE' column.  The first property is always the 'name'
1309*eda14cbcSMatt Macy 		 * property that was tacked on either by /sbin/zfs's
1310*eda14cbcSMatt Macy 		 * zfs_do_get() or when calling zprop_expand_list(), so we
1311*eda14cbcSMatt Macy 		 * ignore its width.  If the user specified the name property
1312*eda14cbcSMatt Macy 		 * to display, then it will be later in the list in any case.
1313*eda14cbcSMatt Macy 		 */
1314*eda14cbcSMatt Macy 		if (pl != cbp->cb_proplist &&
1315*eda14cbcSMatt Macy 		    pl->pl_width > cbp->cb_colwidths[GET_COL_VALUE])
1316*eda14cbcSMatt Macy 			cbp->cb_colwidths[GET_COL_VALUE] = pl->pl_width;
1317*eda14cbcSMatt Macy 
1318*eda14cbcSMatt Macy 		/* 'RECEIVED' column. */
1319*eda14cbcSMatt Macy 		if (pl != cbp->cb_proplist &&
1320*eda14cbcSMatt Macy 		    pl->pl_recvd_width > cbp->cb_colwidths[GET_COL_RECVD])
1321*eda14cbcSMatt Macy 			cbp->cb_colwidths[GET_COL_RECVD] = pl->pl_recvd_width;
1322*eda14cbcSMatt Macy 
1323*eda14cbcSMatt Macy 		/*
1324*eda14cbcSMatt Macy 		 * 'NAME' and 'SOURCE' columns
1325*eda14cbcSMatt Macy 		 */
1326*eda14cbcSMatt Macy 		if (pl->pl_prop == (type == ZFS_TYPE_POOL ? ZPOOL_PROP_NAME :
1327*eda14cbcSMatt Macy 		    ZFS_PROP_NAME) &&
1328*eda14cbcSMatt Macy 		    pl->pl_width > cbp->cb_colwidths[GET_COL_NAME]) {
1329*eda14cbcSMatt Macy 			cbp->cb_colwidths[GET_COL_NAME] = pl->pl_width;
1330*eda14cbcSMatt Macy 			cbp->cb_colwidths[GET_COL_SOURCE] = pl->pl_width +
1331*eda14cbcSMatt Macy 			    strlen(dgettext(TEXT_DOMAIN, "inherited from"));
1332*eda14cbcSMatt Macy 		}
1333*eda14cbcSMatt Macy 	}
1334*eda14cbcSMatt Macy 
1335*eda14cbcSMatt Macy 	/*
1336*eda14cbcSMatt Macy 	 * Now go through and print the headers.
1337*eda14cbcSMatt Macy 	 */
1338*eda14cbcSMatt Macy 	for (i = 0; i < ZFS_GET_NCOLS; i++) {
1339*eda14cbcSMatt Macy 		switch (cbp->cb_columns[i]) {
1340*eda14cbcSMatt Macy 		case GET_COL_NAME:
1341*eda14cbcSMatt Macy 			title = dgettext(TEXT_DOMAIN, "NAME");
1342*eda14cbcSMatt Macy 			break;
1343*eda14cbcSMatt Macy 		case GET_COL_PROPERTY:
1344*eda14cbcSMatt Macy 			title = dgettext(TEXT_DOMAIN, "PROPERTY");
1345*eda14cbcSMatt Macy 			break;
1346*eda14cbcSMatt Macy 		case GET_COL_VALUE:
1347*eda14cbcSMatt Macy 			title = dgettext(TEXT_DOMAIN, "VALUE");
1348*eda14cbcSMatt Macy 			break;
1349*eda14cbcSMatt Macy 		case GET_COL_RECVD:
1350*eda14cbcSMatt Macy 			title = dgettext(TEXT_DOMAIN, "RECEIVED");
1351*eda14cbcSMatt Macy 			break;
1352*eda14cbcSMatt Macy 		case GET_COL_SOURCE:
1353*eda14cbcSMatt Macy 			title = dgettext(TEXT_DOMAIN, "SOURCE");
1354*eda14cbcSMatt Macy 			break;
1355*eda14cbcSMatt Macy 		default:
1356*eda14cbcSMatt Macy 			title = NULL;
1357*eda14cbcSMatt Macy 		}
1358*eda14cbcSMatt Macy 
1359*eda14cbcSMatt Macy 		if (title != NULL) {
1360*eda14cbcSMatt Macy 			if (i == (ZFS_GET_NCOLS - 1) ||
1361*eda14cbcSMatt Macy 			    cbp->cb_columns[i + 1] == GET_COL_NONE)
1362*eda14cbcSMatt Macy 				(void) printf("%s", title);
1363*eda14cbcSMatt Macy 			else
1364*eda14cbcSMatt Macy 				(void) printf("%-*s  ",
1365*eda14cbcSMatt Macy 				    cbp->cb_colwidths[cbp->cb_columns[i]],
1366*eda14cbcSMatt Macy 				    title);
1367*eda14cbcSMatt Macy 		}
1368*eda14cbcSMatt Macy 	}
1369*eda14cbcSMatt Macy 	(void) printf("\n");
1370*eda14cbcSMatt Macy }
1371*eda14cbcSMatt Macy 
1372*eda14cbcSMatt Macy /*
1373*eda14cbcSMatt Macy  * Display a single line of output, according to the settings in the callback
1374*eda14cbcSMatt Macy  * structure.
1375*eda14cbcSMatt Macy  */
1376*eda14cbcSMatt Macy void
1377*eda14cbcSMatt Macy zprop_print_one_property(const char *name, zprop_get_cbdata_t *cbp,
1378*eda14cbcSMatt Macy     const char *propname, const char *value, zprop_source_t sourcetype,
1379*eda14cbcSMatt Macy     const char *source, const char *recvd_value)
1380*eda14cbcSMatt Macy {
1381*eda14cbcSMatt Macy 	int i;
1382*eda14cbcSMatt Macy 	const char *str = NULL;
1383*eda14cbcSMatt Macy 	char buf[128];
1384*eda14cbcSMatt Macy 
1385*eda14cbcSMatt Macy 	/*
1386*eda14cbcSMatt Macy 	 * Ignore those source types that the user has chosen to ignore.
1387*eda14cbcSMatt Macy 	 */
1388*eda14cbcSMatt Macy 	if ((sourcetype & cbp->cb_sources) == 0)
1389*eda14cbcSMatt Macy 		return;
1390*eda14cbcSMatt Macy 
1391*eda14cbcSMatt Macy 	if (cbp->cb_first)
1392*eda14cbcSMatt Macy 		zprop_print_headers(cbp, cbp->cb_type);
1393*eda14cbcSMatt Macy 
1394*eda14cbcSMatt Macy 	for (i = 0; i < ZFS_GET_NCOLS; i++) {
1395*eda14cbcSMatt Macy 		switch (cbp->cb_columns[i]) {
1396*eda14cbcSMatt Macy 		case GET_COL_NAME:
1397*eda14cbcSMatt Macy 			str = name;
1398*eda14cbcSMatt Macy 			break;
1399*eda14cbcSMatt Macy 
1400*eda14cbcSMatt Macy 		case GET_COL_PROPERTY:
1401*eda14cbcSMatt Macy 			str = propname;
1402*eda14cbcSMatt Macy 			break;
1403*eda14cbcSMatt Macy 
1404*eda14cbcSMatt Macy 		case GET_COL_VALUE:
1405*eda14cbcSMatt Macy 			str = value;
1406*eda14cbcSMatt Macy 			break;
1407*eda14cbcSMatt Macy 
1408*eda14cbcSMatt Macy 		case GET_COL_SOURCE:
1409*eda14cbcSMatt Macy 			switch (sourcetype) {
1410*eda14cbcSMatt Macy 			case ZPROP_SRC_NONE:
1411*eda14cbcSMatt Macy 				str = "-";
1412*eda14cbcSMatt Macy 				break;
1413*eda14cbcSMatt Macy 
1414*eda14cbcSMatt Macy 			case ZPROP_SRC_DEFAULT:
1415*eda14cbcSMatt Macy 				str = "default";
1416*eda14cbcSMatt Macy 				break;
1417*eda14cbcSMatt Macy 
1418*eda14cbcSMatt Macy 			case ZPROP_SRC_LOCAL:
1419*eda14cbcSMatt Macy 				str = "local";
1420*eda14cbcSMatt Macy 				break;
1421*eda14cbcSMatt Macy 
1422*eda14cbcSMatt Macy 			case ZPROP_SRC_TEMPORARY:
1423*eda14cbcSMatt Macy 				str = "temporary";
1424*eda14cbcSMatt Macy 				break;
1425*eda14cbcSMatt Macy 
1426*eda14cbcSMatt Macy 			case ZPROP_SRC_INHERITED:
1427*eda14cbcSMatt Macy 				(void) snprintf(buf, sizeof (buf),
1428*eda14cbcSMatt Macy 				    "inherited from %s", source);
1429*eda14cbcSMatt Macy 				str = buf;
1430*eda14cbcSMatt Macy 				break;
1431*eda14cbcSMatt Macy 			case ZPROP_SRC_RECEIVED:
1432*eda14cbcSMatt Macy 				str = "received";
1433*eda14cbcSMatt Macy 				break;
1434*eda14cbcSMatt Macy 
1435*eda14cbcSMatt Macy 			default:
1436*eda14cbcSMatt Macy 				str = NULL;
1437*eda14cbcSMatt Macy 				assert(!"unhandled zprop_source_t");
1438*eda14cbcSMatt Macy 			}
1439*eda14cbcSMatt Macy 			break;
1440*eda14cbcSMatt Macy 
1441*eda14cbcSMatt Macy 		case GET_COL_RECVD:
1442*eda14cbcSMatt Macy 			str = (recvd_value == NULL ? "-" : recvd_value);
1443*eda14cbcSMatt Macy 			break;
1444*eda14cbcSMatt Macy 
1445*eda14cbcSMatt Macy 		default:
1446*eda14cbcSMatt Macy 			continue;
1447*eda14cbcSMatt Macy 		}
1448*eda14cbcSMatt Macy 
1449*eda14cbcSMatt Macy 		if (i == (ZFS_GET_NCOLS - 1) ||
1450*eda14cbcSMatt Macy 		    cbp->cb_columns[i + 1] == GET_COL_NONE)
1451*eda14cbcSMatt Macy 			(void) printf("%s", str);
1452*eda14cbcSMatt Macy 		else if (cbp->cb_scripted)
1453*eda14cbcSMatt Macy 			(void) printf("%s\t", str);
1454*eda14cbcSMatt Macy 		else
1455*eda14cbcSMatt Macy 			(void) printf("%-*s  ",
1456*eda14cbcSMatt Macy 			    cbp->cb_colwidths[cbp->cb_columns[i]],
1457*eda14cbcSMatt Macy 			    str);
1458*eda14cbcSMatt Macy 	}
1459*eda14cbcSMatt Macy 
1460*eda14cbcSMatt Macy 	(void) printf("\n");
1461*eda14cbcSMatt Macy }
1462*eda14cbcSMatt Macy 
1463*eda14cbcSMatt Macy /*
1464*eda14cbcSMatt Macy  * Given a numeric suffix, convert the value into a number of bits that the
1465*eda14cbcSMatt Macy  * resulting value must be shifted.
1466*eda14cbcSMatt Macy  */
1467*eda14cbcSMatt Macy static int
1468*eda14cbcSMatt Macy str2shift(libzfs_handle_t *hdl, const char *buf)
1469*eda14cbcSMatt Macy {
1470*eda14cbcSMatt Macy 	const char *ends = "BKMGTPEZ";
1471*eda14cbcSMatt Macy 	int i;
1472*eda14cbcSMatt Macy 
1473*eda14cbcSMatt Macy 	if (buf[0] == '\0')
1474*eda14cbcSMatt Macy 		return (0);
1475*eda14cbcSMatt Macy 	for (i = 0; i < strlen(ends); i++) {
1476*eda14cbcSMatt Macy 		if (toupper(buf[0]) == ends[i])
1477*eda14cbcSMatt Macy 			break;
1478*eda14cbcSMatt Macy 	}
1479*eda14cbcSMatt Macy 	if (i == strlen(ends)) {
1480*eda14cbcSMatt Macy 		if (hdl)
1481*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1482*eda14cbcSMatt Macy 			    "invalid numeric suffix '%s'"), buf);
1483*eda14cbcSMatt Macy 		return (-1);
1484*eda14cbcSMatt Macy 	}
1485*eda14cbcSMatt Macy 
1486*eda14cbcSMatt Macy 	/*
1487*eda14cbcSMatt Macy 	 * Allow 'G' = 'GB' = 'GiB', case-insensitively.
1488*eda14cbcSMatt Macy 	 * However, 'BB' and 'BiB' are disallowed.
1489*eda14cbcSMatt Macy 	 */
1490*eda14cbcSMatt Macy 	if (buf[1] == '\0' ||
1491*eda14cbcSMatt Macy 	    (toupper(buf[0]) != 'B' &&
1492*eda14cbcSMatt Macy 	    ((toupper(buf[1]) == 'B' && buf[2] == '\0') ||
1493*eda14cbcSMatt Macy 	    (toupper(buf[1]) == 'I' && toupper(buf[2]) == 'B' &&
1494*eda14cbcSMatt Macy 	    buf[3] == '\0'))))
1495*eda14cbcSMatt Macy 		return (10 * i);
1496*eda14cbcSMatt Macy 
1497*eda14cbcSMatt Macy 	if (hdl)
1498*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1499*eda14cbcSMatt Macy 		    "invalid numeric suffix '%s'"), buf);
1500*eda14cbcSMatt Macy 	return (-1);
1501*eda14cbcSMatt Macy }
1502*eda14cbcSMatt Macy 
1503*eda14cbcSMatt Macy /*
1504*eda14cbcSMatt Macy  * Convert a string of the form '100G' into a real number.  Used when setting
1505*eda14cbcSMatt Macy  * properties or creating a volume.  'buf' is used to place an extended error
1506*eda14cbcSMatt Macy  * message for the caller to use.
1507*eda14cbcSMatt Macy  */
1508*eda14cbcSMatt Macy int
1509*eda14cbcSMatt Macy zfs_nicestrtonum(libzfs_handle_t *hdl, const char *value, uint64_t *num)
1510*eda14cbcSMatt Macy {
1511*eda14cbcSMatt Macy 	char *end;
1512*eda14cbcSMatt Macy 	int shift;
1513*eda14cbcSMatt Macy 
1514*eda14cbcSMatt Macy 	*num = 0;
1515*eda14cbcSMatt Macy 
1516*eda14cbcSMatt Macy 	/* Check to see if this looks like a number.  */
1517*eda14cbcSMatt Macy 	if ((value[0] < '0' || value[0] > '9') && value[0] != '.') {
1518*eda14cbcSMatt Macy 		if (hdl)
1519*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1520*eda14cbcSMatt Macy 			    "bad numeric value '%s'"), value);
1521*eda14cbcSMatt Macy 		return (-1);
1522*eda14cbcSMatt Macy 	}
1523*eda14cbcSMatt Macy 
1524*eda14cbcSMatt Macy 	/* Rely on strtoull() to process the numeric portion.  */
1525*eda14cbcSMatt Macy 	errno = 0;
1526*eda14cbcSMatt Macy 	*num = strtoull(value, &end, 10);
1527*eda14cbcSMatt Macy 
1528*eda14cbcSMatt Macy 	/*
1529*eda14cbcSMatt Macy 	 * Check for ERANGE, which indicates that the value is too large to fit
1530*eda14cbcSMatt Macy 	 * in a 64-bit value.
1531*eda14cbcSMatt Macy 	 */
1532*eda14cbcSMatt Macy 	if (errno == ERANGE) {
1533*eda14cbcSMatt Macy 		if (hdl)
1534*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1535*eda14cbcSMatt Macy 			    "numeric value is too large"));
1536*eda14cbcSMatt Macy 		return (-1);
1537*eda14cbcSMatt Macy 	}
1538*eda14cbcSMatt Macy 
1539*eda14cbcSMatt Macy 	/*
1540*eda14cbcSMatt Macy 	 * If we have a decimal value, then do the computation with floating
1541*eda14cbcSMatt Macy 	 * point arithmetic.  Otherwise, use standard arithmetic.
1542*eda14cbcSMatt Macy 	 */
1543*eda14cbcSMatt Macy 	if (*end == '.') {
1544*eda14cbcSMatt Macy 		double fval = strtod(value, &end);
1545*eda14cbcSMatt Macy 
1546*eda14cbcSMatt Macy 		if ((shift = str2shift(hdl, end)) == -1)
1547*eda14cbcSMatt Macy 			return (-1);
1548*eda14cbcSMatt Macy 
1549*eda14cbcSMatt Macy 		fval *= pow(2, shift);
1550*eda14cbcSMatt Macy 
1551*eda14cbcSMatt Macy 		/*
1552*eda14cbcSMatt Macy 		 * UINT64_MAX is not exactly representable as a double.
1553*eda14cbcSMatt Macy 		 * The closest representation is UINT64_MAX + 1, so we
1554*eda14cbcSMatt Macy 		 * use a >= comparison instead of > for the bounds check.
1555*eda14cbcSMatt Macy 		 */
1556*eda14cbcSMatt Macy 		if (fval >= (double)UINT64_MAX) {
1557*eda14cbcSMatt Macy 			if (hdl)
1558*eda14cbcSMatt Macy 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1559*eda14cbcSMatt Macy 				    "numeric value is too large"));
1560*eda14cbcSMatt Macy 			return (-1);
1561*eda14cbcSMatt Macy 		}
1562*eda14cbcSMatt Macy 
1563*eda14cbcSMatt Macy 		*num = (uint64_t)fval;
1564*eda14cbcSMatt Macy 	} else {
1565*eda14cbcSMatt Macy 		if ((shift = str2shift(hdl, end)) == -1)
1566*eda14cbcSMatt Macy 			return (-1);
1567*eda14cbcSMatt Macy 
1568*eda14cbcSMatt Macy 		/* Check for overflow */
1569*eda14cbcSMatt Macy 		if (shift >= 64 || (*num << shift) >> shift != *num) {
1570*eda14cbcSMatt Macy 			if (hdl)
1571*eda14cbcSMatt Macy 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1572*eda14cbcSMatt Macy 				    "numeric value is too large"));
1573*eda14cbcSMatt Macy 			return (-1);
1574*eda14cbcSMatt Macy 		}
1575*eda14cbcSMatt Macy 
1576*eda14cbcSMatt Macy 		*num <<= shift;
1577*eda14cbcSMatt Macy 	}
1578*eda14cbcSMatt Macy 
1579*eda14cbcSMatt Macy 	return (0);
1580*eda14cbcSMatt Macy }
1581*eda14cbcSMatt Macy 
1582*eda14cbcSMatt Macy /*
1583*eda14cbcSMatt Macy  * Given a propname=value nvpair to set, parse any numeric properties
1584*eda14cbcSMatt Macy  * (index, boolean, etc) if they are specified as strings and add the
1585*eda14cbcSMatt Macy  * resulting nvpair to the returned nvlist.
1586*eda14cbcSMatt Macy  *
1587*eda14cbcSMatt Macy  * At the DSL layer, all properties are either 64-bit numbers or strings.
1588*eda14cbcSMatt Macy  * We want the user to be able to ignore this fact and specify properties
1589*eda14cbcSMatt Macy  * as native values (numbers, for example) or as strings (to simplify
1590*eda14cbcSMatt Macy  * command line utilities).  This also handles converting index types
1591*eda14cbcSMatt Macy  * (compression, checksum, etc) from strings to their on-disk index.
1592*eda14cbcSMatt Macy  */
1593*eda14cbcSMatt Macy int
1594*eda14cbcSMatt Macy zprop_parse_value(libzfs_handle_t *hdl, nvpair_t *elem, int prop,
1595*eda14cbcSMatt Macy     zfs_type_t type, nvlist_t *ret, char **svalp, uint64_t *ivalp,
1596*eda14cbcSMatt Macy     const char *errbuf)
1597*eda14cbcSMatt Macy {
1598*eda14cbcSMatt Macy 	data_type_t datatype = nvpair_type(elem);
1599*eda14cbcSMatt Macy 	zprop_type_t proptype;
1600*eda14cbcSMatt Macy 	const char *propname;
1601*eda14cbcSMatt Macy 	char *value;
1602*eda14cbcSMatt Macy 	boolean_t isnone = B_FALSE;
1603*eda14cbcSMatt Macy 	boolean_t isauto = B_FALSE;
1604*eda14cbcSMatt Macy 	int err = 0;
1605*eda14cbcSMatt Macy 
1606*eda14cbcSMatt Macy 	if (type == ZFS_TYPE_POOL) {
1607*eda14cbcSMatt Macy 		proptype = zpool_prop_get_type(prop);
1608*eda14cbcSMatt Macy 		propname = zpool_prop_to_name(prop);
1609*eda14cbcSMatt Macy 	} else {
1610*eda14cbcSMatt Macy 		proptype = zfs_prop_get_type(prop);
1611*eda14cbcSMatt Macy 		propname = zfs_prop_to_name(prop);
1612*eda14cbcSMatt Macy 	}
1613*eda14cbcSMatt Macy 
1614*eda14cbcSMatt Macy 	/*
1615*eda14cbcSMatt Macy 	 * Convert any properties to the internal DSL value types.
1616*eda14cbcSMatt Macy 	 */
1617*eda14cbcSMatt Macy 	*svalp = NULL;
1618*eda14cbcSMatt Macy 	*ivalp = 0;
1619*eda14cbcSMatt Macy 
1620*eda14cbcSMatt Macy 	switch (proptype) {
1621*eda14cbcSMatt Macy 	case PROP_TYPE_STRING:
1622*eda14cbcSMatt Macy 		if (datatype != DATA_TYPE_STRING) {
1623*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1624*eda14cbcSMatt Macy 			    "'%s' must be a string"), nvpair_name(elem));
1625*eda14cbcSMatt Macy 			goto error;
1626*eda14cbcSMatt Macy 		}
1627*eda14cbcSMatt Macy 		err = nvpair_value_string(elem, svalp);
1628*eda14cbcSMatt Macy 		if (err != 0) {
1629*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1630*eda14cbcSMatt Macy 			    "'%s' is invalid"), nvpair_name(elem));
1631*eda14cbcSMatt Macy 			goto error;
1632*eda14cbcSMatt Macy 		}
1633*eda14cbcSMatt Macy 		if (strlen(*svalp) >= ZFS_MAXPROPLEN) {
1634*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1635*eda14cbcSMatt Macy 			    "'%s' is too long"), nvpair_name(elem));
1636*eda14cbcSMatt Macy 			goto error;
1637*eda14cbcSMatt Macy 		}
1638*eda14cbcSMatt Macy 		break;
1639*eda14cbcSMatt Macy 
1640*eda14cbcSMatt Macy 	case PROP_TYPE_NUMBER:
1641*eda14cbcSMatt Macy 		if (datatype == DATA_TYPE_STRING) {
1642*eda14cbcSMatt Macy 			(void) nvpair_value_string(elem, &value);
1643*eda14cbcSMatt Macy 			if (strcmp(value, "none") == 0) {
1644*eda14cbcSMatt Macy 				isnone = B_TRUE;
1645*eda14cbcSMatt Macy 			} else if (strcmp(value, "auto") == 0) {
1646*eda14cbcSMatt Macy 				isauto = B_TRUE;
1647*eda14cbcSMatt Macy 			} else if (zfs_nicestrtonum(hdl, value, ivalp) != 0) {
1648*eda14cbcSMatt Macy 				goto error;
1649*eda14cbcSMatt Macy 			}
1650*eda14cbcSMatt Macy 		} else if (datatype == DATA_TYPE_UINT64) {
1651*eda14cbcSMatt Macy 			(void) nvpair_value_uint64(elem, ivalp);
1652*eda14cbcSMatt Macy 		} else {
1653*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1654*eda14cbcSMatt Macy 			    "'%s' must be a number"), nvpair_name(elem));
1655*eda14cbcSMatt Macy 			goto error;
1656*eda14cbcSMatt Macy 		}
1657*eda14cbcSMatt Macy 
1658*eda14cbcSMatt Macy 		/*
1659*eda14cbcSMatt Macy 		 * Quota special: force 'none' and don't allow 0.
1660*eda14cbcSMatt Macy 		 */
1661*eda14cbcSMatt Macy 		if ((type & ZFS_TYPE_DATASET) && *ivalp == 0 && !isnone &&
1662*eda14cbcSMatt Macy 		    (prop == ZFS_PROP_QUOTA || prop == ZFS_PROP_REFQUOTA)) {
1663*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1664*eda14cbcSMatt Macy 			    "use 'none' to disable quota/refquota"));
1665*eda14cbcSMatt Macy 			goto error;
1666*eda14cbcSMatt Macy 		}
1667*eda14cbcSMatt Macy 
1668*eda14cbcSMatt Macy 		/*
1669*eda14cbcSMatt Macy 		 * Special handling for "*_limit=none". In this case it's not
1670*eda14cbcSMatt Macy 		 * 0 but UINT64_MAX.
1671*eda14cbcSMatt Macy 		 */
1672*eda14cbcSMatt Macy 		if ((type & ZFS_TYPE_DATASET) && isnone &&
1673*eda14cbcSMatt Macy 		    (prop == ZFS_PROP_FILESYSTEM_LIMIT ||
1674*eda14cbcSMatt Macy 		    prop == ZFS_PROP_SNAPSHOT_LIMIT)) {
1675*eda14cbcSMatt Macy 			*ivalp = UINT64_MAX;
1676*eda14cbcSMatt Macy 		}
1677*eda14cbcSMatt Macy 
1678*eda14cbcSMatt Macy 		/*
1679*eda14cbcSMatt Macy 		 * Special handling for setting 'refreservation' to 'auto'.  Use
1680*eda14cbcSMatt Macy 		 * UINT64_MAX to tell the caller to use zfs_fix_auto_resv().
1681*eda14cbcSMatt Macy 		 * 'auto' is only allowed on volumes.
1682*eda14cbcSMatt Macy 		 */
1683*eda14cbcSMatt Macy 		if (isauto) {
1684*eda14cbcSMatt Macy 			switch (prop) {
1685*eda14cbcSMatt Macy 			case ZFS_PROP_REFRESERVATION:
1686*eda14cbcSMatt Macy 				if ((type & ZFS_TYPE_VOLUME) == 0) {
1687*eda14cbcSMatt Macy 					zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1688*eda14cbcSMatt Macy 					    "'%s=auto' only allowed on "
1689*eda14cbcSMatt Macy 					    "volumes"), nvpair_name(elem));
1690*eda14cbcSMatt Macy 					goto error;
1691*eda14cbcSMatt Macy 				}
1692*eda14cbcSMatt Macy 				*ivalp = UINT64_MAX;
1693*eda14cbcSMatt Macy 				break;
1694*eda14cbcSMatt Macy 			default:
1695*eda14cbcSMatt Macy 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1696*eda14cbcSMatt Macy 				    "'auto' is invalid value for '%s'"),
1697*eda14cbcSMatt Macy 				    nvpair_name(elem));
1698*eda14cbcSMatt Macy 				goto error;
1699*eda14cbcSMatt Macy 			}
1700*eda14cbcSMatt Macy 		}
1701*eda14cbcSMatt Macy 
1702*eda14cbcSMatt Macy 		break;
1703*eda14cbcSMatt Macy 
1704*eda14cbcSMatt Macy 	case PROP_TYPE_INDEX:
1705*eda14cbcSMatt Macy 		if (datatype != DATA_TYPE_STRING) {
1706*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1707*eda14cbcSMatt Macy 			    "'%s' must be a string"), nvpair_name(elem));
1708*eda14cbcSMatt Macy 			goto error;
1709*eda14cbcSMatt Macy 		}
1710*eda14cbcSMatt Macy 
1711*eda14cbcSMatt Macy 		(void) nvpair_value_string(elem, &value);
1712*eda14cbcSMatt Macy 
1713*eda14cbcSMatt Macy 		if (zprop_string_to_index(prop, value, ivalp, type) != 0) {
1714*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1715*eda14cbcSMatt Macy 			    "'%s' must be one of '%s'"), propname,
1716*eda14cbcSMatt Macy 			    zprop_values(prop, type));
1717*eda14cbcSMatt Macy 			goto error;
1718*eda14cbcSMatt Macy 		}
1719*eda14cbcSMatt Macy 		break;
1720*eda14cbcSMatt Macy 
1721*eda14cbcSMatt Macy 	default:
1722*eda14cbcSMatt Macy 		abort();
1723*eda14cbcSMatt Macy 	}
1724*eda14cbcSMatt Macy 
1725*eda14cbcSMatt Macy 	/*
1726*eda14cbcSMatt Macy 	 * Add the result to our return set of properties.
1727*eda14cbcSMatt Macy 	 */
1728*eda14cbcSMatt Macy 	if (*svalp != NULL) {
1729*eda14cbcSMatt Macy 		if (nvlist_add_string(ret, propname, *svalp) != 0) {
1730*eda14cbcSMatt Macy 			(void) no_memory(hdl);
1731*eda14cbcSMatt Macy 			return (-1);
1732*eda14cbcSMatt Macy 		}
1733*eda14cbcSMatt Macy 	} else {
1734*eda14cbcSMatt Macy 		if (nvlist_add_uint64(ret, propname, *ivalp) != 0) {
1735*eda14cbcSMatt Macy 			(void) no_memory(hdl);
1736*eda14cbcSMatt Macy 			return (-1);
1737*eda14cbcSMatt Macy 		}
1738*eda14cbcSMatt Macy 	}
1739*eda14cbcSMatt Macy 
1740*eda14cbcSMatt Macy 	return (0);
1741*eda14cbcSMatt Macy error:
1742*eda14cbcSMatt Macy 	(void) zfs_error(hdl, EZFS_BADPROP, errbuf);
1743*eda14cbcSMatt Macy 	return (-1);
1744*eda14cbcSMatt Macy }
1745*eda14cbcSMatt Macy 
1746*eda14cbcSMatt Macy static int
1747*eda14cbcSMatt Macy addlist(libzfs_handle_t *hdl, char *propname, zprop_list_t **listp,
1748*eda14cbcSMatt Macy     zfs_type_t type)
1749*eda14cbcSMatt Macy {
1750*eda14cbcSMatt Macy 	int prop;
1751*eda14cbcSMatt Macy 	zprop_list_t *entry;
1752*eda14cbcSMatt Macy 
1753*eda14cbcSMatt Macy 	prop = zprop_name_to_prop(propname, type);
1754*eda14cbcSMatt Macy 
1755*eda14cbcSMatt Macy 	if (prop != ZPROP_INVAL && !zprop_valid_for_type(prop, type, B_FALSE))
1756*eda14cbcSMatt Macy 		prop = ZPROP_INVAL;
1757*eda14cbcSMatt Macy 
1758*eda14cbcSMatt Macy 	/*
1759*eda14cbcSMatt Macy 	 * When no property table entry can be found, return failure if
1760*eda14cbcSMatt Macy 	 * this is a pool property or if this isn't a user-defined
1761*eda14cbcSMatt Macy 	 * dataset property,
1762*eda14cbcSMatt Macy 	 */
1763*eda14cbcSMatt Macy 	if (prop == ZPROP_INVAL && ((type == ZFS_TYPE_POOL &&
1764*eda14cbcSMatt Macy 	    !zpool_prop_feature(propname) &&
1765*eda14cbcSMatt Macy 	    !zpool_prop_unsupported(propname)) ||
1766*eda14cbcSMatt Macy 	    (type == ZFS_TYPE_DATASET && !zfs_prop_user(propname) &&
1767*eda14cbcSMatt Macy 	    !zfs_prop_userquota(propname) && !zfs_prop_written(propname)))) {
1768*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1769*eda14cbcSMatt Macy 		    "invalid property '%s'"), propname);
1770*eda14cbcSMatt Macy 		return (zfs_error(hdl, EZFS_BADPROP,
1771*eda14cbcSMatt Macy 		    dgettext(TEXT_DOMAIN, "bad property list")));
1772*eda14cbcSMatt Macy 	}
1773*eda14cbcSMatt Macy 
1774*eda14cbcSMatt Macy 	if ((entry = zfs_alloc(hdl, sizeof (zprop_list_t))) == NULL)
1775*eda14cbcSMatt Macy 		return (-1);
1776*eda14cbcSMatt Macy 
1777*eda14cbcSMatt Macy 	entry->pl_prop = prop;
1778*eda14cbcSMatt Macy 	if (prop == ZPROP_INVAL) {
1779*eda14cbcSMatt Macy 		if ((entry->pl_user_prop = zfs_strdup(hdl, propname)) ==
1780*eda14cbcSMatt Macy 		    NULL) {
1781*eda14cbcSMatt Macy 			free(entry);
1782*eda14cbcSMatt Macy 			return (-1);
1783*eda14cbcSMatt Macy 		}
1784*eda14cbcSMatt Macy 		entry->pl_width = strlen(propname);
1785*eda14cbcSMatt Macy 	} else {
1786*eda14cbcSMatt Macy 		entry->pl_width = zprop_width(prop, &entry->pl_fixed,
1787*eda14cbcSMatt Macy 		    type);
1788*eda14cbcSMatt Macy 	}
1789*eda14cbcSMatt Macy 
1790*eda14cbcSMatt Macy 	*listp = entry;
1791*eda14cbcSMatt Macy 
1792*eda14cbcSMatt Macy 	return (0);
1793*eda14cbcSMatt Macy }
1794*eda14cbcSMatt Macy 
1795*eda14cbcSMatt Macy /*
1796*eda14cbcSMatt Macy  * Given a comma-separated list of properties, construct a property list
1797*eda14cbcSMatt Macy  * containing both user-defined and native properties.  This function will
1798*eda14cbcSMatt Macy  * return a NULL list if 'all' is specified, which can later be expanded
1799*eda14cbcSMatt Macy  * by zprop_expand_list().
1800*eda14cbcSMatt Macy  */
1801*eda14cbcSMatt Macy int
1802*eda14cbcSMatt Macy zprop_get_list(libzfs_handle_t *hdl, char *props, zprop_list_t **listp,
1803*eda14cbcSMatt Macy     zfs_type_t type)
1804*eda14cbcSMatt Macy {
1805*eda14cbcSMatt Macy 	*listp = NULL;
1806*eda14cbcSMatt Macy 
1807*eda14cbcSMatt Macy 	/*
1808*eda14cbcSMatt Macy 	 * If 'all' is specified, return a NULL list.
1809*eda14cbcSMatt Macy 	 */
1810*eda14cbcSMatt Macy 	if (strcmp(props, "all") == 0)
1811*eda14cbcSMatt Macy 		return (0);
1812*eda14cbcSMatt Macy 
1813*eda14cbcSMatt Macy 	/*
1814*eda14cbcSMatt Macy 	 * If no props were specified, return an error.
1815*eda14cbcSMatt Macy 	 */
1816*eda14cbcSMatt Macy 	if (props[0] == '\0') {
1817*eda14cbcSMatt Macy 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1818*eda14cbcSMatt Macy 		    "no properties specified"));
1819*eda14cbcSMatt Macy 		return (zfs_error(hdl, EZFS_BADPROP, dgettext(TEXT_DOMAIN,
1820*eda14cbcSMatt Macy 		    "bad property list")));
1821*eda14cbcSMatt Macy 	}
1822*eda14cbcSMatt Macy 
1823*eda14cbcSMatt Macy 	/*
1824*eda14cbcSMatt Macy 	 * It would be nice to use getsubopt() here, but the inclusion of column
1825*eda14cbcSMatt Macy 	 * aliases makes this more effort than it's worth.
1826*eda14cbcSMatt Macy 	 */
1827*eda14cbcSMatt Macy 	while (*props != '\0') {
1828*eda14cbcSMatt Macy 		size_t len;
1829*eda14cbcSMatt Macy 		char *p;
1830*eda14cbcSMatt Macy 		char c;
1831*eda14cbcSMatt Macy 
1832*eda14cbcSMatt Macy 		if ((p = strchr(props, ',')) == NULL) {
1833*eda14cbcSMatt Macy 			len = strlen(props);
1834*eda14cbcSMatt Macy 			p = props + len;
1835*eda14cbcSMatt Macy 		} else {
1836*eda14cbcSMatt Macy 			len = p - props;
1837*eda14cbcSMatt Macy 		}
1838*eda14cbcSMatt Macy 
1839*eda14cbcSMatt Macy 		/*
1840*eda14cbcSMatt Macy 		 * Check for empty options.
1841*eda14cbcSMatt Macy 		 */
1842*eda14cbcSMatt Macy 		if (len == 0) {
1843*eda14cbcSMatt Macy 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1844*eda14cbcSMatt Macy 			    "empty property name"));
1845*eda14cbcSMatt Macy 			return (zfs_error(hdl, EZFS_BADPROP,
1846*eda14cbcSMatt Macy 			    dgettext(TEXT_DOMAIN, "bad property list")));
1847*eda14cbcSMatt Macy 		}
1848*eda14cbcSMatt Macy 
1849*eda14cbcSMatt Macy 		/*
1850*eda14cbcSMatt Macy 		 * Check all regular property names.
1851*eda14cbcSMatt Macy 		 */
1852*eda14cbcSMatt Macy 		c = props[len];
1853*eda14cbcSMatt Macy 		props[len] = '\0';
1854*eda14cbcSMatt Macy 
1855*eda14cbcSMatt Macy 		if (strcmp(props, "space") == 0) {
1856*eda14cbcSMatt Macy 			static char *spaceprops[] = {
1857*eda14cbcSMatt Macy 				"name", "avail", "used", "usedbysnapshots",
1858*eda14cbcSMatt Macy 				"usedbydataset", "usedbyrefreservation",
1859*eda14cbcSMatt Macy 				"usedbychildren", NULL
1860*eda14cbcSMatt Macy 			};
1861*eda14cbcSMatt Macy 			int i;
1862*eda14cbcSMatt Macy 
1863*eda14cbcSMatt Macy 			for (i = 0; spaceprops[i]; i++) {
1864*eda14cbcSMatt Macy 				if (addlist(hdl, spaceprops[i], listp, type))
1865*eda14cbcSMatt Macy 					return (-1);
1866*eda14cbcSMatt Macy 				listp = &(*listp)->pl_next;
1867*eda14cbcSMatt Macy 			}
1868*eda14cbcSMatt Macy 		} else {
1869*eda14cbcSMatt Macy 			if (addlist(hdl, props, listp, type))
1870*eda14cbcSMatt Macy 				return (-1);
1871*eda14cbcSMatt Macy 			listp = &(*listp)->pl_next;
1872*eda14cbcSMatt Macy 		}
1873*eda14cbcSMatt Macy 
1874*eda14cbcSMatt Macy 		props = p;
1875*eda14cbcSMatt Macy 		if (c == ',')
1876*eda14cbcSMatt Macy 			props++;
1877*eda14cbcSMatt Macy 	}
1878*eda14cbcSMatt Macy 
1879*eda14cbcSMatt Macy 	return (0);
1880*eda14cbcSMatt Macy }
1881*eda14cbcSMatt Macy 
1882*eda14cbcSMatt Macy void
1883*eda14cbcSMatt Macy zprop_free_list(zprop_list_t *pl)
1884*eda14cbcSMatt Macy {
1885*eda14cbcSMatt Macy 	zprop_list_t *next;
1886*eda14cbcSMatt Macy 
1887*eda14cbcSMatt Macy 	while (pl != NULL) {
1888*eda14cbcSMatt Macy 		next = pl->pl_next;
1889*eda14cbcSMatt Macy 		free(pl->pl_user_prop);
1890*eda14cbcSMatt Macy 		free(pl);
1891*eda14cbcSMatt Macy 		pl = next;
1892*eda14cbcSMatt Macy 	}
1893*eda14cbcSMatt Macy }
1894*eda14cbcSMatt Macy 
1895*eda14cbcSMatt Macy typedef struct expand_data {
1896*eda14cbcSMatt Macy 	zprop_list_t	**last;
1897*eda14cbcSMatt Macy 	libzfs_handle_t	*hdl;
1898*eda14cbcSMatt Macy 	zfs_type_t type;
1899*eda14cbcSMatt Macy } expand_data_t;
1900*eda14cbcSMatt Macy 
1901*eda14cbcSMatt Macy static int
1902*eda14cbcSMatt Macy zprop_expand_list_cb(int prop, void *cb)
1903*eda14cbcSMatt Macy {
1904*eda14cbcSMatt Macy 	zprop_list_t *entry;
1905*eda14cbcSMatt Macy 	expand_data_t *edp = cb;
1906*eda14cbcSMatt Macy 
1907*eda14cbcSMatt Macy 	if ((entry = zfs_alloc(edp->hdl, sizeof (zprop_list_t))) == NULL)
1908*eda14cbcSMatt Macy 		return (ZPROP_INVAL);
1909*eda14cbcSMatt Macy 
1910*eda14cbcSMatt Macy 	entry->pl_prop = prop;
1911*eda14cbcSMatt Macy 	entry->pl_width = zprop_width(prop, &entry->pl_fixed, edp->type);
1912*eda14cbcSMatt Macy 	entry->pl_all = B_TRUE;
1913*eda14cbcSMatt Macy 
1914*eda14cbcSMatt Macy 	*(edp->last) = entry;
1915*eda14cbcSMatt Macy 	edp->last = &entry->pl_next;
1916*eda14cbcSMatt Macy 
1917*eda14cbcSMatt Macy 	return (ZPROP_CONT);
1918*eda14cbcSMatt Macy }
1919*eda14cbcSMatt Macy 
1920*eda14cbcSMatt Macy int
1921*eda14cbcSMatt Macy zprop_expand_list(libzfs_handle_t *hdl, zprop_list_t **plp, zfs_type_t type)
1922*eda14cbcSMatt Macy {
1923*eda14cbcSMatt Macy 	zprop_list_t *entry;
1924*eda14cbcSMatt Macy 	zprop_list_t **last;
1925*eda14cbcSMatt Macy 	expand_data_t exp;
1926*eda14cbcSMatt Macy 
1927*eda14cbcSMatt Macy 	if (*plp == NULL) {
1928*eda14cbcSMatt Macy 		/*
1929*eda14cbcSMatt Macy 		 * If this is the very first time we've been called for an 'all'
1930*eda14cbcSMatt Macy 		 * specification, expand the list to include all native
1931*eda14cbcSMatt Macy 		 * properties.
1932*eda14cbcSMatt Macy 		 */
1933*eda14cbcSMatt Macy 		last = plp;
1934*eda14cbcSMatt Macy 
1935*eda14cbcSMatt Macy 		exp.last = last;
1936*eda14cbcSMatt Macy 		exp.hdl = hdl;
1937*eda14cbcSMatt Macy 		exp.type = type;
1938*eda14cbcSMatt Macy 
1939*eda14cbcSMatt Macy 		if (zprop_iter_common(zprop_expand_list_cb, &exp, B_FALSE,
1940*eda14cbcSMatt Macy 		    B_FALSE, type) == ZPROP_INVAL)
1941*eda14cbcSMatt Macy 			return (-1);
1942*eda14cbcSMatt Macy 
1943*eda14cbcSMatt Macy 		/*
1944*eda14cbcSMatt Macy 		 * Add 'name' to the beginning of the list, which is handled
1945*eda14cbcSMatt Macy 		 * specially.
1946*eda14cbcSMatt Macy 		 */
1947*eda14cbcSMatt Macy 		if ((entry = zfs_alloc(hdl, sizeof (zprop_list_t))) == NULL)
1948*eda14cbcSMatt Macy 			return (-1);
1949*eda14cbcSMatt Macy 
1950*eda14cbcSMatt Macy 		entry->pl_prop = (type == ZFS_TYPE_POOL) ?  ZPOOL_PROP_NAME :
1951*eda14cbcSMatt Macy 		    ZFS_PROP_NAME;
1952*eda14cbcSMatt Macy 		entry->pl_width = zprop_width(entry->pl_prop,
1953*eda14cbcSMatt Macy 		    &entry->pl_fixed, type);
1954*eda14cbcSMatt Macy 		entry->pl_all = B_TRUE;
1955*eda14cbcSMatt Macy 		entry->pl_next = *plp;
1956*eda14cbcSMatt Macy 		*plp = entry;
1957*eda14cbcSMatt Macy 	}
1958*eda14cbcSMatt Macy 	return (0);
1959*eda14cbcSMatt Macy }
1960*eda14cbcSMatt Macy 
1961*eda14cbcSMatt Macy int
1962*eda14cbcSMatt Macy zprop_iter(zprop_func func, void *cb, boolean_t show_all, boolean_t ordered,
1963*eda14cbcSMatt Macy     zfs_type_t type)
1964*eda14cbcSMatt Macy {
1965*eda14cbcSMatt Macy 	return (zprop_iter_common(func, cb, show_all, ordered, type));
1966*eda14cbcSMatt Macy }
1967*eda14cbcSMatt Macy 
1968*eda14cbcSMatt Macy /*
1969*eda14cbcSMatt Macy  * Fill given version buffer with zfs userland version
1970*eda14cbcSMatt Macy  */
1971*eda14cbcSMatt Macy void
1972*eda14cbcSMatt Macy zfs_version_userland(char *version, int len)
1973*eda14cbcSMatt Macy {
1974*eda14cbcSMatt Macy 	(void) strlcpy(version, ZFS_META_ALIAS, len);
1975*eda14cbcSMatt Macy }
1976*eda14cbcSMatt Macy 
1977*eda14cbcSMatt Macy /*
1978*eda14cbcSMatt Macy  * Prints both zfs userland and kernel versions
1979*eda14cbcSMatt Macy  * Returns 0 on success, and -1 on error (with errno set)
1980*eda14cbcSMatt Macy  */
1981*eda14cbcSMatt Macy int
1982*eda14cbcSMatt Macy zfs_version_print(void)
1983*eda14cbcSMatt Macy {
1984*eda14cbcSMatt Macy 	char zver_userland[128];
1985*eda14cbcSMatt Macy 	char zver_kernel[128];
1986*eda14cbcSMatt Macy 
1987*eda14cbcSMatt Macy 	zfs_version_userland(zver_userland, sizeof (zver_userland));
1988*eda14cbcSMatt Macy 
1989*eda14cbcSMatt Macy 	(void) printf("%s\n", zver_userland);
1990*eda14cbcSMatt Macy 
1991*eda14cbcSMatt Macy 	if (zfs_version_kernel(zver_kernel, sizeof (zver_kernel)) == -1) {
1992*eda14cbcSMatt Macy 		fprintf(stderr, "zfs_version_kernel() failed: %s\n",
1993*eda14cbcSMatt Macy 		    strerror(errno));
1994*eda14cbcSMatt Macy 		return (-1);
1995*eda14cbcSMatt Macy 	}
1996*eda14cbcSMatt Macy 
1997*eda14cbcSMatt Macy 	(void) printf("zfs-kmod-%s\n", zver_kernel);
1998*eda14cbcSMatt Macy 
1999*eda14cbcSMatt Macy 	return (0);
2000*eda14cbcSMatt Macy }
2001*eda14cbcSMatt Macy 
2002*eda14cbcSMatt Macy /*
2003*eda14cbcSMatt Macy  * Return 1 if the user requested ANSI color output, and our terminal supports
2004*eda14cbcSMatt Macy  * it.  Return 0 for no color.
2005*eda14cbcSMatt Macy  */
2006*eda14cbcSMatt Macy static int
2007*eda14cbcSMatt Macy use_color(void)
2008*eda14cbcSMatt Macy {
2009*eda14cbcSMatt Macy 	static int use_color = -1;
2010*eda14cbcSMatt Macy 	char *term;
2011*eda14cbcSMatt Macy 
2012*eda14cbcSMatt Macy 	/*
2013*eda14cbcSMatt Macy 	 * Optimization:
2014*eda14cbcSMatt Macy 	 *
2015*eda14cbcSMatt Macy 	 * For each zpool invocation, we do a single check to see if we should
2016*eda14cbcSMatt Macy 	 * be using color or not, and cache that value for the lifetime of the
2017*eda14cbcSMatt Macy 	 * the zpool command.  That makes it cheap to call use_color() when
2018*eda14cbcSMatt Macy 	 * we're printing with color.  We assume that the settings are not going
2019*eda14cbcSMatt Macy 	 * to change during the invocation of a zpool command (the user isn't
2020*eda14cbcSMatt Macy 	 * going to change the ZFS_COLOR value while zpool is running, for
2021*eda14cbcSMatt Macy 	 * example).
2022*eda14cbcSMatt Macy 	 */
2023*eda14cbcSMatt Macy 	if (use_color != -1) {
2024*eda14cbcSMatt Macy 		/*
2025*eda14cbcSMatt Macy 		 * We've already figured out if we should be using color or
2026*eda14cbcSMatt Macy 		 * not.  Return the cached value.
2027*eda14cbcSMatt Macy 		 */
2028*eda14cbcSMatt Macy 		return (use_color);
2029*eda14cbcSMatt Macy 	}
2030*eda14cbcSMatt Macy 
2031*eda14cbcSMatt Macy 	term = getenv("TERM");
2032*eda14cbcSMatt Macy 	/*
2033*eda14cbcSMatt Macy 	 * The user sets the ZFS_COLOR env var set to enable zpool ANSI color
2034*eda14cbcSMatt Macy 	 * output.  However if NO_COLOR is set (https://no-color.org/) then
2035*eda14cbcSMatt Macy 	 * don't use it.  Also, don't use color if terminal doesn't support
2036*eda14cbcSMatt Macy 	 * it.
2037*eda14cbcSMatt Macy 	 */
2038*eda14cbcSMatt Macy 	if (libzfs_envvar_is_set("ZFS_COLOR") &&
2039*eda14cbcSMatt Macy 	    !libzfs_envvar_is_set("NO_COLOR") &&
2040*eda14cbcSMatt Macy 	    isatty(STDOUT_FILENO) && term && strcmp("dumb", term) != 0 &&
2041*eda14cbcSMatt Macy 	    strcmp("unknown", term) != 0) {
2042*eda14cbcSMatt Macy 		/* Color supported */
2043*eda14cbcSMatt Macy 		use_color = 1;
2044*eda14cbcSMatt Macy 	} else {
2045*eda14cbcSMatt Macy 		use_color = 0;
2046*eda14cbcSMatt Macy 	}
2047*eda14cbcSMatt Macy 
2048*eda14cbcSMatt Macy 	return (use_color);
2049*eda14cbcSMatt Macy }
2050*eda14cbcSMatt Macy 
2051*eda14cbcSMatt Macy /*
2052*eda14cbcSMatt Macy  * color_start() and color_end() are used for when you want to colorize a block
2053*eda14cbcSMatt Macy  * of text.  For example:
2054*eda14cbcSMatt Macy  *
2055*eda14cbcSMatt Macy  * color_start(ANSI_RED_FG)
2056*eda14cbcSMatt Macy  * printf("hello");
2057*eda14cbcSMatt Macy  * printf("world");
2058*eda14cbcSMatt Macy  * color_end();
2059*eda14cbcSMatt Macy  */
2060*eda14cbcSMatt Macy void
2061*eda14cbcSMatt Macy color_start(char *color)
2062*eda14cbcSMatt Macy {
2063*eda14cbcSMatt Macy 	if (use_color())
2064*eda14cbcSMatt Macy 		printf("%s", color);
2065*eda14cbcSMatt Macy }
2066*eda14cbcSMatt Macy 
2067*eda14cbcSMatt Macy void
2068*eda14cbcSMatt Macy color_end(void)
2069*eda14cbcSMatt Macy {
2070*eda14cbcSMatt Macy 	if (use_color())
2071*eda14cbcSMatt Macy 		printf(ANSI_RESET);
2072*eda14cbcSMatt Macy }
2073*eda14cbcSMatt Macy 
2074*eda14cbcSMatt Macy /* printf() with a color.  If color is NULL, then do a normal printf. */
2075*eda14cbcSMatt Macy int
2076*eda14cbcSMatt Macy printf_color(char *color, char *format, ...)
2077*eda14cbcSMatt Macy {
2078*eda14cbcSMatt Macy 	va_list aptr;
2079*eda14cbcSMatt Macy 	int rc;
2080*eda14cbcSMatt Macy 
2081*eda14cbcSMatt Macy 	if (color)
2082*eda14cbcSMatt Macy 		color_start(color);
2083*eda14cbcSMatt Macy 
2084*eda14cbcSMatt Macy 	va_start(aptr, format);
2085*eda14cbcSMatt Macy 	rc = vprintf(format, aptr);
2086*eda14cbcSMatt Macy 	va_end(aptr);
2087*eda14cbcSMatt Macy 
2088*eda14cbcSMatt Macy 	if (color)
2089*eda14cbcSMatt Macy 		color_end();
2090*eda14cbcSMatt Macy 
2091*eda14cbcSMatt Macy 	return (rc);
2092*eda14cbcSMatt Macy }
2093