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