1 /* $OpenBSD: xmalloc.c,v 1.36 2019/11/12 22:32:48 djm Exp $ */ 2 /* 3 * Author: Tatu Ylonen <ylo@cs.hut.fi> 4 * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland 5 * All rights reserved 6 * Versions of malloc and friends that check their results, and never return 7 * failure (they call fatal if they encounter an error). 8 * 9 * As far as I am concerned, the code I have written for this software 10 * can be used freely for any purpose. Any derived versions of this 11 * software must be clearly marked as such, and if the derived work is 12 * incompatible with the protocol description in the RFC file, it must be 13 * called by a name other than "ssh" or "Secure Shell". 14 */ 15 16 #include "includes.h" 17 18 #include <stdarg.h> 19 #ifdef HAVE_STDINT_H 20 # include <stdint.h> 21 #endif 22 #include <stdio.h> 23 #include <stdlib.h> 24 #include <string.h> 25 26 #include "xmalloc.h" 27 #include "log.h" 28 29 #if defined(__OpenBSD__) 30 char *malloc_options = "S"; 31 #endif /* __OpenBSD__ */ 32 33 void * 34 xmalloc(size_t size) 35 { 36 void *ptr; 37 38 if (size == 0) 39 fatal("xmalloc: zero size"); 40 ptr = malloc(size); 41 if (ptr == NULL) 42 fatal("xmalloc: out of memory (allocating %zu bytes)", size); 43 return ptr; 44 } 45 46 void * 47 xcalloc(size_t nmemb, size_t size) 48 { 49 void *ptr; 50 51 if (size == 0 || nmemb == 0) 52 fatal("xcalloc: zero size"); 53 if (SIZE_MAX / nmemb < size) 54 fatal("xcalloc: nmemb * size > SIZE_MAX"); 55 ptr = calloc(nmemb, size); 56 if (ptr == NULL) 57 fatal("xcalloc: out of memory (allocating %zu bytes)", 58 size * nmemb); 59 return ptr; 60 } 61 62 void * 63 xreallocarray(void *ptr, size_t nmemb, size_t size) 64 { 65 void *new_ptr; 66 67 new_ptr = reallocarray(ptr, nmemb, size); 68 if (new_ptr == NULL) 69 fatal("xreallocarray: out of memory (%zu elements of %zu bytes)", 70 nmemb, size); 71 return new_ptr; 72 } 73 74 void * 75 xrecallocarray(void *ptr, size_t onmemb, size_t nmemb, size_t size) 76 { 77 void *new_ptr; 78 79 new_ptr = recallocarray(ptr, onmemb, nmemb, size); 80 if (new_ptr == NULL) 81 fatal("xrecallocarray: out of memory (%zu elements of %zu bytes)", 82 nmemb, size); 83 return new_ptr; 84 } 85 86 char * 87 xstrdup(const char *str) 88 { 89 size_t len; 90 char *cp; 91 92 len = strlen(str) + 1; 93 cp = xmalloc(len); 94 strlcpy(cp, str, len); 95 return cp; 96 } 97 98 int 99 xvasprintf(char **ret, const char *fmt, va_list ap) 100 { 101 int i; 102 103 i = vasprintf(ret, fmt, ap); 104 if (i < 0 || *ret == NULL) 105 fatal("xvasprintf: could not allocate memory"); 106 return i; 107 } 108 109 int 110 xasprintf(char **ret, const char *fmt, ...) 111 { 112 va_list ap; 113 int i; 114 115 va_start(ap, fmt); 116 i = xvasprintf(ret, fmt, ap); 117 va_end(ap); 118 return i; 119 } 120