logos-blockchain/lssa
1
0
Fork 0
mirror of https://github.com/logos-blockchain/lssa.git synced 2026-01-02 13:23:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor call stack execution loop

Browse Source
This commit is contained in:
Sergio Chouhy 2025-11-27 09:54:14 -03:00
parent d1d2292028
commit 577fad6d5f
1 changed files with 22 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
nssa/src/public_transaction/transaction.rs
View File

@ -2,7 +2,7 @@ use std::collections::{HashMap, HashSet};
use nssa_core::{
account::{Account, AccountId, AccountWithMetadata},
program::{DEFAULT_PROGRAM_ID, validate_execution},
program::{ChainedCall, DEFAULT_PROGRAM_ID, validate_execution},
};
use sha2::{Digest, digest::FixedOutput};
@ -88,7 +88,7 @@ impl PublicTransaction {
}
// Build pre_states for execution
let mut input_pre_states: Vec<_> = message
let input_pre_states: Vec<_> = message
.account_ids
.iter()
.map(|account_id| {
@ -102,17 +102,27 @@ impl PublicTransaction {
let mut state_diff: HashMap<AccountId, Account> = HashMap::new();
let mut program_id = message.program_id;
let mut instruction_data = message.instruction_data.clone();
let mut chained_calls = Vec::new();
let initial_call = ChainedCall {
program_id: message.program_id,
instruction_data: message.instruction_data.clone(),
pre_states: input_pre_states,
};
let mut chained_calls = Vec::from_iter([initial_call]);
let mut chain_calls_counter = 0;
while let Some(chained_call) = chained_calls.pop() {
if chain_calls_counter > MAX_NUMBER_CHAINED_CALLS {
return Err(NssaError::MaxChainedCallsDepthExceeded);
}
for _i in 0..MAX_NUMBER_CHAINED_CALLS {
// Check the `program_id` corresponds to a deployed program
let Some(program) = state.programs().get(&program_id) else {
let Some(program) = state.programs().get(&chained_call.program_id) else {
return Err(NssaError::InvalidInput("Unknown program".into()));
};
let mut program_output = program.execute(&input_pre_states, &instruction_data)?;
let mut program_output =
program.execute(&chained_call.pre_states, &chained_call.instruction_data)?;
for pre in program_output.pre_states.iter() {
let account_id = pre.account_id;
@ -137,7 +147,7 @@ impl PublicTransaction {
if !validate_execution(
&program_output.pre_states,
&program_output.post_states,
program_id,
chained_call.program_id,
) {
return Err(NssaError::InvalidProgramBehavior);
}
@ -145,7 +155,7 @@ impl PublicTransaction {
// The invoked program claims the accounts with default program id.
for post in program_output.post_states.iter_mut() {
if post.program_owner == DEFAULT_PROGRAM_ID {
post.program_owner = program_id;
post.program_owner = chained_call.program_id;
}
}
@ -159,21 +169,10 @@ impl PublicTransaction {
}
chained_calls.extend_from_slice(&program_output.chained_calls);
if let Some(next_chained_call) = chained_calls.pop() {
program_id = next_chained_call.program_id;
instruction_data = next_chained_call.instruction_data;
input_pre_states = next_chained_call.pre_states;
} else {
break;
};
chain_calls_counter += 1;
}
if chained_calls.is_empty() {
Ok(state_diff)
} else {
Err(NssaError::MaxChainedCallsDepthExceeded)
}
Ok(state_diff)
}
}