#include<math.h>
#include<stdlib.h>
#include<stdio.h>
+#include<string.h>
#include <fstream>
#include <sys/time.h>
int key_size=256;
int nb_test=1;
+int ctr=0;
+
+
+
-const int h=16;
-const int h2=h*h;
}
+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) {
+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];
+ 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) {
uchar *X=new uchar[h2];
uchar *fX=new uchar[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 a=0;a<h2;a++) {
+ for(int a=0;a<h2;a+=4) {
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+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+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+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+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) {
- 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]];
}
}
+int main(int argc, char** argv) {
+
+
+ int h=32;
+ int lena=0;
+ int size_buf=1;
-void Dynamickeygenerationnew(uchar *Secretkey, uchar *counter) {
+
+
+ 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])); //CTR ? 1 otherwise CBC like
+ 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;
- 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;
- 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];
+ }
}
for (int i = 0; i < 256 ; i++) {
-// mix[i]=(int)secret_key.BytePtr()[i]^(int)mycounter.BytePtr()[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++) {
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);
+ 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;
- 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();
- 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;
-
- 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++) {
+ if(buffer[i]!=seq[i])
+ equal=false;
+ }
+ cout<<"RESULT CORRECT: "<<equal<<endl;
}
+
- Dynamickeygenerationnew(Secretkey, counter);
return 0;
}
