From 665ad1389e2a74e9de389a24f0bbaed4e5a5c4fd Mon Sep 17 00:00:00 2001
From: Lukas Tassanyi <lukas.tassanyi@iota.org>
Date: Sun, 9 Dec 2018 15:15:46 +0100
Subject: [PATCH] make amount of rounds configurable

---
 src/main/java/com/iota/curl/IotaCurlHash.java  | 12 +++++++++---
 src/main/java/com/iota/curl/IotaCurlMiner.java |  2 +-
 src/main/java/com/iota/curl/miner/Miner.java   |  2 +-
 src/test/java/com/iota/curl/HashTest.java      |  4 ++--
 4 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/iota/curl/IotaCurlHash.java b/src/main/java/com/iota/curl/IotaCurlHash.java
index e8209da..b2b836c 100644
--- a/src/main/java/com/iota/curl/IotaCurlHash.java
+++ b/src/main/java/com/iota/curl/IotaCurlHash.java
@@ -32,12 +32,18 @@ private static final class CurlState {
 
     private final CurlState curlState = new CurlState();
 
-    public static String iotaCurlHash(final String tx, final int len) {
-        final IotaCurlHash ctx = new IotaCurlHash();
+    private final int rounds;
+
+    public static String iotaCurlHash(final String tx, final int len, int rounds) {
+        final IotaCurlHash ctx = new IotaCurlHash(rounds);
         ctx.doAbsorb(tx.toCharArray(), len);
         return ctx.doFinalize();
     }
 
+    protected IotaCurlHash(int rounds) {
+        this.rounds = rounds;
+    }
+
     /**
      * Absorb given trytes.
      */
@@ -51,7 +57,7 @@ protected void doAbsorb(final char [] trytes, final int len) {
     protected void doHashTransform(int [] state1) {
         int [] state2 = Arrays.copyOf(state1, state1.length);
 
-        for(int r=0; r<27; r++) {
+        for(int r=0; r<rounds; r++) {
             state1[0] = IotaCurlUtils.TRUTH_TABLE[state2[0] + (state2[364] << 2) + 5];
 
             // state1[0] = IotaCurlUtils.invokeBinaryTruthOn(state2[0], state2[364]);
diff --git a/src/main/java/com/iota/curl/IotaCurlMiner.java b/src/main/java/com/iota/curl/IotaCurlMiner.java
index 1a2c927..426f593 100644
--- a/src/main/java/com/iota/curl/IotaCurlMiner.java
+++ b/src/main/java/com/iota/curl/IotaCurlMiner.java
@@ -121,7 +121,7 @@ private long doWork(final int minWeightMagnitude, long offset) {
 
     private final char[] powInit(final String tx) {
 
-        final IotaCurlHash ctx = new IotaCurlHash();
+        final IotaCurlHash ctx = new IotaCurlHash(27);
         final char[] trx = tx.toCharArray();
         ctx.doAbsorb(trx, TX_HEADER_SZ);
 
diff --git a/src/main/java/com/iota/curl/miner/Miner.java b/src/main/java/com/iota/curl/miner/Miner.java
index 20f9120..0bfb040 100644
--- a/src/main/java/com/iota/curl/miner/Miner.java
+++ b/src/main/java/com/iota/curl/miner/Miner.java
@@ -63,7 +63,7 @@ public static void main(final String ... args) {
         stdout("txTrytes: %s", tx);
 
         // Print hash.
-        final String hash = IotaCurlHash.iotaCurlHash(tx, TX_LENGTH);
+        final String hash = IotaCurlHash.iotaCurlHash(tx, TX_LENGTH, 27);
         stdout("hash: %s", hash);
     }
 
diff --git a/src/test/java/com/iota/curl/HashTest.java b/src/test/java/com/iota/curl/HashTest.java
index d2b9de4..c6b9fe7 100644
--- a/src/test/java/com/iota/curl/HashTest.java
+++ b/src/test/java/com/iota/curl/HashTest.java
@@ -17,7 +17,7 @@ public class HashTest {
 
     @Test
     public void shouldHash() {
-        final String hashed = IotaCurlHash.iotaCurlHash(tx, 2673);
+        final String hashed = IotaCurlHash.iotaCurlHash(tx, 2673, 27);
         System.err.println(hashed);
         System.err.println(hash);
         Assert.assertEquals(hash, hashed);
@@ -25,7 +25,7 @@ public void shouldHash() {
 
     @Test
     public void shouldDoHashTransform() {
-        IotaCurlHash hash = new IotaCurlHash();
+        IotaCurlHash hash = new IotaCurlHash(27);
         int [] state = IntStream.generate(() -> 0).limit(1000).toArray();
         System.err.println(Arrays.toString(state));
         hash.doHashTransform(state);