From f9df17d8d0efcd8ef92c419f0b89cc1bac2c73a5 Mon Sep 17 00:00:00 2001
From: Rob Walker <rob@solana.com>
Date: Mon, 2 Dec 2019 13:47:20 -0800
Subject: [PATCH] track bank size (#7144)

* track bank size

* support non-linux a bit better
---
 genesis/Cargo.toml               |  8 ++++
 genesis/src/address_generator.rs |  1 +
 genesis/src/genesis_accounts.rs  |  2 +-
 genesis/src/lib.rs               |  5 +++
 runtime/src/bank.rs              | 64 +++++++++++++++++++++++++++++++-
 5 files changed, 78 insertions(+), 2 deletions(-)

diff --git a/genesis/Cargo.toml b/genesis/Cargo.toml
index 2e66e9a85b..9aa62dab3e 100644
--- a/genesis/Cargo.toml
+++ b/genesis/Cargo.toml
@@ -24,3 +24,11 @@ solana-stake-program = { path = "../programs/stake", version = "0.22.0" }
 solana-storage-program = { path = "../programs/storage", version = "0.22.0" }
 solana-vote-program = { path = "../programs/vote", version = "0.22.0" }
 tempfile = "3.1.0"
+
+[[bin]]
+name = "solana-genesis"
+path = "src/main.rs"
+
+
+[lib]
+name = "solana_genesis"
diff --git a/genesis/src/address_generator.rs b/genesis/src/address_generator.rs
index 06ea7c204c..828eb8e7fa 100644
--- a/genesis/src/address_generator.rs
+++ b/genesis/src/address_generator.rs
@@ -24,6 +24,7 @@ impl AddressGenerator {
             .as_ref(),
         )
     }
+    #[allow(clippy::should_implement_trait)]
     pub fn next(&mut self) -> Pubkey {
         let nth = self.nth;
         self.nth += 1;
diff --git a/genesis/src/genesis_accounts.rs b/genesis/src/genesis_accounts.rs
index 919f2a8505..559819383f 100644
--- a/genesis/src/genesis_accounts.rs
+++ b/genesis/src/genesis_accounts.rs
@@ -614,7 +614,7 @@ fn add_stakes(
         .sum::<u64>()
 }
 
-pub(crate) fn add_genesis_accounts(genesis_config: &mut GenesisConfig) -> u64 {
+pub fn add_genesis_accounts(genesis_config: &mut GenesisConfig) -> u64 {
     add_stakes(
         genesis_config,
         &BATCH_ONE_STAKER_INFOS,
diff --git a/genesis/src/lib.rs b/genesis/src/lib.rs
index cae7466fdc..19eab13672 100644
--- a/genesis/src/lib.rs
+++ b/genesis/src/lib.rs
@@ -1,3 +1,8 @@
+pub mod address_generator;
+pub mod genesis_accounts;
+pub mod stakes;
+pub mod unlocks;
+
 use serde::{Deserialize, Serialize};
 
 /// An account where the data is encoded as a Base64 string.
diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index af11632135..7956718cb1 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -1683,7 +1683,7 @@ mod tests {
         system_instruction,
         sysvar::{fees::Fees, rewards::Rewards},
     };
-    use solana_stake_program::stake_state::{Delegation, Stake};
+    use solana_stake_program::stake_state::{Authorized, Delegation, Lockup, Stake};
     use solana_vote_program::{
         vote_instruction,
         vote_state::{self, Vote, VoteInit, VoteState, MAX_LOCKOUT_HISTORY},
@@ -3921,4 +3921,66 @@ mod tests {
         }
         assert_ne!(bank1.hash_internal_state(), hash1);
     }
+
+    #[ignore]
+    #[test]
+    fn test_banks_leak() {
+        fn add_lotsa_stake_accounts(genesis_config: &mut GenesisConfig) {
+            const LOTSA: usize = 4_096;
+
+            (0..LOTSA).for_each(|_| {
+                let pubkey = Pubkey::new_rand();
+                genesis_config.add_account(
+                    pubkey,
+                    solana_stake_program::stake_state::create_lockup_stake_account(
+                        &Authorized::auto(&pubkey),
+                        &Lockup::default(),
+                        &Rent::default(),
+                        42,
+                    ),
+                );
+            });
+        }
+        solana_logger::setup();
+        let (mut genesis_config, _) = create_genesis_config(100_000_000);
+        add_lotsa_stake_accounts(&mut genesis_config);
+        let mut bank = std::sync::Arc::new(Bank::new(&genesis_config));
+        let mut num_banks = 0;
+        let pid = std::process::id();
+        #[cfg(not(target_os = "linux"))]
+        error!(
+            "\nYou can run this to watch RAM:\n   while read -p 'banks: '; do echo $(( $(ps -h -o %z --pid={})/$REPLY));done", pid
+        );
+        loop {
+            num_banks += 1;
+            bank = std::sync::Arc::new(new_from_parent(&bank));
+            if num_banks % 100 == 0 {
+                #[cfg(target_os = "linux")]
+                {
+                    let pages_consumed = std::fs::read_to_string(format!("/proc/{}/statm", pid))
+                        .unwrap()
+                        .split_whitespace()
+                        .next()
+                        .unwrap()
+                        .parse::<usize>()
+                        .unwrap();
+                    error!(
+                        "at {} banks: {} mem or {}kB/bank",
+                        num_banks,
+                        pages_consumed * 4096,
+                        (pages_consumed * 4) / num_banks
+                    );
+                }
+                #[cfg(not(target_os = "linux"))]
+                {
+                    error!(
+                        "{} banks, get mem usage for pid {} from ps or activity monitor, sleeping for 5 sec",
+                        num_banks,
+                        pid
+                    );
+                    std::thread::sleep(Duration::new(5, 0));
+                }
+            }
+        }
+    }
 }