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 <stdarg.h> 17 #include <stdint.h> 18 #include <stdio.h> 19 #include <stdlib.h> 20 #include <string.h> 21 22 #include "xmalloc.h" 23 #include "log.h" 24 25 char *malloc_options = "S"; 26 27 void * 28 xmalloc(size_t size) 29 { 30 void *ptr; 31 32 if (size == 0) 33 fatal("xmalloc: zero size"); 34 ptr = malloc(size); 35 if (ptr == NULL) 36 fatal("xmalloc: out of memory (allocating %zu bytes)", size); 37 return ptr; 38 } 39 40 void * 41 xcalloc(size_t nmemb, size_t size) 42 { 43 void *ptr; 44 45 if (size == 0 || nmemb == 0) 46 fatal("xcalloc: zero size"); 47 if (SIZE_MAX / nmemb < size) 48 fatal("xcalloc: nmemb * size > SIZE_MAX"); 49 ptr = calloc(nmemb, size); 50 if (ptr == NULL) 51 fatal("xcalloc: out of memory (allocating %zu bytes)", 52 size * nmemb); 53 return ptr; 54 } 55 56 void * 57 xreallocarray(void *ptr, size_t nmemb, size_t size) 58 { 59 void *new_ptr; 60 61 new_ptr = reallocarray(ptr, nmemb, size); 62 if (new_ptr == NULL) 63 fatal("xreallocarray: out of memory (%zu elements of %zu bytes)", 64 nmemb, size); 65 return new_ptr; 66 } 67 68 void * 69 xrecallocarray(void *ptr, size_t onmemb, size_t nmemb, size_t size) 70 { 71 void *new_ptr; 72 73 new_ptr = recallocarray(ptr, onmemb, nmemb, size); 74 if (new_ptr == NULL) 75 fatal("xrecallocarray: out of memory (%zu elements of %zu bytes)", 76 nmemb, size); 77 return new_ptr; 78 } 79 80 char * 81 xstrdup(const char *str) 82 { 83 size_t len; 84 char *cp; 85 86 len = strlen(str) + 1; 87 cp = xmalloc(len); 88 strlcpy(cp, str, len); 89 return cp; 90 } 91 92 int 93 xvasprintf(char **ret, const char *fmt, va_list ap) 94 { 95 int i; 96 97 i = vasprintf(ret, fmt, ap); 98 if (i < 0 || *ret == NULL) 99 fatal("xvasprintf: could not allocate memory"); 100 return i; 101 } 102 103 int 104 xasprintf(char **ret, const char *fmt, ...) 105 { 106 va_list ap; 107 int i; 108 109 va_start(ap, fmt); 110 i = xvasprintf(ret, fmt, ap); 111 va_end(ap); 112 return i; 113 } 114