76 lines
1.8 KiB
Rust
76 lines
1.8 KiB
Rust
//! storage program
|
|
//! Receive mining proofs from miners, validate the answers
|
|
//! and give reward for good proofs.
|
|
|
|
use bincode::deserialize;
|
|
use hash::Hash;
|
|
use solana_sdk::account::Account;
|
|
use solana_sdk::pubkey::Pubkey;
|
|
use transaction::Transaction;
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
pub enum StorageProgram {
|
|
SubmitMiningProof { sha_state: Hash },
|
|
}
|
|
|
|
pub enum StorageError {
|
|
InvalidUserData,
|
|
}
|
|
|
|
const STORAGE_PROGRAM_ID: [u8; 32] = [
|
|
130, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
|
0,
|
|
];
|
|
|
|
impl StorageProgram {
|
|
pub fn check_id(program_id: &Pubkey) -> bool {
|
|
program_id.as_ref() == STORAGE_PROGRAM_ID
|
|
}
|
|
|
|
pub fn id() -> Pubkey {
|
|
Pubkey::new(&STORAGE_PROGRAM_ID)
|
|
}
|
|
|
|
pub fn get_balance(account: &Account) -> u64 {
|
|
account.tokens
|
|
}
|
|
|
|
pub fn process_transaction(
|
|
tx: &Transaction,
|
|
pix: usize,
|
|
_accounts: &mut [&mut Account],
|
|
) -> Result<(), StorageError> {
|
|
if let Ok(syscall) = deserialize(tx.userdata(pix)) {
|
|
match syscall {
|
|
StorageProgram::SubmitMiningProof { sha_state } => {
|
|
info!("Mining proof submitted with state {:?}", sha_state);
|
|
return Ok(());
|
|
}
|
|
}
|
|
} else {
|
|
return Err(StorageError::InvalidUserData);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::*;
|
|
use signature::Keypair;
|
|
use signature::KeypairUtil;
|
|
|
|
#[test]
|
|
fn test_storage_tx() {
|
|
let keypair = Keypair::new();
|
|
let tx = Transaction::new(
|
|
&keypair,
|
|
&[],
|
|
StorageProgram::id(),
|
|
&(),
|
|
Default::default(),
|
|
0,
|
|
);
|
|
assert!(StorageProgram::process_transaction(&tx, 0, &mut []).is_err());
|
|
}
|
|
}
|