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