950 lines
20 KiB
C
950 lines
20 KiB
C
/*---------------------------------------------
|
|
NIST AES Algorithm Validation Suite
|
|
Test Program
|
|
|
|
Copyright
|
|
V-ONE Corporation
|
|
20250 Century Blvd, Suite 300
|
|
Germantown, MD 20874
|
|
U.S.A.
|
|
----------------------------------------------*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <assert.h>
|
|
|
|
#include <openssl/aes.h>
|
|
#include <openssl/evp.h>
|
|
#include <openssl/fips.h>
|
|
#include <openssl/err.h>
|
|
|
|
#define AES_BLOCK_SIZE 16
|
|
|
|
#define VERBOSE 0
|
|
|
|
/*-----------------------------------------------*/
|
|
|
|
int AESTest(EVP_CIPHER_CTX *ctx,
|
|
char *amode, int akeysz, unsigned char *aKey,
|
|
unsigned char *iVec,
|
|
int dir, /* 0 = decrypt, 1 = encrypt */
|
|
unsigned char *plaintext, unsigned char *ciphertext, int len)
|
|
{
|
|
const EVP_CIPHER *cipher = NULL;
|
|
int ret = 1;
|
|
int kt = 0;
|
|
|
|
if (ctx)
|
|
memset(ctx, 0, sizeof(EVP_CIPHER_CTX));
|
|
|
|
if (strcasecmp(amode, "CBC") == 0)
|
|
kt = 1000;
|
|
else if (strcasecmp(amode, "ECB") == 0)
|
|
kt = 2000;
|
|
else if (strcasecmp(amode, "CFB128") == 0)
|
|
kt = 3000;
|
|
else if (strncasecmp(amode, "OFB", 3) == 0)
|
|
kt = 4000;
|
|
else if(!strcasecmp(amode,"CFB1"))
|
|
kt=5000;
|
|
else if(!strcasecmp(amode,"CFB8"))
|
|
kt=6000;
|
|
else
|
|
{
|
|
printf("Unknown mode: %s\n", amode);
|
|
exit(1);
|
|
}
|
|
if (ret)
|
|
{
|
|
if ((akeysz != 128) && (akeysz != 192) && (akeysz != 256))
|
|
{
|
|
printf("Invalid key size: %d\n", akeysz);
|
|
ret = 0;
|
|
}
|
|
else
|
|
{
|
|
kt += akeysz;
|
|
switch (kt)
|
|
{
|
|
case 1128: /* CBC 128 */
|
|
cipher = EVP_aes_128_cbc();
|
|
break;
|
|
case 1192: /* CBC 192 */
|
|
cipher = EVP_aes_192_cbc();
|
|
break;
|
|
case 1256: /* CBC 256 */
|
|
cipher = EVP_aes_256_cbc();
|
|
break;
|
|
case 2128: /* ECB 128 */
|
|
cipher = EVP_aes_128_ecb();
|
|
break;
|
|
case 2192: /* ECB 192 */
|
|
cipher = EVP_aes_192_ecb();
|
|
break;
|
|
case 2256: /* ECB 256 */
|
|
cipher = EVP_aes_256_ecb();
|
|
break;
|
|
case 3128: /* CFB 128 */
|
|
cipher = EVP_aes_128_cfb();
|
|
break;
|
|
case 3192: /* CFB 192 */
|
|
cipher = EVP_aes_192_cfb();
|
|
break;
|
|
case 3256: /* CFB 256 */
|
|
cipher = EVP_aes_256_cfb();
|
|
break;
|
|
case 4128: /* OFB 128 */
|
|
cipher = EVP_aes_128_ofb();
|
|
break;
|
|
case 4192: /* OFB 192 */
|
|
cipher = EVP_aes_192_ofb();
|
|
break;
|
|
case 4256: /* OFB 256 */
|
|
cipher = EVP_aes_256_ofb();
|
|
break;
|
|
case 5128:
|
|
cipher=EVP_aes_128_cfb1();
|
|
break;
|
|
case 5192:
|
|
cipher=EVP_aes_192_cfb1();
|
|
break;
|
|
case 5256:
|
|
cipher=EVP_aes_256_cfb1();
|
|
break;
|
|
case 6128:
|
|
cipher=EVP_aes_128_cfb8();
|
|
break;
|
|
case 6192:
|
|
cipher=EVP_aes_192_cfb8();
|
|
break;
|
|
case 6256:
|
|
cipher=EVP_aes_256_cfb8();
|
|
break;
|
|
default:
|
|
printf("Didn't handle mode %d\n",kt);
|
|
exit(1);
|
|
}
|
|
if (dir)
|
|
{ /* encrypt */
|
|
if(!EVP_CipherInit(ctx, cipher, aKey, iVec, AES_ENCRYPT))
|
|
{
|
|
ERR_print_errors_fp(stderr);
|
|
exit(1);
|
|
}
|
|
|
|
EVP_Cipher(ctx, ciphertext, (unsigned char*)plaintext, len);
|
|
}
|
|
else
|
|
{ /* decrypt */
|
|
if(!EVP_CipherInit(ctx, cipher, aKey, iVec, AES_DECRYPT))
|
|
{
|
|
ERR_print_errors_fp(stderr);
|
|
exit(1);
|
|
}
|
|
EVP_Cipher(ctx, (unsigned char*)plaintext, ciphertext, len);
|
|
}
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/*-----------------------------------------------*/
|
|
|
|
int hex2bin(char *in, int len, unsigned char *out)
|
|
{
|
|
int n1, n2;
|
|
unsigned char ch;
|
|
|
|
for (n1 = 0, n2 = 0; n1 < len; )
|
|
{ /* first byte */
|
|
if ((in[n1] >= '0') && (in[n1] <= '9'))
|
|
ch = in[n1++] - '0';
|
|
else if ((in[n1] >= 'A') && (in[n1] <= 'F'))
|
|
ch = in[n1++] - 'A' + 10;
|
|
else if ((in[n1] >= 'a') && (in[n1] <= 'f'))
|
|
ch = in[n1++] - 'a' + 10;
|
|
else
|
|
return -1;
|
|
if(len == 1)
|
|
{
|
|
out[n2++]=ch;
|
|
break;
|
|
}
|
|
out[n2] = ch << 4;
|
|
/* second byte */
|
|
if ((in[n1] >= '0') && (in[n1] <= '9'))
|
|
ch = in[n1++] - '0';
|
|
else if ((in[n1] >= 'A') && (in[n1] <= 'F'))
|
|
ch = in[n1++] - 'A' + 10;
|
|
else if ((in[n1] >= 'a') && (in[n1] <= 'f'))
|
|
ch = in[n1++] - 'a' + 10;
|
|
else
|
|
return -1;
|
|
out[n2++] |= ch;
|
|
}
|
|
return n2;
|
|
}
|
|
|
|
/*-----------------------------------------------*/
|
|
|
|
int bin2hex(unsigned char *in, int len, char *out)
|
|
{
|
|
int n1, n2;
|
|
unsigned char ch;
|
|
|
|
for (n1 = 0, n2 = 0; n1 < len; ++n1)
|
|
{
|
|
/* first nibble */
|
|
ch = in[n1] >> 4;
|
|
if (ch <= 0x09)
|
|
out[n2++] = ch + '0';
|
|
else
|
|
out[n2++] = ch - 10 + 'a';
|
|
/* second nibble */
|
|
ch = in[n1] & 0x0f;
|
|
if (ch <= 0x09)
|
|
out[n2++] = ch + '0';
|
|
else
|
|
out[n2++] = ch - 10 + 'a';
|
|
}
|
|
return n2;
|
|
}
|
|
|
|
/* NB: this return the number of _bits_ read */
|
|
int bint2bin(const char *in, int len, unsigned char *out)
|
|
{
|
|
int n;
|
|
|
|
memset(out,0,len);
|
|
for(n=0 ; n < len ; ++n)
|
|
if(in[n] == '1')
|
|
out[n/8]|=(0x80 >> (n%8));
|
|
return len;
|
|
}
|
|
|
|
int bin2bint(const unsigned char *in,int len,char *out)
|
|
{
|
|
int n;
|
|
|
|
for(n=0 ; n < len ; ++n)
|
|
out[n]=(in[n/8]&(0x80 >> (n%8))) ? '1' : '0';
|
|
return n;
|
|
}
|
|
|
|
/*-----------------------------------------------*/
|
|
|
|
void PrintValue(char *tag, unsigned char *val, int len)
|
|
{
|
|
#if VERBOSE
|
|
char obuf[2048];
|
|
int olen;
|
|
olen = bin2hex(val, len, obuf);
|
|
printf("%s = %.*s\n", tag, olen, obuf);
|
|
#endif
|
|
}
|
|
|
|
void OutputValue(char *tag, unsigned char *val, int len, FILE *rfp,int bitmode)
|
|
{
|
|
char obuf[2048];
|
|
int olen;
|
|
|
|
if(bitmode)
|
|
olen=bin2bint(val,len,obuf);
|
|
else
|
|
olen=bin2hex(val,len,obuf);
|
|
|
|
fprintf(rfp, "%s = %.*s\n", tag, olen, obuf);
|
|
#if VERBOSE
|
|
printf("%s = %.*s\n", tag, olen, obuf);
|
|
#endif
|
|
}
|
|
|
|
/*-----------------------------------------------*/
|
|
char *t_tag[2] = {"PLAINTEXT", "CIPHERTEXT"};
|
|
char *t_mode[6] = {"CBC","ECB","OFB","CFB1","CFB8","CFB128"};
|
|
enum Mode {CBC, ECB, OFB, CFB1, CFB8, CFB128};
|
|
enum XCrypt {XDECRYPT, XENCRYPT};
|
|
|
|
/*=============================*/
|
|
/* Monte Carlo Tests */
|
|
/*-----------------------------*/
|
|
|
|
/*#define gb(a,b) (((a)[(b)/8] >> ((b)%8))&1)*/
|
|
/*#define sb(a,b,v) ((a)[(b)/8]=((a)[(b)/8]&~(1 << ((b)%8)))|(!!(v) << ((b)%8)))*/
|
|
|
|
#define gb(a,b) (((a)[(b)/8] >> (7-(b)%8))&1)
|
|
#define sb(a,b,v) ((a)[(b)/8]=((a)[(b)/8]&~(1 << (7-(b)%8)))|(!!(v) << (7-(b)%8)))
|
|
|
|
int do_mct(char *amode,
|
|
int akeysz, unsigned char *aKey,unsigned char *iVec,
|
|
int dir, unsigned char *text, int len,
|
|
FILE *rfp)
|
|
{
|
|
int ret = 0;
|
|
unsigned char key[101][32];
|
|
unsigned char iv[101][AES_BLOCK_SIZE];
|
|
unsigned char ptext[1001][32];
|
|
unsigned char ctext[1001][32];
|
|
unsigned char ciphertext[64+4];
|
|
int i, j, n, n1, n2;
|
|
int imode = 0, nkeysz = akeysz/8;
|
|
EVP_CIPHER_CTX ctx;
|
|
|
|
if (len > 32)
|
|
{
|
|
printf("\n>>>> Length exceeds 32 for %s %d <<<<\n\n",
|
|
amode, akeysz);
|
|
return -1;
|
|
}
|
|
for (imode = 0; imode < 6; ++imode)
|
|
if (strcmp(amode, t_mode[imode]) == 0)
|
|
break;
|
|
if (imode == 6)
|
|
{
|
|
printf("Unrecognized mode: %s\n", amode);
|
|
return -1;
|
|
}
|
|
|
|
memcpy(key[0], aKey, nkeysz);
|
|
if (iVec)
|
|
memcpy(iv[0], iVec, AES_BLOCK_SIZE);
|
|
if (dir == XENCRYPT)
|
|
memcpy(ptext[0], text, len);
|
|
else
|
|
memcpy(ctext[0], text, len);
|
|
for (i = 0; i < 100; ++i)
|
|
{
|
|
/* printf("Iteration %d\n", i); */
|
|
if (i > 0)
|
|
{
|
|
OutputValue("KEY",key[i],nkeysz,rfp,0);
|
|
if (imode != ECB) /* ECB */
|
|
OutputValue("IV",iv[i],AES_BLOCK_SIZE,rfp,0);
|
|
/* Output Ciphertext | Plaintext */
|
|
OutputValue(t_tag[dir^1],dir ? ptext[0] : ctext[0],len,rfp,
|
|
imode == CFB1);
|
|
}
|
|
for (j = 0; j < 1000; ++j)
|
|
{
|
|
switch (imode)
|
|
{
|
|
case ECB:
|
|
if (j == 0)
|
|
{ /* set up encryption */
|
|
ret = AESTest(&ctx, amode, akeysz, key[i], NULL,
|
|
dir, /* 0 = decrypt, 1 = encrypt */
|
|
ptext[j], ctext[j], len);
|
|
if (dir == XENCRYPT)
|
|
memcpy(ptext[j+1], ctext[j], len);
|
|
else
|
|
memcpy(ctext[j+1], ptext[j], len);
|
|
}
|
|
else
|
|
{
|
|
if (dir == XENCRYPT)
|
|
{
|
|
EVP_Cipher(&ctx, ctext[j], ptext[j], len);
|
|
memcpy(ptext[j+1], ctext[j], len);
|
|
}
|
|
else
|
|
{
|
|
EVP_Cipher(&ctx, ptext[j], ctext[j], len);
|
|
memcpy(ctext[j+1], ptext[j], len);
|
|
}
|
|
}
|
|
break;
|
|
|
|
case CBC:
|
|
case OFB:
|
|
case CFB128:
|
|
if (j == 0)
|
|
{
|
|
ret = AESTest(&ctx, amode, akeysz, key[i], iv[i],
|
|
dir, /* 0 = decrypt, 1 = encrypt */
|
|
ptext[j], ctext[j], len);
|
|
if (dir == XENCRYPT)
|
|
memcpy(ptext[j+1], iv[i], len);
|
|
else
|
|
memcpy(ctext[j+1], iv[i], len);
|
|
}
|
|
else
|
|
{
|
|
if (dir == XENCRYPT)
|
|
{
|
|
EVP_Cipher(&ctx, ctext[j], ptext[j], len);
|
|
memcpy(ptext[j+1], ctext[j-1], len);
|
|
}
|
|
else
|
|
{
|
|
EVP_Cipher(&ctx, ptext[j], ctext[j], len);
|
|
memcpy(ctext[j+1], ptext[j-1], len);
|
|
}
|
|
}
|
|
break;
|
|
|
|
case CFB8:
|
|
if (j == 0)
|
|
{
|
|
ret = AESTest(&ctx, amode, akeysz, key[i], iv[i],
|
|
dir, /* 0 = decrypt, 1 = encrypt */
|
|
ptext[j], ctext[j], len);
|
|
}
|
|
else
|
|
{
|
|
if (dir == XENCRYPT)
|
|
EVP_Cipher(&ctx, ctext[j], ptext[j], len);
|
|
else
|
|
EVP_Cipher(&ctx, ptext[j], ctext[j], len);
|
|
}
|
|
if (dir == XENCRYPT)
|
|
{
|
|
if (j < 16)
|
|
memcpy(ptext[j+1], &iv[i][j], len);
|
|
else
|
|
memcpy(ptext[j+1], ctext[j-16], len);
|
|
}
|
|
else
|
|
{
|
|
if (j < 16)
|
|
memcpy(ctext[j+1], &iv[i][j], len);
|
|
else
|
|
memcpy(ctext[j+1], ptext[j-16], len);
|
|
}
|
|
break;
|
|
|
|
case CFB1:
|
|
if(j == 0)
|
|
{
|
|
/* compensate for wrong endianness of input file */
|
|
if(i == 0)
|
|
ptext[0][0]<<=7;
|
|
ret=AESTest(&ctx,amode,akeysz,key[i],iv[i],dir,
|
|
ptext[j], ctext[j], len);
|
|
}
|
|
else
|
|
{
|
|
if (dir == XENCRYPT)
|
|
EVP_Cipher(&ctx, ctext[j], ptext[j], len);
|
|
else
|
|
EVP_Cipher(&ctx, ptext[j], ctext[j], len);
|
|
|
|
}
|
|
if(dir == XENCRYPT)
|
|
{
|
|
if(j < 128)
|
|
sb(ptext[j+1],0,gb(iv[i],j));
|
|
else
|
|
sb(ptext[j+1],0,gb(ctext[j-128],0));
|
|
}
|
|
else
|
|
{
|
|
if(j < 128)
|
|
sb(ctext[j+1],0,gb(iv[i],j));
|
|
else
|
|
sb(ctext[j+1],0,gb(ptext[j-128],0));
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
--j; /* reset to last of range */
|
|
/* Output Ciphertext | Plaintext */
|
|
OutputValue(t_tag[dir],dir ? ctext[j] : ptext[j],len,rfp,
|
|
imode == CFB1);
|
|
fprintf(rfp, "\n"); /* add separator */
|
|
|
|
/* Compute next KEY */
|
|
if (dir == XENCRYPT)
|
|
{
|
|
if (imode == CFB8)
|
|
{ /* ct = CT[j-15] || CT[j-14] || ... || CT[j] */
|
|
for (n1 = 0, n2 = nkeysz-1; n1 < nkeysz; ++n1, --n2)
|
|
ciphertext[n1] = ctext[j-n2][0];
|
|
}
|
|
else if(imode == CFB1)
|
|
{
|
|
for(n1=0,n2=akeysz-1 ; n1 < akeysz ; ++n1,--n2)
|
|
sb(ciphertext,n1,gb(ctext[j-n2],0));
|
|
}
|
|
else
|
|
switch (akeysz)
|
|
{
|
|
case 128:
|
|
memcpy(ciphertext, ctext[j], 16);
|
|
break;
|
|
case 192:
|
|
memcpy(ciphertext, ctext[j-1]+8, 8);
|
|
memcpy(ciphertext+8, ctext[j], 16);
|
|
break;
|
|
case 256:
|
|
memcpy(ciphertext, ctext[j-1], 16);
|
|
memcpy(ciphertext+16, ctext[j], 16);
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (imode == CFB8)
|
|
{ /* ct = CT[j-15] || CT[j-14] || ... || CT[j] */
|
|
for (n1 = 0, n2 = nkeysz-1; n1 < nkeysz; ++n1, --n2)
|
|
ciphertext[n1] = ptext[j-n2][0];
|
|
}
|
|
else if(imode == CFB1)
|
|
{
|
|
for(n1=0,n2=akeysz-1 ; n1 < akeysz ; ++n1,--n2)
|
|
sb(ciphertext,n1,gb(ptext[j-n2],0));
|
|
}
|
|
else
|
|
switch (akeysz)
|
|
{
|
|
case 128:
|
|
memcpy(ciphertext, ptext[j], 16);
|
|
break;
|
|
case 192:
|
|
memcpy(ciphertext, ptext[j-1]+8, 8);
|
|
memcpy(ciphertext+8, ptext[j], 16);
|
|
break;
|
|
case 256:
|
|
memcpy(ciphertext, ptext[j-1], 16);
|
|
memcpy(ciphertext+16, ptext[j], 16);
|
|
break;
|
|
}
|
|
}
|
|
/* Compute next key: Key[i+1] = Key[i] xor ct */
|
|
for (n = 0; n < nkeysz; ++n)
|
|
key[i+1][n] = key[i][n] ^ ciphertext[n];
|
|
|
|
/* Compute next IV and text */
|
|
if (dir == XENCRYPT)
|
|
{
|
|
switch (imode)
|
|
{
|
|
case ECB:
|
|
memcpy(ptext[0], ctext[j], AES_BLOCK_SIZE);
|
|
break;
|
|
case CBC:
|
|
case OFB:
|
|
case CFB128:
|
|
memcpy(iv[i+1], ctext[j], AES_BLOCK_SIZE);
|
|
memcpy(ptext[0], ctext[j-1], AES_BLOCK_SIZE);
|
|
break;
|
|
case CFB8:
|
|
/* IV[i+1] = ct */
|
|
for (n1 = 0, n2 = 15; n1 < 16; ++n1, --n2)
|
|
iv[i+1][n1] = ctext[j-n2][0];
|
|
ptext[0][0] = ctext[j-16][0];
|
|
break;
|
|
case CFB1:
|
|
for(n1=0,n2=127 ; n1 < 128 ; ++n1,--n2)
|
|
sb(iv[i+1],n1,gb(ctext[j-n2],0));
|
|
ptext[0][0]=ctext[j-128][0]&0x80;
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
switch (imode)
|
|
{
|
|
case ECB:
|
|
memcpy(ctext[0], ptext[j], AES_BLOCK_SIZE);
|
|
break;
|
|
case CBC:
|
|
case OFB:
|
|
case CFB128:
|
|
memcpy(iv[i+1], ptext[j], AES_BLOCK_SIZE);
|
|
memcpy(ctext[0], ptext[j-1], AES_BLOCK_SIZE);
|
|
break;
|
|
case CFB8:
|
|
for (n1 = 0, n2 = 15; n < 16; ++n1, --n2)
|
|
iv[i+1][n1] = ptext[j-n2][0];
|
|
ctext[0][0] = ptext[j-16][0];
|
|
break;
|
|
case CFB1:
|
|
for(n1=0,n2=127 ; n1 < 128 ; ++n1,--n2)
|
|
sb(iv[i+1],n1,gb(ptext[j-n2],0));
|
|
ctext[0][0]=ptext[j-128][0]&0x80;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
/*================================================*/
|
|
/*----------------------------
|
|
# Config info for v-one
|
|
# AESVS MMT test data for ECB
|
|
# State : Encrypt and Decrypt
|
|
# Key Length : 256
|
|
# Fri Aug 30 04:07:22 PM
|
|
----------------------------*/
|
|
|
|
int proc_file(char *rqfile)
|
|
{
|
|
char afn[256], rfn[256];
|
|
FILE *afp = NULL, *rfp = NULL;
|
|
char ibuf[2048];
|
|
int ilen, len, ret = 0;
|
|
char algo[8] = "";
|
|
char amode[8] = "";
|
|
char atest[8] = "";
|
|
int akeysz = 0;
|
|
unsigned char iVec[20], aKey[40];
|
|
int dir = -1, err = 0, step = 0;
|
|
unsigned char plaintext[2048];
|
|
unsigned char ciphertext[2048];
|
|
char *rp;
|
|
EVP_CIPHER_CTX ctx;
|
|
|
|
if (!rqfile || !(*rqfile))
|
|
{
|
|
printf("No req file\n");
|
|
return -1;
|
|
}
|
|
strcpy(afn, rqfile);
|
|
|
|
if ((afp = fopen(afn, "r")) == NULL)
|
|
{
|
|
printf("Cannot open file: %s, %s\n",
|
|
afn, strerror(errno));
|
|
return -1;
|
|
}
|
|
strcpy(rfn,afn);
|
|
rp=strstr(rfn,"req/");
|
|
assert(rp);
|
|
memcpy(rp,"rsp",3);
|
|
rp = strstr(rfn, ".req");
|
|
memcpy(rp, ".rsp", 4);
|
|
if ((rfp = fopen(rfn, "w")) == NULL)
|
|
{
|
|
printf("Cannot open file: %s, %s\n",
|
|
rfn, strerror(errno));
|
|
fclose(afp);
|
|
afp = NULL;
|
|
return -1;
|
|
}
|
|
while (!err && (fgets(ibuf, sizeof(ibuf), afp)) != NULL)
|
|
{
|
|
ilen = strlen(ibuf);
|
|
/* printf("step=%d ibuf=%s",step,ibuf); */
|
|
switch (step)
|
|
{
|
|
case 0: /* read preamble */
|
|
if (ibuf[0] == '\n')
|
|
{ /* end of preamble */
|
|
if ((*algo == '\0') ||
|
|
(*amode == '\0') ||
|
|
(akeysz == 0))
|
|
{
|
|
printf("Missing Algorithm, Mode or KeySize (%s/%s/%d)\n",
|
|
algo,amode,akeysz);
|
|
err = 1;
|
|
}
|
|
else
|
|
{
|
|
fputs(ibuf, rfp);
|
|
++ step;
|
|
}
|
|
}
|
|
else if (ibuf[0] != '#')
|
|
{
|
|
printf("Invalid preamble item: %s\n", ibuf);
|
|
err = 1;
|
|
}
|
|
else
|
|
{ /* process preamble */
|
|
char *xp, *pp = ibuf+2;
|
|
int n;
|
|
if (akeysz)
|
|
{ /* insert current time & date */
|
|
time_t rtim = time(0);
|
|
fprintf(rfp, "# %s", ctime(&rtim));
|
|
}
|
|
else
|
|
{
|
|
fputs(ibuf, rfp);
|
|
if (strncmp(pp, "AESVS ", 6) == 0)
|
|
{
|
|
strcpy(algo, "AES");
|
|
/* get test type */
|
|
pp += 6;
|
|
xp = strchr(pp, ' ');
|
|
n = xp-pp;
|
|
strncpy(atest, pp, n);
|
|
atest[n] = '\0';
|
|
/* get mode */
|
|
xp = strrchr(pp, ' '); /* get mode" */
|
|
n = strlen(xp+1)-1;
|
|
strncpy(amode, xp+1, n);
|
|
amode[n] = '\0';
|
|
/* amode[3] = '\0'; */
|
|
printf("Test = %s, Mode = %s\n", atest, amode);
|
|
}
|
|
else if (strncasecmp(pp, "Key Length : ", 13) == 0)
|
|
{
|
|
akeysz = atoi(pp+13);
|
|
printf("Key size = %d\n", akeysz);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
case 1: /* [ENCRYPT] | [DECRYPT] */
|
|
if (ibuf[0] == '[')
|
|
{
|
|
fputs(ibuf, rfp);
|
|
++step;
|
|
if (strncasecmp(ibuf, "[ENCRYPT]", 9) == 0)
|
|
dir = 1;
|
|
else if (strncasecmp(ibuf, "[DECRYPT]", 9) == 0)
|
|
dir = 0;
|
|
else
|
|
{
|
|
printf("Invalid keyword: %s\n", ibuf);
|
|
err = 1;
|
|
}
|
|
break;
|
|
}
|
|
else if (dir == -1)
|
|
{
|
|
err = 1;
|
|
printf("Missing ENCRYPT/DECRYPT keyword\n");
|
|
break;
|
|
}
|
|
else
|
|
step = 2;
|
|
|
|
case 2: /* KEY = xxxx */
|
|
fputs(ibuf, rfp);
|
|
if(*ibuf == '\n')
|
|
break;
|
|
if(!strncasecmp(ibuf,"COUNT = ",8))
|
|
break;
|
|
|
|
if (strncasecmp(ibuf, "KEY = ", 6) != 0)
|
|
{
|
|
printf("Missing KEY\n");
|
|
err = 1;
|
|
}
|
|
else
|
|
{
|
|
len = hex2bin((char*)ibuf+6, strlen(ibuf+6)-1, aKey);
|
|
if (len < 0)
|
|
{
|
|
printf("Invalid KEY\n");
|
|
err =1;
|
|
break;
|
|
}
|
|
PrintValue("KEY", aKey, len);
|
|
if (strcmp(amode, "ECB") == 0)
|
|
{
|
|
memset(iVec, 0, sizeof(iVec));
|
|
step = (dir)? 4: 5; /* no ivec for ECB */
|
|
}
|
|
else
|
|
++step;
|
|
}
|
|
break;
|
|
|
|
case 3: /* IV = xxxx */
|
|
fputs(ibuf, rfp);
|
|
if (strncasecmp(ibuf, "IV = ", 5) != 0)
|
|
{
|
|
printf("Missing IV\n");
|
|
err = 1;
|
|
}
|
|
else
|
|
{
|
|
len = hex2bin((char*)ibuf+5, strlen(ibuf+5)-1, iVec);
|
|
if (len < 0)
|
|
{
|
|
printf("Invalid IV\n");
|
|
err =1;
|
|
break;
|
|
}
|
|
PrintValue("IV", iVec, len);
|
|
step = (dir)? 4: 5;
|
|
}
|
|
break;
|
|
|
|
case 4: /* PLAINTEXT = xxxx */
|
|
fputs(ibuf, rfp);
|
|
if (strncasecmp(ibuf, "PLAINTEXT = ", 12) != 0)
|
|
{
|
|
printf("Missing PLAINTEXT\n");
|
|
err = 1;
|
|
}
|
|
else
|
|
{
|
|
int nn = strlen(ibuf+12);
|
|
if(!strcmp(amode,"CFB1"))
|
|
len=bint2bin(ibuf+12,nn-1,plaintext);
|
|
else
|
|
len=hex2bin(ibuf+12, nn-1,plaintext);
|
|
if (len < 0)
|
|
{
|
|
printf("Invalid PLAINTEXT: %s", ibuf+12);
|
|
err =1;
|
|
break;
|
|
}
|
|
if (len >= sizeof(plaintext))
|
|
{
|
|
printf("Buffer overflow\n");
|
|
}
|
|
PrintValue("PLAINTEXT", (unsigned char*)plaintext, len);
|
|
if (strcmp(atest, "MCT") == 0) /* Monte Carlo Test */
|
|
{
|
|
if(do_mct(amode, akeysz, aKey, iVec,
|
|
dir, (unsigned char*)plaintext, len,
|
|
rfp) < 0)
|
|
exit(1);
|
|
}
|
|
else
|
|
{
|
|
ret = AESTest(&ctx, amode, akeysz, aKey, iVec,
|
|
dir, /* 0 = decrypt, 1 = encrypt */
|
|
plaintext, ciphertext, len);
|
|
OutputValue("CIPHERTEXT",ciphertext,len,rfp,
|
|
!strcmp(amode,"CFB1"));
|
|
}
|
|
step = 6;
|
|
}
|
|
break;
|
|
|
|
case 5: /* CIPHERTEXT = xxxx */
|
|
fputs(ibuf, rfp);
|
|
if (strncasecmp(ibuf, "CIPHERTEXT = ", 13) != 0)
|
|
{
|
|
printf("Missing KEY\n");
|
|
err = 1;
|
|
}
|
|
else
|
|
{
|
|
if(!strcmp(amode,"CFB1"))
|
|
len=bint2bin(ibuf+13,strlen(ibuf+13)-1,ciphertext);
|
|
else
|
|
len = hex2bin(ibuf+13,strlen(ibuf+13)-1,ciphertext);
|
|
if (len < 0)
|
|
{
|
|
printf("Invalid CIPHERTEXT\n");
|
|
err =1;
|
|
break;
|
|
}
|
|
|
|
PrintValue("CIPHERTEXT", ciphertext, len);
|
|
if (strcmp(atest, "MCT") == 0) /* Monte Carlo Test */
|
|
{
|
|
do_mct(amode, akeysz, aKey, iVec,
|
|
dir, ciphertext, len, rfp);
|
|
}
|
|
else
|
|
{
|
|
ret = AESTest(&ctx, amode, akeysz, aKey, iVec,
|
|
dir, /* 0 = decrypt, 1 = encrypt */
|
|
plaintext, ciphertext, len);
|
|
OutputValue("PLAINTEXT",(unsigned char *)plaintext,len,rfp,
|
|
!strcmp(amode,"CFB1"));
|
|
}
|
|
step = 6;
|
|
}
|
|
break;
|
|
|
|
case 6:
|
|
if (ibuf[0] != '\n')
|
|
{
|
|
err = 1;
|
|
printf("Missing terminator\n");
|
|
}
|
|
else if (strcmp(atest, "MCT") != 0)
|
|
{ /* MCT already added terminating nl */
|
|
fputs(ibuf, rfp);
|
|
}
|
|
step = 1;
|
|
break;
|
|
}
|
|
}
|
|
if (rfp)
|
|
fclose(rfp);
|
|
if (afp)
|
|
fclose(afp);
|
|
return err;
|
|
}
|
|
|
|
/*--------------------------------------------------
|
|
Processes either a single file or
|
|
a set of files whose names are passed in a file.
|
|
A single file is specified as:
|
|
aes_test -f xxx.req
|
|
A set of files is specified as:
|
|
aes_test -d xxxxx.xxx
|
|
The default is: -d req.txt
|
|
--------------------------------------------------*/
|
|
int main(int argc, char **argv)
|
|
{
|
|
char *rqlist = "req.txt";
|
|
FILE *fp = NULL;
|
|
char fn[250] = "", rfn[256] = "";
|
|
int f_opt = 0, d_opt = 1;
|
|
|
|
#ifdef FIPS
|
|
FIPS_mode_set(1);
|
|
#endif
|
|
ERR_load_crypto_strings();
|
|
if (argc > 1)
|
|
{
|
|
if (strcasecmp(argv[1], "-d") == 0)
|
|
{
|
|
d_opt = 1;
|
|
}
|
|
else if (strcasecmp(argv[1], "-f") == 0)
|
|
{
|
|
f_opt = 1;
|
|
d_opt = 0;
|
|
}
|
|
else
|
|
{
|
|
printf("Invalid parameter: %s\n", argv[1]);
|
|
return 0;
|
|
}
|
|
if (argc < 3)
|
|
{
|
|
printf("Missing parameter\n");
|
|
return 0;
|
|
}
|
|
if (d_opt)
|
|
rqlist = argv[2];
|
|
else
|
|
strcpy(fn, argv[2]);
|
|
}
|
|
if (d_opt)
|
|
{ /* list of files (directory) */
|
|
if (!(fp = fopen(rqlist, "r")))
|
|
{
|
|
printf("Cannot open req list file\n");
|
|
return -1;
|
|
}
|
|
while (fgets(fn, sizeof(fn), fp))
|
|
{
|
|
strtok(fn, "\r\n");
|
|
strcpy(rfn, fn);
|
|
printf("Processing: %s\n", rfn);
|
|
if (proc_file(rfn))
|
|
{
|
|
printf(">>> Processing failed for: %s <<<\n", rfn);
|
|
exit(1);
|
|
}
|
|
}
|
|
fclose(fp);
|
|
}
|
|
else /* single file */
|
|
{
|
|
printf("Processing: %s\n", fn);
|
|
if (proc_file(fn))
|
|
{
|
|
printf(">>> Processing failed for: %s <<<\n", fn);
|
|
}
|
|
}
|
|
return 0;
|
|
}
|