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 0).limit(1000).toArray(); System.err.println(Arrays.toString(state)); hash.doHashTransform(state);