blob: 5cd1c3e842b8fe6d7695353bad15efe94f851ccf [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 *
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001342 * Calculate l - r where l and r have the same size.
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001343 * This function operates modulo (2^ciL)^n and returns the carry
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001344 * (1 if there was a wraparound, i.e. if `l < r`, and 0 otherwise).
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001345 *
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001346 * d may be aliased to l or r.
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001347 *
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001348 * \param n Number of limbs of \p d, \p l and \p r.
1349 * \param[out] d The result of the subtraction.
1350 * \param[in] l The left operand.
1351 * \param[in] r The right operand.
1352 *
1353 * \return 1 if `l < r`.
1354 * 0 if `l >= r`.
Paul Bakker5121ce52009-01-03 21:22:43 +00001355 */
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001356static mbedtls_mpi_uint mpi_sub_hlp( size_t n,
1357 mbedtls_mpi_uint *d,
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001358 const mbedtls_mpi_uint *l,
1359 const mbedtls_mpi_uint *r )
Paul Bakker5121ce52009-01-03 21:22:43 +00001360{
Paul Bakker23986e52011-04-24 08:57:21 +00001361 size_t i;
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001362 mbedtls_mpi_uint c = 0, t, z;
Paul Bakker5121ce52009-01-03 21:22:43 +00001363
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001364 for( i = 0; i < n; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00001365 {
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001366 z = ( l[i] < c ); t = l[i] - c;
1367 c = ( t < r[i] ) + z; d[i] = t - r[i];
Paul Bakker5121ce52009-01-03 21:22:43 +00001368 }
1369
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001370 return( c );
Paul Bakker5121ce52009-01-03 21:22:43 +00001371}
1372
1373/*
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001374 * Unsigned subtraction: X = |A| - |B| (HAC 14.9, 14.10)
Paul Bakker5121ce52009-01-03 21:22:43 +00001375 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001376int mbedtls_mpi_sub_abs( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001377{
Janos Follath24eed8d2019-11-22 13:21:35 +00001378 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001379 size_t n;
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001380 mbedtls_mpi_uint carry;
Hanno Becker73d7d792018-12-11 10:35:51 +00001381 MPI_VALIDATE_RET( X != NULL );
1382 MPI_VALIDATE_RET( A != NULL );
1383 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001384
Paul Bakker23986e52011-04-24 08:57:21 +00001385 for( n = B->n; n > 0; n-- )
1386 if( B->p[n - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001387 break;
1388
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001389 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, A->n ) );
1390
1391 /* Set the high limbs of X to match A. Don't touch the lower limbs
1392 * because X might be aliased to B, and we must not overwrite the
1393 * significant digits of B. */
1394 if( A->n > n )
1395 memcpy( X->p + n, A->p + n, ( A->n - n ) * ciL );
1396 if( X->n > A->n )
1397 memset( X->p + A->n, 0, ( X->n - A->n ) * ciL );
1398
1399 carry = mpi_sub_hlp( n, X->p, A->p, B->p );
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001400 if( carry != 0 )
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001401 {
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001402 /* Propagate the carry to the first nonzero limb of X. */
1403 for( ; n < X->n && X->p[n] == 0; n++ )
1404 --X->p[n];
1405 /* If we ran out of space for the carry, it means that the result
1406 * is negative. */
1407 if( n == X->n )
Gilles Peskine89b41302020-07-23 01:16:46 +02001408 {
1409 ret = MBEDTLS_ERR_MPI_NEGATIVE_VALUE;
1410 goto cleanup;
1411 }
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001412 --X->p[n];
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001413 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001414
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001415 /* X should always be positive as a result of unsigned subtractions. */
1416 X->s = 1;
1417
Paul Bakker5121ce52009-01-03 21:22:43 +00001418cleanup:
Paul Bakker5121ce52009-01-03 21:22:43 +00001419 return( ret );
1420}
1421
1422/*
1423 * Signed addition: X = A + B
1424 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001425int mbedtls_mpi_add_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001426{
Hanno Becker73d7d792018-12-11 10:35:51 +00001427 int ret, s;
1428 MPI_VALIDATE_RET( X != NULL );
1429 MPI_VALIDATE_RET( A != NULL );
1430 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001431
Hanno Becker73d7d792018-12-11 10:35:51 +00001432 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001433 if( A->s * B->s < 0 )
1434 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001435 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001436 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001437 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001438 X->s = s;
1439 }
1440 else
1441 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001442 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001443 X->s = -s;
1444 }
1445 }
1446 else
1447 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001448 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001449 X->s = s;
1450 }
1451
1452cleanup:
1453
1454 return( ret );
1455}
1456
1457/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001458 * Signed subtraction: X = A - B
Paul Bakker5121ce52009-01-03 21:22:43 +00001459 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001460int mbedtls_mpi_sub_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001461{
Hanno Becker73d7d792018-12-11 10:35:51 +00001462 int ret, s;
1463 MPI_VALIDATE_RET( X != NULL );
1464 MPI_VALIDATE_RET( A != NULL );
1465 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001466
Hanno Becker73d7d792018-12-11 10:35:51 +00001467 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001468 if( A->s * B->s > 0 )
1469 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001470 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001471 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001472 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001473 X->s = s;
1474 }
1475 else
1476 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001477 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001478 X->s = -s;
1479 }
1480 }
1481 else
1482 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001483 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001484 X->s = s;
1485 }
1486
1487cleanup:
1488
1489 return( ret );
1490}
1491
1492/*
1493 * Signed addition: X = A + b
1494 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001495int mbedtls_mpi_add_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001496{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001497 mbedtls_mpi _B;
1498 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001499 MPI_VALIDATE_RET( X != NULL );
1500 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001501
1502 p[0] = ( b < 0 ) ? -b : b;
1503 _B.s = ( b < 0 ) ? -1 : 1;
1504 _B.n = 1;
1505 _B.p = p;
1506
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001507 return( mbedtls_mpi_add_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001508}
1509
1510/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001511 * Signed subtraction: X = A - b
Paul Bakker5121ce52009-01-03 21:22:43 +00001512 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001513int mbedtls_mpi_sub_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001514{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001515 mbedtls_mpi _B;
1516 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001517 MPI_VALIDATE_RET( X != NULL );
1518 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001519
1520 p[0] = ( b < 0 ) ? -b : b;
1521 _B.s = ( b < 0 ) ? -1 : 1;
1522 _B.n = 1;
1523 _B.p = p;
1524
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001525 return( mbedtls_mpi_sub_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001526}
1527
1528/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001529 * Helper for mbedtls_mpi multiplication
Paul Bakkerfc4f46f2013-06-24 19:23:56 +02001530 */
1531static
1532#if defined(__APPLE__) && defined(__arm__)
1533/*
1534 * Apple LLVM version 4.2 (clang-425.0.24) (based on LLVM 3.2svn)
1535 * appears to need this to prevent bad ARM code generation at -O3.
1536 */
1537__attribute__ ((noinline))
1538#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001539void 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 +00001540{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001541 mbedtls_mpi_uint c = 0, t = 0;
Paul Bakker5121ce52009-01-03 21:22:43 +00001542
1543#if defined(MULADDC_HUIT)
1544 for( ; i >= 8; i -= 8 )
1545 {
1546 MULADDC_INIT
1547 MULADDC_HUIT
1548 MULADDC_STOP
1549 }
1550
1551 for( ; i > 0; i-- )
1552 {
1553 MULADDC_INIT
1554 MULADDC_CORE
1555 MULADDC_STOP
1556 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001557#else /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001558 for( ; i >= 16; i -= 16 )
1559 {
1560 MULADDC_INIT
1561 MULADDC_CORE MULADDC_CORE
1562 MULADDC_CORE MULADDC_CORE
1563 MULADDC_CORE MULADDC_CORE
1564 MULADDC_CORE MULADDC_CORE
1565
1566 MULADDC_CORE MULADDC_CORE
1567 MULADDC_CORE MULADDC_CORE
1568 MULADDC_CORE MULADDC_CORE
1569 MULADDC_CORE MULADDC_CORE
1570 MULADDC_STOP
1571 }
1572
1573 for( ; i >= 8; i -= 8 )
1574 {
1575 MULADDC_INIT
1576 MULADDC_CORE MULADDC_CORE
1577 MULADDC_CORE MULADDC_CORE
1578
1579 MULADDC_CORE MULADDC_CORE
1580 MULADDC_CORE MULADDC_CORE
1581 MULADDC_STOP
1582 }
1583
1584 for( ; i > 0; i-- )
1585 {
1586 MULADDC_INIT
1587 MULADDC_CORE
1588 MULADDC_STOP
1589 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001590#endif /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001591
1592 t++;
1593
1594 do {
1595 *d += c; c = ( *d < c ); d++;
1596 }
1597 while( c != 0 );
1598}
1599
1600/*
1601 * Baseline multiplication: X = A * B (HAC 14.12)
1602 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001603int mbedtls_mpi_mul_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001604{
Janos Follath24eed8d2019-11-22 13:21:35 +00001605 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001606 size_t i, j;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001607 mbedtls_mpi TA, TB;
Hanno Becker73d7d792018-12-11 10:35:51 +00001608 MPI_VALIDATE_RET( X != NULL );
1609 MPI_VALIDATE_RET( A != NULL );
1610 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001611
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001612 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001613
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001614 if( X == A ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) ); A = &TA; }
1615 if( X == B ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) ); B = &TB; }
Paul Bakker5121ce52009-01-03 21:22:43 +00001616
Paul Bakker23986e52011-04-24 08:57:21 +00001617 for( i = A->n; i > 0; i-- )
1618 if( A->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001619 break;
1620
Paul Bakker23986e52011-04-24 08:57:21 +00001621 for( j = B->n; j > 0; j-- )
1622 if( B->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001623 break;
1624
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001625 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + j ) );
1626 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001627
Alexey Skalozub8e75e682016-01-13 21:59:27 +02001628 for( ; j > 0; j-- )
1629 mpi_mul_hlp( i, A->p, X->p + j - 1, B->p[j - 1] );
Paul Bakker5121ce52009-01-03 21:22:43 +00001630
1631 X->s = A->s * B->s;
1632
1633cleanup:
1634
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001635 mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TA );
Paul Bakker5121ce52009-01-03 21:22:43 +00001636
1637 return( ret );
1638}
1639
1640/*
1641 * Baseline multiplication: X = A * b
1642 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001643int mbedtls_mpi_mul_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001644{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001645 mbedtls_mpi _B;
1646 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001647 MPI_VALIDATE_RET( X != NULL );
1648 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001649
1650 _B.s = 1;
1651 _B.n = 1;
1652 _B.p = p;
1653 p[0] = b;
1654
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001655 return( mbedtls_mpi_mul_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001656}
1657
1658/*
Simon Butcherf5ba0452015-12-27 23:01:55 +00001659 * Unsigned integer divide - double mbedtls_mpi_uint dividend, u1/u0, and
1660 * mbedtls_mpi_uint divisor, d
Simon Butcher15b15d12015-11-26 19:35:03 +00001661 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001662static mbedtls_mpi_uint mbedtls_int_div_int( mbedtls_mpi_uint u1,
1663 mbedtls_mpi_uint u0, mbedtls_mpi_uint d, mbedtls_mpi_uint *r )
Simon Butcher15b15d12015-11-26 19:35:03 +00001664{
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001665#if defined(MBEDTLS_HAVE_UDBL)
1666 mbedtls_t_udbl dividend, quotient;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001667#else
Simon Butcher9803d072016-01-03 00:24:34 +00001668 const mbedtls_mpi_uint radix = (mbedtls_mpi_uint) 1 << biH;
1669 const mbedtls_mpi_uint uint_halfword_mask = ( (mbedtls_mpi_uint) 1 << biH ) - 1;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001670 mbedtls_mpi_uint d0, d1, q0, q1, rAX, r0, quotient;
1671 mbedtls_mpi_uint u0_msw, u0_lsw;
Simon Butcher9803d072016-01-03 00:24:34 +00001672 size_t s;
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001673#endif
1674
Simon Butcher15b15d12015-11-26 19:35:03 +00001675 /*
1676 * Check for overflow
1677 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001678 if( 0 == d || u1 >= d )
Simon Butcher15b15d12015-11-26 19:35:03 +00001679 {
Simon Butcherf5ba0452015-12-27 23:01:55 +00001680 if (r != NULL) *r = ~0;
Simon Butcher15b15d12015-11-26 19:35:03 +00001681
Simon Butcherf5ba0452015-12-27 23:01:55 +00001682 return ( ~0 );
Simon Butcher15b15d12015-11-26 19:35:03 +00001683 }
1684
1685#if defined(MBEDTLS_HAVE_UDBL)
Simon Butcher15b15d12015-11-26 19:35:03 +00001686 dividend = (mbedtls_t_udbl) u1 << biL;
1687 dividend |= (mbedtls_t_udbl) u0;
1688 quotient = dividend / d;
1689 if( quotient > ( (mbedtls_t_udbl) 1 << biL ) - 1 )
1690 quotient = ( (mbedtls_t_udbl) 1 << biL ) - 1;
1691
1692 if( r != NULL )
Simon Butcher9803d072016-01-03 00:24:34 +00001693 *r = (mbedtls_mpi_uint)( dividend - (quotient * d ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001694
1695 return (mbedtls_mpi_uint) quotient;
1696#else
Simon Butcher15b15d12015-11-26 19:35:03 +00001697
1698 /*
1699 * Algorithm D, Section 4.3.1 - The Art of Computer Programming
1700 * Vol. 2 - Seminumerical Algorithms, Knuth
1701 */
1702
1703 /*
1704 * Normalize the divisor, d, and dividend, u0, u1
1705 */
1706 s = mbedtls_clz( d );
1707 d = d << s;
1708
1709 u1 = u1 << s;
Simon Butcher9803d072016-01-03 00:24:34 +00001710 u1 |= ( u0 >> ( biL - s ) ) & ( -(mbedtls_mpi_sint)s >> ( biL - 1 ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001711 u0 = u0 << s;
1712
1713 d1 = d >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001714 d0 = d & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001715
1716 u0_msw = u0 >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001717 u0_lsw = u0 & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001718
1719 /*
1720 * Find the first quotient and remainder
1721 */
1722 q1 = u1 / d1;
1723 r0 = u1 - d1 * q1;
1724
1725 while( q1 >= radix || ( q1 * d0 > radix * r0 + u0_msw ) )
1726 {
1727 q1 -= 1;
1728 r0 += d1;
1729
1730 if ( r0 >= radix ) break;
1731 }
1732
Simon Butcherf5ba0452015-12-27 23:01:55 +00001733 rAX = ( u1 * radix ) + ( u0_msw - q1 * d );
Simon Butcher15b15d12015-11-26 19:35:03 +00001734 q0 = rAX / d1;
1735 r0 = rAX - q0 * d1;
1736
1737 while( q0 >= radix || ( q0 * d0 > radix * r0 + u0_lsw ) )
1738 {
1739 q0 -= 1;
1740 r0 += d1;
1741
1742 if ( r0 >= radix ) break;
1743 }
1744
1745 if (r != NULL)
Simon Butcherf5ba0452015-12-27 23:01:55 +00001746 *r = ( rAX * radix + u0_lsw - q0 * d ) >> s;
Simon Butcher15b15d12015-11-26 19:35:03 +00001747
1748 quotient = q1 * radix + q0;
1749
1750 return quotient;
1751#endif
1752}
1753
1754/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001755 * Division by mbedtls_mpi: A = Q * B + R (HAC 14.20)
Paul Bakker5121ce52009-01-03 21:22:43 +00001756 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001757int mbedtls_mpi_div_mpi( mbedtls_mpi *Q, mbedtls_mpi *R, const mbedtls_mpi *A,
1758 const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001759{
Janos Follath24eed8d2019-11-22 13:21:35 +00001760 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001761 size_t i, n, t, k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001762 mbedtls_mpi X, Y, Z, T1, T2;
Alexander Kd19a1932019-11-01 18:20:42 +03001763 mbedtls_mpi_uint TP2[3];
Hanno Becker73d7d792018-12-11 10:35:51 +00001764 MPI_VALIDATE_RET( A != NULL );
1765 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001766
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001767 if( mbedtls_mpi_cmp_int( B, 0 ) == 0 )
1768 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001769
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001770 mbedtls_mpi_init( &X ); mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001771 mbedtls_mpi_init( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001772 /*
1773 * Avoid dynamic memory allocations for constant-size T2.
1774 *
1775 * T2 is used for comparison only and the 3 limbs are assigned explicitly,
1776 * so nobody increase the size of the MPI and we're safe to use an on-stack
1777 * buffer.
1778 */
Alexander K35d6d462019-10-31 14:46:45 +03001779 T2.s = 1;
Alexander Kd19a1932019-11-01 18:20:42 +03001780 T2.n = sizeof( TP2 ) / sizeof( *TP2 );
1781 T2.p = TP2;
Paul Bakker5121ce52009-01-03 21:22:43 +00001782
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001783 if( mbedtls_mpi_cmp_abs( A, B ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001784 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001785 if( Q != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_lset( Q, 0 ) );
1786 if( R != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001787 return( 0 );
1788 }
1789
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001790 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &X, A ) );
1791 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001792 X.s = Y.s = 1;
1793
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001794 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &Z, A->n + 2 ) );
1795 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Z, 0 ) );
1796 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T1, 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001797
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001798 k = mbedtls_mpi_bitlen( &Y ) % biL;
Paul Bakkerf9688572011-05-05 10:00:45 +00001799 if( k < biL - 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001800 {
1801 k = biL - 1 - k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001802 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &X, k ) );
1803 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, k ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001804 }
1805 else k = 0;
1806
1807 n = X.n - 1;
1808 t = Y.n - 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001809 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001810
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001811 while( mbedtls_mpi_cmp_mpi( &X, &Y ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001812 {
1813 Z.p[n - t]++;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001814 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001815 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001816 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001817
1818 for( i = n; i > t ; i-- )
1819 {
1820 if( X.p[i] >= Y.p[t] )
1821 Z.p[i - t - 1] = ~0;
1822 else
1823 {
Simon Butcher15b15d12015-11-26 19:35:03 +00001824 Z.p[i - t - 1] = mbedtls_int_div_int( X.p[i], X.p[i - 1],
1825 Y.p[t], NULL);
Paul Bakker5121ce52009-01-03 21:22:43 +00001826 }
1827
Alexander K35d6d462019-10-31 14:46:45 +03001828 T2.p[0] = ( i < 2 ) ? 0 : X.p[i - 2];
1829 T2.p[1] = ( i < 1 ) ? 0 : X.p[i - 1];
1830 T2.p[2] = X.p[i];
1831
Paul Bakker5121ce52009-01-03 21:22:43 +00001832 Z.p[i - t - 1]++;
1833 do
1834 {
1835 Z.p[i - t - 1]--;
1836
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001837 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &T1, 0 ) );
Paul Bakker66d5d072014-06-17 16:39:18 +02001838 T1.p[0] = ( t < 1 ) ? 0 : Y.p[t - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001839 T1.p[1] = Y.p[t];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001840 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &T1, Z.p[i - t - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001841 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001842 while( mbedtls_mpi_cmp_mpi( &T1, &T2 ) > 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001843
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001844 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &Y, Z.p[i - t - 1] ) );
1845 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1846 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001847
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001848 if( mbedtls_mpi_cmp_int( &X, 0 ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001849 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001850 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T1, &Y ) );
1851 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1852 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001853 Z.p[i - t - 1]--;
1854 }
1855 }
1856
1857 if( Q != NULL )
1858 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001859 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( Q, &Z ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001860 Q->s = A->s * B->s;
1861 }
1862
1863 if( R != NULL )
1864 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001865 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &X, k ) );
Paul Bakkerf02c5642012-11-13 10:25:21 +00001866 X.s = A->s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001867 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, &X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001868
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001869 if( mbedtls_mpi_cmp_int( R, 0 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001870 R->s = 1;
1871 }
1872
1873cleanup:
1874
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001875 mbedtls_mpi_free( &X ); mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001876 mbedtls_mpi_free( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001877 mbedtls_platform_zeroize( TP2, sizeof( TP2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001878
1879 return( ret );
1880}
1881
1882/*
1883 * Division by int: A = Q * b + R
Paul Bakker5121ce52009-01-03 21:22:43 +00001884 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001885int mbedtls_mpi_div_int( mbedtls_mpi *Q, mbedtls_mpi *R,
1886 const mbedtls_mpi *A,
1887 mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001888{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001889 mbedtls_mpi _B;
1890 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001891 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001892
1893 p[0] = ( b < 0 ) ? -b : b;
1894 _B.s = ( b < 0 ) ? -1 : 1;
1895 _B.n = 1;
1896 _B.p = p;
1897
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001898 return( mbedtls_mpi_div_mpi( Q, R, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001899}
1900
1901/*
1902 * Modulo: R = A mod B
1903 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001904int mbedtls_mpi_mod_mpi( mbedtls_mpi *R, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001905{
Janos Follath24eed8d2019-11-22 13:21:35 +00001906 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker73d7d792018-12-11 10:35:51 +00001907 MPI_VALIDATE_RET( R != NULL );
1908 MPI_VALIDATE_RET( A != NULL );
1909 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001910
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001911 if( mbedtls_mpi_cmp_int( B, 0 ) < 0 )
1912 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001913
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001914 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( NULL, R, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001915
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001916 while( mbedtls_mpi_cmp_int( R, 0 ) < 0 )
1917 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001918
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001919 while( mbedtls_mpi_cmp_mpi( R, B ) >= 0 )
1920 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001921
1922cleanup:
1923
1924 return( ret );
1925}
1926
1927/*
1928 * Modulo: r = A mod b
1929 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001930int mbedtls_mpi_mod_int( mbedtls_mpi_uint *r, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001931{
Paul Bakker23986e52011-04-24 08:57:21 +00001932 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001933 mbedtls_mpi_uint x, y, z;
Hanno Becker73d7d792018-12-11 10:35:51 +00001934 MPI_VALIDATE_RET( r != NULL );
1935 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001936
1937 if( b == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001938 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
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_NEGATIVE_VALUE );
Paul Bakker5121ce52009-01-03 21:22:43 +00001942
1943 /*
1944 * handle trivial cases
1945 */
1946 if( b == 1 )
1947 {
1948 *r = 0;
1949 return( 0 );
1950 }
1951
1952 if( b == 2 )
1953 {
1954 *r = A->p[0] & 1;
1955 return( 0 );
1956 }
1957
1958 /*
1959 * general case
1960 */
Paul Bakker23986e52011-04-24 08:57:21 +00001961 for( i = A->n, y = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001962 {
Paul Bakker23986e52011-04-24 08:57:21 +00001963 x = A->p[i - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001964 y = ( y << biH ) | ( x >> biH );
1965 z = y / b;
1966 y -= z * b;
1967
1968 x <<= biH;
1969 y = ( y << biH ) | ( x >> biH );
1970 z = y / b;
1971 y -= z * b;
1972 }
1973
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001974 /*
1975 * If A is negative, then the current y represents a negative value.
1976 * Flipping it to the positive side.
1977 */
1978 if( A->s < 0 && y != 0 )
1979 y = b - y;
1980
Paul Bakker5121ce52009-01-03 21:22:43 +00001981 *r = y;
1982
1983 return( 0 );
1984}
1985
1986/*
1987 * Fast Montgomery initialization (thanks to Tom St Denis)
1988 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001989static void mpi_montg_init( mbedtls_mpi_uint *mm, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00001990{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001991 mbedtls_mpi_uint x, m0 = N->p[0];
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01001992 unsigned int i;
Paul Bakker5121ce52009-01-03 21:22:43 +00001993
1994 x = m0;
1995 x += ( ( m0 + 2 ) & 4 ) << 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00001996
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01001997 for( i = biL; i >= 8; i /= 2 )
1998 x *= ( 2 - ( m0 * x ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001999
2000 *mm = ~x + 1;
2001}
2002
Gilles Peskine2a82f722020-06-04 15:00:49 +02002003/** Montgomery multiplication: A = A * B * R^-1 mod N (HAC 14.36)
2004 *
2005 * \param[in,out] A One of the numbers to multiply.
Gilles Peskine221626f2020-06-08 22:37:50 +02002006 * It must have at least as many limbs as N
2007 * (A->n >= N->n), and any limbs beyond n are ignored.
Gilles Peskine2a82f722020-06-04 15:00:49 +02002008 * On successful completion, A contains the result of
2009 * the multiplication A * B * R^-1 mod N where
2010 * R = (2^ciL)^n.
2011 * \param[in] B One of the numbers to multiply.
2012 * It must be nonzero and must not have more limbs than N
2013 * (B->n <= N->n).
2014 * \param[in] N The modulo. N must be odd.
2015 * \param mm The value calculated by `mpi_montg_init(&mm, N)`.
2016 * This is -N^-1 mod 2^ciL.
2017 * \param[in,out] T A bignum for temporary storage.
2018 * It must be at least twice the limb size of N plus 2
2019 * (T->n >= 2 * (N->n + 1)).
2020 * Its initial content is unused and
2021 * its final content is indeterminate.
2022 * Note that unlike the usual convention in the library
2023 * for `const mbedtls_mpi*`, the content of T can change.
Paul Bakker5121ce52009-01-03 21:22:43 +00002024 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002025static 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 +02002026 const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002027{
Paul Bakker23986e52011-04-24 08:57:21 +00002028 size_t i, n, m;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002029 mbedtls_mpi_uint u0, u1, *d;
Paul Bakker5121ce52009-01-03 21:22:43 +00002030
2031 memset( T->p, 0, T->n * ciL );
2032
2033 d = T->p;
2034 n = N->n;
2035 m = ( B->n < n ) ? B->n : n;
2036
2037 for( i = 0; i < n; i++ )
2038 {
2039 /*
2040 * T = (T + u0*B + u1*N) / 2^biL
2041 */
2042 u0 = A->p[i];
2043 u1 = ( d[0] + u0 * B->p[0] ) * mm;
2044
2045 mpi_mul_hlp( m, B->p, d, u0 );
2046 mpi_mul_hlp( n, N->p, d, u1 );
2047
2048 *d++ = u0; d[n + 1] = 0;
2049 }
2050
Gilles Peskine221626f2020-06-08 22:37:50 +02002051 /* At this point, d is either the desired result or the desired result
2052 * plus N. We now potentially subtract N, avoiding leaking whether the
2053 * subtraction is performed through side channels. */
Paul Bakker5121ce52009-01-03 21:22:43 +00002054
Gilles Peskine221626f2020-06-08 22:37:50 +02002055 /* Copy the n least significant limbs of d to A, so that
2056 * A = d if d < N (recall that N has n limbs). */
2057 memcpy( A->p, d, n * ciL );
Gilles Peskine09ec10a2020-06-09 10:39:38 +02002058 /* If d >= N then we want to set A to d - N. To prevent timing attacks,
Gilles Peskine221626f2020-06-08 22:37:50 +02002059 * do the calculation without using conditional tests. */
2060 /* Set d to d0 + (2^biL)^n - N where d0 is the current value of d. */
Gilles Peskine132c0972020-06-04 21:05:24 +02002061 d[n] += 1;
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02002062 d[n] -= mpi_sub_hlp( n, d, d, N->p );
Gilles Peskine221626f2020-06-08 22:37:50 +02002063 /* If d0 < N then d < (2^biL)^n
2064 * so d[n] == 0 and we want to keep A as it is.
2065 * If d0 >= N then d >= (2^biL)^n, and d <= (2^biL)^n + N < 2 * (2^biL)^n
2066 * so d[n] == 1 and we want to set A to the result of the subtraction
2067 * which is d - (2^biL)^n, i.e. the n least significant limbs of d.
2068 * This exactly corresponds to a conditional assignment. */
2069 mpi_safe_cond_assign( n, A->p, d, (unsigned char) d[n] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002070}
2071
2072/*
2073 * Montgomery reduction: A = A * R^-1 mod N
Gilles Peskine2a82f722020-06-04 15:00:49 +02002074 *
2075 * See mpi_montmul() regarding constraints and guarantees on the parameters.
Paul Bakker5121ce52009-01-03 21:22:43 +00002076 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002077static void mpi_montred( mbedtls_mpi *A, const mbedtls_mpi *N,
2078 mbedtls_mpi_uint mm, const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002079{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002080 mbedtls_mpi_uint z = 1;
2081 mbedtls_mpi U;
Paul Bakker5121ce52009-01-03 21:22:43 +00002082
Paul Bakker8ddb6452013-02-27 14:56:33 +01002083 U.n = U.s = (int) z;
Paul Bakker5121ce52009-01-03 21:22:43 +00002084 U.p = &z;
2085
Gilles Peskine4e91d472020-06-04 20:55:15 +02002086 mpi_montmul( A, &U, N, mm, T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002087}
2088
2089/*
2090 * Sliding-window exponentiation: X = A^E mod N (HAC 14.85)
2091 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002092int mbedtls_mpi_exp_mod( mbedtls_mpi *X, const mbedtls_mpi *A,
2093 const mbedtls_mpi *E, const mbedtls_mpi *N,
2094 mbedtls_mpi *_RR )
Paul Bakker5121ce52009-01-03 21:22:43 +00002095{
Janos Follath24eed8d2019-11-22 13:21:35 +00002096 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002097 size_t wbits, wsize, one = 1;
2098 size_t i, j, nblimbs;
2099 size_t bufsize, nbits;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002100 mbedtls_mpi_uint ei, mm, state;
Daniel Otte388f9b22020-08-21 12:34:29 +02002101 mbedtls_mpi RR, T, W[ 1 << MBEDTLS_MPI_WINDOW_SIZE ], Apos;
Paul Bakkerf6198c12012-05-16 08:02:29 +00002102 int neg;
Paul Bakker5121ce52009-01-03 21:22:43 +00002103
Hanno Becker73d7d792018-12-11 10:35:51 +00002104 MPI_VALIDATE_RET( X != NULL );
2105 MPI_VALIDATE_RET( A != NULL );
2106 MPI_VALIDATE_RET( E != NULL );
2107 MPI_VALIDATE_RET( N != NULL );
2108
Hanno Becker8d1dd1b2017-09-28 11:02:24 +01002109 if( mbedtls_mpi_cmp_int( N, 0 ) <= 0 || ( N->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002110 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002111
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002112 if( mbedtls_mpi_cmp_int( E, 0 ) < 0 )
2113 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002114
2115 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002116 * Init temps and window size
2117 */
2118 mpi_montg_init( &mm, N );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002119 mbedtls_mpi_init( &RR ); mbedtls_mpi_init( &T );
2120 mbedtls_mpi_init( &Apos );
Paul Bakker5121ce52009-01-03 21:22:43 +00002121 memset( W, 0, sizeof( W ) );
2122
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002123 i = mbedtls_mpi_bitlen( E );
Paul Bakker5121ce52009-01-03 21:22:43 +00002124
2125 wsize = ( i > 671 ) ? 6 : ( i > 239 ) ? 5 :
2126 ( i > 79 ) ? 4 : ( i > 23 ) ? 3 : 1;
2127
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002128#if( MBEDTLS_MPI_WINDOW_SIZE < 6 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002129 if( wsize > MBEDTLS_MPI_WINDOW_SIZE )
2130 wsize = MBEDTLS_MPI_WINDOW_SIZE;
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002131#endif
Paul Bakkerb6d5f082011-11-25 11:52:11 +00002132
Paul Bakker5121ce52009-01-03 21:22:43 +00002133 j = N->n + 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002134 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
2135 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[1], j ) );
2136 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T, j * 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002137
2138 /*
Paul Bakker50546922012-05-19 08:40:49 +00002139 * Compensate for negative A (and correct at the end)
2140 */
2141 neg = ( A->s == -1 );
Paul Bakker50546922012-05-19 08:40:49 +00002142 if( neg )
2143 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002144 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Apos, A ) );
Paul Bakker50546922012-05-19 08:40:49 +00002145 Apos.s = 1;
2146 A = &Apos;
2147 }
2148
2149 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002150 * If 1st call, pre-compute R^2 mod N
2151 */
2152 if( _RR == NULL || _RR->p == NULL )
2153 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002154 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &RR, 1 ) );
2155 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &RR, N->n * 2 * biL ) );
2156 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &RR, &RR, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002157
2158 if( _RR != NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002159 memcpy( _RR, &RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002160 }
2161 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002162 memcpy( &RR, _RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002163
2164 /*
2165 * W[1] = A * R^2 * R^-1 mod N = A * R mod N
2166 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002167 if( mbedtls_mpi_cmp_mpi( A, N ) >= 0 )
2168 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &W[1], A, N ) );
Paul Bakkerc2024f42014-01-23 20:38:35 +01002169 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002170 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[1], A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002171
Gilles Peskine4e91d472020-06-04 20:55:15 +02002172 mpi_montmul( &W[1], &RR, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002173
2174 /*
2175 * X = R^2 * R^-1 mod N = R mod N
2176 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002177 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &RR ) );
Gilles Peskine4e91d472020-06-04 20:55:15 +02002178 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002179
2180 if( wsize > 1 )
2181 {
2182 /*
2183 * W[1 << (wsize - 1)] = W[1] ^ (wsize - 1)
2184 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002185 j = one << ( wsize - 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002186
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002187 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[j], N->n + 1 ) );
2188 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[j], &W[1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002189
2190 for( i = 0; i < wsize - 1; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002191 mpi_montmul( &W[j], &W[j], N, mm, &T );
Paul Bakker0d7702c2013-10-29 16:18:35 +01002192
Paul Bakker5121ce52009-01-03 21:22:43 +00002193 /*
2194 * W[i] = W[i - 1] * W[1]
2195 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002196 for( i = j + 1; i < ( one << wsize ); i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002197 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002198 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[i], N->n + 1 ) );
2199 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[i], &W[i - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002200
Gilles Peskine4e91d472020-06-04 20:55:15 +02002201 mpi_montmul( &W[i], &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002202 }
2203 }
2204
2205 nblimbs = E->n;
2206 bufsize = 0;
2207 nbits = 0;
2208 wbits = 0;
2209 state = 0;
2210
2211 while( 1 )
2212 {
2213 if( bufsize == 0 )
2214 {
Paul Bakker0d7702c2013-10-29 16:18:35 +01002215 if( nblimbs == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002216 break;
2217
Paul Bakker0d7702c2013-10-29 16:18:35 +01002218 nblimbs--;
2219
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002220 bufsize = sizeof( mbedtls_mpi_uint ) << 3;
Paul Bakker5121ce52009-01-03 21:22:43 +00002221 }
2222
2223 bufsize--;
2224
2225 ei = (E->p[nblimbs] >> bufsize) & 1;
2226
2227 /*
2228 * skip leading 0s
2229 */
2230 if( ei == 0 && state == 0 )
2231 continue;
2232
2233 if( ei == 0 && state == 1 )
2234 {
2235 /*
2236 * out of window, square X
2237 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002238 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002239 continue;
2240 }
2241
2242 /*
2243 * add ei to current window
2244 */
2245 state = 2;
2246
2247 nbits++;
Paul Bakker66d5d072014-06-17 16:39:18 +02002248 wbits |= ( ei << ( wsize - nbits ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002249
2250 if( nbits == wsize )
2251 {
2252 /*
2253 * X = X^wsize R^-1 mod N
2254 */
2255 for( i = 0; i < wsize; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002256 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002257
2258 /*
2259 * X = X * W[wbits] R^-1 mod N
2260 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002261 mpi_montmul( X, &W[wbits], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002262
2263 state--;
2264 nbits = 0;
2265 wbits = 0;
2266 }
2267 }
2268
2269 /*
2270 * process the remaining bits
2271 */
2272 for( i = 0; i < nbits; i++ )
2273 {
Gilles Peskine4e91d472020-06-04 20:55:15 +02002274 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002275
2276 wbits <<= 1;
2277
Paul Bakker66d5d072014-06-17 16:39:18 +02002278 if( ( wbits & ( one << wsize ) ) != 0 )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002279 mpi_montmul( X, &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002280 }
2281
2282 /*
2283 * X = A^E * R * R^-1 mod N = A^E mod N
2284 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002285 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002286
Hanno Beckera4af1c42017-04-18 09:07:45 +01002287 if( neg && E->n != 0 && ( E->p[0] & 1 ) != 0 )
Paul Bakkerf6198c12012-05-16 08:02:29 +00002288 {
2289 X->s = -1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002290 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( X, N, X ) );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002291 }
2292
Paul Bakker5121ce52009-01-03 21:22:43 +00002293cleanup:
2294
Paul Bakker66d5d072014-06-17 16:39:18 +02002295 for( i = ( one << ( wsize - 1 ) ); i < ( one << wsize ); i++ )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002296 mbedtls_mpi_free( &W[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002297
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002298 mbedtls_mpi_free( &W[1] ); mbedtls_mpi_free( &T ); mbedtls_mpi_free( &Apos );
Paul Bakker6c591fa2011-05-05 11:49:20 +00002299
Paul Bakker75a28602014-03-31 12:08:17 +02002300 if( _RR == NULL || _RR->p == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002301 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002302
2303 return( ret );
2304}
2305
Paul Bakker5121ce52009-01-03 21:22:43 +00002306/*
2307 * Greatest common divisor: G = gcd(A, B) (HAC 14.54)
2308 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002309int mbedtls_mpi_gcd( mbedtls_mpi *G, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00002310{
Janos Follath24eed8d2019-11-22 13:21:35 +00002311 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002312 size_t lz, lzt;
Alexander Ke8ad49f2019-08-16 16:16:07 +03002313 mbedtls_mpi TA, TB;
Paul Bakker5121ce52009-01-03 21:22:43 +00002314
Hanno Becker73d7d792018-12-11 10:35:51 +00002315 MPI_VALIDATE_RET( G != NULL );
2316 MPI_VALIDATE_RET( A != NULL );
2317 MPI_VALIDATE_RET( B != NULL );
2318
Alexander Ke8ad49f2019-08-16 16:16:07 +03002319 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002320
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002321 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) );
2322 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002323
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002324 lz = mbedtls_mpi_lsb( &TA );
2325 lzt = mbedtls_mpi_lsb( &TB );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002326
Paul Bakker66d5d072014-06-17 16:39:18 +02002327 if( lzt < lz )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002328 lz = lzt;
2329
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002330 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, lz ) );
2331 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, lz ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002332
Paul Bakker5121ce52009-01-03 21:22:43 +00002333 TA.s = TB.s = 1;
2334
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002335 while( mbedtls_mpi_cmp_int( &TA, 0 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002336 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002337 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, mbedtls_mpi_lsb( &TA ) ) );
2338 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, mbedtls_mpi_lsb( &TB ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002339
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002340 if( mbedtls_mpi_cmp_mpi( &TA, &TB ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002341 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002342 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TA, &TA, &TB ) );
2343 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002344 }
2345 else
2346 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002347 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TB, &TB, &TA ) );
2348 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002349 }
2350 }
2351
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002352 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &TB, lz ) );
2353 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( G, &TB ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002354
2355cleanup:
2356
Alexander Ke8ad49f2019-08-16 16:16:07 +03002357 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002358
2359 return( ret );
2360}
2361
Paul Bakker33dc46b2014-04-30 16:11:39 +02002362/*
2363 * Fill X with size bytes of random.
2364 *
2365 * Use a temporary bytes representation to make sure the result is the same
Paul Bakkerc37b0ac2014-05-01 14:19:23 +02002366 * regardless of the platform endianness (useful when f_rng is actually
Paul Bakker33dc46b2014-04-30 16:11:39 +02002367 * deterministic, eg for tests).
2368 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002369int mbedtls_mpi_fill_random( mbedtls_mpi *X, size_t size,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002370 int (*f_rng)(void *, unsigned char *, size_t),
2371 void *p_rng )
Paul Bakker287781a2011-03-26 13:18:49 +00002372{
Janos Follath24eed8d2019-11-22 13:21:35 +00002373 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker6dab6202019-01-02 16:42:29 +00002374 size_t const limbs = CHARS_TO_LIMBS( size );
Hanno Beckerda1655a2017-10-18 14:21:44 +01002375 size_t const overhead = ( limbs * ciL ) - size;
2376 unsigned char *Xp;
2377
Hanno Becker8ce11a32018-12-19 16:18:52 +00002378 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002379 MPI_VALIDATE_RET( f_rng != NULL );
Paul Bakker33dc46b2014-04-30 16:11:39 +02002380
Hanno Beckerda1655a2017-10-18 14:21:44 +01002381 /* Ensure that target MPI has exactly the necessary number of limbs */
2382 if( X->n != limbs )
2383 {
2384 mbedtls_mpi_free( X );
2385 mbedtls_mpi_init( X );
2386 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
2387 }
2388 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker287781a2011-03-26 13:18:49 +00002389
Hanno Beckerda1655a2017-10-18 14:21:44 +01002390 Xp = (unsigned char*) X->p;
2391 f_rng( p_rng, Xp + overhead, size );
2392
Hanno Becker2be8a552018-10-25 12:40:09 +01002393 mpi_bigendian_to_host( X->p, limbs );
Paul Bakker287781a2011-03-26 13:18:49 +00002394
2395cleanup:
2396 return( ret );
2397}
2398
Paul Bakker5121ce52009-01-03 21:22:43 +00002399/*
2400 * Modular inverse: X = A^-1 mod N (HAC 14.61 / 14.64)
2401 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002402int mbedtls_mpi_inv_mod( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002403{
Janos Follath24eed8d2019-11-22 13:21:35 +00002404 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002405 mbedtls_mpi G, TA, TU, U1, U2, TB, TV, V1, V2;
Hanno Becker73d7d792018-12-11 10:35:51 +00002406 MPI_VALIDATE_RET( X != NULL );
2407 MPI_VALIDATE_RET( A != NULL );
2408 MPI_VALIDATE_RET( N != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00002409
Hanno Becker4bcb4912017-04-18 15:49:39 +01002410 if( mbedtls_mpi_cmp_int( N, 1 ) <= 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002411 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002412
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002413 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TU ); mbedtls_mpi_init( &U1 ); mbedtls_mpi_init( &U2 );
2414 mbedtls_mpi_init( &G ); mbedtls_mpi_init( &TB ); mbedtls_mpi_init( &TV );
2415 mbedtls_mpi_init( &V1 ); mbedtls_mpi_init( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002416
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002417 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &G, A, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002418
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002419 if( mbedtls_mpi_cmp_int( &G, 1 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002420 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002421 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002422 goto cleanup;
2423 }
2424
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002425 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &TA, A, N ) );
2426 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TU, &TA ) );
2427 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, N ) );
2428 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TV, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002429
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002430 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U1, 1 ) );
2431 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U2, 0 ) );
2432 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V1, 0 ) );
2433 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002434
2435 do
2436 {
2437 while( ( TU.p[0] & 1 ) == 0 )
2438 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002439 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TU, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002440
2441 if( ( U1.p[0] & 1 ) != 0 || ( U2.p[0] & 1 ) != 0 )
2442 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002443 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &U1, &U1, &TB ) );
2444 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002445 }
2446
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002447 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U1, 1 ) );
2448 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002449 }
2450
2451 while( ( TV.p[0] & 1 ) == 0 )
2452 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002453 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TV, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002454
2455 if( ( V1.p[0] & 1 ) != 0 || ( V2.p[0] & 1 ) != 0 )
2456 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002457 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, &TB ) );
2458 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002459 }
2460
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002461 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V1, 1 ) );
2462 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002463 }
2464
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002465 if( mbedtls_mpi_cmp_mpi( &TU, &TV ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002466 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002467 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TU, &TU, &TV ) );
2468 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U1, &U1, &V1 ) );
2469 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &V2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002470 }
2471 else
2472 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002473 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TV, &TV, &TU ) );
2474 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, &U1 ) );
2475 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &U2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002476 }
2477 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002478 while( mbedtls_mpi_cmp_int( &TU, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002479
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002480 while( mbedtls_mpi_cmp_int( &V1, 0 ) < 0 )
2481 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002482
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002483 while( mbedtls_mpi_cmp_mpi( &V1, N ) >= 0 )
2484 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002485
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002486 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &V1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002487
2488cleanup:
2489
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002490 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TU ); mbedtls_mpi_free( &U1 ); mbedtls_mpi_free( &U2 );
2491 mbedtls_mpi_free( &G ); mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TV );
2492 mbedtls_mpi_free( &V1 ); mbedtls_mpi_free( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002493
2494 return( ret );
2495}
2496
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002497#if defined(MBEDTLS_GENPRIME)
Paul Bakkerd9374b02012-11-02 11:02:58 +00002498
Paul Bakker5121ce52009-01-03 21:22:43 +00002499static const int small_prime[] =
2500{
2501 3, 5, 7, 11, 13, 17, 19, 23,
2502 29, 31, 37, 41, 43, 47, 53, 59,
2503 61, 67, 71, 73, 79, 83, 89, 97,
2504 101, 103, 107, 109, 113, 127, 131, 137,
2505 139, 149, 151, 157, 163, 167, 173, 179,
2506 181, 191, 193, 197, 199, 211, 223, 227,
2507 229, 233, 239, 241, 251, 257, 263, 269,
2508 271, 277, 281, 283, 293, 307, 311, 313,
2509 317, 331, 337, 347, 349, 353, 359, 367,
2510 373, 379, 383, 389, 397, 401, 409, 419,
2511 421, 431, 433, 439, 443, 449, 457, 461,
2512 463, 467, 479, 487, 491, 499, 503, 509,
2513 521, 523, 541, 547, 557, 563, 569, 571,
2514 577, 587, 593, 599, 601, 607, 613, 617,
2515 619, 631, 641, 643, 647, 653, 659, 661,
2516 673, 677, 683, 691, 701, 709, 719, 727,
2517 733, 739, 743, 751, 757, 761, 769, 773,
2518 787, 797, 809, 811, 821, 823, 827, 829,
2519 839, 853, 857, 859, 863, 877, 881, 883,
2520 887, 907, 911, 919, 929, 937, 941, 947,
2521 953, 967, 971, 977, 983, 991, 997, -103
2522};
2523
2524/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002525 * Small divisors test (X must be positive)
2526 *
2527 * Return values:
2528 * 0: no small factor (possible prime, more tests needed)
2529 * 1: certain prime
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002530 * MBEDTLS_ERR_MPI_NOT_ACCEPTABLE: certain non-prime
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002531 * other negative: error
Paul Bakker5121ce52009-01-03 21:22:43 +00002532 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002533static int mpi_check_small_factors( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +00002534{
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002535 int ret = 0;
2536 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002537 mbedtls_mpi_uint r;
Paul Bakker5121ce52009-01-03 21:22:43 +00002538
Paul Bakker5121ce52009-01-03 21:22:43 +00002539 if( ( X->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002540 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002541
2542 for( i = 0; small_prime[i] > 0; i++ )
2543 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002544 if( mbedtls_mpi_cmp_int( X, small_prime[i] ) <= 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002545 return( 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002546
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002547 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, small_prime[i] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002548
2549 if( r == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002550 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002551 }
2552
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002553cleanup:
2554 return( ret );
2555}
2556
2557/*
2558 * Miller-Rabin pseudo-primality test (HAC 4.24)
2559 */
Janos Follathda31fa12018-09-03 14:45:23 +01002560static int mpi_miller_rabin( const mbedtls_mpi *X, size_t rounds,
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002561 int (*f_rng)(void *, unsigned char *, size_t),
2562 void *p_rng )
2563{
Pascal Junodb99183d2015-03-11 16:49:45 +01002564 int ret, count;
Janos Follathda31fa12018-09-03 14:45:23 +01002565 size_t i, j, k, s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002566 mbedtls_mpi W, R, T, A, RR;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002567
Hanno Becker8ce11a32018-12-19 16:18:52 +00002568 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002569 MPI_VALIDATE_RET( f_rng != NULL );
2570
2571 mbedtls_mpi_init( &W ); mbedtls_mpi_init( &R );
2572 mbedtls_mpi_init( &T ); mbedtls_mpi_init( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002573 mbedtls_mpi_init( &RR );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002574
Paul Bakker5121ce52009-01-03 21:22:43 +00002575 /*
2576 * W = |X| - 1
2577 * R = W >> lsb( W )
2578 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002579 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( &W, X, 1 ) );
2580 s = mbedtls_mpi_lsb( &W );
2581 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &R, &W ) );
2582 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &R, s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002583
Janos Follathda31fa12018-09-03 14:45:23 +01002584 for( i = 0; i < rounds; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002585 {
2586 /*
2587 * pick a random A, 1 < A < |X| - 1
2588 */
Pascal Junodb99183d2015-03-11 16:49:45 +01002589 count = 0;
2590 do {
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002591 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( &A, X->n * ciL, f_rng, p_rng ) );
Pascal Junodb99183d2015-03-11 16:49:45 +01002592
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002593 j = mbedtls_mpi_bitlen( &A );
2594 k = mbedtls_mpi_bitlen( &W );
Pascal Junodb99183d2015-03-11 16:49:45 +01002595 if (j > k) {
Darryl Greene3f95ed2018-10-02 13:21:35 +01002596 A.p[A.n - 1] &= ( (mbedtls_mpi_uint) 1 << ( k - ( A.n - 1 ) * biL - 1 ) ) - 1;
Pascal Junodb99183d2015-03-11 16:49:45 +01002597 }
2598
2599 if (count++ > 30) {
Jens Wiklanderf08aa3e2019-01-17 13:30:57 +01002600 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
2601 goto cleanup;
Pascal Junodb99183d2015-03-11 16:49:45 +01002602 }
2603
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002604 } while ( mbedtls_mpi_cmp_mpi( &A, &W ) >= 0 ||
2605 mbedtls_mpi_cmp_int( &A, 1 ) <= 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002606
2607 /*
2608 * A = A^R mod |X|
2609 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002610 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &A, &A, &R, X, &RR ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002611
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002612 if( mbedtls_mpi_cmp_mpi( &A, &W ) == 0 ||
2613 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002614 continue;
2615
2616 j = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002617 while( j < s && mbedtls_mpi_cmp_mpi( &A, &W ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002618 {
2619 /*
2620 * A = A * A mod |X|
2621 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002622 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &T, &A, &A ) );
2623 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &A, &T, X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002624
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002625 if( mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002626 break;
2627
2628 j++;
2629 }
2630
2631 /*
2632 * not prime if A != |X| - 1 or A == 1
2633 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002634 if( mbedtls_mpi_cmp_mpi( &A, &W ) != 0 ||
2635 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002636 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002637 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002638 break;
2639 }
2640 }
2641
2642cleanup:
Hanno Becker73d7d792018-12-11 10:35:51 +00002643 mbedtls_mpi_free( &W ); mbedtls_mpi_free( &R );
2644 mbedtls_mpi_free( &T ); mbedtls_mpi_free( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002645 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002646
2647 return( ret );
2648}
2649
2650/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002651 * Pseudo-primality test: small factors, then Miller-Rabin
2652 */
Janos Follatha0b67c22018-09-18 14:48:23 +01002653int mbedtls_mpi_is_prime_ext( const mbedtls_mpi *X, int rounds,
2654 int (*f_rng)(void *, unsigned char *, size_t),
2655 void *p_rng )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002656{
Janos Follath24eed8d2019-11-22 13:21:35 +00002657 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002658 mbedtls_mpi XX;
Hanno Becker8ce11a32018-12-19 16:18:52 +00002659 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002660 MPI_VALIDATE_RET( f_rng != NULL );
Manuel Pégourié-Gonnard7f4ed672014-10-14 20:56:02 +02002661
2662 XX.s = 1;
2663 XX.n = X->n;
2664 XX.p = X->p;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002665
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002666 if( mbedtls_mpi_cmp_int( &XX, 0 ) == 0 ||
2667 mbedtls_mpi_cmp_int( &XX, 1 ) == 0 )
2668 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002669
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002670 if( mbedtls_mpi_cmp_int( &XX, 2 ) == 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002671 return( 0 );
2672
2673 if( ( ret = mpi_check_small_factors( &XX ) ) != 0 )
2674 {
2675 if( ret == 1 )
2676 return( 0 );
2677
2678 return( ret );
2679 }
2680
Janos Follathda31fa12018-09-03 14:45:23 +01002681 return( mpi_miller_rabin( &XX, rounds, f_rng, p_rng ) );
Janos Follathf301d232018-08-14 13:34:01 +01002682}
2683
Janos Follatha0b67c22018-09-18 14:48:23 +01002684#if !defined(MBEDTLS_DEPRECATED_REMOVED)
Janos Follathf301d232018-08-14 13:34:01 +01002685/*
2686 * Pseudo-primality test, error probability 2^-80
2687 */
2688int mbedtls_mpi_is_prime( const mbedtls_mpi *X,
2689 int (*f_rng)(void *, unsigned char *, size_t),
2690 void *p_rng )
2691{
Hanno Becker8ce11a32018-12-19 16:18:52 +00002692 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002693 MPI_VALIDATE_RET( f_rng != NULL );
2694
Janos Follatha0b67c22018-09-18 14:48:23 +01002695 /*
2696 * In the past our key generation aimed for an error rate of at most
2697 * 2^-80. Since this function is deprecated, aim for the same certainty
2698 * here as well.
2699 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002700 return( mbedtls_mpi_is_prime_ext( X, 40, f_rng, p_rng ) );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002701}
Janos Follatha0b67c22018-09-18 14:48:23 +01002702#endif
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002703
2704/*
Paul Bakker5121ce52009-01-03 21:22:43 +00002705 * Prime number generation
Jethro Beekman66689272018-02-14 19:24:10 -08002706 *
Janos Follathf301d232018-08-14 13:34:01 +01002707 * To generate an RSA key in a way recommended by FIPS 186-4, both primes must
2708 * be either 1024 bits or 1536 bits long, and flags must contain
2709 * MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR.
Paul Bakker5121ce52009-01-03 21:22:43 +00002710 */
Janos Follath7c025a92018-08-14 11:08:41 +01002711int mbedtls_mpi_gen_prime( mbedtls_mpi *X, size_t nbits, int flags,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002712 int (*f_rng)(void *, unsigned char *, size_t),
2713 void *p_rng )
Paul Bakker5121ce52009-01-03 21:22:43 +00002714{
Jethro Beekman66689272018-02-14 19:24:10 -08002715#ifdef MBEDTLS_HAVE_INT64
2716// ceil(2^63.5)
2717#define CEIL_MAXUINT_DIV_SQRT2 0xb504f333f9de6485ULL
2718#else
2719// ceil(2^31.5)
2720#define CEIL_MAXUINT_DIV_SQRT2 0xb504f334U
2721#endif
2722 int ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker23986e52011-04-24 08:57:21 +00002723 size_t k, n;
Janos Follathda31fa12018-09-03 14:45:23 +01002724 int rounds;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002725 mbedtls_mpi_uint r;
2726 mbedtls_mpi Y;
Paul Bakker5121ce52009-01-03 21:22:43 +00002727
Hanno Becker8ce11a32018-12-19 16:18:52 +00002728 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002729 MPI_VALIDATE_RET( f_rng != NULL );
2730
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002731 if( nbits < 3 || nbits > MBEDTLS_MPI_MAX_BITS )
2732 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002733
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002734 mbedtls_mpi_init( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002735
2736 n = BITS_TO_LIMBS( nbits );
2737
Janos Follathda31fa12018-09-03 14:45:23 +01002738 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR ) == 0 )
2739 {
2740 /*
2741 * 2^-80 error probability, number of rounds chosen per HAC, table 4.4
2742 */
2743 rounds = ( ( nbits >= 1300 ) ? 2 : ( nbits >= 850 ) ? 3 :
2744 ( nbits >= 650 ) ? 4 : ( nbits >= 350 ) ? 8 :
2745 ( nbits >= 250 ) ? 12 : ( nbits >= 150 ) ? 18 : 27 );
2746 }
2747 else
2748 {
2749 /*
2750 * 2^-100 error probability, number of rounds computed based on HAC,
2751 * fact 4.48
2752 */
2753 rounds = ( ( nbits >= 1450 ) ? 4 : ( nbits >= 1150 ) ? 5 :
2754 ( nbits >= 1000 ) ? 6 : ( nbits >= 850 ) ? 7 :
2755 ( nbits >= 750 ) ? 8 : ( nbits >= 500 ) ? 13 :
2756 ( nbits >= 250 ) ? 28 : ( nbits >= 150 ) ? 40 : 51 );
2757 }
2758
Jethro Beekman66689272018-02-14 19:24:10 -08002759 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002760 {
Jethro Beekman66689272018-02-14 19:24:10 -08002761 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( X, n * ciL, f_rng, p_rng ) );
2762 /* make sure generated number is at least (nbits-1)+0.5 bits (FIPS 186-4 §B.3.3 steps 4.4, 5.5) */
2763 if( X->p[n-1] < CEIL_MAXUINT_DIV_SQRT2 ) continue;
2764
2765 k = n * biL;
2766 if( k > nbits ) MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( X, k - nbits ) );
2767 X->p[0] |= 1;
2768
Janos Follath7c025a92018-08-14 11:08:41 +01002769 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_DH ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002770 {
Janos Follatha0b67c22018-09-18 14:48:23 +01002771 ret = mbedtls_mpi_is_prime_ext( X, rounds, f_rng, p_rng );
Jethro Beekman66689272018-02-14 19:24:10 -08002772
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002773 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
Paul Bakker5121ce52009-01-03 21:22:43 +00002774 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002775 }
Jethro Beekman66689272018-02-14 19:24:10 -08002776 else
Paul Bakker5121ce52009-01-03 21:22:43 +00002777 {
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002778 /*
Jethro Beekman66689272018-02-14 19:24:10 -08002779 * An necessary condition for Y and X = 2Y + 1 to be prime
2780 * is X = 2 mod 3 (which is equivalent to Y = 2 mod 3).
2781 * Make sure it is satisfied, while keeping X = 3 mod 4
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002782 */
Jethro Beekman66689272018-02-14 19:24:10 -08002783
2784 X->p[0] |= 2;
2785
2786 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, 3 ) );
2787 if( r == 0 )
2788 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 8 ) );
2789 else if( r == 1 )
2790 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 4 ) );
2791
2792 /* Set Y = (X-1) / 2, which is X / 2 because X is odd */
2793 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, X ) );
2794 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, 1 ) );
2795
2796 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002797 {
Jethro Beekman66689272018-02-14 19:24:10 -08002798 /*
2799 * First, check small factors for X and Y
2800 * before doing Miller-Rabin on any of them
2801 */
2802 if( ( ret = mpi_check_small_factors( X ) ) == 0 &&
2803 ( ret = mpi_check_small_factors( &Y ) ) == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002804 ( ret = mpi_miller_rabin( X, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002805 == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002806 ( ret = mpi_miller_rabin( &Y, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002807 == 0 )
Jethro Beekman66689272018-02-14 19:24:10 -08002808 goto cleanup;
2809
2810 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
2811 goto cleanup;
2812
2813 /*
2814 * Next candidates. We want to preserve Y = (X-1) / 2 and
2815 * Y = 1 mod 2 and Y = 2 mod 3 (eq X = 3 mod 4 and X = 2 mod 3)
2816 * so up Y by 6 and X by 12.
2817 */
2818 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 12 ) );
2819 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( &Y, &Y, 6 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002820 }
Paul Bakker5121ce52009-01-03 21:22:43 +00002821 }
2822 }
2823
2824cleanup:
2825
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002826 mbedtls_mpi_free( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002827
2828 return( ret );
2829}
2830
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002831#endif /* MBEDTLS_GENPRIME */
Paul Bakker5121ce52009-01-03 21:22:43 +00002832
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002833#if defined(MBEDTLS_SELF_TEST)
Paul Bakker5121ce52009-01-03 21:22:43 +00002834
Paul Bakker23986e52011-04-24 08:57:21 +00002835#define GCD_PAIR_COUNT 3
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002836
2837static const int gcd_pairs[GCD_PAIR_COUNT][3] =
2838{
2839 { 693, 609, 21 },
2840 { 1764, 868, 28 },
2841 { 768454923, 542167814, 1 }
2842};
2843
Paul Bakker5121ce52009-01-03 21:22:43 +00002844/*
2845 * Checkup routine
2846 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002847int mbedtls_mpi_self_test( int verbose )
Paul Bakker5121ce52009-01-03 21:22:43 +00002848{
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002849 int ret, i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002850 mbedtls_mpi A, E, N, X, Y, U, V;
Paul Bakker5121ce52009-01-03 21:22:43 +00002851
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002852 mbedtls_mpi_init( &A ); mbedtls_mpi_init( &E ); mbedtls_mpi_init( &N ); mbedtls_mpi_init( &X );
2853 mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &U ); mbedtls_mpi_init( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002854
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002855 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &A, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002856 "EFE021C2645FD1DC586E69184AF4A31E" \
2857 "D5F53E93B5F123FA41680867BA110131" \
2858 "944FE7952E2517337780CB0DB80E61AA" \
2859 "E7C8DDC6C5C6AADEB34EB38A2F40D5E6" ) );
2860
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002861 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &E, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002862 "B2E7EFD37075B9F03FF989C7C5051C20" \
2863 "34D2A323810251127E7BF8625A4F49A5" \
2864 "F3E27F4DA8BD59C47D6DAABA4C8127BD" \
2865 "5B5C25763222FEFCCFC38B832366C29E" ) );
2866
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002867 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &N, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002868 "0066A198186C18C10B2F5ED9B522752A" \
2869 "9830B69916E535C8F047518A889A43A5" \
2870 "94B6BED27A168D31D4A52F88925AA8F5" ) );
2871
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002872 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002873
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002874 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002875 "602AB7ECA597A3D6B56FF9829A5E8B85" \
2876 "9E857EA95A03512E2BAE7391688D264A" \
2877 "A5663B0341DB9CCFD2C4C5F421FEC814" \
2878 "8001B72E848A38CAE1C65F78E56ABDEF" \
2879 "E12D3C039B8A02D6BE593F0BBBDA56F1" \
2880 "ECF677152EF804370C1A305CAF3B5BF1" \
2881 "30879B56C61DE584A0F53A2447A51E" ) );
2882
2883 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002884 mbedtls_printf( " MPI test #1 (mul_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002885
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002886 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002887 {
2888 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002889 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002890
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002891 ret = 1;
2892 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002893 }
2894
2895 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002896 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002897
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002898 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( &X, &Y, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002899
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002900 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002901 "256567336059E52CAE22925474705F39A94" ) );
2902
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002903 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &V, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002904 "6613F26162223DF488E9CD48CC132C7A" \
2905 "0AC93C701B001B092E4E5B9F73BCD27B" \
2906 "9EE50D0657C77F374E903CDFA4C642" ) );
2907
2908 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002909 mbedtls_printf( " MPI test #2 (div_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002910
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002911 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 ||
2912 mbedtls_mpi_cmp_mpi( &Y, &V ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002913 {
2914 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002915 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002916
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002917 ret = 1;
2918 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002919 }
2920
2921 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002922 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002923
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002924 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &X, &A, &E, &N, NULL ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002925
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002926 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002927 "36E139AEA55215609D2816998ED020BB" \
2928 "BD96C37890F65171D948E9BC7CBAA4D9" \
2929 "325D24D6A3C12710F10A09FA08AB87" ) );
2930
2931 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002932 mbedtls_printf( " MPI test #3 (exp_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002933
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002934 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002935 {
2936 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002937 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002938
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002939 ret = 1;
2940 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002941 }
2942
2943 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002944 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002945
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002946 MBEDTLS_MPI_CHK( mbedtls_mpi_inv_mod( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002947
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002948 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002949 "003A0AAEDD7E784FC07D8F9EC6E3BFD5" \
2950 "C3DBA76456363A10869622EAC2DD84EC" \
2951 "C5B8A74DAC4D09E03B5E0BE779F2DF61" ) );
2952
2953 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002954 mbedtls_printf( " MPI test #4 (inv_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002955
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002956 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002957 {
2958 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002959 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002960
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002961 ret = 1;
2962 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002963 }
2964
2965 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002966 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002967
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002968 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002969 mbedtls_printf( " MPI test #5 (simple gcd): " );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002970
Paul Bakker66d5d072014-06-17 16:39:18 +02002971 for( i = 0; i < GCD_PAIR_COUNT; i++ )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002972 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002973 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &X, gcd_pairs[i][0] ) );
2974 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Y, gcd_pairs[i][1] ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002975
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002976 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &A, &X, &Y ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002977
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002978 if( mbedtls_mpi_cmp_int( &A, gcd_pairs[i][2] ) != 0 )
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002979 {
2980 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002981 mbedtls_printf( "failed at %d\n", i );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002982
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002983 ret = 1;
2984 goto cleanup;
2985 }
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002986 }
2987
2988 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002989 mbedtls_printf( "passed\n" );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002990
Paul Bakker5121ce52009-01-03 21:22:43 +00002991cleanup:
2992
2993 if( ret != 0 && verbose != 0 )
Kenneth Soerensen518d4352020-04-01 17:22:45 +02002994 mbedtls_printf( "Unexpected error, return code = %08X\n", (unsigned int) ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00002995
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002996 mbedtls_mpi_free( &A ); mbedtls_mpi_free( &E ); mbedtls_mpi_free( &N ); mbedtls_mpi_free( &X );
2997 mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &U ); mbedtls_mpi_free( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002998
2999 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003000 mbedtls_printf( "\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00003001
3002 return( ret );
3003}
3004
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003005#endif /* MBEDTLS_SELF_TEST */
Paul Bakker5121ce52009-01-03 21:22:43 +00003006
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003007#endif /* MBEDTLS_BIGNUM_C */