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