wolfSSL/wolfssl
1
0
Fork 1
mirror of https://github.com/wolfSSL/wolfssl.git synced 2025-07-30 18:57:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

test.c:aes_test(): add WOLFSSL_SMALL_STACK codepaths for WOLFSSL_AESNI test.

Browse Source
This commit is contained in:
Daniel Pouzzner
2020-12-08 00:35:41 -06:00
parent 1c0df61247
commit 53c6d33695
1 changed files with 46 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
wolfcrypt/test/test.c
View File

@ -8122,9 +8122,22 @@ static int aes_test(void)
0x65,0x73,0x20,0x4a,0x61,0x63,0x6b,0x20 0x65,0x73,0x20,0x4a,0x61,0x63,0x6b,0x20
}; };
static const byte bigKey[] = "0123456789abcdeffedcba9876543210"; static const byte bigKey[] = "0123456789abcdeffedcba9876543210";
word32 keySz, msgSz;
#ifdef WOLFSSL_SMALL_STACK
byte *bigCipher = (byte *)XMALLOC(sizeof(bigMsg), NULL, DYNAMIC_TYPE_TMP_BUFFER);
byte *bigPlain = (byte *)XMALLOC(sizeof(bigMsg), NULL, DYNAMIC_TYPE_TMP_BUFFER);
if ((! bigCipher) ||
(! bigPlain)) {
if (bigCipher)
XFREE(bigCipher, NULL, DYNAMIC_TYPE_TMP_BUFFER);
return -5947;
}
#else
byte bigCipher[sizeof(bigMsg)]; byte bigCipher[sizeof(bigMsg)];
byte bigPlain[sizeof(bigMsg)]; byte bigPlain[sizeof(bigMsg)];
word32 keySz, msgSz; #endif
/* Iterate from one AES_BLOCK_SIZE of bigMsg through the whole /* Iterate from one AES_BLOCK_SIZE of bigMsg through the whole
* message by AES_BLOCK_SIZE for each size of AES key. */ * message by AES_BLOCK_SIZE for each size of AES key. */
@ -8133,33 +8146,53 @@ static int aes_test(void)
msgSz <= sizeof(bigMsg); msgSz <= sizeof(bigMsg);
msgSz += AES_BLOCK_SIZE) { msgSz += AES_BLOCK_SIZE) {
XMEMSET(bigCipher, 0, sizeof(bigCipher)); XMEMSET(bigCipher, 0, sizeof(bigMsg));
XMEMSET(bigPlain, 0, sizeof(bigPlain)); XMEMSET(bigPlain, 0, sizeof(bigMsg));
ret = wc_AesSetKey(&enc, bigKey, keySz, iv, AES_ENCRYPTION); ret = wc_AesSetKey(&enc, bigKey, keySz, iv, AES_ENCRYPTION);
if (ret != 0) if (ret != 0) {
return -5908; ret = -5908;
break;
}
ret = wc_AesSetKey(&dec, bigKey, keySz, iv, AES_DECRYPTION); ret = wc_AesSetKey(&dec, bigKey, keySz, iv, AES_DECRYPTION);
if (ret != 0) if (ret != 0) {
return -5909; ret = -5909;
break;
}
ret = wc_AesCbcEncrypt(&enc, bigCipher, bigMsg, msgSz); ret = wc_AesCbcEncrypt(&enc, bigCipher, bigMsg, msgSz);
#if defined(WOLFSSL_ASYNC_CRYPT) #if defined(WOLFSSL_ASYNC_CRYPT)
ret = wc_AsyncWait(ret, &enc.asyncDev, WC_ASYNC_FLAG_NONE); ret = wc_AsyncWait(ret, &enc.asyncDev, WC_ASYNC_FLAG_NONE);
#endif #endif
if (ret != 0) if (ret != 0) {
return -5910; ret = -5910;
break;
}
ret = wc_AesCbcDecrypt(&dec, bigPlain, bigCipher, msgSz); ret = wc_AesCbcDecrypt(&dec, bigPlain, bigCipher, msgSz);
#if defined(WOLFSSL_ASYNC_CRYPT) #if defined(WOLFSSL_ASYNC_CRYPT)
ret = wc_AsyncWait(ret, &dec.asyncDev, WC_ASYNC_FLAG_NONE); ret = wc_AsyncWait(ret, &dec.asyncDev, WC_ASYNC_FLAG_NONE);
#endif #endif
if (ret != 0) if (ret != 0) {
return -5911; ret = -5911;
break;
}
if (XMEMCMP(bigPlain, bigMsg, msgSz)) if (XMEMCMP(bigPlain, bigMsg, msgSz)) {
return -5912; ret = -5912;
break;
}
} }
if (ret != 0)
break;
} }
#ifdef WOLFSSL_SMALL_STACK
XFREE(bigCipher, NULL, DYNAMIC_TYPE_TMP_BUFFER);
XFREE(bigPlain, NULL, DYNAMIC_TYPE_TMP_BUFFER);
#endif
if (ret != 0)
return ret;
} }
#endif /* WOLFSSL_AESNI && HAVE_AES_DECRYPT */ #endif /* WOLFSSL_AESNI && HAVE_AES_DECRYPT */