06651 Added missing rehash for VirtualMap leaves (#6653)

Signed-off-by: Ivan Malygin <ivan@swirldslabs.com>
hashgraph · May 19, 2023 · aa0e9b5 · aa0e9b5
1 parent 9f1a8e0
commit aa0e9b5
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...PlatformTestingTool/src/main/java/com/swirlds/demo/platform/PlatformTestingToolState.java b/...PlatformTestingTool/src/main/java/com/swirlds/demo/platform/PlatformTestingToolState.java
@@ -1271,10 +1271,22 @@ public void init(
 
         if (trigger == InitTrigger.GENESIS) {
             genesisInit();
+        } else {
+            logger.info(LOGM_STARTUP, "Doing full rehash for the initialized VirtualMaps");
+            fullLeafRehash(getVirtualMap());
+            fullLeafRehash(getVirtualMapForSmartContracts());
+            fullLeafRehash(getVirtualMapForSmartContractsByteCode());
+            logger.info(LOGM_STARTUP, "Full rehash is complete");
         }
         this.invalidateHash();
     }
 
+    private void fullLeafRehash(VirtualMap<?, ?> virtualMap) {
+        if (virtualMap != null) {
+            virtualMap.fullLeafRehash();
+        }
+    }
+
     private void expandSignatures(final Transaction trans) {
         if (getConfig().isAppendSig()) {
             try {