From 9e250c72f9c70af3a3a7448e50c23a7ad0ea1bd5 Mon Sep 17 00:00:00 2001
From: couturie <couturie@extinction>
Date: Wed, 6 Sep 2017 14:04:26 +0200
Subject: [PATCH] new

---
 OneRoundIoT/OneRound/Makefile           |  11 +-
 OneRoundIoT/OneRound/one_round_par.cpp  |   2 +
 OneRoundIoT/OneRound/one_round_par2.cpp | 778 ++++++++++++++++++++++++
 3 files changed, 790 insertions(+), 1 deletion(-)
 create mode 100644 OneRoundIoT/OneRound/one_round_par2.cpp

diff --git a/OneRoundIoT/OneRound/Makefile b/OneRoundIoT/OneRound/Makefile
index 9da9adc..8f0c0cf 100644
--- a/OneRoundIoT/OneRound/Makefile
+++ b/OneRoundIoT/OneRound/Makefile
@@ -20,14 +20,23 @@ endif
 one_round_new.o: one_round_new.cpp 
 	$(CXX) -c -o $@ $< $(CFLAGS)
 
+
+
 one_round_par.o: one_round_par.cpp 
 	$(CXX)  -fopenmp -c -o $@ $< $(CFLAGS)
 
+one_round_par2.o: one_round_par2.cpp 
+	$(CXX)  -fopenmp -c -o $@ $< $(CFLAGS)
+
 one_round_new: pixmap_io.o one_round_new.o 
 	$(CXX)  -o $@ $^ $(CFLAGS)
 
 one_round_par: pixmap_io.o one_round_par.o 
 	$(CXX) -fopenmp -o $@ $^ $(CFLAGS)
 
+
+one_round_par2: pixmap_io.o one_round_par2.o 
+	$(CXX) -fopenmp -o $@ $^ $(CFLAGS)
+
 clean:
-	rm -rf *.o one_round_new
+	rm -rf *.o one_round_new one_round_par one_round_par2
diff --git a/OneRoundIoT/OneRound/one_round_par.cpp b/OneRoundIoT/OneRound/one_round_par.cpp
index bd23683..8519fe2 100644
--- a/OneRoundIoT/OneRound/one_round_par.cpp
+++ b/OneRoundIoT/OneRound/one_round_par.cpp
@@ -144,6 +144,7 @@ void rotate(uchar *RM1, uchar *RM2, int size, int n)
 }
 
 
+
 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, int num) {
 
@@ -267,6 +268,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, int num) {
 
diff --git a/OneRoundIoT/OneRound/one_round_par2.cpp b/OneRoundIoT/OneRound/one_round_par2.cpp
new file mode 100644
index 0000000..e42a6d5
--- /dev/null
+++ b/OneRoundIoT/OneRound/one_round_par2.cpp
@@ -0,0 +1,778 @@
+//gcc pixmap_io.c  -c 
+//g++ -O3 one_round_new.cpp pixmap_io.o  -o one_round_new -std=c++11   
+
+#include <iostream>
+#include <list>
+#include<math.h>
+#include<stdlib.h>
+#include<stdio.h>
+#include<string.h>
+#include <fstream>
+#include <sys/time.h>
+#include <omp.h>
+
+/*#include <cryptopp/hex.h>
+#include <cryptopp/sha.h>
+#include <cryptopp/osrng.h>
+#include <cryptopp/secblock.h>
+*/
+
+
+extern "C" {
+  int load_RGB_pixmap(char *filename, int *width, int *height, unsigned char**R_data, unsigned char**G_data, unsigned char**B_data);
+  void store_RGB_pixmap(char *filename, unsigned char *R_data, unsigned char *G_data, unsigned char *B_data, int width, int height);
+}
+
+
+//using namespace CryptoPP;
+using namespace std;
+
+
+int key_size=256;
+int nb_test=1;
+int ctr=0;
+
+
+
+
+
+
+
+typedef unsigned char   uchar;
+
+
+double TimeStart()
+{
+  struct timeval tstart;
+  gettimeofday(&tstart,0);
+  return( (double) (tstart.tv_sec + tstart.tv_usec*1e-6) );
+}
+
+double TimeStop(double t)
+{
+  struct timeval tend;
+
+  gettimeofday(&tend,0);
+  t = (double) (tend.tv_sec + tend.tv_usec*1e-6) - t;
+  return (t);
+}
+
+
+
+
+
+
+void inverse_tables(uchar *tab, int size_tab,uchar *inv_perm_tabs) {
+
+  for(int i=0;i<size_tab;i++) {
+    inv_perm_tabs[tab[i]] = i;
+  }
+
+}
+
+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) {
+
+  for(int i=0;i<256;i++) {
+    sc[i]=i;
+  }
+
+
+  uchar j0 = 0;
+  for(int i0=0; i0<256; i0++) {
+    j0 = (j0 + sc[i0] + key[i0%size_DK] )&0xFF;
+    uchar tmp = sc[i0];
+    sc[i0] = sc[j0 ];
+    sc[j0] = tmp;
+  }
+}
+
+
+
+void rc4keyperm(uchar *key,int len, int rp,int *sc, int size_DK) {
+
+  //sc=1:len;
+
+
+  
+  for (int i=0;i<len;i++) {
+    sc[i]=i;
+  }
+  for (int it = 0; it < rp; it++) {
+    int j0 = 1;
+    for(int i0 = 0; i0<len; i0++) {
+      j0 = (j0 + sc[i0] + sc[j0] + key[i0%size_DK] )% len;
+      int tmp = sc[i0];
+      sc[i0] = sc[j0];
+      sc[j0] = tmp;
+    }
+
+  }
+}
+
+void prga(uchar *sc, int ldata, uchar *r) {
+  uchar i0=0;
+  uchar j0=0;
+
+  for (int it=0; it<ldata; it++) {
+    i0 = ((i0+1)&0xFE); //%255);
+    j0 = (j0 + sc[i0])&0xFF;
+    uchar tmp = sc[i0];
+    sc[i0] = sc[j0];
+    sc[j0] = tmp;
+    r[it]=sc[(sc[i0]+sc[j0])&0xFF];
+  }
+}
+
+void rotate(uchar *RM1, uchar *RM2, int size, int n)
+{
+  int i;
+   for (i = 0; i< size-n; i++)
+     RM2[i+n] = RM1[i];
+
+   for (i = 0; i< n; i++)
+     RM2[i] = RM1[size-n-1+i];
+}
+
+
+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, int num) {
+
+
+  uchar X[h2];
+  uchar fX[h2*num];
+
+
+  
+
+
+  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]
+   }
+
+
+  int loc_len=len/num;
+  
+#pragma omp parallel for
+  for(int p=0;p<num;p++) {
+
+    int id=omp_get_thread_num();
+    uchar RM2[h2];
+    for(int a=0;a<h2;a++) {
+      RM2[a]=RM1[id*h2+a];           //Warning according to the size of h2, we can be outsize of Sbox1[a]
+    }
+
+  
+    int offset=p*loc_len;
+
+    
+    for(int it=offset;it<offset+loc_len;it++) {
+    int ind1,ind2;
+
+
+    //cout<<id<<" "<<it<<endl;
+    
+    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[id*h2+a]=X[a];
+      fX[id*h2+a+1]=X[a+1];
+      fX[id*h2+a+2]=X[a+2];
+      fX[id*h2+a+3]=X[a+3];
+    }
+
+
+
+    
+    /*   if(it<513) {
+      for(int a=0;a<h2;a++)
+	printf("%d ",fX[a]);
+      printf("\n");
+      }*/
+    
+    *(int*)&fX[id*h2+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[id*h2+a]=fX[id*h2+a]^RM1[id*h2+a];
+      fX[id*h2+a+1]=fX[id*h2+a+1]^RM1[id*h2+a+1];
+      fX[id*h2+a+2]=fX[id*h2+a+2]^RM1[id*h2+a+2];
+      fX[id*h2+a+3]=fX[id*h2+a+3]^RM1[id*h2+a+3];
+      }*/
+
+
+
+    
+    for(int a=0;a<h2;a+=4) {
+      fX[id*h2+a]=Sbox2[fX[id*h2+a]];
+      fX[id*h2+a+1]=Sbox2[fX[id*h2+a+1]];
+      fX[id*h2+a+2]=Sbox2[fX[id*h2+a+2]];
+      fX[id*h2+a+3]=Sbox2[fX[id*h2+a+3]];
+    }
+
+//    rotate(RM1, &RM2[id*h2], h2, Pbox[it]%h2);
+    
+    for(int a=0;a<h2;a+=4) {
+      RM2[a]=RM2[PboxRM[a]];
+      RM2[a+1]=RM2[PboxRM[a+1]];
+      RM2[a+2]=RM2[PboxRM[a+2]];
+      RM2[a+3]=RM2[PboxRM[a+3]];
+    }
+    
+    for(int a=0;a<h2;a+=4) {
+      fX[id*h2+a]=fX[id*h2+a]^RM2[id*h2+a];
+      fX[id*h2+a+1]=fX[id*h2+a+1]^RM2[id*h2+a+1];
+      fX[id*h2+a+2]=fX[id*h2+a+2]^RM2[id*h2+a+2];
+      fX[id*h2+a+3]=fX[id*h2+a+3]^RM2[id*h2+a+3];
+    }
+
+
+    
+     for(int a=0;a<h2;a+=4) {
+      fX[id*h2+a]=fX[id*h2+a]^seq_in[ind2+a];
+      fX[id*h2+a+1]=fX[id*h2+a+1]^seq_in[ind2+a+1];
+      fX[id*h2+a+2]=fX[id*h2+a+2]^seq_in[ind2+a+2];
+      fX[id*h2+a+3]=fX[id*h2+a+3]^seq_in[ind2+a+3];
+    }
+
+ 
+    for(int a=0;a<h2;a+=4) {
+      seq_out[ind1+a]=fX[id*h2+a];
+      seq_out[ind1+a+1]=fX[id*h2+a+1];
+      seq_out[ind1+a+2]=fX[id*h2+a+2];
+      seq_out[ind1+a+3]=fX[id*h2+a+3];
+    }
+    
+    /*for(int a=0;a<h2;a+=4) {
+      RM1[id*h2+a]=RM1[id*h2+PboxRM[a]];
+      RM1[id*h2+a+1]=RM1[id*h2+PboxRM[a+1]];
+      RM1[id*h2+a+2]=RM1[id*h2+PboxRM[a+2]];
+      RM1[id*h2+a+3]=RM1[id*h2+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, int num) {
+
+
+/*  uchar *X=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*num];
+  uchar fX[h2*num];
+  uchar RM2[h2*num];
+//  unsigned int *lX=(unsigned int*)X;
+//  unsigned int *lseq_in=(unsigned int*)seq_in;
+  
+#pragma omp parallel for 
+  for(int it=0;it<len;it++) {
+    int ind1=it*h2;
+    int ind2=Pbox[it]*h2;
+    int id=omp_get_thread_num();
+ 
+    for(int a=0;a<h2;a+=4) {
+      X[id*h2+a]=seq_in[ind2+a];
+      X[id*h2+a+1]=seq_in[ind2+a+1];
+      X[id*h2+a+2]=seq_in[ind2+a+2];
+      X[id*h2+a+3]=seq_in[ind2+a+3];
+    }
+
+    for(int a=0;a<h2;a+=4){
+      fX[id*h2+a]=Sbox1[X[id*h2+a]];
+      fX[id*h2+a+1]=Sbox1[X[id*h2+a+1]];
+      fX[id*h2+a+2]=Sbox1[X[id*h2+a+2]];
+      fX[id*h2+a+3]=Sbox1[X[id*h2+a+3]];
+    }
+
+
+/*    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];
+      }*/
+    rotate(RM1, &RM2[id*h2], h2, Pbox[it]%h2);
+    for(int a=0;a<h2;a+=4) {
+      fX[id*h2+a]=fX[id*h2+a]^RM2[id*h2+a];
+      fX[id*h2+a+1]=fX[id*h2+a+1]^RM2[id*h2+a+1];
+      fX[id*h2+a+2]=fX[id*h2+a+2]^RM2[id*h2+a+2];
+      fX[id*h2+a+3]=fX[id*h2+a+3]^RM2[id*h2+a+3];
+    }
+
+    for(int a=0;a<h2;a+=4) {
+      seq_out[ind1+a]=Sbox2[fX[id*h2+a]];
+      seq_out[ind1+a+1]=Sbox2[fX[id*h2+a+1]];
+      seq_out[ind1+a+2]=Sbox2[fX[id*h2+a+2]];
+      seq_out[ind1+a+3]=Sbox2[fX[id*h2+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 decrypt(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, int *PboxRM, uchar *Inv_Sbox1, uchar *Inv_Sbox2, int debug, int num) {
+
+
+  /*uchar *fX=new uchar[h2];
+  uchar *Inv_Sbox1=new uchar[256];
+  uchar *Inv_Sbox2=new uchar[256];
+  */
+  uchar fX[h2*num];
+  uchar RM2[h2*num];
+
+#pragma omp parallel for 
+  for(int it=0;it<len;it++) {
+
+    int ind1=it*h2;
+    int ind2=Pbox[it]*h2;
+    int id=omp_get_thread_num();
+
+
+
+    for(int a=0;a<h2;a+=4) {
+      fX[id*h2+a]=seq_in[ind1+a];
+      fX[id*h2+a+1]=seq_in[ind1+a+1];
+      fX[id*h2+a+2]=seq_in[ind1+a+2];
+      fX[id*h2+a+3]=seq_in[ind1+a+3];
+	    
+    }
+    for(int a=0;a<h2;a+=4) {
+      fX[id*h2+a]=Inv_Sbox2[fX[id*h2+a]];
+      fX[id*h2+a+1]=Inv_Sbox2[fX[id*h2+a+1]];
+      fX[id*h2+a+2]=Inv_Sbox2[fX[id*h2+a+2]];
+      fX[id*h2+a+3]=Inv_Sbox2[fX[id*h2+a+3]];
+    }
+    /*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) {
+      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]];
+      }*/
+
+
+    rotate(RM1, &RM2[id*h2], h2, Pbox[it]%h2);
+    for(int a=0;a<h2;a+=4) {
+      fX[id*h2+a]=fX[id*h2+a]^RM2[id*h2+a];
+      fX[id*h2+a+1]=fX[id*h2+a+1]^RM2[id*h2+a+1];
+      fX[id*h2+a+2]=fX[id*h2+a+2]^RM2[id*h2+a+2];
+      fX[id*h2+a+3]=fX[id*h2+a+3]^RM2[id*h2+a+3];      
+    }
+
+    
+    for(int a=0;a<h2;a+=4) {
+      seq_out[ind2+a]=Inv_Sbox1[fX[id*h2+a]];
+      seq_out[ind2+a+1]=Inv_Sbox1[fX[id*h2+a+1]];
+      seq_out[ind2+a+2]=Inv_Sbox1[fX[id*h2+a+2]];
+      seq_out[ind2+a+3]=Inv_Sbox1[fX[id*h2+a+3]];
+    }
+
+     
+  }
+
+
+}
+
+
+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;
+  }
+
+  
+  int size = 64;
+  uchar DK[size];
+
+
+
+
+  int width;
+  int height;
+
+  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();
+    }
+  }
+
+
+
+  
+  
+  uchar* seq= new uchar[imsize];
+  uchar* seq2= new uchar[imsize];
+
+  int oneD=width*height;
+  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];
+    }
+  }
+
+
+
+  
+
+  int total_len=imsize;
+  int rp=1;
+  int len= total_len/h2;
+
+
+  
+  uchar *mix=new uchar[256];
+
+
+
+    
+  for (int i = 0; i < 256 ; i++) {
+    mix[i]=Secretkey[i]^counter[i];
+  }
+
+  
+//  cout<<"hash "<<endl;
+  for (int i = 0; i < 64 ; i++) {
+//    DK[i]=digest[i];
+    DK[i]=mix[i];
+  }
+
+
+
+  
+  uchar Sbox1[256];
+  rc4key(DK, Sbox1, 16);
+
+  uchar Sbox2[256];
+  rc4key(&DK[16], Sbox2, 16);
+  uchar Inv_Sbox1[256];
+  uchar Inv_Sbox2[256];
+  inverse_tables(Sbox1,256,Inv_Sbox1);
+  inverse_tables(Sbox2,256,Inv_Sbox2);
+
+
+
+  
+  uchar sc[256];
+  rc4key(&DK[32], sc, 16);
+
+  int num=omp_get_max_threads();
+  cout<<"num "<<num<<endl;
+  
+  uchar RM1[num*(h * h)];
+  uchar RM2[num*(h * h)];
+  prga(sc, num*(h * h), RM1);
+  for(int i=0;i<num*h2;i++) {
+    RM2[i]=RM1[i];
+  }
+  
+
+
+  
+  
+    
+  
+  uchar keyp[16];
+  for (int i = 48; i < 64; i++)
+    keyp[i-48] = DK[i];
+
+//  cout<<len<<endl;
+  int *Pbox=new int[len];
+
+
+  
+  int *PboxRM=new int[h2];
+
+  rc4keyperm(keyp, len, rp, Pbox, 16);
+
+//  printf("len %d\n",len);
+  for(int i=0;i<len;i++) {
+//    printf("%d \n",Pbox[i]);
+  }
+  
+
+
+ 
+  double time=0;
+  double t=TimeStart();
+
+  int i;
+  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,num);
+      else
+	encrypt<4*4>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0,num);
+      
+    }
+    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,num);
+      else
+	encrypt<8*8>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0,num);
+      
+    }
+    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,num);
+      else
+	encrypt<16*16>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0,num);
+      
+    }
+    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,num);
+      else
+	encrypt<32*32>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0,num);
+      
+    }
+    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,num);
+      else
+	encrypt<64*64>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0,num);
+      
+    }
+    break;
+  case 128: 
+    for(i=0;i<nb_test;i++)
+    {
+      if(ctr)
+	encrypt_ctr<128*128>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1,num);
+      else
+	encrypt<128*128>(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0,num);
+      
+    }
+    break;
+  }
+  time+=TimeStop(t);
+  cout<<"Time encrypt "<<time<<endl;
+
+
+  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);
+  }
+  
+
+
+
+  for(int i=0;i<imsize;i++) {
+    seq[i]=0;
+  }
+
+  
+  time=0;
+  t=TimeStart();
+  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,num);
+      else
+	decrypt<4*4>(seq2,seq,len,RM2,Pbox,PboxRM,Inv_Sbox1,Inv_Sbox2,0,num);
+    }
+    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,num);
+      else
+	decrypt<8*8>(seq2,seq,len,RM2,Pbox,PboxRM,Inv_Sbox1,Inv_Sbox2,0,num);
+    }
+    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,num);
+      else
+	decrypt<16*16>(seq2,seq,len,RM2,Pbox,PboxRM,Inv_Sbox1,Inv_Sbox2,0,num);
+    }
+    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,num);
+      else
+	decrypt<32*32>(seq2,seq,len,RM2,Pbox,PboxRM,Inv_Sbox1,Inv_Sbox2,0,num);
+    }
+    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,num);
+      else
+	decrypt<64*64>(seq2,seq,len,RM2,Pbox,PboxRM,Inv_Sbox1,Inv_Sbox2,0,num);
+    }
+    break;
+  case 128:
+    for(i=0;i<nb_test;i++) {
+      if(ctr)
+	encrypt_ctr<128*128>(seq2, seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0,num);
+      else
+	decrypt<128*128>(seq2,seq,len,RM2,Pbox,PboxRM,Inv_Sbox1,Inv_Sbox2,0,num);
+    }
+    break;  
+
+    
+  }
+
+  time+=TimeStop(t);
+  cout<<"Time decrypt "<<time<<endl;
+
+  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 {
+    bool equal=true;
+    for(int i=0;i<imsize;i++) {
+      //cout<<"sol"<<(int)buffer[i]<<" "<<(int)seq[i]<<" "<<endl;
+      if(buffer[i]!=seq[i]) {
+	equal=false;
+      }
+    }
+    cout<<"RESULT CORRECT: "<<equal<<endl;
+  }
+  
+
+
+  return 0;
+}
-- 
2.39.5