#include <cutter.h>

#include <freefare.h>

#include "mifare_desfire_auto_authenticate.h"

uint8_t key_data_null[8]  = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
uint8_t key_data_des[8]   = { 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H' };
uint8_t key_data_3des[16] = { 'C', 'a', 'r', 'd', ' ', 'M', 'a', 's', 't', 'e', 'r', ' ', 'K', 'e', 'y', '!' };
uint8_t key_data_aes[16]  = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
uint8_t key_data_3k3des[24]  = { 0x00, 0x01, 0x00, 0x01, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
				 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
			       };

const uint8_t key_data_aes_version = 0x42;

void
mifare_desfire_auto_authenticate(FreefareTag tag, uint8_t key_no)
{
    /* Determine which key is currently the master one */
    uint8_t key_version;
    int res = mifare_desfire_get_key_version(tag, key_no, &key_version);
    cut_assert_equal_int(0, res, cut_message("mifare_desfire_get_key_version()"));

    MifareDESFireKey key;

    switch (key_version) {
    case 0x00:
	key = mifare_desfire_des_key_new_with_version(key_data_null);
	break;
    case 0x42:
	key = mifare_desfire_aes_key_new_with_version(key_data_aes, key_data_aes_version);
	break;
    case 0xAA:
	key = mifare_desfire_des_key_new_with_version(key_data_des);
	break;
    case 0xC7:
	key = mifare_desfire_3des_key_new_with_version(key_data_3des);
	break;
    case 0x55:
	key = mifare_desfire_3k3des_key_new_with_version(key_data_3k3des);
	break;
    default:
	cut_fail("Unknown master key.");
    }

    cut_assert_not_null(key, cut_message("Cannot allocate key"));

    /* Authenticate with this key */
    switch (key_version) {
    case 0x00:
    case 0xAA:
    case 0xC7:
	res = mifare_desfire_authenticate(tag, key_no, key);
	break;
    case 0x55:
	res = mifare_desfire_authenticate_iso(tag, key_no, key);
	break;
    case 0x42:
	res = mifare_desfire_authenticate_aes(tag, key_no, key);
	break;
    }
    cut_assert_equal_int(0, res, cut_message("mifare_desfire_authenticate()"));

    mifare_desfire_key_free(key);
}