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 2009 Sun Microsystems, Inc. All rights reserved.
23 * Use is subject to license terms.
24 */
25
26 #include <errno.h>
27 #include <libgen.h>
28 #include <libintl.h>
29 #include <stdio.h>
30 #include <stdlib.h>
31 #include <strings.h>
32
33 #include "zpool_util.h"
34
35 /*
36 * Utility function to guarantee malloc() success.
37 */
38 void *
safe_malloc(size_t size)39 safe_malloc(size_t size)
40 {
41 void *data;
42
43 if ((data = calloc(1, size)) == NULL) {
44 (void) fprintf(stderr, "internal error: out of memory\n");
45 exit(1);
46 }
47
48 return (data);
49 }
50
51 /*
52 * Display an out of memory error message and abort the current program.
53 */
54 void
zpool_no_memory(void)55 zpool_no_memory(void)
56 {
57 assert(errno == ENOMEM);
58 (void) fprintf(stderr,
59 gettext("internal error: out of memory\n"));
60 exit(1);
61 }
62
63 /*
64 * Return the number of logs in supplied nvlist
65 */
66 uint_t
num_logs(nvlist_t * nv)67 num_logs(nvlist_t *nv)
68 {
69 uint_t nlogs = 0;
70 uint_t c, children;
71 nvlist_t **child;
72
73 if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_CHILDREN,
74 &child, &children) != 0)
75 return (0);
76
77 for (c = 0; c < children; c++) {
78 uint64_t is_log = B_FALSE;
79
80 (void) nvlist_lookup_uint64(child[c], ZPOOL_CONFIG_IS_LOG,
81 &is_log);
82 if (is_log)
83 nlogs++;
84 }
85 return (nlogs);
86 }
87