1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

//! rewards_pools
//! * initialize genesis with rewards pools
//! * keep track of rewards
//! * own mining pools
use crate::stake_state::StakeState;
use rand::{thread_rng, Rng};
use solana_sdk::{
    account::Account,
    hash::{hash, Hash},
    pubkey::Pubkey,
};

// base rewards pool ID
solana_sdk::declare_id!("StakeRewards1111111111111111111111111111111");

// to cut down on collisions for redemptions, we make multiple accounts
pub const NUM_REWARDS_POOLS: usize = 256;

pub fn random_id() -> Pubkey {
    let mut id = Hash::new(id().as_ref());

    for _i in 0..thread_rng().gen_range(0, NUM_REWARDS_POOLS) {
        id = hash(id.as_ref());
    }

    Pubkey::new(id.as_ref())
}

pub fn create_genesis_accounts() -> Vec<(Pubkey, Account)> {
    let mut accounts = Vec::with_capacity(NUM_REWARDS_POOLS);
    let mut pubkey = id();

    for _i in 0..NUM_REWARDS_POOLS {
        accounts.push((
            pubkey,
            Account::new_data(std::u64::MAX, &StakeState::RewardsPool, &crate::id()).unwrap(),
        ));
        pubkey = Pubkey::new(hash(pubkey.as_ref()).as_ref());
    }
    accounts
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test() {
        let accounts = create_genesis_accounts();

        for _i in 0..NUM_REWARDS_POOLS {
            let id = random_id();
            assert!(accounts.iter().position(|x| x.0 == id).is_some());
        }
    }
}