289 lines
		
	
	
		
			9.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			289 lines
		
	
	
		
			9.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 *  Low-level modular bignum functions
 | 
						|
 *
 | 
						|
 *  Copyright The Mbed TLS Contributors
 | 
						|
 *  SPDX-License-Identifier: Apache-2.0
 | 
						|
 *
 | 
						|
 *  Licensed under the Apache License, Version 2.0 (the "License"); you may
 | 
						|
 *  not use this file except in compliance with the License.
 | 
						|
 *  You may obtain a copy of the License at
 | 
						|
 *
 | 
						|
 *  http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 *
 | 
						|
 *  Unless required by applicable law or agreed to in writing, software
 | 
						|
 *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 | 
						|
 *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
 *  See the License for the specific language governing permissions and
 | 
						|
 *  limitations under the License.
 | 
						|
 */
 | 
						|
 | 
						|
#include "common.h"
 | 
						|
 | 
						|
#if defined(MBEDTLS_BIGNUM_C)
 | 
						|
 | 
						|
#include <string.h>
 | 
						|
 | 
						|
#include "mbedtls/error.h"
 | 
						|
#include "mbedtls/platform_util.h"
 | 
						|
 | 
						|
#include "mbedtls/platform.h"
 | 
						|
 | 
						|
#include "bignum_core.h"
 | 
						|
#include "bignum_mod_raw.h"
 | 
						|
#include "bignum_mod.h"
 | 
						|
#include "constant_time_internal.h"
 | 
						|
 | 
						|
#include "bignum_mod_raw_invasive.h"
 | 
						|
 | 
						|
void mbedtls_mpi_mod_raw_cond_assign(mbedtls_mpi_uint *X,
 | 
						|
                                     const mbedtls_mpi_uint *A,
 | 
						|
                                     const mbedtls_mpi_mod_modulus *N,
 | 
						|
                                     unsigned char assign)
 | 
						|
{
 | 
						|
    mbedtls_mpi_core_cond_assign(X, A, N->limbs, mbedtls_ct_bool(assign));
 | 
						|
}
 | 
						|
 | 
						|
void mbedtls_mpi_mod_raw_cond_swap(mbedtls_mpi_uint *X,
 | 
						|
                                   mbedtls_mpi_uint *Y,
 | 
						|
                                   const mbedtls_mpi_mod_modulus *N,
 | 
						|
                                   unsigned char swap)
 | 
						|
{
 | 
						|
    mbedtls_mpi_core_cond_swap(X, Y, N->limbs, mbedtls_ct_bool(swap));
 | 
						|
}
 | 
						|
 | 
						|
int mbedtls_mpi_mod_raw_read(mbedtls_mpi_uint *X,
 | 
						|
                             const mbedtls_mpi_mod_modulus *N,
 | 
						|
                             const unsigned char *input,
 | 
						|
                             size_t input_length,
 | 
						|
                             mbedtls_mpi_mod_ext_rep ext_rep)
 | 
						|
{
 | 
						|
    int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
 | 
						|
 | 
						|
    switch (ext_rep) {
 | 
						|
        case MBEDTLS_MPI_MOD_EXT_REP_LE:
 | 
						|
            ret = mbedtls_mpi_core_read_le(X, N->limbs,
 | 
						|
                                           input, input_length);
 | 
						|
            break;
 | 
						|
        case MBEDTLS_MPI_MOD_EXT_REP_BE:
 | 
						|
            ret = mbedtls_mpi_core_read_be(X, N->limbs,
 | 
						|
                                           input, input_length);
 | 
						|
            break;
 | 
						|
        default:
 | 
						|
            return MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
 | 
						|
    }
 | 
						|
 | 
						|
    if (ret != 0) {
 | 
						|
        goto cleanup;
 | 
						|
    }
 | 
						|
 | 
						|
    if (!mbedtls_mpi_core_lt_ct(X, N->p, N->limbs)) {
 | 
						|
        ret = MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
 | 
						|
        goto cleanup;
 | 
						|
    }
 | 
						|
 | 
						|
cleanup:
 | 
						|
 | 
						|
    return ret;
 | 
						|
}
 | 
						|
 | 
						|
int mbedtls_mpi_mod_raw_write(const mbedtls_mpi_uint *A,
 | 
						|
                              const mbedtls_mpi_mod_modulus *N,
 | 
						|
                              unsigned char *output,
 | 
						|
                              size_t output_length,
 | 
						|
                              mbedtls_mpi_mod_ext_rep ext_rep)
 | 
						|
{
 | 
						|
    switch (ext_rep) {
 | 
						|
        case MBEDTLS_MPI_MOD_EXT_REP_LE:
 | 
						|
            return mbedtls_mpi_core_write_le(A, N->limbs,
 | 
						|
                                             output, output_length);
 | 
						|
        case MBEDTLS_MPI_MOD_EXT_REP_BE:
 | 
						|
            return mbedtls_mpi_core_write_be(A, N->limbs,
 | 
						|
                                             output, output_length);
 | 
						|
        default:
 | 
						|
            return MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
void mbedtls_mpi_mod_raw_sub(mbedtls_mpi_uint *X,
 | 
						|
                             const mbedtls_mpi_uint *A,
 | 
						|
                             const mbedtls_mpi_uint *B,
 | 
						|
                             const mbedtls_mpi_mod_modulus *N)
 | 
						|
{
 | 
						|
    mbedtls_mpi_uint c = mbedtls_mpi_core_sub(X, A, B, N->limbs);
 | 
						|
 | 
						|
    (void) mbedtls_mpi_core_add_if(X, N->p, N->limbs, (unsigned) c);
 | 
						|
}
 | 
						|
 | 
						|
MBEDTLS_STATIC_TESTABLE
 | 
						|
void mbedtls_mpi_mod_raw_fix_quasi_reduction(mbedtls_mpi_uint *X,
 | 
						|
                                             const mbedtls_mpi_mod_modulus *N)
 | 
						|
{
 | 
						|
    mbedtls_mpi_uint c = mbedtls_mpi_core_sub(X, X, N->p, N->limbs);
 | 
						|
 | 
						|
    (void) mbedtls_mpi_core_add_if(X, N->p, N->limbs, (unsigned) c);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
void mbedtls_mpi_mod_raw_mul(mbedtls_mpi_uint *X,
 | 
						|
                             const mbedtls_mpi_uint *A,
 | 
						|
                             const mbedtls_mpi_uint *B,
 | 
						|
                             const mbedtls_mpi_mod_modulus *N,
 | 
						|
                             mbedtls_mpi_uint *T)
 | 
						|
{
 | 
						|
    /* Standard (A * B) multiplication stored into pre-allocated T
 | 
						|
     * buffer of fixed limb size of (2N + 1).
 | 
						|
     *
 | 
						|
     * The space may not not fully filled by when
 | 
						|
     * MBEDTLS_MPI_MOD_REP_OPT_RED is used. */
 | 
						|
    const size_t T_limbs = BITS_TO_LIMBS(N->bits) * 2;
 | 
						|
    switch (N->int_rep) {
 | 
						|
        case MBEDTLS_MPI_MOD_REP_MONTGOMERY:
 | 
						|
            mbedtls_mpi_core_montmul(X, A, B, N->limbs, N->p, N->limbs,
 | 
						|
                                     N->rep.mont.mm, T);
 | 
						|
            break;
 | 
						|
        case MBEDTLS_MPI_MOD_REP_OPT_RED:
 | 
						|
            mbedtls_mpi_core_mul(T, A, N->limbs, B, N->limbs);
 | 
						|
 | 
						|
            /* Optimised Reduction */
 | 
						|
            (*N->rep.ored.modp)(T, T_limbs);
 | 
						|
 | 
						|
            /* Convert back to canonical representation */
 | 
						|
            mbedtls_mpi_mod_raw_fix_quasi_reduction(T, N);
 | 
						|
            memcpy(X, T, N->limbs * sizeof(mbedtls_mpi_uint));
 | 
						|
            break;
 | 
						|
        default:
 | 
						|
            break;
 | 
						|
    }
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
size_t mbedtls_mpi_mod_raw_inv_prime_working_limbs(size_t AN_limbs)
 | 
						|
{
 | 
						|
    /* mbedtls_mpi_mod_raw_inv_prime() needs a temporary for the exponent,
 | 
						|
     * which will be the same size as the modulus and input (AN_limbs),
 | 
						|
     * and additional space to pass to mbedtls_mpi_core_exp_mod(). */
 | 
						|
    return AN_limbs +
 | 
						|
           mbedtls_mpi_core_exp_mod_working_limbs(AN_limbs, AN_limbs);
 | 
						|
}
 | 
						|
 | 
						|
void mbedtls_mpi_mod_raw_inv_prime(mbedtls_mpi_uint *X,
 | 
						|
                                   const mbedtls_mpi_uint *A,
 | 
						|
                                   const mbedtls_mpi_uint *N,
 | 
						|
                                   size_t AN_limbs,
 | 
						|
                                   const mbedtls_mpi_uint *RR,
 | 
						|
                                   mbedtls_mpi_uint *T)
 | 
						|
{
 | 
						|
    /* Inversion by power: g^|G| = 1 => g^(-1) = g^(|G|-1), and
 | 
						|
     *                       |G| = N - 1, so we want
 | 
						|
     *                 g^(|G|-1) = g^(N - 2)
 | 
						|
     */
 | 
						|
 | 
						|
    /* Use the first AN_limbs of T to hold N - 2 */
 | 
						|
    mbedtls_mpi_uint *Nminus2 = T;
 | 
						|
    (void) mbedtls_mpi_core_sub_int(Nminus2, N, 2, AN_limbs);
 | 
						|
 | 
						|
    /* Rest of T is given to exp_mod for its working space */
 | 
						|
    mbedtls_mpi_core_exp_mod(X,
 | 
						|
                             A, N, AN_limbs, Nminus2, AN_limbs,
 | 
						|
                             RR, T + AN_limbs);
 | 
						|
}
 | 
						|
 | 
						|
void mbedtls_mpi_mod_raw_add(mbedtls_mpi_uint *X,
 | 
						|
                             const mbedtls_mpi_uint *A,
 | 
						|
                             const mbedtls_mpi_uint *B,
 | 
						|
                             const mbedtls_mpi_mod_modulus *N)
 | 
						|
{
 | 
						|
    mbedtls_mpi_uint carry, borrow;
 | 
						|
    carry  = mbedtls_mpi_core_add(X, A, B, N->limbs);
 | 
						|
    borrow = mbedtls_mpi_core_sub(X, X, N->p, N->limbs);
 | 
						|
    (void) mbedtls_mpi_core_add_if(X, N->p, N->limbs, (unsigned) (carry ^ borrow));
 | 
						|
}
 | 
						|
 | 
						|
int mbedtls_mpi_mod_raw_canonical_to_modulus_rep(
 | 
						|
    mbedtls_mpi_uint *X,
 | 
						|
    const mbedtls_mpi_mod_modulus *N)
 | 
						|
{
 | 
						|
    switch (N->int_rep) {
 | 
						|
        case MBEDTLS_MPI_MOD_REP_MONTGOMERY:
 | 
						|
            return mbedtls_mpi_mod_raw_to_mont_rep(X, N);
 | 
						|
        case MBEDTLS_MPI_MOD_REP_OPT_RED:
 | 
						|
            return 0;
 | 
						|
        default:
 | 
						|
            return MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int mbedtls_mpi_mod_raw_modulus_to_canonical_rep(
 | 
						|
    mbedtls_mpi_uint *X,
 | 
						|
    const mbedtls_mpi_mod_modulus *N)
 | 
						|
{
 | 
						|
    switch (N->int_rep) {
 | 
						|
        case MBEDTLS_MPI_MOD_REP_MONTGOMERY:
 | 
						|
            return mbedtls_mpi_mod_raw_from_mont_rep(X, N);
 | 
						|
        case MBEDTLS_MPI_MOD_REP_OPT_RED:
 | 
						|
            return 0;
 | 
						|
        default:
 | 
						|
            return MBEDTLS_ERR_MPI_BAD_INPUT_DATA;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
int mbedtls_mpi_mod_raw_random(mbedtls_mpi_uint *X,
 | 
						|
                               mbedtls_mpi_uint min,
 | 
						|
                               const mbedtls_mpi_mod_modulus *N,
 | 
						|
                               int (*f_rng)(void *, unsigned char *, size_t),
 | 
						|
                               void *p_rng)
 | 
						|
{
 | 
						|
    int ret = mbedtls_mpi_core_random(X, min, N->p, N->limbs, f_rng, p_rng);
 | 
						|
    if (ret != 0) {
 | 
						|
        return ret;
 | 
						|
    }
 | 
						|
    return mbedtls_mpi_mod_raw_canonical_to_modulus_rep(X, N);
 | 
						|
}
 | 
						|
 | 
						|
int mbedtls_mpi_mod_raw_to_mont_rep(mbedtls_mpi_uint *X,
 | 
						|
                                    const mbedtls_mpi_mod_modulus *N)
 | 
						|
{
 | 
						|
    mbedtls_mpi_uint *T;
 | 
						|
    const size_t t_limbs = mbedtls_mpi_core_montmul_working_limbs(N->limbs);
 | 
						|
 | 
						|
    if ((T = (mbedtls_mpi_uint *) mbedtls_calloc(t_limbs, ciL)) == NULL) {
 | 
						|
        return MBEDTLS_ERR_MPI_ALLOC_FAILED;
 | 
						|
    }
 | 
						|
 | 
						|
    mbedtls_mpi_core_to_mont_rep(X, X, N->p, N->limbs,
 | 
						|
                                 N->rep.mont.mm, N->rep.mont.rr, T);
 | 
						|
 | 
						|
    mbedtls_zeroize_and_free(T, t_limbs * ciL);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
int mbedtls_mpi_mod_raw_from_mont_rep(mbedtls_mpi_uint *X,
 | 
						|
                                      const mbedtls_mpi_mod_modulus *N)
 | 
						|
{
 | 
						|
    const size_t t_limbs = mbedtls_mpi_core_montmul_working_limbs(N->limbs);
 | 
						|
    mbedtls_mpi_uint *T;
 | 
						|
 | 
						|
    if ((T = (mbedtls_mpi_uint *) mbedtls_calloc(t_limbs, ciL)) == NULL) {
 | 
						|
        return MBEDTLS_ERR_MPI_ALLOC_FAILED;
 | 
						|
    }
 | 
						|
 | 
						|
    mbedtls_mpi_core_from_mont_rep(X, X, N->p, N->limbs, N->rep.mont.mm, T);
 | 
						|
 | 
						|
    mbedtls_zeroize_and_free(T, t_limbs * ciL);
 | 
						|
    return 0;
 | 
						|
}
 | 
						|
 | 
						|
void mbedtls_mpi_mod_raw_neg(mbedtls_mpi_uint *X,
 | 
						|
                             const mbedtls_mpi_uint *A,
 | 
						|
                             const mbedtls_mpi_mod_modulus *N)
 | 
						|
{
 | 
						|
    mbedtls_mpi_core_sub(X, N->p, A, N->limbs);
 | 
						|
 | 
						|
    /* If A=0 initially, then X=N now. Detect this by
 | 
						|
     * subtracting N and catching the carry. */
 | 
						|
    mbedtls_mpi_uint borrow = mbedtls_mpi_core_sub(X, X, N->p, N->limbs);
 | 
						|
    (void) mbedtls_mpi_core_add_if(X, N->p, N->limbs, (unsigned) borrow);
 | 
						|
}
 | 
						|
 | 
						|
#endif /* MBEDTLS_BIGNUM_C */
 |