[Cipher_code.git] / Hash / rc4_hash.cpp

//gcc pixmap_io.c  -c 
//g++ -O3 one_round_hash_new.cpp pixmap_io.o  -o one_round_hash_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 <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 rc4key(uchar *key, uchar *sc, int h) {

  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%h] + key[j0%h] );
    uchar tmp = sc[i0];
    sc[i0] = sc[j0 ];
    sc[j0] = tmp;
  }
}






void prga(uchar *sc, uchar *block, int h, uchar *r) {
  uchar i0=0;
  uchar j0=0;

  for (int it=0; it<h; it++) {
    i0 = i0+1+block[(i0+1)%h];
    j0 = j0+1+block[(j0+1)%h];
    uchar tmp = sc[i0];
    sc[i0] = sc[j0];
    sc[j0] = tmp;
    r[it]=sc[(uchar)(sc[i0]+sc[j0])];
  }
}



void myhash(uchar* seq_in, int len, uchar* out, int h) {



  
  int ind2;

  uchar X[h];
  uchar out2[h];


  uchar S[256];

  
  rc4key(out, S, h);


/*  cout<<"S"<<endl;
  for(int i=0;i<256;i++)
    cout<<(int)S[i]<<" ";
  cout<<endl;
*/

/*  cout<<"out"<<endl;
  for(int i=0;i<h;i++)
    cout<<(int)out[i]<<" ";
  cout<<endl;

  cout<<len<<endl;
*/
  
  for(int it=0;it<len;it++) {
   
    ind2=it*h;

    // Mix with dynamic RM
    
    for(int a=0;a<h;a+=4) {
      X[a]=out[a]^seq_in[ind2+a];
      X[a+1]=out[a+1]^seq_in[ind2+a+1];
      X[a+2]=out[a+2]^seq_in[ind2+a+2];
      X[a+3]=out[a+3]^seq_in[ind2+a+3];
    }

    
/*    for(int i=0;i<h;i++)
      cout<<(int)X[i]<<endl;
*/
    
    prga(S, X, h, out2);



    
    for(int a=0;a<h;a+=4) {
      out[a]=S[out2[a]^X[a]];
      out[a+1]=S[out2[a+1]^X[a+1]];
      out[a+2]=S[out2[a+2]^X[a+2]];
      out[a+3]=S[out2[a+3]^X[a+3]];
    }
    
    

    
  }


}





  



  
int main(int argc, char** argv) {


  int h=16;
  int lena=0;
  int size_buf=1;
  int change=0;

  
  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
    if(strncmp(argv[i],"c",1)==0) change = atoi(&(argv[i][1]));          //Use Lena or buffer
  }


  cout<<size_buf<<endl;
      


  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 {
    imsize=size_buf;
    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;
  if(lena) {
    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];
    }
  }
  else {
    oneD=imsize;
    for(int i=0;i<oneD;i++) {
//      seq[i]=buffer[i];
      seq[i]=i+1;
    }
  }


  if(change==1) {
    
    seq[14]++;
  }
  if(change==2) {
    
    seq[15]++;
  }


  
  

  int total_len=imsize;
  int rp=1;
  int len= total_len/h;
  cout<<len<<endl;

  





  uchar res[h];
  for(int i=0;i<h;i++)
    res[i]=i+1;




  double time=0;

  cout<<"imsize "<<imsize<<endl;
/*
  cout<<"seq"<<endl;
  for(int i=0;i<imsize;i++){
    cout<<(int)seq[i]<<" ";
  }
  cout<<endl;
*/
  cout<<"res"<<endl;
  for(int i=0;i<h;i++){
    cout<<(int)res[i]<<" ";
  }
  cout<<endl;

  
  time=0;
  double t=TimeStart();
  for(int i=0;i<nb_test;i++)
  {

    myhash(seq, len,res,h);
  }


  
  
  time+=TimeStop(t);
  cout<<"Hash Time  "<<time<<endl;
  cout<<(double)imsize*nb_test/time<<"\t";

  for(int i=0;i<h;i++){
    cout<<(int)res[i]<<" ";
  }
  cout<<endl;

  

  return 0;
}