d0zingcat/ore
1
0
Fork 0
mirror of https://github.com/d0zingcat/ore.git synced 2026-05-14 07:26:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

use strict treasury_tokens loader

Browse Source
This commit is contained in:
Hardhat Chad
2024-07-02 13:08:35 +00:00
parent b55b6d338d
commit 3b1039b469
5 changed files with 31 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
api/src/consts.rs
View File

@@ -131,3 +131,16 @@ pub const TREASURY_ADDRESS: Pubkey =
/// The bump of the treasury account, for cpis.
pub const TREASURY_BUMP: u8 = ed25519::derive_program_address(&[TREASURY], &PROGRAM_ID).1;
/// The address of the treasury token account.
pub const TREASURY_TOKENS_ADDRESS: Pubkey = Pubkey::new_from_array(
ed25519::derive_program_address(
&[
unsafe { &*(&TREASURY_ADDRESS as *const Pubkey as *const [u8; 32]) },
unsafe { &*(&spl_token::id() as *const Pubkey as *const [u8; 32]) },
unsafe { &*(&MINT_ADDRESS as *const Pubkey as *const [u8; 32]) },
],
unsafe { &*(&spl_token::id() as *const Pubkey as *const [u8; 32]) },
)
.0,
);

14
api/src/loaders.rs
View File

@@ -252,6 +252,20 @@ pub fn load_treasury<'a, 'info>(
Ok(())
}
/// Errors if:
/// - Address does not match the expected treasury tokens address.
/// - Cannot load as a token account
pub fn load_treasury_tokens<'a, 'info>(
info: &'a AccountInfo<'info>,
is_writable: bool,
) -> Result<(), ProgramError> {
if info.key.ne(&TREASURY_TOKENS_ADDRESS) {
return Err(ProgramError::InvalidSeeds);
}
load_token_account(info, Some(&TREASURY_ADDRESS), &MINT_ADDRESS, is_writable)
}
/// Errors if:
/// - Owner is not SPL token program.
/// - Address does not match the expected mint address.