blob: 391ff87769da82e57b626749ef2c29638667bd2a [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"
Hanno Beckeraef9cc42022-04-11 06:36:29 +010041#include "bignum_internal.h"
Janos Follath4614b9a2022-07-21 15:34:47 +010042#include "bignum_core.h"
Chris Jones4c5819c2021-03-03 17:45:34 +000043#include "bn_mul.h"
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050044#include "mbedtls/platform_util.h"
Janos Follath24eed8d2019-11-22 13:21:35 +000045#include "mbedtls/error.h"
Gabor Mezei22c9a6f2021-10-20 12:09:35 +020046#include "constant_time_internal.h"
Paul Bakker5121ce52009-01-03 21:22:43 +000047
Dave Rodgman351c71b2021-12-06 17:50:53 +000048#include <limits.h>
Rich Evans00ab4702015-02-06 13:43:58 +000049#include <string.h>
50
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020051#if defined(MBEDTLS_PLATFORM_C)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000052#include "mbedtls/platform.h"
Paul Bakker6e339b52013-07-03 13:37:05 +020053#else
Rich Evans00ab4702015-02-06 13:43:58 +000054#include <stdio.h>
55#include <stdlib.h>
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020056#define mbedtls_printf printf
Manuel Pégourié-Gonnard7551cb92015-05-26 16:04:06 +020057#define mbedtls_calloc calloc
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020058#define mbedtls_free free
Paul Bakker6e339b52013-07-03 13:37:05 +020059#endif
60
Manuel Pégourié-Gonnard2d708342015-10-05 15:23:11 +010061#define MPI_SIZE_T_MAX ( (size_t) -1 ) /* SIZE_T_MAX is not standard */
62
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050063/* Implementation that should never be optimized out by the compiler */
Andres Amaya Garcia6698d2f2018-04-24 08:39:07 -050064static void mbedtls_mpi_zeroize( mbedtls_mpi_uint *v, size_t n )
65{
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050066 mbedtls_platform_zeroize( v, ciL * n );
67}
68
Paul Bakker5121ce52009-01-03 21:22:43 +000069/*
Paul Bakker6c591fa2011-05-05 11:49:20 +000070 * Initialize one MPI
Paul Bakker5121ce52009-01-03 21:22:43 +000071 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020072void mbedtls_mpi_init( mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +000073{
Hanno Becker73d7d792018-12-11 10:35:51 +000074 MPI_VALIDATE( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +000075
Paul Bakker6c591fa2011-05-05 11:49:20 +000076 X->s = 1;
77 X->n = 0;
78 X->p = NULL;
Paul Bakker5121ce52009-01-03 21:22:43 +000079}
80
81/*
Paul Bakker6c591fa2011-05-05 11:49:20 +000082 * Unallocate one MPI
Paul Bakker5121ce52009-01-03 21:22:43 +000083 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020084void mbedtls_mpi_free( mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +000085{
Paul Bakker6c591fa2011-05-05 11:49:20 +000086 if( X == NULL )
87 return;
Paul Bakker5121ce52009-01-03 21:22:43 +000088
Paul Bakker6c591fa2011-05-05 11:49:20 +000089 if( X->p != NULL )
Paul Bakker5121ce52009-01-03 21:22:43 +000090 {
Alexey Skalozube17a8da2016-01-13 17:19:33 +020091 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020092 mbedtls_free( X->p );
Paul Bakker5121ce52009-01-03 21:22:43 +000093 }
94
Paul Bakker6c591fa2011-05-05 11:49:20 +000095 X->s = 1;
96 X->n = 0;
97 X->p = NULL;
Paul Bakker5121ce52009-01-03 21:22:43 +000098}
99
100/*
101 * Enlarge to the specified number of limbs
102 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200103int mbedtls_mpi_grow( mbedtls_mpi *X, size_t nblimbs )
Paul Bakker5121ce52009-01-03 21:22:43 +0000104{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200105 mbedtls_mpi_uint *p;
Hanno Becker73d7d792018-12-11 10:35:51 +0000106 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000107
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200108 if( nblimbs > MBEDTLS_MPI_MAX_LIMBS )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200109 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Paul Bakkerf9688572011-05-05 10:00:45 +0000110
Paul Bakker5121ce52009-01-03 21:22:43 +0000111 if( X->n < nblimbs )
112 {
Simon Butcher29176892016-05-20 00:19:09 +0100113 if( ( p = (mbedtls_mpi_uint*)mbedtls_calloc( nblimbs, ciL ) ) == NULL )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200114 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Paul Bakker5121ce52009-01-03 21:22:43 +0000115
Paul Bakker5121ce52009-01-03 21:22:43 +0000116 if( X->p != NULL )
117 {
118 memcpy( p, X->p, X->n * ciL );
Alexey Skalozube17a8da2016-01-13 17:19:33 +0200119 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200120 mbedtls_free( X->p );
Paul Bakker5121ce52009-01-03 21:22:43 +0000121 }
122
123 X->n = nblimbs;
124 X->p = p;
125 }
126
127 return( 0 );
128}
129
130/*
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100131 * Resize down as much as possible,
132 * while keeping at least the specified number of limbs
133 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200134int mbedtls_mpi_shrink( mbedtls_mpi *X, size_t nblimbs )
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100135{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200136 mbedtls_mpi_uint *p;
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100137 size_t i;
Hanno Becker73d7d792018-12-11 10:35:51 +0000138 MPI_VALIDATE_RET( X != NULL );
139
140 if( nblimbs > MBEDTLS_MPI_MAX_LIMBS )
141 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100142
Gilles Peskinee2f563e2020-01-20 21:17:43 +0100143 /* Actually resize up if there are currently fewer than nblimbs limbs. */
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100144 if( X->n <= nblimbs )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200145 return( mbedtls_mpi_grow( X, nblimbs ) );
Gilles Peskine322752b2020-01-21 13:59:51 +0100146 /* After this point, then X->n > nblimbs and in particular X->n > 0. */
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100147
148 for( i = X->n - 1; i > 0; i-- )
149 if( X->p[i] != 0 )
150 break;
151 i++;
152
153 if( i < nblimbs )
154 i = nblimbs;
155
Simon Butcher29176892016-05-20 00:19:09 +0100156 if( ( p = (mbedtls_mpi_uint*)mbedtls_calloc( i, ciL ) ) == NULL )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200157 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100158
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100159 if( X->p != NULL )
160 {
161 memcpy( p, X->p, i * ciL );
Alexey Skalozube17a8da2016-01-13 17:19:33 +0200162 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200163 mbedtls_free( X->p );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100164 }
165
166 X->n = i;
167 X->p = p;
168
169 return( 0 );
170}
171
Gilles Peskineed32b572021-06-02 22:17:52 +0200172/* Resize X to have exactly n limbs and set it to 0. */
173static int mbedtls_mpi_resize_clear( mbedtls_mpi *X, size_t limbs )
174{
175 if( limbs == 0 )
176 {
177 mbedtls_mpi_free( X );
178 return( 0 );
179 }
180 else if( X->n == limbs )
181 {
182 memset( X->p, 0, limbs * ciL );
183 X->s = 1;
184 return( 0 );
185 }
186 else
187 {
188 mbedtls_mpi_free( X );
189 return( mbedtls_mpi_grow( X, limbs ) );
190 }
191}
192
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100193/*
Gilles Peskine3da1a8f2021-06-08 23:17:42 +0200194 * Copy the contents of Y into X.
195 *
196 * This function is not constant-time. Leading zeros in Y may be removed.
197 *
198 * Ensure that X does not shrink. This is not guaranteed by the public API,
199 * but some code in the bignum module relies on this property, for example
200 * in mbedtls_mpi_exp_mod().
Paul Bakker5121ce52009-01-03 21:22:43 +0000201 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200202int mbedtls_mpi_copy( mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +0000203{
Gilles Peskine4e4be7c2018-03-21 16:29:03 +0100204 int ret = 0;
Paul Bakker23986e52011-04-24 08:57:21 +0000205 size_t i;
Hanno Becker73d7d792018-12-11 10:35:51 +0000206 MPI_VALIDATE_RET( X != NULL );
207 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000208
209 if( X == Y )
210 return( 0 );
211
Gilles Peskinedb420622020-01-20 21:12:50 +0100212 if( Y->n == 0 )
Manuel Pégourié-Gonnardf4999932013-08-12 17:02:59 +0200213 {
Gilles Peskine3da1a8f2021-06-08 23:17:42 +0200214 if( X->n != 0 )
215 {
216 X->s = 1;
217 memset( X->p, 0, X->n * ciL );
218 }
Manuel Pégourié-Gonnardf4999932013-08-12 17:02:59 +0200219 return( 0 );
220 }
221
Paul Bakker5121ce52009-01-03 21:22:43 +0000222 for( i = Y->n - 1; i > 0; i-- )
223 if( Y->p[i] != 0 )
224 break;
225 i++;
226
227 X->s = Y->s;
228
Gilles Peskine4e4be7c2018-03-21 16:29:03 +0100229 if( X->n < i )
230 {
231 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i ) );
232 }
233 else
234 {
235 memset( X->p + i, 0, ( X->n - i ) * ciL );
236 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000237
Paul Bakker5121ce52009-01-03 21:22:43 +0000238 memcpy( X->p, Y->p, i * ciL );
239
240cleanup:
241
242 return( ret );
243}
244
245/*
246 * Swap the contents of X and Y
247 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200248void mbedtls_mpi_swap( mbedtls_mpi *X, mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +0000249{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200250 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000251 MPI_VALIDATE( X != NULL );
252 MPI_VALIDATE( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000253
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200254 memcpy( &T, X, sizeof( mbedtls_mpi ) );
255 memcpy( X, Y, sizeof( mbedtls_mpi ) );
256 memcpy( Y, &T, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000257}
258
259/*
260 * Set value from integer
261 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200262int mbedtls_mpi_lset( mbedtls_mpi *X, mbedtls_mpi_sint z )
Paul Bakker5121ce52009-01-03 21:22:43 +0000263{
Janos Follath24eed8d2019-11-22 13:21:35 +0000264 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker73d7d792018-12-11 10:35:51 +0000265 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000266
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200267 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000268 memset( X->p, 0, X->n * ciL );
269
270 X->p[0] = ( z < 0 ) ? -z : z;
271 X->s = ( z < 0 ) ? -1 : 1;
272
273cleanup:
274
275 return( ret );
276}
277
278/*
Paul Bakker2f5947e2011-05-18 15:47:11 +0000279 * Get a specific bit
280 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200281int mbedtls_mpi_get_bit( const mbedtls_mpi *X, size_t pos )
Paul Bakker2f5947e2011-05-18 15:47:11 +0000282{
Hanno Becker73d7d792018-12-11 10:35:51 +0000283 MPI_VALIDATE_RET( X != NULL );
284
Paul Bakker2f5947e2011-05-18 15:47:11 +0000285 if( X->n * biL <= pos )
286 return( 0 );
287
Paul Bakkerd8bb8262014-06-17 14:06:49 +0200288 return( ( X->p[pos / biL] >> ( pos % biL ) ) & 0x01 );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000289}
290
Gilles Peskine11cdb052018-11-20 16:47:47 +0100291/* Get a specific byte, without range checks. */
292#define GET_BYTE( X, i ) \
293 ( ( ( X )->p[( i ) / ciL] >> ( ( ( i ) % ciL ) * 8 ) ) & 0xff )
294
Paul Bakker2f5947e2011-05-18 15:47:11 +0000295/*
296 * Set a bit to a specific value of 0 or 1
297 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200298int mbedtls_mpi_set_bit( mbedtls_mpi *X, size_t pos, unsigned char val )
Paul Bakker2f5947e2011-05-18 15:47:11 +0000299{
300 int ret = 0;
301 size_t off = pos / biL;
302 size_t idx = pos % biL;
Hanno Becker73d7d792018-12-11 10:35:51 +0000303 MPI_VALIDATE_RET( X != NULL );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000304
305 if( val != 0 && val != 1 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200306 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker9af723c2014-05-01 13:03:14 +0200307
Paul Bakker2f5947e2011-05-18 15:47:11 +0000308 if( X->n * biL <= pos )
309 {
310 if( val == 0 )
Paul Bakkerd8bb8262014-06-17 14:06:49 +0200311 return( 0 );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000312
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200313 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, off + 1 ) );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000314 }
315
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200316 X->p[off] &= ~( (mbedtls_mpi_uint) 0x01 << idx );
317 X->p[off] |= (mbedtls_mpi_uint) val << idx;
Paul Bakker2f5947e2011-05-18 15:47:11 +0000318
319cleanup:
Paul Bakker9af723c2014-05-01 13:03:14 +0200320
Paul Bakker2f5947e2011-05-18 15:47:11 +0000321 return( ret );
322}
323
324/*
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200325 * Return the number of less significant zero-bits
Paul Bakker5121ce52009-01-03 21:22:43 +0000326 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200327size_t mbedtls_mpi_lsb( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000328{
Paul Bakker23986e52011-04-24 08:57:21 +0000329 size_t i, j, count = 0;
Hanno Beckerf25ee7f2018-12-19 16:51:02 +0000330 MBEDTLS_INTERNAL_VALIDATE_RET( X != NULL, 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000331
332 for( i = 0; i < X->n; i++ )
Paul Bakkerf9688572011-05-05 10:00:45 +0000333 for( j = 0; j < biL; j++, count++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000334 if( ( ( X->p[i] >> j ) & 1 ) != 0 )
335 return( count );
336
337 return( 0 );
338}
339
340/*
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200341 * Return the number of bits
Paul Bakker5121ce52009-01-03 21:22:43 +0000342 */
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200343size_t mbedtls_mpi_bitlen( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000344{
Janos Follath4670f882022-07-21 18:25:42 +0100345 return mbedtls_mpi_core_bitlen( X->p, X->n );
Paul Bakker5121ce52009-01-03 21:22:43 +0000346}
347
348/*
349 * Return the total size in bytes
350 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200351size_t mbedtls_mpi_size( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000352{
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200353 return( ( mbedtls_mpi_bitlen( X ) + 7 ) >> 3 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000354}
355
356/*
357 * Convert an ASCII character to digit value
358 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200359static int mpi_get_digit( mbedtls_mpi_uint *d, int radix, char c )
Paul Bakker5121ce52009-01-03 21:22:43 +0000360{
361 *d = 255;
362
363 if( c >= 0x30 && c <= 0x39 ) *d = c - 0x30;
364 if( c >= 0x41 && c <= 0x46 ) *d = c - 0x37;
365 if( c >= 0x61 && c <= 0x66 ) *d = c - 0x57;
366
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200367 if( *d >= (mbedtls_mpi_uint) radix )
368 return( MBEDTLS_ERR_MPI_INVALID_CHARACTER );
Paul Bakker5121ce52009-01-03 21:22:43 +0000369
370 return( 0 );
371}
372
373/*
374 * Import from an ASCII string
375 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200376int mbedtls_mpi_read_string( mbedtls_mpi *X, int radix, const char *s )
Paul Bakker5121ce52009-01-03 21:22:43 +0000377{
Janos Follath24eed8d2019-11-22 13:21:35 +0000378 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +0000379 size_t i, j, slen, n;
Gilles Peskine80f56732021-04-03 18:26:13 +0200380 int sign = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200381 mbedtls_mpi_uint d;
382 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000383 MPI_VALIDATE_RET( X != NULL );
384 MPI_VALIDATE_RET( s != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000385
386 if( radix < 2 || radix > 16 )
Hanno Becker54c91dd2018-12-12 13:37:06 +0000387 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000388
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200389 mbedtls_mpi_init( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000390
Gilles Peskine7cba8592021-06-08 18:32:34 +0200391 if( s[0] == 0 )
392 {
393 mbedtls_mpi_free( X );
394 return( 0 );
395 }
396
Gilles Peskine80f56732021-04-03 18:26:13 +0200397 if( s[0] == '-' )
398 {
399 ++s;
400 sign = -1;
401 }
402
Paul Bakkerff60ee62010-03-16 21:09:09 +0000403 slen = strlen( s );
404
Paul Bakker5121ce52009-01-03 21:22:43 +0000405 if( radix == 16 )
406 {
Manuel Pégourié-Gonnard2d708342015-10-05 15:23:11 +0100407 if( slen > MPI_SIZE_T_MAX >> 2 )
Manuel Pégourié-Gonnard58fb4952015-09-28 13:48:04 +0200408 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
409
Paul Bakkerff60ee62010-03-16 21:09:09 +0000410 n = BITS_TO_LIMBS( slen << 2 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000411
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200412 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, n ) );
413 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000414
Paul Bakker23986e52011-04-24 08:57:21 +0000415 for( i = slen, j = 0; i > 0; i--, j++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000416 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200417 MBEDTLS_MPI_CHK( mpi_get_digit( &d, radix, s[i - 1] ) );
Paul Bakker66d5d072014-06-17 16:39:18 +0200418 X->p[j / ( 2 * ciL )] |= d << ( ( j % ( 2 * ciL ) ) << 2 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000419 }
420 }
421 else
422 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200423 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000424
Paul Bakkerff60ee62010-03-16 21:09:09 +0000425 for( i = 0; i < slen; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000426 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200427 MBEDTLS_MPI_CHK( mpi_get_digit( &d, radix, s[i] ) );
428 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T, X, radix ) );
Gilles Peskine80f56732021-04-03 18:26:13 +0200429 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, &T, d ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000430 }
431 }
432
Gilles Peskine80f56732021-04-03 18:26:13 +0200433 if( sign < 0 && mbedtls_mpi_bitlen( X ) != 0 )
434 X->s = -1;
435
Paul Bakker5121ce52009-01-03 21:22:43 +0000436cleanup:
437
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200438 mbedtls_mpi_free( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000439
440 return( ret );
441}
442
443/*
Ron Eldora16fa292018-11-20 14:07:01 +0200444 * Helper to write the digits high-order first.
Paul Bakker5121ce52009-01-03 21:22:43 +0000445 */
Ron Eldora16fa292018-11-20 14:07:01 +0200446static int mpi_write_hlp( mbedtls_mpi *X, int radix,
447 char **p, const size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000448{
Janos Follath24eed8d2019-11-22 13:21:35 +0000449 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200450 mbedtls_mpi_uint r;
Ron Eldora16fa292018-11-20 14:07:01 +0200451 size_t length = 0;
452 char *p_end = *p + buflen;
Paul Bakker5121ce52009-01-03 21:22:43 +0000453
Ron Eldora16fa292018-11-20 14:07:01 +0200454 do
455 {
456 if( length >= buflen )
457 {
458 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
459 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000460
Ron Eldora16fa292018-11-20 14:07:01 +0200461 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, radix ) );
462 MBEDTLS_MPI_CHK( mbedtls_mpi_div_int( X, NULL, X, radix ) );
463 /*
464 * Write the residue in the current position, as an ASCII character.
465 */
466 if( r < 0xA )
467 *(--p_end) = (char)( '0' + r );
468 else
469 *(--p_end) = (char)( 'A' + ( r - 0xA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000470
Ron Eldora16fa292018-11-20 14:07:01 +0200471 length++;
472 } while( mbedtls_mpi_cmp_int( X, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000473
Ron Eldora16fa292018-11-20 14:07:01 +0200474 memmove( *p, p_end, length );
475 *p += length;
Paul Bakker5121ce52009-01-03 21:22:43 +0000476
477cleanup:
478
479 return( ret );
480}
481
482/*
483 * Export into an ASCII string
484 */
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100485int mbedtls_mpi_write_string( const mbedtls_mpi *X, int radix,
486 char *buf, size_t buflen, size_t *olen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000487{
Paul Bakker23986e52011-04-24 08:57:21 +0000488 int ret = 0;
489 size_t n;
Paul Bakker5121ce52009-01-03 21:22:43 +0000490 char *p;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200491 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000492 MPI_VALIDATE_RET( X != NULL );
493 MPI_VALIDATE_RET( olen != NULL );
494 MPI_VALIDATE_RET( buflen == 0 || buf != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000495
496 if( radix < 2 || radix > 16 )
Hanno Becker54c91dd2018-12-12 13:37:06 +0000497 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000498
Hanno Becker23cfea02019-02-04 09:45:07 +0000499 n = mbedtls_mpi_bitlen( X ); /* Number of bits necessary to present `n`. */
500 if( radix >= 4 ) n >>= 1; /* Number of 4-adic digits necessary to present
501 * `n`. If radix > 4, this might be a strict
502 * overapproximation of the number of
503 * radix-adic digits needed to present `n`. */
504 if( radix >= 16 ) n >>= 1; /* Number of hexadecimal digits necessary to
505 * present `n`. */
506
Janos Follath80470622019-03-06 13:43:02 +0000507 n += 1; /* Terminating null byte */
Hanno Becker23cfea02019-02-04 09:45:07 +0000508 n += 1; /* Compensate for the divisions above, which round down `n`
509 * in case it's not even. */
510 n += 1; /* Potential '-'-sign. */
511 n += ( n & 1 ); /* Make n even to have enough space for hexadecimal writing,
512 * which always uses an even number of hex-digits. */
Paul Bakker5121ce52009-01-03 21:22:43 +0000513
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100514 if( buflen < n )
Paul Bakker5121ce52009-01-03 21:22:43 +0000515 {
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100516 *olen = n;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200517 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000518 }
519
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100520 p = buf;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200521 mbedtls_mpi_init( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000522
523 if( X->s == -1 )
Hanno Beckerc983c812019-02-01 16:41:30 +0000524 {
Paul Bakker5121ce52009-01-03 21:22:43 +0000525 *p++ = '-';
Hanno Beckerc983c812019-02-01 16:41:30 +0000526 buflen--;
527 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000528
529 if( radix == 16 )
530 {
Paul Bakker23986e52011-04-24 08:57:21 +0000531 int c;
532 size_t i, j, k;
Paul Bakker5121ce52009-01-03 21:22:43 +0000533
Paul Bakker23986e52011-04-24 08:57:21 +0000534 for( i = X->n, k = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000535 {
Paul Bakker23986e52011-04-24 08:57:21 +0000536 for( j = ciL; j > 0; j-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000537 {
Paul Bakker23986e52011-04-24 08:57:21 +0000538 c = ( X->p[i - 1] >> ( ( j - 1 ) << 3) ) & 0xFF;
Paul Bakker5121ce52009-01-03 21:22:43 +0000539
Paul Bakker6c343d72014-07-10 14:36:19 +0200540 if( c == 0 && k == 0 && ( i + j ) != 2 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000541 continue;
542
Paul Bakker98fe5ea2012-10-24 11:17:48 +0000543 *(p++) = "0123456789ABCDEF" [c / 16];
Paul Bakkerd2c167e2012-10-30 07:49:19 +0000544 *(p++) = "0123456789ABCDEF" [c % 16];
Paul Bakker5121ce52009-01-03 21:22:43 +0000545 k = 1;
546 }
547 }
548 }
549 else
550 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200551 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T, X ) );
Paul Bakkerce40a6d2009-06-23 19:46:08 +0000552
553 if( T.s == -1 )
554 T.s = 1;
555
Ron Eldora16fa292018-11-20 14:07:01 +0200556 MBEDTLS_MPI_CHK( mpi_write_hlp( &T, radix, &p, buflen ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000557 }
558
559 *p++ = '\0';
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100560 *olen = p - buf;
Paul Bakker5121ce52009-01-03 21:22:43 +0000561
562cleanup:
563
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200564 mbedtls_mpi_free( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000565
566 return( ret );
567}
568
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200569#if defined(MBEDTLS_FS_IO)
Paul Bakker5121ce52009-01-03 21:22:43 +0000570/*
571 * Read X from an opened file
572 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200573int mbedtls_mpi_read_file( mbedtls_mpi *X, int radix, FILE *fin )
Paul Bakker5121ce52009-01-03 21:22:43 +0000574{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200575 mbedtls_mpi_uint d;
Paul Bakker23986e52011-04-24 08:57:21 +0000576 size_t slen;
Paul Bakker5121ce52009-01-03 21:22:43 +0000577 char *p;
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000578 /*
Paul Bakkercb37aa52011-11-30 16:00:20 +0000579 * Buffer should have space for (short) label and decimal formatted MPI,
580 * newline characters and '\0'
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000581 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200582 char s[ MBEDTLS_MPI_RW_BUFFER_SIZE ];
Paul Bakker5121ce52009-01-03 21:22:43 +0000583
Hanno Becker73d7d792018-12-11 10:35:51 +0000584 MPI_VALIDATE_RET( X != NULL );
585 MPI_VALIDATE_RET( fin != NULL );
586
587 if( radix < 2 || radix > 16 )
588 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
589
Paul Bakker5121ce52009-01-03 21:22:43 +0000590 memset( s, 0, sizeof( s ) );
591 if( fgets( s, sizeof( s ) - 1, fin ) == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200592 return( MBEDTLS_ERR_MPI_FILE_IO_ERROR );
Paul Bakker5121ce52009-01-03 21:22:43 +0000593
594 slen = strlen( s );
Paul Bakkercb37aa52011-11-30 16:00:20 +0000595 if( slen == sizeof( s ) - 2 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200596 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
Paul Bakkercb37aa52011-11-30 16:00:20 +0000597
Hanno Beckerb2034b72017-04-26 11:46:46 +0100598 if( slen > 0 && s[slen - 1] == '\n' ) { slen--; s[slen] = '\0'; }
599 if( slen > 0 && s[slen - 1] == '\r' ) { slen--; s[slen] = '\0'; }
Paul Bakker5121ce52009-01-03 21:22:43 +0000600
601 p = s + slen;
Hanno Beckerb2034b72017-04-26 11:46:46 +0100602 while( p-- > s )
Paul Bakker5121ce52009-01-03 21:22:43 +0000603 if( mpi_get_digit( &d, radix, *p ) != 0 )
604 break;
605
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200606 return( mbedtls_mpi_read_string( X, radix, p + 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000607}
608
609/*
610 * Write X into an opened file (or stdout if fout == NULL)
611 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200612int mbedtls_mpi_write_file( const char *p, const mbedtls_mpi *X, int radix, FILE *fout )
Paul Bakker5121ce52009-01-03 21:22:43 +0000613{
Janos Follath24eed8d2019-11-22 13:21:35 +0000614 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +0000615 size_t n, slen, plen;
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000616 /*
Paul Bakker5531c6d2012-09-26 19:20:46 +0000617 * Buffer should have space for (short) label and decimal formatted MPI,
618 * newline characters and '\0'
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000619 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200620 char s[ MBEDTLS_MPI_RW_BUFFER_SIZE ];
Hanno Becker73d7d792018-12-11 10:35:51 +0000621 MPI_VALIDATE_RET( X != NULL );
622
623 if( radix < 2 || radix > 16 )
624 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000625
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100626 memset( s, 0, sizeof( s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000627
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100628 MBEDTLS_MPI_CHK( mbedtls_mpi_write_string( X, radix, s, sizeof( s ) - 2, &n ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000629
630 if( p == NULL ) p = "";
631
632 plen = strlen( p );
633 slen = strlen( s );
634 s[slen++] = '\r';
635 s[slen++] = '\n';
636
637 if( fout != NULL )
638 {
639 if( fwrite( p, 1, plen, fout ) != plen ||
640 fwrite( s, 1, slen, fout ) != slen )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200641 return( MBEDTLS_ERR_MPI_FILE_IO_ERROR );
Paul Bakker5121ce52009-01-03 21:22:43 +0000642 }
643 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200644 mbedtls_printf( "%s%s", p, s );
Paul Bakker5121ce52009-01-03 21:22:43 +0000645
646cleanup:
647
648 return( ret );
649}
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200650#endif /* MBEDTLS_FS_IO */
Paul Bakker5121ce52009-01-03 21:22:43 +0000651
652/*
Janos Follatha778a942019-02-13 10:28:28 +0000653 * Import X from unsigned binary data, little endian
Przemyslaw Stekiel76960a72022-02-21 13:42:09 +0100654 *
655 * This function is guaranteed to return an MPI with exactly the necessary
656 * number of limbs (in particular, it does not skip 0s in the input).
Janos Follatha778a942019-02-13 10:28:28 +0000657 */
658int mbedtls_mpi_read_binary_le( mbedtls_mpi *X,
659 const unsigned char *buf, size_t buflen )
660{
Janos Follath24eed8d2019-11-22 13:21:35 +0000661 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Janos Follatha778a942019-02-13 10:28:28 +0000662 size_t const limbs = CHARS_TO_LIMBS( buflen );
663
664 /* Ensure that target MPI has exactly the necessary number of limbs */
Gilles Peskineed32b572021-06-02 22:17:52 +0200665 MBEDTLS_MPI_CHK( mbedtls_mpi_resize_clear( X, limbs ) );
Janos Follatha778a942019-02-13 10:28:28 +0000666
Janos Follath5f016652022-07-22 16:18:41 +0100667 MBEDTLS_MPI_CHK( mbedtls_mpi_core_read_le( X->p, X->n, buf, buflen ) );
Janos Follatha778a942019-02-13 10:28:28 +0000668
669cleanup:
670
Janos Follath171a7ef2019-02-15 16:17:45 +0000671 /*
672 * This function is also used to import keys. However, wiping the buffers
673 * upon failure is not necessary because failure only can happen before any
674 * input is copied.
675 */
Janos Follatha778a942019-02-13 10:28:28 +0000676 return( ret );
677}
678
679/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000680 * Import X from unsigned binary data, big endian
Przemyslaw Stekiel76960a72022-02-21 13:42:09 +0100681 *
682 * This function is guaranteed to return an MPI with exactly the necessary
683 * number of limbs (in particular, it does not skip 0s in the input).
Paul Bakker5121ce52009-01-03 21:22:43 +0000684 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200685int mbedtls_mpi_read_binary( mbedtls_mpi *X, const unsigned char *buf, size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000686{
Janos Follath24eed8d2019-11-22 13:21:35 +0000687 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Janos Follath5f016652022-07-22 16:18:41 +0100688 size_t const limbs = CHARS_TO_LIMBS( buflen );
Paul Bakker5121ce52009-01-03 21:22:43 +0000689
Hanno Becker8ce11a32018-12-19 16:18:52 +0000690 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +0000691 MPI_VALIDATE_RET( buflen == 0 || buf != NULL );
692
Hanno Becker073c1992017-10-17 15:17:27 +0100693 /* Ensure that target MPI has exactly the necessary number of limbs */
Gilles Peskineed32b572021-06-02 22:17:52 +0200694 MBEDTLS_MPI_CHK( mbedtls_mpi_resize_clear( X, limbs ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000695
Janos Follath5f016652022-07-22 16:18:41 +0100696 MBEDTLS_MPI_CHK( mbedtls_mpi_core_read_be( X->p, X->n, buf, buflen ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000697
698cleanup:
699
Janos Follath171a7ef2019-02-15 16:17:45 +0000700 /*
701 * This function is also used to import keys. However, wiping the buffers
702 * upon failure is not necessary because failure only can happen before any
703 * input is copied.
704 */
Paul Bakker5121ce52009-01-03 21:22:43 +0000705 return( ret );
706}
707
708/*
Janos Follathe344d0f2019-02-19 16:17:40 +0000709 * Export X into unsigned binary data, little endian
710 */
711int mbedtls_mpi_write_binary_le( const mbedtls_mpi *X,
712 unsigned char *buf, size_t buflen )
713{
Janos Follath5f016652022-07-22 16:18:41 +0100714 return( mbedtls_mpi_core_write_le( X->p, X->n, buf, buflen) );
Janos Follathe344d0f2019-02-19 16:17:40 +0000715}
716
717/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000718 * Export X into unsigned binary data, big endian
719 */
Gilles Peskine11cdb052018-11-20 16:47:47 +0100720int mbedtls_mpi_write_binary( const mbedtls_mpi *X,
721 unsigned char *buf, size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000722{
Janos Follath5f016652022-07-22 16:18:41 +0100723 return( mbedtls_mpi_core_write_be( X->p, X->n, buf, buflen ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000724}
725
726/*
727 * Left-shift: X <<= count
728 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200729int mbedtls_mpi_shift_l( mbedtls_mpi *X, size_t count )
Paul Bakker5121ce52009-01-03 21:22:43 +0000730{
Janos Follath24eed8d2019-11-22 13:21:35 +0000731 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +0000732 size_t i, v0, t1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200733 mbedtls_mpi_uint r0 = 0, r1;
Hanno Becker73d7d792018-12-11 10:35:51 +0000734 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000735
736 v0 = count / (biL );
737 t1 = count & (biL - 1);
738
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200739 i = mbedtls_mpi_bitlen( X ) + count;
Paul Bakker5121ce52009-01-03 21:22:43 +0000740
Paul Bakkerf9688572011-05-05 10:00:45 +0000741 if( X->n * biL < i )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200742 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, BITS_TO_LIMBS( i ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000743
744 ret = 0;
745
746 /*
747 * shift by count / limb_size
748 */
749 if( v0 > 0 )
750 {
Paul Bakker23986e52011-04-24 08:57:21 +0000751 for( i = X->n; i > v0; i-- )
752 X->p[i - 1] = X->p[i - v0 - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +0000753
Paul Bakker23986e52011-04-24 08:57:21 +0000754 for( ; i > 0; i-- )
755 X->p[i - 1] = 0;
Paul Bakker5121ce52009-01-03 21:22:43 +0000756 }
757
758 /*
759 * shift by count % limb_size
760 */
761 if( t1 > 0 )
762 {
763 for( i = v0; i < X->n; i++ )
764 {
765 r1 = X->p[i] >> (biL - t1);
766 X->p[i] <<= t1;
767 X->p[i] |= r0;
768 r0 = r1;
769 }
770 }
771
772cleanup:
773
774 return( ret );
775}
776
777/*
778 * Right-shift: X >>= count
779 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200780int mbedtls_mpi_shift_r( mbedtls_mpi *X, size_t count )
Paul Bakker5121ce52009-01-03 21:22:43 +0000781{
Paul Bakker23986e52011-04-24 08:57:21 +0000782 size_t i, v0, v1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200783 mbedtls_mpi_uint r0 = 0, r1;
Hanno Becker73d7d792018-12-11 10:35:51 +0000784 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000785
786 v0 = count / biL;
787 v1 = count & (biL - 1);
788
Manuel Pégourié-Gonnarde44ec102012-11-17 12:42:51 +0100789 if( v0 > X->n || ( v0 == X->n && v1 > 0 ) )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200790 return mbedtls_mpi_lset( X, 0 );
Manuel Pégourié-Gonnarde44ec102012-11-17 12:42:51 +0100791
Paul Bakker5121ce52009-01-03 21:22:43 +0000792 /*
793 * shift by count / limb_size
794 */
795 if( v0 > 0 )
796 {
797 for( i = 0; i < X->n - v0; i++ )
798 X->p[i] = X->p[i + v0];
799
800 for( ; i < X->n; i++ )
801 X->p[i] = 0;
802 }
803
804 /*
805 * shift by count % limb_size
806 */
807 if( v1 > 0 )
808 {
Paul Bakker23986e52011-04-24 08:57:21 +0000809 for( i = X->n; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000810 {
Paul Bakker23986e52011-04-24 08:57:21 +0000811 r1 = X->p[i - 1] << (biL - v1);
812 X->p[i - 1] >>= v1;
813 X->p[i - 1] |= r0;
Paul Bakker5121ce52009-01-03 21:22:43 +0000814 r0 = r1;
815 }
816 }
817
818 return( 0 );
819}
820
821/*
822 * Compare unsigned values
823 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200824int mbedtls_mpi_cmp_abs( const mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +0000825{
Paul Bakker23986e52011-04-24 08:57:21 +0000826 size_t i, j;
Hanno Becker73d7d792018-12-11 10:35:51 +0000827 MPI_VALIDATE_RET( X != NULL );
828 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000829
Paul Bakker23986e52011-04-24 08:57:21 +0000830 for( i = X->n; i > 0; i-- )
831 if( X->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000832 break;
833
Paul Bakker23986e52011-04-24 08:57:21 +0000834 for( j = Y->n; j > 0; j-- )
835 if( Y->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000836 break;
837
Paul Bakker23986e52011-04-24 08:57:21 +0000838 if( i == 0 && j == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000839 return( 0 );
840
841 if( i > j ) return( 1 );
842 if( j > i ) return( -1 );
843
Paul Bakker23986e52011-04-24 08:57:21 +0000844 for( ; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000845 {
Paul Bakker23986e52011-04-24 08:57:21 +0000846 if( X->p[i - 1] > Y->p[i - 1] ) return( 1 );
847 if( X->p[i - 1] < Y->p[i - 1] ) return( -1 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000848 }
849
850 return( 0 );
851}
852
853/*
854 * Compare signed values
855 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200856int mbedtls_mpi_cmp_mpi( const mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +0000857{
Paul Bakker23986e52011-04-24 08:57:21 +0000858 size_t i, j;
Hanno Becker73d7d792018-12-11 10:35:51 +0000859 MPI_VALIDATE_RET( X != NULL );
860 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000861
Paul Bakker23986e52011-04-24 08:57:21 +0000862 for( i = X->n; i > 0; i-- )
863 if( X->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000864 break;
865
Paul Bakker23986e52011-04-24 08:57:21 +0000866 for( j = Y->n; j > 0; j-- )
867 if( Y->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000868 break;
869
Paul Bakker23986e52011-04-24 08:57:21 +0000870 if( i == 0 && j == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000871 return( 0 );
872
873 if( i > j ) return( X->s );
Paul Bakker0c8f73b2012-03-22 14:08:57 +0000874 if( j > i ) return( -Y->s );
Paul Bakker5121ce52009-01-03 21:22:43 +0000875
876 if( X->s > 0 && Y->s < 0 ) return( 1 );
877 if( Y->s > 0 && X->s < 0 ) return( -1 );
878
Paul Bakker23986e52011-04-24 08:57:21 +0000879 for( ; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000880 {
Paul Bakker23986e52011-04-24 08:57:21 +0000881 if( X->p[i - 1] > Y->p[i - 1] ) return( X->s );
882 if( X->p[i - 1] < Y->p[i - 1] ) return( -X->s );
Paul Bakker5121ce52009-01-03 21:22:43 +0000883 }
884
885 return( 0 );
886}
887
Janos Follathee6abce2019-09-05 14:47:19 +0100888/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000889 * Compare signed values
890 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200891int mbedtls_mpi_cmp_int( const mbedtls_mpi *X, mbedtls_mpi_sint z )
Paul Bakker5121ce52009-01-03 21:22:43 +0000892{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200893 mbedtls_mpi Y;
894 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +0000895 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000896
897 *p = ( z < 0 ) ? -z : z;
898 Y.s = ( z < 0 ) ? -1 : 1;
899 Y.n = 1;
900 Y.p = p;
901
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200902 return( mbedtls_mpi_cmp_mpi( X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000903}
904
905/*
906 * Unsigned addition: X = |A| + |B| (HAC 14.7)
907 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200908int mbedtls_mpi_add_abs( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +0000909{
Janos Follath24eed8d2019-11-22 13:21:35 +0000910 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +0000911 size_t i, j;
Janos Follath6c922682015-10-30 17:43:11 +0100912 mbedtls_mpi_uint *o, *p, c, tmp;
Hanno Becker73d7d792018-12-11 10:35:51 +0000913 MPI_VALIDATE_RET( X != NULL );
914 MPI_VALIDATE_RET( A != NULL );
915 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000916
917 if( X == B )
918 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200919 const mbedtls_mpi *T = A; A = X; B = T;
Paul Bakker5121ce52009-01-03 21:22:43 +0000920 }
921
922 if( X != A )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200923 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, A ) );
Paul Bakker9af723c2014-05-01 13:03:14 +0200924
Paul Bakkerf7ca7b92009-06-20 10:31:06 +0000925 /*
926 * X should always be positive as a result of unsigned additions.
927 */
928 X->s = 1;
Paul Bakker5121ce52009-01-03 21:22:43 +0000929
Paul Bakker23986e52011-04-24 08:57:21 +0000930 for( j = B->n; j > 0; j-- )
931 if( B->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000932 break;
933
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200934 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000935
936 o = B->p; p = X->p; c = 0;
937
Janos Follath6c922682015-10-30 17:43:11 +0100938 /*
939 * tmp is used because it might happen that p == o
940 */
Paul Bakker23986e52011-04-24 08:57:21 +0000941 for( i = 0; i < j; i++, o++, p++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000942 {
Janos Follath6c922682015-10-30 17:43:11 +0100943 tmp= *o;
Paul Bakker5121ce52009-01-03 21:22:43 +0000944 *p += c; c = ( *p < c );
Janos Follath6c922682015-10-30 17:43:11 +0100945 *p += tmp; c += ( *p < tmp );
Paul Bakker5121ce52009-01-03 21:22:43 +0000946 }
947
948 while( c != 0 )
949 {
950 if( i >= X->n )
951 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200952 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000953 p = X->p + i;
954 }
955
Paul Bakker2d319fd2012-09-16 21:34:26 +0000956 *p += c; c = ( *p < c ); i++; p++;
Paul Bakker5121ce52009-01-03 21:22:43 +0000957 }
958
959cleanup:
960
961 return( ret );
962}
963
Gilles Peskine09ec10a2020-06-09 10:39:38 +0200964/**
Gilles Peskinec097e9e2020-06-08 21:58:22 +0200965 * Helper for mbedtls_mpi subtraction.
966 *
Gilles Peskine1acf7cb2020-07-23 01:03:22 +0200967 * Calculate l - r where l and r have the same size.
Gilles Peskinec097e9e2020-06-08 21:58:22 +0200968 * This function operates modulo (2^ciL)^n and returns the carry
Gilles Peskine1acf7cb2020-07-23 01:03:22 +0200969 * (1 if there was a wraparound, i.e. if `l < r`, and 0 otherwise).
Gilles Peskinec097e9e2020-06-08 21:58:22 +0200970 *
Gilles Peskine1acf7cb2020-07-23 01:03:22 +0200971 * d may be aliased to l or r.
Gilles Peskinec097e9e2020-06-08 21:58:22 +0200972 *
Gilles Peskine1acf7cb2020-07-23 01:03:22 +0200973 * \param n Number of limbs of \p d, \p l and \p r.
974 * \param[out] d The result of the subtraction.
975 * \param[in] l The left operand.
976 * \param[in] r The right operand.
977 *
978 * \return 1 if `l < r`.
979 * 0 if `l >= r`.
Paul Bakker5121ce52009-01-03 21:22:43 +0000980 */
Gilles Peskinec097e9e2020-06-08 21:58:22 +0200981static mbedtls_mpi_uint mpi_sub_hlp( size_t n,
982 mbedtls_mpi_uint *d,
Gilles Peskine1acf7cb2020-07-23 01:03:22 +0200983 const mbedtls_mpi_uint *l,
984 const mbedtls_mpi_uint *r )
Paul Bakker5121ce52009-01-03 21:22:43 +0000985{
Paul Bakker23986e52011-04-24 08:57:21 +0000986 size_t i;
Gilles Peskine1acf7cb2020-07-23 01:03:22 +0200987 mbedtls_mpi_uint c = 0, t, z;
Paul Bakker5121ce52009-01-03 21:22:43 +0000988
Gilles Peskine1acf7cb2020-07-23 01:03:22 +0200989 for( i = 0; i < n; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000990 {
Gilles Peskine1acf7cb2020-07-23 01:03:22 +0200991 z = ( l[i] < c ); t = l[i] - c;
992 c = ( t < r[i] ) + z; d[i] = t - r[i];
Paul Bakker5121ce52009-01-03 21:22:43 +0000993 }
994
Gilles Peskinec097e9e2020-06-08 21:58:22 +0200995 return( c );
Paul Bakker5121ce52009-01-03 21:22:43 +0000996}
997
998/*
Gilles Peskine0e5faf62020-06-08 22:50:35 +0200999 * Unsigned subtraction: X = |A| - |B| (HAC 14.9, 14.10)
Paul Bakker5121ce52009-01-03 21:22:43 +00001000 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001001int mbedtls_mpi_sub_abs( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001002{
Janos Follath24eed8d2019-11-22 13:21:35 +00001003 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001004 size_t n;
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001005 mbedtls_mpi_uint carry;
Hanno Becker73d7d792018-12-11 10:35:51 +00001006 MPI_VALIDATE_RET( X != NULL );
1007 MPI_VALIDATE_RET( A != NULL );
1008 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001009
Paul Bakker23986e52011-04-24 08:57:21 +00001010 for( n = B->n; n > 0; n-- )
1011 if( B->p[n - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001012 break;
Gilles Peskinec8a91772021-01-27 22:30:43 +01001013 if( n > A->n )
1014 {
1015 /* B >= (2^ciL)^n > A */
1016 ret = MBEDTLS_ERR_MPI_NEGATIVE_VALUE;
1017 goto cleanup;
1018 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001019
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001020 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, A->n ) );
1021
1022 /* Set the high limbs of X to match A. Don't touch the lower limbs
1023 * because X might be aliased to B, and we must not overwrite the
1024 * significant digits of B. */
1025 if( A->n > n )
1026 memcpy( X->p + n, A->p + n, ( A->n - n ) * ciL );
1027 if( X->n > A->n )
1028 memset( X->p + A->n, 0, ( X->n - A->n ) * ciL );
1029
1030 carry = mpi_sub_hlp( n, X->p, A->p, B->p );
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001031 if( carry != 0 )
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001032 {
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001033 /* Propagate the carry to the first nonzero limb of X. */
1034 for( ; n < X->n && X->p[n] == 0; n++ )
1035 --X->p[n];
1036 /* If we ran out of space for the carry, it means that the result
1037 * is negative. */
1038 if( n == X->n )
Gilles Peskine89b41302020-07-23 01:16:46 +02001039 {
1040 ret = MBEDTLS_ERR_MPI_NEGATIVE_VALUE;
1041 goto cleanup;
1042 }
Gilles Peskine0e5faf62020-06-08 22:50:35 +02001043 --X->p[n];
Gilles Peskinec097e9e2020-06-08 21:58:22 +02001044 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001045
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001046 /* X should always be positive as a result of unsigned subtractions. */
1047 X->s = 1;
1048
Paul Bakker5121ce52009-01-03 21:22:43 +00001049cleanup:
Paul Bakker5121ce52009-01-03 21:22:43 +00001050 return( ret );
1051}
1052
1053/*
1054 * Signed addition: X = A + B
1055 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001056int mbedtls_mpi_add_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001057{
Hanno Becker73d7d792018-12-11 10:35:51 +00001058 int ret, s;
1059 MPI_VALIDATE_RET( X != NULL );
1060 MPI_VALIDATE_RET( A != NULL );
1061 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001062
Hanno Becker73d7d792018-12-11 10:35:51 +00001063 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001064 if( A->s * B->s < 0 )
1065 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001066 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001067 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001068 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001069 X->s = s;
1070 }
1071 else
1072 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001073 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001074 X->s = -s;
1075 }
1076 }
1077 else
1078 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001079 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001080 X->s = s;
1081 }
1082
1083cleanup:
1084
1085 return( ret );
1086}
1087
1088/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001089 * Signed subtraction: X = A - B
Paul Bakker5121ce52009-01-03 21:22:43 +00001090 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001091int mbedtls_mpi_sub_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001092{
Hanno Becker73d7d792018-12-11 10:35:51 +00001093 int ret, s;
1094 MPI_VALIDATE_RET( X != NULL );
1095 MPI_VALIDATE_RET( A != NULL );
1096 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001097
Hanno Becker73d7d792018-12-11 10:35:51 +00001098 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001099 if( A->s * B->s > 0 )
1100 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001101 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001102 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001103 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001104 X->s = s;
1105 }
1106 else
1107 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001108 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001109 X->s = -s;
1110 }
1111 }
1112 else
1113 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001114 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001115 X->s = s;
1116 }
1117
1118cleanup:
1119
1120 return( ret );
1121}
1122
1123/*
1124 * Signed addition: X = A + b
1125 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001126int mbedtls_mpi_add_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001127{
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001128 mbedtls_mpi B;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001129 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001130 MPI_VALIDATE_RET( X != NULL );
1131 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001132
1133 p[0] = ( b < 0 ) ? -b : b;
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001134 B.s = ( b < 0 ) ? -1 : 1;
1135 B.n = 1;
1136 B.p = p;
Paul Bakker5121ce52009-01-03 21:22:43 +00001137
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001138 return( mbedtls_mpi_add_mpi( X, A, &B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001139}
1140
1141/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001142 * Signed subtraction: X = A - b
Paul Bakker5121ce52009-01-03 21:22:43 +00001143 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001144int mbedtls_mpi_sub_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001145{
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001146 mbedtls_mpi B;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001147 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001148 MPI_VALIDATE_RET( X != NULL );
1149 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001150
1151 p[0] = ( b < 0 ) ? -b : b;
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001152 B.s = ( b < 0 ) ? -1 : 1;
1153 B.n = 1;
1154 B.p = p;
Paul Bakker5121ce52009-01-03 21:22:43 +00001155
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001156 return( mbedtls_mpi_sub_mpi( X, A, &B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001157}
1158
Hanno Becker284d7782022-04-11 09:19:24 +01001159mbedtls_mpi_uint mbedtls_mpi_core_mla( mbedtls_mpi_uint *d, size_t d_len,
Hanno Beckeraef9cc42022-04-11 06:36:29 +01001160 const mbedtls_mpi_uint *s, size_t s_len,
1161 mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001162{
Hanno Beckere7f14a32022-04-06 06:11:26 +01001163 mbedtls_mpi_uint c = 0; /* carry */
Hanno Becker5d4ceeb2022-04-11 09:46:47 +01001164 size_t excess_len = d_len - s_len;
Hanno Beckerdefe5692022-04-06 06:12:09 +01001165
Hanno Becker63eb28c2022-04-06 11:30:51 +01001166 size_t steps_x8 = s_len / 8;
1167 size_t steps_x1 = s_len & 7;
1168
1169 while( steps_x8-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001170 {
Hanno Beckereacf3b92022-04-06 11:25:22 +01001171 MULADDC_X8_INIT
1172 MULADDC_X8_CORE
1173 MULADDC_X8_STOP
Paul Bakker5121ce52009-01-03 21:22:43 +00001174 }
1175
Hanno Becker63eb28c2022-04-06 11:30:51 +01001176 while( steps_x1-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001177 {
Hanno Beckereacf3b92022-04-06 11:25:22 +01001178 MULADDC_X1_INIT
1179 MULADDC_X1_CORE
1180 MULADDC_X1_STOP
Paul Bakker5121ce52009-01-03 21:22:43 +00001181 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001182
Hanno Becker284d7782022-04-11 09:19:24 +01001183 while( excess_len-- )
Gilles Peskine8e464c42020-07-24 00:08:38 +02001184 {
Paul Bakker5121ce52009-01-03 21:22:43 +00001185 *d += c; c = ( *d < c ); d++;
1186 }
Hanno Beckerdefe5692022-04-06 06:12:09 +01001187
1188 return( c );
Paul Bakker5121ce52009-01-03 21:22:43 +00001189}
1190
1191/*
1192 * Baseline multiplication: X = A * B (HAC 14.12)
1193 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001194int mbedtls_mpi_mul_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001195{
Janos Follath24eed8d2019-11-22 13:21:35 +00001196 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker1772e052022-04-13 06:51:40 +01001197 size_t i, j;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001198 mbedtls_mpi TA, TB;
Hanno Beckerda763de2022-04-13 06:50:02 +01001199 int result_is_zero = 0;
Hanno Becker73d7d792018-12-11 10:35:51 +00001200 MPI_VALIDATE_RET( X != NULL );
1201 MPI_VALIDATE_RET( A != NULL );
1202 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001203
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001204 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001205
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001206 if( X == A ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) ); A = &TA; }
1207 if( X == B ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) ); B = &TB; }
Paul Bakker5121ce52009-01-03 21:22:43 +00001208
Hanno Beckerda763de2022-04-13 06:50:02 +01001209 for( i = A->n; i > 0; i-- )
1210 if( A->p[i - 1] != 0 )
1211 break;
1212 if( i == 0 )
1213 result_is_zero = 1;
1214
1215 for( j = B->n; j > 0; j-- )
1216 if( B->p[j - 1] != 0 )
1217 break;
1218 if( j == 0 )
1219 result_is_zero = 1;
1220
1221 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + j ) );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001222 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001223
Hanno Becker1772e052022-04-13 06:51:40 +01001224 for( size_t k = 0; k < j; k++ )
Hanno Beckerfee261a2022-04-06 06:20:22 +01001225 {
1226 /* We know that there cannot be any carry-out since we're
1227 * iterating from bottom to top. */
Hanno Beckerda763de2022-04-13 06:50:02 +01001228 (void) mbedtls_mpi_core_mla( X->p + k, i + 1,
1229 A->p, i,
Hanno Beckeraef9cc42022-04-11 06:36:29 +01001230 B->p[k] );
Hanno Beckerfee261a2022-04-06 06:20:22 +01001231 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001232
Hanno Beckerda763de2022-04-13 06:50:02 +01001233 /* If the result is 0, we don't shortcut the operation, which reduces
1234 * but does not eliminate side channels leaking the zero-ness. We do
1235 * need to take care to set the sign bit properly since the library does
1236 * not fully support an MPI object with a value of 0 and s == -1. */
1237 if( result_is_zero )
1238 X->s = 1;
1239 else
1240 X->s = A->s * B->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001241
1242cleanup:
1243
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001244 mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TA );
Paul Bakker5121ce52009-01-03 21:22:43 +00001245
1246 return( ret );
1247}
1248
1249/*
1250 * Baseline multiplication: X = A * b
1251 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001252int mbedtls_mpi_mul_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001253{
Hanno Becker73d7d792018-12-11 10:35:51 +00001254 MPI_VALIDATE_RET( X != NULL );
1255 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001256
Hanno Becker35771312022-04-14 11:52:11 +01001257 size_t n = A->n;
1258 while( n > 0 && A->p[n - 1] == 0 )
1259 --n;
1260
Hanno Becker74a11a32022-04-06 06:27:00 +01001261 /* The general method below doesn't work if b==0. */
Hanno Becker35771312022-04-14 11:52:11 +01001262 if( b == 0 || n == 0 )
Paul Elliott986b55a2021-04-20 21:46:29 +01001263 return( mbedtls_mpi_lset( X, 0 ) );
Gilles Peskine8fd95c62020-07-23 21:58:50 +02001264
Hanno Beckeraef9cc42022-04-11 06:36:29 +01001265 /* Calculate A*b as A + A*(b-1) to take advantage of mbedtls_mpi_core_mla */
Gilles Peskine8fd95c62020-07-23 21:58:50 +02001266 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Gilles Peskinecd0dbf32020-07-24 00:09:04 +02001267 /* In general, A * b requires 1 limb more than b. If
1268 * A->p[n - 1] * b / b == A->p[n - 1], then A * b fits in the same
1269 * number of limbs as A and the call to grow() is not required since
Gilles Peskinee1bba7c2021-03-10 23:44:10 +01001270 * copy() will take care of the growth if needed. However, experimentally,
1271 * making the call to grow() unconditional causes slightly fewer
Gilles Peskinecd0dbf32020-07-24 00:09:04 +02001272 * calls to calloc() in ECP code, presumably because it reuses the
1273 * same mpi for a while and this way the mpi is more likely to directly
Hanno Becker9137b9c2022-04-12 10:51:54 +01001274 * grow to its final size.
1275 *
1276 * Note that calculating A*b as 0 + A*b doesn't work as-is because
1277 * A,X can be the same. */
Hanno Becker35771312022-04-14 11:52:11 +01001278 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, n + 1 ) );
Gilles Peskine8fd95c62020-07-23 21:58:50 +02001279 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, A ) );
Hanno Becker35771312022-04-14 11:52:11 +01001280 mbedtls_mpi_core_mla( X->p, X->n, A->p, n, b - 1 );
Gilles Peskine8fd95c62020-07-23 21:58:50 +02001281
1282cleanup:
1283 return( ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00001284}
1285
1286/*
Simon Butcherf5ba0452015-12-27 23:01:55 +00001287 * Unsigned integer divide - double mbedtls_mpi_uint dividend, u1/u0, and
1288 * mbedtls_mpi_uint divisor, d
Simon Butcher15b15d12015-11-26 19:35:03 +00001289 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001290static mbedtls_mpi_uint mbedtls_int_div_int( mbedtls_mpi_uint u1,
1291 mbedtls_mpi_uint u0, mbedtls_mpi_uint d, mbedtls_mpi_uint *r )
Simon Butcher15b15d12015-11-26 19:35:03 +00001292{
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001293#if defined(MBEDTLS_HAVE_UDBL)
1294 mbedtls_t_udbl dividend, quotient;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001295#else
Simon Butcher9803d072016-01-03 00:24:34 +00001296 const mbedtls_mpi_uint radix = (mbedtls_mpi_uint) 1 << biH;
1297 const mbedtls_mpi_uint uint_halfword_mask = ( (mbedtls_mpi_uint) 1 << biH ) - 1;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001298 mbedtls_mpi_uint d0, d1, q0, q1, rAX, r0, quotient;
1299 mbedtls_mpi_uint u0_msw, u0_lsw;
Simon Butcher9803d072016-01-03 00:24:34 +00001300 size_t s;
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001301#endif
1302
Simon Butcher15b15d12015-11-26 19:35:03 +00001303 /*
1304 * Check for overflow
1305 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001306 if( 0 == d || u1 >= d )
Simon Butcher15b15d12015-11-26 19:35:03 +00001307 {
Simon Butcherf5ba0452015-12-27 23:01:55 +00001308 if (r != NULL) *r = ~0;
Simon Butcher15b15d12015-11-26 19:35:03 +00001309
Simon Butcherf5ba0452015-12-27 23:01:55 +00001310 return ( ~0 );
Simon Butcher15b15d12015-11-26 19:35:03 +00001311 }
1312
1313#if defined(MBEDTLS_HAVE_UDBL)
Simon Butcher15b15d12015-11-26 19:35:03 +00001314 dividend = (mbedtls_t_udbl) u1 << biL;
1315 dividend |= (mbedtls_t_udbl) u0;
1316 quotient = dividend / d;
1317 if( quotient > ( (mbedtls_t_udbl) 1 << biL ) - 1 )
1318 quotient = ( (mbedtls_t_udbl) 1 << biL ) - 1;
1319
1320 if( r != NULL )
Simon Butcher9803d072016-01-03 00:24:34 +00001321 *r = (mbedtls_mpi_uint)( dividend - (quotient * d ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001322
1323 return (mbedtls_mpi_uint) quotient;
1324#else
Simon Butcher15b15d12015-11-26 19:35:03 +00001325
1326 /*
1327 * Algorithm D, Section 4.3.1 - The Art of Computer Programming
1328 * Vol. 2 - Seminumerical Algorithms, Knuth
1329 */
1330
1331 /*
1332 * Normalize the divisor, d, and dividend, u0, u1
1333 */
Janos Follath4670f882022-07-21 18:25:42 +01001334 s = mbedtls_mpi_core_clz( d );
Simon Butcher15b15d12015-11-26 19:35:03 +00001335 d = d << s;
1336
1337 u1 = u1 << s;
Simon Butcher9803d072016-01-03 00:24:34 +00001338 u1 |= ( u0 >> ( biL - s ) ) & ( -(mbedtls_mpi_sint)s >> ( biL - 1 ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001339 u0 = u0 << s;
1340
1341 d1 = d >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001342 d0 = d & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001343
1344 u0_msw = u0 >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001345 u0_lsw = u0 & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001346
1347 /*
1348 * Find the first quotient and remainder
1349 */
1350 q1 = u1 / d1;
1351 r0 = u1 - d1 * q1;
1352
1353 while( q1 >= radix || ( q1 * d0 > radix * r0 + u0_msw ) )
1354 {
1355 q1 -= 1;
1356 r0 += d1;
1357
1358 if ( r0 >= radix ) break;
1359 }
1360
Simon Butcherf5ba0452015-12-27 23:01:55 +00001361 rAX = ( u1 * radix ) + ( u0_msw - q1 * d );
Simon Butcher15b15d12015-11-26 19:35:03 +00001362 q0 = rAX / d1;
1363 r0 = rAX - q0 * d1;
1364
1365 while( q0 >= radix || ( q0 * d0 > radix * r0 + u0_lsw ) )
1366 {
1367 q0 -= 1;
1368 r0 += d1;
1369
1370 if ( r0 >= radix ) break;
1371 }
1372
1373 if (r != NULL)
Simon Butcherf5ba0452015-12-27 23:01:55 +00001374 *r = ( rAX * radix + u0_lsw - q0 * d ) >> s;
Simon Butcher15b15d12015-11-26 19:35:03 +00001375
1376 quotient = q1 * radix + q0;
1377
1378 return quotient;
1379#endif
1380}
1381
1382/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001383 * Division by mbedtls_mpi: A = Q * B + R (HAC 14.20)
Paul Bakker5121ce52009-01-03 21:22:43 +00001384 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001385int mbedtls_mpi_div_mpi( mbedtls_mpi *Q, mbedtls_mpi *R, const mbedtls_mpi *A,
1386 const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001387{
Janos Follath24eed8d2019-11-22 13:21:35 +00001388 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001389 size_t i, n, t, k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001390 mbedtls_mpi X, Y, Z, T1, T2;
Alexander Kd19a1932019-11-01 18:20:42 +03001391 mbedtls_mpi_uint TP2[3];
Hanno Becker73d7d792018-12-11 10:35:51 +00001392 MPI_VALIDATE_RET( A != NULL );
1393 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001394
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001395 if( mbedtls_mpi_cmp_int( B, 0 ) == 0 )
1396 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001397
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001398 mbedtls_mpi_init( &X ); mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001399 mbedtls_mpi_init( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001400 /*
1401 * Avoid dynamic memory allocations for constant-size T2.
1402 *
1403 * T2 is used for comparison only and the 3 limbs are assigned explicitly,
1404 * so nobody increase the size of the MPI and we're safe to use an on-stack
1405 * buffer.
1406 */
Alexander K35d6d462019-10-31 14:46:45 +03001407 T2.s = 1;
Alexander Kd19a1932019-11-01 18:20:42 +03001408 T2.n = sizeof( TP2 ) / sizeof( *TP2 );
1409 T2.p = TP2;
Paul Bakker5121ce52009-01-03 21:22:43 +00001410
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001411 if( mbedtls_mpi_cmp_abs( A, B ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001412 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001413 if( Q != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_lset( Q, 0 ) );
1414 if( R != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001415 return( 0 );
1416 }
1417
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001418 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &X, A ) );
1419 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001420 X.s = Y.s = 1;
1421
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001422 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &Z, A->n + 2 ) );
1423 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Z, 0 ) );
Gilles Peskine2536aa72020-07-24 00:12:59 +02001424 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T1, A->n + 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001425
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001426 k = mbedtls_mpi_bitlen( &Y ) % biL;
Paul Bakkerf9688572011-05-05 10:00:45 +00001427 if( k < biL - 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001428 {
1429 k = biL - 1 - k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001430 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &X, k ) );
1431 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, k ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001432 }
1433 else k = 0;
1434
1435 n = X.n - 1;
1436 t = Y.n - 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001437 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001438
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001439 while( mbedtls_mpi_cmp_mpi( &X, &Y ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001440 {
1441 Z.p[n - t]++;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001442 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001443 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001444 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001445
1446 for( i = n; i > t ; i-- )
1447 {
1448 if( X.p[i] >= Y.p[t] )
1449 Z.p[i - t - 1] = ~0;
1450 else
1451 {
Simon Butcher15b15d12015-11-26 19:35:03 +00001452 Z.p[i - t - 1] = mbedtls_int_div_int( X.p[i], X.p[i - 1],
1453 Y.p[t], NULL);
Paul Bakker5121ce52009-01-03 21:22:43 +00001454 }
1455
Alexander K35d6d462019-10-31 14:46:45 +03001456 T2.p[0] = ( i < 2 ) ? 0 : X.p[i - 2];
1457 T2.p[1] = ( i < 1 ) ? 0 : X.p[i - 1];
1458 T2.p[2] = X.p[i];
1459
Paul Bakker5121ce52009-01-03 21:22:43 +00001460 Z.p[i - t - 1]++;
1461 do
1462 {
1463 Z.p[i - t - 1]--;
1464
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001465 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &T1, 0 ) );
Paul Bakker66d5d072014-06-17 16:39:18 +02001466 T1.p[0] = ( t < 1 ) ? 0 : Y.p[t - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001467 T1.p[1] = Y.p[t];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001468 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &T1, Z.p[i - t - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001469 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001470 while( mbedtls_mpi_cmp_mpi( &T1, &T2 ) > 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001471
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001472 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &Y, Z.p[i - t - 1] ) );
1473 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1474 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001475
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001476 if( mbedtls_mpi_cmp_int( &X, 0 ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001477 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001478 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T1, &Y ) );
1479 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1480 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001481 Z.p[i - t - 1]--;
1482 }
1483 }
1484
1485 if( Q != NULL )
1486 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001487 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( Q, &Z ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001488 Q->s = A->s * B->s;
1489 }
1490
1491 if( R != NULL )
1492 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001493 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &X, k ) );
Paul Bakkerf02c5642012-11-13 10:25:21 +00001494 X.s = A->s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001495 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, &X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001496
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001497 if( mbedtls_mpi_cmp_int( R, 0 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001498 R->s = 1;
1499 }
1500
1501cleanup:
1502
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001503 mbedtls_mpi_free( &X ); mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &Z );
Alexander K35d6d462019-10-31 14:46:45 +03001504 mbedtls_mpi_free( &T1 );
Alexander Kd19a1932019-11-01 18:20:42 +03001505 mbedtls_platform_zeroize( TP2, sizeof( TP2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001506
1507 return( ret );
1508}
1509
1510/*
1511 * Division by int: A = Q * b + R
Paul Bakker5121ce52009-01-03 21:22:43 +00001512 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001513int mbedtls_mpi_div_int( mbedtls_mpi *Q, mbedtls_mpi *R,
1514 const mbedtls_mpi *A,
1515 mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001516{
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001517 mbedtls_mpi B;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001518 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001519 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001520
1521 p[0] = ( b < 0 ) ? -b : b;
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001522 B.s = ( b < 0 ) ? -1 : 1;
1523 B.n = 1;
1524 B.p = p;
Paul Bakker5121ce52009-01-03 21:22:43 +00001525
Yuto Takano36c8ddc2021-07-05 09:10:52 +01001526 return( mbedtls_mpi_div_mpi( Q, R, A, &B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001527}
1528
1529/*
1530 * Modulo: R = A mod B
1531 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001532int mbedtls_mpi_mod_mpi( mbedtls_mpi *R, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001533{
Janos Follath24eed8d2019-11-22 13:21:35 +00001534 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker73d7d792018-12-11 10:35:51 +00001535 MPI_VALIDATE_RET( R != NULL );
1536 MPI_VALIDATE_RET( A != NULL );
1537 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001538
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001539 if( mbedtls_mpi_cmp_int( B, 0 ) < 0 )
1540 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001541
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001542 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( NULL, R, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001543
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001544 while( mbedtls_mpi_cmp_int( R, 0 ) < 0 )
1545 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001546
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001547 while( mbedtls_mpi_cmp_mpi( R, B ) >= 0 )
1548 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001549
1550cleanup:
1551
1552 return( ret );
1553}
1554
1555/*
1556 * Modulo: r = A mod b
1557 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001558int mbedtls_mpi_mod_int( mbedtls_mpi_uint *r, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001559{
Paul Bakker23986e52011-04-24 08:57:21 +00001560 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001561 mbedtls_mpi_uint x, y, z;
Hanno Becker73d7d792018-12-11 10:35:51 +00001562 MPI_VALIDATE_RET( r != NULL );
1563 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001564
1565 if( b == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001566 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001567
1568 if( b < 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001569 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakker5121ce52009-01-03 21:22:43 +00001570
1571 /*
1572 * handle trivial cases
1573 */
Gilles Peskineae25bb02022-06-09 19:32:46 +02001574 if( b == 1 || A->n == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001575 {
1576 *r = 0;
1577 return( 0 );
1578 }
1579
1580 if( b == 2 )
1581 {
1582 *r = A->p[0] & 1;
1583 return( 0 );
1584 }
1585
1586 /*
1587 * general case
1588 */
Paul Bakker23986e52011-04-24 08:57:21 +00001589 for( i = A->n, y = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001590 {
Paul Bakker23986e52011-04-24 08:57:21 +00001591 x = A->p[i - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001592 y = ( y << biH ) | ( x >> biH );
1593 z = y / b;
1594 y -= z * b;
1595
1596 x <<= biH;
1597 y = ( y << biH ) | ( x >> biH );
1598 z = y / b;
1599 y -= z * b;
1600 }
1601
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001602 /*
1603 * If A is negative, then the current y represents a negative value.
1604 * Flipping it to the positive side.
1605 */
1606 if( A->s < 0 && y != 0 )
1607 y = b - y;
1608
Paul Bakker5121ce52009-01-03 21:22:43 +00001609 *r = y;
1610
1611 return( 0 );
1612}
1613
1614/*
1615 * Fast Montgomery initialization (thanks to Tom St Denis)
1616 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001617static void mpi_montg_init( mbedtls_mpi_uint *mm, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00001618{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001619 mbedtls_mpi_uint x, m0 = N->p[0];
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01001620 unsigned int i;
Paul Bakker5121ce52009-01-03 21:22:43 +00001621
1622 x = m0;
1623 x += ( ( m0 + 2 ) & 4 ) << 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00001624
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01001625 for( i = biL; i >= 8; i /= 2 )
1626 x *= ( 2 - ( m0 * x ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001627
1628 *mm = ~x + 1;
1629}
1630
Gilles Peskine2a82f722020-06-04 15:00:49 +02001631/** Montgomery multiplication: A = A * B * R^-1 mod N (HAC 14.36)
1632 *
1633 * \param[in,out] A One of the numbers to multiply.
Gilles Peskine221626f2020-06-08 22:37:50 +02001634 * It must have at least as many limbs as N
1635 * (A->n >= N->n), and any limbs beyond n are ignored.
Gilles Peskine2a82f722020-06-04 15:00:49 +02001636 * On successful completion, A contains the result of
1637 * the multiplication A * B * R^-1 mod N where
1638 * R = (2^ciL)^n.
1639 * \param[in] B One of the numbers to multiply.
1640 * It must be nonzero and must not have more limbs than N
1641 * (B->n <= N->n).
1642 * \param[in] N The modulo. N must be odd.
1643 * \param mm The value calculated by `mpi_montg_init(&mm, N)`.
1644 * This is -N^-1 mod 2^ciL.
1645 * \param[in,out] T A bignum for temporary storage.
Hanno Beckere1417022022-04-06 06:45:45 +01001646 * It must be at least twice the limb size of N plus 1
1647 * (T->n >= 2 * N->n + 1).
Gilles Peskine2a82f722020-06-04 15:00:49 +02001648 * Its initial content is unused and
1649 * its final content is indeterminate.
1650 * Note that unlike the usual convention in the library
1651 * for `const mbedtls_mpi*`, the content of T can change.
Paul Bakker5121ce52009-01-03 21:22:43 +00001652 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02001653static 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 +02001654 const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00001655{
Hanno Becker0235f752022-04-12 10:54:46 +01001656 size_t n, m;
1657 mbedtls_mpi_uint *d;
Paul Bakker5121ce52009-01-03 21:22:43 +00001658
1659 memset( T->p, 0, T->n * ciL );
1660
1661 d = T->p;
1662 n = N->n;
1663 m = ( B->n < n ) ? B->n : n;
1664
Hanno Becker0235f752022-04-12 10:54:46 +01001665 for( size_t i = 0; i < n; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00001666 {
Hanno Becker0235f752022-04-12 10:54:46 +01001667 mbedtls_mpi_uint u0, u1;
1668
Paul Bakker5121ce52009-01-03 21:22:43 +00001669 /*
1670 * T = (T + u0*B + u1*N) / 2^biL
1671 */
1672 u0 = A->p[i];
1673 u1 = ( d[0] + u0 * B->p[0] ) * mm;
1674
Hanno Beckeraef9cc42022-04-11 06:36:29 +01001675 (void) mbedtls_mpi_core_mla( d, n + 2,
1676 B->p, m,
1677 u0 );
1678 (void) mbedtls_mpi_core_mla( d, n + 2,
1679 N->p, n,
1680 u1 );
Hanno Beckere1417022022-04-06 06:45:45 +01001681 d++;
Paul Bakker5121ce52009-01-03 21:22:43 +00001682 }
1683
Gilles Peskine221626f2020-06-08 22:37:50 +02001684 /* At this point, d is either the desired result or the desired result
1685 * plus N. We now potentially subtract N, avoiding leaking whether the
1686 * subtraction is performed through side channels. */
Paul Bakker5121ce52009-01-03 21:22:43 +00001687
Gilles Peskine221626f2020-06-08 22:37:50 +02001688 /* Copy the n least significant limbs of d to A, so that
1689 * A = d if d < N (recall that N has n limbs). */
1690 memcpy( A->p, d, n * ciL );
Gilles Peskine09ec10a2020-06-09 10:39:38 +02001691 /* If d >= N then we want to set A to d - N. To prevent timing attacks,
Gilles Peskine221626f2020-06-08 22:37:50 +02001692 * do the calculation without using conditional tests. */
1693 /* Set d to d0 + (2^biL)^n - N where d0 is the current value of d. */
Gilles Peskine132c0972020-06-04 21:05:24 +02001694 d[n] += 1;
Gilles Peskine1acf7cb2020-07-23 01:03:22 +02001695 d[n] -= mpi_sub_hlp( n, d, d, N->p );
Gilles Peskine221626f2020-06-08 22:37:50 +02001696 /* If d0 < N then d < (2^biL)^n
1697 * so d[n] == 0 and we want to keep A as it is.
1698 * If d0 >= N then d >= (2^biL)^n, and d <= (2^biL)^n + N < 2 * (2^biL)^n
1699 * so d[n] == 1 and we want to set A to the result of the subtraction
1700 * which is d - (2^biL)^n, i.e. the n least significant limbs of d.
1701 * This exactly corresponds to a conditional assignment. */
Gabor Mezei90437e32021-10-20 11:59:27 +02001702 mbedtls_ct_mpi_uint_cond_assign( n, A->p, d, (unsigned char) d[n] );
Paul Bakker5121ce52009-01-03 21:22:43 +00001703}
1704
1705/*
1706 * Montgomery reduction: A = A * R^-1 mod N
Gilles Peskine2a82f722020-06-04 15:00:49 +02001707 *
1708 * See mpi_montmul() regarding constraints and guarantees on the parameters.
Paul Bakker5121ce52009-01-03 21:22:43 +00001709 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02001710static void mpi_montred( mbedtls_mpi *A, const mbedtls_mpi *N,
1711 mbedtls_mpi_uint mm, const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00001712{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001713 mbedtls_mpi_uint z = 1;
1714 mbedtls_mpi U;
Paul Bakker5121ce52009-01-03 21:22:43 +00001715
Paul Bakker8ddb6452013-02-27 14:56:33 +01001716 U.n = U.s = (int) z;
Paul Bakker5121ce52009-01-03 21:22:43 +00001717 U.p = &z;
1718
Gilles Peskine4e91d472020-06-04 20:55:15 +02001719 mpi_montmul( A, &U, N, mm, T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001720}
1721
Manuel Pégourié-Gonnard1297ef32021-03-09 11:22:20 +01001722/**
1723 * Select an MPI from a table without leaking the index.
1724 *
1725 * This is functionally equivalent to mbedtls_mpi_copy(R, T[idx]) except it
1726 * reads the entire table in order to avoid leaking the value of idx to an
1727 * attacker able to observe memory access patterns.
1728 *
1729 * \param[out] R Where to write the selected MPI.
1730 * \param[in] T The table to read from.
1731 * \param[in] T_size The number of elements in the table.
1732 * \param[in] idx The index of the element to select;
1733 * this must satisfy 0 <= idx < T_size.
1734 *
1735 * \return \c 0 on success, or a negative error code.
1736 */
1737static int mpi_select( mbedtls_mpi *R, const mbedtls_mpi *T, size_t T_size, size_t idx )
1738{
1739 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
1740
1741 for( size_t i = 0; i < T_size; i++ )
Manuel Pégourié-Gonnard92413ef2021-06-03 10:42:46 +02001742 {
1743 MBEDTLS_MPI_CHK( mbedtls_mpi_safe_cond_assign( R, &T[i],
Gabor Mezei90437e32021-10-20 11:59:27 +02001744 (unsigned char) mbedtls_ct_size_bool_eq( i, idx ) ) );
Manuel Pégourié-Gonnard92413ef2021-06-03 10:42:46 +02001745 }
Manuel Pégourié-Gonnard1297ef32021-03-09 11:22:20 +01001746
1747cleanup:
1748 return( ret );
1749}
1750
Paul Bakker5121ce52009-01-03 21:22:43 +00001751/*
1752 * Sliding-window exponentiation: X = A^E mod N (HAC 14.85)
1753 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001754int mbedtls_mpi_exp_mod( mbedtls_mpi *X, const mbedtls_mpi *A,
1755 const mbedtls_mpi *E, const mbedtls_mpi *N,
Yuto Takano538a0cb2021-07-14 10:20:09 +01001756 mbedtls_mpi *prec_RR )
Paul Bakker5121ce52009-01-03 21:22:43 +00001757{
Janos Follath24eed8d2019-11-22 13:21:35 +00001758 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001759 size_t wbits, wsize, one = 1;
1760 size_t i, j, nblimbs;
1761 size_t bufsize, nbits;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001762 mbedtls_mpi_uint ei, mm, state;
Manuel Pégourié-Gonnard1297ef32021-03-09 11:22:20 +01001763 mbedtls_mpi RR, T, W[ 1 << MBEDTLS_MPI_WINDOW_SIZE ], WW, Apos;
Paul Bakkerf6198c12012-05-16 08:02:29 +00001764 int neg;
Paul Bakker5121ce52009-01-03 21:22:43 +00001765
Hanno Becker73d7d792018-12-11 10:35:51 +00001766 MPI_VALIDATE_RET( X != NULL );
1767 MPI_VALIDATE_RET( A != NULL );
1768 MPI_VALIDATE_RET( E != NULL );
1769 MPI_VALIDATE_RET( N != NULL );
1770
Hanno Becker8d1dd1b2017-09-28 11:02:24 +01001771 if( mbedtls_mpi_cmp_int( N, 0 ) <= 0 || ( N->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001772 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00001773
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001774 if( mbedtls_mpi_cmp_int( E, 0 ) < 0 )
1775 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakkerf6198c12012-05-16 08:02:29 +00001776
Chris Jones9246d042020-11-25 15:12:39 +00001777 if( mbedtls_mpi_bitlen( E ) > MBEDTLS_MPI_MAX_BITS ||
1778 mbedtls_mpi_bitlen( N ) > MBEDTLS_MPI_MAX_BITS )
1779 return ( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
1780
Paul Bakkerf6198c12012-05-16 08:02:29 +00001781 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00001782 * Init temps and window size
1783 */
1784 mpi_montg_init( &mm, N );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001785 mbedtls_mpi_init( &RR ); mbedtls_mpi_init( &T );
1786 mbedtls_mpi_init( &Apos );
Manuel Pégourié-Gonnard1297ef32021-03-09 11:22:20 +01001787 mbedtls_mpi_init( &WW );
Paul Bakker5121ce52009-01-03 21:22:43 +00001788 memset( W, 0, sizeof( W ) );
1789
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001790 i = mbedtls_mpi_bitlen( E );
Paul Bakker5121ce52009-01-03 21:22:43 +00001791
1792 wsize = ( i > 671 ) ? 6 : ( i > 239 ) ? 5 :
1793 ( i > 79 ) ? 4 : ( i > 23 ) ? 3 : 1;
1794
Peter Kolbuse6bcad32018-12-11 14:01:44 -06001795#if( MBEDTLS_MPI_WINDOW_SIZE < 6 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001796 if( wsize > MBEDTLS_MPI_WINDOW_SIZE )
1797 wsize = MBEDTLS_MPI_WINDOW_SIZE;
Peter Kolbuse6bcad32018-12-11 14:01:44 -06001798#endif
Paul Bakkerb6d5f082011-11-25 11:52:11 +00001799
Paul Bakker5121ce52009-01-03 21:22:43 +00001800 j = N->n + 1;
Gilles Peskine3da1a8f2021-06-08 23:17:42 +02001801 /* All W[i] and X must have at least N->n limbs for the mpi_montmul()
1802 * and mpi_montred() calls later. Here we ensure that W[1] and X are
1803 * large enough, and later we'll grow other W[i] to the same length.
1804 * They must not be shrunk midway through this function!
1805 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001806 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
1807 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[1], j ) );
1808 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T, j * 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001809
1810 /*
Paul Bakker50546922012-05-19 08:40:49 +00001811 * Compensate for negative A (and correct at the end)
1812 */
1813 neg = ( A->s == -1 );
Paul Bakker50546922012-05-19 08:40:49 +00001814 if( neg )
1815 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001816 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Apos, A ) );
Paul Bakker50546922012-05-19 08:40:49 +00001817 Apos.s = 1;
1818 A = &Apos;
1819 }
1820
1821 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00001822 * If 1st call, pre-compute R^2 mod N
1823 */
Yuto Takano538a0cb2021-07-14 10:20:09 +01001824 if( prec_RR == NULL || prec_RR->p == NULL )
Paul Bakker5121ce52009-01-03 21:22:43 +00001825 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001826 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &RR, 1 ) );
1827 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &RR, N->n * 2 * biL ) );
1828 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &RR, &RR, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001829
Yuto Takano538a0cb2021-07-14 10:20:09 +01001830 if( prec_RR != NULL )
1831 memcpy( prec_RR, &RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001832 }
1833 else
Yuto Takano538a0cb2021-07-14 10:20:09 +01001834 memcpy( &RR, prec_RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001835
1836 /*
1837 * W[1] = A * R^2 * R^-1 mod N = A * R mod N
1838 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001839 if( mbedtls_mpi_cmp_mpi( A, N ) >= 0 )
Gilles Peskine3da1a8f2021-06-08 23:17:42 +02001840 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001841 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &W[1], A, N ) );
Gilles Peskine3da1a8f2021-06-08 23:17:42 +02001842 /* This should be a no-op because W[1] is already that large before
1843 * mbedtls_mpi_mod_mpi(), but it's necessary to avoid an overflow
1844 * in mpi_montmul() below, so let's make sure. */
Gilles Peskine2aa3f162021-06-15 21:22:48 +02001845 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[1], N->n + 1 ) );
Gilles Peskine3da1a8f2021-06-08 23:17:42 +02001846 }
Paul Bakkerc2024f42014-01-23 20:38:35 +01001847 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001848 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[1], A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001849
Gilles Peskine3da1a8f2021-06-08 23:17:42 +02001850 /* Note that this is safe because W[1] always has at least N->n limbs
1851 * (it grew above and was preserved by mbedtls_mpi_copy()). */
Gilles Peskine4e91d472020-06-04 20:55:15 +02001852 mpi_montmul( &W[1], &RR, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001853
1854 /*
1855 * X = R^2 * R^-1 mod N = R mod N
1856 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001857 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &RR ) );
Gilles Peskine4e91d472020-06-04 20:55:15 +02001858 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001859
1860 if( wsize > 1 )
1861 {
1862 /*
1863 * W[1 << (wsize - 1)] = W[1] ^ (wsize - 1)
1864 */
Paul Bakker66d5d072014-06-17 16:39:18 +02001865 j = one << ( wsize - 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001866
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001867 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[j], N->n + 1 ) );
1868 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[j], &W[1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001869
1870 for( i = 0; i < wsize - 1; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02001871 mpi_montmul( &W[j], &W[j], N, mm, &T );
Paul Bakker0d7702c2013-10-29 16:18:35 +01001872
Paul Bakker5121ce52009-01-03 21:22:43 +00001873 /*
1874 * W[i] = W[i - 1] * W[1]
1875 */
Paul Bakker66d5d072014-06-17 16:39:18 +02001876 for( i = j + 1; i < ( one << wsize ); i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00001877 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001878 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[i], N->n + 1 ) );
1879 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[i], &W[i - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001880
Gilles Peskine4e91d472020-06-04 20:55:15 +02001881 mpi_montmul( &W[i], &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001882 }
1883 }
1884
1885 nblimbs = E->n;
1886 bufsize = 0;
1887 nbits = 0;
1888 wbits = 0;
1889 state = 0;
1890
1891 while( 1 )
1892 {
1893 if( bufsize == 0 )
1894 {
Paul Bakker0d7702c2013-10-29 16:18:35 +01001895 if( nblimbs == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001896 break;
1897
Paul Bakker0d7702c2013-10-29 16:18:35 +01001898 nblimbs--;
1899
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001900 bufsize = sizeof( mbedtls_mpi_uint ) << 3;
Paul Bakker5121ce52009-01-03 21:22:43 +00001901 }
1902
1903 bufsize--;
1904
1905 ei = (E->p[nblimbs] >> bufsize) & 1;
1906
1907 /*
1908 * skip leading 0s
1909 */
1910 if( ei == 0 && state == 0 )
1911 continue;
1912
1913 if( ei == 0 && state == 1 )
1914 {
1915 /*
1916 * out of window, square X
1917 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02001918 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001919 continue;
1920 }
1921
1922 /*
1923 * add ei to current window
1924 */
1925 state = 2;
1926
1927 nbits++;
Paul Bakker66d5d072014-06-17 16:39:18 +02001928 wbits |= ( ei << ( wsize - nbits ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001929
1930 if( nbits == wsize )
1931 {
1932 /*
1933 * X = X^wsize R^-1 mod N
1934 */
1935 for( i = 0; i < wsize; i++ )
Gilles Peskine4e91d472020-06-04 20:55:15 +02001936 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001937
1938 /*
1939 * X = X * W[wbits] R^-1 mod N
1940 */
Manuel Pégourié-Gonnarde22176e2021-06-10 09:34:00 +02001941 MBEDTLS_MPI_CHK( mpi_select( &WW, W, (size_t) 1 << wsize, wbits ) );
Manuel Pégourié-Gonnard1297ef32021-03-09 11:22:20 +01001942 mpi_montmul( X, &WW, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001943
1944 state--;
1945 nbits = 0;
1946 wbits = 0;
1947 }
1948 }
1949
1950 /*
1951 * process the remaining bits
1952 */
1953 for( i = 0; i < nbits; i++ )
1954 {
Gilles Peskine4e91d472020-06-04 20:55:15 +02001955 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001956
1957 wbits <<= 1;
1958
Paul Bakker66d5d072014-06-17 16:39:18 +02001959 if( ( wbits & ( one << wsize ) ) != 0 )
Gilles Peskine4e91d472020-06-04 20:55:15 +02001960 mpi_montmul( X, &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001961 }
1962
1963 /*
1964 * X = A^E * R * R^-1 mod N = A^E mod N
1965 */
Gilles Peskine4e91d472020-06-04 20:55:15 +02001966 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00001967
Hanno Beckera4af1c42017-04-18 09:07:45 +01001968 if( neg && E->n != 0 && ( E->p[0] & 1 ) != 0 )
Paul Bakkerf6198c12012-05-16 08:02:29 +00001969 {
1970 X->s = -1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001971 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( X, N, X ) );
Paul Bakkerf6198c12012-05-16 08:02:29 +00001972 }
1973
Paul Bakker5121ce52009-01-03 21:22:43 +00001974cleanup:
1975
Paul Bakker66d5d072014-06-17 16:39:18 +02001976 for( i = ( one << ( wsize - 1 ) ); i < ( one << wsize ); i++ )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001977 mbedtls_mpi_free( &W[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +00001978
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001979 mbedtls_mpi_free( &W[1] ); mbedtls_mpi_free( &T ); mbedtls_mpi_free( &Apos );
Manuel Pégourié-Gonnard1297ef32021-03-09 11:22:20 +01001980 mbedtls_mpi_free( &WW );
Paul Bakker6c591fa2011-05-05 11:49:20 +00001981
Yuto Takano538a0cb2021-07-14 10:20:09 +01001982 if( prec_RR == NULL || prec_RR->p == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001983 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00001984
1985 return( ret );
1986}
1987
Paul Bakker5121ce52009-01-03 21:22:43 +00001988/*
1989 * Greatest common divisor: G = gcd(A, B) (HAC 14.54)
1990 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001991int mbedtls_mpi_gcd( mbedtls_mpi *G, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001992{
Janos Follath24eed8d2019-11-22 13:21:35 +00001993 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Paul Bakker23986e52011-04-24 08:57:21 +00001994 size_t lz, lzt;
Alexander Ke8ad49f2019-08-16 16:16:07 +03001995 mbedtls_mpi TA, TB;
Paul Bakker5121ce52009-01-03 21:22:43 +00001996
Hanno Becker73d7d792018-12-11 10:35:51 +00001997 MPI_VALIDATE_RET( G != NULL );
1998 MPI_VALIDATE_RET( A != NULL );
1999 MPI_VALIDATE_RET( B != NULL );
2000
Alexander Ke8ad49f2019-08-16 16:16:07 +03002001 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002002
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002003 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) );
2004 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002005
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002006 lz = mbedtls_mpi_lsb( &TA );
2007 lzt = mbedtls_mpi_lsb( &TB );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002008
Gilles Peskine27253bc2021-06-09 13:26:43 +02002009 /* The loop below gives the correct result when A==0 but not when B==0.
2010 * So have a special case for B==0. Leverage the fact that we just
2011 * calculated the lsb and lsb(B)==0 iff B is odd or 0 to make the test
2012 * slightly more efficient than cmp_int(). */
2013 if( lzt == 0 && mbedtls_mpi_get_bit( &TB, 0 ) == 0 )
2014 {
2015 ret = mbedtls_mpi_copy( G, A );
2016 goto cleanup;
2017 }
2018
Paul Bakker66d5d072014-06-17 16:39:18 +02002019 if( lzt < lz )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002020 lz = lzt;
2021
Paul Bakker5121ce52009-01-03 21:22:43 +00002022 TA.s = TB.s = 1;
2023
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002024 /* We mostly follow the procedure described in HAC 14.54, but with some
2025 * minor differences:
2026 * - Sequences of multiplications or divisions by 2 are grouped into a
2027 * single shift operation.
Gilles Peskineb09c7ee2021-06-21 18:58:39 +02002028 * - The procedure in HAC assumes that 0 < TB <= TA.
2029 * - The condition TB <= TA is not actually necessary for correctness.
2030 * TA and TB have symmetric roles except for the loop termination
2031 * condition, and the shifts at the beginning of the loop body
2032 * remove any significance from the ordering of TA vs TB before
2033 * the shifts.
2034 * - If TA = 0, the loop goes through 0 iterations and the result is
2035 * correctly TB.
2036 * - The case TB = 0 was short-circuited above.
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002037 *
2038 * For the correctness proof below, decompose the original values of
2039 * A and B as
2040 * A = sa * 2^a * A' with A'=0 or A' odd, and sa = +-1
2041 * B = sb * 2^b * B' with B'=0 or B' odd, and sb = +-1
2042 * Then gcd(A, B) = 2^{min(a,b)} * gcd(A',B'),
2043 * and gcd(A',B') is odd or 0.
2044 *
2045 * At the beginning, we have TA = |A| and TB = |B| so gcd(A,B) = gcd(TA,TB).
2046 * The code maintains the following invariant:
2047 * gcd(A,B) = 2^k * gcd(TA,TB) for some k (I)
Gilles Peskine4df3f1f2021-06-15 22:09:39 +02002048 */
2049
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002050 /* Proof that the loop terminates:
2051 * At each iteration, either the right-shift by 1 is made on a nonzero
2052 * value and the nonnegative integer bitlen(TA) + bitlen(TB) decreases
2053 * by at least 1, or the right-shift by 1 is made on zero and then
2054 * TA becomes 0 which ends the loop (TB cannot be 0 if it is right-shifted
2055 * since in that case TB is calculated from TB-TA with the condition TB>TA).
2056 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002057 while( mbedtls_mpi_cmp_int( &TA, 0 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002058 {
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002059 /* Divisions by 2 preserve the invariant (I). */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002060 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, mbedtls_mpi_lsb( &TA ) ) );
2061 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, mbedtls_mpi_lsb( &TB ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002062
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002063 /* Set either TA or TB to |TA-TB|/2. Since TA and TB are both odd,
2064 * TA-TB is even so the division by 2 has an integer result.
2065 * Invariant (I) is preserved since any odd divisor of both TA and TB
2066 * also divides |TA-TB|/2, and any odd divisor of both TA and |TA-TB|/2
Shaun Case8b0ecbc2021-12-20 21:14:10 -08002067 * also divides TB, and any odd divisor of both TB and |TA-TB|/2 also
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002068 * divides TA.
2069 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002070 if( mbedtls_mpi_cmp_mpi( &TA, &TB ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002071 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002072 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TA, &TA, &TB ) );
2073 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002074 }
2075 else
2076 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002077 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TB, &TB, &TA ) );
2078 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002079 }
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002080 /* Note that one of TA or TB is still odd. */
Paul Bakker5121ce52009-01-03 21:22:43 +00002081 }
2082
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002083 /* By invariant (I), gcd(A,B) = 2^k * gcd(TA,TB) for some k.
2084 * At the loop exit, TA = 0, so gcd(TA,TB) = TB.
2085 * - If there was at least one loop iteration, then one of TA or TB is odd,
2086 * and TA = 0, so TB is odd and gcd(TA,TB) = gcd(A',B'). In this case,
2087 * lz = min(a,b) so gcd(A,B) = 2^lz * TB.
2088 * - If there was no loop iteration, then A was 0, and gcd(A,B) = B.
Gilles Peskine4d3fd362021-06-21 11:40:38 +02002089 * In this case, lz = 0 and B = TB so gcd(A,B) = B = 2^lz * TB as well.
Gilles Peskine2a63c5b2021-06-16 13:42:04 +02002090 */
2091
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002092 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &TB, lz ) );
2093 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( G, &TB ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002094
2095cleanup:
2096
Alexander Ke8ad49f2019-08-16 16:16:07 +03002097 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002098
2099 return( ret );
2100}
2101
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002102/* Fill X with n_bytes random bytes.
2103 * X must already have room for those bytes.
Gilles Peskineafb2bd22021-06-03 11:51:09 +02002104 * The ordering of the bytes returned from the RNG is suitable for
2105 * deterministic ECDSA (see RFC 6979 §3.3 and mbedtls_mpi_random()).
Gilles Peskineebe9b6a2021-04-13 21:55:35 +02002106 * The size and sign of X are unchanged.
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002107 * n_bytes must not be 0.
2108 */
2109static int mpi_fill_random_internal(
2110 mbedtls_mpi *X, size_t n_bytes,
2111 int (*f_rng)(void *, unsigned char *, size_t), void *p_rng )
2112{
2113 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
2114 const size_t limbs = CHARS_TO_LIMBS( n_bytes );
2115 const size_t overhead = ( limbs * ciL ) - n_bytes;
2116
2117 if( X->n < limbs )
2118 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002119
Gilles Peskineebe9b6a2021-04-13 21:55:35 +02002120 memset( X->p, 0, overhead );
2121 memset( (unsigned char *) X->p + limbs * ciL, 0, ( X->n - limbs ) * ciL );
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002122 MBEDTLS_MPI_CHK( f_rng( p_rng, (unsigned char *) X->p + overhead, n_bytes ) );
Janos Follath4670f882022-07-21 18:25:42 +01002123 mbedtls_mpi_core_bigendian_to_host( X->p, limbs );
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002124
2125cleanup:
2126 return( ret );
2127}
2128
Paul Bakker33dc46b2014-04-30 16:11:39 +02002129/*
2130 * Fill X with size bytes of random.
2131 *
2132 * Use a temporary bytes representation to make sure the result is the same
Paul Bakkerc37b0ac2014-05-01 14:19:23 +02002133 * regardless of the platform endianness (useful when f_rng is actually
Paul Bakker33dc46b2014-04-30 16:11:39 +02002134 * deterministic, eg for tests).
2135 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002136int mbedtls_mpi_fill_random( mbedtls_mpi *X, size_t size,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002137 int (*f_rng)(void *, unsigned char *, size_t),
2138 void *p_rng )
Paul Bakker287781a2011-03-26 13:18:49 +00002139{
Janos Follath24eed8d2019-11-22 13:21:35 +00002140 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Hanno Becker6dab6202019-01-02 16:42:29 +00002141 size_t const limbs = CHARS_TO_LIMBS( size );
Hanno Beckerda1655a2017-10-18 14:21:44 +01002142
Hanno Becker8ce11a32018-12-19 16:18:52 +00002143 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002144 MPI_VALIDATE_RET( f_rng != NULL );
Paul Bakker33dc46b2014-04-30 16:11:39 +02002145
Hanno Beckerda1655a2017-10-18 14:21:44 +01002146 /* Ensure that target MPI has exactly the necessary number of limbs */
Gilles Peskineed32b572021-06-02 22:17:52 +02002147 MBEDTLS_MPI_CHK( mbedtls_mpi_resize_clear( X, limbs ) );
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002148 if( size == 0 )
2149 return( 0 );
Paul Bakker287781a2011-03-26 13:18:49 +00002150
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002151 ret = mpi_fill_random_internal( X, size, f_rng, p_rng );
Paul Bakker287781a2011-03-26 13:18:49 +00002152
2153cleanup:
2154 return( ret );
2155}
2156
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002157int mbedtls_mpi_random( mbedtls_mpi *X,
2158 mbedtls_mpi_sint min,
2159 const mbedtls_mpi *N,
2160 int (*f_rng)(void *, unsigned char *, size_t),
2161 void *p_rng )
2162{
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002163 int ret = MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
Gilles Peskinee5381682021-04-13 21:23:25 +02002164 int count;
Gilles Peskine5b0589e2021-04-13 21:09:10 +02002165 unsigned lt_lower = 1, lt_upper = 0;
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002166 size_t n_bits = mbedtls_mpi_bitlen( N );
2167 size_t n_bytes = ( n_bits + 7 ) / 8;
Gilles Peskine5b0589e2021-04-13 21:09:10 +02002168 mbedtls_mpi lower_bound;
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002169
Gilles Peskine1e918f42021-03-29 22:14:51 +02002170 if( min < 0 )
2171 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
2172 if( mbedtls_mpi_cmp_int( N, min ) <= 0 )
2173 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
2174
Gilles Peskinee5381682021-04-13 21:23:25 +02002175 /*
2176 * When min == 0, each try has at worst a probability 1/2 of failing
2177 * (the msb has a probability 1/2 of being 0, and then the result will
2178 * be < N), so after 30 tries failure probability is a most 2**(-30).
2179 *
2180 * When N is just below a power of 2, as is the case when generating
Gilles Peskinee842e582021-04-15 11:45:19 +02002181 * a random scalar on most elliptic curves, 1 try is enough with
Gilles Peskinee5381682021-04-13 21:23:25 +02002182 * overwhelming probability. When N is just above a power of 2,
Gilles Peskinee842e582021-04-15 11:45:19 +02002183 * as when generating a random scalar on secp224k1, each try has
Gilles Peskinee5381682021-04-13 21:23:25 +02002184 * a probability of failing that is almost 1/2.
2185 *
2186 * The probabilities are almost the same if min is nonzero but negligible
2187 * compared to N. This is always the case when N is crypto-sized, but
2188 * it's convenient to support small N for testing purposes. When N
2189 * is small, use a higher repeat count, otherwise the probability of
2190 * failure is macroscopic.
2191 */
Gilles Peskine87823d72021-06-02 21:18:59 +02002192 count = ( n_bytes > 4 ? 30 : 250 );
Gilles Peskinee5381682021-04-13 21:23:25 +02002193
Gilles Peskine5b0589e2021-04-13 21:09:10 +02002194 mbedtls_mpi_init( &lower_bound );
2195
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002196 /* Ensure that target MPI has exactly the same number of limbs
2197 * as the upper bound, even if the upper bound has leading zeros.
2198 * This is necessary for the mbedtls_mpi_lt_mpi_ct() check. */
Gilles Peskineed32b572021-06-02 22:17:52 +02002199 MBEDTLS_MPI_CHK( mbedtls_mpi_resize_clear( X, N->n ) );
Gilles Peskine5b0589e2021-04-13 21:09:10 +02002200 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &lower_bound, N->n ) );
2201 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &lower_bound, min ) );
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002202
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002203 /*
2204 * Match the procedure given in RFC 6979 §3.3 (deterministic ECDSA)
2205 * when f_rng is a suitably parametrized instance of HMAC_DRBG:
2206 * - use the same byte ordering;
2207 * - keep the leftmost n_bits bits of the generated octet string;
2208 * - try until result is in the desired range.
2209 * This also avoids any bias, which is especially important for ECDSA.
2210 */
2211 do
2212 {
Gilles Peskine1a7df4e2021-04-01 15:57:18 +02002213 MBEDTLS_MPI_CHK( mpi_fill_random_internal( X, n_bytes, f_rng, p_rng ) );
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002214 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( X, 8 * n_bytes - n_bits ) );
2215
Gilles Peskinee5381682021-04-13 21:23:25 +02002216 if( --count == 0 )
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002217 {
2218 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
2219 goto cleanup;
2220 }
2221
Gilles Peskine5b0589e2021-04-13 21:09:10 +02002222 MBEDTLS_MPI_CHK( mbedtls_mpi_lt_mpi_ct( X, &lower_bound, &lt_lower ) );
2223 MBEDTLS_MPI_CHK( mbedtls_mpi_lt_mpi_ct( X, N, &lt_upper ) );
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002224 }
Gilles Peskine5b0589e2021-04-13 21:09:10 +02002225 while( lt_lower != 0 || lt_upper == 0 );
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002226
2227cleanup:
Gilles Peskine5b0589e2021-04-13 21:09:10 +02002228 mbedtls_mpi_free( &lower_bound );
Gilles Peskine02ac93a2021-03-29 22:02:55 +02002229 return( ret );
2230}
2231
Paul Bakker5121ce52009-01-03 21:22:43 +00002232/*
2233 * Modular inverse: X = A^-1 mod N (HAC 14.61 / 14.64)
2234 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002235int mbedtls_mpi_inv_mod( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002236{
Janos Follath24eed8d2019-11-22 13:21:35 +00002237 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002238 mbedtls_mpi G, TA, TU, U1, U2, TB, TV, V1, V2;
Hanno Becker73d7d792018-12-11 10:35:51 +00002239 MPI_VALIDATE_RET( X != NULL );
2240 MPI_VALIDATE_RET( A != NULL );
2241 MPI_VALIDATE_RET( N != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00002242
Hanno Becker4bcb4912017-04-18 15:49:39 +01002243 if( mbedtls_mpi_cmp_int( N, 1 ) <= 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002244 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002245
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002246 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TU ); mbedtls_mpi_init( &U1 ); mbedtls_mpi_init( &U2 );
2247 mbedtls_mpi_init( &G ); mbedtls_mpi_init( &TB ); mbedtls_mpi_init( &TV );
2248 mbedtls_mpi_init( &V1 ); mbedtls_mpi_init( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002249
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002250 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &G, A, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002251
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002252 if( mbedtls_mpi_cmp_int( &G, 1 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002253 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002254 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002255 goto cleanup;
2256 }
2257
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002258 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &TA, A, N ) );
2259 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TU, &TA ) );
2260 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, N ) );
2261 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TV, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002262
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002263 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U1, 1 ) );
2264 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U2, 0 ) );
2265 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V1, 0 ) );
2266 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002267
2268 do
2269 {
2270 while( ( TU.p[0] & 1 ) == 0 )
2271 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002272 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TU, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002273
2274 if( ( U1.p[0] & 1 ) != 0 || ( U2.p[0] & 1 ) != 0 )
2275 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002276 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &U1, &U1, &TB ) );
2277 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002278 }
2279
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002280 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U1, 1 ) );
2281 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002282 }
2283
2284 while( ( TV.p[0] & 1 ) == 0 )
2285 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002286 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TV, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002287
2288 if( ( V1.p[0] & 1 ) != 0 || ( V2.p[0] & 1 ) != 0 )
2289 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002290 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, &TB ) );
2291 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002292 }
2293
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002294 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V1, 1 ) );
2295 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002296 }
2297
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002298 if( mbedtls_mpi_cmp_mpi( &TU, &TV ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002299 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002300 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TU, &TU, &TV ) );
2301 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U1, &U1, &V1 ) );
2302 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &V2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002303 }
2304 else
2305 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002306 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TV, &TV, &TU ) );
2307 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, &U1 ) );
2308 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &U2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002309 }
2310 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002311 while( mbedtls_mpi_cmp_int( &TU, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002312
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002313 while( mbedtls_mpi_cmp_int( &V1, 0 ) < 0 )
2314 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002315
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002316 while( mbedtls_mpi_cmp_mpi( &V1, N ) >= 0 )
2317 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002318
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002319 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &V1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002320
2321cleanup:
2322
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002323 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TU ); mbedtls_mpi_free( &U1 ); mbedtls_mpi_free( &U2 );
2324 mbedtls_mpi_free( &G ); mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TV );
2325 mbedtls_mpi_free( &V1 ); mbedtls_mpi_free( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002326
2327 return( ret );
2328}
2329
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002330#if defined(MBEDTLS_GENPRIME)
Paul Bakkerd9374b02012-11-02 11:02:58 +00002331
Paul Bakker5121ce52009-01-03 21:22:43 +00002332static const int small_prime[] =
2333{
2334 3, 5, 7, 11, 13, 17, 19, 23,
2335 29, 31, 37, 41, 43, 47, 53, 59,
2336 61, 67, 71, 73, 79, 83, 89, 97,
2337 101, 103, 107, 109, 113, 127, 131, 137,
2338 139, 149, 151, 157, 163, 167, 173, 179,
2339 181, 191, 193, 197, 199, 211, 223, 227,
2340 229, 233, 239, 241, 251, 257, 263, 269,
2341 271, 277, 281, 283, 293, 307, 311, 313,
2342 317, 331, 337, 347, 349, 353, 359, 367,
2343 373, 379, 383, 389, 397, 401, 409, 419,
2344 421, 431, 433, 439, 443, 449, 457, 461,
2345 463, 467, 479, 487, 491, 499, 503, 509,
2346 521, 523, 541, 547, 557, 563, 569, 571,
2347 577, 587, 593, 599, 601, 607, 613, 617,
2348 619, 631, 641, 643, 647, 653, 659, 661,
2349 673, 677, 683, 691, 701, 709, 719, 727,
2350 733, 739, 743, 751, 757, 761, 769, 773,
2351 787, 797, 809, 811, 821, 823, 827, 829,
2352 839, 853, 857, 859, 863, 877, 881, 883,
2353 887, 907, 911, 919, 929, 937, 941, 947,
2354 953, 967, 971, 977, 983, 991, 997, -103
2355};
2356
2357/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002358 * Small divisors test (X must be positive)
2359 *
2360 * Return values:
2361 * 0: no small factor (possible prime, more tests needed)
2362 * 1: certain prime
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002363 * MBEDTLS_ERR_MPI_NOT_ACCEPTABLE: certain non-prime
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002364 * other negative: error
Paul Bakker5121ce52009-01-03 21:22:43 +00002365 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002366static int mpi_check_small_factors( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +00002367{
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002368 int ret = 0;
2369 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002370 mbedtls_mpi_uint r;
Paul Bakker5121ce52009-01-03 21:22:43 +00002371
Paul Bakker5121ce52009-01-03 21:22:43 +00002372 if( ( X->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002373 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002374
2375 for( i = 0; small_prime[i] > 0; i++ )
2376 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002377 if( mbedtls_mpi_cmp_int( X, small_prime[i] ) <= 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002378 return( 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002379
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002380 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, small_prime[i] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002381
2382 if( r == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002383 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002384 }
2385
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002386cleanup:
2387 return( ret );
2388}
2389
2390/*
2391 * Miller-Rabin pseudo-primality test (HAC 4.24)
2392 */
Janos Follathda31fa12018-09-03 14:45:23 +01002393static int mpi_miller_rabin( const mbedtls_mpi *X, size_t rounds,
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002394 int (*f_rng)(void *, unsigned char *, size_t),
2395 void *p_rng )
2396{
Pascal Junodb99183d2015-03-11 16:49:45 +01002397 int ret, count;
Janos Follathda31fa12018-09-03 14:45:23 +01002398 size_t i, j, k, s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002399 mbedtls_mpi W, R, T, A, RR;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002400
Hanno Becker8ce11a32018-12-19 16:18:52 +00002401 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002402 MPI_VALIDATE_RET( f_rng != NULL );
2403
2404 mbedtls_mpi_init( &W ); mbedtls_mpi_init( &R );
2405 mbedtls_mpi_init( &T ); mbedtls_mpi_init( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002406 mbedtls_mpi_init( &RR );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002407
Paul Bakker5121ce52009-01-03 21:22:43 +00002408 /*
2409 * W = |X| - 1
2410 * R = W >> lsb( W )
2411 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002412 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( &W, X, 1 ) );
2413 s = mbedtls_mpi_lsb( &W );
2414 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &R, &W ) );
2415 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &R, s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002416
Janos Follathda31fa12018-09-03 14:45:23 +01002417 for( i = 0; i < rounds; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002418 {
2419 /*
2420 * pick a random A, 1 < A < |X| - 1
2421 */
Pascal Junodb99183d2015-03-11 16:49:45 +01002422 count = 0;
2423 do {
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002424 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( &A, X->n * ciL, f_rng, p_rng ) );
Pascal Junodb99183d2015-03-11 16:49:45 +01002425
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002426 j = mbedtls_mpi_bitlen( &A );
2427 k = mbedtls_mpi_bitlen( &W );
Pascal Junodb99183d2015-03-11 16:49:45 +01002428 if (j > k) {
Darryl Greene3f95ed2018-10-02 13:21:35 +01002429 A.p[A.n - 1] &= ( (mbedtls_mpi_uint) 1 << ( k - ( A.n - 1 ) * biL - 1 ) ) - 1;
Pascal Junodb99183d2015-03-11 16:49:45 +01002430 }
2431
2432 if (count++ > 30) {
Jens Wiklanderf08aa3e2019-01-17 13:30:57 +01002433 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
2434 goto cleanup;
Pascal Junodb99183d2015-03-11 16:49:45 +01002435 }
2436
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002437 } while ( mbedtls_mpi_cmp_mpi( &A, &W ) >= 0 ||
2438 mbedtls_mpi_cmp_int( &A, 1 ) <= 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002439
2440 /*
2441 * A = A^R mod |X|
2442 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002443 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &A, &A, &R, X, &RR ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002444
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002445 if( mbedtls_mpi_cmp_mpi( &A, &W ) == 0 ||
2446 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002447 continue;
2448
2449 j = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002450 while( j < s && mbedtls_mpi_cmp_mpi( &A, &W ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002451 {
2452 /*
2453 * A = A * A mod |X|
2454 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002455 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &T, &A, &A ) );
2456 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &A, &T, X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002457
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002458 if( mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002459 break;
2460
2461 j++;
2462 }
2463
2464 /*
2465 * not prime if A != |X| - 1 or A == 1
2466 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002467 if( mbedtls_mpi_cmp_mpi( &A, &W ) != 0 ||
2468 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002469 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002470 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002471 break;
2472 }
2473 }
2474
2475cleanup:
Hanno Becker73d7d792018-12-11 10:35:51 +00002476 mbedtls_mpi_free( &W ); mbedtls_mpi_free( &R );
2477 mbedtls_mpi_free( &T ); mbedtls_mpi_free( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002478 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002479
2480 return( ret );
2481}
2482
2483/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002484 * Pseudo-primality test: small factors, then Miller-Rabin
2485 */
Janos Follatha0b67c22018-09-18 14:48:23 +01002486int mbedtls_mpi_is_prime_ext( const mbedtls_mpi *X, int rounds,
2487 int (*f_rng)(void *, unsigned char *, size_t),
2488 void *p_rng )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002489{
Janos Follath24eed8d2019-11-22 13:21:35 +00002490 int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002491 mbedtls_mpi XX;
Hanno Becker8ce11a32018-12-19 16:18:52 +00002492 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002493 MPI_VALIDATE_RET( f_rng != NULL );
Manuel Pégourié-Gonnard7f4ed672014-10-14 20:56:02 +02002494
2495 XX.s = 1;
2496 XX.n = X->n;
2497 XX.p = X->p;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002498
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002499 if( mbedtls_mpi_cmp_int( &XX, 0 ) == 0 ||
2500 mbedtls_mpi_cmp_int( &XX, 1 ) == 0 )
2501 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002502
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002503 if( mbedtls_mpi_cmp_int( &XX, 2 ) == 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002504 return( 0 );
2505
2506 if( ( ret = mpi_check_small_factors( &XX ) ) != 0 )
2507 {
2508 if( ret == 1 )
2509 return( 0 );
2510
2511 return( ret );
2512 }
2513
Janos Follathda31fa12018-09-03 14:45:23 +01002514 return( mpi_miller_rabin( &XX, rounds, f_rng, p_rng ) );
Janos Follathf301d232018-08-14 13:34:01 +01002515}
2516
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002517/*
Paul Bakker5121ce52009-01-03 21:22:43 +00002518 * Prime number generation
Jethro Beekman66689272018-02-14 19:24:10 -08002519 *
Janos Follathf301d232018-08-14 13:34:01 +01002520 * To generate an RSA key in a way recommended by FIPS 186-4, both primes must
2521 * be either 1024 bits or 1536 bits long, and flags must contain
2522 * MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR.
Paul Bakker5121ce52009-01-03 21:22:43 +00002523 */
Janos Follath7c025a92018-08-14 11:08:41 +01002524int mbedtls_mpi_gen_prime( mbedtls_mpi *X, size_t nbits, int flags,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002525 int (*f_rng)(void *, unsigned char *, size_t),
2526 void *p_rng )
Paul Bakker5121ce52009-01-03 21:22:43 +00002527{
Jethro Beekman66689272018-02-14 19:24:10 -08002528#ifdef MBEDTLS_HAVE_INT64
2529// ceil(2^63.5)
2530#define CEIL_MAXUINT_DIV_SQRT2 0xb504f333f9de6485ULL
2531#else
2532// ceil(2^31.5)
2533#define CEIL_MAXUINT_DIV_SQRT2 0xb504f334U
2534#endif
2535 int ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker23986e52011-04-24 08:57:21 +00002536 size_t k, n;
Janos Follathda31fa12018-09-03 14:45:23 +01002537 int rounds;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002538 mbedtls_mpi_uint r;
2539 mbedtls_mpi Y;
Paul Bakker5121ce52009-01-03 21:22:43 +00002540
Hanno Becker8ce11a32018-12-19 16:18:52 +00002541 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002542 MPI_VALIDATE_RET( f_rng != NULL );
2543
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002544 if( nbits < 3 || nbits > MBEDTLS_MPI_MAX_BITS )
2545 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002546
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002547 mbedtls_mpi_init( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002548
2549 n = BITS_TO_LIMBS( nbits );
2550
Janos Follathda31fa12018-09-03 14:45:23 +01002551 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR ) == 0 )
2552 {
2553 /*
2554 * 2^-80 error probability, number of rounds chosen per HAC, table 4.4
2555 */
2556 rounds = ( ( nbits >= 1300 ) ? 2 : ( nbits >= 850 ) ? 3 :
2557 ( nbits >= 650 ) ? 4 : ( nbits >= 350 ) ? 8 :
2558 ( nbits >= 250 ) ? 12 : ( nbits >= 150 ) ? 18 : 27 );
2559 }
2560 else
2561 {
2562 /*
2563 * 2^-100 error probability, number of rounds computed based on HAC,
2564 * fact 4.48
2565 */
2566 rounds = ( ( nbits >= 1450 ) ? 4 : ( nbits >= 1150 ) ? 5 :
2567 ( nbits >= 1000 ) ? 6 : ( nbits >= 850 ) ? 7 :
2568 ( nbits >= 750 ) ? 8 : ( nbits >= 500 ) ? 13 :
2569 ( nbits >= 250 ) ? 28 : ( nbits >= 150 ) ? 40 : 51 );
2570 }
2571
Jethro Beekman66689272018-02-14 19:24:10 -08002572 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002573 {
Jethro Beekman66689272018-02-14 19:24:10 -08002574 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( X, n * ciL, f_rng, p_rng ) );
2575 /* make sure generated number is at least (nbits-1)+0.5 bits (FIPS 186-4 §B.3.3 steps 4.4, 5.5) */
2576 if( X->p[n-1] < CEIL_MAXUINT_DIV_SQRT2 ) continue;
2577
2578 k = n * biL;
2579 if( k > nbits ) MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( X, k - nbits ) );
2580 X->p[0] |= 1;
2581
Janos Follath7c025a92018-08-14 11:08:41 +01002582 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_DH ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002583 {
Janos Follatha0b67c22018-09-18 14:48:23 +01002584 ret = mbedtls_mpi_is_prime_ext( X, rounds, f_rng, p_rng );
Jethro Beekman66689272018-02-14 19:24:10 -08002585
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002586 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
Paul Bakker5121ce52009-01-03 21:22:43 +00002587 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002588 }
Jethro Beekman66689272018-02-14 19:24:10 -08002589 else
Paul Bakker5121ce52009-01-03 21:22:43 +00002590 {
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002591 /*
Tom Cosgrovece7f18c2022-07-28 05:50:56 +01002592 * A necessary condition for Y and X = 2Y + 1 to be prime
Jethro Beekman66689272018-02-14 19:24:10 -08002593 * is X = 2 mod 3 (which is equivalent to Y = 2 mod 3).
2594 * Make sure it is satisfied, while keeping X = 3 mod 4
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002595 */
Jethro Beekman66689272018-02-14 19:24:10 -08002596
2597 X->p[0] |= 2;
2598
2599 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, 3 ) );
2600 if( r == 0 )
2601 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 8 ) );
2602 else if( r == 1 )
2603 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 4 ) );
2604
2605 /* Set Y = (X-1) / 2, which is X / 2 because X is odd */
2606 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, X ) );
2607 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, 1 ) );
2608
2609 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002610 {
Jethro Beekman66689272018-02-14 19:24:10 -08002611 /*
2612 * First, check small factors for X and Y
2613 * before doing Miller-Rabin on any of them
2614 */
2615 if( ( ret = mpi_check_small_factors( X ) ) == 0 &&
2616 ( ret = mpi_check_small_factors( &Y ) ) == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002617 ( ret = mpi_miller_rabin( X, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002618 == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002619 ( ret = mpi_miller_rabin( &Y, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002620 == 0 )
Jethro Beekman66689272018-02-14 19:24:10 -08002621 goto cleanup;
2622
2623 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
2624 goto cleanup;
2625
2626 /*
2627 * Next candidates. We want to preserve Y = (X-1) / 2 and
2628 * Y = 1 mod 2 and Y = 2 mod 3 (eq X = 3 mod 4 and X = 2 mod 3)
2629 * so up Y by 6 and X by 12.
2630 */
2631 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 12 ) );
2632 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( &Y, &Y, 6 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002633 }
Paul Bakker5121ce52009-01-03 21:22:43 +00002634 }
2635 }
2636
2637cleanup:
2638
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002639 mbedtls_mpi_free( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002640
2641 return( ret );
2642}
2643
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002644#endif /* MBEDTLS_GENPRIME */
Paul Bakker5121ce52009-01-03 21:22:43 +00002645
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002646#if defined(MBEDTLS_SELF_TEST)
Paul Bakker5121ce52009-01-03 21:22:43 +00002647
Paul Bakker23986e52011-04-24 08:57:21 +00002648#define GCD_PAIR_COUNT 3
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002649
2650static const int gcd_pairs[GCD_PAIR_COUNT][3] =
2651{
2652 { 693, 609, 21 },
2653 { 1764, 868, 28 },
2654 { 768454923, 542167814, 1 }
2655};
2656
Paul Bakker5121ce52009-01-03 21:22:43 +00002657/*
2658 * Checkup routine
2659 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002660int mbedtls_mpi_self_test( int verbose )
Paul Bakker5121ce52009-01-03 21:22:43 +00002661{
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002662 int ret, i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002663 mbedtls_mpi A, E, N, X, Y, U, V;
Paul Bakker5121ce52009-01-03 21:22:43 +00002664
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002665 mbedtls_mpi_init( &A ); mbedtls_mpi_init( &E ); mbedtls_mpi_init( &N ); mbedtls_mpi_init( &X );
2666 mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &U ); mbedtls_mpi_init( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002667
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002668 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &A, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002669 "EFE021C2645FD1DC586E69184AF4A31E" \
2670 "D5F53E93B5F123FA41680867BA110131" \
2671 "944FE7952E2517337780CB0DB80E61AA" \
2672 "E7C8DDC6C5C6AADEB34EB38A2F40D5E6" ) );
2673
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002674 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &E, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002675 "B2E7EFD37075B9F03FF989C7C5051C20" \
2676 "34D2A323810251127E7BF8625A4F49A5" \
2677 "F3E27F4DA8BD59C47D6DAABA4C8127BD" \
2678 "5B5C25763222FEFCCFC38B832366C29E" ) );
2679
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002680 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &N, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002681 "0066A198186C18C10B2F5ED9B522752A" \
2682 "9830B69916E535C8F047518A889A43A5" \
2683 "94B6BED27A168D31D4A52F88925AA8F5" ) );
2684
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002685 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002686
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002687 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002688 "602AB7ECA597A3D6B56FF9829A5E8B85" \
2689 "9E857EA95A03512E2BAE7391688D264A" \
2690 "A5663B0341DB9CCFD2C4C5F421FEC814" \
2691 "8001B72E848A38CAE1C65F78E56ABDEF" \
2692 "E12D3C039B8A02D6BE593F0BBBDA56F1" \
2693 "ECF677152EF804370C1A305CAF3B5BF1" \
2694 "30879B56C61DE584A0F53A2447A51E" ) );
2695
2696 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002697 mbedtls_printf( " MPI test #1 (mul_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002698
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002699 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002700 {
2701 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002702 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002703
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002704 ret = 1;
2705 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002706 }
2707
2708 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002709 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002710
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002711 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( &X, &Y, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002712
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002713 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002714 "256567336059E52CAE22925474705F39A94" ) );
2715
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002716 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &V, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002717 "6613F26162223DF488E9CD48CC132C7A" \
2718 "0AC93C701B001B092E4E5B9F73BCD27B" \
2719 "9EE50D0657C77F374E903CDFA4C642" ) );
2720
2721 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002722 mbedtls_printf( " MPI test #2 (div_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002723
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002724 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 ||
2725 mbedtls_mpi_cmp_mpi( &Y, &V ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002726 {
2727 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002728 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002729
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002730 ret = 1;
2731 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002732 }
2733
2734 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002735 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002736
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002737 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &X, &A, &E, &N, NULL ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002738
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002739 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002740 "36E139AEA55215609D2816998ED020BB" \
2741 "BD96C37890F65171D948E9BC7CBAA4D9" \
2742 "325D24D6A3C12710F10A09FA08AB87" ) );
2743
2744 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002745 mbedtls_printf( " MPI test #3 (exp_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002746
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002747 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002748 {
2749 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002750 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002751
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002752 ret = 1;
2753 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002754 }
2755
2756 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002757 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002758
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002759 MBEDTLS_MPI_CHK( mbedtls_mpi_inv_mod( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002760
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002761 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002762 "003A0AAEDD7E784FC07D8F9EC6E3BFD5" \
2763 "C3DBA76456363A10869622EAC2DD84EC" \
2764 "C5B8A74DAC4D09E03B5E0BE779F2DF61" ) );
2765
2766 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002767 mbedtls_printf( " MPI test #4 (inv_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002768
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002769 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002770 {
2771 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002772 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002773
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002774 ret = 1;
2775 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002776 }
2777
2778 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002779 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002780
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002781 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002782 mbedtls_printf( " MPI test #5 (simple gcd): " );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002783
Paul Bakker66d5d072014-06-17 16:39:18 +02002784 for( i = 0; i < GCD_PAIR_COUNT; i++ )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002785 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002786 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &X, gcd_pairs[i][0] ) );
2787 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Y, gcd_pairs[i][1] ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002788
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002789 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &A, &X, &Y ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002790
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002791 if( mbedtls_mpi_cmp_int( &A, gcd_pairs[i][2] ) != 0 )
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002792 {
2793 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002794 mbedtls_printf( "failed at %d\n", i );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002795
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002796 ret = 1;
2797 goto cleanup;
2798 }
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002799 }
2800
2801 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002802 mbedtls_printf( "passed\n" );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002803
Paul Bakker5121ce52009-01-03 21:22:43 +00002804cleanup:
2805
2806 if( ret != 0 && verbose != 0 )
Kenneth Soerensen518d4352020-04-01 17:22:45 +02002807 mbedtls_printf( "Unexpected error, return code = %08X\n", (unsigned int) ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00002808
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002809 mbedtls_mpi_free( &A ); mbedtls_mpi_free( &E ); mbedtls_mpi_free( &N ); mbedtls_mpi_free( &X );
2810 mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &U ); mbedtls_mpi_free( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002811
2812 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002813 mbedtls_printf( "\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002814
2815 return( ret );
2816}
2817
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002818#endif /* MBEDTLS_SELF_TEST */
Paul Bakker5121ce52009-01-03 21:22:43 +00002819
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002820#endif /* MBEDTLS_BIGNUM_C */