OneRoundIoT/OneRound/rc4_hash2.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, uchar *X, int ldata, uchar *r, int h) {
 125   uchar i0=0;
 126   uchar j0=0;
 127
 128   for (int it=0; it<ldata; it++) {
 129     i0 = X[(i0+1)&(h-1)];
 130     j0 = (j0 + sc[i0]);
 131     uchar tmp = sc[i0];
 132     sc[i0] = sc[j0];
 133     sc[j0] = tmp;
 134     r[it]=sc[i0];//sc[(sc[i0]+sc[j0])&255];
 135   }
 136 }
 137
 138 inline uchar  circ(uchar x,int n) {return (x << n) | (x >> (8 - n));}
 139
 140
 141 uint64_t xorshift64( const uint64_t state)
 142 {
 143   uint64_t x = state;
 144   x^= x << 13;
 145   x^= x >> 7;
 146   x^= x << 17;
 147   return x;
 148 }
 149
 150
 151 static inline uint64_t splitmix64(uint64_t index) {
 152   uint64_t z = (index + UINT64_C(0x9E3779B97F4A7C15));
 153   z = (z ^ (z >> 30)) * UINT64_C(0xBF58476D1CE4E5B9);
 154   z = (z ^ (z >> 27)) * UINT64_C(0x94D049BB133111EB);
 155   return z ^ (z >> 31);
 156 }
 157
 158
 159
 160
 161
 162
 163
 164 //the proposed hash function, which is based on DSD structure. Sensitivity is ensured by employing the binary diffusion
 165
 166 void hash_DSD_BIN(uchar* seq_in, uchar* RM1,int len, uchar *S, int h) {
 167
 168
 169   // Goal: Calculate the hash value
 170   // Output: RM (hash value)
 171   uchar X[h];
 172   int ind1,ind2;
 173
 174
 175   uint64_t *rm=(uint64_t*)RM1;
 176   //  uint64_t *xx=(uint64_t*)X;
 177   uint64_t *ss=(uint64_t*)seq_in;
 178
 179
 180
 181   for(int it=0;it<len;it++) {
 182
 183     ind1=(it*h)>>3;
 184     // Mix with dynamic RM
 185     uint64_t sum=0;
 186
 187
 188     for(int a=0;a<(h>>3);a++) {
 189        rm[a]=rm[a]^ss[ind1+a];
 190        sum^=rm[a];
 191      }
 192
 193
 194
 195
 196      rm[0]=xorshift64(sum);
 197      for(int a=1;a<(h>>3);a++) {
 198        rm[a]=xorshift64(rm[a-1]);
 199        //rm[a]=splitmix64(rm[a-1]);
 200      }
 201
 202
 203   }
 204
 205
 206 }
 207
 208
 209
 210
 211
 212
 213
 214
 215
 216
 217 int main(int argc, char** argv) {
 218
 219
 220   int h=16;
 221   int lena=0;
 222   int size_buf=1;
 223   int change=0;
 224
 225
 226   for(int i=1; i<argc; i++){
 227     if(strncmp(argv[i],"nb",2)==0)    nb_test = atoi(&(argv[i][2]));    //nb of test
 228     if(strncmp(argv[i],"h",1)==0) h = atoi(&(argv[i][1]));          //size of block
 229     if(strncmp(argv[i],"sizebuf",7)==0) size_buf = atoi(&(argv[i][7]));          //SIZE of the buffer
 230     if(strncmp(argv[i],"lena",4)==0) lena = atoi(&(argv[i][4]));          //Use Lena or buffer
 231     if(strncmp(argv[i],"c",1)==0) change = atoi(&(argv[i][1]));          //Use Lena or buffer
 232   }
 233
 234
 235   cout<<size_buf<<endl;
 236   int seed=12;//time(NULL);
 237   cout<<seed<<endl;
 238   srand48(seed);
 239
 240   uchar Secretkey[key_size];
 241
 242   uchar counter[key_size];
 243
 244   for(int i=0;i<key_size;i++) {
 245     Secretkey[i]=lrand48()&0xFF;
 246     counter[i]=lrand48()&0xFF;
 247   }
 248
 249   int size = 64;
 250   uchar DK[size];
 251
 252
 253
 254
 255   int width;
 256   int height;
 257
 258   uchar *data_R, *data_G, *data_B;
 259   int imsize;
 260   uchar *buffer;
 261
 262   if(lena==1) {
 263     load_RGB_pixmap("lena.ppm", &width, &height, &data_R, &data_G, &data_B);
 264     imsize=width*height*3;
 265 //  load_RGB_pixmap("No_ecb_mode_picture.ppm", &width, &height, &data_R, &data_G, &data_B);
 266   }
 267   else {
 268     imsize=size_buf;
 269     buffer=new uchar[imsize];
 270     for(int i=0;i<imsize;i++) {
 271       buffer[i]=lrand48();
 272     }
 273   }
 274
 275
 276
 277
 278   uchar* seq= new uchar[imsize];
 279   uchar* seq2= new uchar[imsize];
 280
 281   int oneD;
 282   if(lena) {
 283     oneD=width*height;
 284     for(int i=0;i<oneD;i++) {
 285       seq[i]=data_R[i];
 286       seq[oneD+i]=data_G[i];
 287       seq[2*oneD+i]=data_B[i];
 288     }
 289   }
 290   else {
 291     oneD=imsize;
 292     for(int i=0;i<oneD;i++) {
 293       seq[i]=buffer[i];
 294     }
 295   }
 296
 297   printf("seq 4 %d\n",seq[4]);
 298   if(change==1) {
 299
 300     seq[4]++;
 301     seq[5]--;
 302   }
 303   if(change==2) {
 304
 305     seq[9]++;
 306     seq[10]--;
 307   }
 308
 309   printf("seq 4 %d\n",seq[4]);
 310
 311
 312
 313
 314   int total_len=imsize;
 315   int rp=1;
 316   int len= total_len/h;
 317   cout<<len<<endl;
 318
 319
 320   uchar *mix=new uchar[256];
 321
 322
 323
 324
 325   for (int i = 0; i < 256 ; i++) {
 326     mix[i]=Secretkey[i]^counter[i];
 327   }
 328
 329
 330 //  cout<<"hash "<<endl;
 331   for (int i = 0; i < 64 ; i++) {
 332 //    DK[i]=digest[i];
 333     DK[i]=mix[i];
 334     //cout<<(int)DK[i]<<" ";
 335   }
 336   //cout<<endl;
 337
 338
 339
 340
 341
 342   uchar Sbox1[256];
 343   uchar sc[256];
 344   uchar RM1[h];
 345
 346
 347
 348   double time=0;
 349   double t=TimeStart();
 350   rc4key(DK, Sbox1, 8);
 351
 352   rc4key(&DK[16], sc, 8);
 353
 354
 355
 356
 357   prga(sc, sc,h,RM1,h);
 358
 359
 360
 361
 362
 363   time+=TimeStop(t);
 364   cout<<"Time initializaton "<<time<<endl;
 365
 366
 367
 368   cout<<"imsize "<<imsize<<endl;
 369
 370 /*  for(int i=0;i<imsize;i++){
 371     cout<<(int)seq[i]<<" ";
 372   }
 373   cout<<endl;
 374 */
 375
 376
 377
 378   time=0;
 379   t=TimeStart();
 380   for(int i=0;i<nb_test;i++)
 381   {
 382     hash_DSD_BIN(seq, RM1,len,Sbox1,h);
 383   }
 384
 385
 386
 387
 388   time+=TimeStop(t);
 389   cout<<"Hash Time  "<<time<<endl;
 390   cout<<(double)imsize*nb_test/time<<"\t";
 391
 392   for(int i=0;i<h;i++){
 393     cout<<(int)RM1[i]<<" ";
 394   }
 395   cout<<endl;
 396
 397   //  cout<<splitmix64(2490)<<endl;
 398   // cout<<splitmix64(2489)<<endl;
 399
 400
 401   return 0;
 402 }