xref: /netbsd-src/external/cddl/osnet/dist/lib/libzfs/common/libzfs_util.c (revision ba2539a9805a0544ff82c0003cc02fe1eee5603d)
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 (c) 2013, Joyent, Inc. All rights reserved.
25  * Copyright (c) 2011, 2015 by Delphix. All rights reserved.
26  * Copyright 2016 Igor Kozhukhov <ikozhukhov@gmail.com>
27  */
28 
29 /*
30  * Internal utility routines for the ZFS library.
31  */
32 
33 #include <sys/param.h>
34 #include <sys/linker.h>
35 #include <sys/module.h>
36 #include <sys/stat.h>
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 <ctype.h>
47 #include <math.h>
48 #include <sys/mnttab.h>
49 #include <sys/mntent.h>
50 #include <sys/types.h>
51 #include <sys/ioctl.h>
52 
53 #ifdef __NetBSD__
54 #include <sys/statvfs.h>
55 #endif
56 
57 #include <libzfs.h>
58 #include <libzfs_core.h>
59 
60 #include "libzfs_impl.h"
61 #include "zfs_prop.h"
62 #include "zfeature_common.h"
63 
64 
65 int
libzfs_errno(libzfs_handle_t * hdl)66 libzfs_errno(libzfs_handle_t *hdl)
67 {
68 	return (hdl->libzfs_error);
69 }
70 
71 const char *
libzfs_error_action(libzfs_handle_t * hdl)72 libzfs_error_action(libzfs_handle_t *hdl)
73 {
74 	return (hdl->libzfs_action);
75 }
76 
77 const char *
libzfs_error_description(libzfs_handle_t * hdl)78 libzfs_error_description(libzfs_handle_t *hdl)
79 {
80 	if (hdl->libzfs_desc[0] != '\0')
81 		return (hdl->libzfs_desc);
82 
83 	switch (hdl->libzfs_error) {
84 	case EZFS_NOMEM:
85 		return (dgettext(TEXT_DOMAIN, "out of memory"));
86 	case EZFS_BADPROP:
87 		return (dgettext(TEXT_DOMAIN, "invalid property value"));
88 	case EZFS_PROPREADONLY:
89 		return (dgettext(TEXT_DOMAIN, "read-only property"));
90 	case EZFS_PROPTYPE:
91 		return (dgettext(TEXT_DOMAIN, "property doesn't apply to "
92 		    "datasets of this type"));
93 	case EZFS_PROPNONINHERIT:
94 		return (dgettext(TEXT_DOMAIN, "property cannot be inherited"));
95 	case EZFS_PROPSPACE:
96 		return (dgettext(TEXT_DOMAIN, "invalid quota or reservation"));
97 	case EZFS_BADTYPE:
98 		return (dgettext(TEXT_DOMAIN, "operation not applicable to "
99 		    "datasets of this type"));
100 	case EZFS_BUSY:
101 		return (dgettext(TEXT_DOMAIN, "pool or dataset is busy"));
102 	case EZFS_EXISTS:
103 		return (dgettext(TEXT_DOMAIN, "pool or dataset exists"));
104 	case EZFS_NOENT:
105 		return (dgettext(TEXT_DOMAIN, "no such pool or dataset"));
106 	case EZFS_BADSTREAM:
107 		return (dgettext(TEXT_DOMAIN, "invalid backup stream"));
108 	case EZFS_DSREADONLY:
109 		return (dgettext(TEXT_DOMAIN, "dataset is read-only"));
110 	case EZFS_VOLTOOBIG:
111 		return (dgettext(TEXT_DOMAIN, "volume size exceeds limit for "
112 		    "this system"));
113 	case EZFS_INVALIDNAME:
114 		return (dgettext(TEXT_DOMAIN, "invalid name"));
115 	case EZFS_BADRESTORE:
116 		return (dgettext(TEXT_DOMAIN, "unable to restore to "
117 		    "destination"));
118 	case EZFS_BADBACKUP:
119 		return (dgettext(TEXT_DOMAIN, "backup failed"));
120 	case EZFS_BADTARGET:
121 		return (dgettext(TEXT_DOMAIN, "invalid target vdev"));
122 	case EZFS_NODEVICE:
123 		return (dgettext(TEXT_DOMAIN, "no such device in pool"));
124 	case EZFS_BADDEV:
125 		return (dgettext(TEXT_DOMAIN, "invalid device"));
126 	case EZFS_NOREPLICAS:
127 		return (dgettext(TEXT_DOMAIN, "no valid replicas"));
128 	case EZFS_RESILVERING:
129 		return (dgettext(TEXT_DOMAIN, "currently resilvering"));
130 	case EZFS_BADVERSION:
131 		return (dgettext(TEXT_DOMAIN, "unsupported version or "
132 		    "feature"));
133 	case EZFS_POOLUNAVAIL:
134 		return (dgettext(TEXT_DOMAIN, "pool is unavailable"));
135 	case EZFS_DEVOVERFLOW:
136 		return (dgettext(TEXT_DOMAIN, "too many devices in one vdev"));
137 	case EZFS_BADPATH:
138 		return (dgettext(TEXT_DOMAIN, "must be an absolute path"));
139 	case EZFS_CROSSTARGET:
140 		return (dgettext(TEXT_DOMAIN, "operation crosses datasets or "
141 		    "pools"));
142 	case EZFS_ZONED:
143 		return (dgettext(TEXT_DOMAIN, "dataset in use by local zone"));
144 	case EZFS_MOUNTFAILED:
145 		return (dgettext(TEXT_DOMAIN, "mount failed"));
146 	case EZFS_UMOUNTFAILED:
147 		return (dgettext(TEXT_DOMAIN, "umount failed"));
148 	case EZFS_UNSHARENFSFAILED:
149 		return (dgettext(TEXT_DOMAIN, "unshare(1M) failed"));
150 	case EZFS_SHARENFSFAILED:
151 		return (dgettext(TEXT_DOMAIN, "share(1M) failed"));
152 	case EZFS_UNSHARESMBFAILED:
153 		return (dgettext(TEXT_DOMAIN, "smb remove share failed"));
154 	case EZFS_SHARESMBFAILED:
155 		return (dgettext(TEXT_DOMAIN, "smb add share failed"));
156 	case EZFS_PERM:
157 		return (dgettext(TEXT_DOMAIN, "permission denied"));
158 	case EZFS_NOSPC:
159 		return (dgettext(TEXT_DOMAIN, "out of space"));
160 	case EZFS_FAULT:
161 		return (dgettext(TEXT_DOMAIN, "bad address"));
162 	case EZFS_IO:
163 		return (dgettext(TEXT_DOMAIN, "I/O error"));
164 	case EZFS_INTR:
165 		return (dgettext(TEXT_DOMAIN, "signal received"));
166 	case EZFS_ISSPARE:
167 		return (dgettext(TEXT_DOMAIN, "device is reserved as a hot "
168 		    "spare"));
169 	case EZFS_INVALCONFIG:
170 		return (dgettext(TEXT_DOMAIN, "invalid vdev configuration"));
171 	case EZFS_RECURSIVE:
172 		return (dgettext(TEXT_DOMAIN, "recursive dataset dependency"));
173 	case EZFS_NOHISTORY:
174 		return (dgettext(TEXT_DOMAIN, "no history available"));
175 	case EZFS_POOLPROPS:
176 		return (dgettext(TEXT_DOMAIN, "failed to retrieve "
177 		    "pool properties"));
178 	case EZFS_POOL_NOTSUP:
179 		return (dgettext(TEXT_DOMAIN, "operation not supported "
180 		    "on this type of pool"));
181 	case EZFS_POOL_INVALARG:
182 		return (dgettext(TEXT_DOMAIN, "invalid argument for "
183 		    "this pool operation"));
184 	case EZFS_NAMETOOLONG:
185 		return (dgettext(TEXT_DOMAIN, "dataset name is too long"));
186 	case EZFS_OPENFAILED:
187 		return (dgettext(TEXT_DOMAIN, "open failed"));
188 	case EZFS_NOCAP:
189 		return (dgettext(TEXT_DOMAIN,
190 		    "disk capacity information could not be retrieved"));
191 	case EZFS_LABELFAILED:
192 		return (dgettext(TEXT_DOMAIN, "write of label failed"));
193 	case EZFS_BADWHO:
194 		return (dgettext(TEXT_DOMAIN, "invalid user/group"));
195 	case EZFS_BADPERM:
196 		return (dgettext(TEXT_DOMAIN, "invalid permission"));
197 	case EZFS_BADPERMSET:
198 		return (dgettext(TEXT_DOMAIN, "invalid permission set name"));
199 	case EZFS_NODELEGATION:
200 		return (dgettext(TEXT_DOMAIN, "delegated administration is "
201 		    "disabled on pool"));
202 	case EZFS_BADCACHE:
203 		return (dgettext(TEXT_DOMAIN, "invalid or missing cache file"));
204 	case EZFS_ISL2CACHE:
205 		return (dgettext(TEXT_DOMAIN, "device is in use as a cache"));
206 	case EZFS_VDEVNOTSUP:
207 		return (dgettext(TEXT_DOMAIN, "vdev specification is not "
208 		    "supported"));
209 	case EZFS_NOTSUP:
210 		return (dgettext(TEXT_DOMAIN, "operation not supported "
211 		    "on this dataset"));
212 	case EZFS_ACTIVE_SPARE:
213 		return (dgettext(TEXT_DOMAIN, "pool has active shared spare "
214 		    "device"));
215 	case EZFS_UNPLAYED_LOGS:
216 		return (dgettext(TEXT_DOMAIN, "log device has unplayed intent "
217 		    "logs"));
218 	case EZFS_REFTAG_RELE:
219 		return (dgettext(TEXT_DOMAIN, "no such tag on this dataset"));
220 	case EZFS_REFTAG_HOLD:
221 		return (dgettext(TEXT_DOMAIN, "tag already exists on this "
222 		    "dataset"));
223 	case EZFS_TAGTOOLONG:
224 		return (dgettext(TEXT_DOMAIN, "tag too long"));
225 	case EZFS_PIPEFAILED:
226 		return (dgettext(TEXT_DOMAIN, "pipe create failed"));
227 	case EZFS_THREADCREATEFAILED:
228 		return (dgettext(TEXT_DOMAIN, "thread create failed"));
229 	case EZFS_POSTSPLIT_ONLINE:
230 		return (dgettext(TEXT_DOMAIN, "disk was split from this pool "
231 		    "into a new one"));
232 	case EZFS_SCRUBBING:
233 		return (dgettext(TEXT_DOMAIN, "currently scrubbing; "
234 		    "use 'zpool scrub -s' to cancel current scrub"));
235 	case EZFS_NO_SCRUB:
236 		return (dgettext(TEXT_DOMAIN, "there is no active scrub"));
237 	case EZFS_DIFF:
238 		return (dgettext(TEXT_DOMAIN, "unable to generate diffs"));
239 	case EZFS_DIFFDATA:
240 		return (dgettext(TEXT_DOMAIN, "invalid diff data"));
241 	case EZFS_POOLREADONLY:
242 		return (dgettext(TEXT_DOMAIN, "pool is read-only"));
243 	case EZFS_UNKNOWN:
244 		return (dgettext(TEXT_DOMAIN, "unknown error"));
245 	default:
246 		assert(hdl->libzfs_error == 0);
247 		return (dgettext(TEXT_DOMAIN, "no error"));
248 	}
249 }
250 
251 /*PRINTFLIKE2*/
252 void
zfs_error_aux(libzfs_handle_t * hdl,const char * fmt,...)253 zfs_error_aux(libzfs_handle_t *hdl, const char *fmt, ...)
254 {
255 	va_list ap;
256 
257 	va_start(ap, fmt);
258 
259 	(void) vsnprintf(hdl->libzfs_desc, sizeof (hdl->libzfs_desc),
260 	    fmt, ap);
261 	hdl->libzfs_desc_active = 1;
262 
263 	va_end(ap);
264 }
265 
266 static void
zfs_verror(libzfs_handle_t * hdl,int error,const char * fmt,va_list ap)267 zfs_verror(libzfs_handle_t *hdl, int error, const char *fmt, va_list ap)
268 {
269 	(void) vsnprintf(hdl->libzfs_action, sizeof (hdl->libzfs_action),
270 	    fmt, ap);
271 	hdl->libzfs_error = error;
272 
273 	if (hdl->libzfs_desc_active)
274 		hdl->libzfs_desc_active = 0;
275 	else
276 		hdl->libzfs_desc[0] = '\0';
277 
278 	if (hdl->libzfs_printerr) {
279 		if (error == EZFS_UNKNOWN) {
280 			(void) fprintf(stderr, dgettext(TEXT_DOMAIN, "internal "
281 			    "error: %s\n"), libzfs_error_description(hdl));
282 			abort();
283 		}
284 
285 		(void) fprintf(stderr, "%s: %s\n", hdl->libzfs_action,
286 		    libzfs_error_description(hdl));
287 		if (error == EZFS_NOMEM)
288 			exit(1);
289 	}
290 }
291 
292 int
zfs_error(libzfs_handle_t * hdl,int error,const char * msg)293 zfs_error(libzfs_handle_t *hdl, int error, const char *msg)
294 {
295 	return (zfs_error_fmt(hdl, error, "%s", msg));
296 }
297 
298 /*PRINTFLIKE3*/
299 int
zfs_error_fmt(libzfs_handle_t * hdl,int error,const char * fmt,...)300 zfs_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
301 {
302 	va_list ap;
303 
304 	va_start(ap, fmt);
305 
306 	zfs_verror(hdl, error, fmt, ap);
307 
308 	va_end(ap);
309 
310 	return (-1);
311 }
312 
313 static int
zfs_common_error(libzfs_handle_t * hdl,int error,const char * fmt,va_list ap)314 zfs_common_error(libzfs_handle_t *hdl, int error, const char *fmt,
315     va_list ap)
316 {
317 	switch (error) {
318 	case EPERM:
319 	case EACCES:
320 		zfs_verror(hdl, EZFS_PERM, fmt, ap);
321 		return (-1);
322 
323 	case ECANCELED:
324 		zfs_verror(hdl, EZFS_NODELEGATION, fmt, ap);
325 		return (-1);
326 
327 	case EIO:
328 		zfs_verror(hdl, EZFS_IO, fmt, ap);
329 		return (-1);
330 
331 	case EFAULT:
332 		zfs_verror(hdl, EZFS_FAULT, fmt, ap);
333 		return (-1);
334 
335 	case EINTR:
336 		zfs_verror(hdl, EZFS_INTR, fmt, ap);
337 		return (-1);
338 	}
339 
340 	return (0);
341 }
342 
343 int
zfs_standard_error(libzfs_handle_t * hdl,int error,const char * msg)344 zfs_standard_error(libzfs_handle_t *hdl, int error, const char *msg)
345 {
346 	return (zfs_standard_error_fmt(hdl, error, "%s", msg));
347 }
348 
349 /*PRINTFLIKE3*/
350 int
zfs_standard_error_fmt(libzfs_handle_t * hdl,int error,const char * fmt,...)351 zfs_standard_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
352 {
353 	va_list ap;
354 
355 	va_start(ap, fmt);
356 
357 	if (zfs_common_error(hdl, error, fmt, ap) != 0) {
358 		va_end(ap);
359 		return (-1);
360 	}
361 
362 	switch (error) {
363 	case ENXIO:
364 	case ENODEV:
365 	case EPIPE:
366 		zfs_verror(hdl, EZFS_IO, fmt, ap);
367 		break;
368 
369 	case ENOENT:
370 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
371 		    "dataset does not exist"));
372 		zfs_verror(hdl, EZFS_NOENT, fmt, ap);
373 		break;
374 
375 	case ENOSPC:
376 	case EDQUOT:
377 		zfs_verror(hdl, EZFS_NOSPC, fmt, ap);
378 		va_end(ap);
379 		return (-1);
380 
381 	case EEXIST:
382 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
383 		    "dataset already exists"));
384 		zfs_verror(hdl, EZFS_EXISTS, fmt, ap);
385 		break;
386 
387 	case EBUSY:
388 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
389 		    "dataset is busy"));
390 		zfs_verror(hdl, EZFS_BUSY, fmt, ap);
391 		break;
392 	case EROFS:
393 		zfs_verror(hdl, EZFS_POOLREADONLY, fmt, ap);
394 		break;
395 	case ENAMETOOLONG:
396 		zfs_verror(hdl, EZFS_NAMETOOLONG, fmt, ap);
397 		break;
398 	case ENOTSUP:
399 		zfs_verror(hdl, EZFS_BADVERSION, fmt, ap);
400 		break;
401 	case EAGAIN:
402 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
403 		    "pool I/O is currently suspended"));
404 		zfs_verror(hdl, EZFS_POOLUNAVAIL, fmt, ap);
405 		break;
406 	default:
407 		zfs_error_aux(hdl, strerror(error));
408 		zfs_verror(hdl, EZFS_UNKNOWN, fmt, ap);
409 		break;
410 	}
411 
412 	va_end(ap);
413 	return (-1);
414 }
415 
416 int
zpool_standard_error(libzfs_handle_t * hdl,int error,const char * msg)417 zpool_standard_error(libzfs_handle_t *hdl, int error, const char *msg)
418 {
419 	return (zpool_standard_error_fmt(hdl, error, "%s", msg));
420 }
421 
422 /*PRINTFLIKE3*/
423 int
zpool_standard_error_fmt(libzfs_handle_t * hdl,int error,const char * fmt,...)424 zpool_standard_error_fmt(libzfs_handle_t *hdl, int error, const char *fmt, ...)
425 {
426 	va_list ap;
427 
428 	va_start(ap, fmt);
429 
430 	if (zfs_common_error(hdl, error, fmt, ap) != 0) {
431 		va_end(ap);
432 		return (-1);
433 	}
434 
435 	switch (error) {
436 	case ENODEV:
437 		zfs_verror(hdl, EZFS_NODEVICE, fmt, ap);
438 		break;
439 
440 	case ENOENT:
441 		zfs_error_aux(hdl,
442 		    dgettext(TEXT_DOMAIN, "no such pool or dataset"));
443 		zfs_verror(hdl, EZFS_NOENT, fmt, ap);
444 		break;
445 
446 	case EEXIST:
447 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
448 		    "pool already exists"));
449 		zfs_verror(hdl, EZFS_EXISTS, fmt, ap);
450 		break;
451 
452 	case EBUSY:
453 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "pool is busy"));
454 		zfs_verror(hdl, EZFS_BUSY, fmt, ap);
455 		break;
456 
457 	case ENXIO:
458 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
459 		    "one or more devices is currently unavailable"));
460 		zfs_verror(hdl, EZFS_BADDEV, fmt, ap);
461 		break;
462 
463 	case ENAMETOOLONG:
464 		zfs_verror(hdl, EZFS_DEVOVERFLOW, fmt, ap);
465 		break;
466 
467 	case ENOTSUP:
468 		zfs_verror(hdl, EZFS_POOL_NOTSUP, fmt, ap);
469 		break;
470 
471 	case EINVAL:
472 		zfs_verror(hdl, EZFS_POOL_INVALARG, fmt, ap);
473 		break;
474 
475 	case ENOSPC:
476 	case EDQUOT:
477 		zfs_verror(hdl, EZFS_NOSPC, fmt, ap);
478 		va_end(ap);
479 		return (-1);
480 
481 	case EAGAIN:
482 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
483 		    "pool I/O is currently suspended"));
484 		zfs_verror(hdl, EZFS_POOLUNAVAIL, fmt, ap);
485 		break;
486 
487 	case EROFS:
488 		zfs_verror(hdl, EZFS_POOLREADONLY, fmt, ap);
489 		break;
490 
491 	default:
492 		zfs_error_aux(hdl, strerror(error));
493 		zfs_verror(hdl, EZFS_UNKNOWN, fmt, ap);
494 	}
495 
496 	va_end(ap);
497 	return (-1);
498 }
499 
500 /*
501  * Display an out of memory error message and abort the current program.
502  */
503 int
no_memory(libzfs_handle_t * hdl)504 no_memory(libzfs_handle_t *hdl)
505 {
506 	return (zfs_error(hdl, EZFS_NOMEM, "internal error"));
507 }
508 
509 /*
510  * A safe form of malloc() which will die if the allocation fails.
511  */
512 void *
zfs_alloc(libzfs_handle_t * hdl,size_t size)513 zfs_alloc(libzfs_handle_t *hdl, size_t size)
514 {
515 	void *data;
516 
517 	if ((data = calloc(1, size)) == NULL)
518 		(void) no_memory(hdl);
519 
520 	return (data);
521 }
522 
523 /*
524  * A safe form of asprintf() which will die if the allocation fails.
525  */
526 /*PRINTFLIKE2*/
527 char *
zfs_asprintf(libzfs_handle_t * hdl,const char * fmt,...)528 zfs_asprintf(libzfs_handle_t *hdl, const char *fmt, ...)
529 {
530 	va_list ap;
531 	char *ret;
532 	int err;
533 
534 	va_start(ap, fmt);
535 
536 	err = vasprintf(&ret, fmt, ap);
537 
538 	va_end(ap);
539 
540 	if (err < 0)
541 		(void) no_memory(hdl);
542 
543 	return (ret);
544 }
545 
546 /*
547  * A safe form of realloc(), which also zeroes newly allocated space.
548  */
549 void *
zfs_realloc(libzfs_handle_t * hdl,void * ptr,size_t oldsize,size_t newsize)550 zfs_realloc(libzfs_handle_t *hdl, void *ptr, size_t oldsize, size_t newsize)
551 {
552 	void *ret;
553 
554 	if ((ret = realloc(ptr, newsize)) == NULL) {
555 		(void) no_memory(hdl);
556 		return (NULL);
557 	}
558 
559 	bzero((char *)ret + oldsize, (newsize - oldsize));
560 	return (ret);
561 }
562 
563 /*
564  * A safe form of strdup() which will die if the allocation fails.
565  */
566 char *
zfs_strdup(libzfs_handle_t * hdl,const char * str)567 zfs_strdup(libzfs_handle_t *hdl, const char *str)
568 {
569 	char *ret;
570 
571 	if ((ret = strdup(str)) == NULL)
572 		(void) no_memory(hdl);
573 
574 	return (ret);
575 }
576 
577 /*
578  * Convert a number to an appropriately human-readable output.
579  */
580 void
zfs_nicenum(uint64_t num,char * buf,size_t buflen)581 zfs_nicenum(uint64_t num, char *buf, size_t buflen)
582 {
583 	uint64_t n = num;
584 	int index = 0;
585 	char u;
586 
587 	while (n >= 1024) {
588 		n /= 1024;
589 		index++;
590 	}
591 
592 	u = " KMGTPE"[index];
593 
594 	if (index == 0) {
595 		(void) snprintf(buf, buflen, "%llu", n);
596 	} else if ((num & ((1ULL << 10 * index) - 1)) == 0) {
597 		/*
598 		 * If this is an even multiple of the base, always display
599 		 * without any decimal precision.
600 		 */
601 		(void) snprintf(buf, buflen, "%llu%c", n, u);
602 	} else {
603 		/*
604 		 * We want to choose a precision that reflects the best choice
605 		 * for fitting in 5 characters.  This can get rather tricky when
606 		 * we have numbers that are very close to an order of magnitude.
607 		 * For example, when displaying 10239 (which is really 9.999K),
608 		 * we want only a single place of precision for 10.0K.  We could
609 		 * develop some complex heuristics for this, but it's much
610 		 * easier just to try each combination in turn.
611 		 */
612 		int i;
613 		for (i = 2; i >= 0; i--) {
614 			if (snprintf(buf, buflen, "%.*f%c", i,
615 			    (double)num / (1ULL << 10 * index), u) <= 5)
616 				break;
617 		}
618 	}
619 }
620 
621 void
libzfs_print_on_error(libzfs_handle_t * hdl,boolean_t printerr)622 libzfs_print_on_error(libzfs_handle_t *hdl, boolean_t printerr)
623 {
624 	hdl->libzfs_printerr = printerr;
625 }
626 
627 static int
libzfs_load(void)628 libzfs_load(void)
629 {
630 #ifdef __FreeBSD__
631 	if (modfind("zfs") < 0) {
632 		/* Not present in kernel, try loading it. */
633 		if (kldload("zfs") < 0 || modfind("zfs") < 0) {
634 			if (errno != EEXIST)
635 				return (-1);
636 		}
637 	}
638 #endif
639 #ifdef __NetBSD__
640 	modctl_load_t cmdargs;
641 
642 	cmdargs.ml_filename = "zfs";
643 	cmdargs.ml_flags = MODCTL_NO_PROP;
644 	cmdargs.ml_props = NULL;
645 	cmdargs.ml_propslen = 0;
646 
647 	if (modctl(MODCTL_LOAD, &cmdargs) < 0 && errno != EEXIST)
648 		return (-1);
649 #endif
650 	return (0);
651 }
652 
653 libzfs_handle_t *
libzfs_init(void)654 libzfs_init(void)
655 {
656 	libzfs_handle_t *hdl;
657 
658 	if ((hdl = calloc(1, sizeof (libzfs_handle_t))) == NULL) {
659 		return (NULL);
660 	}
661 
662 	if (libzfs_load() < 0) {
663 		free(hdl);
664 		return (NULL);
665 	}
666 
667 	if ((hdl->libzfs_fd = open(ZFS_DEV, O_RDWR)) < 0) {
668 		free(hdl);
669 		return (NULL);
670 	}
671 
672 	if ((hdl->libzfs_mnttab = fopen(MNTTAB, "r")) == NULL) {
673 		(void) close(hdl->libzfs_fd);
674 		free(hdl);
675 		return (NULL);
676 	}
677 
678 	hdl->libzfs_sharetab = fopen(ZFS_EXPORTS_PATH, "r");
679 
680 	if (libzfs_core_init() != 0) {
681 		(void) close(hdl->libzfs_fd);
682 		(void) fclose(hdl->libzfs_mnttab);
683 		(void) fclose(hdl->libzfs_sharetab);
684 		free(hdl);
685 		return (NULL);
686 	}
687 
688 	zfs_prop_init();
689 	zpool_prop_init();
690 	zpool_feature_init();
691 	libzfs_mnttab_init(hdl);
692 
693 	return (hdl);
694 }
695 
696 void
libzfs_fini(libzfs_handle_t * hdl)697 libzfs_fini(libzfs_handle_t *hdl)
698 {
699 	(void) close(hdl->libzfs_fd);
700 	if (hdl->libzfs_mnttab)
701 		(void) fclose(hdl->libzfs_mnttab);
702 	if (hdl->libzfs_sharetab)
703 		(void) fclose(hdl->libzfs_sharetab);
704 	zfs_uninit_libshare(hdl);
705 	zpool_free_handles(hdl);
706 #ifdef illumos
707 	libzfs_fru_clear(hdl, B_TRUE);
708 #endif
709 	namespace_clear(hdl);
710 	libzfs_mnttab_fini(hdl);
711 	libzfs_core_fini();
712 	free(hdl);
713 }
714 
715 libzfs_handle_t *
zpool_get_handle(zpool_handle_t * zhp)716 zpool_get_handle(zpool_handle_t *zhp)
717 {
718 	return (zhp->zpool_hdl);
719 }
720 
721 libzfs_handle_t *
zfs_get_handle(zfs_handle_t * zhp)722 zfs_get_handle(zfs_handle_t *zhp)
723 {
724 	return (zhp->zfs_hdl);
725 }
726 
727 zpool_handle_t *
zfs_get_pool_handle(const zfs_handle_t * zhp)728 zfs_get_pool_handle(const zfs_handle_t *zhp)
729 {
730 	return (zhp->zpool_hdl);
731 }
732 
733 /*
734  * Given a name, determine whether or not it's a valid path
735  * (starts with '/' or "./").  If so, walk the mnttab trying
736  * to match the device number.  If not, treat the path as an
737  * fs/vol/snap name.
738  */
739 zfs_handle_t *
zfs_path_to_zhandle(libzfs_handle_t * hdl,char * path,zfs_type_t argtype)740 zfs_path_to_zhandle(libzfs_handle_t *hdl, char *path, zfs_type_t argtype)
741 {
742 	struct stat64 statbuf;
743 	struct extmnttab entry;
744 	int ret;
745 
746 	if (path[0] != '/' && strncmp(path, "./", strlen("./")) != 0) {
747 		/*
748 		 * It's not a valid path, assume it's a name of type 'argtype'.
749 		 */
750 		return (zfs_open(hdl, path, argtype));
751 	}
752 
753 	if (stat64(path, &statbuf) != 0) {
754 		(void) fprintf(stderr, "%s: %s\n", path, strerror(errno));
755 		return (NULL);
756 	}
757 
758 #ifdef illumos
759 	rewind(hdl->libzfs_mnttab);
760 	while ((ret = getextmntent(hdl->libzfs_mnttab, &entry, 0)) == 0) {
761 		if (makedevice(entry.mnt_major, entry.mnt_minor) ==
762 		    statbuf.st_dev) {
763 			break;
764 		}
765 	}
766 #endif	/* illumos */
767 #ifdef __FreeBSD__
768 	{
769 		struct statfs sfs;
770 
771 		ret = statfs(path, &sfs);
772 		if (ret == 0)
773 			statfs2mnttab(&sfs, &entry);
774 		else {
775 			(void) fprintf(stderr, "%s: %s\n", path,
776 			    strerror(errno));
777 		}
778 	}
779 #endif	/* __FreeBSD__ */
780 #ifdef __NetBSD__
781 	{
782 		struct statvfs sfs;
783 
784 		ret = statvfs(path, &sfs);
785 		if (ret == 0)
786 			statvfs2mnttab(&sfs, &entry);
787 		else {
788 			(void) fprintf(stderr, "%s: %s\n", path,
789 			    strerror(errno));
790 		}
791 	}
792 #endif	/* __NetBSD__ */
793 	if (ret != 0) {
794 		return (NULL);
795 	}
796 
797 	if (strcmp(entry.mnt_fstype, MNTTYPE_ZFS) != 0) {
798 		(void) fprintf(stderr, gettext("'%s': not a ZFS filesystem\n"),
799 		    path);
800 		return (NULL);
801 	}
802 
803 	return (zfs_open(hdl, entry.mnt_special, ZFS_TYPE_FILESYSTEM));
804 }
805 
806 /*
807  * Initialize the zc_nvlist_dst member to prepare for receiving an nvlist from
808  * an ioctl().
809  */
810 int
zcmd_alloc_dst_nvlist(libzfs_handle_t * hdl,zfs_cmd_t * zc,size_t len)811 zcmd_alloc_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, size_t len)
812 {
813 	if (len == 0)
814 		len = 16 * 1024;
815 	zc->zc_nvlist_dst_size = len;
816 	zc->zc_nvlist_dst =
817 	    (uint64_t)(uintptr_t)zfs_alloc(hdl, zc->zc_nvlist_dst_size);
818 	if (zc->zc_nvlist_dst == 0)
819 		return (-1);
820 
821 	return (0);
822 }
823 
824 /*
825  * Called when an ioctl() which returns an nvlist fails with ENOMEM.  This will
826  * expand the nvlist to the size specified in 'zc_nvlist_dst_size', which was
827  * filled in by the kernel to indicate the actual required size.
828  */
829 int
zcmd_expand_dst_nvlist(libzfs_handle_t * hdl,zfs_cmd_t * zc)830 zcmd_expand_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc)
831 {
832 	free((void *)(uintptr_t)zc->zc_nvlist_dst);
833 	zc->zc_nvlist_dst =
834 	    (uint64_t)(uintptr_t)zfs_alloc(hdl, zc->zc_nvlist_dst_size);
835 	if (zc->zc_nvlist_dst == 0)
836 		return (-1);
837 
838 	return (0);
839 }
840 
841 /*
842  * Called to free the src and dst nvlists stored in the command structure.
843  */
844 void
zcmd_free_nvlists(zfs_cmd_t * zc)845 zcmd_free_nvlists(zfs_cmd_t *zc)
846 {
847 	free((void *)(uintptr_t)zc->zc_nvlist_conf);
848 	free((void *)(uintptr_t)zc->zc_nvlist_src);
849 	free((void *)(uintptr_t)zc->zc_nvlist_dst);
850 	zc->zc_nvlist_conf = NULL;
851 	zc->zc_nvlist_src = NULL;
852 	zc->zc_nvlist_dst = NULL;
853 }
854 
855 static int
zcmd_write_nvlist_com(libzfs_handle_t * hdl,uint64_t * outnv,uint64_t * outlen,nvlist_t * nvl)856 zcmd_write_nvlist_com(libzfs_handle_t *hdl, uint64_t *outnv, uint64_t *outlen,
857     nvlist_t *nvl)
858 {
859 	char *packed;
860 	size_t len;
861 
862 	verify(nvlist_size(nvl, &len, NV_ENCODE_NATIVE) == 0);
863 
864 	if ((packed = zfs_alloc(hdl, len)) == NULL)
865 		return (-1);
866 
867 	verify(nvlist_pack(nvl, &packed, &len, NV_ENCODE_NATIVE, 0) == 0);
868 
869 	*outnv = (uint64_t)(uintptr_t)packed;
870 	*outlen = len;
871 
872 	return (0);
873 }
874 
875 int
zcmd_write_conf_nvlist(libzfs_handle_t * hdl,zfs_cmd_t * zc,nvlist_t * nvl)876 zcmd_write_conf_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, nvlist_t *nvl)
877 {
878 	return (zcmd_write_nvlist_com(hdl, &zc->zc_nvlist_conf,
879 	    &zc->zc_nvlist_conf_size, nvl));
880 }
881 
882 int
zcmd_write_src_nvlist(libzfs_handle_t * hdl,zfs_cmd_t * zc,nvlist_t * nvl)883 zcmd_write_src_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, nvlist_t *nvl)
884 {
885 	return (zcmd_write_nvlist_com(hdl, &zc->zc_nvlist_src,
886 	    &zc->zc_nvlist_src_size, nvl));
887 }
888 
889 /*
890  * Unpacks an nvlist from the ZFS ioctl command structure.
891  */
892 int
zcmd_read_dst_nvlist(libzfs_handle_t * hdl,zfs_cmd_t * zc,nvlist_t ** nvlp)893 zcmd_read_dst_nvlist(libzfs_handle_t *hdl, zfs_cmd_t *zc, nvlist_t **nvlp)
894 {
895 	if (nvlist_unpack((void *)(uintptr_t)zc->zc_nvlist_dst,
896 	    zc->zc_nvlist_dst_size, nvlp, 0) != 0)
897 		return (no_memory(hdl));
898 
899 	return (0);
900 }
901 
902 int
zfs_ioctl(libzfs_handle_t * hdl,int request,zfs_cmd_t * zc)903 zfs_ioctl(libzfs_handle_t *hdl, int request, zfs_cmd_t *zc)
904 {
905 	return (ioctl(hdl->libzfs_fd, request, zc));
906 }
907 
908 /*
909  * ================================================================
910  * API shared by zfs and zpool property management
911  * ================================================================
912  */
913 
914 static void
zprop_print_headers(zprop_get_cbdata_t * cbp,zfs_type_t type)915 zprop_print_headers(zprop_get_cbdata_t *cbp, zfs_type_t type)
916 {
917 	zprop_list_t *pl = cbp->cb_proplist;
918 	int i;
919 	char *title;
920 	size_t len;
921 
922 	cbp->cb_first = B_FALSE;
923 	if (cbp->cb_scripted)
924 		return;
925 
926 	/*
927 	 * Start with the length of the column headers.
928 	 */
929 	cbp->cb_colwidths[GET_COL_NAME] = strlen(dgettext(TEXT_DOMAIN, "NAME"));
930 	cbp->cb_colwidths[GET_COL_PROPERTY] = strlen(dgettext(TEXT_DOMAIN,
931 	    "PROPERTY"));
932 	cbp->cb_colwidths[GET_COL_VALUE] = strlen(dgettext(TEXT_DOMAIN,
933 	    "VALUE"));
934 	cbp->cb_colwidths[GET_COL_RECVD] = strlen(dgettext(TEXT_DOMAIN,
935 	    "RECEIVED"));
936 	cbp->cb_colwidths[GET_COL_SOURCE] = strlen(dgettext(TEXT_DOMAIN,
937 	    "SOURCE"));
938 
939 	/* first property is always NAME */
940 	assert(cbp->cb_proplist->pl_prop ==
941 	    ((type == ZFS_TYPE_POOL) ?  ZPOOL_PROP_NAME : ZFS_PROP_NAME));
942 
943 	/*
944 	 * Go through and calculate the widths for each column.  For the
945 	 * 'source' column, we kludge it up by taking the worst-case scenario of
946 	 * inheriting from the longest name.  This is acceptable because in the
947 	 * majority of cases 'SOURCE' is the last column displayed, and we don't
948 	 * use the width anyway.  Note that the 'VALUE' column can be oversized,
949 	 * if the name of the property is much longer than any values we find.
950 	 */
951 	for (pl = cbp->cb_proplist; pl != NULL; pl = pl->pl_next) {
952 		/*
953 		 * 'PROPERTY' column
954 		 */
955 		if (pl->pl_prop != ZPROP_INVAL) {
956 			const char *propname = (type == ZFS_TYPE_POOL) ?
957 			    zpool_prop_to_name(pl->pl_prop) :
958 			    zfs_prop_to_name(pl->pl_prop);
959 
960 			len = strlen(propname);
961 			if (len > cbp->cb_colwidths[GET_COL_PROPERTY])
962 				cbp->cb_colwidths[GET_COL_PROPERTY] = len;
963 		} else {
964 			len = strlen(pl->pl_user_prop);
965 			if (len > cbp->cb_colwidths[GET_COL_PROPERTY])
966 				cbp->cb_colwidths[GET_COL_PROPERTY] = len;
967 		}
968 
969 		/*
970 		 * 'VALUE' column.  The first property is always the 'name'
971 		 * property that was tacked on either by /sbin/zfs's
972 		 * zfs_do_get() or when calling zprop_expand_list(), so we
973 		 * ignore its width.  If the user specified the name property
974 		 * to display, then it will be later in the list in any case.
975 		 */
976 		if (pl != cbp->cb_proplist &&
977 		    pl->pl_width > cbp->cb_colwidths[GET_COL_VALUE])
978 			cbp->cb_colwidths[GET_COL_VALUE] = pl->pl_width;
979 
980 		/* 'RECEIVED' column. */
981 		if (pl != cbp->cb_proplist &&
982 		    pl->pl_recvd_width > cbp->cb_colwidths[GET_COL_RECVD])
983 			cbp->cb_colwidths[GET_COL_RECVD] = pl->pl_recvd_width;
984 
985 		/*
986 		 * 'NAME' and 'SOURCE' columns
987 		 */
988 		if (pl->pl_prop == (type == ZFS_TYPE_POOL ? ZPOOL_PROP_NAME :
989 		    ZFS_PROP_NAME) &&
990 		    pl->pl_width > cbp->cb_colwidths[GET_COL_NAME]) {
991 			cbp->cb_colwidths[GET_COL_NAME] = pl->pl_width;
992 			cbp->cb_colwidths[GET_COL_SOURCE] = pl->pl_width +
993 			    strlen(dgettext(TEXT_DOMAIN, "inherited from"));
994 		}
995 	}
996 
997 	/*
998 	 * Now go through and print the headers.
999 	 */
1000 	for (i = 0; i < ZFS_GET_NCOLS; i++) {
1001 		switch (cbp->cb_columns[i]) {
1002 		case GET_COL_NAME:
1003 			title = dgettext(TEXT_DOMAIN, "NAME");
1004 			break;
1005 		case GET_COL_PROPERTY:
1006 			title = dgettext(TEXT_DOMAIN, "PROPERTY");
1007 			break;
1008 		case GET_COL_VALUE:
1009 			title = dgettext(TEXT_DOMAIN, "VALUE");
1010 			break;
1011 		case GET_COL_RECVD:
1012 			title = dgettext(TEXT_DOMAIN, "RECEIVED");
1013 			break;
1014 		case GET_COL_SOURCE:
1015 			title = dgettext(TEXT_DOMAIN, "SOURCE");
1016 			break;
1017 		default:
1018 			title = NULL;
1019 		}
1020 
1021 		if (title != NULL) {
1022 			if (i == (ZFS_GET_NCOLS - 1) ||
1023 			    cbp->cb_columns[i + 1] == GET_COL_NONE)
1024 				(void) printf("%s", title);
1025 			else
1026 				(void) printf("%-*s  ",
1027 				    cbp->cb_colwidths[cbp->cb_columns[i]],
1028 				    title);
1029 		}
1030 	}
1031 	(void) printf("\n");
1032 }
1033 
1034 /*
1035  * Display a single line of output, according to the settings in the callback
1036  * structure.
1037  */
1038 void
zprop_print_one_property(const char * name,zprop_get_cbdata_t * cbp,const char * propname,const char * value,zprop_source_t sourcetype,const char * source,const char * recvd_value)1039 zprop_print_one_property(const char *name, zprop_get_cbdata_t *cbp,
1040     const char *propname, const char *value, zprop_source_t sourcetype,
1041     const char *source, const char *recvd_value)
1042 {
1043 	int i;
1044 	const char *str = NULL;
1045 	char buf[128];
1046 
1047 	/*
1048 	 * Ignore those source types that the user has chosen to ignore.
1049 	 */
1050 	if ((sourcetype & cbp->cb_sources) == 0)
1051 		return;
1052 
1053 	if (cbp->cb_first)
1054 		zprop_print_headers(cbp, cbp->cb_type);
1055 
1056 	for (i = 0; i < ZFS_GET_NCOLS; i++) {
1057 		switch (cbp->cb_columns[i]) {
1058 		case GET_COL_NAME:
1059 			str = name;
1060 			break;
1061 
1062 		case GET_COL_PROPERTY:
1063 			str = propname;
1064 			break;
1065 
1066 		case GET_COL_VALUE:
1067 			str = value;
1068 			break;
1069 
1070 		case GET_COL_SOURCE:
1071 			switch (sourcetype) {
1072 			case ZPROP_SRC_NONE:
1073 				str = "-";
1074 				break;
1075 
1076 			case ZPROP_SRC_DEFAULT:
1077 				str = "default";
1078 				break;
1079 
1080 			case ZPROP_SRC_LOCAL:
1081 				str = "local";
1082 				break;
1083 
1084 			case ZPROP_SRC_TEMPORARY:
1085 				str = "temporary";
1086 				break;
1087 
1088 			case ZPROP_SRC_INHERITED:
1089 				(void) snprintf(buf, sizeof (buf),
1090 				    "inherited from %s", source);
1091 				str = buf;
1092 				break;
1093 			case ZPROP_SRC_RECEIVED:
1094 				str = "received";
1095 				break;
1096 
1097 			default:
1098 				str = NULL;
1099 				assert(!"unhandled zprop_source_t");
1100 			}
1101 			break;
1102 
1103 		case GET_COL_RECVD:
1104 			str = (recvd_value == NULL ? "-" : recvd_value);
1105 			break;
1106 
1107 		default:
1108 			continue;
1109 		}
1110 
1111 		if (cbp->cb_columns[i + 1] == GET_COL_NONE)
1112 			(void) printf("%s", str);
1113 		else if (cbp->cb_scripted)
1114 			(void) printf("%s\t", str);
1115 		else
1116 			(void) printf("%-*s  ",
1117 			    cbp->cb_colwidths[cbp->cb_columns[i]],
1118 			    str);
1119 	}
1120 
1121 	(void) printf("\n");
1122 }
1123 
1124 /*
1125  * Given a numeric suffix, convert the value into a number of bits that the
1126  * resulting value must be shifted.
1127  */
1128 static int
str2shift(libzfs_handle_t * hdl,const char * buf)1129 str2shift(libzfs_handle_t *hdl, const char *buf)
1130 {
1131 	const char *ends = "BKMGTPEZ";
1132 	int i;
1133 
1134 	if (buf[0] == '\0')
1135 		return (0);
1136 	for (i = 0; i < strlen(ends); i++) {
1137 		if (toupper(buf[0]) == ends[i])
1138 			break;
1139 	}
1140 	if (i == strlen(ends)) {
1141 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1142 		    "invalid numeric suffix '%s'"), buf);
1143 		return (-1);
1144 	}
1145 
1146 	/*
1147 	 * We want to allow trailing 'b' characters for 'GB' or 'Mb'.  But don't
1148 	 * allow 'BB' - that's just weird.
1149 	 */
1150 	if (buf[1] == '\0' || (toupper(buf[1]) == 'B' && buf[2] == '\0' &&
1151 	    toupper(buf[0]) != 'B'))
1152 		return (10*i);
1153 
1154 	zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1155 	    "invalid numeric suffix '%s'"), buf);
1156 	return (-1);
1157 }
1158 
1159 /*
1160  * Convert a string of the form '100G' into a real number.  Used when setting
1161  * properties or creating a volume.  'buf' is used to place an extended error
1162  * message for the caller to use.
1163  */
1164 int
zfs_nicestrtonum(libzfs_handle_t * hdl,const char * value,uint64_t * num)1165 zfs_nicestrtonum(libzfs_handle_t *hdl, const char *value, uint64_t *num)
1166 {
1167 	char *end;
1168 	int shift;
1169 
1170 	*num = 0;
1171 
1172 	/* Check to see if this looks like a number.  */
1173 	if ((value[0] < '0' || value[0] > '9') && value[0] != '.') {
1174 		if (hdl)
1175 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1176 			    "bad numeric value '%s'"), value);
1177 		return (-1);
1178 	}
1179 
1180 	/* Rely on strtoull() to process the numeric portion.  */
1181 	errno = 0;
1182 	*num = strtoull(value, &end, 10);
1183 
1184 	/*
1185 	 * Check for ERANGE, which indicates that the value is too large to fit
1186 	 * in a 64-bit value.
1187 	 */
1188 	if (errno == ERANGE) {
1189 		if (hdl)
1190 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1191 			    "numeric value is too large"));
1192 		return (-1);
1193 	}
1194 
1195 	/*
1196 	 * If we have a decimal value, then do the computation with floating
1197 	 * point arithmetic.  Otherwise, use standard arithmetic.
1198 	 */
1199 	if (*end == '.') {
1200 		double fval = strtod(value, &end);
1201 
1202 		if ((shift = str2shift(hdl, end)) == -1)
1203 			return (-1);
1204 
1205 		fval *= pow(2, shift);
1206 
1207 		if (fval > UINT64_MAX) {
1208 			if (hdl)
1209 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1210 				    "numeric value is too large"));
1211 			return (-1);
1212 		}
1213 
1214 		*num = (uint64_t)fval;
1215 	} else {
1216 		if ((shift = str2shift(hdl, end)) == -1)
1217 			return (-1);
1218 
1219 		/* Check for overflow */
1220 		if (shift >= 64 || (*num << shift) >> shift != *num) {
1221 			if (hdl)
1222 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1223 				    "numeric value is too large"));
1224 			return (-1);
1225 		}
1226 
1227 		*num <<= shift;
1228 	}
1229 
1230 	return (0);
1231 }
1232 
1233 /*
1234  * Given a propname=value nvpair to set, parse any numeric properties
1235  * (index, boolean, etc) if they are specified as strings and add the
1236  * resulting nvpair to the returned nvlist.
1237  *
1238  * At the DSL layer, all properties are either 64-bit numbers or strings.
1239  * We want the user to be able to ignore this fact and specify properties
1240  * as native values (numbers, for example) or as strings (to simplify
1241  * command line utilities).  This also handles converting index types
1242  * (compression, checksum, etc) from strings to their on-disk index.
1243  */
1244 int
zprop_parse_value(libzfs_handle_t * hdl,nvpair_t * elem,int prop,zfs_type_t type,nvlist_t * ret,char ** svalp,uint64_t * ivalp,const char * errbuf)1245 zprop_parse_value(libzfs_handle_t *hdl, nvpair_t *elem, int prop,
1246     zfs_type_t type, nvlist_t *ret, char **svalp, uint64_t *ivalp,
1247     const char *errbuf)
1248 {
1249 	data_type_t datatype = nvpair_type(elem);
1250 	zprop_type_t proptype;
1251 	const char *propname;
1252 	char *value;
1253 	boolean_t isnone = B_FALSE;
1254 
1255 	if (type == ZFS_TYPE_POOL) {
1256 		proptype = zpool_prop_get_type(prop);
1257 		propname = zpool_prop_to_name(prop);
1258 	} else {
1259 		proptype = zfs_prop_get_type(prop);
1260 		propname = zfs_prop_to_name(prop);
1261 	}
1262 
1263 	/*
1264 	 * Convert any properties to the internal DSL value types.
1265 	 */
1266 	*svalp = NULL;
1267 	*ivalp = 0;
1268 
1269 	switch (proptype) {
1270 	case PROP_TYPE_STRING:
1271 		if (datatype != DATA_TYPE_STRING) {
1272 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1273 			    "'%s' must be a string"), nvpair_name(elem));
1274 			goto error;
1275 		}
1276 		(void) nvpair_value_string(elem, svalp);
1277 		if (strlen(*svalp) >= ZFS_MAXPROPLEN) {
1278 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1279 			    "'%s' is too long"), nvpair_name(elem));
1280 			goto error;
1281 		}
1282 		break;
1283 
1284 	case PROP_TYPE_NUMBER:
1285 		if (datatype == DATA_TYPE_STRING) {
1286 			(void) nvpair_value_string(elem, &value);
1287 			if (strcmp(value, "none") == 0) {
1288 				isnone = B_TRUE;
1289 			} else if (zfs_nicestrtonum(hdl, value, ivalp)
1290 			    != 0) {
1291 				goto error;
1292 			}
1293 		} else if (datatype == DATA_TYPE_UINT64) {
1294 			(void) nvpair_value_uint64(elem, ivalp);
1295 		} else {
1296 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1297 			    "'%s' must be a number"), nvpair_name(elem));
1298 			goto error;
1299 		}
1300 
1301 		/*
1302 		 * Quota special: force 'none' and don't allow 0.
1303 		 */
1304 		if ((type & ZFS_TYPE_DATASET) && *ivalp == 0 && !isnone &&
1305 		    (prop == ZFS_PROP_QUOTA || prop == ZFS_PROP_REFQUOTA)) {
1306 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1307 			    "use 'none' to disable quota/refquota"));
1308 			goto error;
1309 		}
1310 
1311 		/*
1312 		 * Special handling for "*_limit=none". In this case it's not
1313 		 * 0 but UINT64_MAX.
1314 		 */
1315 		if ((type & ZFS_TYPE_DATASET) && isnone &&
1316 		    (prop == ZFS_PROP_FILESYSTEM_LIMIT ||
1317 		    prop == ZFS_PROP_SNAPSHOT_LIMIT)) {
1318 			*ivalp = UINT64_MAX;
1319 		}
1320 		break;
1321 
1322 	case PROP_TYPE_INDEX:
1323 		if (datatype != DATA_TYPE_STRING) {
1324 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1325 			    "'%s' must be a string"), nvpair_name(elem));
1326 			goto error;
1327 		}
1328 
1329 		(void) nvpair_value_string(elem, &value);
1330 
1331 		if (zprop_string_to_index(prop, value, ivalp, type) != 0) {
1332 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1333 			    "'%s' must be one of '%s'"), propname,
1334 			    zprop_values(prop, type));
1335 			goto error;
1336 		}
1337 		break;
1338 
1339 	default:
1340 		abort();
1341 	}
1342 
1343 	/*
1344 	 * Add the result to our return set of properties.
1345 	 */
1346 	if (*svalp != NULL) {
1347 		if (nvlist_add_string(ret, propname, *svalp) != 0) {
1348 			(void) no_memory(hdl);
1349 			return (-1);
1350 		}
1351 	} else {
1352 		if (nvlist_add_uint64(ret, propname, *ivalp) != 0) {
1353 			(void) no_memory(hdl);
1354 			return (-1);
1355 		}
1356 	}
1357 
1358 	return (0);
1359 error:
1360 	(void) zfs_error(hdl, EZFS_BADPROP, errbuf);
1361 	return (-1);
1362 }
1363 
1364 static int
addlist(libzfs_handle_t * hdl,char * propname,zprop_list_t ** listp,zfs_type_t type)1365 addlist(libzfs_handle_t *hdl, char *propname, zprop_list_t **listp,
1366     zfs_type_t type)
1367 {
1368 	int prop;
1369 	zprop_list_t *entry;
1370 
1371 	prop = zprop_name_to_prop(propname, type);
1372 
1373 	if (prop != ZPROP_INVAL && !zprop_valid_for_type(prop, type))
1374 		prop = ZPROP_INVAL;
1375 
1376 	/*
1377 	 * When no property table entry can be found, return failure if
1378 	 * this is a pool property or if this isn't a user-defined
1379 	 * dataset property,
1380 	 */
1381 	if (prop == ZPROP_INVAL && ((type == ZFS_TYPE_POOL &&
1382 	    !zpool_prop_feature(propname) &&
1383 	    !zpool_prop_unsupported(propname)) ||
1384 	    (type == ZFS_TYPE_DATASET && !zfs_prop_user(propname) &&
1385 	    !zfs_prop_userquota(propname) && !zfs_prop_written(propname)))) {
1386 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1387 		    "invalid property '%s'"), propname);
1388 		return (zfs_error(hdl, EZFS_BADPROP,
1389 		    dgettext(TEXT_DOMAIN, "bad property list")));
1390 	}
1391 
1392 	if ((entry = zfs_alloc(hdl, sizeof (zprop_list_t))) == NULL)
1393 		return (-1);
1394 
1395 	entry->pl_prop = prop;
1396 	if (prop == ZPROP_INVAL) {
1397 		if ((entry->pl_user_prop = zfs_strdup(hdl, propname)) ==
1398 		    NULL) {
1399 			free(entry);
1400 			return (-1);
1401 		}
1402 		entry->pl_width = strlen(propname);
1403 	} else {
1404 		entry->pl_width = zprop_width(prop, &entry->pl_fixed,
1405 		    type);
1406 	}
1407 
1408 	*listp = entry;
1409 
1410 	return (0);
1411 }
1412 
1413 /*
1414  * Given a comma-separated list of properties, construct a property list
1415  * containing both user-defined and native properties.  This function will
1416  * return a NULL list if 'all' is specified, which can later be expanded
1417  * by zprop_expand_list().
1418  */
1419 int
zprop_get_list(libzfs_handle_t * hdl,char * props,zprop_list_t ** listp,zfs_type_t type)1420 zprop_get_list(libzfs_handle_t *hdl, char *props, zprop_list_t **listp,
1421     zfs_type_t type)
1422 {
1423 	*listp = NULL;
1424 
1425 	/*
1426 	 * If 'all' is specified, return a NULL list.
1427 	 */
1428 	if (strcmp(props, "all") == 0)
1429 		return (0);
1430 
1431 	/*
1432 	 * If no props were specified, return an error.
1433 	 */
1434 	if (props[0] == '\0') {
1435 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1436 		    "no properties specified"));
1437 		return (zfs_error(hdl, EZFS_BADPROP, dgettext(TEXT_DOMAIN,
1438 		    "bad property list")));
1439 	}
1440 
1441 	/*
1442 	 * It would be nice to use getsubopt() here, but the inclusion of column
1443 	 * aliases makes this more effort than it's worth.
1444 	 */
1445 	while (*props != '\0') {
1446 		size_t len;
1447 		char *p;
1448 		char c;
1449 
1450 		if ((p = strchr(props, ',')) == NULL) {
1451 			len = strlen(props);
1452 			p = props + len;
1453 		} else {
1454 			len = p - props;
1455 		}
1456 
1457 		/*
1458 		 * Check for empty options.
1459 		 */
1460 		if (len == 0) {
1461 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
1462 			    "empty property name"));
1463 			return (zfs_error(hdl, EZFS_BADPROP,
1464 			    dgettext(TEXT_DOMAIN, "bad property list")));
1465 		}
1466 
1467 		/*
1468 		 * Check all regular property names.
1469 		 */
1470 		c = props[len];
1471 		props[len] = '\0';
1472 
1473 		if (strcmp(props, "space") == 0) {
1474 			static char *spaceprops[] = {
1475 				"name", "avail", "used", "usedbysnapshots",
1476 				"usedbydataset", "usedbyrefreservation",
1477 				"usedbychildren", NULL
1478 			};
1479 			int i;
1480 
1481 			for (i = 0; spaceprops[i]; i++) {
1482 				if (addlist(hdl, spaceprops[i], listp, type))
1483 					return (-1);
1484 				listp = &(*listp)->pl_next;
1485 			}
1486 		} else {
1487 			if (addlist(hdl, props, listp, type))
1488 				return (-1);
1489 			listp = &(*listp)->pl_next;
1490 		}
1491 
1492 		props = p;
1493 		if (c == ',')
1494 			props++;
1495 	}
1496 
1497 	return (0);
1498 }
1499 
1500 void
zprop_free_list(zprop_list_t * pl)1501 zprop_free_list(zprop_list_t *pl)
1502 {
1503 	zprop_list_t *next;
1504 
1505 	while (pl != NULL) {
1506 		next = pl->pl_next;
1507 		free(pl->pl_user_prop);
1508 		free(pl);
1509 		pl = next;
1510 	}
1511 }
1512 
1513 typedef struct expand_data {
1514 	zprop_list_t	**last;
1515 	libzfs_handle_t	*hdl;
1516 	zfs_type_t type;
1517 } expand_data_t;
1518 
1519 int
zprop_expand_list_cb(int prop,void * cb)1520 zprop_expand_list_cb(int prop, void *cb)
1521 {
1522 	zprop_list_t *entry;
1523 	expand_data_t *edp = cb;
1524 
1525 	if ((entry = zfs_alloc(edp->hdl, sizeof (zprop_list_t))) == NULL)
1526 		return (ZPROP_INVAL);
1527 
1528 	entry->pl_prop = prop;
1529 	entry->pl_width = zprop_width(prop, &entry->pl_fixed, edp->type);
1530 	entry->pl_all = B_TRUE;
1531 
1532 	*(edp->last) = entry;
1533 	edp->last = &entry->pl_next;
1534 
1535 	return (ZPROP_CONT);
1536 }
1537 
1538 int
zprop_expand_list(libzfs_handle_t * hdl,zprop_list_t ** plp,zfs_type_t type)1539 zprop_expand_list(libzfs_handle_t *hdl, zprop_list_t **plp, zfs_type_t type)
1540 {
1541 	zprop_list_t *entry;
1542 	zprop_list_t **last;
1543 	expand_data_t exp;
1544 
1545 	if (*plp == NULL) {
1546 		/*
1547 		 * If this is the very first time we've been called for an 'all'
1548 		 * specification, expand the list to include all native
1549 		 * properties.
1550 		 */
1551 		last = plp;
1552 
1553 		exp.last = last;
1554 		exp.hdl = hdl;
1555 		exp.type = type;
1556 
1557 		if (zprop_iter_common(zprop_expand_list_cb, &exp, B_FALSE,
1558 		    B_FALSE, type) == ZPROP_INVAL)
1559 			return (-1);
1560 
1561 		/*
1562 		 * Add 'name' to the beginning of the list, which is handled
1563 		 * specially.
1564 		 */
1565 		if ((entry = zfs_alloc(hdl, sizeof (zprop_list_t))) == NULL)
1566 			return (-1);
1567 
1568 		entry->pl_prop = (type == ZFS_TYPE_POOL) ?  ZPOOL_PROP_NAME :
1569 		    ZFS_PROP_NAME;
1570 		entry->pl_width = zprop_width(entry->pl_prop,
1571 		    &entry->pl_fixed, type);
1572 		entry->pl_all = B_TRUE;
1573 		entry->pl_next = *plp;
1574 		*plp = entry;
1575 	}
1576 	return (0);
1577 }
1578 
1579 int
zprop_iter(zprop_func func,void * cb,boolean_t show_all,boolean_t ordered,zfs_type_t type)1580 zprop_iter(zprop_func func, void *cb, boolean_t show_all, boolean_t ordered,
1581     zfs_type_t type)
1582 {
1583 	return (zprop_iter_common(func, cb, show_all, ordered, type));
1584 }
1585