148 lines
4.2 KiB
C
148 lines
4.2 KiB
C
/*
|
|
==============================================================================
|
|
Name : CCMtest.c
|
|
Author : polfosol
|
|
Version : 1.6.1.2
|
|
Copyright : copyright © 2022 - polfosol
|
|
Description : illustrating how to validate NIST's vectors for AES-CCM mode
|
|
==============================================================================
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include "../micro_aes.h"
|
|
|
|
#define TESTFILEPATH "CCM_VNT128.rsp"
|
|
|
|
static void str2bytes(const char* hex, uint8_t* bytes)
|
|
{
|
|
unsigned shl = 0;
|
|
for (--bytes; *hex; ++hex)
|
|
{
|
|
if (*hex < '0' || 'f' < *hex) continue;
|
|
if ((shl ^= 4) != 0) *++bytes = 0;
|
|
*bytes |= (*hex % 16 + (*hex > '9') * 9) << shl;
|
|
}
|
|
}
|
|
|
|
static void bytes2str(const uint8_t* bytes, char* str, const size_t len)
|
|
{
|
|
const char offset = 0x27; /* offset must be 7 for uppercase */
|
|
size_t i = len + len, shr = 0;
|
|
for (str[i] = 0; i--; shr ^= 4)
|
|
{
|
|
str[i] = bytes[i / 2] >> shr & 0xF | '0';
|
|
if (str[i] > '9') str[i] += offset;
|
|
}
|
|
}
|
|
|
|
static int ciphertest(uint8_t* key, uint8_t* iv, uint8_t* p, uint8_t* a, uint8_t* c,
|
|
size_t np, size_t na, char* r)
|
|
{
|
|
char sk[2*AES_KEY_SIZE + 1], si[33], sp[80], sc[96], sa[80], msg[30];
|
|
uint8_t tmp[64], t = 0;
|
|
sprintf(msg, "%s", "passed the test");
|
|
|
|
AES_CCM_encrypt(key, iv, p, np, a, na, tmp, tmp + np);
|
|
if (memcmp(c, tmp, np + CCM_TAG_LEN))
|
|
{
|
|
sprintf(msg, "%s", "encrypt failure");
|
|
t = 1;
|
|
}
|
|
memset(tmp, 0xcc , sizeof tmp);
|
|
t |= AES_CCM_decrypt(key, iv, c, np, a, na, CCM_TAG_LEN, tmp) ? 2 : 0;
|
|
if (t > 1)
|
|
{
|
|
sprintf(msg, "%sdecrypt failure", t & 1 ? "encrypt & " : "");
|
|
}
|
|
bytes2str(key, sk, AES_KEY_SIZE);
|
|
bytes2str(iv, si, CCM_NONCE_LEN);
|
|
bytes2str(p, sp, np);
|
|
bytes2str(a, sa, na);
|
|
bytes2str(c, sc, np + CCM_TAG_LEN);
|
|
sprintf(r, "%s\nK: %s\ni: %s\nP: %s\nA: %s\nC: %s", msg, sk, si, sp, sa, sc);
|
|
return t;
|
|
}
|
|
|
|
int main()
|
|
{
|
|
const char *linehdr[] = { "Key = ", "Nonce = ", "Adata = ", "Payload = ", "CT = " };
|
|
char buffer[0x800], *value = "";
|
|
size_t pass = 0, df = 0, ef = 0, sk = 0, sn = 0, sp = 0, sc = 0, sa = 0;
|
|
uint8_t i, n = 0, key[AES_KEY_SIZE], iv[16], p[64], c[80], a[64];
|
|
FILE *fp, *fs, *ferr;
|
|
|
|
fp = fopen(TESTFILEPATH, "r");
|
|
fs = fopen("passed.log", "w");
|
|
ferr = fopen("failed.log", "w");
|
|
|
|
if (fp == NULL)
|
|
{
|
|
printf("File not found: %s\n", TESTFILEPATH);
|
|
return 1;
|
|
}
|
|
if (!fs || !ferr) return 1;
|
|
|
|
while (fgets(buffer, sizeof buffer, fp) != NULL)
|
|
{
|
|
buffer[strcspn(buffer, "\n")] = 0;
|
|
if (strlen(buffer) < 4) continue;
|
|
for (i = 0; i < 5; i++)
|
|
{
|
|
if (strncmp(buffer, linehdr[i], strlen(linehdr[i])) == 0)
|
|
{
|
|
value = strrchr(buffer, ' ') + 1;
|
|
break;
|
|
}
|
|
}
|
|
switch (i)
|
|
{
|
|
case 0:
|
|
sk = strlen(value) / 2;
|
|
if (sk == AES_KEY_SIZE) str2bytes(value, key);
|
|
break;
|
|
case 1:
|
|
sn = strlen(value) / 2;
|
|
if (sn == CCM_NONCE_LEN) str2bytes(value, iv);
|
|
break;
|
|
case 2:
|
|
sa = strlen(value) / 2;
|
|
str2bytes(value, a);
|
|
break;
|
|
case 3:
|
|
sp = strlen(value) / 2;
|
|
str2bytes(value, p);
|
|
++n;
|
|
break;
|
|
case 4:
|
|
sc = strlen(value) / 2 - CCM_TAG_LEN;
|
|
str2bytes(value, c);
|
|
++n;
|
|
break;
|
|
}
|
|
if (n == 2)
|
|
{
|
|
if (sk == AES_KEY_SIZE && sn == CCM_NONCE_LEN && sp == sc)
|
|
{
|
|
n = ciphertest(key, iv, p, a, c, sp, sa, buffer);
|
|
fprintf(n ? ferr : fs, "%s\n", buffer); /* save the log */
|
|
if (n == 0) ++pass;
|
|
else
|
|
{
|
|
if (n & 1) ++ef;
|
|
if (n & 2) ++df;
|
|
}
|
|
}
|
|
n = 0;
|
|
}
|
|
}
|
|
printf ("test cases: %d\nsuccessful: %d\nfailed encrypt: %d, failed decrypt: %d\n",
|
|
pass + (ef > df ? ef : df), pass, ef, df);
|
|
|
|
fclose(fp); fclose(fs); fclose(ferr);
|
|
if (ef + df == 0)
|
|
{
|
|
remove("passed.log"); remove("failed.log");
|
|
}
|
|
return 0;
|
|
}
|