@@ -307,7 +307,9 @@ static struct xmalloc_slot xmalloc_pool[] = {
307307
308308
309309#elif defined(WOLFBOOT_SIGN_RSA2048) || defined(WOLFBOOT_SIGN_RSA4096) || \
310- defined(WOLFBOOT_SIGN_RSA3072)
310+ defined(WOLFBOOT_SIGN_RSA3072) || \
311+ defined(WOLFBOOT_SIGN_RSAPSS2048) || defined(WOLFBOOT_SIGN_RSAPSS4096) || \
312+ defined(WOLFBOOT_SIGN_RSAPSS3072)
311313
312314#if defined(WOLFBOOT_HASH_SHA256 ) || defined(WOLFBOOT_HASH_SHA384 )
313315static uint32_t sha_block [HASH_BLOCK_SIZE ];
@@ -321,7 +323,7 @@ static uint32_t sha_block[HASH_BLOCK_SIZE];
321323static uint8_t asncheck_buf [ASNCHECK_BUF_SIZE ];
322324
323325#ifndef USE_FAST_MATH
324- #ifdef WOLFBOOT_SIGN_RSA2048
326+ #if defined( WOLFBOOT_SIGN_RSA2048 ) || defined( WOLFBOOT_SIGN_RSAPSS2048 )
325327 #define MP_SCHEME "SP RSA2048"
326328 #define MP_INT_DYNAMIC_SIZE MP_INT_SIZEOF(MP_BITS_CNT(2048))
327329 #define MP_BIGINT_MODEXP_SIZE (MP_INT_DYNAMIC_SIZE * 4)
@@ -334,7 +336,7 @@ static uint8_t asncheck_buf[ASNCHECK_BUF_SIZE];
334336 #define MPDIGIT_BUF1_SIZE (MP_DIGIT_SIZE * (72 * 4 + 3))
335337 static uint8_t mp_digit_buf1 [MPDIGIT_BUF1_SIZE ];
336338 #endif
337- #elif defined WOLFBOOT_SIGN_RSA3072
339+ #elif defined( WOLFBOOT_SIGN_RSA3072 ) || defined( WOLFBOOT_SIGN_RSAPSS3072 )
338340 #define MP_SCHEME "SP RSA3072"
339341 #define MP_INT_DYNAMIC_SIZE MP_INT_SIZEOF(MP_BITS_CNT(3072))
340342 #define MP_BIGINT_MODEXP_SIZE (MP_INT_DYNAMIC_SIZE * 4)
0 commit comments