xref: /openbsd/usr.bin/rcs/xmalloc.c (revision d415bd75)
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