blob: 4413752633c5cea28e4f57ec2717a8a99befea0e [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
Gilles Peskine8e464c42020-07-24 00:08:38 +02001610 while( c != 0 )
1611 {
Paul Bakker5121ce52009-01-03 21:22:43 +00001612 *d += c; c = ( *d < c ); d++;
1613 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001614}
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;
Gilles Peskinecd0dbf32020-07-24 00:09:04 +02001679 /* In general, A * b requires 1 limb more than b. If
1680 * A->p[n - 1] * b / b == A->p[n - 1], then A * b fits in the same
1681 * number of limbs as A and the call to grow() is not required since
1682 * copy() will take care of the growth. However, experimentally,
1683 * making the call to grow() conditional causes slightly fewer
1684 * calls to calloc() in ECP code, presumably because it reuses the
1685 * same mpi for a while and this way the mpi is more likely to directly
1686 * grow to its final size. */
Gilles Peskine8fd95c62020-07-23 21:58:50 +02001687 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, n + 1 ) );
1688 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, A ) );
1689 mpi_mul_hlp( n, A->p, X->p, b - 1 );
1690
1691cleanup:
1692 return( ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00001693}
1694
1695/*
Simon Butcherf5ba0452015-12-27 23:01:55 +00001696 * Unsigned integer divide - double mbedtls_mpi_uint dividend, u1/u0, and
1697 * mbedtls_mpi_uint divisor, d
Simon Butcher15b15d12015-11-26 19:35:03 +00001698 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001699static mbedtls_mpi_uint mbedtls_int_div_int( mbedtls_mpi_uint u1,
1700 mbedtls_mpi_uint u0, mbedtls_mpi_uint d, mbedtls_mpi_uint *r )
Simon Butcher15b15d12015-11-26 19:35:03 +00001701{
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001702#if defined(MBEDTLS_HAVE_UDBL)
1703 mbedtls_t_udbl dividend, quotient;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001704#else
Simon Butcher9803d072016-01-03 00:24:34 +00001705 const mbedtls_mpi_uint radix = (mbedtls_mpi_uint) 1 << biH;
1706 const mbedtls_mpi_uint uint_halfword_mask = ( (mbedtls_mpi_uint) 1 << biH ) - 1;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001707 mbedtls_mpi_uint d0, d1, q0, q1, rAX, r0, quotient;
1708 mbedtls_mpi_uint u0_msw, u0_lsw;
Simon Butcher9803d072016-01-03 00:24:34 +00001709 size_t s;
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001710#endif
1711
Simon Butcher15b15d12015-11-26 19:35:03 +00001712 /*
1713 * Check for overflow
1714 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001715 if( 0 == d || u1 >= d )
Simon Butcher15b15d12015-11-26 19:35:03 +00001716 {
Simon Butcherf5ba0452015-12-27 23:01:55 +00001717 if (r != NULL) *r = ~0;
Simon Butcher15b15d12015-11-26 19:35:03 +00001718
Simon Butcherf5ba0452015-12-27 23:01:55 +00001719 return ( ~0 );
Simon Butcher15b15d12015-11-26 19:35:03 +00001720 }
1721
1722#if defined(MBEDTLS_HAVE_UDBL)
Simon Butcher15b15d12015-11-26 19:35:03 +00001723 dividend = (mbedtls_t_udbl) u1 << biL;
1724 dividend |= (mbedtls_t_udbl) u0;
1725 quotient = dividend / d;
1726 if( quotient > ( (mbedtls_t_udbl) 1 << biL ) - 1 )
1727 quotient = ( (mbedtls_t_udbl) 1 << biL ) - 1;
1728
1729 if( r != NULL )
Simon Butcher9803d072016-01-03 00:24:34 +00001730 *r = (mbedtls_mpi_uint)( dividend - (quotient * d ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001731
1732 return (mbedtls_mpi_uint) quotient;
1733#else
Simon Butcher15b15d12015-11-26 19:35:03 +00001734
1735 /*
1736 * Algorithm D, Section 4.3.1 - The Art of Computer Programming
1737 * Vol. 2 - Seminumerical Algorithms, Knuth
1738 */
1739
1740 /*
1741 * Normalize the divisor, d, and dividend, u0, u1
1742 */
1743 s = mbedtls_clz( d );
1744 d = d << s;
1745
1746 u1 = u1 << s;
Simon Butcher9803d072016-01-03 00:24:34 +00001747 u1 |= ( u0 >> ( biL - s ) ) & ( -(mbedtls_mpi_sint)s >> ( biL - 1 ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001748 u0 = u0 << s;
1749
1750 d1 = d >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001751 d0 = d & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001752
1753 u0_msw = u0 >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001754 u0_lsw = u0 & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001755
1756 /*
1757 * Find the first quotient and remainder
1758 */
1759 q1 = u1 / d1;
1760 r0 = u1 - d1 * q1;
1761
1762 while( q1 >= radix || ( q1 * d0 > radix * r0 + u0_msw ) )
1763 {
1764 q1 -= 1;
1765 r0 += d1;
1766
1767 if ( r0 >= radix ) break;
1768 }
1769
Simon Butcherf5ba0452015-12-27 23:01:55 +00001770 rAX = ( u1 * radix ) + ( u0_msw - q1 * d );
Simon Butcher15b15d12015-11-26 19:35:03 +00001771 q0 = rAX / d1;
1772 r0 = rAX - q0 * d1;
1773
1774 while( q0 >= radix || ( q0 * d0 > radix * r0 + u0_lsw ) )
1775 {
1776 q0 -= 1;
1777 r0 += d1;
1778
1779 if ( r0 >= radix ) break;
1780 }
1781
1782 if (r != NULL)
Simon Butcherf5ba0452015-12-27 23:01:55 +00001783 *r = ( rAX * radix + u0_lsw - q0 * d ) >> s;
Simon Butcher15b15d12015-11-26 19:35:03 +00001784
1785 quotient = q1 * radix + q0;
1786
1787 return quotient;
1788#endif
1789}
1790
1791/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001792 * Division by mbedtls_mpi: A = Q * B + R (HAC 14.20)
Paul Bakker5121ce52009-01-03 21:22:43 +00001793 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001794int mbedtls_mpi_div_mpi( mbedtls_mpi *Q, mbedtls_mpi *R, const mbedtls_mpi *A,
1795 const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001796{
Janos Follath24eed8d2019-11-22 13:21:35 +00001797 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001798 size_t i, n, t, k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001799 mbedtls_mpi X, Y, Z, T1, T2;
Alexander Kd19a1932019-11-01 18:20:42 +03001800 mbedtls_mpi_uint TP2[3];
Hanno Becker73d7d792018-12-11 10:35:51 +00001801 MPI_VALIDATE_RET( A != NULL );
1802 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001803
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001804 if( mbedtls_mpi_cmp_int( B, 0 ) == 0 )
1805 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001806
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001807 mbedtls_mpi_init( &X ); mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001808 mbedtls_mpi_init( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001809 /*
1810 * Avoid dynamic memory allocations for constant-size T2.
1811 *
1812 * T2 is used for comparison only and the 3 limbs are assigned explicitly,
1813 * so nobody increase the size of the MPI and we're safe to use an on-stack
1814 * buffer.
1815 */
Alexander K35d6d462019-10-31 14:46:45 +03001816 T2.s = 1;
Alexander Kd19a1932019-11-01 18:20:42 +03001817 T2.n = sizeof( TP2 ) / sizeof( *TP2 );
1818 T2.p = TP2;
Paul Bakker5121ce52009-01-03 21:22:43 +00001819
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001820 if( mbedtls_mpi_cmp_abs( A, B ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001821 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001822 if( Q != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_lset( Q, 0 ) );
1823 if( R != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001824 return( 0 );
1825 }
1826
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001827 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &X, A ) );
1828 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001829 X.s = Y.s = 1;
1830
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001831 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &Z, A->n + 2 ) );
1832 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Z, 0 ) );
1833 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T1, 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001834
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001835 k = mbedtls_mpi_bitlen( &Y ) % biL;
Paul Bakkerf9688572011-05-05 10:00:45 +00001836 if( k < biL - 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001837 {
1838 k = biL - 1 - k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001839 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &X, k ) );
1840 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, k ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001841 }
1842 else k = 0;
1843
1844 n = X.n - 1;
1845 t = Y.n - 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001846 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001847
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001848 while( mbedtls_mpi_cmp_mpi( &X, &Y ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001849 {
1850 Z.p[n - t]++;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001851 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001852 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001853 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001854
1855 for( i = n; i > t ; i-- )
1856 {
1857 if( X.p[i] >= Y.p[t] )
1858 Z.p[i - t - 1] = ~0;
1859 else
1860 {
Simon Butcher15b15d12015-11-26 19:35:03 +00001861 Z.p[i - t - 1] = mbedtls_int_div_int( X.p[i], X.p[i - 1],
1862 Y.p[t], NULL);
Paul Bakker5121ce52009-01-03 21:22:43 +00001863 }
1864
Alexander K35d6d462019-10-31 14:46:45 +03001865 T2.p[0] = ( i < 2 ) ? 0 : X.p[i - 2];
1866 T2.p[1] = ( i < 1 ) ? 0 : X.p[i - 1];
1867 T2.p[2] = X.p[i];
1868
Paul Bakker5121ce52009-01-03 21:22:43 +00001869 Z.p[i - t - 1]++;
1870 do
1871 {
1872 Z.p[i - t - 1]--;
1873
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001874 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &T1, 0 ) );
Paul Bakker66d5d072014-06-17 16:39:18 +02001875 T1.p[0] = ( t < 1 ) ? 0 : Y.p[t - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001876 T1.p[1] = Y.p[t];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001877 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &T1, Z.p[i - t - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001878 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001879 while( mbedtls_mpi_cmp_mpi( &T1, &T2 ) > 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001880
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001881 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &Y, Z.p[i - t - 1] ) );
1882 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1883 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001884
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001885 if( mbedtls_mpi_cmp_int( &X, 0 ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001886 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001887 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T1, &Y ) );
1888 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1889 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001890 Z.p[i - t - 1]--;
1891 }
1892 }
1893
1894 if( Q != NULL )
1895 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001896 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( Q, &Z ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001897 Q->s = A->s * B->s;
1898 }
1899
1900 if( R != NULL )
1901 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001902 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &X, k ) );
Paul Bakkerf02c5642012-11-13 10:25:21 +00001903 X.s = A->s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001904 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, &X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001905
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001906 if( mbedtls_mpi_cmp_int( R, 0 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001907 R->s = 1;
1908 }
1909
1910cleanup:
1911
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001912 mbedtls_mpi_free( &X ); mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001913 mbedtls_mpi_free( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001914 mbedtls_platform_zeroize( TP2, sizeof( TP2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001915
1916 return( ret );
1917}
1918
1919/*
1920 * Division by int: A = Q * b + R
Paul Bakker5121ce52009-01-03 21:22:43 +00001921 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001922int mbedtls_mpi_div_int( mbedtls_mpi *Q, mbedtls_mpi *R,
1923 const mbedtls_mpi *A,
1924 mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001925{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001926 mbedtls_mpi _B;
1927 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001928 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001929
1930 p[0] = ( b < 0 ) ? -b : b;
1931 _B.s = ( b < 0 ) ? -1 : 1;
1932 _B.n = 1;
1933 _B.p = p;
1934
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001935 return( mbedtls_mpi_div_mpi( Q, R, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001936}
1937
1938/*
1939 * Modulo: R = A mod B
1940 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001941int mbedtls_mpi_mod_mpi( mbedtls_mpi *R, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001942{
Janos Follath24eed8d2019-11-22 13:21:35 +00001943 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker73d7d792018-12-11 10:35:51 +00001944 MPI_VALIDATE_RET( R != NULL );
1945 MPI_VALIDATE_RET( A != NULL );
1946 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001947
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001948 if( mbedtls_mpi_cmp_int( B, 0 ) < 0 )
1949 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001950
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001951 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( NULL, R, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001952
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001953 while( mbedtls_mpi_cmp_int( R, 0 ) < 0 )
1954 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001955
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001956 while( mbedtls_mpi_cmp_mpi( R, B ) >= 0 )
1957 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001958
1959cleanup:
1960
1961 return( ret );
1962}
1963
1964/*
1965 * Modulo: r = A mod b
1966 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001967int mbedtls_mpi_mod_int( mbedtls_mpi_uint *r, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001968{
Paul Bakker23986e52011-04-24 08:57:21 +00001969 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001970 mbedtls_mpi_uint x, y, z;
Hanno Becker73d7d792018-12-11 10:35:51 +00001971 MPI_VALIDATE_RET( r != NULL );
1972 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001973
1974 if( b == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001975 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001976
1977 if( b < 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001978 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakker5121ce52009-01-03 21:22:43 +00001979
1980 /*
1981 * handle trivial cases
1982 */
1983 if( b == 1 )
1984 {
1985 *r = 0;
1986 return( 0 );
1987 }
1988
1989 if( b == 2 )
1990 {
1991 *r = A->p[0] & 1;
1992 return( 0 );
1993 }
1994
1995 /*
1996 * general case
1997 */
Paul Bakker23986e52011-04-24 08:57:21 +00001998 for( i = A->n, y = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001999 {
Paul Bakker23986e52011-04-24 08:57:21 +00002000 x = A->p[i - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00002001 y = ( y << biH ) | ( x >> biH );
2002 z = y / b;
2003 y -= z * b;
2004
2005 x <<= biH;
2006 y = ( y << biH ) | ( x >> biH );
2007 z = y / b;
2008 y -= z * b;
2009 }
2010
Paul Bakkerce40a6d2009-06-23 19:46:08 +00002011 /*
2012 * If A is negative, then the current y represents a negative value.
2013 * Flipping it to the positive side.
2014 */
2015 if( A->s < 0 && y != 0 )
2016 y = b - y;
2017
Paul Bakker5121ce52009-01-03 21:22:43 +00002018 *r = y;
2019
2020 return( 0 );
2021}
2022
2023/*
2024 * Fast Montgomery initialization (thanks to Tom St Denis)
2025 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002026static void mpi_montg_init( mbedtls_mpi_uint *mm, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002027{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002028 mbedtls_mpi_uint x, m0 = N->p[0];
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01002029 unsigned int i;
Paul Bakker5121ce52009-01-03 21:22:43 +00002030
2031 x = m0;
2032 x += ( ( m0 + 2 ) & 4 ) << 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00002033
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01002034 for( i = biL; i >= 8; i /= 2 )
2035 x *= ( 2 - ( m0 * x ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002036
2037 *mm = ~x + 1;
2038}
2039
Gilles Peskine2a82f722020-06-04 15:00:49 +02002040/** Montgomery multiplication: A = A * B * R^-1 mod N (HAC 14.36)
2041 *
2042 * \param[in,out] A One of the numbers to multiply.
Gilles Peskine221626f2020-06-08 22:37:50 +02002043 * It must have at least as many limbs as N
2044 * (A->n >= N->n), and any limbs beyond n are ignored.
Gilles Peskine2a82f722020-06-04 15:00:49 +02002045 * On successful completion, A contains the result of
2046 * the multiplication A * B * R^-1 mod N where
2047 * R = (2^ciL)^n.
2048 * \param[in] B One of the numbers to multiply.
2049 * It must be nonzero and must not have more limbs than N
2050 * (B->n <= N->n).
2051 * \param[in] N The modulo. N must be odd.
2052 * \param mm The value calculated by `mpi_montg_init(&mm, N)`.
2053 * This is -N^-1 mod 2^ciL.
2054 * \param[in,out] T A bignum for temporary storage.
2055 * It must be at least twice the limb size of N plus 2
2056 * (T->n >= 2 * (N->n + 1)).
2057 * Its initial content is unused and
2058 * its final content is indeterminate.
2059 * Note that unlike the usual convention in the library
2060 * for `const mbedtls_mpi*`, the content of T can change.
Paul Bakker5121ce52009-01-03 21:22:43 +00002061 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002062static 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 +02002063 const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002064{
Paul Bakker23986e52011-04-24 08:57:21 +00002065 size_t i, n, m;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002066 mbedtls_mpi_uint u0, u1, *d;
Paul Bakker5121ce52009-01-03 21:22:43 +00002067
2068 memset( T->p, 0, T->n * ciL );
2069
2070 d = T->p;
2071 n = N->n;
2072 m = ( B->n < n ) ? B->n : n;
2073
2074 for( i = 0; i < n; i++ )
2075 {
2076 /*
2077 * T = (T + u0*B + u1*N) / 2^biL
2078 */
2079 u0 = A->p[i];
2080 u1 = ( d[0] + u0 * B->p[0] ) * mm;
2081
2082 mpi_mul_hlp( m, B->p, d, u0 );
2083 mpi_mul_hlp( n, N->p, d, u1 );
2084
2085 *d++ = u0; d[n + 1] = 0;
2086 }
2087
Gilles Peskine221626f2020-06-08 22:37:50 +02002088 /* At this point, d is either the desired result or the desired result
2089 * plus N. We now potentially subtract N, avoiding leaking whether the
2090 * subtraction is performed through side channels. */
Paul Bakker5121ce52009-01-03 21:22:43 +00002091
Gilles Peskine221626f2020-06-08 22:37:50 +02002092 /* Copy the n least significant limbs of d to A, so that
2093 * A = d if d < N (recall that N has n limbs). */
2094 memcpy( A->p, d, n * ciL );
Gilles Peskine09ec10a2020-06-09 10:39:38 +02002095 /* If d >= N then we want to set A to d - N. To prevent timing attacks,
Gilles Peskine221626f2020-06-08 22:37:50 +02002096 * do the calculation without using conditional tests. */
2097 /* Set d to d0 + (2^biL)^n - N where d0 is the current value of d. */
Gilles Peskine132c0972020-06-04 21:05:24 +02002098 d[n] += 1;
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02002099 d[n] -= mpi_sub_hlp( n, d, d, N->p );
Gilles Peskine221626f2020-06-08 22:37:50 +02002100 /* If d0 < N then d < (2^biL)^n
2101 * so d[n] == 0 and we want to keep A as it is.
2102 * If d0 >= N then d >= (2^biL)^n, and d <= (2^biL)^n + N < 2 * (2^biL)^n
2103 * so d[n] == 1 and we want to set A to the result of the subtraction
2104 * which is d - (2^biL)^n, i.e. the n least significant limbs of d.
2105 * This exactly corresponds to a conditional assignment. */
2106 mpi_safe_cond_assign( n, A->p, d, (unsigned char) d[n] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002107}
2108
2109/*
2110 * Montgomery reduction: A = A * R^-1 mod N
Gilles Peskine2a82f722020-06-04 15:00:49 +02002111 *
2112 * See mpi_montmul() regarding constraints and guarantees on the parameters.
Paul Bakker5121ce52009-01-03 21:22:43 +00002113 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002114static void mpi_montred( mbedtls_mpi *A, const mbedtls_mpi *N,
2115 mbedtls_mpi_uint mm, const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002116{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002117 mbedtls_mpi_uint z = 1;
2118 mbedtls_mpi U;
Paul Bakker5121ce52009-01-03 21:22:43 +00002119
Paul Bakker8ddb6452013-02-27 14:56:33 +01002120 U.n = U.s = (int) z;
Paul Bakker5121ce52009-01-03 21:22:43 +00002121 U.p = &z;
2122
Gilles Peskine4e91d472020-06-04 20:55:15 +02002123 mpi_montmul( A, &U, N, mm, T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002124}
2125
2126/*
2127 * Sliding-window exponentiation: X = A^E mod N (HAC 14.85)
2128 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002129int mbedtls_mpi_exp_mod( mbedtls_mpi *X, const mbedtls_mpi *A,
2130 const mbedtls_mpi *E, const mbedtls_mpi *N,
2131 mbedtls_mpi *_RR )
Paul Bakker5121ce52009-01-03 21:22:43 +00002132{
Janos Follath24eed8d2019-11-22 13:21:35 +00002133 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002134 size_t wbits, wsize, one = 1;
2135 size_t i, j, nblimbs;
2136 size_t bufsize, nbits;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002137 mbedtls_mpi_uint ei, mm, state;
Daniel Otte388f9b22020-08-21 12:34:29 +02002138 mbedtls_mpi RR, T, W[ 1 << MBEDTLS_MPI_WINDOW_SIZE ], Apos;
Paul Bakkerf6198c12012-05-16 08:02:29 +00002139 int neg;
Paul Bakker5121ce52009-01-03 21:22:43 +00002140
Hanno Becker73d7d792018-12-11 10:35:51 +00002141 MPI_VALIDATE_RET( X != NULL );
2142 MPI_VALIDATE_RET( A != NULL );
2143 MPI_VALIDATE_RET( E != NULL );
2144 MPI_VALIDATE_RET( N != NULL );
2145
Hanno Becker8d1dd1b2017-09-28 11:02:24 +01002146 if( mbedtls_mpi_cmp_int( N, 0 ) <= 0 || ( N->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002147 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002148
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002149 if( mbedtls_mpi_cmp_int( E, 0 ) < 0 )
2150 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002151
2152 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002153 * Init temps and window size
2154 */
2155 mpi_montg_init( &mm, N );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002156 mbedtls_mpi_init( &RR ); mbedtls_mpi_init( &T );
2157 mbedtls_mpi_init( &Apos );
Paul Bakker5121ce52009-01-03 21:22:43 +00002158 memset( W, 0, sizeof( W ) );
2159
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002160 i = mbedtls_mpi_bitlen( E );
Paul Bakker5121ce52009-01-03 21:22:43 +00002161
2162 wsize = ( i > 671 ) ? 6 : ( i > 239 ) ? 5 :
2163 ( i > 79 ) ? 4 : ( i > 23 ) ? 3 : 1;
2164
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002165#if( MBEDTLS_MPI_WINDOW_SIZE < 6 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002166 if( wsize > MBEDTLS_MPI_WINDOW_SIZE )
2167 wsize = MBEDTLS_MPI_WINDOW_SIZE;
Peter Kolbuse6bcad32018-12-11 14:01:44 -06002168#endif
Paul Bakkerb6d5f082011-11-25 11:52:11 +00002169
Paul Bakker5121ce52009-01-03 21:22:43 +00002170 j = N->n + 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002171 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
2172 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[1], j ) );
2173 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T, j * 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002174
2175 /*
Paul Bakker50546922012-05-19 08:40:49 +00002176 * Compensate for negative A (and correct at the end)
2177 */
2178 neg = ( A->s == -1 );
Paul Bakker50546922012-05-19 08:40:49 +00002179 if( neg )
2180 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002181 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Apos, A ) );
Paul Bakker50546922012-05-19 08:40:49 +00002182 Apos.s = 1;
2183 A = &Apos;
2184 }
2185
2186 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002187 * If 1st call, pre-compute R^2 mod N
2188 */
2189 if( _RR == NULL || _RR->p == NULL )
2190 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002191 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &RR, 1 ) );
2192 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &RR, N->n * 2 * biL ) );
2193 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &RR, &RR, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002194
2195 if( _RR != NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002196 memcpy( _RR, &RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002197 }
2198 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002199 memcpy( &RR, _RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002200
2201 /*
2202 * W[1] = A * R^2 * R^-1 mod N = A * R mod N
2203 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002204 if( mbedtls_mpi_cmp_mpi( A, N ) >= 0 )
2205 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &W[1], A, N ) );
Paul Bakkerc2024f42014-01-23 20:38:35 +01002206 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002207 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[1], A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002208
Gilles Peskine4e91d472020-06-04 20:55:15 +02002209 mpi_montmul( &W[1], &RR, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002210
2211 /*
2212 * X = R^2 * R^-1 mod N = R mod N
2213 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002214 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &RR ) );
Gilles Peskine4e91d472020-06-04 20:55:15 +02002215 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002216
2217 if( wsize > 1 )
2218 {
2219 /*
2220 * W[1 << (wsize - 1)] = W[1] ^ (wsize - 1)
2221 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002222 j = one << ( wsize - 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002223
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002224 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[j], N->n + 1 ) );
2225 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[j], &W[1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002226
2227 for( i = 0; i < wsize - 1; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002228 mpi_montmul( &W[j], &W[j], N, mm, &T );
Paul Bakker0d7702c2013-10-29 16:18:35 +01002229
Paul Bakker5121ce52009-01-03 21:22:43 +00002230 /*
2231 * W[i] = W[i - 1] * W[1]
2232 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002233 for( i = j + 1; i < ( one << wsize ); i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002234 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002235 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[i], N->n + 1 ) );
2236 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[i], &W[i - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002237
Gilles Peskine4e91d472020-06-04 20:55:15 +02002238 mpi_montmul( &W[i], &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002239 }
2240 }
2241
2242 nblimbs = E->n;
2243 bufsize = 0;
2244 nbits = 0;
2245 wbits = 0;
2246 state = 0;
2247
2248 while( 1 )
2249 {
2250 if( bufsize == 0 )
2251 {
Paul Bakker0d7702c2013-10-29 16:18:35 +01002252 if( nblimbs == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002253 break;
2254
Paul Bakker0d7702c2013-10-29 16:18:35 +01002255 nblimbs--;
2256
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002257 bufsize = sizeof( mbedtls_mpi_uint ) << 3;
Paul Bakker5121ce52009-01-03 21:22:43 +00002258 }
2259
2260 bufsize--;
2261
2262 ei = (E->p[nblimbs] >> bufsize) & 1;
2263
2264 /*
2265 * skip leading 0s
2266 */
2267 if( ei == 0 && state == 0 )
2268 continue;
2269
2270 if( ei == 0 && state == 1 )
2271 {
2272 /*
2273 * out of window, square X
2274 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002275 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002276 continue;
2277 }
2278
2279 /*
2280 * add ei to current window
2281 */
2282 state = 2;
2283
2284 nbits++;
Paul Bakker66d5d072014-06-17 16:39:18 +02002285 wbits |= ( ei << ( wsize - nbits ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002286
2287 if( nbits == wsize )
2288 {
2289 /*
2290 * X = X^wsize R^-1 mod N
2291 */
2292 for( i = 0; i < wsize; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002293 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002294
2295 /*
2296 * X = X * W[wbits] R^-1 mod N
2297 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002298 mpi_montmul( X, &W[wbits], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002299
2300 state--;
2301 nbits = 0;
2302 wbits = 0;
2303 }
2304 }
2305
2306 /*
2307 * process the remaining bits
2308 */
2309 for( i = 0; i < nbits; i++ )
2310 {
Gilles Peskine4e91d472020-06-04 20:55:15 +02002311 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002312
2313 wbits <<= 1;
2314
Paul Bakker66d5d072014-06-17 16:39:18 +02002315 if( ( wbits & ( one << wsize ) ) != 0 )
Gilles Peskine4e91d472020-06-04 20:55:15 +02002316 mpi_montmul( X, &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002317 }
2318
2319 /*
2320 * X = A^E * R * R^-1 mod N = A^E mod N
2321 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02002322 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002323
Hanno Beckera4af1c42017-04-18 09:07:45 +01002324 if( neg && E->n != 0 && ( E->p[0] & 1 ) != 0 )
Paul Bakkerf6198c12012-05-16 08:02:29 +00002325 {
2326 X->s = -1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002327 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( X, N, X ) );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002328 }
2329
Paul Bakker5121ce52009-01-03 21:22:43 +00002330cleanup:
2331
Paul Bakker66d5d072014-06-17 16:39:18 +02002332 for( i = ( one << ( wsize - 1 ) ); i < ( one << wsize ); i++ )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002333 mbedtls_mpi_free( &W[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002334
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002335 mbedtls_mpi_free( &W[1] ); mbedtls_mpi_free( &T ); mbedtls_mpi_free( &Apos );
Paul Bakker6c591fa2011-05-05 11:49:20 +00002336
Paul Bakker75a28602014-03-31 12:08:17 +02002337 if( _RR == NULL || _RR->p == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002338 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002339
2340 return( ret );
2341}
2342
Paul Bakker5121ce52009-01-03 21:22:43 +00002343/*
2344 * Greatest common divisor: G = gcd(A, B) (HAC 14.54)
2345 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002346int mbedtls_mpi_gcd( mbedtls_mpi *G, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00002347{
Janos Follath24eed8d2019-11-22 13:21:35 +00002348 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00002349 size_t lz, lzt;
Alexander Ke8ad49f2019-08-16 16:16:07 +03002350 mbedtls_mpi TA, TB;
Paul Bakker5121ce52009-01-03 21:22:43 +00002351
Hanno Becker73d7d792018-12-11 10:35:51 +00002352 MPI_VALIDATE_RET( G != NULL );
2353 MPI_VALIDATE_RET( A != NULL );
2354 MPI_VALIDATE_RET( B != NULL );
2355
Alexander Ke8ad49f2019-08-16 16:16:07 +03002356 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002357
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002358 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) );
2359 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002360
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002361 lz = mbedtls_mpi_lsb( &TA );
2362 lzt = mbedtls_mpi_lsb( &TB );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002363
Paul Bakker66d5d072014-06-17 16:39:18 +02002364 if( lzt < lz )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002365 lz = lzt;
2366
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002367 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, lz ) );
2368 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, lz ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002369
Paul Bakker5121ce52009-01-03 21:22:43 +00002370 TA.s = TB.s = 1;
2371
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002372 while( mbedtls_mpi_cmp_int( &TA, 0 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002373 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002374 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, mbedtls_mpi_lsb( &TA ) ) );
2375 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, mbedtls_mpi_lsb( &TB ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002376
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002377 if( mbedtls_mpi_cmp_mpi( &TA, &TB ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002378 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002379 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TA, &TA, &TB ) );
2380 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002381 }
2382 else
2383 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002384 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TB, &TB, &TA ) );
2385 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002386 }
2387 }
2388
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002389 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &TB, lz ) );
2390 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( G, &TB ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002391
2392cleanup:
2393
Alexander Ke8ad49f2019-08-16 16:16:07 +03002394 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002395
2396 return( ret );
2397}
2398
Paul Bakker33dc46b2014-04-30 16:11:39 +02002399/*
2400 * Fill X with size bytes of random.
2401 *
2402 * Use a temporary bytes representation to make sure the result is the same
Paul Bakkerc37b0ac2014-05-01 14:19:23 +02002403 * regardless of the platform endianness (useful when f_rng is actually
Paul Bakker33dc46b2014-04-30 16:11:39 +02002404 * deterministic, eg for tests).
2405 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002406int mbedtls_mpi_fill_random( mbedtls_mpi *X, size_t size,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002407 int (*f_rng)(void *, unsigned char *, size_t),
2408 void *p_rng )
Paul Bakker287781a2011-03-26 13:18:49 +00002409{
Janos Follath24eed8d2019-11-22 13:21:35 +00002410 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker6dab6202019-01-02 16:42:29 +00002411 size_t const limbs = CHARS_TO_LIMBS( size );
Hanno Beckerda1655a2017-10-18 14:21:44 +01002412 size_t const overhead = ( limbs * ciL ) - size;
2413 unsigned char *Xp;
2414
Hanno Becker8ce11a32018-12-19 16:18:52 +00002415 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002416 MPI_VALIDATE_RET( f_rng != NULL );
Paul Bakker33dc46b2014-04-30 16:11:39 +02002417
Hanno Beckerda1655a2017-10-18 14:21:44 +01002418 /* Ensure that target MPI has exactly the necessary number of limbs */
2419 if( X->n != limbs )
2420 {
2421 mbedtls_mpi_free( X );
2422 mbedtls_mpi_init( X );
2423 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
2424 }
2425 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker287781a2011-03-26 13:18:49 +00002426
Hanno Beckerda1655a2017-10-18 14:21:44 +01002427 Xp = (unsigned char*) X->p;
2428 f_rng( p_rng, Xp + overhead, size );
2429
Hanno Becker2be8a552018-10-25 12:40:09 +01002430 mpi_bigendian_to_host( X->p, limbs );
Paul Bakker287781a2011-03-26 13:18:49 +00002431
2432cleanup:
2433 return( ret );
2434}
2435
Paul Bakker5121ce52009-01-03 21:22:43 +00002436/*
2437 * Modular inverse: X = A^-1 mod N (HAC 14.61 / 14.64)
2438 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002439int mbedtls_mpi_inv_mod( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002440{
Janos Follath24eed8d2019-11-22 13:21:35 +00002441 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002442 mbedtls_mpi G, TA, TU, U1, U2, TB, TV, V1, V2;
Hanno Becker73d7d792018-12-11 10:35:51 +00002443 MPI_VALIDATE_RET( X != NULL );
2444 MPI_VALIDATE_RET( A != NULL );
2445 MPI_VALIDATE_RET( N != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00002446
Hanno Becker4bcb4912017-04-18 15:49:39 +01002447 if( mbedtls_mpi_cmp_int( N, 1 ) <= 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002448 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002449
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002450 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TU ); mbedtls_mpi_init( &U1 ); mbedtls_mpi_init( &U2 );
2451 mbedtls_mpi_init( &G ); mbedtls_mpi_init( &TB ); mbedtls_mpi_init( &TV );
2452 mbedtls_mpi_init( &V1 ); mbedtls_mpi_init( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002453
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002454 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &G, A, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002455
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002456 if( mbedtls_mpi_cmp_int( &G, 1 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002457 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002458 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002459 goto cleanup;
2460 }
2461
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002462 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &TA, A, N ) );
2463 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TU, &TA ) );
2464 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, N ) );
2465 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TV, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002466
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002467 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U1, 1 ) );
2468 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U2, 0 ) );
2469 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V1, 0 ) );
2470 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002471
2472 do
2473 {
2474 while( ( TU.p[0] & 1 ) == 0 )
2475 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002476 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TU, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002477
2478 if( ( U1.p[0] & 1 ) != 0 || ( U2.p[0] & 1 ) != 0 )
2479 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002480 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &U1, &U1, &TB ) );
2481 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002482 }
2483
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002484 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U1, 1 ) );
2485 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002486 }
2487
2488 while( ( TV.p[0] & 1 ) == 0 )
2489 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002490 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TV, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002491
2492 if( ( V1.p[0] & 1 ) != 0 || ( V2.p[0] & 1 ) != 0 )
2493 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002494 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, &TB ) );
2495 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002496 }
2497
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002498 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V1, 1 ) );
2499 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002500 }
2501
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002502 if( mbedtls_mpi_cmp_mpi( &TU, &TV ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002503 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002504 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TU, &TU, &TV ) );
2505 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U1, &U1, &V1 ) );
2506 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &V2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002507 }
2508 else
2509 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002510 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TV, &TV, &TU ) );
2511 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, &U1 ) );
2512 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &U2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002513 }
2514 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002515 while( mbedtls_mpi_cmp_int( &TU, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002516
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002517 while( mbedtls_mpi_cmp_int( &V1, 0 ) < 0 )
2518 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002519
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002520 while( mbedtls_mpi_cmp_mpi( &V1, N ) >= 0 )
2521 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002522
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002523 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &V1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002524
2525cleanup:
2526
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002527 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TU ); mbedtls_mpi_free( &U1 ); mbedtls_mpi_free( &U2 );
2528 mbedtls_mpi_free( &G ); mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TV );
2529 mbedtls_mpi_free( &V1 ); mbedtls_mpi_free( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002530
2531 return( ret );
2532}
2533
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002534#if defined(MBEDTLS_GENPRIME)
Paul Bakkerd9374b02012-11-02 11:02:58 +00002535
Paul Bakker5121ce52009-01-03 21:22:43 +00002536static const int small_prime[] =
2537{
2538 3, 5, 7, 11, 13, 17, 19, 23,
2539 29, 31, 37, 41, 43, 47, 53, 59,
2540 61, 67, 71, 73, 79, 83, 89, 97,
2541 101, 103, 107, 109, 113, 127, 131, 137,
2542 139, 149, 151, 157, 163, 167, 173, 179,
2543 181, 191, 193, 197, 199, 211, 223, 227,
2544 229, 233, 239, 241, 251, 257, 263, 269,
2545 271, 277, 281, 283, 293, 307, 311, 313,
2546 317, 331, 337, 347, 349, 353, 359, 367,
2547 373, 379, 383, 389, 397, 401, 409, 419,
2548 421, 431, 433, 439, 443, 449, 457, 461,
2549 463, 467, 479, 487, 491, 499, 503, 509,
2550 521, 523, 541, 547, 557, 563, 569, 571,
2551 577, 587, 593, 599, 601, 607, 613, 617,
2552 619, 631, 641, 643, 647, 653, 659, 661,
2553 673, 677, 683, 691, 701, 709, 719, 727,
2554 733, 739, 743, 751, 757, 761, 769, 773,
2555 787, 797, 809, 811, 821, 823, 827, 829,
2556 839, 853, 857, 859, 863, 877, 881, 883,
2557 887, 907, 911, 919, 929, 937, 941, 947,
2558 953, 967, 971, 977, 983, 991, 997, -103
2559};
2560
2561/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002562 * Small divisors test (X must be positive)
2563 *
2564 * Return values:
2565 * 0: no small factor (possible prime, more tests needed)
2566 * 1: certain prime
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002567 * MBEDTLS_ERR_MPI_NOT_ACCEPTABLE: certain non-prime
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002568 * other negative: error
Paul Bakker5121ce52009-01-03 21:22:43 +00002569 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002570static int mpi_check_small_factors( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +00002571{
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002572 int ret = 0;
2573 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002574 mbedtls_mpi_uint r;
Paul Bakker5121ce52009-01-03 21:22:43 +00002575
Paul Bakker5121ce52009-01-03 21:22:43 +00002576 if( ( X->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002577 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002578
2579 for( i = 0; small_prime[i] > 0; i++ )
2580 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002581 if( mbedtls_mpi_cmp_int( X, small_prime[i] ) <= 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002582 return( 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002583
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002584 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, small_prime[i] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002585
2586 if( r == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002587 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002588 }
2589
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002590cleanup:
2591 return( ret );
2592}
2593
2594/*
2595 * Miller-Rabin pseudo-primality test (HAC 4.24)
2596 */
Janos Follathda31fa12018-09-03 14:45:23 +01002597static int mpi_miller_rabin( const mbedtls_mpi *X, size_t rounds,
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002598 int (*f_rng)(void *, unsigned char *, size_t),
2599 void *p_rng )
2600{
Pascal Junodb99183d2015-03-11 16:49:45 +01002601 int ret, count;
Janos Follathda31fa12018-09-03 14:45:23 +01002602 size_t i, j, k, s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002603 mbedtls_mpi W, R, T, A, RR;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002604
Hanno Becker8ce11a32018-12-19 16:18:52 +00002605 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002606 MPI_VALIDATE_RET( f_rng != NULL );
2607
2608 mbedtls_mpi_init( &W ); mbedtls_mpi_init( &R );
2609 mbedtls_mpi_init( &T ); mbedtls_mpi_init( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002610 mbedtls_mpi_init( &RR );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002611
Paul Bakker5121ce52009-01-03 21:22:43 +00002612 /*
2613 * W = |X| - 1
2614 * R = W >> lsb( W )
2615 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002616 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( &W, X, 1 ) );
2617 s = mbedtls_mpi_lsb( &W );
2618 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &R, &W ) );
2619 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &R, s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002620
Janos Follathda31fa12018-09-03 14:45:23 +01002621 for( i = 0; i < rounds; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002622 {
2623 /*
2624 * pick a random A, 1 < A < |X| - 1
2625 */
Pascal Junodb99183d2015-03-11 16:49:45 +01002626 count = 0;
2627 do {
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002628 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( &A, X->n * ciL, f_rng, p_rng ) );
Pascal Junodb99183d2015-03-11 16:49:45 +01002629
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002630 j = mbedtls_mpi_bitlen( &A );
2631 k = mbedtls_mpi_bitlen( &W );
Pascal Junodb99183d2015-03-11 16:49:45 +01002632 if (j > k) {
Darryl Greene3f95ed2018-10-02 13:21:35 +01002633 A.p[A.n - 1] &= ( (mbedtls_mpi_uint) 1 << ( k - ( A.n - 1 ) * biL - 1 ) ) - 1;
Pascal Junodb99183d2015-03-11 16:49:45 +01002634 }
2635
2636 if (count++ > 30) {
Jens Wiklanderf08aa3e2019-01-17 13:30:57 +01002637 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
2638 goto cleanup;
Pascal Junodb99183d2015-03-11 16:49:45 +01002639 }
2640
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002641 } while ( mbedtls_mpi_cmp_mpi( &A, &W ) >= 0 ||
2642 mbedtls_mpi_cmp_int( &A, 1 ) <= 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002643
2644 /*
2645 * A = A^R mod |X|
2646 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002647 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &A, &A, &R, X, &RR ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002648
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002649 if( mbedtls_mpi_cmp_mpi( &A, &W ) == 0 ||
2650 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002651 continue;
2652
2653 j = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002654 while( j < s && mbedtls_mpi_cmp_mpi( &A, &W ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002655 {
2656 /*
2657 * A = A * A mod |X|
2658 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002659 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &T, &A, &A ) );
2660 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &A, &T, X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002661
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002662 if( mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002663 break;
2664
2665 j++;
2666 }
2667
2668 /*
2669 * not prime if A != |X| - 1 or A == 1
2670 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002671 if( mbedtls_mpi_cmp_mpi( &A, &W ) != 0 ||
2672 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002673 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002674 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002675 break;
2676 }
2677 }
2678
2679cleanup:
Hanno Becker73d7d792018-12-11 10:35:51 +00002680 mbedtls_mpi_free( &W ); mbedtls_mpi_free( &R );
2681 mbedtls_mpi_free( &T ); mbedtls_mpi_free( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002682 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002683
2684 return( ret );
2685}
2686
2687/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002688 * Pseudo-primality test: small factors, then Miller-Rabin
2689 */
Janos Follatha0b67c22018-09-18 14:48:23 +01002690int mbedtls_mpi_is_prime_ext( const mbedtls_mpi *X, int rounds,
2691 int (*f_rng)(void *, unsigned char *, size_t),
2692 void *p_rng )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002693{
Janos Follath24eed8d2019-11-22 13:21:35 +00002694 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002695 mbedtls_mpi XX;
Hanno Becker8ce11a32018-12-19 16:18:52 +00002696 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002697 MPI_VALIDATE_RET( f_rng != NULL );
Manuel Pégourié-Gonnard7f4ed672014-10-14 20:56:02 +02002698
2699 XX.s = 1;
2700 XX.n = X->n;
2701 XX.p = X->p;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002702
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002703 if( mbedtls_mpi_cmp_int( &XX, 0 ) == 0 ||
2704 mbedtls_mpi_cmp_int( &XX, 1 ) == 0 )
2705 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002706
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002707 if( mbedtls_mpi_cmp_int( &XX, 2 ) == 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002708 return( 0 );
2709
2710 if( ( ret = mpi_check_small_factors( &XX ) ) != 0 )
2711 {
2712 if( ret == 1 )
2713 return( 0 );
2714
2715 return( ret );
2716 }
2717
Janos Follathda31fa12018-09-03 14:45:23 +01002718 return( mpi_miller_rabin( &XX, rounds, f_rng, p_rng ) );
Janos Follathf301d232018-08-14 13:34:01 +01002719}
2720
Janos Follatha0b67c22018-09-18 14:48:23 +01002721#if !defined(MBEDTLS_DEPRECATED_REMOVED)
Janos Follathf301d232018-08-14 13:34:01 +01002722/*
2723 * Pseudo-primality test, error probability 2^-80
2724 */
2725int mbedtls_mpi_is_prime( const mbedtls_mpi *X,
2726 int (*f_rng)(void *, unsigned char *, size_t),
2727 void *p_rng )
2728{
Hanno Becker8ce11a32018-12-19 16:18:52 +00002729 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002730 MPI_VALIDATE_RET( f_rng != NULL );
2731
Janos Follatha0b67c22018-09-18 14:48:23 +01002732 /*
2733 * In the past our key generation aimed for an error rate of at most
2734 * 2^-80. Since this function is deprecated, aim for the same certainty
2735 * here as well.
2736 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002737 return( mbedtls_mpi_is_prime_ext( X, 40, f_rng, p_rng ) );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002738}
Janos Follatha0b67c22018-09-18 14:48:23 +01002739#endif
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002740
2741/*
Paul Bakker5121ce52009-01-03 21:22:43 +00002742 * Prime number generation
Jethro Beekman66689272018-02-14 19:24:10 -08002743 *
Janos Follathf301d232018-08-14 13:34:01 +01002744 * To generate an RSA key in a way recommended by FIPS 186-4, both primes must
2745 * be either 1024 bits or 1536 bits long, and flags must contain
2746 * MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR.
Paul Bakker5121ce52009-01-03 21:22:43 +00002747 */
Janos Follath7c025a92018-08-14 11:08:41 +01002748int mbedtls_mpi_gen_prime( mbedtls_mpi *X, size_t nbits, int flags,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002749 int (*f_rng)(void *, unsigned char *, size_t),
2750 void *p_rng )
Paul Bakker5121ce52009-01-03 21:22:43 +00002751{
Jethro Beekman66689272018-02-14 19:24:10 -08002752#ifdef MBEDTLS_HAVE_INT64
2753// ceil(2^63.5)
2754#define CEIL_MAXUINT_DIV_SQRT2 0xb504f333f9de6485ULL
2755#else
2756// ceil(2^31.5)
2757#define CEIL_MAXUINT_DIV_SQRT2 0xb504f334U
2758#endif
2759 int ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker23986e52011-04-24 08:57:21 +00002760 size_t k, n;
Janos Follathda31fa12018-09-03 14:45:23 +01002761 int rounds;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002762 mbedtls_mpi_uint r;
2763 mbedtls_mpi Y;
Paul Bakker5121ce52009-01-03 21:22:43 +00002764
Hanno Becker8ce11a32018-12-19 16:18:52 +00002765 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002766 MPI_VALIDATE_RET( f_rng != NULL );
2767
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002768 if( nbits < 3 || nbits > MBEDTLS_MPI_MAX_BITS )
2769 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002770
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002771 mbedtls_mpi_init( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002772
2773 n = BITS_TO_LIMBS( nbits );
2774
Janos Follathda31fa12018-09-03 14:45:23 +01002775 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR ) == 0 )
2776 {
2777 /*
2778 * 2^-80 error probability, number of rounds chosen per HAC, table 4.4
2779 */
2780 rounds = ( ( nbits >= 1300 ) ? 2 : ( nbits >= 850 ) ? 3 :
2781 ( nbits >= 650 ) ? 4 : ( nbits >= 350 ) ? 8 :
2782 ( nbits >= 250 ) ? 12 : ( nbits >= 150 ) ? 18 : 27 );
2783 }
2784 else
2785 {
2786 /*
2787 * 2^-100 error probability, number of rounds computed based on HAC,
2788 * fact 4.48
2789 */
2790 rounds = ( ( nbits >= 1450 ) ? 4 : ( nbits >= 1150 ) ? 5 :
2791 ( nbits >= 1000 ) ? 6 : ( nbits >= 850 ) ? 7 :
2792 ( nbits >= 750 ) ? 8 : ( nbits >= 500 ) ? 13 :
2793 ( nbits >= 250 ) ? 28 : ( nbits >= 150 ) ? 40 : 51 );
2794 }
2795
Jethro Beekman66689272018-02-14 19:24:10 -08002796 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002797 {
Jethro Beekman66689272018-02-14 19:24:10 -08002798 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( X, n * ciL, f_rng, p_rng ) );
2799 /* make sure generated number is at least (nbits-1)+0.5 bits (FIPS 186-4 §B.3.3 steps 4.4, 5.5) */
2800 if( X->p[n-1] < CEIL_MAXUINT_DIV_SQRT2 ) continue;
2801
2802 k = n * biL;
2803 if( k > nbits ) MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( X, k - nbits ) );
2804 X->p[0] |= 1;
2805
Janos Follath7c025a92018-08-14 11:08:41 +01002806 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_DH ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002807 {
Janos Follatha0b67c22018-09-18 14:48:23 +01002808 ret = mbedtls_mpi_is_prime_ext( X, rounds, f_rng, p_rng );
Jethro Beekman66689272018-02-14 19:24:10 -08002809
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002810 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
Paul Bakker5121ce52009-01-03 21:22:43 +00002811 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002812 }
Jethro Beekman66689272018-02-14 19:24:10 -08002813 else
Paul Bakker5121ce52009-01-03 21:22:43 +00002814 {
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002815 /*
Jethro Beekman66689272018-02-14 19:24:10 -08002816 * An necessary condition for Y and X = 2Y + 1 to be prime
2817 * is X = 2 mod 3 (which is equivalent to Y = 2 mod 3).
2818 * Make sure it is satisfied, while keeping X = 3 mod 4
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002819 */
Jethro Beekman66689272018-02-14 19:24:10 -08002820
2821 X->p[0] |= 2;
2822
2823 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, 3 ) );
2824 if( r == 0 )
2825 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 8 ) );
2826 else if( r == 1 )
2827 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 4 ) );
2828
2829 /* Set Y = (X-1) / 2, which is X / 2 because X is odd */
2830 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, X ) );
2831 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, 1 ) );
2832
2833 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002834 {
Jethro Beekman66689272018-02-14 19:24:10 -08002835 /*
2836 * First, check small factors for X and Y
2837 * before doing Miller-Rabin on any of them
2838 */
2839 if( ( ret = mpi_check_small_factors( X ) ) == 0 &&
2840 ( ret = mpi_check_small_factors( &Y ) ) == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002841 ( ret = mpi_miller_rabin( X, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002842 == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002843 ( ret = mpi_miller_rabin( &Y, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002844 == 0 )
Jethro Beekman66689272018-02-14 19:24:10 -08002845 goto cleanup;
2846
2847 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
2848 goto cleanup;
2849
2850 /*
2851 * Next candidates. We want to preserve Y = (X-1) / 2 and
2852 * Y = 1 mod 2 and Y = 2 mod 3 (eq X = 3 mod 4 and X = 2 mod 3)
2853 * so up Y by 6 and X by 12.
2854 */
2855 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 12 ) );
2856 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( &Y, &Y, 6 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002857 }
Paul Bakker5121ce52009-01-03 21:22:43 +00002858 }
2859 }
2860
2861cleanup:
2862
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002863 mbedtls_mpi_free( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002864
2865 return( ret );
2866}
2867
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002868#endif /* MBEDTLS_GENPRIME */
Paul Bakker5121ce52009-01-03 21:22:43 +00002869
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002870#if defined(MBEDTLS_SELF_TEST)
Paul Bakker5121ce52009-01-03 21:22:43 +00002871
Paul Bakker23986e52011-04-24 08:57:21 +00002872#define GCD_PAIR_COUNT 3
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002873
2874static const int gcd_pairs[GCD_PAIR_COUNT][3] =
2875{
2876 { 693, 609, 21 },
2877 { 1764, 868, 28 },
2878 { 768454923, 542167814, 1 }
2879};
2880
Paul Bakker5121ce52009-01-03 21:22:43 +00002881/*
2882 * Checkup routine
2883 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002884int mbedtls_mpi_self_test( int verbose )
Paul Bakker5121ce52009-01-03 21:22:43 +00002885{
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002886 int ret, i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002887 mbedtls_mpi A, E, N, X, Y, U, V;
Paul Bakker5121ce52009-01-03 21:22:43 +00002888
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002889 mbedtls_mpi_init( &A ); mbedtls_mpi_init( &E ); mbedtls_mpi_init( &N ); mbedtls_mpi_init( &X );
2890 mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &U ); mbedtls_mpi_init( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002891
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002892 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &A, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002893 "EFE021C2645FD1DC586E69184AF4A31E" \
2894 "D5F53E93B5F123FA41680867BA110131" \
2895 "944FE7952E2517337780CB0DB80E61AA" \
2896 "E7C8DDC6C5C6AADEB34EB38A2F40D5E6" ) );
2897
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002898 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &E, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002899 "B2E7EFD37075B9F03FF989C7C5051C20" \
2900 "34D2A323810251127E7BF8625A4F49A5" \
2901 "F3E27F4DA8BD59C47D6DAABA4C8127BD" \
2902 "5B5C25763222FEFCCFC38B832366C29E" ) );
2903
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002904 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &N, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002905 "0066A198186C18C10B2F5ED9B522752A" \
2906 "9830B69916E535C8F047518A889A43A5" \
2907 "94B6BED27A168D31D4A52F88925AA8F5" ) );
2908
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002909 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002910
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002911 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002912 "602AB7ECA597A3D6B56FF9829A5E8B85" \
2913 "9E857EA95A03512E2BAE7391688D264A" \
2914 "A5663B0341DB9CCFD2C4C5F421FEC814" \
2915 "8001B72E848A38CAE1C65F78E56ABDEF" \
2916 "E12D3C039B8A02D6BE593F0BBBDA56F1" \
2917 "ECF677152EF804370C1A305CAF3B5BF1" \
2918 "30879B56C61DE584A0F53A2447A51E" ) );
2919
2920 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002921 mbedtls_printf( " MPI test #1 (mul_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002922
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002923 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002924 {
2925 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002926 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002927
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002928 ret = 1;
2929 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002930 }
2931
2932 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002933 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002934
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002935 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( &X, &Y, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002936
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002937 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002938 "256567336059E52CAE22925474705F39A94" ) );
2939
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002940 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &V, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002941 "6613F26162223DF488E9CD48CC132C7A" \
2942 "0AC93C701B001B092E4E5B9F73BCD27B" \
2943 "9EE50D0657C77F374E903CDFA4C642" ) );
2944
2945 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002946 mbedtls_printf( " MPI test #2 (div_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002947
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002948 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 ||
2949 mbedtls_mpi_cmp_mpi( &Y, &V ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002950 {
2951 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002952 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002953
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002954 ret = 1;
2955 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002956 }
2957
2958 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002959 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002960
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002961 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &X, &A, &E, &N, NULL ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002962
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002963 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002964 "36E139AEA55215609D2816998ED020BB" \
2965 "BD96C37890F65171D948E9BC7CBAA4D9" \
2966 "325D24D6A3C12710F10A09FA08AB87" ) );
2967
2968 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002969 mbedtls_printf( " MPI test #3 (exp_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002970
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002971 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002972 {
2973 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002974 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002975
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002976 ret = 1;
2977 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002978 }
2979
2980 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002981 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002982
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002983 MBEDTLS_MPI_CHK( mbedtls_mpi_inv_mod( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002984
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002985 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002986 "003A0AAEDD7E784FC07D8F9EC6E3BFD5" \
2987 "C3DBA76456363A10869622EAC2DD84EC" \
2988 "C5B8A74DAC4D09E03B5E0BE779F2DF61" ) );
2989
2990 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002991 mbedtls_printf( " MPI test #4 (inv_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002992
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002993 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002994 {
2995 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002996 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002997
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002998 ret = 1;
2999 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00003000 }
3001
3002 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003003 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00003004
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003005 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003006 mbedtls_printf( " MPI test #5 (simple gcd): " );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003007
Paul Bakker66d5d072014-06-17 16:39:18 +02003008 for( i = 0; i < GCD_PAIR_COUNT; i++ )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003009 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003010 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &X, gcd_pairs[i][0] ) );
3011 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Y, gcd_pairs[i][1] ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003012
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003013 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &A, &X, &Y ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003014
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003015 if( mbedtls_mpi_cmp_int( &A, gcd_pairs[i][2] ) != 0 )
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01003016 {
3017 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003018 mbedtls_printf( "failed at %d\n", i );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003019
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01003020 ret = 1;
3021 goto cleanup;
3022 }
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003023 }
3024
3025 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003026 mbedtls_printf( "passed\n" );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00003027
Paul Bakker5121ce52009-01-03 21:22:43 +00003028cleanup:
3029
3030 if( ret != 0 && verbose != 0 )
Kenneth Soerensen518d4352020-04-01 17:22:45 +02003031 mbedtls_printf( "Unexpected error, return code = %08X\n", (unsigned int) ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00003032
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003033 mbedtls_mpi_free( &A ); mbedtls_mpi_free( &E ); mbedtls_mpi_free( &N ); mbedtls_mpi_free( &X );
3034 mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &U ); mbedtls_mpi_free( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00003035
3036 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003037 mbedtls_printf( "\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00003038
3039 return( ret );
3040}
3041
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003042#endif /* MBEDTLS_SELF_TEST */
Paul Bakker5121ce52009-01-03 21:22:43 +00003043
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02003044#endif /* MBEDTLS_BIGNUM_C */