From 08325d5e1ec4c416c5908d9f6c5ca250deefc237 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Rapha=C3=ABl=20Couturier?= <raphael.couturier@univ-fcomte.fr>
Date: Sat, 24 Aug 2019 22:17:31 +0200
Subject: [PATCH] update

---
 OneRoundIoT/NEW/Makefile   |  8 ++++++-
 OneRoundIoT/NEW/scprng.cpp | 45 ++++++++++++++++++++++++++++----------
 2 files changed, 40 insertions(+), 13 deletions(-)

diff --git a/OneRoundIoT/NEW/Makefile b/OneRoundIoT/NEW/Makefile
index 1e5720f..993c8ac 100644
--- a/OneRoundIoT/NEW/Makefile
+++ b/OneRoundIoT/NEW/Makefile
@@ -20,12 +20,18 @@ endif
 scprng.o: scprng.cpp 
 	$(CXX) -c -o $@ $< $(CFLAGS)
 
+scprng_old.o: scprng_old.cpp 
+	$(CXX) -c -o $@ $< $(CFLAGS)
+
 
 
 scprng: pixmap_io.o scprng.o 
 	$(CXX)  -o $@ $^ $(CFLAGS)
 
+scprng_old: pixmap_io.o scprng_old.o 
+	$(CXX)  -o $@ $^ $(CFLAGS)
+
 
 
 clean:
-	rm -rf *.o scprng
+	rm -rf *.o scprng scprng_old
diff --git a/OneRoundIoT/NEW/scprng.cpp b/OneRoundIoT/NEW/scprng.cpp
index e0cc4c4..18a9686 100644
--- a/OneRoundIoT/NEW/scprng.cpp
+++ b/OneRoundIoT/NEW/scprng.cpp
@@ -192,29 +192,35 @@ void rc4keyperm(uchar *key,int len, int rp,int *sc, int size_DK) {
   }
 }
 
+#define ROR64(x,r) (((x)>>(r))|((x)<<(64-(r))))
+#define ROL64(x,r) (((x)<<(r))|((x)>>(64-(r))))
+#define R(x,y,k) (x=ROR64(x,18), x+=y, x^=k, y=ROL64(y,13), y^=x)
 
-void scprng(uint64_t *plain, uint64_t* cipher, int bufsize, int nb_bloc, uint64_t *Val, uchar *Sbox1, uchar *Sbox2, uchar * Sbox3, uchar *Sbox4, int *Pbox, int *Pbox2, int *Pbox3, int *Pbox4, uchar *DK, int delta) {
-  int update=0;
 
 
-  for(int nb=0;nb<nb_bloc;nb++) {
+void scprng(uint64_t *plain, uint64_t* cipher, int bufsize, int nb_bloc, uint64_t *Val, uint64_t* Val2,uchar *Sbox1, uchar *Sbox2, uchar * Sbox3, uchar *Sbox4, int *Pbox, int *Pbox2, int *Pbox3, int *Pbox4, uchar *DK, int delta) {
+  int update=0;
+
+  
   
+  for(int nb=0;nb<nb_bloc;nb++) {
+
+    // #pragma omp parallel   
     for(int j=0;j<bufsize;j++) {
       //Val[j]=splitmix64_stateless(Val[j])^Val[Pbox[j]];
       //Val[j]=xorshift64(Val[j])^Val[Pbox[j]];  //fail
 //      Val[j]=xorshift64(Val[j])^Val[Pbox[j]]^Val[Pbox2[j]];
-      Val[j]=xorshift64(Val[j]);
-      
+
+      Val2[j]=xorshift64(Val[j])^Val[Pbox[j]]^Val[Pbox2[j]];      
       //Val[j]=xoroshiro128plus(&xoro[j])^Val[Pbox[j]];
       //Val[j]=jsf(&ctx[j])^Val[Pbox[j]];  //good
       //Val[j]=sfc(&sfcd[j])^Val[Pbox[j]];  //good
     }
-    for(int j=0;j<bufsize;j++) {
-      Val[j]=Val[j]^Val[Pbox[j]]^Val[Pbox2[j]];
-    }
+
 
     
     for(int j=0;j<bufsize;j++) {
+      Val[j]=ROR64(Val2[j],Pbox[j]&63);
       cipher[nb*bufsize+j]=Val[j]^plain[nb*bufsize+j];
     }
   
@@ -223,10 +229,16 @@ void scprng(uint64_t *plain, uint64_t* cipher, int bufsize, int nb_bloc, uint64_
     
     if(update==delta) {
       update=0;
-      uchar *ptr=(uchar*)Val;
+      /*      uchar *ptr=(uchar*)Val;
       for(int j=0;j<bufsize*8;j++)
 	ptr[j]^=Sbox2[Sbox1[ptr[j]+DK[j&63]]];
+      */
 
+      for(int j=0;j<bufsize;j++) {
+	Val[j]=R(Val[j],Val[Pbox[j]],Val[Pbox2[j]]);
+      }
+
+      /*      
       for(int j=0;j<256;j++)
 	Sbox1[j]=Sbox3[Sbox1[j]];
 
@@ -235,7 +247,7 @@ void scprng(uint64_t *plain, uint64_t* cipher, int bufsize, int nb_bloc, uint64_
 
       for(int j=0;j<256;j++)
 	Sbox3[j]=Sbox4[Sbox3[j]];
-
+      */
 
       for(int j=0;j<bufsize;j++)
 	Pbox[j]=Pbox3[Pbox[j]];
@@ -359,6 +371,7 @@ int main(int argc, char **argv) {
   
   
   uint64_t Val[h];
+  uint64_t Val2[h];
   for(int i=0;i<h;i++) {
     Val[Pbox[i]]=splitmix64_stateless(i+DK[i&63]);
   }
@@ -376,7 +389,7 @@ int main(int argc, char **argv) {
 
 
   for(uint iter=0;iter<nb_test;iter++) {
-    scprng(SEQ, SEQ2, h, nb_bloc, Val,Sbox1, Sbox2, Sbox3, Sbox4, Pbox, Pbox2, Pbox3, Pbox4, DK, delta);
+    scprng(SEQ, SEQ2, h, nb_bloc, Val,Val2,Sbox1, Sbox2, Sbox3, Sbox4, Pbox, Pbox2, Pbox3, Pbox4, DK, delta);
   }
 
   double time=TimeStop(t);
@@ -392,12 +405,20 @@ int main(int argc, char **argv) {
   }
 
 
+
   //reinit of parameters
+
   rc4key(DK, Sbox1, 8);
   rc4key(&DK[8], Sbox2, 8);
+  rc4key(&DK[16], Sbox3, 8);
+  rc4key(&DK[24], Sbox4, 8);
 
   rc4keyperm(&DK[16], h, 1, Pbox, 16);
   rc4keyperm(&DK[32], h, 1, Pbox2, 16);
+  rc4keyperm(&DK[8], h, 1, Pbox3, 16);
+  rc4keyperm(&DK[48], h, 1, Pbox4, 16);
+
+
   for(int i=0;i<h;i++) {
     Val[Pbox[i]]=splitmix64_stateless(i+DK[i&63]);
   }
@@ -406,7 +427,7 @@ int main(int argc, char **argv) {
 
 
   for(uint iter=0;iter<nb_test;iter++) {
-    scprng(SEQ2, SEQ, h, nb_bloc, Val,Sbox1, Sbox2, Sbox3, Sbox4, Pbox, Pbox2, Pbox3, Pbox4, DK, delta);
+    scprng(SEQ2, SEQ, h, nb_bloc, Val,Val2,Sbox1, Sbox2, Sbox3, Sbox4, Pbox, Pbox2, Pbox3, Pbox4, DK, delta);
   }
 
 
-- 
2.39.5