blob: 0eb212560a6c5c7e1a7077e674dfd6662760d497 [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
Gilles Peskinea5d8d892020-07-23 21:27:15 +02001528/** Helper for mbedtls_mpi multiplication.
1529 *
1530 * Add \p b * \p s to \p d.
1531 *
1532 * \param i The number of limbs of \p s.
1533 * \param[in] s A bignum to multiply, of size \p i.
1534 * It may overlap with \p d, but only if
1535 * \p d <= \p s.
1536 * Its leading limb must not be \c 0.
1537 * \param[in,out] d The bignum to add to.
1538 * It must be sufficiently large to store the
1539 * result of the multiplication. This means
1540 * \p i + 1 limbs if \p d[\p i - 1] started as 0 and \p b
1541 * is not known a priori.
1542 * \param b A scalar to multiply.
Paul Bakkerfc4f46f2013-06-24 19:23:56 +02001543 */
1544static
1545#if defined(__APPLE__) && defined(__arm__)
1546/*
1547 * Apple LLVM version 4.2 (clang-425.0.24) (based on LLVM 3.2svn)
1548 * appears to need this to prevent bad ARM code generation at -O3.
1549 */
1550__attribute__ ((noinline))
1551#endif
Gilles Peskinea5d8d892020-07-23 21:27:15 +02001552void mpi_mul_hlp( size_t i,
1553 const mbedtls_mpi_uint *s,
1554 mbedtls_mpi_uint *d,
1555 mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001556{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001557 mbedtls_mpi_uint c = 0, t = 0;
Paul Bakker5121ce52009-01-03 21:22:43 +00001558
1559#if defined(MULADDC_HUIT)
1560 for( ; i >= 8; i -= 8 )
1561 {
1562 MULADDC_INIT
1563 MULADDC_HUIT
1564 MULADDC_STOP
1565 }
1566
1567 for( ; i > 0; i-- )
1568 {
1569 MULADDC_INIT
1570 MULADDC_CORE
1571 MULADDC_STOP
1572 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001573#else /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001574 for( ; i >= 16; i -= 16 )
1575 {
1576 MULADDC_INIT
1577 MULADDC_CORE MULADDC_CORE
1578 MULADDC_CORE MULADDC_CORE
1579 MULADDC_CORE MULADDC_CORE
1580 MULADDC_CORE MULADDC_CORE
1581
1582 MULADDC_CORE MULADDC_CORE
1583 MULADDC_CORE MULADDC_CORE
1584 MULADDC_CORE MULADDC_CORE
1585 MULADDC_CORE MULADDC_CORE
1586 MULADDC_STOP
1587 }
1588
1589 for( ; i >= 8; i -= 8 )
1590 {
1591 MULADDC_INIT
1592 MULADDC_CORE MULADDC_CORE
1593 MULADDC_CORE MULADDC_CORE
1594
1595 MULADDC_CORE MULADDC_CORE
1596 MULADDC_CORE MULADDC_CORE
1597 MULADDC_STOP
1598 }
1599
1600 for( ; i > 0; i-- )
1601 {
1602 MULADDC_INIT
1603 MULADDC_CORE
1604 MULADDC_STOP
1605 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001606#endif /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001607
1608 t++;
1609
1610 do {
1611 *d += c; c = ( *d < c ); d++;
1612 }
1613 while( c != 0 );
1614}
1615
1616/*
1617 * Baseline multiplication: X = A * B (HAC 14.12)
1618 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001619int mbedtls_mpi_mul_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001620{
Janos Follath24eed8d2019-11-22 13:21:35 +00001621 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001622 size_t i, j;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001623 mbedtls_mpi TA, TB;
Hanno Becker73d7d792018-12-11 10:35:51 +00001624 MPI_VALIDATE_RET( X != NULL );
1625 MPI_VALIDATE_RET( A != NULL );
1626 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001627
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001628 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001629
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001630 if( X == A ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) ); A = &TA; }
1631 if( X == B ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) ); B = &TB; }
Paul Bakker5121ce52009-01-03 21:22:43 +00001632
Paul Bakker23986e52011-04-24 08:57:21 +00001633 for( i = A->n; i > 0; i-- )
1634 if( A->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001635 break;
1636
Paul Bakker23986e52011-04-24 08:57:21 +00001637 for( j = B->n; j > 0; j-- )
1638 if( B->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001639 break;
1640
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001641 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + j ) );
1642 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001643
Alexey Skalozub8e75e682016-01-13 21:59:27 +02001644 for( ; j > 0; j-- )
1645 mpi_mul_hlp( i, A->p, X->p + j - 1, B->p[j - 1] );
Paul Bakker5121ce52009-01-03 21:22:43 +00001646
1647 X->s = A->s * B->s;
1648
1649cleanup:
1650
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001651 mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TA );
Paul Bakker5121ce52009-01-03 21:22:43 +00001652
1653 return( ret );
1654}
1655
1656/*
1657 * Baseline multiplication: X = A * b
1658 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001659int mbedtls_mpi_mul_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001660{
Hanno Becker73d7d792018-12-11 10:35:51 +00001661 MPI_VALIDATE_RET( X != NULL );
1662 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001663
Gilles Peskine8fd95c62020-07-23 21:58:50 +02001664 /* mpi_mul_hlp can't deal with a leading 0. */
1665 size_t n = A->n;
1666 while( n > 0 && A->p[n - 1] == 0 )
1667 --n;
Paul Bakker5121ce52009-01-03 21:22:43 +00001668
Gilles Peskine8fd95c62020-07-23 21:58:50 +02001669 /* The general method below doesn't work if n==0 or b==0. By chance
1670 * calculating the result is trivial in those cases. */
1671 if( b == 0 || n == 0 )
1672 {
1673 mbedtls_mpi_lset( X, 0 );
1674 return( 0 );
1675 }
1676
1677 /* Calculate X*b as A + A*(b-1) to take advantage of mpi_mul_hlp */
1678 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
1679 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, n + 1 ) );
1680 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, A ) );
1681 mpi_mul_hlp( n, A->p, X->p, b - 1 );
1682
1683cleanup:
1684 return( ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00001685}
1686
1687/*
Simon Butcherf5ba0452015-12-27 23:01:55 +00001688 * Unsigned integer divide - double mbedtls_mpi_uint dividend, u1/u0, and
1689 * mbedtls_mpi_uint divisor, d
Simon Butcher15b15d12015-11-26 19:35:03 +00001690 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001691static mbedtls_mpi_uint mbedtls_int_div_int( mbedtls_mpi_uint u1,
1692 mbedtls_mpi_uint u0, mbedtls_mpi_uint d, mbedtls_mpi_uint *r )
Simon Butcher15b15d12015-11-26 19:35:03 +00001693{
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001694#if defined(MBEDTLS_HAVE_UDBL)
1695 mbedtls_t_udbl dividend, quotient;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001696#else
Simon Butcher9803d072016-01-03 00:24:34 +00001697 const mbedtls_mpi_uint radix = (mbedtls_mpi_uint) 1 << biH;
1698 const mbedtls_mpi_uint uint_halfword_mask = ( (mbedtls_mpi_uint) 1 << biH ) - 1;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001699 mbedtls_mpi_uint d0, d1, q0, q1, rAX, r0, quotient;
1700 mbedtls_mpi_uint u0_msw, u0_lsw;
Simon Butcher9803d072016-01-03 00:24:34 +00001701 size_t s;
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001702#endif
1703
Simon Butcher15b15d12015-11-26 19:35:03 +00001704 /*
1705 * Check for overflow
1706 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001707 if( 0 == d || u1 >= d )
Simon Butcher15b15d12015-11-26 19:35:03 +00001708 {
Simon Butcherf5ba0452015-12-27 23:01:55 +00001709 if (r != NULL) *r = ~0;
Simon Butcher15b15d12015-11-26 19:35:03 +00001710
Simon Butcherf5ba0452015-12-27 23:01:55 +00001711 return ( ~0 );
Simon Butcher15b15d12015-11-26 19:35:03 +00001712 }
1713
1714#if defined(MBEDTLS_HAVE_UDBL)
Simon Butcher15b15d12015-11-26 19:35:03 +00001715 dividend = (mbedtls_t_udbl) u1 << biL;
1716 dividend |= (mbedtls_t_udbl) u0;
1717 quotient = dividend / d;
1718 if( quotient > ( (mbedtls_t_udbl) 1 << biL ) - 1 )
1719 quotient = ( (mbedtls_t_udbl) 1 << biL ) - 1;
1720
1721 if( r != NULL )
Simon Butcher9803d072016-01-03 00:24:34 +00001722 *r = (mbedtls_mpi_uint)( dividend - (quotient * d ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001723
1724 return (mbedtls_mpi_uint) quotient;
1725#else
Simon Butcher15b15d12015-11-26 19:35:03 +00001726
1727 /*
1728 * Algorithm D, Section 4.3.1 - The Art of Computer Programming
1729 * Vol. 2 - Seminumerical Algorithms, Knuth
1730 */
1731
1732 /*
1733 * Normalize the divisor, d, and dividend, u0, u1
1734 */
1735 s = mbedtls_clz( d );
1736 d = d << s;
1737
1738 u1 = u1 << s;
Simon Butcher9803d072016-01-03 00:24:34 +00001739 u1 |= ( u0 >> ( biL - s ) ) & ( -(mbedtls_mpi_sint)s >> ( biL - 1 ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001740 u0 = u0 << s;
1741
1742 d1 = d >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001743 d0 = d & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001744
1745 u0_msw = u0 >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001746 u0_lsw = u0 & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001747
1748 /*
1749 * Find the first quotient and remainder
1750 */
1751 q1 = u1 / d1;
1752 r0 = u1 - d1 * q1;
1753
1754 while( q1 >= radix || ( q1 * d0 > radix * r0 + u0_msw ) )
1755 {
1756 q1 -= 1;
1757 r0 += d1;
1758
1759 if ( r0 >= radix ) break;
1760 }
1761
Simon Butcherf5ba0452015-12-27 23:01:55 +00001762 rAX = ( u1 * radix ) + ( u0_msw - q1 * d );
Simon Butcher15b15d12015-11-26 19:35:03 +00001763 q0 = rAX / d1;
1764 r0 = rAX - q0 * d1;
1765
1766 while( q0 >= radix || ( q0 * d0 > radix * r0 + u0_lsw ) )
1767 {
1768 q0 -= 1;
1769 r0 += d1;
1770
1771 if ( r0 >= radix ) break;
1772 }
1773
1774 if (r != NULL)
Simon Butcherf5ba0452015-12-27 23:01:55 +00001775 *r = ( rAX * radix + u0_lsw - q0 * d ) >> s;
Simon Butcher15b15d12015-11-26 19:35:03 +00001776
1777 quotient = q1 * radix + q0;
1778
1779 return quotient;
1780#endif
1781}
1782
1783/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001784 * Division by mbedtls_mpi: A = Q * B + R (HAC 14.20)
Paul Bakker5121ce52009-01-03 21:22:43 +00001785 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001786int mbedtls_mpi_div_mpi( mbedtls_mpi *Q, mbedtls_mpi *R, const mbedtls_mpi *A,
1787 const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001788{
Janos Follath24eed8d2019-11-22 13:21:35 +00001789 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001790 size_t i, n, t, k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001791 mbedtls_mpi X, Y, Z, T1, T2;
Alexander Kd19a1932019-11-01 18:20:42 +03001792 mbedtls_mpi_uint TP2[3];
Hanno Becker73d7d792018-12-11 10:35:51 +00001793 MPI_VALIDATE_RET( A != NULL );
1794 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001795
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001796 if( mbedtls_mpi_cmp_int( B, 0 ) == 0 )
1797 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001798
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001799 mbedtls_mpi_init( &X ); mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001800 mbedtls_mpi_init( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001801 /*
1802 * Avoid dynamic memory allocations for constant-size T2.
1803 *
1804 * T2 is used for comparison only and the 3 limbs are assigned explicitly,
1805 * so nobody increase the size of the MPI and we're safe to use an on-stack
1806 * buffer.
1807 */
Alexander K35d6d462019-10-31 14:46:45 +03001808 T2.s = 1;
Alexander Kd19a1932019-11-01 18:20:42 +03001809 T2.n = sizeof( TP2 ) / sizeof( *TP2 );
1810 T2.p = TP2;
Paul Bakker5121ce52009-01-03 21:22:43 +00001811
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001812 if( mbedtls_mpi_cmp_abs( A, B ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001813 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001814 if( Q != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_lset( Q, 0 ) );
1815 if( R != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001816 return( 0 );
1817 }
1818
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001819 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &X, A ) );
1820 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001821 X.s = Y.s = 1;
1822
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001823 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &Z, A->n + 2 ) );
1824 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Z, 0 ) );
1825 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T1, 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001826
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001827 k = mbedtls_mpi_bitlen( &Y ) % biL;
Paul Bakkerf9688572011-05-05 10:00:45 +00001828 if( k < biL - 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001829 {
1830 k = biL - 1 - k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001831 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &X, k ) );
1832 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, k ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001833 }
1834 else k = 0;
1835
1836 n = X.n - 1;
1837 t = Y.n - 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001838 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001839
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001840 while( mbedtls_mpi_cmp_mpi( &X, &Y ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001841 {
1842 Z.p[n - t]++;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001843 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001844 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001845 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001846
1847 for( i = n; i > t ; i-- )
1848 {
1849 if( X.p[i] >= Y.p[t] )
1850 Z.p[i - t - 1] = ~0;
1851 else
1852 {
Simon Butcher15b15d12015-11-26 19:35:03 +00001853 Z.p[i - t - 1] = mbedtls_int_div_int( X.p[i], X.p[i - 1],
1854 Y.p[t], NULL);
Paul Bakker5121ce52009-01-03 21:22:43 +00001855 }
1856
Alexander K35d6d462019-10-31 14:46:45 +03001857 T2.p[0] = ( i < 2 ) ? 0 : X.p[i - 2];
1858 T2.p[1] = ( i < 1 ) ? 0 : X.p[i - 1];
1859 T2.p[2] = X.p[i];
1860
Paul Bakker5121ce52009-01-03 21:22:43 +00001861 Z.p[i - t - 1]++;
1862 do
1863 {
1864 Z.p[i - t - 1]--;
1865
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001866 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &T1, 0 ) );
Paul Bakker66d5d072014-06-17 16:39:18 +02001867 T1.p[0] = ( t < 1 ) ? 0 : Y.p[t - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001868 T1.p[1] = Y.p[t];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001869 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &T1, Z.p[i - t - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001870 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001871 while( mbedtls_mpi_cmp_mpi( &T1, &T2 ) > 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001872
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001873 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &Y, Z.p[i - t - 1] ) );
1874 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1875 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001876
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001877 if( mbedtls_mpi_cmp_int( &X, 0 ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001878 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001879 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T1, &Y ) );
1880 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1881 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001882 Z.p[i - t - 1]--;
1883 }
1884 }
1885
1886 if( Q != NULL )
1887 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001888 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( Q, &Z ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001889 Q->s = A->s * B->s;
1890 }
1891
1892 if( R != NULL )
1893 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001894 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &X, k ) );
Paul Bakkerf02c5642012-11-13 10:25:21 +00001895 X.s = A->s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001896 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, &X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001897
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001898 if( mbedtls_mpi_cmp_int( R, 0 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001899 R->s = 1;
1900 }
1901
1902cleanup:
1903
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001904 mbedtls_mpi_free( &X ); mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001905 mbedtls_mpi_free( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001906 mbedtls_platform_zeroize( TP2, sizeof( TP2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001907
1908 return( ret );
1909}
1910
1911/*
1912 * Division by int: A = Q * b + R
Paul Bakker5121ce52009-01-03 21:22:43 +00001913 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001914int mbedtls_mpi_div_int( mbedtls_mpi *Q, mbedtls_mpi *R,
1915 const mbedtls_mpi *A,
1916 mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001917{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001918 mbedtls_mpi _B;
1919 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001920 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001921
1922 p[0] = ( b < 0 ) ? -b : b;
1923 _B.s = ( b < 0 ) ? -1 : 1;
1924 _B.n = 1;
1925 _B.p = p;
1926
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001927 return( mbedtls_mpi_div_mpi( Q, R, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001928}
1929
1930/*
1931 * Modulo: R = A mod B
1932 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001933int mbedtls_mpi_mod_mpi( mbedtls_mpi *R, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001934{
Janos Follath24eed8d2019-11-22 13:21:35 +00001935 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker73d7d792018-12-11 10:35:51 +00001936 MPI_VALIDATE_RET( R != NULL );
1937 MPI_VALIDATE_RET( A != NULL );
1938 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001939
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001940 if( mbedtls_mpi_cmp_int( B, 0 ) < 0 )
1941 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001942
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001943 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( NULL, R, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001944
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001945 while( mbedtls_mpi_cmp_int( R, 0 ) < 0 )
1946 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001947
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001948 while( mbedtls_mpi_cmp_mpi( R, B ) >= 0 )
1949 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001950
1951cleanup:
1952
1953 return( ret );
1954}
1955
1956/*
1957 * Modulo: r = A mod b
1958 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001959int mbedtls_mpi_mod_int( mbedtls_mpi_uint *r, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001960{
Paul Bakker23986e52011-04-24 08:57:21 +00001961 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001962 mbedtls_mpi_uint x, y, z;
Hanno Becker73d7d792018-12-11 10:35:51 +00001963 MPI_VALIDATE_RET( r != NULL );
1964 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001965
1966 if( b == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001967 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001968
1969 if( b < 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001970 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakker5121ce52009-01-03 21:22:43 +00001971
1972 /*
1973 * handle trivial cases
1974 */
1975 if( b == 1 )
1976 {
1977 *r = 0;
1978 return( 0 );
1979 }
1980
1981 if( b == 2 )
1982 {
1983 *r = A->p[0] & 1;
1984 return( 0 );
1985 }
1986
1987 /*
1988 * general case
1989 */
Paul Bakker23986e52011-04-24 08:57:21 +00001990 for( i = A->n, y = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001991 {
Paul Bakker23986e52011-04-24 08:57:21 +00001992 x = A->p[i - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001993 y = ( y << biH ) | ( x >> biH );
1994 z = y / b;
1995 y -= z * b;
1996
1997 x <<= biH;
1998 y = ( y << biH ) | ( x >> biH );
1999 z = y / b;
2000 y -= z * b;
2001 }
2002
Paul Bakkerce40a6d2009-06-23 19:46:08 +00002003 /*
2004 * If A is negative, then the current y represents a negative value.
2005 * Flipping it to the positive side.
2006 */
2007 if( A->s < 0 && y != 0 )
2008 y = b - y;
2009
Paul Bakker5121ce52009-01-03 21:22:43 +00002010 *r = y;
2011
2012 return( 0 );
2013}
2014
2015/*
2016 * Fast Montgomery initialization (thanks to Tom St Denis)
2017 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002018static void mpi_montg_init( mbedtls_mpi_uint *mm, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002019{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002020 mbedtls_mpi_uint x, m0 = N->p[0];
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01002021 unsigned int i;
Paul Bakker5121ce52009-01-03 21:22:43 +00002022
2023 x = m0;
2024 x += ( ( m0 + 2 ) & 4 ) << 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00002025
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01002026 for( i = biL; i >= 8; i /= 2 )
2027 x *= ( 2 - ( m0 * x ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002028
2029 *mm = ~x + 1;
2030}
2031
Gilles Peskine2a82f722020-06-04 15:00:49 +02002032/** Montgomery multiplication: A = A * B * R^-1 mod N (HAC 14.36)
2033 *
2034 * \param[in,out] A One of the numbers to multiply.
Gilles Peskine221626f2020-06-08 22:37:50 +02002035 * It must have at least as many limbs as N
2036 * (A->n >= N->n), and any limbs beyond n are ignored.
Gilles Peskine2a82f722020-06-04 15:00:49 +02002037 * On successful completion, A contains the result of
2038 * the multiplication A * B * R^-1 mod N where
2039 * R = (2^ciL)^n.
2040 * \param[in] B One of the numbers to multiply.
2041 * It must be nonzero and must not have more limbs than N
2042 * (B->n <= N->n).
2043 * \param[in] N The modulo. N must be odd.
2044 * \param mm The value calculated by `mpi_montg_init(&mm, N)`.
2045 * This is -N^-1 mod 2^ciL.
2046 * \param[in,out] T A bignum for temporary storage.
2047 * It must be at least twice the limb size of N plus 2
2048 * (T->n >= 2 * (N->n + 1)).
2049 * Its initial content is unused and
2050 * its final content is indeterminate.
2051 * Note that unlike the usual convention in the library
2052 * for `const mbedtls_mpi*`, the content of T can change.
Paul Bakker5121ce52009-01-03 21:22:43 +00002053 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002054static 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 +02002055 const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002056{
Paul Bakker23986e52011-04-24 08:57:21 +00002057 size_t i, n, m;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002058 mbedtls_mpi_uint u0, u1, *d;
Paul Bakker5121ce52009-01-03 21:22:43 +00002059
2060 memset( T->p, 0, T->n * ciL );
2061
2062 d = T->p;
2063 n = N->n;
2064 m = ( B->n < n ) ? B->n : n;
2065
2066 for( i = 0; i < n; i++ )
2067 {
2068 /*
2069 * T = (T + u0*B + u1*N) / 2^biL
2070 */
2071 u0 = A->p[i];
2072 u1 = ( d[0] + u0 * B->p[0] ) * mm;
2073
2074 mpi_mul_hlp( m, B->p, d, u0 );
2075 mpi_mul_hlp( n, N->p, d, u1 );
2076
2077 *d++ = u0; d[n + 1] = 0;
2078 }
2079
Gilles Peskine221626f2020-06-08 22:37:50 +02002080 /* At this point, d is either the desired result or the desired result
2081 * plus N. We now potentially subtract N, avoiding leaking whether the
2082 * subtraction is performed through side channels. */
Paul Bakker5121ce52009-01-03 21:22:43 +00002083
Gilles Peskine221626f2020-06-08 22:37:50 +02002084 /* Copy the n least significant limbs of d to A, so that
2085 * A = d if d < N (recall that N has n limbs). */
2086 memcpy( A->p, d, n * ciL );
Gilles Peskine09ec10a2020-06-09 10:39:38 +02002087 /* If d >= N then we want to set A to d - N. To prevent timing attacks,
Gilles Peskine221626f2020-06-08 22:37:50 +02002088 * do the calculation without using conditional tests. */
2089 /* Set d to d0 + (2^biL)^n - N where d0 is the current value of d. */
Gilles Peskine132c0972020-06-04 21:05:24 +02002090 d[n] += 1;
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02002091 d[n] -= mpi_sub_hlp( n, d, d, N->p );
Gilles Peskine221626f2020-06-08 22:37:50 +02002092 /* If d0 < N then d < (2^biL)^n
2093 * so d[n] == 0 and we want to keep A as it is.
2094 * If d0 >= N then d >= (2^biL)^n, and d <= (2^biL)^n + N < 2 * (2^biL)^n
2095 * so d[n] == 1 and we want to set A to the result of the subtraction
2096 * which is d - (2^biL)^n, i.e. the n least significant limbs of d.
2097 * This exactly corresponds to a conditional assignment. */
2098 mpi_safe_cond_assign( n, A->p, d, (unsigned char) d[n] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002099}
2100
2101/*
2102 * Montgomery reduction: A = A * R^-1 mod N
Gilles Peskine2a82f722020-06-04 15:00:49 +02002103 *
2104 * See mpi_montmul() regarding constraints and guarantees on the parameters.
Paul Bakker5121ce52009-01-03 21:22:43 +00002105 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002106static void mpi_montred( mbedtls_mpi *A, const mbedtls_mpi *N,
2107 mbedtls_mpi_uint mm, const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002108{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002109 mbedtls_mpi_uint z = 1;
2110 mbedtls_mpi U;
Paul Bakker5121ce52009-01-03 21:22:43 +00002111
Paul Bakker8ddb6452013-02-27 14:56:33 +01002112 U.n = U.s = (int) z;
Paul Bakker5121ce52009-01-03 21:22:43 +00002113 U.p = &z;
2114
Gilles Peskine4e91d472020-06-04 20:55:15 +02002115 mpi_montmul( A, &U, N, mm, T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002116}
2117
2118/*
2119 * Sliding-window exponentiation: X = A^E mod N (HAC 14.85)
2120 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002121int mbedtls_mpi_exp_mod( mbedtls_mpi *X, const mbedtls_mpi *A,
2122 const mbedtls_mpi *E, const mbedtls_mpi *N,
2123 mbedtls_mpi *_RR )
Paul Bakker5121ce52009-01-03 21:22:43 +00002124{
Janos Follath24eed8d2019-11-22 13:21:35 +00002125 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002126 size_t wbits, wsize, one = 1;
2127 size_t i, j, nblimbs;
2128 size_t bufsize, nbits;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002129 mbedtls_mpi_uint ei, mm, state;
Daniel Otte388f9b22020-08-21 12:34:29 +02002130 mbedtls_mpi RR, T, W[ 1 << MBEDTLS_MPI_WINDOW_SIZE ], Apos;
Paul Bakkerf6198c12012-05-16 08:02:29 +00002131 int neg;
Paul Bakker5121ce52009-01-03 21:22:43 +00002132
Hanno Becker73d7d792018-12-11 10:35:51 +00002133 MPI_VALIDATE_RET( X != NULL );
2134 MPI_VALIDATE_RET( A != NULL );
2135 MPI_VALIDATE_RET( E != NULL );
2136 MPI_VALIDATE_RET( N != NULL );
2137
Hanno Becker8d1dd1b2017-09-28 11:02:24 +01002138 if( mbedtls_mpi_cmp_int( N, 0 ) <= 0 || ( N->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002139 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002140
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002141 if( mbedtls_mpi_cmp_int( E, 0 ) < 0 )
2142 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002143
2144 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002145 * Init temps and window size
2146 */
2147 mpi_montg_init( &mm, N );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002148 mbedtls_mpi_init( &RR ); mbedtls_mpi_init( &T );
2149 mbedtls_mpi_init( &Apos );
Paul Bakker5121ce52009-01-03 21:22:43 +00002150 memset( W, 0, sizeof( W ) );
2151
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002152 i = mbedtls_mpi_bitlen( E );
Paul Bakker5121ce52009-01-03 21:22:43 +00002153
2154 wsize = ( i > 671 ) ? 6 : ( i > 239 ) ? 5 :
2155 ( i > 79 ) ? 4 : ( i > 23 ) ? 3 : 1;
2156
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002157#if( MBEDTLS_MPI_WINDOW_SIZE < 6 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002158 if( wsize > MBEDTLS_MPI_WINDOW_SIZE )
2159 wsize = MBEDTLS_MPI_WINDOW_SIZE;
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002160#endif
Paul Bakkerb6d5f082011-11-25 11:52:11 +00002161
Paul Bakker5121ce52009-01-03 21:22:43 +00002162 j = N->n + 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002163 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
2164 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[1], j ) );
2165 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T, j * 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002166
2167 /*
Paul Bakker50546922012-05-19 08:40:49 +00002168 * Compensate for negative A (and correct at the end)
2169 */
2170 neg = ( A->s == -1 );
Paul Bakker50546922012-05-19 08:40:49 +00002171 if( neg )
2172 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002173 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Apos, A ) );
Paul Bakker50546922012-05-19 08:40:49 +00002174 Apos.s = 1;
2175 A = &Apos;
2176 }
2177
2178 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002179 * If 1st call, pre-compute R^2 mod N
2180 */
2181 if( _RR == NULL || _RR->p == NULL )
2182 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002183 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &RR, 1 ) );
2184 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &RR, N->n * 2 * biL ) );
2185 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &RR, &RR, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002186
2187 if( _RR != NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002188 memcpy( _RR, &RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002189 }
2190 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002191 memcpy( &RR, _RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002192
2193 /*
2194 * W[1] = A * R^2 * R^-1 mod N = A * R mod N
2195 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002196 if( mbedtls_mpi_cmp_mpi( A, N ) >= 0 )
2197 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &W[1], A, N ) );
Paul Bakkerc2024f42014-01-23 20:38:35 +01002198 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002199 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[1], A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002200
Gilles Peskine4e91d472020-06-04 20:55:15 +02002201 mpi_montmul( &W[1], &RR, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002202
2203 /*
2204 * X = R^2 * R^-1 mod N = R mod N
2205 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002206 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &RR ) );
Gilles Peskine4e91d472020-06-04 20:55:15 +02002207 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002208
2209 if( wsize > 1 )
2210 {
2211 /*
2212 * W[1 << (wsize - 1)] = W[1] ^ (wsize - 1)
2213 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002214 j = one << ( wsize - 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002215
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002216 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[j], N->n + 1 ) );
2217 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[j], &W[1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002218
2219 for( i = 0; i < wsize - 1; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002220 mpi_montmul( &W[j], &W[j], N, mm, &T );
Paul Bakker0d7702c2013-10-29 16:18:35 +01002221
Paul Bakker5121ce52009-01-03 21:22:43 +00002222 /*
2223 * W[i] = W[i - 1] * W[1]
2224 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002225 for( i = j + 1; i < ( one << wsize ); i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002226 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002227 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[i], N->n + 1 ) );
2228 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[i], &W[i - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002229
Gilles Peskine4e91d472020-06-04 20:55:15 +02002230 mpi_montmul( &W[i], &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002231 }
2232 }
2233
2234 nblimbs = E->n;
2235 bufsize = 0;
2236 nbits = 0;
2237 wbits = 0;
2238 state = 0;
2239
2240 while( 1 )
2241 {
2242 if( bufsize == 0 )
2243 {
Paul Bakker0d7702c2013-10-29 16:18:35 +01002244 if( nblimbs == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002245 break;
2246
Paul Bakker0d7702c2013-10-29 16:18:35 +01002247 nblimbs--;
2248
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002249 bufsize = sizeof( mbedtls_mpi_uint ) << 3;
Paul Bakker5121ce52009-01-03 21:22:43 +00002250 }
2251
2252 bufsize--;
2253
2254 ei = (E->p[nblimbs] >> bufsize) & 1;
2255
2256 /*
2257 * skip leading 0s
2258 */
2259 if( ei == 0 && state == 0 )
2260 continue;
2261
2262 if( ei == 0 && state == 1 )
2263 {
2264 /*
2265 * out of window, square X
2266 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002267 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002268 continue;
2269 }
2270
2271 /*
2272 * add ei to current window
2273 */
2274 state = 2;
2275
2276 nbits++;
Paul Bakker66d5d072014-06-17 16:39:18 +02002277 wbits |= ( ei << ( wsize - nbits ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002278
2279 if( nbits == wsize )
2280 {
2281 /*
2282 * X = X^wsize R^-1 mod N
2283 */
2284 for( i = 0; i < wsize; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002285 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002286
2287 /*
2288 * X = X * W[wbits] R^-1 mod N
2289 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002290 mpi_montmul( X, &W[wbits], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002291
2292 state--;
2293 nbits = 0;
2294 wbits = 0;
2295 }
2296 }
2297
2298 /*
2299 * process the remaining bits
2300 */
2301 for( i = 0; i < nbits; i++ )
2302 {
Gilles Peskine4e91d472020-06-04 20:55:15 +02002303 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002304
2305 wbits <<= 1;
2306
Paul Bakker66d5d072014-06-17 16:39:18 +02002307 if( ( wbits & ( one << wsize ) ) != 0 )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002308 mpi_montmul( X, &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002309 }
2310
2311 /*
2312 * X = A^E * R * R^-1 mod N = A^E mod N
2313 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002314 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002315
Hanno Beckera4af1c42017-04-18 09:07:45 +01002316 if( neg && E->n != 0 && ( E->p[0] & 1 ) != 0 )
Paul Bakkerf6198c12012-05-16 08:02:29 +00002317 {
2318 X->s = -1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002319 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( X, N, X ) );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002320 }
2321
Paul Bakker5121ce52009-01-03 21:22:43 +00002322cleanup:
2323
Paul Bakker66d5d072014-06-17 16:39:18 +02002324 for( i = ( one << ( wsize - 1 ) ); i < ( one << wsize ); i++ )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002325 mbedtls_mpi_free( &W[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002326
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002327 mbedtls_mpi_free( &W[1] ); mbedtls_mpi_free( &T ); mbedtls_mpi_free( &Apos );
Paul Bakker6c591fa2011-05-05 11:49:20 +00002328
Paul Bakker75a28602014-03-31 12:08:17 +02002329 if( _RR == NULL || _RR->p == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002330 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002331
2332 return( ret );
2333}
2334
Paul Bakker5121ce52009-01-03 21:22:43 +00002335/*
2336 * Greatest common divisor: G = gcd(A, B) (HAC 14.54)
2337 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002338int mbedtls_mpi_gcd( mbedtls_mpi *G, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00002339{
Janos Follath24eed8d2019-11-22 13:21:35 +00002340 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002341 size_t lz, lzt;
Alexander Ke8ad49f2019-08-16 16:16:07 +03002342 mbedtls_mpi TA, TB;
Paul Bakker5121ce52009-01-03 21:22:43 +00002343
Hanno Becker73d7d792018-12-11 10:35:51 +00002344 MPI_VALIDATE_RET( G != NULL );
2345 MPI_VALIDATE_RET( A != NULL );
2346 MPI_VALIDATE_RET( B != NULL );
2347
Alexander Ke8ad49f2019-08-16 16:16:07 +03002348 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002349
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002350 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) );
2351 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002352
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002353 lz = mbedtls_mpi_lsb( &TA );
2354 lzt = mbedtls_mpi_lsb( &TB );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002355
Paul Bakker66d5d072014-06-17 16:39:18 +02002356 if( lzt < lz )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002357 lz = lzt;
2358
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002359 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, lz ) );
2360 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, lz ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002361
Paul Bakker5121ce52009-01-03 21:22:43 +00002362 TA.s = TB.s = 1;
2363
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002364 while( mbedtls_mpi_cmp_int( &TA, 0 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002365 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002366 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, mbedtls_mpi_lsb( &TA ) ) );
2367 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, mbedtls_mpi_lsb( &TB ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002368
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002369 if( mbedtls_mpi_cmp_mpi( &TA, &TB ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002370 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002371 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TA, &TA, &TB ) );
2372 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002373 }
2374 else
2375 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002376 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TB, &TB, &TA ) );
2377 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002378 }
2379 }
2380
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002381 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &TB, lz ) );
2382 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( G, &TB ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002383
2384cleanup:
2385
Alexander Ke8ad49f2019-08-16 16:16:07 +03002386 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002387
2388 return( ret );
2389}
2390
Paul Bakker33dc46b2014-04-30 16:11:39 +02002391/*
2392 * Fill X with size bytes of random.
2393 *
2394 * Use a temporary bytes representation to make sure the result is the same
Paul Bakkerc37b0ac2014-05-01 14:19:23 +02002395 * regardless of the platform endianness (useful when f_rng is actually
Paul Bakker33dc46b2014-04-30 16:11:39 +02002396 * deterministic, eg for tests).
2397 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002398int mbedtls_mpi_fill_random( mbedtls_mpi *X, size_t size,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002399 int (*f_rng)(void *, unsigned char *, size_t),
2400 void *p_rng )
Paul Bakker287781a2011-03-26 13:18:49 +00002401{
Janos Follath24eed8d2019-11-22 13:21:35 +00002402 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker6dab6202019-01-02 16:42:29 +00002403 size_t const limbs = CHARS_TO_LIMBS( size );
Hanno Beckerda1655a2017-10-18 14:21:44 +01002404 size_t const overhead = ( limbs * ciL ) - size;
2405 unsigned char *Xp;
2406
Hanno Becker8ce11a32018-12-19 16:18:52 +00002407 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002408 MPI_VALIDATE_RET( f_rng != NULL );
Paul Bakker33dc46b2014-04-30 16:11:39 +02002409
Hanno Beckerda1655a2017-10-18 14:21:44 +01002410 /* Ensure that target MPI has exactly the necessary number of limbs */
2411 if( X->n != limbs )
2412 {
2413 mbedtls_mpi_free( X );
2414 mbedtls_mpi_init( X );
2415 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
2416 }
2417 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker287781a2011-03-26 13:18:49 +00002418
Hanno Beckerda1655a2017-10-18 14:21:44 +01002419 Xp = (unsigned char*) X->p;
2420 f_rng( p_rng, Xp + overhead, size );
2421
Hanno Becker2be8a552018-10-25 12:40:09 +01002422 mpi_bigendian_to_host( X->p, limbs );
Paul Bakker287781a2011-03-26 13:18:49 +00002423
2424cleanup:
2425 return( ret );
2426}
2427
Paul Bakker5121ce52009-01-03 21:22:43 +00002428/*
2429 * Modular inverse: X = A^-1 mod N (HAC 14.61 / 14.64)
2430 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002431int mbedtls_mpi_inv_mod( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002432{
Janos Follath24eed8d2019-11-22 13:21:35 +00002433 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002434 mbedtls_mpi G, TA, TU, U1, U2, TB, TV, V1, V2;
Hanno Becker73d7d792018-12-11 10:35:51 +00002435 MPI_VALIDATE_RET( X != NULL );
2436 MPI_VALIDATE_RET( A != NULL );
2437 MPI_VALIDATE_RET( N != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00002438
Hanno Becker4bcb4912017-04-18 15:49:39 +01002439 if( mbedtls_mpi_cmp_int( N, 1 ) <= 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002440 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002441
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002442 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TU ); mbedtls_mpi_init( &U1 ); mbedtls_mpi_init( &U2 );
2443 mbedtls_mpi_init( &G ); mbedtls_mpi_init( &TB ); mbedtls_mpi_init( &TV );
2444 mbedtls_mpi_init( &V1 ); mbedtls_mpi_init( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002445
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002446 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &G, A, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002447
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002448 if( mbedtls_mpi_cmp_int( &G, 1 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002449 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002450 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002451 goto cleanup;
2452 }
2453
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002454 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &TA, A, N ) );
2455 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TU, &TA ) );
2456 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, N ) );
2457 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TV, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002458
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002459 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U1, 1 ) );
2460 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U2, 0 ) );
2461 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V1, 0 ) );
2462 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002463
2464 do
2465 {
2466 while( ( TU.p[0] & 1 ) == 0 )
2467 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002468 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TU, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002469
2470 if( ( U1.p[0] & 1 ) != 0 || ( U2.p[0] & 1 ) != 0 )
2471 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002472 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &U1, &U1, &TB ) );
2473 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002474 }
2475
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002476 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U1, 1 ) );
2477 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002478 }
2479
2480 while( ( TV.p[0] & 1 ) == 0 )
2481 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002482 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TV, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002483
2484 if( ( V1.p[0] & 1 ) != 0 || ( V2.p[0] & 1 ) != 0 )
2485 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002486 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, &TB ) );
2487 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002488 }
2489
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002490 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V1, 1 ) );
2491 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002492 }
2493
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002494 if( mbedtls_mpi_cmp_mpi( &TU, &TV ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002495 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002496 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TU, &TU, &TV ) );
2497 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U1, &U1, &V1 ) );
2498 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &V2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002499 }
2500 else
2501 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002502 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TV, &TV, &TU ) );
2503 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, &U1 ) );
2504 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &U2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002505 }
2506 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002507 while( mbedtls_mpi_cmp_int( &TU, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002508
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002509 while( mbedtls_mpi_cmp_int( &V1, 0 ) < 0 )
2510 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002511
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002512 while( mbedtls_mpi_cmp_mpi( &V1, N ) >= 0 )
2513 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002514
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002515 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &V1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002516
2517cleanup:
2518
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002519 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TU ); mbedtls_mpi_free( &U1 ); mbedtls_mpi_free( &U2 );
2520 mbedtls_mpi_free( &G ); mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TV );
2521 mbedtls_mpi_free( &V1 ); mbedtls_mpi_free( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002522
2523 return( ret );
2524}
2525
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002526#if defined(MBEDTLS_GENPRIME)
Paul Bakkerd9374b02012-11-02 11:02:58 +00002527
Paul Bakker5121ce52009-01-03 21:22:43 +00002528static const int small_prime[] =
2529{
2530 3, 5, 7, 11, 13, 17, 19, 23,
2531 29, 31, 37, 41, 43, 47, 53, 59,
2532 61, 67, 71, 73, 79, 83, 89, 97,
2533 101, 103, 107, 109, 113, 127, 131, 137,
2534 139, 149, 151, 157, 163, 167, 173, 179,
2535 181, 191, 193, 197, 199, 211, 223, 227,
2536 229, 233, 239, 241, 251, 257, 263, 269,
2537 271, 277, 281, 283, 293, 307, 311, 313,
2538 317, 331, 337, 347, 349, 353, 359, 367,
2539 373, 379, 383, 389, 397, 401, 409, 419,
2540 421, 431, 433, 439, 443, 449, 457, 461,
2541 463, 467, 479, 487, 491, 499, 503, 509,
2542 521, 523, 541, 547, 557, 563, 569, 571,
2543 577, 587, 593, 599, 601, 607, 613, 617,
2544 619, 631, 641, 643, 647, 653, 659, 661,
2545 673, 677, 683, 691, 701, 709, 719, 727,
2546 733, 739, 743, 751, 757, 761, 769, 773,
2547 787, 797, 809, 811, 821, 823, 827, 829,
2548 839, 853, 857, 859, 863, 877, 881, 883,
2549 887, 907, 911, 919, 929, 937, 941, 947,
2550 953, 967, 971, 977, 983, 991, 997, -103
2551};
2552
2553/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002554 * Small divisors test (X must be positive)
2555 *
2556 * Return values:
2557 * 0: no small factor (possible prime, more tests needed)
2558 * 1: certain prime
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002559 * MBEDTLS_ERR_MPI_NOT_ACCEPTABLE: certain non-prime
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002560 * other negative: error
Paul Bakker5121ce52009-01-03 21:22:43 +00002561 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002562static int mpi_check_small_factors( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +00002563{
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002564 int ret = 0;
2565 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002566 mbedtls_mpi_uint r;
Paul Bakker5121ce52009-01-03 21:22:43 +00002567
Paul Bakker5121ce52009-01-03 21:22:43 +00002568 if( ( X->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002569 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002570
2571 for( i = 0; small_prime[i] > 0; i++ )
2572 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002573 if( mbedtls_mpi_cmp_int( X, small_prime[i] ) <= 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002574 return( 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002575
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002576 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, small_prime[i] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002577
2578 if( r == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002579 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002580 }
2581
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002582cleanup:
2583 return( ret );
2584}
2585
2586/*
2587 * Miller-Rabin pseudo-primality test (HAC 4.24)
2588 */
Janos Follathda31fa12018-09-03 14:45:23 +01002589static int mpi_miller_rabin( const mbedtls_mpi *X, size_t rounds,
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002590 int (*f_rng)(void *, unsigned char *, size_t),
2591 void *p_rng )
2592{
Pascal Junodb99183d2015-03-11 16:49:45 +01002593 int ret, count;
Janos Follathda31fa12018-09-03 14:45:23 +01002594 size_t i, j, k, s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002595 mbedtls_mpi W, R, T, A, RR;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002596
Hanno Becker8ce11a32018-12-19 16:18:52 +00002597 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002598 MPI_VALIDATE_RET( f_rng != NULL );
2599
2600 mbedtls_mpi_init( &W ); mbedtls_mpi_init( &R );
2601 mbedtls_mpi_init( &T ); mbedtls_mpi_init( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002602 mbedtls_mpi_init( &RR );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002603
Paul Bakker5121ce52009-01-03 21:22:43 +00002604 /*
2605 * W = |X| - 1
2606 * R = W >> lsb( W )
2607 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002608 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( &W, X, 1 ) );
2609 s = mbedtls_mpi_lsb( &W );
2610 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &R, &W ) );
2611 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &R, s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002612
Janos Follathda31fa12018-09-03 14:45:23 +01002613 for( i = 0; i < rounds; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002614 {
2615 /*
2616 * pick a random A, 1 < A < |X| - 1
2617 */
Pascal Junodb99183d2015-03-11 16:49:45 +01002618 count = 0;
2619 do {
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002620 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( &A, X->n * ciL, f_rng, p_rng ) );
Pascal Junodb99183d2015-03-11 16:49:45 +01002621
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002622 j = mbedtls_mpi_bitlen( &A );
2623 k = mbedtls_mpi_bitlen( &W );
Pascal Junodb99183d2015-03-11 16:49:45 +01002624 if (j > k) {
Darryl Greene3f95ed2018-10-02 13:21:35 +01002625 A.p[A.n - 1] &= ( (mbedtls_mpi_uint) 1 << ( k - ( A.n - 1 ) * biL - 1 ) ) - 1;
Pascal Junodb99183d2015-03-11 16:49:45 +01002626 }
2627
2628 if (count++ > 30) {
Jens Wiklanderf08aa3e2019-01-17 13:30:57 +01002629 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
2630 goto cleanup;
Pascal Junodb99183d2015-03-11 16:49:45 +01002631 }
2632
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002633 } while ( mbedtls_mpi_cmp_mpi( &A, &W ) >= 0 ||
2634 mbedtls_mpi_cmp_int( &A, 1 ) <= 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002635
2636 /*
2637 * A = A^R mod |X|
2638 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002639 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &A, &A, &R, X, &RR ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002640
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002641 if( mbedtls_mpi_cmp_mpi( &A, &W ) == 0 ||
2642 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002643 continue;
2644
2645 j = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002646 while( j < s && mbedtls_mpi_cmp_mpi( &A, &W ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002647 {
2648 /*
2649 * A = A * A mod |X|
2650 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002651 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &T, &A, &A ) );
2652 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &A, &T, X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002653
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002654 if( mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002655 break;
2656
2657 j++;
2658 }
2659
2660 /*
2661 * not prime if A != |X| - 1 or A == 1
2662 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002663 if( mbedtls_mpi_cmp_mpi( &A, &W ) != 0 ||
2664 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002665 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002666 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002667 break;
2668 }
2669 }
2670
2671cleanup:
Hanno Becker73d7d792018-12-11 10:35:51 +00002672 mbedtls_mpi_free( &W ); mbedtls_mpi_free( &R );
2673 mbedtls_mpi_free( &T ); mbedtls_mpi_free( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002674 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002675
2676 return( ret );
2677}
2678
2679/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002680 * Pseudo-primality test: small factors, then Miller-Rabin
2681 */
Janos Follatha0b67c22018-09-18 14:48:23 +01002682int mbedtls_mpi_is_prime_ext( const mbedtls_mpi *X, int rounds,
2683 int (*f_rng)(void *, unsigned char *, size_t),
2684 void *p_rng )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002685{
Janos Follath24eed8d2019-11-22 13:21:35 +00002686 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002687 mbedtls_mpi XX;
Hanno Becker8ce11a32018-12-19 16:18:52 +00002688 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002689 MPI_VALIDATE_RET( f_rng != NULL );
Manuel Pégourié-Gonnard7f4ed672014-10-14 20:56:02 +02002690
2691 XX.s = 1;
2692 XX.n = X->n;
2693 XX.p = X->p;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002694
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002695 if( mbedtls_mpi_cmp_int( &XX, 0 ) == 0 ||
2696 mbedtls_mpi_cmp_int( &XX, 1 ) == 0 )
2697 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002698
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002699 if( mbedtls_mpi_cmp_int( &XX, 2 ) == 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002700 return( 0 );
2701
2702 if( ( ret = mpi_check_small_factors( &XX ) ) != 0 )
2703 {
2704 if( ret == 1 )
2705 return( 0 );
2706
2707 return( ret );
2708 }
2709
Janos Follathda31fa12018-09-03 14:45:23 +01002710 return( mpi_miller_rabin( &XX, rounds, f_rng, p_rng ) );
Janos Follathf301d232018-08-14 13:34:01 +01002711}
2712
Janos Follatha0b67c22018-09-18 14:48:23 +01002713#if !defined(MBEDTLS_DEPRECATED_REMOVED)
Janos Follathf301d232018-08-14 13:34:01 +01002714/*
2715 * Pseudo-primality test, error probability 2^-80
2716 */
2717int mbedtls_mpi_is_prime( const mbedtls_mpi *X,
2718 int (*f_rng)(void *, unsigned char *, size_t),
2719 void *p_rng )
2720{
Hanno Becker8ce11a32018-12-19 16:18:52 +00002721 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002722 MPI_VALIDATE_RET( f_rng != NULL );
2723
Janos Follatha0b67c22018-09-18 14:48:23 +01002724 /*
2725 * In the past our key generation aimed for an error rate of at most
2726 * 2^-80. Since this function is deprecated, aim for the same certainty
2727 * here as well.
2728 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002729 return( mbedtls_mpi_is_prime_ext( X, 40, f_rng, p_rng ) );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002730}
Janos Follatha0b67c22018-09-18 14:48:23 +01002731#endif
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002732
2733/*
Paul Bakker5121ce52009-01-03 21:22:43 +00002734 * Prime number generation
Jethro Beekman66689272018-02-14 19:24:10 -08002735 *
Janos Follathf301d232018-08-14 13:34:01 +01002736 * To generate an RSA key in a way recommended by FIPS 186-4, both primes must
2737 * be either 1024 bits or 1536 bits long, and flags must contain
2738 * MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR.
Paul Bakker5121ce52009-01-03 21:22:43 +00002739 */
Janos Follath7c025a92018-08-14 11:08:41 +01002740int mbedtls_mpi_gen_prime( mbedtls_mpi *X, size_t nbits, int flags,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002741 int (*f_rng)(void *, unsigned char *, size_t),
2742 void *p_rng )
Paul Bakker5121ce52009-01-03 21:22:43 +00002743{
Jethro Beekman66689272018-02-14 19:24:10 -08002744#ifdef MBEDTLS_HAVE_INT64
2745// ceil(2^63.5)
2746#define CEIL_MAXUINT_DIV_SQRT2 0xb504f333f9de6485ULL
2747#else
2748// ceil(2^31.5)
2749#define CEIL_MAXUINT_DIV_SQRT2 0xb504f334U
2750#endif
2751 int ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker23986e52011-04-24 08:57:21 +00002752 size_t k, n;
Janos Follathda31fa12018-09-03 14:45:23 +01002753 int rounds;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002754 mbedtls_mpi_uint r;
2755 mbedtls_mpi Y;
Paul Bakker5121ce52009-01-03 21:22:43 +00002756
Hanno Becker8ce11a32018-12-19 16:18:52 +00002757 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002758 MPI_VALIDATE_RET( f_rng != NULL );
2759
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002760 if( nbits < 3 || nbits > MBEDTLS_MPI_MAX_BITS )
2761 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002762
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002763 mbedtls_mpi_init( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002764
2765 n = BITS_TO_LIMBS( nbits );
2766
Janos Follathda31fa12018-09-03 14:45:23 +01002767 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR ) == 0 )
2768 {
2769 /*
2770 * 2^-80 error probability, number of rounds chosen per HAC, table 4.4
2771 */
2772 rounds = ( ( nbits >= 1300 ) ? 2 : ( nbits >= 850 ) ? 3 :
2773 ( nbits >= 650 ) ? 4 : ( nbits >= 350 ) ? 8 :
2774 ( nbits >= 250 ) ? 12 : ( nbits >= 150 ) ? 18 : 27 );
2775 }
2776 else
2777 {
2778 /*
2779 * 2^-100 error probability, number of rounds computed based on HAC,
2780 * fact 4.48
2781 */
2782 rounds = ( ( nbits >= 1450 ) ? 4 : ( nbits >= 1150 ) ? 5 :
2783 ( nbits >= 1000 ) ? 6 : ( nbits >= 850 ) ? 7 :
2784 ( nbits >= 750 ) ? 8 : ( nbits >= 500 ) ? 13 :
2785 ( nbits >= 250 ) ? 28 : ( nbits >= 150 ) ? 40 : 51 );
2786 }
2787
Jethro Beekman66689272018-02-14 19:24:10 -08002788 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002789 {
Jethro Beekman66689272018-02-14 19:24:10 -08002790 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( X, n * ciL, f_rng, p_rng ) );
2791 /* make sure generated number is at least (nbits-1)+0.5 bits (FIPS 186-4 §B.3.3 steps 4.4, 5.5) */
2792 if( X->p[n-1] < CEIL_MAXUINT_DIV_SQRT2 ) continue;
2793
2794 k = n * biL;
2795 if( k > nbits ) MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( X, k - nbits ) );
2796 X->p[0] |= 1;
2797
Janos Follath7c025a92018-08-14 11:08:41 +01002798 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_DH ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002799 {
Janos Follatha0b67c22018-09-18 14:48:23 +01002800 ret = mbedtls_mpi_is_prime_ext( X, rounds, f_rng, p_rng );
Jethro Beekman66689272018-02-14 19:24:10 -08002801
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002802 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
Paul Bakker5121ce52009-01-03 21:22:43 +00002803 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002804 }
Jethro Beekman66689272018-02-14 19:24:10 -08002805 else
Paul Bakker5121ce52009-01-03 21:22:43 +00002806 {
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002807 /*
Jethro Beekman66689272018-02-14 19:24:10 -08002808 * An necessary condition for Y and X = 2Y + 1 to be prime
2809 * is X = 2 mod 3 (which is equivalent to Y = 2 mod 3).
2810 * Make sure it is satisfied, while keeping X = 3 mod 4
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002811 */
Jethro Beekman66689272018-02-14 19:24:10 -08002812
2813 X->p[0] |= 2;
2814
2815 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, 3 ) );
2816 if( r == 0 )
2817 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 8 ) );
2818 else if( r == 1 )
2819 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 4 ) );
2820
2821 /* Set Y = (X-1) / 2, which is X / 2 because X is odd */
2822 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, X ) );
2823 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, 1 ) );
2824
2825 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002826 {
Jethro Beekman66689272018-02-14 19:24:10 -08002827 /*
2828 * First, check small factors for X and Y
2829 * before doing Miller-Rabin on any of them
2830 */
2831 if( ( ret = mpi_check_small_factors( X ) ) == 0 &&
2832 ( ret = mpi_check_small_factors( &Y ) ) == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002833 ( ret = mpi_miller_rabin( X, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002834 == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002835 ( ret = mpi_miller_rabin( &Y, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002836 == 0 )
Jethro Beekman66689272018-02-14 19:24:10 -08002837 goto cleanup;
2838
2839 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
2840 goto cleanup;
2841
2842 /*
2843 * Next candidates. We want to preserve Y = (X-1) / 2 and
2844 * Y = 1 mod 2 and Y = 2 mod 3 (eq X = 3 mod 4 and X = 2 mod 3)
2845 * so up Y by 6 and X by 12.
2846 */
2847 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 12 ) );
2848 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( &Y, &Y, 6 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002849 }
Paul Bakker5121ce52009-01-03 21:22:43 +00002850 }
2851 }
2852
2853cleanup:
2854
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002855 mbedtls_mpi_free( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002856
2857 return( ret );
2858}
2859
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002860#endif /* MBEDTLS_GENPRIME */
Paul Bakker5121ce52009-01-03 21:22:43 +00002861
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002862#if defined(MBEDTLS_SELF_TEST)
Paul Bakker5121ce52009-01-03 21:22:43 +00002863
Paul Bakker23986e52011-04-24 08:57:21 +00002864#define GCD_PAIR_COUNT 3
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002865
2866static const int gcd_pairs[GCD_PAIR_COUNT][3] =
2867{
2868 { 693, 609, 21 },
2869 { 1764, 868, 28 },
2870 { 768454923, 542167814, 1 }
2871};
2872
Paul Bakker5121ce52009-01-03 21:22:43 +00002873/*
2874 * Checkup routine
2875 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002876int mbedtls_mpi_self_test( int verbose )
Paul Bakker5121ce52009-01-03 21:22:43 +00002877{
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002878 int ret, i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002879 mbedtls_mpi A, E, N, X, Y, U, V;
Paul Bakker5121ce52009-01-03 21:22:43 +00002880
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002881 mbedtls_mpi_init( &A ); mbedtls_mpi_init( &E ); mbedtls_mpi_init( &N ); mbedtls_mpi_init( &X );
2882 mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &U ); mbedtls_mpi_init( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002883
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002884 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &A, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002885 "EFE021C2645FD1DC586E69184AF4A31E" \
2886 "D5F53E93B5F123FA41680867BA110131" \
2887 "944FE7952E2517337780CB0DB80E61AA" \
2888 "E7C8DDC6C5C6AADEB34EB38A2F40D5E6" ) );
2889
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002890 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &E, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002891 "B2E7EFD37075B9F03FF989C7C5051C20" \
2892 "34D2A323810251127E7BF8625A4F49A5" \
2893 "F3E27F4DA8BD59C47D6DAABA4C8127BD" \
2894 "5B5C25763222FEFCCFC38B832366C29E" ) );
2895
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002896 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &N, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002897 "0066A198186C18C10B2F5ED9B522752A" \
2898 "9830B69916E535C8F047518A889A43A5" \
2899 "94B6BED27A168D31D4A52F88925AA8F5" ) );
2900
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002901 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002902
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002903 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002904 "602AB7ECA597A3D6B56FF9829A5E8B85" \
2905 "9E857EA95A03512E2BAE7391688D264A" \
2906 "A5663B0341DB9CCFD2C4C5F421FEC814" \
2907 "8001B72E848A38CAE1C65F78E56ABDEF" \
2908 "E12D3C039B8A02D6BE593F0BBBDA56F1" \
2909 "ECF677152EF804370C1A305CAF3B5BF1" \
2910 "30879B56C61DE584A0F53A2447A51E" ) );
2911
2912 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002913 mbedtls_printf( " MPI test #1 (mul_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002914
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002915 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002916 {
2917 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002918 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002919
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002920 ret = 1;
2921 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002922 }
2923
2924 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002925 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002926
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002927 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( &X, &Y, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002928
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002929 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002930 "256567336059E52CAE22925474705F39A94" ) );
2931
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002932 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &V, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002933 "6613F26162223DF488E9CD48CC132C7A" \
2934 "0AC93C701B001B092E4E5B9F73BCD27B" \
2935 "9EE50D0657C77F374E903CDFA4C642" ) );
2936
2937 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002938 mbedtls_printf( " MPI test #2 (div_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002939
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002940 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 ||
2941 mbedtls_mpi_cmp_mpi( &Y, &V ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002942 {
2943 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002944 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002945
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002946 ret = 1;
2947 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002948 }
2949
2950 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002951 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002952
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002953 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &X, &A, &E, &N, NULL ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002954
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002955 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002956 "36E139AEA55215609D2816998ED020BB" \
2957 "BD96C37890F65171D948E9BC7CBAA4D9" \
2958 "325D24D6A3C12710F10A09FA08AB87" ) );
2959
2960 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002961 mbedtls_printf( " MPI test #3 (exp_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002962
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002963 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002964 {
2965 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002966 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002967
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002968 ret = 1;
2969 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002970 }
2971
2972 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002973 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002974
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002975 MBEDTLS_MPI_CHK( mbedtls_mpi_inv_mod( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002976
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002977 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002978 "003A0AAEDD7E784FC07D8F9EC6E3BFD5" \
2979 "C3DBA76456363A10869622EAC2DD84EC" \
2980 "C5B8A74DAC4D09E03B5E0BE779F2DF61" ) );
2981
2982 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002983 mbedtls_printf( " MPI test #4 (inv_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002984
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002985 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002986 {
2987 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002988 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002989
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002990 ret = 1;
2991 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002992 }
2993
2994 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002995 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002996
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002997 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002998 mbedtls_printf( " MPI test #5 (simple gcd): " );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002999
Paul Bakker66d5d072014-06-17 16:39:18 +02003000 for( i = 0; i < GCD_PAIR_COUNT; i++ )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003001 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003002 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &X, gcd_pairs[i][0] ) );
3003 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Y, gcd_pairs[i][1] ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003004
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003005 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &A, &X, &Y ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003006
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003007 if( mbedtls_mpi_cmp_int( &A, gcd_pairs[i][2] ) != 0 )
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01003008 {
3009 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003010 mbedtls_printf( "failed at %d\n", i );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003011
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01003012 ret = 1;
3013 goto cleanup;
3014 }
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003015 }
3016
3017 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003018 mbedtls_printf( "passed\n" );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003019
Paul Bakker5121ce52009-01-03 21:22:43 +00003020cleanup:
3021
3022 if( ret != 0 && verbose != 0 )
Kenneth Soerensen518d4352020-04-01 17:22:45 +02003023 mbedtls_printf( "Unexpected error, return code = %08X\n", (unsigned int) ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00003024
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003025 mbedtls_mpi_free( &A ); mbedtls_mpi_free( &E ); mbedtls_mpi_free( &N ); mbedtls_mpi_free( &X );
3026 mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &U ); mbedtls_mpi_free( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00003027
3028 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003029 mbedtls_printf( "\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00003030
3031 return( ret );
3032}
3033
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003034#endif /* MBEDTLS_SELF_TEST */
Paul Bakker5121ce52009-01-03 21:22:43 +00003035
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003036#endif /* MBEDTLS_BIGNUM_C */