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/*
23 * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
24 * Copyright 2012 Milan Jurik. All rights reserved.
25 */
26
27#include "availdevs.h"
28#include <libzfs.h>
29#include <libzfs_jni_diskmgt.h>
30#include <libzfs_jni_ipool.h>
31#include <libxml/parser.h>
32
33/*
34 * Function prototypes
35 */
36
37static void handle_error(const char *, va_list);
38static void set_uint64_prop(xmlNodePtr, const char *, uint64_t);
39static int add_disk_to_xml(dmgt_disk_t *, void *);
40static int add_pool_to_xml(nvlist_t *, void *);
41static xmlDocPtr create_doc();
42int main();
43
44/*
45 * Static functions
46 */
47
48static void
49handle_error(const char *fmt, va_list ap)
50{
51	(void) vfprintf(stderr, fmt, ap);
52	(void) fprintf(stderr, "\n");
53}
54
55static void
56set_uint64_prop(xmlNodePtr node, const char *attr, uint64_t value)
57{
58	static char tmp[64];
59	(void) snprintf(tmp, sizeof (tmp), "%llu", value);
60	(void) xmlSetProp(node, (xmlChar *)attr, (xmlChar *)tmp);
61}
62
63static int
64add_disk_to_xml(dmgt_disk_t *dp, void *data)
65{
66	int i;
67	xmlNodePtr available = *((xmlNodePtr *)data);
68
69	xmlNodePtr disk = xmlNewChild(
70	    available, NULL, (xmlChar *)ELEMENT_DISK, NULL);
71	(void) xmlSetProp(disk,
72	    (xmlChar *)ATTR_DISK_NAME, (xmlChar *)dp->name);
73
74	set_uint64_prop(disk, ATTR_DISK_SIZE, dp->size);
75
76	(void) xmlSetProp(disk, (xmlChar *)ATTR_DISK_INUSE, (xmlChar *)
77	    (dp->in_use ? VAL_ATTR_TRUE : VAL_ATTR_FALSE));
78
79	if (dp->aliases != NULL) {
80		for (i = 0; dp->aliases[i] != NULL; i++) {
81			xmlNodePtr alias = xmlNewChild(
82			    disk, NULL, (xmlChar *)ELEMENT_ALIAS, NULL);
83			(void) xmlSetProp(alias,
84			    (xmlChar *)ATTR_ALIAS_NAME,
85			    (xmlChar *)dp->aliases[i]);
86		}
87	}
88
89	if (dp->slices != NULL) {
90		for (i = 0; dp->slices[i] != NULL; i++) {
91			dmgt_slice_t *sp = dp->slices[i];
92			xmlNodePtr slice = xmlNewChild(
93			    disk, NULL, (xmlChar *)ELEMENT_SLICE, NULL);
94			(void) xmlSetProp(slice,
95			    (xmlChar *)ATTR_SLICE_NAME, (xmlChar *)sp->name);
96
97			set_uint64_prop(slice, ATTR_SLICE_SIZE, sp->size);
98			set_uint64_prop(slice, ATTR_SLICE_START, sp->start);
99
100			if (sp->used_name != NULL) {
101				(void) xmlSetProp(slice,
102				    (xmlChar *)ATTR_SLICE_USED_NAME,
103				    (xmlChar *)sp->used_name);
104			}
105
106			if (sp->used_by != NULL) {
107				(void) xmlSetProp(slice,
108				    (xmlChar *)ATTR_SLICE_USED_BY,
109				    (xmlChar *)sp->used_by);
110			}
111		}
112	}
113
114	return (0);
115}
116
117static int
118add_pool_to_xml(nvlist_t *config, void *data)
119{
120	char *c;
121	char *name;
122	uint64_t guid;
123	uint64_t version;
124	uint64_t state;
125	nvlist_t *devices;
126	uint_t n;
127	vdev_stat_t *vs;
128	xmlNodePtr pool;
129	xmlNodePtr importable = *((xmlNodePtr *)data);
130
131	if (nvlist_lookup_string(config, ZPOOL_CONFIG_POOL_NAME, &name) ||
132	    nvlist_lookup_uint64(config, ZPOOL_CONFIG_POOL_GUID, &guid) ||
133	    nvlist_lookup_uint64(config, ZPOOL_CONFIG_VERSION, &version) ||
134	    nvlist_lookup_uint64(config, ZPOOL_CONFIG_POOL_STATE, &state) ||
135	    nvlist_lookup_nvlist(config, ZPOOL_CONFIG_VDEV_TREE, &devices) ||
136	    nvlist_lookup_uint64_array(
137	    devices, ZPOOL_CONFIG_VDEV_STATS, (uint64_t **)&vs, &n)) {
138		return (-1);
139	}
140
141	pool = xmlNewChild(importable, NULL, (xmlChar *)ELEMENT_POOL, NULL);
142	(void) xmlSetProp(pool, (xmlChar *)ATTR_POOL_NAME, (xmlChar *)name);
143
144	set_uint64_prop(pool, ATTR_POOL_ID, guid);
145	set_uint64_prop(pool, ATTR_POOL_VERSION, version);
146	set_uint64_prop(pool, ATTR_POOL_USED, vs->vs_alloc);
147	set_uint64_prop(pool, ATTR_POOL_SIZE, vs->vs_space);
148	set_uint64_prop(pool, ATTR_POOL_REPLACEMENT_SIZE, vs->vs_rsize);
149	set_uint64_prop(pool, ATTR_POOL_READ_BYTES,
150	    vs->vs_bytes[ZIO_TYPE_READ]);
151	set_uint64_prop(pool, ATTR_POOL_WRITE_BYTES,
152	    vs->vs_bytes[ZIO_TYPE_WRITE]);
153	set_uint64_prop(pool, ATTR_POOL_READ_OPERATIONS,
154	    vs->vs_ops[ZIO_TYPE_READ]);
155	set_uint64_prop(pool, ATTR_POOL_WRITE_OPERATIONS,
156	    vs->vs_ops[ZIO_TYPE_WRITE]);
157	set_uint64_prop(pool, ATTR_POOL_READ_ERRORS, vs->vs_read_errors);
158	set_uint64_prop(pool, ATTR_POOL_WRITE_ERRORS, vs->vs_write_errors);
159	set_uint64_prop(pool, ATTR_POOL_CHECKSUM_ERRORS,
160	    vs->vs_checksum_errors);
161
162	(void) xmlSetProp(pool, (xmlChar *)ATTR_DEVICE_STATE,
163	    (xmlChar *)zjni_vdev_state_to_str(vs->vs_state));
164
165	(void) xmlSetProp(pool, (xmlChar *)ATTR_DEVICE_STATUS,
166	    (xmlChar *)zjni_vdev_aux_to_str(vs->vs_aux));
167
168	(void) xmlSetProp(pool, (xmlChar *)ATTR_POOL_STATE,
169	    (xmlChar *)zjni_pool_state_to_str(state));
170
171	(void) xmlSetProp(pool, (xmlChar *)ATTR_POOL_STATUS, (xmlChar *)
172	    zjni_pool_status_to_str(zpool_import_status(config, &c, NULL)));
173
174	return (0);
175}
176
177static xmlDocPtr
178create_doc(void)
179{
180	/* Create the XML document */
181	xmlDocPtr doc = xmlNewDoc((xmlChar *)"1.0");
182
183	/* Create the root node */
184	xmlNodePtr root = xmlNewDocNode(
185	    doc, NULL, (xmlChar *)ELEMENT_ROOT, NULL);
186	(void) xmlAddChild((xmlNodePtr) doc, (xmlNodePtr)root);
187
188	return (doc);
189}
190
191/*
192 * Main entry to availdisks.
193 *
194 * @return      0 on successful exit, non-zero otherwise
195 */
196int
197main(int argc, char **argv)
198{
199	int error = 0;
200	int get_pools = 0;
201	int get_devices = 0;
202
203	/* Examine first arg */
204	int c = getopt(argc, argv, CLI_OPTSTRING);
205	switch (c) {
206		case CLI_ARG_ALL:
207			get_devices = 1;
208			get_pools = 1;
209			break;
210
211		case CLI_ARG_DEVICES:
212			get_devices = 1;
213			break;
214
215		case CLI_ARG_POOLS:
216			get_pools = 1;
217			break;
218
219		default:
220			return (1);
221	}
222
223	argc -= optind;
224	argv += optind;
225
226	if (get_pools || get_devices) {
227		xmlDocPtr doc = create_doc();
228		xmlNodePtr root = xmlDocGetRootElement(doc);
229
230		if (get_devices) {
231			/* Create the available node */
232			xmlNodePtr available = xmlNewChild(root, NULL,
233			    (xmlChar *)ELEMENT_AVAILABLE, NULL);
234
235			/* libzfs_jni_diskmgt.o error handler */
236			dmgt_set_error_handler(handle_error);
237
238			error = dmgt_avail_disk_iter(
239			    add_disk_to_xml, &available);
240		}
241
242		if (get_pools && !error) {
243			/* Create the importable node */
244			xmlNodePtr importable = xmlNewChild(root, NULL,
245			    (xmlChar *)ELEMENT_IMPORTABLE, NULL);
246
247			error = zjni_ipool_iter(
248			    argc, argv, add_pool_to_xml, &importable);
249		}
250
251		if (!error) {
252			/* Print out XML */
253			(void) xmlDocFormatDump(stdout, doc, 1);
254		}
255
256		xmlFreeDoc(doc);
257	}
258
259	return (error != 0);
260}
261