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

Discriminator

Browse Source
This commit is contained in:
Hardhat Chad
2024-02-15 18:00:30 +00:00
parent c2692c89d2
commit 9476fc5b21
11 changed files with 107 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/processor/create_proof.rs
View File

@@ -5,7 +5,13 @@ use solana_program::{
program_error::ProgramError, pubkey::Pubkey, system_program,
};
use crate::{instruction::CreateProofArgs, loaders::*, state::Proof, utils::create_pda, PROOF};
use crate::{
instruction::CreateProofArgs,
loaders::*,
state::{Discriminator, Proof},
utils::create_pda,
PROOF,
};
pub fn process_create_proof<'a, 'info>(
_program_id: &Pubkey,
@@ -27,12 +33,13 @@ pub fn process_create_proof<'a, 'info>(
create_pda(
proof_info,
&crate::id(),
size_of::<Proof>(),
8 + size_of::<Proof>(),
&[PROOF, signer.key.as_ref(), &[args.bump]],
system_program,
signer,
)?;
let mut proof_data = proof_info.data.borrow_mut();
proof_data[0] = Proof::discriminator() as u8;
let mut proof = Proof::try_from_bytes_mut(&mut proof_data)?;
proof.bump = args.bump as u64;
proof.authority = *signer.key;

20
src/processor/initialize.rs
View File

@@ -8,6 +8,7 @@ use solana_program::{
};
use spl_token::state::Mint;
use crate::state::Discriminator;
use crate::{instruction::*, BUS, INITIAL_DIFFICULTY, MINT_ADDRESS, TREASURY_ADDRESS};
use crate::{
loaders::*,
@@ -70,31 +71,30 @@ pub fn process_initialize<'a, 'info>(
create_pda(
bus_infos[i],
&crate::id(),
size_of::<Bus>(),
8 + size_of::<Bus>(),
&[BUS, &[i as u8], &[bus_bumps[i]]],
system_program,
signer,
)?;
bus_infos[i].try_borrow_mut_data()?.copy_from_slice(
Bus {
bump: bus_bumps[i] as u32,
id: i as u32,
available_rewards: 0,
}
.to_bytes(),
);
let mut bus_data = bus_infos[i].try_borrow_mut_data()?;
bus_data[0] = Bus::discriminator() as u8;
let mut bus = Bus::try_from_bytes_mut(&mut bus_data)?;
bus.bump = bus_bumps[i] as u32;
bus.id = i as u32;
bus.available_rewards = 0;
}
// Initialize treasury
create_pda(
treasury_info,
&crate::id(),
size_of::<Treasury>(),
8 + size_of::<Treasury>(),
&[TREASURY, &[args.treasury_bump]],
system_program,
signer,
)?;
let mut treasury_data = treasury_info.data.borrow_mut();
treasury_data[0] = Treasury::discriminator() as u8;
let mut treasury = Treasury::try_from_bytes_mut(&mut treasury_data)?;
treasury.bump = args.treasury_bump as u64;
treasury.admin = *signer.key;