wolfSSL/wolfssl
1
0
Fork 1
mirror of https://github.com/wolfSSL/wolfssl.git synced 2025-07-30 10:47:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

Handle case where bits is 0 properly.

Browse Source
This commit is contained in:
Hayden Roche
2022-03-30 16:56:53 -07:00
parent 2077690bf0
commit 3f03f4e40a
1 changed files with 53 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

96
src/ssl.c
View File

@ -54566,55 +54566,65 @@ int wolfSSL_BN_rand(WOLFSSL_BIGNUM* bn, int bits, int top, int bottom)
} }
if (ret == WOLFSSL_SUCCESS) { if (ret == WOLFSSL_SUCCESS) {
buff = (byte*)XMALLOC(len, NULL, DYNAMIC_TYPE_TMP_BUFFER); if (len == 0) {
if (buff == NULL) { mp_zero((mp_int*)bn->internal);
WOLFSSL_MSG("Failed to allocate buffer.");
XFREE(buff, NULL, DYNAMIC_TYPE_TMP_BUFFER);
ret = WOLFSSL_FAILURE;
} }
} else {
buff = (byte*)XMALLOC(len, NULL, DYNAMIC_TYPE_TMP_BUFFER);
if (ret == WOLFSSL_SUCCESS && initGlobalRNG == 0 && if (buff == NULL) {
wolfSSL_RAND_Init() != WOLFSSL_SUCCESS) { WOLFSSL_MSG("Failed to allocate buffer.");
WOLFSSL_MSG("Failed to use global RNG."); XFREE(buff, NULL, DYNAMIC_TYPE_TMP_BUFFER);
ret = WOLFSSL_FAILURE;
}
if (ret == WOLFSSL_SUCCESS && wc_RNG_GenerateBlock(rng, buff, len) != 0) {
WOLFSSL_MSG("wc_RNG_GenerateBlock failed");
ret = WOLFSSL_FAILURE;
}
if (ret == WOLFSSL_SUCCESS &&
mp_read_unsigned_bin((mp_int*)bn->internal,buff,len) != MP_OKAY) {
WOLFSSL_MSG("mp_read_unsigned_bin failed");
ret = WOLFSSL_FAILURE;
}
if (ret == WOLFSSL_SUCCESS) {
/* Truncate to requested bit length. */
mp_rshb((mp_int*)bn->internal, 8 - (bits % 8));
if (top == 0) {
if (mp_set_bit((mp_int*)bn->internal, bits - 1) != MP_OKAY) {
WOLFSSL_MSG("Failed to set top bit");
ret = WOLFSSL_FAILURE; ret = WOLFSSL_FAILURE;
} }
}
else if (top > 0) { if (ret == WOLFSSL_SUCCESS && initGlobalRNG == 0 &&
if (mp_set_bit((mp_int*)bn->internal, bits - 1) != MP_OKAY || wolfSSL_RAND_Init() != WOLFSSL_SUCCESS) {
mp_set_bit((mp_int*)bn->internal, bits - 2) != MP_OKAY) { WOLFSSL_MSG("Failed to use global RNG.");
WOLFSSL_MSG("Failed to set top 2 bits");
ret = WOLFSSL_FAILURE; ret = WOLFSSL_FAILURE;
} }
}
}
if (ret == WOLFSSL_SUCCESS && bottom &&
mp_set_bit((mp_int*)bn->internal, 0) != MP_OKAY) {
WOLFSSL_MSG("Failed to set 0th bit");
ret = WOLFSSL_FAILURE;
}
if (buff != NULL) { if (ret == WOLFSSL_SUCCESS &&
XFREE(buff, NULL, DYNAMIC_TYPE_TMP_BUFFER); wc_RNG_GenerateBlock(rng, buff, len) != 0) {
WOLFSSL_MSG("wc_RNG_GenerateBlock failed");
ret = WOLFSSL_FAILURE;
}
if (ret == WOLFSSL_SUCCESS &&
mp_read_unsigned_bin((mp_int*)bn->internal,buff,len)
!= MP_OKAY) {
WOLFSSL_MSG("mp_read_unsigned_bin failed");
ret = WOLFSSL_FAILURE;
}
if (ret == WOLFSSL_SUCCESS) {
/* Truncate to requested bit length. */
mp_rshb((mp_int*)bn->internal, 8 - (bits % 8));
if (top == 0) {
if (mp_set_bit((mp_int*)bn->internal, bits - 1)
!= MP_OKAY) {
WOLFSSL_MSG("Failed to set top bit");
ret = WOLFSSL_FAILURE;
}
}
else if (top > 0) {
if (mp_set_bit((mp_int*)bn->internal, bits - 1)
!= MP_OKAY ||
mp_set_bit((mp_int*)bn->internal, bits - 2)
!= MP_OKAY) {
WOLFSSL_MSG("Failed to set top 2 bits");
ret = WOLFSSL_FAILURE;
}
}
}
if (ret == WOLFSSL_SUCCESS && bottom &&
mp_set_bit((mp_int*)bn->internal, 0) != MP_OKAY) {
WOLFSSL_MSG("Failed to set 0th bit");
ret = WOLFSSL_FAILURE;
}
if (buff != NULL) {
XFREE(buff, NULL, DYNAMIC_TYPE_TMP_BUFFER);
}
}
} }
WOLFSSL_LEAVE("wolfSSL_BN_rand", ret); WOLFSSL_LEAVE("wolfSSL_BN_rand", ret);