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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
//! This account contains the current cluster rent
//!
use crate::{
    account::{Account, KeyedAccount},
    account_info::AccountInfo,
    instruction::InstructionError,
    rent_calculator::RentCalculator,
    sysvar,
};
use bincode::serialized_size;

///  rent account pubkey
const ID: [u8; 32] = [
    6, 167, 213, 23, 25, 44, 92, 81, 33, 140, 201, 76, 61, 74, 241, 127, 88, 218, 238, 8, 155, 161,
    253, 68, 227, 219, 217, 138, 0, 0, 0, 0,
];

crate::solana_sysvar_id!(ID, "SysvarRent111111111111111111111111111111111");

#[repr(C)]
#[derive(Serialize, Deserialize, Debug, Default)]
pub struct Rent {
    pub rent_calculator: RentCalculator,
}

impl Rent {
    pub fn from_account(account: &Account) -> Option<Self> {
        account.deserialize_data().ok()
    }
    pub fn to_account(&self, account: &mut Account) -> Option<()> {
        account.serialize_data(self).ok()
    }
    pub fn from_account_info(account: &AccountInfo) -> Option<Self> {
        account.deserialize_data().ok()
    }
    pub fn to_account_info(&self, account: &mut AccountInfo) -> Option<()> {
        account.serialize_data(self).ok()
    }
    pub fn size_of() -> usize {
        serialized_size(&Rent::default()).unwrap() as usize
    }
}

pub fn create_account(lamports: u64, rent_calculator: &RentCalculator) -> Account {
    Account::new_data(
        lamports,
        &Rent {
            rent_calculator: *rent_calculator,
        },
        &sysvar::id(),
    )
    .unwrap()
}

pub fn from_keyed_account(account: &KeyedAccount) -> Result<Rent, InstructionError> {
    if !check_id(account.unsigned_key()) {
        return Err(InstructionError::InvalidArgument);
    }
    Rent::from_account(account.account).ok_or(InstructionError::InvalidArgument)
}

pub fn verify_rent_exemption(
    account: &KeyedAccount,
    rent_sysvar_account: &KeyedAccount,
) -> Result<(), InstructionError> {
    if !from_keyed_account(rent_sysvar_account)?
        .rent_calculator
        .is_exempt(account.account.lamports, account.account.data.len())
    {
        Err(InstructionError::InsufficientFunds)
    } else {
        Ok(())
    }
}

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

    #[test]
    fn test_rent_create_account() {
        let lamports = 42;
        let account = create_account(lamports, &RentCalculator::default());
        let rent = Rent::from_account(&account).unwrap();
        assert_eq!(rent.rent_calculator, RentCalculator::default());
    }
}