make_malloc.c revision 1.18 
1/*	$NetBSD: make_malloc.c,v 1.18 2020/09/02 06:10:44 rillig 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.18 2020/09/02 06:10:44 rillig Exp $");
32#endif
33
34#include <stdio.h>
35#include <stdlib.h>
36#include <string.h>
37#include <errno.h>
38
39#include "make.h"
40
41#ifndef USE_EMALLOC
42static MAKE_ATTR_DEAD void enomem(void);
43
44/* die when out of memory. */
45static MAKE_ATTR_DEAD void
46enomem(void)
47{
48	(void)fprintf(stderr, "%s: %s.\n", progname, strerror(ENOMEM));
49	exit(2);
50}
51
52/* malloc, but die on error. */
53void *
54bmake_malloc(size_t len)
55{
56	void *p;
57
58	if ((p = malloc(len)) == NULL)
59		enomem();
60	return p;
61}
62
63/* strdup, but die on error. */
64char *
65bmake_strdup(const char *str)
66{
67	size_t len;
68	char *p;
69
70	len = strlen(str) + 1;
71	if ((p = malloc(len)) == NULL)
72		enomem();
73	return memcpy(p, str, len);
74}
75
76/* Allocate a string starting from str with exactly len characters. */
77char *
78bmake_strldup(const char *str, size_t len)
79{
80	char *p = bmake_malloc(len + 1);
81	memcpy(p, str, len);
82	p[len] = '\0';
83	return p;
84}
85
86/* realloc, but die on error. */
87void *
88bmake_realloc(void *ptr, size_t size)
89{
90	if ((ptr = realloc(ptr, size)) == NULL)
91		enomem();
92	return ptr;
93}
94#endif
95
96/* Allocate a string from start up to but excluding end. */
97char *
98bmake_strsedup(const char *start, const char *end)
99{
100	return bmake_strldup(start, (size_t)(end - start));
101}
102