xref: /titanic_51/usr/src/cmd/zpool/zpool_util.c (revision 88ecc943b4eb72f7c4fbbd8435997b85ef171fc3)
1fa9e4066Sahrens /*
2fa9e4066Sahrens  * CDDL HEADER START
3fa9e4066Sahrens  *
4fa9e4066Sahrens  * The contents of this file are subject to the terms of the
55ad82045Snd150628  * Common Development and Distribution License (the "License").
65ad82045Snd150628  * You may not use this file except in compliance with the License.
7fa9e4066Sahrens  *
8fa9e4066Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9fa9e4066Sahrens  * or http://www.opensolaris.org/os/licensing.
10fa9e4066Sahrens  * See the License for the specific language governing permissions
11fa9e4066Sahrens  * and limitations under the License.
12fa9e4066Sahrens  *
13fa9e4066Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
14fa9e4066Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15fa9e4066Sahrens  * If applicable, add the following below this CDDL HEADER, with the
16fa9e4066Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
17fa9e4066Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
18fa9e4066Sahrens  *
19fa9e4066Sahrens  * CDDL HEADER END
20fa9e4066Sahrens  */
21fa9e4066Sahrens /*
22*26fd7700SKrishnendu Sadhukhan - Sun Microsystems  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
23fa9e4066Sahrens  * Use is subject to license terms.
24fa9e4066Sahrens  */
25fa9e4066Sahrens 
26fa9e4066Sahrens #include <errno.h>
27fa9e4066Sahrens #include <libgen.h>
28fa9e4066Sahrens #include <libintl.h>
29fa9e4066Sahrens #include <stdio.h>
30fa9e4066Sahrens #include <stdlib.h>
31fa9e4066Sahrens #include <strings.h>
32fa9e4066Sahrens 
33fa9e4066Sahrens #include "zpool_util.h"
34fa9e4066Sahrens 
35fa9e4066Sahrens /*
36fa9e4066Sahrens  * Utility function to guarantee malloc() success.
37fa9e4066Sahrens  */
38fa9e4066Sahrens void *
39fa9e4066Sahrens safe_malloc(size_t size)
40fa9e4066Sahrens {
41fa9e4066Sahrens 	void *data;
42fa9e4066Sahrens 
43fa9e4066Sahrens 	if ((data = calloc(1, size)) == NULL) {
44fa9e4066Sahrens 		(void) fprintf(stderr, "internal error: out of memory\n");
45fa9e4066Sahrens 		exit(1);
46fa9e4066Sahrens 	}
47fa9e4066Sahrens 
48fa9e4066Sahrens 	return (data);
49fa9e4066Sahrens }
50fa9e4066Sahrens 
51fa9e4066Sahrens /*
52fa9e4066Sahrens  * Display an out of memory error message and abort the current program.
53fa9e4066Sahrens  */
54fa9e4066Sahrens void
555ad82045Snd150628 zpool_no_memory(void)
56fa9e4066Sahrens {
57fa9e4066Sahrens 	assert(errno == ENOMEM);
58fa9e4066Sahrens 	(void) fprintf(stderr,
59fa9e4066Sahrens 	    gettext("internal error: out of memory\n"));
60fa9e4066Sahrens 	exit(1);
61fa9e4066Sahrens }
628654d025Sperrin 
638654d025Sperrin /*
648654d025Sperrin  * Return the number of logs in supplied nvlist
658654d025Sperrin  */
668654d025Sperrin uint_t
678654d025Sperrin num_logs(nvlist_t *nv)
688654d025Sperrin {
698654d025Sperrin 	uint_t nlogs = 0;
708654d025Sperrin 	uint_t c, children;
718654d025Sperrin 	nvlist_t **child;
728654d025Sperrin 
738654d025Sperrin 	if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_CHILDREN,
748654d025Sperrin 	    &child, &children) != 0)
758654d025Sperrin 		return (0);
768654d025Sperrin 
778654d025Sperrin 	for (c = 0; c < children; c++) {
788654d025Sperrin 		uint64_t is_log = B_FALSE;
798654d025Sperrin 
808654d025Sperrin 		(void) nvlist_lookup_uint64(child[c], ZPOOL_CONFIG_IS_LOG,
818654d025Sperrin 		    &is_log);
828654d025Sperrin 		if (is_log)
838654d025Sperrin 			nlogs++;
848654d025Sperrin 	}
858654d025Sperrin 	return (nlogs);
868654d025Sperrin }
87