blob: 64bf095cee7c73f32385335803f2bd89cfbbd4e1 [file] [log] [blame]
Paul Bakker5121ce52009-01-03 21:22:43 +00001/*
2 * Multi-precision integer library
3 *
Manuel Pégourié-Gonnard6fb81872015-07-27 11:11:48 +02004 * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved
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 Bakkerb96f1542010-07-18 20:36:00 +000018 *
Manuel Pégourié-Gonnardfe446432015-03-06 13:17:10 +000019 * This file is part of mbed TLS (https://tls.mbed.org)
Paul Bakker5121ce52009-01-03 21:22:43 +000020 */
Simon Butcher15b15d12015-11-26 19:35:03 +000021
Paul Bakker5121ce52009-01-03 21:22:43 +000022/*
Simon Butcher15b15d12015-11-26 19:35:03 +000023 * The following sources were referenced in the design of this Multi-precision
24 * Integer library:
Paul Bakker5121ce52009-01-03 21:22:43 +000025 *
Simon Butcher15b15d12015-11-26 19:35:03 +000026 * [1] Handbook of Applied Cryptography - 1997
27 * Menezes, van Oorschot and Vanstone
28 *
29 * [2] Multi-Precision Math
30 * Tom St Denis
31 * https://github.com/libtom/libtommath/blob/develop/tommath.pdf
32 *
33 * [3] GNU Multi-Precision Arithmetic Library
34 * https://gmplib.org/manual/index.html
35 *
Simon Butcherf5ba0452015-12-27 23:01:55 +000036 */
Paul Bakker5121ce52009-01-03 21:22:43 +000037
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020038#if !defined(MBEDTLS_CONFIG_FILE)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000039#include "mbedtls/config.h"
Manuel Pégourié-Gonnardcef4ad22014-04-29 12:39:06 +020040#else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020041#include MBEDTLS_CONFIG_FILE
Manuel Pégourié-Gonnardcef4ad22014-04-29 12:39:06 +020042#endif
Paul Bakker5121ce52009-01-03 21:22:43 +000043
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020044#if defined(MBEDTLS_BIGNUM_C)
Paul Bakker5121ce52009-01-03 21:22:43 +000045
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000046#include "mbedtls/bignum.h"
47#include "mbedtls/bn_mul.h"
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050048#include "mbedtls/platform_util.h"
Paul Bakker5121ce52009-01-03 21:22:43 +000049
Rich Evans00ab4702015-02-06 13:43:58 +000050#include <string.h>
51
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020052#if defined(MBEDTLS_PLATFORM_C)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000053#include "mbedtls/platform.h"
Paul Bakker6e339b52013-07-03 13:37:05 +020054#else
Rich Evans00ab4702015-02-06 13:43:58 +000055#include <stdio.h>
56#include <stdlib.h>
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020057#define mbedtls_printf printf
Manuel Pégourié-Gonnard7551cb92015-05-26 16:04:06 +020058#define mbedtls_calloc calloc
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020059#define mbedtls_free free
Paul Bakker6e339b52013-07-03 13:37:05 +020060#endif
61
Hanno Becker73d7d792018-12-11 10:35:51 +000062#define MPI_VALIDATE_RET( cond ) \
63 MBEDTLS_INTERNAL_VALIDATE_RET( cond, MBEDTLS_ERR_MPI_BAD_INPUT_DATA )
64#define MPI_VALIDATE( cond ) \
65 MBEDTLS_INTERNAL_VALIDATE( cond )
66
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020067#define ciL (sizeof(mbedtls_mpi_uint)) /* chars in limb */
Paul Bakker5121ce52009-01-03 21:22:43 +000068#define biL (ciL << 3) /* bits in limb */
69#define biH (ciL << 2) /* half limb size */
70
Manuel Pégourié-Gonnard2d708342015-10-05 15:23:11 +010071#define MPI_SIZE_T_MAX ( (size_t) -1 ) /* SIZE_T_MAX is not standard */
72
Paul Bakker5121ce52009-01-03 21:22:43 +000073/*
74 * Convert between bits/chars and number of limbs
Manuel Pégourié-Gonnard58fb4952015-09-28 13:48:04 +020075 * Divide first in order to avoid potential overflows
Paul Bakker5121ce52009-01-03 21:22:43 +000076 */
Manuel Pégourié-Gonnard58fb4952015-09-28 13:48:04 +020077#define BITS_TO_LIMBS(i) ( (i) / biL + ( (i) % biL != 0 ) )
78#define CHARS_TO_LIMBS(i) ( (i) / ciL + ( (i) % ciL != 0 ) )
Paul Bakker5121ce52009-01-03 21:22:43 +000079
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050080/* Implementation that should never be optimized out by the compiler */
Andres Amaya Garcia6698d2f2018-04-24 08:39:07 -050081static void mbedtls_mpi_zeroize( mbedtls_mpi_uint *v, size_t n )
82{
Andres Amaya Garcia1f6301b2018-04-17 09:51:09 -050083 mbedtls_platform_zeroize( v, ciL * n );
84}
85
Paul Bakker5121ce52009-01-03 21:22:43 +000086/*
Paul Bakker6c591fa2011-05-05 11:49:20 +000087 * Initialize one MPI
Paul Bakker5121ce52009-01-03 21:22:43 +000088 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020089void mbedtls_mpi_init( mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +000090{
Hanno Becker73d7d792018-12-11 10:35:51 +000091 MPI_VALIDATE( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +000092
Paul Bakker6c591fa2011-05-05 11:49:20 +000093 X->s = 1;
94 X->n = 0;
95 X->p = NULL;
Paul Bakker5121ce52009-01-03 21:22:43 +000096}
97
98/*
Paul Bakker6c591fa2011-05-05 11:49:20 +000099 * Unallocate one MPI
Paul Bakker5121ce52009-01-03 21:22:43 +0000100 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200101void mbedtls_mpi_free( mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000102{
Paul Bakker6c591fa2011-05-05 11:49:20 +0000103 if( X == NULL )
104 return;
Paul Bakker5121ce52009-01-03 21:22:43 +0000105
Paul Bakker6c591fa2011-05-05 11:49:20 +0000106 if( X->p != NULL )
Paul Bakker5121ce52009-01-03 21:22:43 +0000107 {
Alexey Skalozube17a8da2016-01-13 17:19:33 +0200108 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200109 mbedtls_free( X->p );
Paul Bakker5121ce52009-01-03 21:22:43 +0000110 }
111
Paul Bakker6c591fa2011-05-05 11:49:20 +0000112 X->s = 1;
113 X->n = 0;
114 X->p = NULL;
Paul Bakker5121ce52009-01-03 21:22:43 +0000115}
116
117/*
118 * Enlarge to the specified number of limbs
119 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200120int mbedtls_mpi_grow( mbedtls_mpi *X, size_t nblimbs )
Paul Bakker5121ce52009-01-03 21:22:43 +0000121{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200122 mbedtls_mpi_uint *p;
Hanno Becker73d7d792018-12-11 10:35:51 +0000123 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000124
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200125 if( nblimbs > MBEDTLS_MPI_MAX_LIMBS )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200126 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Paul Bakkerf9688572011-05-05 10:00:45 +0000127
Paul Bakker5121ce52009-01-03 21:22:43 +0000128 if( X->n < nblimbs )
129 {
Simon Butcher29176892016-05-20 00:19:09 +0100130 if( ( p = (mbedtls_mpi_uint*)mbedtls_calloc( nblimbs, ciL ) ) == NULL )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200131 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Paul Bakker5121ce52009-01-03 21:22:43 +0000132
Paul Bakker5121ce52009-01-03 21:22:43 +0000133 if( X->p != NULL )
134 {
135 memcpy( p, X->p, X->n * ciL );
Alexey Skalozube17a8da2016-01-13 17:19:33 +0200136 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200137 mbedtls_free( X->p );
Paul Bakker5121ce52009-01-03 21:22:43 +0000138 }
139
140 X->n = nblimbs;
141 X->p = p;
142 }
143
144 return( 0 );
145}
146
147/*
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100148 * Resize down as much as possible,
149 * while keeping at least the specified number of limbs
150 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200151int mbedtls_mpi_shrink( mbedtls_mpi *X, size_t nblimbs )
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100152{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200153 mbedtls_mpi_uint *p;
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100154 size_t i;
Hanno Becker73d7d792018-12-11 10:35:51 +0000155 MPI_VALIDATE_RET( X != NULL );
156
157 if( nblimbs > MBEDTLS_MPI_MAX_LIMBS )
158 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100159
Gilles Peskine27c15c72020-01-20 21:17:43 +0100160 /* Actually resize up if there are currently fewer than nblimbs limbs. */
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100161 if( X->n <= nblimbs )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200162 return( mbedtls_mpi_grow( X, nblimbs ) );
Gilles Peskine56427c22020-01-21 13:59:51 +0100163 /* After this point, then X->n > nblimbs and in particular X->n > 0. */
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100164
165 for( i = X->n - 1; i > 0; i-- )
166 if( X->p[i] != 0 )
167 break;
168 i++;
169
170 if( i < nblimbs )
171 i = nblimbs;
172
Simon Butcher29176892016-05-20 00:19:09 +0100173 if( ( p = (mbedtls_mpi_uint*)mbedtls_calloc( i, ciL ) ) == NULL )
Manuel Pégourié-Gonnard6a8ca332015-05-28 09:33:39 +0200174 return( MBEDTLS_ERR_MPI_ALLOC_FAILED );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100175
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100176 if( X->p != NULL )
177 {
178 memcpy( p, X->p, i * ciL );
Alexey Skalozube17a8da2016-01-13 17:19:33 +0200179 mbedtls_mpi_zeroize( X->p, X->n );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200180 mbedtls_free( X->p );
Manuel Pégourié-Gonnard58681632013-11-21 10:39:37 +0100181 }
182
183 X->n = i;
184 X->p = p;
185
186 return( 0 );
187}
188
189/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000190 * Copy the contents of Y into X
191 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200192int mbedtls_mpi_copy( mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +0000193{
Gilles Peskine4e4be7c2018-03-21 16:29:03 +0100194 int ret = 0;
Paul Bakker23986e52011-04-24 08:57:21 +0000195 size_t i;
Hanno Becker73d7d792018-12-11 10:35:51 +0000196 MPI_VALIDATE_RET( X != NULL );
197 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000198
199 if( X == Y )
200 return( 0 );
201
Gilles Peskine3e9f5222020-01-20 21:12:50 +0100202 if( Y->n == 0 )
Manuel Pégourié-Gonnardf4999932013-08-12 17:02:59 +0200203 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200204 mbedtls_mpi_free( X );
Manuel Pégourié-Gonnardf4999932013-08-12 17:02:59 +0200205 return( 0 );
206 }
207
Paul Bakker5121ce52009-01-03 21:22:43 +0000208 for( i = Y->n - 1; i > 0; i-- )
209 if( Y->p[i] != 0 )
210 break;
211 i++;
212
213 X->s = Y->s;
214
Gilles Peskine4e4be7c2018-03-21 16:29:03 +0100215 if( X->n < i )
216 {
217 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i ) );
218 }
219 else
220 {
221 memset( X->p + i, 0, ( X->n - i ) * ciL );
222 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000223
Paul Bakker5121ce52009-01-03 21:22:43 +0000224 memcpy( X->p, Y->p, i * ciL );
225
226cleanup:
227
228 return( ret );
229}
230
231/*
232 * Swap the contents of X and Y
233 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200234void mbedtls_mpi_swap( mbedtls_mpi *X, mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +0000235{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200236 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000237 MPI_VALIDATE( X != NULL );
238 MPI_VALIDATE( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000239
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200240 memcpy( &T, X, sizeof( mbedtls_mpi ) );
241 memcpy( X, Y, sizeof( mbedtls_mpi ) );
242 memcpy( Y, &T, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000243}
244
245/*
Gilles Peskinec81c5882020-06-04 19:14:58 +0200246 * Conditionally assign dest = src, without leaking information
247 * about whether the assignment was made or not.
248 * dest and src must be arrays of limbs of size n.
249 * assign must be 0 or 1.
250 */
251static void mpi_safe_cond_assign( size_t n,
252 mbedtls_mpi_uint *dest,
253 const mbedtls_mpi_uint *src,
254 unsigned char assign )
255{
256 size_t i;
257 for( i = 0; i < n; i++ )
258 dest[i] = dest[i] * ( 1 - assign ) + src[i] * assign;
259}
260
261/*
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100262 * Conditionally assign X = Y, without leaking information
Manuel Pégourié-Gonnard96c7a922013-11-25 18:28:53 +0100263 * about whether the assignment was made or not.
264 * (Leaking information about the respective sizes of X and Y is ok however.)
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100265 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200266int mbedtls_mpi_safe_cond_assign( mbedtls_mpi *X, const mbedtls_mpi *Y, unsigned char assign )
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100267{
268 int ret = 0;
269 size_t i;
Hanno Becker73d7d792018-12-11 10:35:51 +0000270 MPI_VALIDATE_RET( X != NULL );
271 MPI_VALIDATE_RET( Y != NULL );
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100272
Pascal Junodb99183d2015-03-11 16:49:45 +0100273 /* make sure assign is 0 or 1 in a time-constant manner */
274 assign = (assign | (unsigned char)-assign) >> 7;
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100275
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200276 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, Y->n ) );
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100277
Paul Bakker66d5d072014-06-17 16:39:18 +0200278 X->s = X->s * ( 1 - assign ) + Y->s * assign;
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100279
Gilles Peskinec81c5882020-06-04 19:14:58 +0200280 mpi_safe_cond_assign( Y->n, X->p, Y->p, assign );
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100281
Gilles Peskinec81c5882020-06-04 19:14:58 +0200282 for( i = Y->n; i < X->n; i++ )
Paul Bakker66d5d072014-06-17 16:39:18 +0200283 X->p[i] *= ( 1 - assign );
Manuel Pégourié-Gonnard71c2c212013-11-21 16:56:39 +0100284
285cleanup:
286 return( ret );
287}
288
289/*
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100290 * Conditionally swap X and Y, without leaking information
291 * about whether the swap was made or not.
292 * Here it is not ok to simply swap the pointers, which whould lead to
293 * different memory access patterns when X and Y are used afterwards.
294 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200295int mbedtls_mpi_safe_cond_swap( mbedtls_mpi *X, mbedtls_mpi *Y, unsigned char swap )
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100296{
297 int ret, s;
298 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200299 mbedtls_mpi_uint tmp;
Hanno Becker73d7d792018-12-11 10:35:51 +0000300 MPI_VALIDATE_RET( X != NULL );
301 MPI_VALIDATE_RET( Y != NULL );
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100302
303 if( X == Y )
304 return( 0 );
305
Pascal Junodb99183d2015-03-11 16:49:45 +0100306 /* make sure swap is 0 or 1 in a time-constant manner */
307 swap = (swap | (unsigned char)-swap) >> 7;
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100308
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200309 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, Y->n ) );
310 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( Y, X->n ) );
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100311
312 s = X->s;
Paul Bakker66d5d072014-06-17 16:39:18 +0200313 X->s = X->s * ( 1 - swap ) + Y->s * swap;
314 Y->s = Y->s * ( 1 - swap ) + s * swap;
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100315
316
317 for( i = 0; i < X->n; i++ )
318 {
319 tmp = X->p[i];
Paul Bakker66d5d072014-06-17 16:39:18 +0200320 X->p[i] = X->p[i] * ( 1 - swap ) + Y->p[i] * swap;
321 Y->p[i] = Y->p[i] * ( 1 - swap ) + tmp * swap;
Manuel Pégourié-Gonnarda60fe892013-12-04 21:41:50 +0100322 }
323
324cleanup:
325 return( ret );
326}
327
328/*
Paul Bakker5121ce52009-01-03 21:22:43 +0000329 * Set value from integer
330 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200331int mbedtls_mpi_lset( mbedtls_mpi *X, mbedtls_mpi_sint z )
Paul Bakker5121ce52009-01-03 21:22:43 +0000332{
333 int ret;
Hanno Becker73d7d792018-12-11 10:35:51 +0000334 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000335
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200336 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000337 memset( X->p, 0, X->n * ciL );
338
339 X->p[0] = ( z < 0 ) ? -z : z;
340 X->s = ( z < 0 ) ? -1 : 1;
341
342cleanup:
343
344 return( ret );
345}
346
347/*
Paul Bakker2f5947e2011-05-18 15:47:11 +0000348 * Get a specific bit
349 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200350int mbedtls_mpi_get_bit( const mbedtls_mpi *X, size_t pos )
Paul Bakker2f5947e2011-05-18 15:47:11 +0000351{
Hanno Becker73d7d792018-12-11 10:35:51 +0000352 MPI_VALIDATE_RET( X != NULL );
353
Paul Bakker2f5947e2011-05-18 15:47:11 +0000354 if( X->n * biL <= pos )
355 return( 0 );
356
Paul Bakkerd8bb8262014-06-17 14:06:49 +0200357 return( ( X->p[pos / biL] >> ( pos % biL ) ) & 0x01 );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000358}
359
Gilles Peskine11cdb052018-11-20 16:47:47 +0100360/* Get a specific byte, without range checks. */
361#define GET_BYTE( X, i ) \
362 ( ( ( X )->p[( i ) / ciL] >> ( ( ( i ) % ciL ) * 8 ) ) & 0xff )
363
Paul Bakker2f5947e2011-05-18 15:47:11 +0000364/*
365 * Set a bit to a specific value of 0 or 1
366 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200367int mbedtls_mpi_set_bit( mbedtls_mpi *X, size_t pos, unsigned char val )
Paul Bakker2f5947e2011-05-18 15:47:11 +0000368{
369 int ret = 0;
370 size_t off = pos / biL;
371 size_t idx = pos % biL;
Hanno Becker73d7d792018-12-11 10:35:51 +0000372 MPI_VALIDATE_RET( X != NULL );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000373
374 if( val != 0 && val != 1 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200375 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker9af723c2014-05-01 13:03:14 +0200376
Paul Bakker2f5947e2011-05-18 15:47:11 +0000377 if( X->n * biL <= pos )
378 {
379 if( val == 0 )
Paul Bakkerd8bb8262014-06-17 14:06:49 +0200380 return( 0 );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000381
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200382 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, off + 1 ) );
Paul Bakker2f5947e2011-05-18 15:47:11 +0000383 }
384
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200385 X->p[off] &= ~( (mbedtls_mpi_uint) 0x01 << idx );
386 X->p[off] |= (mbedtls_mpi_uint) val << idx;
Paul Bakker2f5947e2011-05-18 15:47:11 +0000387
388cleanup:
Paul Bakker9af723c2014-05-01 13:03:14 +0200389
Paul Bakker2f5947e2011-05-18 15:47:11 +0000390 return( ret );
391}
392
393/*
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200394 * Return the number of less significant zero-bits
Paul Bakker5121ce52009-01-03 21:22:43 +0000395 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200396size_t mbedtls_mpi_lsb( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000397{
Paul Bakker23986e52011-04-24 08:57:21 +0000398 size_t i, j, count = 0;
Hanno Beckerf25ee7f2018-12-19 16:51:02 +0000399 MBEDTLS_INTERNAL_VALIDATE_RET( X != NULL, 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000400
401 for( i = 0; i < X->n; i++ )
Paul Bakkerf9688572011-05-05 10:00:45 +0000402 for( j = 0; j < biL; j++, count++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000403 if( ( ( X->p[i] >> j ) & 1 ) != 0 )
404 return( count );
405
406 return( 0 );
407}
408
409/*
Simon Butcher15b15d12015-11-26 19:35:03 +0000410 * Count leading zero bits in a given integer
411 */
412static size_t mbedtls_clz( const mbedtls_mpi_uint x )
413{
414 size_t j;
Manuel Pégourié-Gonnarde3e8edf2015-12-01 09:31:52 +0100415 mbedtls_mpi_uint mask = (mbedtls_mpi_uint) 1 << (biL - 1);
Simon Butcher15b15d12015-11-26 19:35:03 +0000416
417 for( j = 0; j < biL; j++ )
418 {
419 if( x & mask ) break;
420
421 mask >>= 1;
422 }
423
424 return j;
425}
426
427/*
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200428 * Return the number of bits
Paul Bakker5121ce52009-01-03 21:22:43 +0000429 */
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200430size_t mbedtls_mpi_bitlen( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000431{
Paul Bakker23986e52011-04-24 08:57:21 +0000432 size_t i, j;
Paul Bakker5121ce52009-01-03 21:22:43 +0000433
Manuel Pégourié-Gonnard770b5e12015-04-29 17:02:01 +0200434 if( X->n == 0 )
435 return( 0 );
436
Paul Bakker5121ce52009-01-03 21:22:43 +0000437 for( i = X->n - 1; i > 0; i-- )
438 if( X->p[i] != 0 )
439 break;
440
Simon Butcher15b15d12015-11-26 19:35:03 +0000441 j = biL - mbedtls_clz( X->p[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +0000442
Paul Bakker23986e52011-04-24 08:57:21 +0000443 return( ( i * biL ) + j );
Paul Bakker5121ce52009-01-03 21:22:43 +0000444}
445
446/*
447 * Return the total size in bytes
448 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200449size_t mbedtls_mpi_size( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +0000450{
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200451 return( ( mbedtls_mpi_bitlen( X ) + 7 ) >> 3 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000452}
453
454/*
455 * Convert an ASCII character to digit value
456 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200457static int mpi_get_digit( mbedtls_mpi_uint *d, int radix, char c )
Paul Bakker5121ce52009-01-03 21:22:43 +0000458{
459 *d = 255;
460
461 if( c >= 0x30 && c <= 0x39 ) *d = c - 0x30;
462 if( c >= 0x41 && c <= 0x46 ) *d = c - 0x37;
463 if( c >= 0x61 && c <= 0x66 ) *d = c - 0x57;
464
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200465 if( *d >= (mbedtls_mpi_uint) radix )
466 return( MBEDTLS_ERR_MPI_INVALID_CHARACTER );
Paul Bakker5121ce52009-01-03 21:22:43 +0000467
468 return( 0 );
469}
470
471/*
472 * Import from an ASCII string
473 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200474int mbedtls_mpi_read_string( mbedtls_mpi *X, int radix, const char *s )
Paul Bakker5121ce52009-01-03 21:22:43 +0000475{
Paul Bakker23986e52011-04-24 08:57:21 +0000476 int ret;
477 size_t i, j, slen, n;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200478 mbedtls_mpi_uint d;
479 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000480 MPI_VALIDATE_RET( X != NULL );
481 MPI_VALIDATE_RET( s != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000482
483 if( radix < 2 || radix > 16 )
Hanno Becker54c91dd2018-12-12 13:37:06 +0000484 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000485
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200486 mbedtls_mpi_init( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000487
Paul Bakkerff60ee62010-03-16 21:09:09 +0000488 slen = strlen( s );
489
Paul Bakker5121ce52009-01-03 21:22:43 +0000490 if( radix == 16 )
491 {
Manuel Pégourié-Gonnard2d708342015-10-05 15:23:11 +0100492 if( slen > MPI_SIZE_T_MAX >> 2 )
Manuel Pégourié-Gonnard58fb4952015-09-28 13:48:04 +0200493 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
494
Paul Bakkerff60ee62010-03-16 21:09:09 +0000495 n = BITS_TO_LIMBS( slen << 2 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000496
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200497 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, n ) );
498 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000499
Paul Bakker23986e52011-04-24 08:57:21 +0000500 for( i = slen, j = 0; i > 0; i--, j++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000501 {
Paul Bakker23986e52011-04-24 08:57:21 +0000502 if( i == 1 && s[i - 1] == '-' )
Paul Bakker5121ce52009-01-03 21:22:43 +0000503 {
504 X->s = -1;
505 break;
506 }
507
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200508 MBEDTLS_MPI_CHK( mpi_get_digit( &d, radix, s[i - 1] ) );
Paul Bakker66d5d072014-06-17 16:39:18 +0200509 X->p[j / ( 2 * ciL )] |= d << ( ( j % ( 2 * ciL ) ) << 2 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000510 }
511 }
512 else
513 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200514 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000515
Paul Bakkerff60ee62010-03-16 21:09:09 +0000516 for( i = 0; i < slen; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +0000517 {
518 if( i == 0 && s[i] == '-' )
519 {
520 X->s = -1;
521 continue;
522 }
523
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200524 MBEDTLS_MPI_CHK( mpi_get_digit( &d, radix, s[i] ) );
525 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T, X, radix ) );
Paul Bakker05feca62009-06-20 08:22:43 +0000526
527 if( X->s == 1 )
528 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200529 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, &T, d ) );
Paul Bakker05feca62009-06-20 08:22:43 +0000530 }
531 else
532 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200533 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( X, &T, d ) );
Paul Bakker05feca62009-06-20 08:22:43 +0000534 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000535 }
536 }
537
538cleanup:
539
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200540 mbedtls_mpi_free( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000541
542 return( ret );
543}
544
545/*
Ron Eldora16fa292018-11-20 14:07:01 +0200546 * Helper to write the digits high-order first.
Paul Bakker5121ce52009-01-03 21:22:43 +0000547 */
Ron Eldora16fa292018-11-20 14:07:01 +0200548static int mpi_write_hlp( mbedtls_mpi *X, int radix,
549 char **p, const size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000550{
551 int ret;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200552 mbedtls_mpi_uint r;
Ron Eldora16fa292018-11-20 14:07:01 +0200553 size_t length = 0;
554 char *p_end = *p + buflen;
Paul Bakker5121ce52009-01-03 21:22:43 +0000555
Ron Eldora16fa292018-11-20 14:07:01 +0200556 do
557 {
558 if( length >= buflen )
559 {
560 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
561 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000562
Ron Eldora16fa292018-11-20 14:07:01 +0200563 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, radix ) );
564 MBEDTLS_MPI_CHK( mbedtls_mpi_div_int( X, NULL, X, radix ) );
565 /*
566 * Write the residue in the current position, as an ASCII character.
567 */
568 if( r < 0xA )
569 *(--p_end) = (char)( '0' + r );
570 else
571 *(--p_end) = (char)( 'A' + ( r - 0xA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000572
Ron Eldora16fa292018-11-20 14:07:01 +0200573 length++;
574 } while( mbedtls_mpi_cmp_int( X, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +0000575
Ron Eldora16fa292018-11-20 14:07:01 +0200576 memmove( *p, p_end, length );
577 *p += length;
Paul Bakker5121ce52009-01-03 21:22:43 +0000578
579cleanup:
580
581 return( ret );
582}
583
584/*
585 * Export into an ASCII string
586 */
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100587int mbedtls_mpi_write_string( const mbedtls_mpi *X, int radix,
588 char *buf, size_t buflen, size_t *olen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000589{
Paul Bakker23986e52011-04-24 08:57:21 +0000590 int ret = 0;
591 size_t n;
Paul Bakker5121ce52009-01-03 21:22:43 +0000592 char *p;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200593 mbedtls_mpi T;
Hanno Becker73d7d792018-12-11 10:35:51 +0000594 MPI_VALIDATE_RET( X != NULL );
595 MPI_VALIDATE_RET( olen != NULL );
596 MPI_VALIDATE_RET( buflen == 0 || buf != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000597
598 if( radix < 2 || radix > 16 )
Hanno Becker54c91dd2018-12-12 13:37:06 +0000599 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000600
Hanno Beckerc1fa6cd2019-02-04 09:45:07 +0000601 n = mbedtls_mpi_bitlen( X ); /* Number of bits necessary to present `n`. */
602 if( radix >= 4 ) n >>= 1; /* Number of 4-adic digits necessary to present
603 * `n`. If radix > 4, this might be a strict
604 * overapproximation of the number of
605 * radix-adic digits needed to present `n`. */
606 if( radix >= 16 ) n >>= 1; /* Number of hexadecimal digits necessary to
607 * present `n`. */
608
Janos Follath870ed002019-03-06 13:43:02 +0000609 n += 1; /* Terminating null byte */
Hanno Beckerc1fa6cd2019-02-04 09:45:07 +0000610 n += 1; /* Compensate for the divisions above, which round down `n`
611 * in case it's not even. */
612 n += 1; /* Potential '-'-sign. */
613 n += ( n & 1 ); /* Make n even to have enough space for hexadecimal writing,
614 * which always uses an even number of hex-digits. */
Paul Bakker5121ce52009-01-03 21:22:43 +0000615
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100616 if( buflen < n )
Paul Bakker5121ce52009-01-03 21:22:43 +0000617 {
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100618 *olen = n;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200619 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000620 }
621
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100622 p = buf;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200623 mbedtls_mpi_init( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000624
625 if( X->s == -1 )
Hanno Beckeraf97cae2019-02-01 16:41:30 +0000626 {
Paul Bakker5121ce52009-01-03 21:22:43 +0000627 *p++ = '-';
Hanno Beckeraf97cae2019-02-01 16:41:30 +0000628 buflen--;
629 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000630
631 if( radix == 16 )
632 {
Paul Bakker23986e52011-04-24 08:57:21 +0000633 int c;
634 size_t i, j, k;
Paul Bakker5121ce52009-01-03 21:22:43 +0000635
Paul Bakker23986e52011-04-24 08:57:21 +0000636 for( i = X->n, k = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000637 {
Paul Bakker23986e52011-04-24 08:57:21 +0000638 for( j = ciL; j > 0; j-- )
Paul Bakker5121ce52009-01-03 21:22:43 +0000639 {
Paul Bakker23986e52011-04-24 08:57:21 +0000640 c = ( X->p[i - 1] >> ( ( j - 1 ) << 3) ) & 0xFF;
Paul Bakker5121ce52009-01-03 21:22:43 +0000641
Paul Bakker6c343d72014-07-10 14:36:19 +0200642 if( c == 0 && k == 0 && ( i + j ) != 2 )
Paul Bakker5121ce52009-01-03 21:22:43 +0000643 continue;
644
Paul Bakker98fe5ea2012-10-24 11:17:48 +0000645 *(p++) = "0123456789ABCDEF" [c / 16];
Paul Bakkerd2c167e2012-10-30 07:49:19 +0000646 *(p++) = "0123456789ABCDEF" [c % 16];
Paul Bakker5121ce52009-01-03 21:22:43 +0000647 k = 1;
648 }
649 }
650 }
651 else
652 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200653 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T, X ) );
Paul Bakkerce40a6d2009-06-23 19:46:08 +0000654
655 if( T.s == -1 )
656 T.s = 1;
657
Ron Eldora16fa292018-11-20 14:07:01 +0200658 MBEDTLS_MPI_CHK( mpi_write_hlp( &T, radix, &p, buflen ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000659 }
660
661 *p++ = '\0';
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100662 *olen = p - buf;
Paul Bakker5121ce52009-01-03 21:22:43 +0000663
664cleanup:
665
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200666 mbedtls_mpi_free( &T );
Paul Bakker5121ce52009-01-03 21:22:43 +0000667
668 return( ret );
669}
670
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200671#if defined(MBEDTLS_FS_IO)
Paul Bakker5121ce52009-01-03 21:22:43 +0000672/*
673 * Read X from an opened file
674 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200675int mbedtls_mpi_read_file( mbedtls_mpi *X, int radix, FILE *fin )
Paul Bakker5121ce52009-01-03 21:22:43 +0000676{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200677 mbedtls_mpi_uint d;
Paul Bakker23986e52011-04-24 08:57:21 +0000678 size_t slen;
Paul Bakker5121ce52009-01-03 21:22:43 +0000679 char *p;
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000680 /*
Paul Bakkercb37aa52011-11-30 16:00:20 +0000681 * Buffer should have space for (short) label and decimal formatted MPI,
682 * newline characters and '\0'
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000683 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200684 char s[ MBEDTLS_MPI_RW_BUFFER_SIZE ];
Paul Bakker5121ce52009-01-03 21:22:43 +0000685
Hanno Becker73d7d792018-12-11 10:35:51 +0000686 MPI_VALIDATE_RET( X != NULL );
687 MPI_VALIDATE_RET( fin != NULL );
688
689 if( radix < 2 || radix > 16 )
690 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
691
Paul Bakker5121ce52009-01-03 21:22:43 +0000692 memset( s, 0, sizeof( s ) );
693 if( fgets( s, sizeof( s ) - 1, fin ) == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200694 return( MBEDTLS_ERR_MPI_FILE_IO_ERROR );
Paul Bakker5121ce52009-01-03 21:22:43 +0000695
696 slen = strlen( s );
Paul Bakkercb37aa52011-11-30 16:00:20 +0000697 if( slen == sizeof( s ) - 2 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200698 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
Paul Bakkercb37aa52011-11-30 16:00:20 +0000699
Hanno Beckerb2034b72017-04-26 11:46:46 +0100700 if( slen > 0 && s[slen - 1] == '\n' ) { slen--; s[slen] = '\0'; }
701 if( slen > 0 && s[slen - 1] == '\r' ) { slen--; s[slen] = '\0'; }
Paul Bakker5121ce52009-01-03 21:22:43 +0000702
703 p = s + slen;
Hanno Beckerb2034b72017-04-26 11:46:46 +0100704 while( p-- > s )
Paul Bakker5121ce52009-01-03 21:22:43 +0000705 if( mpi_get_digit( &d, radix, *p ) != 0 )
706 break;
707
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200708 return( mbedtls_mpi_read_string( X, radix, p + 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000709}
710
711/*
712 * Write X into an opened file (or stdout if fout == NULL)
713 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200714int mbedtls_mpi_write_file( const char *p, const mbedtls_mpi *X, int radix, FILE *fout )
Paul Bakker5121ce52009-01-03 21:22:43 +0000715{
Paul Bakker23986e52011-04-24 08:57:21 +0000716 int ret;
717 size_t n, slen, plen;
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000718 /*
Paul Bakker5531c6d2012-09-26 19:20:46 +0000719 * Buffer should have space for (short) label and decimal formatted MPI,
720 * newline characters and '\0'
Paul Bakkerfe3256e2011-11-25 12:11:43 +0000721 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200722 char s[ MBEDTLS_MPI_RW_BUFFER_SIZE ];
Hanno Becker73d7d792018-12-11 10:35:51 +0000723 MPI_VALIDATE_RET( X != NULL );
724
725 if( radix < 2 || radix > 16 )
726 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +0000727
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100728 memset( s, 0, sizeof( s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000729
Manuel Pégourié-Gonnardf79b4252015-06-02 15:41:48 +0100730 MBEDTLS_MPI_CHK( mbedtls_mpi_write_string( X, radix, s, sizeof( s ) - 2, &n ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000731
732 if( p == NULL ) p = "";
733
734 plen = strlen( p );
735 slen = strlen( s );
736 s[slen++] = '\r';
737 s[slen++] = '\n';
738
739 if( fout != NULL )
740 {
741 if( fwrite( p, 1, plen, fout ) != plen ||
742 fwrite( s, 1, slen, fout ) != slen )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200743 return( MBEDTLS_ERR_MPI_FILE_IO_ERROR );
Paul Bakker5121ce52009-01-03 21:22:43 +0000744 }
745 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200746 mbedtls_printf( "%s%s", p, s );
Paul Bakker5121ce52009-01-03 21:22:43 +0000747
748cleanup:
749
750 return( ret );
751}
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200752#endif /* MBEDTLS_FS_IO */
Paul Bakker5121ce52009-01-03 21:22:43 +0000753
Hanno Beckerda1655a2017-10-18 14:21:44 +0100754
755/* Convert a big-endian byte array aligned to the size of mbedtls_mpi_uint
756 * into the storage form used by mbedtls_mpi. */
Hanno Beckerf8720072018-11-08 11:53:49 +0000757
758static mbedtls_mpi_uint mpi_uint_bigendian_to_host_c( mbedtls_mpi_uint x )
759{
760 uint8_t i;
Hanno Becker92c98932019-05-01 17:09:11 +0100761 unsigned char *x_ptr;
Hanno Beckerf8720072018-11-08 11:53:49 +0000762 mbedtls_mpi_uint tmp = 0;
Hanno Becker92c98932019-05-01 17:09:11 +0100763
764 for( i = 0, x_ptr = (unsigned char*) &x; i < ciL; i++, x_ptr++ )
765 {
766 tmp <<= CHAR_BIT;
767 tmp |= (mbedtls_mpi_uint) *x_ptr;
768 }
769
Hanno Beckerf8720072018-11-08 11:53:49 +0000770 return( tmp );
771}
772
773static mbedtls_mpi_uint mpi_uint_bigendian_to_host( mbedtls_mpi_uint x )
774{
775#if defined(__BYTE_ORDER__)
776
777/* Nothing to do on bigendian systems. */
778#if ( __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__ )
779 return( x );
780#endif /* __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__ */
781
782#if ( __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ )
783
784/* For GCC and Clang, have builtins for byte swapping. */
Hanno Becker9f6d16a2019-01-02 17:15:06 +0000785#if defined(__GNUC__) && defined(__GNUC_PREREQ)
786#if __GNUC_PREREQ(4,3)
Hanno Beckerf8720072018-11-08 11:53:49 +0000787#define have_bswap
788#endif
Hanno Becker9f6d16a2019-01-02 17:15:06 +0000789#endif
790
791#if defined(__clang__) && defined(__has_builtin)
792#if __has_builtin(__builtin_bswap32) && \
793 __has_builtin(__builtin_bswap64)
794#define have_bswap
795#endif
796#endif
797
Hanno Beckerf8720072018-11-08 11:53:49 +0000798#if defined(have_bswap)
799 /* The compiler is hopefully able to statically evaluate this! */
800 switch( sizeof(mbedtls_mpi_uint) )
801 {
802 case 4:
803 return( __builtin_bswap32(x) );
804 case 8:
805 return( __builtin_bswap64(x) );
806 }
807#endif
808#endif /* __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ */
809#endif /* __BYTE_ORDER__ */
810
811 /* Fall back to C-based reordering if we don't know the byte order
812 * or we couldn't use a compiler-specific builtin. */
813 return( mpi_uint_bigendian_to_host_c( x ) );
814}
815
Hanno Becker2be8a552018-10-25 12:40:09 +0100816static void mpi_bigendian_to_host( mbedtls_mpi_uint * const p, size_t limbs )
Hanno Beckerda1655a2017-10-18 14:21:44 +0100817{
Hanno Beckerda1655a2017-10-18 14:21:44 +0100818 mbedtls_mpi_uint *cur_limb_left;
819 mbedtls_mpi_uint *cur_limb_right;
Hanno Becker2be8a552018-10-25 12:40:09 +0100820 if( limbs == 0 )
821 return;
Hanno Beckerda1655a2017-10-18 14:21:44 +0100822
823 /*
824 * Traverse limbs and
825 * - adapt byte-order in each limb
826 * - swap the limbs themselves.
827 * For that, simultaneously traverse the limbs from left to right
828 * and from right to left, as long as the left index is not bigger
829 * than the right index (it's not a problem if limbs is odd and the
830 * indices coincide in the last iteration).
831 */
Hanno Beckerda1655a2017-10-18 14:21:44 +0100832 for( cur_limb_left = p, cur_limb_right = p + ( limbs - 1 );
833 cur_limb_left <= cur_limb_right;
834 cur_limb_left++, cur_limb_right-- )
835 {
Hanno Beckerf8720072018-11-08 11:53:49 +0000836 mbedtls_mpi_uint tmp;
837 /* Note that if cur_limb_left == cur_limb_right,
838 * this code effectively swaps the bytes only once. */
839 tmp = mpi_uint_bigendian_to_host( *cur_limb_left );
840 *cur_limb_left = mpi_uint_bigendian_to_host( *cur_limb_right );
841 *cur_limb_right = tmp;
Hanno Beckerda1655a2017-10-18 14:21:44 +0100842 }
Hanno Beckerda1655a2017-10-18 14:21:44 +0100843}
844
Paul Bakker5121ce52009-01-03 21:22:43 +0000845/*
846 * Import X from unsigned binary data, big endian
847 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200848int mbedtls_mpi_read_binary( mbedtls_mpi *X, const unsigned char *buf, size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000849{
Paul Bakker23986e52011-04-24 08:57:21 +0000850 int ret;
Hanno Beckerda1655a2017-10-18 14:21:44 +0100851 size_t const limbs = CHARS_TO_LIMBS( buflen );
852 size_t const overhead = ( limbs * ciL ) - buflen;
853 unsigned char *Xp;
Paul Bakker5121ce52009-01-03 21:22:43 +0000854
Hanno Becker8ce11a32018-12-19 16:18:52 +0000855 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +0000856 MPI_VALIDATE_RET( buflen == 0 || buf != NULL );
857
Hanno Becker073c1992017-10-17 15:17:27 +0100858 /* Ensure that target MPI has exactly the necessary number of limbs */
859 if( X->n != limbs )
860 {
861 mbedtls_mpi_free( X );
862 mbedtls_mpi_init( X );
863 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
864 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200865 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000866
Hanno Becker0e810b92019-01-03 17:13:11 +0000867 /* Avoid calling `memcpy` with NULL source argument,
868 * even if buflen is 0. */
869 if( buf != NULL )
870 {
871 Xp = (unsigned char*) X->p;
872 memcpy( Xp + overhead, buf, buflen );
Hanno Beckerda1655a2017-10-18 14:21:44 +0100873
Hanno Becker0e810b92019-01-03 17:13:11 +0000874 mpi_bigendian_to_host( X->p, limbs );
875 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000876
877cleanup:
878
879 return( ret );
880}
881
882/*
883 * Export X into unsigned binary data, big endian
884 */
Gilles Peskine11cdb052018-11-20 16:47:47 +0100885int mbedtls_mpi_write_binary( const mbedtls_mpi *X,
886 unsigned char *buf, size_t buflen )
Paul Bakker5121ce52009-01-03 21:22:43 +0000887{
Hanno Becker73d7d792018-12-11 10:35:51 +0000888 size_t stored_bytes;
Gilles Peskine11cdb052018-11-20 16:47:47 +0100889 size_t bytes_to_copy;
890 unsigned char *p;
891 size_t i;
Paul Bakker5121ce52009-01-03 21:22:43 +0000892
Hanno Becker73d7d792018-12-11 10:35:51 +0000893 MPI_VALIDATE_RET( X != NULL );
894 MPI_VALIDATE_RET( buflen == 0 || buf != NULL );
895
896 stored_bytes = X->n * ciL;
897
Gilles Peskine11cdb052018-11-20 16:47:47 +0100898 if( stored_bytes < buflen )
899 {
900 /* There is enough space in the output buffer. Write initial
901 * null bytes and record the position at which to start
902 * writing the significant bytes. In this case, the execution
903 * trace of this function does not depend on the value of the
904 * number. */
905 bytes_to_copy = stored_bytes;
906 p = buf + buflen - stored_bytes;
907 memset( buf, 0, buflen - stored_bytes );
908 }
909 else
910 {
911 /* The output buffer is smaller than the allocated size of X.
912 * However X may fit if its leading bytes are zero. */
913 bytes_to_copy = buflen;
914 p = buf;
915 for( i = bytes_to_copy; i < stored_bytes; i++ )
916 {
917 if( GET_BYTE( X, i ) != 0 )
918 return( MBEDTLS_ERR_MPI_BUFFER_TOO_SMALL );
919 }
920 }
Paul Bakker5121ce52009-01-03 21:22:43 +0000921
Gilles Peskine11cdb052018-11-20 16:47:47 +0100922 for( i = 0; i < bytes_to_copy; i++ )
923 p[bytes_to_copy - i - 1] = GET_BYTE( X, i );
Paul Bakker5121ce52009-01-03 21:22:43 +0000924
925 return( 0 );
926}
927
928/*
929 * Left-shift: X <<= count
930 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200931int mbedtls_mpi_shift_l( mbedtls_mpi *X, size_t count )
Paul Bakker5121ce52009-01-03 21:22:43 +0000932{
Paul Bakker23986e52011-04-24 08:57:21 +0000933 int ret;
934 size_t i, v0, t1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200935 mbedtls_mpi_uint r0 = 0, r1;
Hanno Becker73d7d792018-12-11 10:35:51 +0000936 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000937
938 v0 = count / (biL );
939 t1 = count & (biL - 1);
940
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +0200941 i = mbedtls_mpi_bitlen( X ) + count;
Paul Bakker5121ce52009-01-03 21:22:43 +0000942
Paul Bakkerf9688572011-05-05 10:00:45 +0000943 if( X->n * biL < i )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200944 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, BITS_TO_LIMBS( i ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +0000945
946 ret = 0;
947
948 /*
949 * shift by count / limb_size
950 */
951 if( v0 > 0 )
952 {
Paul Bakker23986e52011-04-24 08:57:21 +0000953 for( i = X->n; i > v0; i-- )
954 X->p[i - 1] = X->p[i - v0 - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +0000955
Paul Bakker23986e52011-04-24 08:57:21 +0000956 for( ; i > 0; i-- )
957 X->p[i - 1] = 0;
Paul Bakker5121ce52009-01-03 21:22:43 +0000958 }
959
960 /*
961 * shift by count % limb_size
962 */
963 if( t1 > 0 )
964 {
965 for( i = v0; i < X->n; i++ )
966 {
967 r1 = X->p[i] >> (biL - t1);
968 X->p[i] <<= t1;
969 X->p[i] |= r0;
970 r0 = r1;
971 }
972 }
973
974cleanup:
975
976 return( ret );
977}
978
979/*
980 * Right-shift: X >>= count
981 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200982int mbedtls_mpi_shift_r( mbedtls_mpi *X, size_t count )
Paul Bakker5121ce52009-01-03 21:22:43 +0000983{
Paul Bakker23986e52011-04-24 08:57:21 +0000984 size_t i, v0, v1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200985 mbedtls_mpi_uint r0 = 0, r1;
Hanno Becker73d7d792018-12-11 10:35:51 +0000986 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +0000987
988 v0 = count / biL;
989 v1 = count & (biL - 1);
990
Manuel Pégourié-Gonnarde44ec102012-11-17 12:42:51 +0100991 if( v0 > X->n || ( v0 == X->n && v1 > 0 ) )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200992 return mbedtls_mpi_lset( X, 0 );
Manuel Pégourié-Gonnarde44ec102012-11-17 12:42:51 +0100993
Paul Bakker5121ce52009-01-03 21:22:43 +0000994 /*
995 * shift by count / limb_size
996 */
997 if( v0 > 0 )
998 {
999 for( i = 0; i < X->n - v0; i++ )
1000 X->p[i] = X->p[i + v0];
1001
1002 for( ; i < X->n; i++ )
1003 X->p[i] = 0;
1004 }
1005
1006 /*
1007 * shift by count % limb_size
1008 */
1009 if( v1 > 0 )
1010 {
Paul Bakker23986e52011-04-24 08:57:21 +00001011 for( i = X->n; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001012 {
Paul Bakker23986e52011-04-24 08:57:21 +00001013 r1 = X->p[i - 1] << (biL - v1);
1014 X->p[i - 1] >>= v1;
1015 X->p[i - 1] |= r0;
Paul Bakker5121ce52009-01-03 21:22:43 +00001016 r0 = r1;
1017 }
1018 }
1019
1020 return( 0 );
1021}
1022
1023/*
1024 * Compare unsigned values
1025 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001026int mbedtls_mpi_cmp_abs( const mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +00001027{
Paul Bakker23986e52011-04-24 08:57:21 +00001028 size_t i, j;
Hanno Becker73d7d792018-12-11 10:35:51 +00001029 MPI_VALIDATE_RET( X != NULL );
1030 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001031
Paul Bakker23986e52011-04-24 08:57:21 +00001032 for( i = X->n; i > 0; i-- )
1033 if( X->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001034 break;
1035
Paul Bakker23986e52011-04-24 08:57:21 +00001036 for( j = Y->n; j > 0; j-- )
1037 if( Y->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001038 break;
1039
Paul Bakker23986e52011-04-24 08:57:21 +00001040 if( i == 0 && j == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001041 return( 0 );
1042
1043 if( i > j ) return( 1 );
1044 if( j > i ) return( -1 );
1045
Paul Bakker23986e52011-04-24 08:57:21 +00001046 for( ; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001047 {
Paul Bakker23986e52011-04-24 08:57:21 +00001048 if( X->p[i - 1] > Y->p[i - 1] ) return( 1 );
1049 if( X->p[i - 1] < Y->p[i - 1] ) return( -1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001050 }
1051
1052 return( 0 );
1053}
1054
1055/*
1056 * Compare signed values
1057 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001058int mbedtls_mpi_cmp_mpi( const mbedtls_mpi *X, const mbedtls_mpi *Y )
Paul Bakker5121ce52009-01-03 21:22:43 +00001059{
Paul Bakker23986e52011-04-24 08:57:21 +00001060 size_t i, j;
Hanno Becker73d7d792018-12-11 10:35:51 +00001061 MPI_VALIDATE_RET( X != NULL );
1062 MPI_VALIDATE_RET( Y != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001063
Paul Bakker23986e52011-04-24 08:57:21 +00001064 for( i = X->n; i > 0; i-- )
1065 if( X->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001066 break;
1067
Paul Bakker23986e52011-04-24 08:57:21 +00001068 for( j = Y->n; j > 0; j-- )
1069 if( Y->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001070 break;
1071
Paul Bakker23986e52011-04-24 08:57:21 +00001072 if( i == 0 && j == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001073 return( 0 );
1074
1075 if( i > j ) return( X->s );
Paul Bakker0c8f73b2012-03-22 14:08:57 +00001076 if( j > i ) return( -Y->s );
Paul Bakker5121ce52009-01-03 21:22:43 +00001077
1078 if( X->s > 0 && Y->s < 0 ) return( 1 );
1079 if( Y->s > 0 && X->s < 0 ) return( -1 );
1080
Paul Bakker23986e52011-04-24 08:57:21 +00001081 for( ; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001082 {
Paul Bakker23986e52011-04-24 08:57:21 +00001083 if( X->p[i - 1] > Y->p[i - 1] ) return( X->s );
1084 if( X->p[i - 1] < Y->p[i - 1] ) return( -X->s );
Paul Bakker5121ce52009-01-03 21:22:43 +00001085 }
1086
1087 return( 0 );
1088}
1089
Janos Follath45ec9902019-10-14 09:09:32 +01001090/** Decide if an integer is less than the other, without branches.
1091 *
1092 * \param x First integer.
1093 * \param y Second integer.
1094 *
1095 * \return 1 if \p x is less than \p y, 0 otherwise
1096 */
Janos Follath867a3ab2019-10-11 14:21:53 +01001097static unsigned ct_lt_mpi_uint( const mbedtls_mpi_uint x,
1098 const mbedtls_mpi_uint y )
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001099{
1100 mbedtls_mpi_uint ret;
1101 mbedtls_mpi_uint cond;
1102
1103 /*
1104 * Check if the most significant bits (MSB) of the operands are different.
1105 */
1106 cond = ( x ^ y );
1107 /*
1108 * If the MSB are the same then the difference x-y will be negative (and
1109 * have its MSB set to 1 during conversion to unsigned) if and only if x<y.
1110 */
1111 ret = ( x - y ) & ~cond;
1112 /*
1113 * If the MSB are different, then the operand with the MSB of 1 is the
1114 * bigger. (That is if y has MSB of 1, then x<y is true and it is false if
1115 * the MSB of y is 0.)
1116 */
1117 ret |= y & cond;
1118
1119
Janos Follath7a34bcf2019-10-14 08:59:14 +01001120 ret = ret >> ( biL - 1 );
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001121
Janos Follath359a01e2019-10-29 15:08:46 +00001122 return (unsigned) ret;
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001123}
1124
1125/*
1126 * Compare signed values in constant time
1127 */
Janos Follath867a3ab2019-10-11 14:21:53 +01001128int mbedtls_mpi_lt_mpi_ct( const mbedtls_mpi *X, const mbedtls_mpi *Y,
1129 unsigned *ret )
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001130{
1131 size_t i;
Janos Follathbd87a592019-10-28 12:23:18 +00001132 /* The value of any of these variables is either 0 or 1 at all times. */
Janos Follath1f21c1d2019-10-28 12:31:34 +00001133 unsigned cond, done, X_is_negative, Y_is_negative;
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001134
1135 MPI_VALIDATE_RET( X != NULL );
1136 MPI_VALIDATE_RET( Y != NULL );
1137 MPI_VALIDATE_RET( ret != NULL );
1138
1139 if( X->n != Y->n )
1140 return MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
1141
1142 /*
Janos Follath58525182019-10-28 12:12:15 +00001143 * Set sign_N to 1 if N >= 0, 0 if N < 0.
1144 * We know that N->s == 1 if N >= 0 and N->s == -1 if N < 0.
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001145 */
Janos Follath1f21c1d2019-10-28 12:31:34 +00001146 X_is_negative = ( X->s & 2 ) >> 1;
1147 Y_is_negative = ( Y->s & 2 ) >> 1;
Janos Follath867a3ab2019-10-11 14:21:53 +01001148
1149 /*
1150 * If the signs are different, then the positive operand is the bigger.
Janos Follath1f21c1d2019-10-28 12:31:34 +00001151 * That is if X is negative (X_is_negative == 1), then X < Y is true and it
1152 * is false if X is positive (X_is_negative == 0).
Janos Follath867a3ab2019-10-11 14:21:53 +01001153 */
Janos Follath1f21c1d2019-10-28 12:31:34 +00001154 cond = ( X_is_negative ^ Y_is_negative );
1155 *ret = cond & X_is_negative;
Janos Follath867a3ab2019-10-11 14:21:53 +01001156
1157 /*
Janos Follathbd87a592019-10-28 12:23:18 +00001158 * This is a constant-time function. We might have the result, but we still
Janos Follath867a3ab2019-10-11 14:21:53 +01001159 * need to go through the loop. Record if we have the result already.
1160 */
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001161 done = cond;
1162
1163 for( i = X->n; i > 0; i-- )
1164 {
1165 /*
Janos Follathe25f1ee2019-11-05 12:24:52 +00001166 * If Y->p[i - 1] < X->p[i - 1] then X < Y is true if and only if both
1167 * X and Y are negative.
Janos Follath867a3ab2019-10-11 14:21:53 +01001168 *
1169 * Again even if we can make a decision, we just mark the result and
1170 * the fact that we are done and continue looping.
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001171 */
Janos Follathe25f1ee2019-11-05 12:24:52 +00001172 cond = ct_lt_mpi_uint( Y->p[i - 1], X->p[i - 1] );
1173 *ret |= cond & ( 1 - done ) & X_is_negative;
Janos Follathfbe4c942019-10-28 12:37:21 +00001174 done |= cond;
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001175
1176 /*
Janos Follathe25f1ee2019-11-05 12:24:52 +00001177 * If X->p[i - 1] < Y->p[i - 1] then X < Y is true if and only if both
1178 * X and Y are positive.
Janos Follath867a3ab2019-10-11 14:21:53 +01001179 *
1180 * Again even if we can make a decision, we just mark the result and
1181 * the fact that we are done and continue looping.
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001182 */
Janos Follathe25f1ee2019-11-05 12:24:52 +00001183 cond = ct_lt_mpi_uint( X->p[i - 1], Y->p[i - 1] );
1184 *ret |= cond & ( 1 - done ) & ( 1 - X_is_negative );
Janos Follathfbe4c942019-10-28 12:37:21 +00001185 done |= cond;
Janos Follathb9f6f9b2019-09-05 14:47:19 +01001186 }
1187
1188 return( 0 );
1189}
1190
Paul Bakker5121ce52009-01-03 21:22:43 +00001191/*
1192 * Compare signed values
1193 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001194int mbedtls_mpi_cmp_int( const mbedtls_mpi *X, mbedtls_mpi_sint z )
Paul Bakker5121ce52009-01-03 21:22:43 +00001195{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001196 mbedtls_mpi Y;
1197 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001198 MPI_VALIDATE_RET( X != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001199
1200 *p = ( z < 0 ) ? -z : z;
1201 Y.s = ( z < 0 ) ? -1 : 1;
1202 Y.n = 1;
1203 Y.p = p;
1204
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001205 return( mbedtls_mpi_cmp_mpi( X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001206}
1207
1208/*
1209 * Unsigned addition: X = |A| + |B| (HAC 14.7)
1210 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001211int mbedtls_mpi_add_abs( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001212{
Paul Bakker23986e52011-04-24 08:57:21 +00001213 int ret;
1214 size_t i, j;
Janos Follath6c922682015-10-30 17:43:11 +01001215 mbedtls_mpi_uint *o, *p, c, tmp;
Hanno Becker73d7d792018-12-11 10:35:51 +00001216 MPI_VALIDATE_RET( X != NULL );
1217 MPI_VALIDATE_RET( A != NULL );
1218 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001219
1220 if( X == B )
1221 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001222 const mbedtls_mpi *T = A; A = X; B = T;
Paul Bakker5121ce52009-01-03 21:22:43 +00001223 }
1224
1225 if( X != A )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001226 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, A ) );
Paul Bakker9af723c2014-05-01 13:03:14 +02001227
Paul Bakkerf7ca7b92009-06-20 10:31:06 +00001228 /*
1229 * X should always be positive as a result of unsigned additions.
1230 */
1231 X->s = 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00001232
Paul Bakker23986e52011-04-24 08:57:21 +00001233 for( j = B->n; j > 0; j-- )
1234 if( B->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001235 break;
1236
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001237 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001238
1239 o = B->p; p = X->p; c = 0;
1240
Janos Follath6c922682015-10-30 17:43:11 +01001241 /*
1242 * tmp is used because it might happen that p == o
1243 */
Paul Bakker23986e52011-04-24 08:57:21 +00001244 for( i = 0; i < j; i++, o++, p++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00001245 {
Janos Follath6c922682015-10-30 17:43:11 +01001246 tmp= *o;
Paul Bakker5121ce52009-01-03 21:22:43 +00001247 *p += c; c = ( *p < c );
Janos Follath6c922682015-10-30 17:43:11 +01001248 *p += tmp; c += ( *p < tmp );
Paul Bakker5121ce52009-01-03 21:22:43 +00001249 }
1250
1251 while( c != 0 )
1252 {
1253 if( i >= X->n )
1254 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001255 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001256 p = X->p + i;
1257 }
1258
Paul Bakker2d319fd2012-09-16 21:34:26 +00001259 *p += c; c = ( *p < c ); i++; p++;
Paul Bakker5121ce52009-01-03 21:22:43 +00001260 }
1261
1262cleanup:
1263
1264 return( ret );
1265}
1266
Gilles Peskinede719d52020-06-09 10:39:38 +02001267/**
Gilles Peskine36acd542020-06-08 21:58:22 +02001268 * Helper for mbedtls_mpi subtraction.
1269 *
1270 * Calculate d - s where d and s have the same size.
1271 * This function operates modulo (2^ciL)^n and returns the carry
1272 * (1 if there was a wraparound, i.e. if `d < s`, and 0 otherwise).
1273 *
1274 * \param n Number of limbs of \p d and \p s.
1275 * \param[in,out] d On input, the left operand.
1276 * On output, the result of the subtraction:
Gilles Peskinede719d52020-06-09 10:39:38 +02001277 * \param[in] s The right operand.
Gilles Peskine36acd542020-06-08 21:58:22 +02001278 *
1279 * \return 1 if `d < s`.
1280 * 0 if `d >= s`.
Paul Bakker5121ce52009-01-03 21:22:43 +00001281 */
Gilles Peskine36acd542020-06-08 21:58:22 +02001282static mbedtls_mpi_uint mpi_sub_hlp( size_t n,
1283 mbedtls_mpi_uint *d,
1284 const mbedtls_mpi_uint *s )
Paul Bakker5121ce52009-01-03 21:22:43 +00001285{
Paul Bakker23986e52011-04-24 08:57:21 +00001286 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001287 mbedtls_mpi_uint c, z;
Paul Bakker5121ce52009-01-03 21:22:43 +00001288
1289 for( i = c = 0; i < n; i++, s++, d++ )
1290 {
1291 z = ( *d < c ); *d -= c;
1292 c = ( *d < *s ) + z; *d -= *s;
1293 }
1294
Gilles Peskine36acd542020-06-08 21:58:22 +02001295 return( c );
Paul Bakker5121ce52009-01-03 21:22:43 +00001296}
1297
1298/*
Gilles Peskine08fd43c2020-06-08 22:50:35 +02001299 * Unsigned subtraction: X = |A| - |B| (HAC 14.9, 14.10)
Paul Bakker5121ce52009-01-03 21:22:43 +00001300 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001301int mbedtls_mpi_sub_abs( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001302{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001303 mbedtls_mpi TB;
Paul Bakker23986e52011-04-24 08:57:21 +00001304 int ret;
1305 size_t n;
Gilles Peskine08fd43c2020-06-08 22:50:35 +02001306 mbedtls_mpi_uint carry;
Hanno Becker73d7d792018-12-11 10:35:51 +00001307 MPI_VALIDATE_RET( X != NULL );
1308 MPI_VALIDATE_RET( A != NULL );
1309 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001310
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001311 mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001312
1313 if( X == B )
1314 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001315 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001316 B = &TB;
1317 }
1318
1319 if( X != A )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001320 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001321
Paul Bakker1ef7a532009-06-20 10:50:55 +00001322 /*
Paul Bakker60b1d102013-10-29 10:02:51 +01001323 * X should always be positive as a result of unsigned subtractions.
Paul Bakker1ef7a532009-06-20 10:50:55 +00001324 */
1325 X->s = 1;
1326
Paul Bakker5121ce52009-01-03 21:22:43 +00001327 ret = 0;
1328
Paul Bakker23986e52011-04-24 08:57:21 +00001329 for( n = B->n; n > 0; n-- )
1330 if( B->p[n - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001331 break;
1332
Gilles Peskine08fd43c2020-06-08 22:50:35 +02001333 carry = mpi_sub_hlp( n, X->p, B->p );
1334 if( carry != 0 )
Gilles Peskine36acd542020-06-08 21:58:22 +02001335 {
Gilles Peskine08fd43c2020-06-08 22:50:35 +02001336 /* Propagate the carry to the first nonzero limb of X. */
1337 for( ; n < X->n && X->p[n] == 0; n++ )
1338 --X->p[n];
1339 /* If we ran out of space for the carry, it means that the result
1340 * is negative. */
1341 if( n == X->n )
1342 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
1343 --X->p[n];
Gilles Peskine36acd542020-06-08 21:58:22 +02001344 }
Paul Bakker5121ce52009-01-03 21:22:43 +00001345
1346cleanup:
1347
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001348 mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001349
1350 return( ret );
1351}
1352
1353/*
1354 * Signed addition: X = A + B
1355 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001356int mbedtls_mpi_add_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001357{
Hanno Becker73d7d792018-12-11 10:35:51 +00001358 int ret, s;
1359 MPI_VALIDATE_RET( X != NULL );
1360 MPI_VALIDATE_RET( A != NULL );
1361 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001362
Hanno Becker73d7d792018-12-11 10:35:51 +00001363 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001364 if( A->s * B->s < 0 )
1365 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001366 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001367 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001368 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001369 X->s = s;
1370 }
1371 else
1372 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001373 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001374 X->s = -s;
1375 }
1376 }
1377 else
1378 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001379 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001380 X->s = s;
1381 }
1382
1383cleanup:
1384
1385 return( ret );
1386}
1387
1388/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001389 * Signed subtraction: X = A - B
Paul Bakker5121ce52009-01-03 21:22:43 +00001390 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001391int mbedtls_mpi_sub_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001392{
Hanno Becker73d7d792018-12-11 10:35:51 +00001393 int ret, s;
1394 MPI_VALIDATE_RET( X != NULL );
1395 MPI_VALIDATE_RET( A != NULL );
1396 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001397
Hanno Becker73d7d792018-12-11 10:35:51 +00001398 s = A->s;
Paul Bakker5121ce52009-01-03 21:22:43 +00001399 if( A->s * B->s > 0 )
1400 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001401 if( mbedtls_mpi_cmp_abs( A, B ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001402 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001403 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001404 X->s = s;
1405 }
1406 else
1407 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001408 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( X, B, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001409 X->s = -s;
1410 }
1411 }
1412 else
1413 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001414 MBEDTLS_MPI_CHK( mbedtls_mpi_add_abs( X, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001415 X->s = s;
1416 }
1417
1418cleanup:
1419
1420 return( ret );
1421}
1422
1423/*
1424 * Signed addition: X = A + b
1425 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001426int mbedtls_mpi_add_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001427{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001428 mbedtls_mpi _B;
1429 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001430 MPI_VALIDATE_RET( X != NULL );
1431 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001432
1433 p[0] = ( b < 0 ) ? -b : b;
1434 _B.s = ( b < 0 ) ? -1 : 1;
1435 _B.n = 1;
1436 _B.p = p;
1437
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001438 return( mbedtls_mpi_add_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001439}
1440
1441/*
Paul Bakker60b1d102013-10-29 10:02:51 +01001442 * Signed subtraction: X = A - b
Paul Bakker5121ce52009-01-03 21:22:43 +00001443 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001444int mbedtls_mpi_sub_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001445{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001446 mbedtls_mpi _B;
1447 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001448 MPI_VALIDATE_RET( X != NULL );
1449 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001450
1451 p[0] = ( b < 0 ) ? -b : b;
1452 _B.s = ( b < 0 ) ? -1 : 1;
1453 _B.n = 1;
1454 _B.p = p;
1455
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001456 return( mbedtls_mpi_sub_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001457}
1458
1459/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001460 * Helper for mbedtls_mpi multiplication
Paul Bakkerfc4f46f2013-06-24 19:23:56 +02001461 */
1462static
1463#if defined(__APPLE__) && defined(__arm__)
1464/*
1465 * Apple LLVM version 4.2 (clang-425.0.24) (based on LLVM 3.2svn)
1466 * appears to need this to prevent bad ARM code generation at -O3.
1467 */
1468__attribute__ ((noinline))
1469#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001470void mpi_mul_hlp( size_t i, mbedtls_mpi_uint *s, mbedtls_mpi_uint *d, mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001471{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001472 mbedtls_mpi_uint c = 0, t = 0;
Paul Bakker5121ce52009-01-03 21:22:43 +00001473
1474#if defined(MULADDC_HUIT)
1475 for( ; i >= 8; i -= 8 )
1476 {
1477 MULADDC_INIT
1478 MULADDC_HUIT
1479 MULADDC_STOP
1480 }
1481
1482 for( ; i > 0; i-- )
1483 {
1484 MULADDC_INIT
1485 MULADDC_CORE
1486 MULADDC_STOP
1487 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001488#else /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001489 for( ; i >= 16; i -= 16 )
1490 {
1491 MULADDC_INIT
1492 MULADDC_CORE MULADDC_CORE
1493 MULADDC_CORE MULADDC_CORE
1494 MULADDC_CORE MULADDC_CORE
1495 MULADDC_CORE MULADDC_CORE
1496
1497 MULADDC_CORE MULADDC_CORE
1498 MULADDC_CORE MULADDC_CORE
1499 MULADDC_CORE MULADDC_CORE
1500 MULADDC_CORE MULADDC_CORE
1501 MULADDC_STOP
1502 }
1503
1504 for( ; i >= 8; i -= 8 )
1505 {
1506 MULADDC_INIT
1507 MULADDC_CORE MULADDC_CORE
1508 MULADDC_CORE MULADDC_CORE
1509
1510 MULADDC_CORE MULADDC_CORE
1511 MULADDC_CORE MULADDC_CORE
1512 MULADDC_STOP
1513 }
1514
1515 for( ; i > 0; i-- )
1516 {
1517 MULADDC_INIT
1518 MULADDC_CORE
1519 MULADDC_STOP
1520 }
Paul Bakker9af723c2014-05-01 13:03:14 +02001521#endif /* MULADDC_HUIT */
Paul Bakker5121ce52009-01-03 21:22:43 +00001522
1523 t++;
1524
1525 do {
1526 *d += c; c = ( *d < c ); d++;
1527 }
1528 while( c != 0 );
1529}
1530
1531/*
1532 * Baseline multiplication: X = A * B (HAC 14.12)
1533 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001534int mbedtls_mpi_mul_mpi( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001535{
Paul Bakker23986e52011-04-24 08:57:21 +00001536 int ret;
1537 size_t i, j;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001538 mbedtls_mpi TA, TB;
Hanno Becker73d7d792018-12-11 10:35:51 +00001539 MPI_VALIDATE_RET( X != NULL );
1540 MPI_VALIDATE_RET( A != NULL );
1541 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001542
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001543 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00001544
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001545 if( X == A ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) ); A = &TA; }
1546 if( X == B ) { MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) ); B = &TB; }
Paul Bakker5121ce52009-01-03 21:22:43 +00001547
Paul Bakker23986e52011-04-24 08:57:21 +00001548 for( i = A->n; i > 0; i-- )
1549 if( A->p[i - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001550 break;
1551
Paul Bakker23986e52011-04-24 08:57:21 +00001552 for( j = B->n; j > 0; j-- )
1553 if( B->p[j - 1] != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001554 break;
1555
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001556 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, i + j ) );
1557 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001558
Alexey Skalozub8e75e682016-01-13 21:59:27 +02001559 for( ; j > 0; j-- )
1560 mpi_mul_hlp( i, A->p, X->p + j - 1, B->p[j - 1] );
Paul Bakker5121ce52009-01-03 21:22:43 +00001561
1562 X->s = A->s * B->s;
1563
1564cleanup:
1565
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001566 mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TA );
Paul Bakker5121ce52009-01-03 21:22:43 +00001567
1568 return( ret );
1569}
1570
1571/*
1572 * Baseline multiplication: X = A * b
1573 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001574int mbedtls_mpi_mul_int( mbedtls_mpi *X, const mbedtls_mpi *A, mbedtls_mpi_uint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001575{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001576 mbedtls_mpi _B;
1577 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001578 MPI_VALIDATE_RET( X != NULL );
1579 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001580
1581 _B.s = 1;
1582 _B.n = 1;
1583 _B.p = p;
1584 p[0] = b;
1585
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001586 return( mbedtls_mpi_mul_mpi( X, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001587}
1588
1589/*
Simon Butcherf5ba0452015-12-27 23:01:55 +00001590 * Unsigned integer divide - double mbedtls_mpi_uint dividend, u1/u0, and
1591 * mbedtls_mpi_uint divisor, d
Simon Butcher15b15d12015-11-26 19:35:03 +00001592 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001593static mbedtls_mpi_uint mbedtls_int_div_int( mbedtls_mpi_uint u1,
1594 mbedtls_mpi_uint u0, mbedtls_mpi_uint d, mbedtls_mpi_uint *r )
Simon Butcher15b15d12015-11-26 19:35:03 +00001595{
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001596#if defined(MBEDTLS_HAVE_UDBL)
1597 mbedtls_t_udbl dividend, quotient;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001598#else
Simon Butcher9803d072016-01-03 00:24:34 +00001599 const mbedtls_mpi_uint radix = (mbedtls_mpi_uint) 1 << biH;
1600 const mbedtls_mpi_uint uint_halfword_mask = ( (mbedtls_mpi_uint) 1 << biH ) - 1;
Simon Butcherf5ba0452015-12-27 23:01:55 +00001601 mbedtls_mpi_uint d0, d1, q0, q1, rAX, r0, quotient;
1602 mbedtls_mpi_uint u0_msw, u0_lsw;
Simon Butcher9803d072016-01-03 00:24:34 +00001603 size_t s;
Manuel Pégourié-Gonnard16308882015-12-01 10:27:00 +01001604#endif
1605
Simon Butcher15b15d12015-11-26 19:35:03 +00001606 /*
1607 * Check for overflow
1608 */
Simon Butcherf5ba0452015-12-27 23:01:55 +00001609 if( 0 == d || u1 >= d )
Simon Butcher15b15d12015-11-26 19:35:03 +00001610 {
Simon Butcherf5ba0452015-12-27 23:01:55 +00001611 if (r != NULL) *r = ~0;
Simon Butcher15b15d12015-11-26 19:35:03 +00001612
Simon Butcherf5ba0452015-12-27 23:01:55 +00001613 return ( ~0 );
Simon Butcher15b15d12015-11-26 19:35:03 +00001614 }
1615
1616#if defined(MBEDTLS_HAVE_UDBL)
Simon Butcher15b15d12015-11-26 19:35:03 +00001617 dividend = (mbedtls_t_udbl) u1 << biL;
1618 dividend |= (mbedtls_t_udbl) u0;
1619 quotient = dividend / d;
1620 if( quotient > ( (mbedtls_t_udbl) 1 << biL ) - 1 )
1621 quotient = ( (mbedtls_t_udbl) 1 << biL ) - 1;
1622
1623 if( r != NULL )
Simon Butcher9803d072016-01-03 00:24:34 +00001624 *r = (mbedtls_mpi_uint)( dividend - (quotient * d ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001625
1626 return (mbedtls_mpi_uint) quotient;
1627#else
Simon Butcher15b15d12015-11-26 19:35:03 +00001628
1629 /*
1630 * Algorithm D, Section 4.3.1 - The Art of Computer Programming
1631 * Vol. 2 - Seminumerical Algorithms, Knuth
1632 */
1633
1634 /*
1635 * Normalize the divisor, d, and dividend, u0, u1
1636 */
1637 s = mbedtls_clz( d );
1638 d = d << s;
1639
1640 u1 = u1 << s;
Simon Butcher9803d072016-01-03 00:24:34 +00001641 u1 |= ( u0 >> ( biL - s ) ) & ( -(mbedtls_mpi_sint)s >> ( biL - 1 ) );
Simon Butcher15b15d12015-11-26 19:35:03 +00001642 u0 = u0 << s;
1643
1644 d1 = d >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001645 d0 = d & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001646
1647 u0_msw = u0 >> biH;
Simon Butcher9803d072016-01-03 00:24:34 +00001648 u0_lsw = u0 & uint_halfword_mask;
Simon Butcher15b15d12015-11-26 19:35:03 +00001649
1650 /*
1651 * Find the first quotient and remainder
1652 */
1653 q1 = u1 / d1;
1654 r0 = u1 - d1 * q1;
1655
1656 while( q1 >= radix || ( q1 * d0 > radix * r0 + u0_msw ) )
1657 {
1658 q1 -= 1;
1659 r0 += d1;
1660
1661 if ( r0 >= radix ) break;
1662 }
1663
Simon Butcherf5ba0452015-12-27 23:01:55 +00001664 rAX = ( u1 * radix ) + ( u0_msw - q1 * d );
Simon Butcher15b15d12015-11-26 19:35:03 +00001665 q0 = rAX / d1;
1666 r0 = rAX - q0 * d1;
1667
1668 while( q0 >= radix || ( q0 * d0 > radix * r0 + u0_lsw ) )
1669 {
1670 q0 -= 1;
1671 r0 += d1;
1672
1673 if ( r0 >= radix ) break;
1674 }
1675
1676 if (r != NULL)
Simon Butcherf5ba0452015-12-27 23:01:55 +00001677 *r = ( rAX * radix + u0_lsw - q0 * d ) >> s;
Simon Butcher15b15d12015-11-26 19:35:03 +00001678
1679 quotient = q1 * radix + q0;
1680
1681 return quotient;
1682#endif
1683}
1684
1685/*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001686 * Division by mbedtls_mpi: A = Q * B + R (HAC 14.20)
Paul Bakker5121ce52009-01-03 21:22:43 +00001687 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001688int mbedtls_mpi_div_mpi( mbedtls_mpi *Q, mbedtls_mpi *R, const mbedtls_mpi *A,
1689 const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001690{
Paul Bakker23986e52011-04-24 08:57:21 +00001691 int ret;
1692 size_t i, n, t, k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001693 mbedtls_mpi X, Y, Z, T1, T2;
Hanno Becker73d7d792018-12-11 10:35:51 +00001694 MPI_VALIDATE_RET( A != NULL );
1695 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001696
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001697 if( mbedtls_mpi_cmp_int( B, 0 ) == 0 )
1698 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001699
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001700 mbedtls_mpi_init( &X ); mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &Z );
1701 mbedtls_mpi_init( &T1 ); mbedtls_mpi_init( &T2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001702
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001703 if( mbedtls_mpi_cmp_abs( A, B ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001704 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001705 if( Q != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_lset( Q, 0 ) );
1706 if( R != NULL ) MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001707 return( 0 );
1708 }
1709
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001710 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &X, A ) );
1711 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001712 X.s = Y.s = 1;
1713
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001714 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &Z, A->n + 2 ) );
1715 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Z, 0 ) );
1716 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T1, 2 ) );
1717 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T2, 3 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001718
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02001719 k = mbedtls_mpi_bitlen( &Y ) % biL;
Paul Bakkerf9688572011-05-05 10:00:45 +00001720 if( k < biL - 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001721 {
1722 k = biL - 1 - k;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001723 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &X, k ) );
1724 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, k ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001725 }
1726 else k = 0;
1727
1728 n = X.n - 1;
1729 t = Y.n - 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001730 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001731
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001732 while( mbedtls_mpi_cmp_mpi( &X, &Y ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001733 {
1734 Z.p[n - t]++;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001735 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &Y ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001736 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001737 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, biL * ( n - t ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001738
1739 for( i = n; i > t ; i-- )
1740 {
1741 if( X.p[i] >= Y.p[t] )
1742 Z.p[i - t - 1] = ~0;
1743 else
1744 {
Simon Butcher15b15d12015-11-26 19:35:03 +00001745 Z.p[i - t - 1] = mbedtls_int_div_int( X.p[i], X.p[i - 1],
1746 Y.p[t], NULL);
Paul Bakker5121ce52009-01-03 21:22:43 +00001747 }
1748
1749 Z.p[i - t - 1]++;
1750 do
1751 {
1752 Z.p[i - t - 1]--;
1753
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001754 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &T1, 0 ) );
Paul Bakker66d5d072014-06-17 16:39:18 +02001755 T1.p[0] = ( t < 1 ) ? 0 : Y.p[t - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001756 T1.p[1] = Y.p[t];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001757 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &T1, Z.p[i - t - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001758
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001759 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &T2, 0 ) );
Paul Bakker66d5d072014-06-17 16:39:18 +02001760 T2.p[0] = ( i < 2 ) ? 0 : X.p[i - 2];
1761 T2.p[1] = ( i < 1 ) ? 0 : X.p[i - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001762 T2.p[2] = X.p[i];
1763 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001764 while( mbedtls_mpi_cmp_mpi( &T1, &T2 ) > 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001765
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001766 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_int( &T1, &Y, Z.p[i - t - 1] ) );
1767 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1768 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001769
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001770 if( mbedtls_mpi_cmp_int( &X, 0 ) < 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001771 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001772 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &T1, &Y ) );
1773 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &T1, biL * ( i - t - 1 ) ) );
1774 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &X, &X, &T1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001775 Z.p[i - t - 1]--;
1776 }
1777 }
1778
1779 if( Q != NULL )
1780 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001781 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( Q, &Z ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001782 Q->s = A->s * B->s;
1783 }
1784
1785 if( R != NULL )
1786 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001787 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &X, k ) );
Paul Bakkerf02c5642012-11-13 10:25:21 +00001788 X.s = A->s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001789 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( R, &X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001790
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001791 if( mbedtls_mpi_cmp_int( R, 0 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00001792 R->s = 1;
1793 }
1794
1795cleanup:
1796
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001797 mbedtls_mpi_free( &X ); mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &Z );
1798 mbedtls_mpi_free( &T1 ); mbedtls_mpi_free( &T2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00001799
1800 return( ret );
1801}
1802
1803/*
1804 * Division by int: A = Q * b + R
Paul Bakker5121ce52009-01-03 21:22:43 +00001805 */
Hanno Becker73d7d792018-12-11 10:35:51 +00001806int mbedtls_mpi_div_int( mbedtls_mpi *Q, mbedtls_mpi *R,
1807 const mbedtls_mpi *A,
1808 mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001809{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001810 mbedtls_mpi _B;
1811 mbedtls_mpi_uint p[1];
Hanno Becker73d7d792018-12-11 10:35:51 +00001812 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001813
1814 p[0] = ( b < 0 ) ? -b : b;
1815 _B.s = ( b < 0 ) ? -1 : 1;
1816 _B.n = 1;
1817 _B.p = p;
1818
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001819 return( mbedtls_mpi_div_mpi( Q, R, A, &_B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001820}
1821
1822/*
1823 * Modulo: R = A mod B
1824 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001825int mbedtls_mpi_mod_mpi( mbedtls_mpi *R, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00001826{
1827 int ret;
Hanno Becker73d7d792018-12-11 10:35:51 +00001828 MPI_VALIDATE_RET( R != NULL );
1829 MPI_VALIDATE_RET( A != NULL );
1830 MPI_VALIDATE_RET( B != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001831
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001832 if( mbedtls_mpi_cmp_int( B, 0 ) < 0 )
1833 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001834
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001835 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( NULL, R, A, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001836
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001837 while( mbedtls_mpi_cmp_int( R, 0 ) < 0 )
1838 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001839
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001840 while( mbedtls_mpi_cmp_mpi( R, B ) >= 0 )
1841 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( R, R, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001842
1843cleanup:
1844
1845 return( ret );
1846}
1847
1848/*
1849 * Modulo: r = A mod b
1850 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001851int mbedtls_mpi_mod_int( mbedtls_mpi_uint *r, const mbedtls_mpi *A, mbedtls_mpi_sint b )
Paul Bakker5121ce52009-01-03 21:22:43 +00001852{
Paul Bakker23986e52011-04-24 08:57:21 +00001853 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001854 mbedtls_mpi_uint x, y, z;
Hanno Becker73d7d792018-12-11 10:35:51 +00001855 MPI_VALIDATE_RET( r != NULL );
1856 MPI_VALIDATE_RET( A != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00001857
1858 if( b == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001859 return( MBEDTLS_ERR_MPI_DIVISION_BY_ZERO );
Paul Bakker5121ce52009-01-03 21:22:43 +00001860
1861 if( b < 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001862 return( MBEDTLS_ERR_MPI_NEGATIVE_VALUE );
Paul Bakker5121ce52009-01-03 21:22:43 +00001863
1864 /*
1865 * handle trivial cases
1866 */
1867 if( b == 1 )
1868 {
1869 *r = 0;
1870 return( 0 );
1871 }
1872
1873 if( b == 2 )
1874 {
1875 *r = A->p[0] & 1;
1876 return( 0 );
1877 }
1878
1879 /*
1880 * general case
1881 */
Paul Bakker23986e52011-04-24 08:57:21 +00001882 for( i = A->n, y = 0; i > 0; i-- )
Paul Bakker5121ce52009-01-03 21:22:43 +00001883 {
Paul Bakker23986e52011-04-24 08:57:21 +00001884 x = A->p[i - 1];
Paul Bakker5121ce52009-01-03 21:22:43 +00001885 y = ( y << biH ) | ( x >> biH );
1886 z = y / b;
1887 y -= z * b;
1888
1889 x <<= biH;
1890 y = ( y << biH ) | ( x >> biH );
1891 z = y / b;
1892 y -= z * b;
1893 }
1894
Paul Bakkerce40a6d2009-06-23 19:46:08 +00001895 /*
1896 * If A is negative, then the current y represents a negative value.
1897 * Flipping it to the positive side.
1898 */
1899 if( A->s < 0 && y != 0 )
1900 y = b - y;
1901
Paul Bakker5121ce52009-01-03 21:22:43 +00001902 *r = y;
1903
1904 return( 0 );
1905}
1906
1907/*
1908 * Fast Montgomery initialization (thanks to Tom St Denis)
1909 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001910static void mpi_montg_init( mbedtls_mpi_uint *mm, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00001911{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001912 mbedtls_mpi_uint x, m0 = N->p[0];
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01001913 unsigned int i;
Paul Bakker5121ce52009-01-03 21:22:43 +00001914
1915 x = m0;
1916 x += ( ( m0 + 2 ) & 4 ) << 1;
Paul Bakker5121ce52009-01-03 21:22:43 +00001917
Manuel Pégourié-Gonnardfdf3f0e2014-03-11 13:47:05 +01001918 for( i = biL; i >= 8; i /= 2 )
1919 x *= ( 2 - ( m0 * x ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00001920
1921 *mm = ~x + 1;
1922}
1923
Gilles Peskine3ce3ddf2020-06-04 15:00:49 +02001924/** Montgomery multiplication: A = A * B * R^-1 mod N (HAC 14.36)
1925 *
1926 * \param[in,out] A One of the numbers to multiply.
Gilles Peskine635a3742020-06-08 22:37:50 +02001927 * It must have at least as many limbs as N
1928 * (A->n >= N->n), and any limbs beyond n are ignored.
Gilles Peskine3ce3ddf2020-06-04 15:00:49 +02001929 * On successful completion, A contains the result of
1930 * the multiplication A * B * R^-1 mod N where
1931 * R = (2^ciL)^n.
1932 * \param[in] B One of the numbers to multiply.
1933 * It must be nonzero and must not have more limbs than N
1934 * (B->n <= N->n).
1935 * \param[in] N The modulo. N must be odd.
1936 * \param mm The value calculated by `mpi_montg_init(&mm, N)`.
1937 * This is -N^-1 mod 2^ciL.
1938 * \param[in,out] T A bignum for temporary storage.
1939 * It must be at least twice the limb size of N plus 2
1940 * (T->n >= 2 * (N->n + 1)).
1941 * Its initial content is unused and
1942 * its final content is indeterminate.
1943 * Note that unlike the usual convention in the library
1944 * for `const mbedtls_mpi*`, the content of T can change.
Paul Bakker5121ce52009-01-03 21:22:43 +00001945 */
Gilles Peskinebdcb3962020-06-04 20:55:15 +02001946static 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 +02001947 const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00001948{
Paul Bakker23986e52011-04-24 08:57:21 +00001949 size_t i, n, m;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02001950 mbedtls_mpi_uint u0, u1, *d;
Paul Bakker5121ce52009-01-03 21:22:43 +00001951
1952 memset( T->p, 0, T->n * ciL );
1953
1954 d = T->p;
1955 n = N->n;
1956 m = ( B->n < n ) ? B->n : n;
1957
1958 for( i = 0; i < n; i++ )
1959 {
1960 /*
1961 * T = (T + u0*B + u1*N) / 2^biL
1962 */
1963 u0 = A->p[i];
1964 u1 = ( d[0] + u0 * B->p[0] ) * mm;
1965
1966 mpi_mul_hlp( m, B->p, d, u0 );
1967 mpi_mul_hlp( n, N->p, d, u1 );
1968
1969 *d++ = u0; d[n + 1] = 0;
1970 }
1971
Gilles Peskine635a3742020-06-08 22:37:50 +02001972 /* At this point, d is either the desired result or the desired result
1973 * plus N. We now potentially subtract N, avoiding leaking whether the
1974 * subtraction is performed through side channels. */
Paul Bakker5121ce52009-01-03 21:22:43 +00001975
Gilles Peskine635a3742020-06-08 22:37:50 +02001976 /* Copy the n least significant limbs of d to A, so that
1977 * A = d if d < N (recall that N has n limbs). */
1978 memcpy( A->p, d, n * ciL );
Gilles Peskinede719d52020-06-09 10:39:38 +02001979 /* If d >= N then we want to set A to d - N. To prevent timing attacks,
Gilles Peskine635a3742020-06-08 22:37:50 +02001980 * do the calculation without using conditional tests. */
1981 /* Set d to d0 + (2^biL)^n - N where d0 is the current value of d. */
Gilles Peskine8f672662020-06-04 21:05:24 +02001982 d[n] += 1;
Gilles Peskine36acd542020-06-08 21:58:22 +02001983 d[n] -= mpi_sub_hlp( n, d, N->p );
Gilles Peskine635a3742020-06-08 22:37:50 +02001984 /* If d0 < N then d < (2^biL)^n
1985 * so d[n] == 0 and we want to keep A as it is.
1986 * If d0 >= N then d >= (2^biL)^n, and d <= (2^biL)^n + N < 2 * (2^biL)^n
1987 * so d[n] == 1 and we want to set A to the result of the subtraction
1988 * which is d - (2^biL)^n, i.e. the n least significant limbs of d.
1989 * This exactly corresponds to a conditional assignment. */
1990 mpi_safe_cond_assign( n, A->p, d, (unsigned char) d[n] );
Paul Bakker5121ce52009-01-03 21:22:43 +00001991}
1992
1993/*
1994 * Montgomery reduction: A = A * R^-1 mod N
Gilles Peskine3ce3ddf2020-06-04 15:00:49 +02001995 *
1996 * See mpi_montmul() regarding constraints and guarantees on the parameters.
Paul Bakker5121ce52009-01-03 21:22:43 +00001997 */
Gilles Peskinebdcb3962020-06-04 20:55:15 +02001998static void mpi_montred( mbedtls_mpi *A, const mbedtls_mpi *N,
1999 mbedtls_mpi_uint mm, const mbedtls_mpi *T )
Paul Bakker5121ce52009-01-03 21:22:43 +00002000{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002001 mbedtls_mpi_uint z = 1;
2002 mbedtls_mpi U;
Paul Bakker5121ce52009-01-03 21:22:43 +00002003
Paul Bakker8ddb6452013-02-27 14:56:33 +01002004 U.n = U.s = (int) z;
Paul Bakker5121ce52009-01-03 21:22:43 +00002005 U.p = &z;
2006
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002007 mpi_montmul( A, &U, N, mm, T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002008}
2009
2010/*
2011 * Sliding-window exponentiation: X = A^E mod N (HAC 14.85)
2012 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002013int mbedtls_mpi_exp_mod( mbedtls_mpi *X, const mbedtls_mpi *A,
2014 const mbedtls_mpi *E, const mbedtls_mpi *N,
2015 mbedtls_mpi *_RR )
Paul Bakker5121ce52009-01-03 21:22:43 +00002016{
Paul Bakker23986e52011-04-24 08:57:21 +00002017 int ret;
2018 size_t wbits, wsize, one = 1;
2019 size_t i, j, nblimbs;
2020 size_t bufsize, nbits;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002021 mbedtls_mpi_uint ei, mm, state;
2022 mbedtls_mpi RR, T, W[ 2 << MBEDTLS_MPI_WINDOW_SIZE ], Apos;
Paul Bakkerf6198c12012-05-16 08:02:29 +00002023 int neg;
Paul Bakker5121ce52009-01-03 21:22:43 +00002024
Hanno Becker73d7d792018-12-11 10:35:51 +00002025 MPI_VALIDATE_RET( X != NULL );
2026 MPI_VALIDATE_RET( A != NULL );
2027 MPI_VALIDATE_RET( E != NULL );
2028 MPI_VALIDATE_RET( N != NULL );
2029
Hanno Becker8d1dd1b2017-09-28 11:02:24 +01002030 if( mbedtls_mpi_cmp_int( N, 0 ) <= 0 || ( N->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002031 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002032
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002033 if( mbedtls_mpi_cmp_int( E, 0 ) < 0 )
2034 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002035
2036 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002037 * Init temps and window size
2038 */
2039 mpi_montg_init( &mm, N );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002040 mbedtls_mpi_init( &RR ); mbedtls_mpi_init( &T );
2041 mbedtls_mpi_init( &Apos );
Paul Bakker5121ce52009-01-03 21:22:43 +00002042 memset( W, 0, sizeof( W ) );
2043
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002044 i = mbedtls_mpi_bitlen( E );
Paul Bakker5121ce52009-01-03 21:22:43 +00002045
2046 wsize = ( i > 671 ) ? 6 : ( i > 239 ) ? 5 :
2047 ( i > 79 ) ? 4 : ( i > 23 ) ? 3 : 1;
2048
Peter Kolbusb83d41d2018-12-11 14:01:44 -06002049#if( MBEDTLS_MPI_WINDOW_SIZE < 6 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002050 if( wsize > MBEDTLS_MPI_WINDOW_SIZE )
2051 wsize = MBEDTLS_MPI_WINDOW_SIZE;
Peter Kolbusb83d41d2018-12-11 14:01:44 -06002052#endif
Paul Bakkerb6d5f082011-11-25 11:52:11 +00002053
Paul Bakker5121ce52009-01-03 21:22:43 +00002054 j = N->n + 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002055 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, j ) );
2056 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[1], j ) );
2057 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &T, j * 2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002058
2059 /*
Paul Bakker50546922012-05-19 08:40:49 +00002060 * Compensate for negative A (and correct at the end)
2061 */
2062 neg = ( A->s == -1 );
Paul Bakker50546922012-05-19 08:40:49 +00002063 if( neg )
2064 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002065 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Apos, A ) );
Paul Bakker50546922012-05-19 08:40:49 +00002066 Apos.s = 1;
2067 A = &Apos;
2068 }
2069
2070 /*
Paul Bakker5121ce52009-01-03 21:22:43 +00002071 * If 1st call, pre-compute R^2 mod N
2072 */
2073 if( _RR == NULL || _RR->p == NULL )
2074 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002075 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &RR, 1 ) );
2076 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &RR, N->n * 2 * biL ) );
2077 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &RR, &RR, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002078
2079 if( _RR != NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002080 memcpy( _RR, &RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002081 }
2082 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002083 memcpy( &RR, _RR, sizeof( mbedtls_mpi ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002084
2085 /*
2086 * W[1] = A * R^2 * R^-1 mod N = A * R mod N
2087 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002088 if( mbedtls_mpi_cmp_mpi( A, N ) >= 0 )
2089 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &W[1], A, N ) );
Paul Bakkerc2024f42014-01-23 20:38:35 +01002090 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002091 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[1], A ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002092
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002093 mpi_montmul( &W[1], &RR, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002094
2095 /*
2096 * X = R^2 * R^-1 mod N = R mod N
2097 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002098 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &RR ) );
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002099 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002100
2101 if( wsize > 1 )
2102 {
2103 /*
2104 * W[1 << (wsize - 1)] = W[1] ^ (wsize - 1)
2105 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002106 j = one << ( wsize - 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002107
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002108 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[j], N->n + 1 ) );
2109 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[j], &W[1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002110
2111 for( i = 0; i < wsize - 1; i++ )
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002112 mpi_montmul( &W[j], &W[j], N, mm, &T );
Paul Bakker0d7702c2013-10-29 16:18:35 +01002113
Paul Bakker5121ce52009-01-03 21:22:43 +00002114 /*
2115 * W[i] = W[i - 1] * W[1]
2116 */
Paul Bakker66d5d072014-06-17 16:39:18 +02002117 for( i = j + 1; i < ( one << wsize ); i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002118 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002119 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( &W[i], N->n + 1 ) );
2120 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &W[i], &W[i - 1] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002121
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002122 mpi_montmul( &W[i], &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002123 }
2124 }
2125
2126 nblimbs = E->n;
2127 bufsize = 0;
2128 nbits = 0;
2129 wbits = 0;
2130 state = 0;
2131
2132 while( 1 )
2133 {
2134 if( bufsize == 0 )
2135 {
Paul Bakker0d7702c2013-10-29 16:18:35 +01002136 if( nblimbs == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002137 break;
2138
Paul Bakker0d7702c2013-10-29 16:18:35 +01002139 nblimbs--;
2140
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002141 bufsize = sizeof( mbedtls_mpi_uint ) << 3;
Paul Bakker5121ce52009-01-03 21:22:43 +00002142 }
2143
2144 bufsize--;
2145
2146 ei = (E->p[nblimbs] >> bufsize) & 1;
2147
2148 /*
2149 * skip leading 0s
2150 */
2151 if( ei == 0 && state == 0 )
2152 continue;
2153
2154 if( ei == 0 && state == 1 )
2155 {
2156 /*
2157 * out of window, square X
2158 */
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002159 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002160 continue;
2161 }
2162
2163 /*
2164 * add ei to current window
2165 */
2166 state = 2;
2167
2168 nbits++;
Paul Bakker66d5d072014-06-17 16:39:18 +02002169 wbits |= ( ei << ( wsize - nbits ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002170
2171 if( nbits == wsize )
2172 {
2173 /*
2174 * X = X^wsize R^-1 mod N
2175 */
2176 for( i = 0; i < wsize; i++ )
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002177 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002178
2179 /*
2180 * X = X * W[wbits] R^-1 mod N
2181 */
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002182 mpi_montmul( X, &W[wbits], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002183
2184 state--;
2185 nbits = 0;
2186 wbits = 0;
2187 }
2188 }
2189
2190 /*
2191 * process the remaining bits
2192 */
2193 for( i = 0; i < nbits; i++ )
2194 {
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002195 mpi_montmul( X, X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002196
2197 wbits <<= 1;
2198
Paul Bakker66d5d072014-06-17 16:39:18 +02002199 if( ( wbits & ( one << wsize ) ) != 0 )
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002200 mpi_montmul( X, &W[1], N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002201 }
2202
2203 /*
2204 * X = A^E * R * R^-1 mod N = A^E mod N
2205 */
Gilles Peskinebdcb3962020-06-04 20:55:15 +02002206 mpi_montred( X, N, mm, &T );
Paul Bakker5121ce52009-01-03 21:22:43 +00002207
Hanno Beckera4af1c42017-04-18 09:07:45 +01002208 if( neg && E->n != 0 && ( E->p[0] & 1 ) != 0 )
Paul Bakkerf6198c12012-05-16 08:02:29 +00002209 {
2210 X->s = -1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002211 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( X, N, X ) );
Paul Bakkerf6198c12012-05-16 08:02:29 +00002212 }
2213
Paul Bakker5121ce52009-01-03 21:22:43 +00002214cleanup:
2215
Paul Bakker66d5d072014-06-17 16:39:18 +02002216 for( i = ( one << ( wsize - 1 ) ); i < ( one << wsize ); i++ )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002217 mbedtls_mpi_free( &W[i] );
Paul Bakker5121ce52009-01-03 21:22:43 +00002218
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002219 mbedtls_mpi_free( &W[1] ); mbedtls_mpi_free( &T ); mbedtls_mpi_free( &Apos );
Paul Bakker6c591fa2011-05-05 11:49:20 +00002220
Paul Bakker75a28602014-03-31 12:08:17 +02002221 if( _RR == NULL || _RR->p == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002222 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002223
2224 return( ret );
2225}
2226
Paul Bakker5121ce52009-01-03 21:22:43 +00002227/*
2228 * Greatest common divisor: G = gcd(A, B) (HAC 14.54)
2229 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002230int mbedtls_mpi_gcd( mbedtls_mpi *G, const mbedtls_mpi *A, const mbedtls_mpi *B )
Paul Bakker5121ce52009-01-03 21:22:43 +00002231{
Paul Bakker23986e52011-04-24 08:57:21 +00002232 int ret;
2233 size_t lz, lzt;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002234 mbedtls_mpi TG, TA, TB;
Paul Bakker5121ce52009-01-03 21:22:43 +00002235
Hanno Becker73d7d792018-12-11 10:35:51 +00002236 MPI_VALIDATE_RET( G != NULL );
2237 MPI_VALIDATE_RET( A != NULL );
2238 MPI_VALIDATE_RET( B != NULL );
2239
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002240 mbedtls_mpi_init( &TG ); mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002241
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002242 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TA, A ) );
2243 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, B ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002244
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002245 lz = mbedtls_mpi_lsb( &TA );
2246 lzt = mbedtls_mpi_lsb( &TB );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002247
Paul Bakker66d5d072014-06-17 16:39:18 +02002248 if( lzt < lz )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002249 lz = lzt;
2250
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002251 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, lz ) );
2252 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, lz ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002253
Paul Bakker5121ce52009-01-03 21:22:43 +00002254 TA.s = TB.s = 1;
2255
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002256 while( mbedtls_mpi_cmp_int( &TA, 0 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002257 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002258 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, mbedtls_mpi_lsb( &TA ) ) );
2259 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, mbedtls_mpi_lsb( &TB ) ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002260
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002261 if( mbedtls_mpi_cmp_mpi( &TA, &TB ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002262 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002263 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TA, &TA, &TB ) );
2264 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TA, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002265 }
2266 else
2267 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002268 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_abs( &TB, &TB, &TA ) );
2269 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TB, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002270 }
2271 }
2272
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002273 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_l( &TB, lz ) );
2274 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( G, &TB ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002275
2276cleanup:
2277
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002278 mbedtls_mpi_free( &TG ); mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TB );
Paul Bakker5121ce52009-01-03 21:22:43 +00002279
2280 return( ret );
2281}
2282
Paul Bakker33dc46b2014-04-30 16:11:39 +02002283/*
2284 * Fill X with size bytes of random.
2285 *
2286 * Use a temporary bytes representation to make sure the result is the same
Paul Bakkerc37b0ac2014-05-01 14:19:23 +02002287 * regardless of the platform endianness (useful when f_rng is actually
Paul Bakker33dc46b2014-04-30 16:11:39 +02002288 * deterministic, eg for tests).
2289 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002290int mbedtls_mpi_fill_random( mbedtls_mpi *X, size_t size,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002291 int (*f_rng)(void *, unsigned char *, size_t),
2292 void *p_rng )
Paul Bakker287781a2011-03-26 13:18:49 +00002293{
Paul Bakker23986e52011-04-24 08:57:21 +00002294 int ret;
Hanno Becker6dab6202019-01-02 16:42:29 +00002295 size_t const limbs = CHARS_TO_LIMBS( size );
Hanno Beckerda1655a2017-10-18 14:21:44 +01002296 size_t const overhead = ( limbs * ciL ) - size;
2297 unsigned char *Xp;
2298
Hanno Becker8ce11a32018-12-19 16:18:52 +00002299 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002300 MPI_VALIDATE_RET( f_rng != NULL );
Paul Bakker33dc46b2014-04-30 16:11:39 +02002301
Hanno Beckerda1655a2017-10-18 14:21:44 +01002302 /* Ensure that target MPI has exactly the necessary number of limbs */
2303 if( X->n != limbs )
2304 {
2305 mbedtls_mpi_free( X );
2306 mbedtls_mpi_init( X );
2307 MBEDTLS_MPI_CHK( mbedtls_mpi_grow( X, limbs ) );
2308 }
2309 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( X, 0 ) );
Paul Bakker287781a2011-03-26 13:18:49 +00002310
Hanno Beckerda1655a2017-10-18 14:21:44 +01002311 Xp = (unsigned char*) X->p;
2312 f_rng( p_rng, Xp + overhead, size );
2313
Hanno Becker2be8a552018-10-25 12:40:09 +01002314 mpi_bigendian_to_host( X->p, limbs );
Paul Bakker287781a2011-03-26 13:18:49 +00002315
2316cleanup:
2317 return( ret );
2318}
2319
Paul Bakker5121ce52009-01-03 21:22:43 +00002320/*
2321 * Modular inverse: X = A^-1 mod N (HAC 14.61 / 14.64)
2322 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002323int mbedtls_mpi_inv_mod( mbedtls_mpi *X, const mbedtls_mpi *A, const mbedtls_mpi *N )
Paul Bakker5121ce52009-01-03 21:22:43 +00002324{
2325 int ret;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002326 mbedtls_mpi G, TA, TU, U1, U2, TB, TV, V1, V2;
Hanno Becker73d7d792018-12-11 10:35:51 +00002327 MPI_VALIDATE_RET( X != NULL );
2328 MPI_VALIDATE_RET( A != NULL );
2329 MPI_VALIDATE_RET( N != NULL );
Paul Bakker5121ce52009-01-03 21:22:43 +00002330
Hanno Becker4bcb4912017-04-18 15:49:39 +01002331 if( mbedtls_mpi_cmp_int( N, 1 ) <= 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002332 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002333
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002334 mbedtls_mpi_init( &TA ); mbedtls_mpi_init( &TU ); mbedtls_mpi_init( &U1 ); mbedtls_mpi_init( &U2 );
2335 mbedtls_mpi_init( &G ); mbedtls_mpi_init( &TB ); mbedtls_mpi_init( &TV );
2336 mbedtls_mpi_init( &V1 ); mbedtls_mpi_init( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002337
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002338 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &G, A, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002339
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002340 if( mbedtls_mpi_cmp_int( &G, 1 ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002341 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002342 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002343 goto cleanup;
2344 }
2345
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002346 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &TA, A, N ) );
2347 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TU, &TA ) );
2348 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TB, N ) );
2349 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &TV, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002350
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002351 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U1, 1 ) );
2352 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &U2, 0 ) );
2353 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V1, 0 ) );
2354 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002355
2356 do
2357 {
2358 while( ( TU.p[0] & 1 ) == 0 )
2359 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002360 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TU, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002361
2362 if( ( U1.p[0] & 1 ) != 0 || ( U2.p[0] & 1 ) != 0 )
2363 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002364 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &U1, &U1, &TB ) );
2365 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002366 }
2367
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002368 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U1, 1 ) );
2369 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &U2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002370 }
2371
2372 while( ( TV.p[0] & 1 ) == 0 )
2373 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002374 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &TV, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002375
2376 if( ( V1.p[0] & 1 ) != 0 || ( V2.p[0] & 1 ) != 0 )
2377 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002378 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, &TB ) );
2379 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &TA ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002380 }
2381
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002382 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V1, 1 ) );
2383 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &V2, 1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002384 }
2385
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002386 if( mbedtls_mpi_cmp_mpi( &TU, &TV ) >= 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002387 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002388 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TU, &TU, &TV ) );
2389 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U1, &U1, &V1 ) );
2390 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &U2, &U2, &V2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002391 }
2392 else
2393 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002394 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &TV, &TV, &TU ) );
2395 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, &U1 ) );
2396 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V2, &V2, &U2 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002397 }
2398 }
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002399 while( mbedtls_mpi_cmp_int( &TU, 0 ) != 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002400
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002401 while( mbedtls_mpi_cmp_int( &V1, 0 ) < 0 )
2402 MBEDTLS_MPI_CHK( mbedtls_mpi_add_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002403
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002404 while( mbedtls_mpi_cmp_mpi( &V1, N ) >= 0 )
2405 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_mpi( &V1, &V1, N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002406
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002407 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( X, &V1 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002408
2409cleanup:
2410
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002411 mbedtls_mpi_free( &TA ); mbedtls_mpi_free( &TU ); mbedtls_mpi_free( &U1 ); mbedtls_mpi_free( &U2 );
2412 mbedtls_mpi_free( &G ); mbedtls_mpi_free( &TB ); mbedtls_mpi_free( &TV );
2413 mbedtls_mpi_free( &V1 ); mbedtls_mpi_free( &V2 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002414
2415 return( ret );
2416}
2417
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002418#if defined(MBEDTLS_GENPRIME)
Paul Bakkerd9374b02012-11-02 11:02:58 +00002419
Paul Bakker5121ce52009-01-03 21:22:43 +00002420static const int small_prime[] =
2421{
2422 3, 5, 7, 11, 13, 17, 19, 23,
2423 29, 31, 37, 41, 43, 47, 53, 59,
2424 61, 67, 71, 73, 79, 83, 89, 97,
2425 101, 103, 107, 109, 113, 127, 131, 137,
2426 139, 149, 151, 157, 163, 167, 173, 179,
2427 181, 191, 193, 197, 199, 211, 223, 227,
2428 229, 233, 239, 241, 251, 257, 263, 269,
2429 271, 277, 281, 283, 293, 307, 311, 313,
2430 317, 331, 337, 347, 349, 353, 359, 367,
2431 373, 379, 383, 389, 397, 401, 409, 419,
2432 421, 431, 433, 439, 443, 449, 457, 461,
2433 463, 467, 479, 487, 491, 499, 503, 509,
2434 521, 523, 541, 547, 557, 563, 569, 571,
2435 577, 587, 593, 599, 601, 607, 613, 617,
2436 619, 631, 641, 643, 647, 653, 659, 661,
2437 673, 677, 683, 691, 701, 709, 719, 727,
2438 733, 739, 743, 751, 757, 761, 769, 773,
2439 787, 797, 809, 811, 821, 823, 827, 829,
2440 839, 853, 857, 859, 863, 877, 881, 883,
2441 887, 907, 911, 919, 929, 937, 941, 947,
2442 953, 967, 971, 977, 983, 991, 997, -103
2443};
2444
2445/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002446 * Small divisors test (X must be positive)
2447 *
2448 * Return values:
2449 * 0: no small factor (possible prime, more tests needed)
2450 * 1: certain prime
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002451 * MBEDTLS_ERR_MPI_NOT_ACCEPTABLE: certain non-prime
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002452 * other negative: error
Paul Bakker5121ce52009-01-03 21:22:43 +00002453 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002454static int mpi_check_small_factors( const mbedtls_mpi *X )
Paul Bakker5121ce52009-01-03 21:22:43 +00002455{
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002456 int ret = 0;
2457 size_t i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002458 mbedtls_mpi_uint r;
Paul Bakker5121ce52009-01-03 21:22:43 +00002459
Paul Bakker5121ce52009-01-03 21:22:43 +00002460 if( ( X->p[0] & 1 ) == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002461 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002462
2463 for( i = 0; small_prime[i] > 0; i++ )
2464 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002465 if( mbedtls_mpi_cmp_int( X, small_prime[i] ) <= 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002466 return( 1 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002467
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002468 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, small_prime[i] ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002469
2470 if( r == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002471 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Paul Bakker5121ce52009-01-03 21:22:43 +00002472 }
2473
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002474cleanup:
2475 return( ret );
2476}
2477
2478/*
2479 * Miller-Rabin pseudo-primality test (HAC 4.24)
2480 */
Janos Follathda31fa12018-09-03 14:45:23 +01002481static int mpi_miller_rabin( const mbedtls_mpi *X, size_t rounds,
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002482 int (*f_rng)(void *, unsigned char *, size_t),
2483 void *p_rng )
2484{
Pascal Junodb99183d2015-03-11 16:49:45 +01002485 int ret, count;
Janos Follathda31fa12018-09-03 14:45:23 +01002486 size_t i, j, k, s;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002487 mbedtls_mpi W, R, T, A, RR;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002488
Hanno Becker8ce11a32018-12-19 16:18:52 +00002489 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002490 MPI_VALIDATE_RET( f_rng != NULL );
2491
2492 mbedtls_mpi_init( &W ); mbedtls_mpi_init( &R );
2493 mbedtls_mpi_init( &T ); mbedtls_mpi_init( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002494 mbedtls_mpi_init( &RR );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002495
Paul Bakker5121ce52009-01-03 21:22:43 +00002496 /*
2497 * W = |X| - 1
2498 * R = W >> lsb( W )
2499 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002500 MBEDTLS_MPI_CHK( mbedtls_mpi_sub_int( &W, X, 1 ) );
2501 s = mbedtls_mpi_lsb( &W );
2502 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &R, &W ) );
2503 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &R, s ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002504
Janos Follathda31fa12018-09-03 14:45:23 +01002505 for( i = 0; i < rounds; i++ )
Paul Bakker5121ce52009-01-03 21:22:43 +00002506 {
2507 /*
2508 * pick a random A, 1 < A < |X| - 1
2509 */
Pascal Junodb99183d2015-03-11 16:49:45 +01002510 count = 0;
2511 do {
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002512 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( &A, X->n * ciL, f_rng, p_rng ) );
Pascal Junodb99183d2015-03-11 16:49:45 +01002513
Manuel Pégourié-Gonnardc0696c22015-06-18 16:47:17 +02002514 j = mbedtls_mpi_bitlen( &A );
2515 k = mbedtls_mpi_bitlen( &W );
Pascal Junodb99183d2015-03-11 16:49:45 +01002516 if (j > k) {
Darryl Greene3f95ed2018-10-02 13:21:35 +01002517 A.p[A.n - 1] &= ( (mbedtls_mpi_uint) 1 << ( k - ( A.n - 1 ) * biL - 1 ) ) - 1;
Pascal Junodb99183d2015-03-11 16:49:45 +01002518 }
2519
2520 if (count++ > 30) {
Jens Wiklanderdfd447e2019-01-17 13:30:57 +01002521 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
2522 goto cleanup;
Pascal Junodb99183d2015-03-11 16:49:45 +01002523 }
2524
Manuel Pégourié-Gonnard53c76c02015-04-17 20:15:36 +02002525 } while ( mbedtls_mpi_cmp_mpi( &A, &W ) >= 0 ||
2526 mbedtls_mpi_cmp_int( &A, 1 ) <= 0 );
Paul Bakker5121ce52009-01-03 21:22:43 +00002527
2528 /*
2529 * A = A^R mod |X|
2530 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002531 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &A, &A, &R, X, &RR ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002532
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002533 if( mbedtls_mpi_cmp_mpi( &A, &W ) == 0 ||
2534 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002535 continue;
2536
2537 j = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002538 while( j < s && mbedtls_mpi_cmp_mpi( &A, &W ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002539 {
2540 /*
2541 * A = A * A mod |X|
2542 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002543 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &T, &A, &A ) );
2544 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_mpi( &A, &T, X ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002545
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002546 if( mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002547 break;
2548
2549 j++;
2550 }
2551
2552 /*
2553 * not prime if A != |X| - 1 or A == 1
2554 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002555 if( mbedtls_mpi_cmp_mpi( &A, &W ) != 0 ||
2556 mbedtls_mpi_cmp_int( &A, 1 ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002557 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002558 ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker5121ce52009-01-03 21:22:43 +00002559 break;
2560 }
2561 }
2562
2563cleanup:
Hanno Becker73d7d792018-12-11 10:35:51 +00002564 mbedtls_mpi_free( &W ); mbedtls_mpi_free( &R );
2565 mbedtls_mpi_free( &T ); mbedtls_mpi_free( &A );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002566 mbedtls_mpi_free( &RR );
Paul Bakker5121ce52009-01-03 21:22:43 +00002567
2568 return( ret );
2569}
2570
2571/*
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002572 * Pseudo-primality test: small factors, then Miller-Rabin
2573 */
Janos Follatha0b67c22018-09-18 14:48:23 +01002574int mbedtls_mpi_is_prime_ext( const mbedtls_mpi *X, int rounds,
2575 int (*f_rng)(void *, unsigned char *, size_t),
2576 void *p_rng )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002577{
2578 int ret;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002579 mbedtls_mpi XX;
Hanno Becker8ce11a32018-12-19 16:18:52 +00002580 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002581 MPI_VALIDATE_RET( f_rng != NULL );
Manuel Pégourié-Gonnard7f4ed672014-10-14 20:56:02 +02002582
2583 XX.s = 1;
2584 XX.n = X->n;
2585 XX.p = X->p;
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002586
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002587 if( mbedtls_mpi_cmp_int( &XX, 0 ) == 0 ||
2588 mbedtls_mpi_cmp_int( &XX, 1 ) == 0 )
2589 return( MBEDTLS_ERR_MPI_NOT_ACCEPTABLE );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002590
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002591 if( mbedtls_mpi_cmp_int( &XX, 2 ) == 0 )
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002592 return( 0 );
2593
2594 if( ( ret = mpi_check_small_factors( &XX ) ) != 0 )
2595 {
2596 if( ret == 1 )
2597 return( 0 );
2598
2599 return( ret );
2600 }
2601
Janos Follathda31fa12018-09-03 14:45:23 +01002602 return( mpi_miller_rabin( &XX, rounds, f_rng, p_rng ) );
Janos Follathf301d232018-08-14 13:34:01 +01002603}
2604
Janos Follatha0b67c22018-09-18 14:48:23 +01002605#if !defined(MBEDTLS_DEPRECATED_REMOVED)
Janos Follathf301d232018-08-14 13:34:01 +01002606/*
2607 * Pseudo-primality test, error probability 2^-80
2608 */
2609int mbedtls_mpi_is_prime( const mbedtls_mpi *X,
2610 int (*f_rng)(void *, unsigned char *, size_t),
2611 void *p_rng )
2612{
Hanno Becker8ce11a32018-12-19 16:18:52 +00002613 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002614 MPI_VALIDATE_RET( f_rng != NULL );
2615
Janos Follatha0b67c22018-09-18 14:48:23 +01002616 /*
2617 * In the past our key generation aimed for an error rate of at most
2618 * 2^-80. Since this function is deprecated, aim for the same certainty
2619 * here as well.
2620 */
Hanno Becker73d7d792018-12-11 10:35:51 +00002621 return( mbedtls_mpi_is_prime_ext( X, 40, f_rng, p_rng ) );
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002622}
Janos Follatha0b67c22018-09-18 14:48:23 +01002623#endif
Manuel Pégourié-Gonnard378fb4b2013-11-22 18:39:18 +01002624
2625/*
Paul Bakker5121ce52009-01-03 21:22:43 +00002626 * Prime number generation
Jethro Beekman66689272018-02-14 19:24:10 -08002627 *
Janos Follathf301d232018-08-14 13:34:01 +01002628 * To generate an RSA key in a way recommended by FIPS 186-4, both primes must
2629 * be either 1024 bits or 1536 bits long, and flags must contain
2630 * MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR.
Paul Bakker5121ce52009-01-03 21:22:43 +00002631 */
Janos Follath7c025a92018-08-14 11:08:41 +01002632int mbedtls_mpi_gen_prime( mbedtls_mpi *X, size_t nbits, int flags,
Paul Bakkera3d195c2011-11-27 21:07:34 +00002633 int (*f_rng)(void *, unsigned char *, size_t),
2634 void *p_rng )
Paul Bakker5121ce52009-01-03 21:22:43 +00002635{
Jethro Beekman66689272018-02-14 19:24:10 -08002636#ifdef MBEDTLS_HAVE_INT64
2637// ceil(2^63.5)
2638#define CEIL_MAXUINT_DIV_SQRT2 0xb504f333f9de6485ULL
2639#else
2640// ceil(2^31.5)
2641#define CEIL_MAXUINT_DIV_SQRT2 0xb504f334U
2642#endif
2643 int ret = MBEDTLS_ERR_MPI_NOT_ACCEPTABLE;
Paul Bakker23986e52011-04-24 08:57:21 +00002644 size_t k, n;
Janos Follathda31fa12018-09-03 14:45:23 +01002645 int rounds;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002646 mbedtls_mpi_uint r;
2647 mbedtls_mpi Y;
Paul Bakker5121ce52009-01-03 21:22:43 +00002648
Hanno Becker8ce11a32018-12-19 16:18:52 +00002649 MPI_VALIDATE_RET( X != NULL );
Hanno Becker73d7d792018-12-11 10:35:51 +00002650 MPI_VALIDATE_RET( f_rng != NULL );
2651
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002652 if( nbits < 3 || nbits > MBEDTLS_MPI_MAX_BITS )
2653 return( MBEDTLS_ERR_MPI_BAD_INPUT_DATA );
Paul Bakker5121ce52009-01-03 21:22:43 +00002654
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002655 mbedtls_mpi_init( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002656
2657 n = BITS_TO_LIMBS( nbits );
2658
Janos Follathda31fa12018-09-03 14:45:23 +01002659 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_LOW_ERR ) == 0 )
2660 {
2661 /*
2662 * 2^-80 error probability, number of rounds chosen per HAC, table 4.4
2663 */
2664 rounds = ( ( nbits >= 1300 ) ? 2 : ( nbits >= 850 ) ? 3 :
2665 ( nbits >= 650 ) ? 4 : ( nbits >= 350 ) ? 8 :
2666 ( nbits >= 250 ) ? 12 : ( nbits >= 150 ) ? 18 : 27 );
2667 }
2668 else
2669 {
2670 /*
2671 * 2^-100 error probability, number of rounds computed based on HAC,
2672 * fact 4.48
2673 */
2674 rounds = ( ( nbits >= 1450 ) ? 4 : ( nbits >= 1150 ) ? 5 :
2675 ( nbits >= 1000 ) ? 6 : ( nbits >= 850 ) ? 7 :
2676 ( nbits >= 750 ) ? 8 : ( nbits >= 500 ) ? 13 :
2677 ( nbits >= 250 ) ? 28 : ( nbits >= 150 ) ? 40 : 51 );
2678 }
2679
Jethro Beekman66689272018-02-14 19:24:10 -08002680 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002681 {
Jethro Beekman66689272018-02-14 19:24:10 -08002682 MBEDTLS_MPI_CHK( mbedtls_mpi_fill_random( X, n * ciL, f_rng, p_rng ) );
2683 /* make sure generated number is at least (nbits-1)+0.5 bits (FIPS 186-4 §B.3.3 steps 4.4, 5.5) */
2684 if( X->p[n-1] < CEIL_MAXUINT_DIV_SQRT2 ) continue;
2685
2686 k = n * biL;
2687 if( k > nbits ) MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( X, k - nbits ) );
2688 X->p[0] |= 1;
2689
Janos Follath7c025a92018-08-14 11:08:41 +01002690 if( ( flags & MBEDTLS_MPI_GEN_PRIME_FLAG_DH ) == 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002691 {
Janos Follatha0b67c22018-09-18 14:48:23 +01002692 ret = mbedtls_mpi_is_prime_ext( X, rounds, f_rng, p_rng );
Jethro Beekman66689272018-02-14 19:24:10 -08002693
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002694 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
Paul Bakker5121ce52009-01-03 21:22:43 +00002695 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002696 }
Jethro Beekman66689272018-02-14 19:24:10 -08002697 else
Paul Bakker5121ce52009-01-03 21:22:43 +00002698 {
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002699 /*
Jethro Beekman66689272018-02-14 19:24:10 -08002700 * An necessary condition for Y and X = 2Y + 1 to be prime
2701 * is X = 2 mod 3 (which is equivalent to Y = 2 mod 3).
2702 * Make sure it is satisfied, while keeping X = 3 mod 4
Manuel Pégourié-Gonnardddf76152013-11-22 19:58:22 +01002703 */
Jethro Beekman66689272018-02-14 19:24:10 -08002704
2705 X->p[0] |= 2;
2706
2707 MBEDTLS_MPI_CHK( mbedtls_mpi_mod_int( &r, X, 3 ) );
2708 if( r == 0 )
2709 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 8 ) );
2710 else if( r == 1 )
2711 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 4 ) );
2712
2713 /* Set Y = (X-1) / 2, which is X / 2 because X is odd */
2714 MBEDTLS_MPI_CHK( mbedtls_mpi_copy( &Y, X ) );
2715 MBEDTLS_MPI_CHK( mbedtls_mpi_shift_r( &Y, 1 ) );
2716
2717 while( 1 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002718 {
Jethro Beekman66689272018-02-14 19:24:10 -08002719 /*
2720 * First, check small factors for X and Y
2721 * before doing Miller-Rabin on any of them
2722 */
2723 if( ( ret = mpi_check_small_factors( X ) ) == 0 &&
2724 ( ret = mpi_check_small_factors( &Y ) ) == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002725 ( ret = mpi_miller_rabin( X, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002726 == 0 &&
Janos Follathda31fa12018-09-03 14:45:23 +01002727 ( ret = mpi_miller_rabin( &Y, rounds, f_rng, p_rng ) )
Janos Follathf301d232018-08-14 13:34:01 +01002728 == 0 )
Jethro Beekman66689272018-02-14 19:24:10 -08002729 goto cleanup;
2730
2731 if( ret != MBEDTLS_ERR_MPI_NOT_ACCEPTABLE )
2732 goto cleanup;
2733
2734 /*
2735 * Next candidates. We want to preserve Y = (X-1) / 2 and
2736 * Y = 1 mod 2 and Y = 2 mod 3 (eq X = 3 mod 4 and X = 2 mod 3)
2737 * so up Y by 6 and X by 12.
2738 */
2739 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( X, X, 12 ) );
2740 MBEDTLS_MPI_CHK( mbedtls_mpi_add_int( &Y, &Y, 6 ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002741 }
Paul Bakker5121ce52009-01-03 21:22:43 +00002742 }
2743 }
2744
2745cleanup:
2746
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002747 mbedtls_mpi_free( &Y );
Paul Bakker5121ce52009-01-03 21:22:43 +00002748
2749 return( ret );
2750}
2751
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002752#endif /* MBEDTLS_GENPRIME */
Paul Bakker5121ce52009-01-03 21:22:43 +00002753
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002754#if defined(MBEDTLS_SELF_TEST)
Paul Bakker5121ce52009-01-03 21:22:43 +00002755
Paul Bakker23986e52011-04-24 08:57:21 +00002756#define GCD_PAIR_COUNT 3
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002757
2758static const int gcd_pairs[GCD_PAIR_COUNT][3] =
2759{
2760 { 693, 609, 21 },
2761 { 1764, 868, 28 },
2762 { 768454923, 542167814, 1 }
2763};
2764
Paul Bakker5121ce52009-01-03 21:22:43 +00002765/*
2766 * Checkup routine
2767 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002768int mbedtls_mpi_self_test( int verbose )
Paul Bakker5121ce52009-01-03 21:22:43 +00002769{
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002770 int ret, i;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002771 mbedtls_mpi A, E, N, X, Y, U, V;
Paul Bakker5121ce52009-01-03 21:22:43 +00002772
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002773 mbedtls_mpi_init( &A ); mbedtls_mpi_init( &E ); mbedtls_mpi_init( &N ); mbedtls_mpi_init( &X );
2774 mbedtls_mpi_init( &Y ); mbedtls_mpi_init( &U ); mbedtls_mpi_init( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002775
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002776 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &A, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002777 "EFE021C2645FD1DC586E69184AF4A31E" \
2778 "D5F53E93B5F123FA41680867BA110131" \
2779 "944FE7952E2517337780CB0DB80E61AA" \
2780 "E7C8DDC6C5C6AADEB34EB38A2F40D5E6" ) );
2781
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002782 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &E, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002783 "B2E7EFD37075B9F03FF989C7C5051C20" \
2784 "34D2A323810251127E7BF8625A4F49A5" \
2785 "F3E27F4DA8BD59C47D6DAABA4C8127BD" \
2786 "5B5C25763222FEFCCFC38B832366C29E" ) );
2787
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002788 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &N, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002789 "0066A198186C18C10B2F5ED9B522752A" \
2790 "9830B69916E535C8F047518A889A43A5" \
2791 "94B6BED27A168D31D4A52F88925AA8F5" ) );
2792
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002793 MBEDTLS_MPI_CHK( mbedtls_mpi_mul_mpi( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002794
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002795 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002796 "602AB7ECA597A3D6B56FF9829A5E8B85" \
2797 "9E857EA95A03512E2BAE7391688D264A" \
2798 "A5663B0341DB9CCFD2C4C5F421FEC814" \
2799 "8001B72E848A38CAE1C65F78E56ABDEF" \
2800 "E12D3C039B8A02D6BE593F0BBBDA56F1" \
2801 "ECF677152EF804370C1A305CAF3B5BF1" \
2802 "30879B56C61DE584A0F53A2447A51E" ) );
2803
2804 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002805 mbedtls_printf( " MPI test #1 (mul_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002806
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002807 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002808 {
2809 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002810 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002811
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002812 ret = 1;
2813 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002814 }
2815
2816 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002817 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002818
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002819 MBEDTLS_MPI_CHK( mbedtls_mpi_div_mpi( &X, &Y, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002820
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002821 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002822 "256567336059E52CAE22925474705F39A94" ) );
2823
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002824 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &V, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002825 "6613F26162223DF488E9CD48CC132C7A" \
2826 "0AC93C701B001B092E4E5B9F73BCD27B" \
2827 "9EE50D0657C77F374E903CDFA4C642" ) );
2828
2829 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002830 mbedtls_printf( " MPI test #2 (div_mpi): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002831
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002832 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 ||
2833 mbedtls_mpi_cmp_mpi( &Y, &V ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002834 {
2835 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002836 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002837
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002838 ret = 1;
2839 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002840 }
2841
2842 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002843 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002844
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002845 MBEDTLS_MPI_CHK( mbedtls_mpi_exp_mod( &X, &A, &E, &N, NULL ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002846
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002847 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002848 "36E139AEA55215609D2816998ED020BB" \
2849 "BD96C37890F65171D948E9BC7CBAA4D9" \
2850 "325D24D6A3C12710F10A09FA08AB87" ) );
2851
2852 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002853 mbedtls_printf( " MPI test #3 (exp_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002854
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002855 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002856 {
2857 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002858 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002859
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002860 ret = 1;
2861 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002862 }
2863
2864 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002865 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002866
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002867 MBEDTLS_MPI_CHK( mbedtls_mpi_inv_mod( &X, &A, &N ) );
Paul Bakker5121ce52009-01-03 21:22:43 +00002868
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002869 MBEDTLS_MPI_CHK( mbedtls_mpi_read_string( &U, 16,
Paul Bakker5121ce52009-01-03 21:22:43 +00002870 "003A0AAEDD7E784FC07D8F9EC6E3BFD5" \
2871 "C3DBA76456363A10869622EAC2DD84EC" \
2872 "C5B8A74DAC4D09E03B5E0BE779F2DF61" ) );
2873
2874 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002875 mbedtls_printf( " MPI test #4 (inv_mod): " );
Paul Bakker5121ce52009-01-03 21:22:43 +00002876
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002877 if( mbedtls_mpi_cmp_mpi( &X, &U ) != 0 )
Paul Bakker5121ce52009-01-03 21:22:43 +00002878 {
2879 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002880 mbedtls_printf( "failed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002881
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002882 ret = 1;
2883 goto cleanup;
Paul Bakker5121ce52009-01-03 21:22:43 +00002884 }
2885
2886 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002887 mbedtls_printf( "passed\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002888
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002889 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002890 mbedtls_printf( " MPI test #5 (simple gcd): " );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002891
Paul Bakker66d5d072014-06-17 16:39:18 +02002892 for( i = 0; i < GCD_PAIR_COUNT; i++ )
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002893 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002894 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &X, gcd_pairs[i][0] ) );
2895 MBEDTLS_MPI_CHK( mbedtls_mpi_lset( &Y, gcd_pairs[i][1] ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002896
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002897 MBEDTLS_MPI_CHK( mbedtls_mpi_gcd( &A, &X, &Y ) );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002898
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002899 if( mbedtls_mpi_cmp_int( &A, gcd_pairs[i][2] ) != 0 )
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002900 {
2901 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002902 mbedtls_printf( "failed at %d\n", i );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002903
Manuel Pégourié-Gonnard9e987ed2014-01-20 10:03:15 +01002904 ret = 1;
2905 goto cleanup;
2906 }
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002907 }
2908
2909 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002910 mbedtls_printf( "passed\n" );
Paul Bakker4e0d7ca2009-01-29 22:24:33 +00002911
Paul Bakker5121ce52009-01-03 21:22:43 +00002912cleanup:
2913
2914 if( ret != 0 && verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002915 mbedtls_printf( "Unexpected error, return code = %08X\n", ret );
Paul Bakker5121ce52009-01-03 21:22:43 +00002916
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002917 mbedtls_mpi_free( &A ); mbedtls_mpi_free( &E ); mbedtls_mpi_free( &N ); mbedtls_mpi_free( &X );
2918 mbedtls_mpi_free( &Y ); mbedtls_mpi_free( &U ); mbedtls_mpi_free( &V );
Paul Bakker5121ce52009-01-03 21:22:43 +00002919
2920 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002921 mbedtls_printf( "\n" );
Paul Bakker5121ce52009-01-03 21:22:43 +00002922
2923 return( ret );
2924}
2925
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002926#endif /* MBEDTLS_SELF_TEST */
Paul Bakker5121ce52009-01-03 21:22:43 +00002927
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +02002928#endif /* MBEDTLS_BIGNUM_C */