IDA/test_mat.cpp

   1 // g++ -std=c++11 -O4 -msse2 -msse3 -msse4 -fopenmp  -O3   test_mat.cpp -o test_mat   -I /home/couturie/tools/armadillo-6.200.5/include/ -lc   -lm -lpthread -lgfortran -DMAX_STACK_ALLOC=2048 -Wall -m64 -DF_INTERFACE_GFORT -fPIC -DSMP_SERVER -DNO_WARMUP -DMAX_CPU_NUMBER=8  -DNO_AFFINITY   -UCOMPLEX -DDOUBLE -I/home/couturie/tools/openblas/include -I/home/couturie/Downloads/OpenBLAS-0.2.15/  /home/couturie/tools/openblas/lib/libopenblas_haswellp-r0.2.15.a
   2
   3
   4
   5 #include <armadillo>
   6 #include <stdlib.h>
   7 #include <stdio.h>
   8
   9 #include <iostream>
  10 #include <fstream>
  11 #include <vector>
  12
  13
  14 #include <iostream>     // std::cout, std::fixed
  15 #include <iomanip>
  16 #include <math.h>
  17
  18 using namespace arma;
  19 using namespace std;
  20 int key_size=256;
  21
  22 typedef unsigned char byte;
  23
  24
  25 void rc4key(byte *key, byte *sc, int size_DK) {
  26
  27   for(int i=0;i<256;i++) {
  28     sc[i]=i;
  29   }
  30
  31
  32   byte j0 = 0;
  33   for(int i0=0; i0<256; i0++) {
  34     j0 = (j0 + sc[i0] + key[i0%size_DK] )&0xFF;
  35     byte tmp = sc[i0];
  36     sc[i0] = sc[j0 ];
  37     sc[j0] = tmp;
  38   }
  39 }
  40
  41
  42 void prga(byte *sc, int ldata, byte *r) {
  43   byte i0=0;
  44   byte j0=0;
  45
  46   for (int it=0; it<ldata; it++) {
  47     i0 = ((i0+1)&0xFE); //%255);
  48     j0 = (j0 + sc[i0])&0xFF;
  49     byte tmp = sc[i0];
  50     sc[i0] = sc[j0];
  51     sc[j0] = tmp;
  52     r[it]=sc[(sc[i0]+sc[j0])&0xFF]%254+1;
  53   }
  54 }
  55
  56
  57 void rc4keyperm(byte *key,int len, int rp,byte *sc, int size_DK) {
  58
  59   //sc=1:len;
  60
  61
  62
  63   for (int i=0;i<len;i++) {
  64     sc[i]=i;
  65   }
  66   for (int it = 0; it < rp; it++) {
  67     int j0 = 1;
  68     for(int i0 = 0; i0<len; i0++) {
  69       j0 = (j0 + sc[i0] + sc[j0] + key[i0%size_DK] )% len;
  70       int tmp = sc[i0];
  71       sc[i0] = sc[j0];
  72       sc[j0] = tmp;
  73     }
  74
  75   }
  76 }
  77
  78
  79 Mat<byte> readFile(int n, int k, int& sizeFile, int &lc) {
  80
  81   ifstream stream("/home/couturie/ajeter/lena.png", ios::in | ios::binary | ios::ate);
  82   sizeFile=stream.tellg();
  83   cout<<sizeFile<<endl;
  84   stream.seekg(0, ios::beg);
  85
  86
  87   lc=ceil(double(sizeFile)/k);
  88   cout<<lc<<endl;
  89
  90   vector<uint8_t> contents2(n*lc,0);
  91   vector<uint8_t> contents((istreambuf_iterator<char>(stream)), istreambuf_iterator<char>());
  92   copy ( contents.begin(), contents.end(), contents2.begin() );
  93
  94
  95
  96
  97   Mat<byte> matData(&contents2[0],1,contents2.size());
  98
  99   cout << "file size: " << contents2.size() << endl;
 100
 101   matData.reshape(n,lc);
 102   cout<<matData.n_rows<<" "<<matData.n_cols<<endl;
 103   return matData;
 104 }
 105
 106
 107 int main( int argc, char *argv[] ) {
 108
 109
 110   int q=2;
 111   int n=8;
 112   int k=4;
 113   int Tb=64;
 114   int l=10;//399*Tb;
 115
 116
 117
 118   int key_size=32;
 119   int seed=time(NULL);
 120   cout<<seed<<endl;
 121   srand48(seed);
 122
 123   byte Secretkey[key_size];
 124
 125   byte counter[key_size];
 126
 127   for(int i=0;i<key_size;i++) {
 128     Secretkey[i]=lrand48()&0xFF;
 129     counter[i]=lrand48()&0xFF;
 130   }
 131
 132
 133   byte DK[key_size];
 134   for (int i = 0; i < key_size ; i++) {
 135     DK[i]=Secretkey[i]^counter[i];
 136   }
 137
 138
 139   byte Sbox[256];
 140   byte RM1[q*n*n/4];
 141   byte sc[256];
 142
 143   rc4key(DK, Sbox, 8);
 144   prga(Sbox, q*n*n/4, RM1);
 145
 146
 147   byte PboxV[l/Tb];
 148
 149   rc4keyperm(&DK[8], l/Tb, 1, PboxV, 16);
 150
 151   cout<<"rprree"<<endl;
 152   Cube<byte> rm1(RM1,1,1,q*n*n/4);
 153   rm1.reshape(n/2,n/2,q);
 154
 155   cout<<"rm1"<<endl;
 156   cout<<rm1<<endl;
 157
 158
 159
 160
 161 // Initialization of IDA and inverse IDA matrix
 162   Cube<byte> IDAmat(n,n,q);
 163   Cube<short> invIDAmat(n,n,q);
 164
 165 //  Construction of the IDA matrices
 166   for(int i=0;i<q;i++) {
 167     Mat<byte> sub(n,n);
 168     sub.submat(0,0,n/2-1,n/2-1)=rm1.slice(i);
 169     sub.submat(0,n/2,n/2-1,n-1)=rm1.slice(i)+eye<Mat<byte>>(n/2,n/2);
 170     sub.submat(n/2,0,n-1,n/2-1)=rm1.slice(i)-eye<Mat<byte>>(n/2,n/2);
 171     sub.submat(n/2,n/2,n-1,n-1)=rm1.slice(i);
 172
 173 /*    cout<<"sub "<<i<<endl;
 174     cout<<sub<<endl;
 175
 176     cout<<"det"<<endl;
 177     mat toto=conv_to<mat>::from(sub);
 178     cout<<det(toto)<<endl;
 179 */
 180     IDAmat.slice(i)=sub;
 181
 182     Mat<short> sub2(n,n);
 183     Mat<short> rm2=conv_to<Mat<short>>::from(rm1.slice(i));
 184     sub2.submat(0,0,n/2-1,n/2-1)=rm2;
 185     sub2.submat(0,n/2,n/2-1,n-1)=-rm2-eye<Mat<short>>(n/2,n/2);
 186     sub2.submat(n/2,0,n-1,n/2-1)=-rm2+eye<Mat<short>>(n/2,n/2);
 187     sub2.submat(n/2,n/2,n-1,n-1)=rm2;
 188
 189     cout<<"det"<<endl;
 190     mat toto=conv_to<mat>::from(sub2);
 191     cout<<det(toto)<<endl;
 192
 193     invIDAmat.slice(i)=sub2;
 194
 195   }
 196   cout<<IDAmat<<endl;
 197
 198   cout<<invIDAmat<<endl;
 199   // rc4key(&DK[8], sc, 16);
 200
 201
 202
 203
 204
 205
 206   arma_rng::set_seed(time(NULL));
 207
 208
 209   int lc;
 210   int sizeFile;
 211   Mat<byte> M=readFile(n,k,sizeFile,lc);
 212
 213 //  Ma<byte> data;
 214 //  data.load("/home/couturie/ajeter/lena.png");
 215
 216   /* Mat<byte> M=conv_to<Mat<byte>>::from(randi(k,l));
 217   M=abs(M);
 218   M=M - floor(M/255)*255;*/
 219
 220 /*  cout<<"M"<<endl;
 221   cout<<M<<endl;
 222   cout<<"C"<<endl;
 223   cout<<C<<endl;
 224 */
 225
 226
 227   Mat<byte> C=IDAmat.slice(0)*M;
 228   M.save("M.bin",raw_binary);
 229   Mat<short> D=invIDAmat.slice(0)*C;
 230
 231
 232   Mat<char> D2=conv_to<Mat<char>>::from(D);
 233 //  cout<<"D2"<<endl;
 234 //  cout<<D2<<endl;
 235 //  D2.save("D2.bin",raw_binary);
 236   D2.reshape(1,n*lc);
 237   Row<char> d2=D2.row(0);
 238   byte* vec=(byte*)d2.memptr();
 239
 240   FILE* pFile = fopen ("myfile.png", "wb");
 241   fwrite (vec , sizeof(byte), sizeFile, pFile);
 242   fclose (pFile);
 243 }