From 1bae5007d1fabd3147e642352fac765684a7aad6 Mon Sep 17 00:00:00 2001
From: couturie <couturie@extinction>
Date: Thu, 7 Sep 2017 20:38:59 +0200
Subject: [PATCH] new

---
 OneRoundIoT/OneRound/one_round_par2.cpp | 45 ++++++++++++++++---------
 1 file changed, 29 insertions(+), 16 deletions(-)

diff --git a/OneRoundIoT/OneRound/one_round_par2.cpp b/OneRoundIoT/OneRound/one_round_par2.cpp
index f579622..efa6969 100644
--- a/OneRoundIoT/OneRound/one_round_par2.cpp
+++ b/OneRoundIoT/OneRound/one_round_par2.cpp
@@ -167,7 +167,7 @@ void encrypt_ctr(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, in
     uchar X[h2];
 
     for(int a=0;a<h2;a++) {
-      X[a]=Sbox1[a&0xFF];           //Warning according to the size of h2, we can be outsize of Sbox1[a]
+      X[a]=Sbox1[(a+10*id)&0xFF];           //Warning according to the size of h2, we can be outsize of Sbox1[a]
     }
 
 
@@ -605,12 +605,6 @@ int main(int argc, char** argv) {
   int num=omp_get_max_threads();
   cout<<"num "<<num<<endl;
   
-  uchar RM1[num*(h * h)];
-  uchar RM2[num*(h * h)];
-  prga(sc, num*(h * h), RM1);
-  for(int i=0;i<num*h2;i++) {
-    RM2[i]=RM1[i];
-  }
   
 
 
@@ -618,10 +612,30 @@ int main(int argc, char** argv) {
   
     
   
-  uchar keyp[16];
-  for (int i = 48; i < 64; i++)
-    keyp[i-48] = DK[i];
 
+
+  uchar RM1[num*(h * h)];
+  uchar RM2[num*(h * h)];
+  /*for(int i=0;i<num;i++) {
+    
+    rc4key(&DK[48+i*16], sc, 16);
+    prga(sc, h2, &RM1[h2*i]);
+    for(int a=0;a<h2;a++) {
+      cout<<(int)RM1[h2*i+a]<<" ";
+    }
+    cout<<endl<<endl;
+  }*/
+
+  rc4key(&DK[48], sc, 16);
+  prga(sc, h2*num, RM1);
+
+  rc4key(&DK[64], sc, 16);
+  prga(sc, h2, RM2);
+
+
+
+
+  
 //  cout<<len<<endl;
   int *Pbox=new int[len];
 
@@ -629,15 +643,14 @@ int main(int argc, char** argv) {
   
   int *PboxRM=new int[h2];
 
-  rc4keyperm(keyp, len, rp, Pbox, 16);
+  rc4keyperm(&DK[48+16*num], len, rp, Pbox, 16);
 
-//  printf("len %d\n",len);
-  for(int i=0;i<len;i++) {
-//    printf("%d \n",Pbox[i]);
-  }
-  
 
+  rc4keyperm(RM2, h2, rp, PboxRM, h2);
 
+  for(int i=0;i<num*h2;i++) {
+    RM2[i]=RM1[i];
+  }
  
   double time=0;
   double t=TimeStart();
-- 
2.39.5