2021-01-05 18:42:33 -08:00
|
|
|
use solana_runtime::{
|
|
|
|
accounts_index::{AccountIndex, IndexKey},
|
|
|
|
bank::Bank,
|
|
|
|
};
|
2020-05-09 11:05:29 -07:00
|
|
|
use solana_sdk::pubkey::Pubkey;
|
|
|
|
use solana_stake_program::stake_state::StakeState;
|
|
|
|
use std::{collections::HashSet, sync::Arc};
|
|
|
|
|
|
|
|
pub struct NonCirculatingSupply {
|
|
|
|
pub lamports: u64,
|
|
|
|
pub accounts: Vec<Pubkey>,
|
|
|
|
}
|
|
|
|
|
2020-05-20 20:04:07 -07:00
|
|
|
pub fn calculate_non_circulating_supply(bank: &Arc<Bank>) -> NonCirculatingSupply {
|
2020-05-09 11:05:29 -07:00
|
|
|
debug!("Updating Bank supply, epoch: {}", bank.epoch());
|
|
|
|
let mut non_circulating_accounts_set: HashSet<Pubkey> = HashSet::new();
|
|
|
|
|
|
|
|
for key in non_circulating_accounts() {
|
|
|
|
non_circulating_accounts_set.insert(key);
|
|
|
|
}
|
2020-05-18 13:17:56 -07:00
|
|
|
let withdraw_authority_list = withdraw_authority();
|
2020-05-09 11:05:29 -07:00
|
|
|
|
|
|
|
let clock = bank.clock();
|
2021-01-05 18:42:33 -08:00
|
|
|
let stake_accounts = if bank
|
|
|
|
.rc
|
|
|
|
.accounts
|
|
|
|
.accounts_db
|
|
|
|
.account_indexes
|
|
|
|
.contains(&AccountIndex::ProgramId)
|
|
|
|
{
|
|
|
|
bank.get_filtered_indexed_accounts(
|
|
|
|
&IndexKey::ProgramId(solana_stake_program::id()),
|
|
|
|
// The program-id account index checks for Account owner on inclusion. However, due to
|
2021-02-18 23:42:09 -08:00
|
|
|
// the current AccountsDb implementation, an account may remain in storage as a
|
2021-01-05 18:42:33 -08:00
|
|
|
// zero-lamport Account::Default() after being wiped and reinitialized in later
|
|
|
|
// updates. We include the redundant filter here to avoid returning these accounts.
|
|
|
|
|account| account.owner == solana_stake_program::id(),
|
|
|
|
)
|
|
|
|
} else {
|
|
|
|
bank.get_program_accounts(&solana_stake_program::id())
|
|
|
|
};
|
2020-05-09 11:05:29 -07:00
|
|
|
for (pubkey, account) in stake_accounts.iter() {
|
2021-03-09 13:06:07 -08:00
|
|
|
let stake_account = StakeState::from(account).unwrap_or_default();
|
2020-05-09 11:05:29 -07:00
|
|
|
match stake_account {
|
|
|
|
StakeState::Initialized(meta) => {
|
2020-07-31 12:37:53 -07:00
|
|
|
if meta.lockup.is_in_force(&clock, None)
|
2020-05-18 13:17:56 -07:00
|
|
|
|| withdraw_authority_list.contains(&meta.authorized.withdrawer)
|
2020-05-09 11:05:29 -07:00
|
|
|
{
|
|
|
|
non_circulating_accounts_set.insert(*pubkey);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
StakeState::Stake(meta, _stake) => {
|
2020-07-31 12:37:53 -07:00
|
|
|
if meta.lockup.is_in_force(&clock, None)
|
2020-05-18 13:17:56 -07:00
|
|
|
|| withdraw_authority_list.contains(&meta.authorized.withdrawer)
|
2020-05-09 11:05:29 -07:00
|
|
|
{
|
|
|
|
non_circulating_accounts_set.insert(*pubkey);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
_ => {}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
let lamports = non_circulating_accounts_set
|
|
|
|
.iter()
|
2020-05-19 18:13:41 -07:00
|
|
|
.map(|pubkey| bank.get_balance(&pubkey))
|
|
|
|
.sum();
|
2020-05-09 11:05:29 -07:00
|
|
|
|
|
|
|
NonCirculatingSupply {
|
|
|
|
lamports,
|
|
|
|
accounts: non_circulating_accounts_set.into_iter().collect(),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Mainnet-beta accounts that should be considered non-circulating
|
|
|
|
solana_sdk::pubkeys!(
|
|
|
|
non_circulating_accounts,
|
|
|
|
[
|
|
|
|
"9huDUZfxoJ7wGMTffUE7vh1xePqef7gyrLJu9NApncqA",
|
|
|
|
"GK2zqSsXLA2rwVZk347RYhh6jJpRsCA69FjLW93ZGi3B",
|
2020-06-11 16:30:25 -07:00
|
|
|
"CWeRmXme7LmbaUWTZWFLt6FMnpzLCHaQLuR2TdgFn4Lq",
|
2020-05-09 11:05:29 -07:00
|
|
|
"HCV5dGFJXRrJ3jhDYA4DCeb9TEDTwGGYXtT3wHksu2Zr",
|
|
|
|
"14FUT96s9swbmH7ZjpDvfEDywnAYy9zaNhv4xvezySGu",
|
|
|
|
"HbZ5FfmKWNHC7uwk6TF1hVi6TCs7dtYfdjEcuPGgzFAg",
|
|
|
|
"C7C8odR8oashR5Feyrq2tJKaXL18id1dSj2zbkDGL2C2",
|
|
|
|
"Eyr9P5XsjK2NUKNCnfu39eqpGoiLFgVAv1LSQgMZCwiQ",
|
|
|
|
"DE1bawNcRJB9rVm3buyMVfr8mBEoyyu73NBovf2oXJsJ",
|
|
|
|
"CakcnaRDHka2gXyfbEd2d3xsvkJkqsLw2akB3zsN1D2S",
|
|
|
|
"7Np41oeYqPefeNQEHSv1UDhYrehxin3NStELsSKCT4K2",
|
|
|
|
"GdnSyH3YtwcxFvQrVVJMm1JhTS4QVX7MFsX56uJLUfiZ",
|
|
|
|
"Mc5XB47H3DKJHym5RLa9mPzWv5snERsF3KNv5AauXK8",
|
|
|
|
"7cvkjYAkUYs4W8XcXsca7cBrEGFeSUjeZmKoNBvEwyri",
|
|
|
|
"AG3m2bAibcY8raMt4oXEGqRHwX4FWKPPJVjZxn1LySDX",
|
|
|
|
"5XdtyEDREHJXXW1CTtCsVjJRjBapAwK78ZquzvnNVRrV",
|
|
|
|
"6yKHERk8rsbmJxvMpPuwPs1ct3hRiP7xaJF2tvnGU6nK",
|
|
|
|
"CHmdL15akDcJgBkY6BP3hzs98Dqr6wbdDC5p8odvtSbq",
|
|
|
|
"FR84wZQy3Y3j2gWz6pgETUiUoJtreMEuWfbg6573UCj9",
|
|
|
|
"5q54XjQ7vDx4y6KphPeE97LUNiYGtP55spjvXAWPGBuf",
|
2020-05-22 15:10:32 -07:00
|
|
|
"3o6xgkJ9sTmDeQWyfj3sxwon18fXJB9PV5LDc8sfgR4a",
|
2020-06-16 19:20:10 -07:00
|
|
|
"GumSE5HsMV5HCwBTv2D2D81yy9x17aDkvobkqAfTRgmo",
|
2020-07-16 11:57:13 -07:00
|
|
|
"AzVV9ZZDxTgW4wWfJmsG6ytaHpQGSe1yz76Nyy84VbQF",
|
2020-09-03 16:45:49 -07:00
|
|
|
"8CUUMKYNGxdgYio5CLHRHyzMEhhVRMcqefgE6dLqnVRK",
|
2020-10-08 17:48:32 -07:00
|
|
|
"CQDYc4ET2mbFhVpgj41gXahL6Exn5ZoPcGAzSHuYxwmE",
|
|
|
|
"5PLJZLJiRR9vf7d1JCCg7UuWjtyN9nkab9uok6TqSyuP",
|
|
|
|
"7xJ9CLtEAcEShw9kW2gSoZkRWL566Dg12cvgzANJwbTr",
|
|
|
|
"BuCEvc9ze8UoAQwwsQLy8d447C8sA4zeVtVpc6m5wQeS",
|
|
|
|
"8ndGYFjav6NDXvzYcxs449Aub3AxYv4vYpk89zRDwgj7",
|
|
|
|
"8W58E8JVJjH1jCy5CeHJQgvwFXTyAVyesuXRZGbcSUGG",
|
|
|
|
"GNiz4Mq886bTNDT3pijGsu2gbw6it7sqrwncro45USeB",
|
|
|
|
"GhsotwFMH6XUrRLJCxcx62h7748N2Uq8mf87hUGkmPhg",
|
|
|
|
"Fgyh8EeYGZtbW8sS33YmNQnzx54WXPrJ5KWNPkCfWPot",
|
|
|
|
"8UVjvYyoqP6sqcctTso3xpCdCfgTMiv3VRh7vraC2eJk",
|
|
|
|
"BhvLngiqqKeZ8rpxch2uGjeCiC88zzewoWPRuoxpp1aS",
|
|
|
|
"63DtkW7zuARcd185EmHAkfF44bDcC2SiTSEj2spLP3iA",
|
|
|
|
"GvpCiTgq9dmEeojCDBivoLoZqc4AkbUDACpqPMwYLWKh",
|
|
|
|
"7Y8smnoUrYKGGuDq2uaFKVxJYhojgg7DVixHyAtGTYEV",
|
|
|
|
"DUS1KxwUhUyDKB4A81E8vdnTe3hSahd92Abtn9CXsEcj",
|
|
|
|
"F9MWFw8cnYVwsRq8Am1PGfFL3cQUZV37mbGoxZftzLjN",
|
|
|
|
"8vqrX3H2BYLaXVintse3gorPEM4TgTwTFZNN1Fm9TdYs",
|
|
|
|
"CUageMFi49kzoDqtdU8NvQ4Bq3sbtJygjKDAXJ45nmAi",
|
|
|
|
"5smrYwb1Hr2T8XMnvsqccTgXxuqQs14iuE8RbHFYf2Cf",
|
|
|
|
"xQadXQiUTCCFhfHjvQx1hyJK6KVWr1w2fD6DT3cdwj7",
|
2020-05-09 11:05:29 -07:00
|
|
|
]
|
|
|
|
);
|
|
|
|
|
|
|
|
// Withdraw authority for autostaked accounts on mainnet-beta
|
|
|
|
solana_sdk::pubkeys!(
|
|
|
|
withdraw_authority,
|
2020-05-18 13:17:56 -07:00
|
|
|
[
|
|
|
|
"8CUUMKYNGxdgYio5CLHRHyzMEhhVRMcqefgE6dLqnVRK",
|
|
|
|
"3FFaheyqtyAXZSYxDzsr5CVKvJuvZD1WE1VEsBtDbRqB",
|
2020-06-12 16:45:20 -07:00
|
|
|
"FdGYQdiRky8NZzN9wZtczTBcWLYYRXrJ3LMDhqDPn5rM",
|
2020-06-24 21:17:49 -07:00
|
|
|
"4e6KwQpyzGQPfgVr5Jn3g5jLjbXB4pKPa2jRLohEb1QA",
|
2020-05-18 13:17:56 -07:00
|
|
|
]
|
2020-05-09 11:05:29 -07:00
|
|
|
);
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::*;
|
|
|
|
use solana_sdk::{
|
2020-08-31 21:51:52 -07:00
|
|
|
account::Account,
|
2021-03-09 13:06:07 -08:00
|
|
|
account::AccountSharedData,
|
2020-08-31 21:51:52 -07:00
|
|
|
epoch_schedule::EpochSchedule,
|
2020-09-08 07:55:09 -07:00
|
|
|
genesis_config::{ClusterType, GenesisConfig},
|
2020-05-09 11:05:29 -07:00
|
|
|
};
|
|
|
|
use solana_stake_program::stake_state::{Authorized, Lockup, Meta, StakeState};
|
|
|
|
use std::{collections::BTreeMap, sync::Arc};
|
|
|
|
|
|
|
|
fn new_from_parent(parent: &Arc<Bank>) -> Bank {
|
|
|
|
Bank::new_from_parent(parent, &Pubkey::default(), parent.slot() + 1)
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_calculate_non_circulating_supply() {
|
|
|
|
let mut accounts: BTreeMap<Pubkey, Account> = BTreeMap::new();
|
|
|
|
let balance = 10;
|
|
|
|
let num_genesis_accounts = 10;
|
|
|
|
for _ in 0..num_genesis_accounts {
|
|
|
|
accounts.insert(
|
2020-10-19 12:12:08 -07:00
|
|
|
solana_sdk::pubkey::new_rand(),
|
2020-05-09 11:05:29 -07:00
|
|
|
Account::new(balance, 0, &Pubkey::default()),
|
|
|
|
);
|
|
|
|
}
|
|
|
|
let non_circulating_accounts = non_circulating_accounts();
|
|
|
|
let num_non_circulating_accounts = non_circulating_accounts.len() as u64;
|
|
|
|
for key in non_circulating_accounts.clone() {
|
|
|
|
accounts.insert(key, Account::new(balance, 0, &Pubkey::default()));
|
|
|
|
}
|
|
|
|
|
|
|
|
let num_stake_accounts = 3;
|
|
|
|
for _ in 0..num_stake_accounts {
|
2020-10-19 12:12:08 -07:00
|
|
|
let pubkey = solana_sdk::pubkey::new_rand();
|
2020-05-09 11:05:29 -07:00
|
|
|
let meta = Meta {
|
|
|
|
authorized: Authorized::auto(&pubkey),
|
|
|
|
lockup: Lockup {
|
|
|
|
epoch: 1,
|
|
|
|
..Lockup::default()
|
|
|
|
},
|
|
|
|
..Meta::default()
|
|
|
|
};
|
|
|
|
let stake_account = Account::new_data_with_space(
|
|
|
|
balance,
|
|
|
|
&StakeState::Initialized(meta),
|
|
|
|
std::mem::size_of::<StakeState>(),
|
|
|
|
&solana_stake_program::id(),
|
|
|
|
)
|
|
|
|
.unwrap();
|
|
|
|
accounts.insert(pubkey, stake_account);
|
|
|
|
}
|
|
|
|
|
|
|
|
let slots_per_epoch = 32;
|
|
|
|
let genesis_config = GenesisConfig {
|
|
|
|
accounts,
|
|
|
|
epoch_schedule: EpochSchedule::new(slots_per_epoch),
|
2020-09-08 07:55:09 -07:00
|
|
|
cluster_type: ClusterType::MainnetBeta,
|
2020-05-09 11:05:29 -07:00
|
|
|
..GenesisConfig::default()
|
|
|
|
};
|
|
|
|
let mut bank = Arc::new(Bank::new(&genesis_config));
|
|
|
|
assert_eq!(
|
|
|
|
bank.capitalization(),
|
|
|
|
(num_genesis_accounts + num_non_circulating_accounts + num_stake_accounts) * balance
|
|
|
|
);
|
|
|
|
|
2020-05-20 20:04:07 -07:00
|
|
|
let non_circulating_supply = calculate_non_circulating_supply(&bank);
|
2020-05-09 11:05:29 -07:00
|
|
|
assert_eq!(
|
|
|
|
non_circulating_supply.lamports,
|
|
|
|
(num_non_circulating_accounts + num_stake_accounts) * balance
|
|
|
|
);
|
|
|
|
assert_eq!(
|
|
|
|
non_circulating_supply.accounts.len(),
|
|
|
|
num_non_circulating_accounts as usize + num_stake_accounts as usize
|
|
|
|
);
|
|
|
|
|
|
|
|
bank = Arc::new(new_from_parent(&bank));
|
|
|
|
let new_balance = 11;
|
|
|
|
for key in non_circulating_accounts {
|
2021-03-09 13:06:07 -08:00
|
|
|
bank.store_account(
|
|
|
|
&key,
|
|
|
|
&AccountSharedData::new(new_balance, 0, &Pubkey::default()),
|
|
|
|
);
|
2020-05-09 11:05:29 -07:00
|
|
|
}
|
2020-05-20 20:04:07 -07:00
|
|
|
let non_circulating_supply = calculate_non_circulating_supply(&bank);
|
2020-05-09 11:05:29 -07:00
|
|
|
assert_eq!(
|
|
|
|
non_circulating_supply.lamports,
|
|
|
|
(num_non_circulating_accounts * new_balance) + (num_stake_accounts * balance)
|
|
|
|
);
|
|
|
|
assert_eq!(
|
|
|
|
non_circulating_supply.accounts.len(),
|
|
|
|
num_non_circulating_accounts as usize + num_stake_accounts as usize
|
|
|
|
);
|
|
|
|
|
|
|
|
// Advance bank an epoch, which should unlock stakes
|
|
|
|
for _ in 0..slots_per_epoch {
|
|
|
|
bank = Arc::new(new_from_parent(&bank));
|
|
|
|
}
|
|
|
|
assert_eq!(bank.epoch(), 1);
|
2020-05-20 20:04:07 -07:00
|
|
|
let non_circulating_supply = calculate_non_circulating_supply(&bank);
|
2020-05-09 11:05:29 -07:00
|
|
|
assert_eq!(
|
|
|
|
non_circulating_supply.lamports,
|
|
|
|
num_non_circulating_accounts * new_balance
|
|
|
|
);
|
|
|
|
assert_eq!(
|
|
|
|
non_circulating_supply.accounts.len(),
|
|
|
|
num_non_circulating_accounts as usize
|
|
|
|
);
|
|
|
|
}
|
|
|
|
}
|