blob: 9a70ccdf1d85b6a66170d82c8e21f55d5b7646ab [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"
41#include "mbedtls/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 )
1414 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
1415 --X->p[n];
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001416 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001417
1418cleanup:
1419
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001420 mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001421
1422 return( ret );
1423}
1424
1425/*
1426 * Signed addition: X = A + B
1427 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001428int mbedtls_mpi_add_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001429{
Hanno Becker73d7d792018-12-11 10:35:51 +00001430 int ret, s;
1431 MPI_VALIDATE_RET( X != NULL );
1432 MPI_VALIDATE_RET( A != NULL );
1433 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001434
Hanno Becker73d7d792018-12-11 10:35:51 +00001435 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001436 if( A->s * B->s < 0 )
1437 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001438 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001439 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001440 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001441 X->s = s;
1442 }
1443 else
1444 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001445 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001446 X->s = -s;
1447 }
1448 }
1449 else
1450 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001451 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001452 X->s = s;
1453 }
1454
1455cleanup:
1456
1457 return( ret );
1458}
1459
1460/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001461 * Signed subtraction: X = A - B
Paul Bakker5121ce52009-01-03 21:22:43 +00001462 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001463int mbedtls_mpi_sub_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001464{
Hanno Becker73d7d792018-12-11 10:35:51 +00001465 int ret, s;
1466 MPI_VALIDATE_RET( X != NULL );
1467 MPI_VALIDATE_RET( A != NULL );
1468 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001469
Hanno Becker73d7d792018-12-11 10:35:51 +00001470 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001471 if( A->s * B->s > 0 )
1472 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001473 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001474 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001475 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001476 X->s = s;
1477 }
1478 else
1479 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001480 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001481 X->s = -s;
1482 }
1483 }
1484 else
1485 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001486 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001487 X->s = s;
1488 }
1489
1490cleanup:
1491
1492 return( ret );
1493}
1494
1495/*
1496 * Signed addition: X = A + b
1497 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001498int mbedtls_mpi_add_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001499{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001500 mbedtls_mpi _B;
1501 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001502 MPI_VALIDATE_RET( X != NULL );
1503 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001504
1505 p[0] = ( b < 0 ) ? -b : b;
1506 _B.s = ( b < 0 ) ? -1 : 1;
1507 _B.n = 1;
1508 _B.p = p;
1509
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001510 return( mbedtls_mpi_add_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001511}
1512
1513/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001514 * Signed subtraction: X = A - b
Paul Bakker5121ce52009-01-03 21:22:43 +00001515 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001516int mbedtls_mpi_sub_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001517{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001518 mbedtls_mpi _B;
1519 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001520 MPI_VALIDATE_RET( X != NULL );
1521 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001522
1523 p[0] = ( b < 0 ) ? -b : b;
1524 _B.s = ( b < 0 ) ? -1 : 1;
1525 _B.n = 1;
1526 _B.p = p;
1527
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001528 return( mbedtls_mpi_sub_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001529}
1530
1531/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001532 * Helper for mbedtls_mpi multiplication
Paul Bakkerfc4f46f2013-06-24 19:23:56 +02001533 */
1534static
1535#if defined(__APPLE__) && defined(__arm__)
1536/*
1537 * Apple LLVM version 4.2 (clang-425.0.24) (based on LLVM 3.2svn)
1538 * appears to need this to prevent bad ARM code generation at -O3.
1539 */
1540__attribute__ ((noinline))
1541#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001542void 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 +00001543{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001544 mbedtls_mpi_uint c = 0, t = 0;
Paul Bakker5121ce52009-01-03 21:22:43 +00001545
1546#if defined(MULADDC_HUIT)
1547 for( ; i >= 8; i -= 8 )
1548 {
1549 MULADDC_INIT
1550 MULADDC_HUIT
1551 MULADDC_STOP
1552 }
1553
1554 for( ; i > 0; i-- )
1555 {
1556 MULADDC_INIT
1557 MULADDC_CORE
1558 MULADDC_STOP
1559 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001560#else /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001561 for( ; i >= 16; i -= 16 )
1562 {
1563 MULADDC_INIT
1564 MULADDC_CORE MULADDC_CORE
1565 MULADDC_CORE MULADDC_CORE
1566 MULADDC_CORE MULADDC_CORE
1567 MULADDC_CORE MULADDC_CORE
1568
1569 MULADDC_CORE MULADDC_CORE
1570 MULADDC_CORE MULADDC_CORE
1571 MULADDC_CORE MULADDC_CORE
1572 MULADDC_CORE MULADDC_CORE
1573 MULADDC_STOP
1574 }
1575
1576 for( ; i >= 8; i -= 8 )
1577 {
1578 MULADDC_INIT
1579 MULADDC_CORE MULADDC_CORE
1580 MULADDC_CORE MULADDC_CORE
1581
1582 MULADDC_CORE MULADDC_CORE
1583 MULADDC_CORE MULADDC_CORE
1584 MULADDC_STOP
1585 }
1586
1587 for( ; i > 0; i-- )
1588 {
1589 MULADDC_INIT
1590 MULADDC_CORE
1591 MULADDC_STOP
1592 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001593#endif /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001594
1595 t++;
1596
1597 do {
1598 *d += c; c = ( *d < c ); d++;
1599 }
1600 while( c != 0 );
1601}
1602
1603/*
1604 * Baseline multiplication: X = A * B (HAC 14.12)
1605 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001606int mbedtls_mpi_mul_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001607{
Janos Follath24eed8d2019-11-22 13:21:35 +00001608 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001609 size_t i, j;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001610 mbedtls_mpi TA, TB;
Hanno Becker73d7d792018-12-11 10:35:51 +00001611 MPI_VALIDATE_RET( X != NULL );
1612 MPI_VALIDATE_RET( A != NULL );
1613 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001614
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001615 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001616
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001617 if( X == A ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) ); A = &TA; }
1618 if( X == B ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) ); B = &TB; }
Paul Bakker5121ce52009-01-03 21:22:43 +00001619
Paul Bakker23986e52011-04-24 08:57:21 +00001620 for( i = A->n; i > 0; i-- )
1621 if( A->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001622 break;
1623
Paul Bakker23986e52011-04-24 08:57:21 +00001624 for( j = B->n; j > 0; j-- )
1625 if( B->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001626 break;
1627
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001628 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + j ) );
1629 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001630
Alexey Skalozub8e75e682016-01-13 21:59:27 +02001631 for( ; j > 0; j-- )
1632 mpi_mul_hlp( i, A->p, X->p + j - 1, B->p[j - 1] );
Paul Bakker5121ce52009-01-03 21:22:43 +00001633
1634 X->s = A->s * B->s;
1635
1636cleanup:
1637
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001638 mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TA );
Paul Bakker5121ce52009-01-03 21:22:43 +00001639
1640 return( ret );
1641}
1642
1643/*
1644 * Baseline multiplication: X = A * b
1645 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001646int mbedtls_mpi_mul_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001647{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001648 mbedtls_mpi _B;
1649 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001650 MPI_VALIDATE_RET( X != NULL );
1651 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001652
1653 _B.s = 1;
1654 _B.n = 1;
1655 _B.p = p;
1656 p[0] = b;
1657
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001658 return( mbedtls_mpi_mul_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001659}
1660
1661/*
Simon Butcherf5ba0452015-12-27 23:01:55 +00001662 * Unsigned integer divide - double mbedtls_mpi_uint dividend, u1/u0, and
1663 * mbedtls_mpi_uint divisor, d
Simon Butcher15b15d12015-11-26 19:35:03 +00001664 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001665static mbedtls_mpi_uint mbedtls_int_div_int( mbedtls_mpi_uint u1,
1666 mbedtls_mpi_uint u0, mbedtls_mpi_uint d, mbedtls_mpi_uint *r )
Simon Butcher15b15d12015-11-26 19:35:03 +00001667{
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001668#if defined(MBEDTLS_HAVE_UDBL)
1669 mbedtls_t_udbl dividend, quotient;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001670#else
Simon Butcher9803d072016-01-03 00:24:34 +00001671 const mbedtls_mpi_uint radix = (mbedtls_mpi_uint) 1 << biH;
1672 const mbedtls_mpi_uint uint_halfword_mask = ( (mbedtls_mpi_uint) 1 << biH ) - 1;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001673 mbedtls_mpi_uint d0, d1, q0, q1, rAX, r0, quotient;
1674 mbedtls_mpi_uint u0_msw, u0_lsw;
Simon Butcher9803d072016-01-03 00:24:34 +00001675 size_t s;
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001676#endif
1677
Simon Butcher15b15d12015-11-26 19:35:03 +00001678 /*
1679 * Check for overflow
1680 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001681 if( 0 == d || u1 >= d )
Simon Butcher15b15d12015-11-26 19:35:03 +00001682 {
Simon Butcherf5ba0452015-12-27 23:01:55 +00001683 if (r != NULL) *r = ~0;
Simon Butcher15b15d12015-11-26 19:35:03 +00001684
Simon Butcherf5ba0452015-12-27 23:01:55 +00001685 return ( ~0 );
Simon Butcher15b15d12015-11-26 19:35:03 +00001686 }
1687
1688#if defined(MBEDTLS_HAVE_UDBL)
Simon Butcher15b15d12015-11-26 19:35:03 +00001689 dividend = (mbedtls_t_udbl) u1 << biL;
1690 dividend |= (mbedtls_t_udbl) u0;
1691 quotient = dividend / d;
1692 if( quotient > ( (mbedtls_t_udbl) 1 << biL ) - 1 )
1693 quotient = ( (mbedtls_t_udbl) 1 << biL ) - 1;
1694
1695 if( r != NULL )
Simon Butcher9803d072016-01-03 00:24:34 +00001696 *r = (mbedtls_mpi_uint)( dividend - (quotient * d ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001697
1698 return (mbedtls_mpi_uint) quotient;
1699#else
Simon Butcher15b15d12015-11-26 19:35:03 +00001700
1701 /*
1702 * Algorithm D, Section 4.3.1 - The Art of Computer Programming
1703 * Vol. 2 - Seminumerical Algorithms, Knuth
1704 */
1705
1706 /*
1707 * Normalize the divisor, d, and dividend, u0, u1
1708 */
1709 s = mbedtls_clz( d );
1710 d = d << s;
1711
1712 u1 = u1 << s;
Simon Butcher9803d072016-01-03 00:24:34 +00001713 u1 |= ( u0 >> ( biL - s ) ) & ( -(mbedtls_mpi_sint)s >> ( biL - 1 ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001714 u0 = u0 << s;
1715
1716 d1 = d >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001717 d0 = d & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001718
1719 u0_msw = u0 >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001720 u0_lsw = u0 & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001721
1722 /*
1723 * Find the first quotient and remainder
1724 */
1725 q1 = u1 / d1;
1726 r0 = u1 - d1 * q1;
1727
1728 while( q1 >= radix || ( q1 * d0 > radix * r0 + u0_msw ) )
1729 {
1730 q1 -= 1;
1731 r0 += d1;
1732
1733 if ( r0 >= radix ) break;
1734 }
1735
Simon Butcherf5ba0452015-12-27 23:01:55 +00001736 rAX = ( u1 * radix ) + ( u0_msw - q1 * d );
Simon Butcher15b15d12015-11-26 19:35:03 +00001737 q0 = rAX / d1;
1738 r0 = rAX - q0 * d1;
1739
1740 while( q0 >= radix || ( q0 * d0 > radix * r0 + u0_lsw ) )
1741 {
1742 q0 -= 1;
1743 r0 += d1;
1744
1745 if ( r0 >= radix ) break;
1746 }
1747
1748 if (r != NULL)
Simon Butcherf5ba0452015-12-27 23:01:55 +00001749 *r = ( rAX * radix + u0_lsw - q0 * d ) >> s;
Simon Butcher15b15d12015-11-26 19:35:03 +00001750
1751 quotient = q1 * radix + q0;
1752
1753 return quotient;
1754#endif
1755}
1756
1757/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001758 * Division by mbedtls_mpi: A = Q * B + R (HAC 14.20)
Paul Bakker5121ce52009-01-03 21:22:43 +00001759 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001760int mbedtls_mpi_div_mpi( mbedtls_mpi *Q, mbedtls_mpi *R, const mbedtls_mpi *A,
1761 const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001762{
Janos Follath24eed8d2019-11-22 13:21:35 +00001763 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001764 size_t i, n, t, k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001765 mbedtls_mpi X, Y, Z, T1, T2;
Alexander Kd19a1932019-11-01 18:20:42 +03001766 mbedtls_mpi_uint TP2[3];
Hanno Becker73d7d792018-12-11 10:35:51 +00001767 MPI_VALIDATE_RET( A != NULL );
1768 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001769
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001770 if( mbedtls_mpi_cmp_int( B, 0 ) == 0 )
1771 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001772
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001773 mbedtls_mpi_init( &X ); mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001774 mbedtls_mpi_init( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001775 /*
1776 * Avoid dynamic memory allocations for constant-size T2.
1777 *
1778 * T2 is used for comparison only and the 3 limbs are assigned explicitly,
1779 * so nobody increase the size of the MPI and we're safe to use an on-stack
1780 * buffer.
1781 */
Alexander K35d6d462019-10-31 14:46:45 +03001782 T2.s = 1;
Alexander Kd19a1932019-11-01 18:20:42 +03001783 T2.n = sizeof( TP2 ) / sizeof( *TP2 );
1784 T2.p = TP2;
Paul Bakker5121ce52009-01-03 21:22:43 +00001785
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001786 if( mbedtls_mpi_cmp_abs( A, B ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001787 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001788 if( Q != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_lset( Q, 0 ) );
1789 if( R != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001790 return( 0 );
1791 }
1792
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001793 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &X, A ) );
1794 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001795 X.s = Y.s = 1;
1796
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001797 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &Z, A->n + 2 ) );
1798 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Z, 0 ) );
1799 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T1, 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001800
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001801 k = mbedtls_mpi_bitlen( &Y ) % biL;
Paul Bakkerf9688572011-05-05 10:00:45 +00001802 if( k < biL - 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001803 {
1804 k = biL - 1 - k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001805 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &X, k ) );
1806 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, k ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001807 }
1808 else k = 0;
1809
1810 n = X.n - 1;
1811 t = Y.n - 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001812 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001813
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001814 while( mbedtls_mpi_cmp_mpi( &X, &Y ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001815 {
1816 Z.p[n - t]++;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001817 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001818 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001819 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001820
1821 for( i = n; i > t ; i-- )
1822 {
1823 if( X.p[i] >= Y.p[t] )
1824 Z.p[i - t - 1] = ~0;
1825 else
1826 {
Simon Butcher15b15d12015-11-26 19:35:03 +00001827 Z.p[i - t - 1] = mbedtls_int_div_int( X.p[i], X.p[i - 1],
1828 Y.p[t], NULL);
Paul Bakker5121ce52009-01-03 21:22:43 +00001829 }
1830
Alexander K35d6d462019-10-31 14:46:45 +03001831 T2.p[0] = ( i < 2 ) ? 0 : X.p[i - 2];
1832 T2.p[1] = ( i < 1 ) ? 0 : X.p[i - 1];
1833 T2.p[2] = X.p[i];
1834
Paul Bakker5121ce52009-01-03 21:22:43 +00001835 Z.p[i - t - 1]++;
1836 do
1837 {
1838 Z.p[i - t - 1]--;
1839
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001840 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &T1, 0 ) );
Paul Bakker66d5d072014-06-17 16:39:18 +02001841 T1.p[0] = ( t < 1 ) ? 0 : Y.p[t - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001842 T1.p[1] = Y.p[t];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001843 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &T1, Z.p[i - t - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001844 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001845 while( mbedtls_mpi_cmp_mpi( &T1, &T2 ) > 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001846
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001847 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &Y, Z.p[i - t - 1] ) );
1848 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1849 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001850
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001851 if( mbedtls_mpi_cmp_int( &X, 0 ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001852 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001853 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T1, &Y ) );
1854 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1855 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001856 Z.p[i - t - 1]--;
1857 }
1858 }
1859
1860 if( Q != NULL )
1861 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001862 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( Q, &Z ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001863 Q->s = A->s * B->s;
1864 }
1865
1866 if( R != NULL )
1867 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001868 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &X, k ) );
Paul Bakkerf02c5642012-11-13 10:25:21 +00001869 X.s = A->s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001870 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, &X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001871
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001872 if( mbedtls_mpi_cmp_int( R, 0 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001873 R->s = 1;
1874 }
1875
1876cleanup:
1877
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001878 mbedtls_mpi_free( &X ); mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001879 mbedtls_mpi_free( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001880 mbedtls_platform_zeroize( TP2, sizeof( TP2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001881
1882 return( ret );
1883}
1884
1885/*
1886 * Division by int: A = Q * b + R
Paul Bakker5121ce52009-01-03 21:22:43 +00001887 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001888int mbedtls_mpi_div_int( mbedtls_mpi *Q, mbedtls_mpi *R,
1889 const mbedtls_mpi *A,
1890 mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001891{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001892 mbedtls_mpi _B;
1893 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001894 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001895
1896 p[0] = ( b < 0 ) ? -b : b;
1897 _B.s = ( b < 0 ) ? -1 : 1;
1898 _B.n = 1;
1899 _B.p = p;
1900
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001901 return( mbedtls_mpi_div_mpi( Q, R, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001902}
1903
1904/*
1905 * Modulo: R = A mod B
1906 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001907int mbedtls_mpi_mod_mpi( mbedtls_mpi *R, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001908{
Janos Follath24eed8d2019-11-22 13:21:35 +00001909 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker73d7d792018-12-11 10:35:51 +00001910 MPI_VALIDATE_RET( R != NULL );
1911 MPI_VALIDATE_RET( A != NULL );
1912 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001913
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001914 if( mbedtls_mpi_cmp_int( B, 0 ) < 0 )
1915 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001916
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001917 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( NULL, R, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001918
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001919 while( mbedtls_mpi_cmp_int( R, 0 ) < 0 )
1920 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001921
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001922 while( mbedtls_mpi_cmp_mpi( R, B ) >= 0 )
1923 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001924
1925cleanup:
1926
1927 return( ret );
1928}
1929
1930/*
1931 * Modulo: r = A mod b
1932 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001933int mbedtls_mpi_mod_int( mbedtls_mpi_uint *r, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001934{
Paul Bakker23986e52011-04-24 08:57:21 +00001935 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001936 mbedtls_mpi_uint x, y, z;
Hanno Becker73d7d792018-12-11 10:35:51 +00001937 MPI_VALIDATE_RET( r != NULL );
1938 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001939
1940 if( b == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001941 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
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_NEGATIVE_VALUE );
Paul Bakker5121ce52009-01-03 21:22:43 +00001945
1946 /*
1947 * handle trivial cases
1948 */
1949 if( b == 1 )
1950 {
1951 *r = 0;
1952 return( 0 );
1953 }
1954
1955 if( b == 2 )
1956 {
1957 *r = A->p[0] & 1;
1958 return( 0 );
1959 }
1960
1961 /*
1962 * general case
1963 */
Paul Bakker23986e52011-04-24 08:57:21 +00001964 for( i = A->n, y = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001965 {
Paul Bakker23986e52011-04-24 08:57:21 +00001966 x = A->p[i - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001967 y = ( y << biH ) | ( x >> biH );
1968 z = y / b;
1969 y -= z * b;
1970
1971 x <<= biH;
1972 y = ( y << biH ) | ( x >> biH );
1973 z = y / b;
1974 y -= z * b;
1975 }
1976
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001977 /*
1978 * If A is negative, then the current y represents a negative value.
1979 * Flipping it to the positive side.
1980 */
1981 if( A->s < 0 && y != 0 )
1982 y = b - y;
1983
Paul Bakker5121ce52009-01-03 21:22:43 +00001984 *r = y;
1985
1986 return( 0 );
1987}
1988
1989/*
1990 * Fast Montgomery initialization (thanks to Tom St Denis)
1991 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001992static void mpi_montg_init( mbedtls_mpi_uint *mm, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00001993{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001994 mbedtls_mpi_uint x, m0 = N->p[0];
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01001995 unsigned int i;
Paul Bakker5121ce52009-01-03 21:22:43 +00001996
1997 x = m0;
1998 x += ( ( m0 + 2 ) & 4 ) << 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00001999
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01002000 for( i = biL; i >= 8; i /= 2 )
2001 x *= ( 2 - ( m0 * x ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002002
2003 *mm = ~x + 1;
2004}
2005
Gilles Peskine2a82f722020-06-04 15:00:49 +02002006/** Montgomery multiplication: A = A * B * R^-1 mod N (HAC 14.36)
2007 *
2008 * \param[in,out] A One of the numbers to multiply.
Gilles Peskine221626f2020-06-08 22:37:50 +02002009 * It must have at least as many limbs as N
2010 * (A->n >= N->n), and any limbs beyond n are ignored.
Gilles Peskine2a82f722020-06-04 15:00:49 +02002011 * On successful completion, A contains the result of
2012 * the multiplication A * B * R^-1 mod N where
2013 * R = (2^ciL)^n.
2014 * \param[in] B One of the numbers to multiply.
2015 * It must be nonzero and must not have more limbs than N
2016 * (B->n <= N->n).
2017 * \param[in] N The modulo. N must be odd.
2018 * \param mm The value calculated by `mpi_montg_init(&mm, N)`.
2019 * This is -N^-1 mod 2^ciL.
2020 * \param[in,out] T A bignum for temporary storage.
2021 * It must be at least twice the limb size of N plus 2
2022 * (T->n >= 2 * (N->n + 1)).
2023 * Its initial content is unused and
2024 * its final content is indeterminate.
2025 * Note that unlike the usual convention in the library
2026 * for `const mbedtls_mpi*`, the content of T can change.
Paul Bakker5121ce52009-01-03 21:22:43 +00002027 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002028static 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 +02002029 const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002030{
Paul Bakker23986e52011-04-24 08:57:21 +00002031 size_t i, n, m;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002032 mbedtls_mpi_uint u0, u1, *d;
Paul Bakker5121ce52009-01-03 21:22:43 +00002033
2034 memset( T->p, 0, T->n * ciL );
2035
2036 d = T->p;
2037 n = N->n;
2038 m = ( B->n < n ) ? B->n : n;
2039
2040 for( i = 0; i < n; i++ )
2041 {
2042 /*
2043 * T = (T + u0*B + u1*N) / 2^biL
2044 */
2045 u0 = A->p[i];
2046 u1 = ( d[0] + u0 * B->p[0] ) * mm;
2047
2048 mpi_mul_hlp( m, B->p, d, u0 );
2049 mpi_mul_hlp( n, N->p, d, u1 );
2050
2051 *d++ = u0; d[n + 1] = 0;
2052 }
2053
Gilles Peskine221626f2020-06-08 22:37:50 +02002054 /* At this point, d is either the desired result or the desired result
2055 * plus N. We now potentially subtract N, avoiding leaking whether the
2056 * subtraction is performed through side channels. */
Paul Bakker5121ce52009-01-03 21:22:43 +00002057
Gilles Peskine221626f2020-06-08 22:37:50 +02002058 /* Copy the n least significant limbs of d to A, so that
2059 * A = d if d < N (recall that N has n limbs). */
2060 memcpy( A->p, d, n * ciL );
Gilles Peskine09ec10a2020-06-09 10:39:38 +02002061 /* If d >= N then we want to set A to d - N. To prevent timing attacks,
Gilles Peskine221626f2020-06-08 22:37:50 +02002062 * do the calculation without using conditional tests. */
2063 /* Set d to d0 + (2^biL)^n - N where d0 is the current value of d. */
Gilles Peskine132c0972020-06-04 21:05:24 +02002064 d[n] += 1;
Gilles Peskinec097e9e2020-06-08 21:58:22 +02002065 d[n] -= mpi_sub_hlp( n, d, N->p );
Gilles Peskine221626f2020-06-08 22:37:50 +02002066 /* If d0 < N then d < (2^biL)^n
2067 * so d[n] == 0 and we want to keep A as it is.
2068 * If d0 >= N then d >= (2^biL)^n, and d <= (2^biL)^n + N < 2 * (2^biL)^n
2069 * so d[n] == 1 and we want to set A to the result of the subtraction
2070 * which is d - (2^biL)^n, i.e. the n least significant limbs of d.
2071 * This exactly corresponds to a conditional assignment. */
2072 mpi_safe_cond_assign( n, A->p, d, (unsigned char) d[n] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002073}
2074
2075/*
2076 * Montgomery reduction: A = A * R^-1 mod N
Gilles Peskine2a82f722020-06-04 15:00:49 +02002077 *
2078 * See mpi_montmul() regarding constraints and guarantees on the parameters.
Paul Bakker5121ce52009-01-03 21:22:43 +00002079 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002080static void mpi_montred( mbedtls_mpi *A, const mbedtls_mpi *N,
2081 mbedtls_mpi_uint mm, const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002082{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002083 mbedtls_mpi_uint z = 1;
2084 mbedtls_mpi U;
Paul Bakker5121ce52009-01-03 21:22:43 +00002085
Paul Bakker8ddb6452013-02-27 14:56:33 +01002086 U.n = U.s = (int) z;
Paul Bakker5121ce52009-01-03 21:22:43 +00002087 U.p = &z;
2088
Gilles Peskine4e91d472020-06-04 20:55:15 +02002089 mpi_montmul( A, &U, N, mm, T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002090}
2091
2092/*
2093 * Sliding-window exponentiation: X = A^E mod N (HAC 14.85)
2094 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002095int mbedtls_mpi_exp_mod( mbedtls_mpi *X, const mbedtls_mpi *A,
2096 const mbedtls_mpi *E, const mbedtls_mpi *N,
2097 mbedtls_mpi *_RR )
Paul Bakker5121ce52009-01-03 21:22:43 +00002098{
Janos Follath24eed8d2019-11-22 13:21:35 +00002099 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002100 size_t wbits, wsize, one = 1;
2101 size_t i, j, nblimbs;
2102 size_t bufsize, nbits;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002103 mbedtls_mpi_uint ei, mm, state;
2104 mbedtls_mpi RR, T, W[ 2 << MBEDTLS_MPI_WINDOW_SIZE ], Apos;
Paul Bakkerf6198c12012-05-16 08:02:29 +00002105 int neg;
Paul Bakker5121ce52009-01-03 21:22:43 +00002106
Hanno Becker73d7d792018-12-11 10:35:51 +00002107 MPI_VALIDATE_RET( X != NULL );
2108 MPI_VALIDATE_RET( A != NULL );
2109 MPI_VALIDATE_RET( E != NULL );
2110 MPI_VALIDATE_RET( N != NULL );
2111
Hanno Becker8d1dd1b2017-09-28 11:02:24 +01002112 if( mbedtls_mpi_cmp_int( N, 0 ) <= 0 || ( N->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002113 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002114
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002115 if( mbedtls_mpi_cmp_int( E, 0 ) < 0 )
2116 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002117
Chris Jones9246d042020-11-25 15:12:39 +00002118 if( mbedtls_mpi_bitlen( E ) > MBEDTLS_MPI_MAX_BITS ||
2119 mbedtls_mpi_bitlen( N ) > MBEDTLS_MPI_MAX_BITS )
2120 return ( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
2121
Paul Bakkerf6198c12012-05-16 08:02:29 +00002122 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002123 * Init temps and window size
2124 */
2125 mpi_montg_init( &mm, N );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002126 mbedtls_mpi_init( &RR ); mbedtls_mpi_init( &T );
2127 mbedtls_mpi_init( &Apos );
Paul Bakker5121ce52009-01-03 21:22:43 +00002128 memset( W, 0, sizeof( W ) );
2129
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002130 i = mbedtls_mpi_bitlen( E );
Paul Bakker5121ce52009-01-03 21:22:43 +00002131
2132 wsize = ( i > 671 ) ? 6 : ( i > 239 ) ? 5 :
2133 ( i > 79 ) ? 4 : ( i > 23 ) ? 3 : 1;
2134
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002135#if( MBEDTLS_MPI_WINDOW_SIZE < 6 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002136 if( wsize > MBEDTLS_MPI_WINDOW_SIZE )
2137 wsize = MBEDTLS_MPI_WINDOW_SIZE;
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002138#endif
Paul Bakkerb6d5f082011-11-25 11:52:11 +00002139
Paul Bakker5121ce52009-01-03 21:22:43 +00002140 j = N->n + 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002141 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
2142 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[1], j ) );
2143 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T, j * 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002144
2145 /*
Paul Bakker50546922012-05-19 08:40:49 +00002146 * Compensate for negative A (and correct at the end)
2147 */
2148 neg = ( A->s == -1 );
Paul Bakker50546922012-05-19 08:40:49 +00002149 if( neg )
2150 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002151 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Apos, A ) );
Paul Bakker50546922012-05-19 08:40:49 +00002152 Apos.s = 1;
2153 A = &Apos;
2154 }
2155
2156 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002157 * If 1st call, pre-compute R^2 mod N
2158 */
2159 if( _RR == NULL || _RR->p == NULL )
2160 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002161 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &RR, 1 ) );
2162 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &RR, N->n * 2 * biL ) );
2163 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &RR, &RR, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002164
2165 if( _RR != NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002166 memcpy( _RR, &RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002167 }
2168 else
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 /*
2172 * W[1] = A * R^2 * R^-1 mod N = A * R mod N
2173 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002174 if( mbedtls_mpi_cmp_mpi( A, N ) >= 0 )
2175 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &W[1], A, N ) );
Paul Bakkerc2024f42014-01-23 20:38:35 +01002176 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002177 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[1], A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002178
Gilles Peskine4e91d472020-06-04 20:55:15 +02002179 mpi_montmul( &W[1], &RR, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002180
2181 /*
2182 * X = R^2 * R^-1 mod N = R mod N
2183 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002184 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &RR ) );
Gilles Peskine4e91d472020-06-04 20:55:15 +02002185 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002186
2187 if( wsize > 1 )
2188 {
2189 /*
2190 * W[1 << (wsize - 1)] = W[1] ^ (wsize - 1)
2191 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002192 j = one << ( wsize - 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002193
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002194 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[j], N->n + 1 ) );
2195 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[j], &W[1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002196
2197 for( i = 0; i < wsize - 1; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002198 mpi_montmul( &W[j], &W[j], N, mm, &T );
Paul Bakker0d7702c2013-10-29 16:18:35 +01002199
Paul Bakker5121ce52009-01-03 21:22:43 +00002200 /*
2201 * W[i] = W[i - 1] * W[1]
2202 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002203 for( i = j + 1; i < ( one << wsize ); i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002204 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002205 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[i], N->n + 1 ) );
2206 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[i], &W[i - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002207
Gilles Peskine4e91d472020-06-04 20:55:15 +02002208 mpi_montmul( &W[i], &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002209 }
2210 }
2211
2212 nblimbs = E->n;
2213 bufsize = 0;
2214 nbits = 0;
2215 wbits = 0;
2216 state = 0;
2217
2218 while( 1 )
2219 {
2220 if( bufsize == 0 )
2221 {
Paul Bakker0d7702c2013-10-29 16:18:35 +01002222 if( nblimbs == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002223 break;
2224
Paul Bakker0d7702c2013-10-29 16:18:35 +01002225 nblimbs--;
2226
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002227 bufsize = sizeof( mbedtls_mpi_uint ) << 3;
Paul Bakker5121ce52009-01-03 21:22:43 +00002228 }
2229
2230 bufsize--;
2231
2232 ei = (E->p[nblimbs] >> bufsize) & 1;
2233
2234 /*
2235 * skip leading 0s
2236 */
2237 if( ei == 0 && state == 0 )
2238 continue;
2239
2240 if( ei == 0 && state == 1 )
2241 {
2242 /*
2243 * out of window, square X
2244 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002245 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002246 continue;
2247 }
2248
2249 /*
2250 * add ei to current window
2251 */
2252 state = 2;
2253
2254 nbits++;
Paul Bakker66d5d072014-06-17 16:39:18 +02002255 wbits |= ( ei << ( wsize - nbits ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002256
2257 if( nbits == wsize )
2258 {
2259 /*
2260 * X = X^wsize R^-1 mod N
2261 */
2262 for( i = 0; i < wsize; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002263 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002264
2265 /*
2266 * X = X * W[wbits] R^-1 mod N
2267 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002268 mpi_montmul( X, &W[wbits], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002269
2270 state--;
2271 nbits = 0;
2272 wbits = 0;
2273 }
2274 }
2275
2276 /*
2277 * process the remaining bits
2278 */
2279 for( i = 0; i < nbits; i++ )
2280 {
Gilles Peskine4e91d472020-06-04 20:55:15 +02002281 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002282
2283 wbits <<= 1;
2284
Paul Bakker66d5d072014-06-17 16:39:18 +02002285 if( ( wbits & ( one << wsize ) ) != 0 )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002286 mpi_montmul( X, &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002287 }
2288
2289 /*
2290 * X = A^E * R * R^-1 mod N = A^E mod N
2291 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002292 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002293
Hanno Beckera4af1c42017-04-18 09:07:45 +01002294 if( neg && E->n != 0 && ( E->p[0] & 1 ) != 0 )
Paul Bakkerf6198c12012-05-16 08:02:29 +00002295 {
2296 X->s = -1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002297 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( X, N, X ) );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002298 }
2299
Paul Bakker5121ce52009-01-03 21:22:43 +00002300cleanup:
2301
Paul Bakker66d5d072014-06-17 16:39:18 +02002302 for( i = ( one << ( wsize - 1 ) ); i < ( one << wsize ); i++ )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002303 mbedtls_mpi_free( &W[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002304
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002305 mbedtls_mpi_free( &W[1] ); mbedtls_mpi_free( &T ); mbedtls_mpi_free( &Apos );
Paul Bakker6c591fa2011-05-05 11:49:20 +00002306
Paul Bakker75a28602014-03-31 12:08:17 +02002307 if( _RR == NULL || _RR->p == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002308 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002309
2310 return( ret );
2311}
2312
Paul Bakker5121ce52009-01-03 21:22:43 +00002313/*
2314 * Greatest common divisor: G = gcd(A, B) (HAC 14.54)
2315 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002316int mbedtls_mpi_gcd( mbedtls_mpi *G, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00002317{
Janos Follath24eed8d2019-11-22 13:21:35 +00002318 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002319 size_t lz, lzt;
Alexander Ke8ad49f2019-08-16 16:16:07 +03002320 mbedtls_mpi TA, TB;
Paul Bakker5121ce52009-01-03 21:22:43 +00002321
Hanno Becker73d7d792018-12-11 10:35:51 +00002322 MPI_VALIDATE_RET( G != NULL );
2323 MPI_VALIDATE_RET( A != NULL );
2324 MPI_VALIDATE_RET( B != NULL );
2325
Alexander Ke8ad49f2019-08-16 16:16:07 +03002326 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002327
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002328 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) );
2329 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002330
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002331 lz = mbedtls_mpi_lsb( &TA );
2332 lzt = mbedtls_mpi_lsb( &TB );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002333
Paul Bakker66d5d072014-06-17 16:39:18 +02002334 if( lzt < lz )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002335 lz = lzt;
2336
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002337 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, lz ) );
2338 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, lz ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002339
Paul Bakker5121ce52009-01-03 21:22:43 +00002340 TA.s = TB.s = 1;
2341
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002342 while( mbedtls_mpi_cmp_int( &TA, 0 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002343 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002344 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, mbedtls_mpi_lsb( &TA ) ) );
2345 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, mbedtls_mpi_lsb( &TB ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002346
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002347 if( mbedtls_mpi_cmp_mpi( &TA, &TB ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002348 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002349 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TA, &TA, &TB ) );
2350 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002351 }
2352 else
2353 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002354 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TB, &TB, &TA ) );
2355 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002356 }
2357 }
2358
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002359 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &TB, lz ) );
2360 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( G, &TB ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002361
2362cleanup:
2363
Alexander Ke8ad49f2019-08-16 16:16:07 +03002364 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002365
2366 return( ret );
2367}
2368
Paul Bakker33dc46b2014-04-30 16:11:39 +02002369/*
2370 * Fill X with size bytes of random.
2371 *
2372 * Use a temporary bytes representation to make sure the result is the same
Paul Bakkerc37b0ac2014-05-01 14:19:23 +02002373 * regardless of the platform endianness (useful when f_rng is actually
Paul Bakker33dc46b2014-04-30 16:11:39 +02002374 * deterministic, eg for tests).
2375 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002376int mbedtls_mpi_fill_random( mbedtls_mpi *X, size_t size,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002377 int (*f_rng)(void *, unsigned char *, size_t),
2378 void *p_rng )
Paul Bakker287781a2011-03-26 13:18:49 +00002379{
Janos Follath24eed8d2019-11-22 13:21:35 +00002380 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker6dab6202019-01-02 16:42:29 +00002381 size_t const limbs = CHARS_TO_LIMBS( size );
Hanno Beckerda1655a2017-10-18 14:21:44 +01002382 size_t const overhead = ( limbs * ciL ) - size;
2383 unsigned char *Xp;
2384
Hanno Becker8ce11a32018-12-19 16:18:52 +00002385 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002386 MPI_VALIDATE_RET( f_rng != NULL );
Paul Bakker33dc46b2014-04-30 16:11:39 +02002387
Hanno Beckerda1655a2017-10-18 14:21:44 +01002388 /* Ensure that target MPI has exactly the necessary number of limbs */
2389 if( X->n != limbs )
2390 {
2391 mbedtls_mpi_free( X );
2392 mbedtls_mpi_init( X );
2393 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
2394 }
2395 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker287781a2011-03-26 13:18:49 +00002396
Hanno Beckerda1655a2017-10-18 14:21:44 +01002397 Xp = (unsigned char*) X->p;
2398 f_rng( p_rng, Xp + overhead, size );
2399
Hanno Becker2be8a552018-10-25 12:40:09 +01002400 mpi_bigendian_to_host( X->p, limbs );
Paul Bakker287781a2011-03-26 13:18:49 +00002401
2402cleanup:
2403 return( ret );
2404}
2405
Paul Bakker5121ce52009-01-03 21:22:43 +00002406/*
2407 * Modular inverse: X = A^-1 mod N (HAC 14.61 / 14.64)
2408 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002409int mbedtls_mpi_inv_mod( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002410{
Janos Follath24eed8d2019-11-22 13:21:35 +00002411 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002412 mbedtls_mpi G, TA, TU, U1, U2, TB, TV, V1, V2;
Hanno Becker73d7d792018-12-11 10:35:51 +00002413 MPI_VALIDATE_RET( X != NULL );
2414 MPI_VALIDATE_RET( A != NULL );
2415 MPI_VALIDATE_RET( N != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00002416
Hanno Becker4bcb4912017-04-18 15:49:39 +01002417 if( mbedtls_mpi_cmp_int( N, 1 ) <= 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002418 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002419
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002420 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TU ); mbedtls_mpi_init( &U1 ); mbedtls_mpi_init( &U2 );
2421 mbedtls_mpi_init( &G ); mbedtls_mpi_init( &TB ); mbedtls_mpi_init( &TV );
2422 mbedtls_mpi_init( &V1 ); mbedtls_mpi_init( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002423
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002424 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &G, A, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002425
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002426 if( mbedtls_mpi_cmp_int( &G, 1 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002427 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002428 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002429 goto cleanup;
2430 }
2431
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002432 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &TA, A, N ) );
2433 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TU, &TA ) );
2434 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, N ) );
2435 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TV, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002436
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002437 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U1, 1 ) );
2438 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U2, 0 ) );
2439 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V1, 0 ) );
2440 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002441
2442 do
2443 {
2444 while( ( TU.p[0] & 1 ) == 0 )
2445 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002446 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TU, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002447
2448 if( ( U1.p[0] & 1 ) != 0 || ( U2.p[0] & 1 ) != 0 )
2449 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002450 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &U1, &U1, &TB ) );
2451 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002452 }
2453
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002454 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U1, 1 ) );
2455 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002456 }
2457
2458 while( ( TV.p[0] & 1 ) == 0 )
2459 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002460 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TV, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002461
2462 if( ( V1.p[0] & 1 ) != 0 || ( V2.p[0] & 1 ) != 0 )
2463 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002464 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, &TB ) );
2465 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002466 }
2467
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002468 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V1, 1 ) );
2469 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002470 }
2471
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002472 if( mbedtls_mpi_cmp_mpi( &TU, &TV ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002473 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002474 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TU, &TU, &TV ) );
2475 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U1, &U1, &V1 ) );
2476 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &V2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002477 }
2478 else
2479 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002480 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TV, &TV, &TU ) );
2481 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, &U1 ) );
2482 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &U2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002483 }
2484 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002485 while( mbedtls_mpi_cmp_int( &TU, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002486
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002487 while( mbedtls_mpi_cmp_int( &V1, 0 ) < 0 )
2488 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002489
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002490 while( mbedtls_mpi_cmp_mpi( &V1, N ) >= 0 )
2491 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002492
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002493 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &V1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002494
2495cleanup:
2496
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002497 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TU ); mbedtls_mpi_free( &U1 ); mbedtls_mpi_free( &U2 );
2498 mbedtls_mpi_free( &G ); mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TV );
2499 mbedtls_mpi_free( &V1 ); mbedtls_mpi_free( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002500
2501 return( ret );
2502}
2503
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002504#if defined(MBEDTLS_GENPRIME)
Paul Bakkerd9374b02012-11-02 11:02:58 +00002505
Paul Bakker5121ce52009-01-03 21:22:43 +00002506static const int small_prime[] =
2507{
2508 3, 5, 7, 11, 13, 17, 19, 23,
2509 29, 31, 37, 41, 43, 47, 53, 59,
2510 61, 67, 71, 73, 79, 83, 89, 97,
2511 101, 103, 107, 109, 113, 127, 131, 137,
2512 139, 149, 151, 157, 163, 167, 173, 179,
2513 181, 191, 193, 197, 199, 211, 223, 227,
2514 229, 233, 239, 241, 251, 257, 263, 269,
2515 271, 277, 281, 283, 293, 307, 311, 313,
2516 317, 331, 337, 347, 349, 353, 359, 367,
2517 373, 379, 383, 389, 397, 401, 409, 419,
2518 421, 431, 433, 439, 443, 449, 457, 461,
2519 463, 467, 479, 487, 491, 499, 503, 509,
2520 521, 523, 541, 547, 557, 563, 569, 571,
2521 577, 587, 593, 599, 601, 607, 613, 617,
2522 619, 631, 641, 643, 647, 653, 659, 661,
2523 673, 677, 683, 691, 701, 709, 719, 727,
2524 733, 739, 743, 751, 757, 761, 769, 773,
2525 787, 797, 809, 811, 821, 823, 827, 829,
2526 839, 853, 857, 859, 863, 877, 881, 883,
2527 887, 907, 911, 919, 929, 937, 941, 947,
2528 953, 967, 971, 977, 983, 991, 997, -103
2529};
2530
2531/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002532 * Small divisors test (X must be positive)
2533 *
2534 * Return values:
2535 * 0: no small factor (possible prime, more tests needed)
2536 * 1: certain prime
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002537 * MBEDTLS_ERR_MPI_NOT_ACCEPTABLE: certain non-prime
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002538 * other negative: error
Paul Bakker5121ce52009-01-03 21:22:43 +00002539 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002540static int mpi_check_small_factors( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +00002541{
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002542 int ret = 0;
2543 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002544 mbedtls_mpi_uint r;
Paul Bakker5121ce52009-01-03 21:22:43 +00002545
Paul Bakker5121ce52009-01-03 21:22:43 +00002546 if( ( X->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002547 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002548
2549 for( i = 0; small_prime[i] > 0; i++ )
2550 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002551 if( mbedtls_mpi_cmp_int( X, small_prime[i] ) <= 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002552 return( 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002553
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002554 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, small_prime[i] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002555
2556 if( r == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002557 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002558 }
2559
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002560cleanup:
2561 return( ret );
2562}
2563
2564/*
2565 * Miller-Rabin pseudo-primality test (HAC 4.24)
2566 */
Janos Follathda31fa12018-09-03 14:45:23 +01002567static int mpi_miller_rabin( const mbedtls_mpi *X, size_t rounds,
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002568 int (*f_rng)(void *, unsigned char *, size_t),
2569 void *p_rng )
2570{
Pascal Junodb99183d2015-03-11 16:49:45 +01002571 int ret, count;
Janos Follathda31fa12018-09-03 14:45:23 +01002572 size_t i, j, k, s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002573 mbedtls_mpi W, R, T, A, RR;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002574
Hanno Becker8ce11a32018-12-19 16:18:52 +00002575 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002576 MPI_VALIDATE_RET( f_rng != NULL );
2577
2578 mbedtls_mpi_init( &W ); mbedtls_mpi_init( &R );
2579 mbedtls_mpi_init( &T ); mbedtls_mpi_init( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002580 mbedtls_mpi_init( &RR );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002581
Paul Bakker5121ce52009-01-03 21:22:43 +00002582 /*
2583 * W = |X| - 1
2584 * R = W >> lsb( W )
2585 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002586 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( &W, X, 1 ) );
2587 s = mbedtls_mpi_lsb( &W );
2588 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &R, &W ) );
2589 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &R, s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002590
Janos Follathda31fa12018-09-03 14:45:23 +01002591 for( i = 0; i < rounds; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002592 {
2593 /*
2594 * pick a random A, 1 < A < |X| - 1
2595 */
Pascal Junodb99183d2015-03-11 16:49:45 +01002596 count = 0;
2597 do {
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002598 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( &A, X->n * ciL, f_rng, p_rng ) );
Pascal Junodb99183d2015-03-11 16:49:45 +01002599
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002600 j = mbedtls_mpi_bitlen( &A );
2601 k = mbedtls_mpi_bitlen( &W );
Pascal Junodb99183d2015-03-11 16:49:45 +01002602 if (j > k) {
Darryl Greene3f95ed2018-10-02 13:21:35 +01002603 A.p[A.n - 1] &= ( (mbedtls_mpi_uint) 1 << ( k - ( A.n - 1 ) * biL - 1 ) ) - 1;
Pascal Junodb99183d2015-03-11 16:49:45 +01002604 }
2605
2606 if (count++ > 30) {
Jens Wiklanderf08aa3e2019-01-17 13:30:57 +01002607 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
2608 goto cleanup;
Pascal Junodb99183d2015-03-11 16:49:45 +01002609 }
2610
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002611 } while ( mbedtls_mpi_cmp_mpi( &A, &W ) >= 0 ||
2612 mbedtls_mpi_cmp_int( &A, 1 ) <= 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002613
2614 /*
2615 * A = A^R mod |X|
2616 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002617 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &A, &A, &R, X, &RR ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002618
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002619 if( mbedtls_mpi_cmp_mpi( &A, &W ) == 0 ||
2620 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002621 continue;
2622
2623 j = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002624 while( j < s && mbedtls_mpi_cmp_mpi( &A, &W ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002625 {
2626 /*
2627 * A = A * A mod |X|
2628 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002629 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &T, &A, &A ) );
2630 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &A, &T, X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002631
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002632 if( mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002633 break;
2634
2635 j++;
2636 }
2637
2638 /*
2639 * not prime if A != |X| - 1 or A == 1
2640 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002641 if( mbedtls_mpi_cmp_mpi( &A, &W ) != 0 ||
2642 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002643 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002644 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002645 break;
2646 }
2647 }
2648
2649cleanup:
Hanno Becker73d7d792018-12-11 10:35:51 +00002650 mbedtls_mpi_free( &W ); mbedtls_mpi_free( &R );
2651 mbedtls_mpi_free( &T ); mbedtls_mpi_free( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002652 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002653
2654 return( ret );
2655}
2656
2657/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002658 * Pseudo-primality test: small factors, then Miller-Rabin
2659 */
Janos Follatha0b67c22018-09-18 14:48:23 +01002660int mbedtls_mpi_is_prime_ext( const mbedtls_mpi *X, int rounds,
2661 int (*f_rng)(void *, unsigned char *, size_t),
2662 void *p_rng )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002663{
Janos Follath24eed8d2019-11-22 13:21:35 +00002664 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002665 mbedtls_mpi XX;
Hanno Becker8ce11a32018-12-19 16:18:52 +00002666 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002667 MPI_VALIDATE_RET( f_rng != NULL );
Manuel Pégourié-Gonnard7f4ed672014-10-14 20:56:02 +02002668
2669 XX.s = 1;
2670 XX.n = X->n;
2671 XX.p = X->p;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002672
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002673 if( mbedtls_mpi_cmp_int( &XX, 0 ) == 0 ||
2674 mbedtls_mpi_cmp_int( &XX, 1 ) == 0 )
2675 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002676
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002677 if( mbedtls_mpi_cmp_int( &XX, 2 ) == 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002678 return( 0 );
2679
2680 if( ( ret = mpi_check_small_factors( &XX ) ) != 0 )
2681 {
2682 if( ret == 1 )
2683 return( 0 );
2684
2685 return( ret );
2686 }
2687
Janos Follathda31fa12018-09-03 14:45:23 +01002688 return( mpi_miller_rabin( &XX, rounds, f_rng, p_rng ) );
Janos Follathf301d232018-08-14 13:34:01 +01002689}
2690
Janos Follatha0b67c22018-09-18 14:48:23 +01002691#if !defined(MBEDTLS_DEPRECATED_REMOVED)
Janos Follathf301d232018-08-14 13:34:01 +01002692/*
2693 * Pseudo-primality test, error probability 2^-80
2694 */
2695int mbedtls_mpi_is_prime( const mbedtls_mpi *X,
2696 int (*f_rng)(void *, unsigned char *, size_t),
2697 void *p_rng )
2698{
Hanno Becker8ce11a32018-12-19 16:18:52 +00002699 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002700 MPI_VALIDATE_RET( f_rng != NULL );
2701
Janos Follatha0b67c22018-09-18 14:48:23 +01002702 /*
2703 * In the past our key generation aimed for an error rate of at most
2704 * 2^-80. Since this function is deprecated, aim for the same certainty
2705 * here as well.
2706 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002707 return( mbedtls_mpi_is_prime_ext( X, 40, f_rng, p_rng ) );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002708}
Janos Follatha0b67c22018-09-18 14:48:23 +01002709#endif
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002710
2711/*
Paul Bakker5121ce52009-01-03 21:22:43 +00002712 * Prime number generation
Jethro Beekman66689272018-02-14 19:24:10 -08002713 *
Janos Follathf301d232018-08-14 13:34:01 +01002714 * To generate an RSA key in a way recommended by FIPS 186-4, both primes must
2715 * be either 1024 bits or 1536 bits long, and flags must contain
2716 * MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR.
Paul Bakker5121ce52009-01-03 21:22:43 +00002717 */
Janos Follath7c025a92018-08-14 11:08:41 +01002718int mbedtls_mpi_gen_prime( mbedtls_mpi *X, size_t nbits, int flags,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002719 int (*f_rng)(void *, unsigned char *, size_t),
2720 void *p_rng )
Paul Bakker5121ce52009-01-03 21:22:43 +00002721{
Jethro Beekman66689272018-02-14 19:24:10 -08002722#ifdef MBEDTLS_HAVE_INT64
2723// ceil(2^63.5)
2724#define CEIL_MAXUINT_DIV_SQRT2 0xb504f333f9de6485ULL
2725#else
2726// ceil(2^31.5)
2727#define CEIL_MAXUINT_DIV_SQRT2 0xb504f334U
2728#endif
2729 int ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker23986e52011-04-24 08:57:21 +00002730 size_t k, n;
Janos Follathda31fa12018-09-03 14:45:23 +01002731 int rounds;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002732 mbedtls_mpi_uint r;
2733 mbedtls_mpi Y;
Paul Bakker5121ce52009-01-03 21:22:43 +00002734
Hanno Becker8ce11a32018-12-19 16:18:52 +00002735 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002736 MPI_VALIDATE_RET( f_rng != NULL );
2737
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002738 if( nbits < 3 || nbits > MBEDTLS_MPI_MAX_BITS )
2739 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002740
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002741 mbedtls_mpi_init( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002742
2743 n = BITS_TO_LIMBS( nbits );
2744
Janos Follathda31fa12018-09-03 14:45:23 +01002745 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR ) == 0 )
2746 {
2747 /*
2748 * 2^-80 error probability, number of rounds chosen per HAC, table 4.4
2749 */
2750 rounds = ( ( nbits >= 1300 ) ? 2 : ( nbits >= 850 ) ? 3 :
2751 ( nbits >= 650 ) ? 4 : ( nbits >= 350 ) ? 8 :
2752 ( nbits >= 250 ) ? 12 : ( nbits >= 150 ) ? 18 : 27 );
2753 }
2754 else
2755 {
2756 /*
2757 * 2^-100 error probability, number of rounds computed based on HAC,
2758 * fact 4.48
2759 */
2760 rounds = ( ( nbits >= 1450 ) ? 4 : ( nbits >= 1150 ) ? 5 :
2761 ( nbits >= 1000 ) ? 6 : ( nbits >= 850 ) ? 7 :
2762 ( nbits >= 750 ) ? 8 : ( nbits >= 500 ) ? 13 :
2763 ( nbits >= 250 ) ? 28 : ( nbits >= 150 ) ? 40 : 51 );
2764 }
2765
Jethro Beekman66689272018-02-14 19:24:10 -08002766 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002767 {
Jethro Beekman66689272018-02-14 19:24:10 -08002768 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( X, n * ciL, f_rng, p_rng ) );
2769 /* make sure generated number is at least (nbits-1)+0.5 bits (FIPS 186-4 §B.3.3 steps 4.4, 5.5) */
2770 if( X->p[n-1] < CEIL_MAXUINT_DIV_SQRT2 ) continue;
2771
2772 k = n * biL;
2773 if( k > nbits ) MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( X, k - nbits ) );
2774 X->p[0] |= 1;
2775
Janos Follath7c025a92018-08-14 11:08:41 +01002776 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_DH ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002777 {
Janos Follatha0b67c22018-09-18 14:48:23 +01002778 ret = mbedtls_mpi_is_prime_ext( X, rounds, f_rng, p_rng );
Jethro Beekman66689272018-02-14 19:24:10 -08002779
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002780 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
Paul Bakker5121ce52009-01-03 21:22:43 +00002781 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002782 }
Jethro Beekman66689272018-02-14 19:24:10 -08002783 else
Paul Bakker5121ce52009-01-03 21:22:43 +00002784 {
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002785 /*
Jethro Beekman66689272018-02-14 19:24:10 -08002786 * An necessary condition for Y and X = 2Y + 1 to be prime
2787 * is X = 2 mod 3 (which is equivalent to Y = 2 mod 3).
2788 * Make sure it is satisfied, while keeping X = 3 mod 4
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002789 */
Jethro Beekman66689272018-02-14 19:24:10 -08002790
2791 X->p[0] |= 2;
2792
2793 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, 3 ) );
2794 if( r == 0 )
2795 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 8 ) );
2796 else if( r == 1 )
2797 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 4 ) );
2798
2799 /* Set Y = (X-1) / 2, which is X / 2 because X is odd */
2800 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, X ) );
2801 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, 1 ) );
2802
2803 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002804 {
Jethro Beekman66689272018-02-14 19:24:10 -08002805 /*
2806 * First, check small factors for X and Y
2807 * before doing Miller-Rabin on any of them
2808 */
2809 if( ( ret = mpi_check_small_factors( X ) ) == 0 &&
2810 ( ret = mpi_check_small_factors( &Y ) ) == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002811 ( ret = mpi_miller_rabin( X, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002812 == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002813 ( ret = mpi_miller_rabin( &Y, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002814 == 0 )
Jethro Beekman66689272018-02-14 19:24:10 -08002815 goto cleanup;
2816
2817 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
2818 goto cleanup;
2819
2820 /*
2821 * Next candidates. We want to preserve Y = (X-1) / 2 and
2822 * Y = 1 mod 2 and Y = 2 mod 3 (eq X = 3 mod 4 and X = 2 mod 3)
2823 * so up Y by 6 and X by 12.
2824 */
2825 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 12 ) );
2826 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( &Y, &Y, 6 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002827 }
Paul Bakker5121ce52009-01-03 21:22:43 +00002828 }
2829 }
2830
2831cleanup:
2832
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002833 mbedtls_mpi_free( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002834
2835 return( ret );
2836}
2837
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002838#endif /* MBEDTLS_GENPRIME */
Paul Bakker5121ce52009-01-03 21:22:43 +00002839
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002840#if defined(MBEDTLS_SELF_TEST)
Paul Bakker5121ce52009-01-03 21:22:43 +00002841
Paul Bakker23986e52011-04-24 08:57:21 +00002842#define GCD_PAIR_COUNT 3
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002843
2844static const int gcd_pairs[GCD_PAIR_COUNT][3] =
2845{
2846 { 693, 609, 21 },
2847 { 1764, 868, 28 },
2848 { 768454923, 542167814, 1 }
2849};
2850
Paul Bakker5121ce52009-01-03 21:22:43 +00002851/*
2852 * Checkup routine
2853 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002854int mbedtls_mpi_self_test( int verbose )
Paul Bakker5121ce52009-01-03 21:22:43 +00002855{
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002856 int ret, i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002857 mbedtls_mpi A, E, N, X, Y, U, V;
Paul Bakker5121ce52009-01-03 21:22:43 +00002858
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002859 mbedtls_mpi_init( &A ); mbedtls_mpi_init( &E ); mbedtls_mpi_init( &N ); mbedtls_mpi_init( &X );
2860 mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &U ); mbedtls_mpi_init( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002861
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002862 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &A, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002863 "EFE021C2645FD1DC586E69184AF4A31E" \
2864 "D5F53E93B5F123FA41680867BA110131" \
2865 "944FE7952E2517337780CB0DB80E61AA" \
2866 "E7C8DDC6C5C6AADEB34EB38A2F40D5E6" ) );
2867
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002868 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &E, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002869 "B2E7EFD37075B9F03FF989C7C5051C20" \
2870 "34D2A323810251127E7BF8625A4F49A5" \
2871 "F3E27F4DA8BD59C47D6DAABA4C8127BD" \
2872 "5B5C25763222FEFCCFC38B832366C29E" ) );
2873
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002874 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &N, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002875 "0066A198186C18C10B2F5ED9B522752A" \
2876 "9830B69916E535C8F047518A889A43A5" \
2877 "94B6BED27A168D31D4A52F88925AA8F5" ) );
2878
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002879 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002880
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002881 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002882 "602AB7ECA597A3D6B56FF9829A5E8B85" \
2883 "9E857EA95A03512E2BAE7391688D264A" \
2884 "A5663B0341DB9CCFD2C4C5F421FEC814" \
2885 "8001B72E848A38CAE1C65F78E56ABDEF" \
2886 "E12D3C039B8A02D6BE593F0BBBDA56F1" \
2887 "ECF677152EF804370C1A305CAF3B5BF1" \
2888 "30879B56C61DE584A0F53A2447A51E" ) );
2889
2890 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002891 mbedtls_printf( " MPI test #1 (mul_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002892
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002893 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002894 {
2895 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002896 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002897
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002898 ret = 1;
2899 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002900 }
2901
2902 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002903 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002904
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002905 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( &X, &Y, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002906
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002907 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002908 "256567336059E52CAE22925474705F39A94" ) );
2909
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002910 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &V, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002911 "6613F26162223DF488E9CD48CC132C7A" \
2912 "0AC93C701B001B092E4E5B9F73BCD27B" \
2913 "9EE50D0657C77F374E903CDFA4C642" ) );
2914
2915 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002916 mbedtls_printf( " MPI test #2 (div_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002917
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002918 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 ||
2919 mbedtls_mpi_cmp_mpi( &Y, &V ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002920 {
2921 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002922 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002923
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002924 ret = 1;
2925 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002926 }
2927
2928 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002929 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002930
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002931 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &X, &A, &E, &N, NULL ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002932
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002933 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002934 "36E139AEA55215609D2816998ED020BB" \
2935 "BD96C37890F65171D948E9BC7CBAA4D9" \
2936 "325D24D6A3C12710F10A09FA08AB87" ) );
2937
2938 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002939 mbedtls_printf( " MPI test #3 (exp_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002940
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002941 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002942 {
2943 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002944 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002945
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002946 ret = 1;
2947 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002948 }
2949
2950 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002951 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002952
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002953 MBEDTLS_MPI_CHK( mbedtls_mpi_inv_mod( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002954
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002955 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002956 "003A0AAEDD7E784FC07D8F9EC6E3BFD5" \
2957 "C3DBA76456363A10869622EAC2DD84EC" \
2958 "C5B8A74DAC4D09E03B5E0BE779F2DF61" ) );
2959
2960 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002961 mbedtls_printf( " MPI test #4 (inv_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002962
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002963 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002964 {
2965 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002966 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002967
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002968 ret = 1;
2969 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002970 }
2971
2972 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002973 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002974
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002975 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002976 mbedtls_printf( " MPI test #5 (simple gcd): " );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002977
Paul Bakker66d5d072014-06-17 16:39:18 +02002978 for( i = 0; i < GCD_PAIR_COUNT; i++ )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002979 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002980 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &X, gcd_pairs[i][0] ) );
2981 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Y, gcd_pairs[i][1] ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002982
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002983 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &A, &X, &Y ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002984
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002985 if( mbedtls_mpi_cmp_int( &A, gcd_pairs[i][2] ) != 0 )
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002986 {
2987 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002988 mbedtls_printf( "failed at %d\n", i );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002989
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002990 ret = 1;
2991 goto cleanup;
2992 }
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002993 }
2994
2995 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002996 mbedtls_printf( "passed\n" );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002997
Paul Bakker5121ce52009-01-03 21:22:43 +00002998cleanup:
2999
3000 if( ret != 0 && verbose != 0 )
Kenneth Soerensen518d4352020-04-01 17:22:45 +02003001 mbedtls_printf( "Unexpected error, return code = %08X\n", (unsigned int) ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00003002
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003003 mbedtls_mpi_free( &A ); mbedtls_mpi_free( &E ); mbedtls_mpi_free( &N ); mbedtls_mpi_free( &X );
3004 mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &U ); mbedtls_mpi_free( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00003005
3006 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003007 mbedtls_printf( "\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00003008
3009 return( ret );
3010}
3011
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003012#endif /* MBEDTLS_SELF_TEST */
Paul Bakker5121ce52009-01-03 21:22:43 +00003013
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003014#endif /* MBEDTLS_BIGNUM_C */