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