1 /* $OpenBSD: xmalloc.c,v 1.10 2016/04/04 16:19:43 nicm Exp $ */ 2 3 /* 4 * Author: Tatu Ylonen <ylo@cs.hut.fi> 5 * Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland 6 * All rights reserved 7 * Versions of malloc and friends that check their results, and never return 8 * failure (they call fatalx if they encounter an error). 9 * 10 * As far as I am concerned, the code I have written for this software 11 * can be used freely for any purpose. Any derived versions of this 12 * software must be clearly marked as such, and if the derived work is 13 * incompatible with the protocol description in the RFC file, it must be 14 * called by a name other than "ssh" or "Secure Shell". 15 */ 16 17 #include <errno.h> 18 #include <limits.h> 19 #include <stdint.h> 20 #include <stdio.h> 21 #include <stdlib.h> 22 #include <string.h> 23 24 #include "tmux.h" 25 26 void * 27 xmalloc(size_t size) 28 { 29 void *ptr; 30 31 if (size == 0) 32 fatalx("xmalloc: zero size"); 33 ptr = malloc(size); 34 if (ptr == NULL) 35 fatalx("xmalloc: allocating %zu bytes: %s", 36 size, strerror(errno)); 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 fatalx("xcalloc: zero size"); 47 ptr = calloc(nmemb, size); 48 if (ptr == NULL) 49 fatalx("xcalloc: allocating %zu * %zu bytes: %s", 50 nmemb, size, strerror(errno)); 51 return ptr; 52 } 53 54 void * 55 xrealloc(void *ptr, size_t size) 56 { 57 return xreallocarray(ptr, 1, size); 58 } 59 60 void * 61 xreallocarray(void *ptr, size_t nmemb, size_t size) 62 { 63 void *new_ptr; 64 65 if (nmemb == 0 || size == 0) 66 fatalx("xreallocarray: zero size"); 67 new_ptr = reallocarray(ptr, nmemb, size); 68 if (new_ptr == NULL) 69 fatalx("xreallocarray: allocating %zu * %zu bytes: %s", 70 nmemb, size, strerror(errno)); 71 return new_ptr; 72 } 73 74 char * 75 xstrdup(const char *str) 76 { 77 char *cp; 78 79 if ((cp = strdup(str)) == NULL) 80 fatalx("xstrdup: %s", strerror(errno)); 81 return cp; 82 } 83 84 int 85 xasprintf(char **ret, const char *fmt, ...) 86 { 87 va_list ap; 88 int i; 89 90 va_start(ap, fmt); 91 i = xvasprintf(ret, fmt, ap); 92 va_end(ap); 93 94 return i; 95 } 96 97 int 98 xvasprintf(char **ret, const char *fmt, va_list ap) 99 { 100 int i; 101 102 i = vasprintf(ret, fmt, ap); 103 104 if (i < 0 || *ret == NULL) 105 fatalx("xasprintf: %s", strerror(errno)); 106 107 return i; 108 } 109 110 int 111 xsnprintf(char *str, size_t len, const char *fmt, ...) 112 { 113 va_list ap; 114 int i; 115 116 va_start(ap, fmt); 117 i = xvsnprintf(str, len, fmt, ap); 118 va_end(ap); 119 120 return i; 121 } 122 123 int 124 xvsnprintf(char *str, size_t len, const char *fmt, va_list ap) 125 { 126 int i; 127 128 if (len > INT_MAX) 129 fatalx("xsnprintf: len > INT_MAX"); 130 131 i = vsnprintf(str, len, fmt, ap); 132 133 if (i < 0 || i >= (int)len) 134 fatalx("xsnprintf: overflow"); 135 136 return i; 137 } 138