When checking primes call only available SP ModExp functions

This commit is contained in:
Sean Parkinson
2019-06-20 18:05:25 +10:00
parent 19cb31d429
commit c82c9b481b
2 changed files with 18 additions and 0 deletions

View File

@ -4443,11 +4443,20 @@ static int mp_prime_miller_rabin (mp_int * a, mp_int * b, int *result)
goto LBL_R;
}
#if defined(WOLFSSL_HAVE_SP_RSA) || defined(WOLFSSL_HAVE_SP_DH)
#ifndef WOLFSSL_SP_NO_2048
if (mp_count_bits(a) == 1024)
err = sp_ModExp_1024(b, &r, a, &y);
else if (mp_count_bits(a) == 2048)
err = sp_ModExp_2048(b, &r, a, &y);
else
#endif
#ifndef WOLFSSL_SP_NO_3072
if (mp_count_bits(a) == 1536)
err = sp_ModExp_1536(b, &r, a, &y);
else if (mp_count_bits(a) == 3072)
err = sp_ModExp_3072(b, &r, a, &y);
else
#endif
#endif
err = mp_exptmod (b, &r, a, &y);
if (err != MP_OKAY)

View File

@ -3553,11 +3553,20 @@ static int fp_prime_miller_rabin_ex(fp_int * a, fp_int * b, int *result,
fp_zero(y);
#if (defined(WOLFSSL_HAVE_SP_RSA) && !defined(WOLFSSL_RSA_PUBLIC_ONLY)) || \
defined(WOLFSSL_HAVE_SP_DH)
#ifndef WOLFSSL_SP_NO_2048
if (fp_count_bits(a) == 1024)
sp_ModExp_1024(b, r, a, y);
else if (fp_count_bits(a) == 2048)
sp_ModExp_2048(b, r, a, y);
else
#endif
#ifndef WOLFSSL_SP_NO_3072
if (fp_count_bits(a) == 1536)
sp_ModExp_1536(b, r, a, y);
else if (fp_count_bits(a) == 3072)
sp_ModExp_3072(b, r, a, y);
else
#endif
#endif
fp_exptmod(b, r, a, y);