1789Sahrens /*
2789Sahrens  * CDDL HEADER START
3789Sahrens  *
4789Sahrens  * The contents of this file are subject to the terms of the
51544Seschrock  * Common Development and Distribution License (the "License").
61544Seschrock  * You may not use this file except in compliance with the License.
7789Sahrens  *
8789Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9789Sahrens  * or http://www.opensolaris.org/os/licensing.
10789Sahrens  * See the License for the specific language governing permissions
11789Sahrens  * and limitations under the License.
12789Sahrens  *
13789Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
14789Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15789Sahrens  * If applicable, add the following below this CDDL HEADER, with the
16789Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
17789Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
18789Sahrens  *
19789Sahrens  * CDDL HEADER END
20789Sahrens  */
21789Sahrens /*
223635Sck153898  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
23789Sahrens  * Use is subject to license terms.
24789Sahrens  */
25789Sahrens 
26789Sahrens #pragma ident	"%Z%%M%	%I%	%E% SMI"
27789Sahrens 
28789Sahrens /*
29789Sahrens  * Internal utility routines for the ZFS library.
30789Sahrens  */
31789Sahrens 
32789Sahrens #include <errno.h>
33789Sahrens #include <fcntl.h>
34789Sahrens #include <libintl.h>
35789Sahrens #include <stdarg.h>
36789Sahrens #include <stdio.h>
37789Sahrens #include <stdlib.h>
38789Sahrens #include <strings.h>
39789Sahrens #include <unistd.h>
40789Sahrens #include <sys/mnttab.h>
413635Sck153898 #include <sys/mntent.h>
423635Sck153898 #include <sys/types.h>
43789Sahrens 
44789Sahrens #include <libzfs.h>
45789Sahrens 
46789Sahrens #include "libzfs_impl.h"
47789Sahrens 
482082Seschrock int
492082Seschrock libzfs_errno(libzfs_handle_t *hdl)
502082Seschrock {
512082Seschrock 	return (hdl->libzfs_error);
522082Seschrock }
53789Sahrens 
542082Seschrock const char *
552082Seschrock libzfs_error_action(libzfs_handle_t *hdl)
562082Seschrock {
572082Seschrock 	return (hdl->libzfs_action);
582082Seschrock }
592082Seschrock 
602082Seschrock const char *
612082Seschrock libzfs_error_description(libzfs_handle_t *hdl)
622082Seschrock {
632082Seschrock 	if (hdl->libzfs_desc[0] != '\0')
642082Seschrock 		return (hdl->libzfs_desc);
65789Sahrens 
662082Seschrock 	switch (hdl->libzfs_error) {
672082Seschrock 	case EZFS_NOMEM:
682082Seschrock 		return (dgettext(TEXT_DOMAIN, "out of memory"));
692082Seschrock 	case EZFS_BADPROP:
702082Seschrock 		return (dgettext(TEXT_DOMAIN, "invalid property value"));
712082Seschrock 	case EZFS_PROPREADONLY:
722082Seschrock 		return (dgettext(TEXT_DOMAIN, "read only property"));
732082Seschrock 	case EZFS_PROPTYPE:
742082Seschrock 		return (dgettext(TEXT_DOMAIN, "property doesn't apply to "
752082Seschrock 		    "datasets of this type"));
762082Seschrock 	case EZFS_PROPNONINHERIT:
772082Seschrock 		return (dgettext(TEXT_DOMAIN, "property cannot be inherited"));
782082Seschrock 	case EZFS_PROPSPACE:
792082Seschrock 		return (dgettext(TEXT_DOMAIN, "invalid quota or reservation"));
802082Seschrock 	case EZFS_BADTYPE:
812082Seschrock 		return (dgettext(TEXT_DOMAIN, "operation not applicable to "
822082Seschrock 		    "datasets of this type"));
832082Seschrock 	case EZFS_BUSY:
842082Seschrock 		return (dgettext(TEXT_DOMAIN, "pool or dataset is busy"));
852082Seschrock 	case EZFS_EXISTS:
862082Seschrock 		return (dgettext(TEXT_DOMAIN, "pool or dataset exists"));
872082Seschrock 	case EZFS_NOENT:
882082Seschrock 		return (dgettext(TEXT_DOMAIN, "no such pool or dataset"));
892082Seschrock 	case EZFS_BADSTREAM:
902082Seschrock 		return (dgettext(TEXT_DOMAIN, "invalid backup stream"));
912082Seschrock 	case EZFS_DSREADONLY:
922082Seschrock 		return (dgettext(TEXT_DOMAIN, "dataset is read only"));
932082Seschrock 	case EZFS_VOLTOOBIG:
942082Seschrock 		return (dgettext(TEXT_DOMAIN, "volume size exceeds limit for "
952082Seschrock 		    "this system"));
962082Seschrock 	case EZFS_VOLHASDATA:
972082Seschrock 		return (dgettext(TEXT_DOMAIN, "volume has data"));
982082Seschrock 	case EZFS_INVALIDNAME:
992082Seschrock 		return (dgettext(TEXT_DOMAIN, "invalid name"));
1002082Seschrock 	case EZFS_BADRESTORE:
1012082Seschrock 		return (dgettext(TEXT_DOMAIN, "unable to restore to "
1022082Seschrock 		    "destination"));
1032082Seschrock 	case EZFS_BADBACKUP:
1042082Seschrock 		return (dgettext(TEXT_DOMAIN, "backup failed"));
1052082Seschrock 	case EZFS_BADTARGET:
1062082Seschrock 		return (dgettext(TEXT_DOMAIN, "invalid target vdev"));
1072082Seschrock 	case EZFS_NODEVICE:
1082082Seschrock 		return (dgettext(TEXT_DOMAIN, "no such device in pool"));
1092082Seschrock 	case EZFS_BADDEV:
1102082Seschrock 		return (dgettext(TEXT_DOMAIN, "invalid device"));
1112082Seschrock 	case EZFS_NOREPLICAS:
1122082Seschrock 		return (dgettext(TEXT_DOMAIN, "no valid replicas"));
1132082Seschrock 	case EZFS_RESILVERING:
1142082Seschrock 		return (dgettext(TEXT_DOMAIN, "currently resilvering"));
1152082Seschrock 	case EZFS_BADVERSION:
1162082Seschrock 		return (dgettext(TEXT_DOMAIN, "unsupported version"));
1172082Seschrock 	case EZFS_POOLUNAVAIL:
1182082Seschrock 		return (dgettext(TEXT_DOMAIN, "pool is unavailable"));
1192082Seschrock 	case EZFS_DEVOVERFLOW:
1202082Seschrock 		return (dgettext(TEXT_DOMAIN, "too many devices in one vdev"));
1212082Seschrock 	case EZFS_BADPATH:
1222082Seschrock 		return (dgettext(TEXT_DOMAIN, "must be an absolute path"));
1232082Seschrock 	case EZFS_CROSSTARGET:
1242082Seschrock 		return (dgettext(TEXT_DOMAIN, "operation crosses datasets or "
1252082Seschrock 		    "pools"));
1262082Seschrock 	case EZFS_ZONED:
1272082Seschrock 		return (dgettext(TEXT_DOMAIN, "dataset in use by local zone"));
1282082Seschrock 	case EZFS_MOUNTFAILED:
1292082Seschrock 		return (dgettext(TEXT_DOMAIN, "mount failed"));
1302082Seschrock 	case EZFS_UMOUNTFAILED:
1312082Seschrock 		return (dgettext(TEXT_DOMAIN, "umount failed"));
1323126Sahl 	case EZFS_UNSHARENFSFAILED:
1332082Seschrock 		return (dgettext(TEXT_DOMAIN, "unshare(1M) failed"));
1343126Sahl 	case EZFS_SHARENFSFAILED:
1352082Seschrock 		return (dgettext(TEXT_DOMAIN, "share(1M) failed"));
1362082Seschrock 	case EZFS_DEVLINKS:
1372082Seschrock 		return (dgettext(TEXT_DOMAIN, "failed to create /dev links"));
1382082Seschrock 	case EZFS_PERM:
1392082Seschrock 		return (dgettext(TEXT_DOMAIN, "permission denied"));
1402082Seschrock 	case EZFS_NOSPC:
1412082Seschrock 		return (dgettext(TEXT_DOMAIN, "out of space"));
1422082Seschrock 	case EZFS_IO:
1432082Seschrock 		return (dgettext(TEXT_DOMAIN, "I/O error"));
1442082Seschrock 	case EZFS_INTR:
1452082Seschrock 		return (dgettext(TEXT_DOMAIN, "signal received"));
1462082Seschrock 	case EZFS_ISSPARE:
1472082Seschrock 		return (dgettext(TEXT_DOMAIN, "device is reserved as a hot "
1482082Seschrock 		    "spare"));
1492082Seschrock 	case EZFS_INVALCONFIG:
1502082Seschrock 		return (dgettext(TEXT_DOMAIN, "invalid vdev configuration"));
1512474Seschrock 	case EZFS_RECURSIVE:
1522474Seschrock 		return (dgettext(TEXT_DOMAIN, "recursive dataset dependency"));
1532926Sek110237 	case EZFS_NOHISTORY:
1542926Sek110237 		return (dgettext(TEXT_DOMAIN, "no history available"));
1553126Sahl 	case EZFS_UNSHAREISCSIFAILED:
1563126Sahl 		return (dgettext(TEXT_DOMAIN,
1573126Sahl 		    "iscsitgtd failed request to unshare"));
1583126Sahl 	case EZFS_SHAREISCSIFAILED:
1593126Sahl 		return (dgettext(TEXT_DOMAIN,
1603126Sahl 		    "iscsitgtd failed request to share"));
1613912Slling 	case EZFS_POOLPROPS:
1623912Slling 		return (dgettext(TEXT_DOMAIN, "failed to retrieve "
1633912Slling 		    "pool properties"));
1643912Slling 	case EZFS_POOL_NOTSUP:
1653912Slling 		return (dgettext(TEXT_DOMAIN, "operation not supported "
1663912Slling 		    "on this type of pool"));
1673912Slling 	case EZFS_POOL_INVALARG:
1683912Slling 		return (dgettext(TEXT_DOMAIN, "invalid argument for "
1693912Slling 		    "this pool operation"));
170*3978Smmusante 	case EZFS_NAMETOOLONG:
171*3978Smmusante 		return (dgettext(TEXT_DOMAIN, "dataset name is too long"));
1722082Seschrock 	case EZFS_UNKNOWN:
1732082Seschrock 		return (dgettext(TEXT_DOMAIN, "unknown error"));
1742082Seschrock 	default:
1752500Seschrock 		assert(hdl->libzfs_error == 0);
1762500Seschrock 		return (dgettext(TEXT_DOMAIN, "no error"));
1772082Seschrock 	}
1782082Seschrock }
1792082Seschrock 
1802082Seschrock /*PRINTFLIKE2*/
181789Sahrens void
1822082Seschrock zfs_error_aux(libzfs_handle_t *hdl, const char *fmt, ...)
183789Sahrens {
184789Sahrens 	va_list ap;
185789Sahrens 
186789Sahrens 	va_start(ap, fmt);
187789Sahrens 
1882082Seschrock 	(void) vsnprintf(hdl->libzfs_desc, sizeof (hdl->libzfs_desc),
1892082Seschrock 	    fmt, ap);
1902082Seschrock 	hdl->libzfs_desc_active = 1;
191789Sahrens 
192789Sahrens 	va_end(ap);
193789Sahrens }
194789Sahrens 
1952082Seschrock static void
1962082Seschrock zfs_verror(libzfs_handle_t *hdl, int error, const char *fmt, va_list ap)
1972082Seschrock {
1982082Seschrock 	(void) vsnprintf(hdl->libzfs_action, sizeof (hdl->libzfs_action),
1992082Seschrock 	    fmt, ap);
2002082Seschrock 	hdl->libzfs_error = error;
2012082Seschrock 
2022082Seschrock 	if (hdl->libzfs_desc_active)
2032082Seschrock 		hdl->libzfs_desc_active = 0;
2042082Seschrock 	else
2052082Seschrock 		hdl->libzfs_desc[0] = '\0';
2062082Seschrock 
2072082Seschrock 	if (hdl->libzfs_printerr) {
2082082Seschrock 		if (error == EZFS_UNKNOWN) {
2092082Seschrock 			(void) fprintf(stderr, dgettext(TEXT_DOMAIN, "internal "
2102082Seschrock 			    "error: %s\n"), libzfs_error_description(hdl));
2112082Seschrock 			abort();
2122082Seschrock 		}
2132082Seschrock 
2142082Seschrock 		(void) fprintf(stderr, "%s: %s\n", hdl->libzfs_action,
2153912Slling 		    libzfs_error_description(hdl));
2162082Seschrock 		if (error == EZFS_NOMEM)
2172082Seschrock 			exit(1);
2182082Seschrock 	}
2192082Seschrock }
2202082Seschrock 
2213237Slling int
2223237Slling zfs_error(libzfs_handle_t *hdl, int error, const char *msg)
2233237Slling {
2243237Slling 	return (zfs_error_fmt(hdl, error, "%s", msg));
2253237Slling }
2263237Slling 
2272082Seschrock /*PRINTFLIKE3*/
2282082Seschrock int
2293237Slling zfs_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
2302082Seschrock {
2312082Seschrock 	va_list ap;
2322082Seschrock 
2332082Seschrock 	va_start(ap, fmt);
2342082Seschrock 
2352082Seschrock 	zfs_verror(hdl, error, fmt, ap);
2362082Seschrock 
2372082Seschrock 	va_end(ap);
2382082Seschrock 
2392082Seschrock 	return (-1);
2402082Seschrock }
2412082Seschrock 
2422082Seschrock static int
2432082Seschrock zfs_common_error(libzfs_handle_t *hdl, int error, const char *fmt,
2442082Seschrock     va_list ap)
2452082Seschrock {
2462082Seschrock 	switch (error) {
2472082Seschrock 	case EPERM:
2482082Seschrock 	case EACCES:
2492082Seschrock 		zfs_verror(hdl, EZFS_PERM, fmt, ap);
2502082Seschrock 		return (-1);
2512082Seschrock 
2522082Seschrock 	case EIO:
2532082Seschrock 		zfs_verror(hdl, EZFS_IO, fmt, ap);
2542082Seschrock 		return (-1);
2552082Seschrock 
2562082Seschrock 	case EINTR:
2572082Seschrock 		zfs_verror(hdl, EZFS_INTR, fmt, ap);
2582082Seschrock 		return (-1);
2592082Seschrock 	}
2602082Seschrock 
2612082Seschrock 	return (0);
2622082Seschrock }
2632082Seschrock 
2643237Slling int
2653237Slling zfs_standard_error(libzfs_handle_t *hdl, int error, const char *msg)
2663237Slling {
2673237Slling 	return (zfs_standard_error_fmt(hdl, error, "%s", msg));
2683237Slling }
2693237Slling 
2702082Seschrock /*PRINTFLIKE3*/
2712082Seschrock int
2723237Slling zfs_standard_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
273789Sahrens {
274789Sahrens 	va_list ap;
275789Sahrens 
276789Sahrens 	va_start(ap, fmt);
277789Sahrens 
2782082Seschrock 	if (zfs_common_error(hdl, error, fmt, ap) != 0) {
2792082Seschrock 		va_end(ap);
2802082Seschrock 		return (-1);
2812082Seschrock 	}
2822082Seschrock 
2832082Seschrock 
2842082Seschrock 	switch (error) {
2852082Seschrock 	case ENXIO:
2862082Seschrock 		zfs_verror(hdl, EZFS_IO, fmt, ap);
2872082Seschrock 		break;
2882082Seschrock 
2892082Seschrock 	case ENOENT:
2902082Seschrock 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
2912082Seschrock 		    "dataset does not exist"));
2922082Seschrock 		zfs_verror(hdl, EZFS_NOENT, fmt, ap);
2932082Seschrock 		break;
2942082Seschrock 
2952082Seschrock 	case ENOSPC:
2962082Seschrock 	case EDQUOT:
2972082Seschrock 		zfs_verror(hdl, EZFS_NOSPC, fmt, ap);
2982082Seschrock 		return (-1);
2992082Seschrock 
3002082Seschrock 	case EEXIST:
3012082Seschrock 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
3022082Seschrock 		    "dataset already exists"));
3032082Seschrock 		zfs_verror(hdl, EZFS_EXISTS, fmt, ap);
3042082Seschrock 		break;
3052082Seschrock 
3062082Seschrock 	case EBUSY:
3072082Seschrock 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
3082082Seschrock 		    "dataset is busy"));
3092082Seschrock 		zfs_verror(hdl, EZFS_BUSY, fmt, ap);
3102082Seschrock 		break;
311*3978Smmusante 
312*3978Smmusante 	case ENAMETOOLONG:
313*3978Smmusante 		zfs_verror(hdl, EZFS_NAMETOOLONG, fmt, ap);
314*3978Smmusante 		break;
315*3978Smmusante 
3162082Seschrock 	default:
3172082Seschrock 		zfs_error_aux(hdl, strerror(errno));
3182082Seschrock 		zfs_verror(hdl, EZFS_UNKNOWN, fmt, ap);
3192082Seschrock 		break;
320789Sahrens 	}
321789Sahrens 
322789Sahrens 	va_end(ap);
3232082Seschrock 	return (-1);
324789Sahrens }
325789Sahrens 
3263237Slling int
3273237Slling zpool_standard_error(libzfs_handle_t *hdl, int error, const char *msg)
3283237Slling {
3293237Slling 	return (zpool_standard_error_fmt(hdl, error, "%s", msg));
3303237Slling }
3313237Slling 
3322082Seschrock /*PRINTFLIKE3*/
3332082Seschrock int
3343237Slling zpool_standard_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
335789Sahrens {
3362082Seschrock 	va_list ap;
3372082Seschrock 
3382082Seschrock 	va_start(ap, fmt);
3392082Seschrock 
3402082Seschrock 	if (zfs_common_error(hdl, error, fmt, ap) != 0) {
3412082Seschrock 		va_end(ap);
3422082Seschrock 		return (-1);
3432082Seschrock 	}
3442082Seschrock 
3452082Seschrock 	switch (error) {
3462082Seschrock 	case ENODEV:
3472082Seschrock 		zfs_verror(hdl, EZFS_NODEVICE, fmt, ap);
3482082Seschrock 		break;
3492082Seschrock 
3502082Seschrock 	case ENOENT:
3513912Slling 		zfs_error_aux(hdl,
3523912Slling 		    dgettext(TEXT_DOMAIN, "no such pool or dataset"));
3532082Seschrock 		zfs_verror(hdl, EZFS_NOENT, fmt, ap);
3542082Seschrock 		break;
3552082Seschrock 
3562082Seschrock 	case EEXIST:
3572082Seschrock 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
3582082Seschrock 		    "pool already exists"));
3592082Seschrock 		zfs_verror(hdl, EZFS_EXISTS, fmt, ap);
3602082Seschrock 		break;
3612082Seschrock 
3622082Seschrock 	case EBUSY:
3632082Seschrock 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "pool is busy"));
3642082Seschrock 		zfs_verror(hdl, EZFS_EXISTS, fmt, ap);
3652082Seschrock 		break;
3662082Seschrock 
3672082Seschrock 	case ENXIO:
3682082Seschrock 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
3692082Seschrock 		    "one or more devices is currently unavailable"));
3702082Seschrock 		zfs_verror(hdl, EZFS_BADDEV, fmt, ap);
3712082Seschrock 		break;
3722082Seschrock 
3732082Seschrock 	case ENAMETOOLONG:
3742082Seschrock 		zfs_verror(hdl, EZFS_DEVOVERFLOW, fmt, ap);
3752082Seschrock 		break;
3762082Seschrock 
3773912Slling 	case ENOTSUP:
3783912Slling 		zfs_verror(hdl, EZFS_POOL_NOTSUP, fmt, ap);
3793912Slling 		break;
3803912Slling 
3813912Slling 	case EINVAL:
3823912Slling 		zfs_verror(hdl, EZFS_POOL_INVALARG, fmt, ap);
3833912Slling 		break;
3843912Slling 
3852082Seschrock 	default:
3862082Seschrock 		zfs_error_aux(hdl, strerror(error));
3872082Seschrock 		zfs_verror(hdl, EZFS_UNKNOWN, fmt, ap);
3882082Seschrock 	}
3892082Seschrock 
3902082Seschrock 	va_end(ap);
3912082Seschrock 	return (-1);
392789Sahrens }
393789Sahrens 
394789Sahrens /*
395789Sahrens  * Display an out of memory error message and abort the current program.
396789Sahrens  */
3972082Seschrock int
3982082Seschrock no_memory(libzfs_handle_t *hdl)
399789Sahrens {
4002082Seschrock 	return (zfs_error(hdl, EZFS_NOMEM, "internal error"));
401789Sahrens }
402789Sahrens 
403789Sahrens /*
404789Sahrens  * A safe form of malloc() which will die if the allocation fails.
405789Sahrens  */
406789Sahrens void *
4072082Seschrock zfs_alloc(libzfs_handle_t *hdl, size_t size)
408789Sahrens {
409789Sahrens 	void *data;
410789Sahrens 
411789Sahrens 	if ((data = calloc(1, size)) == NULL)
4122082Seschrock 		(void) no_memory(hdl);
413789Sahrens 
414789Sahrens 	return (data);
415789Sahrens }
416789Sahrens 
417789Sahrens /*
4182676Seschrock  * A safe form of realloc(), which also zeroes newly allocated space.
4192676Seschrock  */
4202676Seschrock void *
4212676Seschrock zfs_realloc(libzfs_handle_t *hdl, void *ptr, size_t oldsize, size_t newsize)
4222676Seschrock {
4232676Seschrock 	void *ret;
4242676Seschrock 
4252676Seschrock 	if ((ret = realloc(ptr, newsize)) == NULL) {
4262676Seschrock 		(void) no_memory(hdl);
4272676Seschrock 		free(ptr);
4282676Seschrock 		return (NULL);
4292676Seschrock 	}
4302676Seschrock 
4312676Seschrock 	bzero((char *)ret + oldsize, (newsize - oldsize));
4322676Seschrock 	return (ret);
4332676Seschrock }
4342676Seschrock 
4352676Seschrock /*
436789Sahrens  * A safe form of strdup() which will die if the allocation fails.
437789Sahrens  */
438789Sahrens char *
4392082Seschrock zfs_strdup(libzfs_handle_t *hdl, const char *str)
440789Sahrens {
441789Sahrens 	char *ret;
442789Sahrens 
443789Sahrens 	if ((ret = strdup(str)) == NULL)
4442082Seschrock 		(void) no_memory(hdl);
445789Sahrens 
446789Sahrens 	return (ret);
447789Sahrens }
448789Sahrens 
449789Sahrens /*
450789Sahrens  * Convert a number to an appropriately human-readable output.
451789Sahrens  */
452789Sahrens void
453789Sahrens zfs_nicenum(uint64_t num, char *buf, size_t buflen)
454789Sahrens {
455789Sahrens 	uint64_t n = num;
456789Sahrens 	int index = 0;
457789Sahrens 	char u;
458789Sahrens 
459789Sahrens 	while (n >= 1024) {
4601162Seschrock 		n /= 1024;
461789Sahrens 		index++;
462789Sahrens 	}
463789Sahrens 
464789Sahrens 	u = " KMGTPE"[index];
465789Sahrens 
4661162Seschrock 	if (index == 0) {
467789Sahrens 		(void) snprintf(buf, buflen, "%llu", n);
4681162Seschrock 	} else if ((num & ((1ULL << 10 * index) - 1)) == 0) {
4691162Seschrock 		/*
4701162Seschrock 		 * If this is an even multiple of the base, always display
4711162Seschrock 		 * without any decimal precision.
4721162Seschrock 		 */
473789Sahrens 		(void) snprintf(buf, buflen, "%llu%c", n, u);
4741162Seschrock 	} else {
4751162Seschrock 		/*
4761162Seschrock 		 * We want to choose a precision that reflects the best choice
4771162Seschrock 		 * for fitting in 5 characters.  This can get rather tricky when
4781162Seschrock 		 * we have numbers that are very close to an order of magnitude.
4791162Seschrock 		 * For example, when displaying 10239 (which is really 9.999K),
4801162Seschrock 		 * we want only a single place of precision for 10.0K.  We could
4811162Seschrock 		 * develop some complex heuristics for this, but it's much
4821162Seschrock 		 * easier just to try each combination in turn.
4831162Seschrock 		 */
4841162Seschrock 		int i;
4851162Seschrock 		for (i = 2; i >= 0; i--) {
4861162Seschrock 			(void) snprintf(buf, buflen, "%.*f%c", i,
4871162Seschrock 			    (double)num / (1ULL << 10 * index), u);
4881162Seschrock 			if (strlen(buf) <= 5)
4891162Seschrock 				break;
4901162Seschrock 		}
4911162Seschrock 	}
492789Sahrens }
4932082Seschrock 
4942082Seschrock void
4952082Seschrock libzfs_print_on_error(libzfs_handle_t *hdl, boolean_t printerr)
4962082Seschrock {
4972082Seschrock 	hdl->libzfs_printerr = printerr;
4982082Seschrock }
4992082Seschrock 
5002082Seschrock libzfs_handle_t *
5012082Seschrock libzfs_init(void)
5022082Seschrock {
5032082Seschrock 	libzfs_handle_t *hdl;
5042082Seschrock 
5052082Seschrock 	if ((hdl = calloc(sizeof (libzfs_handle_t), 1)) == NULL) {
5062082Seschrock 		return (NULL);
5072082Seschrock 	}
5082082Seschrock 
5092500Seschrock 	if ((hdl->libzfs_fd = open(ZFS_DEV, O_RDWR)) < 0) {
5102082Seschrock 		free(hdl);
5112082Seschrock 		return (NULL);
5122082Seschrock 	}
5132082Seschrock 
5142082Seschrock 	if ((hdl->libzfs_mnttab = fopen(MNTTAB, "r")) == NULL) {
5152082Seschrock 		(void) close(hdl->libzfs_fd);
5162082Seschrock 		free(hdl);
5172082Seschrock 		return (NULL);
5182082Seschrock 	}
5192082Seschrock 
5202082Seschrock 	hdl->libzfs_sharetab = fopen("/etc/dfs/sharetab", "r");
5212082Seschrock 
5222082Seschrock 	return (hdl);
5232082Seschrock }
5242082Seschrock 
5252082Seschrock void
5262082Seschrock libzfs_fini(libzfs_handle_t *hdl)
5272082Seschrock {
5282082Seschrock 	(void) close(hdl->libzfs_fd);
5292082Seschrock 	if (hdl->libzfs_mnttab)
5302082Seschrock 		(void) fclose(hdl->libzfs_mnttab);
5312082Seschrock 	if (hdl->libzfs_sharetab)
5322082Seschrock 		(void) fclose(hdl->libzfs_sharetab);
5332082Seschrock 	namespace_clear(hdl);
5342082Seschrock 	free(hdl);
5352082Seschrock }
5362082Seschrock 
5372082Seschrock libzfs_handle_t *
5382082Seschrock zpool_get_handle(zpool_handle_t *zhp)
5392082Seschrock {
5402082Seschrock 	return (zhp->zpool_hdl);
5412082Seschrock }
5422082Seschrock 
5432082Seschrock libzfs_handle_t *
5442082Seschrock zfs_get_handle(zfs_handle_t *zhp)
5452082Seschrock {
5462082Seschrock 	return (zhp->zfs_hdl);
5472082Seschrock }
5482676Seschrock 
5492676Seschrock /*
5503635Sck153898  * Given a name, determine whether or not it's a valid path
5513635Sck153898  * (starts with '/' or "./").  If so, walk the mnttab trying
5523635Sck153898  * to match the device number.  If not, treat the path as an
5533635Sck153898  * fs/vol/snap name.
5543635Sck153898  */
5553635Sck153898 zfs_handle_t *
5563635Sck153898 zfs_path_to_zhandle(libzfs_handle_t *hdl, char *path, zfs_type_t argtype)
5573635Sck153898 {
5583635Sck153898 	struct stat64 statbuf;
5593635Sck153898 	struct extmnttab entry;
5603635Sck153898 	int ret;
5613635Sck153898 
5623635Sck153898 	if (path[0] != '/' && strncmp(path, "./", strlen("./")) != 0) {
5633635Sck153898 		/*
5643635Sck153898 		 * It's not a valid path, assume it's a name of type 'argtype'.
5653635Sck153898 		 */
5663635Sck153898 		return (zfs_open(hdl, path, argtype));
5673635Sck153898 	}
5683635Sck153898 
5693635Sck153898 	if (stat64(path, &statbuf) != 0) {
5703635Sck153898 		(void) fprintf(stderr, "%s: %s\n", path, strerror(errno));
5713635Sck153898 		return (NULL);
5723635Sck153898 	}
5733635Sck153898 
5743635Sck153898 	rewind(hdl->libzfs_mnttab);
5753635Sck153898 	while ((ret = getextmntent(hdl->libzfs_mnttab, &entry, 0)) == 0) {
5763635Sck153898 		if (makedevice(entry.mnt_major, entry.mnt_minor) ==
5773635Sck153898 		    statbuf.st_dev) {
5783635Sck153898 			break;
5793635Sck153898 		}
5803635Sck153898 	}
5813635Sck153898 	if (ret != 0) {
5823635Sck153898 		return (NULL);
5833635Sck153898 	}
5843635Sck153898 
5853635Sck153898 	if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0) {
5863635Sck153898 		(void) fprintf(stderr, gettext("'%s': not a ZFS filesystem\n"),
5873635Sck153898 		    path);
5883635Sck153898 		return (NULL);
5893635Sck153898 	}
5903635Sck153898 
5913635Sck153898 	return (zfs_open(hdl, entry.mnt_special, ZFS_TYPE_FILESYSTEM));
5923635Sck153898 }
5933635Sck153898 
5943635Sck153898 /*
5952676Seschrock  * Initialize the zc_nvlist_dst member to prepare for receiving an nvlist from
5962676Seschrock  * an ioctl().
5972676Seschrock  */
5982676Seschrock int
5992676Seschrock zcmd_alloc_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, size_t len)
6002676Seschrock {
6012676Seschrock 	if (len == 0)
6022885Sahrens 		len = 2048;
6032676Seschrock 	zc->zc_nvlist_dst_size = len;
6042676Seschrock 	if ((zc->zc_nvlist_dst = (uint64_t)(uintptr_t)
6052676Seschrock 	    zfs_alloc(hdl, zc->zc_nvlist_dst_size)) == NULL)
6062676Seschrock 		return (-1);
6072676Seschrock 
6082676Seschrock 	return (0);
6092676Seschrock }
6102676Seschrock 
6112676Seschrock /*
6122676Seschrock  * Called when an ioctl() which returns an nvlist fails with ENOMEM.  This will
6132676Seschrock  * expand the nvlist to the size specified in 'zc_nvlist_dst_size', which was
6142676Seschrock  * filled in by the kernel to indicate the actual required size.
6152676Seschrock  */
6162676Seschrock int
6172676Seschrock zcmd_expand_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc)
6182676Seschrock {
6192676Seschrock 	free((void *)(uintptr_t)zc->zc_nvlist_dst);
6202676Seschrock 	if ((zc->zc_nvlist_dst = (uint64_t)(uintptr_t)
6212676Seschrock 	    zfs_alloc(hdl, zc->zc_nvlist_dst_size))
6222676Seschrock 	    == NULL)
6232676Seschrock 		return (-1);
6242676Seschrock 
6252676Seschrock 	return (0);
6262676Seschrock }
6272676Seschrock 
6282676Seschrock /*
6292885Sahrens  * Called to free the src and dst nvlists stored in the command structure.
6302676Seschrock  */
6312676Seschrock void
6322676Seschrock zcmd_free_nvlists(zfs_cmd_t *zc)
6332676Seschrock {
6342676Seschrock 	free((void *)(uintptr_t)zc->zc_nvlist_src);
6352676Seschrock 	free((void *)(uintptr_t)zc->zc_nvlist_dst);
6362676Seschrock }
6372676Seschrock 
6382676Seschrock int
6392676Seschrock zcmd_write_src_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, nvlist_t *nvl,
6402676Seschrock     size_t *size)
6412676Seschrock {
6422676Seschrock 	char *packed;
6432676Seschrock 	size_t len;
6442676Seschrock 
6452676Seschrock 	verify(nvlist_size(nvl, &len, NV_ENCODE_NATIVE) == 0);
6462676Seschrock 
6472676Seschrock 	if ((packed = zfs_alloc(hdl, len)) == NULL)
6482676Seschrock 		return (-1);
6492676Seschrock 
6502676Seschrock 	verify(nvlist_pack(nvl, &packed, &len, NV_ENCODE_NATIVE, 0) == 0);
6512676Seschrock 
6522676Seschrock 	zc->zc_nvlist_src = (uint64_t)(uintptr_t)packed;
6532676Seschrock 	zc->zc_nvlist_src_size = len;
6542676Seschrock 
6552676Seschrock 	if (size)
6562676Seschrock 		*size = len;
6572676Seschrock 	return (0);
6582676Seschrock }
6592676Seschrock 
6602676Seschrock /*
6612676Seschrock  * Unpacks an nvlist from the ZFS ioctl command structure.
6622676Seschrock  */
6632676Seschrock int
6642676Seschrock zcmd_read_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, nvlist_t **nvlp)
6652676Seschrock {
6662676Seschrock 	if (nvlist_unpack((void *)(uintptr_t)zc->zc_nvlist_dst,
6672676Seschrock 	    zc->zc_nvlist_dst_size, nvlp, 0) != 0)
6682676Seschrock 		return (no_memory(hdl));
6692676Seschrock 
6702676Seschrock 	return (0);
6712676Seschrock }
6723912Slling 
6733912Slling static void
6743912Slling zfs_print_prop_headers(libzfs_get_cbdata_t *cbp)
6753912Slling {
6763912Slling 	zfs_proplist_t *pl = cbp->cb_proplist;
6773912Slling 	int i;
6783912Slling 	char *title;
6793912Slling 	size_t len;
6803912Slling 
6813912Slling 	cbp->cb_first = B_FALSE;
6823912Slling 	if (cbp->cb_scripted)
6833912Slling 		return;
6843912Slling 
6853912Slling 	/*
6863912Slling 	 * Start with the length of the column headers.
6873912Slling 	 */
6883912Slling 	cbp->cb_colwidths[GET_COL_NAME] = strlen(dgettext(TEXT_DOMAIN, "NAME"));
6893912Slling 	cbp->cb_colwidths[GET_COL_PROPERTY] = strlen(dgettext(TEXT_DOMAIN,
6903912Slling 	    "PROPERTY"));
6913912Slling 	cbp->cb_colwidths[GET_COL_VALUE] = strlen(dgettext(TEXT_DOMAIN,
6923912Slling 	    "VALUE"));
6933912Slling 	cbp->cb_colwidths[GET_COL_SOURCE] = strlen(dgettext(TEXT_DOMAIN,
6943912Slling 	    "SOURCE"));
6953912Slling 
6963912Slling 	/*
6973912Slling 	 * Go through and calculate the widths for each column.  For the
6983912Slling 	 * 'source' column, we kludge it up by taking the worst-case scenario of
6993912Slling 	 * inheriting from the longest name.  This is acceptable because in the
7003912Slling 	 * majority of cases 'SOURCE' is the last column displayed, and we don't
7013912Slling 	 * use the width anyway.  Note that the 'VALUE' column can be oversized,
7023912Slling 	 * if the name of the property is much longer the any values we find.
7033912Slling 	 */
7043912Slling 	for (pl = cbp->cb_proplist; pl != NULL; pl = pl->pl_next) {
7053912Slling 		/*
7063912Slling 		 * 'PROPERTY' column
7073912Slling 		 */
7083912Slling 		if (pl->pl_prop != ZFS_PROP_INVAL) {
7093912Slling 			len = strlen(zfs_prop_to_name(pl->pl_prop));
7103912Slling 			if (len > cbp->cb_colwidths[GET_COL_PROPERTY])
7113912Slling 				cbp->cb_colwidths[GET_COL_PROPERTY] = len;
7123912Slling 		} else {
7133912Slling 			len = strlen(pl->pl_user_prop);
7143912Slling 			if (len > cbp->cb_colwidths[GET_COL_PROPERTY])
7153912Slling 				cbp->cb_colwidths[GET_COL_PROPERTY] = len;
7163912Slling 		}
7173912Slling 
7183912Slling 		/*
7193912Slling 		 * 'VALUE' column
7203912Slling 		 */
7213912Slling 		if ((pl->pl_prop != ZFS_PROP_NAME || !pl->pl_all) &&
7223912Slling 		    pl->pl_width > cbp->cb_colwidths[GET_COL_VALUE])
7233912Slling 			cbp->cb_colwidths[GET_COL_VALUE] = pl->pl_width;
7243912Slling 
7253912Slling 		/*
7263912Slling 		 * 'NAME' and 'SOURCE' columns
7273912Slling 		 */
7283912Slling 		if (pl->pl_prop == ZFS_PROP_NAME &&
7293912Slling 		    pl->pl_width > cbp->cb_colwidths[GET_COL_NAME]) {
7303912Slling 			cbp->cb_colwidths[GET_COL_NAME] = pl->pl_width;
7313912Slling 			cbp->cb_colwidths[GET_COL_SOURCE] = pl->pl_width +
7323912Slling 			    strlen(dgettext(TEXT_DOMAIN, "inherited from"));
7333912Slling 		}
7343912Slling 	}
7353912Slling 
7363912Slling 	/*
7373912Slling 	 * Now go through and print the headers.
7383912Slling 	 */
7393912Slling 	for (i = 0; i < 4; i++) {
7403912Slling 		switch (cbp->cb_columns[i]) {
7413912Slling 		case GET_COL_NAME:
7423912Slling 			title = dgettext(TEXT_DOMAIN, "NAME");
7433912Slling 			break;
7443912Slling 		case GET_COL_PROPERTY:
7453912Slling 			title = dgettext(TEXT_DOMAIN, "PROPERTY");
7463912Slling 			break;
7473912Slling 		case GET_COL_VALUE:
7483912Slling 			title = dgettext(TEXT_DOMAIN, "VALUE");
7493912Slling 			break;
7503912Slling 		case GET_COL_SOURCE:
7513912Slling 			title = dgettext(TEXT_DOMAIN, "SOURCE");
7523912Slling 			break;
7533912Slling 		default:
7543912Slling 			title = NULL;
7553912Slling 		}
7563912Slling 
7573912Slling 		if (title != NULL) {
7583912Slling 			if (i == 3 || cbp->cb_columns[i + 1] == 0)
7593912Slling 				(void) printf("%s", title);
7603912Slling 			else
7613912Slling 				(void) printf("%-*s  ",
7623912Slling 				    cbp->cb_colwidths[cbp->cb_columns[i]],
7633912Slling 				    title);
7643912Slling 		}
7653912Slling 	}
7663912Slling 	(void) printf("\n");
7673912Slling }
7683912Slling 
7693912Slling /*
7703912Slling  * Display a single line of output, according to the settings in the callback
7713912Slling  * structure.
7723912Slling  */
7733912Slling void
7743912Slling libzfs_print_one_property(const char *name, libzfs_get_cbdata_t *cbp,
7753912Slling     const char *propname, const char *value, zfs_source_t sourcetype,
7763912Slling     const char *source)
7773912Slling {
7783912Slling 	int i;
7793912Slling 	const char *str;
7803912Slling 	char buf[128];
7813912Slling 
7823912Slling 	/*
7833912Slling 	 * Ignore those source types that the user has chosen to ignore.
7843912Slling 	 */
7853912Slling 	if ((sourcetype & cbp->cb_sources) == 0)
7863912Slling 		return;
7873912Slling 
7883912Slling 	if (cbp->cb_first)
7893912Slling 		zfs_print_prop_headers(cbp);
7903912Slling 
7913912Slling 	for (i = 0; i < 4; i++) {
7923912Slling 		switch (cbp->cb_columns[i]) {
7933912Slling 		case GET_COL_NAME:
7943912Slling 			str = name;
7953912Slling 			break;
7963912Slling 
7973912Slling 		case GET_COL_PROPERTY:
7983912Slling 			str = propname;
7993912Slling 			break;
8003912Slling 
8013912Slling 		case GET_COL_VALUE:
8023912Slling 			str = value;
8033912Slling 			break;
8043912Slling 
8053912Slling 		case GET_COL_SOURCE:
8063912Slling 			switch (sourcetype) {
8073912Slling 			case ZFS_SRC_NONE:
8083912Slling 				str = "-";
8093912Slling 				break;
8103912Slling 
8113912Slling 			case ZFS_SRC_DEFAULT:
8123912Slling 				str = "default";
8133912Slling 				break;
8143912Slling 
8153912Slling 			case ZFS_SRC_LOCAL:
8163912Slling 				str = "local";
8173912Slling 				break;
8183912Slling 
8193912Slling 			case ZFS_SRC_TEMPORARY:
8203912Slling 				str = "temporary";
8213912Slling 				break;
8223912Slling 
8233912Slling 			case ZFS_SRC_INHERITED:
8243912Slling 				(void) snprintf(buf, sizeof (buf),
8253912Slling 				    "inherited from %s", source);
8263912Slling 				str = buf;
8273912Slling 				break;
8283912Slling 			}
8293912Slling 			break;
8303912Slling 
8313912Slling 		default:
8323912Slling 			continue;
8333912Slling 		}
8343912Slling 
8353912Slling 		if (cbp->cb_columns[i + 1] == 0)
8363912Slling 			(void) printf("%s", str);
8373912Slling 		else if (cbp->cb_scripted)
8383912Slling 			(void) printf("%s\t", str);
8393912Slling 		else
8403912Slling 			(void) printf("%-*s  ",
8413912Slling 			    cbp->cb_colwidths[cbp->cb_columns[i]],
8423912Slling 			    str);
8433912Slling 
8443912Slling 	}
8453912Slling 
8463912Slling 	(void) printf("\n");
8473912Slling }
848