Skip to content

Commit e5d27b6

Browse files
JacobBarthelmehJacobBarthelmeh
committed
use heap hints where possible in benchmark
1 parent 5151a69 commit e5d27b6

1 file changed

Lines changed: 21 additions & 19 deletions

File tree

‎wolfcrypt/benchmark/benchmark.c‎

Lines changed: 21 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -2551,7 +2551,7 @@ typedef enum bench_stat_type {
25512551

25522552
if (bstat == NULL) {
25532553
/* allocate new and put on list */
2554-
bstat = (bench_stats_t*)XMALLOC(sizeof(bench_stats_t), NULL,
2554+
bstat = (bench_stats_t*)XMALLOC(sizeof(bench_stats_t), HEAP_HINT,
25552555
DYNAMIC_TYPE_INFO);
25562556
if (bstat) {
25572557
XMEMSET(bstat, 0, sizeof(bench_stats_t));
@@ -3719,7 +3719,7 @@ static WC_INLINE void bench_stats_free(void)
37193719
bench_stats_t* bstat;
37203720
for (bstat = bench_stats_head; bstat != NULL; ) {
37213721
bench_stats_t* next = bstat->next;
3722-
XFREE(bstat, NULL, DYNAMIC_TYPE_INFO);
3722+
XFREE(bstat, HEAP_HINT, DYNAMIC_TYPE_INFO);
37233723
bstat = next;
37243724
}
37253725
bench_stats_head = NULL;
@@ -6470,7 +6470,7 @@ static void bench_aesofb_internal(const byte* key,
64706470

64716471
bench_stats_prepare();
64726472

6473-
ret = wc_AesInit(&enc, NULL, INVALID_DEVID);
6473+
ret = wc_AesInit(&enc, HEAP_HINT, INVALID_DEVID);
64746474
if (ret != 0) {
64756475
printf("AesInit failed at L%d, ret = %d\n", __LINE__, ret);
64766476
return;
@@ -11539,7 +11539,7 @@ static void bench_lms_keygen(enum wc_LmsParm parm, byte* pub)
1153911539
return;
1154011540
}
1154111541

11542-
ret = wc_LmsKey_Init(&key, NULL, INVALID_DEVID);
11542+
ret = wc_LmsKey_Init(&key, HEAP_HINT, devId);
1154311543
if (ret) {
1154411544
printf("wc_LmsKey_Init failed: %d\n", ret);
1154511545
wc_FreeRng(&rng);
@@ -11555,7 +11555,7 @@ static void bench_lms_keygen(enum wc_LmsParm parm, byte* pub)
1155511555

1155611556
wc_LmsKey_Free(&key);
1155711557

11558-
ret = wc_LmsKey_Init(&key, NULL, INVALID_DEVID);
11558+
ret = wc_LmsKey_Init(&key, HEAP_HINT, devId);
1155911559
if (ret) {
1156011560
printf("wc_LmsKey_Init failed: %d\n", ret);
1156111561
goto exit_lms_keygen;
@@ -11644,7 +11644,7 @@ static void bench_lms_sign_verify(enum wc_LmsParm parm, byte* pub)
1164411644

1164511645
bench_stats_prepare();
1164611646

11647-
ret = wc_LmsKey_Init(&key, NULL, INVALID_DEVID);
11647+
ret = wc_LmsKey_Init(&key, HEAP_HINT, devId);
1164811648
if (ret) {
1164911649
printf("wc_LmsKey_Init failed: %d\n", ret);
1165011650
goto exit_lms_sign_verify;
@@ -12124,7 +12124,7 @@ static void bench_xmss_sign_verify(const char * params)
1212412124

1212512125
freeRng = 1;
1212612126

12127-
ret = wc_XmssKey_Init(&key, NULL, INVALID_DEVID);
12127+
ret = wc_XmssKey_Init(&key, HEAP_HINT, devId);
1212812128
if (ret != 0) {
1212912129
printf("wc_XmssKey_Init failed: %d\n", ret);
1213012130
goto exit_xmss_sign_verify;
@@ -12602,7 +12602,8 @@ void bench_slhdsa(int param)
1260212602
WC_ALLOC_VAR_EX(sig, byte, WC_SLHDSA_MAX_SIG_LEN, HEAP_HINT,
1260312603
DYNAMIC_TYPE_TMP_BUFFER, goto exit);
1260412604

12605-
ret = wc_SlhDsaKey_Init(key, (enum SlhDsaParam)param, NULL, INVALID_DEVID);
12605+
ret = wc_SlhDsaKey_Init(key, (enum SlhDsaParam)param, HEAP_HINT,
12606+
INVALID_DEVID);
1260612607
if (ret != 0) {
1260712608
goto exit;
1260812609
}
@@ -12664,7 +12665,8 @@ void bench_slhdsa(int param)
1266412665
goto exit;
1266512666
}
1266612667

12667-
ret = wc_SlhDsaKey_Init(key_vfy, (enum SlhDsaParam)param, NULL, INVALID_DEVID);
12668+
ret = wc_SlhDsaKey_Init(key_vfy, (enum SlhDsaParam)param, HEAP_HINT,
12669+
INVALID_DEVID);
1266812670
if (ret != 0) {
1266912671
goto exit;
1267012672
}
@@ -14215,7 +14217,7 @@ void bench_eccsiKeyGen(void)
1421514217
bench_stats_start(&count, &start);
1421614218
do {
1421714219
for (i = 0; i < genTimes; i++) {
14218-
wc_InitEccsiKey(genKey, NULL, INVALID_DEVID);
14220+
wc_InitEccsiKey(genKey, HEAP_HINT, devId);
1421914221
ret = wc_MakeEccsiKey(genKey, &gRng);
1422014222
wc_FreeEccsiKey(genKey);
1422114223
if (ret != 0) {
@@ -14260,7 +14262,7 @@ void bench_eccsiPairGen(void)
1426014262

1426114263
(void)mp_init(ssk);
1426214264
pvt = wc_ecc_new_point();
14263-
wc_InitEccsiKey(genKey, NULL, INVALID_DEVID);
14265+
wc_InitEccsiKey(genKey, HEAP_HINT, devId);
1426414266
(void)wc_MakeEccsiKey(genKey, &gRng);
1426514267

1426614268
/* RSK Gen */
@@ -14319,7 +14321,7 @@ void bench_eccsiValidate(void)
1431914321

1432014322
(void)mp_init(ssk);
1432114323
pvt = wc_ecc_new_point();
14322-
wc_InitEccsiKey(genKey, NULL, INVALID_DEVID);
14324+
wc_InitEccsiKey(genKey, HEAP_HINT, devId);
1432314325
(void)wc_MakeEccsiKey(genKey, &gRng);
1432414326
(void)wc_MakeEccsiPair(genKey, &gRng, WC_HASH_TYPE_SHA256, id, sizeof(id),
1432514327
ssk, pvt);
@@ -14384,7 +14386,7 @@ void bench_eccsi(void)
1438414386

1438514387
(void)mp_init(ssk);
1438614388
pvt = wc_ecc_new_point();
14387-
(void)wc_InitEccsiKey(genKey, NULL, INVALID_DEVID);
14389+
(void)wc_InitEccsiKey(genKey, HEAP_HINT, devId);
1438814390
(void)wc_MakeEccsiKey(genKey, &gRng);
1438914391
(void)wc_MakeEccsiPair(genKey, &gRng, WC_HASH_TYPE_SHA256, id, sizeof(id),
1439014392
ssk, pvt);
@@ -14475,7 +14477,7 @@ void bench_sakkeKeyGen(void)
1447514477
bench_stats_start(&count, &start);
1447614478
do {
1447714479
for (i = 0; i < genTimes; i++) {
14478-
wc_InitSakkeKey_ex(genKey, 128, ECC_SAKKE_1, NULL, INVALID_DEVID);
14480+
wc_InitSakkeKey_ex(genKey, 128, ECC_SAKKE_1, HEAP_HINT, devId);
1447914481
ret = wc_MakeSakkeKey(genKey, &gRng);
1448014482
if (ret != 0) {
1448114483
printf("wc_MakeSakkeKey failed: %d\n", ret);
@@ -14517,7 +14519,7 @@ void bench_sakkeRskGen(void)
1451714519
WC_ALLOC_VAR(genKey, SakkeKey, 1, HEAP_HINT);
1451814520

1451914521
rsk = wc_ecc_new_point();
14520-
wc_InitSakkeKey_ex(genKey, 128, ECC_SAKKE_1, NULL, INVALID_DEVID);
14522+
wc_InitSakkeKey_ex(genKey, 128, ECC_SAKKE_1, HEAP_HINT, devId);
1452114523
(void)wc_MakeSakkeKey(genKey, &gRng);
1452214524

1452314525
/* RSK Gen */
@@ -14570,7 +14572,7 @@ void bench_sakkeValidate(void)
1457014572
WC_ALLOC_VAR(genKey, SakkeKey, 1, HEAP_HINT);
1457114573

1457214574
rsk = wc_ecc_new_point();
14573-
(void)wc_InitSakkeKey_ex(genKey, 128, ECC_SAKKE_1, NULL, INVALID_DEVID);
14575+
(void)wc_InitSakkeKey_ex(genKey, 128, ECC_SAKKE_1, HEAP_HINT, devId);
1457414576
(void)wc_MakeSakkeKey(genKey, &gRng);
1457514577
(void)wc_MakeSakkeRsk(genKey, id, sizeof(id), rsk);
1457614578
(void)wc_ValidateSakkeRsk(genKey, id, sizeof(id), rsk, &valid);
@@ -14634,7 +14636,7 @@ void bench_sakke(void)
1463414636
XMEMCPY(ssv, ssv_init, sizeof ssv);
1463514637

1463614638
rsk = wc_ecc_new_point();
14637-
(void)wc_InitSakkeKey_ex(genKey, 128, ECC_SAKKE_1, NULL, INVALID_DEVID);
14639+
(void)wc_InitSakkeKey_ex(genKey, 128, ECC_SAKKE_1, HEAP_HINT, devId);
1463814640
(void)wc_MakeSakkeKey(genKey, &gRng);
1463914641
(void)wc_MakeSakkeRsk(genKey, id, sizeof(id), rsk);
1464014642
(void)wc_SetSakkeRsk(genKey, rsk, NULL, 0);
@@ -14861,9 +14863,9 @@ void bench_falconKeySign(byte level)
1486114863

1486214864
bench_stats_prepare();
1486314865

14864-
ret = wc_falcon_init(&key);
14866+
ret = wc_falcon_init_ex(&key, HEAP_HINT, devId);
1486514867
if (ret != 0) {
14866-
printf("wc_falcon_init failed %d\n", ret);
14868+
printf("wc_falcon_init_ex failed %d\n", ret);
1486714869
return;
1486814870
}
1486914871

0 commit comments

Comments
 (0)