xref: /titanic_51/usr/src/cmd/zpool/zpool_util.c (revision fa9e4066f08beec538e775443c5be79dd423fcab)
1*fa9e4066Sahrens /*
2*fa9e4066Sahrens  * CDDL HEADER START
3*fa9e4066Sahrens  *
4*fa9e4066Sahrens  * The contents of this file are subject to the terms of the
5*fa9e4066Sahrens  * Common Development and Distribution License, Version 1.0 only
6*fa9e4066Sahrens  * (the "License").  You may not use this file except in compliance
7*fa9e4066Sahrens  * with the License.
8*fa9e4066Sahrens  *
9*fa9e4066Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*fa9e4066Sahrens  * or http://www.opensolaris.org/os/licensing.
11*fa9e4066Sahrens  * See the License for the specific language governing permissions
12*fa9e4066Sahrens  * and limitations under the License.
13*fa9e4066Sahrens  *
14*fa9e4066Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
15*fa9e4066Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*fa9e4066Sahrens  * If applicable, add the following below this CDDL HEADER, with the
17*fa9e4066Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
18*fa9e4066Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
19*fa9e4066Sahrens  *
20*fa9e4066Sahrens  * CDDL HEADER END
21*fa9e4066Sahrens  */
22*fa9e4066Sahrens /*
23*fa9e4066Sahrens  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
24*fa9e4066Sahrens  * Use is subject to license terms.
25*fa9e4066Sahrens  */
26*fa9e4066Sahrens 
27*fa9e4066Sahrens #pragma ident	"%Z%%M%	%I%	%E% SMI"
28*fa9e4066Sahrens 
29*fa9e4066Sahrens #include <errno.h>
30*fa9e4066Sahrens #include <libgen.h>
31*fa9e4066Sahrens #include <libintl.h>
32*fa9e4066Sahrens #include <stdio.h>
33*fa9e4066Sahrens #include <stdlib.h>
34*fa9e4066Sahrens #include <strings.h>
35*fa9e4066Sahrens 
36*fa9e4066Sahrens #include "zpool_util.h"
37*fa9e4066Sahrens 
38*fa9e4066Sahrens /*
39*fa9e4066Sahrens  * Utility function to guarantee malloc() success.
40*fa9e4066Sahrens  */
41*fa9e4066Sahrens void *
42*fa9e4066Sahrens safe_malloc(size_t size)
43*fa9e4066Sahrens {
44*fa9e4066Sahrens 	void *data;
45*fa9e4066Sahrens 
46*fa9e4066Sahrens 	if ((data = calloc(1, size)) == NULL) {
47*fa9e4066Sahrens 		(void) fprintf(stderr, "internal error: out of memory\n");
48*fa9e4066Sahrens 		exit(1);
49*fa9e4066Sahrens 	}
50*fa9e4066Sahrens 
51*fa9e4066Sahrens 	return (data);
52*fa9e4066Sahrens }
53*fa9e4066Sahrens 
54*fa9e4066Sahrens /*
55*fa9e4066Sahrens  * Same as above, but for strdup()
56*fa9e4066Sahrens  */
57*fa9e4066Sahrens char *
58*fa9e4066Sahrens safe_strdup(const char *str)
59*fa9e4066Sahrens {
60*fa9e4066Sahrens 	char *ret;
61*fa9e4066Sahrens 
62*fa9e4066Sahrens 	if ((ret = strdup(str)) == NULL) {
63*fa9e4066Sahrens 		(void) fprintf(stderr, "internal error: out of memory\n");
64*fa9e4066Sahrens 		exit(1);
65*fa9e4066Sahrens 	}
66*fa9e4066Sahrens 
67*fa9e4066Sahrens 	return (ret);
68*fa9e4066Sahrens }
69*fa9e4066Sahrens 
70*fa9e4066Sahrens /*
71*fa9e4066Sahrens  * Display an out of memory error message and abort the current program.
72*fa9e4066Sahrens  */
73*fa9e4066Sahrens void
74*fa9e4066Sahrens no_memory(void)
75*fa9e4066Sahrens {
76*fa9e4066Sahrens 	assert(errno == ENOMEM);
77*fa9e4066Sahrens 	(void) fprintf(stderr,
78*fa9e4066Sahrens 	    gettext("internal error: out of memory\n"));
79*fa9e4066Sahrens 	exit(1);
80*fa9e4066Sahrens }
81*fa9e4066Sahrens 
82*fa9e4066Sahrens /*
83*fa9e4066Sahrens  * Given a vdev, return the name to display in iostat.  If the vdev has a path,
84*fa9e4066Sahrens  * we use that, stripping off any leading "/dev/dsk/"; if not, we use the type.
85*fa9e4066Sahrens  */
86*fa9e4066Sahrens const char *
87*fa9e4066Sahrens vdev_get_name(nvlist_t *nv)
88*fa9e4066Sahrens {
89*fa9e4066Sahrens 	char *path, *type;
90*fa9e4066Sahrens 
91*fa9e4066Sahrens 	if (nvlist_lookup_string(nv, ZPOOL_CONFIG_PATH, &path) == 0) {
92*fa9e4066Sahrens 		if (strncmp(path, "/dev/dsk/", 9) == 0)
93*fa9e4066Sahrens 			return (path + 9);
94*fa9e4066Sahrens 		return (path);
95*fa9e4066Sahrens 	}
96*fa9e4066Sahrens 
97*fa9e4066Sahrens 	verify(nvlist_lookup_string(nv, ZPOOL_CONFIG_TYPE, &type) == 0);
98*fa9e4066Sahrens 
99*fa9e4066Sahrens 	return (type);
100*fa9e4066Sahrens }
101