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

Disallow closing the top stake account

Browse Source
This commit is contained in:
Hardhat Chad
2024-07-05 15:25:49 +00:00
parent cbfc7afc2b
commit bcba77eb7c
2 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
program/src/close.rs
View File

@@ -1,4 +1,8 @@
use ore_api::{loaders::*, state::Proof};
use ore_api::{
error::OreError,
loaders::*,
state::{Config, Proof},
};
use solana_program::{
account_info::AccountInfo, entrypoint::ProgramResult, program_error::ProgramError,
pubkey::Pubkey, system_program,
@@ -20,13 +24,21 @@ pub fn process_close<'a, 'info>(
_data: &[u8],
) -> ProgramResult {
// Load accounts
let [signer, proof_info, system_program] = accounts else {
let [signer, config_info, proof_info, system_program] = accounts else {
return Err(ProgramError::NotEnoughAccountKeys);
};
load_signer(signer)?;
load_config(config_info, false)?;
load_proof(proof_info, signer.key, true)?;
load_program(system_program, system_program::id())?;
// Validate the account is not the crowned top staker.
let config_data = config_info.data.borrow();
let config = Config::try_from_bytes(&config_data)?;
if config.top_staker.eq(proof_info.key) {
return Err(OreError::CannotClose.into());
}
// Validate balance is zero
let proof_data = proof_info.data.borrow();
let proof = Proof::try_from_bytes(&proof_data)?;