1210650Sdfr/*- 2210650Sdfr * Copyright (c) 2010 Doug Rabson 3225609Savg * Copyright (c) 2011 Andriy Gapon 4226611Spjd * Copyright (c) 2011 Pawel Jakub Dawidek <pawel@dawidek.net> 5210650Sdfr * All rights reserved. 6210650Sdfr * 7210650Sdfr * Redistribution and use in source and binary forms, with or without 8210650Sdfr * modification, are permitted provided that the following conditions 9210650Sdfr * are met: 10210650Sdfr * 1. Redistributions of source code must retain the above copyright 11210650Sdfr * notice, this list of conditions and the following disclaimer. 12210650Sdfr * 2. Redistributions in binary form must reproduce the above copyright 13210650Sdfr * notice, this list of conditions and the following disclaimer in the 14210650Sdfr * documentation and/or other materials provided with the distribution. 15210650Sdfr * 16210650Sdfr * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 17210650Sdfr * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 18210650Sdfr * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 19210650Sdfr * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20210650Sdfr * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21210650Sdfr * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 22210650Sdfr * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23210650Sdfr * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24210650Sdfr * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25210650Sdfr * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26210650Sdfr * SUCH DAMAGE. 27210650Sdfr */ 28210650Sdfr/* $FreeBSD: stable/11/tools/tools/zfsboottest/zfsboottest.c 322215 2017-08-08 05:15:20Z ngie $ */ 29210650Sdfr 30210650Sdfr#include <sys/param.h> 31210650Sdfr#include <sys/queue.h> 32226611Spjd#include <err.h> 33225529Savg#include <errno.h> 34210650Sdfr#include <fcntl.h> 35226611Spjd#include <md5.h> 36210650Sdfr#include <stdint.h> 37210650Sdfr#include <stdio.h> 38210650Sdfr#include <string.h> 39210650Sdfr#include <stdarg.h> 40210650Sdfr#include <stddef.h> 41210650Sdfr#include <stdlib.h> 42225529Savg#include <unistd.h> 43210650Sdfr 44210650Sdfr#define NBBY 8 45210650Sdfr 46322215Sngieint 47210650Sdfrpager_output(const char *line) 48210650Sdfr{ 49226611Spjd 50225529Savg fprintf(stderr, "%s", line); 51322215Sngie return (0); 52210650Sdfr} 53210650Sdfr 54225609Savg#define ZFS_TEST 55225609Savg#define printf(...) fprintf(stderr, __VA_ARGS__) 56253067Savg#include "libzfs.h" 57210650Sdfr#include "zfsimpl.c" 58225609Savg#undef printf 59210650Sdfr 60210650Sdfrstatic int 61210650Sdfrvdev_read(vdev_t *vdev, void *priv, off_t off, void *buf, size_t bytes) 62210650Sdfr{ 63226611Spjd int fd = *(int *)priv; 64210650Sdfr 65210650Sdfr if (pread(fd, buf, bytes, off) != bytes) 66225529Savg return (-1); 67225529Savg return (0); 68210650Sdfr} 69210650Sdfr 70210650Sdfrstatic int 71210650Sdfrzfs_read(spa_t *spa, dnode_phys_t *dn, void *buf, size_t size, off_t off) 72210650Sdfr{ 73210650Sdfr const znode_phys_t *zp = (const znode_phys_t *) dn->dn_bonus; 74210650Sdfr size_t n; 75210650Sdfr int rc; 76210650Sdfr 77210650Sdfr n = size; 78210650Sdfr if (off + n > zp->zp_size) 79210650Sdfr n = zp->zp_size - off; 80225529Savg 81210650Sdfr rc = dnode_read(spa, dn, off, buf, n); 82226611Spjd if (rc != 0) 83225529Savg return (-rc); 84210650Sdfr 85210650Sdfr return (n); 86210650Sdfr} 87210650Sdfr 88210650Sdfrint 89210650Sdfrmain(int argc, char** argv) 90210650Sdfr{ 91226611Spjd char buf[512], hash[33]; 92226611Spjd MD5_CTX ctx; 93225529Savg struct stat sb; 94235392Savg struct zfsmount zfsmnt; 95225529Savg dnode_phys_t dn; 96235392Savg#if 0 97235392Savg uint64_t rootobj; 98235392Savg#endif 99225529Savg spa_t *spa; 100225609Savg off_t off; 101225609Savg ssize_t n; 102226611Spjd int i, failures, *fd; 103210650Sdfr 104210650Sdfr zfs_init(); 105210650Sdfr if (argc == 1) { 106210650Sdfr static char *av[] = { 107226611Spjd "zfsboottest", 108226611Spjd "/dev/gpt/system0", 109226611Spjd "/dev/gpt/system1", 110226611Spjd "-", 111226611Spjd "/boot/zfsloader", 112226611Spjd "/boot/support.4th", 113226611Spjd "/boot/kernel/kernel", 114210650Sdfr NULL, 115210650Sdfr }; 116226611Spjd argc = sizeof(av) / sizeof(av[0]) - 1; 117210650Sdfr argv = av; 118210650Sdfr } 119226611Spjd for (i = 1; i < argc; i++) { 120226611Spjd if (strcmp(argv[i], "-") == 0) 121226611Spjd break; 122226611Spjd } 123226611Spjd fd = malloc(sizeof(fd[0]) * (i - 1)); 124226611Spjd if (fd == NULL) 125226611Spjd errx(1, "Unable to allocate memory."); 126226611Spjd for (i = 1; i < argc; i++) { 127226611Spjd if (strcmp(argv[i], "-") == 0) 128226611Spjd break; 129226611Spjd fd[i - 1] = open(argv[i], O_RDONLY); 130226611Spjd if (fd[i - 1] == -1) { 131226611Spjd warn("open(%s) failed", argv[i]); 132210650Sdfr continue; 133226611Spjd } 134226611Spjd if (vdev_probe(vdev_read, &fd[i - 1], NULL) != 0) { 135226611Spjd warnx("vdev_probe(%s) failed", argv[i]); 136226611Spjd close(fd[i - 1]); 137226611Spjd } 138210650Sdfr } 139210650Sdfr 140290452Savg STAILQ_FOREACH(spa, &zfs_pools, spa_link) { 141290452Savg if (zfs_spa_init(spa)) { 142290452Savg fprintf(stderr, "can't init pool %s\n", spa->spa_name); 143290452Savg exit(1); 144290452Savg } 145290452Savg } 146290452Savg 147290452Savg spa_all_status(); 148290452Savg 149210650Sdfr spa = STAILQ_FIRST(&zfs_pools); 150225529Savg if (spa == NULL) { 151225529Savg fprintf(stderr, "no pools\n"); 152210650Sdfr exit(1); 153225529Savg } 154210650Sdfr 155235392Savg#if 0 156253067Savg uint64_t rootobj; 157235392Savg if (zfs_get_root(spa, &rootobj)) { 158235392Savg fprintf(stderr, "can't get root\n"); 159235392Savg exit(1); 160235392Savg } 161235392Savg 162235392Savg if (zfs_mount(spa, rootobj, &zfsmnt)) { 163235392Savg#else 164235392Savg if (zfs_mount(spa, 0, &zfsmnt)) { 165235392Savg fprintf(stderr, "can't mount\n"); 166235392Savg exit(1); 167253067Savg#endif 168235392Savg } 169235392Savg 170226611Spjd printf("\n"); 171226611Spjd for (++i, failures = 0; i < argc; i++) { 172235392Savg if (zfs_lookup(&zfsmnt, argv[i], &dn)) { 173226611Spjd fprintf(stderr, "%s: can't lookup\n", argv[i]); 174226611Spjd failures++; 175226611Spjd continue; 176226611Spjd } 177225529Savg 178226611Spjd if (zfs_dnode_stat(spa, &dn, &sb)) { 179226611Spjd fprintf(stderr, "%s: can't stat\n", argv[i]); 180226611Spjd failures++; 181226611Spjd continue; 182226611Spjd } 183226611Spjd 184226611Spjd off = 0; 185226611Spjd MD5Init(&ctx); 186226611Spjd do { 187226611Spjd n = sb.st_size - off; 188226611Spjd n = n > sizeof(buf) ? sizeof(buf) : n; 189226611Spjd n = zfs_read(spa, &dn, buf, n, off); 190226611Spjd if (n < 0) { 191226611Spjd fprintf(stderr, "%s: zfs_read failed\n", 192226611Spjd argv[i]); 193226611Spjd failures++; 194226611Spjd break; 195226611Spjd } 196226611Spjd MD5Update(&ctx, buf, n); 197226611Spjd off += n; 198226611Spjd } while (off < sb.st_size); 199226611Spjd if (off < sb.st_size) 200226611Spjd continue; 201226611Spjd MD5End(&ctx, hash); 202226611Spjd printf("%s %s\n", hash, argv[i]); 203225529Savg } 204225529Savg 205226611Spjd return (failures == 0 ? 0 : 1); 206210650Sdfr} 207