Merge pull request #3197 from ethanlooney/19th_branch

Added asn.c unit tests
This commit is contained in:
Chris Conlon 2020-08-07 09:25:50 -06:00 committed by GitHub
commit b03e1dd2a9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -20595,7 +20595,598 @@ static int test_wc_ecc_is_valid_idx (void)
} /* END test_wc_ecc_is_valid_idx */
/*
* Testing ToTraditional
*/
static int test_ToTraditional (void)
{
int ret = 0;
#if defined(WOLFSSL_TEST_CERT) || defined(OPENSSL_EXTRA) || \
defined(OPENSSL_EXTRA_X509_SMALL)
XFILE f;
byte input[TWOK_BUF];
word32 sz;
printf(testingFmt, "ToTraditional()");
f = XFOPEN("./certs/server-keyPkcs8.der", "rb");
AssertTrue((f != XBADFILE));
sz = (word32)XFREAD(input, 1, sizeof(input), f);
XFCLOSE(f);
/* Good case */
ret = ToTraditional(input, sz);
if (ret > 0) {
ret = 0;
}
/* Bad cases */
if (ret == 0) {
ret = ToTraditional(NULL, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = ToTraditional(NULL, sz);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = ToTraditional(input, 0);
if (ret == ASN_PARSE_E) {
ret = 0;
}
}
printf(resultFmt, ret == 0 ? passed : failed);
#endif
return ret;
}/* End test_ToTraditional*/
/*
* Testing wc_Ed25519KeyToDer
*/
static int test_wc_EccPrivateKeyToDer (void)
{
int ret = 0;
#if defined(HAVE_ECC) && !defined(WC_NO_RNG)
byte output[ONEK_BUF];
ecc_key eccKey;
WC_RNG rng;
word32 inLen;
printf(testingFmt, "wc_EccPrivateKeyToDer()");
ret = wc_InitRng(&rng);
if (ret == 0) {
ret = wc_ecc_init(&eccKey);
if (ret == 0) {
ret = wc_ecc_make_key(&rng, KEY14, &eccKey);
}
inLen = (word32)sizeof(output);
/* Bad Cases */
if (ret == 0) {
ret = wc_EccPrivateKeyToDer(NULL, NULL, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_EccPrivateKeyToDer(NULL, output, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_EccPrivateKeyToDer(&eccKey, NULL, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_EccPrivateKeyToDer(&eccKey, output, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
/*Good Case */
if (ret == 0) {
ret = wc_EccPrivateKeyToDer(&eccKey, output, inLen);
if (ret > 0) {
ret = 0;
}
}
wc_ecc_free(&eccKey);
}
wc_FreeRng(&rng);
printf(resultFmt, ret == 0 ? passed : failed);
#endif
return ret;
}/* End test_wc_EccPrivateKeyToDer*/
/*
* Testing wc_Ed25519KeyToDer
*/
static int test_wc_Ed25519KeyToDer (void)
{
int ret = 0;
#if defined(HAVE_ED25519) && (defined(WOLFSSL_CERT_GEN) || \
defined(WOLFSSL_KEY_GEN))
byte output[ONEK_BUF];
ed25519_key ed25519Key;
WC_RNG rng;
word32 inLen;
printf(testingFmt, "wc_Ed25519KeyToDer()");
ret = wc_InitRng(&rng);
if (ret == 0) {
ret = wc_ed25519_init(&ed25519Key);
if (ret == 0) {
ret = wc_ed25519_make_key(&rng, ED25519_KEY_SIZE, &ed25519Key);
}
inLen = (word32)sizeof(output);
/* Bad Cases */
if (ret == 0) {
ret = wc_Ed25519KeyToDer(NULL, NULL, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed25519KeyToDer(NULL, output, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed25519KeyToDer(&ed25519Key, NULL, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed25519KeyToDer(&ed25519Key, output, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
/* Good Case */
if (ret == 0) {
ret = wc_Ed25519KeyToDer(&ed25519Key, output, inLen);
if (ret > 0) {
ret = 0;
}
}
wc_ed25519_free(&ed25519Key);
}
wc_FreeRng(&rng);
printf(resultFmt, ret == 0 ? passed : failed);
#endif
return ret;
}/* End test_wc_Ed25519KeyToDer*/
/*
* Testing wc_Ed25519PrivateKeyToDer
*/
static int test_wc_Ed25519PrivateKeyToDer (void)
{
int ret = 0;
#if defined(HAVE_ED25519) && (defined(WOLFSSL_CERT_GEN) || \
defined(WOLFSSL_KEY_GEN))
byte output[ONEK_BUF];
ed25519_key ed25519PrivKey;
WC_RNG rng;
word32 inLen;
printf(testingFmt, "wc_Ed25519PrivateKeyToDer()");
ret = wc_InitRng(&rng);
if (ret == 0) {
ret = wc_ed25519_init(&ed25519PrivKey);
if (ret == 0) {
ret = wc_ed25519_make_key(&rng, ED25519_KEY_SIZE, &ed25519PrivKey);
}
inLen = (word32)sizeof(output);
/* Bad Cases */
if (ret == 0) {
ret = wc_Ed25519PrivateKeyToDer(NULL, NULL, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed25519PrivateKeyToDer(NULL, output, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed25519PrivateKeyToDer(&ed25519PrivKey, NULL, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed25519PrivateKeyToDer(&ed25519PrivKey, output, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
/* Good Case */
if (ret == 0) {
ret = wc_Ed25519PrivateKeyToDer(&ed25519PrivKey, output, inLen);
if (ret > 0) {
ret = 0;
}
}
wc_ed25519_free(&ed25519PrivKey);
}
wc_FreeRng(&rng);
printf(resultFmt, ret == 0 ? passed : failed);
#endif
return ret;
}/* End test_wc_Ed25519PrivateKeyToDer*/
/*
* Testing wc_Ed448KeyToDer
*/
static int test_wc_Ed448KeyToDer (void)
{
int ret = 0;
#if defined(HAVE_ED448) && (defined(WOLFSSL_CERT_GEN) || \
defined(WOLFSSL_KEY_GEN))
byte output[ONEK_BUF];
ed448_key ed448Key;
WC_RNG rng;
word32 inLen;
printf(testingFmt, "wc_Ed448KeyToDer()");
ret = wc_InitRng(&rng);
if (ret == 0) {
ret = wc_ed448_init(&ed448Key);
if (ret == 0) {
ret = wc_ed448_make_key(&rng, ED448_KEY_SIZE, &ed448Key);
}
inLen = sizeof(output);
/* Bad Cases */
if (ret == 0) {
ret = wc_Ed448KeyToDer(NULL, NULL, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed448KeyToDer(NULL, output, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed448KeyToDer(&ed448Key, NULL, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed448KeyToDer(&ed448Key, output, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
/* Good Case */
if (ret == 0) {
ret = wc_Ed448KeyToDer(&ed448Key, output, inLen);
if (ret > 0) {
ret = 0;
}
}
wc_ed448_free(&ed448Key);
}
wc_FreeRng(&rng);
printf(resultFmt, ret == 0 ? passed : failed);
#endif
return ret;
}/* End test_wc_Ed448KeyToDer*/
/*
* Testing wc_Ed448PrivateKeyToDer
*/
static int test_wc_Ed448PrivateKeyToDer (void)
{
int ret = 0;
#if defined(HAVE_ED448) && (defined(WOLFSSL_CERT_GEN) || \
defined(WOLFSSL_KEY_GEN))
byte output[ONEK_BUF];
ed448_key ed448PrivKey;
WC_RNG rng;
word32 inLen;
printf(testingFmt, "wc_Ed448PrivateKeyToDer()");
ret = wc_InitRng(&rng);
if (ret == 0) {
ret = wc_ed448_init(&ed448PrivKey);
if (ret == 0) {
ret = wc_ed448_make_key(&rng, ED448_KEY_SIZE, &ed448PrivKey);
}
inLen = sizeof(output);
/* Bad Cases */
if (ret == 0) {
ret = wc_Ed448PrivateKeyToDer(NULL, NULL, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed448PrivateKeyToDer(NULL, output, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed448PrivateKeyToDer(&ed448PrivKey, NULL, inLen);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
if (ret == 0) {
ret = wc_Ed448PrivateKeyToDer(&ed448PrivKey, output, 0);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
/* Good case */
if (ret == 0) {
ret = wc_Ed448PrivateKeyToDer(&ed448PrivKey, output, inLen);
if (ret > 0) {
ret = 0;
}
}
wc_ed448_free(&ed448PrivKey);
}
wc_FreeRng(&rng);
printf(resultFmt, ret == 0 ? passed : failed);
#endif
return ret;
}/* End test_wc_Ed448PrivateKeyToDer*/
/*
* Testing wc_SetSubjectBuffer
*/
static int test_wc_SetSubjectBuffer (void)
{
int ret = 0;
#if defined(WOLFSSL_CERT_GEN)
Cert cert;
FILE* file;
byte* der;
word32 derSz;
printf(testingFmt, "wc_SetSubjectBuffer()");
derSz = FOURK_BUF;
der = XMALLOC(FOURK_BUF, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER);
if (der == NULL) {
ret = -1;
}
if (ret == 0) {
file = XFOPEN("./certs/ca-cert.der", "rb");
if (file != NULL) {
derSz = (word32)XFREAD(der, 1, FOURK_BUF, file);
XFCLOSE(file);
}
}
if (ret == 0) {
ret = wc_InitCert(&cert);
}
if (ret == 0) {
ret = wc_SetSubjectBuffer(&cert, der, derSz);
}
if (ret == 0) {
ret = wc_SetSubjectBuffer(NULL, der, derSz);
if (ret == BAD_FUNC_ARG) {
ret = 0;
}
}
XFREE(der, HEAP_HINT, DYNAMIC_TYPE_TMP_BUFFER);
printf(resultFmt, ret == 0 ? passed : failed);
#endif
return ret;
}/* End test_wc_SetSubjectBuffer*/
/*
* Testing wc_SetSubjectKeyIdFromPublicKey_ex
*/
static int test_wc_SetSubjectKeyIdFromPublicKey_ex (void)
{
int ret = 0;
#if defined(WOLFSSL_CERT_EXT) && defined(WOLFSSL_CERT_GEN)
WC_RNG rng;
Cert cert;
#if defined(HAVE_ED25519)
ed25519_key ed25519Key;
#endif
#if !defined(NO_RSA) && defined(HAVE_RSA)
RsaKey rsaKey;
int bits = 2048;
#endif
ecc_key eccKey;
#if defined(HAVE_ED448)
ed448_key ed448Key;
#endif
printf(testingFmt, "wc_SetSubjectKeyIdFromPublicKey_ex()");
#ifndef HAVE_FIPS
ret = wc_InitRng_ex(&rng, HEAP_HINT, devId);
#else
ret = wc_InitRng(&rng);
#endif
wc_InitCert(&cert);
#if defined(HAVE_ED25519)
if (ret == 0) { /*ED25519*/
ret = wc_ed25519_init(&ed25519Key);
if (ret == 0) {
wc_ed25519_make_key(&rng, ED25519_KEY_SIZE, &ed25519Key);
}
if (ret == 0) {
ret = wc_SetSubjectKeyIdFromPublicKey_ex(&cert, ED25519_TYPE,
&ed25519Key);
}
wc_ed25519_free(&ed25519Key);
}
#endif
#if !defined(NO_RSA) && defined(HAVE_RSA) && defined(WOLFSSL_KEY_GEN)
if (ret == 0) { /*RSA*/
ret = wc_InitRsaKey(&rsaKey, NULL);
if (ret == 0) {
MAKE_RSA_KEY(&rsaKey, bits, WC_RSA_EXPONENT, &rng);
}
if (ret == 0) {
ret = wc_SetSubjectKeyIdFromPublicKey_ex(&cert, RSA_TYPE, &rsaKey);
}
wc_FreeRsaKey(&rsaKey);
}
#endif
if (ret == 0) { /*ECC*/
ret = wc_ecc_init(&eccKey);
if (ret == 0) {
wc_ecc_make_key(&rng, KEY14, &eccKey);
}
if (ret == 0) {
ret = wc_SetSubjectKeyIdFromPublicKey_ex(&cert, ECC_TYPE, &eccKey);
}
wc_ecc_free(&eccKey);
}
#if defined(HAVE_ED448) && (defined(WOLFSSL_CERT_GEN) || \
defined(WOLFSSL_KEY_GEN))
if (ret == 0) { /*ED448*/
ret = wc_ed448_init(&ed448Key);
if (ret == 0) {
wc_ed448_make_key(&rng, ED448_KEY_SIZE, &ed448Key);
}
if (ret == 0) {
ret = wc_SetSubjectKeyIdFromPublicKey_ex(&cert, ED448_TYPE,
&ed448Key);
}
wc_ed448_free(&ed448Key);
}
#endif
printf(resultFmt, ret == 0 ? passed : failed);
wc_FreeRng(&rng);
#endif
return ret;
}/* End test_wc_SetSubjectKeyIdFromPublicKey_ex*/
/*
* Testing wc_SetAuthKeyIdFromPublicKey_ex
*/
static int test_wc_SetAuthKeyIdFromPublicKey_ex (void)
{
int ret = 0;
#if defined(WOLFSSL_CERT_EXT) && defined(WOLFSSL_CERT_GEN)
WC_RNG rng;
Cert cert;
#if defined(HAVE_ED25519)
ed25519_key ed25519Key;
#endif
#if !defined(NO_RSA) && defined(HAVE_RSA)
RsaKey rsaKey;
int bits = 2048;
#endif
ecc_key eccKey;
#if defined(HAVE_ED448)
ed448_key ed448Key;
#endif
printf(testingFmt, "wc_SetAuthKeyIdFromPublicKey_ex()");
#ifndef HAVE_FIPS
ret = wc_InitRng_ex(&rng, HEAP_HINT, devId);
#else
ret = wc_InitRng(&rng);
#endif
wc_InitCert(&cert);
#if defined(HAVE_ED25519)
if (ret == 0) { /*ED25519*/
ret = wc_ed25519_init(&ed25519Key);
if (ret == 0) {
wc_ed25519_make_key(&rng, ED25519_KEY_SIZE, &ed25519Key);
}
if (ret == 0) {
ret = wc_SetAuthKeyIdFromPublicKey_ex(&cert, ED25519_TYPE,
&ed25519Key);
}
wc_ed25519_free(&ed25519Key);
}
#endif
#if !defined(NO_RSA) && defined(HAVE_RSA) && defined(WOLFSSL_KEY_GEN)
if (ret == 0) { /*RSA*/
ret = wc_InitRsaKey(&rsaKey, NULL);
if (ret == 0) {
MAKE_RSA_KEY(&rsaKey, bits, WC_RSA_EXPONENT, &rng);
}
if (ret == 0) {
ret = wc_SetAuthKeyIdFromPublicKey_ex(&cert, RSA_TYPE, &rsaKey);
}
wc_FreeRsaKey(&rsaKey);
}
#endif
if (ret == 0) { /*ECC*/
ret = wc_ecc_init(&eccKey);
if (ret == 0) {
wc_ecc_make_key(&rng, KEY14, &eccKey);
}
if (ret == 0) {
ret = wc_SetAuthKeyIdFromPublicKey_ex(&cert, ECC_TYPE, &eccKey);
}
wc_ecc_free(&eccKey);
}
#if defined(HAVE_ED448) && (defined(WOLFSSL_CERT_GEN) || \
defined(WOLFSSL_KEY_GEN))
if (ret == 0) { /*ED448*/
ret = wc_ed448_init(&ed448Key);
if (ret == 0) {
wc_ed448_make_key(&rng, ED448_KEY_SIZE, &ed448Key);
}
if (ret == 0) {
ret = wc_SetAuthKeyIdFromPublicKey_ex(&cert, ED448_TYPE,
&ed448Key);
}
wc_ed448_free(&ed448Key);
}
#endif
printf(resultFmt, ret == 0 ? passed : failed);
wc_FreeRng(&rng);
#endif /*defined(WOLFSSL_CERT_EXT) && defined(WOLFSSL_CERT_GEN)*/
return ret;
}/* End test_wc_SetAuthKeyIdFromPublicKey_ex*/
/*
* Testing wc_PKCS7_New()
*/
@ -36049,6 +36640,17 @@ void ApiTest(void)
AssertIntEQ(test_wc_ecc_mulmod(), 0);
AssertIntEQ(test_wc_ecc_is_valid_idx(), 0);
AssertIntEQ(test_ToTraditional(), 0);
AssertIntEQ(test_wc_EccPrivateKeyToDer(), 0);
AssertIntEQ(test_wc_Ed25519KeyToDer(), 0);
AssertIntEQ(test_wc_Ed25519PrivateKeyToDer(), 0);
AssertIntEQ(test_wc_Ed448KeyToDer(), 0);
AssertIntEQ(test_wc_Ed448PrivateKeyToDer(), 0);
AssertIntEQ(test_wc_SetAuthKeyIdFromPublicKey_ex(), 0);
AssertIntEQ(test_wc_SetSubjectBuffer(), 0);
AssertIntEQ(test_wc_SetSubjectKeyIdFromPublicKey_ex(), 0);
test_wc_PKCS7_New();
test_wc_PKCS7_Init();
test_wc_PKCS7_InitWithCert();