2019-08-15 14:35:48 -07:00
|
|
|
//! config for staking
|
|
|
|
//! carries variables that the stake program cares about
|
2021-06-15 09:04:00 -07:00
|
|
|
use bincode::deserialize;
|
|
|
|
use solana_config_program::{create_config_account, get_config_data};
|
2019-08-15 14:35:48 -07:00
|
|
|
use solana_sdk::{
|
2021-04-28 14:30:06 -07:00
|
|
|
account::{AccountSharedData, ReadableAccount, WritableAccount},
|
2021-03-09 13:06:07 -08:00
|
|
|
genesis_config::GenesisConfig,
|
|
|
|
instruction::InstructionError,
|
2020-10-26 09:14:57 -07:00
|
|
|
keyed_account::KeyedAccount,
|
2021-06-15 09:04:00 -07:00
|
|
|
stake::config::{self, Config},
|
2019-08-15 14:35:48 -07:00
|
|
|
};
|
|
|
|
|
2021-06-15 09:04:00 -07:00
|
|
|
#[deprecated(
|
|
|
|
since = "1.8.0",
|
|
|
|
note = "Please use `solana_sdk::stake::config` or `solana_program::stake::config` instead"
|
|
|
|
)]
|
|
|
|
pub use solana_sdk::stake::config::*;
|
2019-08-15 14:35:48 -07:00
|
|
|
|
2021-06-15 09:04:00 -07:00
|
|
|
pub fn from<T: ReadableAccount>(account: &T) -> Option<Config> {
|
2021-06-18 06:34:46 -07:00
|
|
|
get_config_data(account.data())
|
2021-06-15 09:04:00 -07:00
|
|
|
.ok()
|
|
|
|
.and_then(|data| deserialize(data).ok())
|
2019-08-15 14:35:48 -07:00
|
|
|
}
|
|
|
|
|
2021-06-15 09:04:00 -07:00
|
|
|
pub fn from_keyed_account(account: &KeyedAccount) -> Result<Config, InstructionError> {
|
|
|
|
if !config::check_id(account.unsigned_key()) {
|
|
|
|
return Err(InstructionError::InvalidArgument);
|
2019-08-15 14:35:48 -07:00
|
|
|
}
|
2021-06-15 09:04:00 -07:00
|
|
|
from(&*account.try_account_ref()?).ok_or(InstructionError::InvalidArgument)
|
2019-08-15 14:35:48 -07:00
|
|
|
}
|
|
|
|
|
2021-06-15 09:04:00 -07:00
|
|
|
pub fn create_account(lamports: u64, config: &Config) -> AccountSharedData {
|
|
|
|
create_config_account(vec![], config, lamports)
|
2019-08-15 14:35:48 -07:00
|
|
|
}
|
|
|
|
|
2019-11-25 15:11:55 -08:00
|
|
|
pub fn add_genesis_account(genesis_config: &mut GenesisConfig) -> u64 {
|
|
|
|
let mut account = create_config_account(vec![], &Config::default(), 0);
|
2021-03-09 14:31:33 -08:00
|
|
|
let lamports = genesis_config.rent.minimum_balance(account.data().len());
|
2019-11-25 15:11:55 -08:00
|
|
|
|
2021-04-28 14:30:06 -07:00
|
|
|
account.set_lamports(lamports.max(1));
|
2019-11-25 15:11:55 -08:00
|
|
|
|
2021-06-15 09:04:00 -07:00
|
|
|
genesis_config.add_account(config::id(), account);
|
2019-11-25 15:11:55 -08:00
|
|
|
|
|
|
|
lamports
|
2019-08-15 14:35:48 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::*;
|
2019-11-25 15:11:55 -08:00
|
|
|
use solana_sdk::pubkey::Pubkey;
|
2020-01-22 09:11:56 -08:00
|
|
|
use std::cell::RefCell;
|
2019-08-15 14:35:48 -07:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test() {
|
2020-05-15 09:35:43 -07:00
|
|
|
let account = RefCell::new(create_account(0, &Config::default()));
|
2021-06-15 09:04:00 -07:00
|
|
|
assert_eq!(from(&account.borrow()), Some(Config::default()));
|
2019-08-15 14:35:48 -07:00
|
|
|
assert_eq!(
|
2020-05-15 09:35:43 -07:00
|
|
|
from_keyed_account(&KeyedAccount::new(&Pubkey::default(), false, &account)),
|
2019-08-15 14:35:48 -07:00
|
|
|
Err(InstructionError::InvalidArgument)
|
|
|
|
);
|
|
|
|
}
|
|
|
|
}
|