1 // SPDX-License-Identifier: CDDL-1.0 2 /* 3 * CDDL HEADER START 4 * 5 * The contents of this file are subject to the terms of the 6 * Common Development and Distribution License (the "License"). 7 * You may not use this file except in compliance with the License. 8 * 9 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 10 * or https://opensource.org/licenses/CDDL-1.0. 11 * See the License for the specific language governing permissions 12 * and limitations under the License. 13 * 14 * When distributing Covered Code, include this CDDL HEADER in each 15 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 16 * If applicable, add the following below this CDDL HEADER, with the 17 * fields enclosed by brackets "[]" replaced with your own identifying 18 * information: Portions Copyright [yyyy] [name of copyright owner] 19 * 20 * CDDL HEADER END 21 */ 22 /* 23 * Copyright 2009 Sun Microsystems, Inc. All rights reserved. 24 * Use is subject to license terms. 25 */ 26 27 #include <errno.h> 28 #include <libgen.h> 29 #include <libintl.h> 30 #include <stdio.h> 31 #include <stdlib.h> 32 #include <string.h> 33 #include <ctype.h> 34 35 #include "zpool_util.h" 36 37 /* 38 * Utility function to guarantee malloc() success. 39 */ 40 void * 41 safe_malloc(size_t size) 42 { 43 void *data; 44 45 if ((data = calloc(1, size)) == NULL) { 46 (void) fprintf(stderr, "internal error: out of memory\n"); 47 exit(1); 48 } 49 50 return (data); 51 } 52 53 /* 54 * Utility function to guarantee realloc() success. 55 */ 56 void * 57 safe_realloc(void *from, size_t size) 58 { 59 void *data; 60 61 if ((data = realloc(from, size)) == NULL) { 62 (void) fprintf(stderr, "internal error: out of memory\n"); 63 exit(1); 64 } 65 66 return (data); 67 } 68 69 /* 70 * Display an out of memory error message and abort the current program. 71 */ 72 void 73 zpool_no_memory(void) 74 { 75 assert(errno == ENOMEM); 76 (void) fprintf(stderr, 77 gettext("internal error: out of memory\n")); 78 exit(1); 79 } 80 81 /* 82 * Return the number of logs in supplied nvlist 83 */ 84 uint_t 85 num_logs(nvlist_t *nv) 86 { 87 uint_t nlogs = 0; 88 uint_t c, children; 89 nvlist_t **child; 90 91 if (nvlist_lookup_nvlist_array(nv, ZPOOL_CONFIG_CHILDREN, 92 &child, &children) != 0) 93 return (0); 94 95 for (c = 0; c < children; c++) { 96 uint64_t is_log = B_FALSE; 97 98 (void) nvlist_lookup_uint64(child[c], ZPOOL_CONFIG_IS_LOG, 99 &is_log); 100 if (is_log) 101 nlogs++; 102 } 103 return (nlogs); 104 } 105 106 /* Find the max element in an array of uint64_t values */ 107 uint64_t 108 array64_max(uint64_t array[], unsigned int len) 109 { 110 uint64_t max = 0; 111 int i; 112 for (i = 0; i < len; i++) 113 max = MAX(max, array[i]); 114 115 return (max); 116 } 117 118 /* 119 * Find highest one bit set. 120 * Returns bit number + 1 of highest bit that is set, otherwise returns 0. 121 */ 122 int 123 highbit64(uint64_t i) 124 { 125 if (i == 0) 126 return (0); 127 128 return (NBBY * sizeof (uint64_t) - __builtin_clzll(i)); 129 } 130 131 /* 132 * Find lowest one bit set. 133 * Returns bit number + 1 of lowest bit that is set, otherwise returns 0. 134 */ 135 int 136 lowbit64(uint64_t i) 137 { 138 if (i == 0) 139 return (0); 140 141 return (__builtin_ffsll(i)); 142 } 143