xref: /freebsd/stand/libsa/zfs/zfsimpl.c (revision c1c4c81fd75a7d0c14011ea1cf6113ecdd80e16a)
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 
3412d62cc2SXin LI #include <sys/endian.h>
35b8902de1SWarner Losh #include <sys/stat.h>
36b8902de1SWarner Losh #include <sys/stdint.h>
37b1b93268SToomas Soome #include <sys/list.h>
383c2db0efSToomas Soome #include <machine/_inttypes.h>
39b8902de1SWarner Losh 
40b8902de1SWarner Losh #include "zfsimpl.h"
41b8902de1SWarner Losh #include "zfssubr.c"
42b8902de1SWarner Losh 
43b8902de1SWarner Losh 
44b8902de1SWarner Losh struct zfsmount {
45b8902de1SWarner Losh 	const spa_t	*spa;
46b8902de1SWarner Losh 	objset_phys_t	objset;
47b8902de1SWarner Losh 	uint64_t	rootobj;
48b8902de1SWarner Losh };
49b8902de1SWarner Losh static struct zfsmount zfsmount __unused;
50b8902de1SWarner Losh 
51b8902de1SWarner Losh /*
52b1b93268SToomas Soome  * The indirect_child_t represents the vdev that we will read from, when we
53b1b93268SToomas Soome  * need to read all copies of the data (e.g. for scrub or reconstruction).
54b1b93268SToomas Soome  * For plain (non-mirror) top-level vdevs (i.e. is_vdev is not a mirror),
55b1b93268SToomas Soome  * ic_vdev is the same as is_vdev.  However, for mirror top-level vdevs,
56b1b93268SToomas Soome  * ic_vdev is a child of the mirror.
57b1b93268SToomas Soome  */
58b1b93268SToomas Soome typedef struct indirect_child {
59b1b93268SToomas Soome 	void *ic_data;
60b1b93268SToomas Soome 	vdev_t *ic_vdev;
61b1b93268SToomas Soome } indirect_child_t;
62b1b93268SToomas Soome 
63b1b93268SToomas Soome /*
64b1b93268SToomas Soome  * The indirect_split_t represents one mapped segment of an i/o to the
65b1b93268SToomas Soome  * indirect vdev. For non-split (contiguously-mapped) blocks, there will be
66b1b93268SToomas Soome  * only one indirect_split_t, with is_split_offset==0 and is_size==io_size.
67b1b93268SToomas Soome  * For split blocks, there will be several of these.
68b1b93268SToomas Soome  */
69b1b93268SToomas Soome typedef struct indirect_split {
70b1b93268SToomas Soome 	list_node_t is_node; /* link on iv_splits */
71b1b93268SToomas Soome 
72b1b93268SToomas Soome 	/*
73b1b93268SToomas Soome 	 * is_split_offset is the offset into the i/o.
74b1b93268SToomas Soome 	 * This is the sum of the previous splits' is_size's.
75b1b93268SToomas Soome 	 */
76b1b93268SToomas Soome 	uint64_t is_split_offset;
77b1b93268SToomas Soome 
78b1b93268SToomas Soome 	vdev_t *is_vdev; /* top-level vdev */
79b1b93268SToomas Soome 	uint64_t is_target_offset; /* offset on is_vdev */
80b1b93268SToomas Soome 	uint64_t is_size;
81b1b93268SToomas Soome 	int is_children; /* number of entries in is_child[] */
82b1b93268SToomas Soome 
83b1b93268SToomas Soome 	/*
84b1b93268SToomas Soome 	 * is_good_child is the child that we are currently using to
85b1b93268SToomas Soome 	 * attempt reconstruction.
86b1b93268SToomas Soome 	 */
87b1b93268SToomas Soome 	int is_good_child;
88b1b93268SToomas Soome 
89b1b93268SToomas Soome 	indirect_child_t is_child[1]; /* variable-length */
90b1b93268SToomas Soome } indirect_split_t;
91b1b93268SToomas Soome 
92b1b93268SToomas Soome /*
93b1b93268SToomas Soome  * The indirect_vsd_t is associated with each i/o to the indirect vdev.
94b1b93268SToomas Soome  * It is the "Vdev-Specific Data" in the zio_t's io_vsd.
95b1b93268SToomas Soome  */
96b1b93268SToomas Soome typedef struct indirect_vsd {
97b1b93268SToomas Soome 	boolean_t iv_split_block;
98b1b93268SToomas Soome 	boolean_t iv_reconstruct;
99b1b93268SToomas Soome 
100b1b93268SToomas Soome 	list_t iv_splits; /* list of indirect_split_t's */
101b1b93268SToomas Soome } indirect_vsd_t;
102b1b93268SToomas Soome 
103b1b93268SToomas Soome /*
104b8902de1SWarner Losh  * List of all vdevs, chained through v_alllink.
105b8902de1SWarner Losh  */
106b8902de1SWarner Losh static vdev_list_t zfs_vdevs;
107b8902de1SWarner Losh 
108b8902de1SWarner Losh /*
109b8902de1SWarner Losh  * List of ZFS features supported for read
110b8902de1SWarner Losh  */
111b8902de1SWarner Losh static const char *features_for_read[] = {
112b8902de1SWarner Losh 	"org.illumos:lz4_compress",
113b8902de1SWarner Losh 	"com.delphix:hole_birth",
114b8902de1SWarner Losh 	"com.delphix:extensible_dataset",
115b8902de1SWarner Losh 	"com.delphix:embedded_data",
116b8902de1SWarner Losh 	"org.open-zfs:large_blocks",
117b8902de1SWarner Losh 	"org.illumos:sha512",
118b8902de1SWarner Losh 	"org.illumos:skein",
119b8902de1SWarner Losh 	"org.zfsonlinux:large_dnode",
1207c52f914SToomas Soome 	"com.joyent:multi_vdev_crash_dump",
1216459a61eSMatt Macy 	"com.delphix:spacemap_histogram",
1226459a61eSMatt Macy 	"com.delphix:zpool_checkpoint",
1236459a61eSMatt Macy 	"com.delphix:spacemap_v2",
1246459a61eSMatt Macy 	"com.datto:encryption",
1256459a61eSMatt Macy 	"org.zfsonlinux:allocation_classes",
1266459a61eSMatt Macy 	"com.datto:resilver_defer",
127b1b93268SToomas Soome 	"com.delphix:device_removal",
128b1b93268SToomas Soome 	"com.delphix:obsolete_counts",
129489912daSAndriy Gapon 	"com.intel:allocation_classes",
130b8902de1SWarner Losh 	NULL
131b8902de1SWarner Losh };
132b8902de1SWarner Losh 
133b8902de1SWarner Losh /*
134b8902de1SWarner Losh  * List of all pools, chained through spa_link.
135b8902de1SWarner Losh  */
136b8902de1SWarner Losh static spa_list_t zfs_pools;
137b8902de1SWarner Losh 
138b8902de1SWarner Losh static const dnode_phys_t *dnode_cache_obj;
139b8902de1SWarner Losh static uint64_t dnode_cache_bn;
140b8902de1SWarner Losh static char *dnode_cache_buf;
141b8902de1SWarner Losh 
142b8902de1SWarner Losh static int zio_read(const spa_t *spa, const blkptr_t *bp, void *buf);
143b8902de1SWarner Losh static int zfs_get_root(const spa_t *spa, uint64_t *objid);
144b8902de1SWarner Losh static int zfs_rlookup(const spa_t *spa, uint64_t objnum, char *result);
145b8902de1SWarner Losh static int zap_lookup(const spa_t *spa, const dnode_phys_t *dnode,
146b8902de1SWarner Losh     const char *name, uint64_t integer_size, uint64_t num_integers,
147b8902de1SWarner Losh     void *value);
148b1b93268SToomas Soome static int objset_get_dnode(const spa_t *, const objset_phys_t *, uint64_t,
149b1b93268SToomas Soome     dnode_phys_t *);
150b1b93268SToomas Soome static int dnode_read(const spa_t *, const dnode_phys_t *, off_t, void *,
151b1b93268SToomas Soome     size_t);
152b1b93268SToomas Soome static int vdev_indirect_read(vdev_t *, const blkptr_t *, void *, off_t,
153b1b93268SToomas Soome     size_t);
154b1b93268SToomas Soome static int vdev_mirror_read(vdev_t *, const blkptr_t *, void *, off_t, size_t);
155b1b93268SToomas Soome vdev_indirect_mapping_t *vdev_indirect_mapping_open(spa_t *, objset_phys_t *,
156b1b93268SToomas Soome     uint64_t);
157b1b93268SToomas Soome vdev_indirect_mapping_entry_phys_t *
158b1b93268SToomas Soome     vdev_indirect_mapping_duplicate_adjacent_entries(vdev_t *, uint64_t,
159b1b93268SToomas Soome     uint64_t, uint64_t *);
160b8902de1SWarner Losh 
161b8902de1SWarner Losh static void
162b8902de1SWarner Losh zfs_init(void)
163b8902de1SWarner Losh {
164b8902de1SWarner Losh 	STAILQ_INIT(&zfs_vdevs);
165b8902de1SWarner Losh 	STAILQ_INIT(&zfs_pools);
166b8902de1SWarner Losh 
167b8902de1SWarner Losh 	dnode_cache_buf = malloc(SPA_MAXBLOCKSIZE);
168b8902de1SWarner Losh 
169b8902de1SWarner Losh 	zfs_init_crc();
170b8902de1SWarner Losh }
171b8902de1SWarner Losh 
172b8902de1SWarner Losh static int
173b8902de1SWarner Losh xdr_int(const unsigned char **xdr, int *ip)
174b8902de1SWarner Losh {
17512d62cc2SXin LI 	*ip = be32dec(*xdr);
176b8902de1SWarner Losh 	(*xdr) += 4;
177b8902de1SWarner Losh 	return (0);
178b8902de1SWarner Losh }
179b8902de1SWarner Losh 
180b8902de1SWarner Losh static int
181b8902de1SWarner Losh xdr_u_int(const unsigned char **xdr, u_int *ip)
182b8902de1SWarner Losh {
18312d62cc2SXin LI 	*ip = be32dec(*xdr);
184b8902de1SWarner Losh 	(*xdr) += 4;
185b8902de1SWarner Losh 	return (0);
186b8902de1SWarner Losh }
187b8902de1SWarner Losh 
188b8902de1SWarner Losh static int
189b8902de1SWarner Losh xdr_uint64_t(const unsigned char **xdr, uint64_t *lp)
190b8902de1SWarner Losh {
191b8902de1SWarner Losh 	u_int hi, lo;
192b8902de1SWarner Losh 
193b8902de1SWarner Losh 	xdr_u_int(xdr, &hi);
194b8902de1SWarner Losh 	xdr_u_int(xdr, &lo);
195b8902de1SWarner Losh 	*lp = (((uint64_t)hi) << 32) | lo;
196b8902de1SWarner Losh 	return (0);
197b8902de1SWarner Losh }
198b8902de1SWarner Losh 
199b8902de1SWarner Losh static int
200b8902de1SWarner Losh nvlist_find(const unsigned char *nvlist, const char *name, int type,
201b8902de1SWarner Losh     int *elementsp, void *valuep)
202b8902de1SWarner Losh {
203b8902de1SWarner Losh 	const unsigned char *p, *pair;
204b8902de1SWarner Losh 	int junk;
205b8902de1SWarner Losh 	int encoded_size, decoded_size;
206b8902de1SWarner Losh 
207b8902de1SWarner Losh 	p = nvlist;
208b8902de1SWarner Losh 	xdr_int(&p, &junk);
209b8902de1SWarner Losh 	xdr_int(&p, &junk);
210b8902de1SWarner Losh 
211b8902de1SWarner Losh 	pair = p;
212b8902de1SWarner Losh 	xdr_int(&p, &encoded_size);
213b8902de1SWarner Losh 	xdr_int(&p, &decoded_size);
214b8902de1SWarner Losh 	while (encoded_size && decoded_size) {
215b8902de1SWarner Losh 		int namelen, pairtype, elements;
216b8902de1SWarner Losh 		const char *pairname;
217b8902de1SWarner Losh 
218b8902de1SWarner Losh 		xdr_int(&p, &namelen);
219b8902de1SWarner Losh 		pairname = (const char *)p;
220b8902de1SWarner Losh 		p += roundup(namelen, 4);
221b8902de1SWarner Losh 		xdr_int(&p, &pairtype);
222b8902de1SWarner Losh 
2232e6bb655SToomas Soome 		if (memcmp(name, pairname, namelen) == 0 && type == pairtype) {
224b8902de1SWarner Losh 			xdr_int(&p, &elements);
225b8902de1SWarner Losh 			if (elementsp)
226b8902de1SWarner Losh 				*elementsp = elements;
227b8902de1SWarner Losh 			if (type == DATA_TYPE_UINT64) {
228b8902de1SWarner Losh 				xdr_uint64_t(&p, (uint64_t *)valuep);
229b8902de1SWarner Losh 				return (0);
230b8902de1SWarner Losh 			} else if (type == DATA_TYPE_STRING) {
231b8902de1SWarner Losh 				int len;
232b8902de1SWarner Losh 				xdr_int(&p, &len);
233b8902de1SWarner Losh 				(*(const char **)valuep) = (const char *)p;
234b8902de1SWarner Losh 				return (0);
2352e6bb655SToomas Soome 			} else if (type == DATA_TYPE_NVLIST ||
2362e6bb655SToomas Soome 			    type == DATA_TYPE_NVLIST_ARRAY) {
237b8902de1SWarner Losh 				(*(const unsigned char **)valuep) =
238b8902de1SWarner Losh 				    (const unsigned char *)p;
239b8902de1SWarner Losh 				return (0);
240b8902de1SWarner Losh 			} else {
241b8902de1SWarner Losh 				return (EIO);
242b8902de1SWarner Losh 			}
243b8902de1SWarner Losh 		} else {
244b8902de1SWarner Losh 			/*
2452e6bb655SToomas Soome 			 * Not the pair we are looking for, skip to the
2462e6bb655SToomas Soome 			 * next one.
247b8902de1SWarner Losh 			 */
248b8902de1SWarner Losh 			p = pair + encoded_size;
249b8902de1SWarner Losh 		}
250b8902de1SWarner Losh 
251b8902de1SWarner Losh 		pair = p;
252b8902de1SWarner Losh 		xdr_int(&p, &encoded_size);
253b8902de1SWarner Losh 		xdr_int(&p, &decoded_size);
254b8902de1SWarner Losh 	}
255b8902de1SWarner Losh 
256b8902de1SWarner Losh 	return (EIO);
257b8902de1SWarner Losh }
258b8902de1SWarner Losh 
259b8902de1SWarner Losh static int
260b8902de1SWarner Losh nvlist_check_features_for_read(const unsigned char *nvlist)
261b8902de1SWarner Losh {
262b8902de1SWarner Losh 	const unsigned char *p, *pair;
263b8902de1SWarner Losh 	int junk;
264b8902de1SWarner Losh 	int encoded_size, decoded_size;
265b8902de1SWarner Losh 	int rc;
266b8902de1SWarner Losh 
267b8902de1SWarner Losh 	rc = 0;
268b8902de1SWarner Losh 
269b8902de1SWarner Losh 	p = nvlist;
270b8902de1SWarner Losh 	xdr_int(&p, &junk);
271b8902de1SWarner Losh 	xdr_int(&p, &junk);
272b8902de1SWarner Losh 
273b8902de1SWarner Losh 	pair = p;
274b8902de1SWarner Losh 	xdr_int(&p, &encoded_size);
275b8902de1SWarner Losh 	xdr_int(&p, &decoded_size);
276b8902de1SWarner Losh 	while (encoded_size && decoded_size) {
277b8902de1SWarner Losh 		int namelen, pairtype;
278b8902de1SWarner Losh 		const char *pairname;
279b8902de1SWarner Losh 		int i, found;
280b8902de1SWarner Losh 
281b8902de1SWarner Losh 		found = 0;
282b8902de1SWarner Losh 
283b8902de1SWarner Losh 		xdr_int(&p, &namelen);
284b8902de1SWarner Losh 		pairname = (const char *)p;
285b8902de1SWarner Losh 		p += roundup(namelen, 4);
286b8902de1SWarner Losh 		xdr_int(&p, &pairtype);
287b8902de1SWarner Losh 
288b8902de1SWarner Losh 		for (i = 0; features_for_read[i] != NULL; i++) {
2892e6bb655SToomas Soome 			if (memcmp(pairname, features_for_read[i],
2902e6bb655SToomas Soome 			    namelen) == 0) {
291b8902de1SWarner Losh 				found = 1;
292b8902de1SWarner Losh 				break;
293b8902de1SWarner Losh 			}
294b8902de1SWarner Losh 		}
295b8902de1SWarner Losh 
296b8902de1SWarner Losh 		if (!found) {
297b8902de1SWarner Losh 			printf("ZFS: unsupported feature: %s\n", pairname);
298b8902de1SWarner Losh 			rc = EIO;
299b8902de1SWarner Losh 		}
300b8902de1SWarner Losh 
301b8902de1SWarner Losh 		p = pair + encoded_size;
302b8902de1SWarner Losh 
303b8902de1SWarner Losh 		pair = p;
304b8902de1SWarner Losh 		xdr_int(&p, &encoded_size);
305b8902de1SWarner Losh 		xdr_int(&p, &decoded_size);
306b8902de1SWarner Losh 	}
307b8902de1SWarner Losh 
308b8902de1SWarner Losh 	return (rc);
309b8902de1SWarner Losh }
310b8902de1SWarner Losh 
311b8902de1SWarner Losh /*
312b8902de1SWarner Losh  * Return the next nvlist in an nvlist array.
313b8902de1SWarner Losh  */
314b8902de1SWarner Losh static const unsigned char *
315b8902de1SWarner Losh nvlist_next(const unsigned char *nvlist)
316b8902de1SWarner Losh {
317b8902de1SWarner Losh 	const unsigned char *p, *pair;
318b8902de1SWarner Losh 	int junk;
319b8902de1SWarner Losh 	int encoded_size, decoded_size;
320b8902de1SWarner Losh 
321b8902de1SWarner Losh 	p = nvlist;
322b8902de1SWarner Losh 	xdr_int(&p, &junk);
323b8902de1SWarner Losh 	xdr_int(&p, &junk);
324b8902de1SWarner Losh 
325b8902de1SWarner Losh 	pair = p;
326b8902de1SWarner Losh 	xdr_int(&p, &encoded_size);
327b8902de1SWarner Losh 	xdr_int(&p, &decoded_size);
328b8902de1SWarner Losh 	while (encoded_size && decoded_size) {
329b8902de1SWarner Losh 		p = pair + encoded_size;
330b8902de1SWarner Losh 
331b8902de1SWarner Losh 		pair = p;
332b8902de1SWarner Losh 		xdr_int(&p, &encoded_size);
333b8902de1SWarner Losh 		xdr_int(&p, &decoded_size);
334b8902de1SWarner Losh 	}
335b8902de1SWarner Losh 
3362e6bb655SToomas Soome 	return (p);
337b8902de1SWarner Losh }
338b8902de1SWarner Losh 
339b8902de1SWarner Losh #ifdef TEST
340b8902de1SWarner Losh 
341b8902de1SWarner Losh static const unsigned char *
342b8902de1SWarner Losh nvlist_print(const unsigned char *nvlist, unsigned int indent)
343b8902de1SWarner Losh {
344b8902de1SWarner Losh 	static const char *typenames[] = {
345b8902de1SWarner Losh 		"DATA_TYPE_UNKNOWN",
346b8902de1SWarner Losh 		"DATA_TYPE_BOOLEAN",
347b8902de1SWarner Losh 		"DATA_TYPE_BYTE",
348b8902de1SWarner Losh 		"DATA_TYPE_INT16",
349b8902de1SWarner Losh 		"DATA_TYPE_UINT16",
350b8902de1SWarner Losh 		"DATA_TYPE_INT32",
351b8902de1SWarner Losh 		"DATA_TYPE_UINT32",
352b8902de1SWarner Losh 		"DATA_TYPE_INT64",
353b8902de1SWarner Losh 		"DATA_TYPE_UINT64",
354b8902de1SWarner Losh 		"DATA_TYPE_STRING",
355b8902de1SWarner Losh 		"DATA_TYPE_BYTE_ARRAY",
356b8902de1SWarner Losh 		"DATA_TYPE_INT16_ARRAY",
357b8902de1SWarner Losh 		"DATA_TYPE_UINT16_ARRAY",
358b8902de1SWarner Losh 		"DATA_TYPE_INT32_ARRAY",
359b8902de1SWarner Losh 		"DATA_TYPE_UINT32_ARRAY",
360b8902de1SWarner Losh 		"DATA_TYPE_INT64_ARRAY",
361b8902de1SWarner Losh 		"DATA_TYPE_UINT64_ARRAY",
362b8902de1SWarner Losh 		"DATA_TYPE_STRING_ARRAY",
363b8902de1SWarner Losh 		"DATA_TYPE_HRTIME",
364b8902de1SWarner Losh 		"DATA_TYPE_NVLIST",
365b8902de1SWarner Losh 		"DATA_TYPE_NVLIST_ARRAY",
366b8902de1SWarner Losh 		"DATA_TYPE_BOOLEAN_VALUE",
367b8902de1SWarner Losh 		"DATA_TYPE_INT8",
368b8902de1SWarner Losh 		"DATA_TYPE_UINT8",
369b8902de1SWarner Losh 		"DATA_TYPE_BOOLEAN_ARRAY",
370b8902de1SWarner Losh 		"DATA_TYPE_INT8_ARRAY",
371b8902de1SWarner Losh 		"DATA_TYPE_UINT8_ARRAY"
372b8902de1SWarner Losh 	};
373b8902de1SWarner Losh 
374b8902de1SWarner Losh 	unsigned int i, j;
375b8902de1SWarner Losh 	const unsigned char *p, *pair;
376b8902de1SWarner Losh 	int junk;
377b8902de1SWarner Losh 	int encoded_size, decoded_size;
378b8902de1SWarner Losh 
379b8902de1SWarner Losh 	p = nvlist;
380b8902de1SWarner Losh 	xdr_int(&p, &junk);
381b8902de1SWarner Losh 	xdr_int(&p, &junk);
382b8902de1SWarner Losh 
383b8902de1SWarner Losh 	pair = p;
384b8902de1SWarner Losh 	xdr_int(&p, &encoded_size);
385b8902de1SWarner Losh 	xdr_int(&p, &decoded_size);
386b8902de1SWarner Losh 	while (encoded_size && decoded_size) {
387b8902de1SWarner Losh 		int namelen, pairtype, elements;
388b8902de1SWarner Losh 		const char *pairname;
389b8902de1SWarner Losh 
390b8902de1SWarner Losh 		xdr_int(&p, &namelen);
391b8902de1SWarner Losh 		pairname = (const char *)p;
392b8902de1SWarner Losh 		p += roundup(namelen, 4);
393b8902de1SWarner Losh 		xdr_int(&p, &pairtype);
394b8902de1SWarner Losh 
395b8902de1SWarner Losh 		for (i = 0; i < indent; i++)
396b8902de1SWarner Losh 			printf(" ");
397b8902de1SWarner Losh 		printf("%s %s", typenames[pairtype], pairname);
398b8902de1SWarner Losh 
399b8902de1SWarner Losh 		xdr_int(&p, &elements);
400b8902de1SWarner Losh 		switch (pairtype) {
401b8902de1SWarner Losh 		case DATA_TYPE_UINT64: {
402b8902de1SWarner Losh 			uint64_t val;
403b8902de1SWarner Losh 			xdr_uint64_t(&p, &val);
404b8902de1SWarner Losh 			printf(" = 0x%jx\n", (uintmax_t)val);
405b8902de1SWarner Losh 			break;
406b8902de1SWarner Losh 		}
407b8902de1SWarner Losh 
408b8902de1SWarner Losh 		case DATA_TYPE_STRING: {
409b8902de1SWarner Losh 			int len;
410b8902de1SWarner Losh 			xdr_int(&p, &len);
411b8902de1SWarner Losh 			printf(" = \"%s\"\n", p);
412b8902de1SWarner Losh 			break;
413b8902de1SWarner Losh 		}
414b8902de1SWarner Losh 
415b8902de1SWarner Losh 		case DATA_TYPE_NVLIST:
416b8902de1SWarner Losh 			printf("\n");
417b8902de1SWarner Losh 			nvlist_print(p, indent + 1);
418b8902de1SWarner Losh 			break;
419b8902de1SWarner Losh 
420b8902de1SWarner Losh 		case DATA_TYPE_NVLIST_ARRAY:
421b8902de1SWarner Losh 			for (j = 0; j < elements; j++) {
422b8902de1SWarner Losh 				printf("[%d]\n", j);
423b8902de1SWarner Losh 				p = nvlist_print(p, indent + 1);
424b8902de1SWarner Losh 				if (j != elements - 1) {
425b8902de1SWarner Losh 					for (i = 0; i < indent; i++)
426b8902de1SWarner Losh 						printf(" ");
4272e6bb655SToomas Soome 					printf("%s %s", typenames[pairtype],
4282e6bb655SToomas Soome 					    pairname);
429b8902de1SWarner Losh 				}
430b8902de1SWarner Losh 			}
431b8902de1SWarner Losh 			break;
432b8902de1SWarner Losh 
433b8902de1SWarner Losh 		default:
434b8902de1SWarner Losh 			printf("\n");
435b8902de1SWarner Losh 		}
436b8902de1SWarner Losh 
437b8902de1SWarner Losh 		p = pair + encoded_size;
438b8902de1SWarner Losh 
439b8902de1SWarner Losh 		pair = p;
440b8902de1SWarner Losh 		xdr_int(&p, &encoded_size);
441b8902de1SWarner Losh 		xdr_int(&p, &decoded_size);
442b8902de1SWarner Losh 	}
443b8902de1SWarner Losh 
4442e6bb655SToomas Soome 	return (p);
445b8902de1SWarner Losh }
446b8902de1SWarner Losh 
447b8902de1SWarner Losh #endif
448b8902de1SWarner Losh 
449b8902de1SWarner Losh static int
450b8902de1SWarner Losh vdev_read_phys(vdev_t *vdev, const blkptr_t *bp, void *buf,
451b8902de1SWarner Losh     off_t offset, size_t size)
452b8902de1SWarner Losh {
453b8902de1SWarner Losh 	size_t psize;
454b8902de1SWarner Losh 	int rc;
455b8902de1SWarner Losh 
456b8902de1SWarner Losh 	if (!vdev->v_phys_read)
457b8902de1SWarner Losh 		return (EIO);
458b8902de1SWarner Losh 
459b8902de1SWarner Losh 	if (bp) {
460b8902de1SWarner Losh 		psize = BP_GET_PSIZE(bp);
461b8902de1SWarner Losh 	} else {
462b8902de1SWarner Losh 		psize = size;
463b8902de1SWarner Losh 	}
464b8902de1SWarner Losh 
465b8902de1SWarner Losh 	rc = vdev->v_phys_read(vdev, vdev->v_read_priv, offset, buf, psize);
4663c2db0efSToomas Soome 	if (rc == 0) {
467b1b93268SToomas Soome 		if (bp != NULL)
4683c2db0efSToomas Soome 			rc = zio_checksum_verify(vdev->v_spa, bp, buf);
4693c2db0efSToomas Soome 	}
470b8902de1SWarner Losh 
4713c2db0efSToomas Soome 	return (rc);
472b8902de1SWarner Losh }
473b8902de1SWarner Losh 
474b1b93268SToomas Soome typedef struct remap_segment {
475b1b93268SToomas Soome 	vdev_t *rs_vd;
476b1b93268SToomas Soome 	uint64_t rs_offset;
477b1b93268SToomas Soome 	uint64_t rs_asize;
478b1b93268SToomas Soome 	uint64_t rs_split_offset;
479b1b93268SToomas Soome 	list_node_t rs_node;
480b1b93268SToomas Soome } remap_segment_t;
481b1b93268SToomas Soome 
482b1b93268SToomas Soome static remap_segment_t *
483b1b93268SToomas Soome rs_alloc(vdev_t *vd, uint64_t offset, uint64_t asize, uint64_t split_offset)
484b1b93268SToomas Soome {
485b1b93268SToomas Soome 	remap_segment_t *rs = malloc(sizeof (remap_segment_t));
486b1b93268SToomas Soome 
487b1b93268SToomas Soome 	if (rs != NULL) {
488b1b93268SToomas Soome 		rs->rs_vd = vd;
489b1b93268SToomas Soome 		rs->rs_offset = offset;
490b1b93268SToomas Soome 		rs->rs_asize = asize;
491b1b93268SToomas Soome 		rs->rs_split_offset = split_offset;
492b1b93268SToomas Soome 	}
493b1b93268SToomas Soome 
494b1b93268SToomas Soome 	return (rs);
495b1b93268SToomas Soome }
496b1b93268SToomas Soome 
497b1b93268SToomas Soome vdev_indirect_mapping_t *
498b1b93268SToomas Soome vdev_indirect_mapping_open(spa_t *spa, objset_phys_t *os,
499b1b93268SToomas Soome     uint64_t mapping_object)
500b1b93268SToomas Soome {
501b1b93268SToomas Soome 	vdev_indirect_mapping_t *vim;
502b1b93268SToomas Soome 	vdev_indirect_mapping_phys_t *vim_phys;
503b1b93268SToomas Soome 	int rc;
504b1b93268SToomas Soome 
505b1b93268SToomas Soome 	vim = calloc(1, sizeof (*vim));
506b1b93268SToomas Soome 	if (vim == NULL)
507b1b93268SToomas Soome 		return (NULL);
508b1b93268SToomas Soome 
509b1b93268SToomas Soome 	vim->vim_dn = calloc(1, sizeof (*vim->vim_dn));
510b1b93268SToomas Soome 	if (vim->vim_dn == NULL) {
511b1b93268SToomas Soome 		free(vim);
512b1b93268SToomas Soome 		return (NULL);
513b1b93268SToomas Soome 	}
514b1b93268SToomas Soome 
515b1b93268SToomas Soome 	rc = objset_get_dnode(spa, os, mapping_object, vim->vim_dn);
516b1b93268SToomas Soome 	if (rc != 0) {
517b1b93268SToomas Soome 		free(vim->vim_dn);
518b1b93268SToomas Soome 		free(vim);
519b1b93268SToomas Soome 		return (NULL);
520b1b93268SToomas Soome 	}
521b1b93268SToomas Soome 
522b1b93268SToomas Soome 	vim->vim_spa = spa;
523b1b93268SToomas Soome 	vim->vim_phys = malloc(sizeof (*vim->vim_phys));
524b1b93268SToomas Soome 	if (vim->vim_phys == NULL) {
525b1b93268SToomas Soome 		free(vim->vim_dn);
526b1b93268SToomas Soome 		free(vim);
527b1b93268SToomas Soome 		return (NULL);
528b1b93268SToomas Soome 	}
529b1b93268SToomas Soome 
530b1b93268SToomas Soome 	vim_phys = (vdev_indirect_mapping_phys_t *)DN_BONUS(vim->vim_dn);
531b1b93268SToomas Soome 	*vim->vim_phys = *vim_phys;
532b1b93268SToomas Soome 
533b1b93268SToomas Soome 	vim->vim_objset = os;
534b1b93268SToomas Soome 	vim->vim_object = mapping_object;
535b1b93268SToomas Soome 	vim->vim_entries = NULL;
536b1b93268SToomas Soome 
537b1b93268SToomas Soome 	vim->vim_havecounts =
538b1b93268SToomas Soome 	    (vim->vim_dn->dn_bonuslen > VDEV_INDIRECT_MAPPING_SIZE_V0);
5393c2db0efSToomas Soome 
540b1b93268SToomas Soome 	return (vim);
541b1b93268SToomas Soome }
542b1b93268SToomas Soome 
543b1b93268SToomas Soome /*
544b1b93268SToomas Soome  * Compare an offset with an indirect mapping entry; there are three
545b1b93268SToomas Soome  * possible scenarios:
546b1b93268SToomas Soome  *
547b1b93268SToomas Soome  *     1. The offset is "less than" the mapping entry; meaning the
548b1b93268SToomas Soome  *        offset is less than the source offset of the mapping entry. In
549b1b93268SToomas Soome  *        this case, there is no overlap between the offset and the
550b1b93268SToomas Soome  *        mapping entry and -1 will be returned.
551b1b93268SToomas Soome  *
552b1b93268SToomas Soome  *     2. The offset is "greater than" the mapping entry; meaning the
553b1b93268SToomas Soome  *        offset is greater than the mapping entry's source offset plus
554b1b93268SToomas Soome  *        the entry's size. In this case, there is no overlap between
555b1b93268SToomas Soome  *        the offset and the mapping entry and 1 will be returned.
556b1b93268SToomas Soome  *
557b1b93268SToomas Soome  *        NOTE: If the offset is actually equal to the entry's offset
558b1b93268SToomas Soome  *        plus size, this is considered to be "greater" than the entry,
559b1b93268SToomas Soome  *        and this case applies (i.e. 1 will be returned). Thus, the
560b1b93268SToomas Soome  *        entry's "range" can be considered to be inclusive at its
561b1b93268SToomas Soome  *        start, but exclusive at its end: e.g. [src, src + size).
562b1b93268SToomas Soome  *
563b1b93268SToomas Soome  *     3. The last case to consider is if the offset actually falls
564b1b93268SToomas Soome  *        within the mapping entry's range. If this is the case, the
565b1b93268SToomas Soome  *        offset is considered to be "equal to" the mapping entry and
566b1b93268SToomas Soome  *        0 will be returned.
567b1b93268SToomas Soome  *
568b1b93268SToomas Soome  *        NOTE: If the offset is equal to the entry's source offset,
569b1b93268SToomas Soome  *        this case applies and 0 will be returned. If the offset is
570b1b93268SToomas Soome  *        equal to the entry's source plus its size, this case does
571b1b93268SToomas Soome  *        *not* apply (see "NOTE" above for scenario 2), and 1 will be
572b1b93268SToomas Soome  *        returned.
573b1b93268SToomas Soome  */
574b1b93268SToomas Soome static int
575b1b93268SToomas Soome dva_mapping_overlap_compare(const void *v_key, const void *v_array_elem)
576b1b93268SToomas Soome {
577b1b93268SToomas Soome 	const uint64_t *key = v_key;
578b1b93268SToomas Soome 	const vdev_indirect_mapping_entry_phys_t *array_elem =
579b1b93268SToomas Soome 	    v_array_elem;
580b1b93268SToomas Soome 	uint64_t src_offset = DVA_MAPPING_GET_SRC_OFFSET(array_elem);
581b1b93268SToomas Soome 
582b1b93268SToomas Soome 	if (*key < src_offset) {
583b1b93268SToomas Soome 		return (-1);
584b1b93268SToomas Soome 	} else if (*key < src_offset + DVA_GET_ASIZE(&array_elem->vimep_dst)) {
585b1b93268SToomas Soome 		return (0);
586b1b93268SToomas Soome 	} else {
587b1b93268SToomas Soome 		return (1);
588b1b93268SToomas Soome 	}
589b1b93268SToomas Soome }
590b1b93268SToomas Soome 
591b1b93268SToomas Soome /*
592b1b93268SToomas Soome  * Return array entry.
593b1b93268SToomas Soome  */
594b1b93268SToomas Soome static vdev_indirect_mapping_entry_phys_t *
595b1b93268SToomas Soome vdev_indirect_mapping_entry(vdev_indirect_mapping_t *vim, uint64_t index)
596b1b93268SToomas Soome {
597b1b93268SToomas Soome 	uint64_t size;
598b1b93268SToomas Soome 	off_t offset = 0;
599b1b93268SToomas Soome 	int rc;
600b1b93268SToomas Soome 
601b1b93268SToomas Soome 	if (vim->vim_phys->vimp_num_entries == 0)
602b1b93268SToomas Soome 		return (NULL);
603b1b93268SToomas Soome 
604b1b93268SToomas Soome 	if (vim->vim_entries == NULL) {
605b1b93268SToomas Soome 		uint64_t bsize;
606b1b93268SToomas Soome 
607b1b93268SToomas Soome 		bsize = vim->vim_dn->dn_datablkszsec << SPA_MINBLOCKSHIFT;
608b1b93268SToomas Soome 		size = vim->vim_phys->vimp_num_entries *
609b1b93268SToomas Soome 		    sizeof (*vim->vim_entries);
610b1b93268SToomas Soome 		if (size > bsize) {
611b1b93268SToomas Soome 			size = bsize / sizeof (*vim->vim_entries);
612b1b93268SToomas Soome 			size *= sizeof (*vim->vim_entries);
613b1b93268SToomas Soome 		}
614b1b93268SToomas Soome 		vim->vim_entries = malloc(size);
615b1b93268SToomas Soome 		if (vim->vim_entries == NULL)
616b1b93268SToomas Soome 			return (NULL);
617b1b93268SToomas Soome 		vim->vim_num_entries = size / sizeof (*vim->vim_entries);
618b1b93268SToomas Soome 		offset = index * sizeof (*vim->vim_entries);
619b1b93268SToomas Soome 	}
620b1b93268SToomas Soome 
621b1b93268SToomas Soome 	/* We have data in vim_entries */
622b1b93268SToomas Soome 	if (offset == 0) {
623b1b93268SToomas Soome 		if (index >= vim->vim_entry_offset &&
624b1b93268SToomas Soome 		    index <= vim->vim_entry_offset + vim->vim_num_entries) {
625b1b93268SToomas Soome 			index -= vim->vim_entry_offset;
626b1b93268SToomas Soome 			return (&vim->vim_entries[index]);
627b1b93268SToomas Soome 		}
628b1b93268SToomas Soome 		offset = index * sizeof (*vim->vim_entries);
629b1b93268SToomas Soome 	}
630b1b93268SToomas Soome 
631b1b93268SToomas Soome 	vim->vim_entry_offset = index;
632b1b93268SToomas Soome 	size = vim->vim_num_entries * sizeof (*vim->vim_entries);
633b1b93268SToomas Soome 	rc = dnode_read(vim->vim_spa, vim->vim_dn, offset, vim->vim_entries,
634b1b93268SToomas Soome 	    size);
635b1b93268SToomas Soome 	if (rc != 0) {
636b1b93268SToomas Soome 		/* Read error, invalidate vim_entries. */
637b1b93268SToomas Soome 		free(vim->vim_entries);
638b1b93268SToomas Soome 		vim->vim_entries = NULL;
639b1b93268SToomas Soome 		return (NULL);
640b1b93268SToomas Soome 	}
641b1b93268SToomas Soome 	index -= vim->vim_entry_offset;
642b1b93268SToomas Soome 	return (&vim->vim_entries[index]);
643b1b93268SToomas Soome }
644b1b93268SToomas Soome 
645b1b93268SToomas Soome /*
646b1b93268SToomas Soome  * Returns the mapping entry for the given offset.
647b1b93268SToomas Soome  *
648b1b93268SToomas Soome  * It's possible that the given offset will not be in the mapping table
649b1b93268SToomas Soome  * (i.e. no mapping entries contain this offset), in which case, the
650b1b93268SToomas Soome  * return value value depends on the "next_if_missing" parameter.
651b1b93268SToomas Soome  *
652b1b93268SToomas Soome  * If the offset is not found in the table and "next_if_missing" is
653b1b93268SToomas Soome  * B_FALSE, then NULL will always be returned. The behavior is intended
654b1b93268SToomas Soome  * to allow consumers to get the entry corresponding to the offset
655b1b93268SToomas Soome  * parameter, iff the offset overlaps with an entry in the table.
656b1b93268SToomas Soome  *
657b1b93268SToomas Soome  * If the offset is not found in the table and "next_if_missing" is
658b1b93268SToomas Soome  * B_TRUE, then the entry nearest to the given offset will be returned,
659b1b93268SToomas Soome  * such that the entry's source offset is greater than the offset
660b1b93268SToomas Soome  * passed in (i.e. the "next" mapping entry in the table is returned, if
661b1b93268SToomas Soome  * the offset is missing from the table). If there are no entries whose
662b1b93268SToomas Soome  * source offset is greater than the passed in offset, NULL is returned.
663b1b93268SToomas Soome  */
664b1b93268SToomas Soome static vdev_indirect_mapping_entry_phys_t *
665b1b93268SToomas Soome vdev_indirect_mapping_entry_for_offset(vdev_indirect_mapping_t *vim,
666b1b93268SToomas Soome     uint64_t offset)
667b1b93268SToomas Soome {
668b1b93268SToomas Soome 	ASSERT(vim->vim_phys->vimp_num_entries > 0);
669b1b93268SToomas Soome 
670b1b93268SToomas Soome 	vdev_indirect_mapping_entry_phys_t *entry;
671b1b93268SToomas Soome 
672b1b93268SToomas Soome 	uint64_t last = vim->vim_phys->vimp_num_entries - 1;
673b1b93268SToomas Soome 	uint64_t base = 0;
674b1b93268SToomas Soome 
675b1b93268SToomas Soome 	/*
676b1b93268SToomas Soome 	 * We don't define these inside of the while loop because we use
677b1b93268SToomas Soome 	 * their value in the case that offset isn't in the mapping.
678b1b93268SToomas Soome 	 */
679b1b93268SToomas Soome 	uint64_t mid;
680b1b93268SToomas Soome 	int result;
681b1b93268SToomas Soome 
682b1b93268SToomas Soome 	while (last >= base) {
683b1b93268SToomas Soome 		mid = base + ((last - base) >> 1);
684b1b93268SToomas Soome 
685b1b93268SToomas Soome 		entry = vdev_indirect_mapping_entry(vim, mid);
686b1b93268SToomas Soome 		if (entry == NULL)
687b1b93268SToomas Soome 			break;
688b1b93268SToomas Soome 		result = dva_mapping_overlap_compare(&offset, entry);
689b1b93268SToomas Soome 
690b1b93268SToomas Soome 		if (result == 0) {
691b1b93268SToomas Soome 			break;
692b1b93268SToomas Soome 		} else if (result < 0) {
693b1b93268SToomas Soome 			last = mid - 1;
694b1b93268SToomas Soome 		} else {
695b1b93268SToomas Soome 			base = mid + 1;
696b1b93268SToomas Soome 		}
697b1b93268SToomas Soome 	}
698b1b93268SToomas Soome 	return (entry);
699b1b93268SToomas Soome }
700b1b93268SToomas Soome 
701b1b93268SToomas Soome /*
702b1b93268SToomas Soome  * Given an indirect vdev and an extent on that vdev, it duplicates the
703b1b93268SToomas Soome  * physical entries of the indirect mapping that correspond to the extent
704b1b93268SToomas Soome  * to a new array and returns a pointer to it. In addition, copied_entries
705b1b93268SToomas Soome  * is populated with the number of mapping entries that were duplicated.
706b1b93268SToomas Soome  *
707b1b93268SToomas Soome  * Finally, since we are doing an allocation, it is up to the caller to
708b1b93268SToomas Soome  * free the array allocated in this function.
709b1b93268SToomas Soome  */
710b1b93268SToomas Soome vdev_indirect_mapping_entry_phys_t *
711b1b93268SToomas Soome vdev_indirect_mapping_duplicate_adjacent_entries(vdev_t *vd, uint64_t offset,
712b1b93268SToomas Soome     uint64_t asize, uint64_t *copied_entries)
713b1b93268SToomas Soome {
714b1b93268SToomas Soome 	vdev_indirect_mapping_entry_phys_t *duplicate_mappings = NULL;
715b1b93268SToomas Soome 	vdev_indirect_mapping_t *vim = vd->v_mapping;
716b1b93268SToomas Soome 	uint64_t entries = 0;
717b1b93268SToomas Soome 
718b1b93268SToomas Soome 	vdev_indirect_mapping_entry_phys_t *first_mapping =
719b1b93268SToomas Soome 	    vdev_indirect_mapping_entry_for_offset(vim, offset);
720b1b93268SToomas Soome 	ASSERT3P(first_mapping, !=, NULL);
721b1b93268SToomas Soome 
722b1b93268SToomas Soome 	vdev_indirect_mapping_entry_phys_t *m = first_mapping;
723b1b93268SToomas Soome 	while (asize > 0) {
724b1b93268SToomas Soome 		uint64_t size = DVA_GET_ASIZE(&m->vimep_dst);
725b1b93268SToomas Soome 		uint64_t inner_offset = offset - DVA_MAPPING_GET_SRC_OFFSET(m);
726b1b93268SToomas Soome 		uint64_t inner_size = MIN(asize, size - inner_offset);
727b1b93268SToomas Soome 
728b1b93268SToomas Soome 		offset += inner_size;
729b1b93268SToomas Soome 		asize -= inner_size;
730b1b93268SToomas Soome 		entries++;
731b1b93268SToomas Soome 		m++;
732b1b93268SToomas Soome 	}
733b1b93268SToomas Soome 
734b1b93268SToomas Soome 	size_t copy_length = entries * sizeof (*first_mapping);
735b1b93268SToomas Soome 	duplicate_mappings = malloc(copy_length);
736b1b93268SToomas Soome 	if (duplicate_mappings != NULL)
737b1b93268SToomas Soome 		bcopy(first_mapping, duplicate_mappings, copy_length);
738b1b93268SToomas Soome 	else
739b1b93268SToomas Soome 		entries = 0;
740b1b93268SToomas Soome 
741b1b93268SToomas Soome 	*copied_entries = entries;
742b1b93268SToomas Soome 
743b1b93268SToomas Soome 	return (duplicate_mappings);
744b1b93268SToomas Soome }
745b1b93268SToomas Soome 
746b1b93268SToomas Soome static vdev_t *
747b1b93268SToomas Soome vdev_lookup_top(spa_t *spa, uint64_t vdev)
748b1b93268SToomas Soome {
749b1b93268SToomas Soome 	vdev_t *rvd;
7503c2db0efSToomas Soome 	vdev_list_t *vlist;
751b1b93268SToomas Soome 
7523c2db0efSToomas Soome 	vlist = &spa->spa_root_vdev->v_children;
7533c2db0efSToomas Soome 	STAILQ_FOREACH(rvd, vlist, v_childlink)
754b1b93268SToomas Soome 		if (rvd->v_id == vdev)
755b1b93268SToomas Soome 			break;
756b1b93268SToomas Soome 
757b1b93268SToomas Soome 	return (rvd);
758b1b93268SToomas Soome }
759b1b93268SToomas Soome 
760b1b93268SToomas Soome /*
761b1b93268SToomas Soome  * This is a callback for vdev_indirect_remap() which allocates an
762b1b93268SToomas Soome  * indirect_split_t for each split segment and adds it to iv_splits.
763b1b93268SToomas Soome  */
764b1b93268SToomas Soome static void
765b1b93268SToomas Soome vdev_indirect_gather_splits(uint64_t split_offset, vdev_t *vd, uint64_t offset,
766b1b93268SToomas Soome     uint64_t size, void *arg)
767b1b93268SToomas Soome {
768b1b93268SToomas Soome 	int n = 1;
769b1b93268SToomas Soome 	zio_t *zio = arg;
770b1b93268SToomas Soome 	indirect_vsd_t *iv = zio->io_vsd;
771b1b93268SToomas Soome 
772b1b93268SToomas Soome 	if (vd->v_read == vdev_indirect_read)
773b1b93268SToomas Soome 		return;
774b1b93268SToomas Soome 
775b1b93268SToomas Soome 	if (vd->v_read == vdev_mirror_read)
776b1b93268SToomas Soome 		n = vd->v_nchildren;
777b1b93268SToomas Soome 
778b1b93268SToomas Soome 	indirect_split_t *is =
779b1b93268SToomas Soome 	    malloc(offsetof(indirect_split_t, is_child[n]));
780b1b93268SToomas Soome 	if (is == NULL) {
781b1b93268SToomas Soome 		zio->io_error = ENOMEM;
782b1b93268SToomas Soome 		return;
783b1b93268SToomas Soome 	}
784b1b93268SToomas Soome 	bzero(is, offsetof(indirect_split_t, is_child[n]));
785b1b93268SToomas Soome 
786b1b93268SToomas Soome 	is->is_children = n;
787b1b93268SToomas Soome 	is->is_size = size;
788b1b93268SToomas Soome 	is->is_split_offset = split_offset;
789b1b93268SToomas Soome 	is->is_target_offset = offset;
790b1b93268SToomas Soome 	is->is_vdev = vd;
791b1b93268SToomas Soome 
792b1b93268SToomas Soome 	/*
793b1b93268SToomas Soome 	 * Note that we only consider multiple copies of the data for
794b1b93268SToomas Soome 	 * *mirror* vdevs.  We don't for "replacing" or "spare" vdevs, even
795b1b93268SToomas Soome 	 * though they use the same ops as mirror, because there's only one
796b1b93268SToomas Soome 	 * "good" copy under the replacing/spare.
797b1b93268SToomas Soome 	 */
798b1b93268SToomas Soome 	if (vd->v_read == vdev_mirror_read) {
799b1b93268SToomas Soome 		int i = 0;
800b1b93268SToomas Soome 		vdev_t *kid;
801b1b93268SToomas Soome 
802b1b93268SToomas Soome 		STAILQ_FOREACH(kid, &vd->v_children, v_childlink) {
803b1b93268SToomas Soome 			is->is_child[i++].ic_vdev = kid;
804b1b93268SToomas Soome 		}
805b1b93268SToomas Soome 	} else {
806b1b93268SToomas Soome 		is->is_child[0].ic_vdev = vd;
807b1b93268SToomas Soome 	}
808b1b93268SToomas Soome 
809b1b93268SToomas Soome 	list_insert_tail(&iv->iv_splits, is);
810b1b93268SToomas Soome }
811b1b93268SToomas Soome 
812b1b93268SToomas Soome static void
813b1b93268SToomas Soome vdev_indirect_remap(vdev_t *vd, uint64_t offset, uint64_t asize, void *arg)
814b1b93268SToomas Soome {
815b1b93268SToomas Soome 	list_t stack;
8163c2db0efSToomas Soome 	spa_t *spa = vd->v_spa;
8176e5555adSToomas Soome 	zio_t *zio = arg;
8184efce32aSToomas Soome 	remap_segment_t *rs;
819b1b93268SToomas Soome 
820b1b93268SToomas Soome 	list_create(&stack, sizeof (remap_segment_t),
821b1b93268SToomas Soome 	    offsetof(remap_segment_t, rs_node));
822b1b93268SToomas Soome 
8234efce32aSToomas Soome 	rs = rs_alloc(vd, offset, asize, 0);
8244efce32aSToomas Soome 	if (rs == NULL) {
8254efce32aSToomas Soome 		printf("vdev_indirect_remap: out of memory.\n");
8264efce32aSToomas Soome 		zio->io_error = ENOMEM;
8274efce32aSToomas Soome 	}
8284efce32aSToomas Soome 	for (; rs != NULL; rs = list_remove_head(&stack)) {
829b1b93268SToomas Soome 		vdev_t *v = rs->rs_vd;
830b1b93268SToomas Soome 		uint64_t num_entries = 0;
831b1b93268SToomas Soome 		/* vdev_indirect_mapping_t *vim = v->v_mapping; */
832b1b93268SToomas Soome 		vdev_indirect_mapping_entry_phys_t *mapping =
833b1b93268SToomas Soome 		    vdev_indirect_mapping_duplicate_adjacent_entries(v,
834b1b93268SToomas Soome 		    rs->rs_offset, rs->rs_asize, &num_entries);
835b1b93268SToomas Soome 
8364efce32aSToomas Soome 		if (num_entries == 0)
8374efce32aSToomas Soome 			zio->io_error = ENOMEM;
8384efce32aSToomas Soome 
839b1b93268SToomas Soome 		for (uint64_t i = 0; i < num_entries; i++) {
840b1b93268SToomas Soome 			vdev_indirect_mapping_entry_phys_t *m = &mapping[i];
841b1b93268SToomas Soome 			uint64_t size = DVA_GET_ASIZE(&m->vimep_dst);
842b1b93268SToomas Soome 			uint64_t dst_offset = DVA_GET_OFFSET(&m->vimep_dst);
843b1b93268SToomas Soome 			uint64_t dst_vdev = DVA_GET_VDEV(&m->vimep_dst);
844b1b93268SToomas Soome 			uint64_t inner_offset = rs->rs_offset -
845b1b93268SToomas Soome 			    DVA_MAPPING_GET_SRC_OFFSET(m);
846b1b93268SToomas Soome 			uint64_t inner_size =
847b1b93268SToomas Soome 			    MIN(rs->rs_asize, size - inner_offset);
848b1b93268SToomas Soome 			vdev_t *dst_v = vdev_lookup_top(spa, dst_vdev);
849b1b93268SToomas Soome 
850b1b93268SToomas Soome 			if (dst_v->v_read == vdev_indirect_read) {
8514efce32aSToomas Soome 				remap_segment_t *o;
8524efce32aSToomas Soome 
8534efce32aSToomas Soome 				o = rs_alloc(dst_v, dst_offset + inner_offset,
8544efce32aSToomas Soome 				    inner_size, rs->rs_split_offset);
8554efce32aSToomas Soome 				if (o == NULL) {
8564efce32aSToomas Soome 					printf("vdev_indirect_remap: "
8574efce32aSToomas Soome 					    "out of memory.\n");
8584efce32aSToomas Soome 					zio->io_error = ENOMEM;
8594efce32aSToomas Soome 					break;
8604efce32aSToomas Soome 				}
8614efce32aSToomas Soome 
8624efce32aSToomas Soome 				list_insert_head(&stack, o);
863b1b93268SToomas Soome 			}
864b1b93268SToomas Soome 			vdev_indirect_gather_splits(rs->rs_split_offset, dst_v,
865b1b93268SToomas Soome 			    dst_offset + inner_offset,
866b1b93268SToomas Soome 			    inner_size, arg);
867b1b93268SToomas Soome 
8686e5555adSToomas Soome 			/*
8696e5555adSToomas Soome 			 * vdev_indirect_gather_splits can have memory
8706e5555adSToomas Soome 			 * allocation error, we can not recover from it.
8716e5555adSToomas Soome 			 */
8726e5555adSToomas Soome 			if (zio->io_error != 0)
8736e5555adSToomas Soome 				break;
874b1b93268SToomas Soome 			rs->rs_offset += inner_size;
875b1b93268SToomas Soome 			rs->rs_asize -= inner_size;
876b1b93268SToomas Soome 			rs->rs_split_offset += inner_size;
877b1b93268SToomas Soome 		}
878b1b93268SToomas Soome 
879b1b93268SToomas Soome 		free(mapping);
880b1b93268SToomas Soome 		free(rs);
8816e5555adSToomas Soome 		if (zio->io_error != 0)
8826e5555adSToomas Soome 			break;
883b1b93268SToomas Soome 	}
884b1b93268SToomas Soome 
885b1b93268SToomas Soome 	list_destroy(&stack);
886b1b93268SToomas Soome }
887b1b93268SToomas Soome 
888b1b93268SToomas Soome static void
889b1b93268SToomas Soome vdev_indirect_map_free(zio_t *zio)
890b1b93268SToomas Soome {
891b1b93268SToomas Soome 	indirect_vsd_t *iv = zio->io_vsd;
892b1b93268SToomas Soome 	indirect_split_t *is;
893b1b93268SToomas Soome 
894b1b93268SToomas Soome 	while ((is = list_head(&iv->iv_splits)) != NULL) {
895b1b93268SToomas Soome 		for (int c = 0; c < is->is_children; c++) {
896b1b93268SToomas Soome 			indirect_child_t *ic = &is->is_child[c];
897b1b93268SToomas Soome 			free(ic->ic_data);
898b1b93268SToomas Soome 		}
899b1b93268SToomas Soome 		list_remove(&iv->iv_splits, is);
900b1b93268SToomas Soome 		free(is);
901b1b93268SToomas Soome 	}
902b1b93268SToomas Soome 	free(iv);
903b1b93268SToomas Soome }
904b1b93268SToomas Soome 
905b1b93268SToomas Soome static int
906b1b93268SToomas Soome vdev_indirect_read(vdev_t *vdev, const blkptr_t *bp, void *buf,
907b1b93268SToomas Soome     off_t offset, size_t bytes)
908b1b93268SToomas Soome {
9093c2db0efSToomas Soome 	zio_t zio;
9103c2db0efSToomas Soome 	spa_t *spa = vdev->v_spa;
9113c2db0efSToomas Soome 	indirect_vsd_t *iv;
912b1b93268SToomas Soome 	indirect_split_t *first;
913b1b93268SToomas Soome 	int rc = EIO;
914b1b93268SToomas Soome 
9153c2db0efSToomas Soome 	iv = calloc(1, sizeof(*iv));
916b1b93268SToomas Soome 	if (iv == NULL)
917b1b93268SToomas Soome 		return (ENOMEM);
918b1b93268SToomas Soome 
919b1b93268SToomas Soome 	list_create(&iv->iv_splits,
920b1b93268SToomas Soome 	    sizeof (indirect_split_t), offsetof(indirect_split_t, is_node));
921b1b93268SToomas Soome 
9223c2db0efSToomas Soome 	bzero(&zio, sizeof(zio));
923b1b93268SToomas Soome 	zio.io_spa = spa;
924b1b93268SToomas Soome 	zio.io_bp = (blkptr_t *)bp;
925b1b93268SToomas Soome 	zio.io_data = buf;
926b1b93268SToomas Soome 	zio.io_size = bytes;
927b1b93268SToomas Soome 	zio.io_offset = offset;
928b1b93268SToomas Soome 	zio.io_vd = vdev;
929b1b93268SToomas Soome 	zio.io_vsd = iv;
930b1b93268SToomas Soome 
931b1b93268SToomas Soome 	if (vdev->v_mapping == NULL) {
932b1b93268SToomas Soome 		vdev_indirect_config_t *vic;
933b1b93268SToomas Soome 
934b1b93268SToomas Soome 		vic = &vdev->vdev_indirect_config;
935b1b93268SToomas Soome 		vdev->v_mapping = vdev_indirect_mapping_open(spa,
936b1b93268SToomas Soome 		    &spa->spa_mos, vic->vic_mapping_object);
937b1b93268SToomas Soome 	}
938b1b93268SToomas Soome 
939b1b93268SToomas Soome 	vdev_indirect_remap(vdev, offset, bytes, &zio);
9406e5555adSToomas Soome 	if (zio.io_error != 0)
9416e5555adSToomas Soome 		return (zio.io_error);
942b1b93268SToomas Soome 
943b1b93268SToomas Soome 	first = list_head(&iv->iv_splits);
944b1b93268SToomas Soome 	if (first->is_size == zio.io_size) {
945b1b93268SToomas Soome 		/*
946b1b93268SToomas Soome 		 * This is not a split block; we are pointing to the entire
947b1b93268SToomas Soome 		 * data, which will checksum the same as the original data.
948b1b93268SToomas Soome 		 * Pass the BP down so that the child i/o can verify the
949b1b93268SToomas Soome 		 * checksum, and try a different location if available
950b1b93268SToomas Soome 		 * (e.g. on a mirror).
951b1b93268SToomas Soome 		 *
952b1b93268SToomas Soome 		 * While this special case could be handled the same as the
953b1b93268SToomas Soome 		 * general (split block) case, doing it this way ensures
954b1b93268SToomas Soome 		 * that the vast majority of blocks on indirect vdevs
955b1b93268SToomas Soome 		 * (which are not split) are handled identically to blocks
956b1b93268SToomas Soome 		 * on non-indirect vdevs.  This allows us to be less strict
957b1b93268SToomas Soome 		 * about performance in the general (but rare) case.
958b1b93268SToomas Soome 		 */
959b1b93268SToomas Soome 		rc = first->is_vdev->v_read(first->is_vdev, zio.io_bp,
960b1b93268SToomas Soome 		    zio.io_data, first->is_target_offset, bytes);
961b1b93268SToomas Soome 	} else {
962b1b93268SToomas Soome 		iv->iv_split_block = B_TRUE;
963b1b93268SToomas Soome 		/*
964b1b93268SToomas Soome 		 * Read one copy of each split segment, from the
965b1b93268SToomas Soome 		 * top-level vdev.  Since we don't know the
966b1b93268SToomas Soome 		 * checksum of each split individually, the child
967b1b93268SToomas Soome 		 * zio can't ensure that we get the right data.
968b1b93268SToomas Soome 		 * E.g. if it's a mirror, it will just read from a
969b1b93268SToomas Soome 		 * random (healthy) leaf vdev.  We have to verify
970b1b93268SToomas Soome 		 * the checksum in vdev_indirect_io_done().
971b1b93268SToomas Soome 		 */
972b1b93268SToomas Soome 		for (indirect_split_t *is = list_head(&iv->iv_splits);
973b1b93268SToomas Soome 		    is != NULL; is = list_next(&iv->iv_splits, is)) {
974b1b93268SToomas Soome 			char *ptr = zio.io_data;
975b1b93268SToomas Soome 
976b1b93268SToomas Soome 			rc = is->is_vdev->v_read(is->is_vdev, zio.io_bp,
977b1b93268SToomas Soome 			    ptr + is->is_split_offset, is->is_target_offset,
978b1b93268SToomas Soome 			    is->is_size);
979b1b93268SToomas Soome 		}
980b1b93268SToomas Soome 		if (zio_checksum_verify(spa, zio.io_bp, zio.io_data))
981b1b93268SToomas Soome 			rc = ECKSUM;
982b1b93268SToomas Soome 		else
983b1b93268SToomas Soome 			rc = 0;
984b1b93268SToomas Soome 	}
985b1b93268SToomas Soome 
986b1b93268SToomas Soome 	vdev_indirect_map_free(&zio);
987b1b93268SToomas Soome 	if (rc == 0)
988b1b93268SToomas Soome 		rc = zio.io_error;
989b1b93268SToomas Soome 
990b1b93268SToomas Soome 	return (rc);
991b1b93268SToomas Soome }
992b1b93268SToomas Soome 
993b8902de1SWarner Losh static int
994b8902de1SWarner Losh vdev_disk_read(vdev_t *vdev, const blkptr_t *bp, void *buf,
995b8902de1SWarner Losh     off_t offset, size_t bytes)
996b8902de1SWarner Losh {
997b8902de1SWarner Losh 
998b8902de1SWarner Losh 	return (vdev_read_phys(vdev, bp, buf,
999b8902de1SWarner Losh 	    offset + VDEV_LABEL_START_SIZE, bytes));
1000b8902de1SWarner Losh }
1001b8902de1SWarner Losh 
1002b8902de1SWarner Losh 
1003b8902de1SWarner Losh static int
1004b8902de1SWarner Losh vdev_mirror_read(vdev_t *vdev, const blkptr_t *bp, void *buf,
1005b8902de1SWarner Losh     off_t offset, size_t bytes)
1006b8902de1SWarner Losh {
1007b8902de1SWarner Losh 	vdev_t *kid;
1008b8902de1SWarner Losh 	int rc;
1009b8902de1SWarner Losh 
1010b8902de1SWarner Losh 	rc = EIO;
1011b8902de1SWarner Losh 	STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
1012b8902de1SWarner Losh 		if (kid->v_state != VDEV_STATE_HEALTHY)
1013b8902de1SWarner Losh 			continue;
1014b8902de1SWarner Losh 		rc = kid->v_read(kid, bp, buf, offset, bytes);
1015b8902de1SWarner Losh 		if (!rc)
1016b8902de1SWarner Losh 			return (0);
1017b8902de1SWarner Losh 	}
1018b8902de1SWarner Losh 
1019b8902de1SWarner Losh 	return (rc);
1020b8902de1SWarner Losh }
1021b8902de1SWarner Losh 
1022b8902de1SWarner Losh static int
1023b8902de1SWarner Losh vdev_replacing_read(vdev_t *vdev, const blkptr_t *bp, void *buf,
1024b8902de1SWarner Losh     off_t offset, size_t bytes)
1025b8902de1SWarner Losh {
1026b8902de1SWarner Losh 	vdev_t *kid;
1027b8902de1SWarner Losh 
1028b8902de1SWarner Losh 	/*
1029b8902de1SWarner Losh 	 * Here we should have two kids:
1030b8902de1SWarner Losh 	 * First one which is the one we are replacing and we can trust
1031b8902de1SWarner Losh 	 * only this one to have valid data, but it might not be present.
1032b8902de1SWarner Losh 	 * Second one is that one we are replacing with. It is most likely
1033b8902de1SWarner Losh 	 * healthy, but we can't trust it has needed data, so we won't use it.
1034b8902de1SWarner Losh 	 */
1035b8902de1SWarner Losh 	kid = STAILQ_FIRST(&vdev->v_children);
1036b8902de1SWarner Losh 	if (kid == NULL)
1037b8902de1SWarner Losh 		return (EIO);
1038b8902de1SWarner Losh 	if (kid->v_state != VDEV_STATE_HEALTHY)
1039b8902de1SWarner Losh 		return (EIO);
1040b8902de1SWarner Losh 	return (kid->v_read(kid, bp, buf, offset, bytes));
1041b8902de1SWarner Losh }
1042b8902de1SWarner Losh 
1043b8902de1SWarner Losh static vdev_t *
1044b8902de1SWarner Losh vdev_find(uint64_t guid)
1045b8902de1SWarner Losh {
1046b8902de1SWarner Losh 	vdev_t *vdev;
1047b8902de1SWarner Losh 
1048b8902de1SWarner Losh 	STAILQ_FOREACH(vdev, &zfs_vdevs, v_alllink)
1049b8902de1SWarner Losh 		if (vdev->v_guid == guid)
1050b8902de1SWarner Losh 			return (vdev);
1051b8902de1SWarner Losh 
1052b8902de1SWarner Losh 	return (0);
1053b8902de1SWarner Losh }
1054b8902de1SWarner Losh 
1055b8902de1SWarner Losh static vdev_t *
1056b8902de1SWarner Losh vdev_create(uint64_t guid, vdev_read_t *_read)
1057b8902de1SWarner Losh {
1058b8902de1SWarner Losh 	vdev_t *vdev;
1059b1b93268SToomas Soome 	vdev_indirect_config_t *vic;
1060b8902de1SWarner Losh 
10613c2db0efSToomas Soome 	vdev = calloc(1, sizeof(vdev_t));
10623c2db0efSToomas Soome 	if (vdev != NULL) {
1063b8902de1SWarner Losh 		STAILQ_INIT(&vdev->v_children);
1064b8902de1SWarner Losh 		vdev->v_guid = guid;
1065b8902de1SWarner Losh 		vdev->v_read = _read;
1066b1b93268SToomas Soome 
10673c2db0efSToomas Soome 		/*
1068653c3383SToomas Soome 		 * root vdev has no read function, we use this fact to
1069653c3383SToomas Soome 		 * skip setting up data we do not need for root vdev.
10703c2db0efSToomas Soome 		 * We only point root vdev from spa.
10713c2db0efSToomas Soome 		 */
10723c2db0efSToomas Soome 		if (_read != NULL) {
1073b1b93268SToomas Soome 			vic = &vdev->vdev_indirect_config;
1074b1b93268SToomas Soome 			vic->vic_prev_indirect_vdev = UINT64_MAX;
1075b8902de1SWarner Losh 			STAILQ_INSERT_TAIL(&zfs_vdevs, vdev, v_alllink);
10763c2db0efSToomas Soome 		}
10773c2db0efSToomas Soome 	}
1078b8902de1SWarner Losh 
1079b8902de1SWarner Losh 	return (vdev);
1080b8902de1SWarner Losh }
1081b8902de1SWarner Losh 
10823c2db0efSToomas Soome static void
10833c2db0efSToomas Soome vdev_set_initial_state(vdev_t *vdev, const unsigned char *nvlist)
1084b8902de1SWarner Losh {
1085b8902de1SWarner Losh 	uint64_t is_offline, is_faulted, is_degraded, is_removed, isnt_present;
10860c0a882cSToomas Soome 	uint64_t is_log;
1087b8902de1SWarner Losh 
10883c2db0efSToomas Soome 	is_offline = is_removed = is_faulted = is_degraded = isnt_present = 0;
10893c2db0efSToomas Soome 	is_log = 0;
10903c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_OFFLINE, DATA_TYPE_UINT64, NULL,
10913c2db0efSToomas Soome 	    &is_offline);
10923c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_REMOVED, DATA_TYPE_UINT64, NULL,
10933c2db0efSToomas Soome 	    &is_removed);
10943c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_FAULTED, DATA_TYPE_UINT64, NULL,
10953c2db0efSToomas Soome 	    &is_faulted);
10963c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_DEGRADED, DATA_TYPE_UINT64,
10973c2db0efSToomas Soome 	    NULL, &is_degraded);
10983c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_NOT_PRESENT, DATA_TYPE_UINT64,
10993c2db0efSToomas Soome 	    NULL, &isnt_present);
11003c2db0efSToomas Soome 	(void) nvlist_find(nvlist, ZPOOL_CONFIG_IS_LOG, DATA_TYPE_UINT64, NULL,
11013c2db0efSToomas Soome 	    &is_log);
11023c2db0efSToomas Soome 
11033c2db0efSToomas Soome 	if (is_offline != 0)
11043c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_OFFLINE;
11053c2db0efSToomas Soome 	else if (is_removed != 0)
11063c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_REMOVED;
11073c2db0efSToomas Soome 	else if (is_faulted != 0)
11083c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_FAULTED;
11093c2db0efSToomas Soome 	else if (is_degraded != 0)
11103c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_DEGRADED;
11113c2db0efSToomas Soome 	else if (isnt_present != 0)
11123c2db0efSToomas Soome 		vdev->v_state = VDEV_STATE_CANT_OPEN;
11133c2db0efSToomas Soome 
1114653c3383SToomas Soome 	vdev->v_islog = is_log != 0;
11153c2db0efSToomas Soome }
11163c2db0efSToomas Soome 
11173c2db0efSToomas Soome static int
11183c2db0efSToomas Soome vdev_init(uint64_t guid, const unsigned char *nvlist, vdev_t **vdevp)
11193c2db0efSToomas Soome {
11203c2db0efSToomas Soome 	uint64_t id, ashift, asize, nparity;
11213c2db0efSToomas Soome 	const char *path;
11223c2db0efSToomas Soome 	const char *type;
11233c2db0efSToomas Soome 	vdev_t *vdev;
11243c2db0efSToomas Soome 
11253c2db0efSToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_ID, DATA_TYPE_UINT64, NULL, &id) ||
11263c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_TYPE, DATA_TYPE_STRING,
1127b8902de1SWarner Losh 	    NULL, &type)) {
1128b8902de1SWarner Losh 		return (ENOENT);
1129b8902de1SWarner Losh 	}
1130b8902de1SWarner Losh 
11312e6bb655SToomas Soome 	if (strcmp(type, VDEV_TYPE_MIRROR) != 0 &&
11322e6bb655SToomas Soome 	    strcmp(type, VDEV_TYPE_DISK) != 0 &&
1133b8902de1SWarner Losh #ifdef ZFS_TEST
11342e6bb655SToomas Soome 	    strcmp(type, VDEV_TYPE_FILE) != 0 &&
1135b8902de1SWarner Losh #endif
11362e6bb655SToomas Soome 	    strcmp(type, VDEV_TYPE_RAIDZ) != 0 &&
11372e6bb655SToomas Soome 	    strcmp(type, VDEV_TYPE_INDIRECT) != 0 &&
11382e6bb655SToomas Soome 	    strcmp(type, VDEV_TYPE_REPLACING) != 0) {
11392e6bb655SToomas Soome 		printf("ZFS: can only boot from disk, mirror, raidz1, "
11402e6bb655SToomas Soome 		    "raidz2 and raidz3 vdevs\n");
1141b8902de1SWarner Losh 		return (EIO);
1142b8902de1SWarner Losh 	}
1143b8902de1SWarner Losh 
11442e6bb655SToomas Soome 	if (strcmp(type, VDEV_TYPE_MIRROR) == 0)
1145b8902de1SWarner Losh 		vdev = vdev_create(guid, vdev_mirror_read);
11462e6bb655SToomas Soome 	else if (strcmp(type, VDEV_TYPE_RAIDZ) == 0)
1147b8902de1SWarner Losh 		vdev = vdev_create(guid, vdev_raidz_read);
11482e6bb655SToomas Soome 	else if (strcmp(type, VDEV_TYPE_REPLACING) == 0)
1149b8902de1SWarner Losh 		vdev = vdev_create(guid, vdev_replacing_read);
11502e6bb655SToomas Soome 	else if (strcmp(type, VDEV_TYPE_INDIRECT) == 0) {
1151b1b93268SToomas Soome 		vdev_indirect_config_t *vic;
1152b1b93268SToomas Soome 
1153b1b93268SToomas Soome 		vdev = vdev_create(guid, vdev_indirect_read);
11543c2db0efSToomas Soome 		if (vdev != NULL) {
1155b1b93268SToomas Soome 			vdev->v_state = VDEV_STATE_HEALTHY;
1156b1b93268SToomas Soome 			vic = &vdev->vdev_indirect_config;
1157b1b93268SToomas Soome 
1158b1b93268SToomas Soome 			nvlist_find(nvlist,
11593c2db0efSToomas Soome 			    ZPOOL_CONFIG_INDIRECT_OBJECT,
11603c2db0efSToomas Soome 			    DATA_TYPE_UINT64,
1161b1b93268SToomas Soome 			    NULL, &vic->vic_mapping_object);
1162b1b93268SToomas Soome 			nvlist_find(nvlist,
11633c2db0efSToomas Soome 			    ZPOOL_CONFIG_INDIRECT_BIRTHS,
11643c2db0efSToomas Soome 			    DATA_TYPE_UINT64,
1165b1b93268SToomas Soome 			    NULL, &vic->vic_births_object);
1166b1b93268SToomas Soome 			nvlist_find(nvlist,
11673c2db0efSToomas Soome 			    ZPOOL_CONFIG_PREV_INDIRECT_VDEV,
11683c2db0efSToomas Soome 			    DATA_TYPE_UINT64,
1169b1b93268SToomas Soome 			    NULL, &vic->vic_prev_indirect_vdev);
1170b8902de1SWarner Losh 		}
11713c2db0efSToomas Soome 	} else {
11723c2db0efSToomas Soome 		vdev = vdev_create(guid, vdev_disk_read);
11733c2db0efSToomas Soome 	}
11743c2db0efSToomas Soome 
11753c2db0efSToomas Soome 	if (vdev == NULL)
11763c2db0efSToomas Soome 		return (ENOMEM);
11773c2db0efSToomas Soome 
11783c2db0efSToomas Soome 	vdev_set_initial_state(vdev, nvlist);
11793c2db0efSToomas Soome 	vdev->v_id = id;
11803c2db0efSToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_ASHIFT,
11813c2db0efSToomas Soome 	    DATA_TYPE_UINT64, NULL, &ashift) == 0)
11823c2db0efSToomas Soome 		vdev->v_ashift = ashift;
11833c2db0efSToomas Soome 
1184abca0bd5SToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_ASIZE,
1185abca0bd5SToomas Soome 	    DATA_TYPE_UINT64, NULL, &asize) == 0) {
1186abca0bd5SToomas Soome 		vdev->v_psize = asize +
1187abca0bd5SToomas Soome 		    VDEV_LABEL_START_SIZE + VDEV_LABEL_END_SIZE;
1188abca0bd5SToomas Soome 	}
11893c2db0efSToomas Soome 
1190b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_NPARITY,
11913c2db0efSToomas Soome 	    DATA_TYPE_UINT64, NULL, &nparity) == 0)
1192b8902de1SWarner Losh 		vdev->v_nparity = nparity;
11933c2db0efSToomas Soome 
1194b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_PATH,
1195b8902de1SWarner Losh 	    DATA_TYPE_STRING, NULL, &path) == 0) {
1196b8902de1SWarner Losh 		if (strncmp(path, "/dev/", 5) == 0)
1197b8902de1SWarner Losh 			path += 5;
1198b8902de1SWarner Losh 		vdev->v_name = strdup(path);
1199b8902de1SWarner Losh 	} else {
1200b1b93268SToomas Soome 		char *name;
1201b1b93268SToomas Soome 
12023c2db0efSToomas Soome 		name = NULL;
12032e6bb655SToomas Soome 		if (strcmp(type, "raidz") == 0) {
1204b1b93268SToomas Soome 			if (vdev->v_nparity < 1 ||
1205b1b93268SToomas Soome 			    vdev->v_nparity > 3) {
1206653c3383SToomas Soome 				printf("ZFS: invalid raidz parity: %d\n",
1207653c3383SToomas Soome 				    vdev->v_nparity);
1208b8902de1SWarner Losh 				return (EIO);
1209b8902de1SWarner Losh 			}
12103c2db0efSToomas Soome 			(void) asprintf(&name, "%s%d-%" PRIu64, type,
1211b1b93268SToomas Soome 			    vdev->v_nparity, id);
1212b8902de1SWarner Losh 		} else {
12133c2db0efSToomas Soome 			(void) asprintf(&name, "%s-%" PRIu64, type, id);
1214b8902de1SWarner Losh 		}
1215b1b93268SToomas Soome 		vdev->v_name = name;
1216b8902de1SWarner Losh 	}
12173c2db0efSToomas Soome 	*vdevp = vdev;
12183c2db0efSToomas Soome 	return (0);
1219b8902de1SWarner Losh }
1220b8902de1SWarner Losh 
1221b8902de1SWarner Losh /*
12223c2db0efSToomas Soome  * Find slot for vdev. We return either NULL to signal to use
12233c2db0efSToomas Soome  * STAILQ_INSERT_HEAD, or we return link element to be used with
12243c2db0efSToomas Soome  * STAILQ_INSERT_AFTER.
1225b8902de1SWarner Losh  */
12263c2db0efSToomas Soome static vdev_t *
12273c2db0efSToomas Soome vdev_find_previous(vdev_t *top_vdev, vdev_t *vdev)
12283c2db0efSToomas Soome {
12293c2db0efSToomas Soome 	vdev_t *v, *previous;
12303c2db0efSToomas Soome 
12313c2db0efSToomas Soome 	if (STAILQ_EMPTY(&top_vdev->v_children))
12323c2db0efSToomas Soome 		return (NULL);
12333c2db0efSToomas Soome 
12343c2db0efSToomas Soome 	previous = NULL;
12353c2db0efSToomas Soome 	STAILQ_FOREACH(v, &top_vdev->v_children, v_childlink) {
12363c2db0efSToomas Soome 		if (v->v_id > vdev->v_id)
12373c2db0efSToomas Soome 			return (previous);
12383c2db0efSToomas Soome 
12393c2db0efSToomas Soome 		if (v->v_id == vdev->v_id)
12403c2db0efSToomas Soome 			return (v);
12413c2db0efSToomas Soome 
12423c2db0efSToomas Soome 		if (v->v_id < vdev->v_id)
12433c2db0efSToomas Soome 			previous = v;
12443c2db0efSToomas Soome 	}
12453c2db0efSToomas Soome 	return (previous);
1246b8902de1SWarner Losh }
1247b8902de1SWarner Losh 
12483c2db0efSToomas Soome static size_t
12493c2db0efSToomas Soome vdev_child_count(vdev_t *vdev)
12503c2db0efSToomas Soome {
12513c2db0efSToomas Soome 	vdev_t *v;
12523c2db0efSToomas Soome 	size_t count;
12533c2db0efSToomas Soome 
12543c2db0efSToomas Soome 	count = 0;
12553c2db0efSToomas Soome 	STAILQ_FOREACH(v, &vdev->v_children, v_childlink) {
12563c2db0efSToomas Soome 		count++;
12573c2db0efSToomas Soome 	}
12583c2db0efSToomas Soome 	return (count);
12593c2db0efSToomas Soome }
12603c2db0efSToomas Soome 
12613c2db0efSToomas Soome /*
12623c2db0efSToomas Soome  * Insert vdev into top_vdev children list. List is ordered by v_id.
12633c2db0efSToomas Soome  */
12643c2db0efSToomas Soome static void
12653c2db0efSToomas Soome vdev_insert(vdev_t *top_vdev, vdev_t *vdev)
12663c2db0efSToomas Soome {
12673c2db0efSToomas Soome 	vdev_t *previous;
12683c2db0efSToomas Soome 	size_t count;
12693c2db0efSToomas Soome 
12703c2db0efSToomas Soome 	/*
12713c2db0efSToomas Soome 	 * The top level vdev can appear in random order, depending how
12723c2db0efSToomas Soome 	 * the firmware is presenting the disk devices.
12733c2db0efSToomas Soome 	 * However, we will insert vdev to create list ordered by v_id,
12743c2db0efSToomas Soome 	 * so we can use either STAILQ_INSERT_HEAD or STAILQ_INSERT_AFTER
12753c2db0efSToomas Soome 	 * as STAILQ does not have insert before.
12763c2db0efSToomas Soome 	 */
12773c2db0efSToomas Soome 	previous = vdev_find_previous(top_vdev, vdev);
12783c2db0efSToomas Soome 
12793c2db0efSToomas Soome 	if (previous == NULL) {
12803c2db0efSToomas Soome 		STAILQ_INSERT_HEAD(&top_vdev->v_children, vdev, v_childlink);
1281653c3383SToomas Soome 	} else if (previous->v_id == vdev->v_id) {
1282653c3383SToomas Soome 		/*
1283653c3383SToomas Soome 		 * This vdev was configured from label config,
1284653c3383SToomas Soome 		 * do not insert duplicate.
1285653c3383SToomas Soome 		 */
12863c2db0efSToomas Soome 		return;
1287653c3383SToomas Soome 	} else {
1288653c3383SToomas Soome 		STAILQ_INSERT_AFTER(&top_vdev->v_children, previous, vdev,
1289653c3383SToomas Soome 		    v_childlink);
12903c2db0efSToomas Soome 	}
12913c2db0efSToomas Soome 
12923c2db0efSToomas Soome 	count = vdev_child_count(top_vdev);
12933c2db0efSToomas Soome 	if (top_vdev->v_nchildren < count)
12943c2db0efSToomas Soome 		top_vdev->v_nchildren = count;
12953c2db0efSToomas Soome }
12963c2db0efSToomas Soome 
12973c2db0efSToomas Soome static int
12983c2db0efSToomas Soome vdev_from_nvlist(spa_t *spa, uint64_t top_guid, const unsigned char *nvlist)
12993c2db0efSToomas Soome {
13003c2db0efSToomas Soome 	vdev_t *top_vdev, *vdev;
13013c2db0efSToomas Soome 	const unsigned char *kids;
13023c2db0efSToomas Soome 	int rc, nkids;
13033c2db0efSToomas Soome 
13043c2db0efSToomas Soome 	/* Get top vdev. */
13053c2db0efSToomas Soome 	top_vdev = vdev_find(top_guid);
13063c2db0efSToomas Soome 	if (top_vdev == NULL) {
13073c2db0efSToomas Soome 		rc = vdev_init(top_guid, nvlist, &top_vdev);
13083c2db0efSToomas Soome 		if (rc != 0)
13093c2db0efSToomas Soome 			return (rc);
13103c2db0efSToomas Soome 		top_vdev->v_spa = spa;
13113c2db0efSToomas Soome 		top_vdev->v_top = top_vdev;
13123c2db0efSToomas Soome 		vdev_insert(spa->spa_root_vdev, top_vdev);
13133c2db0efSToomas Soome 	}
13143c2db0efSToomas Soome 
13153c2db0efSToomas Soome 	/* Add children if there are any. */
1316b8902de1SWarner Losh 	rc = nvlist_find(nvlist, ZPOOL_CONFIG_CHILDREN, DATA_TYPE_NVLIST_ARRAY,
1317b8902de1SWarner Losh 	    &nkids, &kids);
1318b8902de1SWarner Losh 	if (rc == 0) {
13193c2db0efSToomas Soome 		for (int i = 0; i < nkids; i++) {
13203c2db0efSToomas Soome 			uint64_t guid;
13213c2db0efSToomas Soome 
13223c2db0efSToomas Soome 			rc = nvlist_find(kids, ZPOOL_CONFIG_GUID,
13233c2db0efSToomas Soome 			    DATA_TYPE_UINT64, NULL, &guid);
13243c2db0efSToomas Soome 			if (rc != 0)
1325b8902de1SWarner Losh 				return (rc);
13263c2db0efSToomas Soome 			rc = vdev_init(guid, kids, &vdev);
13273c2db0efSToomas Soome 			if (rc != 0)
13283c2db0efSToomas Soome 				return (rc);
13293c2db0efSToomas Soome 
13303c2db0efSToomas Soome 			vdev->v_spa = spa;
13313c2db0efSToomas Soome 			vdev->v_top = top_vdev;
13323c2db0efSToomas Soome 			vdev_insert(top_vdev, vdev);
13333c2db0efSToomas Soome 
1334b8902de1SWarner Losh 			kids = nvlist_next(kids);
1335b8902de1SWarner Losh 		}
1336b8902de1SWarner Losh 	} else {
1337653c3383SToomas Soome 		/*
1338653c3383SToomas Soome 		 * When there are no children, nvlist_find() does return
1339653c3383SToomas Soome 		 * error, reset it because leaf devices have no children.
1340653c3383SToomas Soome 		 */
13413c2db0efSToomas Soome 		rc = 0;
1342b8902de1SWarner Losh 	}
1343b8902de1SWarner Losh 
13443c2db0efSToomas Soome 	return (rc);
13453c2db0efSToomas Soome }
13463c2db0efSToomas Soome 
13473c2db0efSToomas Soome static int
13483c2db0efSToomas Soome vdev_init_from_label(spa_t *spa, const unsigned char *nvlist)
13493c2db0efSToomas Soome {
13503c2db0efSToomas Soome 	uint64_t pool_guid, top_guid;
13513c2db0efSToomas Soome 	const unsigned char *vdevs;
13523c2db0efSToomas Soome 
13533c2db0efSToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_POOL_GUID, DATA_TYPE_UINT64,
13543c2db0efSToomas Soome 	    NULL, &pool_guid) ||
13553c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_TOP_GUID, DATA_TYPE_UINT64,
13563c2db0efSToomas Soome 	    NULL, &top_guid) ||
13573c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_VDEV_TREE, DATA_TYPE_NVLIST,
13583c2db0efSToomas Soome 	    NULL, &vdevs)) {
13593c2db0efSToomas Soome 		printf("ZFS: can't find vdev details\n");
13603c2db0efSToomas Soome 		return (ENOENT);
13613c2db0efSToomas Soome 	}
13623c2db0efSToomas Soome 
13633c2db0efSToomas Soome 	return (vdev_from_nvlist(spa, top_guid, vdevs));
1364b8902de1SWarner Losh }
1365b8902de1SWarner Losh 
1366b8902de1SWarner Losh static void
1367b8902de1SWarner Losh vdev_set_state(vdev_t *vdev)
1368b8902de1SWarner Losh {
1369b8902de1SWarner Losh 	vdev_t *kid;
1370b8902de1SWarner Losh 	int good_kids;
1371b8902de1SWarner Losh 	int bad_kids;
1372b8902de1SWarner Losh 
13733c2db0efSToomas Soome 	STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
13743c2db0efSToomas Soome 		vdev_set_state(kid);
13753c2db0efSToomas Soome 	}
13763c2db0efSToomas Soome 
1377b8902de1SWarner Losh 	/*
1378b8902de1SWarner Losh 	 * A mirror or raidz is healthy if all its kids are healthy. A
1379b8902de1SWarner Losh 	 * mirror is degraded if any of its kids is healthy; a raidz
1380b8902de1SWarner Losh 	 * is degraded if at most nparity kids are offline.
1381b8902de1SWarner Losh 	 */
1382b8902de1SWarner Losh 	if (STAILQ_FIRST(&vdev->v_children)) {
1383b8902de1SWarner Losh 		good_kids = 0;
1384b8902de1SWarner Losh 		bad_kids = 0;
1385b8902de1SWarner Losh 		STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
1386b8902de1SWarner Losh 			if (kid->v_state == VDEV_STATE_HEALTHY)
1387b8902de1SWarner Losh 				good_kids++;
1388b8902de1SWarner Losh 			else
1389b8902de1SWarner Losh 				bad_kids++;
1390b8902de1SWarner Losh 		}
1391b8902de1SWarner Losh 		if (bad_kids == 0) {
1392b8902de1SWarner Losh 			vdev->v_state = VDEV_STATE_HEALTHY;
1393b8902de1SWarner Losh 		} else {
1394b8902de1SWarner Losh 			if (vdev->v_read == vdev_mirror_read) {
1395b8902de1SWarner Losh 				if (good_kids) {
1396b8902de1SWarner Losh 					vdev->v_state = VDEV_STATE_DEGRADED;
1397b8902de1SWarner Losh 				} else {
1398b8902de1SWarner Losh 					vdev->v_state = VDEV_STATE_OFFLINE;
1399b8902de1SWarner Losh 				}
1400b8902de1SWarner Losh 			} else if (vdev->v_read == vdev_raidz_read) {
1401b8902de1SWarner Losh 				if (bad_kids > vdev->v_nparity) {
1402b8902de1SWarner Losh 					vdev->v_state = VDEV_STATE_OFFLINE;
1403b8902de1SWarner Losh 				} else {
1404b8902de1SWarner Losh 					vdev->v_state = VDEV_STATE_DEGRADED;
1405b8902de1SWarner Losh 				}
1406b8902de1SWarner Losh 			}
1407b8902de1SWarner Losh 		}
1408b8902de1SWarner Losh 	}
1409b8902de1SWarner Losh }
1410b8902de1SWarner Losh 
14113c2db0efSToomas Soome static int
14123c2db0efSToomas Soome vdev_update_from_nvlist(uint64_t top_guid, const unsigned char *nvlist)
14133c2db0efSToomas Soome {
14143c2db0efSToomas Soome 	vdev_t *vdev;
14153c2db0efSToomas Soome 	const unsigned char *kids;
14163c2db0efSToomas Soome 	int rc, nkids;
14173c2db0efSToomas Soome 
14183c2db0efSToomas Soome 	/* Update top vdev. */
14193c2db0efSToomas Soome 	vdev = vdev_find(top_guid);
14203c2db0efSToomas Soome 	if (vdev != NULL)
14213c2db0efSToomas Soome 		vdev_set_initial_state(vdev, nvlist);
14223c2db0efSToomas Soome 
14233c2db0efSToomas Soome 	/* Update children if there are any. */
14243c2db0efSToomas Soome 	rc = nvlist_find(nvlist, ZPOOL_CONFIG_CHILDREN, DATA_TYPE_NVLIST_ARRAY,
14253c2db0efSToomas Soome 	    &nkids, &kids);
14263c2db0efSToomas Soome 	if (rc == 0) {
14273c2db0efSToomas Soome 		for (int i = 0; i < nkids; i++) {
14283c2db0efSToomas Soome 			uint64_t guid;
14293c2db0efSToomas Soome 
14303c2db0efSToomas Soome 			rc = nvlist_find(kids, ZPOOL_CONFIG_GUID,
14313c2db0efSToomas Soome 			    DATA_TYPE_UINT64, NULL, &guid);
14323c2db0efSToomas Soome 			if (rc != 0)
14333c2db0efSToomas Soome 				break;
14343c2db0efSToomas Soome 
14353c2db0efSToomas Soome 			vdev = vdev_find(guid);
14363c2db0efSToomas Soome 			if (vdev != NULL)
14373c2db0efSToomas Soome 				vdev_set_initial_state(vdev, kids);
14383c2db0efSToomas Soome 
14393c2db0efSToomas Soome 			kids = nvlist_next(kids);
14403c2db0efSToomas Soome 		}
14413c2db0efSToomas Soome 	} else {
14423c2db0efSToomas Soome 		rc = 0;
14433c2db0efSToomas Soome 	}
14443c2db0efSToomas Soome 
14453c2db0efSToomas Soome 	return (rc);
14463c2db0efSToomas Soome }
14473c2db0efSToomas Soome 
14483c2db0efSToomas Soome static int
14493c2db0efSToomas Soome vdev_init_from_nvlist(spa_t *spa, const unsigned char *nvlist)
14503c2db0efSToomas Soome {
14513c2db0efSToomas Soome 	uint64_t pool_guid, vdev_children;
14523c2db0efSToomas Soome 	const unsigned char *vdevs, *kids;
14533c2db0efSToomas Soome 	int rc, nkids;
14543c2db0efSToomas Soome 
14553c2db0efSToomas Soome 	if (nvlist_find(nvlist, ZPOOL_CONFIG_POOL_GUID, DATA_TYPE_UINT64,
14563c2db0efSToomas Soome 	    NULL, &pool_guid) ||
14573c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_VDEV_CHILDREN, DATA_TYPE_UINT64,
14583c2db0efSToomas Soome 	    NULL, &vdev_children) ||
14593c2db0efSToomas Soome 	    nvlist_find(nvlist, ZPOOL_CONFIG_VDEV_TREE, DATA_TYPE_NVLIST,
14603c2db0efSToomas Soome 	    NULL, &vdevs)) {
14613c2db0efSToomas Soome 		printf("ZFS: can't find vdev details\n");
14623c2db0efSToomas Soome 		return (ENOENT);
14633c2db0efSToomas Soome 	}
14643c2db0efSToomas Soome 
14653c2db0efSToomas Soome 	/* Wrong guid?! */
14663c2db0efSToomas Soome 	if (spa->spa_guid != pool_guid)
1467653c3383SToomas Soome 		return (EINVAL);
14683c2db0efSToomas Soome 
14693c2db0efSToomas Soome 	spa->spa_root_vdev->v_nchildren = vdev_children;
14703c2db0efSToomas Soome 
14713c2db0efSToomas Soome 	rc = nvlist_find(vdevs, ZPOOL_CONFIG_CHILDREN, DATA_TYPE_NVLIST_ARRAY,
14723c2db0efSToomas Soome 	    &nkids, &kids);
14733c2db0efSToomas Soome 
14743c2db0efSToomas Soome 	/*
14753c2db0efSToomas Soome 	 * MOS config has at least one child for root vdev.
14763c2db0efSToomas Soome 	 */
14773c2db0efSToomas Soome 	if (rc != 0)
1478653c3383SToomas Soome 		return (rc);
14793c2db0efSToomas Soome 
14803c2db0efSToomas Soome 	for (int i = 0; i < nkids; i++) {
14813c2db0efSToomas Soome 		uint64_t guid;
14823c2db0efSToomas Soome 		vdev_t *vdev;
14833c2db0efSToomas Soome 
14843c2db0efSToomas Soome 		rc = nvlist_find(kids, ZPOOL_CONFIG_GUID, DATA_TYPE_UINT64,
14853c2db0efSToomas Soome 		    NULL, &guid);
14863c2db0efSToomas Soome 		if (rc != 0)
14873c2db0efSToomas Soome 			break;
14883c2db0efSToomas Soome 		vdev = vdev_find(guid);
14893c2db0efSToomas Soome 		/*
14903c2db0efSToomas Soome 		 * Top level vdev is missing, create it.
14913c2db0efSToomas Soome 		 */
14923c2db0efSToomas Soome 		if (vdev == NULL)
14933c2db0efSToomas Soome 			rc = vdev_from_nvlist(spa, guid, kids);
14943c2db0efSToomas Soome 		else
14953c2db0efSToomas Soome 			rc = vdev_update_from_nvlist(guid, kids);
14963c2db0efSToomas Soome 		if (rc != 0)
14973c2db0efSToomas Soome 			break;
14983c2db0efSToomas Soome 		kids = nvlist_next(kids);
14993c2db0efSToomas Soome 	}
15003c2db0efSToomas Soome 
15013c2db0efSToomas Soome 	/*
15023c2db0efSToomas Soome 	 * Re-evaluate top-level vdev state.
15033c2db0efSToomas Soome 	 */
15043c2db0efSToomas Soome 	vdev_set_state(spa->spa_root_vdev);
15053c2db0efSToomas Soome 
15063c2db0efSToomas Soome 	return (rc);
15073c2db0efSToomas Soome }
15083c2db0efSToomas Soome 
1509b8902de1SWarner Losh static spa_t *
1510b8902de1SWarner Losh spa_find_by_guid(uint64_t guid)
1511b8902de1SWarner Losh {
1512b8902de1SWarner Losh 	spa_t *spa;
1513b8902de1SWarner Losh 
1514b8902de1SWarner Losh 	STAILQ_FOREACH(spa, &zfs_pools, spa_link)
1515b8902de1SWarner Losh 		if (spa->spa_guid == guid)
1516b8902de1SWarner Losh 			return (spa);
1517b8902de1SWarner Losh 
15182e6bb655SToomas Soome 	return (NULL);
1519b8902de1SWarner Losh }
1520b8902de1SWarner Losh 
1521b8902de1SWarner Losh static spa_t *
1522b8902de1SWarner Losh spa_find_by_name(const char *name)
1523b8902de1SWarner Losh {
1524b8902de1SWarner Losh 	spa_t *spa;
1525b8902de1SWarner Losh 
1526b8902de1SWarner Losh 	STAILQ_FOREACH(spa, &zfs_pools, spa_link)
15272e6bb655SToomas Soome 		if (strcmp(spa->spa_name, name) == 0)
1528b8902de1SWarner Losh 			return (spa);
1529b8902de1SWarner Losh 
15302e6bb655SToomas Soome 	return (NULL);
1531b8902de1SWarner Losh }
1532b8902de1SWarner Losh 
1533b8902de1SWarner Losh #ifdef BOOT2
1534b8902de1SWarner Losh static spa_t *
1535b8902de1SWarner Losh spa_get_primary(void)
1536b8902de1SWarner Losh {
1537b8902de1SWarner Losh 
1538b8902de1SWarner Losh 	return (STAILQ_FIRST(&zfs_pools));
1539b8902de1SWarner Losh }
1540b8902de1SWarner Losh 
1541b8902de1SWarner Losh static vdev_t *
1542b8902de1SWarner Losh spa_get_primary_vdev(const spa_t *spa)
1543b8902de1SWarner Losh {
1544b8902de1SWarner Losh 	vdev_t *vdev;
1545b8902de1SWarner Losh 	vdev_t *kid;
1546b8902de1SWarner Losh 
1547b8902de1SWarner Losh 	if (spa == NULL)
1548b8902de1SWarner Losh 		spa = spa_get_primary();
1549b8902de1SWarner Losh 	if (spa == NULL)
1550b8902de1SWarner Losh 		return (NULL);
15513c2db0efSToomas Soome 	vdev = spa->spa_root_vdev;
1552b8902de1SWarner Losh 	if (vdev == NULL)
1553b8902de1SWarner Losh 		return (NULL);
1554b8902de1SWarner Losh 	for (kid = STAILQ_FIRST(&vdev->v_children); kid != NULL;
1555b8902de1SWarner Losh 	    kid = STAILQ_FIRST(&vdev->v_children))
1556b8902de1SWarner Losh 		vdev = kid;
1557b8902de1SWarner Losh 	return (vdev);
1558b8902de1SWarner Losh }
1559b8902de1SWarner Losh #endif
1560b8902de1SWarner Losh 
1561b8902de1SWarner Losh static spa_t *
1562b8902de1SWarner Losh spa_create(uint64_t guid, const char *name)
1563b8902de1SWarner Losh {
1564b8902de1SWarner Losh 	spa_t *spa;
1565b8902de1SWarner Losh 
156693a2d4c9SToomas Soome 	if ((spa = calloc(1, sizeof(spa_t))) == NULL)
1567b8902de1SWarner Losh 		return (NULL);
1568b8902de1SWarner Losh 	if ((spa->spa_name = strdup(name)) == NULL) {
1569b8902de1SWarner Losh 		free(spa);
1570b8902de1SWarner Losh 		return (NULL);
1571b8902de1SWarner Losh 	}
1572b8902de1SWarner Losh 	spa->spa_guid = guid;
15733c2db0efSToomas Soome 	spa->spa_root_vdev = vdev_create(guid, NULL);
15743c2db0efSToomas Soome 	if (spa->spa_root_vdev == NULL) {
15753c2db0efSToomas Soome 		free(spa->spa_name);
15763c2db0efSToomas Soome 		free(spa);
15773c2db0efSToomas Soome 		return (NULL);
15783c2db0efSToomas Soome 	}
15793c2db0efSToomas Soome 	spa->spa_root_vdev->v_name = strdup("root");
1580b8902de1SWarner Losh 	STAILQ_INSERT_TAIL(&zfs_pools, spa, spa_link);
1581b8902de1SWarner Losh 
1582b8902de1SWarner Losh 	return (spa);
1583b8902de1SWarner Losh }
1584b8902de1SWarner Losh 
1585b8902de1SWarner Losh static const char *
1586b8902de1SWarner Losh state_name(vdev_state_t state)
1587b8902de1SWarner Losh {
1588b8902de1SWarner Losh 	static const char *names[] = {
1589b8902de1SWarner Losh 		"UNKNOWN",
1590b8902de1SWarner Losh 		"CLOSED",
1591b8902de1SWarner Losh 		"OFFLINE",
1592b8902de1SWarner Losh 		"REMOVED",
1593b8902de1SWarner Losh 		"CANT_OPEN",
1594b8902de1SWarner Losh 		"FAULTED",
1595b8902de1SWarner Losh 		"DEGRADED",
1596b8902de1SWarner Losh 		"ONLINE"
1597b8902de1SWarner Losh 	};
15982e6bb655SToomas Soome 	return (names[state]);
1599b8902de1SWarner Losh }
1600b8902de1SWarner Losh 
1601b8902de1SWarner Losh #ifdef BOOT2
1602b8902de1SWarner Losh 
1603b8902de1SWarner Losh #define pager_printf printf
1604b8902de1SWarner Losh 
1605b8902de1SWarner Losh #else
1606b8902de1SWarner Losh 
1607b8902de1SWarner Losh static int
1608b8902de1SWarner Losh pager_printf(const char *fmt, ...)
1609b8902de1SWarner Losh {
1610b8902de1SWarner Losh 	char line[80];
1611b8902de1SWarner Losh 	va_list args;
1612b8902de1SWarner Losh 
1613b8902de1SWarner Losh 	va_start(args, fmt);
16143c2db0efSToomas Soome 	vsnprintf(line, sizeof(line), fmt, args);
1615b8902de1SWarner Losh 	va_end(args);
1616b8902de1SWarner Losh 	return (pager_output(line));
1617b8902de1SWarner Losh }
1618b8902de1SWarner Losh 
1619b8902de1SWarner Losh #endif
1620b8902de1SWarner Losh 
1621b8902de1SWarner Losh #define	STATUS_FORMAT	"        %s %s\n"
1622b8902de1SWarner Losh 
1623b8902de1SWarner Losh static int
1624b8902de1SWarner Losh print_state(int indent, const char *name, vdev_state_t state)
1625b8902de1SWarner Losh {
1626b8902de1SWarner Losh 	int i;
16273c2db0efSToomas Soome 	char buf[512];
1628b8902de1SWarner Losh 
1629b8902de1SWarner Losh 	buf[0] = 0;
1630b8902de1SWarner Losh 	for (i = 0; i < indent; i++)
1631b8902de1SWarner Losh 		strcat(buf, "  ");
1632b8902de1SWarner Losh 	strcat(buf, name);
1633b8902de1SWarner Losh 	return (pager_printf(STATUS_FORMAT, buf, state_name(state)));
1634b8902de1SWarner Losh }
1635b8902de1SWarner Losh 
1636b8902de1SWarner Losh static int
1637b8902de1SWarner Losh vdev_status(vdev_t *vdev, int indent)
1638b8902de1SWarner Losh {
1639b8902de1SWarner Losh 	vdev_t *kid;
1640b8902de1SWarner Losh 	int ret;
16410c0a882cSToomas Soome 
16420c0a882cSToomas Soome 	if (vdev->v_islog) {
16430c0a882cSToomas Soome 		(void) pager_output("        logs\n");
16440c0a882cSToomas Soome 		indent++;
16450c0a882cSToomas Soome 	}
16460c0a882cSToomas Soome 
1647b8902de1SWarner Losh 	ret = print_state(indent, vdev->v_name, vdev->v_state);
1648b8902de1SWarner Losh 	if (ret != 0)
1649b8902de1SWarner Losh 		return (ret);
1650b8902de1SWarner Losh 
1651b8902de1SWarner Losh 	STAILQ_FOREACH(kid, &vdev->v_children, v_childlink) {
1652b8902de1SWarner Losh 		ret = vdev_status(kid, indent + 1);
1653b8902de1SWarner Losh 		if (ret != 0)
1654b8902de1SWarner Losh 			return (ret);
1655b8902de1SWarner Losh 	}
1656b8902de1SWarner Losh 	return (ret);
1657b8902de1SWarner Losh }
1658b8902de1SWarner Losh 
1659b8902de1SWarner Losh static int
1660b8902de1SWarner Losh spa_status(spa_t *spa)
1661b8902de1SWarner Losh {
1662b8902de1SWarner Losh 	static char bootfs[ZFS_MAXNAMELEN];
1663b8902de1SWarner Losh 	uint64_t rootid;
16643c2db0efSToomas Soome 	vdev_list_t *vlist;
1665b8902de1SWarner Losh 	vdev_t *vdev;
1666b8902de1SWarner Losh 	int good_kids, bad_kids, degraded_kids, ret;
1667b8902de1SWarner Losh 	vdev_state_t state;
1668b8902de1SWarner Losh 
1669b8902de1SWarner Losh 	ret = pager_printf("  pool: %s\n", spa->spa_name);
1670b8902de1SWarner Losh 	if (ret != 0)
1671b8902de1SWarner Losh 		return (ret);
1672b8902de1SWarner Losh 
1673b8902de1SWarner Losh 	if (zfs_get_root(spa, &rootid) == 0 &&
1674b8902de1SWarner Losh 	    zfs_rlookup(spa, rootid, bootfs) == 0) {
1675b8902de1SWarner Losh 		if (bootfs[0] == '\0')
1676b8902de1SWarner Losh 			ret = pager_printf("bootfs: %s\n", spa->spa_name);
1677b8902de1SWarner Losh 		else
1678b8902de1SWarner Losh 			ret = pager_printf("bootfs: %s/%s\n", spa->spa_name,
1679b8902de1SWarner Losh 			    bootfs);
1680b8902de1SWarner Losh 		if (ret != 0)
1681b8902de1SWarner Losh 			return (ret);
1682b8902de1SWarner Losh 	}
1683b8902de1SWarner Losh 	ret = pager_printf("config:\n\n");
1684b8902de1SWarner Losh 	if (ret != 0)
1685b8902de1SWarner Losh 		return (ret);
1686b8902de1SWarner Losh 	ret = pager_printf(STATUS_FORMAT, "NAME", "STATE");
1687b8902de1SWarner Losh 	if (ret != 0)
1688b8902de1SWarner Losh 		return (ret);
1689b8902de1SWarner Losh 
1690b8902de1SWarner Losh 	good_kids = 0;
1691b8902de1SWarner Losh 	degraded_kids = 0;
1692b8902de1SWarner Losh 	bad_kids = 0;
16933c2db0efSToomas Soome 	vlist = &spa->spa_root_vdev->v_children;
16943c2db0efSToomas Soome 	STAILQ_FOREACH(vdev, vlist, v_childlink) {
1695b8902de1SWarner Losh 		if (vdev->v_state == VDEV_STATE_HEALTHY)
1696b8902de1SWarner Losh 			good_kids++;
1697b8902de1SWarner Losh 		else if (vdev->v_state == VDEV_STATE_DEGRADED)
1698b8902de1SWarner Losh 			degraded_kids++;
1699b8902de1SWarner Losh 		else
1700b8902de1SWarner Losh 			bad_kids++;
1701b8902de1SWarner Losh 	}
1702b8902de1SWarner Losh 
1703b8902de1SWarner Losh 	state = VDEV_STATE_CLOSED;
1704b8902de1SWarner Losh 	if (good_kids > 0 && (degraded_kids + bad_kids) == 0)
1705b8902de1SWarner Losh 		state = VDEV_STATE_HEALTHY;
1706b8902de1SWarner Losh 	else if ((good_kids + degraded_kids) > 0)
1707b8902de1SWarner Losh 		state = VDEV_STATE_DEGRADED;
1708b8902de1SWarner Losh 
1709b8902de1SWarner Losh 	ret = print_state(0, spa->spa_name, state);
1710b8902de1SWarner Losh 	if (ret != 0)
1711b8902de1SWarner Losh 		return (ret);
17123c2db0efSToomas Soome 
17133c2db0efSToomas Soome 	STAILQ_FOREACH(vdev, vlist, v_childlink) {
1714b8902de1SWarner Losh 		ret = vdev_status(vdev, 1);
1715b8902de1SWarner Losh 		if (ret != 0)
1716b8902de1SWarner Losh 			return (ret);
1717b8902de1SWarner Losh 	}
1718b8902de1SWarner Losh 	return (ret);
1719b8902de1SWarner Losh }
1720b8902de1SWarner Losh 
1721b8902de1SWarner Losh static int
1722b8902de1SWarner Losh spa_all_status(void)
1723b8902de1SWarner Losh {
1724b8902de1SWarner Losh 	spa_t *spa;
1725b8902de1SWarner Losh 	int first = 1, ret = 0;
1726b8902de1SWarner Losh 
1727b8902de1SWarner Losh 	STAILQ_FOREACH(spa, &zfs_pools, spa_link) {
1728b8902de1SWarner Losh 		if (!first) {
1729b8902de1SWarner Losh 			ret = pager_printf("\n");
1730b8902de1SWarner Losh 			if (ret != 0)
1731b8902de1SWarner Losh 				return (ret);
1732b8902de1SWarner Losh 		}
1733b8902de1SWarner Losh 		first = 0;
1734b8902de1SWarner Losh 		ret = spa_status(spa);
1735b8902de1SWarner Losh 		if (ret != 0)
1736b8902de1SWarner Losh 			return (ret);
1737b8902de1SWarner Losh 	}
1738b8902de1SWarner Losh 	return (ret);
1739b8902de1SWarner Losh }
1740b8902de1SWarner Losh 
1741b8902de1SWarner Losh static uint64_t
1742b8902de1SWarner Losh vdev_label_offset(uint64_t psize, int l, uint64_t offset)
1743b8902de1SWarner Losh {
1744b8902de1SWarner Losh 	uint64_t label_offset;
1745b8902de1SWarner Losh 
1746b8902de1SWarner Losh 	if (l < VDEV_LABELS / 2)
1747b8902de1SWarner Losh 		label_offset = 0;
1748b8902de1SWarner Losh 	else
1749b8902de1SWarner Losh 		label_offset = psize - VDEV_LABELS * sizeof (vdev_label_t);
1750b8902de1SWarner Losh 
1751b8902de1SWarner Losh 	return (offset + l * sizeof (vdev_label_t) + label_offset);
1752b8902de1SWarner Losh }
1753b8902de1SWarner Losh 
1754b8902de1SWarner Losh static int
175579a4bf89SToomas Soome vdev_uberblock_compare(const uberblock_t *ub1, const uberblock_t *ub2)
175679a4bf89SToomas Soome {
175779a4bf89SToomas Soome 	unsigned int seq1 = 0;
175879a4bf89SToomas Soome 	unsigned int seq2 = 0;
175979a4bf89SToomas Soome 	int cmp = AVL_CMP(ub1->ub_txg, ub2->ub_txg);
176079a4bf89SToomas Soome 
176179a4bf89SToomas Soome 	if (cmp != 0)
176279a4bf89SToomas Soome 		return (cmp);
176379a4bf89SToomas Soome 
176479a4bf89SToomas Soome 	cmp = AVL_CMP(ub1->ub_timestamp, ub2->ub_timestamp);
176579a4bf89SToomas Soome 	if (cmp != 0)
176679a4bf89SToomas Soome 		return (cmp);
176779a4bf89SToomas Soome 
176879a4bf89SToomas Soome 	if (MMP_VALID(ub1) && MMP_SEQ_VALID(ub1))
176979a4bf89SToomas Soome 		seq1 = MMP_SEQ(ub1);
177079a4bf89SToomas Soome 
177179a4bf89SToomas Soome 	if (MMP_VALID(ub2) && MMP_SEQ_VALID(ub2))
177279a4bf89SToomas Soome 		seq2 = MMP_SEQ(ub2);
177379a4bf89SToomas Soome 
177479a4bf89SToomas Soome 	return (AVL_CMP(seq1, seq2));
177579a4bf89SToomas Soome }
177679a4bf89SToomas Soome 
177779a4bf89SToomas Soome static int
177879a4bf89SToomas Soome uberblock_verify(uberblock_t *ub)
177979a4bf89SToomas Soome {
178079a4bf89SToomas Soome 	if (ub->ub_magic == BSWAP_64((uint64_t)UBERBLOCK_MAGIC)) {
178179a4bf89SToomas Soome 		byteswap_uint64_array(ub, sizeof (uberblock_t));
178279a4bf89SToomas Soome 	}
178379a4bf89SToomas Soome 
178479a4bf89SToomas Soome 	if (ub->ub_magic != UBERBLOCK_MAGIC ||
178579a4bf89SToomas Soome 	    !SPA_VERSION_IS_SUPPORTED(ub->ub_version))
178679a4bf89SToomas Soome 		return (EINVAL);
178779a4bf89SToomas Soome 
178879a4bf89SToomas Soome 	return (0);
178979a4bf89SToomas Soome }
179079a4bf89SToomas Soome 
179179a4bf89SToomas Soome static int
179279a4bf89SToomas Soome vdev_label_read(vdev_t *vd, int l, void *buf, uint64_t offset,
179379a4bf89SToomas Soome     size_t size)
179479a4bf89SToomas Soome {
179579a4bf89SToomas Soome 	blkptr_t bp;
179679a4bf89SToomas Soome 	off_t off;
179779a4bf89SToomas Soome 
179879a4bf89SToomas Soome 	off = vdev_label_offset(vd->v_psize, l, offset);
179979a4bf89SToomas Soome 
180079a4bf89SToomas Soome 	BP_ZERO(&bp);
180179a4bf89SToomas Soome 	BP_SET_LSIZE(&bp, size);
180279a4bf89SToomas Soome 	BP_SET_PSIZE(&bp, size);
180379a4bf89SToomas Soome 	BP_SET_CHECKSUM(&bp, ZIO_CHECKSUM_LABEL);
180479a4bf89SToomas Soome 	BP_SET_COMPRESS(&bp, ZIO_COMPRESS_OFF);
180579a4bf89SToomas Soome 	DVA_SET_OFFSET(BP_IDENTITY(&bp), off);
180679a4bf89SToomas Soome 	ZIO_SET_CHECKSUM(&bp.blk_cksum, off, 0, 0, 0);
180779a4bf89SToomas Soome 
180879a4bf89SToomas Soome 	return (vdev_read_phys(vd, &bp, buf, off, size));
180979a4bf89SToomas Soome }
181079a4bf89SToomas Soome 
181179a4bf89SToomas Soome static unsigned char *
181279a4bf89SToomas Soome vdev_label_read_config(vdev_t *vd, uint64_t txg)
181379a4bf89SToomas Soome {
181479a4bf89SToomas Soome 	vdev_phys_t *label;
181579a4bf89SToomas Soome 	uint64_t best_txg = 0;
181679a4bf89SToomas Soome 	uint64_t label_txg = 0;
181779a4bf89SToomas Soome 	uint64_t asize;
181879a4bf89SToomas Soome 	unsigned char *nvl;
181979a4bf89SToomas Soome 	size_t nvl_size;
182079a4bf89SToomas Soome 	int error;
182179a4bf89SToomas Soome 
182279a4bf89SToomas Soome 	label = malloc(sizeof (vdev_phys_t));
182379a4bf89SToomas Soome 	if (label == NULL)
182479a4bf89SToomas Soome 		return (NULL);
182579a4bf89SToomas Soome 
182679a4bf89SToomas Soome 	nvl_size = VDEV_PHYS_SIZE - sizeof (zio_eck_t) - 4;
182779a4bf89SToomas Soome 	nvl = malloc(nvl_size);
18283cac32d1SToomas Soome 	if (nvl == NULL)
18293cac32d1SToomas Soome 		goto done;
183079a4bf89SToomas Soome 
183179a4bf89SToomas Soome 	for (int l = 0; l < VDEV_LABELS; l++) {
183279a4bf89SToomas Soome 		const unsigned char *nvlist;
183379a4bf89SToomas Soome 
183479a4bf89SToomas Soome 		if (vdev_label_read(vd, l, label,
183579a4bf89SToomas Soome 		    offsetof(vdev_label_t, vl_vdev_phys),
183679a4bf89SToomas Soome 		    sizeof (vdev_phys_t)))
183779a4bf89SToomas Soome 			continue;
183879a4bf89SToomas Soome 
183979a4bf89SToomas Soome 		if (label->vp_nvlist[0] != NV_ENCODE_XDR)
184079a4bf89SToomas Soome 			continue;
184179a4bf89SToomas Soome 
184279a4bf89SToomas Soome 		nvlist = (const unsigned char *) label->vp_nvlist + 4;
184379a4bf89SToomas Soome 		error = nvlist_find(nvlist, ZPOOL_CONFIG_POOL_TXG,
184479a4bf89SToomas Soome 		    DATA_TYPE_UINT64, NULL, &label_txg);
18459a3ea759SToomas Soome 		if (error != 0 || label_txg == 0) {
18469a3ea759SToomas Soome 			memcpy(nvl, nvlist, nvl_size);
18473cac32d1SToomas Soome 			goto done;
18489a3ea759SToomas Soome 		}
184979a4bf89SToomas Soome 
185079a4bf89SToomas Soome 		if (label_txg <= txg && label_txg > best_txg) {
185179a4bf89SToomas Soome 			best_txg = label_txg;
185279a4bf89SToomas Soome 			memcpy(nvl, nvlist, nvl_size);
185379a4bf89SToomas Soome 
185479a4bf89SToomas Soome 			/*
185579a4bf89SToomas Soome 			 * Use asize from pool config. We need this
185679a4bf89SToomas Soome 			 * because we can get bad value from BIOS.
185779a4bf89SToomas Soome 			 */
185879a4bf89SToomas Soome 			if (nvlist_find(nvlist, ZPOOL_CONFIG_ASIZE,
185979a4bf89SToomas Soome 			    DATA_TYPE_UINT64, NULL, &asize) == 0) {
186079a4bf89SToomas Soome 				vd->v_psize = asize +
186179a4bf89SToomas Soome 				    VDEV_LABEL_START_SIZE + VDEV_LABEL_END_SIZE;
186279a4bf89SToomas Soome 			}
186379a4bf89SToomas Soome 		}
186479a4bf89SToomas Soome 	}
186579a4bf89SToomas Soome 
186679a4bf89SToomas Soome 	if (best_txg == 0) {
186779a4bf89SToomas Soome 		free(nvl);
186879a4bf89SToomas Soome 		nvl = NULL;
186979a4bf89SToomas Soome 	}
18703cac32d1SToomas Soome done:
18713cac32d1SToomas Soome 	free(label);
187279a4bf89SToomas Soome 	return (nvl);
187379a4bf89SToomas Soome }
187479a4bf89SToomas Soome 
187579a4bf89SToomas Soome static void
187679a4bf89SToomas Soome vdev_uberblock_load(vdev_t *vd, uberblock_t *ub)
187779a4bf89SToomas Soome {
187879a4bf89SToomas Soome 	uberblock_t *buf;
187979a4bf89SToomas Soome 
188079a4bf89SToomas Soome 	buf = malloc(VDEV_UBERBLOCK_SIZE(vd));
188179a4bf89SToomas Soome 	if (buf == NULL)
188279a4bf89SToomas Soome 		return;
188379a4bf89SToomas Soome 
188479a4bf89SToomas Soome 	for (int l = 0; l < VDEV_LABELS; l++) {
188579a4bf89SToomas Soome 		for (int n = 0; n < VDEV_UBERBLOCK_COUNT(vd); n++) {
188679a4bf89SToomas Soome 			if (vdev_label_read(vd, l, buf,
188779a4bf89SToomas Soome 			    VDEV_UBERBLOCK_OFFSET(vd, n),
188879a4bf89SToomas Soome 			    VDEV_UBERBLOCK_SIZE(vd)))
188979a4bf89SToomas Soome 				continue;
189079a4bf89SToomas Soome 			if (uberblock_verify(buf) != 0)
189179a4bf89SToomas Soome 				continue;
189279a4bf89SToomas Soome 
189379a4bf89SToomas Soome 			if (vdev_uberblock_compare(buf, ub) > 0)
189479a4bf89SToomas Soome 				*ub = *buf;
189579a4bf89SToomas Soome 		}
189679a4bf89SToomas Soome 	}
189779a4bf89SToomas Soome 	free(buf);
189879a4bf89SToomas Soome }
189979a4bf89SToomas Soome 
190079a4bf89SToomas Soome static int
1901b8902de1SWarner Losh vdev_probe(vdev_phys_read_t *_read, void *read_priv, spa_t **spap)
1902b8902de1SWarner Losh {
1903c082b391SToomas Soome 	vdev_t vtmp;
1904b8902de1SWarner Losh 	spa_t *spa;
19053c2db0efSToomas Soome 	vdev_t *vdev;
190679a4bf89SToomas Soome 	unsigned char *nvlist;
1907b8902de1SWarner Losh 	uint64_t val;
19083c2db0efSToomas Soome 	uint64_t guid, vdev_children;
1909b8902de1SWarner Losh 	uint64_t pool_txg, pool_guid;
1910b8902de1SWarner Losh 	const char *pool_name;
1911b8902de1SWarner Losh 	const unsigned char *features;
19123c2db0efSToomas Soome 	int rc;
1913b8902de1SWarner Losh 
1914b8902de1SWarner Losh 	/*
1915b8902de1SWarner Losh 	 * Load the vdev label and figure out which
1916b8902de1SWarner Losh 	 * uberblock is most current.
1917b8902de1SWarner Losh 	 */
1918c082b391SToomas Soome 	memset(&vtmp, 0, sizeof(vtmp));
1919b8902de1SWarner Losh 	vtmp.v_phys_read = _read;
1920b8902de1SWarner Losh 	vtmp.v_read_priv = read_priv;
1921abca0bd5SToomas Soome 	vtmp.v_psize = P2ALIGN(ldi_get_size(read_priv),
1922b8902de1SWarner Losh 	    (uint64_t)sizeof (vdev_label_t));
1923b8902de1SWarner Losh 
192479a4bf89SToomas Soome 	/* Test for minimum device size. */
1925abca0bd5SToomas Soome 	if (vtmp.v_psize < SPA_MINDEVSIZE)
1926b8902de1SWarner Losh 		return (EIO);
1927b8902de1SWarner Losh 
192879a4bf89SToomas Soome 	nvlist = vdev_label_read_config(&vtmp, UINT64_MAX);
192979a4bf89SToomas Soome 	if (nvlist == NULL)
1930b8902de1SWarner Losh 		return (EIO);
1931b8902de1SWarner Losh 
1932b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_VERSION, DATA_TYPE_UINT64,
1933b8902de1SWarner Losh 	    NULL, &val) != 0) {
193479a4bf89SToomas Soome 		free(nvlist);
1935b8902de1SWarner Losh 		return (EIO);
1936b8902de1SWarner Losh 	}
1937b8902de1SWarner Losh 
1938b8902de1SWarner Losh 	if (!SPA_VERSION_IS_SUPPORTED(val)) {
1939b8902de1SWarner Losh 		printf("ZFS: unsupported ZFS version %u (should be %u)\n",
1940b8902de1SWarner Losh 		    (unsigned)val, (unsigned)SPA_VERSION);
194179a4bf89SToomas Soome 		free(nvlist);
1942b8902de1SWarner Losh 		return (EIO);
1943b8902de1SWarner Losh 	}
1944b8902de1SWarner Losh 
1945b8902de1SWarner Losh 	/* Check ZFS features for read */
1946b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_FEATURES_FOR_READ,
1947b8902de1SWarner Losh 	    DATA_TYPE_NVLIST, NULL, &features) == 0 &&
1948b8902de1SWarner Losh 	    nvlist_check_features_for_read(features) != 0) {
194979a4bf89SToomas Soome 		free(nvlist);
1950b8902de1SWarner Losh 		return (EIO);
1951b8902de1SWarner Losh 	}
1952b8902de1SWarner Losh 
1953b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_POOL_STATE, DATA_TYPE_UINT64,
1954b8902de1SWarner Losh 	    NULL, &val) != 0) {
195579a4bf89SToomas Soome 		free(nvlist);
1956b8902de1SWarner Losh 		return (EIO);
1957b8902de1SWarner Losh 	}
1958b8902de1SWarner Losh 
1959b8902de1SWarner Losh 	if (val == POOL_STATE_DESTROYED) {
1960b8902de1SWarner Losh 		/* We don't boot only from destroyed pools. */
196179a4bf89SToomas Soome 		free(nvlist);
1962b8902de1SWarner Losh 		return (EIO);
1963b8902de1SWarner Losh 	}
1964b8902de1SWarner Losh 
1965b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_POOL_TXG, DATA_TYPE_UINT64,
1966b8902de1SWarner Losh 	    NULL, &pool_txg) != 0 ||
1967b8902de1SWarner Losh 	    nvlist_find(nvlist, ZPOOL_CONFIG_POOL_GUID, DATA_TYPE_UINT64,
1968b8902de1SWarner Losh 	    NULL, &pool_guid) != 0 ||
1969b8902de1SWarner Losh 	    nvlist_find(nvlist, ZPOOL_CONFIG_POOL_NAME, DATA_TYPE_STRING,
1970b8902de1SWarner Losh 	    NULL, &pool_name) != 0) {
1971b8902de1SWarner Losh 		/*
1972b8902de1SWarner Losh 		 * Cache and spare devices end up here - just ignore
1973b8902de1SWarner Losh 		 * them.
1974b8902de1SWarner Losh 		 */
197579a4bf89SToomas Soome 		free(nvlist);
1976b8902de1SWarner Losh 		return (EIO);
1977b8902de1SWarner Losh 	}
1978b8902de1SWarner Losh 
1979b8902de1SWarner Losh 	/*
1980b8902de1SWarner Losh 	 * Create the pool if this is the first time we've seen it.
1981b8902de1SWarner Losh 	 */
1982b8902de1SWarner Losh 	spa = spa_find_by_guid(pool_guid);
1983b8902de1SWarner Losh 	if (spa == NULL) {
19843c2db0efSToomas Soome 		nvlist_find(nvlist, ZPOOL_CONFIG_VDEV_CHILDREN,
19853c2db0efSToomas Soome 		    DATA_TYPE_UINT64, NULL, &vdev_children);
1986b8902de1SWarner Losh 		spa = spa_create(pool_guid, pool_name);
198779a4bf89SToomas Soome 		if (spa == NULL) {
198879a4bf89SToomas Soome 			free(nvlist);
1989b8902de1SWarner Losh 			return (ENOMEM);
1990b8902de1SWarner Losh 		}
19913c2db0efSToomas Soome 		spa->spa_root_vdev->v_nchildren = vdev_children;
199279a4bf89SToomas Soome 	}
19933c2db0efSToomas Soome 	if (pool_txg > spa->spa_txg)
1994b8902de1SWarner Losh 		spa->spa_txg = pool_txg;
1995b8902de1SWarner Losh 
1996b8902de1SWarner Losh 	/*
1997b8902de1SWarner Losh 	 * Get the vdev tree and create our in-core copy of it.
1998b8902de1SWarner Losh 	 * If we already have a vdev with this guid, this must
1999b8902de1SWarner Losh 	 * be some kind of alias (overlapping slices, dangerously dedicated
2000b8902de1SWarner Losh 	 * disks etc).
2001b8902de1SWarner Losh 	 */
2002b8902de1SWarner Losh 	if (nvlist_find(nvlist, ZPOOL_CONFIG_GUID, DATA_TYPE_UINT64,
2003b8902de1SWarner Losh 	    NULL, &guid) != 0) {
200479a4bf89SToomas Soome 		free(nvlist);
2005b8902de1SWarner Losh 		return (EIO);
2006b8902de1SWarner Losh 	}
2007b8902de1SWarner Losh 	vdev = vdev_find(guid);
200879a4bf89SToomas Soome 	/* Has this vdev already been inited? */
200979a4bf89SToomas Soome 	if (vdev && vdev->v_phys_read) {
201079a4bf89SToomas Soome 		free(nvlist);
2011b8902de1SWarner Losh 		return (EIO);
201279a4bf89SToomas Soome 	}
2013b8902de1SWarner Losh 
20143c2db0efSToomas Soome 	rc = vdev_init_from_label(spa, nvlist);
201579a4bf89SToomas Soome 	free(nvlist);
2016b8902de1SWarner Losh 	if (rc != 0)
2017b8902de1SWarner Losh 		return (rc);
2018b8902de1SWarner Losh 
2019b8902de1SWarner Losh 	/*
2020b8902de1SWarner Losh 	 * We should already have created an incomplete vdev for this
2021b8902de1SWarner Losh 	 * vdev. Find it and initialise it with our read proc.
2022b8902de1SWarner Losh 	 */
2023b8902de1SWarner Losh 	vdev = vdev_find(guid);
20243c2db0efSToomas Soome 	if (vdev != NULL) {
2025b8902de1SWarner Losh 		vdev->v_phys_read = _read;
2026b8902de1SWarner Losh 		vdev->v_read_priv = read_priv;
2027abca0bd5SToomas Soome 		vdev->v_psize = vtmp.v_psize;
20283c2db0efSToomas Soome 		/*
20293c2db0efSToomas Soome 		 * If no other state is set, mark vdev healthy.
20303c2db0efSToomas Soome 		 */
20313c2db0efSToomas Soome 		if (vdev->v_state == VDEV_STATE_UNKNOWN)
20323c2db0efSToomas Soome 			vdev->v_state = VDEV_STATE_HEALTHY;
2033b8902de1SWarner Losh 	} else {
2034b8902de1SWarner Losh 		printf("ZFS: inconsistent nvlist contents\n");
2035b8902de1SWarner Losh 		return (EIO);
2036b8902de1SWarner Losh 	}
2037b8902de1SWarner Losh 
20380c0a882cSToomas Soome 	if (vdev->v_islog)
20390c0a882cSToomas Soome 		spa->spa_with_log = vdev->v_islog;
20400c0a882cSToomas Soome 
20410c0a882cSToomas Soome 	/*
2042b8902de1SWarner Losh 	 * Re-evaluate top-level vdev state.
2043b8902de1SWarner Losh 	 */
20443c2db0efSToomas Soome 	vdev_set_state(vdev->v_top);
2045b8902de1SWarner Losh 
2046b8902de1SWarner Losh 	/*
2047b8902de1SWarner Losh 	 * Ok, we are happy with the pool so far. Lets find
2048b8902de1SWarner Losh 	 * the best uberblock and then we can actually access
2049b8902de1SWarner Losh 	 * the contents of the pool.
2050b8902de1SWarner Losh 	 */
205179a4bf89SToomas Soome 	vdev_uberblock_load(vdev, &spa->spa_uberblock);
2052b8902de1SWarner Losh 
2053b8902de1SWarner Losh 	if (spap != NULL)
2054b8902de1SWarner Losh 		*spap = spa;
2055b8902de1SWarner Losh 	return (0);
2056b8902de1SWarner Losh }
2057b8902de1SWarner Losh 
2058b8902de1SWarner Losh static int
2059b8902de1SWarner Losh ilog2(int n)
2060b8902de1SWarner Losh {
2061b8902de1SWarner Losh 	int v;
2062b8902de1SWarner Losh 
2063b8902de1SWarner Losh 	for (v = 0; v < 32; v++)
2064b8902de1SWarner Losh 		if (n == (1 << v))
20652e6bb655SToomas Soome 			return (v);
20662e6bb655SToomas Soome 	return (-1);
2067b8902de1SWarner Losh }
2068b8902de1SWarner Losh 
2069b8902de1SWarner Losh static int
2070b8902de1SWarner Losh zio_read_gang(const spa_t *spa, const blkptr_t *bp, void *buf)
2071b8902de1SWarner Losh {
2072b8902de1SWarner Losh 	blkptr_t gbh_bp;
2073b8902de1SWarner Losh 	zio_gbh_phys_t zio_gb;
2074b8902de1SWarner Losh 	char *pbuf;
2075b8902de1SWarner Losh 	int i;
2076b8902de1SWarner Losh 
2077b8902de1SWarner Losh 	/* Artificial BP for gang block header. */
2078b8902de1SWarner Losh 	gbh_bp = *bp;
2079b8902de1SWarner Losh 	BP_SET_PSIZE(&gbh_bp, SPA_GANGBLOCKSIZE);
2080b8902de1SWarner Losh 	BP_SET_LSIZE(&gbh_bp, SPA_GANGBLOCKSIZE);
2081b8902de1SWarner Losh 	BP_SET_CHECKSUM(&gbh_bp, ZIO_CHECKSUM_GANG_HEADER);
2082b8902de1SWarner Losh 	BP_SET_COMPRESS(&gbh_bp, ZIO_COMPRESS_OFF);
2083b8902de1SWarner Losh 	for (i = 0; i < SPA_DVAS_PER_BP; i++)
2084b8902de1SWarner Losh 		DVA_SET_GANG(&gbh_bp.blk_dva[i], 0);
2085b8902de1SWarner Losh 
2086b8902de1SWarner Losh 	/* Read gang header block using the artificial BP. */
2087b8902de1SWarner Losh 	if (zio_read(spa, &gbh_bp, &zio_gb))
2088b8902de1SWarner Losh 		return (EIO);
2089b8902de1SWarner Losh 
2090b8902de1SWarner Losh 	pbuf = buf;
2091b8902de1SWarner Losh 	for (i = 0; i < SPA_GBH_NBLKPTRS; i++) {
2092b8902de1SWarner Losh 		blkptr_t *gbp = &zio_gb.zg_blkptr[i];
2093b8902de1SWarner Losh 
2094b8902de1SWarner Losh 		if (BP_IS_HOLE(gbp))
2095b8902de1SWarner Losh 			continue;
2096b8902de1SWarner Losh 		if (zio_read(spa, gbp, pbuf))
2097b8902de1SWarner Losh 			return (EIO);
2098b8902de1SWarner Losh 		pbuf += BP_GET_PSIZE(gbp);
2099b8902de1SWarner Losh 	}
2100b8902de1SWarner Losh 
2101b8902de1SWarner Losh 	if (zio_checksum_verify(spa, bp, buf))
2102b8902de1SWarner Losh 		return (EIO);
2103b8902de1SWarner Losh 	return (0);
2104b8902de1SWarner Losh }
2105b8902de1SWarner Losh 
2106b8902de1SWarner Losh static int
2107b8902de1SWarner Losh zio_read(const spa_t *spa, const blkptr_t *bp, void *buf)
2108b8902de1SWarner Losh {
2109b8902de1SWarner Losh 	int cpfunc = BP_GET_COMPRESS(bp);
2110b8902de1SWarner Losh 	uint64_t align, size;
2111b8902de1SWarner Losh 	void *pbuf;
2112b8902de1SWarner Losh 	int i, error;
2113b8902de1SWarner Losh 
2114b8902de1SWarner Losh 	/*
2115b8902de1SWarner Losh 	 * Process data embedded in block pointer
2116b8902de1SWarner Losh 	 */
2117b8902de1SWarner Losh 	if (BP_IS_EMBEDDED(bp)) {
2118b8902de1SWarner Losh 		ASSERT(BPE_GET_ETYPE(bp) == BP_EMBEDDED_TYPE_DATA);
2119b8902de1SWarner Losh 
2120b8902de1SWarner Losh 		size = BPE_GET_PSIZE(bp);
2121b8902de1SWarner Losh 		ASSERT(size <= BPE_PAYLOAD_SIZE);
2122b8902de1SWarner Losh 
2123b8902de1SWarner Losh 		if (cpfunc != ZIO_COMPRESS_OFF)
2124*c1c4c81fSToomas Soome 			pbuf = malloc(size);
2125b8902de1SWarner Losh 		else
2126b8902de1SWarner Losh 			pbuf = buf;
2127b8902de1SWarner Losh 
2128*c1c4c81fSToomas Soome 		if (pbuf == NULL)
2129*c1c4c81fSToomas Soome 			return (ENOMEM);
2130*c1c4c81fSToomas Soome 
2131b8902de1SWarner Losh 		decode_embedded_bp_compressed(bp, pbuf);
2132b8902de1SWarner Losh 		error = 0;
2133b8902de1SWarner Losh 
2134b8902de1SWarner Losh 		if (cpfunc != ZIO_COMPRESS_OFF) {
2135b8902de1SWarner Losh 			error = zio_decompress_data(cpfunc, pbuf,
2136b8902de1SWarner Losh 			    size, buf, BP_GET_LSIZE(bp));
2137*c1c4c81fSToomas Soome 			free(pbuf);
2138b8902de1SWarner Losh 		}
2139b8902de1SWarner Losh 		if (error != 0)
21402e6bb655SToomas Soome 			printf("ZFS: i/o error - unable to decompress "
21412e6bb655SToomas Soome 			    "block pointer data, error %d\n", error);
2142b8902de1SWarner Losh 		return (error);
2143b8902de1SWarner Losh 	}
2144b8902de1SWarner Losh 
2145b8902de1SWarner Losh 	error = EIO;
2146b8902de1SWarner Losh 
2147b8902de1SWarner Losh 	for (i = 0; i < SPA_DVAS_PER_BP; i++) {
2148b8902de1SWarner Losh 		const dva_t *dva = &bp->blk_dva[i];
2149b8902de1SWarner Losh 		vdev_t *vdev;
21503c2db0efSToomas Soome 		vdev_list_t *vlist;
21513c2db0efSToomas Soome 		uint64_t vdevid;
2152b8902de1SWarner Losh 		off_t offset;
2153b8902de1SWarner Losh 
2154b8902de1SWarner Losh 		if (!dva->dva_word[0] && !dva->dva_word[1])
2155b8902de1SWarner Losh 			continue;
2156b8902de1SWarner Losh 
2157b8902de1SWarner Losh 		vdevid = DVA_GET_VDEV(dva);
2158b8902de1SWarner Losh 		offset = DVA_GET_OFFSET(dva);
21593c2db0efSToomas Soome 		vlist = &spa->spa_root_vdev->v_children;
21603c2db0efSToomas Soome 		STAILQ_FOREACH(vdev, vlist, v_childlink) {
2161b8902de1SWarner Losh 			if (vdev->v_id == vdevid)
2162b8902de1SWarner Losh 				break;
2163b8902de1SWarner Losh 		}
2164b8902de1SWarner Losh 		if (!vdev || !vdev->v_read)
2165b8902de1SWarner Losh 			continue;
2166b8902de1SWarner Losh 
2167b8902de1SWarner Losh 		size = BP_GET_PSIZE(bp);
2168b8902de1SWarner Losh 		if (vdev->v_read == vdev_raidz_read) {
21693c2db0efSToomas Soome 			align = 1ULL << vdev->v_ashift;
2170b8902de1SWarner Losh 			if (P2PHASE(size, align) != 0)
2171b8902de1SWarner Losh 				size = P2ROUNDUP(size, align);
2172b8902de1SWarner Losh 		}
2173b8902de1SWarner Losh 		if (size != BP_GET_PSIZE(bp) || cpfunc != ZIO_COMPRESS_OFF)
2174*c1c4c81fSToomas Soome 			pbuf = malloc(size);
2175b8902de1SWarner Losh 		else
2176b8902de1SWarner Losh 			pbuf = buf;
2177b8902de1SWarner Losh 
2178*c1c4c81fSToomas Soome 		if (pbuf == NULL) {
2179*c1c4c81fSToomas Soome 			error = ENOMEM;
2180*c1c4c81fSToomas Soome 			break;
2181*c1c4c81fSToomas Soome 		}
2182*c1c4c81fSToomas Soome 
2183b8902de1SWarner Losh 		if (DVA_GET_GANG(dva))
2184b8902de1SWarner Losh 			error = zio_read_gang(spa, bp, pbuf);
2185b8902de1SWarner Losh 		else
2186b8902de1SWarner Losh 			error = vdev->v_read(vdev, bp, pbuf, offset, size);
2187b8902de1SWarner Losh 		if (error == 0) {
2188b8902de1SWarner Losh 			if (cpfunc != ZIO_COMPRESS_OFF)
2189b8902de1SWarner Losh 				error = zio_decompress_data(cpfunc, pbuf,
2190b8902de1SWarner Losh 				    BP_GET_PSIZE(bp), buf, BP_GET_LSIZE(bp));
2191b8902de1SWarner Losh 			else if (size != BP_GET_PSIZE(bp))
2192b8902de1SWarner Losh 				bcopy(pbuf, buf, BP_GET_PSIZE(bp));
2193b8902de1SWarner Losh 		}
2194b8902de1SWarner Losh 		if (buf != pbuf)
2195*c1c4c81fSToomas Soome 			free(pbuf);
2196b8902de1SWarner Losh 		if (error == 0)
2197b8902de1SWarner Losh 			break;
2198b8902de1SWarner Losh 	}
2199b8902de1SWarner Losh 	if (error != 0)
2200b8902de1SWarner Losh 		printf("ZFS: i/o error - all block copies unavailable\n");
2201*c1c4c81fSToomas Soome 
2202b8902de1SWarner Losh 	return (error);
2203b8902de1SWarner Losh }
2204b8902de1SWarner Losh 
2205b8902de1SWarner Losh static int
22062e6bb655SToomas Soome dnode_read(const spa_t *spa, const dnode_phys_t *dnode, off_t offset,
22072e6bb655SToomas Soome     void *buf, size_t buflen)
2208b8902de1SWarner Losh {
2209b8902de1SWarner Losh 	int ibshift = dnode->dn_indblkshift - SPA_BLKPTRSHIFT;
2210b8902de1SWarner Losh 	int bsize = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2211b8902de1SWarner Losh 	int nlevels = dnode->dn_nlevels;
2212b8902de1SWarner Losh 	int i, rc;
2213b8902de1SWarner Losh 
2214b8902de1SWarner Losh 	if (bsize > SPA_MAXBLOCKSIZE) {
2215b8902de1SWarner Losh 		printf("ZFS: I/O error - blocks larger than %llu are not "
2216b8902de1SWarner Losh 		    "supported\n", SPA_MAXBLOCKSIZE);
2217b8902de1SWarner Losh 		return (EIO);
2218b8902de1SWarner Losh 	}
2219b8902de1SWarner Losh 
2220b8902de1SWarner Losh 	/*
2221b8902de1SWarner Losh 	 * Note: bsize may not be a power of two here so we need to do an
2222b8902de1SWarner Losh 	 * actual divide rather than a bitshift.
2223b8902de1SWarner Losh 	 */
2224b8902de1SWarner Losh 	while (buflen > 0) {
2225b8902de1SWarner Losh 		uint64_t bn = offset / bsize;
2226b8902de1SWarner Losh 		int boff = offset % bsize;
2227b8902de1SWarner Losh 		int ibn;
2228b8902de1SWarner Losh 		const blkptr_t *indbp;
2229b8902de1SWarner Losh 		blkptr_t bp;
2230b8902de1SWarner Losh 
2231b8902de1SWarner Losh 		if (bn > dnode->dn_maxblkid)
2232b8902de1SWarner Losh 			return (EIO);
2233b8902de1SWarner Losh 
2234b8902de1SWarner Losh 		if (dnode == dnode_cache_obj && bn == dnode_cache_bn)
2235b8902de1SWarner Losh 			goto cached;
2236b8902de1SWarner Losh 
2237b8902de1SWarner Losh 		indbp = dnode->dn_blkptr;
2238b8902de1SWarner Losh 		for (i = 0; i < nlevels; i++) {
2239b8902de1SWarner Losh 			/*
2240b8902de1SWarner Losh 			 * Copy the bp from the indirect array so that
2241b8902de1SWarner Losh 			 * we can re-use the scratch buffer for multi-level
2242b8902de1SWarner Losh 			 * objects.
2243b8902de1SWarner Losh 			 */
2244b8902de1SWarner Losh 			ibn = bn >> ((nlevels - i - 1) * ibshift);
2245b8902de1SWarner Losh 			ibn &= ((1 << ibshift) - 1);
2246b8902de1SWarner Losh 			bp = indbp[ibn];
2247b8902de1SWarner Losh 			if (BP_IS_HOLE(&bp)) {
2248b8902de1SWarner Losh 				memset(dnode_cache_buf, 0, bsize);
2249b8902de1SWarner Losh 				break;
2250b8902de1SWarner Losh 			}
2251b8902de1SWarner Losh 			rc = zio_read(spa, &bp, dnode_cache_buf);
2252b8902de1SWarner Losh 			if (rc)
2253b8902de1SWarner Losh 				return (rc);
2254b8902de1SWarner Losh 			indbp = (const blkptr_t *) dnode_cache_buf;
2255b8902de1SWarner Losh 		}
2256b8902de1SWarner Losh 		dnode_cache_obj = dnode;
2257b8902de1SWarner Losh 		dnode_cache_bn = bn;
2258b8902de1SWarner Losh 	cached:
2259b8902de1SWarner Losh 
2260b8902de1SWarner Losh 		/*
2261b8902de1SWarner Losh 		 * The buffer contains our data block. Copy what we
2262b8902de1SWarner Losh 		 * need from it and loop.
2263b8902de1SWarner Losh 		 */
2264b8902de1SWarner Losh 		i = bsize - boff;
2265b8902de1SWarner Losh 		if (i > buflen) i = buflen;
2266b8902de1SWarner Losh 		memcpy(buf, &dnode_cache_buf[boff], i);
2267b8902de1SWarner Losh 		buf = ((char *)buf) + i;
2268b8902de1SWarner Losh 		offset += i;
2269b8902de1SWarner Losh 		buflen -= i;
2270b8902de1SWarner Losh 	}
2271b8902de1SWarner Losh 
2272b8902de1SWarner Losh 	return (0);
2273b8902de1SWarner Losh }
2274b8902de1SWarner Losh 
2275b8902de1SWarner Losh /*
2276380977d5SToomas Soome  * Lookup a value in a microzap directory.
2277b8902de1SWarner Losh  */
2278b8902de1SWarner Losh static int
22794d297e70SToomas Soome mzap_lookup(const mzap_phys_t *mz, size_t size, const char *name,
22804d297e70SToomas Soome     uint64_t *value)
2281b8902de1SWarner Losh {
2282b8902de1SWarner Losh 	const mzap_ent_phys_t *mze;
2283b8902de1SWarner Losh 	int chunks, i;
2284b8902de1SWarner Losh 
2285b8902de1SWarner Losh 	/*
2286b8902de1SWarner Losh 	 * Microzap objects use exactly one block. Read the whole
2287b8902de1SWarner Losh 	 * thing.
2288b8902de1SWarner Losh 	 */
2289b8902de1SWarner Losh 	chunks = size / MZAP_ENT_LEN - 1;
2290b8902de1SWarner Losh 	for (i = 0; i < chunks; i++) {
2291b8902de1SWarner Losh 		mze = &mz->mz_chunk[i];
22922e6bb655SToomas Soome 		if (strcmp(mze->mze_name, name) == 0) {
2293b8902de1SWarner Losh 			*value = mze->mze_value;
2294b8902de1SWarner Losh 			return (0);
2295b8902de1SWarner Losh 		}
2296b8902de1SWarner Losh 	}
2297b8902de1SWarner Losh 
2298b8902de1SWarner Losh 	return (ENOENT);
2299b8902de1SWarner Losh }
2300b8902de1SWarner Losh 
2301b8902de1SWarner Losh /*
2302b8902de1SWarner Losh  * Compare a name with a zap leaf entry. Return non-zero if the name
2303b8902de1SWarner Losh  * matches.
2304b8902de1SWarner Losh  */
2305b8902de1SWarner Losh static int
23062e6bb655SToomas Soome fzap_name_equal(const zap_leaf_t *zl, const zap_leaf_chunk_t *zc,
23072e6bb655SToomas Soome     const char *name)
2308b8902de1SWarner Losh {
2309b8902de1SWarner Losh 	size_t namelen;
2310b8902de1SWarner Losh 	const zap_leaf_chunk_t *nc;
2311b8902de1SWarner Losh 	const char *p;
2312b8902de1SWarner Losh 
2313b8902de1SWarner Losh 	namelen = zc->l_entry.le_name_numints;
2314b8902de1SWarner Losh 
2315b8902de1SWarner Losh 	nc = &ZAP_LEAF_CHUNK(zl, zc->l_entry.le_name_chunk);
2316b8902de1SWarner Losh 	p = name;
2317b8902de1SWarner Losh 	while (namelen > 0) {
2318b8902de1SWarner Losh 		size_t len;
23192e6bb655SToomas Soome 
2320b8902de1SWarner Losh 		len = namelen;
2321b8902de1SWarner Losh 		if (len > ZAP_LEAF_ARRAY_BYTES)
2322b8902de1SWarner Losh 			len = ZAP_LEAF_ARRAY_BYTES;
2323b8902de1SWarner Losh 		if (memcmp(p, nc->l_array.la_array, len))
2324b8902de1SWarner Losh 			return (0);
2325b8902de1SWarner Losh 		p += len;
2326b8902de1SWarner Losh 		namelen -= len;
2327b8902de1SWarner Losh 		nc = &ZAP_LEAF_CHUNK(zl, nc->l_array.la_next);
2328b8902de1SWarner Losh 	}
2329b8902de1SWarner Losh 
23302e6bb655SToomas Soome 	return (1);
2331b8902de1SWarner Losh }
2332b8902de1SWarner Losh 
2333b8902de1SWarner Losh /*
2334b8902de1SWarner Losh  * Extract a uint64_t value from a zap leaf entry.
2335b8902de1SWarner Losh  */
2336b8902de1SWarner Losh static uint64_t
2337b8902de1SWarner Losh fzap_leaf_value(const zap_leaf_t *zl, const zap_leaf_chunk_t *zc)
2338b8902de1SWarner Losh {
2339b8902de1SWarner Losh 	const zap_leaf_chunk_t *vc;
2340b8902de1SWarner Losh 	int i;
2341b8902de1SWarner Losh 	uint64_t value;
2342b8902de1SWarner Losh 	const uint8_t *p;
2343b8902de1SWarner Losh 
2344b8902de1SWarner Losh 	vc = &ZAP_LEAF_CHUNK(zl, zc->l_entry.le_value_chunk);
2345b8902de1SWarner Losh 	for (i = 0, value = 0, p = vc->l_array.la_array; i < 8; i++) {
2346b8902de1SWarner Losh 		value = (value << 8) | p[i];
2347b8902de1SWarner Losh 	}
2348b8902de1SWarner Losh 
23492e6bb655SToomas Soome 	return (value);
2350b8902de1SWarner Losh }
2351b8902de1SWarner Losh 
2352b8902de1SWarner Losh static void
2353b8902de1SWarner Losh stv(int len, void *addr, uint64_t value)
2354b8902de1SWarner Losh {
2355b8902de1SWarner Losh 	switch (len) {
2356b8902de1SWarner Losh 	case 1:
2357b8902de1SWarner Losh 		*(uint8_t *)addr = value;
2358b8902de1SWarner Losh 		return;
2359b8902de1SWarner Losh 	case 2:
2360b8902de1SWarner Losh 		*(uint16_t *)addr = value;
2361b8902de1SWarner Losh 		return;
2362b8902de1SWarner Losh 	case 4:
2363b8902de1SWarner Losh 		*(uint32_t *)addr = value;
2364b8902de1SWarner Losh 		return;
2365b8902de1SWarner Losh 	case 8:
2366b8902de1SWarner Losh 		*(uint64_t *)addr = value;
2367b8902de1SWarner Losh 		return;
2368b8902de1SWarner Losh 	}
2369b8902de1SWarner Losh }
2370b8902de1SWarner Losh 
2371b8902de1SWarner Losh /*
2372b8902de1SWarner Losh  * Extract a array from a zap leaf entry.
2373b8902de1SWarner Losh  */
2374b8902de1SWarner Losh static void
2375b8902de1SWarner Losh fzap_leaf_array(const zap_leaf_t *zl, const zap_leaf_chunk_t *zc,
2376b8902de1SWarner Losh     uint64_t integer_size, uint64_t num_integers, void *buf)
2377b8902de1SWarner Losh {
2378b8902de1SWarner Losh 	uint64_t array_int_len = zc->l_entry.le_value_intlen;
2379b8902de1SWarner Losh 	uint64_t value = 0;
2380b8902de1SWarner Losh 	uint64_t *u64 = buf;
2381b8902de1SWarner Losh 	char *p = buf;
2382b8902de1SWarner Losh 	int len = MIN(zc->l_entry.le_value_numints, num_integers);
2383b8902de1SWarner Losh 	int chunk = zc->l_entry.le_value_chunk;
2384b8902de1SWarner Losh 	int byten = 0;
2385b8902de1SWarner Losh 
2386b8902de1SWarner Losh 	if (integer_size == 8 && len == 1) {
2387b8902de1SWarner Losh 		*u64 = fzap_leaf_value(zl, zc);
2388b8902de1SWarner Losh 		return;
2389b8902de1SWarner Losh 	}
2390b8902de1SWarner Losh 
2391b8902de1SWarner Losh 	while (len > 0) {
2392b8902de1SWarner Losh 		struct zap_leaf_array *la = &ZAP_LEAF_CHUNK(zl, chunk).l_array;
2393b8902de1SWarner Losh 		int i;
2394b8902de1SWarner Losh 
2395b8902de1SWarner Losh 		ASSERT3U(chunk, <, ZAP_LEAF_NUMCHUNKS(zl));
2396b8902de1SWarner Losh 		for (i = 0; i < ZAP_LEAF_ARRAY_BYTES && len > 0; i++) {
2397b8902de1SWarner Losh 			value = (value << 8) | la->la_array[i];
2398b8902de1SWarner Losh 			byten++;
2399b8902de1SWarner Losh 			if (byten == array_int_len) {
2400b8902de1SWarner Losh 				stv(integer_size, p, value);
2401b8902de1SWarner Losh 				byten = 0;
2402b8902de1SWarner Losh 				len--;
2403b8902de1SWarner Losh 				if (len == 0)
2404b8902de1SWarner Losh 					return;
2405b8902de1SWarner Losh 				p += integer_size;
2406b8902de1SWarner Losh 			}
2407b8902de1SWarner Losh 		}
2408b8902de1SWarner Losh 		chunk = la->la_next;
2409b8902de1SWarner Losh 	}
2410b8902de1SWarner Losh }
2411b8902de1SWarner Losh 
24123c2db0efSToomas Soome static int
24133c2db0efSToomas Soome fzap_check_size(uint64_t integer_size, uint64_t num_integers)
24143c2db0efSToomas Soome {
24153c2db0efSToomas Soome 
24163c2db0efSToomas Soome 	switch (integer_size) {
24173c2db0efSToomas Soome 	case 1:
24183c2db0efSToomas Soome 	case 2:
24193c2db0efSToomas Soome 	case 4:
24203c2db0efSToomas Soome 	case 8:
24213c2db0efSToomas Soome 		break;
24223c2db0efSToomas Soome 	default:
24233c2db0efSToomas Soome 		return (EINVAL);
24243c2db0efSToomas Soome 	}
24253c2db0efSToomas Soome 
24263c2db0efSToomas Soome 	if (integer_size * num_integers > ZAP_MAXVALUELEN)
24273c2db0efSToomas Soome 		return (E2BIG);
24283c2db0efSToomas Soome 
24293c2db0efSToomas Soome 	return (0);
24303c2db0efSToomas Soome }
24313c2db0efSToomas Soome 
24324d297e70SToomas Soome static void
24334d297e70SToomas Soome zap_leaf_free(zap_leaf_t *leaf)
24344d297e70SToomas Soome {
24354d297e70SToomas Soome 	free(leaf->l_phys);
24364d297e70SToomas Soome 	free(leaf);
24374d297e70SToomas Soome }
24384d297e70SToomas Soome 
2439b8902de1SWarner Losh static int
24404d297e70SToomas Soome zap_get_leaf_byblk(fat_zap_t *zap, uint64_t blk, zap_leaf_t **lp)
24414d297e70SToomas Soome {
24424d297e70SToomas Soome 	int bs = FZAP_BLOCK_SHIFT(zap);
24434d297e70SToomas Soome 	int err;
24444d297e70SToomas Soome 
24454d297e70SToomas Soome 	*lp = malloc(sizeof(**lp));
24464d297e70SToomas Soome 	if (*lp == NULL)
24474d297e70SToomas Soome 		return (ENOMEM);
24484d297e70SToomas Soome 
24494d297e70SToomas Soome 	(*lp)->l_bs = bs;
24504d297e70SToomas Soome 	(*lp)->l_phys = malloc(1 << bs);
24514d297e70SToomas Soome 
24524d297e70SToomas Soome 	if ((*lp)->l_phys == NULL) {
24534d297e70SToomas Soome 		free(*lp);
24544d297e70SToomas Soome 		return (ENOMEM);
24554d297e70SToomas Soome 	}
24564d297e70SToomas Soome 	err = dnode_read(zap->zap_spa, zap->zap_dnode, blk << bs, (*lp)->l_phys,
24574d297e70SToomas Soome 	    1 << bs);
24584d297e70SToomas Soome 	if (err != 0) {
24594d297e70SToomas Soome 		zap_leaf_free(*lp);
24604d297e70SToomas Soome 	}
24614d297e70SToomas Soome 	return (err);
24624d297e70SToomas Soome }
24634d297e70SToomas Soome 
24644d297e70SToomas Soome static int
24654d297e70SToomas Soome zap_table_load(fat_zap_t *zap, zap_table_phys_t *tbl, uint64_t idx,
24664d297e70SToomas Soome     uint64_t *valp)
24674d297e70SToomas Soome {
24684d297e70SToomas Soome 	int bs = FZAP_BLOCK_SHIFT(zap);
24694d297e70SToomas Soome 	uint64_t blk = idx >> (bs - 3);
24704d297e70SToomas Soome 	uint64_t off = idx & ((1 << (bs - 3)) - 1);
24714d297e70SToomas Soome 	uint64_t *buf;
24724d297e70SToomas Soome 	int rc;
24734d297e70SToomas Soome 
24744d297e70SToomas Soome 	buf = malloc(1 << zap->zap_block_shift);
24754d297e70SToomas Soome 	if (buf == NULL)
24764d297e70SToomas Soome 		return (ENOMEM);
24774d297e70SToomas Soome 	rc = dnode_read(zap->zap_spa, zap->zap_dnode, (tbl->zt_blk + blk) << bs,
24784d297e70SToomas Soome 	    buf, 1 << zap->zap_block_shift);
24794d297e70SToomas Soome 	if (rc == 0)
24804d297e70SToomas Soome 		*valp = buf[off];
24814d297e70SToomas Soome 	free(buf);
24824d297e70SToomas Soome 	return (rc);
24834d297e70SToomas Soome }
24844d297e70SToomas Soome 
24854d297e70SToomas Soome static int
24864d297e70SToomas Soome zap_idx_to_blk(fat_zap_t *zap, uint64_t idx, uint64_t *valp)
24874d297e70SToomas Soome {
24884d297e70SToomas Soome 	if (zap->zap_phys->zap_ptrtbl.zt_numblks == 0) {
24894d297e70SToomas Soome 		*valp = ZAP_EMBEDDED_PTRTBL_ENT(zap, idx);
24904d297e70SToomas Soome 		return (0);
24914d297e70SToomas Soome 	} else {
24924d297e70SToomas Soome 		return (zap_table_load(zap, &zap->zap_phys->zap_ptrtbl,
24934d297e70SToomas Soome 		    idx, valp));
24944d297e70SToomas Soome 	}
24954d297e70SToomas Soome }
24964d297e70SToomas Soome 
24974d297e70SToomas Soome #define	ZAP_HASH_IDX(hash, n)	(((n) == 0) ? 0 : ((hash) >> (64 - (n))))
24984d297e70SToomas Soome static int
24994d297e70SToomas Soome zap_deref_leaf(fat_zap_t *zap, uint64_t h, zap_leaf_t **lp)
25004d297e70SToomas Soome {
25014d297e70SToomas Soome 	uint64_t idx, blk;
25024d297e70SToomas Soome 	int err;
25034d297e70SToomas Soome 
25044d297e70SToomas Soome 	idx = ZAP_HASH_IDX(h, zap->zap_phys->zap_ptrtbl.zt_shift);
25054d297e70SToomas Soome 	err = zap_idx_to_blk(zap, idx, &blk);
25064d297e70SToomas Soome 	if (err != 0)
25074d297e70SToomas Soome 		return (err);
25084d297e70SToomas Soome 	return (zap_get_leaf_byblk(zap, blk, lp));
25094d297e70SToomas Soome }
25104d297e70SToomas Soome 
25114d297e70SToomas Soome #define	CHAIN_END	0xffff	/* end of the chunk chain */
25124d297e70SToomas Soome #define	LEAF_HASH(l, h) \
25134d297e70SToomas Soome 	((ZAP_LEAF_HASH_NUMENTRIES(l)-1) & \
25144d297e70SToomas Soome 	((h) >> \
25154d297e70SToomas Soome 	(64 - ZAP_LEAF_HASH_SHIFT(l) - (l)->l_phys->l_hdr.lh_prefix_len)))
25164d297e70SToomas Soome #define	LEAF_HASH_ENTPTR(l, h)	(&(l)->l_phys->l_hash[LEAF_HASH(l, h)])
25174d297e70SToomas Soome 
25184d297e70SToomas Soome static int
25194d297e70SToomas Soome zap_leaf_lookup(zap_leaf_t *zl, uint64_t hash, const char *name,
2520b8902de1SWarner Losh     uint64_t integer_size, uint64_t num_integers, void *value)
2521b8902de1SWarner Losh {
25224d297e70SToomas Soome 	int rc;
25234d297e70SToomas Soome 	uint16_t *chunkp;
25244d297e70SToomas Soome 	struct zap_leaf_entry *le;
25254d297e70SToomas Soome 
25264d297e70SToomas Soome 	/*
25274d297e70SToomas Soome 	 * Make sure this chunk matches our hash.
25284d297e70SToomas Soome 	 */
25294d297e70SToomas Soome 	if (zl->l_phys->l_hdr.lh_prefix_len > 0 &&
25304d297e70SToomas Soome 	    zl->l_phys->l_hdr.lh_prefix !=
25314d297e70SToomas Soome 	    hash >> (64 - zl->l_phys->l_hdr.lh_prefix_len))
25324d297e70SToomas Soome 		return (EIO);
25334d297e70SToomas Soome 
25344d297e70SToomas Soome 	rc = ENOENT;
25354d297e70SToomas Soome 	for (chunkp = LEAF_HASH_ENTPTR(zl, hash);
25364d297e70SToomas Soome 	    *chunkp != CHAIN_END; chunkp = &le->le_next) {
25374d297e70SToomas Soome 		zap_leaf_chunk_t *zc;
25384d297e70SToomas Soome 		uint16_t chunk = *chunkp;
25394d297e70SToomas Soome 
25404d297e70SToomas Soome 		le = ZAP_LEAF_ENTRY(zl, chunk);
25414d297e70SToomas Soome 		if (le->le_hash != hash)
25424d297e70SToomas Soome 			continue;
25434d297e70SToomas Soome 		zc = &ZAP_LEAF_CHUNK(zl, chunk);
25444d297e70SToomas Soome 		if (fzap_name_equal(zl, zc, name)) {
25454d297e70SToomas Soome 			if (zc->l_entry.le_value_intlen > integer_size) {
25464d297e70SToomas Soome 				rc = EINVAL;
25474d297e70SToomas Soome 			} else {
25484d297e70SToomas Soome 				fzap_leaf_array(zl, zc, integer_size,
25494d297e70SToomas Soome 				    num_integers, value);
25504d297e70SToomas Soome 				rc = 0;
25514d297e70SToomas Soome 			}
25524d297e70SToomas Soome 			break;
25534d297e70SToomas Soome 		}
25544d297e70SToomas Soome 	}
25554d297e70SToomas Soome 	return (rc);
25564d297e70SToomas Soome }
25574d297e70SToomas Soome 
25584d297e70SToomas Soome /*
25594d297e70SToomas Soome  * Lookup a value in a fatzap directory.
25604d297e70SToomas Soome  */
25614d297e70SToomas Soome static int
25624d297e70SToomas Soome fzap_lookup(const spa_t *spa, const dnode_phys_t *dnode, zap_phys_t *zh,
25634d297e70SToomas Soome     const char *name, uint64_t integer_size, uint64_t num_integers,
25644d297e70SToomas Soome     void *value)
25654d297e70SToomas Soome {
2566b8902de1SWarner Losh 	int bsize = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2567b8902de1SWarner Losh 	fat_zap_t z;
25684d297e70SToomas Soome 	zap_leaf_t *zl;
2569b8902de1SWarner Losh 	uint64_t hash;
2570b8902de1SWarner Losh 	int rc;
2571b8902de1SWarner Losh 
25724d297e70SToomas Soome 	if (zh->zap_magic != ZAP_MAGIC)
2573b8902de1SWarner Losh 		return (EIO);
2574b8902de1SWarner Losh 
25753c2db0efSToomas Soome 	if ((rc = fzap_check_size(integer_size, num_integers)) != 0)
25763c2db0efSToomas Soome 		return (rc);
25773c2db0efSToomas Soome 
2578b8902de1SWarner Losh 	z.zap_block_shift = ilog2(bsize);
25794d297e70SToomas Soome 	z.zap_phys = zh;
25804d297e70SToomas Soome 	z.zap_spa = spa;
25814d297e70SToomas Soome 	z.zap_dnode = dnode;
2582b8902de1SWarner Losh 
25834d297e70SToomas Soome 	hash = zap_hash(zh->zap_salt, name);
25844d297e70SToomas Soome 	rc = zap_deref_leaf(&z, hash, &zl);
25854d297e70SToomas Soome 	if (rc != 0)
2586b8902de1SWarner Losh 		return (rc);
2587b8902de1SWarner Losh 
25884d297e70SToomas Soome 	rc = zap_leaf_lookup(zl, hash, name, integer_size, num_integers, value);
2589b8902de1SWarner Losh 
25904d297e70SToomas Soome 	zap_leaf_free(zl);
25914d297e70SToomas Soome 	return (rc);
2592b8902de1SWarner Losh }
2593b8902de1SWarner Losh 
2594b8902de1SWarner Losh /*
2595b8902de1SWarner Losh  * Lookup a name in a zap object and return its value as a uint64_t.
2596b8902de1SWarner Losh  */
2597b8902de1SWarner Losh static int
2598b8902de1SWarner Losh zap_lookup(const spa_t *spa, const dnode_phys_t *dnode, const char *name,
2599b8902de1SWarner Losh     uint64_t integer_size, uint64_t num_integers, void *value)
2600b8902de1SWarner Losh {
2601b8902de1SWarner Losh 	int rc;
26024d297e70SToomas Soome 	zap_phys_t *zap;
2603b8902de1SWarner Losh 	size_t size = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2604b8902de1SWarner Losh 
26054d297e70SToomas Soome 	zap = malloc(size);
26064d297e70SToomas Soome 	if (zap == NULL)
26074d297e70SToomas Soome 		return (ENOMEM);
2608b8902de1SWarner Losh 
26094d297e70SToomas Soome 	rc = dnode_read(spa, dnode, 0, zap, size);
26104d297e70SToomas Soome 	if (rc)
26114d297e70SToomas Soome 		goto done;
26124d297e70SToomas Soome 
26134d297e70SToomas Soome 	switch (zap->zap_block_type) {
26144d297e70SToomas Soome 	case ZBT_MICRO:
26154d297e70SToomas Soome 		rc = mzap_lookup((const mzap_phys_t *)zap, size, name, value);
26164d297e70SToomas Soome 		break;
26174d297e70SToomas Soome 	case ZBT_HEADER:
26184d297e70SToomas Soome 		rc = fzap_lookup(spa, dnode, zap, name, integer_size,
26194d297e70SToomas Soome 		    num_integers, value);
26204d297e70SToomas Soome 		break;
26214d297e70SToomas Soome 	default:
26224d297e70SToomas Soome 		printf("ZFS: invalid zap_type=%" PRIx64 "\n",
26234d297e70SToomas Soome 		    zap->zap_block_type);
26244d297e70SToomas Soome 		rc = EIO;
2625b8902de1SWarner Losh 	}
26264d297e70SToomas Soome done:
26274d297e70SToomas Soome 	free(zap);
26284d297e70SToomas Soome 	return (rc);
2629b8902de1SWarner Losh }
2630b8902de1SWarner Losh 
2631b8902de1SWarner Losh /*
26324d297e70SToomas Soome  * List a microzap directory.
2633b8902de1SWarner Losh  */
2634b8902de1SWarner Losh static int
26354d297e70SToomas Soome mzap_list(const mzap_phys_t *mz, size_t size,
26364d297e70SToomas Soome     int (*callback)(const char *, uint64_t))
2637b8902de1SWarner Losh {
2638b8902de1SWarner Losh 	const mzap_ent_phys_t *mze;
2639b8902de1SWarner Losh 	int chunks, i, rc;
2640b8902de1SWarner Losh 
2641b8902de1SWarner Losh 	/*
2642b8902de1SWarner Losh 	 * Microzap objects use exactly one block. Read the whole
2643b8902de1SWarner Losh 	 * thing.
2644b8902de1SWarner Losh 	 */
26454d297e70SToomas Soome 	rc = 0;
2646b8902de1SWarner Losh 	chunks = size / MZAP_ENT_LEN - 1;
2647b8902de1SWarner Losh 	for (i = 0; i < chunks; i++) {
2648b8902de1SWarner Losh 		mze = &mz->mz_chunk[i];
2649b8902de1SWarner Losh 		if (mze->mze_name[0]) {
2650b8902de1SWarner Losh 			rc = callback(mze->mze_name, mze->mze_value);
2651b8902de1SWarner Losh 			if (rc != 0)
26524d297e70SToomas Soome 				break;
2653b8902de1SWarner Losh 		}
2654b8902de1SWarner Losh 	}
2655b8902de1SWarner Losh 
26564d297e70SToomas Soome 	return (rc);
2657b8902de1SWarner Losh }
2658b8902de1SWarner Losh 
2659b8902de1SWarner Losh /*
26604d297e70SToomas Soome  * List a fatzap directory.
2661b8902de1SWarner Losh  */
2662b8902de1SWarner Losh static int
26634d297e70SToomas Soome fzap_list(const spa_t *spa, const dnode_phys_t *dnode, zap_phys_t *zh,
26642e6bb655SToomas Soome     int (*callback)(const char *, uint64_t))
2665b8902de1SWarner Losh {
2666b8902de1SWarner Losh 	int bsize = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2667b8902de1SWarner Losh 	fat_zap_t z;
26684d297e70SToomas Soome 	uint64_t i;
26694d297e70SToomas Soome 	int j, rc;
2670b8902de1SWarner Losh 
26714d297e70SToomas Soome 	if (zh->zap_magic != ZAP_MAGIC)
2672b8902de1SWarner Losh 		return (EIO);
2673b8902de1SWarner Losh 
2674b8902de1SWarner Losh 	z.zap_block_shift = ilog2(bsize);
26754d297e70SToomas Soome 	z.zap_phys = zh;
2676b8902de1SWarner Losh 
2677b8902de1SWarner Losh 	/*
2678b8902de1SWarner Losh 	 * This assumes that the leaf blocks start at block 1. The
2679b8902de1SWarner Losh 	 * documentation isn't exactly clear on this.
2680b8902de1SWarner Losh 	 */
2681b8902de1SWarner Losh 	zap_leaf_t zl;
2682b8902de1SWarner Losh 	zl.l_bs = z.zap_block_shift;
26834d297e70SToomas Soome 	zl.l_phys = malloc(bsize);
26844d297e70SToomas Soome 	if (zl.l_phys == NULL)
26854d297e70SToomas Soome 		return (ENOMEM);
26864d297e70SToomas Soome 
26874d297e70SToomas Soome 	for (i = 0; i < zh->zap_num_leafs; i++) {
26883c2db0efSToomas Soome 		off_t off = ((off_t)(i + 1)) << zl.l_bs;
2689b8902de1SWarner Losh 		char name[256], *p;
2690b8902de1SWarner Losh 		uint64_t value;
2691b8902de1SWarner Losh 
26924d297e70SToomas Soome 		if (dnode_read(spa, dnode, off, zl.l_phys, bsize)) {
26934d297e70SToomas Soome 			free(zl.l_phys);
2694b8902de1SWarner Losh 			return (EIO);
26954d297e70SToomas Soome 		}
2696b8902de1SWarner Losh 
2697b8902de1SWarner Losh 		for (j = 0; j < ZAP_LEAF_NUMCHUNKS(&zl); j++) {
2698b8902de1SWarner Losh 			zap_leaf_chunk_t *zc, *nc;
2699b8902de1SWarner Losh 			int namelen;
2700b8902de1SWarner Losh 
2701b8902de1SWarner Losh 			zc = &ZAP_LEAF_CHUNK(&zl, j);
2702b8902de1SWarner Losh 			if (zc->l_entry.le_type != ZAP_CHUNK_ENTRY)
2703b8902de1SWarner Losh 				continue;
2704b8902de1SWarner Losh 			namelen = zc->l_entry.le_name_numints;
2705b8902de1SWarner Losh 			if (namelen > sizeof(name))
2706b8902de1SWarner Losh 				namelen = sizeof(name);
2707b8902de1SWarner Losh 
2708b8902de1SWarner Losh 			/*
2709b8902de1SWarner Losh 			 * Paste the name back together.
2710b8902de1SWarner Losh 			 */
2711b8902de1SWarner Losh 			nc = &ZAP_LEAF_CHUNK(&zl, zc->l_entry.le_name_chunk);
2712b8902de1SWarner Losh 			p = name;
2713b8902de1SWarner Losh 			while (namelen > 0) {
2714b8902de1SWarner Losh 				int len;
2715b8902de1SWarner Losh 				len = namelen;
2716b8902de1SWarner Losh 				if (len > ZAP_LEAF_ARRAY_BYTES)
2717b8902de1SWarner Losh 					len = ZAP_LEAF_ARRAY_BYTES;
2718b8902de1SWarner Losh 				memcpy(p, nc->l_array.la_array, len);
2719b8902de1SWarner Losh 				p += len;
2720b8902de1SWarner Losh 				namelen -= len;
2721b8902de1SWarner Losh 				nc = &ZAP_LEAF_CHUNK(&zl, nc->l_array.la_next);
2722b8902de1SWarner Losh 			}
2723b8902de1SWarner Losh 
2724b8902de1SWarner Losh 			/*
2725b8902de1SWarner Losh 			 * Assume the first eight bytes of the value are
2726b8902de1SWarner Losh 			 * a uint64_t.
2727b8902de1SWarner Losh 			 */
2728b8902de1SWarner Losh 			value = fzap_leaf_value(&zl, zc);
2729b8902de1SWarner Losh 
27302e6bb655SToomas Soome 			/* printf("%s 0x%jx\n", name, (uintmax_t)value); */
2731b8902de1SWarner Losh 			rc = callback((const char *)name, value);
27324d297e70SToomas Soome 			if (rc != 0) {
27334d297e70SToomas Soome 				free(zl.l_phys);
2734b8902de1SWarner Losh 				return (rc);
2735b8902de1SWarner Losh 			}
2736b8902de1SWarner Losh 		}
27374d297e70SToomas Soome 	}
2738b8902de1SWarner Losh 
27394d297e70SToomas Soome 	free(zl.l_phys);
2740b8902de1SWarner Losh 	return (0);
2741b8902de1SWarner Losh }
2742b8902de1SWarner Losh 
2743b8902de1SWarner Losh static int zfs_printf(const char *name, uint64_t value __unused)
2744b8902de1SWarner Losh {
2745b8902de1SWarner Losh 
2746b8902de1SWarner Losh 	printf("%s\n", name);
2747b8902de1SWarner Losh 
2748b8902de1SWarner Losh 	return (0);
2749b8902de1SWarner Losh }
2750b8902de1SWarner Losh 
2751b8902de1SWarner Losh /*
2752b8902de1SWarner Losh  * List a zap directory.
2753b8902de1SWarner Losh  */
2754b8902de1SWarner Losh static int
2755b8902de1SWarner Losh zap_list(const spa_t *spa, const dnode_phys_t *dnode)
2756b8902de1SWarner Losh {
27574d297e70SToomas Soome 	zap_phys_t *zap;
2758380977d5SToomas Soome 	size_t size = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
27594d297e70SToomas Soome 	int rc;
2760b8902de1SWarner Losh 
27614d297e70SToomas Soome 	zap = malloc(size);
27624d297e70SToomas Soome 	if (zap == NULL)
27634d297e70SToomas Soome 		return (ENOMEM);
2764b8902de1SWarner Losh 
27654d297e70SToomas Soome 	rc = dnode_read(spa, dnode, 0, zap, size);
27664d297e70SToomas Soome 	if (rc == 0) {
27674d297e70SToomas Soome 		if (zap->zap_block_type == ZBT_MICRO)
27684d297e70SToomas Soome 			rc = mzap_list((const mzap_phys_t *)zap, size,
27694d297e70SToomas Soome 			    zfs_printf);
2770b8902de1SWarner Losh 		else
27714d297e70SToomas Soome 			rc = fzap_list(spa, dnode, zap, zfs_printf);
27724d297e70SToomas Soome 	}
27734d297e70SToomas Soome 	free(zap);
27744d297e70SToomas Soome 	return (rc);
2775b8902de1SWarner Losh }
2776b8902de1SWarner Losh 
2777b8902de1SWarner Losh static int
27782e6bb655SToomas Soome objset_get_dnode(const spa_t *spa, const objset_phys_t *os, uint64_t objnum,
27792e6bb655SToomas Soome     dnode_phys_t *dnode)
2780b8902de1SWarner Losh {
2781b8902de1SWarner Losh 	off_t offset;
2782b8902de1SWarner Losh 
2783b8902de1SWarner Losh 	offset = objnum * sizeof(dnode_phys_t);
2784b8902de1SWarner Losh 	return dnode_read(spa, &os->os_meta_dnode, offset,
2785b8902de1SWarner Losh 		dnode, sizeof(dnode_phys_t));
2786b8902de1SWarner Losh }
2787b8902de1SWarner Losh 
27884d297e70SToomas Soome /*
27894d297e70SToomas Soome  * Lookup a name in a microzap directory.
27904d297e70SToomas Soome  */
2791b8902de1SWarner Losh static int
27924d297e70SToomas Soome mzap_rlookup(const mzap_phys_t *mz, size_t size, char *name, uint64_t value)
2793b8902de1SWarner Losh {
2794b8902de1SWarner Losh 	const mzap_ent_phys_t *mze;
2795b8902de1SWarner Losh 	int chunks, i;
2796b8902de1SWarner Losh 
2797b8902de1SWarner Losh 	/*
2798b8902de1SWarner Losh 	 * Microzap objects use exactly one block. Read the whole
2799b8902de1SWarner Losh 	 * thing.
2800b8902de1SWarner Losh 	 */
2801b8902de1SWarner Losh 	chunks = size / MZAP_ENT_LEN - 1;
2802b8902de1SWarner Losh 	for (i = 0; i < chunks; i++) {
2803b8902de1SWarner Losh 		mze = &mz->mz_chunk[i];
2804b8902de1SWarner Losh 		if (value == mze->mze_value) {
2805b8902de1SWarner Losh 			strcpy(name, mze->mze_name);
2806b8902de1SWarner Losh 			return (0);
2807b8902de1SWarner Losh 		}
2808b8902de1SWarner Losh 	}
2809b8902de1SWarner Losh 
2810b8902de1SWarner Losh 	return (ENOENT);
2811b8902de1SWarner Losh }
2812b8902de1SWarner Losh 
2813b8902de1SWarner Losh static void
2814b8902de1SWarner Losh fzap_name_copy(const zap_leaf_t *zl, const zap_leaf_chunk_t *zc, char *name)
2815b8902de1SWarner Losh {
2816b8902de1SWarner Losh 	size_t namelen;
2817b8902de1SWarner Losh 	const zap_leaf_chunk_t *nc;
2818b8902de1SWarner Losh 	char *p;
2819b8902de1SWarner Losh 
2820b8902de1SWarner Losh 	namelen = zc->l_entry.le_name_numints;
2821b8902de1SWarner Losh 
2822b8902de1SWarner Losh 	nc = &ZAP_LEAF_CHUNK(zl, zc->l_entry.le_name_chunk);
2823b8902de1SWarner Losh 	p = name;
2824b8902de1SWarner Losh 	while (namelen > 0) {
2825b8902de1SWarner Losh 		size_t len;
2826b8902de1SWarner Losh 		len = namelen;
2827b8902de1SWarner Losh 		if (len > ZAP_LEAF_ARRAY_BYTES)
2828b8902de1SWarner Losh 			len = ZAP_LEAF_ARRAY_BYTES;
2829b8902de1SWarner Losh 		memcpy(p, nc->l_array.la_array, len);
2830b8902de1SWarner Losh 		p += len;
2831b8902de1SWarner Losh 		namelen -= len;
2832b8902de1SWarner Losh 		nc = &ZAP_LEAF_CHUNK(zl, nc->l_array.la_next);
2833b8902de1SWarner Losh 	}
2834b8902de1SWarner Losh 
2835b8902de1SWarner Losh 	*p = '\0';
2836b8902de1SWarner Losh }
2837b8902de1SWarner Losh 
2838b8902de1SWarner Losh static int
28394d297e70SToomas Soome fzap_rlookup(const spa_t *spa, const dnode_phys_t *dnode, zap_phys_t *zh,
28404d297e70SToomas Soome     char *name, uint64_t value)
2841b8902de1SWarner Losh {
2842b8902de1SWarner Losh 	int bsize = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
2843b8902de1SWarner Losh 	fat_zap_t z;
28444d297e70SToomas Soome 	uint64_t i;
28454d297e70SToomas Soome 	int j, rc;
2846b8902de1SWarner Losh 
28474d297e70SToomas Soome 	if (zh->zap_magic != ZAP_MAGIC)
2848b8902de1SWarner Losh 		return (EIO);
2849b8902de1SWarner Losh 
2850b8902de1SWarner Losh 	z.zap_block_shift = ilog2(bsize);
28514d297e70SToomas Soome 	z.zap_phys = zh;
2852b8902de1SWarner Losh 
2853b8902de1SWarner Losh 	/*
2854b8902de1SWarner Losh 	 * This assumes that the leaf blocks start at block 1. The
2855b8902de1SWarner Losh 	 * documentation isn't exactly clear on this.
2856b8902de1SWarner Losh 	 */
2857b8902de1SWarner Losh 	zap_leaf_t zl;
2858b8902de1SWarner Losh 	zl.l_bs = z.zap_block_shift;
28594d297e70SToomas Soome 	zl.l_phys = malloc(bsize);
28604d297e70SToomas Soome 	if (zl.l_phys == NULL)
28614d297e70SToomas Soome 		return (ENOMEM);
28624d297e70SToomas Soome 
28634d297e70SToomas Soome 	for (i = 0; i < zh->zap_num_leafs; i++) {
28643c2db0efSToomas Soome 		off_t off = ((off_t)(i + 1)) << zl.l_bs;
2865b8902de1SWarner Losh 
28664d297e70SToomas Soome 		rc = dnode_read(spa, dnode, off, zl.l_phys, bsize);
28674d297e70SToomas Soome 		if (rc != 0)
28684d297e70SToomas Soome 			goto done;
2869b8902de1SWarner Losh 
2870b8902de1SWarner Losh 		for (j = 0; j < ZAP_LEAF_NUMCHUNKS(&zl); j++) {
2871b8902de1SWarner Losh 			zap_leaf_chunk_t *zc;
2872b8902de1SWarner Losh 
2873b8902de1SWarner Losh 			zc = &ZAP_LEAF_CHUNK(&zl, j);
2874b8902de1SWarner Losh 			if (zc->l_entry.le_type != ZAP_CHUNK_ENTRY)
2875b8902de1SWarner Losh 				continue;
2876b8902de1SWarner Losh 			if (zc->l_entry.le_value_intlen != 8 ||
2877b8902de1SWarner Losh 			    zc->l_entry.le_value_numints != 1)
2878b8902de1SWarner Losh 				continue;
2879b8902de1SWarner Losh 
2880b8902de1SWarner Losh 			if (fzap_leaf_value(&zl, zc) == value) {
2881b8902de1SWarner Losh 				fzap_name_copy(&zl, zc, name);
28824d297e70SToomas Soome 				goto done;
2883b8902de1SWarner Losh 			}
2884b8902de1SWarner Losh 		}
2885b8902de1SWarner Losh 	}
2886b8902de1SWarner Losh 
28874d297e70SToomas Soome 	rc = ENOENT;
28884d297e70SToomas Soome done:
28894d297e70SToomas Soome 	free(zl.l_phys);
28904d297e70SToomas Soome 	return (rc);
2891b8902de1SWarner Losh }
2892b8902de1SWarner Losh 
2893b8902de1SWarner Losh static int
28942e6bb655SToomas Soome zap_rlookup(const spa_t *spa, const dnode_phys_t *dnode, char *name,
28952e6bb655SToomas Soome     uint64_t value)
2896b8902de1SWarner Losh {
28974d297e70SToomas Soome 	zap_phys_t *zap;
2898380977d5SToomas Soome 	size_t size = dnode->dn_datablkszsec << SPA_MINBLOCKSHIFT;
28994d297e70SToomas Soome 	int rc;
2900b8902de1SWarner Losh 
29014d297e70SToomas Soome 	zap = malloc(size);
29024d297e70SToomas Soome 	if (zap == NULL)
29034d297e70SToomas Soome 		return (ENOMEM);
2904b8902de1SWarner Losh 
29054d297e70SToomas Soome 	rc = dnode_read(spa, dnode, 0, zap, size);
29064d297e70SToomas Soome 	if (rc == 0) {
29074d297e70SToomas Soome 		if (zap->zap_block_type == ZBT_MICRO)
29084d297e70SToomas Soome 			rc = mzap_rlookup((const mzap_phys_t *)zap, size,
29094d297e70SToomas Soome 			    name, value);
2910b8902de1SWarner Losh 		else
29114d297e70SToomas Soome 			rc = fzap_rlookup(spa, dnode, zap, name, value);
29124d297e70SToomas Soome 	}
29134d297e70SToomas Soome 	free(zap);
29144d297e70SToomas Soome 	return (rc);
2915b8902de1SWarner Losh }
2916b8902de1SWarner Losh 
2917b8902de1SWarner Losh static int
2918b8902de1SWarner Losh zfs_rlookup(const spa_t *spa, uint64_t objnum, char *result)
2919b8902de1SWarner Losh {
2920b8902de1SWarner Losh 	char name[256];
2921b8902de1SWarner Losh 	char component[256];
2922b8902de1SWarner Losh 	uint64_t dir_obj, parent_obj, child_dir_zapobj;
2923b8902de1SWarner Losh 	dnode_phys_t child_dir_zap, dataset, dir, parent;
2924b8902de1SWarner Losh 	dsl_dir_phys_t *dd;
2925b8902de1SWarner Losh 	dsl_dataset_phys_t *ds;
2926b8902de1SWarner Losh 	char *p;
2927b8902de1SWarner Losh 	int len;
2928b8902de1SWarner Losh 
2929b8902de1SWarner Losh 	p = &name[sizeof(name) - 1];
2930b8902de1SWarner Losh 	*p = '\0';
2931b8902de1SWarner Losh 
2932b8902de1SWarner Losh 	if (objset_get_dnode(spa, &spa->spa_mos, objnum, &dataset)) {
2933b8902de1SWarner Losh 		printf("ZFS: can't find dataset %ju\n", (uintmax_t)objnum);
2934b8902de1SWarner Losh 		return (EIO);
2935b8902de1SWarner Losh 	}
2936b8902de1SWarner Losh 	ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
2937b8902de1SWarner Losh 	dir_obj = ds->ds_dir_obj;
2938b8902de1SWarner Losh 
2939b8902de1SWarner Losh 	for (;;) {
2940b8902de1SWarner Losh 		if (objset_get_dnode(spa, &spa->spa_mos, dir_obj, &dir) != 0)
2941b8902de1SWarner Losh 			return (EIO);
2942b8902de1SWarner Losh 		dd = (dsl_dir_phys_t *)&dir.dn_bonus;
2943b8902de1SWarner Losh 
2944b8902de1SWarner Losh 		/* Actual loop condition. */
2945b8902de1SWarner Losh 		parent_obj = dd->dd_parent_obj;
2946b8902de1SWarner Losh 		if (parent_obj == 0)
2947b8902de1SWarner Losh 			break;
2948b8902de1SWarner Losh 
29492e6bb655SToomas Soome 		if (objset_get_dnode(spa, &spa->spa_mos, parent_obj,
29502e6bb655SToomas Soome 		    &parent) != 0)
2951b8902de1SWarner Losh 			return (EIO);
2952b8902de1SWarner Losh 		dd = (dsl_dir_phys_t *)&parent.dn_bonus;
2953b8902de1SWarner Losh 		child_dir_zapobj = dd->dd_child_dir_zapobj;
29542e6bb655SToomas Soome 		if (objset_get_dnode(spa, &spa->spa_mos, child_dir_zapobj,
29552e6bb655SToomas Soome 		    &child_dir_zap) != 0)
2956b8902de1SWarner Losh 			return (EIO);
2957b8902de1SWarner Losh 		if (zap_rlookup(spa, &child_dir_zap, component, dir_obj) != 0)
2958b8902de1SWarner Losh 			return (EIO);
2959b8902de1SWarner Losh 
2960b8902de1SWarner Losh 		len = strlen(component);
2961b8902de1SWarner Losh 		p -= len;
2962b8902de1SWarner Losh 		memcpy(p, component, len);
2963b8902de1SWarner Losh 		--p;
2964b8902de1SWarner Losh 		*p = '/';
2965b8902de1SWarner Losh 
2966b8902de1SWarner Losh 		/* Actual loop iteration. */
2967b8902de1SWarner Losh 		dir_obj = parent_obj;
2968b8902de1SWarner Losh 	}
2969b8902de1SWarner Losh 
2970b8902de1SWarner Losh 	if (*p != '\0')
2971b8902de1SWarner Losh 		++p;
2972b8902de1SWarner Losh 	strcpy(result, p);
2973b8902de1SWarner Losh 
2974b8902de1SWarner Losh 	return (0);
2975b8902de1SWarner Losh }
2976b8902de1SWarner Losh 
2977b8902de1SWarner Losh static int
2978b8902de1SWarner Losh zfs_lookup_dataset(const spa_t *spa, const char *name, uint64_t *objnum)
2979b8902de1SWarner Losh {
2980b8902de1SWarner Losh 	char element[256];
2981b8902de1SWarner Losh 	uint64_t dir_obj, child_dir_zapobj;
2982b8902de1SWarner Losh 	dnode_phys_t child_dir_zap, dir;
2983b8902de1SWarner Losh 	dsl_dir_phys_t *dd;
2984b8902de1SWarner Losh 	const char *p, *q;
2985b8902de1SWarner Losh 
29862e6bb655SToomas Soome 	if (objset_get_dnode(spa, &spa->spa_mos,
29872e6bb655SToomas Soome 	    DMU_POOL_DIRECTORY_OBJECT, &dir))
2988b8902de1SWarner Losh 		return (EIO);
2989b8902de1SWarner Losh 	if (zap_lookup(spa, &dir, DMU_POOL_ROOT_DATASET, sizeof (dir_obj),
2990b8902de1SWarner Losh 	    1, &dir_obj))
2991b8902de1SWarner Losh 		return (EIO);
2992b8902de1SWarner Losh 
2993b8902de1SWarner Losh 	p = name;
2994b8902de1SWarner Losh 	for (;;) {
2995b8902de1SWarner Losh 		if (objset_get_dnode(spa, &spa->spa_mos, dir_obj, &dir))
2996b8902de1SWarner Losh 			return (EIO);
2997b8902de1SWarner Losh 		dd = (dsl_dir_phys_t *)&dir.dn_bonus;
2998b8902de1SWarner Losh 
2999b8902de1SWarner Losh 		while (*p == '/')
3000b8902de1SWarner Losh 			p++;
3001b8902de1SWarner Losh 		/* Actual loop condition #1. */
3002b8902de1SWarner Losh 		if (*p == '\0')
3003b8902de1SWarner Losh 			break;
3004b8902de1SWarner Losh 
3005b8902de1SWarner Losh 		q = strchr(p, '/');
3006b8902de1SWarner Losh 		if (q) {
3007b8902de1SWarner Losh 			memcpy(element, p, q - p);
3008b8902de1SWarner Losh 			element[q - p] = '\0';
3009b8902de1SWarner Losh 			p = q + 1;
3010b8902de1SWarner Losh 		} else {
3011b8902de1SWarner Losh 			strcpy(element, p);
3012b8902de1SWarner Losh 			p += strlen(p);
3013b8902de1SWarner Losh 		}
3014b8902de1SWarner Losh 
3015b8902de1SWarner Losh 		child_dir_zapobj = dd->dd_child_dir_zapobj;
30162e6bb655SToomas Soome 		if (objset_get_dnode(spa, &spa->spa_mos, child_dir_zapobj,
30172e6bb655SToomas Soome 		    &child_dir_zap) != 0)
3018b8902de1SWarner Losh 			return (EIO);
3019b8902de1SWarner Losh 
3020b8902de1SWarner Losh 		/* Actual loop condition #2. */
3021b8902de1SWarner Losh 		if (zap_lookup(spa, &child_dir_zap, element, sizeof (dir_obj),
3022b8902de1SWarner Losh 		    1, &dir_obj) != 0)
3023b8902de1SWarner Losh 			return (ENOENT);
3024b8902de1SWarner Losh 	}
3025b8902de1SWarner Losh 
3026b8902de1SWarner Losh 	*objnum = dd->dd_head_dataset_obj;
3027b8902de1SWarner Losh 	return (0);
3028b8902de1SWarner Losh }
3029b8902de1SWarner Losh 
3030b8902de1SWarner Losh #ifndef BOOT2
3031b8902de1SWarner Losh static int
3032b8902de1SWarner Losh zfs_list_dataset(const spa_t *spa, uint64_t objnum/*, int pos, char *entry*/)
3033b8902de1SWarner Losh {
3034b8902de1SWarner Losh 	uint64_t dir_obj, child_dir_zapobj;
3035b8902de1SWarner Losh 	dnode_phys_t child_dir_zap, dir, dataset;
3036b8902de1SWarner Losh 	dsl_dataset_phys_t *ds;
3037b8902de1SWarner Losh 	dsl_dir_phys_t *dd;
3038b8902de1SWarner Losh 
3039b8902de1SWarner Losh 	if (objset_get_dnode(spa, &spa->spa_mos, objnum, &dataset)) {
3040b8902de1SWarner Losh 		printf("ZFS: can't find dataset %ju\n", (uintmax_t)objnum);
3041b8902de1SWarner Losh 		return (EIO);
3042b8902de1SWarner Losh 	}
3043b8902de1SWarner Losh 	ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3044b8902de1SWarner Losh 	dir_obj = ds->ds_dir_obj;
3045b8902de1SWarner Losh 
3046b8902de1SWarner Losh 	if (objset_get_dnode(spa, &spa->spa_mos, dir_obj, &dir)) {
3047b8902de1SWarner Losh 		printf("ZFS: can't find dirobj %ju\n", (uintmax_t)dir_obj);
3048b8902de1SWarner Losh 		return (EIO);
3049b8902de1SWarner Losh 	}
3050b8902de1SWarner Losh 	dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3051b8902de1SWarner Losh 
3052b8902de1SWarner Losh 	child_dir_zapobj = dd->dd_child_dir_zapobj;
30532e6bb655SToomas Soome 	if (objset_get_dnode(spa, &spa->spa_mos, child_dir_zapobj,
30542e6bb655SToomas Soome 	    &child_dir_zap) != 0) {
3055b8902de1SWarner Losh 		printf("ZFS: can't find child zap %ju\n", (uintmax_t)dir_obj);
3056b8902de1SWarner Losh 		return (EIO);
3057b8902de1SWarner Losh 	}
3058b8902de1SWarner Losh 
3059b8902de1SWarner Losh 	return (zap_list(spa, &child_dir_zap) != 0);
3060b8902de1SWarner Losh }
3061b8902de1SWarner Losh 
3062b8902de1SWarner Losh int
30632e6bb655SToomas Soome zfs_callback_dataset(const spa_t *spa, uint64_t objnum,
30642e6bb655SToomas Soome     int (*callback)(const char *, uint64_t))
3065b8902de1SWarner Losh {
30664d297e70SToomas Soome 	uint64_t dir_obj, child_dir_zapobj;
3067b8902de1SWarner Losh 	dnode_phys_t child_dir_zap, dir, dataset;
3068b8902de1SWarner Losh 	dsl_dataset_phys_t *ds;
3069b8902de1SWarner Losh 	dsl_dir_phys_t *dd;
30704d297e70SToomas Soome 	zap_phys_t *zap;
30714d297e70SToomas Soome 	size_t size;
3072b8902de1SWarner Losh 	int err;
3073b8902de1SWarner Losh 
3074b8902de1SWarner Losh 	err = objset_get_dnode(spa, &spa->spa_mos, objnum, &dataset);
3075b8902de1SWarner Losh 	if (err != 0) {
3076b8902de1SWarner Losh 		printf("ZFS: can't find dataset %ju\n", (uintmax_t)objnum);
3077b8902de1SWarner Losh 		return (err);
3078b8902de1SWarner Losh 	}
3079b8902de1SWarner Losh 	ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3080b8902de1SWarner Losh 	dir_obj = ds->ds_dir_obj;
3081b8902de1SWarner Losh 
3082b8902de1SWarner Losh 	err = objset_get_dnode(spa, &spa->spa_mos, dir_obj, &dir);
3083b8902de1SWarner Losh 	if (err != 0) {
3084b8902de1SWarner Losh 		printf("ZFS: can't find dirobj %ju\n", (uintmax_t)dir_obj);
3085b8902de1SWarner Losh 		return (err);
3086b8902de1SWarner Losh 	}
3087b8902de1SWarner Losh 	dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3088b8902de1SWarner Losh 
3089b8902de1SWarner Losh 	child_dir_zapobj = dd->dd_child_dir_zapobj;
30902e6bb655SToomas Soome 	err = objset_get_dnode(spa, &spa->spa_mos, child_dir_zapobj,
30912e6bb655SToomas Soome 	    &child_dir_zap);
3092b8902de1SWarner Losh 	if (err != 0) {
3093b8902de1SWarner Losh 		printf("ZFS: can't find child zap %ju\n", (uintmax_t)dir_obj);
3094b8902de1SWarner Losh 		return (err);
3095b8902de1SWarner Losh 	}
3096b8902de1SWarner Losh 
3097380977d5SToomas Soome 	size = child_dir_zap.dn_datablkszsec << SPA_MINBLOCKSHIFT;
30984d297e70SToomas Soome 	zap = malloc(size);
30994d297e70SToomas Soome 	if (zap != NULL) {
31004d297e70SToomas Soome 		err = dnode_read(spa, &child_dir_zap, 0, zap, size);
3101b8902de1SWarner Losh 		if (err != 0)
31024d297e70SToomas Soome 			goto done;
3103b8902de1SWarner Losh 
31044d297e70SToomas Soome 		if (zap->zap_block_type == ZBT_MICRO)
31054d297e70SToomas Soome 			err = mzap_list((const mzap_phys_t *)zap, size,
31064d297e70SToomas Soome 			    callback);
3107b8902de1SWarner Losh 		else
31084d297e70SToomas Soome 			err = fzap_list(spa, &child_dir_zap, zap, callback);
3109380977d5SToomas Soome 	} else {
3110380977d5SToomas Soome 		err = ENOMEM;
31114d297e70SToomas Soome 	}
31124d297e70SToomas Soome done:
31134d297e70SToomas Soome 	free(zap);
31144d297e70SToomas Soome 	return (err);
3115b8902de1SWarner Losh }
3116b8902de1SWarner Losh #endif
3117b8902de1SWarner Losh 
3118b8902de1SWarner Losh /*
3119b8902de1SWarner Losh  * Find the object set given the object number of its dataset object
3120b8902de1SWarner Losh  * and return its details in *objset
3121b8902de1SWarner Losh  */
3122b8902de1SWarner Losh static int
3123b8902de1SWarner Losh zfs_mount_dataset(const spa_t *spa, uint64_t objnum, objset_phys_t *objset)
3124b8902de1SWarner Losh {
3125b8902de1SWarner Losh 	dnode_phys_t dataset;
3126b8902de1SWarner Losh 	dsl_dataset_phys_t *ds;
3127b8902de1SWarner Losh 
3128b8902de1SWarner Losh 	if (objset_get_dnode(spa, &spa->spa_mos, objnum, &dataset)) {
3129b8902de1SWarner Losh 		printf("ZFS: can't find dataset %ju\n", (uintmax_t)objnum);
3130b8902de1SWarner Losh 		return (EIO);
3131b8902de1SWarner Losh 	}
3132b8902de1SWarner Losh 
3133b8902de1SWarner Losh 	ds = (dsl_dataset_phys_t *)&dataset.dn_bonus;
3134b8902de1SWarner Losh 	if (zio_read(spa, &ds->ds_bp, objset)) {
3135b8902de1SWarner Losh 		printf("ZFS: can't read object set for dataset %ju\n",
3136b8902de1SWarner Losh 		    (uintmax_t)objnum);
3137b8902de1SWarner Losh 		return (EIO);
3138b8902de1SWarner Losh 	}
3139b8902de1SWarner Losh 
3140b8902de1SWarner Losh 	return (0);
3141b8902de1SWarner Losh }
3142b8902de1SWarner Losh 
3143b8902de1SWarner Losh /*
3144b8902de1SWarner Losh  * Find the object set pointed to by the BOOTFS property or the root
3145b8902de1SWarner Losh  * dataset if there is none and return its details in *objset
3146b8902de1SWarner Losh  */
3147b8902de1SWarner Losh static int
3148b8902de1SWarner Losh zfs_get_root(const spa_t *spa, uint64_t *objid)
3149b8902de1SWarner Losh {
3150b8902de1SWarner Losh 	dnode_phys_t dir, propdir;
3151b8902de1SWarner Losh 	uint64_t props, bootfs, root;
3152b8902de1SWarner Losh 
3153b8902de1SWarner Losh 	*objid = 0;
3154b8902de1SWarner Losh 
3155b8902de1SWarner Losh 	/*
3156b8902de1SWarner Losh 	 * Start with the MOS directory object.
3157b8902de1SWarner Losh 	 */
31582e6bb655SToomas Soome 	if (objset_get_dnode(spa, &spa->spa_mos,
31592e6bb655SToomas Soome 	    DMU_POOL_DIRECTORY_OBJECT, &dir)) {
3160b8902de1SWarner Losh 		printf("ZFS: can't read MOS object directory\n");
3161b8902de1SWarner Losh 		return (EIO);
3162b8902de1SWarner Losh 	}
3163b8902de1SWarner Losh 
3164b8902de1SWarner Losh 	/*
3165b8902de1SWarner Losh 	 * Lookup the pool_props and see if we can find a bootfs.
3166b8902de1SWarner Losh 	 */
31672e6bb655SToomas Soome 	if (zap_lookup(spa, &dir, DMU_POOL_PROPS,
31682e6bb655SToomas Soome 	    sizeof(props), 1, &props) == 0 &&
31692e6bb655SToomas Soome 	    objset_get_dnode(spa, &spa->spa_mos, props, &propdir) == 0 &&
31702e6bb655SToomas Soome 	    zap_lookup(spa, &propdir, "bootfs",
31713c2db0efSToomas Soome 	    sizeof(bootfs), 1, &bootfs) == 0 && bootfs != 0) {
3172b8902de1SWarner Losh 		*objid = bootfs;
3173b8902de1SWarner Losh 		return (0);
3174b8902de1SWarner Losh 	}
3175b8902de1SWarner Losh 	/*
3176b8902de1SWarner Losh 	 * Lookup the root dataset directory
3177b8902de1SWarner Losh 	 */
31782e6bb655SToomas Soome 	if (zap_lookup(spa, &dir, DMU_POOL_ROOT_DATASET,
31792e6bb655SToomas Soome 	    sizeof(root), 1, &root) ||
31802e6bb655SToomas Soome 	    objset_get_dnode(spa, &spa->spa_mos, root, &dir)) {
3181b8902de1SWarner Losh 		printf("ZFS: can't find root dsl_dir\n");
3182b8902de1SWarner Losh 		return (EIO);
3183b8902de1SWarner Losh 	}
3184b8902de1SWarner Losh 
3185b8902de1SWarner Losh 	/*
3186b8902de1SWarner Losh 	 * Use the information from the dataset directory's bonus buffer
3187b8902de1SWarner Losh 	 * to find the dataset object and from that the object set itself.
3188b8902de1SWarner Losh 	 */
3189b8902de1SWarner Losh 	dsl_dir_phys_t *dd = (dsl_dir_phys_t *)&dir.dn_bonus;
3190b8902de1SWarner Losh 	*objid = dd->dd_head_dataset_obj;
3191b8902de1SWarner Losh 	return (0);
3192b8902de1SWarner Losh }
3193b8902de1SWarner Losh 
3194b8902de1SWarner Losh static int
3195b8902de1SWarner Losh zfs_mount(const spa_t *spa, uint64_t rootobj, struct zfsmount *mount)
3196b8902de1SWarner Losh {
3197b8902de1SWarner Losh 
3198b8902de1SWarner Losh 	mount->spa = spa;
3199b8902de1SWarner Losh 
3200b8902de1SWarner Losh 	/*
3201b8902de1SWarner Losh 	 * Find the root object set if not explicitly provided
3202b8902de1SWarner Losh 	 */
3203b8902de1SWarner Losh 	if (rootobj == 0 && zfs_get_root(spa, &rootobj)) {
3204b8902de1SWarner Losh 		printf("ZFS: can't find root filesystem\n");
3205b8902de1SWarner Losh 		return (EIO);
3206b8902de1SWarner Losh 	}
3207b8902de1SWarner Losh 
3208b8902de1SWarner Losh 	if (zfs_mount_dataset(spa, rootobj, &mount->objset)) {
3209b8902de1SWarner Losh 		printf("ZFS: can't open root filesystem\n");
3210b8902de1SWarner Losh 		return (EIO);
3211b8902de1SWarner Losh 	}
3212b8902de1SWarner Losh 
3213b8902de1SWarner Losh 	mount->rootobj = rootobj;
3214b8902de1SWarner Losh 
3215b8902de1SWarner Losh 	return (0);
3216b8902de1SWarner Losh }
3217b8902de1SWarner Losh 
3218b8902de1SWarner Losh /*
3219b8902de1SWarner Losh  * callback function for feature name checks.
3220b8902de1SWarner Losh  */
3221b8902de1SWarner Losh static int
3222b8902de1SWarner Losh check_feature(const char *name, uint64_t value)
3223b8902de1SWarner Losh {
3224b8902de1SWarner Losh 	int i;
3225b8902de1SWarner Losh 
3226b8902de1SWarner Losh 	if (value == 0)
3227b8902de1SWarner Losh 		return (0);
3228b8902de1SWarner Losh 	if (name[0] == '\0')
3229b8902de1SWarner Losh 		return (0);
3230b8902de1SWarner Losh 
3231b8902de1SWarner Losh 	for (i = 0; features_for_read[i] != NULL; i++) {
3232b8902de1SWarner Losh 		if (strcmp(name, features_for_read[i]) == 0)
3233b8902de1SWarner Losh 			return (0);
3234b8902de1SWarner Losh 	}
3235b8902de1SWarner Losh 	printf("ZFS: unsupported feature: %s\n", name);
3236b8902de1SWarner Losh 	return (EIO);
3237b8902de1SWarner Losh }
3238b8902de1SWarner Losh 
3239b8902de1SWarner Losh /*
3240b8902de1SWarner Losh  * Checks whether the MOS features that are active are supported.
3241b8902de1SWarner Losh  */
3242b8902de1SWarner Losh static int
3243b8902de1SWarner Losh check_mos_features(const spa_t *spa)
3244b8902de1SWarner Losh {
3245b8902de1SWarner Losh 	dnode_phys_t dir;
32464d297e70SToomas Soome 	zap_phys_t *zap;
32474d297e70SToomas Soome 	uint64_t objnum;
3248b8902de1SWarner Losh 	size_t size;
3249b8902de1SWarner Losh 	int rc;
3250b8902de1SWarner Losh 
3251b8902de1SWarner Losh 	if ((rc = objset_get_dnode(spa, &spa->spa_mos, DMU_OT_OBJECT_DIRECTORY,
3252b8902de1SWarner Losh 	    &dir)) != 0)
3253b8902de1SWarner Losh 		return (rc);
3254b8902de1SWarner Losh 	if ((rc = zap_lookup(spa, &dir, DMU_POOL_FEATURES_FOR_READ,
3255b8902de1SWarner Losh 	    sizeof (objnum), 1, &objnum)) != 0) {
3256b8902de1SWarner Losh 		/*
3257b8902de1SWarner Losh 		 * It is older pool without features. As we have already
3258b8902de1SWarner Losh 		 * tested the label, just return without raising the error.
3259b8902de1SWarner Losh 		 */
3260b8902de1SWarner Losh 		return (0);
3261b8902de1SWarner Losh 	}
3262b8902de1SWarner Losh 
3263b8902de1SWarner Losh 	if ((rc = objset_get_dnode(spa, &spa->spa_mos, objnum, &dir)) != 0)
3264b8902de1SWarner Losh 		return (rc);
3265b8902de1SWarner Losh 
3266b8902de1SWarner Losh 	if (dir.dn_type != DMU_OTN_ZAP_METADATA)
3267b8902de1SWarner Losh 		return (EIO);
3268b8902de1SWarner Losh 
3269380977d5SToomas Soome 	size = dir.dn_datablkszsec << SPA_MINBLOCKSHIFT;
32704d297e70SToomas Soome 	zap = malloc(size);
32714d297e70SToomas Soome 	if (zap == NULL)
32724d297e70SToomas Soome 		return (ENOMEM);
32734d297e70SToomas Soome 
32744d297e70SToomas Soome 	if (dnode_read(spa, &dir, 0, zap, size)) {
32754d297e70SToomas Soome 		free(zap);
3276b8902de1SWarner Losh 		return (EIO);
32774d297e70SToomas Soome 	}
3278b8902de1SWarner Losh 
32794d297e70SToomas Soome 	if (zap->zap_block_type == ZBT_MICRO)
32804d297e70SToomas Soome 		rc = mzap_list((const mzap_phys_t *)zap, size, check_feature);
3281b8902de1SWarner Losh 	else
32824d297e70SToomas Soome 		rc = fzap_list(spa, &dir, zap, check_feature);
3283b8902de1SWarner Losh 
32844d297e70SToomas Soome 	free(zap);
3285b8902de1SWarner Losh 	return (rc);
3286b8902de1SWarner Losh }
3287b8902de1SWarner Losh 
3288b8902de1SWarner Losh static int
3289b1b93268SToomas Soome load_nvlist(spa_t *spa, uint64_t obj, unsigned char **value)
3290b1b93268SToomas Soome {
3291b1b93268SToomas Soome 	dnode_phys_t dir;
3292b1b93268SToomas Soome 	size_t size;
3293b1b93268SToomas Soome 	int rc;
3294b1b93268SToomas Soome 	unsigned char *nv;
3295b1b93268SToomas Soome 
3296b1b93268SToomas Soome 	*value = NULL;
3297b1b93268SToomas Soome 	if ((rc = objset_get_dnode(spa, &spa->spa_mos, obj, &dir)) != 0)
3298b1b93268SToomas Soome 		return (rc);
3299b1b93268SToomas Soome 	if (dir.dn_type != DMU_OT_PACKED_NVLIST &&
3300b1b93268SToomas Soome 	    dir.dn_bonustype != DMU_OT_PACKED_NVLIST_SIZE) {
3301b1b93268SToomas Soome 		return (EIO);
3302b1b93268SToomas Soome 	}
3303b1b93268SToomas Soome 
3304b1b93268SToomas Soome 	if (dir.dn_bonuslen != sizeof (uint64_t))
3305b1b93268SToomas Soome 		return (EIO);
3306b1b93268SToomas Soome 
3307b1b93268SToomas Soome 	size = *(uint64_t *)DN_BONUS(&dir);
3308b1b93268SToomas Soome 	nv = malloc(size);
3309b1b93268SToomas Soome 	if (nv == NULL)
3310b1b93268SToomas Soome 		return (ENOMEM);
3311b1b93268SToomas Soome 
3312b1b93268SToomas Soome 	rc = dnode_read(spa, &dir, 0, nv, size);
3313b1b93268SToomas Soome 	if (rc != 0) {
3314b1b93268SToomas Soome 		free(nv);
3315b1b93268SToomas Soome 		nv = NULL;
3316b1b93268SToomas Soome 		return (rc);
3317b1b93268SToomas Soome 	}
3318b1b93268SToomas Soome 	*value = nv;
3319b1b93268SToomas Soome 	return (rc);
3320b1b93268SToomas Soome }
3321b1b93268SToomas Soome 
3322b1b93268SToomas Soome static int
3323b8902de1SWarner Losh zfs_spa_init(spa_t *spa)
3324b8902de1SWarner Losh {
3325b8902de1SWarner Losh 	dnode_phys_t dir;
3326b1b93268SToomas Soome 	uint64_t config_object;
3327b1b93268SToomas Soome 	unsigned char *nvlist;
33283c2db0efSToomas Soome 	int rc;
3329b8902de1SWarner Losh 
3330b8902de1SWarner Losh 	if (zio_read(spa, &spa->spa_uberblock.ub_rootbp, &spa->spa_mos)) {
3331b8902de1SWarner Losh 		printf("ZFS: can't read MOS of pool %s\n", spa->spa_name);
3332b8902de1SWarner Losh 		return (EIO);
3333b8902de1SWarner Losh 	}
3334b8902de1SWarner Losh 	if (spa->spa_mos.os_type != DMU_OST_META) {
3335b8902de1SWarner Losh 		printf("ZFS: corrupted MOS of pool %s\n", spa->spa_name);
3336b8902de1SWarner Losh 		return (EIO);
3337b8902de1SWarner Losh 	}
3338b8902de1SWarner Losh 
3339b8902de1SWarner Losh 	if (objset_get_dnode(spa, &spa->spa_mos, DMU_POOL_DIRECTORY_OBJECT,
3340b8902de1SWarner Losh 	    &dir)) {
3341b8902de1SWarner Losh 		printf("ZFS: failed to read pool %s directory object\n",
3342b8902de1SWarner Losh 		    spa->spa_name);
3343b8902de1SWarner Losh 		return (EIO);
3344b8902de1SWarner Losh 	}
3345b8902de1SWarner Losh 	/* this is allowed to fail, older pools do not have salt */
3346b8902de1SWarner Losh 	rc = zap_lookup(spa, &dir, DMU_POOL_CHECKSUM_SALT, 1,
3347b8902de1SWarner Losh 	    sizeof (spa->spa_cksum_salt.zcs_bytes),
3348b8902de1SWarner Losh 	    spa->spa_cksum_salt.zcs_bytes);
3349b8902de1SWarner Losh 
3350b8902de1SWarner Losh 	rc = check_mos_features(spa);
3351b8902de1SWarner Losh 	if (rc != 0) {
3352b8902de1SWarner Losh 		printf("ZFS: pool %s is not supported\n", spa->spa_name);
3353b1b93268SToomas Soome 		return (rc);
3354b8902de1SWarner Losh 	}
3355b8902de1SWarner Losh 
3356b1b93268SToomas Soome 	rc = zap_lookup(spa, &dir, DMU_POOL_CONFIG,
3357b1b93268SToomas Soome 	    sizeof (config_object), 1, &config_object);
3358b1b93268SToomas Soome 	if (rc != 0) {
3359b1b93268SToomas Soome 		printf("ZFS: can not read MOS %s\n", DMU_POOL_CONFIG);
3360b1b93268SToomas Soome 		return (EIO);
3361b1b93268SToomas Soome 	}
3362b1b93268SToomas Soome 	rc = load_nvlist(spa, config_object, &nvlist);
3363b1b93268SToomas Soome 	if (rc != 0)
3364b1b93268SToomas Soome 		return (rc);
3365b1b93268SToomas Soome 
3366653c3383SToomas Soome 	/*
3367653c3383SToomas Soome 	 * Update vdevs from MOS config. Note, we do skip encoding bytes
3368653c3383SToomas Soome 	 * here. See also vdev_label_read_config().
3369653c3383SToomas Soome 	 */
33703c2db0efSToomas Soome 	rc = vdev_init_from_nvlist(spa, nvlist + 4);
3371b1b93268SToomas Soome 	free(nvlist);
3372b8902de1SWarner Losh 	return (rc);
3373b8902de1SWarner Losh }
3374b8902de1SWarner Losh 
3375b8902de1SWarner Losh static int
3376b8902de1SWarner Losh zfs_dnode_stat(const spa_t *spa, dnode_phys_t *dn, struct stat *sb)
3377b8902de1SWarner Losh {
3378b8902de1SWarner Losh 
3379b8902de1SWarner Losh 	if (dn->dn_bonustype != DMU_OT_SA) {
3380b8902de1SWarner Losh 		znode_phys_t *zp = (znode_phys_t *)dn->dn_bonus;
3381b8902de1SWarner Losh 
3382b8902de1SWarner Losh 		sb->st_mode = zp->zp_mode;
3383b8902de1SWarner Losh 		sb->st_uid = zp->zp_uid;
3384b8902de1SWarner Losh 		sb->st_gid = zp->zp_gid;
3385b8902de1SWarner Losh 		sb->st_size = zp->zp_size;
3386b8902de1SWarner Losh 	} else {
3387b8902de1SWarner Losh 		sa_hdr_phys_t *sahdrp;
3388b8902de1SWarner Losh 		int hdrsize;
3389b8902de1SWarner Losh 		size_t size = 0;
3390b8902de1SWarner Losh 		void *buf = NULL;
3391b8902de1SWarner Losh 
3392b8902de1SWarner Losh 		if (dn->dn_bonuslen != 0)
3393b8902de1SWarner Losh 			sahdrp = (sa_hdr_phys_t *)DN_BONUS(dn);
3394b8902de1SWarner Losh 		else {
3395b8902de1SWarner Losh 			if ((dn->dn_flags & DNODE_FLAG_SPILL_BLKPTR) != 0) {
3396b8902de1SWarner Losh 				blkptr_t *bp = DN_SPILL_BLKPTR(dn);
3397b8902de1SWarner Losh 				int error;
3398b8902de1SWarner Losh 
3399b8902de1SWarner Losh 				size = BP_GET_LSIZE(bp);
3400*c1c4c81fSToomas Soome 				buf = malloc(size);
3401*c1c4c81fSToomas Soome 				if (buf == NULL)
3402*c1c4c81fSToomas Soome 					error = ENOMEM;
3403*c1c4c81fSToomas Soome 				else
3404b8902de1SWarner Losh 					error = zio_read(spa, bp, buf);
3405*c1c4c81fSToomas Soome 
3406b8902de1SWarner Losh 				if (error != 0) {
3407*c1c4c81fSToomas Soome 					free(buf);
3408b8902de1SWarner Losh 					return (error);
3409b8902de1SWarner Losh 				}
3410b8902de1SWarner Losh 				sahdrp = buf;
3411b8902de1SWarner Losh 			} else {
3412b8902de1SWarner Losh 				return (EIO);
3413b8902de1SWarner Losh 			}
3414b8902de1SWarner Losh 		}
3415b8902de1SWarner Losh 		hdrsize = SA_HDR_SIZE(sahdrp);
3416b8902de1SWarner Losh 		sb->st_mode = *(uint64_t *)((char *)sahdrp + hdrsize +
3417b8902de1SWarner Losh 		    SA_MODE_OFFSET);
3418b8902de1SWarner Losh 		sb->st_uid = *(uint64_t *)((char *)sahdrp + hdrsize +
3419b8902de1SWarner Losh 		    SA_UID_OFFSET);
3420b8902de1SWarner Losh 		sb->st_gid = *(uint64_t *)((char *)sahdrp + hdrsize +
3421b8902de1SWarner Losh 		    SA_GID_OFFSET);
3422b8902de1SWarner Losh 		sb->st_size = *(uint64_t *)((char *)sahdrp + hdrsize +
3423b8902de1SWarner Losh 		    SA_SIZE_OFFSET);
3424*c1c4c81fSToomas Soome 		free(buf);
3425b8902de1SWarner Losh 	}
3426b8902de1SWarner Losh 
3427b8902de1SWarner Losh 	return (0);
3428b8902de1SWarner Losh }
3429b8902de1SWarner Losh 
3430b8902de1SWarner Losh static int
3431b8902de1SWarner Losh zfs_dnode_readlink(const spa_t *spa, dnode_phys_t *dn, char *path, size_t psize)
3432b8902de1SWarner Losh {
3433b8902de1SWarner Losh 	int rc = 0;
3434b8902de1SWarner Losh 
3435b8902de1SWarner Losh 	if (dn->dn_bonustype == DMU_OT_SA) {
3436b8902de1SWarner Losh 		sa_hdr_phys_t *sahdrp = NULL;
3437b8902de1SWarner Losh 		size_t size = 0;
3438b8902de1SWarner Losh 		void *buf = NULL;
3439b8902de1SWarner Losh 		int hdrsize;
3440b8902de1SWarner Losh 		char *p;
3441b8902de1SWarner Losh 
3442*c1c4c81fSToomas Soome 		if (dn->dn_bonuslen != 0) {
3443b8902de1SWarner Losh 			sahdrp = (sa_hdr_phys_t *)DN_BONUS(dn);
3444*c1c4c81fSToomas Soome 		} else {
3445b8902de1SWarner Losh 			blkptr_t *bp;
3446b8902de1SWarner Losh 
3447b8902de1SWarner Losh 			if ((dn->dn_flags & DNODE_FLAG_SPILL_BLKPTR) == 0)
3448b8902de1SWarner Losh 				return (EIO);
3449b8902de1SWarner Losh 			bp = DN_SPILL_BLKPTR(dn);
3450b8902de1SWarner Losh 
3451b8902de1SWarner Losh 			size = BP_GET_LSIZE(bp);
3452*c1c4c81fSToomas Soome 			buf = malloc(size);
3453*c1c4c81fSToomas Soome 			if (buf == NULL)
3454*c1c4c81fSToomas Soome 				rc = ENOMEM;
3455*c1c4c81fSToomas Soome 			else
3456b8902de1SWarner Losh 				rc = zio_read(spa, bp, buf);
3457b8902de1SWarner Losh 			if (rc != 0) {
3458*c1c4c81fSToomas Soome 				free(buf);
3459b8902de1SWarner Losh 				return (rc);
3460b8902de1SWarner Losh 			}
3461b8902de1SWarner Losh 			sahdrp = buf;
3462b8902de1SWarner Losh 		}
3463b8902de1SWarner Losh 		hdrsize = SA_HDR_SIZE(sahdrp);
3464b8902de1SWarner Losh 		p = (char *)((uintptr_t)sahdrp + hdrsize + SA_SYMLINK_OFFSET);
3465b8902de1SWarner Losh 		memcpy(path, p, psize);
3466*c1c4c81fSToomas Soome 		free(buf);
3467b8902de1SWarner Losh 		return (0);
3468b8902de1SWarner Losh 	}
3469b8902de1SWarner Losh 	/*
3470b8902de1SWarner Losh 	 * Second test is purely to silence bogus compiler
3471b8902de1SWarner Losh 	 * warning about accessing past the end of dn_bonus.
3472b8902de1SWarner Losh 	 */
3473b8902de1SWarner Losh 	if (psize + sizeof(znode_phys_t) <= dn->dn_bonuslen &&
3474b8902de1SWarner Losh 	    sizeof(znode_phys_t) <= sizeof(dn->dn_bonus)) {
3475b8902de1SWarner Losh 		memcpy(path, &dn->dn_bonus[sizeof(znode_phys_t)], psize);
3476b8902de1SWarner Losh 	} else {
3477b8902de1SWarner Losh 		rc = dnode_read(spa, dn, 0, path, psize);
3478b8902de1SWarner Losh 	}
3479b8902de1SWarner Losh 	return (rc);
3480b8902de1SWarner Losh }
3481b8902de1SWarner Losh 
3482b8902de1SWarner Losh struct obj_list {
3483b8902de1SWarner Losh 	uint64_t		objnum;
3484b8902de1SWarner Losh 	STAILQ_ENTRY(obj_list)	entry;
3485b8902de1SWarner Losh };
3486b8902de1SWarner Losh 
3487b8902de1SWarner Losh /*
3488b8902de1SWarner Losh  * Lookup a file and return its dnode.
3489b8902de1SWarner Losh  */
3490b8902de1SWarner Losh static int
3491b8902de1SWarner Losh zfs_lookup(const struct zfsmount *mount, const char *upath, dnode_phys_t *dnode)
3492b8902de1SWarner Losh {
3493b8902de1SWarner Losh 	int rc;
3494b8902de1SWarner Losh 	uint64_t objnum;
3495b8902de1SWarner Losh 	const spa_t *spa;
3496b8902de1SWarner Losh 	dnode_phys_t dn;
3497b8902de1SWarner Losh 	const char *p, *q;
3498b8902de1SWarner Losh 	char element[256];
3499b8902de1SWarner Losh 	char path[1024];
3500b8902de1SWarner Losh 	int symlinks_followed = 0;
3501b8902de1SWarner Losh 	struct stat sb;
3502b8902de1SWarner Losh 	struct obj_list *entry, *tentry;
3503b8902de1SWarner Losh 	STAILQ_HEAD(, obj_list) on_cache = STAILQ_HEAD_INITIALIZER(on_cache);
3504b8902de1SWarner Losh 
3505b8902de1SWarner Losh 	spa = mount->spa;
3506b8902de1SWarner Losh 	if (mount->objset.os_type != DMU_OST_ZFS) {
3507b8902de1SWarner Losh 		printf("ZFS: unexpected object set type %ju\n",
3508b8902de1SWarner Losh 		    (uintmax_t)mount->objset.os_type);
3509b8902de1SWarner Losh 		return (EIO);
3510b8902de1SWarner Losh 	}
3511b8902de1SWarner Losh 
3512b8902de1SWarner Losh 	if ((entry = malloc(sizeof(struct obj_list))) == NULL)
3513b8902de1SWarner Losh 		return (ENOMEM);
3514b8902de1SWarner Losh 
3515b8902de1SWarner Losh 	/*
3516b8902de1SWarner Losh 	 * Get the root directory dnode.
3517b8902de1SWarner Losh 	 */
3518b8902de1SWarner Losh 	rc = objset_get_dnode(spa, &mount->objset, MASTER_NODE_OBJ, &dn);
3519b8902de1SWarner Losh 	if (rc) {
3520b8902de1SWarner Losh 		free(entry);
3521b8902de1SWarner Losh 		return (rc);
3522b8902de1SWarner Losh 	}
3523b8902de1SWarner Losh 
3524b8902de1SWarner Losh 	rc = zap_lookup(spa, &dn, ZFS_ROOT_OBJ, sizeof(objnum), 1, &objnum);
3525b8902de1SWarner Losh 	if (rc) {
3526b8902de1SWarner Losh 		free(entry);
3527b8902de1SWarner Losh 		return (rc);
3528b8902de1SWarner Losh 	}
3529b8902de1SWarner Losh 	entry->objnum = objnum;
3530b8902de1SWarner Losh 	STAILQ_INSERT_HEAD(&on_cache, entry, entry);
3531b8902de1SWarner Losh 
3532b8902de1SWarner Losh 	rc = objset_get_dnode(spa, &mount->objset, objnum, &dn);
3533b8902de1SWarner Losh 	if (rc != 0)
3534b8902de1SWarner Losh 		goto done;
3535b8902de1SWarner Losh 
3536b8902de1SWarner Losh 	p = upath;
3537b8902de1SWarner Losh 	while (p && *p) {
3538b8902de1SWarner Losh 		rc = objset_get_dnode(spa, &mount->objset, objnum, &dn);
3539b8902de1SWarner Losh 		if (rc != 0)
3540b8902de1SWarner Losh 			goto done;
3541b8902de1SWarner Losh 
3542b8902de1SWarner Losh 		while (*p == '/')
3543b8902de1SWarner Losh 			p++;
3544b8902de1SWarner Losh 		if (*p == '\0')
3545b8902de1SWarner Losh 			break;
3546b8902de1SWarner Losh 		q = p;
3547b8902de1SWarner Losh 		while (*q != '\0' && *q != '/')
3548b8902de1SWarner Losh 			q++;
3549b8902de1SWarner Losh 
3550b8902de1SWarner Losh 		/* skip dot */
3551b8902de1SWarner Losh 		if (p + 1 == q && p[0] == '.') {
3552b8902de1SWarner Losh 			p++;
3553b8902de1SWarner Losh 			continue;
3554b8902de1SWarner Losh 		}
3555b8902de1SWarner Losh 		/* double dot */
3556b8902de1SWarner Losh 		if (p + 2 == q && p[0] == '.' && p[1] == '.') {
3557b8902de1SWarner Losh 			p += 2;
3558b8902de1SWarner Losh 			if (STAILQ_FIRST(&on_cache) ==
3559b8902de1SWarner Losh 			    STAILQ_LAST(&on_cache, obj_list, entry)) {
3560b8902de1SWarner Losh 				rc = ENOENT;
3561b8902de1SWarner Losh 				goto done;
3562b8902de1SWarner Losh 			}
3563b8902de1SWarner Losh 			entry = STAILQ_FIRST(&on_cache);
3564b8902de1SWarner Losh 			STAILQ_REMOVE_HEAD(&on_cache, entry);
3565b8902de1SWarner Losh 			free(entry);
3566b8902de1SWarner Losh 			objnum = (STAILQ_FIRST(&on_cache))->objnum;
3567b8902de1SWarner Losh 			continue;
3568b8902de1SWarner Losh 		}
3569b8902de1SWarner Losh 		if (q - p + 1 > sizeof(element)) {
3570b8902de1SWarner Losh 			rc = ENAMETOOLONG;
3571b8902de1SWarner Losh 			goto done;
3572b8902de1SWarner Losh 		}
3573b8902de1SWarner Losh 		memcpy(element, p, q - p);
3574b8902de1SWarner Losh 		element[q - p] = 0;
3575b8902de1SWarner Losh 		p = q;
3576b8902de1SWarner Losh 
3577b8902de1SWarner Losh 		if ((rc = zfs_dnode_stat(spa, &dn, &sb)) != 0)
3578b8902de1SWarner Losh 			goto done;
3579b8902de1SWarner Losh 		if (!S_ISDIR(sb.st_mode)) {
3580b8902de1SWarner Losh 			rc = ENOTDIR;
3581b8902de1SWarner Losh 			goto done;
3582b8902de1SWarner Losh 		}
3583b8902de1SWarner Losh 
3584b8902de1SWarner Losh 		rc = zap_lookup(spa, &dn, element, sizeof (objnum), 1, &objnum);
3585b8902de1SWarner Losh 		if (rc)
3586b8902de1SWarner Losh 			goto done;
3587b8902de1SWarner Losh 		objnum = ZFS_DIRENT_OBJ(objnum);
3588b8902de1SWarner Losh 
3589b8902de1SWarner Losh 		if ((entry = malloc(sizeof(struct obj_list))) == NULL) {
3590b8902de1SWarner Losh 			rc = ENOMEM;
3591b8902de1SWarner Losh 			goto done;
3592b8902de1SWarner Losh 		}
3593b8902de1SWarner Losh 		entry->objnum = objnum;
3594b8902de1SWarner Losh 		STAILQ_INSERT_HEAD(&on_cache, entry, entry);
3595b8902de1SWarner Losh 		rc = objset_get_dnode(spa, &mount->objset, objnum, &dn);
3596b8902de1SWarner Losh 		if (rc)
3597b8902de1SWarner Losh 			goto done;
3598b8902de1SWarner Losh 
3599b8902de1SWarner Losh 		/*
3600b8902de1SWarner Losh 		 * Check for symlink.
3601b8902de1SWarner Losh 		 */
3602b8902de1SWarner Losh 		rc = zfs_dnode_stat(spa, &dn, &sb);
3603b8902de1SWarner Losh 		if (rc)
3604b8902de1SWarner Losh 			goto done;
3605b8902de1SWarner Losh 		if (S_ISLNK(sb.st_mode)) {
3606b8902de1SWarner Losh 			if (symlinks_followed > 10) {
3607b8902de1SWarner Losh 				rc = EMLINK;
3608b8902de1SWarner Losh 				goto done;
3609b8902de1SWarner Losh 			}
3610b8902de1SWarner Losh 			symlinks_followed++;
3611b8902de1SWarner Losh 
3612b8902de1SWarner Losh 			/*
3613b8902de1SWarner Losh 			 * Read the link value and copy the tail of our
3614b8902de1SWarner Losh 			 * current path onto the end.
3615b8902de1SWarner Losh 			 */
3616b8902de1SWarner Losh 			if (sb.st_size + strlen(p) + 1 > sizeof(path)) {
3617b8902de1SWarner Losh 				rc = ENAMETOOLONG;
3618b8902de1SWarner Losh 				goto done;
3619b8902de1SWarner Losh 			}
3620b8902de1SWarner Losh 			strcpy(&path[sb.st_size], p);
3621b8902de1SWarner Losh 
3622b8902de1SWarner Losh 			rc = zfs_dnode_readlink(spa, &dn, path, sb.st_size);
3623b8902de1SWarner Losh 			if (rc != 0)
3624b8902de1SWarner Losh 				goto done;
3625b8902de1SWarner Losh 
3626b8902de1SWarner Losh 			/*
3627b8902de1SWarner Losh 			 * Restart with the new path, starting either at
3628b8902de1SWarner Losh 			 * the root or at the parent depending whether or
3629b8902de1SWarner Losh 			 * not the link is relative.
3630b8902de1SWarner Losh 			 */
3631b8902de1SWarner Losh 			p = path;
3632b8902de1SWarner Losh 			if (*p == '/') {
3633b8902de1SWarner Losh 				while (STAILQ_FIRST(&on_cache) !=
3634b8902de1SWarner Losh 				    STAILQ_LAST(&on_cache, obj_list, entry)) {
3635b8902de1SWarner Losh 					entry = STAILQ_FIRST(&on_cache);
3636b8902de1SWarner Losh 					STAILQ_REMOVE_HEAD(&on_cache, entry);
3637b8902de1SWarner Losh 					free(entry);
3638b8902de1SWarner Losh 				}
3639b8902de1SWarner Losh 			} else {
3640b8902de1SWarner Losh 				entry = STAILQ_FIRST(&on_cache);
3641b8902de1SWarner Losh 				STAILQ_REMOVE_HEAD(&on_cache, entry);
3642b8902de1SWarner Losh 				free(entry);
3643b8902de1SWarner Losh 			}
3644b8902de1SWarner Losh 			objnum = (STAILQ_FIRST(&on_cache))->objnum;
3645b8902de1SWarner Losh 		}
3646b8902de1SWarner Losh 	}
3647b8902de1SWarner Losh 
3648b8902de1SWarner Losh 	*dnode = dn;
3649b8902de1SWarner Losh done:
3650b8902de1SWarner Losh 	STAILQ_FOREACH_SAFE(entry, &on_cache, entry, tentry)
3651b8902de1SWarner Losh 		free(entry);
3652b8902de1SWarner Losh 	return (rc);
3653b8902de1SWarner Losh }
3654