fixing #8113

core/src/consensus.rs

@@ -83,6 +83,40 @@ impl Tower {
         }
     }
 
+    pub fn observed_vote_distribution(
+        vote_accounts: &HashMap<Pubkey, (u64, Account)>,
+    ) -> HashMap<Slot, HashMap<Pubkey, Lockout>> {
+        let mut votes = HashMap::new();
+        for (key, (lamports, account)) in vote_accounts.iter() {
+            let vote_state = VoteState::from(&account);
+            if vote_state.is_none() {
+                datapoint_warn!(
+                    "tower_warn",
+                    (
+                        "warn",
+                        format!("Unable to get vote_state from account {}", key),
+                        String
+                    ),
+                );
+                continue;
+            }
+            let mut vote_state = vote_state.unwrap();
+            for vote in &vote_state.votes {
+                votes
+                    .entry(vote.slot())
+                    .or_default()
+                    .insert(key, vote.lockout());
+            }
+            if let Some(slot) = vote_state.root_slot {
+                votes
+                    .entry(slot)
+                    .or_default()
+                    .insert(key, MAX_LOCKOUT_HISTORY);
+            }
+        }
+        votes
+    }
+
     pub fn collect_vote_lockouts<F>(
         &self,
         bank_slot: u64,