2019-08-23 14:04:53 -07:00
|
|
|
//! calculate and collect rent from Accounts
|
2019-11-18 12:01:27 -08:00
|
|
|
use solana_sdk::{
|
2020-10-21 06:19:25 -07:00
|
|
|
account::Account, clock::Epoch, epoch_schedule::EpochSchedule, genesis_config::GenesisConfig,
|
|
|
|
incinerator, pubkey::Pubkey, rent::Rent, sysvar,
|
2019-11-18 12:01:27 -08:00
|
|
|
};
|
2019-08-23 14:04:53 -07:00
|
|
|
|
2020-08-11 08:04:32 -07:00
|
|
|
#[derive(Serialize, Deserialize, Clone, PartialEq, Debug, AbiExample)]
|
2019-08-23 14:04:53 -07:00
|
|
|
pub struct RentCollector {
|
|
|
|
pub epoch: Epoch,
|
|
|
|
pub epoch_schedule: EpochSchedule,
|
|
|
|
pub slots_per_year: f64,
|
2019-10-30 16:25:12 -07:00
|
|
|
pub rent: Rent,
|
2019-08-23 14:04:53 -07:00
|
|
|
}
|
|
|
|
|
2020-08-11 08:04:32 -07:00
|
|
|
impl Default for RentCollector {
|
|
|
|
fn default() -> Self {
|
|
|
|
Self {
|
|
|
|
epoch: Epoch::default(),
|
|
|
|
epoch_schedule: EpochSchedule::default(),
|
|
|
|
// derive default value using GenesisConfig::default()
|
|
|
|
slots_per_year: GenesisConfig::default().slots_per_year(),
|
|
|
|
rent: Rent::default(),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-08-23 14:04:53 -07:00
|
|
|
impl RentCollector {
|
|
|
|
pub fn new(
|
|
|
|
epoch: Epoch,
|
|
|
|
epoch_schedule: &EpochSchedule,
|
|
|
|
slots_per_year: f64,
|
2019-10-30 16:25:12 -07:00
|
|
|
rent: &Rent,
|
2019-08-23 14:04:53 -07:00
|
|
|
) -> Self {
|
|
|
|
Self {
|
|
|
|
epoch,
|
|
|
|
epoch_schedule: *epoch_schedule,
|
|
|
|
slots_per_year,
|
2019-10-30 16:25:12 -07:00
|
|
|
rent: *rent,
|
2019-08-23 14:04:53 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-10-21 06:19:25 -07:00
|
|
|
pub fn clone_with_epoch(&self, epoch: Epoch) -> Self {
|
2019-08-23 14:04:53 -07:00
|
|
|
Self {
|
|
|
|
epoch,
|
|
|
|
..self.clone()
|
|
|
|
}
|
|
|
|
}
|
2020-08-16 22:22:16 -07:00
|
|
|
|
2019-08-23 14:04:53 -07:00
|
|
|
// updates this account's lamports and status and returns
|
|
|
|
// the account rent collected, if any
|
|
|
|
//
|
2020-08-11 08:04:32 -07:00
|
|
|
#[must_use = "add to Bank::collected_rent"]
|
2020-10-21 06:19:25 -07:00
|
|
|
pub fn collect_from_existing_account(
|
|
|
|
&self,
|
|
|
|
address: &Pubkey,
|
|
|
|
account: &mut Account,
|
|
|
|
rent_fix_enabled: bool,
|
|
|
|
) -> u64 {
|
2020-04-30 22:04:08 -07:00
|
|
|
if account.executable
|
|
|
|
|| account.rent_epoch > self.epoch
|
|
|
|
|| sysvar::check_id(&account.owner)
|
|
|
|
|| *address == incinerator::id()
|
2020-03-27 08:28:18 -07:00
|
|
|
{
|
2019-11-13 21:26:49 -08:00
|
|
|
0
|
2019-08-23 14:04:53 -07:00
|
|
|
} else {
|
|
|
|
let slots_elapsed: u64 = (account.rent_epoch..=self.epoch)
|
|
|
|
.map(|epoch| self.epoch_schedule.get_slots_in_epoch(epoch + 1))
|
|
|
|
.sum();
|
|
|
|
|
2020-06-08 17:38:14 -07:00
|
|
|
// avoid infinite rent in rust 1.45
|
|
|
|
let years_elapsed = if self.slots_per_year != 0.0 {
|
|
|
|
slots_elapsed as f64 / self.slots_per_year
|
|
|
|
} else {
|
|
|
|
0.0
|
|
|
|
};
|
|
|
|
|
|
|
|
let (rent_due, exempt) =
|
|
|
|
self.rent
|
|
|
|
.due(account.lamports, account.data.len(), years_elapsed);
|
2019-08-23 14:04:53 -07:00
|
|
|
|
|
|
|
if exempt || rent_due != 0 {
|
|
|
|
if account.lamports > rent_due {
|
2020-08-16 22:22:16 -07:00
|
|
|
account.rent_epoch = self.epoch
|
2020-10-21 06:19:25 -07:00
|
|
|
+ if rent_fix_enabled && exempt {
|
2020-08-16 22:22:16 -07:00
|
|
|
// Rent isn't collected for the next epoch
|
|
|
|
// Make sure to check exempt status later in curent epoch again
|
|
|
|
0
|
|
|
|
} else {
|
|
|
|
// Rent is collected for next epoch
|
|
|
|
1
|
|
|
|
};
|
2019-08-23 14:04:53 -07:00
|
|
|
account.lamports -= rent_due;
|
2019-11-13 21:26:49 -08:00
|
|
|
rent_due
|
2019-08-23 14:04:53 -07:00
|
|
|
} else {
|
2019-11-13 21:26:49 -08:00
|
|
|
let rent_charged = account.lamports;
|
|
|
|
*account = Account::default();
|
|
|
|
rent_charged
|
2019-08-23 14:04:53 -07:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// maybe collect rent later, leave account alone
|
2019-11-13 21:26:49 -08:00
|
|
|
0
|
2019-08-23 14:04:53 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-08-11 08:04:32 -07:00
|
|
|
|
|
|
|
#[must_use = "add to Bank::collected_rent"]
|
2020-10-21 06:19:25 -07:00
|
|
|
pub fn collect_from_created_account(
|
|
|
|
&self,
|
|
|
|
address: &Pubkey,
|
|
|
|
account: &mut Account,
|
|
|
|
enable_new_behavior: bool,
|
|
|
|
) -> u64 {
|
2020-08-11 08:04:32 -07:00
|
|
|
// initialize rent_epoch as created at this epoch
|
|
|
|
account.rent_epoch = self.epoch;
|
2020-10-21 06:19:25 -07:00
|
|
|
self.collect_from_existing_account(address, account, enable_new_behavior)
|
2020-08-11 08:04:32 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_collect_from_account_created_and_existing() {
|
|
|
|
let old_lamports = 1000;
|
|
|
|
let old_epoch = 1;
|
|
|
|
let new_epoch = 3;
|
|
|
|
|
|
|
|
let (mut created_account, mut existing_account) = {
|
2020-12-13 17:26:34 -08:00
|
|
|
let account = Account {
|
|
|
|
lamports: old_lamports,
|
|
|
|
rent_epoch: old_epoch,
|
|
|
|
..Account::default()
|
|
|
|
};
|
2020-08-11 08:04:32 -07:00
|
|
|
|
|
|
|
(account.clone(), account)
|
|
|
|
};
|
|
|
|
|
2020-10-21 06:19:25 -07:00
|
|
|
let rent_collector = RentCollector::default().clone_with_epoch(new_epoch);
|
2020-08-11 08:04:32 -07:00
|
|
|
|
2020-08-16 22:22:16 -07:00
|
|
|
// collect rent on a newly-created account
|
2020-10-21 06:19:25 -07:00
|
|
|
let collected = rent_collector.collect_from_created_account(
|
2020-10-19 12:12:08 -07:00
|
|
|
&solana_sdk::pubkey::new_rand(),
|
2020-10-21 06:19:25 -07:00
|
|
|
&mut created_account,
|
|
|
|
true,
|
|
|
|
);
|
2020-08-11 08:04:32 -07:00
|
|
|
assert!(created_account.lamports < old_lamports);
|
|
|
|
assert_eq!(created_account.lamports + collected, old_lamports);
|
|
|
|
assert_ne!(created_account.rent_epoch, old_epoch);
|
|
|
|
|
2020-08-16 22:22:16 -07:00
|
|
|
// collect rent on a already-existing account
|
2020-10-21 06:19:25 -07:00
|
|
|
let collected = rent_collector.collect_from_existing_account(
|
2020-10-19 12:23:14 -07:00
|
|
|
&solana_sdk::pubkey::new_rand(),
|
2020-10-21 06:19:25 -07:00
|
|
|
&mut existing_account,
|
|
|
|
true,
|
|
|
|
);
|
2020-08-11 08:04:32 -07:00
|
|
|
assert!(existing_account.lamports < old_lamports);
|
|
|
|
assert_eq!(existing_account.lamports + collected, old_lamports);
|
|
|
|
assert_ne!(existing_account.rent_epoch, old_epoch);
|
|
|
|
|
2020-08-16 22:22:16 -07:00
|
|
|
// newly created account should be collected for less rent; thus more remaining balance
|
2020-08-11 08:04:32 -07:00
|
|
|
assert!(created_account.lamports > existing_account.lamports);
|
|
|
|
assert_eq!(created_account.rent_epoch, existing_account.rent_epoch);
|
|
|
|
}
|
2020-08-16 22:22:16 -07:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_rent_exempt_temporal_escape() {
|
|
|
|
let mut account = Account::default();
|
|
|
|
let epoch = 3;
|
|
|
|
let huge_lamports = 123_456_789_012;
|
|
|
|
let tiny_lamports = 789_012;
|
|
|
|
let mut collected;
|
2020-10-19 12:12:08 -07:00
|
|
|
let pubkey = solana_sdk::pubkey::new_rand();
|
2020-08-16 22:22:16 -07:00
|
|
|
|
|
|
|
account.lamports = huge_lamports;
|
|
|
|
assert_eq!(account.rent_epoch, 0);
|
|
|
|
|
|
|
|
// create a tested rent collector
|
2020-10-21 06:19:25 -07:00
|
|
|
let rent_collector = RentCollector::default().clone_with_epoch(epoch);
|
2020-08-16 22:22:16 -07:00
|
|
|
|
|
|
|
// first mark account as being collected while being rent-exempt
|
2020-10-21 06:19:25 -07:00
|
|
|
collected = rent_collector.collect_from_existing_account(&pubkey, &mut account, true);
|
2020-08-16 22:22:16 -07:00
|
|
|
assert_eq!(account.lamports, huge_lamports);
|
|
|
|
assert_eq!(collected, 0);
|
|
|
|
|
|
|
|
// decrease the balance not to be rent-exempt
|
|
|
|
account.lamports = tiny_lamports;
|
|
|
|
|
|
|
|
// ... and trigger another rent collection on the same epoch and check that rent is working
|
2020-10-21 06:19:25 -07:00
|
|
|
collected = rent_collector.collect_from_existing_account(&pubkey, &mut account, true);
|
2020-08-16 22:22:16 -07:00
|
|
|
assert_eq!(account.lamports, tiny_lamports - collected);
|
|
|
|
assert_ne!(collected, 0);
|
|
|
|
}
|
2019-08-23 14:04:53 -07:00
|
|
|
}
|