Use safer sizeof variant in malloc
[openssl.git] / crypto / buffer / buffer.c
1 /* crypto/buffer/buffer.c */
2 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
3  * All rights reserved.
4  *
5  * This package is an SSL implementation written
6  * by Eric Young (eay@cryptsoft.com).
7  * The implementation was written so as to conform with Netscapes SSL.
8  *
9  * This library is free for commercial and non-commercial use as long as
10  * the following conditions are aheared to.  The following conditions
11  * apply to all code found in this distribution, be it the RC4, RSA,
12  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
13  * included with this distribution is covered by the same copyright terms
14  * except that the holder is Tim Hudson (tjh@cryptsoft.com).
15  *
16  * Copyright remains Eric Young's, and as such any Copyright notices in
17  * the code are not to be removed.
18  * If this package is used in a product, Eric Young should be given attribution
19  * as the author of the parts of the library used.
20  * This can be in the form of a textual message at program startup or
21  * in documentation (online or textual) provided with the package.
22  *
23  * Redistribution and use in source and binary forms, with or without
24  * modification, are permitted provided that the following conditions
25  * are met:
26  * 1. Redistributions of source code must retain the copyright
27  *    notice, this list of conditions and the following disclaimer.
28  * 2. Redistributions in binary form must reproduce the above copyright
29  *    notice, this list of conditions and the following disclaimer in the
30  *    documentation and/or other materials provided with the distribution.
31  * 3. All advertising materials mentioning features or use of this software
32  *    must display the following acknowledgement:
33  *    "This product includes cryptographic software written by
34  *     Eric Young (eay@cryptsoft.com)"
35  *    The word 'cryptographic' can be left out if the rouines from the library
36  *    being used are not cryptographic related :-).
37  * 4. If you include any Windows specific code (or a derivative thereof) from
38  *    the apps directory (application code) you must include an acknowledgement:
39  *    "This product includes software written by Tim Hudson (tjh@cryptsoft.com)"
40  *
41  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
42  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
43  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
44  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
45  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
46  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
47  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
48  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
49  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
50  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
51  * SUCH DAMAGE.
52  *
53  * The licence and distribution terms for any publically available version or
54  * derivative of this code cannot be changed.  i.e. this code cannot simply be
55  * copied and put under another distribution licence
56  * [including the GNU Public Licence.]
57  */
58
59 #include <stdio.h>
60 #include "cryptlib.h"
61 #include <openssl/buffer.h>
62
63 /*
64  * LIMIT_BEFORE_EXPANSION is the maximum n such that (n+3)/3*4 < 2**31. That
65  * function is applied in several functions in this file and this limit
66  * ensures that the result fits in an int.
67  */
68 #define LIMIT_BEFORE_EXPANSION 0x5ffffffc
69
70 BUF_MEM *BUF_MEM_new(void)
71 {
72     BUF_MEM *ret;
73
74     ret = OPENSSL_malloc(sizeof(*ret));
75     if (ret == NULL) {
76         BUFerr(BUF_F_BUF_MEM_NEW, ERR_R_MALLOC_FAILURE);
77         return (NULL);
78     }
79     ret->length = 0;
80     ret->max = 0;
81     ret->data = NULL;
82     return (ret);
83 }
84
85 void BUF_MEM_free(BUF_MEM *a)
86 {
87     if (a == NULL)
88         return;
89
90     if (a->data != NULL) {
91         memset(a->data, 0, (unsigned int)a->max);
92         OPENSSL_free(a->data);
93     }
94     OPENSSL_free(a);
95 }
96
97 size_t BUF_MEM_grow(BUF_MEM *str, size_t len)
98 {
99     char *ret;
100     size_t n;
101
102     if (str->length >= len) {
103         str->length = len;
104         return (len);
105     }
106     if (str->max >= len) {
107         memset(&str->data[str->length], 0, len - str->length);
108         str->length = len;
109         return (len);
110     }
111     /* This limit is sufficient to ensure (len+3)/3*4 < 2**31 */
112     if (len > LIMIT_BEFORE_EXPANSION) {
113         BUFerr(BUF_F_BUF_MEM_GROW, ERR_R_MALLOC_FAILURE);
114         return 0;
115     }
116     n = (len + 3) / 3 * 4;
117     ret = OPENSSL_realloc(str->data, n);
118     if (ret == NULL) {
119         BUFerr(BUF_F_BUF_MEM_GROW, ERR_R_MALLOC_FAILURE);
120         len = 0;
121     } else {
122         str->data = ret;
123         str->max = n;
124         memset(&str->data[str->length], 0, len - str->length);
125         str->length = len;
126     }
127     return (len);
128 }
129
130 size_t BUF_MEM_grow_clean(BUF_MEM *str, size_t len)
131 {
132     char *ret;
133     size_t n;
134
135     if (str->length >= len) {
136         memset(&str->data[len], 0, str->length - len);
137         str->length = len;
138         return (len);
139     }
140     if (str->max >= len) {
141         memset(&str->data[str->length], 0, len - str->length);
142         str->length = len;
143         return (len);
144     }
145     /* This limit is sufficient to ensure (len+3)/3*4 < 2**31 */
146     if (len > LIMIT_BEFORE_EXPANSION) {
147         BUFerr(BUF_F_BUF_MEM_GROW_CLEAN, ERR_R_MALLOC_FAILURE);
148         return 0;
149     }
150     n = (len + 3) / 3 * 4;
151     ret = OPENSSL_realloc_clean(str->data, str->max, n);
152     if (ret == NULL) {
153         BUFerr(BUF_F_BUF_MEM_GROW_CLEAN, ERR_R_MALLOC_FAILURE);
154         len = 0;
155     } else {
156         str->data = ret;
157         str->max = n;
158         memset(&str->data[str->length], 0, len - str->length);
159         str->length = len;
160     }
161     return (len);
162 }
163
164 void BUF_reverse(unsigned char *out, unsigned char *in, size_t size)
165 {
166     size_t i;
167     if (in) {
168         out += size - 1;
169         for (i = 0; i < size; i++)
170             *out-- = *in++;
171     } else {
172         unsigned char *q;
173         char c;
174         q = out + size - 1;
175         for (i = 0; i < size / 2; i++) {
176             c = *q;
177             *q-- = *out;
178             *out++ = c;
179         }
180     }
181 }