Add poh hash

firedancer-io · Jan 13, 2025 · c95ed3d · c95ed3d
1 parent 01b050a
commit c95ed3d
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/src/test_suite/block/codec_utils.py b/src/test_suite/block/codec_utils.py
@@ -22,6 +22,15 @@ def decode_input(context: block_pb.BlockContext):
     for i in range(len(context.acct_states)):
         decode_acct_state(context.acct_states[i])
 
+    # POH hash
+    context.slot_ctx.poh = fd58.dec32(context.slot_ctx.poh)
+
+    # Vote accounts
+    for i in range(len(context.epoch_ctx.vote_accounts)):
+        context.epoch_ctx.vote_accounts[i].pubkey = fd58.dec32(
+            context.epoch_ctx.vote_accounts[i].pubkey
+        )
+
     # Blockhash queue
     for i in range(len(context.blockhash_queue)):
         context.blockhash_queue[i] = fd58.dec32(context.blockhash_queue[i])
@@ -42,6 +51,15 @@ def encode_input(context: block_pb.BlockContext):
     for i in range(len(context.acct_states)):
         encode_acct_state(context.acct_states[i])
 
+    # POH hash
+    context.slot_ctx.poh = fd58.enc32(context.slot_ctx.poh)
+
+    # Vote accounts
+    for i in range(len(context.epoch_ctx.vote_accounts)):
+        context.epoch_ctx.vote_accounts[i].pubkey = fd58.enc32(
+            context.epoch_ctx.vote_accounts[i].pubkey
+        )
+
     # Blockhash queue
     for i in range(len(context.blockhash_queue)):
         context.blockhash_queue[i] = fd58.enc32(context.blockhash_queue[i])
@@ -58,4 +76,6 @@ def encode_output(effects: block_pb.BlockEffects):
     for i in range(len(effects.acct_states)):
         encode_acct_state(effects.acct_states[i])
 
-    # TODO: lt_hash, account delta hash
+    effects.bank_hash = fd58.enc32(effects.bank_hash)
+    effects.lt_hash = fd58.enc32(effects.lt_hash)
+    effects.account_delta_hash = fd58.enc32(effects.account_delta_hash)