make_malloc.c revision 1.8 
1/*	$NetBSD: make_malloc.c,v 1.8 2012/06/10 21:44:12 wiz Exp $	*/
2
3/*-
4 * Copyright (c) 2009 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29#ifdef MAKE_NATIVE
30#include <sys/cdefs.h>
31__RCSID("$NetBSD: make_malloc.c,v 1.8 2012/06/10 21:44:12 wiz Exp $");
32#endif
33
34#include <stdio.h>
35#include <stdlib.h>
36#include <string.h>
37#include <errno.h>
38
39#include "make_malloc.h"
40
41#ifndef USE_EMALLOC
42static void enomem(void) __dead;
43
44/*
45 * enomem --
46 *	die when out of memory.
47 */
48extern char *progname;
49
50static void
51enomem(void)
52{
53	(void)fprintf(stderr, "%s: %s.\n", progname, strerror(ENOMEM));
54	exit(2);
55}
56
57/*
58 * bmake_malloc --
59 *	malloc, but die on error.
60 */
61void *
62bmake_malloc(size_t len)
63{
64	void *p;
65
66	if ((p = malloc(len)) == NULL)
67		enomem();
68	return(p);
69}
70
71/*
72 * bmake_strdup --
73 *	strdup, but die on error.
74 */
75char *
76bmake_strdup(const char *str)
77{
78	size_t len;
79	char *p;
80
81	len = strlen(str) + 1;
82	if ((p = malloc(len)) == NULL)
83		enomem();
84	return memcpy(p, str, len);
85}
86
87/*
88 * bmake_strndup --
89 *	strndup, but die on error.
90 */
91char *
92bmake_strndup(const char *str, size_t max_len)
93{
94	size_t len;
95	char *p;
96
97	if (str == NULL)
98		return NULL;
99
100	len = strlen(str);
101	if (len > max_len)
102		len = max_len;
103	p = bmake_malloc(len + 1);
104	memcpy(p, str, len);
105	p[len] = '\0';
106
107	return(p);
108}
109
110/*
111 * bmake_realloc --
112 *	realloc, but die on error.
113 */
114void *
115bmake_realloc(void *ptr, size_t size)
116{
117	if ((ptr = realloc(ptr, size)) == NULL)
118		enomem();
119	return(ptr);
120}
121#endif
122