Home | History | Annotate | Line # | Download | only in common
mem.c revision 1.11 
      1 /*	$NetBSD: mem.c,v 1.11 2020/12/29 11:35:11 rillig Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 1994, 1995 Jochen Pohl
      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  * 3. All advertising materials mentioning features or use of this software
     16  *    must display the following acknowledgement:
     17  *      This product includes software developed by Jochen Pohl for
     18  *	The NetBSD Project.
     19  * 4. The name of the author may not be used to endorse or promote products
     20  *    derived from this software without specific prior written permission.
     21  *
     22  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     23  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     24  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     25  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     26  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     27  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     28  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     29  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     30  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     31  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     32  */
     33 
     34 #if HAVE_NBTOOL_CONFIG_H
     35 #include "nbtool_config.h"
     36 #endif
     37 
     38 #include <sys/cdefs.h>
     39 #if defined(__RCSID) && !defined(lint)
     40 __RCSID("$NetBSD: mem.c,v 1.11 2020/12/29 11:35:11 rillig Exp $");
     41 #endif
     42 
     43 #include <sys/param.h>
     44 #include <sys/types.h>
     45 #include <sys/mman.h>
     46 #include <stdarg.h>
     47 #include <stdlib.h>
     48 #include <string.h>
     49 
     50 #include "lint.h"
     51 
     52 void *
     53 xmalloc(size_t s)
     54 {
     55 	void	*p;
     56 
     57 	if ((p = malloc(s)) == NULL)
     58 		nomem();
     59 	return p;
     60 }
     61 
     62 void *
     63 xcalloc(size_t n, size_t s)
     64 {
     65 	void	*p;
     66 
     67 	if ((p = calloc(n, s)) == NULL)
     68 		nomem();
     69 	return p;
     70 }
     71 
     72 void *
     73 xrealloc(void *p, size_t s)
     74 {
     75 	void *n;
     76 
     77 	if ((n = realloc(p, s)) == NULL) {
     78 		free(p);
     79 		nomem();
     80 	}
     81 	p = n;
     82 	return p;
     83 }
     84 
     85 char *
     86 xstrdup(const char *s)
     87 {
     88 	char	*s2;
     89 
     90 	if ((s2 = strdup(s)) == NULL)
     91 		nomem();
     92 	return s2;
     93 }
     94 
     95 void
     96 nomem(void)
     97 {
     98 
     99 	errx(1, "virtual memory exhausted");
    100 }
    101 
    102 void
    103 xasprintf(char **buf, const char *fmt, ...)
    104 {
    105 	int e;
    106 	va_list ap;
    107 
    108 	va_start(ap, fmt);
    109 	e = vasprintf(buf, fmt, ap);
    110 	va_end(ap);
    111 	if (e < 0)
    112 		nomem();
    113 }
    114 
    115 #if defined(MAP_ANONYMOUS) && !defined(MAP_ANON)
    116 #define	MAP_ANON	MAP_ANONYMOUS
    117 #endif
    118 
    119 void *
    120 xmapalloc(size_t len)
    121 {
    122 	static const int prot = PROT_READ | PROT_WRITE;
    123 	static int fd = -1;
    124 	void *p;
    125 #ifdef MAP_ANON
    126 	static const int flags = MAP_ANON | MAP_PRIVATE;
    127 #else
    128 	static const int flags = MAP_PRIVATE;
    129 
    130 	if (fd == -1) {
    131 		if ((fd = open("/dev/zero", O_RDWR)) == -1)
    132 			err(1, "Cannot open `/dev/zero'");
    133 	}
    134 #endif
    135 	p = mmap(NULL, len, prot, flags, fd, (off_t)0);
    136 	if (p == (void *)-1)
    137 		err(1, "Cannot map memory for %lu bytes", (unsigned long)len);
    138 	return p;
    139 }
    140