use {
crate::program::spl_token_2022::{
error::TokenError,
extension::{set_account_type, AccountType, ExtensionType, StateWithExtensions},
processor::Processor,
state::Account,
},
solana_sdk::{
account_info::{next_account_info, AccountInfo},
entrypoint::ProgramResult,
msg,
program::invoke,
pubkey::Pubkey,
system_instruction,
sysvar::{rent::Rent, Sysvar},
},
};
pub fn process_reallocate(
program_id: &Pubkey,
accounts: &[AccountInfo],
new_extension_types: Vec<ExtensionType>,
) -> ProgramResult {
let account_info_iter = &mut accounts.iter();
let token_account_info = next_account_info(account_info_iter)?;
let payer_info = next_account_info(account_info_iter)?;
let system_program_info = next_account_info(account_info_iter)?;
let authority_info = next_account_info(account_info_iter)?;
let authority_info_data_len = authority_info.data_len();
let mut current_extension_types = {
let token_account = token_account_info.data.borrow();
let account = StateWithExtensions::<Account>::unpack(&token_account)?;
Processor::validate_owner(
program_id,
&account.base.owner,
authority_info,
authority_info_data_len,
account_info_iter.as_slice(),
)?;
account.get_extension_types()?
};
if new_extension_types
.iter()
.any(|extension_type| extension_type.get_account_type() != AccountType::Account)
{
return Err(TokenError::InvalidState.into());
}
current_extension_types.extend_from_slice(&new_extension_types);
let needed_account_len = ExtensionType::get_account_len::<Account>(¤t_extension_types);
if token_account_info.data_len() >= needed_account_len {
return Ok(());
}
msg!(
"account needs realloc, +{:?} bytes",
needed_account_len - token_account_info.data_len()
);
token_account_info.realloc(needed_account_len, false)?;
let rent = Rent::get()?;
let new_minimum_balance = rent.minimum_balance(needed_account_len);
let lamports_diff = new_minimum_balance.saturating_sub(token_account_info.lamports());
invoke(
&system_instruction::transfer(payer_info.key, token_account_info.key, lamports_diff),
&[
payer_info.clone(),
token_account_info.clone(),
system_program_info.clone(),
],
)?;
let mut token_account = token_account_info.data.borrow_mut();
set_account_type::<Account>(&mut token_account)?;
Ok(())
}