/* Copyright (c) 2016 Jaycon Systems, LLC. All Rights Reserved */

#include <stdint.h>

#include "nrf.h"
#include "nrf_gpio.h"

#include "cmac.h"

#define ERROR_LED_PIN 12
#define ERROR_LED_ACTIVE LOW

#if (ERROR_LED_ACTIVE == LOW)
#define ERROR_LED_OFF nrf_gpio_pin_set(ERROR_LED_PIN)
#define ERROR_LED_ON  nrf_gpio_pin_clear(ERROR_LED_PIN)
#else
#define ERROR_LED_OFF nrf_gpio_pin_clear(ERROR_LED_PIN)
#define ERROR_LED_ON  nrf_gpio_pin_set(ERROR_LED_PIN)
#endif /* ERROR_LET_ACTIVE */

void test_cmac(void) {
  /* NIST Examples */
  block_t k = { .ui8={0x2b, 0x7e, 0x15, 0x16, 0x28, 0xae, 0xd2, 0xa6,
		      0xab, 0xf7, 0x15, 0x88, 0x09, 0xcf, 0x4f, 0x3c}};
  cmac_aes128_init(&k);
  /* Test Null Message, NIST Example */
  uint8_t msg1[] = "";
  block_t case1 = {.ui8={0xbb, 0x1d, 0x69, 0x29, 0xe9, 0x59, 0x37, 0x28,
  			 0x7f, 0xa3, 0x7d, 0x12, 0x9b, 0x75, 0x67, 0x46}};
  block_t tag = cmac_aes128(msg1, 0);
  if (!block_cmp(&tag, &case1)) {
    ERROR_LED_ON;
    while(1);
  }
  /* 16b example */
  uint8_t msg2[] = {0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96,
  		    0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a};
  block_t case2 = {.ui8={0x07, 0x0a, 0x16, 0xb4, 0x6b, 0x4d, 0x41, 0x44,
  			0xf7, 0x9b, 0xdd, 0x9d, 0xd0, 0x4a, 0x28, 0x7c}};
  tag = cmac_aes128(msg2, sizeof(msg2));
  if (!block_cmp(&tag, &case2)) {
    ERROR_LED_ON;
    while(1);
  }
  /* 40b example */
  uint8_t msg3[] = {0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96,
  		    0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a,
  		    0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c,
  		    0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e, 0x51,
  		    0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11};
  block_t case3 = {.ui8={0xdf, 0xa6, 0x67, 0x47, 0xde, 0x9a, 0xe6, 0x30,
  			 0x30, 0xca, 0x32, 0x61, 0x14, 0x97, 0xc8, 0x27}};
  tag = cmac_aes128(msg3, sizeof(msg3));
  if (!block_cmp(&tag, &case3)) {
    ERROR_LED_ON;
    while(1);
  }
  /* 64b example */
  uint8_t msg4[] = {0x6b, 0xc1, 0xbe, 0xe2, 0x2e, 0x40, 0x9f, 0x96,
  		    0xe9, 0x3d, 0x7e, 0x11, 0x73, 0x93, 0x17, 0x2a,
  		    0xae, 0x2d, 0x8a, 0x57, 0x1e, 0x03, 0xac, 0x9c,
  		    0x9e, 0xb7, 0x6f, 0xac, 0x45, 0xaf, 0x8e, 0x51,
  		    0x30, 0xc8, 0x1c, 0x46, 0xa3, 0x5c, 0xe4, 0x11,
  		    0xe5, 0xfb, 0xc1, 0x19, 0x1a, 0x0a, 0x52, 0xef,
  		    0xf6, 0x9f, 0x24, 0x45, 0xdf, 0x4f, 0x9b, 0x17,
  		    0xad, 0x2b, 0x41, 0x7b, 0xe6, 0x6c, 0x37, 0x10};
  block_t case4 = {.ui8={0x51, 0xf0, 0xbe, 0xbf, 0x7e, 0x3b, 0x9d, 0x92,
  			 0xfc, 0x49, 0x74, 0x17, 0x79, 0x36, 0x3c, 0xfe}};
  tag = cmac_aes128(msg4, sizeof(msg4));
  if (!block_cmp(&tag, &case4)) {
    ERROR_LED_ON;
    while(1);
  }
}

int main(void) {
  nrf_gpio_cfg_output(ERROR_LED_PIN);
  ERROR_LED_OFF;
  for (;;) {
    test_cmac();
  }
}