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