1 | /* Written by Ulf Moeller for the OpenSSL project. */ |
2 | /* ==================================================================== |
3 | * Copyright (c) 1998-2004 The OpenSSL Project. All rights reserved. |
4 | * |
5 | * Redistribution and use in source and binary forms, with or without |
6 | * modification, are permitted provided that the following conditions |
7 | * are met: |
8 | * |
9 | * 1. Redistributions of source code must retain the above copyright |
10 | * notice, this list of conditions and the following disclaimer. |
11 | * |
12 | * 2. Redistributions in binary form must reproduce the above copyright |
13 | * notice, this list of conditions and the following disclaimer in |
14 | * the documentation and/or other materials provided with the |
15 | * distribution. |
16 | * |
17 | * 3. All advertising materials mentioning features or use of this |
18 | * software must display the following acknowledgment: |
19 | * "This product includes software developed by the OpenSSL Project |
20 | * for use in the OpenSSL Toolkit. (http://www.openssl.org/)" |
21 | * |
22 | * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to |
23 | * endorse or promote products derived from this software without |
24 | * prior written permission. For written permission, please contact |
25 | * openssl-core@openssl.org. |
26 | * |
27 | * 5. Products derived from this software may not be called "OpenSSL" |
28 | * nor may "OpenSSL" appear in their names without prior written |
29 | * permission of the OpenSSL Project. |
30 | * |
31 | * 6. Redistributions of any form whatsoever must retain the following |
32 | * acknowledgment: |
33 | * "This product includes software developed by the OpenSSL Project |
34 | * for use in the OpenSSL Toolkit (http://www.openssl.org/)" |
35 | * |
36 | * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY |
37 | * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
38 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
39 | * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE OpenSSL PROJECT OR |
40 | * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
41 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
42 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
43 | * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
44 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, |
45 | * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
46 | * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED |
47 | * OF THE POSSIBILITY OF SUCH DAMAGE. |
48 | * ==================================================================== |
49 | * |
50 | * This product includes cryptographic software written by Eric Young |
51 | * (eay@cryptsoft.com). This product includes software written by Tim |
52 | * Hudson (tjh@cryptsoft.com). */ |
53 | |
54 | |
55 | #include <openssl/bn.h> |
56 | |
57 | #include <assert.h> |
58 | #include <string.h> |
59 | |
60 | #include <openssl/err.h> |
61 | #include <openssl/mem.h> |
62 | |
63 | #include "../../internal.h" |
64 | |
65 | |
66 | // The stack frame info is resizing, set a first-time expansion size; |
67 | #define BN_CTX_START_FRAMES 32 |
68 | |
69 | |
70 | // BN_STACK |
71 | |
72 | // A |BN_STACK| is a stack of |size_t| values. |
73 | typedef struct { |
74 | // Array of indexes into |ctx->bignums|. |
75 | size_t *indexes; |
76 | // Number of stack frames, and the size of the allocated array |
77 | size_t depth, size; |
78 | } BN_STACK; |
79 | |
80 | static void BN_STACK_init(BN_STACK *); |
81 | static void BN_STACK_cleanup(BN_STACK *); |
82 | static int BN_STACK_push(BN_STACK *, size_t idx); |
83 | static size_t BN_STACK_pop(BN_STACK *); |
84 | |
85 | |
86 | // BN_CTX |
87 | |
88 | DEFINE_STACK_OF(BIGNUM) |
89 | |
90 | // The opaque BN_CTX type |
91 | struct bignum_ctx { |
92 | // bignums is the stack of |BIGNUM|s managed by this |BN_CTX|. |
93 | STACK_OF(BIGNUM) *bignums; |
94 | // stack is the stack of |BN_CTX_start| frames. It is the value of |used| at |
95 | // the time |BN_CTX_start| was called. |
96 | BN_STACK stack; |
97 | // used is the number of |BIGNUM|s from |bignums| that have been used. |
98 | size_t used; |
99 | // error is one if any operation on this |BN_CTX| failed. All subsequent |
100 | // operations will fail. |
101 | char error; |
102 | // defer_error is one if an operation on this |BN_CTX| has failed, but no |
103 | // error has been pushed to the queue yet. This is used to defer errors from |
104 | // |BN_CTX_start| to |BN_CTX_get|. |
105 | char defer_error; |
106 | }; |
107 | |
108 | BN_CTX *BN_CTX_new(void) { |
109 | BN_CTX *ret = OPENSSL_malloc(sizeof(BN_CTX)); |
110 | if (!ret) { |
111 | OPENSSL_PUT_ERROR(BN, ERR_R_MALLOC_FAILURE); |
112 | return NULL; |
113 | } |
114 | |
115 | // Initialise the structure |
116 | ret->bignums = NULL; |
117 | BN_STACK_init(&ret->stack); |
118 | ret->used = 0; |
119 | ret->error = 0; |
120 | ret->defer_error = 0; |
121 | return ret; |
122 | } |
123 | |
124 | void BN_CTX_free(BN_CTX *ctx) { |
125 | if (ctx == NULL) { |
126 | return; |
127 | } |
128 | |
129 | sk_BIGNUM_pop_free(ctx->bignums, BN_free); |
130 | BN_STACK_cleanup(&ctx->stack); |
131 | OPENSSL_free(ctx); |
132 | } |
133 | |
134 | void BN_CTX_start(BN_CTX *ctx) { |
135 | if (ctx->error) { |
136 | // Once an operation has failed, |ctx->stack| no longer matches the number |
137 | // of |BN_CTX_end| calls to come. Do nothing. |
138 | return; |
139 | } |
140 | |
141 | if (!BN_STACK_push(&ctx->stack, ctx->used)) { |
142 | ctx->error = 1; |
143 | // |BN_CTX_start| cannot fail, so defer the error to |BN_CTX_get|. |
144 | ctx->defer_error = 1; |
145 | } |
146 | } |
147 | |
148 | BIGNUM *BN_CTX_get(BN_CTX *ctx) { |
149 | // Once any operation has failed, they all do. |
150 | if (ctx->error) { |
151 | if (ctx->defer_error) { |
152 | OPENSSL_PUT_ERROR(BN, BN_R_TOO_MANY_TEMPORARY_VARIABLES); |
153 | ctx->defer_error = 0; |
154 | } |
155 | return NULL; |
156 | } |
157 | |
158 | if (ctx->bignums == NULL) { |
159 | ctx->bignums = sk_BIGNUM_new_null(); |
160 | if (ctx->bignums == NULL) { |
161 | OPENSSL_PUT_ERROR(BN, ERR_R_MALLOC_FAILURE); |
162 | ctx->error = 1; |
163 | return NULL; |
164 | } |
165 | } |
166 | |
167 | if (ctx->used == sk_BIGNUM_num(ctx->bignums)) { |
168 | BIGNUM *bn = BN_new(); |
169 | if (bn == NULL || !sk_BIGNUM_push(ctx->bignums, bn)) { |
170 | OPENSSL_PUT_ERROR(BN, BN_R_TOO_MANY_TEMPORARY_VARIABLES); |
171 | BN_free(bn); |
172 | ctx->error = 1; |
173 | return NULL; |
174 | } |
175 | } |
176 | |
177 | BIGNUM *ret = sk_BIGNUM_value(ctx->bignums, ctx->used); |
178 | BN_zero(ret); |
179 | // This is bounded by |sk_BIGNUM_num|, so it cannot overflow. |
180 | ctx->used++; |
181 | return ret; |
182 | } |
183 | |
184 | void BN_CTX_end(BN_CTX *ctx) { |
185 | if (ctx->error) { |
186 | // Once an operation has failed, |ctx->stack| no longer matches the number |
187 | // of |BN_CTX_end| calls to come. Do nothing. |
188 | return; |
189 | } |
190 | |
191 | ctx->used = BN_STACK_pop(&ctx->stack); |
192 | } |
193 | |
194 | |
195 | // BN_STACK |
196 | |
197 | static void BN_STACK_init(BN_STACK *st) { |
198 | st->indexes = NULL; |
199 | st->depth = st->size = 0; |
200 | } |
201 | |
202 | static void BN_STACK_cleanup(BN_STACK *st) { |
203 | OPENSSL_free(st->indexes); |
204 | } |
205 | |
206 | static int BN_STACK_push(BN_STACK *st, size_t idx) { |
207 | if (st->depth == st->size) { |
208 | // This function intentionally does not push to the error queue on error. |
209 | // Error-reporting is deferred to |BN_CTX_get|. |
210 | size_t new_size = st->size != 0 ? st->size * 3 / 2 : BN_CTX_START_FRAMES; |
211 | if (new_size <= st->size || new_size > ((size_t)-1) / sizeof(size_t)) { |
212 | return 0; |
213 | } |
214 | size_t *new_indexes = |
215 | OPENSSL_realloc(st->indexes, new_size * sizeof(size_t)); |
216 | if (new_indexes == NULL) { |
217 | return 0; |
218 | } |
219 | st->indexes = new_indexes; |
220 | st->size = new_size; |
221 | } |
222 | |
223 | st->indexes[st->depth] = idx; |
224 | st->depth++; |
225 | return 1; |
226 | } |
227 | |
228 | static size_t BN_STACK_pop(BN_STACK *st) { |
229 | assert(st->depth > 0); |
230 | st->depth--; |
231 | return st->indexes[st->depth]; |
232 | } |
233 | |