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