zfs.c revision 350338
1/*- 2 * Copyright (c) 2007 Doug Rabson 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 * 26 * $FreeBSD: stable/11/stand/libsa/zfs/zfs.c 350338 2019-07-26 01:35:06Z kevans $ 27 */ 28 29#include <sys/cdefs.h> 30__FBSDID("$FreeBSD: stable/11/stand/libsa/zfs/zfs.c 350338 2019-07-26 01:35:06Z kevans $"); 31 32/* 33 * Stand-alone file reading package. 34 */ 35 36#include <stand.h> 37#include <sys/disk.h> 38#include <sys/param.h> 39#include <sys/time.h> 40#include <sys/queue.h> 41#include <disk.h> 42#include <part.h> 43#include <stddef.h> 44#include <stdarg.h> 45#include <string.h> 46#include <bootstrap.h> 47 48#include "libzfs.h" 49 50#include "zfsimpl.c" 51 52/* Define the range of indexes to be populated with ZFS Boot Environments */ 53#define ZFS_BE_FIRST 4 54#define ZFS_BE_LAST 8 55 56static int zfs_open(const char *path, struct open_file *f); 57static int zfs_close(struct open_file *f); 58static int zfs_read(struct open_file *f, void *buf, size_t size, size_t *resid); 59static off_t zfs_seek(struct open_file *f, off_t offset, int where); 60static int zfs_stat(struct open_file *f, struct stat *sb); 61static int zfs_readdir(struct open_file *f, struct dirent *d); 62 63static void zfs_bootenv_initial(const char *); 64 65struct devsw zfs_dev; 66 67struct fs_ops zfs_fsops = { 68 "zfs", 69 zfs_open, 70 zfs_close, 71 zfs_read, 72 null_write, 73 zfs_seek, 74 zfs_stat, 75 zfs_readdir 76}; 77 78/* 79 * In-core open file. 80 */ 81struct file { 82 off_t f_seekp; /* seek pointer */ 83 dnode_phys_t f_dnode; 84 uint64_t f_zap_type; /* zap type for readdir */ 85 uint64_t f_num_leafs; /* number of fzap leaf blocks */ 86 zap_leaf_phys_t *f_zap_leaf; /* zap leaf buffer */ 87}; 88 89static int zfs_env_index; 90static int zfs_env_count; 91 92SLIST_HEAD(zfs_be_list, zfs_be_entry) zfs_be_head = SLIST_HEAD_INITIALIZER(zfs_be_head); 93struct zfs_be_list *zfs_be_headp; 94struct zfs_be_entry { 95 const char *name; 96 SLIST_ENTRY(zfs_be_entry) entries; 97} *zfs_be, *zfs_be_tmp; 98 99/* 100 * Open a file. 101 */ 102static int 103zfs_open(const char *upath, struct open_file *f) 104{ 105 struct zfsmount *mount = (struct zfsmount *)f->f_devdata; 106 struct file *fp; 107 int rc; 108 109 if (f->f_dev != &zfs_dev) 110 return (EINVAL); 111 112 /* allocate file system specific data structure */ 113 fp = malloc(sizeof(struct file)); 114 bzero(fp, sizeof(struct file)); 115 f->f_fsdata = (void *)fp; 116 117 rc = zfs_lookup(mount, upath, &fp->f_dnode); 118 fp->f_seekp = 0; 119 if (rc) { 120 f->f_fsdata = NULL; 121 free(fp); 122 } 123 return (rc); 124} 125 126static int 127zfs_close(struct open_file *f) 128{ 129 struct file *fp = (struct file *)f->f_fsdata; 130 131 dnode_cache_obj = NULL; 132 f->f_fsdata = (void *)0; 133 if (fp == (struct file *)0) 134 return (0); 135 136 free(fp); 137 return (0); 138} 139 140/* 141 * Copy a portion of a file into kernel memory. 142 * Cross block boundaries when necessary. 143 */ 144static int 145zfs_read(struct open_file *f, void *start, size_t size, size_t *resid /* out */) 146{ 147 const spa_t *spa = ((struct zfsmount *)f->f_devdata)->spa; 148 struct file *fp = (struct file *)f->f_fsdata; 149 struct stat sb; 150 size_t n; 151 int rc; 152 153 rc = zfs_stat(f, &sb); 154 if (rc) 155 return (rc); 156 n = size; 157 if (fp->f_seekp + n > sb.st_size) 158 n = sb.st_size - fp->f_seekp; 159 160 rc = dnode_read(spa, &fp->f_dnode, fp->f_seekp, start, n); 161 if (rc) 162 return (rc); 163 164 if (0) { 165 int i; 166 for (i = 0; i < n; i++) 167 putchar(((char*) start)[i]); 168 } 169 fp->f_seekp += n; 170 if (resid) 171 *resid = size - n; 172 173 return (0); 174} 175 176static off_t 177zfs_seek(struct open_file *f, off_t offset, int where) 178{ 179 struct file *fp = (struct file *)f->f_fsdata; 180 181 switch (where) { 182 case SEEK_SET: 183 fp->f_seekp = offset; 184 break; 185 case SEEK_CUR: 186 fp->f_seekp += offset; 187 break; 188 case SEEK_END: 189 { 190 struct stat sb; 191 int error; 192 193 error = zfs_stat(f, &sb); 194 if (error != 0) { 195 errno = error; 196 return (-1); 197 } 198 fp->f_seekp = sb.st_size - offset; 199 break; 200 } 201 default: 202 errno = EINVAL; 203 return (-1); 204 } 205 return (fp->f_seekp); 206} 207 208static int 209zfs_stat(struct open_file *f, struct stat *sb) 210{ 211 const spa_t *spa = ((struct zfsmount *)f->f_devdata)->spa; 212 struct file *fp = (struct file *)f->f_fsdata; 213 214 return (zfs_dnode_stat(spa, &fp->f_dnode, sb)); 215} 216 217static int 218zfs_readdir(struct open_file *f, struct dirent *d) 219{ 220 const spa_t *spa = ((struct zfsmount *)f->f_devdata)->spa; 221 struct file *fp = (struct file *)f->f_fsdata; 222 mzap_ent_phys_t mze; 223 struct stat sb; 224 size_t bsize = fp->f_dnode.dn_datablkszsec << SPA_MINBLOCKSHIFT; 225 int rc; 226 227 rc = zfs_stat(f, &sb); 228 if (rc) 229 return (rc); 230 if (!S_ISDIR(sb.st_mode)) 231 return (ENOTDIR); 232 233 /* 234 * If this is the first read, get the zap type. 235 */ 236 if (fp->f_seekp == 0) { 237 rc = dnode_read(spa, &fp->f_dnode, 238 0, &fp->f_zap_type, sizeof(fp->f_zap_type)); 239 if (rc) 240 return (rc); 241 242 if (fp->f_zap_type == ZBT_MICRO) { 243 fp->f_seekp = offsetof(mzap_phys_t, mz_chunk); 244 } else { 245 rc = dnode_read(spa, &fp->f_dnode, 246 offsetof(zap_phys_t, zap_num_leafs), 247 &fp->f_num_leafs, 248 sizeof(fp->f_num_leafs)); 249 if (rc) 250 return (rc); 251 252 fp->f_seekp = bsize; 253 fp->f_zap_leaf = (zap_leaf_phys_t *)malloc(bsize); 254 rc = dnode_read(spa, &fp->f_dnode, 255 fp->f_seekp, 256 fp->f_zap_leaf, 257 bsize); 258 if (rc) 259 return (rc); 260 } 261 } 262 263 if (fp->f_zap_type == ZBT_MICRO) { 264 mzap_next: 265 if (fp->f_seekp >= bsize) 266 return (ENOENT); 267 268 rc = dnode_read(spa, &fp->f_dnode, 269 fp->f_seekp, &mze, sizeof(mze)); 270 if (rc) 271 return (rc); 272 fp->f_seekp += sizeof(mze); 273 274 if (!mze.mze_name[0]) 275 goto mzap_next; 276 277 d->d_fileno = ZFS_DIRENT_OBJ(mze.mze_value); 278 d->d_type = ZFS_DIRENT_TYPE(mze.mze_value); 279 strcpy(d->d_name, mze.mze_name); 280 d->d_namlen = strlen(d->d_name); 281 return (0); 282 } else { 283 zap_leaf_t zl; 284 zap_leaf_chunk_t *zc, *nc; 285 int chunk; 286 size_t namelen; 287 char *p; 288 uint64_t value; 289 290 /* 291 * Initialise this so we can use the ZAP size 292 * calculating macros. 293 */ 294 zl.l_bs = ilog2(bsize); 295 zl.l_phys = fp->f_zap_leaf; 296 297 /* 298 * Figure out which chunk we are currently looking at 299 * and consider seeking to the next leaf. We use the 300 * low bits of f_seekp as a simple chunk index. 301 */ 302 fzap_next: 303 chunk = fp->f_seekp & (bsize - 1); 304 if (chunk == ZAP_LEAF_NUMCHUNKS(&zl)) { 305 fp->f_seekp = rounddown2(fp->f_seekp, bsize) + bsize; 306 chunk = 0; 307 308 /* 309 * Check for EOF and read the new leaf. 310 */ 311 if (fp->f_seekp >= bsize * fp->f_num_leafs) 312 return (ENOENT); 313 314 rc = dnode_read(spa, &fp->f_dnode, 315 fp->f_seekp, 316 fp->f_zap_leaf, 317 bsize); 318 if (rc) 319 return (rc); 320 } 321 322 zc = &ZAP_LEAF_CHUNK(&zl, chunk); 323 fp->f_seekp++; 324 if (zc->l_entry.le_type != ZAP_CHUNK_ENTRY) 325 goto fzap_next; 326 327 namelen = zc->l_entry.le_name_numints; 328 if (namelen > sizeof(d->d_name)) 329 namelen = sizeof(d->d_name); 330 331 /* 332 * Paste the name back together. 333 */ 334 nc = &ZAP_LEAF_CHUNK(&zl, zc->l_entry.le_name_chunk); 335 p = d->d_name; 336 while (namelen > 0) { 337 int len; 338 len = namelen; 339 if (len > ZAP_LEAF_ARRAY_BYTES) 340 len = ZAP_LEAF_ARRAY_BYTES; 341 memcpy(p, nc->l_array.la_array, len); 342 p += len; 343 namelen -= len; 344 nc = &ZAP_LEAF_CHUNK(&zl, nc->l_array.la_next); 345 } 346 d->d_name[sizeof(d->d_name) - 1] = 0; 347 348 /* 349 * Assume the first eight bytes of the value are 350 * a uint64_t. 351 */ 352 value = fzap_leaf_value(&zl, zc); 353 354 d->d_fileno = ZFS_DIRENT_OBJ(value); 355 d->d_type = ZFS_DIRENT_TYPE(value); 356 d->d_namlen = strlen(d->d_name); 357 358 return (0); 359 } 360} 361 362static int 363vdev_read(vdev_t *vdev, void *priv, off_t offset, void *buf, size_t bytes) 364{ 365 int fd, ret; 366 size_t res, head, tail, total_size, full_sec_size; 367 unsigned secsz, do_tail_read; 368 off_t start_sec; 369 char *outbuf, *bouncebuf; 370 371 fd = (uintptr_t) priv; 372 outbuf = (char *) buf; 373 bouncebuf = NULL; 374 375 ret = ioctl(fd, DIOCGSECTORSIZE, &secsz); 376 if (ret != 0) 377 return (ret); 378 379 /* 380 * Handling reads of arbitrary offset and size - multi-sector case 381 * and single-sector case. 382 * 383 * Multi-sector Case 384 * (do_tail_read = true if tail > 0) 385 * 386 * |<----------------------total_size--------------------->| 387 * | | 388 * |<--head-->|<--------------bytes------------>|<--tail-->| 389 * | | | | 390 * | | |<~full_sec_size~>| | | 391 * +------------------+ +------------------+ 392 * | |0101010| . . . |0101011| | 393 * +------------------+ +------------------+ 394 * start_sec start_sec + n 395 * 396 * 397 * Single-sector Case 398 * (do_tail_read = false) 399 * 400 * |<------total_size = secsz----->| 401 * | | 402 * |<-head->|<---bytes--->|<-tail->| 403 * +-------------------------------+ 404 * | |0101010101010| | 405 * +-------------------------------+ 406 * start_sec 407 */ 408 start_sec = offset / secsz; 409 head = offset % secsz; 410 total_size = roundup2(head + bytes, secsz); 411 tail = total_size - (head + bytes); 412 do_tail_read = ((tail > 0) && (head + bytes > secsz)); 413 full_sec_size = total_size; 414 if (head > 0) 415 full_sec_size -= secsz; 416 if (do_tail_read) 417 full_sec_size -= secsz; 418 419 /* Return of partial sector data requires a bounce buffer. */ 420 if ((head > 0) || do_tail_read) { 421 bouncebuf = zfs_alloc(secsz); 422 if (bouncebuf == NULL) { 423 printf("vdev_read: out of memory\n"); 424 return (ENOMEM); 425 } 426 } 427 428 if (lseek(fd, start_sec * secsz, SEEK_SET) == -1) 429 return (errno); 430 431 /* Partial data return from first sector */ 432 if (head > 0) { 433 res = read(fd, bouncebuf, secsz); 434 if (res != secsz) { 435 ret = EIO; 436 goto error; 437 } 438 memcpy(outbuf, bouncebuf + head, min(secsz - head, bytes)); 439 outbuf += min(secsz - head, bytes); 440 } 441 442 /* Full data return from read sectors */ 443 if (full_sec_size > 0) { 444 res = read(fd, outbuf, full_sec_size); 445 if (res != full_sec_size) { 446 ret = EIO; 447 goto error; 448 } 449 outbuf += full_sec_size; 450 } 451 452 /* Partial data return from last sector */ 453 if (do_tail_read) { 454 res = read(fd, bouncebuf, secsz); 455 if (res != secsz) { 456 ret = EIO; 457 goto error; 458 } 459 memcpy(outbuf, bouncebuf, secsz - tail); 460 } 461 462 ret = 0; 463error: 464 if (bouncebuf != NULL) 465 zfs_free(bouncebuf, secsz); 466 return (ret); 467} 468 469static int 470zfs_dev_init(void) 471{ 472 spa_t *spa; 473 spa_t *next; 474 spa_t *prev; 475 476 zfs_init(); 477 if (archsw.arch_zfs_probe == NULL) 478 return (ENXIO); 479 archsw.arch_zfs_probe(); 480 481 prev = NULL; 482 spa = STAILQ_FIRST(&zfs_pools); 483 while (spa != NULL) { 484 next = STAILQ_NEXT(spa, spa_link); 485 if (zfs_spa_init(spa)) { 486 if (prev == NULL) 487 STAILQ_REMOVE_HEAD(&zfs_pools, spa_link); 488 else 489 STAILQ_REMOVE_AFTER(&zfs_pools, prev, spa_link); 490 } else 491 prev = spa; 492 spa = next; 493 } 494 return (0); 495} 496 497struct zfs_probe_args { 498 int fd; 499 const char *devname; 500 uint64_t *pool_guid; 501 u_int secsz; 502}; 503 504static int 505zfs_diskread(void *arg, void *buf, size_t blocks, uint64_t offset) 506{ 507 struct zfs_probe_args *ppa; 508 509 ppa = (struct zfs_probe_args *)arg; 510 return (vdev_read(NULL, (void *)(uintptr_t)ppa->fd, 511 offset * ppa->secsz, buf, blocks * ppa->secsz)); 512} 513 514static int 515zfs_probe(int fd, uint64_t *pool_guid) 516{ 517 spa_t *spa; 518 int ret; 519 520 spa = NULL; 521 ret = vdev_probe(vdev_read, (void *)(uintptr_t)fd, &spa); 522 if (ret == 0 && pool_guid != NULL) 523 *pool_guid = spa->spa_guid; 524 return (ret); 525} 526 527static int 528zfs_probe_partition(void *arg, const char *partname, 529 const struct ptable_entry *part) 530{ 531 struct zfs_probe_args *ppa, pa; 532 struct ptable *table; 533 char devname[32]; 534 int ret; 535 536 /* Probe only freebsd-zfs and freebsd partitions */ 537 if (part->type != PART_FREEBSD && 538 part->type != PART_FREEBSD_ZFS) 539 return (0); 540 541 ppa = (struct zfs_probe_args *)arg; 542 strncpy(devname, ppa->devname, strlen(ppa->devname) - 1); 543 devname[strlen(ppa->devname) - 1] = '\0'; 544 sprintf(devname, "%s%s:", devname, partname); 545 pa.fd = open(devname, O_RDONLY); 546 if (pa.fd == -1) 547 return (0); 548 ret = zfs_probe(pa.fd, ppa->pool_guid); 549 if (ret == 0) 550 return (0); 551 /* Do we have BSD label here? */ 552 if (part->type == PART_FREEBSD) { 553 pa.devname = devname; 554 pa.pool_guid = ppa->pool_guid; 555 pa.secsz = ppa->secsz; 556 table = ptable_open(&pa, part->end - part->start + 1, 557 ppa->secsz, zfs_diskread); 558 if (table != NULL) { 559 ptable_iterate(table, &pa, zfs_probe_partition); 560 ptable_close(table); 561 } 562 } 563 close(pa.fd); 564 return (0); 565} 566 567int 568zfs_probe_dev(const char *devname, uint64_t *pool_guid) 569{ 570 struct disk_devdesc *dev; 571 struct ptable *table; 572 struct zfs_probe_args pa; 573 uint64_t mediasz; 574 int ret; 575 576 if (pool_guid) 577 *pool_guid = 0; 578 pa.fd = open(devname, O_RDONLY); 579 if (pa.fd == -1) 580 return (ENXIO); 581 /* Probe the whole disk */ 582 ret = zfs_probe(pa.fd, pool_guid); 583 if (ret == 0) 584 return (0); 585 if (archsw.arch_getdev((void **)&dev, devname, NULL) == 0) { 586 int partition = dev->d_partition; 587 int slice = dev->d_slice; 588 589 free(dev); 590 if (partition != -1 && slice != -1) { 591 ret = zfs_probe(pa.fd, pool_guid); 592 if (ret == 0) 593 return (0); 594 } 595 } 596 597 /* Probe each partition */ 598 ret = ioctl(pa.fd, DIOCGMEDIASIZE, &mediasz); 599 if (ret == 0) 600 ret = ioctl(pa.fd, DIOCGSECTORSIZE, &pa.secsz); 601 if (ret == 0) { 602 pa.devname = devname; 603 pa.pool_guid = pool_guid; 604 table = ptable_open(&pa, mediasz / pa.secsz, pa.secsz, 605 zfs_diskread); 606 if (table != NULL) { 607 ptable_iterate(table, &pa, zfs_probe_partition); 608 ptable_close(table); 609 } 610 } 611 close(pa.fd); 612 if (pool_guid && *pool_guid == 0) 613 ret = ENXIO; 614 return (ret); 615} 616 617/* 618 * Print information about ZFS pools 619 */ 620static int 621zfs_dev_print(int verbose) 622{ 623 spa_t *spa; 624 char line[80]; 625 int ret = 0; 626 627 if (STAILQ_EMPTY(&zfs_pools)) 628 return (0); 629 630 printf("%s devices:", zfs_dev.dv_name); 631 if ((ret = pager_output("\n")) != 0) 632 return (ret); 633 634 if (verbose) { 635 return (spa_all_status()); 636 } 637 STAILQ_FOREACH(spa, &zfs_pools, spa_link) { 638 snprintf(line, sizeof(line), " zfs:%s\n", spa->spa_name); 639 ret = pager_output(line); 640 if (ret != 0) 641 break; 642 } 643 return (ret); 644} 645 646/* 647 * Attempt to open the pool described by (dev) for use by (f). 648 */ 649static int 650zfs_dev_open(struct open_file *f, ...) 651{ 652 va_list args; 653 struct zfs_devdesc *dev; 654 struct zfsmount *mount; 655 spa_t *spa; 656 int rv; 657 658 va_start(args, f); 659 dev = va_arg(args, struct zfs_devdesc *); 660 va_end(args); 661 662 if (dev->pool_guid == 0) 663 spa = STAILQ_FIRST(&zfs_pools); 664 else 665 spa = spa_find_by_guid(dev->pool_guid); 666 if (!spa) 667 return (ENXIO); 668 mount = malloc(sizeof(*mount)); 669 rv = zfs_mount(spa, dev->root_guid, mount); 670 if (rv != 0) { 671 free(mount); 672 return (rv); 673 } 674 if (mount->objset.os_type != DMU_OST_ZFS) { 675 printf("Unexpected object set type %ju\n", 676 (uintmax_t)mount->objset.os_type); 677 free(mount); 678 return (EIO); 679 } 680 f->f_devdata = mount; 681 free(dev); 682 return (0); 683} 684 685static int 686zfs_dev_close(struct open_file *f) 687{ 688 689 free(f->f_devdata); 690 f->f_devdata = NULL; 691 return (0); 692} 693 694static int 695zfs_dev_strategy(void *devdata, int rw, daddr_t dblk, size_t size, char *buf, size_t *rsize) 696{ 697 698 return (ENOSYS); 699} 700 701struct devsw zfs_dev = { 702 .dv_name = "zfs", 703 .dv_type = DEVT_ZFS, 704 .dv_init = zfs_dev_init, 705 .dv_strategy = zfs_dev_strategy, 706 .dv_open = zfs_dev_open, 707 .dv_close = zfs_dev_close, 708 .dv_ioctl = noioctl, 709 .dv_print = zfs_dev_print, 710 .dv_cleanup = NULL 711}; 712 713int 714zfs_parsedev(struct zfs_devdesc *dev, const char *devspec, const char **path) 715{ 716 static char rootname[ZFS_MAXNAMELEN]; 717 static char poolname[ZFS_MAXNAMELEN]; 718 spa_t *spa; 719 const char *end; 720 const char *np; 721 const char *sep; 722 int rv; 723 724 np = devspec; 725 if (*np != ':') 726 return (EINVAL); 727 np++; 728 end = strrchr(np, ':'); 729 if (end == NULL) 730 return (EINVAL); 731 sep = strchr(np, '/'); 732 if (sep == NULL || sep >= end) 733 sep = end; 734 memcpy(poolname, np, sep - np); 735 poolname[sep - np] = '\0'; 736 if (sep < end) { 737 sep++; 738 memcpy(rootname, sep, end - sep); 739 rootname[end - sep] = '\0'; 740 } 741 else 742 rootname[0] = '\0'; 743 744 spa = spa_find_by_name(poolname); 745 if (!spa) 746 return (ENXIO); 747 dev->pool_guid = spa->spa_guid; 748 rv = zfs_lookup_dataset(spa, rootname, &dev->root_guid); 749 if (rv != 0) 750 return (rv); 751 if (path != NULL) 752 *path = (*end == '\0') ? end : end + 1; 753 dev->dd.d_dev = &zfs_dev; 754 return (0); 755} 756 757char * 758zfs_fmtdev(void *vdev) 759{ 760 static char rootname[ZFS_MAXNAMELEN]; 761 static char buf[2 * ZFS_MAXNAMELEN + 8]; 762 struct zfs_devdesc *dev = (struct zfs_devdesc *)vdev; 763 spa_t *spa; 764 765 buf[0] = '\0'; 766 if (dev->dd.d_dev->dv_type != DEVT_ZFS) 767 return (buf); 768 769 if (dev->pool_guid == 0) { 770 spa = STAILQ_FIRST(&zfs_pools); 771 dev->pool_guid = spa->spa_guid; 772 } else 773 spa = spa_find_by_guid(dev->pool_guid); 774 if (spa == NULL) { 775 printf("ZFS: can't find pool by guid\n"); 776 return (buf); 777 } 778 if (dev->root_guid == 0 && zfs_get_root(spa, &dev->root_guid)) { 779 printf("ZFS: can't find root filesystem\n"); 780 return (buf); 781 } 782 if (zfs_rlookup(spa, dev->root_guid, rootname)) { 783 printf("ZFS: can't find filesystem by guid\n"); 784 return (buf); 785 } 786 787 if (rootname[0] == '\0') 788 sprintf(buf, "%s:%s:", dev->dd.d_dev->dv_name, spa->spa_name); 789 else 790 sprintf(buf, "%s:%s/%s:", dev->dd.d_dev->dv_name, spa->spa_name, 791 rootname); 792 return (buf); 793} 794 795int 796zfs_list(const char *name) 797{ 798 static char poolname[ZFS_MAXNAMELEN]; 799 uint64_t objid; 800 spa_t *spa; 801 const char *dsname; 802 int len; 803 int rv; 804 805 len = strlen(name); 806 dsname = strchr(name, '/'); 807 if (dsname != NULL) { 808 len = dsname - name; 809 dsname++; 810 } else 811 dsname = ""; 812 memcpy(poolname, name, len); 813 poolname[len] = '\0'; 814 815 spa = spa_find_by_name(poolname); 816 if (!spa) 817 return (ENXIO); 818 rv = zfs_lookup_dataset(spa, dsname, &objid); 819 if (rv != 0) 820 return (rv); 821 822 return (zfs_list_dataset(spa, objid)); 823} 824 825void 826init_zfs_bootenv(const char *currdev_in) 827{ 828 char *beroot, *currdev; 829 int currdev_len; 830 831 currdev = NULL; 832 currdev_len = strlen(currdev_in); 833 if (currdev_len == 0) 834 return; 835 if (strncmp(currdev_in, "zfs:", 4) != 0) 836 return; 837 currdev = strdup(currdev_in); 838 if (currdev == NULL) 839 return; 840 /* Remove the trailing : */ 841 currdev[currdev_len - 1] = '\0'; 842 setenv("zfs_be_active", currdev, 1); 843 setenv("zfs_be_currpage", "1", 1); 844 /* Remove the last element (current bootenv) */ 845 beroot = strrchr(currdev, '/'); 846 if (beroot != NULL) 847 beroot[0] = '\0'; 848 beroot = strchr(currdev, ':') + 1; 849 setenv("zfs_be_root", beroot, 1); 850 zfs_bootenv_initial(beroot); 851 free(currdev); 852} 853 854static void 855zfs_bootenv_initial(const char *name) 856{ 857 char poolname[ZFS_MAXNAMELEN], *dsname; 858 char envname[32], envval[256]; 859 uint64_t objid; 860 spa_t *spa; 861 int bootenvs_idx, len, rv; 862 863 SLIST_INIT(&zfs_be_head); 864 zfs_env_count = 0; 865 len = strlen(name); 866 dsname = strchr(name, '/'); 867 if (dsname != NULL) { 868 len = dsname - name; 869 dsname++; 870 } else 871 dsname = ""; 872 strlcpy(poolname, name, len + 1); 873 spa = spa_find_by_name(poolname); 874 if (spa == NULL) 875 return; 876 rv = zfs_lookup_dataset(spa, dsname, &objid); 877 if (rv != 0) 878 return; 879 rv = zfs_callback_dataset(spa, objid, zfs_belist_add); 880 bootenvs_idx = 0; 881 /* Populate the initial environment variables */ 882 SLIST_FOREACH_SAFE(zfs_be, &zfs_be_head, entries, zfs_be_tmp) { 883 /* Enumerate all bootenvs for general usage */ 884 snprintf(envname, sizeof(envname), "bootenvs[%d]", bootenvs_idx); 885 snprintf(envval, sizeof(envval), "zfs:%s/%s", name, zfs_be->name); 886 rv = setenv(envname, envval, 1); 887 if (rv != 0) 888 break; 889 bootenvs_idx++; 890 } 891 snprintf(envval, sizeof(envval), "%d", bootenvs_idx); 892 setenv("bootenvs_count", envval, 1); 893 894 /* Clean up the SLIST of ZFS BEs */ 895 while (!SLIST_EMPTY(&zfs_be_head)) { 896 zfs_be = SLIST_FIRST(&zfs_be_head); 897 SLIST_REMOVE_HEAD(&zfs_be_head, entries); 898 free(zfs_be); 899 } 900 901 return; 902 903} 904 905int 906zfs_bootenv(const char *name) 907{ 908 static char poolname[ZFS_MAXNAMELEN], *dsname, *root; 909 char becount[4]; 910 uint64_t objid; 911 spa_t *spa; 912 int len, rv, pages, perpage, currpage; 913 914 if (name == NULL) 915 return (EINVAL); 916 if ((root = getenv("zfs_be_root")) == NULL) 917 return (EINVAL); 918 919 if (strcmp(name, root) != 0) { 920 if (setenv("zfs_be_root", name, 1) != 0) 921 return (ENOMEM); 922 } 923 924 SLIST_INIT(&zfs_be_head); 925 zfs_env_count = 0; 926 len = strlen(name); 927 dsname = strchr(name, '/'); 928 if (dsname != NULL) { 929 len = dsname - name; 930 dsname++; 931 } else 932 dsname = ""; 933 memcpy(poolname, name, len); 934 poolname[len] = '\0'; 935 936 spa = spa_find_by_name(poolname); 937 if (!spa) 938 return (ENXIO); 939 rv = zfs_lookup_dataset(spa, dsname, &objid); 940 if (rv != 0) 941 return (rv); 942 rv = zfs_callback_dataset(spa, objid, zfs_belist_add); 943 944 /* Calculate and store the number of pages of BEs */ 945 perpage = (ZFS_BE_LAST - ZFS_BE_FIRST + 1); 946 pages = (zfs_env_count / perpage) + ((zfs_env_count % perpage) > 0 ? 1 : 0); 947 snprintf(becount, 4, "%d", pages); 948 if (setenv("zfs_be_pages", becount, 1) != 0) 949 return (ENOMEM); 950 951 /* Roll over the page counter if it has exceeded the maximum */ 952 currpage = strtol(getenv("zfs_be_currpage"), NULL, 10); 953 if (currpage > pages) { 954 if (setenv("zfs_be_currpage", "1", 1) != 0) 955 return (ENOMEM); 956 } 957 958 /* Populate the menu environment variables */ 959 zfs_set_env(); 960 961 /* Clean up the SLIST of ZFS BEs */ 962 while (!SLIST_EMPTY(&zfs_be_head)) { 963 zfs_be = SLIST_FIRST(&zfs_be_head); 964 SLIST_REMOVE_HEAD(&zfs_be_head, entries); 965 free(zfs_be); 966 } 967 968 return (rv); 969} 970 971int 972zfs_belist_add(const char *name, uint64_t value __unused) 973{ 974 975 /* Skip special datasets that start with a $ character */ 976 if (strncmp(name, "$", 1) == 0) { 977 return (0); 978 } 979 /* Add the boot environment to the head of the SLIST */ 980 zfs_be = malloc(sizeof(struct zfs_be_entry)); 981 if (zfs_be == NULL) { 982 return (ENOMEM); 983 } 984 zfs_be->name = name; 985 SLIST_INSERT_HEAD(&zfs_be_head, zfs_be, entries); 986 zfs_env_count++; 987 988 return (0); 989} 990 991int 992zfs_set_env(void) 993{ 994 char envname[32], envval[256]; 995 char *beroot, *pagenum; 996 int rv, page, ctr; 997 998 beroot = getenv("zfs_be_root"); 999 if (beroot == NULL) { 1000 return (1); 1001 } 1002 1003 pagenum = getenv("zfs_be_currpage"); 1004 if (pagenum != NULL) { 1005 page = strtol(pagenum, NULL, 10); 1006 } else { 1007 page = 1; 1008 } 1009 1010 ctr = 1; 1011 rv = 0; 1012 zfs_env_index = ZFS_BE_FIRST; 1013 SLIST_FOREACH_SAFE(zfs_be, &zfs_be_head, entries, zfs_be_tmp) { 1014 /* Skip to the requested page number */ 1015 if (ctr <= ((ZFS_BE_LAST - ZFS_BE_FIRST + 1) * (page - 1))) { 1016 ctr++; 1017 continue; 1018 } 1019 1020 snprintf(envname, sizeof(envname), "bootenvmenu_caption[%d]", zfs_env_index); 1021 snprintf(envval, sizeof(envval), "%s", zfs_be->name); 1022 rv = setenv(envname, envval, 1); 1023 if (rv != 0) { 1024 break; 1025 } 1026 1027 snprintf(envname, sizeof(envname), "bootenvansi_caption[%d]", zfs_env_index); 1028 rv = setenv(envname, envval, 1); 1029 if (rv != 0){ 1030 break; 1031 } 1032 1033 snprintf(envname, sizeof(envname), "bootenvmenu_command[%d]", zfs_env_index); 1034 rv = setenv(envname, "set_bootenv", 1); 1035 if (rv != 0){ 1036 break; 1037 } 1038 1039 snprintf(envname, sizeof(envname), "bootenv_root[%d]", zfs_env_index); 1040 snprintf(envval, sizeof(envval), "zfs:%s/%s", beroot, zfs_be->name); 1041 rv = setenv(envname, envval, 1); 1042 if (rv != 0){ 1043 break; 1044 } 1045 1046 zfs_env_index++; 1047 if (zfs_env_index > ZFS_BE_LAST) { 1048 break; 1049 } 1050 1051 } 1052 1053 for (; zfs_env_index <= ZFS_BE_LAST; zfs_env_index++) { 1054 snprintf(envname, sizeof(envname), "bootenvmenu_caption[%d]", zfs_env_index); 1055 (void)unsetenv(envname); 1056 snprintf(envname, sizeof(envname), "bootenvansi_caption[%d]", zfs_env_index); 1057 (void)unsetenv(envname); 1058 snprintf(envname, sizeof(envname), "bootenvmenu_command[%d]", zfs_env_index); 1059 (void)unsetenv(envname); 1060 snprintf(envname, sizeof(envname), "bootenv_root[%d]", zfs_env_index); 1061 (void)unsetenv(envname); 1062 } 1063 1064 return (rv); 1065} 1066