File: | build-scan/../src/basic/utf8.c |
Warning: | line 287, column 37 Use of zero-allocated memory |
Press '?' to see keyboard shortcuts
Keyboard shortcuts:
1 | /* SPDX-License-Identifier: LGPL-2.1+ */ | ||||
2 | |||||
3 | /* Parts of this file are based on the GLIB utf8 validation functions. The | ||||
4 | * original license text follows. */ | ||||
5 | |||||
6 | /* gutf8.c - Operations on UTF-8 strings. | ||||
7 | * | ||||
8 | * Copyright (C) 1999 Tom Tromey | ||||
9 | * Copyright (C) 2000 Red Hat, Inc. | ||||
10 | * | ||||
11 | * This library is free software; you can redistribute it and/or | ||||
12 | * modify it under the terms of the GNU Library General Public | ||||
13 | * License as published by the Free Software Foundation; either | ||||
14 | * version 2 of the License, or (at your option) any later version. | ||||
15 | * | ||||
16 | * This library is distributed in the hope that it will be useful, | ||||
17 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
18 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||||
19 | * Library General Public License for more details. | ||||
20 | * | ||||
21 | * You should have received a copy of the GNU Library General Public | ||||
22 | * License along with this library; if not, write to the Free Software | ||||
23 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||||
24 | */ | ||||
25 | |||||
26 | #include <errno(*__errno_location ()).h> | ||||
27 | #include <stdbool.h> | ||||
28 | #include <stdlib.h> | ||||
29 | #include <string.h> | ||||
30 | |||||
31 | #include "alloc-util.h" | ||||
32 | #include "gunicode.h" | ||||
33 | #include "hexdecoct.h" | ||||
34 | #include "macro.h" | ||||
35 | #include "utf8.h" | ||||
36 | |||||
37 | bool_Bool unichar_is_valid(char32_t ch) { | ||||
38 | |||||
39 | if (ch >= 0x110000) /* End of unicode space */ | ||||
40 | return false0; | ||||
41 | if ((ch & 0xFFFFF800) == 0xD800) /* Reserved area for UTF-16 */ | ||||
42 | return false0; | ||||
43 | if ((ch >= 0xFDD0) && (ch <= 0xFDEF)) /* Reserved */ | ||||
44 | return false0; | ||||
45 | if ((ch & 0xFFFE) == 0xFFFE) /* BOM (Byte Order Mark) */ | ||||
46 | return false0; | ||||
47 | |||||
48 | return true1; | ||||
49 | } | ||||
50 | |||||
51 | static bool_Bool unichar_is_control(char32_t ch) { | ||||
52 | |||||
53 | /* | ||||
54 | 0 to ' '-1 is the C0 range. | ||||
55 | DEL=0x7F, and DEL+1 to 0x9F is C1 range. | ||||
56 | '\t' is in C0 range, but more or less harmless and commonly used. | ||||
57 | */ | ||||
58 | |||||
59 | return (ch < ' ' && !IN_SET(ch, '\t', '\n')({ _Bool _found = 0; static __attribute__ ((unused)) char _static_assert__macros_need_to_be_extended [20 - sizeof((int[]){'\t', '\n'})/sizeof(int)]; switch(ch) { case '\t': case '\n': _found = 1; break; default: break; } _found ; })) || | ||||
60 | (0x7F <= ch && ch <= 0x9F); | ||||
61 | } | ||||
62 | |||||
63 | /* count of characters used to encode one unicode char */ | ||||
64 | static int utf8_encoded_expected_len(const char *str) { | ||||
65 | unsigned char c; | ||||
66 | |||||
67 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 67, __PRETTY_FUNCTION__ ); } while (0); | ||||
68 | |||||
69 | c = (unsigned char) str[0]; | ||||
70 | if (c < 0x80) | ||||
71 | return 1; | ||||
72 | if ((c & 0xe0) == 0xc0) | ||||
73 | return 2; | ||||
74 | if ((c & 0xf0) == 0xe0) | ||||
75 | return 3; | ||||
76 | if ((c & 0xf8) == 0xf0) | ||||
77 | return 4; | ||||
78 | if ((c & 0xfc) == 0xf8) | ||||
79 | return 5; | ||||
80 | if ((c & 0xfe) == 0xfc) | ||||
81 | return 6; | ||||
82 | |||||
83 | return 0; | ||||
84 | } | ||||
85 | |||||
86 | /* decode one unicode char */ | ||||
87 | int utf8_encoded_to_unichar(const char *str, char32_t *ret_unichar) { | ||||
88 | char32_t unichar; | ||||
89 | int len, i; | ||||
90 | |||||
91 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 91, __PRETTY_FUNCTION__ ); } while (0); | ||||
92 | |||||
93 | len = utf8_encoded_expected_len(str); | ||||
94 | |||||
95 | switch (len) { | ||||
96 | case 1: | ||||
97 | *ret_unichar = (char32_t)str[0]; | ||||
98 | return 0; | ||||
99 | case 2: | ||||
100 | unichar = str[0] & 0x1f; | ||||
101 | break; | ||||
102 | case 3: | ||||
103 | unichar = (char32_t)str[0] & 0x0f; | ||||
104 | break; | ||||
105 | case 4: | ||||
106 | unichar = (char32_t)str[0] & 0x07; | ||||
107 | break; | ||||
108 | case 5: | ||||
109 | unichar = (char32_t)str[0] & 0x03; | ||||
110 | break; | ||||
111 | case 6: | ||||
112 | unichar = (char32_t)str[0] & 0x01; | ||||
113 | break; | ||||
114 | default: | ||||
115 | return -EINVAL22; | ||||
116 | } | ||||
117 | |||||
118 | for (i = 1; i < len; i++) { | ||||
119 | if (((char32_t)str[i] & 0xc0) != 0x80) | ||||
120 | return -EINVAL22; | ||||
121 | unichar <<= 6; | ||||
122 | unichar |= (char32_t)str[i] & 0x3f; | ||||
123 | } | ||||
124 | |||||
125 | *ret_unichar = unichar; | ||||
126 | |||||
127 | return 0; | ||||
128 | } | ||||
129 | |||||
130 | bool_Bool utf8_is_printable_newline(const char* str, size_t length, bool_Bool newline) { | ||||
131 | const char *p; | ||||
132 | |||||
133 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 133, __PRETTY_FUNCTION__ ); } while (0); | ||||
134 | |||||
135 | for (p = str; length;) { | ||||
136 | int encoded_len, r; | ||||
137 | char32_t val; | ||||
138 | |||||
139 | encoded_len = utf8_encoded_valid_unichar(p); | ||||
140 | if (encoded_len < 0 || | ||||
141 | (size_t) encoded_len > length) | ||||
142 | return false0; | ||||
143 | |||||
144 | r = utf8_encoded_to_unichar(p, &val); | ||||
145 | if (r < 0 || | ||||
146 | unichar_is_control(val) || | ||||
147 | (!newline && val == '\n')) | ||||
148 | return false0; | ||||
149 | |||||
150 | length -= encoded_len; | ||||
151 | p += encoded_len; | ||||
152 | } | ||||
153 | |||||
154 | return true1; | ||||
155 | } | ||||
156 | |||||
157 | const char *utf8_is_valid(const char *str) { | ||||
158 | const uint8_t *p; | ||||
159 | |||||
160 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 160, __PRETTY_FUNCTION__ ); } while (0); | ||||
161 | |||||
162 | for (p = (const uint8_t*) str; *p; ) { | ||||
163 | int len; | ||||
164 | |||||
165 | len = utf8_encoded_valid_unichar((const char *)p); | ||||
166 | if (len < 0) | ||||
167 | return NULL((void*)0); | ||||
168 | |||||
169 | p += len; | ||||
170 | } | ||||
171 | |||||
172 | return str; | ||||
173 | } | ||||
174 | |||||
175 | char *utf8_escape_invalid(const char *str) { | ||||
176 | char *p, *s; | ||||
177 | |||||
178 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 178, __PRETTY_FUNCTION__ ); } while (0); | ||||
179 | |||||
180 | p = s = malloc(strlen(str) * 4 + 1); | ||||
181 | if (!p) | ||||
182 | return NULL((void*)0); | ||||
183 | |||||
184 | while (*str) { | ||||
185 | int len; | ||||
186 | |||||
187 | len = utf8_encoded_valid_unichar(str); | ||||
188 | if (len > 0) { | ||||
189 | s = mempcpy(s, str, len); | ||||
190 | str += len; | ||||
191 | } else { | ||||
192 | s = stpcpy(s, UTF8_REPLACEMENT_CHARACTER"\xef\xbf\xbd"); | ||||
193 | str += 1; | ||||
194 | } | ||||
195 | } | ||||
196 | |||||
197 | *s = '\0'; | ||||
198 | |||||
199 | return p; | ||||
200 | } | ||||
201 | |||||
202 | char *utf8_escape_non_printable(const char *str) { | ||||
203 | char *p, *s; | ||||
204 | |||||
205 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 205, __PRETTY_FUNCTION__ ); } while (0); | ||||
206 | |||||
207 | p = s = malloc(strlen(str) * 4 + 1); | ||||
208 | if (!p) | ||||
209 | return NULL((void*)0); | ||||
210 | |||||
211 | while (*str) { | ||||
212 | int len; | ||||
213 | |||||
214 | len = utf8_encoded_valid_unichar(str); | ||||
215 | if (len > 0) { | ||||
216 | if (utf8_is_printable(str, len)utf8_is_printable_newline(str, len, 1)) { | ||||
217 | s = mempcpy(s, str, len); | ||||
218 | str += len; | ||||
219 | } else { | ||||
220 | while (len > 0) { | ||||
221 | *(s++) = '\\'; | ||||
222 | *(s++) = 'x'; | ||||
223 | *(s++) = hexchar((int) *str >> 4); | ||||
224 | *(s++) = hexchar((int) *str); | ||||
225 | |||||
226 | str += 1; | ||||
227 | len--; | ||||
228 | } | ||||
229 | } | ||||
230 | } else { | ||||
231 | s = stpcpy(s, UTF8_REPLACEMENT_CHARACTER"\xef\xbf\xbd"); | ||||
232 | str += 1; | ||||
233 | } | ||||
234 | } | ||||
235 | |||||
236 | *s = '\0'; | ||||
237 | |||||
238 | return p; | ||||
239 | } | ||||
240 | |||||
241 | char *ascii_is_valid(const char *str) { | ||||
242 | const char *p; | ||||
243 | |||||
244 | /* Check whether the string consists of valid ASCII bytes, | ||||
245 | * i.e values between 0 and 127, inclusive. */ | ||||
246 | |||||
247 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 247, __PRETTY_FUNCTION__ ); } while (0); | ||||
248 | |||||
249 | for (p = str; *p; p++) | ||||
250 | if ((unsigned char) *p >= 128) | ||||
251 | return NULL((void*)0); | ||||
252 | |||||
253 | return (char*) str; | ||||
254 | } | ||||
255 | |||||
256 | char *ascii_is_valid_n(const char *str, size_t len) { | ||||
257 | size_t i; | ||||
258 | |||||
259 | /* Very similar to ascii_is_valid(), but checks exactly len | ||||
260 | * bytes and rejects any NULs in that range. */ | ||||
261 | |||||
262 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 262, __PRETTY_FUNCTION__ ); } while (0); | ||||
263 | |||||
264 | for (i = 0; i < len; i++) | ||||
265 | if ((unsigned char) str[i] >= 128 || str[i] == 0) | ||||
266 | return NULL((void*)0); | ||||
267 | |||||
268 | return (char*) str; | ||||
269 | } | ||||
270 | |||||
271 | /** | ||||
272 | * utf8_encode_unichar() - Encode single UCS-4 character as UTF-8 | ||||
273 | * @out_utf8: output buffer of at least 4 bytes or NULL | ||||
274 | * @g: UCS-4 character to encode | ||||
275 | * | ||||
276 | * This encodes a single UCS-4 character as UTF-8 and writes it into @out_utf8. | ||||
277 | * The length of the character is returned. It is not zero-terminated! If the | ||||
278 | * output buffer is NULL, only the length is returned. | ||||
279 | * | ||||
280 | * Returns: The length in bytes that the UTF-8 representation does or would | ||||
281 | * occupy. | ||||
282 | */ | ||||
283 | size_t utf8_encode_unichar(char *out_utf8, char32_t g) { | ||||
284 | |||||
285 | if (g < (1 << 7)) { | ||||
286 | if (out_utf8
| ||||
287 | out_utf8[0] = g & 0x7f; | ||||
| |||||
288 | return 1; | ||||
289 | } else if (g < (1 << 11)) { | ||||
290 | if (out_utf8) { | ||||
291 | out_utf8[0] = 0xc0 | ((g >> 6) & 0x1f); | ||||
292 | out_utf8[1] = 0x80 | (g & 0x3f); | ||||
293 | } | ||||
294 | return 2; | ||||
295 | } else if (g < (1 << 16)) { | ||||
296 | if (out_utf8) { | ||||
297 | out_utf8[0] = 0xe0 | ((g >> 12) & 0x0f); | ||||
298 | out_utf8[1] = 0x80 | ((g >> 6) & 0x3f); | ||||
299 | out_utf8[2] = 0x80 | (g & 0x3f); | ||||
300 | } | ||||
301 | return 3; | ||||
302 | } else if (g < (1 << 21)) { | ||||
303 | if (out_utf8) { | ||||
304 | out_utf8[0] = 0xf0 | ((g >> 18) & 0x07); | ||||
305 | out_utf8[1] = 0x80 | ((g >> 12) & 0x3f); | ||||
306 | out_utf8[2] = 0x80 | ((g >> 6) & 0x3f); | ||||
307 | out_utf8[3] = 0x80 | (g & 0x3f); | ||||
308 | } | ||||
309 | return 4; | ||||
310 | } | ||||
311 | |||||
312 | return 0; | ||||
313 | } | ||||
314 | |||||
315 | char *utf16_to_utf8(const void *s, size_t length) { | ||||
316 | const uint8_t *f; | ||||
317 | char *r, *t; | ||||
318 | |||||
319 | r = new(char, (length * 4 + 1) / 2 + 1)((char*) malloc_multiply(sizeof(char), ((length * 4 + 1) / 2 + 1))); | ||||
| |||||
320 | if (!r) | ||||
321 | return NULL((void*)0); | ||||
322 | |||||
323 | f = s; | ||||
324 | t = r; | ||||
325 | |||||
326 | while (f < (const uint8_t*) s + length) { | ||||
327 | char16_t w1, w2; | ||||
328 | |||||
329 | /* see RFC 2781 section 2.2 */ | ||||
330 | |||||
331 | w1 = f[1] << 8 | f[0]; | ||||
332 | f += 2; | ||||
333 | |||||
334 | if (!utf16_is_surrogate(w1)) { | ||||
335 | t += utf8_encode_unichar(t, w1); | ||||
336 | |||||
337 | continue; | ||||
338 | } | ||||
339 | |||||
340 | if (utf16_is_trailing_surrogate(w1)) | ||||
341 | continue; | ||||
342 | else if (f >= (const uint8_t*) s + length) | ||||
343 | break; | ||||
344 | |||||
345 | w2 = f[1] << 8 | f[0]; | ||||
346 | f += 2; | ||||
347 | |||||
348 | if (!utf16_is_trailing_surrogate(w2)) { | ||||
349 | f -= 2; | ||||
350 | continue; | ||||
351 | } | ||||
352 | |||||
353 | t += utf8_encode_unichar(t, utf16_surrogate_pair_to_unichar(w1, w2)); | ||||
354 | } | ||||
355 | |||||
356 | *t = 0; | ||||
357 | return r; | ||||
358 | } | ||||
359 | |||||
360 | /* expected size used to encode one unicode char */ | ||||
361 | static int utf8_unichar_to_encoded_len(char32_t unichar) { | ||||
362 | |||||
363 | if (unichar < 0x80) | ||||
364 | return 1; | ||||
365 | if (unichar < 0x800) | ||||
366 | return 2; | ||||
367 | if (unichar < 0x10000) | ||||
368 | return 3; | ||||
369 | if (unichar < 0x200000) | ||||
370 | return 4; | ||||
371 | if (unichar < 0x4000000) | ||||
372 | return 5; | ||||
373 | |||||
374 | return 6; | ||||
375 | } | ||||
376 | |||||
377 | /* validate one encoded unicode char and return its length */ | ||||
378 | int utf8_encoded_valid_unichar(const char *str) { | ||||
379 | int len, i, r; | ||||
380 | char32_t unichar; | ||||
381 | |||||
382 | assert(str)do { if ((__builtin_expect(!!(!(str)),0))) log_assert_failed_realm (LOG_REALM_SYSTEMD, ("str"), "../src/basic/utf8.c", 382, __PRETTY_FUNCTION__ ); } while (0); | ||||
383 | |||||
384 | len = utf8_encoded_expected_len(str); | ||||
385 | if (len == 0) | ||||
386 | return -EINVAL22; | ||||
387 | |||||
388 | /* ascii is valid */ | ||||
389 | if (len == 1) | ||||
390 | return 1; | ||||
391 | |||||
392 | /* check if expected encoded chars are available */ | ||||
393 | for (i = 0; i < len; i++) | ||||
394 | if ((str[i] & 0x80) != 0x80) | ||||
395 | return -EINVAL22; | ||||
396 | |||||
397 | r = utf8_encoded_to_unichar(str, &unichar); | ||||
398 | if (r < 0) | ||||
399 | return r; | ||||
400 | |||||
401 | /* check if encoded length matches encoded value */ | ||||
402 | if (utf8_unichar_to_encoded_len(unichar) != len) | ||||
403 | return -EINVAL22; | ||||
404 | |||||
405 | /* check if value has valid range */ | ||||
406 | if (!unichar_is_valid(unichar)) | ||||
407 | return -EINVAL22; | ||||
408 | |||||
409 | return len; | ||||
410 | } | ||||
411 | |||||
412 | size_t utf8_n_codepoints(const char *str) { | ||||
413 | size_t n = 0; | ||||
414 | |||||
415 | /* Returns the number of UTF-8 codepoints in this string, or (size_t) -1 if the string is not valid UTF-8. */ | ||||
416 | |||||
417 | while (*str != 0) { | ||||
418 | int k; | ||||
419 | |||||
420 | k = utf8_encoded_valid_unichar(str); | ||||
421 | if (k < 0) | ||||
422 | return (size_t) -1; | ||||
423 | |||||
424 | str += k; | ||||
425 | n++; | ||||
426 | } | ||||
427 | |||||
428 | return n; | ||||
429 | } | ||||
430 | |||||
431 | size_t utf8_console_width(const char *str) { | ||||
432 | size_t n = 0; | ||||
433 | |||||
434 | /* Returns the approximate width a string will take on screen when printed on a character cell | ||||
435 | * terminal/console. */ | ||||
436 | |||||
437 | while (*str != 0) { | ||||
438 | char32_t c; | ||||
439 | |||||
440 | if (utf8_encoded_to_unichar(str, &c) < 0) | ||||
441 | return (size_t) -1; | ||||
442 | |||||
443 | str = utf8_next_char(str)(char *)((str) + utf8_skip_data[*(const unsigned char *)(str) ]); | ||||
444 | |||||
445 | n += unichar_iswide(c) ? 2 : 1; | ||||
446 | } | ||||
447 | |||||
448 | return n; | ||||
449 | } |
1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
2 | #pragma once |
3 | |
4 | #include <alloca.h> |
5 | #include <stddef.h> |
6 | #include <stdlib.h> |
7 | #include <string.h> |
8 | |
9 | #include "macro.h" |
10 | |
11 | #define new(t, n)((t*) malloc_multiply(sizeof(t), (n))) ((t*) malloc_multiply(sizeof(t), (n))) |
12 | |
13 | #define new0(t, n)((t*) calloc((n), sizeof(t))) ((t*) calloc((n), sizeof(t))) |
14 | |
15 | #define newa(t, n)({ do { if ((__builtin_expect(!!(!(!size_multiply_overflow(sizeof (t), n))),0))) log_assert_failed_realm(LOG_REALM_SYSTEMD, ("!size_multiply_overflow(sizeof(t), n)" ), "../src/basic/alloc-util.h", 15, __PRETTY_FUNCTION__); } while (0); (t*) __builtin_alloca (sizeof(t)*(n)); }) \ |
16 | ({ \ |
17 | assert(!size_multiply_overflow(sizeof(t), n))do { if ((__builtin_expect(!!(!(!size_multiply_overflow(sizeof (t), n))),0))) log_assert_failed_realm(LOG_REALM_SYSTEMD, ("!size_multiply_overflow(sizeof(t), n)" ), "../src/basic/alloc-util.h", 17, __PRETTY_FUNCTION__); } while (0); \ |
18 | (t*) alloca(sizeof(t)*(n))__builtin_alloca (sizeof(t)*(n)); \ |
19 | }) |
20 | |
21 | #define newa0(t, n)({ do { if ((__builtin_expect(!!(!(!size_multiply_overflow(sizeof (t), n))),0))) log_assert_failed_realm(LOG_REALM_SYSTEMD, ("!size_multiply_overflow(sizeof(t), n)" ), "../src/basic/alloc-util.h", 21, __PRETTY_FUNCTION__); } while (0); (t*) ({ char *_new_; size_t _len_ = sizeof(t)*(n); _new_ = __builtin_alloca (_len_); (void *) memset(_new_, 0, _len_) ; }); }) \ |
22 | ({ \ |
23 | assert(!size_multiply_overflow(sizeof(t), n))do { if ((__builtin_expect(!!(!(!size_multiply_overflow(sizeof (t), n))),0))) log_assert_failed_realm(LOG_REALM_SYSTEMD, ("!size_multiply_overflow(sizeof(t), n)" ), "../src/basic/alloc-util.h", 23, __PRETTY_FUNCTION__); } while (0); \ |
24 | (t*) alloca0(sizeof(t)*(n))({ char *_new_; size_t _len_ = sizeof(t)*(n); _new_ = __builtin_alloca (_len_); (void *) memset(_new_, 0, _len_); }); \ |
25 | }) |
26 | |
27 | #define newdup(t, p, n)((t*) memdup_multiply(p, sizeof(t), (n))) ((t*) memdup_multiply(p, sizeof(t), (n))) |
28 | |
29 | #define newdup_suffix0(t, p, n)((t*) memdup_suffix0_multiply(p, sizeof(t), (n))) ((t*) memdup_suffix0_multiply(p, sizeof(t), (n))) |
30 | |
31 | #define malloc0(n)(calloc(1, (n))) (calloc(1, (n))) |
32 | |
33 | static inline void *mfree(void *memory) { |
34 | free(memory); |
35 | return NULL((void*)0); |
36 | } |
37 | |
38 | #define free_and_replace(a, b)({ free(a); (a) = (b); (b) = ((void*)0); 0; }) \ |
39 | ({ \ |
40 | free(a); \ |
41 | (a) = (b); \ |
42 | (b) = NULL((void*)0); \ |
43 | 0; \ |
44 | }) |
45 | |
46 | void* memdup(const void *p, size_t l) _alloc_(2); |
47 | void* memdup_suffix0(const void *p, size_t l) _alloc_(2); |
48 | |
49 | static inline void freep(void *p) { |
50 | free(*(void**) p); |
51 | } |
52 | |
53 | #define _cleanup_free___attribute__((cleanup(freep))) _cleanup_(freep)__attribute__((cleanup(freep))) |
54 | |
55 | static inline bool_Bool size_multiply_overflow(size_t size, size_t need) { |
56 | return _unlikely_(need != 0 && size > (SIZE_MAX / need))(__builtin_expect(!!(need != 0 && size > ((18446744073709551615UL ) / need)),0)); |
57 | } |
58 | |
59 | _malloc___attribute__ ((malloc)) _alloc_(1, 2) static inline void *malloc_multiply(size_t size, size_t need) { |
60 | if (size_multiply_overflow(size, need)) |
61 | return NULL((void*)0); |
62 | |
63 | return malloc(size * need); |
64 | } |
65 | |
66 | #if !HAVE_REALLOCARRAY1 |
67 | _alloc_(2, 3) static inline void *reallocarray(void *p, size_t need, size_t size) { |
68 | if (size_multiply_overflow(size, need)) |
69 | return NULL((void*)0); |
70 | |
71 | return realloc(p, size * need); |
72 | } |
73 | #endif |
74 | |
75 | _alloc_(2, 3) static inline void *memdup_multiply(const void *p, size_t size, size_t need) { |
76 | if (size_multiply_overflow(size, need)) |
77 | return NULL((void*)0); |
78 | |
79 | return memdup(p, size * need); |
80 | } |
81 | |
82 | _alloc_(2, 3) static inline void *memdup_suffix0_multiply(const void *p, size_t size, size_t need) { |
83 | if (size_multiply_overflow(size, need)) |
84 | return NULL((void*)0); |
85 | |
86 | return memdup_suffix0(p, size * need); |
87 | } |
88 | |
89 | void* greedy_realloc(void **p, size_t *allocated, size_t need, size_t size); |
90 | void* greedy_realloc0(void **p, size_t *allocated, size_t need, size_t size); |
91 | |
92 | #define GREEDY_REALLOC(array, allocated, need)greedy_realloc((void**) &(array), &(allocated), (need ), sizeof((array)[0])) \ |
93 | greedy_realloc((void**) &(array), &(allocated), (need), sizeof((array)[0])) |
94 | |
95 | #define GREEDY_REALLOC0(array, allocated, need)greedy_realloc0((void**) &(array), &(allocated), (need ), sizeof((array)[0])) \ |
96 | greedy_realloc0((void**) &(array), &(allocated), (need), sizeof((array)[0])) |
97 | |
98 | #define alloca0(n)({ char *_new_; size_t _len_ = n; _new_ = __builtin_alloca (_len_ ); (void *) memset(_new_, 0, _len_); }) \ |
99 | ({ \ |
100 | char *_new_; \ |
101 | size_t _len_ = n; \ |
102 | _new_ = alloca(_len_)__builtin_alloca (_len_); \ |
103 | (void *) memset(_new_, 0, _len_); \ |
104 | }) |
105 | |
106 | /* It's not clear what alignment glibc/gcc alloca() guarantee, hence provide a guaranteed safe version */ |
107 | #define alloca_align(size, align)({ void *_ptr_; size_t _mask_ = (align) - 1; _ptr_ = __builtin_alloca ((size) + _mask_); (void*)(((uintptr_t)_ptr_ + _mask_) & ~_mask_); }) \ |
108 | ({ \ |
109 | void *_ptr_; \ |
110 | size_t _mask_ = (align) - 1; \ |
111 | _ptr_ = alloca((size) + _mask_)__builtin_alloca ((size) + _mask_); \ |
112 | (void*)(((uintptr_t)_ptr_ + _mask_) & ~_mask_); \ |
113 | }) |
114 | |
115 | #define alloca0_align(size, align)({ void *_new_; size_t _size_ = (size); _new_ = ({ void *_ptr_ ; size_t _mask_ = ((align)) - 1; _ptr_ = __builtin_alloca ((_size_ ) + _mask_); (void*)(((uintptr_t)_ptr_ + _mask_) & ~_mask_ ); }); (void*)memset(_new_, 0, _size_); }) \ |
116 | ({ \ |
117 | void *_new_; \ |
118 | size_t _size_ = (size); \ |
119 | _new_ = alloca_align(_size_, (align))({ void *_ptr_; size_t _mask_ = ((align)) - 1; _ptr_ = __builtin_alloca ((_size_) + _mask_); (void*)(((uintptr_t)_ptr_ + _mask_) & ~_mask_); }); \ |
120 | (void*)memset(_new_, 0, _size_); \ |
121 | }) |
122 | |
123 | /* Takes inspiration from Rusts's Option::take() method: reads and returns a pointer, but at the same time resets it to |
124 | * NULL. See: https://doc.rust-lang.org/std/option/enum.Option.html#method.take */ |
125 | #define TAKE_PTR(ptr)({ typeof(ptr) _ptr_ = (ptr); (ptr) = ((void*)0); _ptr_; }) \ |
126 | ({ \ |
127 | typeof(ptr) _ptr_ = (ptr); \ |
128 | (ptr) = NULL((void*)0); \ |
129 | _ptr_; \ |
130 | }) |