blob: 1f6444e4ca1ac5e64337f0df47144636cd56737c [file] [log] [blame]
Paul Bakker5121ce52009-01-03 21:22:43 +00001/*
2 * Multi-precision integer library
3 *
Bence Szépkúti1e148272020-08-07 13:07:28 +02004 * Copyright The Mbed TLS Contributors
Manuel Pégourié-Gonnard37ff1402015-09-04 14:21:07 +02005 * SPDX-License-Identifier: Apache-2.0
6 *
7 * Licensed under the Apache License, Version 2.0 (the "License"); you may
8 * not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
10 *
11 * http://www.apache.org/licenses/LICENSE-2.0
12 *
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
15 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
Paul Bakker5121ce52009-01-03 21:22:43 +000018 */
Simon Butcher15b15d12015-11-26 19:35:03 +000019
Paul Bakker5121ce52009-01-03 21:22:43 +000020/*
Simon Butcher15b15d12015-11-26 19:35:03 +000021 * The following sources were referenced in the design of this Multi-precision
22 * Integer library:
Paul Bakker5121ce52009-01-03 21:22:43 +000023 *
Simon Butcher15b15d12015-11-26 19:35:03 +000024 * [1] Handbook of Applied Cryptography - 1997
25 * Menezes, van Oorschot and Vanstone
26 *
27 * [2] Multi-Precision Math
28 * Tom St Denis
29 * https://github.com/libtom/libtommath/blob/develop/tommath.pdf
30 *
31 * [3] GNU Multi-Precision Arithmetic Library
32 * https://gmplib.org/manual/index.html
33 *
Simon Butcherf5ba0452015-12-27 23:01:55 +000034 */
Paul Bakker5121ce52009-01-03 21:22:43 +000035
Gilles Peskinedb09ef62020-06-03 01:43:33 +020036#include "common.h"
Paul Bakker5121ce52009-01-03 21:22:43 +000037
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020038#if defined(MBEDTLS_BIGNUM_C)
Paul Bakker5121ce52009-01-03 21:22:43 +000039
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000040#include "mbedtls/bignum.h"
Chris Jones4c5819c2021-03-03 17:45:34 +000041#include "bn_mul.h"
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050042#include "mbedtls/platform_util.h"
Janos Follath24eed8d2019-11-22 13:21:35 +000043#include "mbedtls/error.h"
Paul Bakker5121ce52009-01-03 21:22:43 +000044
Rich Evans00ab4702015-02-06 13:43:58 +000045#include <string.h>
46
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020047#if defined(MBEDTLS_PLATFORM_C)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000048#include "mbedtls/platform.h"
Paul Bakker6e339b52013-07-03 13:37:05 +020049#else
Rich Evans00ab4702015-02-06 13:43:58 +000050#include <stdio.h>
51#include <stdlib.h>
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020052#define mbedtls_printf printf
Manuel Pégourié-Gonnard7551cb92015-05-26 16:04:06 +020053#define mbedtls_calloc calloc
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020054#define mbedtls_free free
Paul Bakker6e339b52013-07-03 13:37:05 +020055#endif
56
Hanno Becker73d7d792018-12-11 10:35:51 +000057#define MPI_VALIDATE_RET( cond ) \
58 MBEDTLS_INTERNAL_VALIDATE_RET( cond, MBEDTLS_ERR_MPI_BAD_INPUT_DATA )
59#define MPI_VALIDATE( cond ) \
60 MBEDTLS_INTERNAL_VALIDATE( cond )
61
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020062#define ciL (sizeof(mbedtls_mpi_uint)) /* chars in limb */
Paul Bakker5121ce52009-01-03 21:22:43 +000063#define biL (ciL << 3) /* bits in limb */
64#define biH (ciL << 2) /* half limb size */
65
Manuel Pégourié-Gonnard2d708342015-10-05 15:23:11 +010066#define MPI_SIZE_T_MAX ( (size_t) -1 ) /* SIZE_T_MAX is not standard */
67
Paul Bakker5121ce52009-01-03 21:22:43 +000068/*
69 * Convert between bits/chars and number of limbs
Manuel Pégourié-Gonnard58fb4952015-09-28 13:48:04 +020070 * Divide first in order to avoid potential overflows
Paul Bakker5121ce52009-01-03 21:22:43 +000071 */
Manuel Pégourié-Gonnard58fb4952015-09-28 13:48:04 +020072#define BITS_TO_LIMBS(i) ( (i) / biL + ( (i) % biL != 0 ) )
73#define CHARS_TO_LIMBS(i) ( (i) / ciL + ( (i) % ciL != 0 ) )
Paul Bakker5121ce52009-01-03 21:22:43 +000074
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050075/* Implementation that should never be optimized out by the compiler */
Andres Amaya Garcia6698d2f2018-04-24 08:39:07 -050076static void mbedtls_mpi_zeroize( mbedtls_mpi_uint *v, size_t n )
77{
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050078 mbedtls_platform_zeroize( v, ciL * n );
79}
80
Paul Bakker5121ce52009-01-03 21:22:43 +000081/*
Paul Bakker6c591fa2011-05-05 11:49:20 +000082 * Initialize one MPI
Paul Bakker5121ce52009-01-03 21:22:43 +000083 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020084void mbedtls_mpi_init( mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +000085{
Hanno Becker73d7d792018-12-11 10:35:51 +000086 MPI_VALIDATE( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +000087
Paul Bakker6c591fa2011-05-05 11:49:20 +000088 X->s = 1;
89 X->n = 0;
90 X->p = NULL;
Paul Bakker5121ce52009-01-03 21:22:43 +000091}
92
93/*
Paul Bakker6c591fa2011-05-05 11:49:20 +000094 * Unallocate one MPI
Paul Bakker5121ce52009-01-03 21:22:43 +000095 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020096void mbedtls_mpi_free( mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +000097{
Paul Bakker6c591fa2011-05-05 11:49:20 +000098 if( X == NULL )
99 return;
Paul Bakker5121ce52009-01-03 21:22:43 +0000100
Paul Bakker6c591fa2011-05-05 11:49:20 +0000101 if( X->p != NULL )
Paul Bakker5121ce52009-01-03 21:22:43 +0000102 {
Alexey Skalozube17a8da2016-01-13 17:19:33 +0200103 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200104 mbedtls_free( X->p );
Paul Bakker5121ce52009-01-03 21:22:43 +0000105 }
106
Paul Bakker6c591fa2011-05-05 11:49:20 +0000107 X->s = 1;
108 X->n = 0;
109 X->p = NULL;
Paul Bakker5121ce52009-01-03 21:22:43 +0000110}
111
112/*
113 * Enlarge to the specified number of limbs
114 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200115int mbedtls_mpi_grow( mbedtls_mpi *X, size_t nblimbs )
Paul Bakker5121ce52009-01-03 21:22:43 +0000116{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200117 mbedtls_mpi_uint *p;
Hanno Becker73d7d792018-12-11 10:35:51 +0000118 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000119
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200120 if( nblimbs > MBEDTLS_MPI_MAX_LIMBS )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200121 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Paul Bakkerf9688572011-05-05 10:00:45 +0000122
Paul Bakker5121ce52009-01-03 21:22:43 +0000123 if( X->n < nblimbs )
124 {
Simon Butcher29176892016-05-20 00:19:09 +0100125 if( ( p = (mbedtls_mpi_uint*)mbedtls_calloc( nblimbs, ciL ) ) == NULL )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200126 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Paul Bakker5121ce52009-01-03 21:22:43 +0000127
Paul Bakker5121ce52009-01-03 21:22:43 +0000128 if( X->p != NULL )
129 {
130 memcpy( p, X->p, X->n * ciL );
Alexey Skalozube17a8da2016-01-13 17:19:33 +0200131 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200132 mbedtls_free( X->p );
Paul Bakker5121ce52009-01-03 21:22:43 +0000133 }
134
135 X->n = nblimbs;
136 X->p = p;
137 }
138
139 return( 0 );
140}
141
142/*
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100143 * Resize down as much as possible,
144 * while keeping at least the specified number of limbs
145 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200146int mbedtls_mpi_shrink( mbedtls_mpi *X, size_t nblimbs )
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100147{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200148 mbedtls_mpi_uint *p;
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100149 size_t i;
Hanno Becker73d7d792018-12-11 10:35:51 +0000150 MPI_VALIDATE_RET( X != NULL );
151
152 if( nblimbs > MBEDTLS_MPI_MAX_LIMBS )
153 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100154
Gilles Peskinee2f563e2020-01-20 21:17:43 +0100155 /* Actually resize up if there are currently fewer than nblimbs limbs. */
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100156 if( X->n <= nblimbs )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200157 return( mbedtls_mpi_grow( X, nblimbs ) );
Gilles Peskine322752b2020-01-21 13:59:51 +0100158 /* After this point, then X->n > nblimbs and in particular X->n > 0. */
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100159
160 for( i = X->n - 1; i > 0; i-- )
161 if( X->p[i] != 0 )
162 break;
163 i++;
164
165 if( i < nblimbs )
166 i = nblimbs;
167
Simon Butcher29176892016-05-20 00:19:09 +0100168 if( ( p = (mbedtls_mpi_uint*)mbedtls_calloc( i, ciL ) ) == NULL )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200169 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100170
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100171 if( X->p != NULL )
172 {
173 memcpy( p, X->p, i * ciL );
Alexey Skalozube17a8da2016-01-13 17:19:33 +0200174 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200175 mbedtls_free( X->p );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100176 }
177
178 X->n = i;
179 X->p = p;
180
181 return( 0 );
182}
183
184/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000185 * Copy the contents of Y into X
186 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200187int mbedtls_mpi_copy( mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +0000188{
Gilles Peskine4e4be7c2018-03-21 16:29:03 +0100189 int ret = 0;
Paul Bakker23986e52011-04-24 08:57:21 +0000190 size_t i;
Hanno Becker73d7d792018-12-11 10:35:51 +0000191 MPI_VALIDATE_RET( X != NULL );
192 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000193
194 if( X == Y )
195 return( 0 );
196
Gilles Peskinedb420622020-01-20 21:12:50 +0100197 if( Y->n == 0 )
Manuel Pégourié-Gonnardf4999932013-08-12 17:02:59 +0200198 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200199 mbedtls_mpi_free( X );
Manuel Pégourié-Gonnardf4999932013-08-12 17:02:59 +0200200 return( 0 );
201 }
202
Paul Bakker5121ce52009-01-03 21:22:43 +0000203 for( i = Y->n - 1; i > 0; i-- )
204 if( Y->p[i] != 0 )
205 break;
206 i++;
207
208 X->s = Y->s;
209
Gilles Peskine4e4be7c2018-03-21 16:29:03 +0100210 if( X->n < i )
211 {
212 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i ) );
213 }
214 else
215 {
216 memset( X->p + i, 0, ( X->n - i ) * ciL );
217 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000218
Paul Bakker5121ce52009-01-03 21:22:43 +0000219 memcpy( X->p, Y->p, i * ciL );
220
221cleanup:
222
223 return( ret );
224}
225
226/*
227 * Swap the contents of X and Y
228 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200229void mbedtls_mpi_swap( mbedtls_mpi *X, mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +0000230{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200231 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000232 MPI_VALIDATE( X != NULL );
233 MPI_VALIDATE( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000234
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200235 memcpy( &T, X, sizeof( mbedtls_mpi ) );
236 memcpy( X, Y, sizeof( mbedtls_mpi ) );
237 memcpy( Y, &T, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000238}
239
240/*
Gilles Peskinef04d11e2020-06-04 19:14:58 +0200241 * Conditionally assign dest = src, without leaking information
242 * about whether the assignment was made or not.
243 * dest and src must be arrays of limbs of size n.
244 * assign must be 0 or 1.
245 */
246static void mpi_safe_cond_assign( size_t n,
247 mbedtls_mpi_uint *dest,
248 const mbedtls_mpi_uint *src,
249 unsigned char assign )
250{
251 size_t i;
252 for( i = 0; i < n; i++ )
253 dest[i] = dest[i] * ( 1 - assign ) + src[i] * assign;
254}
255
256/*
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100257 * Conditionally assign X = Y, without leaking information
Manuel Pégourié-Gonnard96c7a922013-11-25 18:28:53 +0100258 * about whether the assignment was made or not.
259 * (Leaking information about the respective sizes of X and Y is ok however.)
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100260 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200261int mbedtls_mpi_safe_cond_assign( mbedtls_mpi *X, const mbedtls_mpi *Y, unsigned char assign )
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100262{
263 int ret = 0;
264 size_t i;
Hanno Becker73d7d792018-12-11 10:35:51 +0000265 MPI_VALIDATE_RET( X != NULL );
266 MPI_VALIDATE_RET( Y != NULL );
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100267
Pascal Junodb99183d2015-03-11 16:49:45 +0100268 /* make sure assign is 0 or 1 in a time-constant manner */
269 assign = (assign | (unsigned char)-assign) >> 7;
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100270
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200271 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, Y->n ) );
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100272
Paul Bakker66d5d072014-06-17 16:39:18 +0200273 X->s = X->s * ( 1 - assign ) + Y->s * assign;
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100274
Gilles Peskinef04d11e2020-06-04 19:14:58 +0200275 mpi_safe_cond_assign( Y->n, X->p, Y->p, assign );
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100276
Gilles Peskinef04d11e2020-06-04 19:14:58 +0200277 for( i = Y->n; i < X->n; i++ )
Paul Bakker66d5d072014-06-17 16:39:18 +0200278 X->p[i] *= ( 1 - assign );
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100279
280cleanup:
281 return( ret );
282}
283
284/*
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100285 * Conditionally swap X and Y, without leaking information
286 * about whether the swap was made or not.
287 * Here it is not ok to simply swap the pointers, which whould lead to
288 * different memory access patterns when X and Y are used afterwards.
289 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200290int mbedtls_mpi_safe_cond_swap( mbedtls_mpi *X, mbedtls_mpi *Y, unsigned char swap )
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100291{
292 int ret, s;
293 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200294 mbedtls_mpi_uint tmp;
Hanno Becker73d7d792018-12-11 10:35:51 +0000295 MPI_VALIDATE_RET( X != NULL );
296 MPI_VALIDATE_RET( Y != NULL );
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100297
298 if( X == Y )
299 return( 0 );
300
Pascal Junodb99183d2015-03-11 16:49:45 +0100301 /* make sure swap is 0 or 1 in a time-constant manner */
302 swap = (swap | (unsigned char)-swap) >> 7;
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100303
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200304 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, Y->n ) );
305 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( Y, X->n ) );
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100306
307 s = X->s;
Paul Bakker66d5d072014-06-17 16:39:18 +0200308 X->s = X->s * ( 1 - swap ) + Y->s * swap;
309 Y->s = Y->s * ( 1 - swap ) + s * swap;
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100310
311
312 for( i = 0; i < X->n; i++ )
313 {
314 tmp = X->p[i];
Paul Bakker66d5d072014-06-17 16:39:18 +0200315 X->p[i] = X->p[i] * ( 1 - swap ) + Y->p[i] * swap;
316 Y->p[i] = Y->p[i] * ( 1 - swap ) + tmp * swap;
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100317 }
318
319cleanup:
320 return( ret );
321}
322
323/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000324 * Set value from integer
325 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200326int mbedtls_mpi_lset( mbedtls_mpi *X, mbedtls_mpi_sint z )
Paul Bakker5121ce52009-01-03 21:22:43 +0000327{
Janos Follath24eed8d2019-11-22 13:21:35 +0000328 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker73d7d792018-12-11 10:35:51 +0000329 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000330
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200331 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000332 memset( X->p, 0, X->n * ciL );
333
334 X->p[0] = ( z < 0 ) ? -z : z;
335 X->s = ( z < 0 ) ? -1 : 1;
336
337cleanup:
338
339 return( ret );
340}
341
342/*
Paul Bakker2f5947e2011-05-18 15:47:11 +0000343 * Get a specific bit
344 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200345int mbedtls_mpi_get_bit( const mbedtls_mpi *X, size_t pos )
Paul Bakker2f5947e2011-05-18 15:47:11 +0000346{
Hanno Becker73d7d792018-12-11 10:35:51 +0000347 MPI_VALIDATE_RET( X != NULL );
348
Paul Bakker2f5947e2011-05-18 15:47:11 +0000349 if( X->n * biL <= pos )
350 return( 0 );
351
Paul Bakkerd8bb8262014-06-17 14:06:49 +0200352 return( ( X->p[pos / biL] >> ( pos % biL ) ) & 0x01 );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000353}
354
Gilles Peskine11cdb052018-11-20 16:47:47 +0100355/* Get a specific byte, without range checks. */
356#define GET_BYTE( X, i ) \
357 ( ( ( X )->p[( i ) / ciL] >> ( ( ( i ) % ciL ) * 8 ) ) & 0xff )
358
Paul Bakker2f5947e2011-05-18 15:47:11 +0000359/*
360 * Set a bit to a specific value of 0 or 1
361 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200362int mbedtls_mpi_set_bit( mbedtls_mpi *X, size_t pos, unsigned char val )
Paul Bakker2f5947e2011-05-18 15:47:11 +0000363{
364 int ret = 0;
365 size_t off = pos / biL;
366 size_t idx = pos % biL;
Hanno Becker73d7d792018-12-11 10:35:51 +0000367 MPI_VALIDATE_RET( X != NULL );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000368
369 if( val != 0 && val != 1 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200370 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker9af723c2014-05-01 13:03:14 +0200371
Paul Bakker2f5947e2011-05-18 15:47:11 +0000372 if( X->n * biL <= pos )
373 {
374 if( val == 0 )
Paul Bakkerd8bb8262014-06-17 14:06:49 +0200375 return( 0 );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000376
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200377 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, off + 1 ) );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000378 }
379
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200380 X->p[off] &= ~( (mbedtls_mpi_uint) 0x01 << idx );
381 X->p[off] |= (mbedtls_mpi_uint) val << idx;
Paul Bakker2f5947e2011-05-18 15:47:11 +0000382
383cleanup:
Paul Bakker9af723c2014-05-01 13:03:14 +0200384
Paul Bakker2f5947e2011-05-18 15:47:11 +0000385 return( ret );
386}
387
388/*
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200389 * Return the number of less significant zero-bits
Paul Bakker5121ce52009-01-03 21:22:43 +0000390 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200391size_t mbedtls_mpi_lsb( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000392{
Paul Bakker23986e52011-04-24 08:57:21 +0000393 size_t i, j, count = 0;
Hanno Beckerf25ee7f2018-12-19 16:51:02 +0000394 MBEDTLS_INTERNAL_VALIDATE_RET( X != NULL, 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000395
396 for( i = 0; i < X->n; i++ )
Paul Bakkerf9688572011-05-05 10:00:45 +0000397 for( j = 0; j < biL; j++, count++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000398 if( ( ( X->p[i] >> j ) & 1 ) != 0 )
399 return( count );
400
401 return( 0 );
402}
403
404/*
Simon Butcher15b15d12015-11-26 19:35:03 +0000405 * Count leading zero bits in a given integer
406 */
407static size_t mbedtls_clz( const mbedtls_mpi_uint x )
408{
409 size_t j;
Manuel Pégourié-Gonnarde3e8edf2015-12-01 09:31:52 +0100410 mbedtls_mpi_uint mask = (mbedtls_mpi_uint) 1 << (biL - 1);
Simon Butcher15b15d12015-11-26 19:35:03 +0000411
412 for( j = 0; j < biL; j++ )
413 {
414 if( x & mask ) break;
415
416 mask >>= 1;
417 }
418
419 return j;
420}
421
422/*
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200423 * Return the number of bits
Paul Bakker5121ce52009-01-03 21:22:43 +0000424 */
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200425size_t mbedtls_mpi_bitlen( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000426{
Paul Bakker23986e52011-04-24 08:57:21 +0000427 size_t i, j;
Paul Bakker5121ce52009-01-03 21:22:43 +0000428
Manuel Pégourié-Gonnard770b5e12015-04-29 17:02:01 +0200429 if( X->n == 0 )
430 return( 0 );
431
Paul Bakker5121ce52009-01-03 21:22:43 +0000432 for( i = X->n - 1; i > 0; i-- )
433 if( X->p[i] != 0 )
434 break;
435
Simon Butcher15b15d12015-11-26 19:35:03 +0000436 j = biL - mbedtls_clz( X->p[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +0000437
Paul Bakker23986e52011-04-24 08:57:21 +0000438 return( ( i * biL ) + j );
Paul Bakker5121ce52009-01-03 21:22:43 +0000439}
440
441/*
442 * Return the total size in bytes
443 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200444size_t mbedtls_mpi_size( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000445{
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200446 return( ( mbedtls_mpi_bitlen( X ) + 7 ) >> 3 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000447}
448
449/*
450 * Convert an ASCII character to digit value
451 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200452static int mpi_get_digit( mbedtls_mpi_uint *d, int radix, char c )
Paul Bakker5121ce52009-01-03 21:22:43 +0000453{
454 *d = 255;
455
456 if( c >= 0x30 && c <= 0x39 ) *d = c - 0x30;
457 if( c >= 0x41 && c <= 0x46 ) *d = c - 0x37;
458 if( c >= 0x61 && c <= 0x66 ) *d = c - 0x57;
459
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200460 if( *d >= (mbedtls_mpi_uint) radix )
461 return( MBEDTLS_ERR_MPI_INVALID_CHARACTER );
Paul Bakker5121ce52009-01-03 21:22:43 +0000462
463 return( 0 );
464}
465
466/*
467 * Import from an ASCII string
468 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200469int mbedtls_mpi_read_string( mbedtls_mpi *X, int radix, const char *s )
Paul Bakker5121ce52009-01-03 21:22:43 +0000470{
Janos Follath24eed8d2019-11-22 13:21:35 +0000471 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +0000472 size_t i, j, slen, n;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200473 mbedtls_mpi_uint d;
474 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000475 MPI_VALIDATE_RET( X != NULL );
476 MPI_VALIDATE_RET( s != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000477
478 if( radix < 2 || radix > 16 )
Hanno Becker54c91dd2018-12-12 13:37:06 +0000479 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000480
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200481 mbedtls_mpi_init( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000482
Paul Bakkerff60ee62010-03-16 21:09:09 +0000483 slen = strlen( s );
484
Paul Bakker5121ce52009-01-03 21:22:43 +0000485 if( radix == 16 )
486 {
Manuel Pégourié-Gonnard2d708342015-10-05 15:23:11 +0100487 if( slen > MPI_SIZE_T_MAX >> 2 )
Manuel Pégourié-Gonnard58fb4952015-09-28 13:48:04 +0200488 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
489
Paul Bakkerff60ee62010-03-16 21:09:09 +0000490 n = BITS_TO_LIMBS( slen << 2 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000491
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200492 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, n ) );
493 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000494
Paul Bakker23986e52011-04-24 08:57:21 +0000495 for( i = slen, j = 0; i > 0; i--, j++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000496 {
Paul Bakker23986e52011-04-24 08:57:21 +0000497 if( i == 1 && s[i - 1] == '-' )
Paul Bakker5121ce52009-01-03 21:22:43 +0000498 {
499 X->s = -1;
500 break;
501 }
502
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200503 MBEDTLS_MPI_CHK( mpi_get_digit( &d, radix, s[i - 1] ) );
Paul Bakker66d5d072014-06-17 16:39:18 +0200504 X->p[j / ( 2 * ciL )] |= d << ( ( j % ( 2 * ciL ) ) << 2 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000505 }
506 }
507 else
508 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200509 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000510
Paul Bakkerff60ee62010-03-16 21:09:09 +0000511 for( i = 0; i < slen; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000512 {
513 if( i == 0 && s[i] == '-' )
514 {
515 X->s = -1;
516 continue;
517 }
518
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200519 MBEDTLS_MPI_CHK( mpi_get_digit( &d, radix, s[i] ) );
520 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T, X, radix ) );
Paul Bakker05feca62009-06-20 08:22:43 +0000521
522 if( X->s == 1 )
523 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200524 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, &T, d ) );
Paul Bakker05feca62009-06-20 08:22:43 +0000525 }
526 else
527 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200528 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( X, &T, d ) );
Paul Bakker05feca62009-06-20 08:22:43 +0000529 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000530 }
531 }
532
533cleanup:
534
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200535 mbedtls_mpi_free( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000536
537 return( ret );
538}
539
540/*
Ron Eldora16fa292018-11-20 14:07:01 +0200541 * Helper to write the digits high-order first.
Paul Bakker5121ce52009-01-03 21:22:43 +0000542 */
Ron Eldora16fa292018-11-20 14:07:01 +0200543static int mpi_write_hlp( mbedtls_mpi *X, int radix,
544 char **p, const size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000545{
Janos Follath24eed8d2019-11-22 13:21:35 +0000546 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200547 mbedtls_mpi_uint r;
Ron Eldora16fa292018-11-20 14:07:01 +0200548 size_t length = 0;
549 char *p_end = *p + buflen;
Paul Bakker5121ce52009-01-03 21:22:43 +0000550
Ron Eldora16fa292018-11-20 14:07:01 +0200551 do
552 {
553 if( length >= buflen )
554 {
555 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
556 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000557
Ron Eldora16fa292018-11-20 14:07:01 +0200558 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, radix ) );
559 MBEDTLS_MPI_CHK( mbedtls_mpi_div_int( X, NULL, X, radix ) );
560 /*
561 * Write the residue in the current position, as an ASCII character.
562 */
563 if( r < 0xA )
564 *(--p_end) = (char)( '0' + r );
565 else
566 *(--p_end) = (char)( 'A' + ( r - 0xA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000567
Ron Eldora16fa292018-11-20 14:07:01 +0200568 length++;
569 } while( mbedtls_mpi_cmp_int( X, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000570
Ron Eldora16fa292018-11-20 14:07:01 +0200571 memmove( *p, p_end, length );
572 *p += length;
Paul Bakker5121ce52009-01-03 21:22:43 +0000573
574cleanup:
575
576 return( ret );
577}
578
579/*
580 * Export into an ASCII string
581 */
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100582int mbedtls_mpi_write_string( const mbedtls_mpi *X, int radix,
583 char *buf, size_t buflen, size_t *olen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000584{
Paul Bakker23986e52011-04-24 08:57:21 +0000585 int ret = 0;
586 size_t n;
Paul Bakker5121ce52009-01-03 21:22:43 +0000587 char *p;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200588 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000589 MPI_VALIDATE_RET( X != NULL );
590 MPI_VALIDATE_RET( olen != NULL );
591 MPI_VALIDATE_RET( buflen == 0 || buf != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000592
593 if( radix < 2 || radix > 16 )
Hanno Becker54c91dd2018-12-12 13:37:06 +0000594 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000595
Hanno Becker23cfea02019-02-04 09:45:07 +0000596 n = mbedtls_mpi_bitlen( X ); /* Number of bits necessary to present `n`. */
597 if( radix >= 4 ) n >>= 1; /* Number of 4-adic digits necessary to present
598 * `n`. If radix > 4, this might be a strict
599 * overapproximation of the number of
600 * radix-adic digits needed to present `n`. */
601 if( radix >= 16 ) n >>= 1; /* Number of hexadecimal digits necessary to
602 * present `n`. */
603
Janos Follath80470622019-03-06 13:43:02 +0000604 n += 1; /* Terminating null byte */
Hanno Becker23cfea02019-02-04 09:45:07 +0000605 n += 1; /* Compensate for the divisions above, which round down `n`
606 * in case it's not even. */
607 n += 1; /* Potential '-'-sign. */
608 n += ( n & 1 ); /* Make n even to have enough space for hexadecimal writing,
609 * which always uses an even number of hex-digits. */
Paul Bakker5121ce52009-01-03 21:22:43 +0000610
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100611 if( buflen < n )
Paul Bakker5121ce52009-01-03 21:22:43 +0000612 {
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100613 *olen = n;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200614 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000615 }
616
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100617 p = buf;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200618 mbedtls_mpi_init( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000619
620 if( X->s == -1 )
Hanno Beckerc983c812019-02-01 16:41:30 +0000621 {
Paul Bakker5121ce52009-01-03 21:22:43 +0000622 *p++ = '-';
Hanno Beckerc983c812019-02-01 16:41:30 +0000623 buflen--;
624 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000625
626 if( radix == 16 )
627 {
Paul Bakker23986e52011-04-24 08:57:21 +0000628 int c;
629 size_t i, j, k;
Paul Bakker5121ce52009-01-03 21:22:43 +0000630
Paul Bakker23986e52011-04-24 08:57:21 +0000631 for( i = X->n, k = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000632 {
Paul Bakker23986e52011-04-24 08:57:21 +0000633 for( j = ciL; j > 0; j-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000634 {
Paul Bakker23986e52011-04-24 08:57:21 +0000635 c = ( X->p[i - 1] >> ( ( j - 1 ) << 3) ) & 0xFF;
Paul Bakker5121ce52009-01-03 21:22:43 +0000636
Paul Bakker6c343d72014-07-10 14:36:19 +0200637 if( c == 0 && k == 0 && ( i + j ) != 2 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000638 continue;
639
Paul Bakker98fe5ea2012-10-24 11:17:48 +0000640 *(p++) = "0123456789ABCDEF" [c / 16];
Paul Bakkerd2c167e2012-10-30 07:49:19 +0000641 *(p++) = "0123456789ABCDEF" [c % 16];
Paul Bakker5121ce52009-01-03 21:22:43 +0000642 k = 1;
643 }
644 }
645 }
646 else
647 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200648 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T, X ) );
Paul Bakkerce40a6d2009-06-23 19:46:08 +0000649
650 if( T.s == -1 )
651 T.s = 1;
652
Ron Eldora16fa292018-11-20 14:07:01 +0200653 MBEDTLS_MPI_CHK( mpi_write_hlp( &T, radix, &p, buflen ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000654 }
655
656 *p++ = '\0';
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100657 *olen = p - buf;
Paul Bakker5121ce52009-01-03 21:22:43 +0000658
659cleanup:
660
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200661 mbedtls_mpi_free( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000662
663 return( ret );
664}
665
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200666#if defined(MBEDTLS_FS_IO)
Paul Bakker5121ce52009-01-03 21:22:43 +0000667/*
668 * Read X from an opened file
669 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200670int mbedtls_mpi_read_file( mbedtls_mpi *X, int radix, FILE *fin )
Paul Bakker5121ce52009-01-03 21:22:43 +0000671{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200672 mbedtls_mpi_uint d;
Paul Bakker23986e52011-04-24 08:57:21 +0000673 size_t slen;
Paul Bakker5121ce52009-01-03 21:22:43 +0000674 char *p;
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000675 /*
Paul Bakkercb37aa52011-11-30 16:00:20 +0000676 * Buffer should have space for (short) label and decimal formatted MPI,
677 * newline characters and '\0'
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000678 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200679 char s[ MBEDTLS_MPI_RW_BUFFER_SIZE ];
Paul Bakker5121ce52009-01-03 21:22:43 +0000680
Hanno Becker73d7d792018-12-11 10:35:51 +0000681 MPI_VALIDATE_RET( X != NULL );
682 MPI_VALIDATE_RET( fin != NULL );
683
684 if( radix < 2 || radix > 16 )
685 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
686
Paul Bakker5121ce52009-01-03 21:22:43 +0000687 memset( s, 0, sizeof( s ) );
688 if( fgets( s, sizeof( s ) - 1, fin ) == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200689 return( MBEDTLS_ERR_MPI_FILE_IO_ERROR );
Paul Bakker5121ce52009-01-03 21:22:43 +0000690
691 slen = strlen( s );
Paul Bakkercb37aa52011-11-30 16:00:20 +0000692 if( slen == sizeof( s ) - 2 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200693 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
Paul Bakkercb37aa52011-11-30 16:00:20 +0000694
Hanno Beckerb2034b72017-04-26 11:46:46 +0100695 if( slen > 0 && s[slen - 1] == '\n' ) { slen--; s[slen] = '\0'; }
696 if( slen > 0 && s[slen - 1] == '\r' ) { slen--; s[slen] = '\0'; }
Paul Bakker5121ce52009-01-03 21:22:43 +0000697
698 p = s + slen;
Hanno Beckerb2034b72017-04-26 11:46:46 +0100699 while( p-- > s )
Paul Bakker5121ce52009-01-03 21:22:43 +0000700 if( mpi_get_digit( &d, radix, *p ) != 0 )
701 break;
702
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200703 return( mbedtls_mpi_read_string( X, radix, p + 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000704}
705
706/*
707 * Write X into an opened file (or stdout if fout == NULL)
708 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200709int mbedtls_mpi_write_file( const char *p, const mbedtls_mpi *X, int radix, FILE *fout )
Paul Bakker5121ce52009-01-03 21:22:43 +0000710{
Janos Follath24eed8d2019-11-22 13:21:35 +0000711 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +0000712 size_t n, slen, plen;
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000713 /*
Paul Bakker5531c6d2012-09-26 19:20:46 +0000714 * Buffer should have space for (short) label and decimal formatted MPI,
715 * newline characters and '\0'
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000716 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200717 char s[ MBEDTLS_MPI_RW_BUFFER_SIZE ];
Hanno Becker73d7d792018-12-11 10:35:51 +0000718 MPI_VALIDATE_RET( X != NULL );
719
720 if( radix < 2 || radix > 16 )
721 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000722
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100723 memset( s, 0, sizeof( s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000724
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100725 MBEDTLS_MPI_CHK( mbedtls_mpi_write_string( X, radix, s, sizeof( s ) - 2, &n ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000726
727 if( p == NULL ) p = "";
728
729 plen = strlen( p );
730 slen = strlen( s );
731 s[slen++] = '\r';
732 s[slen++] = '\n';
733
734 if( fout != NULL )
735 {
736 if( fwrite( p, 1, plen, fout ) != plen ||
737 fwrite( s, 1, slen, fout ) != slen )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200738 return( MBEDTLS_ERR_MPI_FILE_IO_ERROR );
Paul Bakker5121ce52009-01-03 21:22:43 +0000739 }
740 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200741 mbedtls_printf( "%s%s", p, s );
Paul Bakker5121ce52009-01-03 21:22:43 +0000742
743cleanup:
744
745 return( ret );
746}
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200747#endif /* MBEDTLS_FS_IO */
Paul Bakker5121ce52009-01-03 21:22:43 +0000748
Hanno Beckerda1655a2017-10-18 14:21:44 +0100749
750/* Convert a big-endian byte array aligned to the size of mbedtls_mpi_uint
751 * into the storage form used by mbedtls_mpi. */
Hanno Beckerf8720072018-11-08 11:53:49 +0000752
753static mbedtls_mpi_uint mpi_uint_bigendian_to_host_c( mbedtls_mpi_uint x )
754{
755 uint8_t i;
Hanno Becker031d6332019-05-01 17:09:11 +0100756 unsigned char *x_ptr;
Hanno Beckerf8720072018-11-08 11:53:49 +0000757 mbedtls_mpi_uint tmp = 0;
Hanno Becker031d6332019-05-01 17:09:11 +0100758
759 for( i = 0, x_ptr = (unsigned char*) &x; i < ciL; i++, x_ptr++ )
760 {
761 tmp <<= CHAR_BIT;
762 tmp |= (mbedtls_mpi_uint) *x_ptr;
763 }
764
Hanno Beckerf8720072018-11-08 11:53:49 +0000765 return( tmp );
766}
767
768static mbedtls_mpi_uint mpi_uint_bigendian_to_host( mbedtls_mpi_uint x )
769{
770#if defined(__BYTE_ORDER__)
771
772/* Nothing to do on bigendian systems. */
773#if ( __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__ )
774 return( x );
775#endif /* __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__ */
776
777#if ( __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ )
778
779/* For GCC and Clang, have builtins for byte swapping. */
Hanno Becker9f6d16a2019-01-02 17:15:06 +0000780#if defined(__GNUC__) && defined(__GNUC_PREREQ)
781#if __GNUC_PREREQ(4,3)
Hanno Beckerf8720072018-11-08 11:53:49 +0000782#define have_bswap
783#endif
Hanno Becker9f6d16a2019-01-02 17:15:06 +0000784#endif
785
786#if defined(__clang__) && defined(__has_builtin)
787#if __has_builtin(__builtin_bswap32) && \
788 __has_builtin(__builtin_bswap64)
789#define have_bswap
790#endif
791#endif
792
Hanno Beckerf8720072018-11-08 11:53:49 +0000793#if defined(have_bswap)
794 /* The compiler is hopefully able to statically evaluate this! */
795 switch( sizeof(mbedtls_mpi_uint) )
796 {
797 case 4:
798 return( __builtin_bswap32(x) );
799 case 8:
800 return( __builtin_bswap64(x) );
801 }
802#endif
803#endif /* __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ */
804#endif /* __BYTE_ORDER__ */
805
806 /* Fall back to C-based reordering if we don't know the byte order
807 * or we couldn't use a compiler-specific builtin. */
808 return( mpi_uint_bigendian_to_host_c( x ) );
809}
810
Hanno Becker2be8a552018-10-25 12:40:09 +0100811static void mpi_bigendian_to_host( mbedtls_mpi_uint * const p, size_t limbs )
Hanno Beckerda1655a2017-10-18 14:21:44 +0100812{
Hanno Beckerda1655a2017-10-18 14:21:44 +0100813 mbedtls_mpi_uint *cur_limb_left;
814 mbedtls_mpi_uint *cur_limb_right;
Hanno Becker2be8a552018-10-25 12:40:09 +0100815 if( limbs == 0 )
816 return;
Hanno Beckerda1655a2017-10-18 14:21:44 +0100817
818 /*
819 * Traverse limbs and
820 * - adapt byte-order in each limb
821 * - swap the limbs themselves.
822 * For that, simultaneously traverse the limbs from left to right
823 * and from right to left, as long as the left index is not bigger
824 * than the right index (it's not a problem if limbs is odd and the
825 * indices coincide in the last iteration).
826 */
Hanno Beckerda1655a2017-10-18 14:21:44 +0100827 for( cur_limb_left = p, cur_limb_right = p + ( limbs - 1 );
828 cur_limb_left <= cur_limb_right;
829 cur_limb_left++, cur_limb_right-- )
830 {
Hanno Beckerf8720072018-11-08 11:53:49 +0000831 mbedtls_mpi_uint tmp;
832 /* Note that if cur_limb_left == cur_limb_right,
833 * this code effectively swaps the bytes only once. */
834 tmp = mpi_uint_bigendian_to_host( *cur_limb_left );
835 *cur_limb_left = mpi_uint_bigendian_to_host( *cur_limb_right );
836 *cur_limb_right = tmp;
Hanno Beckerda1655a2017-10-18 14:21:44 +0100837 }
Hanno Beckerda1655a2017-10-18 14:21:44 +0100838}
839
Paul Bakker5121ce52009-01-03 21:22:43 +0000840/*
Janos Follatha778a942019-02-13 10:28:28 +0000841 * Import X from unsigned binary data, little endian
842 */
843int mbedtls_mpi_read_binary_le( mbedtls_mpi *X,
844 const unsigned char *buf, size_t buflen )
845{
Janos Follath24eed8d2019-11-22 13:21:35 +0000846 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Janos Follatha778a942019-02-13 10:28:28 +0000847 size_t i;
848 size_t const limbs = CHARS_TO_LIMBS( buflen );
849
850 /* Ensure that target MPI has exactly the necessary number of limbs */
851 if( X->n != limbs )
852 {
853 mbedtls_mpi_free( X );
854 mbedtls_mpi_init( X );
855 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
856 }
857
858 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
859
860 for( i = 0; i < buflen; i++ )
861 X->p[i / ciL] |= ((mbedtls_mpi_uint) buf[i]) << ((i % ciL) << 3);
862
863cleanup:
864
Janos Follath171a7ef2019-02-15 16:17:45 +0000865 /*
866 * This function is also used to import keys. However, wiping the buffers
867 * upon failure is not necessary because failure only can happen before any
868 * input is copied.
869 */
Janos Follatha778a942019-02-13 10:28:28 +0000870 return( ret );
871}
872
873/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000874 * Import X from unsigned binary data, big endian
875 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200876int mbedtls_mpi_read_binary( mbedtls_mpi *X, const unsigned char *buf, size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000877{
Janos Follath24eed8d2019-11-22 13:21:35 +0000878 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Beckerda1655a2017-10-18 14:21:44 +0100879 size_t const limbs = CHARS_TO_LIMBS( buflen );
880 size_t const overhead = ( limbs * ciL ) - buflen;
881 unsigned char *Xp;
Paul Bakker5121ce52009-01-03 21:22:43 +0000882
Hanno Becker8ce11a32018-12-19 16:18:52 +0000883 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +0000884 MPI_VALIDATE_RET( buflen == 0 || buf != NULL );
885
Hanno Becker073c1992017-10-17 15:17:27 +0100886 /* Ensure that target MPI has exactly the necessary number of limbs */
887 if( X->n != limbs )
888 {
889 mbedtls_mpi_free( X );
890 mbedtls_mpi_init( X );
891 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
892 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200893 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000894
Hanno Becker0e810b92019-01-03 17:13:11 +0000895 /* Avoid calling `memcpy` with NULL source argument,
896 * even if buflen is 0. */
897 if( buf != NULL )
898 {
899 Xp = (unsigned char*) X->p;
900 memcpy( Xp + overhead, buf, buflen );
Hanno Beckerda1655a2017-10-18 14:21:44 +0100901
Hanno Becker0e810b92019-01-03 17:13:11 +0000902 mpi_bigendian_to_host( X->p, limbs );
903 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000904
905cleanup:
906
Janos Follath171a7ef2019-02-15 16:17:45 +0000907 /*
908 * This function is also used to import keys. However, wiping the buffers
909 * upon failure is not necessary because failure only can happen before any
910 * input is copied.
911 */
Paul Bakker5121ce52009-01-03 21:22:43 +0000912 return( ret );
913}
914
915/*
Janos Follathe344d0f2019-02-19 16:17:40 +0000916 * Export X into unsigned binary data, little endian
917 */
918int mbedtls_mpi_write_binary_le( const mbedtls_mpi *X,
919 unsigned char *buf, size_t buflen )
920{
921 size_t stored_bytes = X->n * ciL;
922 size_t bytes_to_copy;
923 size_t i;
924
925 if( stored_bytes < buflen )
926 {
927 bytes_to_copy = stored_bytes;
928 }
929 else
930 {
931 bytes_to_copy = buflen;
932
933 /* The output buffer is smaller than the allocated size of X.
934 * However X may fit if its leading bytes are zero. */
935 for( i = bytes_to_copy; i < stored_bytes; i++ )
936 {
937 if( GET_BYTE( X, i ) != 0 )
938 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
939 }
940 }
941
942 for( i = 0; i < bytes_to_copy; i++ )
943 buf[i] = GET_BYTE( X, i );
944
945 if( stored_bytes < buflen )
946 {
947 /* Write trailing 0 bytes */
948 memset( buf + stored_bytes, 0, buflen - stored_bytes );
949 }
950
951 return( 0 );
952}
953
954/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000955 * Export X into unsigned binary data, big endian
956 */
Gilles Peskine11cdb052018-11-20 16:47:47 +0100957int mbedtls_mpi_write_binary( const mbedtls_mpi *X,
958 unsigned char *buf, size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000959{
Hanno Becker73d7d792018-12-11 10:35:51 +0000960 size_t stored_bytes;
Gilles Peskine11cdb052018-11-20 16:47:47 +0100961 size_t bytes_to_copy;
962 unsigned char *p;
963 size_t i;
Paul Bakker5121ce52009-01-03 21:22:43 +0000964
Hanno Becker73d7d792018-12-11 10:35:51 +0000965 MPI_VALIDATE_RET( X != NULL );
966 MPI_VALIDATE_RET( buflen == 0 || buf != NULL );
967
968 stored_bytes = X->n * ciL;
969
Gilles Peskine11cdb052018-11-20 16:47:47 +0100970 if( stored_bytes < buflen )
971 {
972 /* There is enough space in the output buffer. Write initial
973 * null bytes and record the position at which to start
974 * writing the significant bytes. In this case, the execution
975 * trace of this function does not depend on the value of the
976 * number. */
977 bytes_to_copy = stored_bytes;
978 p = buf + buflen - stored_bytes;
979 memset( buf, 0, buflen - stored_bytes );
980 }
981 else
982 {
983 /* The output buffer is smaller than the allocated size of X.
984 * However X may fit if its leading bytes are zero. */
985 bytes_to_copy = buflen;
986 p = buf;
987 for( i = bytes_to_copy; i < stored_bytes; i++ )
988 {
989 if( GET_BYTE( X, i ) != 0 )
990 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
991 }
992 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000993
Gilles Peskine11cdb052018-11-20 16:47:47 +0100994 for( i = 0; i < bytes_to_copy; i++ )
995 p[bytes_to_copy - i - 1] = GET_BYTE( X, i );
Paul Bakker5121ce52009-01-03 21:22:43 +0000996
997 return( 0 );
998}
999
1000/*
1001 * Left-shift: X <<= count
1002 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001003int mbedtls_mpi_shift_l( mbedtls_mpi *X, size_t count )
Paul Bakker5121ce52009-01-03 21:22:43 +00001004{
Janos Follath24eed8d2019-11-22 13:21:35 +00001005 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001006 size_t i, v0, t1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001007 mbedtls_mpi_uint r0 = 0, r1;
Hanno Becker73d7d792018-12-11 10:35:51 +00001008 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001009
1010 v0 = count / (biL );
1011 t1 = count & (biL - 1);
1012
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001013 i = mbedtls_mpi_bitlen( X ) + count;
Paul Bakker5121ce52009-01-03 21:22:43 +00001014
Paul Bakkerf9688572011-05-05 10:00:45 +00001015 if( X->n * biL < i )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001016 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, BITS_TO_LIMBS( i ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001017
1018 ret = 0;
1019
1020 /*
1021 * shift by count / limb_size
1022 */
1023 if( v0 > 0 )
1024 {
Paul Bakker23986e52011-04-24 08:57:21 +00001025 for( i = X->n; i > v0; i-- )
1026 X->p[i - 1] = X->p[i - v0 - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001027
Paul Bakker23986e52011-04-24 08:57:21 +00001028 for( ; i > 0; i-- )
1029 X->p[i - 1] = 0;
Paul Bakker5121ce52009-01-03 21:22:43 +00001030 }
1031
1032 /*
1033 * shift by count % limb_size
1034 */
1035 if( t1 > 0 )
1036 {
1037 for( i = v0; i < X->n; i++ )
1038 {
1039 r1 = X->p[i] >> (biL - t1);
1040 X->p[i] <<= t1;
1041 X->p[i] |= r0;
1042 r0 = r1;
1043 }
1044 }
1045
1046cleanup:
1047
1048 return( ret );
1049}
1050
1051/*
1052 * Right-shift: X >>= count
1053 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001054int mbedtls_mpi_shift_r( mbedtls_mpi *X, size_t count )
Paul Bakker5121ce52009-01-03 21:22:43 +00001055{
Paul Bakker23986e52011-04-24 08:57:21 +00001056 size_t i, v0, v1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001057 mbedtls_mpi_uint r0 = 0, r1;
Hanno Becker73d7d792018-12-11 10:35:51 +00001058 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001059
1060 v0 = count / biL;
1061 v1 = count & (biL - 1);
1062
Manuel Pégourié-Gonnarde44ec102012-11-17 12:42:51 +01001063 if( v0 > X->n || ( v0 == X->n && v1 > 0 ) )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001064 return mbedtls_mpi_lset( X, 0 );
Manuel Pégourié-Gonnarde44ec102012-11-17 12:42:51 +01001065
Paul Bakker5121ce52009-01-03 21:22:43 +00001066 /*
1067 * shift by count / limb_size
1068 */
1069 if( v0 > 0 )
1070 {
1071 for( i = 0; i < X->n - v0; i++ )
1072 X->p[i] = X->p[i + v0];
1073
1074 for( ; i < X->n; i++ )
1075 X->p[i] = 0;
1076 }
1077
1078 /*
1079 * shift by count % limb_size
1080 */
1081 if( v1 > 0 )
1082 {
Paul Bakker23986e52011-04-24 08:57:21 +00001083 for( i = X->n; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001084 {
Paul Bakker23986e52011-04-24 08:57:21 +00001085 r1 = X->p[i - 1] << (biL - v1);
1086 X->p[i - 1] >>= v1;
1087 X->p[i - 1] |= r0;
Paul Bakker5121ce52009-01-03 21:22:43 +00001088 r0 = r1;
1089 }
1090 }
1091
1092 return( 0 );
1093}
1094
1095/*
1096 * Compare unsigned values
1097 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001098int mbedtls_mpi_cmp_abs( const mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +00001099{
Paul Bakker23986e52011-04-24 08:57:21 +00001100 size_t i, j;
Hanno Becker73d7d792018-12-11 10:35:51 +00001101 MPI_VALIDATE_RET( X != NULL );
1102 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001103
Paul Bakker23986e52011-04-24 08:57:21 +00001104 for( i = X->n; i > 0; i-- )
1105 if( X->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001106 break;
1107
Paul Bakker23986e52011-04-24 08:57:21 +00001108 for( j = Y->n; j > 0; j-- )
1109 if( Y->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001110 break;
1111
Paul Bakker23986e52011-04-24 08:57:21 +00001112 if( i == 0 && j == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001113 return( 0 );
1114
1115 if( i > j ) return( 1 );
1116 if( j > i ) return( -1 );
1117
Paul Bakker23986e52011-04-24 08:57:21 +00001118 for( ; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001119 {
Paul Bakker23986e52011-04-24 08:57:21 +00001120 if( X->p[i - 1] > Y->p[i - 1] ) return( 1 );
1121 if( X->p[i - 1] < Y->p[i - 1] ) return( -1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001122 }
1123
1124 return( 0 );
1125}
1126
1127/*
1128 * Compare signed values
1129 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001130int mbedtls_mpi_cmp_mpi( const mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +00001131{
Paul Bakker23986e52011-04-24 08:57:21 +00001132 size_t i, j;
Hanno Becker73d7d792018-12-11 10:35:51 +00001133 MPI_VALIDATE_RET( X != NULL );
1134 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001135
Paul Bakker23986e52011-04-24 08:57:21 +00001136 for( i = X->n; i > 0; i-- )
1137 if( X->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001138 break;
1139
Paul Bakker23986e52011-04-24 08:57:21 +00001140 for( j = Y->n; j > 0; j-- )
1141 if( Y->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001142 break;
1143
Paul Bakker23986e52011-04-24 08:57:21 +00001144 if( i == 0 && j == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001145 return( 0 );
1146
1147 if( i > j ) return( X->s );
Paul Bakker0c8f73b2012-03-22 14:08:57 +00001148 if( j > i ) return( -Y->s );
Paul Bakker5121ce52009-01-03 21:22:43 +00001149
1150 if( X->s > 0 && Y->s < 0 ) return( 1 );
1151 if( Y->s > 0 && X->s < 0 ) return( -1 );
1152
Paul Bakker23986e52011-04-24 08:57:21 +00001153 for( ; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001154 {
Paul Bakker23986e52011-04-24 08:57:21 +00001155 if( X->p[i - 1] > Y->p[i - 1] ) return( X->s );
1156 if( X->p[i - 1] < Y->p[i - 1] ) return( -X->s );
Paul Bakker5121ce52009-01-03 21:22:43 +00001157 }
1158
1159 return( 0 );
1160}
1161
Janos Follath3f6f0e42019-10-14 09:09:32 +01001162/** Decide if an integer is less than the other, without branches.
1163 *
1164 * \param x First integer.
1165 * \param y Second integer.
1166 *
1167 * \return 1 if \p x is less than \p y, 0 otherwise
1168 */
Janos Follath0e5532d2019-10-11 14:21:53 +01001169static unsigned ct_lt_mpi_uint( const mbedtls_mpi_uint x,
1170 const mbedtls_mpi_uint y )
Janos Follathee6abce2019-09-05 14:47:19 +01001171{
1172 mbedtls_mpi_uint ret;
1173 mbedtls_mpi_uint cond;
1174
1175 /*
1176 * Check if the most significant bits (MSB) of the operands are different.
1177 */
1178 cond = ( x ^ y );
1179 /*
1180 * If the MSB are the same then the difference x-y will be negative (and
1181 * have its MSB set to 1 during conversion to unsigned) if and only if x<y.
1182 */
1183 ret = ( x - y ) & ~cond;
1184 /*
1185 * If the MSB are different, then the operand with the MSB of 1 is the
1186 * bigger. (That is if y has MSB of 1, then x<y is true and it is false if
1187 * the MSB of y is 0.)
1188 */
1189 ret |= y & cond;
1190
1191
Janos Follatha0f732b2019-10-14 08:59:14 +01001192 ret = ret >> ( biL - 1 );
Janos Follathee6abce2019-09-05 14:47:19 +01001193
Janos Follath67ce6472019-10-29 15:08:46 +00001194 return (unsigned) ret;
Janos Follathee6abce2019-09-05 14:47:19 +01001195}
1196
1197/*
1198 * Compare signed values in constant time
1199 */
Janos Follath0e5532d2019-10-11 14:21:53 +01001200int mbedtls_mpi_lt_mpi_ct( const mbedtls_mpi *X, const mbedtls_mpi *Y,
1201 unsigned *ret )
Janos Follathee6abce2019-09-05 14:47:19 +01001202{
1203 size_t i;
Janos Follathbb5147f2019-10-28 12:23:18 +00001204 /* The value of any of these variables is either 0 or 1 at all times. */
Janos Follath5e614ce2019-10-28 12:31:34 +00001205 unsigned cond, done, X_is_negative, Y_is_negative;
Janos Follathee6abce2019-09-05 14:47:19 +01001206
1207 MPI_VALIDATE_RET( X != NULL );
1208 MPI_VALIDATE_RET( Y != NULL );
1209 MPI_VALIDATE_RET( ret != NULL );
1210
1211 if( X->n != Y->n )
1212 return MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
1213
1214 /*
Janos Follath73ba9ec2019-10-28 12:12:15 +00001215 * Set sign_N to 1 if N >= 0, 0 if N < 0.
1216 * We know that N->s == 1 if N >= 0 and N->s == -1 if N < 0.
Janos Follathee6abce2019-09-05 14:47:19 +01001217 */
Janos Follath5e614ce2019-10-28 12:31:34 +00001218 X_is_negative = ( X->s & 2 ) >> 1;
1219 Y_is_negative = ( Y->s & 2 ) >> 1;
Janos Follath0e5532d2019-10-11 14:21:53 +01001220
1221 /*
1222 * If the signs are different, then the positive operand is the bigger.
Janos Follath5e614ce2019-10-28 12:31:34 +00001223 * That is if X is negative (X_is_negative == 1), then X < Y is true and it
1224 * is false if X is positive (X_is_negative == 0).
Janos Follath0e5532d2019-10-11 14:21:53 +01001225 */
Janos Follath5e614ce2019-10-28 12:31:34 +00001226 cond = ( X_is_negative ^ Y_is_negative );
1227 *ret = cond & X_is_negative;
Janos Follath0e5532d2019-10-11 14:21:53 +01001228
1229 /*
Janos Follathbb5147f2019-10-28 12:23:18 +00001230 * This is a constant-time function. We might have the result, but we still
Janos Follath0e5532d2019-10-11 14:21:53 +01001231 * need to go through the loop. Record if we have the result already.
1232 */
Janos Follathee6abce2019-09-05 14:47:19 +01001233 done = cond;
1234
1235 for( i = X->n; i > 0; i-- )
1236 {
1237 /*
Janos Follath30702422019-11-05 12:24:52 +00001238 * If Y->p[i - 1] < X->p[i - 1] then X < Y is true if and only if both
1239 * X and Y are negative.
Janos Follath0e5532d2019-10-11 14:21:53 +01001240 *
1241 * Again even if we can make a decision, we just mark the result and
1242 * the fact that we are done and continue looping.
Janos Follathee6abce2019-09-05 14:47:19 +01001243 */
Janos Follath30702422019-11-05 12:24:52 +00001244 cond = ct_lt_mpi_uint( Y->p[i - 1], X->p[i - 1] );
1245 *ret |= cond & ( 1 - done ) & X_is_negative;
Janos Follathc50e6d52019-10-28 12:37:21 +00001246 done |= cond;
Janos Follathee6abce2019-09-05 14:47:19 +01001247
1248 /*
Janos Follath30702422019-11-05 12:24:52 +00001249 * If X->p[i - 1] < Y->p[i - 1] then X < Y is true if and only if both
1250 * X and Y are positive.
Janos Follath0e5532d2019-10-11 14:21:53 +01001251 *
1252 * Again even if we can make a decision, we just mark the result and
1253 * the fact that we are done and continue looping.
Janos Follathee6abce2019-09-05 14:47:19 +01001254 */
Janos Follath30702422019-11-05 12:24:52 +00001255 cond = ct_lt_mpi_uint( X->p[i - 1], Y->p[i - 1] );
1256 *ret |= cond & ( 1 - done ) & ( 1 - X_is_negative );
Janos Follathc50e6d52019-10-28 12:37:21 +00001257 done |= cond;
Janos Follathee6abce2019-09-05 14:47:19 +01001258 }
1259
1260 return( 0 );
1261}
1262
Paul Bakker5121ce52009-01-03 21:22:43 +00001263/*
1264 * Compare signed values
1265 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001266int mbedtls_mpi_cmp_int( const mbedtls_mpi *X, mbedtls_mpi_sint z )
Paul Bakker5121ce52009-01-03 21:22:43 +00001267{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001268 mbedtls_mpi Y;
1269 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001270 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001271
1272 *p = ( z < 0 ) ? -z : z;
1273 Y.s = ( z < 0 ) ? -1 : 1;
1274 Y.n = 1;
1275 Y.p = p;
1276
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001277 return( mbedtls_mpi_cmp_mpi( X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001278}
1279
1280/*
1281 * Unsigned addition: X = |A| + |B| (HAC 14.7)
1282 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001283int mbedtls_mpi_add_abs( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001284{
Janos Follath24eed8d2019-11-22 13:21:35 +00001285 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001286 size_t i, j;
Janos Follath6c922682015-10-30 17:43:11 +01001287 mbedtls_mpi_uint *o, *p, c, tmp;
Hanno Becker73d7d792018-12-11 10:35:51 +00001288 MPI_VALIDATE_RET( X != NULL );
1289 MPI_VALIDATE_RET( A != NULL );
1290 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001291
1292 if( X == B )
1293 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001294 const mbedtls_mpi *T = A; A = X; B = T;
Paul Bakker5121ce52009-01-03 21:22:43 +00001295 }
1296
1297 if( X != A )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001298 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, A ) );
Paul Bakker9af723c2014-05-01 13:03:14 +02001299
Paul Bakkerf7ca7b92009-06-20 10:31:06 +00001300 /*
1301 * X should always be positive as a result of unsigned additions.
1302 */
1303 X->s = 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00001304
Paul Bakker23986e52011-04-24 08:57:21 +00001305 for( j = B->n; j > 0; j-- )
1306 if( B->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001307 break;
1308
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001309 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001310
1311 o = B->p; p = X->p; c = 0;
1312
Janos Follath6c922682015-10-30 17:43:11 +01001313 /*
1314 * tmp is used because it might happen that p == o
1315 */
Paul Bakker23986e52011-04-24 08:57:21 +00001316 for( i = 0; i < j; i++, o++, p++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00001317 {
Janos Follath6c922682015-10-30 17:43:11 +01001318 tmp= *o;
Paul Bakker5121ce52009-01-03 21:22:43 +00001319 *p += c; c = ( *p < c );
Janos Follath6c922682015-10-30 17:43:11 +01001320 *p += tmp; c += ( *p < tmp );
Paul Bakker5121ce52009-01-03 21:22:43 +00001321 }
1322
1323 while( c != 0 )
1324 {
1325 if( i >= X->n )
1326 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001327 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001328 p = X->p + i;
1329 }
1330
Paul Bakker2d319fd2012-09-16 21:34:26 +00001331 *p += c; c = ( *p < c ); i++; p++;
Paul Bakker5121ce52009-01-03 21:22:43 +00001332 }
1333
1334cleanup:
1335
1336 return( ret );
1337}
1338
Gilles Peskine09ec10a2020-06-09 10:39:38 +02001339/**
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001340 * Helper for mbedtls_mpi subtraction.
1341 *
1342 * Calculate d - s where d and s have the same size.
1343 * This function operates modulo (2^ciL)^n and returns the carry
1344 * (1 if there was a wraparound, i.e. if `d < s`, and 0 otherwise).
1345 *
1346 * \param n Number of limbs of \p d and \p s.
1347 * \param[in,out] d On input, the left operand.
1348 * On output, the result of the subtraction:
Gilles Peskine09ec10a2020-06-09 10:39:38 +02001349 * \param[in] s The right operand.
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001350 *
1351 * \return 1 if `d < s`.
1352 * 0 if `d >= s`.
Paul Bakker5121ce52009-01-03 21:22:43 +00001353 */
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001354static mbedtls_mpi_uint mpi_sub_hlp( size_t n,
1355 mbedtls_mpi_uint *d,
1356 const mbedtls_mpi_uint *s )
Paul Bakker5121ce52009-01-03 21:22:43 +00001357{
Paul Bakker23986e52011-04-24 08:57:21 +00001358 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001359 mbedtls_mpi_uint c, z;
Paul Bakker5121ce52009-01-03 21:22:43 +00001360
1361 for( i = c = 0; i < n; i++, s++, d++ )
1362 {
1363 z = ( *d < c ); *d -= c;
1364 c = ( *d < *s ) + z; *d -= *s;
1365 }
1366
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001367 return( c );
Paul Bakker5121ce52009-01-03 21:22:43 +00001368}
1369
1370/*
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001371 * Unsigned subtraction: X = |A| - |B| (HAC 14.9, 14.10)
Paul Bakker5121ce52009-01-03 21:22:43 +00001372 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001373int mbedtls_mpi_sub_abs( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001374{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001375 mbedtls_mpi TB;
Janos Follath24eed8d2019-11-22 13:21:35 +00001376 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001377 size_t n;
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001378 mbedtls_mpi_uint carry;
Hanno Becker73d7d792018-12-11 10:35:51 +00001379 MPI_VALIDATE_RET( X != NULL );
1380 MPI_VALIDATE_RET( A != NULL );
1381 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001382
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001383 mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001384
1385 if( X == B )
1386 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001387 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001388 B = &TB;
1389 }
1390
1391 if( X != A )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001392 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001393
Paul Bakker1ef7a532009-06-20 10:50:55 +00001394 /*
Paul Bakker60b1d102013-10-29 10:02:51 +01001395 * X should always be positive as a result of unsigned subtractions.
Paul Bakker1ef7a532009-06-20 10:50:55 +00001396 */
1397 X->s = 1;
1398
Paul Bakker5121ce52009-01-03 21:22:43 +00001399 ret = 0;
1400
Paul Bakker23986e52011-04-24 08:57:21 +00001401 for( n = B->n; n > 0; n-- )
1402 if( B->p[n - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001403 break;
1404
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001405 carry = mpi_sub_hlp( n, X->p, B->p );
1406 if( carry != 0 )
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001407 {
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001408 /* Propagate the carry to the first nonzero limb of X. */
1409 for( ; n < X->n && X->p[n] == 0; n++ )
1410 --X->p[n];
1411 /* If we ran out of space for the carry, it means that the result
1412 * is negative. */
1413 if( n == X->n )
Gilles Peskine89b41302020-07-23 01:16:46 +02001414 {
1415 ret = MBEDTLS_ERR_MPI_NEGATIVE_VALUE;
1416 goto cleanup;
1417 }
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001418 --X->p[n];
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001419 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001420
1421cleanup:
1422
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001423 mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001424
1425 return( ret );
1426}
1427
1428/*
1429 * Signed addition: X = A + B
1430 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001431int mbedtls_mpi_add_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001432{
Hanno Becker73d7d792018-12-11 10:35:51 +00001433 int ret, s;
1434 MPI_VALIDATE_RET( X != NULL );
1435 MPI_VALIDATE_RET( A != NULL );
1436 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001437
Hanno Becker73d7d792018-12-11 10:35:51 +00001438 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001439 if( A->s * B->s < 0 )
1440 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001441 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001442 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001443 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001444 X->s = s;
1445 }
1446 else
1447 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001448 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001449 X->s = -s;
1450 }
1451 }
1452 else
1453 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001454 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001455 X->s = s;
1456 }
1457
1458cleanup:
1459
1460 return( ret );
1461}
1462
1463/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001464 * Signed subtraction: X = A - B
Paul Bakker5121ce52009-01-03 21:22:43 +00001465 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001466int mbedtls_mpi_sub_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001467{
Hanno Becker73d7d792018-12-11 10:35:51 +00001468 int ret, s;
1469 MPI_VALIDATE_RET( X != NULL );
1470 MPI_VALIDATE_RET( A != NULL );
1471 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001472
Hanno Becker73d7d792018-12-11 10:35:51 +00001473 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001474 if( A->s * B->s > 0 )
1475 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001476 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001477 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001478 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001479 X->s = s;
1480 }
1481 else
1482 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001483 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001484 X->s = -s;
1485 }
1486 }
1487 else
1488 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001489 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001490 X->s = s;
1491 }
1492
1493cleanup:
1494
1495 return( ret );
1496}
1497
1498/*
1499 * Signed addition: X = A + b
1500 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001501int mbedtls_mpi_add_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001502{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001503 mbedtls_mpi _B;
1504 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001505 MPI_VALIDATE_RET( X != NULL );
1506 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001507
1508 p[0] = ( b < 0 ) ? -b : b;
1509 _B.s = ( b < 0 ) ? -1 : 1;
1510 _B.n = 1;
1511 _B.p = p;
1512
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001513 return( mbedtls_mpi_add_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001514}
1515
1516/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001517 * Signed subtraction: X = A - b
Paul Bakker5121ce52009-01-03 21:22:43 +00001518 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001519int mbedtls_mpi_sub_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001520{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001521 mbedtls_mpi _B;
1522 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001523 MPI_VALIDATE_RET( X != NULL );
1524 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001525
1526 p[0] = ( b < 0 ) ? -b : b;
1527 _B.s = ( b < 0 ) ? -1 : 1;
1528 _B.n = 1;
1529 _B.p = p;
1530
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001531 return( mbedtls_mpi_sub_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001532}
1533
1534/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001535 * Helper for mbedtls_mpi multiplication
Paul Bakkerfc4f46f2013-06-24 19:23:56 +02001536 */
1537static
1538#if defined(__APPLE__) && defined(__arm__)
1539/*
1540 * Apple LLVM version 4.2 (clang-425.0.24) (based on LLVM 3.2svn)
1541 * appears to need this to prevent bad ARM code generation at -O3.
1542 */
1543__attribute__ ((noinline))
1544#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001545void mpi_mul_hlp( size_t i, mbedtls_mpi_uint *s, mbedtls_mpi_uint *d, mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001546{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001547 mbedtls_mpi_uint c = 0, t = 0;
Paul Bakker5121ce52009-01-03 21:22:43 +00001548
1549#if defined(MULADDC_HUIT)
1550 for( ; i >= 8; i -= 8 )
1551 {
1552 MULADDC_INIT
1553 MULADDC_HUIT
1554 MULADDC_STOP
1555 }
1556
1557 for( ; i > 0; i-- )
1558 {
1559 MULADDC_INIT
1560 MULADDC_CORE
1561 MULADDC_STOP
1562 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001563#else /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001564 for( ; i >= 16; i -= 16 )
1565 {
1566 MULADDC_INIT
1567 MULADDC_CORE MULADDC_CORE
1568 MULADDC_CORE MULADDC_CORE
1569 MULADDC_CORE MULADDC_CORE
1570 MULADDC_CORE MULADDC_CORE
1571
1572 MULADDC_CORE MULADDC_CORE
1573 MULADDC_CORE MULADDC_CORE
1574 MULADDC_CORE MULADDC_CORE
1575 MULADDC_CORE MULADDC_CORE
1576 MULADDC_STOP
1577 }
1578
1579 for( ; i >= 8; i -= 8 )
1580 {
1581 MULADDC_INIT
1582 MULADDC_CORE MULADDC_CORE
1583 MULADDC_CORE MULADDC_CORE
1584
1585 MULADDC_CORE MULADDC_CORE
1586 MULADDC_CORE MULADDC_CORE
1587 MULADDC_STOP
1588 }
1589
1590 for( ; i > 0; i-- )
1591 {
1592 MULADDC_INIT
1593 MULADDC_CORE
1594 MULADDC_STOP
1595 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001596#endif /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001597
1598 t++;
1599
1600 do {
1601 *d += c; c = ( *d < c ); d++;
1602 }
1603 while( c != 0 );
1604}
1605
1606/*
1607 * Baseline multiplication: X = A * B (HAC 14.12)
1608 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001609int mbedtls_mpi_mul_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001610{
Janos Follath24eed8d2019-11-22 13:21:35 +00001611 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001612 size_t i, j;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001613 mbedtls_mpi TA, TB;
Hanno Becker73d7d792018-12-11 10:35:51 +00001614 MPI_VALIDATE_RET( X != NULL );
1615 MPI_VALIDATE_RET( A != NULL );
1616 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001617
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001618 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001619
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001620 if( X == A ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) ); A = &TA; }
1621 if( X == B ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) ); B = &TB; }
Paul Bakker5121ce52009-01-03 21:22:43 +00001622
Paul Bakker23986e52011-04-24 08:57:21 +00001623 for( i = A->n; i > 0; i-- )
1624 if( A->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001625 break;
1626
Paul Bakker23986e52011-04-24 08:57:21 +00001627 for( j = B->n; j > 0; j-- )
1628 if( B->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001629 break;
1630
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001631 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + j ) );
1632 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001633
Alexey Skalozub8e75e682016-01-13 21:59:27 +02001634 for( ; j > 0; j-- )
1635 mpi_mul_hlp( i, A->p, X->p + j - 1, B->p[j - 1] );
Paul Bakker5121ce52009-01-03 21:22:43 +00001636
1637 X->s = A->s * B->s;
1638
1639cleanup:
1640
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001641 mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TA );
Paul Bakker5121ce52009-01-03 21:22:43 +00001642
1643 return( ret );
1644}
1645
1646/*
1647 * Baseline multiplication: X = A * b
1648 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001649int mbedtls_mpi_mul_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001650{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001651 mbedtls_mpi _B;
1652 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001653 MPI_VALIDATE_RET( X != NULL );
1654 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001655
1656 _B.s = 1;
1657 _B.n = 1;
1658 _B.p = p;
1659 p[0] = b;
1660
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001661 return( mbedtls_mpi_mul_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001662}
1663
1664/*
Simon Butcherf5ba0452015-12-27 23:01:55 +00001665 * Unsigned integer divide - double mbedtls_mpi_uint dividend, u1/u0, and
1666 * mbedtls_mpi_uint divisor, d
Simon Butcher15b15d12015-11-26 19:35:03 +00001667 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001668static mbedtls_mpi_uint mbedtls_int_div_int( mbedtls_mpi_uint u1,
1669 mbedtls_mpi_uint u0, mbedtls_mpi_uint d, mbedtls_mpi_uint *r )
Simon Butcher15b15d12015-11-26 19:35:03 +00001670{
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001671#if defined(MBEDTLS_HAVE_UDBL)
1672 mbedtls_t_udbl dividend, quotient;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001673#else
Simon Butcher9803d072016-01-03 00:24:34 +00001674 const mbedtls_mpi_uint radix = (mbedtls_mpi_uint) 1 << biH;
1675 const mbedtls_mpi_uint uint_halfword_mask = ( (mbedtls_mpi_uint) 1 << biH ) - 1;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001676 mbedtls_mpi_uint d0, d1, q0, q1, rAX, r0, quotient;
1677 mbedtls_mpi_uint u0_msw, u0_lsw;
Simon Butcher9803d072016-01-03 00:24:34 +00001678 size_t s;
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001679#endif
1680
Simon Butcher15b15d12015-11-26 19:35:03 +00001681 /*
1682 * Check for overflow
1683 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001684 if( 0 == d || u1 >= d )
Simon Butcher15b15d12015-11-26 19:35:03 +00001685 {
Simon Butcherf5ba0452015-12-27 23:01:55 +00001686 if (r != NULL) *r = ~0;
Simon Butcher15b15d12015-11-26 19:35:03 +00001687
Simon Butcherf5ba0452015-12-27 23:01:55 +00001688 return ( ~0 );
Simon Butcher15b15d12015-11-26 19:35:03 +00001689 }
1690
1691#if defined(MBEDTLS_HAVE_UDBL)
Simon Butcher15b15d12015-11-26 19:35:03 +00001692 dividend = (mbedtls_t_udbl) u1 << biL;
1693 dividend |= (mbedtls_t_udbl) u0;
1694 quotient = dividend / d;
1695 if( quotient > ( (mbedtls_t_udbl) 1 << biL ) - 1 )
1696 quotient = ( (mbedtls_t_udbl) 1 << biL ) - 1;
1697
1698 if( r != NULL )
Simon Butcher9803d072016-01-03 00:24:34 +00001699 *r = (mbedtls_mpi_uint)( dividend - (quotient * d ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001700
1701 return (mbedtls_mpi_uint) quotient;
1702#else
Simon Butcher15b15d12015-11-26 19:35:03 +00001703
1704 /*
1705 * Algorithm D, Section 4.3.1 - The Art of Computer Programming
1706 * Vol. 2 - Seminumerical Algorithms, Knuth
1707 */
1708
1709 /*
1710 * Normalize the divisor, d, and dividend, u0, u1
1711 */
1712 s = mbedtls_clz( d );
1713 d = d << s;
1714
1715 u1 = u1 << s;
Simon Butcher9803d072016-01-03 00:24:34 +00001716 u1 |= ( u0 >> ( biL - s ) ) & ( -(mbedtls_mpi_sint)s >> ( biL - 1 ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001717 u0 = u0 << s;
1718
1719 d1 = d >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001720 d0 = d & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001721
1722 u0_msw = u0 >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001723 u0_lsw = u0 & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001724
1725 /*
1726 * Find the first quotient and remainder
1727 */
1728 q1 = u1 / d1;
1729 r0 = u1 - d1 * q1;
1730
1731 while( q1 >= radix || ( q1 * d0 > radix * r0 + u0_msw ) )
1732 {
1733 q1 -= 1;
1734 r0 += d1;
1735
1736 if ( r0 >= radix ) break;
1737 }
1738
Simon Butcherf5ba0452015-12-27 23:01:55 +00001739 rAX = ( u1 * radix ) + ( u0_msw - q1 * d );
Simon Butcher15b15d12015-11-26 19:35:03 +00001740 q0 = rAX / d1;
1741 r0 = rAX - q0 * d1;
1742
1743 while( q0 >= radix || ( q0 * d0 > radix * r0 + u0_lsw ) )
1744 {
1745 q0 -= 1;
1746 r0 += d1;
1747
1748 if ( r0 >= radix ) break;
1749 }
1750
1751 if (r != NULL)
Simon Butcherf5ba0452015-12-27 23:01:55 +00001752 *r = ( rAX * radix + u0_lsw - q0 * d ) >> s;
Simon Butcher15b15d12015-11-26 19:35:03 +00001753
1754 quotient = q1 * radix + q0;
1755
1756 return quotient;
1757#endif
1758}
1759
1760/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001761 * Division by mbedtls_mpi: A = Q * B + R (HAC 14.20)
Paul Bakker5121ce52009-01-03 21:22:43 +00001762 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001763int mbedtls_mpi_div_mpi( mbedtls_mpi *Q, mbedtls_mpi *R, const mbedtls_mpi *A,
1764 const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001765{
Janos Follath24eed8d2019-11-22 13:21:35 +00001766 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001767 size_t i, n, t, k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001768 mbedtls_mpi X, Y, Z, T1, T2;
Alexander Kd19a1932019-11-01 18:20:42 +03001769 mbedtls_mpi_uint TP2[3];
Hanno Becker73d7d792018-12-11 10:35:51 +00001770 MPI_VALIDATE_RET( A != NULL );
1771 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001772
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001773 if( mbedtls_mpi_cmp_int( B, 0 ) == 0 )
1774 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001775
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001776 mbedtls_mpi_init( &X ); mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001777 mbedtls_mpi_init( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001778 /*
1779 * Avoid dynamic memory allocations for constant-size T2.
1780 *
1781 * T2 is used for comparison only and the 3 limbs are assigned explicitly,
1782 * so nobody increase the size of the MPI and we're safe to use an on-stack
1783 * buffer.
1784 */
Alexander K35d6d462019-10-31 14:46:45 +03001785 T2.s = 1;
Alexander Kd19a1932019-11-01 18:20:42 +03001786 T2.n = sizeof( TP2 ) / sizeof( *TP2 );
1787 T2.p = TP2;
Paul Bakker5121ce52009-01-03 21:22:43 +00001788
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001789 if( mbedtls_mpi_cmp_abs( A, B ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001790 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001791 if( Q != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_lset( Q, 0 ) );
1792 if( R != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001793 return( 0 );
1794 }
1795
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001796 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &X, A ) );
1797 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001798 X.s = Y.s = 1;
1799
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001800 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &Z, A->n + 2 ) );
1801 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Z, 0 ) );
1802 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T1, 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001803
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001804 k = mbedtls_mpi_bitlen( &Y ) % biL;
Paul Bakkerf9688572011-05-05 10:00:45 +00001805 if( k < biL - 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001806 {
1807 k = biL - 1 - k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001808 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &X, k ) );
1809 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, k ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001810 }
1811 else k = 0;
1812
1813 n = X.n - 1;
1814 t = Y.n - 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001815 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001816
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001817 while( mbedtls_mpi_cmp_mpi( &X, &Y ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001818 {
1819 Z.p[n - t]++;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001820 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001821 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001822 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001823
1824 for( i = n; i > t ; i-- )
1825 {
1826 if( X.p[i] >= Y.p[t] )
1827 Z.p[i - t - 1] = ~0;
1828 else
1829 {
Simon Butcher15b15d12015-11-26 19:35:03 +00001830 Z.p[i - t - 1] = mbedtls_int_div_int( X.p[i], X.p[i - 1],
1831 Y.p[t], NULL);
Paul Bakker5121ce52009-01-03 21:22:43 +00001832 }
1833
Alexander K35d6d462019-10-31 14:46:45 +03001834 T2.p[0] = ( i < 2 ) ? 0 : X.p[i - 2];
1835 T2.p[1] = ( i < 1 ) ? 0 : X.p[i - 1];
1836 T2.p[2] = X.p[i];
1837
Paul Bakker5121ce52009-01-03 21:22:43 +00001838 Z.p[i - t - 1]++;
1839 do
1840 {
1841 Z.p[i - t - 1]--;
1842
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001843 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &T1, 0 ) );
Paul Bakker66d5d072014-06-17 16:39:18 +02001844 T1.p[0] = ( t < 1 ) ? 0 : Y.p[t - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001845 T1.p[1] = Y.p[t];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001846 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &T1, Z.p[i - t - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001847 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001848 while( mbedtls_mpi_cmp_mpi( &T1, &T2 ) > 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001849
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001850 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &Y, Z.p[i - t - 1] ) );
1851 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1852 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001853
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001854 if( mbedtls_mpi_cmp_int( &X, 0 ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001855 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001856 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T1, &Y ) );
1857 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1858 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001859 Z.p[i - t - 1]--;
1860 }
1861 }
1862
1863 if( Q != NULL )
1864 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001865 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( Q, &Z ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001866 Q->s = A->s * B->s;
1867 }
1868
1869 if( R != NULL )
1870 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001871 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &X, k ) );
Paul Bakkerf02c5642012-11-13 10:25:21 +00001872 X.s = A->s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001873 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, &X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001874
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001875 if( mbedtls_mpi_cmp_int( R, 0 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001876 R->s = 1;
1877 }
1878
1879cleanup:
1880
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001881 mbedtls_mpi_free( &X ); mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001882 mbedtls_mpi_free( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001883 mbedtls_platform_zeroize( TP2, sizeof( TP2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001884
1885 return( ret );
1886}
1887
1888/*
1889 * Division by int: A = Q * b + R
Paul Bakker5121ce52009-01-03 21:22:43 +00001890 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001891int mbedtls_mpi_div_int( mbedtls_mpi *Q, mbedtls_mpi *R,
1892 const mbedtls_mpi *A,
1893 mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001894{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001895 mbedtls_mpi _B;
1896 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001897 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001898
1899 p[0] = ( b < 0 ) ? -b : b;
1900 _B.s = ( b < 0 ) ? -1 : 1;
1901 _B.n = 1;
1902 _B.p = p;
1903
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001904 return( mbedtls_mpi_div_mpi( Q, R, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001905}
1906
1907/*
1908 * Modulo: R = A mod B
1909 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001910int mbedtls_mpi_mod_mpi( mbedtls_mpi *R, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001911{
Janos Follath24eed8d2019-11-22 13:21:35 +00001912 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker73d7d792018-12-11 10:35:51 +00001913 MPI_VALIDATE_RET( R != NULL );
1914 MPI_VALIDATE_RET( A != NULL );
1915 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001916
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001917 if( mbedtls_mpi_cmp_int( B, 0 ) < 0 )
1918 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001919
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001920 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( NULL, R, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001921
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001922 while( mbedtls_mpi_cmp_int( R, 0 ) < 0 )
1923 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001924
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001925 while( mbedtls_mpi_cmp_mpi( R, B ) >= 0 )
1926 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001927
1928cleanup:
1929
1930 return( ret );
1931}
1932
1933/*
1934 * Modulo: r = A mod b
1935 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001936int mbedtls_mpi_mod_int( mbedtls_mpi_uint *r, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001937{
Paul Bakker23986e52011-04-24 08:57:21 +00001938 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001939 mbedtls_mpi_uint x, y, z;
Hanno Becker73d7d792018-12-11 10:35:51 +00001940 MPI_VALIDATE_RET( r != NULL );
1941 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001942
1943 if( b == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001944 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001945
1946 if( b < 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001947 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakker5121ce52009-01-03 21:22:43 +00001948
1949 /*
1950 * handle trivial cases
1951 */
1952 if( b == 1 )
1953 {
1954 *r = 0;
1955 return( 0 );
1956 }
1957
1958 if( b == 2 )
1959 {
1960 *r = A->p[0] & 1;
1961 return( 0 );
1962 }
1963
1964 /*
1965 * general case
1966 */
Paul Bakker23986e52011-04-24 08:57:21 +00001967 for( i = A->n, y = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001968 {
Paul Bakker23986e52011-04-24 08:57:21 +00001969 x = A->p[i - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001970 y = ( y << biH ) | ( x >> biH );
1971 z = y / b;
1972 y -= z * b;
1973
1974 x <<= biH;
1975 y = ( y << biH ) | ( x >> biH );
1976 z = y / b;
1977 y -= z * b;
1978 }
1979
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001980 /*
1981 * If A is negative, then the current y represents a negative value.
1982 * Flipping it to the positive side.
1983 */
1984 if( A->s < 0 && y != 0 )
1985 y = b - y;
1986
Paul Bakker5121ce52009-01-03 21:22:43 +00001987 *r = y;
1988
1989 return( 0 );
1990}
1991
1992/*
1993 * Fast Montgomery initialization (thanks to Tom St Denis)
1994 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001995static void mpi_montg_init( mbedtls_mpi_uint *mm, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00001996{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001997 mbedtls_mpi_uint x, m0 = N->p[0];
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01001998 unsigned int i;
Paul Bakker5121ce52009-01-03 21:22:43 +00001999
2000 x = m0;
2001 x += ( ( m0 + 2 ) & 4 ) << 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00002002
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01002003 for( i = biL; i >= 8; i /= 2 )
2004 x *= ( 2 - ( m0 * x ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002005
2006 *mm = ~x + 1;
2007}
2008
Gilles Peskine2a82f722020-06-04 15:00:49 +02002009/** Montgomery multiplication: A = A * B * R^-1 mod N (HAC 14.36)
2010 *
2011 * \param[in,out] A One of the numbers to multiply.
Gilles Peskine221626f2020-06-08 22:37:50 +02002012 * It must have at least as many limbs as N
2013 * (A->n >= N->n), and any limbs beyond n are ignored.
Gilles Peskine2a82f722020-06-04 15:00:49 +02002014 * On successful completion, A contains the result of
2015 * the multiplication A * B * R^-1 mod N where
2016 * R = (2^ciL)^n.
2017 * \param[in] B One of the numbers to multiply.
2018 * It must be nonzero and must not have more limbs than N
2019 * (B->n <= N->n).
2020 * \param[in] N The modulo. N must be odd.
2021 * \param mm The value calculated by `mpi_montg_init(&mm, N)`.
2022 * This is -N^-1 mod 2^ciL.
2023 * \param[in,out] T A bignum for temporary storage.
2024 * It must be at least twice the limb size of N plus 2
2025 * (T->n >= 2 * (N->n + 1)).
2026 * Its initial content is unused and
2027 * its final content is indeterminate.
2028 * Note that unlike the usual convention in the library
2029 * for `const mbedtls_mpi*`, the content of T can change.
Paul Bakker5121ce52009-01-03 21:22:43 +00002030 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002031static void mpi_montmul( mbedtls_mpi *A, const mbedtls_mpi *B, const mbedtls_mpi *N, mbedtls_mpi_uint mm,
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002032 const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002033{
Paul Bakker23986e52011-04-24 08:57:21 +00002034 size_t i, n, m;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002035 mbedtls_mpi_uint u0, u1, *d;
Paul Bakker5121ce52009-01-03 21:22:43 +00002036
2037 memset( T->p, 0, T->n * ciL );
2038
2039 d = T->p;
2040 n = N->n;
2041 m = ( B->n < n ) ? B->n : n;
2042
2043 for( i = 0; i < n; i++ )
2044 {
2045 /*
2046 * T = (T + u0*B + u1*N) / 2^biL
2047 */
2048 u0 = A->p[i];
2049 u1 = ( d[0] + u0 * B->p[0] ) * mm;
2050
2051 mpi_mul_hlp( m, B->p, d, u0 );
2052 mpi_mul_hlp( n, N->p, d, u1 );
2053
2054 *d++ = u0; d[n + 1] = 0;
2055 }
2056
Gilles Peskine221626f2020-06-08 22:37:50 +02002057 /* At this point, d is either the desired result or the desired result
2058 * plus N. We now potentially subtract N, avoiding leaking whether the
2059 * subtraction is performed through side channels. */
Paul Bakker5121ce52009-01-03 21:22:43 +00002060
Gilles Peskine221626f2020-06-08 22:37:50 +02002061 /* Copy the n least significant limbs of d to A, so that
2062 * A = d if d < N (recall that N has n limbs). */
2063 memcpy( A->p, d, n * ciL );
Gilles Peskine09ec10a2020-06-09 10:39:38 +02002064 /* If d >= N then we want to set A to d - N. To prevent timing attacks,
Gilles Peskine221626f2020-06-08 22:37:50 +02002065 * do the calculation without using conditional tests. */
2066 /* Set d to d0 + (2^biL)^n - N where d0 is the current value of d. */
Gilles Peskine132c0972020-06-04 21:05:24 +02002067 d[n] += 1;
Gilles Peskinec097e9e2020-06-08 21:58:22 +02002068 d[n] -= mpi_sub_hlp( n, d, N->p );
Gilles Peskine221626f2020-06-08 22:37:50 +02002069 /* If d0 < N then d < (2^biL)^n
2070 * so d[n] == 0 and we want to keep A as it is.
2071 * If d0 >= N then d >= (2^biL)^n, and d <= (2^biL)^n + N < 2 * (2^biL)^n
2072 * so d[n] == 1 and we want to set A to the result of the subtraction
2073 * which is d - (2^biL)^n, i.e. the n least significant limbs of d.
2074 * This exactly corresponds to a conditional assignment. */
2075 mpi_safe_cond_assign( n, A->p, d, (unsigned char) d[n] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002076}
2077
2078/*
2079 * Montgomery reduction: A = A * R^-1 mod N
Gilles Peskine2a82f722020-06-04 15:00:49 +02002080 *
2081 * See mpi_montmul() regarding constraints and guarantees on the parameters.
Paul Bakker5121ce52009-01-03 21:22:43 +00002082 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002083static void mpi_montred( mbedtls_mpi *A, const mbedtls_mpi *N,
2084 mbedtls_mpi_uint mm, const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002085{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002086 mbedtls_mpi_uint z = 1;
2087 mbedtls_mpi U;
Paul Bakker5121ce52009-01-03 21:22:43 +00002088
Paul Bakker8ddb6452013-02-27 14:56:33 +01002089 U.n = U.s = (int) z;
Paul Bakker5121ce52009-01-03 21:22:43 +00002090 U.p = &z;
2091
Gilles Peskine4e91d472020-06-04 20:55:15 +02002092 mpi_montmul( A, &U, N, mm, T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002093}
2094
2095/*
2096 * Sliding-window exponentiation: X = A^E mod N (HAC 14.85)
2097 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002098int mbedtls_mpi_exp_mod( mbedtls_mpi *X, const mbedtls_mpi *A,
2099 const mbedtls_mpi *E, const mbedtls_mpi *N,
2100 mbedtls_mpi *_RR )
Paul Bakker5121ce52009-01-03 21:22:43 +00002101{
Janos Follath24eed8d2019-11-22 13:21:35 +00002102 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002103 size_t wbits, wsize, one = 1;
2104 size_t i, j, nblimbs;
2105 size_t bufsize, nbits;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002106 mbedtls_mpi_uint ei, mm, state;
Daniel Otte388f9b22020-08-21 12:34:29 +02002107 mbedtls_mpi RR, T, W[ 1 << MBEDTLS_MPI_WINDOW_SIZE ], Apos;
Paul Bakkerf6198c12012-05-16 08:02:29 +00002108 int neg;
Paul Bakker5121ce52009-01-03 21:22:43 +00002109
Hanno Becker73d7d792018-12-11 10:35:51 +00002110 MPI_VALIDATE_RET( X != NULL );
2111 MPI_VALIDATE_RET( A != NULL );
2112 MPI_VALIDATE_RET( E != NULL );
2113 MPI_VALIDATE_RET( N != NULL );
2114
Hanno Becker8d1dd1b2017-09-28 11:02:24 +01002115 if( mbedtls_mpi_cmp_int( N, 0 ) <= 0 || ( N->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002116 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002117
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002118 if( mbedtls_mpi_cmp_int( E, 0 ) < 0 )
2119 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002120
Chris Jones9246d042020-11-25 15:12:39 +00002121 if( mbedtls_mpi_bitlen( E ) > MBEDTLS_MPI_MAX_BITS ||
2122 mbedtls_mpi_bitlen( N ) > MBEDTLS_MPI_MAX_BITS )
2123 return ( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
2124
Paul Bakkerf6198c12012-05-16 08:02:29 +00002125 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002126 * Init temps and window size
2127 */
2128 mpi_montg_init( &mm, N );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002129 mbedtls_mpi_init( &RR ); mbedtls_mpi_init( &T );
2130 mbedtls_mpi_init( &Apos );
Paul Bakker5121ce52009-01-03 21:22:43 +00002131 memset( W, 0, sizeof( W ) );
2132
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002133 i = mbedtls_mpi_bitlen( E );
Paul Bakker5121ce52009-01-03 21:22:43 +00002134
2135 wsize = ( i > 671 ) ? 6 : ( i > 239 ) ? 5 :
2136 ( i > 79 ) ? 4 : ( i > 23 ) ? 3 : 1;
2137
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002138#if( MBEDTLS_MPI_WINDOW_SIZE < 6 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002139 if( wsize > MBEDTLS_MPI_WINDOW_SIZE )
2140 wsize = MBEDTLS_MPI_WINDOW_SIZE;
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002141#endif
Paul Bakkerb6d5f082011-11-25 11:52:11 +00002142
Paul Bakker5121ce52009-01-03 21:22:43 +00002143 j = N->n + 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002144 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
2145 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[1], j ) );
2146 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T, j * 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002147
2148 /*
Paul Bakker50546922012-05-19 08:40:49 +00002149 * Compensate for negative A (and correct at the end)
2150 */
2151 neg = ( A->s == -1 );
Paul Bakker50546922012-05-19 08:40:49 +00002152 if( neg )
2153 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002154 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Apos, A ) );
Paul Bakker50546922012-05-19 08:40:49 +00002155 Apos.s = 1;
2156 A = &Apos;
2157 }
2158
2159 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002160 * If 1st call, pre-compute R^2 mod N
2161 */
2162 if( _RR == NULL || _RR->p == NULL )
2163 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002164 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &RR, 1 ) );
2165 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &RR, N->n * 2 * biL ) );
2166 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &RR, &RR, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002167
2168 if( _RR != NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002169 memcpy( _RR, &RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002170 }
2171 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002172 memcpy( &RR, _RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002173
2174 /*
2175 * W[1] = A * R^2 * R^-1 mod N = A * R mod N
2176 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002177 if( mbedtls_mpi_cmp_mpi( A, N ) >= 0 )
2178 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &W[1], A, N ) );
Paul Bakkerc2024f42014-01-23 20:38:35 +01002179 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002180 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[1], A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002181
Gilles Peskine4e91d472020-06-04 20:55:15 +02002182 mpi_montmul( &W[1], &RR, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002183
2184 /*
2185 * X = R^2 * R^-1 mod N = R mod N
2186 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002187 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &RR ) );
Gilles Peskine4e91d472020-06-04 20:55:15 +02002188 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002189
2190 if( wsize > 1 )
2191 {
2192 /*
2193 * W[1 << (wsize - 1)] = W[1] ^ (wsize - 1)
2194 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002195 j = one << ( wsize - 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002196
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002197 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[j], N->n + 1 ) );
2198 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[j], &W[1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002199
2200 for( i = 0; i < wsize - 1; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002201 mpi_montmul( &W[j], &W[j], N, mm, &T );
Paul Bakker0d7702c2013-10-29 16:18:35 +01002202
Paul Bakker5121ce52009-01-03 21:22:43 +00002203 /*
2204 * W[i] = W[i - 1] * W[1]
2205 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002206 for( i = j + 1; i < ( one << wsize ); i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002207 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002208 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[i], N->n + 1 ) );
2209 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[i], &W[i - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002210
Gilles Peskine4e91d472020-06-04 20:55:15 +02002211 mpi_montmul( &W[i], &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002212 }
2213 }
2214
2215 nblimbs = E->n;
2216 bufsize = 0;
2217 nbits = 0;
2218 wbits = 0;
2219 state = 0;
2220
2221 while( 1 )
2222 {
2223 if( bufsize == 0 )
2224 {
Paul Bakker0d7702c2013-10-29 16:18:35 +01002225 if( nblimbs == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002226 break;
2227
Paul Bakker0d7702c2013-10-29 16:18:35 +01002228 nblimbs--;
2229
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002230 bufsize = sizeof( mbedtls_mpi_uint ) << 3;
Paul Bakker5121ce52009-01-03 21:22:43 +00002231 }
2232
2233 bufsize--;
2234
2235 ei = (E->p[nblimbs] >> bufsize) & 1;
2236
2237 /*
2238 * skip leading 0s
2239 */
2240 if( ei == 0 && state == 0 )
2241 continue;
2242
2243 if( ei == 0 && state == 1 )
2244 {
2245 /*
2246 * out of window, square X
2247 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002248 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002249 continue;
2250 }
2251
2252 /*
2253 * add ei to current window
2254 */
2255 state = 2;
2256
2257 nbits++;
Paul Bakker66d5d072014-06-17 16:39:18 +02002258 wbits |= ( ei << ( wsize - nbits ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002259
2260 if( nbits == wsize )
2261 {
2262 /*
2263 * X = X^wsize R^-1 mod N
2264 */
2265 for( i = 0; i < wsize; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002266 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002267
2268 /*
2269 * X = X * W[wbits] R^-1 mod N
2270 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002271 mpi_montmul( X, &W[wbits], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002272
2273 state--;
2274 nbits = 0;
2275 wbits = 0;
2276 }
2277 }
2278
2279 /*
2280 * process the remaining bits
2281 */
2282 for( i = 0; i < nbits; i++ )
2283 {
Gilles Peskine4e91d472020-06-04 20:55:15 +02002284 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002285
2286 wbits <<= 1;
2287
Paul Bakker66d5d072014-06-17 16:39:18 +02002288 if( ( wbits & ( one << wsize ) ) != 0 )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002289 mpi_montmul( X, &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002290 }
2291
2292 /*
2293 * X = A^E * R * R^-1 mod N = A^E mod N
2294 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002295 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002296
Hanno Beckera4af1c42017-04-18 09:07:45 +01002297 if( neg && E->n != 0 && ( E->p[0] & 1 ) != 0 )
Paul Bakkerf6198c12012-05-16 08:02:29 +00002298 {
2299 X->s = -1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002300 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( X, N, X ) );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002301 }
2302
Paul Bakker5121ce52009-01-03 21:22:43 +00002303cleanup:
2304
Paul Bakker66d5d072014-06-17 16:39:18 +02002305 for( i = ( one << ( wsize - 1 ) ); i < ( one << wsize ); i++ )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002306 mbedtls_mpi_free( &W[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002307
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002308 mbedtls_mpi_free( &W[1] ); mbedtls_mpi_free( &T ); mbedtls_mpi_free( &Apos );
Paul Bakker6c591fa2011-05-05 11:49:20 +00002309
Paul Bakker75a28602014-03-31 12:08:17 +02002310 if( _RR == NULL || _RR->p == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002311 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002312
2313 return( ret );
2314}
2315
Paul Bakker5121ce52009-01-03 21:22:43 +00002316/*
2317 * Greatest common divisor: G = gcd(A, B) (HAC 14.54)
2318 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002319int mbedtls_mpi_gcd( mbedtls_mpi *G, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00002320{
Janos Follath24eed8d2019-11-22 13:21:35 +00002321 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002322 size_t lz, lzt;
Alexander Ke8ad49f2019-08-16 16:16:07 +03002323 mbedtls_mpi TA, TB;
Paul Bakker5121ce52009-01-03 21:22:43 +00002324
Hanno Becker73d7d792018-12-11 10:35:51 +00002325 MPI_VALIDATE_RET( G != NULL );
2326 MPI_VALIDATE_RET( A != NULL );
2327 MPI_VALIDATE_RET( B != NULL );
2328
Alexander Ke8ad49f2019-08-16 16:16:07 +03002329 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002330
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002331 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) );
2332 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002333
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002334 lz = mbedtls_mpi_lsb( &TA );
2335 lzt = mbedtls_mpi_lsb( &TB );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002336
Paul Bakker66d5d072014-06-17 16:39:18 +02002337 if( lzt < lz )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002338 lz = lzt;
2339
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002340 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, lz ) );
2341 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, lz ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002342
Paul Bakker5121ce52009-01-03 21:22:43 +00002343 TA.s = TB.s = 1;
2344
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002345 while( mbedtls_mpi_cmp_int( &TA, 0 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002346 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002347 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, mbedtls_mpi_lsb( &TA ) ) );
2348 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, mbedtls_mpi_lsb( &TB ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002349
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002350 if( mbedtls_mpi_cmp_mpi( &TA, &TB ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002351 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002352 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TA, &TA, &TB ) );
2353 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002354 }
2355 else
2356 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002357 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TB, &TB, &TA ) );
2358 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002359 }
2360 }
2361
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002362 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &TB, lz ) );
2363 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( G, &TB ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002364
2365cleanup:
2366
Alexander Ke8ad49f2019-08-16 16:16:07 +03002367 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002368
2369 return( ret );
2370}
2371
Paul Bakker33dc46b2014-04-30 16:11:39 +02002372/*
2373 * Fill X with size bytes of random.
2374 *
2375 * Use a temporary bytes representation to make sure the result is the same
Paul Bakkerc37b0ac2014-05-01 14:19:23 +02002376 * regardless of the platform endianness (useful when f_rng is actually
Paul Bakker33dc46b2014-04-30 16:11:39 +02002377 * deterministic, eg for tests).
2378 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002379int mbedtls_mpi_fill_random( mbedtls_mpi *X, size_t size,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002380 int (*f_rng)(void *, unsigned char *, size_t),
2381 void *p_rng )
Paul Bakker287781a2011-03-26 13:18:49 +00002382{
Janos Follath24eed8d2019-11-22 13:21:35 +00002383 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker6dab6202019-01-02 16:42:29 +00002384 size_t const limbs = CHARS_TO_LIMBS( size );
Hanno Beckerda1655a2017-10-18 14:21:44 +01002385 size_t const overhead = ( limbs * ciL ) - size;
2386 unsigned char *Xp;
2387
Hanno Becker8ce11a32018-12-19 16:18:52 +00002388 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002389 MPI_VALIDATE_RET( f_rng != NULL );
Paul Bakker33dc46b2014-04-30 16:11:39 +02002390
Hanno Beckerda1655a2017-10-18 14:21:44 +01002391 /* Ensure that target MPI has exactly the necessary number of limbs */
2392 if( X->n != limbs )
2393 {
2394 mbedtls_mpi_free( X );
2395 mbedtls_mpi_init( X );
2396 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
2397 }
2398 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker287781a2011-03-26 13:18:49 +00002399
Hanno Beckerda1655a2017-10-18 14:21:44 +01002400 Xp = (unsigned char*) X->p;
Gilles Peskine436400e2020-11-25 16:15:14 +01002401 MBEDTLS_MPI_CHK( f_rng( p_rng, Xp + overhead, size ) );
Hanno Beckerda1655a2017-10-18 14:21:44 +01002402
Hanno Becker2be8a552018-10-25 12:40:09 +01002403 mpi_bigendian_to_host( X->p, limbs );
Paul Bakker287781a2011-03-26 13:18:49 +00002404
2405cleanup:
2406 return( ret );
2407}
2408
Paul Bakker5121ce52009-01-03 21:22:43 +00002409/*
2410 * Modular inverse: X = A^-1 mod N (HAC 14.61 / 14.64)
2411 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002412int mbedtls_mpi_inv_mod( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002413{
Janos Follath24eed8d2019-11-22 13:21:35 +00002414 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002415 mbedtls_mpi G, TA, TU, U1, U2, TB, TV, V1, V2;
Hanno Becker73d7d792018-12-11 10:35:51 +00002416 MPI_VALIDATE_RET( X != NULL );
2417 MPI_VALIDATE_RET( A != NULL );
2418 MPI_VALIDATE_RET( N != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00002419
Hanno Becker4bcb4912017-04-18 15:49:39 +01002420 if( mbedtls_mpi_cmp_int( N, 1 ) <= 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002421 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002422
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002423 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TU ); mbedtls_mpi_init( &U1 ); mbedtls_mpi_init( &U2 );
2424 mbedtls_mpi_init( &G ); mbedtls_mpi_init( &TB ); mbedtls_mpi_init( &TV );
2425 mbedtls_mpi_init( &V1 ); mbedtls_mpi_init( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002426
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002427 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &G, A, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002428
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002429 if( mbedtls_mpi_cmp_int( &G, 1 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002430 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002431 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002432 goto cleanup;
2433 }
2434
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002435 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &TA, A, N ) );
2436 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TU, &TA ) );
2437 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, N ) );
2438 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TV, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002439
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002440 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U1, 1 ) );
2441 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U2, 0 ) );
2442 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V1, 0 ) );
2443 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002444
2445 do
2446 {
2447 while( ( TU.p[0] & 1 ) == 0 )
2448 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002449 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TU, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002450
2451 if( ( U1.p[0] & 1 ) != 0 || ( U2.p[0] & 1 ) != 0 )
2452 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002453 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &U1, &U1, &TB ) );
2454 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002455 }
2456
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002457 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U1, 1 ) );
2458 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002459 }
2460
2461 while( ( TV.p[0] & 1 ) == 0 )
2462 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002463 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TV, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002464
2465 if( ( V1.p[0] & 1 ) != 0 || ( V2.p[0] & 1 ) != 0 )
2466 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002467 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, &TB ) );
2468 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002469 }
2470
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002471 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V1, 1 ) );
2472 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002473 }
2474
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002475 if( mbedtls_mpi_cmp_mpi( &TU, &TV ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002476 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002477 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TU, &TU, &TV ) );
2478 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U1, &U1, &V1 ) );
2479 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &V2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002480 }
2481 else
2482 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002483 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TV, &TV, &TU ) );
2484 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, &U1 ) );
2485 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &U2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002486 }
2487 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002488 while( mbedtls_mpi_cmp_int( &TU, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002489
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002490 while( mbedtls_mpi_cmp_int( &V1, 0 ) < 0 )
2491 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002492
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002493 while( mbedtls_mpi_cmp_mpi( &V1, N ) >= 0 )
2494 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002495
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002496 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &V1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002497
2498cleanup:
2499
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002500 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TU ); mbedtls_mpi_free( &U1 ); mbedtls_mpi_free( &U2 );
2501 mbedtls_mpi_free( &G ); mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TV );
2502 mbedtls_mpi_free( &V1 ); mbedtls_mpi_free( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002503
2504 return( ret );
2505}
2506
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002507#if defined(MBEDTLS_GENPRIME)
Paul Bakkerd9374b02012-11-02 11:02:58 +00002508
Paul Bakker5121ce52009-01-03 21:22:43 +00002509static const int small_prime[] =
2510{
2511 3, 5, 7, 11, 13, 17, 19, 23,
2512 29, 31, 37, 41, 43, 47, 53, 59,
2513 61, 67, 71, 73, 79, 83, 89, 97,
2514 101, 103, 107, 109, 113, 127, 131, 137,
2515 139, 149, 151, 157, 163, 167, 173, 179,
2516 181, 191, 193, 197, 199, 211, 223, 227,
2517 229, 233, 239, 241, 251, 257, 263, 269,
2518 271, 277, 281, 283, 293, 307, 311, 313,
2519 317, 331, 337, 347, 349, 353, 359, 367,
2520 373, 379, 383, 389, 397, 401, 409, 419,
2521 421, 431, 433, 439, 443, 449, 457, 461,
2522 463, 467, 479, 487, 491, 499, 503, 509,
2523 521, 523, 541, 547, 557, 563, 569, 571,
2524 577, 587, 593, 599, 601, 607, 613, 617,
2525 619, 631, 641, 643, 647, 653, 659, 661,
2526 673, 677, 683, 691, 701, 709, 719, 727,
2527 733, 739, 743, 751, 757, 761, 769, 773,
2528 787, 797, 809, 811, 821, 823, 827, 829,
2529 839, 853, 857, 859, 863, 877, 881, 883,
2530 887, 907, 911, 919, 929, 937, 941, 947,
2531 953, 967, 971, 977, 983, 991, 997, -103
2532};
2533
2534/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002535 * Small divisors test (X must be positive)
2536 *
2537 * Return values:
2538 * 0: no small factor (possible prime, more tests needed)
2539 * 1: certain prime
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002540 * MBEDTLS_ERR_MPI_NOT_ACCEPTABLE: certain non-prime
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002541 * other negative: error
Paul Bakker5121ce52009-01-03 21:22:43 +00002542 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002543static int mpi_check_small_factors( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +00002544{
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002545 int ret = 0;
2546 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002547 mbedtls_mpi_uint r;
Paul Bakker5121ce52009-01-03 21:22:43 +00002548
Paul Bakker5121ce52009-01-03 21:22:43 +00002549 if( ( X->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002550 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002551
2552 for( i = 0; small_prime[i] > 0; i++ )
2553 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002554 if( mbedtls_mpi_cmp_int( X, small_prime[i] ) <= 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002555 return( 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002556
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002557 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, small_prime[i] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002558
2559 if( r == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002560 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002561 }
2562
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002563cleanup:
2564 return( ret );
2565}
2566
2567/*
2568 * Miller-Rabin pseudo-primality test (HAC 4.24)
2569 */
Janos Follathda31fa12018-09-03 14:45:23 +01002570static int mpi_miller_rabin( const mbedtls_mpi *X, size_t rounds,
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002571 int (*f_rng)(void *, unsigned char *, size_t),
2572 void *p_rng )
2573{
Pascal Junodb99183d2015-03-11 16:49:45 +01002574 int ret, count;
Janos Follathda31fa12018-09-03 14:45:23 +01002575 size_t i, j, k, s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002576 mbedtls_mpi W, R, T, A, RR;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002577
Hanno Becker8ce11a32018-12-19 16:18:52 +00002578 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002579 MPI_VALIDATE_RET( f_rng != NULL );
2580
2581 mbedtls_mpi_init( &W ); mbedtls_mpi_init( &R );
2582 mbedtls_mpi_init( &T ); mbedtls_mpi_init( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002583 mbedtls_mpi_init( &RR );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002584
Paul Bakker5121ce52009-01-03 21:22:43 +00002585 /*
2586 * W = |X| - 1
2587 * R = W >> lsb( W )
2588 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002589 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( &W, X, 1 ) );
2590 s = mbedtls_mpi_lsb( &W );
2591 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &R, &W ) );
2592 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &R, s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002593
Janos Follathda31fa12018-09-03 14:45:23 +01002594 for( i = 0; i < rounds; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002595 {
2596 /*
2597 * pick a random A, 1 < A < |X| - 1
2598 */
Pascal Junodb99183d2015-03-11 16:49:45 +01002599 count = 0;
2600 do {
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002601 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( &A, X->n * ciL, f_rng, p_rng ) );
Pascal Junodb99183d2015-03-11 16:49:45 +01002602
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002603 j = mbedtls_mpi_bitlen( &A );
2604 k = mbedtls_mpi_bitlen( &W );
Pascal Junodb99183d2015-03-11 16:49:45 +01002605 if (j > k) {
Darryl Greene3f95ed2018-10-02 13:21:35 +01002606 A.p[A.n - 1] &= ( (mbedtls_mpi_uint) 1 << ( k - ( A.n - 1 ) * biL - 1 ) ) - 1;
Pascal Junodb99183d2015-03-11 16:49:45 +01002607 }
2608
2609 if (count++ > 30) {
Jens Wiklanderf08aa3e2019-01-17 13:30:57 +01002610 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
2611 goto cleanup;
Pascal Junodb99183d2015-03-11 16:49:45 +01002612 }
2613
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002614 } while ( mbedtls_mpi_cmp_mpi( &A, &W ) >= 0 ||
2615 mbedtls_mpi_cmp_int( &A, 1 ) <= 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002616
2617 /*
2618 * A = A^R mod |X|
2619 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002620 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &A, &A, &R, X, &RR ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002621
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002622 if( mbedtls_mpi_cmp_mpi( &A, &W ) == 0 ||
2623 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002624 continue;
2625
2626 j = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002627 while( j < s && mbedtls_mpi_cmp_mpi( &A, &W ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002628 {
2629 /*
2630 * A = A * A mod |X|
2631 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002632 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &T, &A, &A ) );
2633 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &A, &T, X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002634
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002635 if( mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002636 break;
2637
2638 j++;
2639 }
2640
2641 /*
2642 * not prime if A != |X| - 1 or A == 1
2643 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002644 if( mbedtls_mpi_cmp_mpi( &A, &W ) != 0 ||
2645 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002646 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002647 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002648 break;
2649 }
2650 }
2651
2652cleanup:
Hanno Becker73d7d792018-12-11 10:35:51 +00002653 mbedtls_mpi_free( &W ); mbedtls_mpi_free( &R );
2654 mbedtls_mpi_free( &T ); mbedtls_mpi_free( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002655 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002656
2657 return( ret );
2658}
2659
2660/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002661 * Pseudo-primality test: small factors, then Miller-Rabin
2662 */
Janos Follatha0b67c22018-09-18 14:48:23 +01002663int mbedtls_mpi_is_prime_ext( const mbedtls_mpi *X, int rounds,
2664 int (*f_rng)(void *, unsigned char *, size_t),
2665 void *p_rng )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002666{
Janos Follath24eed8d2019-11-22 13:21:35 +00002667 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002668 mbedtls_mpi XX;
Hanno Becker8ce11a32018-12-19 16:18:52 +00002669 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002670 MPI_VALIDATE_RET( f_rng != NULL );
Manuel Pégourié-Gonnard7f4ed672014-10-14 20:56:02 +02002671
2672 XX.s = 1;
2673 XX.n = X->n;
2674 XX.p = X->p;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002675
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002676 if( mbedtls_mpi_cmp_int( &XX, 0 ) == 0 ||
2677 mbedtls_mpi_cmp_int( &XX, 1 ) == 0 )
2678 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002679
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002680 if( mbedtls_mpi_cmp_int( &XX, 2 ) == 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002681 return( 0 );
2682
2683 if( ( ret = mpi_check_small_factors( &XX ) ) != 0 )
2684 {
2685 if( ret == 1 )
2686 return( 0 );
2687
2688 return( ret );
2689 }
2690
Janos Follathda31fa12018-09-03 14:45:23 +01002691 return( mpi_miller_rabin( &XX, rounds, f_rng, p_rng ) );
Janos Follathf301d232018-08-14 13:34:01 +01002692}
2693
Janos Follatha0b67c22018-09-18 14:48:23 +01002694#if !defined(MBEDTLS_DEPRECATED_REMOVED)
Janos Follathf301d232018-08-14 13:34:01 +01002695/*
2696 * Pseudo-primality test, error probability 2^-80
2697 */
2698int mbedtls_mpi_is_prime( const mbedtls_mpi *X,
2699 int (*f_rng)(void *, unsigned char *, size_t),
2700 void *p_rng )
2701{
Hanno Becker8ce11a32018-12-19 16:18:52 +00002702 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002703 MPI_VALIDATE_RET( f_rng != NULL );
2704
Janos Follatha0b67c22018-09-18 14:48:23 +01002705 /*
2706 * In the past our key generation aimed for an error rate of at most
2707 * 2^-80. Since this function is deprecated, aim for the same certainty
2708 * here as well.
2709 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002710 return( mbedtls_mpi_is_prime_ext( X, 40, f_rng, p_rng ) );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002711}
Janos Follatha0b67c22018-09-18 14:48:23 +01002712#endif
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002713
2714/*
Paul Bakker5121ce52009-01-03 21:22:43 +00002715 * Prime number generation
Jethro Beekman66689272018-02-14 19:24:10 -08002716 *
Janos Follathf301d232018-08-14 13:34:01 +01002717 * To generate an RSA key in a way recommended by FIPS 186-4, both primes must
2718 * be either 1024 bits or 1536 bits long, and flags must contain
2719 * MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR.
Paul Bakker5121ce52009-01-03 21:22:43 +00002720 */
Janos Follath7c025a92018-08-14 11:08:41 +01002721int mbedtls_mpi_gen_prime( mbedtls_mpi *X, size_t nbits, int flags,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002722 int (*f_rng)(void *, unsigned char *, size_t),
2723 void *p_rng )
Paul Bakker5121ce52009-01-03 21:22:43 +00002724{
Jethro Beekman66689272018-02-14 19:24:10 -08002725#ifdef MBEDTLS_HAVE_INT64
2726// ceil(2^63.5)
2727#define CEIL_MAXUINT_DIV_SQRT2 0xb504f333f9de6485ULL
2728#else
2729// ceil(2^31.5)
2730#define CEIL_MAXUINT_DIV_SQRT2 0xb504f334U
2731#endif
2732 int ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker23986e52011-04-24 08:57:21 +00002733 size_t k, n;
Janos Follathda31fa12018-09-03 14:45:23 +01002734 int rounds;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002735 mbedtls_mpi_uint r;
2736 mbedtls_mpi Y;
Paul Bakker5121ce52009-01-03 21:22:43 +00002737
Hanno Becker8ce11a32018-12-19 16:18:52 +00002738 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002739 MPI_VALIDATE_RET( f_rng != NULL );
2740
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002741 if( nbits < 3 || nbits > MBEDTLS_MPI_MAX_BITS )
2742 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002743
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002744 mbedtls_mpi_init( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002745
2746 n = BITS_TO_LIMBS( nbits );
2747
Janos Follathda31fa12018-09-03 14:45:23 +01002748 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR ) == 0 )
2749 {
2750 /*
2751 * 2^-80 error probability, number of rounds chosen per HAC, table 4.4
2752 */
2753 rounds = ( ( nbits >= 1300 ) ? 2 : ( nbits >= 850 ) ? 3 :
2754 ( nbits >= 650 ) ? 4 : ( nbits >= 350 ) ? 8 :
2755 ( nbits >= 250 ) ? 12 : ( nbits >= 150 ) ? 18 : 27 );
2756 }
2757 else
2758 {
2759 /*
2760 * 2^-100 error probability, number of rounds computed based on HAC,
2761 * fact 4.48
2762 */
2763 rounds = ( ( nbits >= 1450 ) ? 4 : ( nbits >= 1150 ) ? 5 :
2764 ( nbits >= 1000 ) ? 6 : ( nbits >= 850 ) ? 7 :
2765 ( nbits >= 750 ) ? 8 : ( nbits >= 500 ) ? 13 :
2766 ( nbits >= 250 ) ? 28 : ( nbits >= 150 ) ? 40 : 51 );
2767 }
2768
Jethro Beekman66689272018-02-14 19:24:10 -08002769 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002770 {
Jethro Beekman66689272018-02-14 19:24:10 -08002771 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( X, n * ciL, f_rng, p_rng ) );
2772 /* make sure generated number is at least (nbits-1)+0.5 bits (FIPS 186-4 §B.3.3 steps 4.4, 5.5) */
2773 if( X->p[n-1] < CEIL_MAXUINT_DIV_SQRT2 ) continue;
2774
2775 k = n * biL;
2776 if( k > nbits ) MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( X, k - nbits ) );
2777 X->p[0] |= 1;
2778
Janos Follath7c025a92018-08-14 11:08:41 +01002779 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_DH ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002780 {
Janos Follatha0b67c22018-09-18 14:48:23 +01002781 ret = mbedtls_mpi_is_prime_ext( X, rounds, f_rng, p_rng );
Jethro Beekman66689272018-02-14 19:24:10 -08002782
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002783 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
Paul Bakker5121ce52009-01-03 21:22:43 +00002784 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002785 }
Jethro Beekman66689272018-02-14 19:24:10 -08002786 else
Paul Bakker5121ce52009-01-03 21:22:43 +00002787 {
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002788 /*
Jethro Beekman66689272018-02-14 19:24:10 -08002789 * An necessary condition for Y and X = 2Y + 1 to be prime
2790 * is X = 2 mod 3 (which is equivalent to Y = 2 mod 3).
2791 * Make sure it is satisfied, while keeping X = 3 mod 4
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002792 */
Jethro Beekman66689272018-02-14 19:24:10 -08002793
2794 X->p[0] |= 2;
2795
2796 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, 3 ) );
2797 if( r == 0 )
2798 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 8 ) );
2799 else if( r == 1 )
2800 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 4 ) );
2801
2802 /* Set Y = (X-1) / 2, which is X / 2 because X is odd */
2803 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, X ) );
2804 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, 1 ) );
2805
2806 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002807 {
Jethro Beekman66689272018-02-14 19:24:10 -08002808 /*
2809 * First, check small factors for X and Y
2810 * before doing Miller-Rabin on any of them
2811 */
2812 if( ( ret = mpi_check_small_factors( X ) ) == 0 &&
2813 ( ret = mpi_check_small_factors( &Y ) ) == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002814 ( ret = mpi_miller_rabin( X, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002815 == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002816 ( ret = mpi_miller_rabin( &Y, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002817 == 0 )
Jethro Beekman66689272018-02-14 19:24:10 -08002818 goto cleanup;
2819
2820 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
2821 goto cleanup;
2822
2823 /*
2824 * Next candidates. We want to preserve Y = (X-1) / 2 and
2825 * Y = 1 mod 2 and Y = 2 mod 3 (eq X = 3 mod 4 and X = 2 mod 3)
2826 * so up Y by 6 and X by 12.
2827 */
2828 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 12 ) );
2829 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( &Y, &Y, 6 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002830 }
Paul Bakker5121ce52009-01-03 21:22:43 +00002831 }
2832 }
2833
2834cleanup:
2835
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002836 mbedtls_mpi_free( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002837
2838 return( ret );
2839}
2840
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002841#endif /* MBEDTLS_GENPRIME */
Paul Bakker5121ce52009-01-03 21:22:43 +00002842
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002843#if defined(MBEDTLS_SELF_TEST)
Paul Bakker5121ce52009-01-03 21:22:43 +00002844
Paul Bakker23986e52011-04-24 08:57:21 +00002845#define GCD_PAIR_COUNT 3
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002846
2847static const int gcd_pairs[GCD_PAIR_COUNT][3] =
2848{
2849 { 693, 609, 21 },
2850 { 1764, 868, 28 },
2851 { 768454923, 542167814, 1 }
2852};
2853
Paul Bakker5121ce52009-01-03 21:22:43 +00002854/*
2855 * Checkup routine
2856 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002857int mbedtls_mpi_self_test( int verbose )
Paul Bakker5121ce52009-01-03 21:22:43 +00002858{
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002859 int ret, i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002860 mbedtls_mpi A, E, N, X, Y, U, V;
Paul Bakker5121ce52009-01-03 21:22:43 +00002861
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002862 mbedtls_mpi_init( &A ); mbedtls_mpi_init( &E ); mbedtls_mpi_init( &N ); mbedtls_mpi_init( &X );
2863 mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &U ); mbedtls_mpi_init( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002864
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002865 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &A, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002866 "EFE021C2645FD1DC586E69184AF4A31E" \
2867 "D5F53E93B5F123FA41680867BA110131" \
2868 "944FE7952E2517337780CB0DB80E61AA" \
2869 "E7C8DDC6C5C6AADEB34EB38A2F40D5E6" ) );
2870
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002871 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &E, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002872 "B2E7EFD37075B9F03FF989C7C5051C20" \
2873 "34D2A323810251127E7BF8625A4F49A5" \
2874 "F3E27F4DA8BD59C47D6DAABA4C8127BD" \
2875 "5B5C25763222FEFCCFC38B832366C29E" ) );
2876
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002877 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &N, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002878 "0066A198186C18C10B2F5ED9B522752A" \
2879 "9830B69916E535C8F047518A889A43A5" \
2880 "94B6BED27A168D31D4A52F88925AA8F5" ) );
2881
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002882 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002883
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002884 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002885 "602AB7ECA597A3D6B56FF9829A5E8B85" \
2886 "9E857EA95A03512E2BAE7391688D264A" \
2887 "A5663B0341DB9CCFD2C4C5F421FEC814" \
2888 "8001B72E848A38CAE1C65F78E56ABDEF" \
2889 "E12D3C039B8A02D6BE593F0BBBDA56F1" \
2890 "ECF677152EF804370C1A305CAF3B5BF1" \
2891 "30879B56C61DE584A0F53A2447A51E" ) );
2892
2893 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002894 mbedtls_printf( " MPI test #1 (mul_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002895
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002896 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002897 {
2898 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002899 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002900
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002901 ret = 1;
2902 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002903 }
2904
2905 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002906 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002907
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002908 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( &X, &Y, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002909
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002910 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002911 "256567336059E52CAE22925474705F39A94" ) );
2912
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002913 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &V, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002914 "6613F26162223DF488E9CD48CC132C7A" \
2915 "0AC93C701B001B092E4E5B9F73BCD27B" \
2916 "9EE50D0657C77F374E903CDFA4C642" ) );
2917
2918 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002919 mbedtls_printf( " MPI test #2 (div_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002920
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002921 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 ||
2922 mbedtls_mpi_cmp_mpi( &Y, &V ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002923 {
2924 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002925 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002926
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002927 ret = 1;
2928 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002929 }
2930
2931 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002932 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002933
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002934 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &X, &A, &E, &N, NULL ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002935
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002936 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002937 "36E139AEA55215609D2816998ED020BB" \
2938 "BD96C37890F65171D948E9BC7CBAA4D9" \
2939 "325D24D6A3C12710F10A09FA08AB87" ) );
2940
2941 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002942 mbedtls_printf( " MPI test #3 (exp_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002943
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002944 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002945 {
2946 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002947 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002948
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002949 ret = 1;
2950 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002951 }
2952
2953 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002954 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002955
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002956 MBEDTLS_MPI_CHK( mbedtls_mpi_inv_mod( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002957
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002958 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002959 "003A0AAEDD7E784FC07D8F9EC6E3BFD5" \
2960 "C3DBA76456363A10869622EAC2DD84EC" \
2961 "C5B8A74DAC4D09E03B5E0BE779F2DF61" ) );
2962
2963 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002964 mbedtls_printf( " MPI test #4 (inv_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002965
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002966 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002967 {
2968 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002969 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002970
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002971 ret = 1;
2972 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002973 }
2974
2975 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002976 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002977
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002978 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002979 mbedtls_printf( " MPI test #5 (simple gcd): " );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002980
Paul Bakker66d5d072014-06-17 16:39:18 +02002981 for( i = 0; i < GCD_PAIR_COUNT; i++ )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002982 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002983 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &X, gcd_pairs[i][0] ) );
2984 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Y, gcd_pairs[i][1] ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002985
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002986 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &A, &X, &Y ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002987
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002988 if( mbedtls_mpi_cmp_int( &A, gcd_pairs[i][2] ) != 0 )
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002989 {
2990 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002991 mbedtls_printf( "failed at %d\n", i );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002992
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002993 ret = 1;
2994 goto cleanup;
2995 }
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002996 }
2997
2998 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002999 mbedtls_printf( "passed\n" );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003000
Paul Bakker5121ce52009-01-03 21:22:43 +00003001cleanup:
3002
3003 if( ret != 0 && verbose != 0 )
Kenneth Soerensen518d4352020-04-01 17:22:45 +02003004 mbedtls_printf( "Unexpected error, return code = %08X\n", (unsigned int) ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00003005
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003006 mbedtls_mpi_free( &A ); mbedtls_mpi_free( &E ); mbedtls_mpi_free( &N ); mbedtls_mpi_free( &X );
3007 mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &U ); mbedtls_mpi_free( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00003008
3009 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003010 mbedtls_printf( "\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00003011
3012 return( ret );
3013}
3014
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003015#endif /* MBEDTLS_SELF_TEST */
Paul Bakker5121ce52009-01-03 21:22:43 +00003016
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003017#endif /* MBEDTLS_BIGNUM_C */