xref: /freebsd/stand/libsa/zfs/zfsimpl.c (revision 5fd34912b4a51f6ebe55d0d832fab4f3c60228e7)
1b8902de1SWarner Losh /*-
2b8902de1SWarner Losh  * Copyright (c) 2007 Doug Rabson
3b8902de1SWarner Losh  * All rights reserved.
4b8902de1SWarner Losh  *
5b8902de1SWarner Losh  * Redistribution and use in source and binary forms, with or without
6b8902de1SWarner Losh  * modification, are permitted provided that the following conditions
7b8902de1SWarner Losh  * are met:
8b8902de1SWarner Losh  * 1. Redistributions of source code must retain the above copyright
9b8902de1SWarner Losh  *    notice, this list of conditions and the following disclaimer.
10b8902de1SWarner Losh  * 2. Redistributions in binary form must reproduce the above copyright
11b8902de1SWarner Losh  *    notice, this list of conditions and the following disclaimer in the
12b8902de1SWarner Losh  *    documentation and/or other materials provided with the distribution.
13b8902de1SWarner Losh  *
14b8902de1SWarner Losh  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15b8902de1SWarner Losh  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16b8902de1SWarner Losh  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17b8902de1SWarner Losh  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18b8902de1SWarner Losh  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19b8902de1SWarner Losh  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20b8902de1SWarner Losh  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21b8902de1SWarner Losh  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22b8902de1SWarner Losh  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23b8902de1SWarner Losh  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24b8902de1SWarner Losh  * SUCH DAMAGE.
25b8902de1SWarner Losh  */
26b8902de1SWarner Losh 
27b8902de1SWarner Losh #include <sys/cdefs.h>
28b8902de1SWarner Losh __FBSDID("$FreeBSD$");
29b8902de1SWarner Losh 
30b8902de1SWarner Losh /*
31b8902de1SWarner Losh  *	Stand-alone ZFS file reader.
32b8902de1SWarner Losh  */
33b8902de1SWarner Losh 
34e307eb94SToomas Soome #include <stdbool.h>
3512d62cc2SXin LI #include <sys/endian.h>
36b8902de1SWarner Losh #include <sys/stat.h>
37b8902de1SWarner Losh #include <sys/stdint.h>
38b1b93268SToomas Soome #include <sys/list.h>
39e307eb94SToomas Soome #include <sys/zfs_bootenv.h>
403c2db0efSToomas Soome #include <machine/_inttypes.h>
41b8902de1SWarner Losh 
42b8902de1SWarner Losh #include "zfsimpl.h"
43b8902de1SWarner Losh #include "zfssubr.c"
44b8902de1SWarner Losh 
459257c69bSWarner Losh #ifdef HAS_ZSTD_ZFS
462fec3ae8SWarner Losh extern int zstd_init(void);
479257c69bSWarner Losh #endif
48b8902de1SWarner Losh 
49b8902de1SWarner Losh struct zfsmount {
50b4cb3fe0SToomas Soome 	char			*path;
51b8902de1SWarner Losh 	const spa_t		*spa;
52b8902de1SWarner Losh 	objset_phys_t		objset;
53b8902de1SWarner Losh 	uint64_t		rootobj;
54b4cb3fe0SToomas Soome 	STAILQ_ENTRY(zfsmount)	next;
55b8902de1SWarner Losh };
56b4cb3fe0SToomas Soome 
57b4cb3fe0SToomas Soome typedef STAILQ_HEAD(zfs_mnt_list, zfsmount) zfs_mnt_list_t;
58b4cb3fe0SToomas Soome static zfs_mnt_list_t zfsmount = STAILQ_HEAD_INITIALIZER(zfsmount);
59b8902de1SWarner Losh 
60b8902de1SWarner Losh /*
61b1b93268SToomas Soome  * The indirect_child_t represents the vdev that we will read from, when we
62b1b93268SToomas Soome  * need to read all copies of the data (e.g. for scrub or reconstruction).
63b1b93268SToomas Soome  * For plain (non-mirror) top-level vdevs (i.e. is_vdev is not a mirror),
64b1b93268SToomas Soome  * ic_vdev is the same as is_vdev.  However, for mirror top-level vdevs,
65b1b93268SToomas Soome  * ic_vdev is a child of the mirror.
66b1b93268SToomas Soome  */
67b1b93268SToomas Soome typedef struct indirect_child {
68b1b93268SToomas Soome 	void *ic_data;
69b1b93268SToomas Soome 	vdev_t *ic_vdev;
70b1b93268SToomas Soome } indirect_child_t;
71b1b93268SToomas Soome 
72b1b93268SToomas Soome /*
73b1b93268SToomas Soome  * The indirect_split_t represents one mapped segment of an i/o to the
74b1b93268SToomas Soome  * indirect vdev. For non-split (contiguously-mapped) blocks, there will be
75b1b93268SToomas Soome  * only one indirect_split_t, with is_split_offset==0 and is_size==io_size.
76b1b93268SToomas Soome  * For split blocks, there will be several of these.
77b1b93268SToomas Soome  */
78b1b93268SToomas Soome typedef struct indirect_split {
79b1b93268SToomas Soome 	list_node_t is_node; /* link on iv_splits */
80b1b93268SToomas Soome 
81b1b93268SToomas Soome 	/*
82b1b93268SToomas Soome 	 * is_split_offset is the offset into the i/o.
83b1b93268SToomas Soome 	 * This is the sum of the previous splits' is_size's.
84b1b93268SToomas Soome 	 */
85b1b93268SToomas Soome 	uint64_t is_split_offset;
86b1b93268SToomas Soome 
87b1b93268SToomas Soome 	vdev_t *is_vdev; /* top-level vdev */
88b1b93268SToomas Soome 	uint64_t is_target_offset; /* offset on is_vdev */
89b1b93268SToomas Soome 	uint64_t is_size;
90b1b93268SToomas Soome 	int is_children; /* number of entries in is_child[] */
91b1b93268SToomas Soome 
92b1b93268SToomas Soome 	/*
93b1b93268SToomas Soome 	 * is_good_child is the child that we are currently using to
94b1b93268SToomas Soome 	 * attempt reconstruction.
95b1b93268SToomas Soome 	 */
96b1b93268SToomas Soome 	int is_good_child;
97b1b93268SToomas Soome 
98b1b93268SToomas Soome 	indirect_child_t is_child[1]; /* variable-length */
99b1b93268SToomas Soome } indirect_split_t;
100b1b93268SToomas Soome 
101b1b93268SToomas Soome /*
102b1b93268SToomas Soome  * The indirect_vsd_t is associated with each i/o to the indirect vdev.
103b1b93268SToomas Soome  * It is the "Vdev-Specific Data" in the zio_t's io_vsd.
104b1b93268SToomas Soome  */
105b1b93268SToomas Soome typedef struct indirect_vsd {
106b1b93268SToomas Soome 	boolean_t iv_split_block;
107b1b93268SToomas Soome 	boolean_t iv_reconstruct;
108b1b93268SToomas Soome 
109b1b93268SToomas Soome 	list_t iv_splits; /* list of indirect_split_t's */
110b1b93268SToomas Soome } indirect_vsd_t;
111b1b93268SToomas Soome 
112b1b93268SToomas Soome /*
113b8902de1SWarner Losh  * List of all vdevs, chained through v_alllink.
114b8902de1SWarner Losh  */
115b8902de1SWarner Losh static vdev_list_t zfs_vdevs;
116b8902de1SWarner Losh 
117b8902de1SWarner Losh /*
118b8902de1SWarner Losh  * List of ZFS features supported for read
119b8902de1SWarner Losh  */
120b8902de1SWarner Losh static const char *features_for_read[] = {
121f993fff6SAlexander Leidinger 	"com.datto:bookmark_v2",
122f993fff6SAlexander Leidinger 	"com.datto:encryption",
123f993fff6SAlexander Leidinger 	"com.datto:resilver_defer",
124f993fff6SAlexander Leidinger 	"com.delphix:bookmark_written",
125f993fff6SAlexander Leidinger 	"com.delphix:device_removal",
126b8902de1SWarner Losh 	"com.delphix:embedded_data",
127f993fff6SAlexander Leidinger 	"com.delphix:extensible_dataset",
128f993fff6SAlexander Leidinger 	"com.delphix:head_errlog",
129f993fff6SAlexander Leidinger 	"com.delphix:hole_birth",
130f993fff6SAlexander Leidinger 	"com.delphix:obsolete_counts",
131f993fff6SAlexander Leidinger 	"com.delphix:spacemap_histogram",
132f993fff6SAlexander Leidinger 	"com.delphix:spacemap_v2",
133f993fff6SAlexander Leidinger 	"com.delphix:zpool_checkpoint",
134f993fff6SAlexander Leidinger 	"com.intel:allocation_classes",
135f993fff6SAlexander Leidinger 	"com.joyent:multi_vdev_crash_dump",
136*5fd34912SWarner Losh 	"com.klarasystems:vdev_zaps_v2",
137f993fff6SAlexander Leidinger 	"org.freebsd:zstd_compress",
138f993fff6SAlexander Leidinger 	"org.illumos:lz4_compress",
139b8902de1SWarner Losh 	"org.illumos:sha512",
140b8902de1SWarner Losh 	"org.illumos:skein",
141f993fff6SAlexander Leidinger 	"org.open-zfs:large_blocks",
14275ad2477SWarner Losh 	"org.openzfs:blake3",
143f993fff6SAlexander Leidinger 	"org.zfsonlinux:allocation_classes",
144f993fff6SAlexander Leidinger 	"org.zfsonlinux:large_dnode",
145b8902de1SWarner Losh 	NULL
146b8902de1SWarner Losh };
147b8902de1SWarner Losh 
148b8902de1SWarner Losh /*
149b8902de1SWarner Losh  * List of all pools, chained through spa_link.
150b8902de1SWarner Losh  */
151b8902de1SWarner Losh static spa_list_t zfs_pools;
152b8902de1SWarner Losh 
153b8902de1SWarner Losh static const dnode_phys_t *dnode_cache_obj;
154b8902de1SWarner Losh static uint64_t dnode_cache_bn;
155b8902de1SWarner Losh static char *dnode_cache_buf;
156b8902de1SWarner Losh 
157b8902de1SWarner Losh static int zio_read(const spa_t *spa, const blkptr_t *bp, void *buf);
158b8902de1SWarner Losh static int zfs_get_root(const spa_t *spa, uint64_t *objid);
159b8902de1SWarner Losh static int zfs_rlookup(const spa_t *spa, uint64_t objnum, char *result);
160b8902de1SWarner Losh static int zap_lookup(const spa_t *spa, const dnode_phys_t *dnode,
161b8902de1SWarner Losh     const char *name, uint64_t integer_size, uint64_t num_integers,
162b8902de1SWarner Losh     void *value);
163b1b93268SToomas Soome static int objset_get_dnode(const spa_t *, const objset_phys_t *, uint64_t,
164b1b93268SToomas Soome     dnode_phys_t *);
165b1b93268SToomas Soome static int dnode_read(const spa_t *, const dnode_phys_t *, off_t, void *,
166b1b93268SToomas Soome     size_t);
167b1b93268SToomas Soome static int vdev_indirect_read(vdev_t *, const blkptr_t *, void *, off_t,
168b1b93268SToomas Soome     size_t);
169b1b93268SToomas Soome static int vdev_mirror_read(vdev_t *, const blkptr_t *, void *, off_t, size_t);
170b1b93268SToomas Soome vdev_indirect_mapping_t *vdev_indirect_mapping_open(spa_t *, objset_phys_t *,
171b1b93268SToomas Soome     uint64_t);
172b1b93268SToomas Soome vdev_indirect_mapping_entry_phys_t *
173b1b93268SToomas Soome     vdev_indirect_mapping_duplicate_adjacent_entries(vdev_t *, uint64_t,
174b1b93268SToomas Soome     uint64_t, uint64_t *);
175b8902de1SWarner Losh 
176b8902de1SWarner Losh static void
177b8902de1SWarner Losh zfs_init(void)
178b8902de1SWarner Losh {
179b8902de1SWarner Losh 	STAILQ_INIT(&zfs_vdevs);
180b8902de1SWarner Losh 	STAILQ_INIT(&zfs_pools);
181b8902de1SWarner Losh 
182b8902de1SWarner Losh 	dnode_cache_buf = malloc(SPA_MAXBLOCKSIZE);
183b8902de1SWarner Losh 
184b8902de1SWarner Losh 	zfs_init_crc();
1859257c69bSWarner Losh #ifdef HAS_ZSTD_ZFS
1862fec3ae8SWarner Losh 	zstd_init();
1879257c69bSWarner Losh #endif
188b8902de1SWarner Losh }
189b8902de1SWarner Losh 
190b8902de1SWarner Losh static int
1913830659eSToomas Soome nvlist_check_features_for_read(nvlist_t *nvl)
192b8902de1SWarner Losh {
1933830659eSToomas Soome 	nvlist_t *features = NULL;
1943830659eSToomas Soome 	nvs_data_t *data;
1953830659eSToomas Soome 	nvp_header_t *nvp;
1963830659eSToomas Soome 	nv_string_t *nvp_name;
197b8902de1SWarner Losh 	int rc;
198b8902de1SWarner Losh 
1993830659eSToomas Soome 	rc = nvlist_find(nvl, ZPOOL_CONFIG_FEATURES_FOR_READ,
2003830659eSToomas Soome 	    DATA_TYPE_NVLIST, NULL, &features, NULL);
201d36341f7SToomas Soome 	switch (rc) {
202d36341f7SToomas Soome 	case 0:
203d36341f7SToomas Soome 		break;		/* Continue with checks */
204d36341f7SToomas Soome 
205d36341f7SToomas Soome 	case ENOENT:
206d36341f7SToomas Soome 		return (0);	/* All features are disabled */
207d36341f7SToomas Soome 
208d36341f7SToomas Soome 	default:
209d36341f7SToomas Soome 		return (rc);	/* Error while reading nvlist */
210d36341f7SToomas Soome 	}
211b8902de1SWarner Losh 
2123830659eSToomas Soome 	data = (nvs_data_t *)features->nv_data;
2133830659eSToomas Soome 	nvp = &data->nvl_pair;	/* first pair in nvlist */
214b8902de1SWarner Losh 
2153830659eSToomas Soome 	while (nvp->encoded_size != 0 && nvp->decoded_size != 0) {
216b8902de1SWarner Losh 		int i, found;
217b8902de1SWarner Losh 
2183830659eSToomas Soome 		nvp_name = (nv_string_t *)((uintptr_t)nvp + sizeof(*nvp));
219b8902de1SWarner Losh 		found = 0;
220b8902de1SWarner Losh 
221b8902de1SWarner Losh 		for (i = 0; features_for_read[i] != NULL; i++) {
2223830659eSToomas Soome 			if (memcmp(nvp_name->nv_data, features_for_read[i],
2233830659eSToomas Soome 			    nvp_name->nv_size) == 0) {
224b8902de1SWarner Losh 				found = 1;
225b8902de1SWarner Losh 				break;
226b8902de1SWarner Losh 			}
227b8902de1SWarner Losh 		}
228b8902de1SWarner Losh 
229b8902de1SWarner Losh 		if (!found) {
2303830659eSToomas Soome 			printf("ZFS: unsupported feature: %.*s\n",
2313830659eSToomas Soome 			    nvp_name->nv_size, nvp_name->nv_data);
232b8902de1SWarner Losh 			rc = EIO;
233b8902de1SWarner Losh 		}
2343830659eSToomas Soome 		nvp = (nvp_header_t *)((uint8_t *)nvp + nvp->encoded_size);
235b8902de1SWarner Losh 	}
2363830659eSToomas Soome 	nvlist_destroy(features);
237b8902de1SWarner Losh 
238b8902de1SWarner Losh 	return (rc);
239b8902de1SWarner Losh }
240b8902de1SWarner Losh 
241b8902de1SWarner Losh static int
242b8902de1SWarner Losh vdev_read_phys(vdev_t *vdev, const blkptr_t *bp, void *buf,
243b8902de1SWarner Losh     off_t offset, size_t size)
244b8902de1SWarner Losh {
245b8902de1SWarner Losh 	size_t psize;
246b8902de1SWarner Losh 	int rc;
247b8902de1SWarner Losh 
248e307eb94SToomas Soome 	if (vdev->v_phys_read == NULL)
249e307eb94SToomas Soome 		return (ENOTSUP);
250b8902de1SWarner Losh 
251b8902de1SWarner Losh 	if (bp) {
252b8902de1SWarner Losh 		psize = BP_GET_PSIZE(bp);
253b8902de1SWarner Losh 	} else {
254b8902de1SWarner Losh 		psize = size;
255b8902de1SWarner Losh 	}
256b8902de1SWarner Losh 
257e307eb94SToomas Soome 	rc = vdev->v_phys_read(vdev, vdev->v_priv, offset, buf, psize);
2583c2db0efSToomas Soome 	if (rc == 0) {
259b1b93268SToomas Soome 		if (bp != NULL)
2603c2db0efSToomas Soome 			rc = zio_checksum_verify(vdev->v_spa, bp, buf);
2613c2db0efSToomas Soome 	}
262b8902de1SWarner Losh 
2633c2db0efSToomas Soome 	return (rc);
264b8902de1SWarner Losh }
265b8902de1SWarner Losh 
266e307eb94SToomas Soome static int
267e307eb94SToomas Soome vdev_write_phys(vdev_t *vdev, void *buf, off_t offset, size_t size)
268e307eb94SToomas Soome {
269e307eb94SToomas Soome 	if (vdev->v_phys_write == NULL)
270e307eb94SToomas Soome 		return (ENOTSUP);
271e307eb94SToomas Soome 
272e307eb94SToomas Soome 	return (vdev->v_phys_write(vdev, offset, buf, size));
273e307eb94SToomas Soome }
274e307eb94SToomas Soome 
275b1b93268SToomas Soome typedef struct remap_segment {
276b1b93268SToomas Soome 	vdev_t *rs_vd;
277b1b93268SToomas Soome 	uint64_t rs_offset;
278b1b93268SToomas Soome 	uint64_t rs_asize;
279b1b93268SToomas Soome 	uint64_t rs_split_offset;
280b1b93268SToomas Soome 	list_node_t rs_node;
281b1b93268SToomas Soome } remap_segment_t;
282b1b93268SToomas Soome 
283b1b93268SToomas Soome static remap_segment_t *
284b1b93268SToomas Soome rs_alloc(vdev_t *vd, uint64_t offset, uint64_t asize, uint64_t split_offset)
285b1b93268SToomas Soome {
286b1b93268SToomas Soome 	remap_segment_t *rs = malloc(sizeof (remap_segment_t));
287b1b93268SToomas Soome 
288b1b93268SToomas Soome 	if (rs != NULL) {
289b1b93268SToomas Soome 		rs->rs_vd = vd;
290b1b93268SToomas Soome 		rs->rs_offset = offset;
291b1b93268SToomas Soome 		rs->rs_asize = asize;
292b1b93268SToomas Soome 		rs->rs_split_offset = split_offset;
293b1b93268SToomas Soome 	}
294b1b93268SToomas Soome 
295b1b93268SToomas Soome 	return (rs);
296b1b93268SToomas Soome }
297b1b93268SToomas Soome 
298b1b93268SToomas Soome vdev_indirect_mapping_t *
299b1b93268SToomas Soome vdev_indirect_mapping_open(spa_t *spa, objset_phys_t *os,
300b1b93268SToomas Soome     uint64_t mapping_object)
301b1b93268SToomas Soome {
302b1b93268SToomas Soome 	vdev_indirect_mapping_t *vim;
303b1b93268SToomas Soome 	vdev_indirect_mapping_phys_t *vim_phys;
304b1b93268SToomas Soome 	int rc;
305b1b93268SToomas Soome 
306b1b93268SToomas Soome 	vim = calloc(1, sizeof (*vim));
307b1b93268SToomas Soome 	if (vim == NULL)
308b1b93268SToomas Soome 		return (NULL);
309b1b93268SToomas Soome 
310b1b93268SToomas Soome 	vim->vim_dn = calloc(1, sizeof (*vim->vim_dn));
311b1b93268SToomas Soome 	if (vim->vim_dn == NULL) {
312b1b93268SToomas Soome 		free(vim);
313b1b93268SToomas Soome 		return (NULL);
314b1b93268SToomas Soome 	}
315b1b93268SToomas Soome 
316b1b93268SToomas Soome 	rc = objset_get_dnode(spa, os, mapping_object, vim->vim_dn);
317b1b93268SToomas Soome 	if (rc != 0) {
318b1b93268SToomas Soome 		free(vim->vim_dn);
319b1b93268SToomas Soome 		free(vim);
320b1b93268SToomas Soome 		return (NULL);
321b1b93268SToomas Soome 	}
322b1b93268SToomas Soome 
323b1b93268SToomas Soome 	vim->vim_spa = spa;
324b1b93268SToomas Soome 	vim->vim_phys = malloc(sizeof (*vim->vim_phys));
325b1b93268SToomas Soome 	if (vim->vim_phys == NULL) {
326b1b93268SToomas Soome 		free(vim->vim_dn);
327b1b93268SToomas Soome 		free(vim);
328b1b93268SToomas Soome 		return (NULL);
329b1b93268SToomas Soome 	}
330b1b93268SToomas Soome 
331b1b93268SToomas Soome 	vim_phys = (vdev_indirect_mapping_phys_t *)DN_BONUS(vim->vim_dn);
332b1b93268SToomas Soome 	*vim->vim_phys = *vim_phys;
333b1b93268SToomas Soome 
334b1b93268SToomas Soome 	vim->vim_objset = os;
335b1b93268SToomas Soome 	vim->vim_object = mapping_object;
336b1b93268SToomas Soome 	vim->vim_entries = NULL;
337b1b93268SToomas Soome 
338b1b93268SToomas Soome 	vim->vim_havecounts =
339b1b93268SToomas Soome 	    (vim->vim_dn->dn_bonuslen > VDEV_INDIRECT_MAPPING_SIZE_V0);
3403c2db0efSToomas Soome 
341b1b93268SToomas Soome 	return (vim);
342b1b93268SToomas Soome }
343b1b93268SToomas Soome 
344b1b93268SToomas Soome /*
345b1b93268SToomas Soome  * Compare an offset with an indirect mapping entry; there are three
346b1b93268SToomas Soome  * possible scenarios:
347b1b93268SToomas Soome  *
348b1b93268SToomas Soome  *     1. The offset is "less than" the mapping entry; meaning the
349b1b93268SToomas Soome  *        offset is less than the source offset of the mapping entry. In
350b1b93268SToomas Soome  *        this case, there is no overlap between the offset and the
351b1b93268SToomas Soome  *        mapping entry and -1 will be returned.
352b1b93268SToomas Soome  *
353b1b93268SToomas Soome  *     2. The offset is "greater than" the mapping entry; meaning the
354b1b93268SToomas Soome  *        offset is greater than the mapping entry's source offset plus
355b1b93268SToomas Soome  *        the entry's size. In this case, there is no overlap between
356b1b93268SToomas Soome  *        the offset and the mapping entry and 1 will be returned.
357b1b93268SToomas Soome  *
358b1b93268SToomas Soome  *        NOTE: If the offset is actually equal to the entry's offset
359b1b93268SToomas Soome  *        plus size, this is considered to be "greater" than the entry,
360b1b93268SToomas Soome  *        and this case applies (i.e. 1 will be returned). Thus, the
361b1b93268SToomas Soome  *        entry's "range" can be considered to be inclusive at its
362b1b93268SToomas Soome  *        start, but exclusive at its end: e.g. [src, src + size).
363b1b93268SToomas Soome  *
364b1b93268SToomas Soome  *     3. The last case to consider is if the offset actually falls
365b1b93268SToomas Soome  *        within the mapping entry's range. If this is the case, the
366b1b93268SToomas Soome  *        offset is considered to be "equal to" the mapping entry and
367b1b93268SToomas Soome  *        0 will be returned.
368b1b93268SToomas Soome  *
369b1b93268SToomas Soome  *        NOTE: If the offset is equal to the entry's source offset,
370b1b93268SToomas Soome  *        this case applies and 0 will be returned. If the offset is
371b1b93268SToomas Soome  *        equal to the entry's source plus its size, this case does
372b1b93268SToomas Soome  *        *not* apply (see "NOTE" above for scenario 2), and 1 will be
373b1b93268SToomas Soome  *        returned.
374b1b93268SToomas Soome  */
375b1b93268SToomas Soome static int
376b1b93268SToomas Soome dva_mapping_overlap_compare(const void *v_key, const void *v_array_elem)
377b1b93268SToomas Soome {
378b1b93268SToomas Soome 	const uint64_t *key = v_key;
379b1b93268SToomas Soome 	const vdev_indirect_mapping_entry_phys_t *array_elem =
380b1b93268SToomas Soome 	    v_array_elem;
381b1b93268SToomas Soome 	uint64_t src_offset = DVA_MAPPING_GET_SRC_OFFSET(array_elem);
382b1b93268SToomas Soome 
383b1b93268SToomas Soome 	if (*key < src_offset) {
384b1b93268SToomas Soome 		return (-1);
385b1b93268SToomas Soome 	} else if (*key < src_offset + DVA_GET_ASIZE(&array_elem->vimep_dst)) {
386b1b93268SToomas Soome 		return (0);
387b1b93268SToomas Soome 	} else {
388b1b93268SToomas Soome 		return (1);
389b1b93268SToomas Soome 	}
390b1b93268SToomas Soome }
391b1b93268SToomas Soome 
392b1b93268SToomas Soome /*
393b1b93268SToomas Soome  * Return array entry.
394b1b93268SToomas Soome  */
395b1b93268SToomas Soome static vdev_indirect_mapping_entry_phys_t *
396b1b93268SToomas Soome vdev_indirect_mapping_entry(vdev_indirect_mapping_t *vim, uint64_t index)
397b1b93268SToomas Soome {
398b1b93268SToomas Soome 	uint64_t size;
399b1b93268SToomas Soome 	off_t offset = 0;
400b1b93268SToomas Soome 	int rc;
401b1b93268SToomas Soome 
402b1b93268SToomas Soome 	if (vim->vim_phys->vimp_num_entries == 0)
403b1b93268SToomas Soome 		return (NULL);
404b1b93268SToomas Soome 
405b1b93268SToomas Soome 	if (vim->vim_entries == NULL) {
406b1b93268SToomas Soome 		uint64_t bsize;
407b1b93268SToomas Soome 
408b1b93268SToomas Soome 		bsize = vim->vim_dn->dn_datablkszsec << SPA_MINBLOCKSHIFT;
409b1b93268SToomas Soome 		size = vim->vim_phys->vimp_num_entries *
410b1b93268SToomas Soome 		    sizeof (*vim->vim_entries);
411b1b93268SToomas Soome 		if (size > bsize) {
412b1b93268SToomas Soome 			size = bsize / sizeof (*vim->vim_entries);
413b1b93268SToomas Soome 			size *= sizeof (*vim->vim_entries);
414b1b93268SToomas Soome 		}
415b1b93268SToomas Soome 		vim->vim_entries = malloc(size);
416b1b93268SToomas Soome 		if (vim->vim_entries == NULL)
417b1b93268SToomas Soome 			return (NULL);
418b1b93268SToomas Soome 		vim->vim_num_entries = size / sizeof (*vim->vim_entries);
419b1b93268SToomas Soome 		offset = index * sizeof (*vim->vim_entries);
420b1b93268SToomas Soome 	}
421b1b93268SToomas Soome 
422b1b93268SToomas Soome 	/* We have data in vim_entries */
423b1b93268SToomas Soome 	if (offset == 0) {
424b1b93268SToomas Soome 		if (index >= vim->vim_entry_offset &&
425b1b93268SToomas Soome 		    index <= vim->vim_entry_offset + vim->vim_num_entries) {
426b1b93268SToomas Soome 			index -= vim->vim_entry_offset;
427b1b93268SToomas Soome 			return (&vim->vim_entries[index]);
428b1b93268SToomas Soome 		}
429b1b93268SToomas Soome 		offset = index * sizeof (*vim->vim_entries);
430b1b93268SToomas Soome 	}
431b1b93268SToomas Soome 
432b1b93268SToomas Soome 	vim->vim_entry_offset = index;
433b1b93268SToomas Soome 	size = vim->vim_num_entries * sizeof (*vim->vim_entries);
434b1b93268SToomas Soome 	rc = dnode_read(vim->vim_spa, vim->vim_dn, offset, vim->vim_entries,
435b1b93268SToomas Soome 	    size);
436b1b93268SToomas Soome 	if (rc != 0) {
437b1b93268SToomas Soome 		/* Read error, invalidate vim_entries. */
438b1b93268SToomas Soome 		free(vim->vim_entries);
439b1b93268SToomas Soome 		vim->vim_entries = NULL;
440b1b93268SToomas Soome 		return (NULL);
441b1b93268SToomas Soome 	}
442b1b93268SToomas Soome 	index -= vim->vim_entry_offset;
443b1b93268SToomas Soome 	return (&vim->vim_entries[index]);
444b1b93268SToomas Soome }
445b1b93268SToomas Soome 
446b1b93268SToomas Soome /*
447b1b93268SToomas Soome  * Returns the mapping entry for the given offset.
448b1b93268SToomas Soome  *
449b1b93268SToomas Soome  * It's possible that the given offset will not be in the mapping table
450b1b93268SToomas Soome  * (i.e. no mapping entries contain this offset), in which case, the
45166095010SGordon Bergling  * return value depends on the "next_if_missing" parameter.
452b1b93268SToomas Soome  *
453b1b93268SToomas Soome  * If the offset is not found in the table and "next_if_missing" is
454b1b93268SToomas Soome  * B_FALSE, then NULL will always be returned. The behavior is intended
455b1b93268SToomas Soome  * to allow consumers to get the entry corresponding to the offset
456b1b93268SToomas Soome  * parameter, iff the offset overlaps with an entry in the table.
457b1b93268SToomas Soome  *
458b1b93268SToomas Soome  * If the offset is not found in the table and "next_if_missing" is
459b1b93268SToomas Soome  * B_TRUE, then the entry nearest to the given offset will be returned,
460b1b93268SToomas Soome  * such that the entry's source offset is greater than the offset
461b1b93268SToomas Soome  * passed in (i.e. the "next" mapping entry in the table is returned, if
462b1b93268SToomas Soome  * the offset is missing from the table). If there are no entries whose
463b1b93268SToomas Soome  * source offset is greater than the passed in offset, NULL is returned.
464b1b93268SToomas Soome  */
465b1b93268SToomas Soome static vdev_indirect_mapping_entry_phys_t *
466b1b93268SToomas Soome vdev_indirect_mapping_entry_for_offset(vdev_indirect_mapping_t *vim,
467b1b93268SToomas Soome     uint64_t offset)
468b1b93268SToomas Soome {
469b1b93268SToomas Soome 	ASSERT(vim->vim_phys->vimp_num_entries > 0);
470b1b93268SToomas Soome 
471b1b93268SToomas Soome 	vdev_indirect_mapping_entry_phys_t *entry;
472b1b93268SToomas Soome 
473b1b93268SToomas Soome 	uint64_t last = vim->vim_phys->vimp_num_entries - 1;
474b1b93268SToomas Soome 	uint64_t base = 0;
475b1b93268SToomas Soome 
476b1b93268SToomas Soome 	/*
477b1b93268SToomas Soome 	 * We don't define these inside of the while loop because we use
478b1b93268SToomas Soome 	 * their value in the case that offset isn't in the mapping.
479b1b93268SToomas Soome 	 */
480b1b93268SToomas Soome 	uint64_t mid;
481b1b93268SToomas Soome 	int result;
482b1b93268SToomas Soome 
483b1b93268SToomas Soome 	while (last >= base) {
484b1b93268SToomas Soome 		mid = base + ((last - base) >> 1);
485b1b93268SToomas Soome 
486b1b93268SToomas Soome 		entry = vdev_indirect_mapping_entry(vim, mid);
487b1b93268SToomas Soome 		if (entry == NULL)
488b1b93268SToomas Soome 			break;
489b1b93268SToomas Soome 		result = dva_mapping_overlap_compare(&offset, entry);
490b1b93268SToomas Soome 
491b1b93268SToomas Soome 		if (result == 0) {
492b1b93268SToomas Soome 			break;
493b1b93268SToomas Soome 		} else if (result < 0) {
494b1b93268SToomas Soome 			last = mid - 1;
495b1b93268SToomas Soome 		} else {
496b1b93268SToomas Soome 			base = mid + 1;
497b1b93268SToomas Soome 		}
498b1b93268SToomas Soome 	}
499b1b93268SToomas Soome 	return (entry);
500b1b93268SToomas Soome }
501b1b93268SToomas Soome 
502b1b93268SToomas Soome /*
503b1b93268SToomas Soome  * Given an indirect vdev and an extent on that vdev, it duplicates the
504b1b93268SToomas Soome  * physical entries of the indirect mapping that correspond to the extent
505b1b93268SToomas Soome  * to a new array and returns a pointer to it. In addition, copied_entries
506b1b93268SToomas Soome  * is populated with the number of mapping entries that were duplicated.
507b1b93268SToomas Soome  *
508b1b93268SToomas Soome  * Finally, since we are doing an allocation, it is up to the caller to
509b1b93268SToomas Soome  * free the array allocated in this function.
510b1b93268SToomas Soome  */
511b1b93268SToomas Soome vdev_indirect_mapping_entry_phys_t *
512b1b93268SToomas Soome vdev_indirect_mapping_duplicate_adjacent_entries(vdev_t *vd, uint64_t offset,
513b1b93268SToomas Soome     uint64_t asize, uint64_t *copied_entries)
514b1b93268SToomas Soome {
515b1b93268SToomas Soome 	vdev_indirect_mapping_entry_phys_t *duplicate_mappings = NULL;
516b1b93268SToomas Soome 	vdev_indirect_mapping_t *vim = vd->v_mapping;
517b1b93268SToomas Soome 	uint64_t entries = 0;
518b1b93268SToomas Soome 
519b1b93268SToomas Soome 	vdev_indirect_mapping_entry_phys_t *first_mapping =
520b1b93268SToomas Soome 	    vdev_indirect_mapping_entry_for_offset(vim, offset);
521b1b93268SToomas Soome 	ASSERT3P(first_mapping, !=, NULL);
522b1b93268SToomas Soome 
523b1b93268SToomas Soome 	vdev_indirect_mapping_entry_phys_t *m = first_mapping;
524b1b93268SToomas Soome 	while (asize > 0) {
525b1b93268SToomas Soome 		uint64_t size = DVA_GET_ASIZE(&m->vimep_dst);
526b1b93268SToomas Soome 		uint64_t inner_offset = offset - DVA_MAPPING_GET_SRC_OFFSET(m);
527b1b93268SToomas Soome 		uint64_t inner_size = MIN(asize, size - inner_offset);
528b1b93268SToomas Soome 
529b1b93268SToomas Soome 		offset += inner_size;
530b1b93268SToomas Soome 		asize -= inner_size;
531b1b93268SToomas Soome 		entries++;
532b1b93268SToomas Soome 		m++;
533b1b93268SToomas Soome 	}
534b1b93268SToomas Soome 
535b1b93268SToomas Soome 	size_t copy_length = entries * sizeof (*first_mapping);
536b1b93268SToomas Soome 	duplicate_mappings = malloc(copy_length);
537b1b93268SToomas Soome 	if (duplicate_mappings != NULL)
538b1b93268SToomas Soome 		bcopy(first_mapping, duplicate_mappings, copy_length);
539b1b93268SToomas Soome 	else
540b1b93268SToomas Soome 		entries = 0;
541b1b93268SToomas Soome 
542b1b93268SToomas Soome 	*copied_entries = entries;
543b1b93268SToomas Soome 
544b1b93268SToomas Soome 	return (duplicate_mappings);
545b1b93268SToomas Soome }
546b1b93268SToomas Soome 
547b1b93268SToomas Soome static vdev_t *
548b1b93268SToomas Soome vdev_lookup_top(spa_t *spa, uint64_t vdev)
549b1b93268SToomas Soome {
550b1b93268SToomas Soome 	vdev_t *rvd;
5513c2db0efSToomas Soome 	vdev_list_t *vlist;
552b1b93268SToomas Soome 
5533c2db0efSToomas Soome 	vlist = &spa->spa_root_vdev->v_children;
5543c2db0efSToomas Soome 	STAILQ_FOREACH(rvd, vlist, v_childlink)
555b1b93268SToomas Soome 		if (rvd->v_id == vdev)
556b1b93268SToomas Soome 			break;
557b1b93268SToomas Soome 
558b1b93268SToomas Soome 	return (rvd);
559b1b93268SToomas Soome }
560b1b93268SToomas Soome 
561b1b93268SToomas Soome /*
562b1b93268SToomas Soome  * This is a callback for vdev_indirect_remap() which allocates an
563b1b93268SToomas Soome  * indirect_split_t for each split segment and adds it to iv_splits.
564b1b93268SToomas Soome  */
565b1b93268SToomas Soome static void
566b1b93268SToomas Soome vdev_indirect_gather_splits(uint64_t split_offset, vdev_t *vd, uint64_t offset,
567b1b93268SToomas Soome     uint64_t size, void *arg)
568b1b93268SToomas Soome {
569b1b93268SToomas Soome 	int n = 1;
570b1b93268SToomas Soome 	zio_t *zio = arg;
571b1b93268SToomas Soome 	indirect_vsd_t *iv = zio->io_vsd;
572b1b93268SToomas Soome 
573b1b93268SToomas Soome 	if (vd->v_read == vdev_indirect_read)
574b1b93268SToomas Soome 		return;
575b1b93268SToomas Soome 
576b1b93268SToomas Soome 	if (vd->v_read == vdev_mirror_read)
577b1b93268SToomas Soome 		n = vd->v_nchildren;
578b1b93268SToomas Soome 
579b1b93268SToomas Soome 	indirect_split_t *is =
580b1b93268SToomas Soome 	    malloc(offsetof(indirect_split_t, is_child[n]));
581b1b93268SToomas Soome 	if (is == NULL) {
582b1b93268SToomas Soome 		zio->io_error = ENOMEM;
583b1b93268SToomas Soome 		return;
584b1b93268SToomas Soome 	}
585b1b93268SToomas Soome 	bzero(is, offsetof(indirect_split_t, is_child[n]));
586b1b93268SToomas Soome 
587b1b93268SToomas Soome 	is->is_children = n;
588b1b93268SToomas Soome 	is->is_size = size;
589b1b93268SToomas Soome 	is->is_split_offset = split_offset;
590b1b93268SToomas Soome 	is->is_target_offset = offset;
591b1b93268SToomas Soome 	is->is_vdev = vd;
592b1b93268SToomas Soome 
593b1b93268SToomas Soome 	/*
594b1b93268SToomas Soome 	 * Note that we only consider multiple copies of the data for
595b1b93268SToomas Soome 	 * *mirror* vdevs.  We don't for "replacing" or "spare" vdevs, even
596b1b93268SToomas Soome 	 * though they use the same ops as mirror, because there's only one
597b1b93268SToomas Soome 	 * "good" copy under the replacing/spare.
598b1b93268SToomas Soome 	 */
599b1b93268SToomas Soome 	if (vd->v_read == vdev_mirror_read) {
600b1b93268SToomas Soome 		int i = 0;
601b1b93268SToomas Soome 		vdev_t *kid;
602b1b93268SToomas Soome 
603b1b93268SToomas Soome 		STAILQ_FOREACH(kid, &vd->v_children, v_childlink) {
604b1b93268SToomas Soome 			is->is_child[i++].ic_vdev = kid;
605b1b93268SToomas Soome 		}
606b1b93268SToomas Soome 	} else {
607b1b93268SToomas Soome 		is->is_child[0].ic_vdev = vd;
608b1b93268SToomas Soome 	}
609b1b93268SToomas Soome 
610b1b93268SToomas Soome 	list_insert_tail(&iv->iv_splits, is);
611b1b93268SToomas Soome }
612b1b93268SToomas Soome 
613b1b93268SToomas Soome static void
614b1b93268SToomas Soome vdev_indirect_remap(vdev_t *vd, uint64_t offset, uint64_t asize, void *arg)
615b1b93268SToomas Soome {
616b1b93268SToomas Soome 	list_t stack;
6173c2db0efSToomas Soome 	spa_t *spa = vd->v_spa;
6186e5555adSToomas Soome 	zio_t *zio = arg;
6194efce32aSToomas Soome 	remap_segment_t *rs;
620b1b93268SToomas Soome 
621b1b93268SToomas Soome 	list_create(&stack, sizeof (remap_segment_t),
622b1b93268SToomas Soome 	    offsetof(remap_segment_t, rs_node));
623b1b93268SToomas Soome 
6244efce32aSToomas Soome 	rs = rs_alloc(vd, offset, asize, 0);
6254efce32aSToomas Soome 	if (rs == NULL) {
6264efce32aSToomas Soome 		printf("vdev_indirect_remap: out of memory.\n");
6274efce32aSToomas Soome 		zio->io_error = ENOMEM;
6284efce32aSToomas Soome 	}
6294efce32aSToomas Soome 	for (; rs != NULL; rs = list_remove_head(&stack)) {
630b1b93268SToomas Soome 		vdev_t *v = rs->rs_vd;
631b1b93268SToomas Soome 		uint64_t num_entries = 0;
632b1b93268SToomas Soome 		/* vdev_indirect_mapping_t *vim = v->v_mapping; */
633b1b93268SToomas Soome 		vdev_indirect_mapping_entry_phys_t *mapping =
634b1b93268SToomas Soome 		    vdev_indirect_mapping_duplicate_adjacent_entries(v,
635b1b93268SToomas Soome 		    rs->rs_offset, rs->rs_asize, &num_entries);
636b1b93268SToomas Soome 
6374efce32aSToomas Soome 		if (num_entries == 0)
6384efce32aSToomas Soome 			zio->io_error = ENOMEM;
6394efce32aSToomas Soome 
640b1b93268SToomas Soome 		for (uint64_t i = 0; i < num_entries; i++) {
641b1b93268SToomas Soome 			vdev_indirect_mapping_entry_phys_t *m = &mapping[i];
642b1b93268SToomas Soome 			uint64_t size = DVA_GET_ASIZE(&m->vimep_dst);
643b1b93268SToomas Soome 			uint64_t dst_offset = DVA_GET_OFFSET(&m->vimep_dst);
644b1b93268SToomas Soome 			uint64_t dst_vdev = DVA_GET_VDEV(&m->vimep_dst);
645b1b93268SToomas Soome 			uint64_t inner_offset = rs->rs_offset -
646b1b93268SToomas Soome 			    DVA_MAPPING_GET_SRC_OFFSET(m);
647b1b93268SToomas Soome 			uint64_t inner_size =
648b1b93268SToomas Soome 			    MIN(rs->rs_asize, size - inner_offset);
649b1b93268SToomas Soome 			vdev_t *dst_v = vdev_lookup_top(spa, dst_vdev);
650b1b93268SToomas Soome 
651b1b93268SToomas Soome 			if (dst_v->v_read == vdev_indirect_read) {
6524efce32aSToomas Soome 				remap_segment_t *o;
6534efce32aSToomas Soome 
6544efce32aSToomas Soome 				o = rs_alloc(dst_v, dst_offset + inner_offset,
6554efce32aSToomas Soome 				    inner_size, rs->rs_split_offset);
6564efce32aSToomas Soome 				if (o == NULL) {
6574efce32aSToomas Soome 					printf("vdev_indirect_remap: "
6584efce32aSToomas Soome 					    "out of memory.\n");
6594efce32aSToomas Soome 					zio->io_error = ENOMEM;
6604efce32aSToomas Soome 					break;
6614efce32aSToomas Soome 				}
6624efce32aSToomas Soome 
6634efce32aSToomas Soome 				list_insert_head(&stack, o);
664b1b93268SToomas Soome 			}
665b1b93268SToomas Soome 			vdev_indirect_gather_splits(rs->rs_split_offset, dst_v,
666b1b93268SToomas Soome 			    dst_offset + inner_offset,
667b1b93268SToomas Soome 			    inner_size, arg);
668b1b93268SToomas Soome 
6696e5555adSToomas Soome 			/*
6706e5555adSToomas Soome 			 * vdev_indirect_gather_splits can have memory
6716e5555adSToomas Soome 			 * allocation error, we can not recover from it.
6726e5555adSToomas Soome 			 */
6736e5555adSToomas Soome 			if (zio->io_error != 0)
6746e5555adSToomas Soome 				break;
675b1b93268SToomas Soome 			rs->rs_offset += inner_size;
676b1b93268SToomas Soome 			rs->rs_asize -= inner_size;
677b1b93268SToomas Soome 			rs->rs_split_offset += inner_size;
678b1b93268SToomas Soome 		}
679b1b93268SToomas Soome 
680b1b93268SToomas Soome 		free(mapping);
681b1b93268SToomas Soome 		free(rs);
6826e5555adSToomas Soome 		if (zio->io_error != 0)
6836e5555adSToomas Soome 			break;
684b1b93268SToomas Soome 	}
685b1b93268SToomas Soome 
686b1b93268SToomas Soome 	list_destroy(&stack);
687b1b93268SToomas Soome }
688b1b93268SToomas Soome 
689b1b93268SToomas Soome static void
690b1b93268SToomas Soome vdev_indirect_map_free(zio_t *zio)
691b1b93268SToomas Soome {
692b1b93268SToomas Soome 	indirect_vsd_t *iv = zio->io_vsd;
693b1b93268SToomas Soome 	indirect_split_t *is;
694b1b93268SToomas Soome 
695b1b93268SToomas Soome 	while ((is = list_head(&iv->iv_splits)) != NULL) {
696b1b93268SToomas Soome 		for (int c = 0; c < is->is_children; c++) {
697b1b93268SToomas Soome 			indirect_child_t *ic = &is->is_child[c];
698b1b93268SToomas Soome 			free(ic->ic_data);
699b1b93268SToomas Soome 		}
700b1b93268SToomas Soome 		list_remove(&iv->iv_splits, is);
701b1b93268SToomas Soome 		free(is);
702b1b93268SToomas Soome 	}
703b1b93268SToomas Soome 	free(iv);
704b1b93268SToomas Soome }
705b1b93268SToomas Soome 
706b1b93268SToomas Soome static int
707b1b93268SToomas Soome vdev_indirect_read(vdev_t *vdev, const blkptr_t *bp, void *buf,
708b1b93268SToomas Soome     off_t offset, size_t bytes)
709b1b93268SToomas Soome {
7103c2db0efSToomas Soome 	zio_t zio;
7113c2db0efSToomas Soome 	spa_t *spa = vdev->v_spa;
7123c2db0efSToomas Soome 	indirect_vsd_t *iv;
713b1b93268SToomas Soome 	indirect_split_t *first;
714b1b93268SToomas Soome 	int rc = EIO;
715b1b93268SToomas Soome 
7163c2db0efSToomas Soome 	iv = calloc(1, sizeof(*iv));
717b1b93268SToomas Soome 	if (iv == NULL)
718b1b93268SToomas Soome 		return (ENOMEM);
719b1b93268SToomas Soome 
720b1b93268SToomas Soome 	list_create(&iv->iv_splits,
721b1b93268SToomas Soome 	    sizeof (indirect_split_t), offsetof(indirect_split_t, is_node));
722b1b93268SToomas Soome 
7233c2db0efSToomas Soome 	bzero(&zio, sizeof(zio));
724b1b93268SToomas Soome 	zio.io_spa = spa;
725b1b93268SToomas Soome 	zio.io_bp = (blkptr_t *)bp;
726b1b93268SToomas Soome 	zio.io_data = buf;
727b1b93268SToomas Soome 	zio.io_size = bytes;
728b1b93268SToomas Soome 	zio.io_offset = offset;
729b1b93268SToomas Soome 	zio.io_vd = vdev;
730b1b93268SToomas Soome 	zio.io_vsd = iv;
731b1b93268SToomas Soome 
732b1b93268SToomas Soome 	if (vdev->v_mapping == NULL) {
733b1b93268SToomas Soome 		vdev_indirect_config_t *vic;
734b1b93268SToomas Soome 
735b1b93268SToomas Soome 		vic = &vdev->vdev_indirect_config;
736b1b93268SToomas Soome 		vdev->v_mapping = vdev_indirect_mapping_open(spa,
737277f38abSMariusz Zaborski 		    spa->spa_mos, vic->vic_mapping_object);
738b1b93268SToomas Soome 	}
739b1b93268SToomas Soome 
740b1b93268SToomas Soome 	vdev_indirect_remap(vdev, offset, bytes, &zio);
7416e5555adSToomas Soome 	if (zio.io_error != 0)
7426e5555adSToomas Soome 		return (zio.io_error);
743b1b93268SToomas Soome 
744b1b93268SToomas Soome 	first = list_head(&iv->iv_splits);
745b1b93268SToomas Soome 	if (first->is_size == zio.io_size) {
746b1b93268SToomas Soome 		/*
747b1b93268SToomas Soome 		 * This is not a split block; we are pointing to the entire
748b1b93268SToomas Soome 		 * data, which will checksum the same as the original data.
749b1b93268SToomas Soome 		 * Pass the BP down so that the child i/o can verify the
750b1b93268SToomas Soome 		 * checksum, and try a different location if available
751b1b93268SToomas Soome 		 * (e.g. on a mirror).
752b1b93268SToomas Soome 		 *
753b1b93268SToomas Soome 		 * While this special case could be handled the same as the
754b1b93268SToomas Soome 		 * general (split block) case, doing it this way ensures
755b1b93268SToomas Soome 		 * that the vast majority of blocks on indirect vdevs
756b1b93268SToomas Soome 		 * (which are not split) are handled identically to blocks
757b1b93268SToomas Soome 		 * on non-indirect vdevs.  This allows us to be less strict
758b1b93268SToomas Soome 		 * about performance in the general (but rare) case.
759b1b93268SToomas Soome 		 */
760b1b93268SToomas Soome 		rc = first->is_vdev->v_read(first->is_vdev, zio.io_bp,
761b1b93268SToomas Soome 		    zio.io_data, first->is_target_offset, bytes);
762b1b93268SToomas Soome 	} else {
763b1b93268SToomas Soome 		iv->iv_split_block = B_TRUE;
764b1b93268SToomas Soome 		/*
765b1b93268SToomas Soome 		 * Read one copy of each split segment, from the
766b1b93268SToomas Soome 		 * top-level vdev.  Since we don't know the
767b1b93268SToomas Soome 		 * checksum of each split individually, the child
768b1b93268SToomas Soome 		 * zio can't ensure that we get the right data.
769b1b93268SToomas Soome 		 * E.g. if it's a mirror, it will just read from a
770b1b93268SToomas Soome 		 * random (healthy) leaf vdev.  We have to verify
771b1b93268SToomas Soome 		 * the checksum in vdev_indirect_io_done().
772b1b93268SToomas Soome 		 */
773b1b93268SToomas Soome 		for (indirect_split_t *is = list_head(&iv->iv_splits);
774b1b93268SToomas Soome 		    is != NULL; is = list_next(&iv->iv_splits, is)) {
775b1b93268SToomas Soome 			char *ptr = zio.io_data;
776b1b93268SToomas Soome 
777b1b93268SToomas Soome 			rc = is->is_vdev->v_read(is->is_vdev, zio.io_bp,
778b1b93268SToomas Soome 			    ptr + is->is_split_offset, is->is_target_offset,
779b1b93268SToomas Soome 			    is->is_size);
780b1b93268SToomas Soome 		}
781b1b93268SToomas Soome 		if (zio_checksum_verify(spa, zio.io_bp, zio.io_data))
782b1b93268SToomas Soome 			rc = ECKSUM;
783b1b93268SToomas Soome 		else
784b1b93268SToomas Soome 			rc = 0;
785b1b93268SToomas Soome 	}
786b1b93268SToomas Soome 
787b1b93268SToomas Soome 	vdev_indirect_map_free(&zio);
788b1b93268SToomas Soome 	if (rc == 0)
789b1b93268SToomas Soome 		rc = zio.io_error;
790b1b93268SToomas Soome 
791b1b93268SToomas Soome 	return (rc);
792b1b93268SToomas Soome }
793b1b93268SToomas Soome 
794b8902de1SWarner Losh static int
795b8902de1SWarner Losh vdev_disk_read(vdev_t *vdev, const blkptr_t *bp, void *buf,
796b8902de1SWarner Losh     off_t offset, size_t bytes)
797b8902de1SWarner Losh {
798b8902de1SWarner Losh 
799b8902de1SWarner Losh 	return (vdev_read_phys(vdev, bp, buf,
800b8902de1SWarner Losh 	    offset + VDEV_LABEL_START_SIZE, bytes));
801b8902de1SWarner Losh }
802b8902de1SWarner Losh 
803b93b14dcSToomas Soome static int
804b93b14dcSToomas Soome vdev_missing_read(vdev_t *vdev __unused, const blkptr_t *bp __unused,
805b93b14dcSToomas Soome     void *buf __unused, off_t offset __unused, size_t bytes __unused)
806b93b14dcSToomas Soome {
807b93b14dcSToomas Soome 
808b93b14dcSToomas Soome 	return (ENOTSUP);
809b93b14dcSToomas Soome }
810b8902de1SWarner Losh 
811b8902de1SWarner Losh static int
812b8902de1SWarner Losh vdev_mirror_read(vdev_t *vdev, const blkptr_t *bp, void *buf,
813b8902de1SWarner Losh     off_t offset, size_t bytes)
814b8902de1SWarner Losh {
815b8902de1SWarner Losh 	vdev_t *kid;
816b8902de1SWarner Losh 	int rc;
817b8902de1SWarner Losh 
818b8902de1SWarner Losh 	rc = EIO;
819b8902de1SWarner Losh 	STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
820b8902de1SWarner Losh 		if (kid->v_state != VDEV_STATE_HEALTHY)
821b8902de1SWarner Losh 			continue;
822b8902de1SWarner Losh 		rc = kid->v_read(kid, bp, buf, offset, bytes);
823b8902de1SWarner Losh 		if (!rc)
824b8902de1SWarner Losh 			return (0);
825b8902de1SWarner Losh 	}
826b8902de1SWarner Losh 
827b8902de1SWarner Losh 	return (rc);
828b8902de1SWarner Losh }
829b8902de1SWarner Losh 
830b8902de1SWarner Losh static int
831b8902de1SWarner Losh vdev_replacing_read(vdev_t *vdev, const blkptr_t *bp, void *buf,
832b8902de1SWarner Losh     off_t offset, size_t bytes)
833b8902de1SWarner Losh {
834b8902de1SWarner Losh 	vdev_t *kid;
835b8902de1SWarner Losh 
836b8902de1SWarner Losh 	/*
837b8902de1SWarner Losh 	 * Here we should have two kids:
838b8902de1SWarner Losh 	 * First one which is the one we are replacing and we can trust
839b8902de1SWarner Losh 	 * only this one to have valid data, but it might not be present.
840b8902de1SWarner Losh 	 * Second one is that one we are replacing with. It is most likely
841b8902de1SWarner Losh 	 * healthy, but we can't trust it has needed data, so we won't use it.
842b8902de1SWarner Losh 	 */
843b8902de1SWarner Losh 	kid = STAILQ_FIRST(&vdev->v_children);
844b8902de1SWarner Losh 	if (kid == NULL)
845b8902de1SWarner Losh 		return (EIO);
846b8902de1SWarner Losh 	if (kid->v_state != VDEV_STATE_HEALTHY)
847b8902de1SWarner Losh 		return (EIO);
848b8902de1SWarner Losh 	return (kid->v_read(kid, bp, buf, offset, bytes));
849b8902de1SWarner Losh }
850b8902de1SWarner Losh 
851b8902de1SWarner Losh static vdev_t *
852b8902de1SWarner Losh vdev_find(uint64_t guid)
853b8902de1SWarner Losh {
854b8902de1SWarner Losh 	vdev_t *vdev;
855b8902de1SWarner Losh 
856b8902de1SWarner Losh 	STAILQ_FOREACH(vdev, &zfs_vdevs, v_alllink)
857b8902de1SWarner Losh 		if (vdev->v_guid == guid)
858b8902de1SWarner Losh 			return (vdev);
859b8902de1SWarner Losh 
860b8902de1SWarner Losh 	return (0);
861b8902de1SWarner Losh }
862b8902de1SWarner Losh 
863b8902de1SWarner Losh static vdev_t *
864b8902de1SWarner Losh vdev_create(uint64_t guid, vdev_read_t *_read)
865b8902de1SWarner Losh {
866b8902de1SWarner Losh 	vdev_t *vdev;
867b1b93268SToomas Soome 	vdev_indirect_config_t *vic;
868b8902de1SWarner Losh 
8693c2db0efSToomas Soome 	vdev = calloc(1, sizeof(vdev_t));
8703c2db0efSToomas Soome 	if (vdev != NULL) {
871b8902de1SWarner Losh 		STAILQ_INIT(&vdev->v_children);
872b8902de1SWarner Losh 		vdev->v_guid = guid;
873b8902de1SWarner Losh 		vdev->v_read = _read;
874b1b93268SToomas Soome 
8753c2db0efSToomas Soome 		/*
876653c3383SToomas Soome 		 * root vdev has no read function, we use this fact to
877653c3383SToomas Soome 		 * skip setting up data we do not need for root vdev.
8783c2db0efSToomas Soome 		 * We only point root vdev from spa.
8793c2db0efSToomas Soome 		 */
8803c2db0efSToomas Soome 		if (_read != NULL) {
881b1b93268SToomas Soome 			vic = &vdev->vdev_indirect_config;
882b1b93268SToomas Soome 			vic->vic_prev_indirect_vdev = UINT64_MAX;
883b8902de1SWarner Losh 			STAILQ_INSERT_TAIL(&zfs_vdevs, vdev, v_alllink);
8843c2db0efSToomas Soome 		}
8853c2db0efSToomas Soome 	}
886b8902de1SWarner Losh 
887b8902de1SWarner Losh 	return (vdev);
888b8902de1SWarner Losh }
889b8902de1SWarner Losh 
8903c2db0efSToomas Soome static void
8913830659eSToomas Soome vdev_set_initial_state(vdev_t *vdev, const nvlist_t *nvlist)
892b8902de1SWarner Losh {
893b8902de1SWarner Losh 	uint64_t is_offline, is_faulted, is_degraded, is_removed, isnt_present;
8940c0a882cSToomas Soome 	uint64_t is_log;
895b8902de1SWarner Losh 
8963c2db0efSToomas Soome 	is_offline = is_removed = is_faulted = is_degraded = isnt_present = 0;
8973c2db0efSToomas Soome 	is_log = 0;
8983c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_OFFLINE, DATA_TYPE_UINT64, NULL,
899c98740b5SToomas Soome 	    &is_offline, NULL);
9003c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_REMOVED, DATA_TYPE_UINT64, NULL,
901c98740b5SToomas Soome 	    &is_removed, NULL);
9023c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_FAULTED, DATA_TYPE_UINT64, NULL,
903c98740b5SToomas Soome 	    &is_faulted, NULL);
9043c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_DEGRADED, DATA_TYPE_UINT64,
905c98740b5SToomas Soome 	    NULL, &is_degraded, NULL);
9063c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_NOT_PRESENT, DATA_TYPE_UINT64,
907c98740b5SToomas Soome 	    NULL, &isnt_present, NULL);
9083c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_IS_LOG, DATA_TYPE_UINT64, NULL,
909c98740b5SToomas Soome 	    &is_log, NULL);
9103c2db0efSToomas Soome 
9113c2db0efSToomas Soome 	if (is_offline != 0)
9123c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_OFFLINE;
9133c2db0efSToomas Soome 	else if (is_removed != 0)
9143c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_REMOVED;
9153c2db0efSToomas Soome 	else if (is_faulted != 0)
9163c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_FAULTED;
9173c2db0efSToomas Soome 	else if (is_degraded != 0)
9183c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_DEGRADED;
9193c2db0efSToomas Soome 	else if (isnt_present != 0)
9203c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_CANT_OPEN;
9213c2db0efSToomas Soome 
922653c3383SToomas Soome 	vdev->v_islog = is_log != 0;
9233c2db0efSToomas Soome }
9243c2db0efSToomas Soome 
9253c2db0efSToomas Soome static int
9263830659eSToomas Soome vdev_init(uint64_t guid, const nvlist_t *nvlist, vdev_t **vdevp)
9273c2db0efSToomas Soome {
9283c2db0efSToomas Soome 	uint64_t id, ashift, asize, nparity;
9293c2db0efSToomas Soome 	const char *path;
9303c2db0efSToomas Soome 	const char *type;
931c98740b5SToomas Soome 	int len, pathlen;
932c98740b5SToomas Soome 	char *name;
9333c2db0efSToomas Soome 	vdev_t *vdev;
9343c2db0efSToomas Soome 
935c98740b5SToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_ID, DATA_TYPE_UINT64, NULL, &id,
936c98740b5SToomas Soome 	    NULL) ||
9373830659eSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_TYPE, DATA_TYPE_STRING, NULL,
9383830659eSToomas Soome 	    &type, &len)) {
939b8902de1SWarner Losh 		return (ENOENT);
940b8902de1SWarner Losh 	}
941b8902de1SWarner Losh 
942c98740b5SToomas Soome 	if (memcmp(type, VDEV_TYPE_MIRROR, len) != 0 &&
943c98740b5SToomas Soome 	    memcmp(type, VDEV_TYPE_DISK, len) != 0 &&
944b8902de1SWarner Losh #ifdef ZFS_TEST
945c98740b5SToomas Soome 	    memcmp(type, VDEV_TYPE_FILE, len) != 0 &&
946b8902de1SWarner Losh #endif
947c98740b5SToomas Soome 	    memcmp(type, VDEV_TYPE_RAIDZ, len) != 0 &&
948c98740b5SToomas Soome 	    memcmp(type, VDEV_TYPE_INDIRECT, len) != 0 &&
949b93b14dcSToomas Soome 	    memcmp(type, VDEV_TYPE_REPLACING, len) != 0 &&
950b93b14dcSToomas Soome 	    memcmp(type, VDEV_TYPE_HOLE, len) != 0) {
9512e6bb655SToomas Soome 		printf("ZFS: can only boot from disk, mirror, raidz1, "
952b93b14dcSToomas Soome 		    "raidz2 and raidz3 vdevs, got: %.*s\n", len, type);
953b8902de1SWarner Losh 		return (EIO);
954b8902de1SWarner Losh 	}
955b8902de1SWarner Losh 
956c98740b5SToomas Soome 	if (memcmp(type, VDEV_TYPE_MIRROR, len) == 0)
957b8902de1SWarner Losh 		vdev = vdev_create(guid, vdev_mirror_read);
958c98740b5SToomas Soome 	else if (memcmp(type, VDEV_TYPE_RAIDZ, len) == 0)
959b8902de1SWarner Losh 		vdev = vdev_create(guid, vdev_raidz_read);
960c98740b5SToomas Soome 	else if (memcmp(type, VDEV_TYPE_REPLACING, len) == 0)
961b8902de1SWarner Losh 		vdev = vdev_create(guid, vdev_replacing_read);
962c98740b5SToomas Soome 	else if (memcmp(type, VDEV_TYPE_INDIRECT, len) == 0) {
963b1b93268SToomas Soome 		vdev_indirect_config_t *vic;
964b1b93268SToomas Soome 
965b1b93268SToomas Soome 		vdev = vdev_create(guid, vdev_indirect_read);
9663c2db0efSToomas Soome 		if (vdev != NULL) {
967b1b93268SToomas Soome 			vdev->v_state = VDEV_STATE_HEALTHY;
968b1b93268SToomas Soome 			vic = &vdev->vdev_indirect_config;
969b1b93268SToomas Soome 
970b1b93268SToomas Soome 			nvlist_find(nvlist,
9713c2db0efSToomas Soome 			    ZPOOL_CONFIG_INDIRECT_OBJECT,
9723c2db0efSToomas Soome 			    DATA_TYPE_UINT64,
973c98740b5SToomas Soome 			    NULL, &vic->vic_mapping_object, NULL);
974b1b93268SToomas Soome 			nvlist_find(nvlist,
9753c2db0efSToomas Soome 			    ZPOOL_CONFIG_INDIRECT_BIRTHS,
9763c2db0efSToomas Soome 			    DATA_TYPE_UINT64,
977c98740b5SToomas Soome 			    NULL, &vic->vic_births_object, NULL);
978b1b93268SToomas Soome 			nvlist_find(nvlist,
9793c2db0efSToomas Soome 			    ZPOOL_CONFIG_PREV_INDIRECT_VDEV,
9803c2db0efSToomas Soome 			    DATA_TYPE_UINT64,
981c98740b5SToomas Soome 			    NULL, &vic->vic_prev_indirect_vdev, NULL);
982b8902de1SWarner Losh 		}
983b93b14dcSToomas Soome 	} else if (memcmp(type, VDEV_TYPE_HOLE, len) == 0) {
984b93b14dcSToomas Soome 		vdev = vdev_create(guid, vdev_missing_read);
9853c2db0efSToomas Soome 	} else {
9863c2db0efSToomas Soome 		vdev = vdev_create(guid, vdev_disk_read);
9873c2db0efSToomas Soome 	}
9883c2db0efSToomas Soome 
9893c2db0efSToomas Soome 	if (vdev == NULL)
9903c2db0efSToomas Soome 		return (ENOMEM);
9913c2db0efSToomas Soome 
9923c2db0efSToomas Soome 	vdev_set_initial_state(vdev, nvlist);
9933c2db0efSToomas Soome 	vdev->v_id = id;
9943c2db0efSToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_ASHIFT,
995c98740b5SToomas Soome 	    DATA_TYPE_UINT64, NULL, &ashift, NULL) == 0)
9963c2db0efSToomas Soome 		vdev->v_ashift = ashift;
9973c2db0efSToomas Soome 
998abca0bd5SToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_ASIZE,
999c98740b5SToomas Soome 	    DATA_TYPE_UINT64, NULL, &asize, NULL) == 0) {
1000abca0bd5SToomas Soome 		vdev->v_psize = asize +
1001abca0bd5SToomas Soome 		    VDEV_LABEL_START_SIZE + VDEV_LABEL_END_SIZE;
1002abca0bd5SToomas Soome 	}
10033c2db0efSToomas Soome 
1004b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_NPARITY,
1005c98740b5SToomas Soome 	    DATA_TYPE_UINT64, NULL, &nparity, NULL) == 0)
1006b8902de1SWarner Losh 		vdev->v_nparity = nparity;
10073c2db0efSToomas Soome 
1008b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_PATH,
1009c98740b5SToomas Soome 	    DATA_TYPE_STRING, NULL, &path, &pathlen) == 0) {
1010c98740b5SToomas Soome 		char prefix[] = "/dev/";
1011b1b93268SToomas Soome 
1012c98740b5SToomas Soome 		len = strlen(prefix);
1013aba2397eSToomas Soome 		if (len < pathlen && memcmp(path, prefix, len) == 0) {
1014c98740b5SToomas Soome 			path += len;
1015c98740b5SToomas Soome 			pathlen -= len;
1016c98740b5SToomas Soome 		}
1017c98740b5SToomas Soome 		name = malloc(pathlen + 1);
1018c98740b5SToomas Soome 		bcopy(path, name, pathlen);
1019c98740b5SToomas Soome 		name[pathlen] = '\0';
1020c98740b5SToomas Soome 		vdev->v_name = name;
1021c98740b5SToomas Soome 	} else {
10223c2db0efSToomas Soome 		name = NULL;
1023c98740b5SToomas Soome 		if (memcmp(type, VDEV_TYPE_RAIDZ, len) == 0) {
1024b1b93268SToomas Soome 			if (vdev->v_nparity < 1 ||
1025b1b93268SToomas Soome 			    vdev->v_nparity > 3) {
1026653c3383SToomas Soome 				printf("ZFS: invalid raidz parity: %d\n",
1027653c3383SToomas Soome 				    vdev->v_nparity);
1028b8902de1SWarner Losh 				return (EIO);
1029b8902de1SWarner Losh 			}
1030c98740b5SToomas Soome 			(void) asprintf(&name, "%.*s%d-%" PRIu64, len, type,
1031b1b93268SToomas Soome 			    vdev->v_nparity, id);
1032b8902de1SWarner Losh 		} else {
1033c98740b5SToomas Soome 			(void) asprintf(&name, "%.*s-%" PRIu64, len, type, id);
1034b8902de1SWarner Losh 		}
1035b1b93268SToomas Soome 		vdev->v_name = name;
1036b8902de1SWarner Losh 	}
10373c2db0efSToomas Soome 	*vdevp = vdev;
10383c2db0efSToomas Soome 	return (0);
1039b8902de1SWarner Losh }
1040b8902de1SWarner Losh 
1041b8902de1SWarner Losh /*
10423c2db0efSToomas Soome  * Find slot for vdev. We return either NULL to signal to use
10433c2db0efSToomas Soome  * STAILQ_INSERT_HEAD, or we return link element to be used with
10443c2db0efSToomas Soome  * STAILQ_INSERT_AFTER.
1045b8902de1SWarner Losh  */
10463c2db0efSToomas Soome static vdev_t *
10473c2db0efSToomas Soome vdev_find_previous(vdev_t *top_vdev, vdev_t *vdev)
10483c2db0efSToomas Soome {
10493c2db0efSToomas Soome 	vdev_t *v, *previous;
10503c2db0efSToomas Soome 
10513c2db0efSToomas Soome 	if (STAILQ_EMPTY(&top_vdev->v_children))
10523c2db0efSToomas Soome 		return (NULL);
10533c2db0efSToomas Soome 
10543c2db0efSToomas Soome 	previous = NULL;
10553c2db0efSToomas Soome 	STAILQ_FOREACH(v, &top_vdev->v_children, v_childlink) {
10563c2db0efSToomas Soome 		if (v->v_id > vdev->v_id)
10573c2db0efSToomas Soome 			return (previous);
10583c2db0efSToomas Soome 
10593c2db0efSToomas Soome 		if (v->v_id == vdev->v_id)
10603c2db0efSToomas Soome 			return (v);
10613c2db0efSToomas Soome 
10623c2db0efSToomas Soome 		if (v->v_id < vdev->v_id)
10633c2db0efSToomas Soome 			previous = v;
10643c2db0efSToomas Soome 	}
10653c2db0efSToomas Soome 	return (previous);
1066b8902de1SWarner Losh }
1067b8902de1SWarner Losh 
10683c2db0efSToomas Soome static size_t
10693c2db0efSToomas Soome vdev_child_count(vdev_t *vdev)
10703c2db0efSToomas Soome {
10713c2db0efSToomas Soome 	vdev_t *v;
10723c2db0efSToomas Soome 	size_t count;
10733c2db0efSToomas Soome 
10743c2db0efSToomas Soome 	count = 0;
10753c2db0efSToomas Soome 	STAILQ_FOREACH(v, &vdev->v_children, v_childlink) {
10763c2db0efSToomas Soome 		count++;
10773c2db0efSToomas Soome 	}
10783c2db0efSToomas Soome 	return (count);
10793c2db0efSToomas Soome }
10803c2db0efSToomas Soome 
10813c2db0efSToomas Soome /*
10823c2db0efSToomas Soome  * Insert vdev into top_vdev children list. List is ordered by v_id.
10833c2db0efSToomas Soome  */
10843c2db0efSToomas Soome static void
10853c2db0efSToomas Soome vdev_insert(vdev_t *top_vdev, vdev_t *vdev)
10863c2db0efSToomas Soome {
10873c2db0efSToomas Soome 	vdev_t *previous;
10883c2db0efSToomas Soome 	size_t count;
10893c2db0efSToomas Soome 
10903c2db0efSToomas Soome 	/*
10913c2db0efSToomas Soome 	 * The top level vdev can appear in random order, depending how
10923c2db0efSToomas Soome 	 * the firmware is presenting the disk devices.
10933c2db0efSToomas Soome 	 * However, we will insert vdev to create list ordered by v_id,
10943c2db0efSToomas Soome 	 * so we can use either STAILQ_INSERT_HEAD or STAILQ_INSERT_AFTER
10953c2db0efSToomas Soome 	 * as STAILQ does not have insert before.
10963c2db0efSToomas Soome 	 */
10973c2db0efSToomas Soome 	previous = vdev_find_previous(top_vdev, vdev);
10983c2db0efSToomas Soome 
10993c2db0efSToomas Soome 	if (previous == NULL) {
11003c2db0efSToomas Soome 		STAILQ_INSERT_HEAD(&top_vdev->v_children, vdev, v_childlink);
1101653c3383SToomas Soome 	} else if (previous->v_id == vdev->v_id) {
1102653c3383SToomas Soome 		/*
1103653c3383SToomas Soome 		 * This vdev was configured from label config,
1104653c3383SToomas Soome 		 * do not insert duplicate.
1105653c3383SToomas Soome 		 */
11063c2db0efSToomas Soome 		return;
1107653c3383SToomas Soome 	} else {
1108653c3383SToomas Soome 		STAILQ_INSERT_AFTER(&top_vdev->v_children, previous, vdev,
1109653c3383SToomas Soome 		    v_childlink);
11103c2db0efSToomas Soome 	}
11113c2db0efSToomas Soome 
11123c2db0efSToomas Soome 	count = vdev_child_count(top_vdev);
11133c2db0efSToomas Soome 	if (top_vdev->v_nchildren < count)
11143c2db0efSToomas Soome 		top_vdev->v_nchildren = count;
11153c2db0efSToomas Soome }
11163c2db0efSToomas Soome 
11173c2db0efSToomas Soome static int
11183830659eSToomas Soome vdev_from_nvlist(spa_t *spa, uint64_t top_guid, const nvlist_t *nvlist)
11193c2db0efSToomas Soome {
11203c2db0efSToomas Soome 	vdev_t *top_vdev, *vdev;
1121e307eb94SToomas Soome 	nvlist_t **kids = NULL;
11223c2db0efSToomas Soome 	int rc, nkids;
11233c2db0efSToomas Soome 
11243c2db0efSToomas Soome 	/* Get top vdev. */
11253c2db0efSToomas Soome 	top_vdev = vdev_find(top_guid);
11263c2db0efSToomas Soome 	if (top_vdev == NULL) {
11273c2db0efSToomas Soome 		rc = vdev_init(top_guid, nvlist, &top_vdev);
11283c2db0efSToomas Soome 		if (rc != 0)
11293c2db0efSToomas Soome 			return (rc);
11303c2db0efSToomas Soome 		top_vdev->v_spa = spa;
11313c2db0efSToomas Soome 		top_vdev->v_top = top_vdev;
11323c2db0efSToomas Soome 		vdev_insert(spa->spa_root_vdev, top_vdev);
11333c2db0efSToomas Soome 	}
11343c2db0efSToomas Soome 
11353c2db0efSToomas Soome 	/* Add children if there are any. */
1136b8902de1SWarner Losh 	rc = nvlist_find(nvlist, ZPOOL_CONFIG_CHILDREN, DATA_TYPE_NVLIST_ARRAY,
1137c98740b5SToomas Soome 	    &nkids, &kids, NULL);
1138b8902de1SWarner Losh 	if (rc == 0) {
11393c2db0efSToomas Soome 		for (int i = 0; i < nkids; i++) {
11403c2db0efSToomas Soome 			uint64_t guid;
11413c2db0efSToomas Soome 
1142e307eb94SToomas Soome 			rc = nvlist_find(kids[i], ZPOOL_CONFIG_GUID,
1143c98740b5SToomas Soome 			    DATA_TYPE_UINT64, NULL, &guid, NULL);
1144e307eb94SToomas Soome 			if (rc != 0)
1145e307eb94SToomas Soome 				goto done;
1146e307eb94SToomas Soome 
1147e307eb94SToomas Soome 			rc = vdev_init(guid, kids[i], &vdev);
1148e307eb94SToomas Soome 			if (rc != 0)
1149e307eb94SToomas Soome 				goto done;
11503c2db0efSToomas Soome 
11513c2db0efSToomas Soome 			vdev->v_spa = spa;
11523c2db0efSToomas Soome 			vdev->v_top = top_vdev;
11533c2db0efSToomas Soome 			vdev_insert(top_vdev, vdev);
1154b8902de1SWarner Losh 		}
1155b8902de1SWarner Losh 	} else {
1156653c3383SToomas Soome 		/*
1157653c3383SToomas Soome 		 * When there are no children, nvlist_find() does return
1158653c3383SToomas Soome 		 * error, reset it because leaf devices have no children.
1159653c3383SToomas Soome 		 */
11603c2db0efSToomas Soome 		rc = 0;
1161b8902de1SWarner Losh 	}
1162e307eb94SToomas Soome done:
1163e307eb94SToomas Soome 	if (kids != NULL) {
1164e307eb94SToomas Soome 		for (int i = 0; i < nkids; i++)
1165e307eb94SToomas Soome 			nvlist_destroy(kids[i]);
1166e307eb94SToomas Soome 		free(kids);
1167e307eb94SToomas Soome 	}
1168b8902de1SWarner Losh 
11693c2db0efSToomas Soome 	return (rc);
11703c2db0efSToomas Soome }
11713c2db0efSToomas Soome 
11723c2db0efSToomas Soome static int
11733830659eSToomas Soome vdev_init_from_label(spa_t *spa, const nvlist_t *nvlist)
11743c2db0efSToomas Soome {
11753c2db0efSToomas Soome 	uint64_t pool_guid, top_guid;
11763830659eSToomas Soome 	nvlist_t *vdevs;
11773830659eSToomas Soome 	int rc;
11783c2db0efSToomas Soome 
11793c2db0efSToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_POOL_GUID, DATA_TYPE_UINT64,
1180c98740b5SToomas Soome 	    NULL, &pool_guid, NULL) ||
11813c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_TOP_GUID, DATA_TYPE_UINT64,
1182c98740b5SToomas Soome 	    NULL, &top_guid, NULL) ||
11833c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_VDEV_TREE, DATA_TYPE_NVLIST,
1184c98740b5SToomas Soome 	    NULL, &vdevs, NULL)) {
11853c2db0efSToomas Soome 		printf("ZFS: can't find vdev details\n");
11863c2db0efSToomas Soome 		return (ENOENT);
11873c2db0efSToomas Soome 	}
11883c2db0efSToomas Soome 
11893830659eSToomas Soome 	rc = vdev_from_nvlist(spa, top_guid, vdevs);
11903830659eSToomas Soome 	nvlist_destroy(vdevs);
11913830659eSToomas Soome 	return (rc);
1192b8902de1SWarner Losh }
1193b8902de1SWarner Losh 
1194b8902de1SWarner Losh static void
1195b8902de1SWarner Losh vdev_set_state(vdev_t *vdev)
1196b8902de1SWarner Losh {
1197b8902de1SWarner Losh 	vdev_t *kid;
1198b8902de1SWarner Losh 	int good_kids;
1199b8902de1SWarner Losh 	int bad_kids;
1200b8902de1SWarner Losh 
12013c2db0efSToomas Soome 	STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
12023c2db0efSToomas Soome 		vdev_set_state(kid);
12033c2db0efSToomas Soome 	}
12043c2db0efSToomas Soome 
1205b8902de1SWarner Losh 	/*
1206b8902de1SWarner Losh 	 * A mirror or raidz is healthy if all its kids are healthy. A
1207b8902de1SWarner Losh 	 * mirror is degraded if any of its kids is healthy; a raidz
1208b8902de1SWarner Losh 	 * is degraded if at most nparity kids are offline.
1209b8902de1SWarner Losh 	 */
1210b8902de1SWarner Losh 	if (STAILQ_FIRST(&vdev->v_children)) {
1211b8902de1SWarner Losh 		good_kids = 0;
1212b8902de1SWarner Losh 		bad_kids = 0;
1213b8902de1SWarner Losh 		STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
1214b8902de1SWarner Losh 			if (kid->v_state == VDEV_STATE_HEALTHY)
1215b8902de1SWarner Losh 				good_kids++;
1216b8902de1SWarner Losh 			else
1217b8902de1SWarner Losh 				bad_kids++;
1218b8902de1SWarner Losh 		}
1219b8902de1SWarner Losh 		if (bad_kids == 0) {
1220b8902de1SWarner Losh 			vdev->v_state = VDEV_STATE_HEALTHY;
1221b8902de1SWarner Losh 		} else {
1222b8902de1SWarner Losh 			if (vdev->v_read == vdev_mirror_read) {
1223b8902de1SWarner Losh 				if (good_kids) {
1224b8902de1SWarner Losh 					vdev->v_state = VDEV_STATE_DEGRADED;
1225b8902de1SWarner Losh 				} else {
1226b8902de1SWarner Losh 					vdev->v_state = VDEV_STATE_OFFLINE;
1227b8902de1SWarner Losh 				}
1228b8902de1SWarner Losh 			} else if (vdev->v_read == vdev_raidz_read) {
1229b8902de1SWarner Losh 				if (bad_kids > vdev->v_nparity) {
1230b8902de1SWarner Losh 					vdev->v_state = VDEV_STATE_OFFLINE;
1231b8902de1SWarner Losh 				} else {
1232b8902de1SWarner Losh 					vdev->v_state = VDEV_STATE_DEGRADED;
1233b8902de1SWarner Losh 				}
1234b8902de1SWarner Losh 			}
1235b8902de1SWarner Losh 		}
1236b8902de1SWarner Losh 	}
1237b8902de1SWarner Losh }
1238b8902de1SWarner Losh 
12393c2db0efSToomas Soome static int
12403830659eSToomas Soome vdev_update_from_nvlist(uint64_t top_guid, const nvlist_t *nvlist)
12413c2db0efSToomas Soome {
12423c2db0efSToomas Soome 	vdev_t *vdev;
1243e307eb94SToomas Soome 	nvlist_t **kids = NULL;
12443c2db0efSToomas Soome 	int rc, nkids;
12453c2db0efSToomas Soome 
12463c2db0efSToomas Soome 	/* Update top vdev. */
12473c2db0efSToomas Soome 	vdev = vdev_find(top_guid);
12483c2db0efSToomas Soome 	if (vdev != NULL)
12493c2db0efSToomas Soome 		vdev_set_initial_state(vdev, nvlist);
12503c2db0efSToomas Soome 
12513c2db0efSToomas Soome 	/* Update children if there are any. */
12523c2db0efSToomas Soome 	rc = nvlist_find(nvlist, ZPOOL_CONFIG_CHILDREN, DATA_TYPE_NVLIST_ARRAY,
1253c98740b5SToomas Soome 	    &nkids, &kids, NULL);
12543c2db0efSToomas Soome 	if (rc == 0) {
12553c2db0efSToomas Soome 		for (int i = 0; i < nkids; i++) {
12563c2db0efSToomas Soome 			uint64_t guid;
12573c2db0efSToomas Soome 
1258e307eb94SToomas Soome 			rc = nvlist_find(kids[i], ZPOOL_CONFIG_GUID,
1259c98740b5SToomas Soome 			    DATA_TYPE_UINT64, NULL, &guid, NULL);
12603c2db0efSToomas Soome 			if (rc != 0)
12613c2db0efSToomas Soome 				break;
12623c2db0efSToomas Soome 
12633c2db0efSToomas Soome 			vdev = vdev_find(guid);
12643c2db0efSToomas Soome 			if (vdev != NULL)
1265e307eb94SToomas Soome 				vdev_set_initial_state(vdev, kids[i]);
12663c2db0efSToomas Soome 		}
12673c2db0efSToomas Soome 	} else {
12683c2db0efSToomas Soome 		rc = 0;
12693c2db0efSToomas Soome 	}
1270e307eb94SToomas Soome 	if (kids != NULL) {
1271e307eb94SToomas Soome 		for (int i = 0; i < nkids; i++)
1272e307eb94SToomas Soome 			nvlist_destroy(kids[i]);
1273e307eb94SToomas Soome 		free(kids);
1274e307eb94SToomas Soome 	}
12753c2db0efSToomas Soome 
12763c2db0efSToomas Soome 	return (rc);
12773c2db0efSToomas Soome }
12783c2db0efSToomas Soome 
12793c2db0efSToomas Soome static int
12803830659eSToomas Soome vdev_init_from_nvlist(spa_t *spa, const nvlist_t *nvlist)
12813c2db0efSToomas Soome {
12823c2db0efSToomas Soome 	uint64_t pool_guid, vdev_children;
1283e307eb94SToomas Soome 	nvlist_t *vdevs = NULL, **kids = NULL;
12843c2db0efSToomas Soome 	int rc, nkids;
12853c2db0efSToomas Soome 
12863c2db0efSToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_POOL_GUID, DATA_TYPE_UINT64,
1287c98740b5SToomas Soome 	    NULL, &pool_guid, NULL) ||
12883c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_VDEV_CHILDREN, DATA_TYPE_UINT64,
1289c98740b5SToomas Soome 	    NULL, &vdev_children, NULL) ||
12903c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_VDEV_TREE, DATA_TYPE_NVLIST,
1291c98740b5SToomas Soome 	    NULL, &vdevs, NULL)) {
12923c2db0efSToomas Soome 		printf("ZFS: can't find vdev details\n");
12933c2db0efSToomas Soome 		return (ENOENT);
12943c2db0efSToomas Soome 	}
12953c2db0efSToomas Soome 
12963c2db0efSToomas Soome 	/* Wrong guid?! */
12973830659eSToomas Soome 	if (spa->spa_guid != pool_guid) {
12983830659eSToomas Soome 		nvlist_destroy(vdevs);
1299653c3383SToomas Soome 		return (EINVAL);
13003830659eSToomas Soome 	}
13013c2db0efSToomas Soome 
13023c2db0efSToomas Soome 	spa->spa_root_vdev->v_nchildren = vdev_children;
13033c2db0efSToomas Soome 
13043c2db0efSToomas Soome 	rc = nvlist_find(vdevs, ZPOOL_CONFIG_CHILDREN, DATA_TYPE_NVLIST_ARRAY,
1305c98740b5SToomas Soome 	    &nkids, &kids, NULL);
13063830659eSToomas Soome 	nvlist_destroy(vdevs);
13073c2db0efSToomas Soome 
13083c2db0efSToomas Soome 	/*
13093c2db0efSToomas Soome 	 * MOS config has at least one child for root vdev.
13103c2db0efSToomas Soome 	 */
13113c2db0efSToomas Soome 	if (rc != 0)
1312653c3383SToomas Soome 		return (rc);
13133c2db0efSToomas Soome 
13143c2db0efSToomas Soome 	for (int i = 0; i < nkids; i++) {
13153c2db0efSToomas Soome 		uint64_t guid;
13163c2db0efSToomas Soome 		vdev_t *vdev;
13173c2db0efSToomas Soome 
1318e307eb94SToomas Soome 		rc = nvlist_find(kids[i], ZPOOL_CONFIG_GUID, DATA_TYPE_UINT64,
1319c98740b5SToomas Soome 		    NULL, &guid, NULL);
13203c2db0efSToomas Soome 		if (rc != 0)
13213c2db0efSToomas Soome 			break;
13223c2db0efSToomas Soome 		vdev = vdev_find(guid);
13233c2db0efSToomas Soome 		/*
13243c2db0efSToomas Soome 		 * Top level vdev is missing, create it.
13253c2db0efSToomas Soome 		 */
13263c2db0efSToomas Soome 		if (vdev == NULL)
1327e307eb94SToomas Soome 			rc = vdev_from_nvlist(spa, guid, kids[i]);
13283c2db0efSToomas Soome 		else
1329e307eb94SToomas Soome 			rc = vdev_update_from_nvlist(guid, kids[i]);
1330045f497cSToomas Soome 		if (rc != 0)
1331045f497cSToomas Soome 			break;
13323c2db0efSToomas Soome 	}
1333e307eb94SToomas Soome 	if (kids != NULL) {
1334e307eb94SToomas Soome 		for (int i = 0; i < nkids; i++)
1335e307eb94SToomas Soome 			nvlist_destroy(kids[i]);
1336e307eb94SToomas Soome 		free(kids);
1337e307eb94SToomas Soome 	}
13383c2db0efSToomas Soome 
13393c2db0efSToomas Soome 	/*
13403c2db0efSToomas Soome 	 * Re-evaluate top-level vdev state.
13413c2db0efSToomas Soome 	 */
13423c2db0efSToomas Soome 	vdev_set_state(spa->spa_root_vdev);
13433c2db0efSToomas Soome 
13443c2db0efSToomas Soome 	return (rc);
13453c2db0efSToomas Soome }
13463c2db0efSToomas Soome 
1347b8902de1SWarner Losh static spa_t *
1348b8902de1SWarner Losh spa_find_by_guid(uint64_t guid)
1349b8902de1SWarner Losh {
1350b8902de1SWarner Losh 	spa_t *spa;
1351b8902de1SWarner Losh 
1352b8902de1SWarner Losh 	STAILQ_FOREACH(spa, &zfs_pools, spa_link)
1353b8902de1SWarner Losh 		if (spa->spa_guid == guid)
1354b8902de1SWarner Losh 			return (spa);
1355b8902de1SWarner Losh 
13562e6bb655SToomas Soome 	return (NULL);
1357b8902de1SWarner Losh }
1358b8902de1SWarner Losh 
1359b8902de1SWarner Losh static spa_t *
1360b8902de1SWarner Losh spa_find_by_name(const char *name)
1361b8902de1SWarner Losh {
1362b8902de1SWarner Losh 	spa_t *spa;
1363b8902de1SWarner Losh 
1364b8902de1SWarner Losh 	STAILQ_FOREACH(spa, &zfs_pools, spa_link)
13652e6bb655SToomas Soome 		if (strcmp(spa->spa_name, name) == 0)
1366b8902de1SWarner Losh 			return (spa);
1367b8902de1SWarner Losh 
13682e6bb655SToomas Soome 	return (NULL);
1369b8902de1SWarner Losh }
1370b8902de1SWarner Losh 
1371b8902de1SWarner Losh static spa_t *
1372b8902de1SWarner Losh spa_create(uint64_t guid, const char *name)
1373b8902de1SWarner Losh {
1374b8902de1SWarner Losh 	spa_t *spa;
1375b8902de1SWarner Losh 
137693a2d4c9SToomas Soome 	if ((spa = calloc(1, sizeof(spa_t))) == NULL)
1377b8902de1SWarner Losh 		return (NULL);
1378b8902de1SWarner Losh 	if ((spa->spa_name = strdup(name)) == NULL) {
1379b8902de1SWarner Losh 		free(spa);
1380b8902de1SWarner Losh 		return (NULL);
1381b8902de1SWarner Losh 	}
1382277f38abSMariusz Zaborski 	spa->spa_uberblock = &spa->spa_uberblock_master;
1383277f38abSMariusz Zaborski 	spa->spa_mos = &spa->spa_mos_master;
1384b8902de1SWarner Losh 	spa->spa_guid = guid;
13853c2db0efSToomas Soome 	spa->spa_root_vdev = vdev_create(guid, NULL);
13863c2db0efSToomas Soome 	if (spa->spa_root_vdev == NULL) {
13873c2db0efSToomas Soome 		free(spa->spa_name);
13883c2db0efSToomas Soome 		free(spa);
13893c2db0efSToomas Soome 		return (NULL);
13903c2db0efSToomas Soome 	}
13913c2db0efSToomas Soome 	spa->spa_root_vdev->v_name = strdup("root");
1392b8902de1SWarner Losh 	STAILQ_INSERT_TAIL(&zfs_pools, spa, spa_link);
1393b8902de1SWarner Losh 
1394b8902de1SWarner Losh 	return (spa);
1395b8902de1SWarner Losh }
1396b8902de1SWarner Losh 
1397b8902de1SWarner Losh static const char *
1398b8902de1SWarner Losh state_name(vdev_state_t state)
1399b8902de1SWarner Losh {
1400b8902de1SWarner Losh 	static const char *names[] = {
1401b8902de1SWarner Losh 		"UNKNOWN",
1402b8902de1SWarner Losh 		"CLOSED",
1403b8902de1SWarner Losh 		"OFFLINE",
1404b8902de1SWarner Losh 		"REMOVED",
1405b8902de1SWarner Losh 		"CANT_OPEN",
1406b8902de1SWarner Losh 		"FAULTED",
1407b8902de1SWarner Losh 		"DEGRADED",
1408b8902de1SWarner Losh 		"ONLINE"
1409b8902de1SWarner Losh 	};
14102e6bb655SToomas Soome 	return (names[state]);
1411b8902de1SWarner Losh }
1412b8902de1SWarner Losh 
1413b8902de1SWarner Losh #ifdef BOOT2
1414b8902de1SWarner Losh 
1415b8902de1SWarner Losh #define pager_printf printf
1416b8902de1SWarner Losh 
1417b8902de1SWarner Losh #else
1418b8902de1SWarner Losh 
1419b8902de1SWarner Losh static int
1420b8902de1SWarner Losh pager_printf(const char *fmt, ...)
1421b8902de1SWarner Losh {
1422b8902de1SWarner Losh 	char line[80];
1423b8902de1SWarner Losh 	va_list args;
1424b8902de1SWarner Losh 
1425b8902de1SWarner Losh 	va_start(args, fmt);
14263c2db0efSToomas Soome 	vsnprintf(line, sizeof(line), fmt, args);
1427b8902de1SWarner Losh 	va_end(args);
1428b8902de1SWarner Losh 	return (pager_output(line));
1429b8902de1SWarner Losh }
1430b8902de1SWarner Losh 
1431b8902de1SWarner Losh #endif
1432b8902de1SWarner Losh 
1433b8902de1SWarner Losh #define	STATUS_FORMAT	"        %s %s\n"
1434b8902de1SWarner Losh 
1435b8902de1SWarner Losh static int
1436b8902de1SWarner Losh print_state(int indent, const char *name, vdev_state_t state)
1437b8902de1SWarner Losh {
1438b8902de1SWarner Losh 	int i;
14393c2db0efSToomas Soome 	char buf[512];
1440b8902de1SWarner Losh 
1441b8902de1SWarner Losh 	buf[0] = 0;
1442b8902de1SWarner Losh 	for (i = 0; i < indent; i++)
1443b8902de1SWarner Losh 		strcat(buf, "  ");
1444b8902de1SWarner Losh 	strcat(buf, name);
1445b8902de1SWarner Losh 	return (pager_printf(STATUS_FORMAT, buf, state_name(state)));
1446b8902de1SWarner Losh }
1447b8902de1SWarner Losh 
1448b8902de1SWarner Losh static int
1449b8902de1SWarner Losh vdev_status(vdev_t *vdev, int indent)
1450b8902de1SWarner Losh {
1451b8902de1SWarner Losh 	vdev_t *kid;
1452b8902de1SWarner Losh 	int ret;
14530c0a882cSToomas Soome 
14540c0a882cSToomas Soome 	if (vdev->v_islog) {
14550c0a882cSToomas Soome 		(void) pager_output("        logs\n");
14560c0a882cSToomas Soome 		indent++;
14570c0a882cSToomas Soome 	}
14580c0a882cSToomas Soome 
1459b8902de1SWarner Losh 	ret = print_state(indent, vdev->v_name, vdev->v_state);
1460b8902de1SWarner Losh 	if (ret != 0)
1461b8902de1SWarner Losh 		return (ret);
1462b8902de1SWarner Losh 
1463b8902de1SWarner Losh 	STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
1464b8902de1SWarner Losh 		ret = vdev_status(kid, indent + 1);
1465b8902de1SWarner Losh 		if (ret != 0)
1466b8902de1SWarner Losh 			return (ret);
1467b8902de1SWarner Losh 	}
1468b8902de1SWarner Losh 	return (ret);
1469b8902de1SWarner Losh }
1470b8902de1SWarner Losh 
1471b8902de1SWarner Losh static int
1472b8902de1SWarner Losh spa_status(spa_t *spa)
1473b8902de1SWarner Losh {
1474b8902de1SWarner Losh 	static char bootfs[ZFS_MAXNAMELEN];
1475b8902de1SWarner Losh 	uint64_t rootid;
14763c2db0efSToomas Soome 	vdev_list_t *vlist;
1477b8902de1SWarner Losh 	vdev_t *vdev;
1478b8902de1SWarner Losh 	int good_kids, bad_kids, degraded_kids, ret;
1479b8902de1SWarner Losh 	vdev_state_t state;
1480b8902de1SWarner Losh 
1481b8902de1SWarner Losh 	ret = pager_printf("  pool: %s\n", spa->spa_name);
1482b8902de1SWarner Losh 	if (ret != 0)
1483b8902de1SWarner Losh 		return (ret);
1484b8902de1SWarner Losh 
1485b8902de1SWarner Losh 	if (zfs_get_root(spa, &rootid) == 0 &&
1486b8902de1SWarner Losh 	    zfs_rlookup(spa, rootid, bootfs) == 0) {
1487b8902de1SWarner Losh 		if (bootfs[0] == '\0')
1488b8902de1SWarner Losh 			ret = pager_printf("bootfs: %s\n", spa->spa_name);
1489b8902de1SWarner Losh 		else
1490b8902de1SWarner Losh 			ret = pager_printf("bootfs: %s/%s\n", spa->spa_name,
1491b8902de1SWarner Losh 			    bootfs);
1492b8902de1SWarner Losh 		if (ret != 0)
1493b8902de1SWarner Losh 			return (ret);
1494b8902de1SWarner Losh 	}
1495b8902de1SWarner Losh 	ret = pager_printf("config:\n\n");
1496b8902de1SWarner Losh 	if (ret != 0)
1497b8902de1SWarner Losh 		return (ret);
1498b8902de1SWarner Losh 	ret = pager_printf(STATUS_FORMAT, "NAME", "STATE");
1499b8902de1SWarner Losh 	if (ret != 0)
1500b8902de1SWarner Losh 		return (ret);
1501b8902de1SWarner Losh 
1502b8902de1SWarner Losh 	good_kids = 0;
1503b8902de1SWarner Losh 	degraded_kids = 0;
1504b8902de1SWarner Losh 	bad_kids = 0;
15053c2db0efSToomas Soome 	vlist = &spa->spa_root_vdev->v_children;
15063c2db0efSToomas Soome 	STAILQ_FOREACH(vdev, vlist, v_childlink) {
1507b8902de1SWarner Losh 		if (vdev->v_state == VDEV_STATE_HEALTHY)
1508b8902de1SWarner Losh 			good_kids++;
1509b8902de1SWarner Losh 		else if (vdev->v_state == VDEV_STATE_DEGRADED)
1510b8902de1SWarner Losh 			degraded_kids++;
1511b8902de1SWarner Losh 		else
1512b8902de1SWarner Losh 			bad_kids++;
1513b8902de1SWarner Losh 	}
1514b8902de1SWarner Losh 
1515b8902de1SWarner Losh 	state = VDEV_STATE_CLOSED;
1516b8902de1SWarner Losh 	if (good_kids > 0 && (degraded_kids + bad_kids) == 0)
1517b8902de1SWarner Losh 		state = VDEV_STATE_HEALTHY;
1518b8902de1SWarner Losh 	else if ((good_kids + degraded_kids) > 0)
1519b8902de1SWarner Losh 		state = VDEV_STATE_DEGRADED;
1520b8902de1SWarner Losh 
1521b8902de1SWarner Losh 	ret = print_state(0, spa->spa_name, state);
1522b8902de1SWarner Losh 	if (ret != 0)
1523b8902de1SWarner Losh 		return (ret);
15243c2db0efSToomas Soome 
15253c2db0efSToomas Soome 	STAILQ_FOREACH(vdev, vlist, v_childlink) {
1526b8902de1SWarner Losh 		ret = vdev_status(vdev, 1);
1527b8902de1SWarner Losh 		if (ret != 0)
1528b8902de1SWarner Losh 			return (ret);
1529b8902de1SWarner Losh 	}
1530b8902de1SWarner Losh 	return (ret);
1531b8902de1SWarner Losh }
1532b8902de1SWarner Losh 
1533b8902de1SWarner Losh static int
1534b8902de1SWarner Losh spa_all_status(void)
1535b8902de1SWarner Losh {
1536b8902de1SWarner Losh 	spa_t *spa;
1537b8902de1SWarner Losh 	int first = 1, ret = 0;
1538b8902de1SWarner Losh 
1539b8902de1SWarner Losh 	STAILQ_FOREACH(spa, &zfs_pools, spa_link) {
1540b8902de1SWarner Losh 		if (!first) {
1541b8902de1SWarner Losh 			ret = pager_printf("\n");
1542b8902de1SWarner Losh 			if (ret != 0)
1543b8902de1SWarner Losh 				return (ret);
1544b8902de1SWarner Losh 		}
1545b8902de1SWarner Losh 		first = 0;
1546b8902de1SWarner Losh 		ret = spa_status(spa);
1547b8902de1SWarner Losh 		if (ret != 0)
1548b8902de1SWarner Losh 			return (ret);
1549b8902de1SWarner Losh 	}
1550b8902de1SWarner Losh 	return (ret);
1551b8902de1SWarner Losh }
1552b8902de1SWarner Losh 
1553b8902de1SWarner Losh static uint64_t
1554b8902de1SWarner Losh vdev_label_offset(uint64_t psize, int l, uint64_t offset)
1555b8902de1SWarner Losh {
1556b8902de1SWarner Losh 	uint64_t label_offset;
1557b8902de1SWarner Losh 
1558b8902de1SWarner Losh 	if (l < VDEV_LABELS / 2)
1559b8902de1SWarner Losh 		label_offset = 0;
1560b8902de1SWarner Losh 	else
1561b8902de1SWarner Losh 		label_offset = psize - VDEV_LABELS * sizeof (vdev_label_t);
1562b8902de1SWarner Losh 
1563b8902de1SWarner Losh 	return (offset + l * sizeof (vdev_label_t) + label_offset);
1564b8902de1SWarner Losh }
1565b8902de1SWarner Losh 
1566b8902de1SWarner Losh static int
156779a4bf89SToomas Soome vdev_uberblock_compare(const uberblock_t *ub1, const uberblock_t *ub2)
156879a4bf89SToomas Soome {
156979a4bf89SToomas Soome 	unsigned int seq1 = 0;
157079a4bf89SToomas Soome 	unsigned int seq2 = 0;
157179a4bf89SToomas Soome 	int cmp = AVL_CMP(ub1->ub_txg, ub2->ub_txg);
157279a4bf89SToomas Soome 
157379a4bf89SToomas Soome 	if (cmp != 0)
157479a4bf89SToomas Soome 		return (cmp);
157579a4bf89SToomas Soome 
157679a4bf89SToomas Soome 	cmp = AVL_CMP(ub1->ub_timestamp, ub2->ub_timestamp);
157779a4bf89SToomas Soome 	if (cmp != 0)
157879a4bf89SToomas Soome 		return (cmp);
157979a4bf89SToomas Soome 
158079a4bf89SToomas Soome 	if (MMP_VALID(ub1) && MMP_SEQ_VALID(ub1))
158179a4bf89SToomas Soome 		seq1 = MMP_SEQ(ub1);
158279a4bf89SToomas Soome 
158379a4bf89SToomas Soome 	if (MMP_VALID(ub2) && MMP_SEQ_VALID(ub2))
158479a4bf89SToomas Soome 		seq2 = MMP_SEQ(ub2);
158579a4bf89SToomas Soome 
158679a4bf89SToomas Soome 	return (AVL_CMP(seq1, seq2));
158779a4bf89SToomas Soome }
158879a4bf89SToomas Soome 
158979a4bf89SToomas Soome static int
159079a4bf89SToomas Soome uberblock_verify(uberblock_t *ub)
159179a4bf89SToomas Soome {
159279a4bf89SToomas Soome 	if (ub->ub_magic == BSWAP_64((uint64_t)UBERBLOCK_MAGIC)) {
159379a4bf89SToomas Soome 		byteswap_uint64_array(ub, sizeof (uberblock_t));
159479a4bf89SToomas Soome 	}
159579a4bf89SToomas Soome 
159679a4bf89SToomas Soome 	if (ub->ub_magic != UBERBLOCK_MAGIC ||
159779a4bf89SToomas Soome 	    !SPA_VERSION_IS_SUPPORTED(ub->ub_version))
159879a4bf89SToomas Soome 		return (EINVAL);
159979a4bf89SToomas Soome 
160079a4bf89SToomas Soome 	return (0);
160179a4bf89SToomas Soome }
160279a4bf89SToomas Soome 
160379a4bf89SToomas Soome static int
160479a4bf89SToomas Soome vdev_label_read(vdev_t *vd, int l, void *buf, uint64_t offset,
160579a4bf89SToomas Soome     size_t size)
160679a4bf89SToomas Soome {
160779a4bf89SToomas Soome 	blkptr_t bp;
160879a4bf89SToomas Soome 	off_t off;
160979a4bf89SToomas Soome 
161079a4bf89SToomas Soome 	off = vdev_label_offset(vd->v_psize, l, offset);
161179a4bf89SToomas Soome 
161279a4bf89SToomas Soome 	BP_ZERO(&bp);
161379a4bf89SToomas Soome 	BP_SET_LSIZE(&bp, size);
161479a4bf89SToomas Soome 	BP_SET_PSIZE(&bp, size);
161579a4bf89SToomas Soome 	BP_SET_CHECKSUM(&bp, ZIO_CHECKSUM_LABEL);
161679a4bf89SToomas Soome 	BP_SET_COMPRESS(&bp, ZIO_COMPRESS_OFF);
161779a4bf89SToomas Soome 	DVA_SET_OFFSET(BP_IDENTITY(&bp), off);
161879a4bf89SToomas Soome 	ZIO_SET_CHECKSUM(&bp.blk_cksum, off, 0, 0, 0);
161979a4bf89SToomas Soome 
162079a4bf89SToomas Soome 	return (vdev_read_phys(vd, &bp, buf, off, size));
162179a4bf89SToomas Soome }
162279a4bf89SToomas Soome 
1623e307eb94SToomas Soome /*
1624e307eb94SToomas Soome  * We do need to be sure we write to correct location.
1625e307eb94SToomas Soome  * Our vdev label does consist of 4 fields:
1626e307eb94SToomas Soome  * pad1 (8k), reserved.
1627e307eb94SToomas Soome  * bootenv (8k), checksummed, previously reserved, may contian garbage.
1628e307eb94SToomas Soome  * vdev_phys (112k), checksummed
1629e307eb94SToomas Soome  * uberblock ring (128k), checksummed.
1630e307eb94SToomas Soome  *
1631e307eb94SToomas Soome  * Since bootenv area may contain garbage, we can not reliably read it, as
1632e307eb94SToomas Soome  * we can get checksum errors.
1633e307eb94SToomas Soome  * Next best thing is vdev_phys - it is just after bootenv. It still may
1634e307eb94SToomas Soome  * be corrupted, but in such case we will miss this one write.
1635e307eb94SToomas Soome  */
1636e307eb94SToomas Soome static int
1637e307eb94SToomas Soome vdev_label_write_validate(vdev_t *vd, int l, uint64_t offset)
1638e307eb94SToomas Soome {
1639e307eb94SToomas Soome 	uint64_t off, o_phys;
1640e307eb94SToomas Soome 	void *buf;
1641e307eb94SToomas Soome 	size_t size = VDEV_PHYS_SIZE;
1642e307eb94SToomas Soome 	int rc;
1643e307eb94SToomas Soome 
1644e307eb94SToomas Soome 	o_phys = offsetof(vdev_label_t, vl_vdev_phys);
1645e307eb94SToomas Soome 	off = vdev_label_offset(vd->v_psize, l, o_phys);
1646e307eb94SToomas Soome 
1647e307eb94SToomas Soome 	/* off should be 8K from bootenv */
1648e307eb94SToomas Soome 	if (vdev_label_offset(vd->v_psize, l, offset) + VDEV_PAD_SIZE != off)
1649e307eb94SToomas Soome 		return (EINVAL);
1650e307eb94SToomas Soome 
1651e307eb94SToomas Soome 	buf = malloc(size);
1652e307eb94SToomas Soome 	if (buf == NULL)
1653e307eb94SToomas Soome 		return (ENOMEM);
1654e307eb94SToomas Soome 
1655e307eb94SToomas Soome 	/* Read vdev_phys */
1656e307eb94SToomas Soome 	rc = vdev_label_read(vd, l, buf, o_phys, size);
1657e307eb94SToomas Soome 	free(buf);
1658e307eb94SToomas Soome 	return (rc);
1659e307eb94SToomas Soome }
1660e307eb94SToomas Soome 
1661e307eb94SToomas Soome static int
1662e307eb94SToomas Soome vdev_label_write(vdev_t *vd, int l, vdev_boot_envblock_t *be, uint64_t offset)
1663e307eb94SToomas Soome {
1664e307eb94SToomas Soome 	zio_checksum_info_t *ci;
1665e307eb94SToomas Soome 	zio_cksum_t cksum;
1666e307eb94SToomas Soome 	off_t off;
1667e307eb94SToomas Soome 	size_t size = VDEV_PAD_SIZE;
1668e307eb94SToomas Soome 	int rc;
1669e307eb94SToomas Soome 
1670e307eb94SToomas Soome 	if (vd->v_phys_write == NULL)
1671e307eb94SToomas Soome 		return (ENOTSUP);
1672e307eb94SToomas Soome 
1673e307eb94SToomas Soome 	off = vdev_label_offset(vd->v_psize, l, offset);
1674e307eb94SToomas Soome 
1675e307eb94SToomas Soome 	rc = vdev_label_write_validate(vd, l, offset);
1676e307eb94SToomas Soome 	if (rc != 0) {
1677e307eb94SToomas Soome 		return (rc);
1678e307eb94SToomas Soome 	}
1679e307eb94SToomas Soome 
1680e307eb94SToomas Soome 	ci = &zio_checksum_table[ZIO_CHECKSUM_LABEL];
1681e307eb94SToomas Soome 	be->vbe_zbt.zec_magic = ZEC_MAGIC;
1682e307eb94SToomas Soome 	zio_checksum_label_verifier(&be->vbe_zbt.zec_cksum, off);
1683e307eb94SToomas Soome 	ci->ci_func[0](be, size, NULL, &cksum);
1684e307eb94SToomas Soome 	be->vbe_zbt.zec_cksum = cksum;
1685e307eb94SToomas Soome 
1686e307eb94SToomas Soome 	return (vdev_write_phys(vd, be, off, size));
1687e307eb94SToomas Soome }
1688e307eb94SToomas Soome 
1689e307eb94SToomas Soome static int
1690e307eb94SToomas Soome vdev_write_bootenv_impl(vdev_t *vdev, vdev_boot_envblock_t *be)
1691e307eb94SToomas Soome {
1692e307eb94SToomas Soome 	vdev_t *kid;
1693e307eb94SToomas Soome 	int rv = 0, rc;
1694e307eb94SToomas Soome 
1695e307eb94SToomas Soome 	STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
1696e307eb94SToomas Soome 		if (kid->v_state != VDEV_STATE_HEALTHY)
1697e307eb94SToomas Soome 			continue;
1698e307eb94SToomas Soome 		rc = vdev_write_bootenv_impl(kid, be);
1699e307eb94SToomas Soome 		if (rv == 0)
1700e307eb94SToomas Soome 			rv = rc;
1701e307eb94SToomas Soome 	}
1702e307eb94SToomas Soome 
1703e307eb94SToomas Soome 	/*
1704e307eb94SToomas Soome 	 * Non-leaf vdevs do not have v_phys_write.
1705e307eb94SToomas Soome 	 */
1706e307eb94SToomas Soome 	if (vdev->v_phys_write == NULL)
1707e307eb94SToomas Soome 		return (rv);
1708e307eb94SToomas Soome 
1709e307eb94SToomas Soome 	for (int l = 0; l < VDEV_LABELS; l++) {
1710e307eb94SToomas Soome 		rc = vdev_label_write(vdev, l, be,
1711e307eb94SToomas Soome 		    offsetof(vdev_label_t, vl_be));
1712e307eb94SToomas Soome 		if (rc != 0) {
1713e307eb94SToomas Soome 			printf("failed to write bootenv to %s label %d: %d\n",
1714e307eb94SToomas Soome 			    vdev->v_name ? vdev->v_name : "unknown", l, rc);
1715e307eb94SToomas Soome 			rv = rc;
1716e307eb94SToomas Soome 		}
1717e307eb94SToomas Soome 	}
1718e307eb94SToomas Soome 	return (rv);
1719e307eb94SToomas Soome }
1720e307eb94SToomas Soome 
1721e307eb94SToomas Soome int
1722e307eb94SToomas Soome vdev_write_bootenv(vdev_t *vdev, nvlist_t *nvl)
1723e307eb94SToomas Soome {
1724e307eb94SToomas Soome 	vdev_boot_envblock_t *be;
1725e307eb94SToomas Soome 	nvlist_t nv, *nvp;
1726e307eb94SToomas Soome 	uint64_t version;
1727e307eb94SToomas Soome 	int rv;
1728e307eb94SToomas Soome 
1729e307eb94SToomas Soome 	if (nvl->nv_size > sizeof(be->vbe_bootenv))
1730e307eb94SToomas Soome 		return (E2BIG);
1731e307eb94SToomas Soome 
1732e307eb94SToomas Soome 	version = VB_RAW;
1733e307eb94SToomas Soome 	nvp = vdev_read_bootenv(vdev);
1734e307eb94SToomas Soome 	if (nvp != NULL) {
1735e307eb94SToomas Soome 		nvlist_find(nvp, BOOTENV_VERSION, DATA_TYPE_UINT64, NULL,
1736e307eb94SToomas Soome 		    &version, NULL);
1737e307eb94SToomas Soome 		nvlist_destroy(nvp);
1738e307eb94SToomas Soome 	}
1739e307eb94SToomas Soome 
1740e307eb94SToomas Soome 	be = calloc(1, sizeof(*be));
1741e307eb94SToomas Soome 	if (be == NULL)
1742e307eb94SToomas Soome 		return (ENOMEM);
1743e307eb94SToomas Soome 
1744e307eb94SToomas Soome 	be->vbe_version = version;
1745e307eb94SToomas Soome 	switch (version) {
1746e307eb94SToomas Soome 	case VB_RAW:
1747e307eb94SToomas Soome 		/*
1748e307eb94SToomas Soome 		 * If there is no envmap, we will just wipe bootenv.
1749e307eb94SToomas Soome 		 */
1750e307eb94SToomas Soome 		nvlist_find(nvl, GRUB_ENVMAP, DATA_TYPE_STRING, NULL,
1751e307eb94SToomas Soome 		    be->vbe_bootenv, NULL);
1752e307eb94SToomas Soome 		rv = 0;
1753e307eb94SToomas Soome 		break;
1754e307eb94SToomas Soome 
1755e307eb94SToomas Soome 	case VB_NVLIST:
1756e307eb94SToomas Soome 		nv.nv_header = nvl->nv_header;
1757e307eb94SToomas Soome 		nv.nv_asize = nvl->nv_asize;
1758e307eb94SToomas Soome 		nv.nv_size = nvl->nv_size;
1759e307eb94SToomas Soome 
1760e307eb94SToomas Soome 		bcopy(&nv.nv_header, be->vbe_bootenv, sizeof(nv.nv_header));
1761e307eb94SToomas Soome 		nv.nv_data = be->vbe_bootenv + sizeof(nvs_header_t);
1762e307eb94SToomas Soome 		bcopy(nvl->nv_data, nv.nv_data, nv.nv_size);
1763e307eb94SToomas Soome 		rv = nvlist_export(&nv);
1764e307eb94SToomas Soome 		break;
1765e307eb94SToomas Soome 
1766e307eb94SToomas Soome 	default:
1767e307eb94SToomas Soome 		rv = EINVAL;
1768e307eb94SToomas Soome 		break;
1769e307eb94SToomas Soome 	}
1770e307eb94SToomas Soome 
1771e307eb94SToomas Soome 	if (rv == 0) {
1772e307eb94SToomas Soome 		be->vbe_version = htobe64(be->vbe_version);
1773e307eb94SToomas Soome 		rv = vdev_write_bootenv_impl(vdev, be);
1774e307eb94SToomas Soome 	}
1775e307eb94SToomas Soome 	free(be);
1776e307eb94SToomas Soome 	return (rv);
1777e307eb94SToomas Soome }
1778e307eb94SToomas Soome 
1779e307eb94SToomas Soome /*
1780e307eb94SToomas Soome  * Read the bootenv area from pool label, return the nvlist from it.
1781e307eb94SToomas Soome  * We return from first successful read.
1782e307eb94SToomas Soome  */
1783e307eb94SToomas Soome nvlist_t *
1784e307eb94SToomas Soome vdev_read_bootenv(vdev_t *vdev)
1785e307eb94SToomas Soome {
1786e307eb94SToomas Soome 	vdev_t *kid;
1787e307eb94SToomas Soome 	nvlist_t *benv;
1788e307eb94SToomas Soome 	vdev_boot_envblock_t *be;
1789e307eb94SToomas Soome 	char *command;
1790e307eb94SToomas Soome 	bool ok;
1791e307eb94SToomas Soome 	int rv;
1792e307eb94SToomas Soome 
1793e307eb94SToomas Soome 	STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
1794e307eb94SToomas Soome 		if (kid->v_state != VDEV_STATE_HEALTHY)
1795e307eb94SToomas Soome 			continue;
1796e307eb94SToomas Soome 
1797e307eb94SToomas Soome 		benv = vdev_read_bootenv(kid);
1798e307eb94SToomas Soome 		if (benv != NULL)
1799e307eb94SToomas Soome 			return (benv);
1800e307eb94SToomas Soome 	}
1801e307eb94SToomas Soome 
1802e307eb94SToomas Soome 	be = malloc(sizeof (*be));
1803e307eb94SToomas Soome 	if (be == NULL)
1804e307eb94SToomas Soome 		return (NULL);
1805e307eb94SToomas Soome 
1806e307eb94SToomas Soome 	rv = 0;
1807e307eb94SToomas Soome 	for (int l = 0; l < VDEV_LABELS; l++) {
1808e307eb94SToomas Soome 		rv = vdev_label_read(vdev, l, be,
1809e307eb94SToomas Soome 		    offsetof(vdev_label_t, vl_be),
1810e307eb94SToomas Soome 		    sizeof (*be));
1811e307eb94SToomas Soome 		if (rv == 0)
1812e307eb94SToomas Soome 			break;
1813e307eb94SToomas Soome 	}
1814e307eb94SToomas Soome 	if (rv != 0) {
1815e307eb94SToomas Soome 		free(be);
1816e307eb94SToomas Soome 		return (NULL);
1817e307eb94SToomas Soome 	}
1818e307eb94SToomas Soome 
1819e307eb94SToomas Soome 	be->vbe_version = be64toh(be->vbe_version);
1820e307eb94SToomas Soome 	switch (be->vbe_version) {
1821e307eb94SToomas Soome 	case VB_RAW:
1822e307eb94SToomas Soome 		/*
1823e307eb94SToomas Soome 		 * we have textual data in vbe_bootenv, create nvlist
1824e307eb94SToomas Soome 		 * with key "envmap".
1825e307eb94SToomas Soome 		 */
1826e307eb94SToomas Soome 		benv = nvlist_create(NV_UNIQUE_NAME);
1827e307eb94SToomas Soome 		if (benv != NULL) {
1828e307eb94SToomas Soome 			if (*be->vbe_bootenv == '\0') {
1829e307eb94SToomas Soome 				nvlist_add_uint64(benv, BOOTENV_VERSION,
1830e307eb94SToomas Soome 				    VB_NVLIST);
1831e307eb94SToomas Soome 				break;
1832e307eb94SToomas Soome 			}
1833e307eb94SToomas Soome 			nvlist_add_uint64(benv, BOOTENV_VERSION, VB_RAW);
1834e307eb94SToomas Soome 			be->vbe_bootenv[sizeof (be->vbe_bootenv) - 1] = '\0';
1835e307eb94SToomas Soome 			nvlist_add_string(benv, GRUB_ENVMAP, be->vbe_bootenv);
1836e307eb94SToomas Soome 		}
1837e307eb94SToomas Soome 		break;
1838e307eb94SToomas Soome 
1839e307eb94SToomas Soome 	case VB_NVLIST:
1840e307eb94SToomas Soome 		benv = nvlist_import(be->vbe_bootenv, sizeof(be->vbe_bootenv));
1841e307eb94SToomas Soome 		break;
1842e307eb94SToomas Soome 
1843e307eb94SToomas Soome 	default:
1844e307eb94SToomas Soome 		command = (char *)be;
1845e307eb94SToomas Soome 		ok = false;
1846e307eb94SToomas Soome 
1847e307eb94SToomas Soome 		/* Check for legacy zfsbootcfg command string */
1848e307eb94SToomas Soome 		for (int i = 0; command[i] != '\0'; i++) {
1849e307eb94SToomas Soome 			if (iscntrl(command[i])) {
1850e307eb94SToomas Soome 				ok = false;
1851e307eb94SToomas Soome 				break;
1852e307eb94SToomas Soome 			} else {
1853e307eb94SToomas Soome 				ok = true;
1854e307eb94SToomas Soome 			}
1855e307eb94SToomas Soome 		}
1856e307eb94SToomas Soome 		benv = nvlist_create(NV_UNIQUE_NAME);
1857e307eb94SToomas Soome 		if (benv != NULL) {
1858e307eb94SToomas Soome 			if (ok)
1859e307eb94SToomas Soome 				nvlist_add_string(benv, FREEBSD_BOOTONCE,
1860e307eb94SToomas Soome 				    command);
1861e307eb94SToomas Soome 			else
1862e307eb94SToomas Soome 				nvlist_add_uint64(benv, BOOTENV_VERSION,
1863e307eb94SToomas Soome 				    VB_NVLIST);
1864e307eb94SToomas Soome 		}
1865e307eb94SToomas Soome 		break;
1866e307eb94SToomas Soome 	}
1867e307eb94SToomas Soome 	free(be);
1868e307eb94SToomas Soome 	return (benv);
1869e307eb94SToomas Soome }
1870e307eb94SToomas Soome 
18711a4b982eSToomas Soome static uint64_t
18721a4b982eSToomas Soome vdev_get_label_asize(nvlist_t *nvl)
18731a4b982eSToomas Soome {
18741a4b982eSToomas Soome 	nvlist_t *vdevs;
18751a4b982eSToomas Soome 	uint64_t asize;
18761a4b982eSToomas Soome 	const char *type;
18771a4b982eSToomas Soome 	int len;
18781a4b982eSToomas Soome 
18791a4b982eSToomas Soome 	asize = 0;
18801a4b982eSToomas Soome 	/* Get vdev tree */
18811a4b982eSToomas Soome 	if (nvlist_find(nvl, ZPOOL_CONFIG_VDEV_TREE, DATA_TYPE_NVLIST,
18821a4b982eSToomas Soome 	    NULL, &vdevs, NULL) != 0)
18831a4b982eSToomas Soome 		return (asize);
18841a4b982eSToomas Soome 
18851a4b982eSToomas Soome 	/*
18861a4b982eSToomas Soome 	 * Get vdev type. We will calculate asize for raidz, mirror and disk.
18871a4b982eSToomas Soome 	 * For raidz, the asize is raw size of all children.
18881a4b982eSToomas Soome 	 */
18891a4b982eSToomas Soome 	if (nvlist_find(vdevs, ZPOOL_CONFIG_TYPE, DATA_TYPE_STRING,
18901a4b982eSToomas Soome 	    NULL, &type, &len) != 0)
18911a4b982eSToomas Soome 		goto done;
18921a4b982eSToomas Soome 
18931a4b982eSToomas Soome 	if (memcmp(type, VDEV_TYPE_MIRROR, len) != 0 &&
18941a4b982eSToomas Soome 	    memcmp(type, VDEV_TYPE_DISK, len) != 0 &&
18951a4b982eSToomas Soome 	    memcmp(type, VDEV_TYPE_RAIDZ, len) != 0)
18961a4b982eSToomas Soome 		goto done;
18971a4b982eSToomas Soome 
18981a4b982eSToomas Soome 	if (nvlist_find(vdevs, ZPOOL_CONFIG_ASIZE, DATA_TYPE_UINT64,
18991a4b982eSToomas Soome 	    NULL, &asize, NULL) != 0)
19001a4b982eSToomas Soome 		goto done;
19011a4b982eSToomas Soome 
19021a4b982eSToomas Soome 	if (memcmp(type, VDEV_TYPE_RAIDZ, len) == 0) {
1903e307eb94SToomas Soome 		nvlist_t **kids;
19041a4b982eSToomas Soome 		int nkids;
19051a4b982eSToomas Soome 
19061a4b982eSToomas Soome 		if (nvlist_find(vdevs, ZPOOL_CONFIG_CHILDREN,
19071a4b982eSToomas Soome 		    DATA_TYPE_NVLIST_ARRAY, &nkids, &kids, NULL) != 0) {
19081a4b982eSToomas Soome 			asize = 0;
19091a4b982eSToomas Soome 			goto done;
19101a4b982eSToomas Soome 		}
19111a4b982eSToomas Soome 
19121a4b982eSToomas Soome 		asize /= nkids;
1913e307eb94SToomas Soome 		for (int i = 0; i < nkids; i++)
1914e307eb94SToomas Soome 			nvlist_destroy(kids[i]);
1915e307eb94SToomas Soome 		free(kids);
19161a4b982eSToomas Soome 	}
19171a4b982eSToomas Soome 
19181a4b982eSToomas Soome 	asize += VDEV_LABEL_START_SIZE + VDEV_LABEL_END_SIZE;
19191a4b982eSToomas Soome done:
19201a4b982eSToomas Soome 	nvlist_destroy(vdevs);
19211a4b982eSToomas Soome 	return (asize);
19221a4b982eSToomas Soome }
19231a4b982eSToomas Soome 
19243830659eSToomas Soome static nvlist_t *
192579a4bf89SToomas Soome vdev_label_read_config(vdev_t *vd, uint64_t txg)
192679a4bf89SToomas Soome {
192779a4bf89SToomas Soome 	vdev_phys_t *label;
192879a4bf89SToomas Soome 	uint64_t best_txg = 0;
192979a4bf89SToomas Soome 	uint64_t label_txg = 0;
193079a4bf89SToomas Soome 	uint64_t asize;
19313830659eSToomas Soome 	nvlist_t *nvl = NULL, *tmp;
193279a4bf89SToomas Soome 	int error;
193379a4bf89SToomas Soome 
193479a4bf89SToomas Soome 	label = malloc(sizeof (vdev_phys_t));
193579a4bf89SToomas Soome 	if (label == NULL)
193679a4bf89SToomas Soome 		return (NULL);
193779a4bf89SToomas Soome 
193879a4bf89SToomas Soome 	for (int l = 0; l < VDEV_LABELS; l++) {
193979a4bf89SToomas Soome 		if (vdev_label_read(vd, l, label,
194079a4bf89SToomas Soome 		    offsetof(vdev_label_t, vl_vdev_phys),
194179a4bf89SToomas Soome 		    sizeof (vdev_phys_t)))
194279a4bf89SToomas Soome 			continue;
194379a4bf89SToomas Soome 
1944e307eb94SToomas Soome 		tmp = nvlist_import(label->vp_nvlist,
1945e307eb94SToomas Soome 		    sizeof(label->vp_nvlist));
19463830659eSToomas Soome 		if (tmp == NULL)
194779a4bf89SToomas Soome 			continue;
194879a4bf89SToomas Soome 
19493830659eSToomas Soome 		error = nvlist_find(tmp, ZPOOL_CONFIG_POOL_TXG,
1950c98740b5SToomas Soome 		    DATA_TYPE_UINT64, NULL, &label_txg, NULL);
19519a3ea759SToomas Soome 		if (error != 0 || label_txg == 0) {
19523830659eSToomas Soome 			nvlist_destroy(nvl);
19533830659eSToomas Soome 			nvl = tmp;
19543cac32d1SToomas Soome 			goto done;
19559a3ea759SToomas Soome 		}
195679a4bf89SToomas Soome 
195779a4bf89SToomas Soome 		if (label_txg <= txg && label_txg > best_txg) {
195879a4bf89SToomas Soome 			best_txg = label_txg;
19593830659eSToomas Soome 			nvlist_destroy(nvl);
19603830659eSToomas Soome 			nvl = tmp;
19613830659eSToomas Soome 			tmp = NULL;
196279a4bf89SToomas Soome 
196379a4bf89SToomas Soome 			/*
196479a4bf89SToomas Soome 			 * Use asize from pool config. We need this
196579a4bf89SToomas Soome 			 * because we can get bad value from BIOS.
196679a4bf89SToomas Soome 			 */
19671a4b982eSToomas Soome 			asize = vdev_get_label_asize(nvl);
19681a4b982eSToomas Soome 			if (asize != 0) {
19691a4b982eSToomas Soome 				vd->v_psize = asize;
197079a4bf89SToomas Soome 			}
197179a4bf89SToomas Soome 		}
19723830659eSToomas Soome 		nvlist_destroy(tmp);
197379a4bf89SToomas Soome 	}
197479a4bf89SToomas Soome 
197579a4bf89SToomas Soome 	if (best_txg == 0) {
19763830659eSToomas Soome 		nvlist_destroy(nvl);
197779a4bf89SToomas Soome 		nvl = NULL;
197879a4bf89SToomas Soome 	}
19793cac32d1SToomas Soome done:
19803cac32d1SToomas Soome 	free(label);
198179a4bf89SToomas Soome 	return (nvl);
198279a4bf89SToomas Soome }
198379a4bf89SToomas Soome 
198479a4bf89SToomas Soome static void
198579a4bf89SToomas Soome vdev_uberblock_load(vdev_t *vd, uberblock_t *ub)
198679a4bf89SToomas Soome {
198779a4bf89SToomas Soome 	uberblock_t *buf;
198879a4bf89SToomas Soome 
198979a4bf89SToomas Soome 	buf = malloc(VDEV_UBERBLOCK_SIZE(vd));
199079a4bf89SToomas Soome 	if (buf == NULL)
199179a4bf89SToomas Soome 		return;
199279a4bf89SToomas Soome 
199379a4bf89SToomas Soome 	for (int l = 0; l < VDEV_LABELS; l++) {
199479a4bf89SToomas Soome 		for (int n = 0; n < VDEV_UBERBLOCK_COUNT(vd); n++) {
199579a4bf89SToomas Soome 			if (vdev_label_read(vd, l, buf,
199679a4bf89SToomas Soome 			    VDEV_UBERBLOCK_OFFSET(vd, n),
199779a4bf89SToomas Soome 			    VDEV_UBERBLOCK_SIZE(vd)))
199879a4bf89SToomas Soome 				continue;
199979a4bf89SToomas Soome 			if (uberblock_verify(buf) != 0)
200079a4bf89SToomas Soome 				continue;
200179a4bf89SToomas Soome 
200279a4bf89SToomas Soome 			if (vdev_uberblock_compare(buf, ub) > 0)
200379a4bf89SToomas Soome 				*ub = *buf;
200479a4bf89SToomas Soome 		}
200579a4bf89SToomas Soome 	}
200679a4bf89SToomas Soome 	free(buf);
200779a4bf89SToomas Soome }
200879a4bf89SToomas Soome 
200979a4bf89SToomas Soome static int
2010e307eb94SToomas Soome vdev_probe(vdev_phys_read_t *_read, vdev_phys_write_t *_write, void *priv,
2011e307eb94SToomas Soome     spa_t **spap)
2012b8902de1SWarner Losh {
2013c082b391SToomas Soome 	vdev_t vtmp;
2014b8902de1SWarner Losh 	spa_t *spa;
20153c2db0efSToomas Soome 	vdev_t *vdev;
20163830659eSToomas Soome 	nvlist_t *nvl;
2017b8902de1SWarner Losh 	uint64_t val;
20183c2db0efSToomas Soome 	uint64_t guid, vdev_children;
2019b8902de1SWarner Losh 	uint64_t pool_txg, pool_guid;
2020b8902de1SWarner Losh 	const char *pool_name;
2021c98740b5SToomas Soome 	int rc, namelen;
2022b8902de1SWarner Losh 
2023b8902de1SWarner Losh 	/*
2024b8902de1SWarner Losh 	 * Load the vdev label and figure out which
2025b8902de1SWarner Losh 	 * uberblock is most current.
2026b8902de1SWarner Losh 	 */
2027c082b391SToomas Soome 	memset(&vtmp, 0, sizeof(vtmp));
2028b8902de1SWarner Losh 	vtmp.v_phys_read = _read;
2029e307eb94SToomas Soome 	vtmp.v_phys_write = _write;
2030e307eb94SToomas Soome 	vtmp.v_priv = priv;
2031e307eb94SToomas Soome 	vtmp.v_psize = P2ALIGN(ldi_get_size(priv),
2032b8902de1SWarner Losh 	    (uint64_t)sizeof (vdev_label_t));
2033b8902de1SWarner Losh 
203479a4bf89SToomas Soome 	/* Test for minimum device size. */
2035abca0bd5SToomas Soome 	if (vtmp.v_psize < SPA_MINDEVSIZE)
2036b8902de1SWarner Losh 		return (EIO);
2037b8902de1SWarner Losh 
20383830659eSToomas Soome 	nvl = vdev_label_read_config(&vtmp, UINT64_MAX);
20393830659eSToomas Soome 	if (nvl == NULL)
2040b8902de1SWarner Losh 		return (EIO);
2041b8902de1SWarner Losh 
20423830659eSToomas Soome 	if (nvlist_find(nvl, ZPOOL_CONFIG_VERSION, DATA_TYPE_UINT64,
2043c98740b5SToomas Soome 	    NULL, &val, NULL) != 0) {
20443830659eSToomas Soome 		nvlist_destroy(nvl);
2045b8902de1SWarner Losh 		return (EIO);
2046b8902de1SWarner Losh 	}
2047b8902de1SWarner Losh 
2048b8902de1SWarner Losh 	if (!SPA_VERSION_IS_SUPPORTED(val)) {
2049b8902de1SWarner Losh 		printf("ZFS: unsupported ZFS version %u (should be %u)\n",
2050b8902de1SWarner Losh 		    (unsigned)val, (unsigned)SPA_VERSION);
20513830659eSToomas Soome 		nvlist_destroy(nvl);
2052b8902de1SWarner Losh 		return (EIO);
2053b8902de1SWarner Losh 	}
2054b8902de1SWarner Losh 
2055b8902de1SWarner Losh 	/* Check ZFS features for read */
20563830659eSToomas Soome 	rc = nvlist_check_features_for_read(nvl);
20573830659eSToomas Soome 	if (rc != 0) {
20583830659eSToomas Soome 		nvlist_destroy(nvl);
2059b8902de1SWarner Losh 		return (EIO);
2060b8902de1SWarner Losh 	}
2061b8902de1SWarner Losh 
20623830659eSToomas Soome 	if (nvlist_find(nvl, ZPOOL_CONFIG_POOL_STATE, DATA_TYPE_UINT64,
2063c98740b5SToomas Soome 	    NULL, &val, NULL) != 0) {
20643830659eSToomas Soome 		nvlist_destroy(nvl);
2065b8902de1SWarner Losh 		return (EIO);
2066b8902de1SWarner Losh 	}
2067b8902de1SWarner Losh 
2068b8902de1SWarner Losh 	if (val == POOL_STATE_DESTROYED) {
2069b8902de1SWarner Losh 		/* We don't boot only from destroyed pools. */
20703830659eSToomas Soome 		nvlist_destroy(nvl);
2071b8902de1SWarner Losh 		return (EIO);
2072b8902de1SWarner Losh 	}
2073b8902de1SWarner Losh 
20743830659eSToomas Soome 	if (nvlist_find(nvl, ZPOOL_CONFIG_POOL_TXG, DATA_TYPE_UINT64,
2075c98740b5SToomas Soome 	    NULL, &pool_txg, NULL) != 0 ||
20763830659eSToomas Soome 	    nvlist_find(nvl, ZPOOL_CONFIG_POOL_GUID, DATA_TYPE_UINT64,
2077c98740b5SToomas Soome 	    NULL, &pool_guid, NULL) != 0 ||
20783830659eSToomas Soome 	    nvlist_find(nvl, ZPOOL_CONFIG_POOL_NAME, DATA_TYPE_STRING,
2079c98740b5SToomas Soome 	    NULL, &pool_name, &namelen) != 0) {
2080b8902de1SWarner Losh 		/*
2081b8902de1SWarner Losh 		 * Cache and spare devices end up here - just ignore
2082b8902de1SWarner Losh 		 * them.
2083b8902de1SWarner Losh 		 */
20843830659eSToomas Soome 		nvlist_destroy(nvl);
2085b8902de1SWarner Losh 		return (EIO);
2086b8902de1SWarner Losh 	}
2087b8902de1SWarner Losh 
2088b8902de1SWarner Losh 	/*
2089b8902de1SWarner Losh 	 * Create the pool if this is the first time we've seen it.
2090b8902de1SWarner Losh 	 */
2091b8902de1SWarner Losh 	spa = spa_find_by_guid(pool_guid);
2092b8902de1SWarner Losh 	if (spa == NULL) {
2093c98740b5SToomas Soome 		char *name;
2094c98740b5SToomas Soome 
20953830659eSToomas Soome 		nvlist_find(nvl, ZPOOL_CONFIG_VDEV_CHILDREN,
2096c98740b5SToomas Soome 		    DATA_TYPE_UINT64, NULL, &vdev_children, NULL);
2097c98740b5SToomas Soome 		name = malloc(namelen + 1);
2098c98740b5SToomas Soome 		if (name == NULL) {
20993830659eSToomas Soome 			nvlist_destroy(nvl);
2100c98740b5SToomas Soome 			return (ENOMEM);
2101c98740b5SToomas Soome 		}
2102c98740b5SToomas Soome 		bcopy(pool_name, name, namelen);
2103c98740b5SToomas Soome 		name[namelen] = '\0';
2104c98740b5SToomas Soome 		spa = spa_create(pool_guid, name);
2105c98740b5SToomas Soome 		free(name);
210679a4bf89SToomas Soome 		if (spa == NULL) {
21073830659eSToomas Soome 			nvlist_destroy(nvl);
2108b8902de1SWarner Losh 			return (ENOMEM);
2109b8902de1SWarner Losh 		}
21103c2db0efSToomas Soome 		spa->spa_root_vdev->v_nchildren = vdev_children;
211179a4bf89SToomas Soome 	}
21123c2db0efSToomas Soome 	if (pool_txg > spa->spa_txg)
2113b8902de1SWarner Losh 		spa->spa_txg = pool_txg;
2114b8902de1SWarner Losh 
2115b8902de1SWarner Losh 	/*
2116b8902de1SWarner Losh 	 * Get the vdev tree and create our in-core copy of it.
2117b8902de1SWarner Losh 	 * If we already have a vdev with this guid, this must
2118b8902de1SWarner Losh 	 * be some kind of alias (overlapping slices, dangerously dedicated
2119b8902de1SWarner Losh 	 * disks etc).
2120b8902de1SWarner Losh 	 */
21213830659eSToomas Soome 	if (nvlist_find(nvl, ZPOOL_CONFIG_GUID, DATA_TYPE_UINT64,
2122c98740b5SToomas Soome 	    NULL, &guid, NULL) != 0) {
21233830659eSToomas Soome 		nvlist_destroy(nvl);
2124b8902de1SWarner Losh 		return (EIO);
2125b8902de1SWarner Losh 	}
2126b8902de1SWarner Losh 	vdev = vdev_find(guid);
212779a4bf89SToomas Soome 	/* Has this vdev already been inited? */
212879a4bf89SToomas Soome 	if (vdev && vdev->v_phys_read) {
21293830659eSToomas Soome 		nvlist_destroy(nvl);
2130b8902de1SWarner Losh 		return (EIO);
213179a4bf89SToomas Soome 	}
2132b8902de1SWarner Losh 
21333830659eSToomas Soome 	rc = vdev_init_from_label(spa, nvl);
21343830659eSToomas Soome 	nvlist_destroy(nvl);
2135b8902de1SWarner Losh 	if (rc != 0)
2136b8902de1SWarner Losh 		return (rc);
2137b8902de1SWarner Losh 
2138b8902de1SWarner Losh 	/*
2139b8902de1SWarner Losh 	 * We should already have created an incomplete vdev for this
2140b8902de1SWarner Losh 	 * vdev. Find it and initialise it with our read proc.
2141b8902de1SWarner Losh 	 */
2142b8902de1SWarner Losh 	vdev = vdev_find(guid);
21433c2db0efSToomas Soome 	if (vdev != NULL) {
2144b8902de1SWarner Losh 		vdev->v_phys_read = _read;
2145e307eb94SToomas Soome 		vdev->v_phys_write = _write;
2146e307eb94SToomas Soome 		vdev->v_priv = priv;
2147abca0bd5SToomas Soome 		vdev->v_psize = vtmp.v_psize;
21483c2db0efSToomas Soome 		/*
21493c2db0efSToomas Soome 		 * If no other state is set, mark vdev healthy.
21503c2db0efSToomas Soome 		 */
21513c2db0efSToomas Soome 		if (vdev->v_state == VDEV_STATE_UNKNOWN)
21523c2db0efSToomas Soome 			vdev->v_state = VDEV_STATE_HEALTHY;
2153b8902de1SWarner Losh 	} else {
2154b8902de1SWarner Losh 		printf("ZFS: inconsistent nvlist contents\n");
2155b8902de1SWarner Losh 		return (EIO);
2156b8902de1SWarner Losh 	}
2157b8902de1SWarner Losh 
21580c0a882cSToomas Soome 	if (vdev->v_islog)
21590c0a882cSToomas Soome 		spa->spa_with_log = vdev->v_islog;
21600c0a882cSToomas Soome 
21610c0a882cSToomas Soome 	/*
2162b8902de1SWarner Losh 	 * Re-evaluate top-level vdev state.
2163b8902de1SWarner Losh 	 */
21643c2db0efSToomas Soome 	vdev_set_state(vdev->v_top);
2165b8902de1SWarner Losh 
2166b8902de1SWarner Losh 	/*
2167b8902de1SWarner Losh 	 * Ok, we are happy with the pool so far. Lets find
2168b8902de1SWarner Losh 	 * the best uberblock and then we can actually access
2169b8902de1SWarner Losh 	 * the contents of the pool.
2170b8902de1SWarner Losh 	 */
2171277f38abSMariusz Zaborski 	vdev_uberblock_load(vdev, spa->spa_uberblock);
2172b8902de1SWarner Losh 
2173b8902de1SWarner Losh 	if (spap != NULL)
2174b8902de1SWarner Losh 		*spap = spa;
2175b8902de1SWarner Losh 	return (0);
2176b8902de1SWarner Losh }
2177b8902de1SWarner Losh 
2178b8902de1SWarner Losh static int
2179b8902de1SWarner Losh ilog2(int n)
2180b8902de1SWarner Losh {
2181b8902de1SWarner Losh 	int v;
2182b8902de1SWarner Losh 
2183b8902de1SWarner Losh 	for (v = 0; v < 32; v++)
2184b8902de1SWarner Losh 		if (n == (1 << v))
21852e6bb655SToomas Soome 			return (v);
21862e6bb655SToomas Soome 	return (-1);
2187b8902de1SWarner Losh }
2188b8902de1SWarner Losh 
2189b8902de1SWarner Losh static int
2190b8902de1SWarner Losh zio_read_gang(const spa_t *spa, const blkptr_t *bp, void *buf)
2191b8902de1SWarner Losh {
2192b8902de1SWarner Losh 	blkptr_t gbh_bp;
2193b8902de1SWarner Losh 	zio_gbh_phys_t zio_gb;
2194b8902de1SWarner Losh 	char *pbuf;
2195b8902de1SWarner Losh 	int i;
2196b8902de1SWarner Losh 
2197b8902de1SWarner Losh 	/* Artificial BP for gang block header. */
2198b8902de1SWarner Losh 	gbh_bp = *bp;
2199b8902de1SWarner Losh 	BP_SET_PSIZE(&gbh_bp, SPA_GANGBLOCKSIZE);
2200b8902de1SWarner Losh 	BP_SET_LSIZE(&gbh_bp, SPA_GANGBLOCKSIZE);
2201b8902de1SWarner Losh 	BP_SET_CHECKSUM(&gbh_bp, ZIO_CHECKSUM_GANG_HEADER);
2202b8902de1SWarner Losh 	BP_SET_COMPRESS(&gbh_bp, ZIO_COMPRESS_OFF);
2203b8902de1SWarner Losh 	for (i = 0; i < SPA_DVAS_PER_BP; i++)
2204b8902de1SWarner Losh 		DVA_SET_GANG(&gbh_bp.blk_dva[i], 0);
2205b8902de1SWarner Losh 
2206b8902de1SWarner Losh 	/* Read gang header block using the artificial BP. */
2207b8902de1SWarner Losh 	if (zio_read(spa, &gbh_bp, &zio_gb))
2208b8902de1SWarner Losh 		return (EIO);
2209b8902de1SWarner Losh 
2210b8902de1SWarner Losh 	pbuf = buf;
2211b8902de1SWarner Losh 	for (i = 0; i < SPA_GBH_NBLKPTRS; i++) {
2212b8902de1SWarner Losh 		blkptr_t *gbp = &zio_gb.zg_blkptr[i];
2213b8902de1SWarner Losh 
2214b8902de1SWarner Losh 		if (BP_IS_HOLE(gbp))
2215b8902de1SWarner Losh 			continue;
2216b8902de1SWarner Losh 		if (zio_read(spa, gbp, pbuf))
2217b8902de1SWarner Losh 			return (EIO);
2218b8902de1SWarner Losh 		pbuf += BP_GET_PSIZE(gbp);
2219b8902de1SWarner Losh 	}
2220b8902de1SWarner Losh 
2221b8902de1SWarner Losh 	if (zio_checksum_verify(spa, bp, buf))
2222b8902de1SWarner Losh 		return (EIO);
2223b8902de1SWarner Losh 	return (0);
2224b8902de1SWarner Losh }
2225b8902de1SWarner Losh 
2226b8902de1SWarner Losh static int
2227b8902de1SWarner Losh zio_read(const spa_t *spa, const blkptr_t *bp, void *buf)
2228b8902de1SWarner Losh {
2229b8902de1SWarner Losh 	int cpfunc = BP_GET_COMPRESS(bp);
2230b8902de1SWarner Losh 	uint64_t align, size;
2231b8902de1SWarner Losh 	void *pbuf;
2232b8902de1SWarner Losh 	int i, error;
2233b8902de1SWarner Losh 
2234b8902de1SWarner Losh 	/*
2235b8902de1SWarner Losh 	 * Process data embedded in block pointer
2236b8902de1SWarner Losh 	 */
2237b8902de1SWarner Losh 	if (BP_IS_EMBEDDED(bp)) {
2238b8902de1SWarner Losh 		ASSERT(BPE_GET_ETYPE(bp) == BP_EMBEDDED_TYPE_DATA);
2239b8902de1SWarner Losh 
2240b8902de1SWarner Losh 		size = BPE_GET_PSIZE(bp);
2241b8902de1SWarner Losh 		ASSERT(size <= BPE_PAYLOAD_SIZE);
2242b8902de1SWarner Losh 
2243b8902de1SWarner Losh 		if (cpfunc != ZIO_COMPRESS_OFF)
2244c1c4c81fSToomas Soome 			pbuf = malloc(size);
2245b8902de1SWarner Losh 		else
2246b8902de1SWarner Losh 			pbuf = buf;
2247b8902de1SWarner Losh 
2248c1c4c81fSToomas Soome 		if (pbuf == NULL)
2249c1c4c81fSToomas Soome 			return (ENOMEM);
2250c1c4c81fSToomas Soome 
2251b8902de1SWarner Losh 		decode_embedded_bp_compressed(bp, pbuf);
2252b8902de1SWarner Losh 		error = 0;
2253b8902de1SWarner Losh 
2254b8902de1SWarner Losh 		if (cpfunc != ZIO_COMPRESS_OFF) {
2255b8902de1SWarner Losh 			error = zio_decompress_data(cpfunc, pbuf,
2256b8902de1SWarner Losh 			    size, buf, BP_GET_LSIZE(bp));
2257c1c4c81fSToomas Soome 			free(pbuf);
2258b8902de1SWarner Losh 		}
2259b8902de1SWarner Losh 		if (error != 0)
22602e6bb655SToomas Soome 			printf("ZFS: i/o error - unable to decompress "
22612e6bb655SToomas Soome 			    "block pointer data, error %d\n", error);
2262b8902de1SWarner Losh 		return (error);
2263b8902de1SWarner Losh 	}
2264b8902de1SWarner Losh 
2265b8902de1SWarner Losh 	error = EIO;
2266b8902de1SWarner Losh 
2267b8902de1SWarner Losh 	for (i = 0; i < SPA_DVAS_PER_BP; i++) {
2268b8902de1SWarner Losh 		const dva_t *dva = &bp->blk_dva[i];
2269b8902de1SWarner Losh 		vdev_t *vdev;
22703c2db0efSToomas Soome 		vdev_list_t *vlist;
22713c2db0efSToomas Soome 		uint64_t vdevid;
2272b8902de1SWarner Losh 		off_t offset;
2273b8902de1SWarner Losh 
2274b8902de1SWarner Losh 		if (!dva->dva_word[0] && !dva->dva_word[1])
2275b8902de1SWarner Losh 			continue;
2276b8902de1SWarner Losh 
2277b8902de1SWarner Losh 		vdevid = DVA_GET_VDEV(dva);
2278b8902de1SWarner Losh 		offset = DVA_GET_OFFSET(dva);
22793c2db0efSToomas Soome 		vlist = &spa->spa_root_vdev->v_children;
22803c2db0efSToomas Soome 		STAILQ_FOREACH(vdev, vlist, v_childlink) {
2281b8902de1SWarner Losh 			if (vdev->v_id == vdevid)
2282b8902de1SWarner Losh 				break;
2283b8902de1SWarner Losh 		}
2284b8902de1SWarner Losh 		if (!vdev || !vdev->v_read)
2285b8902de1SWarner Losh 			continue;
2286b8902de1SWarner Losh 
2287b8902de1SWarner Losh 		size = BP_GET_PSIZE(bp);
2288b8902de1SWarner Losh 		if (vdev->v_read == vdev_raidz_read) {
22893c2db0efSToomas Soome 			align = 1ULL << vdev->v_ashift;
2290b8902de1SWarner Losh 			if (P2PHASE(size, align) != 0)
2291b8902de1SWarner Losh 				size = P2ROUNDUP(size, align);
2292b8902de1SWarner Losh 		}
2293b8902de1SWarner Losh 		if (size != BP_GET_PSIZE(bp) || cpfunc != ZIO_COMPRESS_OFF)
2294c1c4c81fSToomas Soome 			pbuf = malloc(size);
2295b8902de1SWarner Losh 		else
2296b8902de1SWarner Losh 			pbuf = buf;
2297b8902de1SWarner Losh 
2298c1c4c81fSToomas Soome 		if (pbuf == NULL) {
2299c1c4c81fSToomas Soome 			error = ENOMEM;
2300c1c4c81fSToomas Soome 			break;
2301c1c4c81fSToomas Soome 		}
2302c1c4c81fSToomas Soome 
2303b8902de1SWarner Losh 		if (DVA_GET_GANG(dva))
2304b8902de1SWarner Losh 			error = zio_read_gang(spa, bp, pbuf);
2305b8902de1SWarner Losh 		else
2306b8902de1SWarner Losh 			error = vdev->v_read(vdev, bp, pbuf, offset, size);
2307b8902de1SWarner Losh 		if (error == 0) {
2308b8902de1SWarner Losh 			if (cpfunc != ZIO_COMPRESS_OFF)
2309b8902de1SWarner Losh 				error = zio_decompress_data(cpfunc, pbuf,
2310b8902de1SWarner Losh 				    BP_GET_PSIZE(bp), buf, BP_GET_LSIZE(bp));
2311b8902de1SWarner Losh 			else if (size != BP_GET_PSIZE(bp))
2312b8902de1SWarner Losh 				bcopy(pbuf, buf, BP_GET_PSIZE(bp));
23133830659eSToomas Soome 		} else {
23143830659eSToomas Soome 			printf("zio_read error: %d\n", error);
2315b8902de1SWarner Losh 		}
2316b8902de1SWarner Losh 		if (buf != pbuf)
2317c1c4c81fSToomas Soome 			free(pbuf);
2318b8902de1SWarner Losh 		if (error == 0)
2319b8902de1SWarner Losh 			break;
2320b8902de1SWarner Losh 	}
2321b8902de1SWarner Losh 	if (error != 0)
2322b8902de1SWarner Losh 		printf("ZFS: i/o error - all block copies unavailable\n");
2323c1c4c81fSToomas Soome 
2324b8902de1SWarner Losh 	return (error);
2325b8902de1SWarner Losh }
2326b8902de1SWarner Losh 
2327b8902de1SWarner Losh static int
23282e6bb655SToomas Soome dnode_read(const spa_t *spa, const dnode_phys_t *dnode, off_t offset,
23292e6bb655SToomas Soome     void *buf, size_t buflen)
2330b8902de1SWarner Losh {
2331b8902de1SWarner Losh 	int ibshift = dnode->dn_indblkshift - SPA_BLKPTRSHIFT;
2332b8902de1SWarner Losh 	int bsize = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2333b8902de1SWarner Losh 	int nlevels = dnode->dn_nlevels;
2334b8902de1SWarner Losh 	int i, rc;
2335b8902de1SWarner Losh 
2336b8902de1SWarner Losh 	if (bsize > SPA_MAXBLOCKSIZE) {
2337b8902de1SWarner Losh 		printf("ZFS: I/O error - blocks larger than %llu are not "
2338b8902de1SWarner Losh 		    "supported\n", SPA_MAXBLOCKSIZE);
2339b8902de1SWarner Losh 		return (EIO);
2340b8902de1SWarner Losh 	}
2341b8902de1SWarner Losh 
2342b8902de1SWarner Losh 	/*
2343914dc91dSKyle Evans 	 * Handle odd block sizes, mirrors dmu_read_impl().  Data can't exist
2344914dc91dSKyle Evans 	 * past the first block, so we'll clip the read to the portion of the
2345914dc91dSKyle Evans 	 * buffer within bsize and zero out the remainder.
2346914dc91dSKyle Evans 	 */
2347914dc91dSKyle Evans 	if (dnode->dn_maxblkid == 0) {
2348914dc91dSKyle Evans 		size_t newbuflen;
2349914dc91dSKyle Evans 
2350914dc91dSKyle Evans 		newbuflen = offset > bsize ? 0 : MIN(buflen, bsize - offset);
2351914dc91dSKyle Evans 		bzero((char *)buf + newbuflen, buflen - newbuflen);
2352914dc91dSKyle Evans 		buflen = newbuflen;
2353914dc91dSKyle Evans 	}
2354914dc91dSKyle Evans 
2355914dc91dSKyle Evans 	/*
2356b8902de1SWarner Losh 	 * Note: bsize may not be a power of two here so we need to do an
2357b8902de1SWarner Losh 	 * actual divide rather than a bitshift.
2358b8902de1SWarner Losh 	 */
2359b8902de1SWarner Losh 	while (buflen > 0) {
2360b8902de1SWarner Losh 		uint64_t bn = offset / bsize;
2361b8902de1SWarner Losh 		int boff = offset % bsize;
2362b8902de1SWarner Losh 		int ibn;
2363b8902de1SWarner Losh 		const blkptr_t *indbp;
2364b8902de1SWarner Losh 		blkptr_t bp;
2365b8902de1SWarner Losh 
2366b8902de1SWarner Losh 		if (bn > dnode->dn_maxblkid)
2367b8902de1SWarner Losh 			return (EIO);
2368b8902de1SWarner Losh 
2369b8902de1SWarner Losh 		if (dnode == dnode_cache_obj && bn == dnode_cache_bn)
2370b8902de1SWarner Losh 			goto cached;
2371b8902de1SWarner Losh 
2372b8902de1SWarner Losh 		indbp = dnode->dn_blkptr;
2373b8902de1SWarner Losh 		for (i = 0; i < nlevels; i++) {
2374b8902de1SWarner Losh 			/*
2375b8902de1SWarner Losh 			 * Copy the bp from the indirect array so that
2376b8902de1SWarner Losh 			 * we can re-use the scratch buffer for multi-level
2377b8902de1SWarner Losh 			 * objects.
2378b8902de1SWarner Losh 			 */
2379b8902de1SWarner Losh 			ibn = bn >> ((nlevels - i - 1) * ibshift);
2380b8902de1SWarner Losh 			ibn &= ((1 << ibshift) - 1);
2381b8902de1SWarner Losh 			bp = indbp[ibn];
2382b8902de1SWarner Losh 			if (BP_IS_HOLE(&bp)) {
2383b8902de1SWarner Losh 				memset(dnode_cache_buf, 0, bsize);
2384b8902de1SWarner Losh 				break;
2385b8902de1SWarner Losh 			}
2386b8902de1SWarner Losh 			rc = zio_read(spa, &bp, dnode_cache_buf);
2387b8902de1SWarner Losh 			if (rc)
2388b8902de1SWarner Losh 				return (rc);
2389b8902de1SWarner Losh 			indbp = (const blkptr_t *) dnode_cache_buf;
2390b8902de1SWarner Losh 		}
2391b8902de1SWarner Losh 		dnode_cache_obj = dnode;
2392b8902de1SWarner Losh 		dnode_cache_bn = bn;
2393b8902de1SWarner Losh 	cached:
2394b8902de1SWarner Losh 
2395b8902de1SWarner Losh 		/*
2396b8902de1SWarner Losh 		 * The buffer contains our data block. Copy what we
2397b8902de1SWarner Losh 		 * need from it and loop.
2398b8902de1SWarner Losh 		 */
2399b8902de1SWarner Losh 		i = bsize - boff;
2400b8902de1SWarner Losh 		if (i > buflen) i = buflen;
2401b8902de1SWarner Losh 		memcpy(buf, &dnode_cache_buf[boff], i);
2402b8902de1SWarner Losh 		buf = ((char *)buf) + i;
2403b8902de1SWarner Losh 		offset += i;
2404b8902de1SWarner Losh 		buflen -= i;
2405b8902de1SWarner Losh 	}
2406b8902de1SWarner Losh 
2407b8902de1SWarner Losh 	return (0);
2408b8902de1SWarner Losh }
2409b8902de1SWarner Losh 
2410b8902de1SWarner Losh /*
2411380977d5SToomas Soome  * Lookup a value in a microzap directory.
2412b8902de1SWarner Losh  */
2413b8902de1SWarner Losh static int
24144d297e70SToomas Soome mzap_lookup(const mzap_phys_t *mz, size_t size, const char *name,
24154d297e70SToomas Soome     uint64_t *value)
2416b8902de1SWarner Losh {
2417b8902de1SWarner Losh 	const mzap_ent_phys_t *mze;
2418b8902de1SWarner Losh 	int chunks, i;
2419b8902de1SWarner Losh 
2420b8902de1SWarner Losh 	/*
2421b8902de1SWarner Losh 	 * Microzap objects use exactly one block. Read the whole
2422b8902de1SWarner Losh 	 * thing.
2423b8902de1SWarner Losh 	 */
2424b8902de1SWarner Losh 	chunks = size / MZAP_ENT_LEN - 1;
2425b8902de1SWarner Losh 	for (i = 0; i < chunks; i++) {
2426b8902de1SWarner Losh 		mze = &mz->mz_chunk[i];
24272e6bb655SToomas Soome 		if (strcmp(mze->mze_name, name) == 0) {
2428b8902de1SWarner Losh 			*value = mze->mze_value;
2429b8902de1SWarner Losh 			return (0);
2430b8902de1SWarner Losh 		}
2431b8902de1SWarner Losh 	}
2432b8902de1SWarner Losh 
2433b8902de1SWarner Losh 	return (ENOENT);
2434b8902de1SWarner Losh }
2435b8902de1SWarner Losh 
2436b8902de1SWarner Losh /*
2437b8902de1SWarner Losh  * Compare a name with a zap leaf entry. Return non-zero if the name
2438b8902de1SWarner Losh  * matches.
2439b8902de1SWarner Losh  */
2440b8902de1SWarner Losh static int
24412e6bb655SToomas Soome fzap_name_equal(const zap_leaf_t *zl, const zap_leaf_chunk_t *zc,
24422e6bb655SToomas Soome     const char *name)
2443b8902de1SWarner Losh {
2444b8902de1SWarner Losh 	size_t namelen;
2445b8902de1SWarner Losh 	const zap_leaf_chunk_t *nc;
2446b8902de1SWarner Losh 	const char *p;
2447b8902de1SWarner Losh 
2448b8902de1SWarner Losh 	namelen = zc->l_entry.le_name_numints;
2449b8902de1SWarner Losh 
2450b8902de1SWarner Losh 	nc = &ZAP_LEAF_CHUNK(zl, zc->l_entry.le_name_chunk);
2451b8902de1SWarner Losh 	p = name;
2452b8902de1SWarner Losh 	while (namelen > 0) {
2453b8902de1SWarner Losh 		size_t len;
24542e6bb655SToomas Soome 
2455b8902de1SWarner Losh 		len = namelen;
2456b8902de1SWarner Losh 		if (len > ZAP_LEAF_ARRAY_BYTES)
2457b8902de1SWarner Losh 			len = ZAP_LEAF_ARRAY_BYTES;
2458b8902de1SWarner Losh 		if (memcmp(p, nc->l_array.la_array, len))
2459b8902de1SWarner Losh 			return (0);
2460b8902de1SWarner Losh 		p += len;
2461b8902de1SWarner Losh 		namelen -= len;
2462b8902de1SWarner Losh 		nc = &ZAP_LEAF_CHUNK(zl, nc->l_array.la_next);
2463b8902de1SWarner Losh 	}
2464b8902de1SWarner Losh 
24652e6bb655SToomas Soome 	return (1);
2466b8902de1SWarner Losh }
2467b8902de1SWarner Losh 
2468b8902de1SWarner Losh /*
2469b8902de1SWarner Losh  * Extract a uint64_t value from a zap leaf entry.
2470b8902de1SWarner Losh  */
2471b8902de1SWarner Losh static uint64_t
2472b8902de1SWarner Losh fzap_leaf_value(const zap_leaf_t *zl, const zap_leaf_chunk_t *zc)
2473b8902de1SWarner Losh {
2474b8902de1SWarner Losh 	const zap_leaf_chunk_t *vc;
2475b8902de1SWarner Losh 	int i;
2476b8902de1SWarner Losh 	uint64_t value;
2477b8902de1SWarner Losh 	const uint8_t *p;
2478b8902de1SWarner Losh 
2479b8902de1SWarner Losh 	vc = &ZAP_LEAF_CHUNK(zl, zc->l_entry.le_value_chunk);
2480b8902de1SWarner Losh 	for (i = 0, value = 0, p = vc->l_array.la_array; i < 8; i++) {
2481b8902de1SWarner Losh 		value = (value << 8) | p[i];
2482b8902de1SWarner Losh 	}
2483b8902de1SWarner Losh 
24842e6bb655SToomas Soome 	return (value);
2485b8902de1SWarner Losh }
2486b8902de1SWarner Losh 
2487b8902de1SWarner Losh static void
2488b8902de1SWarner Losh stv(int len, void *addr, uint64_t value)
2489b8902de1SWarner Losh {
2490b8902de1SWarner Losh 	switch (len) {
2491b8902de1SWarner Losh 	case 1:
2492b8902de1SWarner Losh 		*(uint8_t *)addr = value;
2493b8902de1SWarner Losh 		return;
2494b8902de1SWarner Losh 	case 2:
2495b8902de1SWarner Losh 		*(uint16_t *)addr = value;
2496b8902de1SWarner Losh 		return;
2497b8902de1SWarner Losh 	case 4:
2498b8902de1SWarner Losh 		*(uint32_t *)addr = value;
2499b8902de1SWarner Losh 		return;
2500b8902de1SWarner Losh 	case 8:
2501b8902de1SWarner Losh 		*(uint64_t *)addr = value;
2502b8902de1SWarner Losh 		return;
2503b8902de1SWarner Losh 	}
2504b8902de1SWarner Losh }
2505b8902de1SWarner Losh 
2506b8902de1SWarner Losh /*
2507b8902de1SWarner Losh  * Extract a array from a zap leaf entry.
2508b8902de1SWarner Losh  */
2509b8902de1SWarner Losh static void
2510b8902de1SWarner Losh fzap_leaf_array(const zap_leaf_t *zl, const zap_leaf_chunk_t *zc,
2511b8902de1SWarner Losh     uint64_t integer_size, uint64_t num_integers, void *buf)
2512b8902de1SWarner Losh {
2513b8902de1SWarner Losh 	uint64_t array_int_len = zc->l_entry.le_value_intlen;
2514b8902de1SWarner Losh 	uint64_t value = 0;
2515b8902de1SWarner Losh 	uint64_t *u64 = buf;
2516b8902de1SWarner Losh 	char *p = buf;
2517b8902de1SWarner Losh 	int len = MIN(zc->l_entry.le_value_numints, num_integers);
2518b8902de1SWarner Losh 	int chunk = zc->l_entry.le_value_chunk;
2519b8902de1SWarner Losh 	int byten = 0;
2520b8902de1SWarner Losh 
2521b8902de1SWarner Losh 	if (integer_size == 8 && len == 1) {
2522b8902de1SWarner Losh 		*u64 = fzap_leaf_value(zl, zc);
2523b8902de1SWarner Losh 		return;
2524b8902de1SWarner Losh 	}
2525b8902de1SWarner Losh 
2526b8902de1SWarner Losh 	while (len > 0) {
2527b8902de1SWarner Losh 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(zl, chunk).l_array;
2528b8902de1SWarner Losh 		int i;
2529b8902de1SWarner Losh 
2530b8902de1SWarner Losh 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(zl));
2531b8902de1SWarner Losh 		for (i = 0; i < ZAP_LEAF_ARRAY_BYTES && len > 0; i++) {
2532b8902de1SWarner Losh 			value = (value << 8) | la->la_array[i];
2533b8902de1SWarner Losh 			byten++;
2534b8902de1SWarner Losh 			if (byten == array_int_len) {
2535b8902de1SWarner Losh 				stv(integer_size, p, value);
2536b8902de1SWarner Losh 				byten = 0;
2537b8902de1SWarner Losh 				len--;
2538b8902de1SWarner Losh 				if (len == 0)
2539b8902de1SWarner Losh 					return;
2540b8902de1SWarner Losh 				p += integer_size;
2541b8902de1SWarner Losh 			}
2542b8902de1SWarner Losh 		}
2543b8902de1SWarner Losh 		chunk = la->la_next;
2544b8902de1SWarner Losh 	}
2545b8902de1SWarner Losh }
2546b8902de1SWarner Losh 
25473c2db0efSToomas Soome static int
25483c2db0efSToomas Soome fzap_check_size(uint64_t integer_size, uint64_t num_integers)
25493c2db0efSToomas Soome {
25503c2db0efSToomas Soome 
25513c2db0efSToomas Soome 	switch (integer_size) {
25523c2db0efSToomas Soome 	case 1:
25533c2db0efSToomas Soome 	case 2:
25543c2db0efSToomas Soome 	case 4:
25553c2db0efSToomas Soome 	case 8:
25563c2db0efSToomas Soome 		break;
25573c2db0efSToomas Soome 	default:
25583c2db0efSToomas Soome 		return (EINVAL);
25593c2db0efSToomas Soome 	}
25603c2db0efSToomas Soome 
25613c2db0efSToomas Soome 	if (integer_size * num_integers > ZAP_MAXVALUELEN)
25623c2db0efSToomas Soome 		return (E2BIG);
25633c2db0efSToomas Soome 
25643c2db0efSToomas Soome 	return (0);
25653c2db0efSToomas Soome }
25663c2db0efSToomas Soome 
25674d297e70SToomas Soome static void
25684d297e70SToomas Soome zap_leaf_free(zap_leaf_t *leaf)
25694d297e70SToomas Soome {
25704d297e70SToomas Soome 	free(leaf->l_phys);
25714d297e70SToomas Soome 	free(leaf);
25724d297e70SToomas Soome }
25734d297e70SToomas Soome 
2574b8902de1SWarner Losh static int
25754d297e70SToomas Soome zap_get_leaf_byblk(fat_zap_t *zap, uint64_t blk, zap_leaf_t **lp)
25764d297e70SToomas Soome {
25774d297e70SToomas Soome 	int bs = FZAP_BLOCK_SHIFT(zap);
25784d297e70SToomas Soome 	int err;
25794d297e70SToomas Soome 
25804d297e70SToomas Soome 	*lp = malloc(sizeof(**lp));
25814d297e70SToomas Soome 	if (*lp == NULL)
25824d297e70SToomas Soome 		return (ENOMEM);
25834d297e70SToomas Soome 
25844d297e70SToomas Soome 	(*lp)->l_bs = bs;
25854d297e70SToomas Soome 	(*lp)->l_phys = malloc(1 << bs);
25864d297e70SToomas Soome 
25874d297e70SToomas Soome 	if ((*lp)->l_phys == NULL) {
25884d297e70SToomas Soome 		free(*lp);
25894d297e70SToomas Soome 		return (ENOMEM);
25904d297e70SToomas Soome 	}
25914d297e70SToomas Soome 	err = dnode_read(zap->zap_spa, zap->zap_dnode, blk << bs, (*lp)->l_phys,
25924d297e70SToomas Soome 	    1 << bs);
25934d297e70SToomas Soome 	if (err != 0) {
25944d297e70SToomas Soome 		zap_leaf_free(*lp);
25954d297e70SToomas Soome 	}
25964d297e70SToomas Soome 	return (err);
25974d297e70SToomas Soome }
25984d297e70SToomas Soome 
25994d297e70SToomas Soome static int
26004d297e70SToomas Soome zap_table_load(fat_zap_t *zap, zap_table_phys_t *tbl, uint64_t idx,
26014d297e70SToomas Soome     uint64_t *valp)
26024d297e70SToomas Soome {
26034d297e70SToomas Soome 	int bs = FZAP_BLOCK_SHIFT(zap);
26044d297e70SToomas Soome 	uint64_t blk = idx >> (bs - 3);
26054d297e70SToomas Soome 	uint64_t off = idx & ((1 << (bs - 3)) - 1);
26064d297e70SToomas Soome 	uint64_t *buf;
26074d297e70SToomas Soome 	int rc;
26084d297e70SToomas Soome 
26094d297e70SToomas Soome 	buf = malloc(1 << zap->zap_block_shift);
26104d297e70SToomas Soome 	if (buf == NULL)
26114d297e70SToomas Soome 		return (ENOMEM);
26124d297e70SToomas Soome 	rc = dnode_read(zap->zap_spa, zap->zap_dnode, (tbl->zt_blk + blk) << bs,
26134d297e70SToomas Soome 	    buf, 1 << zap->zap_block_shift);
26144d297e70SToomas Soome 	if (rc == 0)
26154d297e70SToomas Soome 		*valp = buf[off];
26164d297e70SToomas Soome 	free(buf);
26174d297e70SToomas Soome 	return (rc);
26184d297e70SToomas Soome }
26194d297e70SToomas Soome 
26204d297e70SToomas Soome static int
26214d297e70SToomas Soome zap_idx_to_blk(fat_zap_t *zap, uint64_t idx, uint64_t *valp)
26224d297e70SToomas Soome {
26234d297e70SToomas Soome 	if (zap->zap_phys->zap_ptrtbl.zt_numblks == 0) {
26244d297e70SToomas Soome 		*valp = ZAP_EMBEDDED_PTRTBL_ENT(zap, idx);
26254d297e70SToomas Soome 		return (0);
26264d297e70SToomas Soome 	} else {
26274d297e70SToomas Soome 		return (zap_table_load(zap, &zap->zap_phys->zap_ptrtbl,
26284d297e70SToomas Soome 		    idx, valp));
26294d297e70SToomas Soome 	}
26304d297e70SToomas Soome }
26314d297e70SToomas Soome 
26324d297e70SToomas Soome #define	ZAP_HASH_IDX(hash, n)	(((n) == 0) ? 0 : ((hash) >> (64 - (n))))
26334d297e70SToomas Soome static int
26344d297e70SToomas Soome zap_deref_leaf(fat_zap_t *zap, uint64_t h, zap_leaf_t **lp)
26354d297e70SToomas Soome {
26364d297e70SToomas Soome 	uint64_t idx, blk;
26374d297e70SToomas Soome 	int err;
26384d297e70SToomas Soome 
26394d297e70SToomas Soome 	idx = ZAP_HASH_IDX(h, zap->zap_phys->zap_ptrtbl.zt_shift);
26404d297e70SToomas Soome 	err = zap_idx_to_blk(zap, idx, &blk);
26414d297e70SToomas Soome 	if (err != 0)
26424d297e70SToomas Soome 		return (err);
26434d297e70SToomas Soome 	return (zap_get_leaf_byblk(zap, blk, lp));
26444d297e70SToomas Soome }
26454d297e70SToomas Soome 
26464d297e70SToomas Soome #define	CHAIN_END	0xffff	/* end of the chunk chain */
26474d297e70SToomas Soome #define	LEAF_HASH(l, h) \
26484d297e70SToomas Soome 	((ZAP_LEAF_HASH_NUMENTRIES(l)-1) & \
26494d297e70SToomas Soome 	((h) >> \
26504d297e70SToomas Soome 	(64 - ZAP_LEAF_HASH_SHIFT(l) - (l)->l_phys->l_hdr.lh_prefix_len)))
26514d297e70SToomas Soome #define	LEAF_HASH_ENTPTR(l, h)	(&(l)->l_phys->l_hash[LEAF_HASH(l, h)])
26524d297e70SToomas Soome 
26534d297e70SToomas Soome static int
26544d297e70SToomas Soome zap_leaf_lookup(zap_leaf_t *zl, uint64_t hash, const char *name,
2655b8902de1SWarner Losh     uint64_t integer_size, uint64_t num_integers, void *value)
2656b8902de1SWarner Losh {
26574d297e70SToomas Soome 	int rc;
26584d297e70SToomas Soome 	uint16_t *chunkp;
26594d297e70SToomas Soome 	struct zap_leaf_entry *le;
26604d297e70SToomas Soome 
26614d297e70SToomas Soome 	/*
26624d297e70SToomas Soome 	 * Make sure this chunk matches our hash.
26634d297e70SToomas Soome 	 */
26644d297e70SToomas Soome 	if (zl->l_phys->l_hdr.lh_prefix_len > 0 &&
26654d297e70SToomas Soome 	    zl->l_phys->l_hdr.lh_prefix !=
26664d297e70SToomas Soome 	    hash >> (64 - zl->l_phys->l_hdr.lh_prefix_len))
26674d297e70SToomas Soome 		return (EIO);
26684d297e70SToomas Soome 
26694d297e70SToomas Soome 	rc = ENOENT;
26704d297e70SToomas Soome 	for (chunkp = LEAF_HASH_ENTPTR(zl, hash);
26714d297e70SToomas Soome 	    *chunkp != CHAIN_END; chunkp = &le->le_next) {
26724d297e70SToomas Soome 		zap_leaf_chunk_t *zc;
26734d297e70SToomas Soome 		uint16_t chunk = *chunkp;
26744d297e70SToomas Soome 
26754d297e70SToomas Soome 		le = ZAP_LEAF_ENTRY(zl, chunk);
26764d297e70SToomas Soome 		if (le->le_hash != hash)
26774d297e70SToomas Soome 			continue;
26784d297e70SToomas Soome 		zc = &ZAP_LEAF_CHUNK(zl, chunk);
26794d297e70SToomas Soome 		if (fzap_name_equal(zl, zc, name)) {
26804d297e70SToomas Soome 			if (zc->l_entry.le_value_intlen > integer_size) {
26814d297e70SToomas Soome 				rc = EINVAL;
26824d297e70SToomas Soome 			} else {
26834d297e70SToomas Soome 				fzap_leaf_array(zl, zc, integer_size,
26844d297e70SToomas Soome 				    num_integers, value);
26854d297e70SToomas Soome 				rc = 0;
26864d297e70SToomas Soome 			}
26874d297e70SToomas Soome 			break;
26884d297e70SToomas Soome 		}
26894d297e70SToomas Soome 	}
26904d297e70SToomas Soome 	return (rc);
26914d297e70SToomas Soome }
26924d297e70SToomas Soome 
26934d297e70SToomas Soome /*
26944d297e70SToomas Soome  * Lookup a value in a fatzap directory.
26954d297e70SToomas Soome  */
26964d297e70SToomas Soome static int
26974d297e70SToomas Soome fzap_lookup(const spa_t *spa, const dnode_phys_t *dnode, zap_phys_t *zh,
26984d297e70SToomas Soome     const char *name, uint64_t integer_size, uint64_t num_integers,
26994d297e70SToomas Soome     void *value)
27004d297e70SToomas Soome {
2701b8902de1SWarner Losh 	int bsize = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2702b8902de1SWarner Losh 	fat_zap_t z;
27034d297e70SToomas Soome 	zap_leaf_t *zl;
2704b8902de1SWarner Losh 	uint64_t hash;
2705b8902de1SWarner Losh 	int rc;
2706b8902de1SWarner Losh 
27074d297e70SToomas Soome 	if (zh->zap_magic != ZAP_MAGIC)
2708b8902de1SWarner Losh 		return (EIO);
2709b8902de1SWarner Losh 
2710277f38abSMariusz Zaborski 	if ((rc = fzap_check_size(integer_size, num_integers)) != 0) {
27113c2db0efSToomas Soome 		return (rc);
2712277f38abSMariusz Zaborski 	}
27133c2db0efSToomas Soome 
2714b8902de1SWarner Losh 	z.zap_block_shift = ilog2(bsize);
27154d297e70SToomas Soome 	z.zap_phys = zh;
27164d297e70SToomas Soome 	z.zap_spa = spa;
27174d297e70SToomas Soome 	z.zap_dnode = dnode;
2718b8902de1SWarner Losh 
27194d297e70SToomas Soome 	hash = zap_hash(zh->zap_salt, name);
27204d297e70SToomas Soome 	rc = zap_deref_leaf(&z, hash, &zl);
27214d297e70SToomas Soome 	if (rc != 0)
2722b8902de1SWarner Losh 		return (rc);
2723b8902de1SWarner Losh 
27244d297e70SToomas Soome 	rc = zap_leaf_lookup(zl, hash, name, integer_size, num_integers, value);
2725b8902de1SWarner Losh 
27264d297e70SToomas Soome 	zap_leaf_free(zl);
27274d297e70SToomas Soome 	return (rc);
2728b8902de1SWarner Losh }
2729b8902de1SWarner Losh 
2730b8902de1SWarner Losh /*
2731b8902de1SWarner Losh  * Lookup a name in a zap object and return its value as a uint64_t.
2732b8902de1SWarner Losh  */
2733b8902de1SWarner Losh static int
2734b8902de1SWarner Losh zap_lookup(const spa_t *spa, const dnode_phys_t *dnode, const char *name,
2735b8902de1SWarner Losh     uint64_t integer_size, uint64_t num_integers, void *value)
2736b8902de1SWarner Losh {
2737b8902de1SWarner Losh 	int rc;
27384d297e70SToomas Soome 	zap_phys_t *zap;
2739b8902de1SWarner Losh 	size_t size = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2740b8902de1SWarner Losh 
27414d297e70SToomas Soome 	zap = malloc(size);
27424d297e70SToomas Soome 	if (zap == NULL)
27434d297e70SToomas Soome 		return (ENOMEM);
2744b8902de1SWarner Losh 
27454d297e70SToomas Soome 	rc = dnode_read(spa, dnode, 0, zap, size);
27464d297e70SToomas Soome 	if (rc)
27474d297e70SToomas Soome 		goto done;
27484d297e70SToomas Soome 
27494d297e70SToomas Soome 	switch (zap->zap_block_type) {
27504d297e70SToomas Soome 	case ZBT_MICRO:
27514d297e70SToomas Soome 		rc = mzap_lookup((const mzap_phys_t *)zap, size, name, value);
27524d297e70SToomas Soome 		break;
27534d297e70SToomas Soome 	case ZBT_HEADER:
27544d297e70SToomas Soome 		rc = fzap_lookup(spa, dnode, zap, name, integer_size,
27554d297e70SToomas Soome 		    num_integers, value);
27564d297e70SToomas Soome 		break;
27574d297e70SToomas Soome 	default:
27584d297e70SToomas Soome 		printf("ZFS: invalid zap_type=%" PRIx64 "\n",
27594d297e70SToomas Soome 		    zap->zap_block_type);
27604d297e70SToomas Soome 		rc = EIO;
2761b8902de1SWarner Losh 	}
27624d297e70SToomas Soome done:
27634d297e70SToomas Soome 	free(zap);
27644d297e70SToomas Soome 	return (rc);
2765b8902de1SWarner Losh }
2766b8902de1SWarner Losh 
2767b8902de1SWarner Losh /*
27684d297e70SToomas Soome  * List a microzap directory.
2769b8902de1SWarner Losh  */
2770b8902de1SWarner Losh static int
27714d297e70SToomas Soome mzap_list(const mzap_phys_t *mz, size_t size,
27724d297e70SToomas Soome     int (*callback)(const char *, uint64_t))
2773b8902de1SWarner Losh {
2774b8902de1SWarner Losh 	const mzap_ent_phys_t *mze;
2775b8902de1SWarner Losh 	int chunks, i, rc;
2776b8902de1SWarner Losh 
2777b8902de1SWarner Losh 	/*
2778b8902de1SWarner Losh 	 * Microzap objects use exactly one block. Read the whole
2779b8902de1SWarner Losh 	 * thing.
2780b8902de1SWarner Losh 	 */
27814d297e70SToomas Soome 	rc = 0;
2782b8902de1SWarner Losh 	chunks = size / MZAP_ENT_LEN - 1;
2783b8902de1SWarner Losh 	for (i = 0; i < chunks; i++) {
2784b8902de1SWarner Losh 		mze = &mz->mz_chunk[i];
2785b8902de1SWarner Losh 		if (mze->mze_name[0]) {
2786b8902de1SWarner Losh 			rc = callback(mze->mze_name, mze->mze_value);
2787b8902de1SWarner Losh 			if (rc != 0)
27884d297e70SToomas Soome 				break;
2789b8902de1SWarner Losh 		}
2790b8902de1SWarner Losh 	}
2791b8902de1SWarner Losh 
27924d297e70SToomas Soome 	return (rc);
2793b8902de1SWarner Losh }
2794b8902de1SWarner Losh 
2795b8902de1SWarner Losh /*
27964d297e70SToomas Soome  * List a fatzap directory.
2797b8902de1SWarner Losh  */
2798b8902de1SWarner Losh static int
27994d297e70SToomas Soome fzap_list(const spa_t *spa, const dnode_phys_t *dnode, zap_phys_t *zh,
28002e6bb655SToomas Soome     int (*callback)(const char *, uint64_t))
2801b8902de1SWarner Losh {
2802b8902de1SWarner Losh 	int bsize = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2803b8902de1SWarner Losh 	fat_zap_t z;
28044d297e70SToomas Soome 	uint64_t i;
28054d297e70SToomas Soome 	int j, rc;
2806b8902de1SWarner Losh 
28074d297e70SToomas Soome 	if (zh->zap_magic != ZAP_MAGIC)
2808b8902de1SWarner Losh 		return (EIO);
2809b8902de1SWarner Losh 
2810b8902de1SWarner Losh 	z.zap_block_shift = ilog2(bsize);
28114d297e70SToomas Soome 	z.zap_phys = zh;
2812b8902de1SWarner Losh 
2813b8902de1SWarner Losh 	/*
2814b8902de1SWarner Losh 	 * This assumes that the leaf blocks start at block 1. The
2815b8902de1SWarner Losh 	 * documentation isn't exactly clear on this.
2816b8902de1SWarner Losh 	 */
2817b8902de1SWarner Losh 	zap_leaf_t zl;
2818b8902de1SWarner Losh 	zl.l_bs = z.zap_block_shift;
28194d297e70SToomas Soome 	zl.l_phys = malloc(bsize);
28204d297e70SToomas Soome 	if (zl.l_phys == NULL)
28214d297e70SToomas Soome 		return (ENOMEM);
28224d297e70SToomas Soome 
28234d297e70SToomas Soome 	for (i = 0; i < zh->zap_num_leafs; i++) {
28243c2db0efSToomas Soome 		off_t off = ((off_t)(i + 1)) << zl.l_bs;
2825b8902de1SWarner Losh 		char name[256], *p;
2826b8902de1SWarner Losh 		uint64_t value;
2827b8902de1SWarner Losh 
28284d297e70SToomas Soome 		if (dnode_read(spa, dnode, off, zl.l_phys, bsize)) {
28294d297e70SToomas Soome 			free(zl.l_phys);
2830b8902de1SWarner Losh 			return (EIO);
28314d297e70SToomas Soome 		}
2832b8902de1SWarner Losh 
2833b8902de1SWarner Losh 		for (j = 0; j < ZAP_LEAF_NUMCHUNKS(&zl); j++) {
2834b8902de1SWarner Losh 			zap_leaf_chunk_t *zc, *nc;
2835b8902de1SWarner Losh 			int namelen;
2836b8902de1SWarner Losh 
2837b8902de1SWarner Losh 			zc = &ZAP_LEAF_CHUNK(&zl, j);
2838b8902de1SWarner Losh 			if (zc->l_entry.le_type != ZAP_CHUNK_ENTRY)
2839b8902de1SWarner Losh 				continue;
2840b8902de1SWarner Losh 			namelen = zc->l_entry.le_name_numints;
2841b8902de1SWarner Losh 			if (namelen > sizeof(name))
2842b8902de1SWarner Losh 				namelen = sizeof(name);
2843b8902de1SWarner Losh 
2844b8902de1SWarner Losh 			/*
2845b8902de1SWarner Losh 			 * Paste the name back together.
2846b8902de1SWarner Losh 			 */
2847b8902de1SWarner Losh 			nc = &ZAP_LEAF_CHUNK(&zl, zc->l_entry.le_name_chunk);
2848b8902de1SWarner Losh 			p = name;
2849b8902de1SWarner Losh 			while (namelen > 0) {
2850b8902de1SWarner Losh 				int len;
2851b8902de1SWarner Losh 				len = namelen;
2852b8902de1SWarner Losh 				if (len > ZAP_LEAF_ARRAY_BYTES)
2853b8902de1SWarner Losh 					len = ZAP_LEAF_ARRAY_BYTES;
2854b8902de1SWarner Losh 				memcpy(p, nc->l_array.la_array, len);
2855b8902de1SWarner Losh 				p += len;
2856b8902de1SWarner Losh 				namelen -= len;
2857b8902de1SWarner Losh 				nc = &ZAP_LEAF_CHUNK(&zl, nc->l_array.la_next);
2858b8902de1SWarner Losh 			}
2859b8902de1SWarner Losh 
2860b8902de1SWarner Losh 			/*
2861b8902de1SWarner Losh 			 * Assume the first eight bytes of the value are
2862b8902de1SWarner Losh 			 * a uint64_t.
2863b8902de1SWarner Losh 			 */
2864b8902de1SWarner Losh 			value = fzap_leaf_value(&zl, zc);
2865b8902de1SWarner Losh 
28662e6bb655SToomas Soome 			/* printf("%s 0x%jx\n", name, (uintmax_t)value); */
2867b8902de1SWarner Losh 			rc = callback((const char *)name, value);
28684d297e70SToomas Soome 			if (rc != 0) {
28694d297e70SToomas Soome 				free(zl.l_phys);
2870b8902de1SWarner Losh 				return (rc);
2871b8902de1SWarner Losh 			}
2872b8902de1SWarner Losh 		}
28734d297e70SToomas Soome 	}
2874b8902de1SWarner Losh 
28754d297e70SToomas Soome 	free(zl.l_phys);
2876b8902de1SWarner Losh 	return (0);
2877b8902de1SWarner Losh }
2878b8902de1SWarner Losh 
2879b8902de1SWarner Losh static int zfs_printf(const char *name, uint64_t value __unused)
2880b8902de1SWarner Losh {
2881b8902de1SWarner Losh 
2882b8902de1SWarner Losh 	printf("%s\n", name);
2883b8902de1SWarner Losh 
2884b8902de1SWarner Losh 	return (0);
2885b8902de1SWarner Losh }
2886b8902de1SWarner Losh 
2887b8902de1SWarner Losh /*
2888b8902de1SWarner Losh  * List a zap directory.
2889b8902de1SWarner Losh  */
2890b8902de1SWarner Losh static int
2891b8902de1SWarner Losh zap_list(const spa_t *spa, const dnode_phys_t *dnode)
2892b8902de1SWarner Losh {
28934d297e70SToomas Soome 	zap_phys_t *zap;
2894380977d5SToomas Soome 	size_t size = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
28954d297e70SToomas Soome 	int rc;
2896b8902de1SWarner Losh 
28974d297e70SToomas Soome 	zap = malloc(size);
28984d297e70SToomas Soome 	if (zap == NULL)
28994d297e70SToomas Soome 		return (ENOMEM);
2900b8902de1SWarner Losh 
29014d297e70SToomas Soome 	rc = dnode_read(spa, dnode, 0, zap, size);
29024d297e70SToomas Soome 	if (rc == 0) {
29034d297e70SToomas Soome 		if (zap->zap_block_type == ZBT_MICRO)
29044d297e70SToomas Soome 			rc = mzap_list((const mzap_phys_t *)zap, size,
29054d297e70SToomas Soome 			    zfs_printf);
2906b8902de1SWarner Losh 		else
29074d297e70SToomas Soome 			rc = fzap_list(spa, dnode, zap, zfs_printf);
29084d297e70SToomas Soome 	}
29094d297e70SToomas Soome 	free(zap);
29104d297e70SToomas Soome 	return (rc);
2911b8902de1SWarner Losh }
2912b8902de1SWarner Losh 
2913b8902de1SWarner Losh static int
29142e6bb655SToomas Soome objset_get_dnode(const spa_t *spa, const objset_phys_t *os, uint64_t objnum,
29152e6bb655SToomas Soome     dnode_phys_t *dnode)
2916b8902de1SWarner Losh {
2917b8902de1SWarner Losh 	off_t offset;
2918b8902de1SWarner Losh 
2919b8902de1SWarner Losh 	offset = objnum * sizeof(dnode_phys_t);
2920b8902de1SWarner Losh 	return dnode_read(spa, &os->os_meta_dnode, offset,
2921b8902de1SWarner Losh 		dnode, sizeof(dnode_phys_t));
2922b8902de1SWarner Losh }
2923b8902de1SWarner Losh 
29244d297e70SToomas Soome /*
29254d297e70SToomas Soome  * Lookup a name in a microzap directory.
29264d297e70SToomas Soome  */
2927b8902de1SWarner Losh static int
29284d297e70SToomas Soome mzap_rlookup(const mzap_phys_t *mz, size_t size, char *name, uint64_t value)
2929b8902de1SWarner Losh {
2930b8902de1SWarner Losh 	const mzap_ent_phys_t *mze;
2931b8902de1SWarner Losh 	int chunks, i;
2932b8902de1SWarner Losh 
2933b8902de1SWarner Losh 	/*
2934b8902de1SWarner Losh 	 * Microzap objects use exactly one block. Read the whole
2935b8902de1SWarner Losh 	 * thing.
2936b8902de1SWarner Losh 	 */
2937b8902de1SWarner Losh 	chunks = size / MZAP_ENT_LEN - 1;
2938b8902de1SWarner Losh 	for (i = 0; i < chunks; i++) {
2939b8902de1SWarner Losh 		mze = &mz->mz_chunk[i];
2940b8902de1SWarner Losh 		if (value == mze->mze_value) {
2941b8902de1SWarner Losh 			strcpy(name, mze->mze_name);
2942b8902de1SWarner Losh 			return (0);
2943b8902de1SWarner Losh 		}
2944b8902de1SWarner Losh 	}
2945b8902de1SWarner Losh 
2946b8902de1SWarner Losh 	return (ENOENT);
2947b8902de1SWarner Losh }
2948b8902de1SWarner Losh 
2949b8902de1SWarner Losh static void
2950b8902de1SWarner Losh fzap_name_copy(const zap_leaf_t *zl, const zap_leaf_chunk_t *zc, char *name)
2951b8902de1SWarner Losh {
2952b8902de1SWarner Losh 	size_t namelen;
2953b8902de1SWarner Losh 	const zap_leaf_chunk_t *nc;
2954b8902de1SWarner Losh 	char *p;
2955b8902de1SWarner Losh 
2956b8902de1SWarner Losh 	namelen = zc->l_entry.le_name_numints;
2957b8902de1SWarner Losh 
2958b8902de1SWarner Losh 	nc = &ZAP_LEAF_CHUNK(zl, zc->l_entry.le_name_chunk);
2959b8902de1SWarner Losh 	p = name;
2960b8902de1SWarner Losh 	while (namelen > 0) {
2961b8902de1SWarner Losh 		size_t len;
2962b8902de1SWarner Losh 		len = namelen;
2963b8902de1SWarner Losh 		if (len > ZAP_LEAF_ARRAY_BYTES)
2964b8902de1SWarner Losh 			len = ZAP_LEAF_ARRAY_BYTES;
2965b8902de1SWarner Losh 		memcpy(p, nc->l_array.la_array, len);
2966b8902de1SWarner Losh 		p += len;
2967b8902de1SWarner Losh 		namelen -= len;
2968b8902de1SWarner Losh 		nc = &ZAP_LEAF_CHUNK(zl, nc->l_array.la_next);
2969b8902de1SWarner Losh 	}
2970b8902de1SWarner Losh 
2971b8902de1SWarner Losh 	*p = '\0';
2972b8902de1SWarner Losh }
2973b8902de1SWarner Losh 
2974b8902de1SWarner Losh static int
29754d297e70SToomas Soome fzap_rlookup(const spa_t *spa, const dnode_phys_t *dnode, zap_phys_t *zh,
29764d297e70SToomas Soome     char *name, uint64_t value)
2977b8902de1SWarner Losh {
2978b8902de1SWarner Losh 	int bsize = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2979b8902de1SWarner Losh 	fat_zap_t z;
29804d297e70SToomas Soome 	uint64_t i;
29814d297e70SToomas Soome 	int j, rc;
2982b8902de1SWarner Losh 
29834d297e70SToomas Soome 	if (zh->zap_magic != ZAP_MAGIC)
2984b8902de1SWarner Losh 		return (EIO);
2985b8902de1SWarner Losh 
2986b8902de1SWarner Losh 	z.zap_block_shift = ilog2(bsize);
29874d297e70SToomas Soome 	z.zap_phys = zh;
2988b8902de1SWarner Losh 
2989b8902de1SWarner Losh 	/*
2990b8902de1SWarner Losh 	 * This assumes that the leaf blocks start at block 1. The
2991b8902de1SWarner Losh 	 * documentation isn't exactly clear on this.
2992b8902de1SWarner Losh 	 */
2993b8902de1SWarner Losh 	zap_leaf_t zl;
2994b8902de1SWarner Losh 	zl.l_bs = z.zap_block_shift;
29954d297e70SToomas Soome 	zl.l_phys = malloc(bsize);
29964d297e70SToomas Soome 	if (zl.l_phys == NULL)
29974d297e70SToomas Soome 		return (ENOMEM);
29984d297e70SToomas Soome 
29994d297e70SToomas Soome 	for (i = 0; i < zh->zap_num_leafs; i++) {
30003c2db0efSToomas Soome 		off_t off = ((off_t)(i + 1)) << zl.l_bs;
3001b8902de1SWarner Losh 
30024d297e70SToomas Soome 		rc = dnode_read(spa, dnode, off, zl.l_phys, bsize);
30034d297e70SToomas Soome 		if (rc != 0)
30044d297e70SToomas Soome 			goto done;
3005b8902de1SWarner Losh 
3006b8902de1SWarner Losh 		for (j = 0; j < ZAP_LEAF_NUMCHUNKS(&zl); j++) {
3007b8902de1SWarner Losh 			zap_leaf_chunk_t *zc;
3008b8902de1SWarner Losh 
3009b8902de1SWarner Losh 			zc = &ZAP_LEAF_CHUNK(&zl, j);
3010b8902de1SWarner Losh 			if (zc->l_entry.le_type != ZAP_CHUNK_ENTRY)
3011b8902de1SWarner Losh 				continue;
3012b8902de1SWarner Losh 			if (zc->l_entry.le_value_intlen != 8 ||
3013b8902de1SWarner Losh 			    zc->l_entry.le_value_numints != 1)
3014b8902de1SWarner Losh 				continue;
3015b8902de1SWarner Losh 
3016b8902de1SWarner Losh 			if (fzap_leaf_value(&zl, zc) == value) {
3017b8902de1SWarner Losh 				fzap_name_copy(&zl, zc, name);
30184d297e70SToomas Soome 				goto done;
3019b8902de1SWarner Losh 			}
3020b8902de1SWarner Losh 		}
3021b8902de1SWarner Losh 	}
3022b8902de1SWarner Losh 
30234d297e70SToomas Soome 	rc = ENOENT;
30244d297e70SToomas Soome done:
30254d297e70SToomas Soome 	free(zl.l_phys);
30264d297e70SToomas Soome 	return (rc);
3027b8902de1SWarner Losh }
3028b8902de1SWarner Losh 
3029b8902de1SWarner Losh static int
30302e6bb655SToomas Soome zap_rlookup(const spa_t *spa, const dnode_phys_t *dnode, char *name,
30312e6bb655SToomas Soome     uint64_t value)
3032b8902de1SWarner Losh {
30334d297e70SToomas Soome 	zap_phys_t *zap;
3034380977d5SToomas Soome 	size_t size = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
30354d297e70SToomas Soome 	int rc;
3036b8902de1SWarner Losh 
30374d297e70SToomas Soome 	zap = malloc(size);
30384d297e70SToomas Soome 	if (zap == NULL)
30394d297e70SToomas Soome 		return (ENOMEM);
3040b8902de1SWarner Losh 
30414d297e70SToomas Soome 	rc = dnode_read(spa, dnode, 0, zap, size);
30424d297e70SToomas Soome 	if (rc == 0) {
30434d297e70SToomas Soome 		if (zap->zap_block_type == ZBT_MICRO)
30444d297e70SToomas Soome 			rc = mzap_rlookup((const mzap_phys_t *)zap, size,
30454d297e70SToomas Soome 			    name, value);
3046b8902de1SWarner Losh 		else
30474d297e70SToomas Soome 			rc = fzap_rlookup(spa, dnode, zap, name, value);
30484d297e70SToomas Soome 	}
30494d297e70SToomas Soome 	free(zap);
30504d297e70SToomas Soome 	return (rc);
3051b8902de1SWarner Losh }
3052b8902de1SWarner Losh 
3053b8902de1SWarner Losh static int
3054b8902de1SWarner Losh zfs_rlookup(const spa_t *spa, uint64_t objnum, char *result)
3055b8902de1SWarner Losh {
3056b8902de1SWarner Losh 	char name[256];
3057b8902de1SWarner Losh 	char component[256];
3058b8902de1SWarner Losh 	uint64_t dir_obj, parent_obj, child_dir_zapobj;
3059a849842fSAllan Jude 	dnode_phys_t child_dir_zap, snapnames_zap, dataset, dir, parent;
3060b8902de1SWarner Losh 	dsl_dir_phys_t *dd;
3061b8902de1SWarner Losh 	dsl_dataset_phys_t *ds;
3062b8902de1SWarner Losh 	char *p;
3063b8902de1SWarner Losh 	int len;
3064a849842fSAllan Jude 	boolean_t issnap = B_FALSE;
3065b8902de1SWarner Losh 
3066b8902de1SWarner Losh 	p = &name[sizeof(name) - 1];
3067b8902de1SWarner Losh 	*p = '\0';
3068b8902de1SWarner Losh 
3069277f38abSMariusz Zaborski 	if (objset_get_dnode(spa, spa->spa_mos, objnum, &dataset)) {
3070b8902de1SWarner Losh 		printf("ZFS: can't find dataset %ju\n", (uintmax_t)objnum);
3071b8902de1SWarner Losh 		return (EIO);
3072b8902de1SWarner Losh 	}
3073b8902de1SWarner Losh 	ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3074b8902de1SWarner Losh 	dir_obj = ds->ds_dir_obj;
3075a849842fSAllan Jude 	if (ds->ds_snapnames_zapobj == 0)
3076a849842fSAllan Jude 		issnap = B_TRUE;
3077b8902de1SWarner Losh 
3078b8902de1SWarner Losh 	for (;;) {
3079277f38abSMariusz Zaborski 		if (objset_get_dnode(spa, spa->spa_mos, dir_obj, &dir) != 0)
3080b8902de1SWarner Losh 			return (EIO);
3081b8902de1SWarner Losh 		dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3082b8902de1SWarner Losh 
3083b8902de1SWarner Losh 		/* Actual loop condition. */
3084b8902de1SWarner Losh 		parent_obj = dd->dd_parent_obj;
3085b8902de1SWarner Losh 		if (parent_obj == 0)
3086b8902de1SWarner Losh 			break;
3087b8902de1SWarner Losh 
3088277f38abSMariusz Zaborski 		if (objset_get_dnode(spa, spa->spa_mos, parent_obj,
30892e6bb655SToomas Soome 		    &parent) != 0)
3090b8902de1SWarner Losh 			return (EIO);
3091b8902de1SWarner Losh 		dd = (dsl_dir_phys_t *)&parent.dn_bonus;
3092a849842fSAllan Jude 		if (issnap == B_TRUE) {
3093a849842fSAllan Jude 			/*
3094a849842fSAllan Jude 			 * The dataset we are looking up is a snapshot
3095a849842fSAllan Jude 			 * the dir_obj is the parent already, we don't want
3096a849842fSAllan Jude 			 * the grandparent just yet. Reset to the parent.
3097a849842fSAllan Jude 			 */
3098a849842fSAllan Jude 			dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3099a849842fSAllan Jude 			/* Lookup the dataset to get the snapname ZAP */
3100a849842fSAllan Jude 			if (objset_get_dnode(spa, spa->spa_mos,
3101a849842fSAllan Jude 			    dd->dd_head_dataset_obj, &dataset))
3102a849842fSAllan Jude 				return (EIO);
3103a849842fSAllan Jude 			ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3104a849842fSAllan Jude 			if (objset_get_dnode(spa, spa->spa_mos,
3105a849842fSAllan Jude 			    ds->ds_snapnames_zapobj, &snapnames_zap) != 0)
3106a849842fSAllan Jude 				return (EIO);
3107a849842fSAllan Jude 			/* Get the name of the snapshot */
3108a849842fSAllan Jude 			if (zap_rlookup(spa, &snapnames_zap, component,
3109a849842fSAllan Jude 			    objnum) != 0)
3110a849842fSAllan Jude 				return (EIO);
3111a849842fSAllan Jude 			len = strlen(component);
3112a849842fSAllan Jude 			p -= len;
3113a849842fSAllan Jude 			memcpy(p, component, len);
3114a849842fSAllan Jude 			--p;
3115a849842fSAllan Jude 			*p = '@';
3116a849842fSAllan Jude 			issnap = B_FALSE;
3117a849842fSAllan Jude 			continue;
3118a849842fSAllan Jude 		}
3119a849842fSAllan Jude 
3120b8902de1SWarner Losh 		child_dir_zapobj = dd->dd_child_dir_zapobj;
3121277f38abSMariusz Zaborski 		if (objset_get_dnode(spa, spa->spa_mos, child_dir_zapobj,
31222e6bb655SToomas Soome 		    &child_dir_zap) != 0)
3123b8902de1SWarner Losh 			return (EIO);
3124b8902de1SWarner Losh 		if (zap_rlookup(spa, &child_dir_zap, component, dir_obj) != 0)
3125b8902de1SWarner Losh 			return (EIO);
3126b8902de1SWarner Losh 
3127b8902de1SWarner Losh 		len = strlen(component);
3128b8902de1SWarner Losh 		p -= len;
3129b8902de1SWarner Losh 		memcpy(p, component, len);
3130b8902de1SWarner Losh 		--p;
3131b8902de1SWarner Losh 		*p = '/';
3132b8902de1SWarner Losh 
3133b8902de1SWarner Losh 		/* Actual loop iteration. */
3134b8902de1SWarner Losh 		dir_obj = parent_obj;
3135b8902de1SWarner Losh 	}
3136b8902de1SWarner Losh 
3137b8902de1SWarner Losh 	if (*p != '\0')
3138b8902de1SWarner Losh 		++p;
3139b8902de1SWarner Losh 	strcpy(result, p);
3140b8902de1SWarner Losh 
3141b8902de1SWarner Losh 	return (0);
3142b8902de1SWarner Losh }
3143b8902de1SWarner Losh 
3144b8902de1SWarner Losh static int
3145b8902de1SWarner Losh zfs_lookup_dataset(const spa_t *spa, const char *name, uint64_t *objnum)
3146b8902de1SWarner Losh {
3147b8902de1SWarner Losh 	char element[256];
3148b8902de1SWarner Losh 	uint64_t dir_obj, child_dir_zapobj;
3149a849842fSAllan Jude 	dnode_phys_t child_dir_zap, snapnames_zap, dir, dataset;
3150b8902de1SWarner Losh 	dsl_dir_phys_t *dd;
3151a849842fSAllan Jude 	dsl_dataset_phys_t *ds;
3152b8902de1SWarner Losh 	const char *p, *q;
3153a849842fSAllan Jude 	boolean_t issnap = B_FALSE;
3154b8902de1SWarner Losh 
3155277f38abSMariusz Zaborski 	if (objset_get_dnode(spa, spa->spa_mos,
31562e6bb655SToomas Soome 	    DMU_POOL_DIRECTORY_OBJECT, &dir))
3157b8902de1SWarner Losh 		return (EIO);
3158b8902de1SWarner Losh 	if (zap_lookup(spa, &dir, DMU_POOL_ROOT_DATASET, sizeof (dir_obj),
3159b8902de1SWarner Losh 	    1, &dir_obj))
3160b8902de1SWarner Losh 		return (EIO);
3161b8902de1SWarner Losh 
3162b8902de1SWarner Losh 	p = name;
3163b8902de1SWarner Losh 	for (;;) {
3164277f38abSMariusz Zaborski 		if (objset_get_dnode(spa, spa->spa_mos, dir_obj, &dir))
3165b8902de1SWarner Losh 			return (EIO);
3166b8902de1SWarner Losh 		dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3167b8902de1SWarner Losh 
3168b8902de1SWarner Losh 		while (*p == '/')
3169b8902de1SWarner Losh 			p++;
3170b8902de1SWarner Losh 		/* Actual loop condition #1. */
3171b8902de1SWarner Losh 		if (*p == '\0')
3172b8902de1SWarner Losh 			break;
3173b8902de1SWarner Losh 
3174b8902de1SWarner Losh 		q = strchr(p, '/');
3175b8902de1SWarner Losh 		if (q) {
3176b8902de1SWarner Losh 			memcpy(element, p, q - p);
3177b8902de1SWarner Losh 			element[q - p] = '\0';
3178b8902de1SWarner Losh 			p = q + 1;
3179b8902de1SWarner Losh 		} else {
3180b8902de1SWarner Losh 			strcpy(element, p);
3181b8902de1SWarner Losh 			p += strlen(p);
3182b8902de1SWarner Losh 		}
3183b8902de1SWarner Losh 
3184a849842fSAllan Jude 		if (issnap == B_TRUE) {
3185a849842fSAllan Jude 		        if (objset_get_dnode(spa, spa->spa_mos,
3186a849842fSAllan Jude 			    dd->dd_head_dataset_obj, &dataset))
3187a849842fSAllan Jude 		                return (EIO);
3188a849842fSAllan Jude 			ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3189a849842fSAllan Jude 			if (objset_get_dnode(spa, spa->spa_mos,
3190a849842fSAllan Jude 			    ds->ds_snapnames_zapobj, &snapnames_zap) != 0)
3191a849842fSAllan Jude 				return (EIO);
3192a849842fSAllan Jude 			/* Actual loop condition #2. */
3193a849842fSAllan Jude 			if (zap_lookup(spa, &snapnames_zap, element,
3194a849842fSAllan Jude 			    sizeof (dir_obj), 1, &dir_obj) != 0)
3195a849842fSAllan Jude 				return (ENOENT);
3196a849842fSAllan Jude 			*objnum = dir_obj;
3197a849842fSAllan Jude 			return (0);
3198a849842fSAllan Jude 		} else if ((q = strchr(element, '@')) != NULL) {
3199a849842fSAllan Jude 			issnap = B_TRUE;
3200a849842fSAllan Jude 			element[q - element] = '\0';
3201a849842fSAllan Jude 			p = q + 1;
3202a849842fSAllan Jude 		}
3203b8902de1SWarner Losh 		child_dir_zapobj = dd->dd_child_dir_zapobj;
3204277f38abSMariusz Zaborski 		if (objset_get_dnode(spa, spa->spa_mos, child_dir_zapobj,
32052e6bb655SToomas Soome 		    &child_dir_zap) != 0)
3206b8902de1SWarner Losh 			return (EIO);
3207b8902de1SWarner Losh 
3208b8902de1SWarner Losh 		/* Actual loop condition #2. */
3209b8902de1SWarner Losh 		if (zap_lookup(spa, &child_dir_zap, element, sizeof (dir_obj),
3210b8902de1SWarner Losh 		    1, &dir_obj) != 0)
3211b8902de1SWarner Losh 			return (ENOENT);
3212b8902de1SWarner Losh 	}
3213b8902de1SWarner Losh 
3214b8902de1SWarner Losh 	*objnum = dd->dd_head_dataset_obj;
3215b8902de1SWarner Losh 	return (0);
3216b8902de1SWarner Losh }
3217b8902de1SWarner Losh 
3218b8902de1SWarner Losh #ifndef BOOT2
3219b8902de1SWarner Losh static int
3220b8902de1SWarner Losh zfs_list_dataset(const spa_t *spa, uint64_t objnum/*, int pos, char *entry*/)
3221b8902de1SWarner Losh {
3222b8902de1SWarner Losh 	uint64_t dir_obj, child_dir_zapobj;
3223b8902de1SWarner Losh 	dnode_phys_t child_dir_zap, dir, dataset;
3224b8902de1SWarner Losh 	dsl_dataset_phys_t *ds;
3225b8902de1SWarner Losh 	dsl_dir_phys_t *dd;
3226b8902de1SWarner Losh 
3227277f38abSMariusz Zaborski 	if (objset_get_dnode(spa, spa->spa_mos, objnum, &dataset)) {
3228b8902de1SWarner Losh 		printf("ZFS: can't find dataset %ju\n", (uintmax_t)objnum);
3229b8902de1SWarner Losh 		return (EIO);
3230b8902de1SWarner Losh 	}
3231b8902de1SWarner Losh 	ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3232b8902de1SWarner Losh 	dir_obj = ds->ds_dir_obj;
3233b8902de1SWarner Losh 
3234277f38abSMariusz Zaborski 	if (objset_get_dnode(spa, spa->spa_mos, dir_obj, &dir)) {
3235b8902de1SWarner Losh 		printf("ZFS: can't find dirobj %ju\n", (uintmax_t)dir_obj);
3236b8902de1SWarner Losh 		return (EIO);
3237b8902de1SWarner Losh 	}
3238b8902de1SWarner Losh 	dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3239b8902de1SWarner Losh 
3240b8902de1SWarner Losh 	child_dir_zapobj = dd->dd_child_dir_zapobj;
3241277f38abSMariusz Zaborski 	if (objset_get_dnode(spa, spa->spa_mos, child_dir_zapobj,
32422e6bb655SToomas Soome 	    &child_dir_zap) != 0) {
3243b8902de1SWarner Losh 		printf("ZFS: can't find child zap %ju\n", (uintmax_t)dir_obj);
3244b8902de1SWarner Losh 		return (EIO);
3245b8902de1SWarner Losh 	}
3246b8902de1SWarner Losh 
3247b8902de1SWarner Losh 	return (zap_list(spa, &child_dir_zap) != 0);
3248b8902de1SWarner Losh }
3249b8902de1SWarner Losh 
3250b8902de1SWarner Losh int
32512e6bb655SToomas Soome zfs_callback_dataset(const spa_t *spa, uint64_t objnum,
32522e6bb655SToomas Soome     int (*callback)(const char *, uint64_t))
3253b8902de1SWarner Losh {
32544d297e70SToomas Soome 	uint64_t dir_obj, child_dir_zapobj;
3255b8902de1SWarner Losh 	dnode_phys_t child_dir_zap, dir, dataset;
3256b8902de1SWarner Losh 	dsl_dataset_phys_t *ds;
3257b8902de1SWarner Losh 	dsl_dir_phys_t *dd;
32584d297e70SToomas Soome 	zap_phys_t *zap;
32594d297e70SToomas Soome 	size_t size;
3260b8902de1SWarner Losh 	int err;
3261b8902de1SWarner Losh 
3262277f38abSMariusz Zaborski 	err = objset_get_dnode(spa, spa->spa_mos, objnum, &dataset);
3263b8902de1SWarner Losh 	if (err != 0) {
3264b8902de1SWarner Losh 		printf("ZFS: can't find dataset %ju\n", (uintmax_t)objnum);
3265b8902de1SWarner Losh 		return (err);
3266b8902de1SWarner Losh 	}
3267b8902de1SWarner Losh 	ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3268b8902de1SWarner Losh 	dir_obj = ds->ds_dir_obj;
3269b8902de1SWarner Losh 
3270277f38abSMariusz Zaborski 	err = objset_get_dnode(spa, spa->spa_mos, dir_obj, &dir);
3271b8902de1SWarner Losh 	if (err != 0) {
3272b8902de1SWarner Losh 		printf("ZFS: can't find dirobj %ju\n", (uintmax_t)dir_obj);
3273b8902de1SWarner Losh 		return (err);
3274b8902de1SWarner Losh 	}
3275b8902de1SWarner Losh 	dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3276b8902de1SWarner Losh 
3277b8902de1SWarner Losh 	child_dir_zapobj = dd->dd_child_dir_zapobj;
3278277f38abSMariusz Zaborski 	err = objset_get_dnode(spa, spa->spa_mos, child_dir_zapobj,
32792e6bb655SToomas Soome 	    &child_dir_zap);
3280b8902de1SWarner Losh 	if (err != 0) {
3281b8902de1SWarner Losh 		printf("ZFS: can't find child zap %ju\n", (uintmax_t)dir_obj);
3282b8902de1SWarner Losh 		return (err);
3283b8902de1SWarner Losh 	}
3284b8902de1SWarner Losh 
3285380977d5SToomas Soome 	size = child_dir_zap.dn_datablkszsec << SPA_MINBLOCKSHIFT;
32864d297e70SToomas Soome 	zap = malloc(size);
32874d297e70SToomas Soome 	if (zap != NULL) {
32884d297e70SToomas Soome 		err = dnode_read(spa, &child_dir_zap, 0, zap, size);
3289b8902de1SWarner Losh 		if (err != 0)
32904d297e70SToomas Soome 			goto done;
3291b8902de1SWarner Losh 
32924d297e70SToomas Soome 		if (zap->zap_block_type == ZBT_MICRO)
32934d297e70SToomas Soome 			err = mzap_list((const mzap_phys_t *)zap, size,
32944d297e70SToomas Soome 			    callback);
3295b8902de1SWarner Losh 		else
32964d297e70SToomas Soome 			err = fzap_list(spa, &child_dir_zap, zap, callback);
3297380977d5SToomas Soome 	} else {
3298380977d5SToomas Soome 		err = ENOMEM;
32994d297e70SToomas Soome 	}
33004d297e70SToomas Soome done:
33014d297e70SToomas Soome 	free(zap);
33024d297e70SToomas Soome 	return (err);
3303b8902de1SWarner Losh }
3304b8902de1SWarner Losh #endif
3305b8902de1SWarner Losh 
3306b8902de1SWarner Losh /*
3307b8902de1SWarner Losh  * Find the object set given the object number of its dataset object
3308b8902de1SWarner Losh  * and return its details in *objset
3309b8902de1SWarner Losh  */
3310b8902de1SWarner Losh static int
3311b8902de1SWarner Losh zfs_mount_dataset(const spa_t *spa, uint64_t objnum, objset_phys_t *objset)
3312b8902de1SWarner Losh {
3313b8902de1SWarner Losh 	dnode_phys_t dataset;
3314b8902de1SWarner Losh 	dsl_dataset_phys_t *ds;
3315b8902de1SWarner Losh 
3316277f38abSMariusz Zaborski 	if (objset_get_dnode(spa, spa->spa_mos, objnum, &dataset)) {
3317b8902de1SWarner Losh 		printf("ZFS: can't find dataset %ju\n", (uintmax_t)objnum);
3318b8902de1SWarner Losh 		return (EIO);
3319b8902de1SWarner Losh 	}
3320b8902de1SWarner Losh 
3321b8902de1SWarner Losh 	ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3322b8902de1SWarner Losh 	if (zio_read(spa, &ds->ds_bp, objset)) {
3323b8902de1SWarner Losh 		printf("ZFS: can't read object set for dataset %ju\n",
3324b8902de1SWarner Losh 		    (uintmax_t)objnum);
3325b8902de1SWarner Losh 		return (EIO);
3326b8902de1SWarner Losh 	}
3327b8902de1SWarner Losh 
3328b8902de1SWarner Losh 	return (0);
3329b8902de1SWarner Losh }
3330b8902de1SWarner Losh 
3331b8902de1SWarner Losh /*
3332b8902de1SWarner Losh  * Find the object set pointed to by the BOOTFS property or the root
3333b8902de1SWarner Losh  * dataset if there is none and return its details in *objset
3334b8902de1SWarner Losh  */
3335b8902de1SWarner Losh static int
3336b8902de1SWarner Losh zfs_get_root(const spa_t *spa, uint64_t *objid)
3337b8902de1SWarner Losh {
3338b8902de1SWarner Losh 	dnode_phys_t dir, propdir;
3339b8902de1SWarner Losh 	uint64_t props, bootfs, root;
3340b8902de1SWarner Losh 
3341b8902de1SWarner Losh 	*objid = 0;
3342b8902de1SWarner Losh 
3343b8902de1SWarner Losh 	/*
3344b8902de1SWarner Losh 	 * Start with the MOS directory object.
3345b8902de1SWarner Losh 	 */
3346277f38abSMariusz Zaborski 	if (objset_get_dnode(spa, spa->spa_mos,
33472e6bb655SToomas Soome 	    DMU_POOL_DIRECTORY_OBJECT, &dir)) {
3348b8902de1SWarner Losh 		printf("ZFS: can't read MOS object directory\n");
3349b8902de1SWarner Losh 		return (EIO);
3350b8902de1SWarner Losh 	}
3351b8902de1SWarner Losh 
3352b8902de1SWarner Losh 	/*
3353b8902de1SWarner Losh 	 * Lookup the pool_props and see if we can find a bootfs.
3354b8902de1SWarner Losh 	 */
33552e6bb655SToomas Soome 	if (zap_lookup(spa, &dir, DMU_POOL_PROPS,
33562e6bb655SToomas Soome 	    sizeof(props), 1, &props) == 0 &&
3357277f38abSMariusz Zaborski 	    objset_get_dnode(spa, spa->spa_mos, props, &propdir) == 0 &&
33582e6bb655SToomas Soome 	    zap_lookup(spa, &propdir, "bootfs",
33593c2db0efSToomas Soome 	    sizeof(bootfs), 1, &bootfs) == 0 && bootfs != 0) {
3360b8902de1SWarner Losh 		*objid = bootfs;
3361b8902de1SWarner Losh 		return (0);
3362b8902de1SWarner Losh 	}
3363b8902de1SWarner Losh 	/*
3364b8902de1SWarner Losh 	 * Lookup the root dataset directory
3365b8902de1SWarner Losh 	 */
33662e6bb655SToomas Soome 	if (zap_lookup(spa, &dir, DMU_POOL_ROOT_DATASET,
33672e6bb655SToomas Soome 	    sizeof(root), 1, &root) ||
3368277f38abSMariusz Zaborski 	    objset_get_dnode(spa, spa->spa_mos, root, &dir)) {
3369b8902de1SWarner Losh 		printf("ZFS: can't find root dsl_dir\n");
3370b8902de1SWarner Losh 		return (EIO);
3371b8902de1SWarner Losh 	}
3372b8902de1SWarner Losh 
3373b8902de1SWarner Losh 	/*
3374b8902de1SWarner Losh 	 * Use the information from the dataset directory's bonus buffer
3375b8902de1SWarner Losh 	 * to find the dataset object and from that the object set itself.
3376b8902de1SWarner Losh 	 */
3377b8902de1SWarner Losh 	dsl_dir_phys_t *dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3378b8902de1SWarner Losh 	*objid = dd->dd_head_dataset_obj;
3379b8902de1SWarner Losh 	return (0);
3380b8902de1SWarner Losh }
3381b8902de1SWarner Losh 
3382b8902de1SWarner Losh static int
3383b4cb3fe0SToomas Soome zfs_mount_impl(const spa_t *spa, uint64_t rootobj, struct zfsmount *mount)
3384b8902de1SWarner Losh {
3385b8902de1SWarner Losh 
3386b8902de1SWarner Losh 	mount->spa = spa;
3387b8902de1SWarner Losh 
3388b8902de1SWarner Losh 	/*
3389b8902de1SWarner Losh 	 * Find the root object set if not explicitly provided
3390b8902de1SWarner Losh 	 */
3391b8902de1SWarner Losh 	if (rootobj == 0 && zfs_get_root(spa, &rootobj)) {
3392b8902de1SWarner Losh 		printf("ZFS: can't find root filesystem\n");
3393b8902de1SWarner Losh 		return (EIO);
3394b8902de1SWarner Losh 	}
3395b8902de1SWarner Losh 
3396b8902de1SWarner Losh 	if (zfs_mount_dataset(spa, rootobj, &mount->objset)) {
3397b8902de1SWarner Losh 		printf("ZFS: can't open root filesystem\n");
3398b8902de1SWarner Losh 		return (EIO);
3399b8902de1SWarner Losh 	}
3400b8902de1SWarner Losh 
3401b8902de1SWarner Losh 	mount->rootobj = rootobj;
3402b8902de1SWarner Losh 
3403b8902de1SWarner Losh 	return (0);
3404b8902de1SWarner Losh }
3405b8902de1SWarner Losh 
3406b8902de1SWarner Losh /*
3407b8902de1SWarner Losh  * callback function for feature name checks.
3408b8902de1SWarner Losh  */
3409b8902de1SWarner Losh static int
3410b8902de1SWarner Losh check_feature(const char *name, uint64_t value)
3411b8902de1SWarner Losh {
3412b8902de1SWarner Losh 	int i;
3413b8902de1SWarner Losh 
3414b8902de1SWarner Losh 	if (value == 0)
3415b8902de1SWarner Losh 		return (0);
3416b8902de1SWarner Losh 	if (name[0] == '\0')
3417b8902de1SWarner Losh 		return (0);
3418b8902de1SWarner Losh 
3419b8902de1SWarner Losh 	for (i = 0; features_for_read[i] != NULL; i++) {
3420b8902de1SWarner Losh 		if (strcmp(name, features_for_read[i]) == 0)
3421b8902de1SWarner Losh 			return (0);
3422b8902de1SWarner Losh 	}
3423b8902de1SWarner Losh 	printf("ZFS: unsupported feature: %s\n", name);
3424b8902de1SWarner Losh 	return (EIO);
3425b8902de1SWarner Losh }
3426b8902de1SWarner Losh 
3427b8902de1SWarner Losh /*
3428b8902de1SWarner Losh  * Checks whether the MOS features that are active are supported.
3429b8902de1SWarner Losh  */
3430b8902de1SWarner Losh static int
3431b8902de1SWarner Losh check_mos_features(const spa_t *spa)
3432b8902de1SWarner Losh {
3433b8902de1SWarner Losh 	dnode_phys_t dir;
34344d297e70SToomas Soome 	zap_phys_t *zap;
34354d297e70SToomas Soome 	uint64_t objnum;
3436b8902de1SWarner Losh 	size_t size;
3437b8902de1SWarner Losh 	int rc;
3438b8902de1SWarner Losh 
3439277f38abSMariusz Zaborski 	if ((rc = objset_get_dnode(spa, spa->spa_mos, DMU_OT_OBJECT_DIRECTORY,
3440b8902de1SWarner Losh 	    &dir)) != 0)
3441b8902de1SWarner Losh 		return (rc);
3442b8902de1SWarner Losh 	if ((rc = zap_lookup(spa, &dir, DMU_POOL_FEATURES_FOR_READ,
3443b8902de1SWarner Losh 	    sizeof (objnum), 1, &objnum)) != 0) {
3444b8902de1SWarner Losh 		/*
3445b8902de1SWarner Losh 		 * It is older pool without features. As we have already
3446b8902de1SWarner Losh 		 * tested the label, just return without raising the error.
3447b8902de1SWarner Losh 		 */
3448b8902de1SWarner Losh 		return (0);
3449b8902de1SWarner Losh 	}
3450b8902de1SWarner Losh 
3451277f38abSMariusz Zaborski 	if ((rc = objset_get_dnode(spa, spa->spa_mos, objnum, &dir)) != 0)
3452b8902de1SWarner Losh 		return (rc);
3453b8902de1SWarner Losh 
3454b8902de1SWarner Losh 	if (dir.dn_type != DMU_OTN_ZAP_METADATA)
3455b8902de1SWarner Losh 		return (EIO);
3456b8902de1SWarner Losh 
3457380977d5SToomas Soome 	size = dir.dn_datablkszsec << SPA_MINBLOCKSHIFT;
34584d297e70SToomas Soome 	zap = malloc(size);
34594d297e70SToomas Soome 	if (zap == NULL)
34604d297e70SToomas Soome 		return (ENOMEM);
34614d297e70SToomas Soome 
34624d297e70SToomas Soome 	if (dnode_read(spa, &dir, 0, zap, size)) {
34634d297e70SToomas Soome 		free(zap);
3464b8902de1SWarner Losh 		return (EIO);
34654d297e70SToomas Soome 	}
3466b8902de1SWarner Losh 
34674d297e70SToomas Soome 	if (zap->zap_block_type == ZBT_MICRO)
34684d297e70SToomas Soome 		rc = mzap_list((const mzap_phys_t *)zap, size, check_feature);
3469b8902de1SWarner Losh 	else
34704d297e70SToomas Soome 		rc = fzap_list(spa, &dir, zap, check_feature);
3471b8902de1SWarner Losh 
34724d297e70SToomas Soome 	free(zap);
3473b8902de1SWarner Losh 	return (rc);
3474b8902de1SWarner Losh }
3475b8902de1SWarner Losh 
3476b8902de1SWarner Losh static int
34773830659eSToomas Soome load_nvlist(spa_t *spa, uint64_t obj, nvlist_t **value)
3478b1b93268SToomas Soome {
3479b1b93268SToomas Soome 	dnode_phys_t dir;
3480b1b93268SToomas Soome 	size_t size;
3481b1b93268SToomas Soome 	int rc;
3482e307eb94SToomas Soome 	char *nv;
3483b1b93268SToomas Soome 
3484b1b93268SToomas Soome 	*value = NULL;
3485277f38abSMariusz Zaborski 	if ((rc = objset_get_dnode(spa, spa->spa_mos, obj, &dir)) != 0)
3486b1b93268SToomas Soome 		return (rc);
3487b1b93268SToomas Soome 	if (dir.dn_type != DMU_OT_PACKED_NVLIST &&
3488b1b93268SToomas Soome 	    dir.dn_bonustype != DMU_OT_PACKED_NVLIST_SIZE) {
3489b1b93268SToomas Soome 		return (EIO);
3490b1b93268SToomas Soome 	}
3491b1b93268SToomas Soome 
3492b1b93268SToomas Soome 	if (dir.dn_bonuslen != sizeof (uint64_t))
3493b1b93268SToomas Soome 		return (EIO);
3494b1b93268SToomas Soome 
3495b1b93268SToomas Soome 	size = *(uint64_t *)DN_BONUS(&dir);
3496b1b93268SToomas Soome 	nv = malloc(size);
3497b1b93268SToomas Soome 	if (nv == NULL)
3498b1b93268SToomas Soome 		return (ENOMEM);
3499b1b93268SToomas Soome 
3500b1b93268SToomas Soome 	rc = dnode_read(spa, &dir, 0, nv, size);
3501b1b93268SToomas Soome 	if (rc != 0) {
3502b1b93268SToomas Soome 		free(nv);
3503b1b93268SToomas Soome 		nv = NULL;
3504b1b93268SToomas Soome 		return (rc);
3505b1b93268SToomas Soome 	}
3506e307eb94SToomas Soome 	*value = nvlist_import(nv, size);
35073830659eSToomas Soome 	free(nv);
3508b1b93268SToomas Soome 	return (rc);
3509b1b93268SToomas Soome }
3510b1b93268SToomas Soome 
3511b1b93268SToomas Soome static int
3512b8902de1SWarner Losh zfs_spa_init(spa_t *spa)
3513b8902de1SWarner Losh {
3514277f38abSMariusz Zaborski 	struct uberblock checkpoint;
3515b8902de1SWarner Losh 	dnode_phys_t dir;
3516b1b93268SToomas Soome 	uint64_t config_object;
35173830659eSToomas Soome 	nvlist_t *nvlist;
35183c2db0efSToomas Soome 	int rc;
3519b8902de1SWarner Losh 
3520277f38abSMariusz Zaborski 	if (zio_read(spa, &spa->spa_uberblock->ub_rootbp, spa->spa_mos)) {
3521b8902de1SWarner Losh 		printf("ZFS: can't read MOS of pool %s\n", spa->spa_name);
3522b8902de1SWarner Losh 		return (EIO);
3523b8902de1SWarner Losh 	}
3524277f38abSMariusz Zaborski 	if (spa->spa_mos->os_type != DMU_OST_META) {
3525b8902de1SWarner Losh 		printf("ZFS: corrupted MOS of pool %s\n", spa->spa_name);
3526b8902de1SWarner Losh 		return (EIO);
3527b8902de1SWarner Losh 	}
3528b8902de1SWarner Losh 
3529277f38abSMariusz Zaborski 	if (objset_get_dnode(spa, &spa->spa_mos_master,
3530277f38abSMariusz Zaborski 	    DMU_POOL_DIRECTORY_OBJECT, &dir)) {
3531b8902de1SWarner Losh 		printf("ZFS: failed to read pool %s directory object\n",
3532b8902de1SWarner Losh 		    spa->spa_name);
3533b8902de1SWarner Losh 		return (EIO);
3534b8902de1SWarner Losh 	}
3535b8902de1SWarner Losh 	/* this is allowed to fail, older pools do not have salt */
3536b8902de1SWarner Losh 	rc = zap_lookup(spa, &dir, DMU_POOL_CHECKSUM_SALT, 1,
3537b8902de1SWarner Losh 	    sizeof (spa->spa_cksum_salt.zcs_bytes),
3538b8902de1SWarner Losh 	    spa->spa_cksum_salt.zcs_bytes);
3539b8902de1SWarner Losh 
3540b8902de1SWarner Losh 	rc = check_mos_features(spa);
3541b8902de1SWarner Losh 	if (rc != 0) {
3542b8902de1SWarner Losh 		printf("ZFS: pool %s is not supported\n", spa->spa_name);
3543b1b93268SToomas Soome 		return (rc);
3544b8902de1SWarner Losh 	}
3545b8902de1SWarner Losh 
3546b1b93268SToomas Soome 	rc = zap_lookup(spa, &dir, DMU_POOL_CONFIG,
3547b1b93268SToomas Soome 	    sizeof (config_object), 1, &config_object);
3548b1b93268SToomas Soome 	if (rc != 0) {
3549b1b93268SToomas Soome 		printf("ZFS: can not read MOS %s\n", DMU_POOL_CONFIG);
3550b1b93268SToomas Soome 		return (EIO);
3551b1b93268SToomas Soome 	}
3552b1b93268SToomas Soome 	rc = load_nvlist(spa, config_object, &nvlist);
3553b1b93268SToomas Soome 	if (rc != 0)
3554b1b93268SToomas Soome 		return (rc);
3555277f38abSMariusz Zaborski 
3556277f38abSMariusz Zaborski 	rc = zap_lookup(spa, &dir, DMU_POOL_ZPOOL_CHECKPOINT,
3557277f38abSMariusz Zaborski 	    sizeof(uint64_t), sizeof(checkpoint) / sizeof(uint64_t),
3558277f38abSMariusz Zaborski 	    &checkpoint);
3559277f38abSMariusz Zaborski 	if (rc == 0 && checkpoint.ub_checkpoint_txg != 0) {
3560277f38abSMariusz Zaborski 		memcpy(&spa->spa_uberblock_checkpoint, &checkpoint,
3561277f38abSMariusz Zaborski 		    sizeof(checkpoint));
3562277f38abSMariusz Zaborski 		if (zio_read(spa, &spa->spa_uberblock_checkpoint.ub_rootbp,
3563277f38abSMariusz Zaborski 		    &spa->spa_mos_checkpoint)) {
3564277f38abSMariusz Zaborski 			printf("ZFS: can not read checkpoint data.\n");
3565277f38abSMariusz Zaborski 			return (EIO);
3566277f38abSMariusz Zaborski 		}
3567277f38abSMariusz Zaborski 	}
3568277f38abSMariusz Zaborski 
3569653c3383SToomas Soome 	/*
3570653c3383SToomas Soome 	 * Update vdevs from MOS config. Note, we do skip encoding bytes
3571653c3383SToomas Soome 	 * here. See also vdev_label_read_config().
3572653c3383SToomas Soome 	 */
35733830659eSToomas Soome 	rc = vdev_init_from_nvlist(spa, nvlist);
35743830659eSToomas Soome 	nvlist_destroy(nvlist);
3575b8902de1SWarner Losh 	return (rc);
3576b8902de1SWarner Losh }
3577b8902de1SWarner Losh 
3578b8902de1SWarner Losh static int
3579b8902de1SWarner Losh zfs_dnode_stat(const spa_t *spa, dnode_phys_t *dn, struct stat *sb)
3580b8902de1SWarner Losh {
3581b8902de1SWarner Losh 
3582b8902de1SWarner Losh 	if (dn->dn_bonustype != DMU_OT_SA) {
3583b8902de1SWarner Losh 		znode_phys_t *zp = (znode_phys_t *)dn->dn_bonus;
3584b8902de1SWarner Losh 
3585b8902de1SWarner Losh 		sb->st_mode = zp->zp_mode;
3586b8902de1SWarner Losh 		sb->st_uid = zp->zp_uid;
3587b8902de1SWarner Losh 		sb->st_gid = zp->zp_gid;
3588b8902de1SWarner Losh 		sb->st_size = zp->zp_size;
3589b8902de1SWarner Losh 	} else {
3590b8902de1SWarner Losh 		sa_hdr_phys_t *sahdrp;
3591b8902de1SWarner Losh 		int hdrsize;
3592b8902de1SWarner Losh 		size_t size = 0;
3593b8902de1SWarner Losh 		void *buf = NULL;
3594b8902de1SWarner Losh 
3595b8902de1SWarner Losh 		if (dn->dn_bonuslen != 0)
3596b8902de1SWarner Losh 			sahdrp = (sa_hdr_phys_t *)DN_BONUS(dn);
3597b8902de1SWarner Losh 		else {
3598b8902de1SWarner Losh 			if ((dn->dn_flags & DNODE_FLAG_SPILL_BLKPTR) != 0) {
3599b8902de1SWarner Losh 				blkptr_t *bp = DN_SPILL_BLKPTR(dn);
3600b8902de1SWarner Losh 				int error;
3601b8902de1SWarner Losh 
3602b8902de1SWarner Losh 				size = BP_GET_LSIZE(bp);
3603c1c4c81fSToomas Soome 				buf = malloc(size);
3604c1c4c81fSToomas Soome 				if (buf == NULL)
3605c1c4c81fSToomas Soome 					error = ENOMEM;
3606c1c4c81fSToomas Soome 				else
3607b8902de1SWarner Losh 					error = zio_read(spa, bp, buf);
3608c1c4c81fSToomas Soome 
3609b8902de1SWarner Losh 				if (error != 0) {
3610c1c4c81fSToomas Soome 					free(buf);
3611b8902de1SWarner Losh 					return (error);
3612b8902de1SWarner Losh 				}
3613b8902de1SWarner Losh 				sahdrp = buf;
3614b8902de1SWarner Losh 			} else {
3615b8902de1SWarner Losh 				return (EIO);
3616b8902de1SWarner Losh 			}
3617b8902de1SWarner Losh 		}
3618b8902de1SWarner Losh 		hdrsize = SA_HDR_SIZE(sahdrp);
3619b8902de1SWarner Losh 		sb->st_mode = *(uint64_t *)((char *)sahdrp + hdrsize +
3620b8902de1SWarner Losh 		    SA_MODE_OFFSET);
3621b8902de1SWarner Losh 		sb->st_uid = *(uint64_t *)((char *)sahdrp + hdrsize +
3622b8902de1SWarner Losh 		    SA_UID_OFFSET);
3623b8902de1SWarner Losh 		sb->st_gid = *(uint64_t *)((char *)sahdrp + hdrsize +
3624b8902de1SWarner Losh 		    SA_GID_OFFSET);
3625b8902de1SWarner Losh 		sb->st_size = *(uint64_t *)((char *)sahdrp + hdrsize +
3626b8902de1SWarner Losh 		    SA_SIZE_OFFSET);
3627c1c4c81fSToomas Soome 		free(buf);
3628b8902de1SWarner Losh 	}
3629b8902de1SWarner Losh 
3630b8902de1SWarner Losh 	return (0);
3631b8902de1SWarner Losh }
3632b8902de1SWarner Losh 
3633b8902de1SWarner Losh static int
3634b8902de1SWarner Losh zfs_dnode_readlink(const spa_t *spa, dnode_phys_t *dn, char *path, size_t psize)
3635b8902de1SWarner Losh {
3636b8902de1SWarner Losh 	int rc = 0;
3637b8902de1SWarner Losh 
3638b8902de1SWarner Losh 	if (dn->dn_bonustype == DMU_OT_SA) {
3639b8902de1SWarner Losh 		sa_hdr_phys_t *sahdrp = NULL;
3640b8902de1SWarner Losh 		size_t size = 0;
3641b8902de1SWarner Losh 		void *buf = NULL;
3642b8902de1SWarner Losh 		int hdrsize;
3643b8902de1SWarner Losh 		char *p;
3644b8902de1SWarner Losh 
3645c1c4c81fSToomas Soome 		if (dn->dn_bonuslen != 0) {
3646b8902de1SWarner Losh 			sahdrp = (sa_hdr_phys_t *)DN_BONUS(dn);
3647c1c4c81fSToomas Soome 		} else {
3648b8902de1SWarner Losh 			blkptr_t *bp;
3649b8902de1SWarner Losh 
3650b8902de1SWarner Losh 			if ((dn->dn_flags & DNODE_FLAG_SPILL_BLKPTR) == 0)
3651b8902de1SWarner Losh 				return (EIO);
3652b8902de1SWarner Losh 			bp = DN_SPILL_BLKPTR(dn);
3653b8902de1SWarner Losh 
3654b8902de1SWarner Losh 			size = BP_GET_LSIZE(bp);
3655c1c4c81fSToomas Soome 			buf = malloc(size);
3656c1c4c81fSToomas Soome 			if (buf == NULL)
3657c1c4c81fSToomas Soome 				rc = ENOMEM;
3658c1c4c81fSToomas Soome 			else
3659b8902de1SWarner Losh 				rc = zio_read(spa, bp, buf);
3660b8902de1SWarner Losh 			if (rc != 0) {
3661c1c4c81fSToomas Soome 				free(buf);
3662b8902de1SWarner Losh 				return (rc);
3663b8902de1SWarner Losh 			}
3664b8902de1SWarner Losh 			sahdrp = buf;
3665b8902de1SWarner Losh 		}
3666b8902de1SWarner Losh 		hdrsize = SA_HDR_SIZE(sahdrp);
3667b8902de1SWarner Losh 		p = (char *)((uintptr_t)sahdrp + hdrsize + SA_SYMLINK_OFFSET);
3668b8902de1SWarner Losh 		memcpy(path, p, psize);
3669c1c4c81fSToomas Soome 		free(buf);
3670b8902de1SWarner Losh 		return (0);
3671b8902de1SWarner Losh 	}
3672b8902de1SWarner Losh 	/*
3673b8902de1SWarner Losh 	 * Second test is purely to silence bogus compiler
3674b8902de1SWarner Losh 	 * warning about accessing past the end of dn_bonus.
3675b8902de1SWarner Losh 	 */
3676b8902de1SWarner Losh 	if (psize + sizeof(znode_phys_t) <= dn->dn_bonuslen &&
3677b8902de1SWarner Losh 	    sizeof(znode_phys_t) <= sizeof(dn->dn_bonus)) {
3678b8902de1SWarner Losh 		memcpy(path, &dn->dn_bonus[sizeof(znode_phys_t)], psize);
3679b8902de1SWarner Losh 	} else {
3680b8902de1SWarner Losh 		rc = dnode_read(spa, dn, 0, path, psize);
3681b8902de1SWarner Losh 	}
3682b8902de1SWarner Losh 	return (rc);
3683b8902de1SWarner Losh }
3684b8902de1SWarner Losh 
3685b8902de1SWarner Losh struct obj_list {
3686b8902de1SWarner Losh 	uint64_t		objnum;
3687b8902de1SWarner Losh 	STAILQ_ENTRY(obj_list)	entry;
3688b8902de1SWarner Losh };
3689b8902de1SWarner Losh 
3690b8902de1SWarner Losh /*
3691b8902de1SWarner Losh  * Lookup a file and return its dnode.
3692b8902de1SWarner Losh  */
3693b8902de1SWarner Losh static int
3694b8902de1SWarner Losh zfs_lookup(const struct zfsmount *mount, const char *upath, dnode_phys_t *dnode)
3695b8902de1SWarner Losh {
3696b8902de1SWarner Losh 	int rc;
3697b8902de1SWarner Losh 	uint64_t objnum;
3698b8902de1SWarner Losh 	const spa_t *spa;
3699b8902de1SWarner Losh 	dnode_phys_t dn;
3700b8902de1SWarner Losh 	const char *p, *q;
3701b8902de1SWarner Losh 	char element[256];
3702b8902de1SWarner Losh 	char path[1024];
3703b8902de1SWarner Losh 	int symlinks_followed = 0;
3704b8902de1SWarner Losh 	struct stat sb;
3705b8902de1SWarner Losh 	struct obj_list *entry, *tentry;
3706b8902de1SWarner Losh 	STAILQ_HEAD(, obj_list) on_cache = STAILQ_HEAD_INITIALIZER(on_cache);
3707b8902de1SWarner Losh 
3708b8902de1SWarner Losh 	spa = mount->spa;
3709b8902de1SWarner Losh 	if (mount->objset.os_type != DMU_OST_ZFS) {
3710b8902de1SWarner Losh 		printf("ZFS: unexpected object set type %ju\n",
3711b8902de1SWarner Losh 		    (uintmax_t)mount->objset.os_type);
3712b8902de1SWarner Losh 		return (EIO);
3713b8902de1SWarner Losh 	}
3714b8902de1SWarner Losh 
3715b8902de1SWarner Losh 	if ((entry = malloc(sizeof(struct obj_list))) == NULL)
3716b8902de1SWarner Losh 		return (ENOMEM);
3717b8902de1SWarner Losh 
3718b8902de1SWarner Losh 	/*
3719b8902de1SWarner Losh 	 * Get the root directory dnode.
3720b8902de1SWarner Losh 	 */
3721b8902de1SWarner Losh 	rc = objset_get_dnode(spa, &mount->objset, MASTER_NODE_OBJ, &dn);
3722b8902de1SWarner Losh 	if (rc) {
3723b8902de1SWarner Losh 		free(entry);
3724b8902de1SWarner Losh 		return (rc);
3725b8902de1SWarner Losh 	}
3726b8902de1SWarner Losh 
3727b8902de1SWarner Losh 	rc = zap_lookup(spa, &dn, ZFS_ROOT_OBJ, sizeof(objnum), 1, &objnum);
3728b8902de1SWarner Losh 	if (rc) {
3729b8902de1SWarner Losh 		free(entry);
3730b8902de1SWarner Losh 		return (rc);
3731b8902de1SWarner Losh 	}
3732b8902de1SWarner Losh 	entry->objnum = objnum;
3733b8902de1SWarner Losh 	STAILQ_INSERT_HEAD(&on_cache, entry, entry);
3734b8902de1SWarner Losh 
3735b8902de1SWarner Losh 	rc = objset_get_dnode(spa, &mount->objset, objnum, &dn);
3736b8902de1SWarner Losh 	if (rc != 0)
3737b8902de1SWarner Losh 		goto done;
3738b8902de1SWarner Losh 
3739b8902de1SWarner Losh 	p = upath;
3740b8902de1SWarner Losh 	while (p && *p) {
3741b8902de1SWarner Losh 		rc = objset_get_dnode(spa, &mount->objset, objnum, &dn);
3742b8902de1SWarner Losh 		if (rc != 0)
3743b8902de1SWarner Losh 			goto done;
3744b8902de1SWarner Losh 
3745b8902de1SWarner Losh 		while (*p == '/')
3746b8902de1SWarner Losh 			p++;
3747b8902de1SWarner Losh 		if (*p == '\0')
3748b8902de1SWarner Losh 			break;
3749b8902de1SWarner Losh 		q = p;
3750b8902de1SWarner Losh 		while (*q != '\0' && *q != '/')
3751b8902de1SWarner Losh 			q++;
3752b8902de1SWarner Losh 
3753b8902de1SWarner Losh 		/* skip dot */
3754b8902de1SWarner Losh 		if (p + 1 == q && p[0] == '.') {
3755b8902de1SWarner Losh 			p++;
3756b8902de1SWarner Losh 			continue;
3757b8902de1SWarner Losh 		}
3758b8902de1SWarner Losh 		/* double dot */
3759b8902de1SWarner Losh 		if (p + 2 == q && p[0] == '.' && p[1] == '.') {
3760b8902de1SWarner Losh 			p += 2;
3761b8902de1SWarner Losh 			if (STAILQ_FIRST(&on_cache) ==
3762b8902de1SWarner Losh 			    STAILQ_LAST(&on_cache, obj_list, entry)) {
3763b8902de1SWarner Losh 				rc = ENOENT;
3764b8902de1SWarner Losh 				goto done;
3765b8902de1SWarner Losh 			}
3766b8902de1SWarner Losh 			entry = STAILQ_FIRST(&on_cache);
3767b8902de1SWarner Losh 			STAILQ_REMOVE_HEAD(&on_cache, entry);
3768b8902de1SWarner Losh 			free(entry);
3769b8902de1SWarner Losh 			objnum = (STAILQ_FIRST(&on_cache))->objnum;
3770b8902de1SWarner Losh 			continue;
3771b8902de1SWarner Losh 		}
3772b8902de1SWarner Losh 		if (q - p + 1 > sizeof(element)) {
3773b8902de1SWarner Losh 			rc = ENAMETOOLONG;
3774b8902de1SWarner Losh 			goto done;
3775b8902de1SWarner Losh 		}
3776b8902de1SWarner Losh 		memcpy(element, p, q - p);
3777b8902de1SWarner Losh 		element[q - p] = 0;
3778b8902de1SWarner Losh 		p = q;
3779b8902de1SWarner Losh 
3780b8902de1SWarner Losh 		if ((rc = zfs_dnode_stat(spa, &dn, &sb)) != 0)
3781b8902de1SWarner Losh 			goto done;
3782b8902de1SWarner Losh 		if (!S_ISDIR(sb.st_mode)) {
3783b8902de1SWarner Losh 			rc = ENOTDIR;
3784b8902de1SWarner Losh 			goto done;
3785b8902de1SWarner Losh 		}
3786b8902de1SWarner Losh 
3787b8902de1SWarner Losh 		rc = zap_lookup(spa, &dn, element, sizeof (objnum), 1, &objnum);
3788b8902de1SWarner Losh 		if (rc)
3789b8902de1SWarner Losh 			goto done;
3790b8902de1SWarner Losh 		objnum = ZFS_DIRENT_OBJ(objnum);
3791b8902de1SWarner Losh 
3792b8902de1SWarner Losh 		if ((entry = malloc(sizeof(struct obj_list))) == NULL) {
3793b8902de1SWarner Losh 			rc = ENOMEM;
3794b8902de1SWarner Losh 			goto done;
3795b8902de1SWarner Losh 		}
3796b8902de1SWarner Losh 		entry->objnum = objnum;
3797b8902de1SWarner Losh 		STAILQ_INSERT_HEAD(&on_cache, entry, entry);
3798b8902de1SWarner Losh 		rc = objset_get_dnode(spa, &mount->objset, objnum, &dn);
3799b8902de1SWarner Losh 		if (rc)
3800b8902de1SWarner Losh 			goto done;
3801b8902de1SWarner Losh 
3802b8902de1SWarner Losh 		/*
3803b8902de1SWarner Losh 		 * Check for symlink.
3804b8902de1SWarner Losh 		 */
3805b8902de1SWarner Losh 		rc = zfs_dnode_stat(spa, &dn, &sb);
3806b8902de1SWarner Losh 		if (rc)
3807b8902de1SWarner Losh 			goto done;
3808b8902de1SWarner Losh 		if (S_ISLNK(sb.st_mode)) {
3809b8902de1SWarner Losh 			if (symlinks_followed > 10) {
3810b8902de1SWarner Losh 				rc = EMLINK;
3811b8902de1SWarner Losh 				goto done;
3812b8902de1SWarner Losh 			}
3813b8902de1SWarner Losh 			symlinks_followed++;
3814b8902de1SWarner Losh 
3815b8902de1SWarner Losh 			/*
3816b8902de1SWarner Losh 			 * Read the link value and copy the tail of our
3817b8902de1SWarner Losh 			 * current path onto the end.
3818b8902de1SWarner Losh 			 */
3819b8902de1SWarner Losh 			if (sb.st_size + strlen(p) + 1 > sizeof(path)) {
3820b8902de1SWarner Losh 				rc = ENAMETOOLONG;
3821b8902de1SWarner Losh 				goto done;
3822b8902de1SWarner Losh 			}
3823b8902de1SWarner Losh 			strcpy(&path[sb.st_size], p);
3824b8902de1SWarner Losh 
3825b8902de1SWarner Losh 			rc = zfs_dnode_readlink(spa, &dn, path, sb.st_size);
3826b8902de1SWarner Losh 			if (rc != 0)
3827b8902de1SWarner Losh 				goto done;
3828b8902de1SWarner Losh 
3829b8902de1SWarner Losh 			/*
3830b8902de1SWarner Losh 			 * Restart with the new path, starting either at
3831b8902de1SWarner Losh 			 * the root or at the parent depending whether or
3832b8902de1SWarner Losh 			 * not the link is relative.
3833b8902de1SWarner Losh 			 */
3834b8902de1SWarner Losh 			p = path;
3835b8902de1SWarner Losh 			if (*p == '/') {
3836b8902de1SWarner Losh 				while (STAILQ_FIRST(&on_cache) !=
3837b8902de1SWarner Losh 				    STAILQ_LAST(&on_cache, obj_list, entry)) {
3838b8902de1SWarner Losh 					entry = STAILQ_FIRST(&on_cache);
3839b8902de1SWarner Losh 					STAILQ_REMOVE_HEAD(&on_cache, entry);
3840b8902de1SWarner Losh 					free(entry);
3841b8902de1SWarner Losh 				}
3842b8902de1SWarner Losh 			} else {
3843b8902de1SWarner Losh 				entry = STAILQ_FIRST(&on_cache);
3844b8902de1SWarner Losh 				STAILQ_REMOVE_HEAD(&on_cache, entry);
3845b8902de1SWarner Losh 				free(entry);
3846b8902de1SWarner Losh 			}
3847b8902de1SWarner Losh 			objnum = (STAILQ_FIRST(&on_cache))->objnum;
3848b8902de1SWarner Losh 		}
3849b8902de1SWarner Losh 	}
3850b8902de1SWarner Losh 
3851b8902de1SWarner Losh 	*dnode = dn;
3852b8902de1SWarner Losh done:
3853b8902de1SWarner Losh 	STAILQ_FOREACH_SAFE(entry, &on_cache, entry, tentry)
3854b8902de1SWarner Losh 		free(entry);
3855b8902de1SWarner Losh 	return (rc);
3856b8902de1SWarner Losh }
38576479bd1bSWarner Losh 
38586479bd1bSWarner Losh /*
38596479bd1bSWarner Losh  * Return either a cached copy of the bootenv, or read each of the vdev children
38606479bd1bSWarner Losh  * looking for the bootenv. Cache what's found and return the results. Returns 0
38616479bd1bSWarner Losh  * when benvp is filled in, and some errno when not.
38626479bd1bSWarner Losh  */
38636479bd1bSWarner Losh static int
38646479bd1bSWarner Losh zfs_get_bootenv_spa(spa_t *spa, nvlist_t **benvp)
38656479bd1bSWarner Losh {
38666479bd1bSWarner Losh 	vdev_t *vd;
38676479bd1bSWarner Losh 	nvlist_t *benv = NULL;
38686479bd1bSWarner Losh 
38696479bd1bSWarner Losh 	if (spa->spa_bootenv == NULL) {
38706479bd1bSWarner Losh 		STAILQ_FOREACH(vd, &spa->spa_root_vdev->v_children,
38716479bd1bSWarner Losh 		    v_childlink) {
38726479bd1bSWarner Losh 			benv = vdev_read_bootenv(vd);
38736479bd1bSWarner Losh 
38746479bd1bSWarner Losh 			if (benv != NULL)
38756479bd1bSWarner Losh 				break;
38766479bd1bSWarner Losh 		}
38776479bd1bSWarner Losh 		spa->spa_bootenv = benv;
38786479bd1bSWarner Losh 	}
38796479bd1bSWarner Losh 	benv = spa->spa_bootenv;
38806479bd1bSWarner Losh 
38816479bd1bSWarner Losh 	if (benv == NULL)
38826479bd1bSWarner Losh 		return (ENOENT);
38836479bd1bSWarner Losh 
38846479bd1bSWarner Losh 	*benvp = benv;
38856479bd1bSWarner Losh 	return (0);
38866479bd1bSWarner Losh }
38874dcae288SWarner Losh 
38884dcae288SWarner Losh /*
38894dcae288SWarner Losh  * Store nvlist to pool label bootenv area. Also updates cached pointer in spa.
38904dcae288SWarner Losh  */
38914dcae288SWarner Losh static int
38924dcae288SWarner Losh zfs_set_bootenv_spa(spa_t *spa, nvlist_t *benv)
38934dcae288SWarner Losh {
38944dcae288SWarner Losh 	vdev_t *vd;
38954dcae288SWarner Losh 
38964dcae288SWarner Losh 	STAILQ_FOREACH(vd, &spa->spa_root_vdev->v_children, v_childlink) {
38974dcae288SWarner Losh 		vdev_write_bootenv(vd, benv);
38984dcae288SWarner Losh 	}
38994dcae288SWarner Losh 
39004dcae288SWarner Losh 	spa->spa_bootenv = benv;
39014dcae288SWarner Losh 	return (0);
39024dcae288SWarner Losh }
3903b765cfa3SWarner Losh 
3904b765cfa3SWarner Losh /*
3905b765cfa3SWarner Losh  * Get bootonce value by key. The bootonce <key, value> pair is removed from the
3906b765cfa3SWarner Losh  * bootenv nvlist and the remaining nvlist is committed back to disk. This process
3907b765cfa3SWarner Losh  * the bootonce flag since we've reached the point in the boot that we've 'used'
3908b765cfa3SWarner Losh  * the BE. For chained boot scenarios, we may reach this point multiple times (but
3909b765cfa3SWarner Losh  * only remove it and return 0 the first time).
3910b765cfa3SWarner Losh  */
3911b765cfa3SWarner Losh static int
3912b765cfa3SWarner Losh zfs_get_bootonce_spa(spa_t *spa, const char *key, char *buf, size_t size)
3913b765cfa3SWarner Losh {
3914b765cfa3SWarner Losh 	nvlist_t *benv;
3915b765cfa3SWarner Losh 	char *result = NULL;
3916b765cfa3SWarner Losh 	int result_size, rv;
3917b765cfa3SWarner Losh 
3918b765cfa3SWarner Losh 	if ((rv = zfs_get_bootenv_spa(spa, &benv)) != 0)
3919b765cfa3SWarner Losh 		return (rv);
3920b765cfa3SWarner Losh 
3921b765cfa3SWarner Losh 	if ((rv = nvlist_find(benv, key, DATA_TYPE_STRING, NULL,
3922b765cfa3SWarner Losh 	    &result, &result_size)) == 0) {
3923b765cfa3SWarner Losh 		if (result_size == 0) {
3924b765cfa3SWarner Losh 			/* ignore empty string */
3925b765cfa3SWarner Losh 			rv = ENOENT;
3926b765cfa3SWarner Losh 		} else if (buf != NULL) {
3927b765cfa3SWarner Losh 			size = MIN((size_t)result_size + 1, size);
3928b765cfa3SWarner Losh 			strlcpy(buf, result, size);
3929b765cfa3SWarner Losh 		}
3930b765cfa3SWarner Losh 		(void)nvlist_remove(benv, key, DATA_TYPE_STRING);
3931b765cfa3SWarner Losh 		(void)zfs_set_bootenv_spa(spa, benv);
3932b765cfa3SWarner Losh 	}
3933b765cfa3SWarner Losh 
3934b765cfa3SWarner Losh 	return (rv);
3935b765cfa3SWarner Losh }
3936