1 |
|
|
/* $OpenBSD: xmalloc.c,v 1.11 2016/11/17 10:06:08 nicm Exp $ */ |
2 |
|
|
|
3 |
|
|
/* |
4 |
|
|
* Author: Tatu Ylonen <ylo@cs.hut.fi> |
5 |
|
|
* Copyright (c) 1995 Tatu Ylonen <ylo@cs.hut.fi>, Espoo, Finland |
6 |
|
|
* All rights reserved |
7 |
|
|
* Versions of malloc and friends that check their results, and never return |
8 |
|
|
* failure (they call fatalx if they encounter an error). |
9 |
|
|
* |
10 |
|
|
* As far as I am concerned, the code I have written for this software |
11 |
|
|
* can be used freely for any purpose. Any derived versions of this |
12 |
|
|
* software must be clearly marked as such, and if the derived work is |
13 |
|
|
* incompatible with the protocol description in the RFC file, it must be |
14 |
|
|
* called by a name other than "ssh" or "Secure Shell". |
15 |
|
|
*/ |
16 |
|
|
|
17 |
|
|
#include <errno.h> |
18 |
|
|
#include <limits.h> |
19 |
|
|
#include <stdint.h> |
20 |
|
|
#include <stdio.h> |
21 |
|
|
#include <stdlib.h> |
22 |
|
|
#include <string.h> |
23 |
|
|
|
24 |
|
|
#include "tmux.h" |
25 |
|
|
|
26 |
|
|
void * |
27 |
|
|
xmalloc(size_t size) |
28 |
|
|
{ |
29 |
|
|
void *ptr; |
30 |
|
|
|
31 |
✗✓ |
176 |
if (size == 0) |
32 |
|
|
fatalx("xmalloc: zero size"); |
33 |
|
88 |
ptr = malloc(size); |
34 |
✗✓ |
88 |
if (ptr == NULL) |
35 |
|
|
fatalx("xmalloc: allocating %zu bytes: %s", |
36 |
|
|
size, strerror(errno)); |
37 |
|
88 |
return ptr; |
38 |
|
|
} |
39 |
|
|
|
40 |
|
|
void * |
41 |
|
|
xcalloc(size_t nmemb, size_t size) |
42 |
|
|
{ |
43 |
|
|
void *ptr; |
44 |
|
|
|
45 |
✗✓ |
1660 |
if (size == 0 || nmemb == 0) |
46 |
|
|
fatalx("xcalloc: zero size"); |
47 |
|
830 |
ptr = calloc(nmemb, size); |
48 |
✗✓ |
830 |
if (ptr == NULL) |
49 |
|
|
fatalx("xcalloc: allocating %zu * %zu bytes: %s", |
50 |
|
|
nmemb, size, strerror(errno)); |
51 |
|
830 |
return ptr; |
52 |
|
|
} |
53 |
|
|
|
54 |
|
|
void * |
55 |
|
|
xrealloc(void *ptr, size_t size) |
56 |
|
|
{ |
57 |
|
|
return xreallocarray(ptr, 1, size); |
58 |
|
|
} |
59 |
|
|
|
60 |
|
|
void * |
61 |
|
|
xreallocarray(void *ptr, size_t nmemb, size_t size) |
62 |
|
|
{ |
63 |
|
|
void *new_ptr; |
64 |
|
|
|
65 |
✗✓ |
180 |
if (nmemb == 0 || size == 0) |
66 |
|
|
fatalx("xreallocarray: zero size"); |
67 |
|
90 |
new_ptr = reallocarray(ptr, nmemb, size); |
68 |
✗✓ |
90 |
if (new_ptr == NULL) |
69 |
|
|
fatalx("xreallocarray: allocating %zu * %zu bytes: %s", |
70 |
|
|
nmemb, size, strerror(errno)); |
71 |
|
90 |
return new_ptr; |
72 |
|
|
} |
73 |
|
|
|
74 |
|
|
char * |
75 |
|
|
xstrdup(const char *str) |
76 |
|
|
{ |
77 |
|
|
char *cp; |
78 |
|
|
|
79 |
✗✓ |
2284 |
if ((cp = strdup(str)) == NULL) |
80 |
|
|
fatalx("xstrdup: %s", strerror(errno)); |
81 |
|
1142 |
return cp; |
82 |
|
|
} |
83 |
|
|
|
84 |
|
|
char * |
85 |
|
|
xstrndup(const char *str, size_t maxlen) |
86 |
|
|
{ |
87 |
|
|
char *cp; |
88 |
|
|
|
89 |
|
|
if ((cp = strndup(str, maxlen)) == NULL) |
90 |
|
|
fatalx("xstrndup: %s", strerror(errno)); |
91 |
|
|
return cp; |
92 |
|
|
} |
93 |
|
|
|
94 |
|
|
int |
95 |
|
|
xasprintf(char **ret, const char *fmt, ...) |
96 |
|
|
{ |
97 |
|
28 |
va_list ap; |
98 |
|
|
int i; |
99 |
|
|
|
100 |
|
14 |
va_start(ap, fmt); |
101 |
|
14 |
i = xvasprintf(ret, fmt, ap); |
102 |
|
14 |
va_end(ap); |
103 |
|
|
|
104 |
|
14 |
return i; |
105 |
|
14 |
} |
106 |
|
|
|
107 |
|
|
int |
108 |
|
|
xvasprintf(char **ret, const char *fmt, va_list ap) |
109 |
|
|
{ |
110 |
|
|
int i; |
111 |
|
|
|
112 |
|
204 |
i = vasprintf(ret, fmt, ap); |
113 |
|
|
|
114 |
✓✗✗✓
|
204 |
if (i < 0 || *ret == NULL) |
115 |
|
|
fatalx("xasprintf: %s", strerror(errno)); |
116 |
|
|
|
117 |
|
102 |
return i; |
118 |
|
|
} |
119 |
|
|
|
120 |
|
|
int |
121 |
|
|
xsnprintf(char *str, size_t len, const char *fmt, ...) |
122 |
|
|
{ |
123 |
|
2 |
va_list ap; |
124 |
|
|
int i; |
125 |
|
|
|
126 |
|
1 |
va_start(ap, fmt); |
127 |
|
1 |
i = xvsnprintf(str, len, fmt, ap); |
128 |
|
1 |
va_end(ap); |
129 |
|
|
|
130 |
|
1 |
return i; |
131 |
|
1 |
} |
132 |
|
|
|
133 |
|
|
int |
134 |
|
|
xvsnprintf(char *str, size_t len, const char *fmt, va_list ap) |
135 |
|
|
{ |
136 |
|
|
int i; |
137 |
|
|
|
138 |
✗✓ |
2 |
if (len > INT_MAX) |
139 |
|
|
fatalx("xsnprintf: len > INT_MAX"); |
140 |
|
|
|
141 |
|
1 |
i = vsnprintf(str, len, fmt, ap); |
142 |
|
|
|
143 |
✓✗✗✓
|
2 |
if (i < 0 || i >= (int)len) |
144 |
|
|
fatalx("xsnprintf: overflow"); |
145 |
|
|
|
146 |
|
1 |
return i; |
147 |
|
|
} |