X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/Cipher_code.git/blobdiff_plain/4fa9c3d0a87911ef71653cb99aa1c70f633a9473..ca940d4b9b6ef86858b8c2e8dcffce5602cbbbae:/OneRoundIoT/OneRound/one_round_new.cpp

diff --git a/OneRoundIoT/OneRound/one_round_new.cpp b/OneRoundIoT/OneRound/one_round_new.cpp
index 59210a8..de1b0d2 100644
--- a/OneRoundIoT/OneRound/one_round_new.cpp
+++ b/OneRoundIoT/OneRound/one_round_new.cpp
@@ -30,9 +30,10 @@ using namespace std;
 int key_size=256;
 int nb_test=1;
 int ctr=0;
+int h=64;
 
-const int h=16;
-const int h2=h*h;
+
+int h2=h*h;
 
 
 
@@ -326,10 +327,14 @@ int main(int argc, char** argv) {
   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]));          //CTR ? 1  otherwise CBC like
   }
 
   printf("nb times %d\n",nb_test);
   printf("ctr %d\n",ctr);
+  printf("h %d\n",h);
+
+  h2=h*h;
   
 /*
   if(argc==2)
@@ -468,8 +473,11 @@ int main(int argc, char** argv) {
   int i;
   for(i=0;i<nb_test;i++)
   {
-//    encrypt(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
-    encrypt_ctr(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1);
+    if(ctr)
+      encrypt_ctr(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,1);
+    else
+      encrypt(seq, seq2,len,RM1,Pbox,PboxRM,Sbox1,Sbox2,0);
+
   }
   
   time+=TimeStop(t);
@@ -487,8 +495,11 @@ int main(int argc, char** argv) {
   time=0;
   t=TimeStart();
   for(i=0;i<nb_test;i++) {
-//    decrypt(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
-    encrypt_ctr(seq2, seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+    if(ctr)
+      encrypt_ctr(seq2, seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+    else
+      decrypt(seq2,seq,len,RM2,Pbox,PboxRM,Sbox1,Sbox2,0);
+
   }
 
   time+=TimeStop(t);