OneRoundIoT/OneRound/one_round_hash_new.cpp

   1 //gcc pixmap_io.c  -c
   2 //g++ -O3 one_round_hash_new.cpp pixmap_io.o  -o one_round_hash_new -std=c++11
   3
   4 //
   5
   6
   7 #include <iostream>
   8 #include <list>
   9 #include<math.h>
  10 #include<stdlib.h>
  11 #include<stdio.h>
  12 #include<string.h>
  13 #include <fstream>
  14 #include <sys/time.h>
  15
  16 /*#include <cryptopp/hex.h>
  17 #include <cryptopp/sha.h>
  18 #include <cryptopp/osrng.h>
  19 #include <cryptopp/secblock.h>
  20 */
  21
  22
  23 extern "C" {
  24   int load_RGB_pixmap(char *filename, int *width, int *height, unsigned char**R_data, unsigned char**G_data, unsigned char**B_data);
  25   void store_RGB_pixmap(char *filename, unsigned char *R_data, unsigned char *G_data, unsigned char *B_data, int width, int height);
  26 }
  27
  28
  29 //using namespace CryptoPP;
  30 using namespace std;
  31
  32
  33 int key_size=256;
  34 int nb_test=1;
  35 int ctr=0;
  36
  37
  38
  39
  40
  41
  42
  43 typedef unsigned char   uchar;
  44
  45
  46 double TimeStart()
  47 {
  48   struct timeval tstart;
  49   gettimeofday(&tstart,0);
  50   return( (double) (tstart.tv_sec + tstart.tv_usec*1e-6) );
  51 }
  52
  53 double TimeStop(double t)
  54 {
  55   struct timeval tend;
  56
  57   gettimeofday(&tend,0);
  58   t = (double) (tend.tv_sec + tend.tv_usec*1e-6) - t;
  59   return (t);
  60 }
  61
  62
  63
  64
  65
  66
  67 void inverse_tables(uchar *tab, int size_tab,uchar *inv_perm_tabs) {
  68
  69   for(int i=0;i<size_tab;i++) {
  70     inv_perm_tabs[tab[i]] = i;
  71   }
  72
  73 }
  74
  75 void inverse_tables_int(int *tab, int size_tab,int *inv_perm_tabs) {
  76
  77   for(int i=0;i<size_tab;i++) {
  78     inv_perm_tabs[tab[i]] = i;
  79   }
  80
  81 }
  82
  83
  84
  85 void rc4key(uchar *key, uchar *sc, int size_DK) {
  86
  87   for(int i=0;i<256;i++) {
  88     sc[i]=i;
  89   }
  90
  91
  92   uchar j0 = 0;
  93   for(int i0=0; i0<256; i0++) {
  94     j0 = (j0 + sc[i0] + key[i0%size_DK] )&0xFF;
  95     uchar tmp = sc[i0];
  96     sc[i0] = sc[j0 ];
  97     sc[j0] = tmp;
  98   }
  99 }
 100
 101
 102
 103 void rc4keyperm(uchar *key,int len, int rp,int *sc, int size_DK) {
 104
 105   //sc=1:len;
 106
 107
 108
 109   for (int i=0;i<len;i++) {
 110     sc[i]=i;
 111   }
 112   for (int it = 0; it < rp; it++) {
 113     int j0 = 1;
 114     for(int i0 = 0; i0<len; i0++) {
 115       j0 = (j0 + sc[i0] + sc[j0] + key[i0%size_DK] )% len;
 116       int tmp = sc[i0];
 117       sc[i0] = sc[j0];
 118       sc[j0] = tmp;
 119     }
 120
 121   }
 122 }
 123
 124 void prga(uchar *sc, int ldata, uchar *r) {
 125   uchar i0=0;
 126   uchar j0=0;
 127
 128   for (int it=0; it<ldata; it++) {
 129     i0 = ((i0+1)&0xFE); //%255);
 130     j0 = (j0 + sc[i0])&0xFF;
 131     uchar tmp = sc[i0];
 132     sc[i0] = sc[j0];
 133     sc[j0] = tmp;
 134     r[it]=sc[(sc[i0]+sc[j0])&0xFF];
 135   }
 136 }
 137
 138 inline uchar  circ(uchar x,int n) {return (x << n) | (x >> (8 - n));}
 139
 140
 141 //the proposed hash function, which is based on DSD structure. Sensitivity is ensured by employing the binary diffusion
 142
 143 void hash_DSD_BIN(uchar* seq_in, uchar* RM1,int len, int *PboxRM, uchar *Sbox1, uchar *Sbox2, int h) {
 144
 145
 146   // Goal: Calculate the hash value
 147   // Output: RM (hash value)
 148
 149 //  uchar *X=new uchar[h2];
 150 //  uchar *fX=new uchar[h2];
 151   uchar X[h];
 152   int ind2;
 153
 154   uchar Y[h];
 155   uchar Z[h];
 156
 157
 158   for(int it=0;it<len;it++) {
 159
 160     ind2=it*h;
 161
 162     // Mix with dynamic RM
 163
 164     for(int a=0;a<h;a+=4) {
 165       X[a]=RM1[a]^seq_in[ind2+a];
 166       X[a+1]=RM1[a+1]^seq_in[ind2+a+1];
 167       X[a+2]=RM1[a+2]^seq_in[ind2+a+2];
 168       X[a+3]=RM1[a+3]^seq_in[ind2+a+3];
 169     }
 170
 171 /*    Y[0]=Sbox1[X[0]^X[h-1]];
 172     for(int a=1;a<h;a++) {
 173       Y[a]=Sbox1[Y[a-1]^X[a-1]];
 174     }
 175
 176     Z[h-1]=Sbox2[Y[h-1]^Y[0]];
 177     for(int a=h-1;a>0;a--) {
 178       Z[a-1]=Sbox2[Z[a]^Y[a]];
 179     }
 180 */
 181
 182     Y[0]=X[0]^X[h-1];
 183     for(int a=1;a<h;a++) {
 184       Y[a]=Y[a-1]^X[a-1];
 185     }
 186
 187     for(int a=0;a<h;a+=4) {
 188       Y[a]=Sbox1[Y[a]];
 189       Y[a+1]=Sbox1[Y[a+1]];
 190       Y[a+2]=Sbox1[Y[a+2]];
 191       Y[a+3]=Sbox1[Y[a+3]];
 192     }
 193
 194
 195
 196
 197
 198     Z[h-1]=Y[h-1]^Y[0];
 199     for(int a=h-1;a>0;a--) {
 200       Z[a-1]=Z[a]^Y[a];
 201     }
 202
 203     /*  for(int a=0;a<h;a+=4) {
 204       Z[a]=Sbox2[Z[a]];
 205       Z[a+1]=Sbox2[Z[a+1]];
 206       Z[a+2]=Sbox2[Z[a+2]];
 207       Z[a+3]=Sbox2[Z[a+3]];
 208
 209       }*/
 210
 211
 212
 213     for(int a=0;a<h;a+=4) {
 214       RM1[a]=Z[a];
 215       RM1[a+1]=Z[a+1];
 216       RM1[a+2]=Z[a+2];
 217       RM1[a+3]=Z[a+3];
 218
 219 /*
 220       RM1[a]=Z[a]^RM1[PboxRM[a]];
 221       RM1[a+1]=Z[a+1]^RM1[PboxRM[a+1]];
 222       RM1[a+2]=Z[a+2]^RM1[PboxRM[a+2]];
 223       RM1[a+3]=Z[a+3]^RM1[PboxRM[a+3]];
 224 */
 225     }
 226
 227   }
 228
 229
 230 }
 231
 232
 233
 234
 235
 236
 237
 238
 239
 240
 241 int main(int argc, char** argv) {
 242
 243
 244   int h=16;
 245   int lena=0;
 246   int size_buf=1;
 247   int change=0;
 248
 249
 250   for(int i=1; i<argc; i++){
 251     if(strncmp(argv[i],"nb",2)==0)    nb_test = atoi(&(argv[i][2]));    //nb of test
 252     if(strncmp(argv[i],"ctr",3)==0) ctr = atoi(&(argv[i][3]));          //CTR ? 1  otherwise CBC like
 253     if(strncmp(argv[i],"h",1)==0) h = atoi(&(argv[i][1]));          //size of block
 254     if(strncmp(argv[i],"sizebuf",7)==0) size_buf = atoi(&(argv[i][7]));          //SIZE of the buffer
 255     if(strncmp(argv[i],"lena",4)==0) lena = atoi(&(argv[i][4]));          //Use Lena or buffer
 256     if(strncmp(argv[i],"c",1)==0) change = atoi(&(argv[i][1]));          //Use Lena or buffer
 257   }
 258
 259
 260   cout<<size_buf<<endl;
 261
 262   int seed=time(NULL);
 263 //  cout<<seed<<endl;
 264   srand48(12);
 265
 266   uchar Secretkey[key_size];
 267
 268   uchar counter[key_size];
 269
 270   for(int i=0;i<key_size;i++) {
 271     Secretkey[i]=lrand48()&0xFF;
 272     counter[i]=lrand48()&0xFF;
 273   }
 274
 275
 276   int size = 64;
 277   uchar DK[size];
 278
 279
 280
 281
 282   int width;
 283   int height;
 284
 285   uchar *data_R, *data_G, *data_B;
 286   int imsize;
 287   uchar *buffer;
 288
 289   if(lena==1) {
 290     load_RGB_pixmap("lena.ppm", &width, &height, &data_R, &data_G, &data_B);
 291     imsize=width*height*3;
 292 //  load_RGB_pixmap("No_ecb_mode_picture.ppm", &width, &height, &data_R, &data_G, &data_B);
 293   }
 294   else {
 295     imsize=size_buf;
 296     buffer=new uchar[imsize];
 297     for(int i=0;i<imsize;i++) {
 298       buffer[i]=lrand48();
 299     }
 300   }
 301
 302
 303
 304
 305   uchar* seq= new uchar[imsize];
 306   uchar* seq2= new uchar[imsize];
 307
 308   int oneD;
 309   if(lena) {
 310     oneD=width*height;
 311     for(int i=0;i<oneD;i++) {
 312       seq[i]=data_R[i];
 313       seq[oneD+i]=data_G[i];
 314       seq[2*oneD+i]=data_B[i];
 315     }
 316   }
 317   else {
 318     oneD=imsize;
 319     for(int i=0;i<oneD;i++) {
 320       seq[i]=buffer[i];
 321     }
 322   }
 323
 324
 325   if(change==1) {
 326
 327     seq[4]++;
 328   }
 329   if(change==2) {
 330
 331     seq[9]++;
 332   }
 333
 334
 335
 336
 337
 338   int total_len=imsize;
 339   int rp=1;
 340   int len= total_len/h;
 341   cout<<len<<endl;
 342
 343
 344   uchar *mix=new uchar[256];
 345
 346
 347
 348
 349   for (int i = 0; i < 256 ; i++) {
 350     mix[i]=Secretkey[i]^counter[i];
 351   }
 352
 353
 354 //  cout<<"hash "<<endl;
 355   for (int i = 0; i < 64 ; i++) {
 356 //    DK[i]=digest[i];
 357     DK[i]=mix[i];
 358   }
 359
 360
 361
 362
 363   int *PboxRM=new int[h];
 364   uchar Sbox1[256];
 365   uchar Sbox2[256];
 366   uchar sc[256];
 367   uchar RM1[h];
 368   uchar RM2[h];
 369
 370
 371
 372
 373   double time=0;
 374   double t=TimeStart();
 375   rc4key(DK, Sbox1, 8);
 376   rc4key(&DK[8], Sbox2, 8);
 377
 378   rc4key(&DK[16], sc, 8);
 379
 380
 381   prga(sc, h, RM1);
 382
 383
 384
 385   rc4keyperm(&DK[24], h, rp, PboxRM, 8);
 386
 387   time+=TimeStop(t);
 388   cout<<"Time initializaton "<<time<<endl;
 389
 390
 391
 392
 393
 394
 395
 396
 397
 398
 399
 400   for(int i=0;i<h;i++){
 401     RM2[i]=RM1[i];
 402   }
 403
 404   cout<<"imsize "<<imsize<<endl;
 405
 406 /*  for(int i=0;i<imsize;i++){
 407     cout<<(int)seq[i]<<" ";
 408   }
 409   cout<<endl;
 410 */
 411
 412   time=0;
 413   t=TimeStart();
 414   for(int i=0;i<nb_test;i++)
 415   {
 416     hash_DSD_BIN(seq, RM1,len,PboxRM,Sbox1,Sbox2,h);
 417   }
 418
 419
 420
 421
 422   time+=TimeStop(t);
 423   cout<<"Hash Time  "<<time<<endl;
 424   cout<<(double)imsize*nb_test/time<<"\t";
 425
 426   for(int i=0;i<h;i++){
 427     cout<<(int)RM1[i]<<" ";
 428   }
 429   cout<<endl;
 430
 431
 432
 433   return 0;
 434 }