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 6d9f431031a209bf115f49f62146ffc8079c9398..6a26c6e4efb6ca2d65bab2fe762edb3c044d609a 100644 (file)
--- a/OneRoundIoT/OneRound/one_round_new.cpp
+++ b/OneRoundIoT/OneRound/one_round_new.cpp
@@ -30,10 +30,10 @@ using namespace std;
 int key_size=256;
 int nb_test=1;
 int ctr=0;
-int h=64;
 
 
-int h2=h*h;
+
+
 
 
 
@@ -135,7 +135,7 @@ 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) {
 
 
@@ -143,9 +143,9 @@ void encrypt_ctr(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, in
   uchar *fX=new uchar[h2];
   int ind1,ind2;
 
-
+  
    for(int a=0;a<h2;a++) {
-     X[a]=Sbox1[a];
+     X[a]=Sbox1[a&0xFF];           //Warning according to the size of h2, we can be outsize of Sbox1[a]
    }
 
    
@@ -161,27 +161,69 @@ void encrypt_ctr(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, in
        
 
 
-    for(int a=0;a<h2;a++){
-      fX[a]=Sbox1[RM1[X[a]]];
+    /*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++) {
+    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++) {
+    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++) {
-      seq_out[ind1+a]=fX[a]^seq_in[ind2+a];
+    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++) {
+    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]];
     }
 
 
@@ -192,7 +234,7 @@ void encrypt_ctr(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, in
 }
 
 
-
+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) {
 
 
@@ -206,61 +248,42 @@ void encrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *P
     int ind1=it*h2;
     int ind2=Pbox[it]*h2;
 
-    for(int a=0;a<h2;a+=1) {
+    for(int a=0;a<h2;a+=4) {
       X[a]=seq_in[ind2+a];
-/*      X[a+1]=seq_in[ind2+a+1];
+      X[a+1]=seq_in[ind2+a+1];
       X[a+2]=seq_in[ind2+a+2];
       X[a+3]=seq_in[ind2+a+3];
-      /*      X[a+4]=seq_in[ind2+a+4];                                                                                                                                    
-      X[a+5]=seq_in[ind2+a+5];                                                                                                                                            
-      X[a+6]=seq_in[ind2+a+6];                                                                                                                                            
-      X[a+7]=seq_in[ind2+a+7];*/
     }
 
-    for(int a=0;a<h2;a+=1){
+    for(int a=0;a<h2;a+=4){
       fX[a]=Sbox1[X[a]];
-/*      fX[a+1]=Sbox1[X[a+1]];
+      fX[a+1]=Sbox1[X[a+1]];
       fX[a+2]=Sbox1[X[a+2]];
       fX[a+3]=Sbox1[X[a+3]];
-      /*      fX[a+4]=Sbox1[X[a+4]];                                                                                                                                      
-      fX[a+5]=Sbox1[X[a+5]];                                                                                                                                              
-      fX[a+6]=Sbox1[X[a+6]];                                                                                                                                              
-      fX[a+7]=Sbox1[X[a+7]];*/
     }
 
 
-    for(int a=0;a<h2;a+=1) {
+    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+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];
-      /*      fX[a+4]=fX[a+4]^RM1[a+4];                                                                                                                                   
-      fX[a+5]=fX[a+5]^RM1[a+5];                                                                                                                                           
-      fX[a+6]=fX[a+6]^RM1[a+6];                                                                                                                                           
-      fX[a+7]=fX[a+7]^RM1[a+7];*/
     }
 
 
-    for(int a=0;a<h2;a+=1) {
+    for(int a=0;a<h2;a+=4) {
       seq_out[ind1+a]=Sbox2[fX[a]];
-/*      seq_out[ind1+a+1]=Sbox2[fX[a+1]];
+      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]];
-      /*      seq_out[ind1+a+4]=Sbox2[fX[a+4]];                                                                                                                           
-      seq_out[ind1+a+5]=Sbox2[fX[a+5]];                                                                                                                                   
-      seq_out[ind1+a+6]=Sbox2[fX[a+6]];                                                                                                                                   
-      seq_out[ind1+a+7]=Sbox2[fX[a+7]];*/
     }
 
-    for(int a=0;a<h2;a+=1) {
+    for(int a=0;a<h2;a+=4) {
       RM1[a]=RM1[PboxRM[a]];
-/*      RM1[a+1]=RM1[PboxRM[a+1]];
+      RM1[a+1]=RM1[PboxRM[a+1]];
       RM1[a+2]=RM1[PboxRM[a+2]];
       RM1[a+3]=RM1[PboxRM[a+3]];
-      /*      RM1[a+4]=RM1[PboxRM[a+4]];                                                                                                                                  
-      RM1[a+5]=RM1[PboxRM[a+5]];                                                                                                                                          
-      RM1[a+6]=RM1[PboxRM[a+6]];                                                                                                                                          
-      RM1[a+7]=RM1[PboxRM[a+7]];*/
+
     }
 
 
@@ -272,7 +295,7 @@ void encrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *P
 }
 
 
-
+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) {
 
 
@@ -296,22 +319,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+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+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+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+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+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]];
     }
 
      
@@ -324,6 +363,8 @@ 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;
+  
   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
@@ -334,18 +375,8 @@ int main(int argc, char** argv) {
   printf("ctr %d\n",ctr);
   printf("h %d\n",h);
 
-  h2=h*h;
+  int h2=h*h;
   
-/*
-  if(argc==2)
-    nb_test=atoi(argv[1]);
-  if(nb_test<=0 || nb_test>10000) {
-    printf("nb tests is not correct\n");
-    exit(0);
-  }
-  else
-    printf("nb tests = %d\n\n",nb_test);
-*/
 
       
   int seed=time(NULL);
@@ -465,18 +496,56 @@ int main(int argc, char** argv) {
   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;
-  for(i=0;i<nb_test;i++)
-  {
-//    encrypt(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
-    encrypt_ctr(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1);
+  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;
   }
-  
   time+=TimeStop(t);
   cout<<"Time encrypt "<<time<<endl;
 
@@ -491,9 +560,39 @@ int main(int argc, char** argv) {
 
   time=0;
   t=TimeStart();
-  for(i=0;i<nb_test;i++) {
-//    decrypt(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
-    encrypt_ctr(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;
   }
 
   time+=TimeStop(t);