d0zingcat/ore
1
0
Fork 0
mirror of https://github.com/d0zingcat/ore.git synced 2026-05-17 23:16:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

add stricter address check

Browse Source
This commit is contained in:
Hardhat Chad
2024-10-11 20:58:56 +00:00
parent 8d2efb8916
commit 24aace2c24
2 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/src/loaders.rs
View File

@@ -6,12 +6,20 @@ use crate::{
};
pub trait OreAccountInfoValidation {
fn is_bus(&self) -> Result<&Self, ProgramError>;
fn is_config(&self) -> Result<&Self, ProgramError>;
fn is_treasury(&self) -> Result<&Self, ProgramError>;
fn is_treasury_tokens(&self) -> Result<&Self, ProgramError>;
}
impl OreAccountInfoValidation for AccountInfo<'_> {
fn is_bus(&self) -> Result<&Self, ProgramError> {
if !BUS_ADDRESSES.contains(self.key) {
return Err(ProgramError::InvalidSeeds);
}
Ok(self)
}
fn is_config(&self) -> Result<&Self, ProgramError> {
self.has_address(&CONFIG_ADDRESS)?
.is_type::<Config>(&crate::ID)