xref: /titanic_51/usr/src/cmd/zpool/zpool_util.c (revision 26fd77009b17f8c8fb32eb362584cfd635e87ad9)
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  * Same as above, but for strdup()
53fa9e4066Sahrens  */
54fa9e4066Sahrens char *
55*26fd7700SKrishnendu Sadhukhan - Sun Microsystems zpool_safe_strdup(const char *str)
56fa9e4066Sahrens {
57fa9e4066Sahrens 	char *ret;
58fa9e4066Sahrens 
59fa9e4066Sahrens 	if ((ret = strdup(str)) == NULL) {
60fa9e4066Sahrens 		(void) fprintf(stderr, "internal error: out of memory\n");
61fa9e4066Sahrens 		exit(1);
62fa9e4066Sahrens 	}
63fa9e4066Sahrens 
64fa9e4066Sahrens 	return (ret);
65fa9e4066Sahrens }
66fa9e4066Sahrens 
67fa9e4066Sahrens /*
68fa9e4066Sahrens  * Display an out of memory error message and abort the current program.
69fa9e4066Sahrens  */
70fa9e4066Sahrens void
715ad82045Snd150628 zpool_no_memory(void)
72fa9e4066Sahrens {
73fa9e4066Sahrens 	assert(errno == ENOMEM);
74fa9e4066Sahrens 	(void) fprintf(stderr,
75fa9e4066Sahrens 	    gettext("internal error: out of memory\n"));
76fa9e4066Sahrens 	exit(1);
77fa9e4066Sahrens }
788654d025Sperrin 
798654d025Sperrin /*
808654d025Sperrin  * Return the number of logs in supplied nvlist
818654d025Sperrin  */
828654d025Sperrin uint_t
838654d025Sperrin num_logs(nvlist_t *nv)
848654d025Sperrin {
858654d025Sperrin 	uint_t nlogs = 0;
868654d025Sperrin 	uint_t c, children;
878654d025Sperrin 	nvlist_t **child;
888654d025Sperrin 
898654d025Sperrin 	if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_CHILDREN,
908654d025Sperrin 	    &child, &children) != 0)
918654d025Sperrin 		return (0);
928654d025Sperrin 
938654d025Sperrin 	for (c = 0; c < children; c++) {
948654d025Sperrin 		uint64_t is_log = B_FALSE;
958654d025Sperrin 
968654d025Sperrin 		(void) nvlist_lookup_uint64(child[c], ZPOOL_CONFIG_IS_LOG,
978654d025Sperrin 		    &is_log);
988654d025Sperrin 		if (is_log)
998654d025Sperrin 			nlogs++;
1008654d025Sperrin 	}
1018654d025Sperrin 	return (nlogs);
1028654d025Sperrin }
103