1 /* $NetBSD: reallocarr.c,v 1.5 2015/08/20 22:27:49 kamil Exp $ */
2
3 /*-
4 * Copyright (c) 2015 Joerg Sonnenberger <joerg@NetBSD.org>.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in
15 * the documentation and/or other materials provided with the
16 * distribution.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
21 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
22 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
23 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
24 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
25 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
26 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
27 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
28 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32 #if !HAVE_REALLOCARR
33
34 #include "config.h"
35
36 #if HAVE_NBTOOL_CONFIG_H
37 #include "nbtool_config.h"
38 #endif
39
40 __RCSID("$NetBSD: reallocarr.c,v 1.5 2015/08/20 22:27:49 kamil Exp $");
41
42 #include <errno.h>
43 /* Old POSIX has SIZE_MAX in limits.h */
44 #include <limits.h>
45 #include <stdint.h>
46 #include <stdlib.h>
47 #include <string.h>
48
49 #ifdef _LIBC
50 #ifdef __weak_alias
__weak_alias(reallocarr,_reallocarr)51 __weak_alias(reallocarr, _reallocarr)
52 #endif
53 #endif
54
55 #define SQRT_SIZE_MAX (((size_t)1) << (sizeof(size_t) * CHAR_BIT / 2))
56
57 int
58 reallocarr(void *ptr, size_t number, size_t size)
59 {
60 int saved_errno, result;
61 void *optr;
62 void *nptr;
63
64 saved_errno = errno;
65 memcpy(&optr, ptr, sizeof(ptr));
66 if (number == 0 || size == 0) {
67 free(optr);
68 nptr = NULL;
69 memcpy(ptr, &nptr, sizeof(ptr));
70 errno = saved_errno;
71 return 0;
72 }
73
74 /*
75 * Try to avoid division here.
76 *
77 * It isn't possible to overflow during multiplication if neither
78 * operand uses any of the most significant half of the bits.
79 */
80 if ((number|size) >= SQRT_SIZE_MAX &&
81 number > SIZE_MAX / size) {
82 errno = saved_errno;
83 return EOVERFLOW;
84 }
85
86 nptr = realloc(optr, number * size);
87 if (nptr == NULL) {
88 result = errno;
89 } else {
90 result = 0;
91 memcpy(ptr, &nptr, sizeof(ptr));
92 }
93 errno = saved_errno;
94 return result;
95 }
96 #endif
97