Skip to content

Commit dba2aa3

Browse files
committed
fix xmalloc RSA PSS macro build error
1 parent 757c7dd commit dba2aa3

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/xmalloc.c

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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)
341343
static uint32_t sha_block[HASH_BLOCK_SIZE];
@@ -349,7 +351,7 @@ static uint32_t sha_block[HASH_BLOCK_SIZE];
349351
static 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

Comments
 (0)