From f479021c0f9fd3f3997b27511cfc643de7b1cbfb Mon Sep 17 00:00:00 2001
From: Pankaj Garg <pankaj@solana.com>
Date: Fri, 22 Mar 2019 17:35:54 -0700
Subject: [PATCH] Update leader slot in poh recorder if we skipped it (#3451)

* reset poh recorder with the original start slot
---
 core/src/replay_stage.rs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/core/src/replay_stage.rs b/core/src/replay_stage.rs
index faf87065aa..fbcdf75f38 100644
--- a/core/src/replay_stage.rs
+++ b/core/src/replay_stage.rs
@@ -282,6 +282,25 @@ impl ReplayStage {
         trace!("{} checking poh slot {}", my_id, poh_slot);
         if blocktree.meta(poh_slot).unwrap().is_some() {
             // We've already broadcasted entries for this slot, skip it
+
+            // Since we are skipping our leader slot, let's tell poh recorder when we should be
+            // leader again
+            if reached_leader_tick {
+                let _ = bank_forks.read().unwrap().get(poh_slot).map(|bank| {
+                    let next_leader_slot =
+                        leader_schedule_utils::next_leader_slot(&my_id, bank.slot(), &bank);
+                    let mut poh = poh_recorder.lock().unwrap();
+                    let start_slot = poh.start_slot();
+                    poh.reset(
+                        bank.tick_height(),
+                        bank.last_blockhash(),
+                        start_slot,
+                        next_leader_slot,
+                        bank.ticks_per_slot(),
+                    );
+                });
+            }
+
             return;
         }
         if bank_forks.read().unwrap().get(poh_slot).is_none() {