refactor(lazy-imt): target depth calculation

privacy-scaling-explorations · Sep 18, 2024 · 409bfee · 409bfee
1 parent a4e7136
commit 409bfee
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/packages/lazy-imt/contracts/InternalLazyIMT.sol b/packages/lazy-imt/contracts/InternalLazyIMT.sol
@@ -221,9 +221,14 @@ library InternalLazyIMT {
         uint40 numberOfLeaves = self.numberOfLeaves;
         require(index < numberOfLeaves, "LazyIMT: leaf must exist");
 
+        // targetDepth = log2_floor(numberOfLeaves)
         uint8 targetDepth = 1;
-        while (uint40(2) ** uint40(targetDepth) < numberOfLeaves) {
-            targetDepth++;
+        {
+            uint40 exp = 2;
+            while (exp < numberOfLeaves) {
+                exp <<= 1;
+                targetDepth++;
+            }
         }
         require(depth >= targetDepth, "LazyIMT: proof depth");
         // pass depth -1 because we don't need the root value