93 lines
2.8 KiB
Rust
93 lines
2.8 KiB
Rust
//! config for staking
|
|
//! carries variables that the stake program cares about
|
|
use bincode::{deserialize, serialized_size};
|
|
use serde_derive::{Deserialize, Serialize};
|
|
use solana_config_api::{create_config_account, get_config_data, ConfigState};
|
|
use solana_sdk::{
|
|
account::{Account, KeyedAccount},
|
|
instruction::InstructionError,
|
|
pubkey::Pubkey,
|
|
};
|
|
|
|
// stake config ID
|
|
const ID: [u8; 32] = [
|
|
6, 161, 216, 23, 165, 2, 5, 11, 104, 7, 145, 230, 206, 109, 184, 142, 30, 91, 113, 80, 246, 31,
|
|
198, 121, 10, 78, 180, 209, 0, 0, 0, 0,
|
|
];
|
|
|
|
solana_sdk::solana_name_id!(ID, "StakeConfig11111111111111111111111111111111");
|
|
|
|
// means that no more tha
|
|
pub const DEFAULT_WARMUP_RATE: f64 = 0.25;
|
|
pub const DEFAULT_COOLDOWN_RATE: f64 = 0.25;
|
|
pub const DEFAULT_SLASH_PENALTY: u8 = ((5 * std::u8::MAX as usize) / 100) as u8;
|
|
|
|
#[derive(Serialize, Deserialize, Debug, PartialEq, Clone, Copy)]
|
|
pub struct Config {
|
|
/// how much stake we can activate per-epoch as a fraction of currently effective stake
|
|
pub warmup_rate: f64,
|
|
/// how much stake we can deactivate as a fraction of currently effective stake
|
|
pub cooldown_rate: f64,
|
|
/// percentage of stake lost when slash, expressed as a portion of std::u8::MAX
|
|
pub slash_penalty: u8,
|
|
}
|
|
|
|
impl Config {
|
|
pub fn from(account: &Account) -> Option<Self> {
|
|
get_config_data(&account.data)
|
|
.ok()
|
|
.and_then(|data| deserialize(data).ok())
|
|
}
|
|
}
|
|
|
|
impl Default for Config {
|
|
fn default() -> Self {
|
|
Self {
|
|
warmup_rate: DEFAULT_WARMUP_RATE,
|
|
cooldown_rate: DEFAULT_COOLDOWN_RATE,
|
|
slash_penalty: DEFAULT_SLASH_PENALTY,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ConfigState for Config {
|
|
fn max_space() -> u64 {
|
|
serialized_size(&Config::default()).unwrap()
|
|
}
|
|
}
|
|
|
|
pub fn create_genesis_account() -> (Pubkey, Account) {
|
|
(id(), create_config_account(vec![], &Config::default(), 100))
|
|
}
|
|
|
|
pub fn create_account(lamports: u64, config: &Config) -> Account {
|
|
create_config_account(vec![], config, lamports)
|
|
}
|
|
|
|
pub fn from_keyed_account(account: &KeyedAccount) -> Result<Config, InstructionError> {
|
|
if !check_id(account.unsigned_key()) {
|
|
return Err(InstructionError::InvalidArgument);
|
|
}
|
|
Config::from(account.account).ok_or(InstructionError::InvalidArgument)
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test() {
|
|
let mut account = create_account(1, &Config::default());
|
|
assert_eq!(Config::from(&account), Some(Config::default()));
|
|
assert_eq!(
|
|
from_keyed_account(&KeyedAccount::new(&Pubkey::default(), false, &mut account)),
|
|
Err(InstructionError::InvalidArgument)
|
|
);
|
|
let (pubkey, mut account) = create_genesis_account();
|
|
assert_eq!(
|
|
from_keyed_account(&KeyedAccount::new(&pubkey, false, &mut account)),
|
|
Ok(Config::default())
|
|
);
|
|
}
|
|
}
|