xref: /titanic_52/usr/src/lib/libzfs/common/libzfs_config.c (revision ba4e3c84e6b9390bbf7df80b5f1d11dec34cc525)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*
22  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 #pragma ident	"%Z%%M%	%I%	%E% SMI"
27 
28 /*
29  * The pool configuration repository is stored in /etc/zfs/zpool.cache as a
30  * single packed nvlist.  While it would be nice to just read in this
31  * file from userland, this wouldn't work from a local zone.  So we have to have
32  * a zpool ioctl to return the complete configuration for all pools.  In the
33  * global zone, this will be identical to reading the file and unpacking it in
34  * userland.
35  */
36 
37 #include <errno.h>
38 #include <sys/stat.h>
39 #include <fcntl.h>
40 #include <stddef.h>
41 #include <string.h>
42 #include <unistd.h>
43 #include <libintl.h>
44 #include <libuutil.h>
45 
46 #include "libzfs_impl.h"
47 
48 typedef struct config_node {
49 	char		*cn_name;
50 	nvlist_t	*cn_config;
51 	uu_avl_node_t	cn_avl;
52 } config_node_t;
53 
54 /* ARGSUSED */
55 static int
56 config_node_compare(const void *a, const void *b, void *unused)
57 {
58 	int ret;
59 
60 	const config_node_t *ca = (config_node_t *)a;
61 	const config_node_t *cb = (config_node_t *)b;
62 
63 	ret = strcmp(ca->cn_name, cb->cn_name);
64 
65 	if (ret < 0)
66 		return (-1);
67 	else if (ret > 0)
68 		return (1);
69 	else
70 		return (0);
71 }
72 
73 void
74 namespace_clear(libzfs_handle_t *hdl)
75 {
76 	if (hdl->libzfs_ns_avl) {
77 		uu_avl_walk_t *walk;
78 		config_node_t *cn;
79 
80 		if ((walk = uu_avl_walk_start(hdl->libzfs_ns_avl,
81 		    UU_WALK_ROBUST)) == NULL)
82 			return;
83 
84 		while ((cn = uu_avl_walk_next(walk)) != NULL) {
85 			uu_avl_remove(hdl->libzfs_ns_avl, cn);
86 			nvlist_free(cn->cn_config);
87 			free(cn->cn_name);
88 			free(cn);
89 		}
90 
91 		uu_avl_walk_end(walk);
92 
93 		uu_avl_destroy(hdl->libzfs_ns_avl);
94 		hdl->libzfs_ns_avl = NULL;
95 	}
96 
97 	if (hdl->libzfs_ns_avlpool) {
98 		uu_avl_pool_destroy(hdl->libzfs_ns_avlpool);
99 		hdl->libzfs_ns_avlpool = NULL;
100 	}
101 }
102 
103 /*
104  * Loads the pool namespace, or re-loads it if the cache has changed.
105  */
106 static int
107 namespace_reload(libzfs_handle_t *hdl)
108 {
109 	nvlist_t *config;
110 	config_node_t *cn;
111 	nvpair_t *elem;
112 	zfs_cmd_t zc = { 0 };
113 	uu_avl_walk_t *walk;
114 
115 	if (hdl->libzfs_ns_gen == 0) {
116 		/*
117 		 * This is the first time we've accessed the configuration
118 		 * cache.  Initialize the AVL tree and then fall through to the
119 		 * common code.
120 		 */
121 		if ((hdl->libzfs_ns_avlpool = uu_avl_pool_create("config_pool",
122 		    sizeof (config_node_t),
123 		    offsetof(config_node_t, cn_avl),
124 		    config_node_compare, UU_DEFAULT)) == NULL)
125 			return (no_memory(hdl));
126 
127 		if ((hdl->libzfs_ns_avl = uu_avl_create(hdl->libzfs_ns_avlpool,
128 		    NULL, UU_DEFAULT)) == NULL)
129 			return (no_memory(hdl));
130 	}
131 
132 	/*
133 	 * Issue the ZFS_IOC_POOL_CONFIGS ioctl.
134 	 * This can fail for one of two reasons:
135 	 *
136 	 * 	EEXIST		The generation counts match, nothing to do.
137 	 * 	ENOMEM		The zc_config_dst buffer isn't large enough to
138 	 * 			hold the config; zc_config_dst_size will have
139 	 *			been modified to tell us how much to allocate.
140 	 */
141 	zc.zc_config_dst_size = 1024;
142 	if ((zc.zc_config_dst = (uint64_t)(uintptr_t)
143 	    zfs_alloc(hdl, zc.zc_config_dst_size)) == NULL)
144 		return (-1);
145 	for (;;) {
146 		zc.zc_cookie = hdl->libzfs_ns_gen;
147 		if (ioctl(hdl->libzfs_fd, ZFS_IOC_POOL_CONFIGS, &zc) != 0) {
148 			switch (errno) {
149 			case EEXIST:
150 				/*
151 				 * The namespace hasn't changed.
152 				 */
153 				free((void *)(uintptr_t)zc.zc_config_dst);
154 				return (0);
155 
156 			case ENOMEM:
157 				free((void *)(uintptr_t)zc.zc_config_dst);
158 				if ((zc.zc_config_dst = (uint64_t)(uintptr_t)
159 				    zfs_alloc(hdl, zc.zc_config_dst_size))
160 				    == NULL)
161 					return (-1);
162 				break;
163 
164 			default:
165 				return (zfs_standard_error(hdl, errno,
166 				    dgettext(TEXT_DOMAIN, "failed to read "
167 				    "pool configuration")));
168 			}
169 		} else {
170 			hdl->libzfs_ns_gen = zc.zc_cookie;
171 			break;
172 		}
173 	}
174 
175 	if (nvlist_unpack((void *)(uintptr_t)zc.zc_config_dst,
176 	    zc.zc_config_dst_size, &config, 0) != 0) {
177 		free((void *)(uintptr_t)zc.zc_config_dst);
178 		return (no_memory(hdl));
179 	}
180 
181 	free((void *)(uintptr_t)zc.zc_config_dst);
182 
183 	/*
184 	 * Clear out any existing configuration information.
185 	 */
186 	if ((walk = uu_avl_walk_start(hdl->libzfs_ns_avl,
187 	    UU_WALK_ROBUST)) == NULL) {
188 		nvlist_free(config);
189 		return (no_memory(hdl));
190 	}
191 
192 	while ((cn = uu_avl_walk_next(walk)) != NULL) {
193 		uu_avl_remove(hdl->libzfs_ns_avl, cn);
194 		nvlist_free(cn->cn_config);
195 		free(cn->cn_name);
196 		free(cn);
197 	}
198 
199 	uu_avl_walk_end(walk);
200 
201 	elem = NULL;
202 	while ((elem = nvlist_next_nvpair(config, elem)) != NULL) {
203 		nvlist_t *child;
204 		uu_avl_index_t where;
205 
206 		if ((cn = zfs_alloc(hdl, sizeof (config_node_t))) == NULL) {
207 			nvlist_free(config);
208 			return (-1);
209 		}
210 
211 		if ((cn->cn_name = zfs_strdup(hdl,
212 		    nvpair_name(elem))) == NULL) {
213 			free(cn);
214 			return (-1);
215 		}
216 
217 		verify(nvpair_value_nvlist(elem, &child) == 0);
218 		if (nvlist_dup(child, &cn->cn_config, 0) != 0) {
219 			nvlist_free(config);
220 			return (no_memory(hdl));
221 		}
222 		verify(uu_avl_find(hdl->libzfs_ns_avl, cn, NULL, &where)
223 		    == NULL);
224 
225 		uu_avl_insert(hdl->libzfs_ns_avl, cn, where);
226 	}
227 
228 	nvlist_free(config);
229 	return (0);
230 }
231 
232 /*
233  * Retrive the configuration for the given pool.  The configuration is a nvlist
234  * describing the vdevs, as well as the statistics associated with each one.
235  */
236 nvlist_t *
237 zpool_get_config(zpool_handle_t *zhp, nvlist_t **oldconfig)
238 {
239 	if (oldconfig)
240 		*oldconfig = zhp->zpool_old_config;
241 	return (zhp->zpool_config);
242 }
243 
244 /*
245  * Refresh the vdev statistics associated with the given pool.  This is used in
246  * iostat to show configuration changes and determine the delta from the last
247  * time the function was called.  This function can fail, in case the pool has
248  * been destroyed.
249  */
250 int
251 zpool_refresh_stats(zpool_handle_t *zhp, boolean_t *missing)
252 {
253 	zfs_cmd_t zc = { 0 };
254 	int error;
255 	nvlist_t *config;
256 
257 	*missing = B_FALSE;
258 	(void) strcpy(zc.zc_name, zhp->zpool_name);
259 
260 	if (zhp->zpool_config_size == 0)
261 		zhp->zpool_config_size = 1 << 16;
262 
263 	zc.zc_config_dst_size = zhp->zpool_config_size;
264 	if ((zc.zc_config_dst = (uint64_t)(uintptr_t)
265 	    zfs_alloc(zhp->zpool_hdl, zc.zc_config_dst_size)) == NULL)
266 		return (-1);
267 
268 	for (;;) {
269 		if (ioctl(zhp->zpool_hdl->libzfs_fd, ZFS_IOC_POOL_STATS,
270 		    &zc) == 0) {
271 			/*
272 			 * The real error is returned in the zc_cookie field.
273 			 */
274 			error = zc.zc_cookie;
275 			break;
276 		}
277 
278 		if (errno == ENOMEM) {
279 			free((void *)(uintptr_t)zc.zc_config_dst);
280 			if ((zc.zc_config_dst = (uint64_t)(uintptr_t)
281 			    zfs_alloc(zhp->zpool_hdl,
282 			    zc.zc_config_dst_size)) == NULL)
283 				return (-1);
284 		} else {
285 			free((void *)(uintptr_t)zc.zc_config_dst);
286 			if (errno == ENOENT || errno == EINVAL)
287 				*missing = B_TRUE;
288 			zhp->zpool_state = POOL_STATE_UNAVAIL;
289 			return (0);
290 		}
291 	}
292 
293 	if (nvlist_unpack((void *)(uintptr_t)zc.zc_config_dst,
294 	    zc.zc_config_dst_size, &config, 0) != 0) {
295 		free((void *)(uintptr_t)zc.zc_config_dst);
296 		return (no_memory(zhp->zpool_hdl));
297 	}
298 
299 	zhp->zpool_config_size = zc.zc_config_dst_size;
300 	free((void *)(uintptr_t)zc.zc_config_dst);
301 
302 	if (set_pool_health(config) != 0) {
303 		nvlist_free(config);
304 		return (no_memory(zhp->zpool_hdl));
305 	}
306 
307 	if (zhp->zpool_config != NULL) {
308 		uint64_t oldtxg, newtxg;
309 
310 		verify(nvlist_lookup_uint64(zhp->zpool_config,
311 		    ZPOOL_CONFIG_POOL_TXG, &oldtxg) == 0);
312 		verify(nvlist_lookup_uint64(config,
313 		    ZPOOL_CONFIG_POOL_TXG, &newtxg) == 0);
314 
315 		if (zhp->zpool_old_config != NULL)
316 			nvlist_free(zhp->zpool_old_config);
317 
318 		if (oldtxg != newtxg) {
319 			nvlist_free(zhp->zpool_config);
320 			zhp->zpool_old_config = NULL;
321 		} else {
322 			zhp->zpool_old_config = zhp->zpool_config;
323 		}
324 	}
325 
326 	zhp->zpool_config = config;
327 	if (error)
328 		zhp->zpool_state = POOL_STATE_UNAVAIL;
329 	else
330 		zhp->zpool_state = POOL_STATE_ACTIVE;
331 
332 	return (0);
333 }
334 
335 /*
336  * Iterate over all pools in the system.
337  */
338 int
339 zpool_iter(libzfs_handle_t *hdl, zpool_iter_f func, void *data)
340 {
341 	config_node_t *cn;
342 	zpool_handle_t *zhp;
343 	int ret;
344 
345 	if (namespace_reload(hdl) != 0)
346 		return (-1);
347 
348 	for (cn = uu_avl_first(hdl->libzfs_ns_avl); cn != NULL;
349 	    cn = uu_avl_next(hdl->libzfs_ns_avl, cn)) {
350 
351 		if (zpool_open_silent(hdl, cn->cn_name, &zhp) != 0)
352 			return (-1);
353 
354 		if (zhp == NULL)
355 			continue;
356 
357 		if ((ret = func(zhp, data)) != 0)
358 			return (ret);
359 	}
360 
361 	return (0);
362 }
363 
364 /*
365  * Iterate over root datasets, calling the given function for each.  The zfs
366  * handle passed each time must be explicitly closed by the callback.
367  */
368 int
369 zfs_iter_root(libzfs_handle_t *hdl, zfs_iter_f func, void *data)
370 {
371 	config_node_t *cn;
372 	zfs_handle_t *zhp;
373 	int ret;
374 
375 	if (namespace_reload(hdl) != 0)
376 		return (-1);
377 
378 	for (cn = uu_avl_first(hdl->libzfs_ns_avl); cn != NULL;
379 	    cn = uu_avl_next(hdl->libzfs_ns_avl, cn)) {
380 
381 		if ((zhp = make_dataset_handle(hdl, cn->cn_name)) == NULL)
382 			continue;
383 
384 		if ((ret = func(zhp, data)) != 0)
385 			return (ret);
386 	}
387 
388 	return (0);
389 }
390