aestest.c

LOKI97加密解密算法,用C语言实现的一种加密解密算法。

    /* Create key structs for encrypt */
    st = makeKey(&enc_key, DIR_DECRYPT, keysize, key);
    if (st != TRUE) { fprintf(stderr, "makeKey failed with code %d\n", st); return FALSE; }

    /* test decrypt */
    st = makeKey(&dec_key, DIR_DECRYPT, keysize, key);
    st = blockDecrypt(&aes_cipher, &dec_key, cipher, BLOCK_SIZE*8, dtemp);
    if (st != TRUE) { fprintf(stderr, "blockDecrypt failed with code %d\n", st); return FALSE; }
    if (memcmp(dtemp, plain, sizeof(dtemp)) != 0) deok = FALSE;
    if (verbose>1) {
        fprintf(stderr,"CT: "); puthex(cipher,16,stderr);
        fprintf(stderr,"decrypt to: "); puthex(dtemp,16,stderr);
    }
    if (verbose) fprintf(stderr,"  decrypt %s\n", (deok?"GOOD" : "FAILED"));

    return (enok && deok);	/* return TRUE only if both en & decrypt ok */
}


/* 
 *  chk_e_ecb_mct(keysize, key, iv, plain, cipher)
 *    check ECB Encrypt MCT given key, iv, plain and cipher
 */
int chk_e_ecb_mct(int keysize, char *key, char *iv, BYTE plain[], BYTE cipher[])
{
    int 		enok = TRUE;		/* success/fail flag */
    BYTE		etemp[BLOCK_SIZE]; 	/* tmp blocks */
    keyInstance		enc_key, dec_key;	/* AES keyInstances */
    cipherInstance	aes_cipher;		/* AES cipherInstance */
    int iters = 10000;				/* num iterations in MCT test */
    int	i, st;

    /* Init AES cipher in MODE_ECB */
    st = cipherInit(&aes_cipher, MODE_ECB, iv);
    if (st != TRUE) { fprintf(stderr, "cipherInit failed with code %d\n", st); return FALSE; }

    /* Create key structs for encrypt */
    st = makeKey(&enc_key, DIR_ENCRYPT, keysize, key);
    if (st != TRUE) { fprintf(stderr, "makeKey failed with code %d\n", st); return FALSE; }

    /* test encrypt */
    if (verbose>1) { fprintf(stderr,"PT: "); puthex(plain,16,stderr); }
    for (i=0; i<iters; i++) {
        st = blockEncrypt(&aes_cipher, &enc_key, plain, BLOCK_SIZE*8, etemp);
        if (st != TRUE) {
	    fprintf(stderr, "blockEncrypt failed with code %d\n", st);
	    break;
	}
	memcpy(plain, etemp, sizeof(etemp));	/* plain = etemp */
    }
    if (verbose>1) {
        fprintf(stderr,"encrypt in mct test to: "); puthex(etemp,16,stderr);
    }
    if (memcmp(etemp, cipher, sizeof(etemp)) != 0) enok = FALSE;

    if (verbose) fprintf(stderr,"  test %s\n", (enok?"GOOD" : "FAILED"));

    return enok;
}


/* 
 *  chk_d_ecb_mct(keysize, key, iv, plain, cipher)
 *    check ECB Decrypt MCT given key, iv, plain and cipher
 */
int chk_d_ecb_mct(int keysize, char *key, char *iv, BYTE plain[], BYTE cipher[])
{
    int 		deok = TRUE;		/* success/fail flag */
    BYTE		dtemp[BLOCK_SIZE];	/* tmp block */
    keyInstance		enc_key, dec_key;	/* AES keyInstances */
    cipherInstance	aes_cipher;		/* AES cipherInstance */
    int iters = 10000;				/* num iterations in MCT test */
    int	i, st;

    /* Init AES cipher in MODE_ECB */
    st = cipherInit(&aes_cipher, MODE_ECB, iv);
    if (st != TRUE) { fprintf(stderr, "cipherInit failed with code %d\n", st); return FALSE; }

    /* Create key structs for decrypt */
    st = makeKey(&dec_key, DIR_DECRYPT, keysize, key);
    if (st != TRUE) { fprintf(stderr, "makeKey failed with code %d\n", st); return FALSE; }

    /* test decrypt */
    if (verbose>1) { fprintf(stderr,"PT: "); puthex(plain,16,stderr); }
    for (i=0; i<iters; i++) {
        st = blockDecrypt(&aes_cipher, &dec_key, cipher, BLOCK_SIZE*8, dtemp);
        if (st != TRUE) {
	    fprintf(stderr, "blockDecrypt failed with code %d\n", st);
	    break;
	}
	memcpy(cipher, dtemp, sizeof(dtemp));	/* cipher = dtemp */
    }
    if (verbose>1) {
        fprintf(stderr,"decrypt in mct test to: "); puthex(dtemp,16,stderr);
    }
    if (memcmp(dtemp, plain, sizeof(dtemp)) != 0) deok = FALSE;

    if (verbose) fprintf(stderr,"  test %s\n", (deok?"GOOD" : "FAILED"));

    return deok;
}


/* 
 *  chk_e_cbc_mct(keysize, key, iv, plain, cipher)
 *    check CBC Encrypt MCT given key, iv, plain and cipher
 */
int chk_e_cbc_mct(int keysize, char *key, char *iv, BYTE plain[], BYTE cipher[])
{
    int 		enok = TRUE;		/* success/fail flag */
    BYTE		prev[BLOCK_SIZE], etemp[BLOCK_SIZE]; /* tmp blocks */
    keyInstance		enc_key, dec_key;	/* AES keyInstances */
    cipherInstance	aes_cipher;		/* AES cipherInstance */
    int iters = 10000;				/* num iterations in MCT test */
    int	i, st;

    charToBYTE(prev,iv,sizeof(prev));		/* init prev = iv */

    /* Init AES cipher in MODE_CBC */
    st = cipherInit(&aes_cipher, MODE_CBC, iv);
    if (st != TRUE) { fprintf(stderr, "cipherInit failed with code %d\n", st); return FALSE; }

    /* Create key structs for encrypt */
    st = makeKey(&enc_key, DIR_ENCRYPT, keysize, key);
    if (st != TRUE) { fprintf(stderr, "makeKey failed with code %d\n", st); return FALSE; }

    /* test encrypt */
    if (verbose>1) { fprintf(stderr,"PT: "); puthex(plain,16,stderr); }
    for (i=0; i<iters; i++) {
        st = blockEncrypt(&aes_cipher, &enc_key, plain, BLOCK_SIZE*8, etemp);
        if (st != TRUE) {
	    fprintf(stderr, "blockEncrypt failed with code %d\n", st);
	    break;
	}
	memcpy(plain, prev, sizeof(prev));	/* plain = prev */
	memcpy(prev, etemp, sizeof(prev));	/* prev = etemp */
    }
    if (verbose>1) {
        fprintf(stderr,"encrypt in mct test to: "); puthex(etemp,16,stderr);
    }
    if (memcmp(etemp, cipher, sizeof(etemp)) != 0) enok = FALSE;

    if (verbose) fprintf(stderr,"  test %s\n", (enok?"GOOD" : "FAILED"));

    return enok;
}


/* 
 *  chk_d_cbc_mct(keysize, key, iv, plain, cipher)
 *    check CBC Decrypt MCT given key, iv, plain and cipher
 */
int chk_d_cbc_mct(int keysize, char *key, char *iv, BYTE plain[], BYTE cipher[])
{
    int 		deok = TRUE;		/* success/fail flag */
    BYTE		dtemp[BLOCK_SIZE];	/* tmp blocks */
    keyInstance		enc_key, dec_key;	/* AES keyInstances */
    cipherInstance	aes_cipher;		/* AES cipherInstance */
    int iters = 10000;				/* num iterations in MCT test */
    int	i, st;

    /* Init AES cipher in CBC MODE_CBC */
    st = cipherInit(&aes_cipher, MODE_CBC, iv);
    if (st != TRUE) { fprintf(stderr, "cipherInit failed with code %d\n", st); return FALSE; }

    /* Create key structs for decrypt */
    st = makeKey(&dec_key, DIR_DECRYPT, keysize, key);
    if (st != TRUE) { fprintf(stderr, "makeKey failed with code %d\n", st); return FALSE; }

    /* test decrypt */
    if (verbose>1) { fprintf(stderr,"PT: "); puthex(plain,16,stderr); }
    for (i=0; i<iters; i++) {
        st = blockDecrypt(&aes_cipher, &dec_key, cipher, BLOCK_SIZE*8, dtemp);
        if (st != TRUE) {
	    fprintf(stderr, "blockDecrypt failed with code %d\n", st);
	    break;
	}
	memcpy(cipher, dtemp, sizeof(dtemp));	/* cipher = dtemp */
    }
    if (verbose>1) {
        fprintf(stderr,"decrypt in mct test to: "); puthex(dtemp,16,stderr);
    }
    if (memcmp(dtemp, plain, sizeof(dtemp)) != 0) deok = FALSE;

    if (verbose) fprintf(stderr,"  test %s\n", (deok?"GOOD" : "FAILED"));

    return deok;
}


/* Returns a BYTE array from a string of hexadecimal digits. */
static BYTE *charToBYTE(BYTE *buf, char *hex, int len) {
    int i = 0, j = 0;
    while (j < len) {
        buf[j++] = (BYTE)((fromHex(hex[i++])<<4) | fromHex(hex[i++]));
    }
    return buf;
}


/* Returns number from 0 to 15 corresponding to hex digit ch */
static int fromHex (char ch) {
    if (ch >= '0' && ch <= '9')
        return ch - '0';
    else if (ch >= 'A' && ch <= 'F')
        return ch - 'A' + 10;
    else if (ch >= 'a' && ch <= 'f')
        return ch - 'a' + 10;
    else
        return 0;
}


/* puthex(out, len, f) - display a len byte value out in hex to file f */
static int puthex(BYTE *out, int len, FILE *f)
{
    int i;
    for(i=0;i<len;i++){
        fprintf(f, "%02X",*out++ & 0xff);
    }
    fputc(' ', f);
}