1 /* $OpenBSD: xmalloc.c,v 1.12 2019年06月28日 05:44:09 deraadt 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 <err.h> 17#include <stdarg.h> 18#include <stdint.h> 19#include <stdio.h> 20#include <stdlib.h> 21#include <string.h> 22 23#include "xmalloc.h" 24 25 void * 26 xmalloc(size_t size) 27{ 28 void *ptr; 29 30 if (size == 0) 31 errx(1, "xmalloc: zero size"); 32 ptr = malloc(size); 33 if (ptr == NULL) 34 err(1, "xmalloc: allocating %zu bytes", size); 35 return ptr; 36} 37 38 void * 39 xcalloc(size_t nmemb, size_t size) 40{ 41 void *ptr; 42 43 if (size == 0 || nmemb == 0) 44 errx(1, "xcalloc: zero size"); 45 ptr = calloc(nmemb, size); 46 if (ptr == NULL) 47 err(1, "xcalloc: allocating %zu * %zu bytes", nmemb, size); 48 return ptr; 49} 50 51 void * 52 xreallocarray(void *ptr, size_t nmemb, size_t size) 53{ 54 void *new_ptr; 55 56 new_ptr = reallocarray(ptr, nmemb, size); 57 if (new_ptr == NULL) 58 err(1, "xreallocarray: allocating %zu * %zu bytes", 59 nmemb, size); 60 return new_ptr; 61} 62 63 char * 64 xstrdup(const char *str) 65{ 66 char *cp; 67 68 if ((cp = strdup(str)) == NULL) 69 err(1, "xstrdup"); 70 return cp; 71} 72 73 int 74 xasprintf(char **ret, const char *fmt, ...) 75{ 76 va_list ap; 77 int i; 78 79 va_start(ap, fmt); 80 i = vasprintf(ret, fmt, ap); 81 va_end(ap); 82 83 if (i == -1) 84 err(1, "xasprintf"); 85 86 return i; 87} 88