xref: /freebsd/sys/contrib/openzfs/lib/libzpool/util.c (revision cfd6422a5217410fbd66f7a7a8a64d9d85e61229)
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 (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
23  * Copyright (c) 2016 by Delphix. All rights reserved.
24  * Copyright 2017 Jason King
25  * Copyright (c) 2017, Intel Corporation.
26  */
27 
28 #include <assert.h>
29 #include <sys/zfs_context.h>
30 #include <sys/avl.h>
31 #include <string.h>
32 #include <stdio.h>
33 #include <stdlib.h>
34 #include <sys/spa.h>
35 #include <sys/fs/zfs.h>
36 #include <sys/zfs_refcount.h>
37 #include <sys/zfs_ioctl.h>
38 #include <dlfcn.h>
39 #include <libzutil.h>
40 
41 /*
42  * Routines needed by more than one client of libzpool.
43  */
44 
45 static void
46 show_vdev_stats(const char *desc, const char *ctype, nvlist_t *nv, int indent)
47 {
48 	vdev_stat_t *vs;
49 	vdev_stat_t *v0 = { 0 };
50 	uint64_t sec;
51 	uint64_t is_log = 0;
52 	nvlist_t **child;
53 	uint_t c, children;
54 	char used[6], avail[6];
55 	char rops[6], wops[6], rbytes[6], wbytes[6], rerr[6], werr[6], cerr[6];
56 
57 	v0 = umem_zalloc(sizeof (*v0), UMEM_NOFAIL);
58 
59 	if (indent == 0 && desc != NULL) {
60 		(void) printf("                           "
61 		    " capacity   operations   bandwidth  ---- errors ----\n");
62 		(void) printf("description                "
63 		    "used avail  read write  read write  read write cksum\n");
64 	}
65 
66 	if (desc != NULL) {
67 		char *suffix = "", *bias = NULL;
68 		char bias_suffix[32];
69 
70 		(void) nvlist_lookup_uint64(nv, ZPOOL_CONFIG_IS_LOG, &is_log);
71 		(void) nvlist_lookup_string(nv, ZPOOL_CONFIG_ALLOCATION_BIAS,
72 		    &bias);
73 		if (nvlist_lookup_uint64_array(nv, ZPOOL_CONFIG_VDEV_STATS,
74 		    (uint64_t **)&vs, &c) != 0)
75 			vs = v0;
76 
77 		if (bias != NULL) {
78 			(void) snprintf(bias_suffix, sizeof (bias_suffix),
79 			    " (%s)", bias);
80 			suffix = bias_suffix;
81 		} else if (is_log) {
82 			suffix = " (log)";
83 		}
84 
85 		sec = MAX(1, vs->vs_timestamp / NANOSEC);
86 
87 		nicenum(vs->vs_alloc, used, sizeof (used));
88 		nicenum(vs->vs_space - vs->vs_alloc, avail, sizeof (avail));
89 		nicenum(vs->vs_ops[ZIO_TYPE_READ] / sec, rops, sizeof (rops));
90 		nicenum(vs->vs_ops[ZIO_TYPE_WRITE] / sec, wops, sizeof (wops));
91 		nicenum(vs->vs_bytes[ZIO_TYPE_READ] / sec, rbytes,
92 		    sizeof (rbytes));
93 		nicenum(vs->vs_bytes[ZIO_TYPE_WRITE] / sec, wbytes,
94 		    sizeof (wbytes));
95 		nicenum(vs->vs_read_errors, rerr, sizeof (rerr));
96 		nicenum(vs->vs_write_errors, werr, sizeof (werr));
97 		nicenum(vs->vs_checksum_errors, cerr, sizeof (cerr));
98 
99 		(void) printf("%*s%s%*s%*s%*s %5s %5s %5s %5s %5s %5s %5s\n",
100 		    indent, "",
101 		    desc,
102 		    (int)(indent+strlen(desc)-25-(vs->vs_space ? 0 : 12)),
103 		    suffix,
104 		    vs->vs_space ? 6 : 0, vs->vs_space ? used : "",
105 		    vs->vs_space ? 6 : 0, vs->vs_space ? avail : "",
106 		    rops, wops, rbytes, wbytes, rerr, werr, cerr);
107 	}
108 	umem_free(v0, sizeof (*v0));
109 
110 	if (nvlist_lookup_nvlist_array(nv, ctype, &child, &children) != 0)
111 		return;
112 
113 	for (c = 0; c < children; c++) {
114 		nvlist_t *cnv = child[c];
115 		char *cname = NULL, *tname;
116 		uint64_t np;
117 		int len;
118 		if (nvlist_lookup_string(cnv, ZPOOL_CONFIG_PATH, &cname) &&
119 		    nvlist_lookup_string(cnv, ZPOOL_CONFIG_TYPE, &cname))
120 			cname = "<unknown>";
121 		len = strlen(cname) + 2;
122 		tname = umem_zalloc(len, UMEM_NOFAIL);
123 		(void) strlcpy(tname, cname, len);
124 		if (nvlist_lookup_uint64(cnv, ZPOOL_CONFIG_NPARITY, &np) == 0)
125 			tname[strlen(tname)] = '0' + np;
126 		show_vdev_stats(tname, ctype, cnv, indent + 2);
127 		umem_free(tname, len);
128 	}
129 }
130 
131 void
132 show_pool_stats(spa_t *spa)
133 {
134 	nvlist_t *config, *nvroot;
135 	char *name;
136 
137 	VERIFY(spa_get_stats(spa_name(spa), &config, NULL, 0) == 0);
138 
139 	VERIFY(nvlist_lookup_nvlist(config, ZPOOL_CONFIG_VDEV_TREE,
140 	    &nvroot) == 0);
141 	VERIFY(nvlist_lookup_string(config, ZPOOL_CONFIG_POOL_NAME,
142 	    &name) == 0);
143 
144 	show_vdev_stats(name, ZPOOL_CONFIG_CHILDREN, nvroot, 0);
145 	show_vdev_stats(NULL, ZPOOL_CONFIG_L2CACHE, nvroot, 0);
146 	show_vdev_stats(NULL, ZPOOL_CONFIG_SPARES, nvroot, 0);
147 
148 	nvlist_free(config);
149 }
150 
151 /*
152  * Sets given global variable in libzpool to given unsigned 32-bit value.
153  * arg: "<variable>=<value>"
154  */
155 int
156 set_global_var(char *arg)
157 {
158 	void *zpoolhdl;
159 	char *varname = arg, *varval;
160 	u_longlong_t val;
161 
162 #ifndef _LITTLE_ENDIAN
163 	/*
164 	 * On big endian systems changing a 64-bit variable would set the high
165 	 * 32 bits instead of the low 32 bits, which could cause unexpected
166 	 * results.
167 	 */
168 	fprintf(stderr, "Setting global variables is only supported on "
169 	    "little-endian systems\n");
170 	return (ENOTSUP);
171 #endif
172 	if (arg != NULL && (varval = strchr(arg, '=')) != NULL) {
173 		*varval = '\0';
174 		varval++;
175 		val = strtoull(varval, NULL, 0);
176 		if (val > UINT32_MAX) {
177 			fprintf(stderr, "Value for global variable '%s' must "
178 			    "be a 32-bit unsigned integer\n", varname);
179 			return (EOVERFLOW);
180 		}
181 	} else {
182 		return (EINVAL);
183 	}
184 
185 	zpoolhdl = dlopen("libzpool.so", RTLD_LAZY);
186 	if (zpoolhdl != NULL) {
187 		uint32_t *var;
188 		var = dlsym(zpoolhdl, varname);
189 		if (var == NULL) {
190 			fprintf(stderr, "Global variable '%s' does not exist "
191 			    "in libzpool.so\n", varname);
192 			return (EINVAL);
193 		}
194 		*var = (uint32_t)val;
195 
196 		dlclose(zpoolhdl);
197 	} else {
198 		fprintf(stderr, "Failed to open libzpool.so to set global "
199 		    "variable\n");
200 		return (EIO);
201 	}
202 
203 	return (0);
204 }
205 
206 static nvlist_t *
207 refresh_config(void *unused, nvlist_t *tryconfig)
208 {
209 	return (spa_tryimport(tryconfig));
210 }
211 
212 static int
213 pool_active(void *unused, const char *name, uint64_t guid,
214     boolean_t *isactive)
215 {
216 	zfs_cmd_t *zcp;
217 	nvlist_t *innvl;
218 	char *packed = NULL;
219 	size_t size = 0;
220 	int fd, ret;
221 
222 	/*
223 	 * Use ZFS_IOC_POOL_SYNC to confirm if a pool is active
224 	 */
225 
226 	fd = open(ZFS_DEV, O_RDWR);
227 	if (fd < 0)
228 		return (-1);
229 
230 	zcp = umem_zalloc(sizeof (zfs_cmd_t), UMEM_NOFAIL);
231 
232 	innvl = fnvlist_alloc();
233 	fnvlist_add_boolean_value(innvl, "force", B_FALSE);
234 
235 	(void) strlcpy(zcp->zc_name, name, sizeof (zcp->zc_name));
236 	packed = fnvlist_pack(innvl, &size);
237 	zcp->zc_nvlist_src = (uint64_t)(uintptr_t)packed;
238 	zcp->zc_nvlist_src_size = size;
239 
240 	ret = zfs_ioctl_fd(fd, ZFS_IOC_POOL_SYNC, zcp);
241 
242 	fnvlist_pack_free(packed, size);
243 	free((void *)(uintptr_t)zcp->zc_nvlist_dst);
244 	nvlist_free(innvl);
245 	umem_free(zcp, sizeof (zfs_cmd_t));
246 
247 	(void) close(fd);
248 
249 	*isactive = (ret == 0);
250 
251 	return (0);
252 }
253 
254 const pool_config_ops_t libzpool_config_ops = {
255 	.pco_refresh_config = refresh_config,
256 	.pco_pool_active = pool_active,
257 };
258