#include <AES.h>
//#include "./printf.h"

AES aes ;

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






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


  








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;
      }
    }
  printf("CHECK %d\n",equal);
  
}

void prekey_test ()
{
  prekey (128) ;
}