From 2f92b92a8a779b28b21a228e71dbea624c8599dc Mon Sep 17 00:00:00 2001
From: Justin Starry <justin@solana.com>
Date: Mon, 30 Sep 2019 21:57:49 -0400
Subject: [PATCH] Expose current stake accounts of a bank for use in cli
 tooling (#6184)

---
 runtime/src/bank.rs   | 5 +++++
 runtime/src/stakes.rs | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index 6a751d754..08fe3e443 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -1485,6 +1485,11 @@ impl Bank {
         self.stakes.read().unwrap().vote_accounts().clone()
     }
 
+    /// current stake accounts for this bank
+    pub fn stake_accounts(&self) -> HashMap<Pubkey, Account> {
+        self.stakes.read().unwrap().stake_accounts().clone()
+    }
+
     /// vote accounts for the specific epoch along with the stake
     ///   attributed to each account
     pub fn epoch_vote_accounts(&self, epoch: Epoch) -> Option<&HashMap<Pubkey, (u64, Account)>> {
diff --git a/runtime/src/stakes.rs b/runtime/src/stakes.rs
index ca940e3ed..bcae8d5bc 100644
--- a/runtime/src/stakes.rs
+++ b/runtime/src/stakes.rs
@@ -167,10 +167,15 @@ impl Stakes {
             }
         }
     }
+
     pub fn vote_accounts(&self) -> &HashMap<Pubkey, (u64, Account)> {
         &self.vote_accounts
     }
 
+    pub fn stake_accounts(&self) -> &HashMap<Pubkey, Account> {
+        &self.stake_accounts
+    }
+
     pub fn rewards_pools(&self) -> impl Iterator<Item = (&Pubkey, &Account)> {
         self.stake_accounts
             .iter()