Cleanup program account def (#5833)

This commit is contained in:
Jack May
2019-09-06 17:32:14 -07:00
committed by GitHub
parent 81c36699c4
commit 1833db51a5
7 changed files with 86 additions and 53 deletions

View File

@@ -1,32 +1,19 @@
//! @brief Solana Rust-based BPF program entrypoint and its parameter types
extern crate alloc;
use crate::pubkey::Pubkey;
use crate::{account_info::AccountInfo, pubkey::Pubkey};
use alloc::vec::Vec;
use core::mem::size_of;
use core::slice::{from_raw_parts, from_raw_parts_mut};
/// Keyed Account
pub struct SolKeyedAccount<'a> {
/// Public key of the account
pub key: &'a Pubkey,
/// Public key of the account
pub is_signer: bool,
/// Number of lamports owned by this account
pub lamports: &'a mut u64,
/// On-chain data within this account
pub data: &'a mut [u8],
/// Program that owns this account
pub owner: &'a Pubkey,
}
/// User implemented program entrypoint
///
/// program_id: Program ID of the currently executing program
/// accounts: Accounts passed as part of the instruction
/// data: Instruction data
pub type ProcessInstruction =
fn(program_id: &Pubkey, accounts: &mut [SolKeyedAccount], data: &[u8]) -> bool;
fn(program_id: &Pubkey, accounts: &mut [AccountInfo], data: &[u8]) -> bool;
/// Programs indicate success with a return value of 0
pub const SUCCESS: u32 = 0;
@@ -42,9 +29,9 @@ macro_rules! entrypoint {
($process_instruction:ident) => {
#[no_mangle]
pub unsafe extern "C" fn entrypoint(input: *mut u8) -> u32 {
unsafe {
let (program_id, mut kas, data) = $crate::entrypoint::deserialize(input);
$process_instruction(&program_id, &mut kas, &data)
unsafe {
let (program_id, mut accounts, data) = $crate::entrypoint::deserialize(input);
$process_instruction(&program_id, &mut accounts, &data)
}
}
};
@@ -52,21 +39,19 @@ macro_rules! entrypoint {
/// Deserialize the input parameters
#[allow(clippy::type_complexity)]
pub unsafe fn deserialize<'a>(
input: *mut u8,
) -> (&'a Pubkey, Vec<SolKeyedAccount<'a>>, &'a [u8]) {
pub unsafe fn deserialize<'a>(input: *mut u8) -> (&'a Pubkey, Vec<AccountInfo<'a>>, &'a [u8]) {
let mut offset: usize = 0;
// Number of KeyedAccounts present
// Number of accounts present
#[allow(clippy::cast_ptr_alignment)]
let num_ka = *(input.add(offset) as *const u64) as usize;
let num_accounts = *(input.add(offset) as *const u64) as usize;
offset += size_of::<u64>();
// KeyedAccounts
// Account Infos
let mut kas = Vec::with_capacity(num_ka);
for _ in 0..num_ka {
let mut accounts = Vec::with_capacity(num_accounts);
for _ in 0..num_accounts {
let is_signer = {
#[allow(clippy::cast_ptr_alignment)]
let is_signer_val = *(input.add(offset) as *const u64);
@@ -91,7 +76,7 @@ pub unsafe fn deserialize<'a>(
let owner: &Pubkey = &*(input.add(offset) as *const Pubkey);
offset += size_of::<Pubkey>();
kas.push(SolKeyedAccount {
accounts.push(AccountInfo {
key,
is_signer,
lamports,
@@ -113,5 +98,5 @@ pub unsafe fn deserialize<'a>(
let program_id: &Pubkey = &*(input.add(offset) as *const Pubkey);
(program_id, kas, data)
(program_id, accounts, data)
}