mpl_utils/
account.rs

1use solana_program::{
2    account_info::AccountInfo,
3    entrypoint::ProgramResult,
4    msg,
5    program::{invoke, invoke_signed},
6    program_error::ProgramError,
7    pubkey::Pubkey,
8    rent::Rent,
9    system_instruction, system_program,
10    sysvar::Sysvar,
11};
12
13/// Create account almost from scratch, lifted from
14/// <https://github.com/solana-labs/solana-program-library/tree/master/associated-token-account/program/src/processor.rs#L51-L98>
15pub fn create_or_allocate_account_raw<'a>(
16    program_id: Pubkey,
17    new_account_info: &AccountInfo<'a>,
18    system_program_info: &AccountInfo<'a>,
19    payer_info: &AccountInfo<'a>,
20    size: usize,
21    signer_seeds: &[&[u8]],
22) -> ProgramResult {
23    let rent = &Rent::get()?;
24    let required_lamports = rent
25        .minimum_balance(size)
26        .max(1)
27        .saturating_sub(new_account_info.lamports());
28
29    if required_lamports > 0 {
30        msg!("Transfer {} lamports to the new account", required_lamports);
31        invoke(
32            &system_instruction::transfer(payer_info.key, new_account_info.key, required_lamports),
33            &[
34                payer_info.clone(),
35                new_account_info.clone(),
36                system_program_info.clone(),
37            ],
38        )?;
39    }
40
41    let accounts = &[new_account_info.clone(), system_program_info.clone()];
42
43    msg!("Allocate space for the account");
44    invoke_signed(
45        &system_instruction::allocate(new_account_info.key, size.try_into().unwrap()),
46        accounts,
47        &[signer_seeds],
48    )?;
49
50    msg!("Assign the account to the owning program");
51    invoke_signed(
52        &system_instruction::assign(new_account_info.key, &program_id),
53        accounts,
54        &[signer_seeds],
55    )?;
56
57    Ok(())
58}
59
60/// Resize an account using realloc, lifted from Solana Cookbook
61pub fn resize_or_reallocate_account_raw<'a>(
62    target_account: &AccountInfo<'a>,
63    funding_account: &AccountInfo<'a>,
64    system_program: &AccountInfo<'a>,
65    new_size: usize,
66) -> ProgramResult {
67    let rent = Rent::get()?;
68    let new_minimum_balance = rent.minimum_balance(new_size);
69    let lamports_diff = new_minimum_balance.abs_diff(target_account.lamports());
70    if new_size == target_account.data_len() {
71        return Ok(());
72    }
73
74    let account_infos = &[
75        funding_account.clone(),
76        target_account.clone(),
77        system_program.clone(),
78    ];
79
80    if new_size > target_account.data_len() {
81        invoke(
82            &system_instruction::transfer(funding_account.key, target_account.key, lamports_diff),
83            account_infos,
84        )?;
85    } else if target_account.owner == system_program.key {
86        invoke(
87            &system_instruction::transfer(target_account.key, funding_account.key, lamports_diff),
88            account_infos,
89        )?;
90    } else {
91        let target_account_lamports = target_account.lamports();
92        (**target_account.try_borrow_mut_lamports()?) = target_account_lamports
93            .checked_sub(lamports_diff)
94            .ok_or(ProgramError::InvalidRealloc)?;
95
96        let funding_account_lamports = funding_account.lamports();
97        (**funding_account.try_borrow_mut_lamports()?) = funding_account_lamports
98            .checked_add(lamports_diff)
99            .ok_or(ProgramError::InvalidRealloc)?;
100    }
101
102    target_account.realloc(new_size, false)
103}
104
105/// Close src_account and transfer lamports to dst_account, lifted from Solana Cookbook
106pub fn close_account_raw<'a>(
107    dest_account_info: &AccountInfo<'a>,
108    src_account_info: &AccountInfo<'a>,
109) -> ProgramResult {
110    let dest_starting_lamports = dest_account_info.lamports();
111    let mut dest_lamports_mut = dest_account_info
112        .lamports
113        .try_borrow_mut()
114        .map_err(|_| ProgramError::AccountBorrowFailed)?;
115    **dest_lamports_mut = dest_starting_lamports
116        .checked_add(src_account_info.lamports())
117        .ok_or(ProgramError::InvalidRealloc)?;
118
119    let mut src_lamports_mut = src_account_info
120        .lamports
121        .try_borrow_mut()
122        .map_err(|_| ProgramError::AccountBorrowFailed)?;
123    **src_lamports_mut = 0;
124
125    src_account_info.assign(&system_program::ID);
126    src_account_info.realloc(0, false).map_err(Into::into)
127}
mpl_utils/account.rs

mpl_utils/
account.rs
