From f6a8395dc6bbfc631a9f53b35fe3970a3749a48a Mon Sep 17 00:00:00 2001
From: couturie <couturie@extinction>
Date: Sat, 19 Aug 2017 08:06:30 +0200
Subject: [PATCH] new

---
 OneRoundIoT/OneRound/one_round_new.cpp | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/OneRoundIoT/OneRound/one_round_new.cpp b/OneRoundIoT/OneRound/one_round_new.cpp
index 9046044..9066eaf 100644
--- a/OneRoundIoT/OneRound/one_round_new.cpp
+++ b/OneRoundIoT/OneRound/one_round_new.cpp
@@ -145,7 +145,7 @@ void encrypt_ctr(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, in
 
   
    for(int a=0;a<h2;a++) {
-     X[a]=Sbox2[a];
+     X[a]=Sbox1[a&0xFF];           //Warning according to the size of h2, we can be outsize of Sbox1[a]
    }
 
    
@@ -161,14 +161,29 @@ void encrypt_ctr(uchar* seq_in, uchar *seq_out, int len,uchar* RM1,int *Pbox, in
        
 
 
-    for(int a=0;a<h2;a+=4){
+    /*for(int a=0;a<h2;a+=4){
       fX[a]=RM1[X[a]];
       fX[a+1]=RM1[X[a+1]];
       fX[a+2]=RM1[X[a+2]];
       fX[a+3]=RM1[X[a+3]];
+      }*/
+
+    for(int a=0;a<h2;a+=4){
+      fX[a]=X[a];
+      fX[a+1]=X[a+1];
+      fX[a+2]=X[a+2];
+      fX[a+3]=X[a+3];
     }
 
-    /*   if(it==0) {
+    /*   if(it<513) {
+      for(int a=0;a<h2;a++)
+	printf("%d ",fX[a]);
+      printf("\n");
+      }*/
+    
+    *(int*)&fX[0]=it;
+
+    /* if(it<513) {
       for(int a=0;a<h2;a++)
 	printf("%d ",fX[a]);
       printf("\n");
-- 
2.39.5