X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/Cipher_code.git/blobdiff_plain/70b995d30a4c75869c4223dbb931802246f52d94..149268e31dcc62cf18eddd1b8ffa37cf81acbbca:/IDA/test_mat2.cpp

diff --git a/IDA/test_mat2.cpp b/IDA/test_mat2.cpp
index e09acbd..474c572 100644
--- a/IDA/test_mat2.cpp
+++ b/IDA/test_mat2.cpp
@@ -9,8 +9,6 @@
 #include <iostream>
 #include <fstream>
 #include <vector>
-
-
 #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);
@@ -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) {
 
@@ -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);
   
-  cout<<"rm1"<<endl;
+  /* cout<<"rm1"<<endl;
   cout<<rm1<<endl;
-
+  */
 
   
 
@@ -147,18 +176,18 @@ 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;
 
-    cout<<"det"<<endl;
+    /*cout<<"det"<<endl;
     mat toto=conv_to<mat>::from(sub2);
     cout<<det(toto)<<endl;
-
+    */
     invIDAmat.slice(i)=sub2;
 
   }
-  cout<<IDAmat<<endl;
-  cout<<invIDAmat<<endl;
+  //cout<<IDAmat<<endl;
+  //cout<<invIDAmat<<endl;
 }
 
-void buildKRankIDA(Cube<byte>& IDAmat, int q, int n, int k, byte* DK) {
+void buildPartialRankIDA(Cube<byte>& IDAmat, int q, int n, int k, byte* DK) {
 
   byte Sbox[256];
   byte RM1[q*n*k];
@@ -168,7 +197,8 @@ void buildKRankIDA(Cube<byte>& IDAmat, int q, int n, int k, byte* DK) {
   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;
 
@@ -184,8 +214,13 @@ void buildKRankIDA(Cube<byte>& IDAmat, int q, int n, int k, byte* DK) {
     IDAmat.slice(i)=sub;
 
 
+    cout<<"det "<<i<<endl;
+    mat toto=conv_to<mat>::from(sub);
+    cout<<det(toto)<<endl;
+
   }
-  cout<<IDAmat<<endl;
+*/
+  // cout<<IDAmat<<endl;
 
 }
 
@@ -194,8 +229,11 @@ void buildKRankIDA(Cube<byte>& IDAmat, int q, int n, int k, byte* DK) {
 int main( int argc, char *argv[] ) {
 
 
+
+  int full=0;
+  
   int q=2;
-  int n=4;
+  int n=8;
   int k=4;
   int Tb=64;
   int l=10;//399*Tb;
@@ -235,31 +273,42 @@ 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]);
-*/
-
 
 
-  Cube<byte> IDAmat(n,k,q);
-  buildKRankIDA(Cube<byte>& IDAmat, q, n, k, &DK[0]);
+  int lc;
+  int sizeFile;
+  Mat<byte> M;
+  if(full)
+    M=readFullFile(n,k,sizeFile,lc);
+  else
+    M=readPartialFile(n,k,sizeFile,lc);
 
-  // rc4key(&DK[8], sc, 16);
-  
+  cout<<"M "<<endl;
+  cout<<M<<endl;
   
 
-  
- 
- 
   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");
@@ -274,20 +323,54 @@ int main( int argc, char *argv[] ) {
   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);
-//  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 {
 
-  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));
+    IDAmat.resize(k,k,q);
+    invIDAmat.resize(k,k,q);
+    buildFullRankIDA(IDAmat, invIDAmat, q, k, &DK[0]);
 
-  mat D=solve(A,B);
+    
+    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<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);