new

[Cipher_code.git] / OneRoundIoT / OneRound / one_round_new.cpp

diff --git a/OneRoundIoT/OneRound/one_round_new.cpp b/OneRoundIoT/OneRound/one_round_new.cpp
index a22285512c213b6b8d0bcf4cddd0d1f4186432a7..24702f25998356e2a10e4a5da95718236b69a9f3 100644 (file)
--- a/OneRoundIoT/OneRound/one_round_new.cpp
+++ b/OneRoundIoT/OneRound/one_round_new.cpp
@@ -6,6 +6,7 @@
 #include<math.h>
 #include<stdlib.h>
 #include<stdio.h>
 #include<math.h>
 #include<stdlib.h>
 #include<stdio.h>

+#include<string.h>

 #include <fstream>
 #include <sys/time.h>
 
 #include <fstream>
 #include <sys/time.h>
 


@@ -28,9 +29,11 @@ using namespace std;
 
 int key_size=256;
 int nb_test=1;
 
 int key_size=256;
 int nb_test=1;

+int ctr=0;
+
+
+

 
 

-const int h=16;
-const int h2=h*h;

 
 
 
 
 
 


@@ -66,6 +69,14 @@ void inverse_tables(uchar *tab, int size_tab,uchar *inv_perm_tabs) {
 
 }
 
 
 }
 

+void inverse_tables_int(int *tab, int size_tab,int *inv_perm_tabs) {
+
+  for(int i=0;i<size_tab;i++) {
+    inv_perm_tabs[tab[i]] = i;
+  }
+
+}
+

 
 
 void rc4key(uchar *key, uchar *sc, int size_DK) {
 
 
 void rc4key(uchar *key, uchar *sc, int size_DK) {


@@ -124,59 +135,191 @@ void prga(uchar *sc, int ldata, uchar *r) {
 
 
 
 
 
 

+template<int h2>
+void encrypt_ctr(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *PboxRM, uchar *Sbox1, uchar *Sbox2, int enc) {
+
+
+//  uchar *X=new uchar[h2];
+//  uchar *fX=new uchar[h2];
+  uchar X[h2];
+  uchar fX[h2];
+  
+  int ind1,ind2;
+
+  
+   for(int a=0;a<h2;a++) {
+     X[a]=Sbox1[a&0xFF];           //Warning according to the size of h2, we can be outsize of Sbox1[a]
+   }
+
+   
+  for(int it=0;it<len;it++) {
+    if(enc) {
+      ind1=it*h2;
+      ind2=Pbox[it]*h2;
+    }
+    else {
+      ind2=it*h2;
+      ind1=Pbox[it]*h2;
+    }
+       
+
+
+    /*for(int a=0;a<h2;a+=4){
+      fX[a]=RM1[X[a]];
+      fX[a+1]=RM1[X[a+1]];
+      fX[a+2]=RM1[X[a+2]];
+      fX[a+3]=RM1[X[a+3]];
+      }*/
+
+    for(int a=0;a<h2;a+=4){
+      fX[a]=X[a];
+      fX[a+1]=X[a+1];
+      fX[a+2]=X[a+2];
+      fX[a+3]=X[a+3];
+    }

 
 

+    /*   if(it<513) {
+      for(int a=0;a<h2;a++)
+       printf("%d ",fX[a]);
+      printf("\n");
+      }*/
+    
+    *(int*)&fX[0]^=it;
+
+    /* if(it<513) {
+      for(int a=0;a<h2;a++)
+       printf("%d ",fX[a]);
+      printf("\n");
+      }*/
+
+    for(int a=0;a<h2;a+=4) {
+      fX[a]=fX[a]^RM1[a];
+      fX[a+1]=fX[a+1]^RM1[a+1];
+      fX[a+2]=fX[a+2]^RM1[a+2];
+      fX[a+3]=fX[a+3]^RM1[a+3];
+    }

 
 

+ 
+    for(int a=0;a<h2;a+=4) {
+      fX[a]=Sbox2[fX[a]];
+      fX[a+1]=Sbox2[fX[a+1]];
+      fX[a+2]=Sbox2[fX[a+2]];
+      fX[a+3]=Sbox2[fX[a+3]];
+    }
+    
+     for(int a=0;a<h2;a+=4) {
+      fX[a]=fX[a]^seq_in[ind2+a];
+      fX[a+1]=fX[a+1]^seq_in[ind2+a+1];
+      fX[a+2]=fX[a+2]^seq_in[ind2+a+2];
+      fX[a+3]=fX[a+3]^seq_in[ind2+a+3];
+    }

 
 

+ 
+    for(int a=0;a<h2;a+=4) {
+      seq_out[ind1+a]=fX[a];
+      seq_out[ind1+a+1]=fX[a+1];
+      seq_out[ind1+a+2]=fX[a+2];
+      seq_out[ind1+a+3]=fX[a+3];
+    }
+    
+    for(int a=0;a<h2;a+=4) {
+      RM1[a]=RM1[PboxRM[a]];
+      RM1[a+1]=RM1[PboxRM[a+1]];
+      RM1[a+2]=RM1[PboxRM[a+2]];
+      RM1[a+3]=RM1[PboxRM[a+3]];
+    }
+
+
+    
+  }
+
+
+}
+
+
+template<int h2>

 void encrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *PboxRM, uchar *Sbox1, uchar *Sbox2, int debug) {
 
 
 void encrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *PboxRM, uchar *Sbox1, uchar *Sbox2, int debug) {
 
 

-  uchar *X=new uchar[h2];
+/*  uchar *X=new uchar[h2];

   uchar *fX=new uchar[h2];
   uchar *fX=new uchar[h2];

+  unsigned int *lX=(unsigned int*)X;
+  unsigned int *lseq_in=(unsigned int*)seq_in;
+*/
+  uchar X[h2];
+  uchar fX[h2];
+//  unsigned int *lX=(unsigned int*)X;
+//  unsigned int *lseq_in=(unsigned int*)seq_in;

   
   

+

   for(int it=0;it<len;it++) {
     int ind1=it*h2;
     int ind2=Pbox[it]*h2;
 
   for(int it=0;it<len;it++) {
     int ind1=it*h2;
     int ind2=Pbox[it]*h2;
 

-    for(int a=0;a<h2;a++) {
+    for(int a=0;a<h2;a+=4) {

       X[a]=seq_in[ind2+a];
       X[a]=seq_in[ind2+a];

+      X[a+1]=seq_in[ind2+a+1];
+      X[a+2]=seq_in[ind2+a+2];
+      X[a+3]=seq_in[ind2+a+3];

     }
 
     }
 

-    for(int a=0;a<h2;a++){
+    for(int a=0;a<h2;a+=4){

       fX[a]=Sbox1[X[a]];
       fX[a]=Sbox1[X[a]];

+      fX[a+1]=Sbox1[X[a+1]];
+      fX[a+2]=Sbox1[X[a+2]];
+      fX[a+3]=Sbox1[X[a+3]];

     }
     }

-    

 
 

-    for(int a=0;a<h2;a++) {
+
+    for(int a=0;a<h2;a+=4) {

       fX[a]=fX[a]^RM1[a];
       fX[a]=fX[a]^RM1[a];

+      fX[a+1]=fX[a+1]^RM1[a+1];
+      fX[a+2]=fX[a+2]^RM1[a+2];
+      fX[a+3]=fX[a+3]^RM1[a+3];

     }
 
 
     }
 
 

-    for(int a=0;a<h2;a++) {
+    for(int a=0;a<h2;a+=4) {

       seq_out[ind1+a]=Sbox2[fX[a]];
       seq_out[ind1+a]=Sbox2[fX[a]];

+      seq_out[ind1+a+1]=Sbox2[fX[a+1]];
+      seq_out[ind1+a+2]=Sbox2[fX[a+2]];
+      seq_out[ind1+a+3]=Sbox2[fX[a+3]];

     }
 
     }
 

-    for(int a=0;a<h2;a++) {
+    for(int a=0;a<h2;a+=4) {

       RM1[a]=RM1[PboxRM[a]];
       RM1[a]=RM1[PboxRM[a]];

+      RM1[a+1]=RM1[PboxRM[a+1]];
+      RM1[a+2]=RM1[PboxRM[a+2]];
+      RM1[a+3]=RM1[PboxRM[a+3]];
+

     }
 
     }
 

-    
+

   }
 
 
   }
 
 

-}

 
 
 
 

+}
+

 
 

+template<int h2>

 void decrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *PboxRM, uchar *Sbox1, uchar *Sbox2, int debug) {
 
 
 void decrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *PboxRM, uchar *Sbox1, uchar *Sbox2, int debug) {
 
 

-  uchar *fX=new uchar[h2];
+  /*uchar *fX=new uchar[h2];
+  uchar *Inv_Sbox1=new uchar[256];
+  uchar *Inv_Sbox2=new uchar[256];
+  */
+  uchar fX[h2];
+  uchar Inv_Sbox1[256];
+  uchar Inv_Sbox2[256];

 
 
 
 

-  uchar *Inv_Sbox1=new uchar[256];
+  

   inverse_tables(Sbox1,256,Inv_Sbox1);
 
   inverse_tables(Sbox1,256,Inv_Sbox1);
 

-  uchar *Inv_Sbox2=new uchar[256];
+ 

   inverse_tables(Sbox2,256,Inv_Sbox2);
   
 
   inverse_tables(Sbox2,256,Inv_Sbox2);
   
 


@@ -190,22 +333,38 @@ void decrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *P
 
 
 
 
 
 

-    for(int a=0;a<h2;a++) {
+    for(int a=0;a<h2;a+=4) {

       fX[a]=seq_in[ind1+a];
       fX[a]=seq_in[ind1+a];

+      fX[a+1]=seq_in[ind1+a+1];
+      fX[a+2]=seq_in[ind1+a+2];
+      fX[a+3]=seq_in[ind1+a+3];
+           

     }
     }

-    for(int a=0;a<h2;a++) {
+    for(int a=0;a<h2;a+=4) {

       fX[a]=Inv_Sbox2[fX[a]];
       fX[a]=Inv_Sbox2[fX[a]];

+      fX[a+1]=Inv_Sbox2[fX[a+1]];
+      fX[a+2]=Inv_Sbox2[fX[a+2]];
+      fX[a+3]=Inv_Sbox2[fX[a+3]];

     }
     }

-    for(int a=0;a<h2;a++) {
+    for(int a=0;a<h2;a+=4) {

       fX[a]=fX[a]^RM1[a];
       fX[a]=fX[a]^RM1[a];

+      fX[a+1]=fX[a+1]^RM1[a+1];
+      fX[a+2]=fX[a+2]^RM1[a+2];
+      fX[a+3]=fX[a+3]^RM1[a+3];

     }
 
     }
 

-    for(int a=0;a<h2;a++) {
+    for(int a=0;a<h2;a+=4) {

       RM1[a]=RM1[PboxRM[a]];
       RM1[a]=RM1[PboxRM[a]];

+      RM1[a+1]=RM1[PboxRM[a+1]];
+      RM1[a+2]=RM1[PboxRM[a+2]];
+      RM1[a+3]=RM1[PboxRM[a+3]];

     }
     
     }
     

-    for(int a=0;a<h2;a++) {
+    for(int a=0;a<h2;a+=4) {

       seq_out[ind2+a]=Inv_Sbox1[fX[a]];
       seq_out[ind2+a]=Inv_Sbox1[fX[a]];

+      seq_out[ind2+a+1]=Inv_Sbox1[fX[a+1]];
+      seq_out[ind2+a+2]=Inv_Sbox1[fX[a+2]];
+      seq_out[ind2+a+3]=Inv_Sbox1[fX[a+3]];

     }
 
      
     }
 
      


@@ -215,8 +374,47 @@ void decrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *P
 }
 
 
 }
 
 

+int main(int argc, char** argv) {
+
+
+  int h=32;
+  int lena=0;
+  int size_buf=1;
+
+
+  
+  for(int i=1; i<argc; i++){
+    if(strncmp(argv[i],"nb",2)==0)    nb_test = atoi(&(argv[i][2]));    //nb of test         
+    if(strncmp(argv[i],"ctr",3)==0) ctr = atoi(&(argv[i][3]));          //CTR ? 1  otherwise CBC like
+    if(strncmp(argv[i],"h",1)==0) h = atoi(&(argv[i][1]));          //size of block
+    if(strncmp(argv[i],"sizebuf",7)==0) size_buf = atoi(&(argv[i][7]));          //SIZE of the buffer
+    if(strncmp(argv[i],"lena",4)==0) lena = atoi(&(argv[i][4]));          //Use Lena or buffer
+  }
+
+  printf("nb times %d\n",nb_test);
+  printf("ctr %d\n",ctr);
+  printf("h %d\n",h);
+  printf("lena %d\n",lena);
+  printf("size_buf %d\n",size_buf);
+
+  int h2=h*h;
+  
+
+      
+  int seed=time(NULL);
+  cout<<seed<<endl;
+  srand48(seed);
+
+  uchar Secretkey[key_size];
+
+  uchar counter[key_size];
+
+  for(int i=0;i<key_size;i++) {
+    Secretkey[i]=lrand48()&0xFF;
+    counter[i]=lrand48()&0xFF;
+  }

 
 

-void Dynamickeygenerationnew(uchar *Secretkey, uchar *counter) {
+  

   int size = 64;
   uchar DK[size];
 
   int size = 64;
   uchar DK[size];
 


@@ -225,22 +423,44 @@ void Dynamickeygenerationnew(uchar *Secretkey, uchar *counter) {
 
   int width;
   int height;
 
   int width;
   int height;

-  uchar *data_R, *data_G, *data_B;
-  load_RGB_pixmap("lena.ppm", &width, &height, &data_R, &data_G, &data_B);

 
 

+  uchar *data_R, *data_G, *data_B;
+  int imsize;
+  uchar *buffer;
+  
+  if(lena==1) {
+    load_RGB_pixmap("lena.ppm", &width, &height, &data_R, &data_G, &data_B);
+    imsize=width*height*3;
+//  load_RGB_pixmap("No_ecb_mode_picture.ppm", &width, &height, &data_R, &data_G, &data_B);
+  }
+  else {
+    width=height=size_buf;
+    imsize=width*height;
+    buffer=new uchar[imsize];
+    for(int i=0;i<imsize;i++) {
+      buffer[i]=lrand48();
+    }
+  }

 
 
 
   
 
 
 
   

-  int imsize=width*height*3;
+  

   uchar* seq= new uchar[imsize];
   uchar* seq2= new uchar[imsize];
 
   int oneD=width*height;
   uchar* seq= new uchar[imsize];
   uchar* seq2= new uchar[imsize];
 
   int oneD=width*height;

-  for(int i=0;i<oneD;i++) {
-    seq[i]=data_R[i];
-    seq[oneD+i]=data_G[i];
-    seq[2*oneD+i]=data_B[i];
+  if(lena) {
+    for(int i=0;i<oneD;i++) {
+      seq[i]=data_R[i];
+      seq[oneD+i]=data_G[i];
+      seq[2*oneD+i]=data_B[i];
+    }
+  }
+  else {
+    for(int i=0;i<oneD;i++) {
+      seq[i]=buffer[i];
+    }

   }
 
 
   }
 
 


@@ -259,14 +479,9 @@ void Dynamickeygenerationnew(uchar *Secretkey, uchar *counter) {
 
     
   for (int i = 0; i < 256 ; i++) {
 
     
   for (int i = 0; i < 256 ; i++) {

-//      mix[i]=(int)secret_key.BytePtr()[i]^(int)mycounter.BytePtr()[i];

     mix[i]=Secretkey[i]^counter[i];
   }
 
     mix[i]=Secretkey[i]^counter[i];
   }
 

-/*  byte digest[64];
-  SHA512().CalculateDigest(digest, mix, 256);
-*/
-

   
   cout<<"hash "<<endl;
   for (int i = 0; i < 64 ; i++) {
   
   cout<<"hash "<<endl;
   for (int i = 0; i < 64 ; i++) {


@@ -324,83 +539,148 @@ void Dynamickeygenerationnew(uchar *Secretkey, uchar *counter) {
   for(int i=0;i<h2;i++){
     RM2[i]=RM1[i];
   }
   for(int i=0;i<h2;i++){
     RM2[i]=RM1[i];
   }

+  int *Inv_Pbox=new int[len];
+  inverse_tables_int(Pbox,len,Inv_Pbox);

   
  
  double time=0;
   double t=TimeStart();
 
   int i;
   
  
  double time=0;
   double t=TimeStart();
 
   int i;

-  for(i=0;i<nb_test;i++)
-  {
-    encrypt(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
+  switch(h) {
+  case 4: 
+    for(i=0;i<nb_test;i++)
+    {
+      if(ctr)
+       encrypt_ctr<4*4>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1);
+      else
+       encrypt<4*4>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
+      
+    }
+    break;
+  case 8: 
+    for(i=0;i<nb_test;i++)
+    {
+      if(ctr)
+       encrypt_ctr<8*8>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1);
+      else
+       encrypt<8*8>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
+      
+    }
+    break;
+  case 16: 
+    for(i=0;i<nb_test;i++)
+    {
+      if(ctr)
+       encrypt_ctr<16*16>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1);
+      else
+       encrypt<16*16>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
+      
+    }
+    break;
+  case 32: 
+    for(i=0;i<nb_test;i++)
+    {
+      if(ctr)
+       encrypt_ctr<32*32>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1);
+      else
+       encrypt<32*32>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
+      
+    }
+    break;
+  case 64: 
+    for(i=0;i<nb_test;i++)
+    {
+      if(ctr)
+       encrypt_ctr<64*64>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1);
+      else
+       encrypt<64*64>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
+      
+    }
+    break;

   }
   }

-  

   time+=TimeStop(t);
   cout<<"Time encrypt "<<time<<endl;
 
 
   time+=TimeStop(t);
   cout<<"Time encrypt "<<time<<endl;
 
 

-  for(int i=0;i<oneD;i++) {
-    data_R[i]=seq2[i];
-    data_G[i]=seq2[oneD+i];
-    data_B[i]=seq2[2*oneD+i];
+  if(lena) {
+    for(int i=0;i<oneD;i++) {
+      data_R[i]=seq2[i];
+      data_G[i]=seq2[oneD+i];
+      data_B[i]=seq2[2*oneD+i];
+    }
+    store_RGB_pixmap("lena2.ppm", data_R, data_G, data_B, width, height);

   }
   }

-  store_RGB_pixmap("lena2.ppm", data_R, data_G, data_B, width, height);                    

   
 
   time=0;
   t=TimeStart();
   
 
   time=0;
   t=TimeStart();

-  for(i=0;i<nb_test;i++) {
-    decrypt(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+  switch(h) {
+  case 4:
+    for(i=0;i<nb_test;i++) {
+      if(ctr)
+       encrypt_ctr<4*4>(seq2, seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+      else
+       decrypt<4*4>(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+    }
+    break;
+  case 8:
+    for(i=0;i<nb_test;i++) {
+      if(ctr)
+       encrypt_ctr<8*8>(seq2, seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+      else
+       decrypt<8*8>(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+    }
+    break;
+  case 16:
+    for(i=0;i<nb_test;i++) {
+      if(ctr)
+       encrypt_ctr<16*16>(seq2, seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+      else
+       decrypt<16*16>(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+    }
+    break;
+  case 32:
+    for(i=0;i<nb_test;i++) {
+      if(ctr)
+       encrypt_ctr<32*32>(seq2, seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+      else
+       decrypt<32*32>(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+    }
+    break;
+  case 64:
+    for(i=0;i<nb_test;i++) {
+      if(ctr)
+       encrypt_ctr<64*64>(seq2, seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+      else
+       decrypt<64*64>(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+    }
+    break;

   }
 
   time+=TimeStop(t);
   cout<<"Time decrypt "<<time<<endl;
 
   }
 
   time+=TimeStop(t);
   cout<<"Time decrypt "<<time<<endl;
 

-  
-  for(int i=0;i<oneD;i++) {
-    data_R[i]=seq[i];
-    data_G[i]=seq[oneD+i];
-    data_B[i]=seq[2*oneD+i];
-  }
-  store_RGB_pixmap("lena3.ppm", data_R, data_G, data_B, width, height);                    
-  
-
-
-
-
-}
-
-
-
-int main(int argc, char** argv) {
-  cout << "Hello, World!" << endl;
-
-
- 
-
-  if(argc==2)
-    nb_test=atoi(argv[1]);
-  if(nb_test<=0 || nb_test>10000) {
-    printf("nb tests is not correct\n");
-    exit(0);
+  if(lena) {
+    for(int i=0;i<oneD;i++) {
+      data_R[i]=seq[i];
+      data_G[i]=seq[oneD+i];
+      data_B[i]=seq[2*oneD+i];
+    }
+    store_RGB_pixmap("lena3.ppm", data_R, data_G, data_B, width, height);

   }
   }

-  else
-    printf("nb tests = %d\n\n",nb_test);
-
-  int seed=time(NULL);
-  cout<<seed<<endl;
-  srand48(seed);
-
-  uchar Secretkey[key_size];
-
-  uchar counter[key_size];
-
-  for(int i=0;i<key_size;i++) {
-    Secretkey[i]=lrand48()&0xFF;
-    counter[i]=lrand48()&0xFF;
+  else {
+    bool equal=true;
+    for(int i=0;i<imsize;i++) {
+      //cout<<(int)buffer[i]<<endl;
+      if(buffer[i]!=seq[i]) {
+       equal=false;
+      }
+    }
+    cout<<"RESULT CORRECT: "<<equal<<endl;

   }
   }

+  

 
 

-  Dynamickeygenerationnew(Secretkey, counter);

 
   return 0;
 }
 
   return 0;
 }