certusone
/
wormhole
mirror of https://github.com/certusone/wormhole.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

solitaire: force deserialization into a new stack frame 

Change-Id: I5ffe4ed6b831f4ddbad4e36fad31487fcd1765cd
This commit is contained in:
Reisen 2021-08-23 22:13:59 +00:00 committed by Hendrik Hofstadt
parent d3acd39c71
commit d884f7be17
1 changed files with 22 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
solana/solitaire/program/src/macros.rs
View File

@ -53,10 +53,30 @@ macro_rules! solitaire {
SolitaireError,
};
$(
// Generated module wrapping instruction handler.
//
// These are needed to force the compiler to generate a new function that has not
// been inlined, this provides a new stack frame. Without this, the stack frame for
// deserialization and the handler is the same as that used by solitaire, leading
// to bust stacks.
pub mod $row {
use super::*;
#[inline(never)]
pub fn execute<'a, 'b: 'a, 'c>(p: &Pubkey, a: &'c [AccountInfo<'b>], d: &[u8]) -> Result<()> {
let ix_data: $kind = BorshDeserialize::try_from_slice(d).map_err(|e| SolitaireError::InstructionDeserializeFailed(e))?;
let mut accounts = FromAccounts::from(p, &mut a.iter(), &())?;
$fn(&ExecutionContext{program_id: p, accounts: a}, &mut accounts, ix_data)?;
Persist::persist(&accounts, p)?;
Ok(())
}
}
)*
/// Generated:
/// This Instruction contains a 1-1 mapping for each enum variant to function call. The
/// function calls can be found below in the `api` module.
#[repr(u8)]
#[derive(BorshSerialize, BorshDeserialize)]
pub enum Instruction {
@ -68,16 +88,7 @@ macro_rules! solitaire {
pub fn dispatch<'a, 'b: 'a, 'c>(p: &Pubkey, a: &'c [AccountInfo<'b>], d: &[u8]) -> Result<()> {
match d[0] {
$(
n if n == Instruction::$row as u8 => {
(move || {
trace!("Dispatch: {}", stringify!($row));
let ix_data: $kind = BorshDeserialize::try_from_slice(&d[1..]).map_err(|e| SolitaireError::InstructionDeserializeFailed(e))?;
let mut accounts: $row = FromAccounts::from(p, &mut a.iter(), &())?;
$fn(&ExecutionContext{program_id: p, accounts: a}, &mut accounts, ix_data)?;
Persist::persist(&accounts, p)?;
Ok(())
})()
},
n if n == Instruction::$row as u8 => $row::execute(p, a, &d[1..]),
)*
other => {