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 */
38void *
39safe_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 */
54void
55zpool_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 */
66uint_t
67num_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