Subversion Repositories tiny_aes

#include <stdio.h>

#include <stdlib.h>

typedef unsigned char byte;

typedef unsigned int word;

void encrypt_128_key_expand_inline_no_branch(word state[], word key[]);

void encrypt_192_key_expand_inline_no_branch(word state[], word key[]);

void encrypt_256_key_expand_inline_no_branch(word state[], word key[]);

word rand_word();

void rand_word_array(word w[], int bit_num);

void print_verilog_hex(word w[], int bit_num);

int main() {

    const int num_case = 5;

    int bit_num;

    int i;

    word state[4];

    word key[8];

    bit_num = 128;

    printf("AES-%d test cases:\n\n", bit_num);

    for(i=0; i<num_case; i++) {

        rand_word_array(state, 128);

        rand_word_array(key, bit_num);

        printf("plaintext: ");

        print_verilog_hex(state, 128);

        printf("\n");

        printf("key:       ");

        print_verilog_hex(key, bit_num);

        printf("\n");

        encrypt_128_key_expand_inline_no_branch(state, key);

        printf("ciphertext:");

        print_verilog_hex(state, 128);

        printf("\n\n");

    }

    bit_num = 192;

    printf("AES-%d test cases:\n\n", bit_num);

    for(i=0; i<num_case; i++) {

        rand_word_array(state, 128);

        rand_word_array(key, bit_num);

        printf("plaintext: ");

        print_verilog_hex(state, 128);

        printf("\n");

        printf("key:       ");

        print_verilog_hex(key, bit_num);

        printf("\n");

        encrypt_192_key_expand_inline_no_branch(state, key);

        printf("ciphertext:");

        print_verilog_hex(state, 128);

        printf("\n\n");

    }

    bit_num = 256;

    printf("AES-%d test cases:\n\n", bit_num);

    for(i=0; i<num_case; i++) {

        rand_word_array(state, 128);

        rand_word_array(key, bit_num);

        printf("plaintext: ");

        print_verilog_hex(state, 128);

        printf("\n");

        printf("key:       ");

        print_verilog_hex(key, bit_num);

        printf("\n");

        encrypt_256_key_expand_inline_no_branch(state, key);

        printf("ciphertext:");

        print_verilog_hex(state, 128);

        printf("\n\n");

    }

    return 0;

}

word rand_word() {

    word w = 0;

    int i;

    for(i=0; i<4; i++) {

        word x = rand() & 255;

        w = (w << 8) | x;

    }

    return w;

}

void rand_word_array(word w[], int bit_num) {

    int word_num = bit_num / 32;

    int i;

    for(i=0; i<word_num; i++)

        w[i] = rand_word();

}

void print_verilog_hex(word w[], int bit_num) {

    int byte_num = bit_num / 8;

    int i;

    byte *b = (byte *)w;

    printf("%d'h", bit_num);

    for(i=0; i<byte_num; i++)

        printf("%02x", b[i]);

}