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