xref: /openssh-portable/xmalloc.c (revision 5d19626a)
1 /* $OpenBSD: xmalloc.c,v 1.22 2006/07/10 16:37:36 stevesk 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 "includes.h"
17 
18 #include <stdarg.h>
19 
20 #include "xmalloc.h"
21 #include "log.h"
22 
23 void *
24 xmalloc(size_t size)
25 {
26 	void *ptr;
27 
28 	if (size == 0)
29 		fatal("xmalloc: zero size");
30 	ptr = malloc(size);
31 	if (ptr == NULL)
32 		fatal("xmalloc: out of memory (allocating %lu bytes)", (u_long) size);
33 	return ptr;
34 }
35 
36 void *
37 xcalloc(size_t nmemb, size_t size)
38 {
39 	void *ptr;
40 
41 	if (size == 0 || nmemb == 0)
42 		fatal("xcalloc: zero size");
43 	if (SIZE_T_MAX / nmemb < size)
44 		fatal("xcalloc: nmemb * size > SIZE_T_MAX");
45 	ptr = calloc(nmemb, size);
46 	if (ptr == NULL)
47 		fatal("xcalloc: out of memory (allocating %lu bytes)",
48 		    (u_long)(size * nmemb));
49 	return ptr;
50 }
51 
52 void *
53 xrealloc(void *ptr, size_t nmemb, size_t size)
54 {
55 	void *new_ptr;
56 	size_t new_size = nmemb * size;
57 
58 	if (new_size == 0)
59 		fatal("xrealloc: zero size");
60 	if (SIZE_T_MAX / nmemb < size)
61 		fatal("xrealloc: nmemb * size > SIZE_T_MAX");
62 	if (ptr == NULL)
63 		new_ptr = malloc(new_size);
64 	else
65 		new_ptr = realloc(ptr, new_size);
66 	if (new_ptr == NULL)
67 		fatal("xrealloc: out of memory (new_size %lu bytes)",
68 		    (u_long) new_size);
69 	return new_ptr;
70 }
71 
72 void
73 xfree(void *ptr)
74 {
75 	if (ptr == NULL)
76 		fatal("xfree: NULL pointer given as argument");
77 	free(ptr);
78 }
79 
80 char *
81 xstrdup(const char *str)
82 {
83 	size_t len;
84 	char *cp;
85 
86 	len = strlen(str) + 1;
87 	cp = xmalloc(len);
88 	strlcpy(cp, str, len);
89 	return cp;
90 }
91 
92 int
93 xasprintf(char **ret, const char *fmt, ...)
94 {
95 	va_list ap;
96 	int i;
97 
98 	va_start(ap, fmt);
99 	i = vasprintf(ret, fmt, ap);
100 	va_end(ap);
101 
102 	if (i < 0 || *ret == NULL)
103 		fatal("xasprintf: could not allocate memory");
104 
105 	return (i);
106 }
107