-//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];
- uchar RM2[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]
- }
-
-#pragma omp parallel for
- for(int it=0;it<len;it++) {
- int ind1,ind2;
- int id=omp_get_thread_num();
-
- //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) {
- 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);
-
- uchar outd[2*(h * h)];
- prga(sc, 2*(h * h), outd);
-
- int num=omp_get_max_threads();
- cout<<"num "<<num<<endl;
-
- uchar RM1[h*h];
- uchar RM2[h*h];
- for(int i=0;i<h2;i++){
- RM1[i]=outd[i];
- RM2[i]=outd[i+h2];
- }
-
-
-
-
-
-
- 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]);
- }
-
- rc4keyperm(RM2, h2, rp, PboxRM, h2);
-
- for(int i=0;i<h2;i++){
- RM2[i]=RM1[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;
-}
