@@ -335,7 +335,9 @@ static struct xmalloc_slot xmalloc_pool[] = {
335335
336336
337337#elif defined(WOLFBOOT_SIGN_RSA2048) || defined(WOLFBOOT_SIGN_RSA4096) || \
338- defined(WOLFBOOT_SIGN_RSA3072)
338+ defined(WOLFBOOT_SIGN_RSA3072) || \
339+ defined(WOLFBOOT_SIGN_RSAPSS2048) || defined(WOLFBOOT_SIGN_RSAPSS4096) || \
340+ defined(WOLFBOOT_SIGN_RSAPSS3072)
339341
340342#if defined(WOLFBOOT_HASH_SHA256 ) || defined(WOLFBOOT_HASH_SHA384 )
341343static uint32_t sha_block [HASH_BLOCK_SIZE ];
@@ -349,7 +351,7 @@ static uint32_t sha_block[HASH_BLOCK_SIZE];
349351static uint8_t asncheck_buf [ASNCHECK_BUF_SIZE ];
350352
351353#ifndef USE_FAST_MATH
352- #ifdef WOLFBOOT_SIGN_RSA2048
354+ #if defined( WOLFBOOT_SIGN_RSA2048 ) || defined( WOLFBOOT_SIGN_RSAPSS2048 )
353355 #define MP_SCHEME "SP RSA2048"
354356 #define MP_INT_DYNAMIC_SIZE MP_INT_SIZEOF(MP_BITS_CNT(2048))
355357 #define MP_BIGINT_MODEXP_SIZE (MP_INT_DYNAMIC_SIZE * 4)
@@ -362,7 +364,7 @@ static uint8_t asncheck_buf[ASNCHECK_BUF_SIZE];
362364 #define MPDIGIT_BUF1_SIZE (MP_DIGIT_SIZE * (72 * 4 + 3))
363365 static uint8_t mp_digit_buf1 [MPDIGIT_BUF1_SIZE ];
364366 #endif
365- #elif defined WOLFBOOT_SIGN_RSA3072
367+ #elif defined( WOLFBOOT_SIGN_RSA3072 ) || defined( WOLFBOOT_SIGN_RSAPSS3072 )
366368 #define MP_SCHEME "SP RSA3072"
367369 #define MP_INT_DYNAMIC_SIZE MP_INT_SIZEOF(MP_BITS_CNT(3072))
368370 #define MP_BIGINT_MODEXP_SIZE (MP_INT_DYNAMIC_SIZE * 4)
0 commit comments