ccmaes_test.c

#include "ccmaes.h"
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h> 
#include <assert.h>
extern bool CCMAES_DEBUG;

void static print_arr(uint8_t* rk, size_t len){
    for(size_t i = 0; i<len; ++i){
        printf("%x ", *(rk+i));
    }
    puts("");
}

int is_equal(uint8_t* expected, uint8_t* actual, size_t len){
    for(size_t i=0; i<len; i++){
        if(expected[i]!=actual[i]){
            return 0;
        }
    }
    return 1;
}


int main(void){
    CCMAES_DEBUG = true;
    //40414243 44454647 48494a4b 4c4d4e4f
    uint8_t key1[16] = {0x40, 0x41, 0x42, 0x43, 
                        0x44, 0x45, 0x46, 0x47, 
                        0x48, 0x49, 0x4a, 0x4b, 
                        0x4c, 0x4d, 0x4e, 0x4f};
    //20212223
    uint8_t p1[4] = {0x20, 0x21, 0x22, 0x23};
    //10111213 141516
    uint8_t n1[7] = {0x10, 0x11, 0x12, 0x13, 
                    0x14, 0x15, 0x16};
    //00010203 04050607
    uint8_t a1[8] = {0x00, 0x01, 0x02, 0x03, 
                    0x04, 0x05, 0x06, 0x07}; 
    //71 62 1 5b 4d ac 25 5d
    uint8_t cipher1[8] = {0x71, 0x62, 0x01, 0x5b, 0x4d, 0xac, 0x25, 0x5d};
    
    uint8_t actual_cipher1[8] = {0};
    printf("CCM AES encrypt:\n");
    ccm_aes_encrypt(actual_cipher1, p1, key1, n1, a1, 7, 8, 4, 4);
    printf("cipher:\n");
    print_arr(actual_cipher1, 8);
    assert(is_equal(cipher1, actual_cipher1, 8));
    printf("PASS CCM AES encrypt test 1\n");

    puts("");
    uint8_t actual_pt1[4] = {0};
    ccm_aes_decrypt(actual_pt1, cipher1, key1, n1, a1, 7, 8, 8, 4);
    printf("plaintext:\n");
    print_arr(actual_pt1, 4);
    assert(is_equal(p1, actual_pt1, 4));
    printf("PASS CCM AES decrypt test 1\n");

    puts("");
    //20212223 24252627 28292a2b 2c2d2e2f
    uint8_t p2[16] = {0x20, 0x21, 0x22, 0x23,
                    0x24, 0x25, 0x26, 0x27, 
                    0x28, 0x29, 0x2a, 0x2b, 
                    0x2c, 0x2d, 0x2e, 0x2f};
    //10111213 14151617
    uint8_t n2[8] = {0x10, 0x11, 0x12, 0x13, 
                    0x14, 0x15, 0x16, 0x17};
    //00010203 04050607 08090a0b 0c0d0e0f
    uint8_t a2[16] = {0x00, 0x01, 0x02, 0x03, 
                    0x04, 0x05, 0x06, 0x07,
                    0x08, 0x09, 0x0a, 0x0b,
                    0x0c, 0x0d, 0x0e, 0x0f}; 
    //d2 a1 f0 e0 51 ea 5f 62 8 1a 77 92 7 3d 59 3d 1f c6 4f bf ac cd
    uint8_t cipher2[22] = {0xd2, 0xa1, 0xf0, 0xe0, 0x51, 0xea, 0x5f, 0x62,
                        0x08, 0x1a, 0x77, 0x92, 0x07, 0x3d, 0x59, 0x3d,
                        0x1f, 0xc6, 0x4f, 0xbf, 0xac, 0xcd};

    uint8_t* actual_cipher2 = malloc(16 * sizeof(uint8_t*));
    printf("CCM AES encrypt:\n");
    ccm_aes_encrypt(actual_cipher2, p2, key1, n2, a2, 8, 16, 16*sizeof(uint8_t), 6);
    printf("cipher:\n");
    print_arr(actual_cipher2, 22);
    assert(is_equal(cipher2, actual_cipher2, 22));
    printf("PASS CCM AES encrypt test 2\n");

    puts("");
    uint8_t* actual_pt2 = malloc(16 * sizeof(uint8_t));
    ccm_aes_decrypt(actual_pt2, cipher2, key1, n2, a2, 8, 16, 22, 6);
    printf("plaintext:\n");
    print_arr(actual_pt2, 16);
    assert(is_equal(p2, actual_pt2, 16));
    printf("PASS CCM AES decrypt test 2\n");

    puts("");
    uint8_t p3[24] = {0x20, 0x21, 0x22, 0x23,
                    0x24, 0x25, 0x26, 0x27, 
                    0x28, 0x29, 0x2a, 0x2b, 
                    0x2c, 0x2d, 0x2e, 0x2f,
                    0x30, 0x31, 0x32, 0x33,
                    0x34, 0x35, 0x36, 0x37};
    uint8_t n3[12] = {0x10, 0x11, 0x12, 0x13, 
                    0x14, 0x15, 0x16, 0x17,
                    0x18, 0x19,0x1a, 0x1b};
    uint8_t a3[20] = {0x00, 0x01, 0x02, 0x03, 
                    0x04, 0x05, 0x06, 0x07,
                    0x08, 0x09, 0x0a, 0x0b,
                    0x0c, 0x0d, 0x0e, 0x0f,
                    0x10, 0x11, 0x12, 0x13};
    //e3 b2 1 a9 f5 b7 1a 7a 
    //9b 1c ea ec cd 97 e7 b 
    //61 76 aa d9 a4 42 8a a5 
    //48 43 92 fb c1 b0 99 51
    uint8_t cipher3[32] = {0xe3, 0xb2, 0x01, 0xa9, 0xf5, 0xb7, 0x1a, 0x7a,
                            0x9b, 0x1c, 0xea, 0xec, 0xcd, 0x97, 0xe7, 0x0b,
                            0x61, 0x76, 0xaa, 0xd9, 0xa4, 0x42, 0x8a, 0xa5,
                            0x48, 0x43, 0x92, 0xfb, 0xc1, 0xb0, 0x99, 0x51};
    uint8_t actual_cipher3[32] = {0};
    printf("CCM AES encrypt:\n");
    ccm_aes_encrypt(actual_cipher3, p3, key1, n3, a3, 12, 20, 24, 8);
    printf("cipher:\n");
    print_arr(actual_cipher3, 32);
    assert(is_equal(cipher3, actual_cipher3, 32));
    printf("PASS CCM AES encrypt test 3\n");


    puts("");
    uint8_t actual_pt3[24] = {0};
    ccm_aes_decrypt(actual_pt3, cipher3, key1, n3, a3, 12, 20, 32, 8);
    printf("plaintext:\n");
    print_arr(actual_pt3, 24);
    assert(is_equal(p3, actual_pt3, 24));
    printf("PASS CCM AES decrypt test 3\n");
}