#include <AES.h>

#include<AES-128_V10.h>
//#include "./printf.h"

AES aes ;

byte *key = (unsigned char*)"0123456789010123";
const int size_mesg=16;






//real iv = iv x2 ex: 01234567 = 0123456701234567
unsigned long long int my_iv = 36753562;


  


unsigned char NwkSkey[16] = {
  0x2B, 0x7E, 0x15, 0x16, 0x28, 0xAE, 0xD2, 0xA6,
  0xAB, 0xF7, 0x15, 0x88, 0x09, 0xCF, 0x4F, 0x3C
};

unsigned char AppSkey[16] = {
  0x2B, 0x7E, 0x15, 0x16, 0x28, 0xAE, 0xD2, 0xA6,
  0xAB, 0xF7, 0x15, 0x88, 0x09, 0xCF, 0x4F, 0x3C
};



void setup ()
{
  Serial.begin (57600) ;
//  printf_begin();
  delay(500);
//  printf("\n===testng mode\n") ;
  
  randomSeed(134);


  

  
//  otfly_test () ;
//  otfly_test256 () ;
}

void loop () 
{
  prekey_test () ;
  delay(2000);
}

void prekey (int bits)
{
  aes.iv_inc();
  byte iv [N_BLOCK] ;


  byte plain[size_mesg];
  byte cipher [size_mesg] ;
  byte check [size_mesg] ;

  randomSeed(334);
  for(int i=0;i<size_mesg;i++) {
    plain[i]=random(255);
  }
  
 

 
  
  unsigned long ms = micros ();
  aes.set_IV(my_iv);
  aes.get_IV(iv);

  aes.do_aes_encrypt(plain,size_mesg,cipher,key,bits,iv);
  Serial.print("Encryption took: ");
  Serial.println(micros() - ms);
  ms = micros ();
  aes.set_IV(my_iv);
  aes.get_IV(iv);
  aes.do_aes_decrypt(cipher,size_mesg,check,key,bits,iv);
  Serial.print("Decryption took: ");
  Serial.println(micros() - ms);
/*  printf("\n\nPLAIN :");
  aes.printArray(plain,(bool)true);
  printf("\nCIPHER:");
  aes.printArray(cipher,(bool)false);
  printf("\nCHECK :");
  aes.printArray(check,(bool)true);
  printf("\nIV    :");
  aes.printArray(iv,16);
  printf("\n============================================================\n");

  */
  bool equal=true;
    for(int i=0;i<size_mesg-1;i++) {
      
      if(check[i]!=plain[i]) {
        printf("%d %d %d\n",plain[i],check[i],i);
        equal=false;
      }
    }
  Serial.print("CHECK ");
  Serial.println(equal);

  for(int i=0;i<16;i++) {
    Serial.print(plain[i]);
    Serial.print(" ");
  }
  Serial.println(); 

  ms = micros ();
  AES_Encrypt(plain,AppSkey);
  int time=micros() - ms;
  Serial.print("new enc took: ");
  Serial.println(time);

  for(int i=0;i<16;i++) {
    Serial.print(plain[i]);
    Serial.print(" ");
  }
  Serial.println(); 
 

   
}

void prekey_test ()
{
  prekey (128) ;
}