udpate oe round

[Cipher_code.git] / IDA / test_mat2.cpp

diff --git a/IDA/test_mat2.cpp b/IDA/test_mat2.cpp
index e61ecacebc5358c43e40b146d2bfbd0f584bce60..474c572b87ad0ade2a78ea09562cc6b867c6030a 100644 (file)
--- a/IDA/test_mat2.cpp
+++ b/IDA/test_mat2.cpp
@@ -9,8 +9,6 @@
 #include <iostream>
 #include <fstream>
 #include <vector>
 #include <iostream>
 #include <fstream>
 #include <vector>

-
-

 #include <iostream>     // std::cout, std::fixed
 #include <iomanip>
 #include <math.h>
 #include <iostream>     // std::cout, std::fixed
 #include <iomanip>
 #include <math.h>


@@ -76,10 +74,10 @@ void rc4keyperm(byte *key,int len, int rp,byte *sc, int size_DK) {
 }
 
 
 }
 
 

-Mat<byte> readFile(int n, int k, int& sizeFile, int &lc) {
+Mat<byte> readFullFile(int n, int k, int& sizeFile, int &lc) {

 
 

-  ifstream stream("/home/couturie/ajeter/lena.png", ios::in | ios::binary | ios::ate);
-//  ifstream stream("/home/couturie/ajeter/lena_small.png", ios::in | ios::binary | ios::ate);
+//  ifstream stream("lena.png", ios::in | ios::binary | ios::ate);
+  ifstream stream("lena_small2.png", ios::in | ios::binary | ios::ate);

   sizeFile=stream.tellg();
   cout<<sizeFile<<endl;
   stream.seekg(0, ios::beg);
   sizeFile=stream.tellg();
   cout<<sizeFile<<endl;
   stream.seekg(0, ios::beg);


@@ -105,6 +103,37 @@ Mat<byte> readFile(int n, int k, int& sizeFile, int &lc) {
 }
 
 
 }
 
 

+Mat<byte> readPartialFile(int n, int k, int& sizeFile, int &lc) {
+
+//  ifstream stream("lena.png", ios::in | ios::binary | ios::ate);
+  ifstream stream("lena_small2.png", ios::in | ios::binary | ios::ate);
+  sizeFile=stream.tellg();
+  cout<<sizeFile<<endl;
+  stream.seekg(0, ios::beg);
+
+  
+  lc=ceil(double(sizeFile)/k);
+  cout<<lc<<endl;
+
+  vector<uint8_t> contents2(k*lc,0);
+  vector<uint8_t> contents((istreambuf_iterator<char>(stream)), istreambuf_iterator<char>());
+  copy ( contents.begin(), contents.end(), contents2.begin() );
+
+  
+
+
+  Mat<byte> matData(&contents2[0],1,contents2.size());
+
+  cout << "file size: " << contents2.size() << endl;
+
+  matData.reshape(k,lc);
+  cout<<matData.n_rows<<" "<<matData.n_cols<<endl;
+  return matData;
+}
+
+
+
+

 
 void buildFullRankIDA(Cube<byte>& IDAmat, Cube<short> &invIDAmat, int q, int n, byte* DK) {
 
 
 void buildFullRankIDA(Cube<byte>& IDAmat, Cube<short> &invIDAmat, int q, int n, byte* DK) {
 


@@ -117,9 +146,9 @@ void buildFullRankIDA(Cube<byte>& IDAmat, Cube<short> &invIDAmat, int q, int n,
   Cube<byte> rm1(RM1,1,1,q*n*n/4);
   rm1.reshape(n/2,n/2,q);
   
   Cube<byte> rm1(RM1,1,1,q*n*n/4);
   rm1.reshape(n/2,n/2,q);
   

-  cout<<"rm1"<<endl;
+  /* cout<<"rm1"<<endl;

   cout<<rm1<<endl;
   cout<<rm1<<endl;

-
+  */

 
   
 
 
   
 


@@ -147,23 +176,64 @@ void buildFullRankIDA(Cube<byte>& IDAmat, Cube<short> &invIDAmat, int q, int n,
     sub2.submat(n/2,0,n-1,n/2-1)=-rm2+eye<Mat<short>>(n/2,n/2);
     sub2.submat(n/2,n/2,n-1,n-1)=rm2;
 
     sub2.submat(n/2,0,n-1,n/2-1)=-rm2+eye<Mat<short>>(n/2,n/2);
     sub2.submat(n/2,n/2,n-1,n-1)=rm2;
 

-    cout<<"det"<<endl;
+    /*cout<<"det"<<endl;

     mat toto=conv_to<mat>::from(sub2);
     cout<<det(toto)<<endl;
     mat toto=conv_to<mat>::from(sub2);
     cout<<det(toto)<<endl;

-
+    */

     invIDAmat.slice(i)=sub2;
 
   }
     invIDAmat.slice(i)=sub2;
 
   }

-  cout<<IDAmat<<endl;
-  cout<<invIDAmat<<endl;
+  //cout<<IDAmat<<endl;
+  //cout<<invIDAmat<<endl;
+}
+
+void buildPartialRankIDA(Cube<byte>& IDAmat, int q, int n, int k, byte* DK) {
+
+  byte Sbox[256];
+  byte RM1[q*n*k];
+ 
+  
+  rc4key(DK, Sbox, 8);
+  prga(Sbox, q*n*k, RM1);
+  Cube<byte> rm1(RM1,1,1,q*n*k);
+  rm1.reshape(n,k,q);
+  IDAmat=rm1;
+/*    
+  cout<<"rm1"<<endl;
+  cout<<rm1<<endl;
+
+
+  
+
+//  Construction of the IDA matrices
+  for(int i=0;i<q;i++) {
+    Mat<byte> sub(n,k);
+    sub.submat(0,0,n-1,k-1)=rm1.slice(i);
+ 
+
+    IDAmat.slice(i)=sub;
+
+
+    cout<<"det "<<i<<endl;
+    mat toto=conv_to<mat>::from(sub);
+    cout<<det(toto)<<endl;
+
+  }
+*/
+  // cout<<IDAmat<<endl;
+

 }
 
 
 }
 
 

+

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

+
+  int full=0;
+  

   int q=2;
   int q=2;

-  int n=4;
+  int n=8;

   int k=4;
   int Tb=64;
   int l=10;//399*Tb;
   int k=4;
   int Tb=64;
   int l=10;//399*Tb;


@@ -204,25 +274,41 @@ int main( int argc, char *argv[] ) {
 
 
 
 
 
 

-// Initialization of IDA and inverse IDA matrix
-  Cube<byte> IDAmat(n,n,q);
-  Cube<short> invIDAmat(n,n,q);
-  buildFullRankIDA(IDAmat, invIDAmat, q, n, &DK[0]);
-  

 
 

-  // rc4key(&DK[8], sc, 16);
-  
-  
+  int lc;
+  int sizeFile;
+  Mat<byte> M;
+  if(full)
+    M=readFullFile(n,k,sizeFile,lc);
+  else
+    M=readPartialFile(n,k,sizeFile,lc);

 
 

+  cout<<"M "<<endl;
+  cout<<M<<endl;

   
   

- 
- 
+

   arma_rng::set_seed(time(NULL));
 
   arma_rng::set_seed(time(NULL));
 

+// Initialization of IDA and inverse IDA matrix
+
+  Cube<byte> IDAmat;
+  Cube<short> invIDAmat;
+
+
+  if(full)
+  {
+    IDAmat.resize(n,n,q);
+    invIDAmat.resize(n,n,q);
+    buildFullRankIDA(IDAmat, invIDAmat, q, n, &DK[0]);
+  }
+  else {
+    IDAmat.resize(n,k,q);
+    buildPartialRankIDA(IDAmat, q, n, k, &DK[0]);
+  }
+
+
+

 
 

-  int lc;
-  int sizeFile;
-  Mat<byte> M=readFile(n,k,sizeFile,lc);

 
 //  Ma<byte> data;
 //  data.load("/home/couturie/ajeter/lena.png");
 
 //  Ma<byte> data;
 //  data.load("/home/couturie/ajeter/lena.png");


@@ -237,20 +323,54 @@ int main( int argc, char *argv[] ) {
   cout<<C<<endl;
 */
 
   cout<<C<<endl;
 */
 

+
+  Mat<byte> C;
+  Mat<char> D2;
+  //full
+//  C=IDAmat.slice(0)*M;
+
+//partial
+  C=IDAmat.slice(0).submat(0,0,n-1,k-1)*M;

   
   

-  Mat<byte> C=IDAmat.slice(0)*M;

   M.save("M.bin",raw_binary);
   M.save("M.bin",raw_binary);

-//  Mat<short> D=invIDAmat.slice(0)*C;
+  
+  if(full) {
+    Mat<short> D=invIDAmat.slice(0)*C;
+
+    
+    /*mat A=conv_to<mat>::from(IDAmat.slice(0).submat(0,0,n-1,n-1));
+    mat B=conv_to<mat>::from(C.submat(0,0,n-1,lc-1));
+    mat D=solve(A,B);
+    */
+    
+
+    D2=conv_to<Mat<char>>::from(D);
+    cout<<D2<<endl;
+  }
+  else {

 
 

+    IDAmat.resize(k,k,q);
+    invIDAmat.resize(k,k,q);
+    buildFullRankIDA(IDAmat, invIDAmat, q, k, &DK[0]);

 
 

-  mat A=conv_to<mat>::from(IDAmat.slice(0).submat(0,0,k-1,n-1));
-  mat B=conv_to<mat>::from(C.submat(0,0,k-1,lc-1));
+    
+    mat A=conv_to<mat>::from(IDAmat.slice(0).submat(0,0,k-1,k-1));
+
+    mat B=conv_to<mat>::from(C.submat(0,0,k-1,lc-1));
+    mat D=solve(A,B);
+//    mat D=inv(A)*B;
+    
+//    cout<<D<<endl;

 
 

-  mat D=solve(A,B);
+//    Mat<short> D=invIDAmat.slice(0)*C.submat(0,0,k-1,lc-1);
+    
+    D2=conv_to<Mat<char>>::from(D);
+    cout<<D2<<endl;
+  }

 
   
 
 
   
 

-  Mat<char> D2=conv_to<Mat<char>>::from(D);
+

 //  cout<<"D2"<<endl;
 //  cout<<D2<<endl;
 //  D2.save("D2.bin",raw_binary);
 //  cout<<"D2"<<endl;
 //  cout<<D2<<endl;
 //  D2.save("D2.bin",raw_binary);