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 * 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 * Same as above, but for strdup() 53 */ 54 char * 55 zpool_safe_strdup(const char *str) 56 { 57 char *ret; 58 59 if ((ret = strdup(str)) == NULL) { 60 (void) fprintf(stderr, "internal error: out of memory\n"); 61 exit(1); 62 } 63 64 return (ret); 65 } 66 67 /* 68 * Display an out of memory error message and abort the current program. 69 */ 70 void 71 zpool_no_memory(void) 72 { 73 assert(errno == ENOMEM); 74 (void) fprintf(stderr, 75 gettext("internal error: out of memory\n")); 76 exit(1); 77 } 78 79 /* 80 * Return the number of logs in supplied nvlist 81 */ 82 uint_t 83 num_logs(nvlist_t *nv) 84 { 85 uint_t nlogs = 0; 86 uint_t c, children; 87 nvlist_t **child; 88 89 if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_CHILDREN, 90 &child, &children) != 0) 91 return (0); 92 93 for (c = 0; c < children; c++) { 94 uint64_t is_log = B_FALSE; 95 96 (void) nvlist_lookup_uint64(child[c], ZPOOL_CONFIG_IS_LOG, 97 &is_log); 98 if (is_log) 99 nlogs++; 100 } 101 return (nlogs); 102 } 103