lib/libzfs/libzfs_util.c

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