denozord/strongswan-ext
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

unit-tests: Use TEST_FUNCTION macro in ntru tests

Browse Source
This commit is contained in:
Tobias Brunner
2014-03-20 15:49:05 +01:00
parent b751f6f25a
commit 6ce5aee0b2
1 changed files with 44 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/libstrongswan/tests/suites/test_ntru.c
+44 -34
View File
@@ -94,7 +94,8 @@ START_TEST(test_ntru_drbg_strength)
entropy = lib->crypto->create_rng(lib->crypto, RNG_STRONG); entropy = lib->crypto->create_rng(lib->crypto, RNG_STRONG);
ck_assert(entropy != NULL); ck_assert(entropy != NULL);
drbg = TEST_ntru_drbg_create(strengths[_i].requested, chunk_empty, entropy); drbg = TEST_FUNCTION(ntru, ntru_drbg_create, strengths[_i].requested,
chunk_empty, entropy);
if (strengths[_i].standard) if (strengths[_i].standard)
{ {
ck_assert(drbg != NULL); ck_assert(drbg != NULL);
@@ -251,7 +252,8 @@ START_TEST(test_ntru_drbg)
out = chunk_alloc(128); out = chunk_alloc(128);
entropy = test_rng_create(drbg_tests[_i].entropy); entropy = test_rng_create(drbg_tests[_i].entropy);
drbg = TEST_ntru_drbg_create(256, drbg_tests[_i].pers_str, entropy); drbg = TEST_FUNCTION(ntru, ntru_drbg_create, 256, drbg_tests[_i].pers_str,
entropy);
ck_assert(drbg != NULL); ck_assert(drbg != NULL);
ck_assert(drbg->reseed(drbg)); ck_assert(drbg->reseed(drbg));
ck_assert(drbg->generate(drbg, 256, 128, out.ptr)); ck_assert(drbg->generate(drbg, 256, 128, out.ptr));
@@ -273,7 +275,7 @@ START_TEST(test_ntru_drbg_reseed)
"libstrongswan.plugins.ntru.max_drbg_requests", 2); "libstrongswan.plugins.ntru.max_drbg_requests", 2);
out = chunk_alloc(128); out = chunk_alloc(128);
entropy = test_rng_create(drbg_tests[0].entropy); entropy = test_rng_create(drbg_tests[0].entropy);
drbg = TEST_ntru_drbg_create(256, chunk_empty, entropy); drbg = TEST_FUNCTION(ntru, ntru_drbg_create, 256, chunk_empty, entropy);
/* bad output parameters */ /* bad output parameters */
ck_assert(!drbg->generate(drbg, 256, 0, out.ptr)); ck_assert(!drbg->generate(drbg, 256, 0, out.ptr));
@@ -291,13 +293,13 @@ START_TEST(test_ntru_drbg_reseed)
drbg->destroy(drbg); drbg->destroy(drbg);
/* no entropy available for DRBG instantiation */ /* no entropy available for DRBG instantiation */
drbg = TEST_ntru_drbg_create(256, chunk_empty, entropy); drbg = TEST_FUNCTION(ntru, ntru_drbg_create, 256, chunk_empty, entropy);
ck_assert(drbg == NULL); ck_assert(drbg == NULL);
entropy->destroy(entropy); entropy->destroy(entropy);
/* one automatic reseeding occurs */ /* one automatic reseeding occurs */
entropy = test_rng_create(drbg_tests[0].entropy); entropy = test_rng_create(drbg_tests[0].entropy);
drbg = TEST_ntru_drbg_create(256, chunk_empty, entropy); drbg = TEST_FUNCTION(ntru, ntru_drbg_create, 256, chunk_empty, entropy);
ck_assert(drbg->generate(drbg, 256, 128, out.ptr)); ck_assert(drbg->generate(drbg, 256, 128, out.ptr));
ck_assert(drbg->generate(drbg, 256, 128, out.ptr)); ck_assert(drbg->generate(drbg, 256, 128, out.ptr));
ck_assert(drbg->generate(drbg, 256, 128, out.ptr)); ck_assert(drbg->generate(drbg, 256, 128, out.ptr));
@@ -550,14 +552,17 @@ START_TEST(test_ntru_mgf1)
mask2.len = mgf1_tests[_i].ml2; mask2.len = mgf1_tests[_i].ml2;
mask3.len = mgf1_tests[_i].ml3; mask3.len = mgf1_tests[_i].ml3;
mgf1 = TEST_ntru_mgf1_create(HASH_UNKNOWN, mgf1_tests[_i].seed, TRUE); mgf1 = TEST_FUNCTION(ntru, ntru_mgf1_create, HASH_UNKNOWN,
mgf1_tests[_i].seed, TRUE);
ck_assert(mgf1 == NULL); ck_assert(mgf1 == NULL);
mgf1 = TEST_ntru_mgf1_create(mgf1_tests[_i].alg, chunk_empty, TRUE); mgf1 = TEST_FUNCTION(ntru, ntru_mgf1_create, mgf1_tests[_i].alg,
chunk_empty, TRUE);
ck_assert(mgf1 == NULL); ck_assert(mgf1 == NULL);
/* return mask in allocated chunk */ /* return mask in allocated chunk */
mgf1 = TEST_ntru_mgf1_create(mgf1_tests[_i].alg, mgf1_tests[_i].seed, TRUE); mgf1 = TEST_FUNCTION(ntru, ntru_mgf1_create, mgf1_tests[_i].alg,
mgf1_tests[_i].seed, TRUE);
ck_assert(mgf1); ck_assert(mgf1);
/* check hash size */ /* check hash size */
@@ -573,15 +578,16 @@ START_TEST(test_ntru_mgf1)
mgf1->destroy(mgf1); mgf1->destroy(mgf1);
/* copy mask to pre-allocated buffer */ /* copy mask to pre-allocated buffer */
mgf1 = TEST_ntru_mgf1_create(mgf1_tests[_i].alg, mgf1_tests[_i].seed, TRUE); mgf1 = TEST_FUNCTION(ntru, ntru_mgf1_create, mgf1_tests[_i].alg,
mgf1_tests[_i].seed, TRUE);
ck_assert(mgf1); ck_assert(mgf1);
ck_assert(mgf1->get_mask(mgf1, mgf1_tests[_i].mask.len, mask.ptr)); ck_assert(mgf1->get_mask(mgf1, mgf1_tests[_i].mask.len, mask.ptr));
ck_assert(chunk_equals(mask, mgf1_tests[_i].mask)); ck_assert(chunk_equals(mask, mgf1_tests[_i].mask));
mgf1->destroy(mgf1); mgf1->destroy(mgf1);
/* get mask in batches without hashing the seed */ /* get mask in batches without hashing the seed */
mgf1 = TEST_ntru_mgf1_create(mgf1_tests[_i].alg, mgf1_tests[_i].hashed_seed, mgf1 = TEST_FUNCTION(ntru, ntru_mgf1_create, mgf1_tests[_i].alg,
FALSE); mgf1_tests[_i].hashed_seed, FALSE);
ck_assert(mgf1); ck_assert(mgf1);
/* first batch */ /* first batch */
@@ -609,16 +615,16 @@ START_TEST(test_ntru_trits)
ntru_trits_t *mask; ntru_trits_t *mask;
chunk_t trits; chunk_t trits;
mask = TEST_ntru_trits_create(mgf1_tests[_i].trits.len, HASH_UNKNOWN, mask = TEST_FUNCTION(ntru, ntru_trits_create, mgf1_tests[_i].trits.len,
mgf1_tests[_i].seed); HASH_UNKNOWN, mgf1_tests[_i].seed);
ck_assert(mask == NULL); ck_assert(mask == NULL);
mask = TEST_ntru_trits_create(mgf1_tests[_i].trits.len, mgf1_tests[_i].alg, mask = TEST_FUNCTION(ntru, ntru_trits_create, mgf1_tests[_i].trits.len,
chunk_empty); mgf1_tests[_i].alg, chunk_empty);
ck_assert(mask == NULL); ck_assert(mask == NULL);
mask = TEST_ntru_trits_create(mgf1_tests[_i].trits.len, mgf1_tests[_i].alg, mask = TEST_FUNCTION(ntru, ntru_trits_create, mgf1_tests[_i].trits.len,
mgf1_tests[_i].seed); mgf1_tests[_i].alg, mgf1_tests[_i].seed);
ck_assert(mask); ck_assert(mask);
trits = chunk_create(mask->get_trits(mask), mask->get_size(mask)); trits = chunk_create(mask->get_trits(mask), mask->get_size(mask));
@@ -626,7 +632,8 @@ START_TEST(test_ntru_trits)
mask->destroy(mask); mask->destroy(mask);
/* generate a multiple of 5 trits */ /* generate a multiple of 5 trits */
mask = TEST_ntru_trits_create(10, mgf1_tests[_i].alg, mgf1_tests[_i].seed); mask = TEST_FUNCTION(ntru, ntru_trits_create, 10, mgf1_tests[_i].alg,
mgf1_tests[_i].seed);
ck_assert(mask); ck_assert(mask);
trits = chunk_create(mask->get_trits(mask), mask->get_size(mask)); trits = chunk_create(mask->get_trits(mask), mask->get_size(mask));
@@ -647,17 +654,17 @@ START_TEST(test_ntru_poly)
seed.len = mgf1_tests[_i].seed_len; seed.len = mgf1_tests[_i].seed_len;
p = &mgf1_tests[_i].poly_test[0]; p = &mgf1_tests[_i].poly_test[0];
poly = TEST_ntru_poly_create_from_seed(HASH_UNKNOWN, seed, p->c_bits, p->N, poly = TEST_FUNCTION(ntru, ntru_poly_create_from_seed, HASH_UNKNOWN, seed,
p->q, p->indices_len, p->indices_len, p->c_bits, p->N, p->q, p->indices_len, p->indices_len,
p->is_product_form); p->is_product_form);
ck_assert(poly == NULL); ck_assert(poly == NULL);
for (n = 0; n < 2; n++) for (n = 0; n < 2; n++)
{ {
p = &mgf1_tests[_i].poly_test[n]; p = &mgf1_tests[_i].poly_test[n];
poly = TEST_ntru_poly_create_from_seed(mgf1_tests[_i].alg, seed, poly = TEST_FUNCTION(ntru, ntru_poly_create_from_seed,
p->c_bits, p->N, p->q, p->indices_len, mgf1_tests[_i].alg, seed, p->c_bits, p->N, p->q,
p->indices_len, p->is_product_form); p->indices_len, p->indices_len, p->is_product_form);
ck_assert(poly != NULL && poly->get_size(poly) == p->indices_size); ck_assert(poly != NULL && poly->get_size(poly) == p->indices_size);
indices = poly->get_indices(poly); indices = poly->get_indices(poly);
@@ -757,8 +764,9 @@ START_TEST(test_ntru_ring_mult)
int i; int i;
t = &ring_mult_tests[_i]; t = &ring_mult_tests[_i];
poly = TEST_ntru_poly_create_from_data(t->indices, t->N, t->q, poly = TEST_FUNCTION(ntru, ntru_poly_create_from_data, t->indices, t->N,
t->indices_len_p, t->indices_len_m, t->is_product_form); t->q, t->indices_len_p, t->indices_len_m,
t->is_product_form);
ck_assert(poly != NULL); ck_assert(poly != NULL);
c = malloc(t->N * sizeof(uint16_t)); c = malloc(t->N * sizeof(uint16_t));
@@ -785,8 +793,9 @@ START_TEST(test_ntru_array)
t = &ring_mult_tests[array_tests[_i]]; t = &ring_mult_tests[array_tests[_i]];
poly = TEST_ntru_poly_create_from_data(t->indices, t->N, t->q, poly = TEST_FUNCTION(ntru, ntru_poly_create_from_data, t->indices, t->N,
t->indices_len_p, t->indices_len_m, t->is_product_form); t->q, t->indices_len_p, t->indices_len_m,
t->is_product_form);
ck_assert(poly != NULL); ck_assert(poly != NULL);
c = malloc(t->N * sizeof(uint16_t)); c = malloc(t->N * sizeof(uint16_t));
@@ -804,8 +813,8 @@ END_TEST
START_TEST(test_ntru_param_set) START_TEST(test_ntru_param_set)
{ {
ck_assert(TEST_ntru_param_set_get_by_id(-1) == NULL); ck_assert(TEST_FUNCTION(ntru, ntru_param_set_get_by_id, -1) == NULL);
ck_assert(TEST_ntru_param_set_get_by_id(16) == NULL); ck_assert(TEST_FUNCTION(ntru, ntru_param_set_get_by_id, 16) == NULL);
} }
END_TEST END_TEST
@@ -1067,14 +1076,15 @@ START_TEST(test_ntru_privkey)
uint32_t strength; uint32_t strength;
chunk_t encoding, privkey_encoding, pubkey_encoding; chunk_t encoding, privkey_encoding, pubkey_encoding;
params = TEST_ntru_param_set_get_by_id(privkey_tests[_i].id); params = TEST_FUNCTION(ntru, ntru_param_set_get_by_id,
privkey_tests[_i].id);
strength = params->sec_strength_len * BITS_PER_BYTE; strength = params->sec_strength_len * BITS_PER_BYTE;
entropy = test_rng_create(privkey_tests[_i].entropy); entropy = test_rng_create(privkey_tests[_i].entropy);
drbg = TEST_ntru_drbg_create(strength, chunk_from_str("IKE NTRU-KE"), drbg = TEST_FUNCTION(ntru, ntru_drbg_create, strength,
entropy); chunk_from_str("IKE NTRU-KE"), entropy);
ck_assert(drbg != NULL); ck_assert(drbg != NULL);
privkey = TEST_ntru_private_key_create(drbg, params); privkey = TEST_FUNCTION(ntru, ntru_private_key_create, drbg, params);
ck_assert(privkey); ck_assert(privkey);
privkey_encoding = privkey->get_encoding(privkey); privkey_encoding = privkey->get_encoding(privkey);