granicus.if.org Git - zfs/blob - cmd/zpool/zpool_vdev.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 /*
  23  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Copyright (c) 2013 by Delphix. All rights reserved.
  25  * Copyright (c) 2016 Intel Corporation.
  26  * Copyright 2016 Igor Kozhukhov <ikozhukhov@gmail.com>.
  27  */
  28
  29 /*
  30  * Functions to convert between a list of vdevs and an nvlist representing the
  31  * configuration.  Each entry in the list can be one of:
  32  *
  33  *      Device vdevs
  34  *              disk=(path=..., devid=...)
  35  *              file=(path=...)
  36  *
  37  *      Group vdevs
  38  *              raidz[1|2]=(...)
  39  *              mirror=(...)
  40  *
  41  *      Hot spares
  42  *
  43  * While the underlying implementation supports it, group vdevs cannot contain
  44  * other group vdevs.  All userland verification of devices is contained within
  45  * this file.  If successful, the nvlist returned can be passed directly to the
  46  * kernel; we've done as much verification as possible in userland.
  47  *
  48  * Hot spares are a special case, and passed down as an array of disk vdevs, at
  49  * the same level as the root of the vdev tree.
  50  *
  51  * The only function exported by this file is 'make_root_vdev'.  The
  52  * function performs several passes:
  53  *
  54  *      1. Construct the vdev specification.  Performs syntax validation and
  55  *         makes sure each device is valid.
  56  *      2. Check for devices in use.  Using libblkid to make sure that no
  57  *         devices are also in use.  Some can be overridden using the 'force'
  58  *         flag, others cannot.
  59  *      3. Check for replication errors if the 'force' flag is not specified.
  60  *         validates that the replication level is consistent across the
  61  *         entire pool.
  62  *      4. Call libzfs to label any whole disks with an EFI label.
  63  */
  64
  65 #include <assert.h>
  66 #include <ctype.h>
  67 #include <devid.h>
  68 #include <errno.h>
  69 #include <fcntl.h>
  70 #include <libintl.h>
  71 #include <libnvpair.h>
  72 #include <limits.h>
  73 #include <scsi/scsi.h>
  74 #include <scsi/sg.h>
  75 #include <stdio.h>
  76 #include <string.h>
  77 #include <unistd.h>
  78 #include <sys/efi_partition.h>
  79 #include <sys/stat.h>
  80 #include <sys/vtoc.h>
  81 #include <sys/mntent.h>
  82 #include <uuid/uuid.h>
  83 #include <blkid/blkid.h>
  84 #include "zpool_util.h"
  85 #include <sys/zfs_context.h>
  86
  87 /*
  88  * For any given vdev specification, we can have multiple errors.  The
  89  * vdev_error() function keeps track of whether we have seen an error yet, and
  90  * prints out a header if its the first error we've seen.
  91  */
  92 boolean_t error_seen;
  93 boolean_t is_force;
  94
  95 typedef struct vdev_disk_db_entry
  96 {
  97         char id[24];
  98         int sector_size;
  99 } vdev_disk_db_entry_t;
 100
 101 /*
 102  * Database of block devices that lie about physical sector sizes.  The
 103  * identification string must be precisely 24 characters to avoid false
 104  * negatives
 105  */
 106 static vdev_disk_db_entry_t vdev_disk_database[] = {
 107         {"ATA     ADATA SSD S396 3", 8192},
 108         {"ATA     APPLE SSD SM128E", 8192},
 109         {"ATA     APPLE SSD SM256E", 8192},
 110         {"ATA     APPLE SSD SM512E", 8192},
 111         {"ATA     APPLE SSD SM768E", 8192},
 112         {"ATA     C400-MTFDDAC064M", 8192},
 113         {"ATA     C400-MTFDDAC128M", 8192},
 114         {"ATA     C400-MTFDDAC256M", 8192},
 115         {"ATA     C400-MTFDDAC512M", 8192},
 116         {"ATA     Corsair Force 3 ", 8192},
 117         {"ATA     Corsair Force GS", 8192},
 118         {"ATA     INTEL SSDSA2CT04", 8192},
 119         {"ATA     INTEL SSDSA2BZ10", 8192},
 120         {"ATA     INTEL SSDSA2BZ20", 8192},
 121         {"ATA     INTEL SSDSA2BZ30", 8192},
 122         {"ATA     INTEL SSDSA2CW04", 8192},
 123         {"ATA     INTEL SSDSA2CW08", 8192},
 124         {"ATA     INTEL SSDSA2CW12", 8192},
 125         {"ATA     INTEL SSDSA2CW16", 8192},
 126         {"ATA     INTEL SSDSA2CW30", 8192},
 127         {"ATA     INTEL SSDSA2CW60", 8192},
 128         {"ATA     INTEL SSDSC2CT06", 8192},
 129         {"ATA     INTEL SSDSC2CT12", 8192},
 130         {"ATA     INTEL SSDSC2CT18", 8192},
 131         {"ATA     INTEL SSDSC2CT24", 8192},
 132         {"ATA     INTEL SSDSC2CW06", 8192},
 133         {"ATA     INTEL SSDSC2CW12", 8192},
 134         {"ATA     INTEL SSDSC2CW18", 8192},
 135         {"ATA     INTEL SSDSC2CW24", 8192},
 136         {"ATA     INTEL SSDSC2CW48", 8192},
 137         {"ATA     KINGSTON SH100S3", 8192},
 138         {"ATA     KINGSTON SH103S3", 8192},
 139         {"ATA     M4-CT064M4SSD2  ", 8192},
 140         {"ATA     M4-CT128M4SSD2  ", 8192},
 141         {"ATA     M4-CT256M4SSD2  ", 8192},
 142         {"ATA     M4-CT512M4SSD2  ", 8192},
 143         {"ATA     OCZ-AGILITY2    ", 8192},
 144         {"ATA     OCZ-AGILITY3    ", 8192},
 145         {"ATA     OCZ-VERTEX2 3.5 ", 8192},
 146         {"ATA     OCZ-VERTEX3     ", 8192},
 147         {"ATA     OCZ-VERTEX3 LT  ", 8192},
 148         {"ATA     OCZ-VERTEX3 MI  ", 8192},
 149         {"ATA     OCZ-VERTEX4     ", 8192},
 150         {"ATA     SAMSUNG MZ7WD120", 8192},
 151         {"ATA     SAMSUNG MZ7WD240", 8192},
 152         {"ATA     SAMSUNG MZ7WD480", 8192},
 153         {"ATA     SAMSUNG MZ7WD960", 8192},
 154         {"ATA     SAMSUNG SSD 830 ", 8192},
 155         {"ATA     Samsung SSD 840 ", 8192},
 156         {"ATA     SanDisk SSD U100", 8192},
 157         {"ATA     TOSHIBA THNSNH06", 8192},
 158         {"ATA     TOSHIBA THNSNH12", 8192},
 159         {"ATA     TOSHIBA THNSNH25", 8192},
 160         {"ATA     TOSHIBA THNSNH51", 8192},
 161         {"ATA     APPLE SSD TS064C", 4096},
 162         {"ATA     APPLE SSD TS128C", 4096},
 163         {"ATA     APPLE SSD TS256C", 4096},
 164         {"ATA     APPLE SSD TS512C", 4096},
 165         {"ATA     INTEL SSDSA2M040", 4096},
 166         {"ATA     INTEL SSDSA2M080", 4096},
 167         {"ATA     INTEL SSDSA2M160", 4096},
 168         {"ATA     INTEL SSDSC2MH12", 4096},
 169         {"ATA     INTEL SSDSC2MH25", 4096},
 170         {"ATA     OCZ CORE_SSD    ", 4096},
 171         {"ATA     OCZ-VERTEX      ", 4096},
 172         {"ATA     SAMSUNG MCCOE32G", 4096},
 173         {"ATA     SAMSUNG MCCOE64G", 4096},
 174         {"ATA     SAMSUNG SSD PM80", 4096},
 175         /* Flash drives optimized for 4KB IOs on larger pages */
 176         {"ATA     INTEL SSDSC2BA10", 4096},
 177         {"ATA     INTEL SSDSC2BA20", 4096},
 178         {"ATA     INTEL SSDSC2BA40", 4096},
 179         {"ATA     INTEL SSDSC2BA80", 4096},
 180         {"ATA     INTEL SSDSC2BB08", 4096},
 181         {"ATA     INTEL SSDSC2BB12", 4096},
 182         {"ATA     INTEL SSDSC2BB16", 4096},
 183         {"ATA     INTEL SSDSC2BB24", 4096},
 184         {"ATA     INTEL SSDSC2BB30", 4096},
 185         {"ATA     INTEL SSDSC2BB40", 4096},
 186         {"ATA     INTEL SSDSC2BB48", 4096},
 187         {"ATA     INTEL SSDSC2BB60", 4096},
 188         {"ATA     INTEL SSDSC2BB80", 4096},
 189         {"ATA     INTEL SSDSC2BW24", 4096},
 190         {"ATA     INTEL SSDSC2BP24", 4096},
 191         {"ATA     INTEL SSDSC2BP48", 4096},
 192         {"NA      SmrtStorSDLKAE9W", 4096},
 193         /* Imported from Open Solaris */
 194         {"ATA     MARVELL SD88SA02", 4096},
 195         /* Advanced format Hard drives */
 196         {"ATA     Hitachi HDS5C303", 4096},
 197         {"ATA     SAMSUNG HD204UI ", 4096},
 198         {"ATA     ST2000DL004 HD20", 4096},
 199         {"ATA     WDC WD10EARS-00M", 4096},
 200         {"ATA     WDC WD10EARS-00S", 4096},
 201         {"ATA     WDC WD10EARS-00Z", 4096},
 202         {"ATA     WDC WD15EARS-00M", 4096},
 203         {"ATA     WDC WD15EARS-00S", 4096},
 204         {"ATA     WDC WD15EARS-00Z", 4096},
 205         {"ATA     WDC WD20EARS-00M", 4096},
 206         {"ATA     WDC WD20EARS-00S", 4096},
 207         {"ATA     WDC WD20EARS-00Z", 4096},
 208         {"ATA     WDC WD1600BEVT-0", 4096},
 209         {"ATA     WDC WD2500BEVT-0", 4096},
 210         {"ATA     WDC WD3200BEVT-0", 4096},
 211         {"ATA     WDC WD5000BEVT-0", 4096},
 212         /* Virtual disks: Assume zvols with default volblocksize */
 213 #if 0
 214         {"ATA     QEMU HARDDISK   ", 8192},
 215         {"IET     VIRTUAL-DISK    ", 8192},
 216         {"OI      COMSTAR         ", 8192},
 217         {"SUN     COMSTAR         ", 8192},
 218         {"NETAPP  LUN             ", 8192},
 219 #endif
 220 };
 221
 222 static const int vdev_disk_database_size =
 223         sizeof (vdev_disk_database) / sizeof (vdev_disk_database[0]);
 224
 225 #define INQ_REPLY_LEN   96
 226 #define INQ_CMD_LEN     6
 227
 228 static boolean_t
 229 check_sector_size_database(char *path, int *sector_size)
 230 {
 231         unsigned char inq_buff[INQ_REPLY_LEN];
 232         unsigned char sense_buffer[32];
 233         unsigned char inq_cmd_blk[INQ_CMD_LEN] =
 234             {INQUIRY, 0, 0, 0, INQ_REPLY_LEN, 0};
 235         sg_io_hdr_t io_hdr;
 236         int error;
 237         int fd;
 238         int i;
 239
 240         /* Prepare INQUIRY command */
 241         memset(&io_hdr, 0, sizeof (sg_io_hdr_t));
 242         io_hdr.interface_id = 'S';
 243         io_hdr.cmd_len = sizeof (inq_cmd_blk);
 244         io_hdr.mx_sb_len = sizeof (sense_buffer);
 245         io_hdr.dxfer_direction = SG_DXFER_FROM_DEV;
 246         io_hdr.dxfer_len = INQ_REPLY_LEN;
 247         io_hdr.dxferp = inq_buff;
 248         io_hdr.cmdp = inq_cmd_blk;
 249         io_hdr.sbp = sense_buffer;
 250         io_hdr.timeout = 10;            /* 10 milliseconds is ample time */
 251
 252         if ((fd = open(path, O_RDONLY|O_DIRECT)) < 0)
 253                 return (B_FALSE);
 254
 255         error = ioctl(fd, SG_IO, (unsigned long) &io_hdr);
 256
 257         (void) close(fd);
 258
 259         if (error < 0)
 260                 return (B_FALSE);
 261
 262         if ((io_hdr.info & SG_INFO_OK_MASK) != SG_INFO_OK)
 263                 return (B_FALSE);
 264
 265         for (i = 0; i < vdev_disk_database_size; i++) {
 266                 if (memcmp(inq_buff + 8, vdev_disk_database[i].id, 24))
 267                         continue;
 268
 269                 *sector_size = vdev_disk_database[i].sector_size;
 270                 return (B_TRUE);
 271         }
 272
 273         return (B_FALSE);
 274 }
 275
 276 /*PRINTFLIKE1*/
 277 static void
 278 vdev_error(const char *fmt, ...)
 279 {
 280         va_list ap;
 281
 282         if (!error_seen) {
 283                 (void) fprintf(stderr, gettext("invalid vdev specification\n"));
 284                 if (!is_force)
 285                         (void) fprintf(stderr, gettext("use '-f' to override "
 286                             "the following errors:\n"));
 287                 else
 288                         (void) fprintf(stderr, gettext("the following errors "
 289                             "must be manually repaired:\n"));
 290                 error_seen = B_TRUE;
 291         }
 292
 293         va_start(ap, fmt);
 294         (void) vfprintf(stderr, fmt, ap);
 295         va_end(ap);
 296 }
 297
 298 /*
 299  * Check that a file is valid.  All we can do in this case is check that it's
 300  * not in use by another pool, and not in use by swap.
 301  */
 302 static int
 303 check_file(const char *file, boolean_t force, boolean_t isspare)
 304 {
 305         char  *name;
 306         int fd;
 307         int ret = 0;
 308         pool_state_t state;
 309         boolean_t inuse;
 310
 311         if ((fd = open(file, O_RDONLY)) < 0)
 312                 return (0);
 313
 314         if (zpool_in_use(g_zfs, fd, &state, &name, &inuse) == 0 && inuse) {
 315                 const char *desc;
 316
 317                 switch (state) {
 318                 case POOL_STATE_ACTIVE:
 319                         desc = gettext("active");
 320                         break;
 321
 322                 case POOL_STATE_EXPORTED:
 323                         desc = gettext("exported");
 324                         break;
 325
 326                 case POOL_STATE_POTENTIALLY_ACTIVE:
 327                         desc = gettext("potentially active");
 328                         break;
 329
 330                 default:
 331                         desc = gettext("unknown");
 332                         break;
 333                 }
 334
 335                 /*
 336                  * Allow hot spares to be shared between pools.
 337                  */
 338                 if (state == POOL_STATE_SPARE && isspare) {
 339                         free(name);
 340                         (void) close(fd);
 341                         return (0);
 342                 }
 343
 344                 if (state == POOL_STATE_ACTIVE ||
 345                     state == POOL_STATE_SPARE || !force) {
 346                         switch (state) {
 347                         case POOL_STATE_SPARE:
 348                                 vdev_error(gettext("%s is reserved as a hot "
 349                                     "spare for pool %s\n"), file, name);
 350                                 break;
 351                         default:
 352                                 vdev_error(gettext("%s is part of %s pool "
 353                                     "'%s'\n"), file, desc, name);
 354                                 break;
 355                         }
 356                         ret = -1;
 357                 }
 358
 359                 free(name);
 360         }
 361
 362         (void) close(fd);
 363         return (ret);
 364 }
 365
 366 static void
 367 check_error(int err)
 368 {
 369         (void) fprintf(stderr, gettext("warning: device in use checking "
 370             "failed: %s\n"), strerror(err));
 371 }
 372
 373 static int
 374 check_slice(const char *path, blkid_cache cache, int force, boolean_t isspare)
 375 {
 376         int err;
 377         char *value;
 378
 379         /* No valid type detected device is safe to use */
 380         value = blkid_get_tag_value(cache, "TYPE", path);
 381         if (value == NULL)
 382                 return (0);
 383
 384         /*
 385          * If libblkid detects a ZFS device, we check the device
 386          * using check_file() to see if it's safe.  The one safe
 387          * case is a spare device shared between multiple pools.
 388          */
 389         if (strcmp(value, "zfs_member") == 0) {
 390                 err = check_file(path, force, isspare);
 391         } else {
 392                 if (force) {
 393                         err = 0;
 394                 } else {
 395                         err = -1;
 396                         vdev_error(gettext("%s contains a filesystem of "
 397                             "type '%s'\n"), path, value);
 398                 }
 399         }
 400
 401         free(value);
 402
 403         return (err);
 404 }
 405
 406 /*
 407  * Validate that a disk including all partitions are safe to use.
 408  *
 409  * For EFI labeled disks this can done relatively easily with the libefi
 410  * library.  The partition numbers are extracted from the label and used
 411  * to generate the expected /dev/ paths.  Each partition can then be
 412  * checked for conflicts.
 413  *
 414  * For non-EFI labeled disks (MBR/EBR/etc) the same process is possible
 415  * but due to the lack of a readily available libraries this scanning is
 416  * not implemented.  Instead only the device path as given is checked.
 417  */
 418 static int
 419 check_disk(const char *path, blkid_cache cache, int force,
 420     boolean_t isspare, boolean_t iswholedisk)
 421 {
 422         struct dk_gpt *vtoc;
 423         char slice_path[MAXPATHLEN];
 424         int err = 0;
 425         int fd, i;
 426
 427         if (!iswholedisk)
 428                 return (check_slice(path, cache, force, isspare));
 429
 430         if ((fd = open(path, O_RDONLY|O_DIRECT)) < 0) {
 431                 check_error(errno);
 432                 return (-1);
 433         }
 434
 435         /*
 436          * Expected to fail for non-EFI labled disks.  Just check the device
 437          * as given and do not attempt to detect and scan partitions.
 438          */
 439         err = efi_alloc_and_read(fd, &vtoc);
 440         if (err) {
 441                 (void) close(fd);
 442                 return (check_slice(path, cache, force, isspare));
 443         }
 444
 445         /*
 446          * The primary efi partition label is damaged however the secondary
 447          * label at the end of the device is intact.  Rather than use this
 448          * label we should play it safe and treat this as a non efi device.
 449          */
 450         if (vtoc->efi_flags & EFI_GPT_PRIMARY_CORRUPT) {
 451                 efi_free(vtoc);
 452                 (void) close(fd);
 453
 454                 if (force) {
 455                         /* Partitions will now be created using the backup */
 456                         return (0);
 457                 } else {
 458                         vdev_error(gettext("%s contains a corrupt primary "
 459                             "EFI label.\n"), path);
 460                         return (-1);
 461                 }
 462         }
 463
 464         for (i = 0; i < vtoc->efi_nparts; i++) {
 465
 466                 if (vtoc->efi_parts[i].p_tag == V_UNASSIGNED ||
 467                     uuid_is_null((uchar_t *)&vtoc->efi_parts[i].p_guid))
 468                         continue;
 469
 470                 if (strncmp(path, UDISK_ROOT, strlen(UDISK_ROOT)) == 0)
 471                         (void) snprintf(slice_path, sizeof (slice_path),
 472                             "%s%s%d", path, "-part", i+1);
 473                 else
 474                         (void) snprintf(slice_path, sizeof (slice_path),
 475                             "%s%s%d", path, isdigit(path[strlen(path)-1]) ?
 476                             "p" : "", i+1);
 477
 478                 err = check_slice(slice_path, cache, force, isspare);
 479                 if (err)
 480                         break;
 481         }
 482
 483         efi_free(vtoc);
 484         (void) close(fd);
 485
 486         return (err);
 487 }
 488
 489 static int
 490 check_device(const char *path, boolean_t force,
 491     boolean_t isspare, boolean_t iswholedisk)
 492 {
 493         blkid_cache cache;
 494         int error;
 495
 496         error = blkid_get_cache(&cache, NULL);
 497         if (error != 0) {
 498                 check_error(error);
 499                 return (-1);
 500         }
 501
 502         error = check_disk(path, cache, force, isspare, iswholedisk);
 503         blkid_put_cache(cache);
 504
 505         return (error);
 506 }
 507
 508 /*
 509  * By "whole disk" we mean an entire physical disk (something we can
 510  * label, toggle the write cache on, etc.) as opposed to the full
 511  * capacity of a pseudo-device such as lofi or did.  We act as if we
 512  * are labeling the disk, which should be a pretty good test of whether
 513  * it's a viable device or not.  Returns B_TRUE if it is and B_FALSE if
 514  * it isn't.
 515  */
 516 static boolean_t
 517 is_whole_disk(const char *path)
 518 {
 519         struct dk_gpt *label;
 520         int fd;
 521
 522         if ((fd = open(path, O_RDONLY|O_DIRECT)) < 0)
 523                 return (B_FALSE);
 524         if (efi_alloc_and_init(fd, EFI_NUMPAR, &label) != 0) {
 525                 (void) close(fd);
 526                 return (B_FALSE);
 527         }
 528         efi_free(label);
 529         (void) close(fd);
 530         return (B_TRUE);
 531 }
 532
 533 /*
 534  * This may be a shorthand device path or it could be total gibberish.
 535  * Check to see if it is a known device available in zfs_vdev_paths.
 536  * As part of this check, see if we've been given an entire disk
 537  * (minus the slice number).
 538  */
 539 static int
 540 is_shorthand_path(const char *arg, char *path, size_t path_size,
 541     struct stat64 *statbuf, boolean_t *wholedisk)
 542 {
 543         int error;
 544
 545         error = zfs_resolve_shortname(arg, path, path_size);
 546         if (error == 0) {
 547                 *wholedisk = is_whole_disk(path);
 548                 if (*wholedisk || (stat64(path, statbuf) == 0))
 549                         return (0);
 550         }
 551
 552         strlcpy(path, arg, path_size);
 553         memset(statbuf, 0, sizeof (*statbuf));
 554         *wholedisk = B_FALSE;
 555
 556         return (error);
 557 }
 558
 559 /*
 560  * Determine if the given path is a hot spare within the given configuration.
 561  * If no configuration is given we rely solely on the label.
 562  */
 563 static boolean_t
 564 is_spare(nvlist_t *config, const char *path)
 565 {
 566         int fd;
 567         pool_state_t state;
 568         char *name = NULL;
 569         nvlist_t *label;
 570         uint64_t guid, spareguid;
 571         nvlist_t *nvroot;
 572         nvlist_t **spares;
 573         uint_t i, nspares;
 574         boolean_t inuse;
 575
 576         if ((fd = open(path, O_RDONLY)) < 0)
 577                 return (B_FALSE);
 578
 579         if (zpool_in_use(g_zfs, fd, &state, &name, &inuse) != 0 ||
 580             !inuse ||
 581             state != POOL_STATE_SPARE ||
 582             zpool_read_label(fd, &label, NULL) != 0) {
 583                 free(name);
 584                 (void) close(fd);
 585                 return (B_FALSE);
 586         }
 587         free(name);
 588         (void) close(fd);
 589
 590         if (config == NULL) {
 591                 nvlist_free(label);
 592                 return (B_TRUE);
 593         }
 594
 595         verify(nvlist_lookup_uint64(label, ZPOOL_CONFIG_GUID, &guid) == 0);
 596         nvlist_free(label);
 597
 598         verify(nvlist_lookup_nvlist(config, ZPOOL_CONFIG_VDEV_TREE,
 599             &nvroot) == 0);
 600         if (nvlist_lookup_nvlist_array(nvroot, ZPOOL_CONFIG_SPARES,
 601             &spares, &nspares) == 0) {
 602                 for (i = 0; i < nspares; i++) {
 603                         verify(nvlist_lookup_uint64(spares[i],
 604                             ZPOOL_CONFIG_GUID, &spareguid) == 0);
 605                         if (spareguid == guid)
 606                                 return (B_TRUE);
 607                 }
 608         }
 609
 610         return (B_FALSE);
 611 }
 612
 613 /*
 614  * Create a leaf vdev.  Determine if this is a file or a device.  If it's a
 615  * device, fill in the device id to make a complete nvlist.  Valid forms for a
 616  * leaf vdev are:
 617  *
 618  *      /dev/xxx        Complete disk path
 619  *      /xxx            Full path to file
 620  *      xxx             Shorthand for <zfs_vdev_paths>/xxx
 621  */
 622 static nvlist_t *
 623 make_leaf_vdev(nvlist_t *props, const char *arg, uint64_t is_log)
 624 {
 625         char path[MAXPATHLEN];
 626         struct stat64 statbuf;
 627         nvlist_t *vdev = NULL;
 628         char *type = NULL;
 629         boolean_t wholedisk = B_FALSE;
 630         uint64_t ashift = 0;
 631         int err;
 632
 633         /*
 634          * Determine what type of vdev this is, and put the full path into
 635          * 'path'.  We detect whether this is a device of file afterwards by
 636          * checking the st_mode of the file.
 637          */
 638         if (arg[0] == '/') {
 639                 /*
 640                  * Complete device or file path.  Exact type is determined by
 641                  * examining the file descriptor afterwards.  Symbolic links
 642                  * are resolved to their real paths for the is_whole_disk()
 643                  * and S_ISBLK/S_ISREG type checks.  However, we are careful
 644                  * to store the given path as ZPOOL_CONFIG_PATH to ensure we
 645                  * can leverage udev's persistent device labels.
 646                  */
 647                 if (realpath(arg, path) == NULL) {
 648                         (void) fprintf(stderr,
 649                             gettext("cannot resolve path '%s'\n"), arg);
 650                         return (NULL);
 651                 }
 652
 653                 wholedisk = is_whole_disk(path);
 654                 if (!wholedisk && (stat64(path, &statbuf) != 0)) {
 655                         (void) fprintf(stderr,
 656                             gettext("cannot open '%s': %s\n"),
 657                             path, strerror(errno));
 658                         return (NULL);
 659                 }
 660
 661                 /* After is_whole_disk() check restore original passed path */
 662                 strlcpy(path, arg, sizeof (path));
 663         } else {
 664                 err = is_shorthand_path(arg, path, sizeof (path),
 665                     &statbuf, &wholedisk);
 666                 if (err != 0) {
 667                         /*
 668                          * If we got ENOENT, then the user gave us
 669                          * gibberish, so try to direct them with a
 670                          * reasonable error message.  Otherwise,
 671                          * regurgitate strerror() since it's the best we
 672                          * can do.
 673                          */
 674                         if (err == ENOENT) {
 675                                 (void) fprintf(stderr,
 676                                     gettext("cannot open '%s': no such "
 677                                     "device in %s\n"), arg, DISK_ROOT);
 678                                 (void) fprintf(stderr,
 679                                     gettext("must be a full path or "
 680                                     "shorthand device name\n"));
 681                                 return (NULL);
 682                         } else {
 683                                 (void) fprintf(stderr,
 684                                     gettext("cannot open '%s': %s\n"),
 685                                     path, strerror(errno));
 686                                 return (NULL);
 687                         }
 688                 }
 689         }
 690
 691         /*
 692          * Determine whether this is a device or a file.
 693          */
 694         if (wholedisk || S_ISBLK(statbuf.st_mode)) {
 695                 type = VDEV_TYPE_DISK;
 696         } else if (S_ISREG(statbuf.st_mode)) {
 697                 type = VDEV_TYPE_FILE;
 698         } else {
 699                 (void) fprintf(stderr, gettext("cannot use '%s': must be a "
 700                     "block device or regular file\n"), path);
 701                 return (NULL);
 702         }
 703
 704         /*
 705          * Finally, we have the complete device or file, and we know that it is
 706          * acceptable to use.  Construct the nvlist to describe this vdev.  All
 707          * vdevs have a 'path' element, and devices also have a 'devid' element.
 708          */
 709         verify(nvlist_alloc(&vdev, NV_UNIQUE_NAME, 0) == 0);
 710         verify(nvlist_add_string(vdev, ZPOOL_CONFIG_PATH, path) == 0);
 711         verify(nvlist_add_string(vdev, ZPOOL_CONFIG_TYPE, type) == 0);
 712         verify(nvlist_add_uint64(vdev, ZPOOL_CONFIG_IS_LOG, is_log) == 0);
 713         if (strcmp(type, VDEV_TYPE_DISK) == 0)
 714                 verify(nvlist_add_uint64(vdev, ZPOOL_CONFIG_WHOLE_DISK,
 715                     (uint64_t)wholedisk) == 0);
 716
 717         /*
 718          * Override defaults if custom properties are provided.
 719          */
 720         if (props != NULL) {
 721                 char *value = NULL;
 722
 723                 if (nvlist_lookup_string(props,
 724                     zpool_prop_to_name(ZPOOL_PROP_ASHIFT), &value) == 0)
 725                         zfs_nicestrtonum(NULL, value, &ashift);
 726         }
 727
 728         /*
 729          * If the device is known to incorrectly report its physical sector
 730          * size explicitly provide the known correct value.
 731          */
 732         if (ashift == 0) {
 733                 int sector_size;
 734
 735                 if (check_sector_size_database(path, &sector_size) == B_TRUE)
 736                         ashift = highbit64(sector_size) - 1;
 737         }
 738
 739         if (ashift > 0)
 740                 (void) nvlist_add_uint64(vdev, ZPOOL_CONFIG_ASHIFT, ashift);
 741
 742         return (vdev);
 743 }
 744
 745 /*
 746  * Go through and verify the replication level of the pool is consistent.
 747  * Performs the following checks:
 748  *
 749  *      For the new spec, verifies that devices in mirrors and raidz are the
 750  *      same size.
 751  *
 752  *      If the current configuration already has inconsistent replication
 753  *      levels, ignore any other potential problems in the new spec.
 754  *
 755  *      Otherwise, make sure that the current spec (if there is one) and the new
 756  *      spec have consistent replication levels.
 757  */
 758 typedef struct replication_level {
 759         char *zprl_type;
 760         uint64_t zprl_children;
 761         uint64_t zprl_parity;
 762 } replication_level_t;
 763
 764 #define ZPOOL_FUZZ      (16 * 1024 * 1024)
 765
 766 /*
 767  * Given a list of toplevel vdevs, return the current replication level.  If
 768  * the config is inconsistent, then NULL is returned.  If 'fatal' is set, then
 769  * an error message will be displayed for each self-inconsistent vdev.
 770  */
 771 static replication_level_t *
 772 get_replication(nvlist_t *nvroot, boolean_t fatal)
 773 {
 774         nvlist_t **top;
 775         uint_t t, toplevels;
 776         nvlist_t **child;
 777         uint_t c, children;
 778         nvlist_t *nv;
 779         char *type;
 780         replication_level_t lastrep = {0};
 781         replication_level_t rep;
 782         replication_level_t *ret;
 783         boolean_t dontreport;
 784
 785         ret = safe_malloc(sizeof (replication_level_t));
 786
 787         verify(nvlist_lookup_nvlist_array(nvroot, ZPOOL_CONFIG_CHILDREN,
 788             &top, &toplevels) == 0);
 789
 790         lastrep.zprl_type = NULL;
 791         for (t = 0; t < toplevels; t++) {
 792                 uint64_t is_log = B_FALSE;
 793
 794                 nv = top[t];
 795
 796                 /*
 797                  * For separate logs we ignore the top level vdev replication
 798                  * constraints.
 799                  */
 800                 (void) nvlist_lookup_uint64(nv, ZPOOL_CONFIG_IS_LOG, &is_log);
 801                 if (is_log)
 802                         continue;
 803
 804                 verify(nvlist_lookup_string(nv, ZPOOL_CONFIG_TYPE,
 805                     &type) == 0);
 806                 if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_CHILDREN,
 807                     &child, &children) != 0) {
 808                         /*
 809                          * This is a 'file' or 'disk' vdev.
 810                          */
 811                         rep.zprl_type = type;
 812                         rep.zprl_children = 1;
 813                         rep.zprl_parity = 0;
 814                 } else {
 815                         uint64_t vdev_size;
 816
 817                         /*
 818                          * This is a mirror or RAID-Z vdev.  Go through and make
 819                          * sure the contents are all the same (files vs. disks),
 820                          * keeping track of the number of elements in the
 821                          * process.
 822                          *
 823                          * We also check that the size of each vdev (if it can
 824                          * be determined) is the same.
 825                          */
 826                         rep.zprl_type = type;
 827                         rep.zprl_children = 0;
 828
 829                         if (strcmp(type, VDEV_TYPE_RAIDZ) == 0) {
 830                                 verify(nvlist_lookup_uint64(nv,
 831                                     ZPOOL_CONFIG_NPARITY,
 832                                     &rep.zprl_parity) == 0);
 833                                 assert(rep.zprl_parity != 0);
 834                         } else {
 835                                 rep.zprl_parity = 0;
 836                         }
 837
 838                         /*
 839                          * The 'dontreport' variable indicates that we've
 840                          * already reported an error for this spec, so don't
 841                          * bother doing it again.
 842                          */
 843                         type = NULL;
 844                         dontreport = 0;
 845                         vdev_size = -1ULL;
 846                         for (c = 0; c < children; c++) {
 847                                 nvlist_t *cnv = child[c];
 848                                 char *path;
 849                                 struct stat64 statbuf;
 850                                 uint64_t size = -1ULL;
 851                                 char *childtype;
 852                                 int fd, err;
 853
 854                                 rep.zprl_children++;
 855
 856                                 verify(nvlist_lookup_string(cnv,
 857                                     ZPOOL_CONFIG_TYPE, &childtype) == 0);
 858
 859                                 /*
 860                                  * If this is a replacing or spare vdev, then
 861                                  * get the real first child of the vdev.
 862                                  */
 863                                 if (strcmp(childtype,
 864                                     VDEV_TYPE_REPLACING) == 0 ||
 865                                     strcmp(childtype, VDEV_TYPE_SPARE) == 0) {
 866                                         nvlist_t **rchild;
 867                                         uint_t rchildren;
 868
 869                                         verify(nvlist_lookup_nvlist_array(cnv,
 870                                             ZPOOL_CONFIG_CHILDREN, &rchild,
 871                                             &rchildren) == 0);
 872                                         assert(rchildren == 2);
 873                                         cnv = rchild[0];
 874
 875                                         verify(nvlist_lookup_string(cnv,
 876                                             ZPOOL_CONFIG_TYPE,
 877                                             &childtype) == 0);
 878                                 }
 879
 880                                 verify(nvlist_lookup_string(cnv,
 881                                     ZPOOL_CONFIG_PATH, &path) == 0);
 882
 883                                 /*
 884                                  * If we have a raidz/mirror that combines disks
 885                                  * with files, report it as an error.
 886                                  */
 887                                 if (!dontreport && type != NULL &&
 888                                     strcmp(type, childtype) != 0) {
 889                                         if (ret != NULL)
 890                                                 free(ret);
 891                                         ret = NULL;
 892                                         if (fatal)
 893                                                 vdev_error(gettext(
 894                                                     "mismatched replication "
 895                                                     "level: %s contains both "
 896                                                     "files and devices\n"),
 897                                                     rep.zprl_type);
 898                                         else
 899                                                 return (NULL);
 900                                         dontreport = B_TRUE;
 901                                 }
 902
 903                                 /*
 904                                  * According to stat(2), the value of 'st_size'
 905                                  * is undefined for block devices and character
 906                                  * devices.  But there is no effective way to
 907                                  * determine the real size in userland.
 908                                  *
 909                                  * Instead, we'll take advantage of an
 910                                  * implementation detail of spec_size().  If the
 911                                  * device is currently open, then we (should)
 912                                  * return a valid size.
 913                                  *
 914                                  * If we still don't get a valid size (indicated
 915                                  * by a size of 0 or MAXOFFSET_T), then ignore
 916                                  * this device altogether.
 917                                  */
 918                                 if ((fd = open(path, O_RDONLY)) >= 0) {
 919                                         err = fstat64(fd, &statbuf);
 920                                         (void) close(fd);
 921                                 } else {
 922                                         err = stat64(path, &statbuf);
 923                                 }
 924
 925                                 if (err != 0 ||
 926                                     statbuf.st_size == 0 ||
 927                                     statbuf.st_size == MAXOFFSET_T)
 928                                         continue;
 929
 930                                 size = statbuf.st_size;
 931
 932                                 /*
 933                                  * Also make sure that devices and
 934                                  * slices have a consistent size.  If
 935                                  * they differ by a significant amount
 936                                  * (~16MB) then report an error.
 937                                  */
 938                                 if (!dontreport &&
 939                                     (vdev_size != -1ULL &&
 940                                     (labs(size - vdev_size) >
 941                                     ZPOOL_FUZZ))) {
 942                                         if (ret != NULL)
 943                                                 free(ret);
 944                                         ret = NULL;
 945                                         if (fatal)
 946                                                 vdev_error(gettext(
 947                                                     "%s contains devices of "
 948                                                     "different sizes\n"),
 949                                                     rep.zprl_type);
 950                                         else
 951                                                 return (NULL);
 952                                         dontreport = B_TRUE;
 953                                 }
 954
 955                                 type = childtype;
 956                                 vdev_size = size;
 957                         }
 958                 }
 959
 960                 /*
 961                  * At this point, we have the replication of the last toplevel
 962                  * vdev in 'rep'.  Compare it to 'lastrep' to see if its
 963                  * different.
 964                  */
 965                 if (lastrep.zprl_type != NULL) {
 966                         if (strcmp(lastrep.zprl_type, rep.zprl_type) != 0) {
 967                                 if (ret != NULL)
 968                                         free(ret);
 969                                 ret = NULL;
 970                                 if (fatal)
 971                                         vdev_error(gettext(
 972                                             "mismatched replication level: "
 973                                             "both %s and %s vdevs are "
 974                                             "present\n"),
 975                                             lastrep.zprl_type, rep.zprl_type);
 976                                 else
 977                                         return (NULL);
 978                         } else if (lastrep.zprl_parity != rep.zprl_parity) {
 979                                 if (ret)
 980                                         free(ret);
 981                                 ret = NULL;
 982                                 if (fatal)
 983                                         vdev_error(gettext(
 984                                             "mismatched replication level: "
 985                                             "both %llu and %llu device parity "
 986                                             "%s vdevs are present\n"),
 987                                             lastrep.zprl_parity,
 988                                             rep.zprl_parity,
 989                                             rep.zprl_type);
 990                                 else
 991                                         return (NULL);
 992                         } else if (lastrep.zprl_children != rep.zprl_children) {
 993                                 if (ret)
 994                                         free(ret);
 995                                 ret = NULL;
 996                                 if (fatal)
 997                                         vdev_error(gettext(
 998                                             "mismatched replication level: "
 999                                             "both %llu-way and %llu-way %s "
1000                                             "vdevs are present\n"),
1001                                             lastrep.zprl_children,
1002                                             rep.zprl_children,
1003                                             rep.zprl_type);
1004                                 else
1005                                         return (NULL);
1006                         }
1007                 }
1008                 lastrep = rep;
1009         }
1010
1011         if (ret != NULL)
1012                 *ret = rep;
1013
1014         return (ret);
1015 }
1016
1017 /*
1018  * Check the replication level of the vdev spec against the current pool.  Calls
1019  * get_replication() to make sure the new spec is self-consistent.  If the pool
1020  * has a consistent replication level, then we ignore any errors.  Otherwise,
1021  * report any difference between the two.
1022  */
1023 static int
1024 check_replication(nvlist_t *config, nvlist_t *newroot)
1025 {
1026         nvlist_t **child;
1027         uint_t  children;
1028         replication_level_t *current = NULL, *new;
1029         int ret;
1030
1031         /*
1032          * If we have a current pool configuration, check to see if it's
1033          * self-consistent.  If not, simply return success.
1034          */
1035         if (config != NULL) {
1036                 nvlist_t *nvroot;
1037
1038                 verify(nvlist_lookup_nvlist(config, ZPOOL_CONFIG_VDEV_TREE,
1039                     &nvroot) == 0);
1040                 if ((current = get_replication(nvroot, B_FALSE)) == NULL)
1041                         return (0);
1042         }
1043         /*
1044          * for spares there may be no children, and therefore no
1045          * replication level to check
1046          */
1047         if ((nvlist_lookup_nvlist_array(newroot, ZPOOL_CONFIG_CHILDREN,
1048             &child, &children) != 0) || (children == 0)) {
1049                 free(current);
1050                 return (0);
1051         }
1052
1053         /*
1054          * If all we have is logs then there's no replication level to check.
1055          */
1056         if (num_logs(newroot) == children) {
1057                 free(current);
1058                 return (0);
1059         }
1060
1061         /*
1062          * Get the replication level of the new vdev spec, reporting any
1063          * inconsistencies found.
1064          */
1065         if ((new = get_replication(newroot, B_TRUE)) == NULL) {
1066                 free(current);
1067                 return (-1);
1068         }
1069
1070         /*
1071          * Check to see if the new vdev spec matches the replication level of
1072          * the current pool.
1073          */
1074         ret = 0;
1075         if (current != NULL) {
1076                 if (strcmp(current->zprl_type, new->zprl_type) != 0) {
1077                         vdev_error(gettext(
1078                             "mismatched replication level: pool uses %s "
1079                             "and new vdev is %s\n"),
1080                             current->zprl_type, new->zprl_type);
1081                         ret = -1;
1082                 } else if (current->zprl_parity != new->zprl_parity) {
1083                         vdev_error(gettext(
1084                             "mismatched replication level: pool uses %llu "
1085                             "device parity and new vdev uses %llu\n"),
1086                             current->zprl_parity, new->zprl_parity);
1087                         ret = -1;
1088                 } else if (current->zprl_children != new->zprl_children) {
1089                         vdev_error(gettext(
1090                             "mismatched replication level: pool uses %llu-way "
1091                             "%s and new vdev uses %llu-way %s\n"),
1092                             current->zprl_children, current->zprl_type,
1093                             new->zprl_children, new->zprl_type);
1094                         ret = -1;
1095                 }
1096         }
1097
1098         free(new);
1099         if (current != NULL)
1100                 free(current);
1101
1102         return (ret);
1103 }
1104
1105 static int
1106 zero_label(char *path)
1107 {
1108         const int size = 4096;
1109         char buf[size];
1110         int err, fd;
1111
1112         if ((fd = open(path, O_WRONLY|O_EXCL)) < 0) {
1113                 (void) fprintf(stderr, gettext("cannot open '%s': %s\n"),
1114                     path, strerror(errno));
1115                 return (-1);
1116         }
1117
1118         memset(buf, 0, size);
1119         err = write(fd, buf, size);
1120         (void) fdatasync(fd);
1121         (void) close(fd);
1122
1123         if (err == -1) {
1124                 (void) fprintf(stderr, gettext("cannot zero first %d bytes "
1125                     "of '%s': %s\n"), size, path, strerror(errno));
1126                 return (-1);
1127         }
1128
1129         if (err != size) {
1130                 (void) fprintf(stderr, gettext("could only zero %d/%d bytes "
1131                     "of '%s'\n"), err, size, path);
1132                 return (-1);
1133         }
1134
1135         return (0);
1136 }
1137
1138 /*
1139  * Go through and find any whole disks in the vdev specification, labelling them
1140  * as appropriate.  When constructing the vdev spec, we were unable to open this
1141  * device in order to provide a devid.  Now that we have labelled the disk and
1142  * know that slice 0 is valid, we can construct the devid now.
1143  *
1144  * If the disk was already labeled with an EFI label, we will have gotten the
1145  * devid already (because we were able to open the whole disk).  Otherwise, we
1146  * need to get the devid after we label the disk.
1147  */
1148 static int
1149 make_disks(zpool_handle_t *zhp, nvlist_t *nv)
1150 {
1151         nvlist_t **child;
1152         uint_t c, children;
1153         char *type, *path;
1154         char devpath[MAXPATHLEN];
1155         char udevpath[MAXPATHLEN];
1156         uint64_t wholedisk;
1157         struct stat64 statbuf;
1158         int is_exclusive = 0;
1159         int fd;
1160         int ret;
1161
1162         verify(nvlist_lookup_string(nv, ZPOOL_CONFIG_TYPE, &type) == 0);
1163
1164         if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_CHILDREN,
1165             &child, &children) != 0) {
1166
1167                 if (strcmp(type, VDEV_TYPE_DISK) != 0)
1168                         return (0);
1169
1170                 /*
1171                  * We have a disk device.  If this is a whole disk write
1172                  * out the efi partition table, otherwise write zero's to
1173                  * the first 4k of the partition.  This is to ensure that
1174                  * libblkid will not misidentify the partition due to a
1175                  * magic value left by the previous filesystem.
1176                  */
1177                 verify(!nvlist_lookup_string(nv, ZPOOL_CONFIG_PATH, &path));
1178                 verify(!nvlist_lookup_uint64(nv, ZPOOL_CONFIG_WHOLE_DISK,
1179                     &wholedisk));
1180
1181                 if (!wholedisk) {
1182                         /*
1183                          * Update device id string for mpath nodes (Linux only)
1184                          */
1185                         if (is_mpath_whole_disk(path))
1186                                 update_vdev_config_dev_strs(nv);
1187
1188                         (void) zero_label(path);
1189                         return (0);
1190                 }
1191
1192                 if (realpath(path, devpath) == NULL) {
1193                         ret = errno;
1194                         (void) fprintf(stderr,
1195                             gettext("cannot resolve path '%s'\n"), path);
1196                         return (ret);
1197                 }
1198
1199                 /*
1200                  * Remove any previously existing symlink from a udev path to
1201                  * the device before labeling the disk.  This ensures that
1202                  * only newly created links are used.  Otherwise there is a
1203                  * window between when udev deletes and recreates the link
1204                  * during which access attempts will fail with ENOENT.
1205                  */
1206                 strlcpy(udevpath, path, MAXPATHLEN);
1207                 (void) zfs_append_partition(udevpath, MAXPATHLEN);
1208
1209                 fd = open(devpath, O_RDWR|O_EXCL);
1210                 if (fd == -1) {
1211                         if (errno == EBUSY)
1212                                 is_exclusive = 1;
1213                 } else {
1214                         (void) close(fd);
1215                 }
1216
1217                 /*
1218                  * If the partition exists, contains a valid spare label,
1219                  * and is opened exclusively there is no need to partition
1220                  * it.  Hot spares have already been partitioned and are
1221                  * held open exclusively by the kernel as a safety measure.
1222                  *
1223                  * If the provided path is for a /dev/disk/ device its
1224                  * symbolic link will be removed, partition table created,
1225                  * and then block until udev creates the new link.
1226                  */
1227                 if (!is_exclusive || !is_spare(NULL, udevpath)) {
1228                         char *devnode = strrchr(devpath, '/') + 1;
1229
1230                         ret = strncmp(udevpath, UDISK_ROOT, strlen(UDISK_ROOT));
1231                         if (ret == 0) {
1232                                 ret = lstat64(udevpath, &statbuf);
1233                                 if (ret == 0 && S_ISLNK(statbuf.st_mode))
1234                                         (void) unlink(udevpath);
1235                         }
1236
1237                         /*
1238                          * When labeling a pool the raw device node name
1239                          * is provided as it appears under /dev/.
1240                          */
1241                         if (zpool_label_disk(g_zfs, zhp, devnode) == -1)
1242                                 return (-1);
1243
1244                         /*
1245                          * Wait for udev to signal the device is available
1246                          * by the provided path.
1247                          */
1248                         ret = zpool_label_disk_wait(udevpath, DISK_LABEL_WAIT);
1249                         if (ret) {
1250                                 (void) fprintf(stderr,
1251                                     gettext("missing link: %s was "
1252                                     "partitioned but %s is missing\n"),
1253                                     devnode, udevpath);
1254                                 return (ret);
1255                         }
1256
1257                         ret = zero_label(udevpath);
1258                         if (ret)
1259                                 return (ret);
1260                 }
1261
1262                 /*
1263                  * Update the path to refer to the partition.  The presence of
1264                  * the 'whole_disk' field indicates to the CLI that we should
1265                  * chop off the partition number when displaying the device in
1266                  * future output.
1267                  */
1268                 verify(nvlist_add_string(nv, ZPOOL_CONFIG_PATH, udevpath) == 0);
1269
1270                 /*
1271                  * Update device id strings for whole disks (Linux only)
1272                  */
1273                 update_vdev_config_dev_strs(nv);
1274
1275                 return (0);
1276         }
1277
1278         for (c = 0; c < children; c++)
1279                 if ((ret = make_disks(zhp, child[c])) != 0)
1280                         return (ret);
1281
1282         if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_SPARES,
1283             &child, &children) == 0)
1284                 for (c = 0; c < children; c++)
1285                         if ((ret = make_disks(zhp, child[c])) != 0)
1286                                 return (ret);
1287
1288         if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_L2CACHE,
1289             &child, &children) == 0)
1290                 for (c = 0; c < children; c++)
1291                         if ((ret = make_disks(zhp, child[c])) != 0)
1292                                 return (ret);
1293
1294         return (0);
1295 }
1296
1297 /*
1298  * Go through and find any devices that are in use.  We rely on libdiskmgt for
1299  * the majority of this task.
1300  */
1301 static boolean_t
1302 is_device_in_use(nvlist_t *config, nvlist_t *nv, boolean_t force,
1303     boolean_t replacing, boolean_t isspare)
1304 {
1305         nvlist_t **child;
1306         uint_t c, children;
1307         char *type, *path;
1308         int ret = 0;
1309         char buf[MAXPATHLEN];
1310         uint64_t wholedisk = B_FALSE;
1311         boolean_t anyinuse = B_FALSE;
1312
1313         verify(nvlist_lookup_string(nv, ZPOOL_CONFIG_TYPE, &type) == 0);
1314
1315         if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_CHILDREN,
1316             &child, &children) != 0) {
1317
1318                 verify(!nvlist_lookup_string(nv, ZPOOL_CONFIG_PATH, &path));
1319                 if (strcmp(type, VDEV_TYPE_DISK) == 0)
1320                         verify(!nvlist_lookup_uint64(nv,
1321                             ZPOOL_CONFIG_WHOLE_DISK, &wholedisk));
1322
1323                 /*
1324                  * As a generic check, we look to see if this is a replace of a
1325                  * hot spare within the same pool.  If so, we allow it
1326                  * regardless of what libblkid or zpool_in_use() says.
1327                  */
1328                 if (replacing) {
1329                         (void) strlcpy(buf, path, sizeof (buf));
1330                         if (wholedisk) {
1331                                 ret = zfs_append_partition(buf,  sizeof (buf));
1332                                 if (ret == -1)
1333                                         return (-1);
1334                         }
1335
1336                         if (is_spare(config, buf))
1337                                 return (B_FALSE);
1338                 }
1339
1340                 if (strcmp(type, VDEV_TYPE_DISK) == 0)
1341                         ret = check_device(path, force, isspare, wholedisk);
1342
1343                 else if (strcmp(type, VDEV_TYPE_FILE) == 0)
1344                         ret = check_file(path, force, isspare);
1345
1346                 return (ret != 0);
1347         }
1348
1349         for (c = 0; c < children; c++)
1350                 if (is_device_in_use(config, child[c], force, replacing,
1351                     B_FALSE))
1352                         anyinuse = B_TRUE;
1353
1354         if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_SPARES,
1355             &child, &children) == 0)
1356                 for (c = 0; c < children; c++)
1357                         if (is_device_in_use(config, child[c], force, replacing,
1358                             B_TRUE))
1359                                 anyinuse = B_TRUE;
1360
1361         if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_L2CACHE,
1362             &child, &children) == 0)
1363                 for (c = 0; c < children; c++)
1364                         if (is_device_in_use(config, child[c], force, replacing,
1365                             B_FALSE))
1366                                 anyinuse = B_TRUE;
1367
1368         return (anyinuse);
1369 }
1370
1371 static const char *
1372 is_grouping(const char *type, int *mindev, int *maxdev)
1373 {
1374         if (strncmp(type, "raidz", 5) == 0) {
1375                 const char *p = type + 5;
1376                 char *end;
1377                 long nparity;
1378
1379                 if (*p == '\0') {
1380                         nparity = 1;
1381                 } else if (*p == '0') {
1382                         return (NULL); /* no zero prefixes allowed */
1383                 } else {
1384                         errno = 0;
1385                         nparity = strtol(p, &end, 10);
1386                         if (errno != 0 || nparity < 1 || nparity >= 255 ||
1387                             *end != '\0')
1388                                 return (NULL);
1389                 }
1390
1391                 if (mindev != NULL)
1392                         *mindev = nparity + 1;
1393                 if (maxdev != NULL)
1394                         *maxdev = 255;
1395                 return (VDEV_TYPE_RAIDZ);
1396         }
1397
1398         if (maxdev != NULL)
1399                 *maxdev = INT_MAX;
1400
1401         if (strcmp(type, "mirror") == 0) {
1402                 if (mindev != NULL)
1403                         *mindev = 2;
1404                 return (VDEV_TYPE_MIRROR);
1405         }
1406
1407         if (strcmp(type, "spare") == 0) {
1408                 if (mindev != NULL)
1409                         *mindev = 1;
1410                 return (VDEV_TYPE_SPARE);
1411         }
1412
1413         if (strcmp(type, "log") == 0) {
1414                 if (mindev != NULL)
1415                         *mindev = 1;
1416                 return (VDEV_TYPE_LOG);
1417         }
1418
1419         if (strcmp(type, "cache") == 0) {
1420                 if (mindev != NULL)
1421                         *mindev = 1;
1422                 return (VDEV_TYPE_L2CACHE);
1423         }
1424
1425         return (NULL);
1426 }
1427
1428 /*
1429  * Construct a syntactically valid vdev specification,
1430  * and ensure that all devices and files exist and can be opened.
1431  * Note: we don't bother freeing anything in the error paths
1432  * because the program is just going to exit anyway.
1433  */
1434 nvlist_t *
1435 construct_spec(nvlist_t *props, int argc, char **argv)
1436 {
1437         nvlist_t *nvroot, *nv, **top, **spares, **l2cache;
1438         int t, toplevels, mindev, maxdev, nspares, nlogs, nl2cache;
1439         const char *type;
1440         uint64_t is_log;
1441         boolean_t seen_logs;
1442
1443         top = NULL;
1444         toplevels = 0;
1445         spares = NULL;
1446         l2cache = NULL;
1447         nspares = 0;
1448         nlogs = 0;
1449         nl2cache = 0;
1450         is_log = B_FALSE;
1451         seen_logs = B_FALSE;
1452         nvroot = NULL;
1453
1454         while (argc > 0) {
1455                 nv = NULL;
1456
1457                 /*
1458                  * If it's a mirror or raidz, the subsequent arguments are
1459                  * its leaves -- until we encounter the next mirror or raidz.
1460                  */
1461                 if ((type = is_grouping(argv[0], &mindev, &maxdev)) != NULL) {
1462                         nvlist_t **child = NULL;
1463                         int c, children = 0;
1464
1465                         if (strcmp(type, VDEV_TYPE_SPARE) == 0) {
1466                                 if (spares != NULL) {
1467                                         (void) fprintf(stderr,
1468                                             gettext("invalid vdev "
1469                                             "specification: 'spare' can be "
1470                                             "specified only once\n"));
1471                                         goto spec_out;
1472                                 }
1473                                 is_log = B_FALSE;
1474                         }
1475
1476                         if (strcmp(type, VDEV_TYPE_LOG) == 0) {
1477                                 if (seen_logs) {
1478                                         (void) fprintf(stderr,
1479                                             gettext("invalid vdev "
1480                                             "specification: 'log' can be "
1481                                             "specified only once\n"));
1482                                         goto spec_out;
1483                                 }
1484                                 seen_logs = B_TRUE;
1485                                 is_log = B_TRUE;
1486                                 argc--;
1487                                 argv++;
1488                                 /*
1489                                  * A log is not a real grouping device.
1490                                  * We just set is_log and continue.
1491                                  */
1492                                 continue;
1493                         }
1494
1495                         if (strcmp(type, VDEV_TYPE_L2CACHE) == 0) {
1496                                 if (l2cache != NULL) {
1497                                         (void) fprintf(stderr,
1498                                             gettext("invalid vdev "
1499                                             "specification: 'cache' can be "
1500                                             "specified only once\n"));
1501                                         goto spec_out;
1502                                 }
1503                                 is_log = B_FALSE;
1504                         }
1505
1506                         if (is_log) {
1507                                 if (strcmp(type, VDEV_TYPE_MIRROR) != 0) {
1508                                         (void) fprintf(stderr,
1509                                             gettext("invalid vdev "
1510                                             "specification: unsupported 'log' "
1511                                             "device: %s\n"), type);
1512                                         goto spec_out;
1513                                 }
1514                                 nlogs++;
1515                         }
1516
1517                         for (c = 1; c < argc; c++) {
1518                                 if (is_grouping(argv[c], NULL, NULL) != NULL)
1519                                         break;
1520                                 children++;
1521                                 child = realloc(child,
1522                                     children * sizeof (nvlist_t *));
1523                                 if (child == NULL)
1524                                         zpool_no_memory();
1525                                 if ((nv = make_leaf_vdev(props, argv[c],
1526                                     B_FALSE)) == NULL) {
1527                                         for (c = 0; c < children - 1; c++)
1528                                                 nvlist_free(child[c]);
1529                                         free(child);
1530                                         goto spec_out;
1531                                 }
1532
1533                                 child[children - 1] = nv;
1534                         }
1535
1536                         if (children < mindev) {
1537                                 (void) fprintf(stderr, gettext("invalid vdev "
1538                                     "specification: %s requires at least %d "
1539                                     "devices\n"), argv[0], mindev);
1540                                 for (c = 0; c < children; c++)
1541                                         nvlist_free(child[c]);
1542                                 free(child);
1543                                 goto spec_out;
1544                         }
1545
1546                         if (children > maxdev) {
1547                                 (void) fprintf(stderr, gettext("invalid vdev "
1548                                     "specification: %s supports no more than "
1549                                     "%d devices\n"), argv[0], maxdev);
1550                                 for (c = 0; c < children; c++)
1551                                         nvlist_free(child[c]);
1552                                 free(child);
1553                                 goto spec_out;
1554                         }
1555
1556                         argc -= c;
1557                         argv += c;
1558
1559                         if (strcmp(type, VDEV_TYPE_SPARE) == 0) {
1560                                 spares = child;
1561                                 nspares = children;
1562                                 continue;
1563                         } else if (strcmp(type, VDEV_TYPE_L2CACHE) == 0) {
1564                                 l2cache = child;
1565                                 nl2cache = children;
1566                                 continue;
1567                         } else {
1568                                 verify(nvlist_alloc(&nv, NV_UNIQUE_NAME,
1569                                     0) == 0);
1570                                 verify(nvlist_add_string(nv, ZPOOL_CONFIG_TYPE,
1571                                     type) == 0);
1572                                 verify(nvlist_add_uint64(nv,
1573                                     ZPOOL_CONFIG_IS_LOG, is_log) == 0);
1574                                 if (strcmp(type, VDEV_TYPE_RAIDZ) == 0) {
1575                                         verify(nvlist_add_uint64(nv,
1576                                             ZPOOL_CONFIG_NPARITY,
1577                                             mindev - 1) == 0);
1578                                 }
1579                                 verify(nvlist_add_nvlist_array(nv,
1580                                     ZPOOL_CONFIG_CHILDREN, child,
1581                                     children) == 0);
1582
1583                                 for (c = 0; c < children; c++)
1584                                         nvlist_free(child[c]);
1585                                 free(child);
1586                         }
1587                 } else {
1588                         /*
1589                          * We have a device.  Pass off to make_leaf_vdev() to
1590                          * construct the appropriate nvlist describing the vdev.
1591                          */
1592                         if ((nv = make_leaf_vdev(props, argv[0],
1593                             is_log)) == NULL)
1594                                 goto spec_out;
1595
1596                         if (is_log)
1597                                 nlogs++;
1598                         argc--;
1599                         argv++;
1600                 }
1601
1602                 toplevels++;
1603                 top = realloc(top, toplevels * sizeof (nvlist_t *));
1604                 if (top == NULL)
1605                         zpool_no_memory();
1606                 top[toplevels - 1] = nv;
1607         }
1608
1609         if (toplevels == 0 && nspares == 0 && nl2cache == 0) {
1610                 (void) fprintf(stderr, gettext("invalid vdev "
1611                     "specification: at least one toplevel vdev must be "
1612                     "specified\n"));
1613                 goto spec_out;
1614         }
1615
1616         if (seen_logs && nlogs == 0) {
1617                 (void) fprintf(stderr, gettext("invalid vdev specification: "
1618                     "log requires at least 1 device\n"));
1619                 goto spec_out;
1620         }
1621
1622         /*
1623          * Finally, create nvroot and add all top-level vdevs to it.
1624          */
1625         verify(nvlist_alloc(&nvroot, NV_UNIQUE_NAME, 0) == 0);
1626         verify(nvlist_add_string(nvroot, ZPOOL_CONFIG_TYPE,
1627             VDEV_TYPE_ROOT) == 0);
1628         verify(nvlist_add_nvlist_array(nvroot, ZPOOL_CONFIG_CHILDREN,
1629             top, toplevels) == 0);
1630         if (nspares != 0)
1631                 verify(nvlist_add_nvlist_array(nvroot, ZPOOL_CONFIG_SPARES,
1632                     spares, nspares) == 0);
1633         if (nl2cache != 0)
1634                 verify(nvlist_add_nvlist_array(nvroot, ZPOOL_CONFIG_L2CACHE,
1635                     l2cache, nl2cache) == 0);
1636
1637 spec_out:
1638         for (t = 0; t < toplevels; t++)
1639                 nvlist_free(top[t]);
1640         for (t = 0; t < nspares; t++)
1641                 nvlist_free(spares[t]);
1642         for (t = 0; t < nl2cache; t++)
1643                 nvlist_free(l2cache[t]);
1644
1645         free(spares);
1646         free(l2cache);
1647         free(top);
1648
1649         return (nvroot);
1650 }
1651
1652 nvlist_t *
1653 split_mirror_vdev(zpool_handle_t *zhp, char *newname, nvlist_t *props,
1654     splitflags_t flags, int argc, char **argv)
1655 {
1656         nvlist_t *newroot = NULL, **child;
1657         uint_t c, children;
1658
1659         if (argc > 0) {
1660                 if ((newroot = construct_spec(props, argc, argv)) == NULL) {
1661                         (void) fprintf(stderr, gettext("Unable to build a "
1662                             "pool from the specified devices\n"));
1663                         return (NULL);
1664                 }
1665
1666                 if (!flags.dryrun && make_disks(zhp, newroot) != 0) {
1667                         nvlist_free(newroot);
1668                         return (NULL);
1669                 }
1670
1671                 /* avoid any tricks in the spec */
1672                 verify(nvlist_lookup_nvlist_array(newroot,
1673                     ZPOOL_CONFIG_CHILDREN, &child, &children) == 0);
1674                 for (c = 0; c < children; c++) {
1675                         char *path;
1676                         const char *type;
1677                         int min, max;
1678
1679                         verify(nvlist_lookup_string(child[c],
1680                             ZPOOL_CONFIG_PATH, &path) == 0);
1681                         if ((type = is_grouping(path, &min, &max)) != NULL) {
1682                                 (void) fprintf(stderr, gettext("Cannot use "
1683                                     "'%s' as a device for splitting\n"), type);
1684                                 nvlist_free(newroot);
1685                                 return (NULL);
1686                         }
1687                 }
1688         }
1689
1690         if (zpool_vdev_split(zhp, newname, &newroot, props, flags) != 0) {
1691                 nvlist_free(newroot);
1692                 return (NULL);
1693         }
1694
1695         return (newroot);
1696 }
1697
1698 /*
1699  * Get and validate the contents of the given vdev specification.  This ensures
1700  * that the nvlist returned is well-formed, that all the devices exist, and that
1701  * they are not currently in use by any other known consumer.  The 'poolconfig'
1702  * parameter is the current configuration of the pool when adding devices
1703  * existing pool, and is used to perform additional checks, such as changing the
1704  * replication level of the pool.  It can be 'NULL' to indicate that this is a
1705  * new pool.  The 'force' flag controls whether devices should be forcefully
1706  * added, even if they appear in use.
1707  */
1708 nvlist_t *
1709 make_root_vdev(zpool_handle_t *zhp, nvlist_t *props, int force, int check_rep,
1710     boolean_t replacing, boolean_t dryrun, int argc, char **argv)
1711 {
1712         nvlist_t *newroot;
1713         nvlist_t *poolconfig = NULL;
1714         is_force = force;
1715
1716         /*
1717          * Construct the vdev specification.  If this is successful, we know
1718          * that we have a valid specification, and that all devices can be
1719          * opened.
1720          */
1721         if ((newroot = construct_spec(props, argc, argv)) == NULL)
1722                 return (NULL);
1723
1724         if (zhp && ((poolconfig = zpool_get_config(zhp, NULL)) == NULL)) {
1725                 nvlist_free(newroot);
1726                 return (NULL);
1727         }
1728
1729         /*
1730          * Validate each device to make sure that its not shared with another
1731          * subsystem.  We do this even if 'force' is set, because there are some
1732          * uses (such as a dedicated dump device) that even '-f' cannot
1733          * override.
1734          */
1735         if (is_device_in_use(poolconfig, newroot, force, replacing, B_FALSE)) {
1736                 nvlist_free(newroot);
1737                 return (NULL);
1738         }
1739
1740         /*
1741          * Check the replication level of the given vdevs and report any errors
1742          * found.  We include the existing pool spec, if any, as we need to
1743          * catch changes against the existing replication level.
1744          */
1745         if (check_rep && check_replication(poolconfig, newroot) != 0) {
1746                 nvlist_free(newroot);
1747                 return (NULL);
1748         }
1749
1750         /*
1751          * Run through the vdev specification and label any whole disks found.
1752          */
1753         if (!dryrun && make_disks(zhp, newroot) != 0) {
1754                 nvlist_free(newroot);
1755                 return (NULL);
1756         }
1757
1758         return (newroot);
1759 }