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