Add built-in programs to InvokeContext (#10383)

automerge

runtime/src/bank.rs

@@ -12,7 +12,7 @@ use crate::{
     blockhash_queue::BlockhashQueue,
     builtin_programs::get_builtin_programs,
     epoch_stakes::{EpochStakes, NodeVoteAccounts},
-    message_processor::{MessageProcessor, ProcessInstruction},
+    message_processor::MessageProcessor,
     nonce_utils,
     rent_collector::RentCollector,
     stakes::Stakes,
@@ -35,6 +35,7 @@ use solana_sdk::{
         Epoch, Slot, SlotCount, SlotIndex, UnixTimestamp, DEFAULT_TICKS_PER_SECOND,
         MAX_PROCESSING_AGE, MAX_RECENT_BLOCKHASHES, SECONDS_PER_DAY,
     },
+    entrypoint_native::ProcessInstruction,
     epoch_info::EpochInfo,
     epoch_schedule::EpochSchedule,
     fee_calculator::{FeeCalculator, FeeRateGovernor},

runtime/src/builtin_programs.rs

@@ -1,5 +1,5 @@
-use crate::{message_processor::ProcessInstruction, system_instruction_processor};
-use solana_sdk::{pubkey::Pubkey, system_program};
+use crate::system_instruction_processor;
+use solana_sdk::{entrypoint_native::ProcessInstruction, pubkey::Pubkey, system_program};
 
 pub struct BuiltinProgram {
     pub name: String,

runtime/src/message_processor.rs

@@ -3,7 +3,7 @@ use serde::{Deserialize, Serialize};
 use solana_sdk::{
     account::{create_keyed_readonly_accounts, Account, KeyedAccount},
     clock::Epoch,
-    entrypoint_native::InvokeContext,
+    entrypoint_native::{InvokeContext, ProcessInstruction},
     instruction::{CompiledInstruction, InstructionError},
     message::Message,
     native_loader,
@@ -151,21 +151,28 @@ impl PreAccount {
     }
 }
 
-#[derive(Debug, Default)]
+#[derive(Default)]
 pub struct ThisInvokeContext {
     pub program_ids: Vec<Pubkey>,
     pub rent: Rent,
     pub pre_accounts: Vec<PreAccount>,
+    pub programs: Vec<(Pubkey, ProcessInstruction)>,
 }
 impl ThisInvokeContext {
     const MAX_INVOCATION_DEPTH: usize = 5;
-    pub fn new(program_id: &Pubkey, rent: Rent, pre_accounts: Vec<PreAccount>) -> Self {
+    pub fn new(
+        program_id: &Pubkey,
+        rent: Rent,
+        pre_accounts: Vec<PreAccount>,
+        programs: Vec<(Pubkey, ProcessInstruction)>,
+    ) -> Self {
         let mut program_ids = Vec::with_capacity(Self::MAX_INVOCATION_DEPTH);
         program_ids.push(*program_id);
         Self {
             program_ids,
             rent,
             pre_accounts,
+            programs,
         }
     }
 }
@@ -207,9 +214,12 @@ impl InvokeContext for ThisInvokeContext {
             .last()
             .ok_or(InstructionError::GenericError)
     }
+
+    fn get_programs(&self) -> &[(Pubkey, ProcessInstruction)] {
+        &self.programs
+    }
 }
 
-pub type ProcessInstruction = fn(&Pubkey, &[KeyedAccount], &[u8]) -> Result<(), InstructionError>;
 pub type ProcessInstructionWithContext =
     fn(&Pubkey, &[KeyedAccount], &[u8], &mut dyn InvokeContext) -> Result<(), InstructionError>;
 
@@ -479,6 +489,7 @@ impl MessageProcessor {
             instruction.program_id(&message.account_keys),
             rent_collector.rent,
             pre_accounts,
+            self.programs.clone(),
         );
         let keyed_accounts =
             Self::create_keyed_accounts(message, instruction, executable_accounts, accounts)?;
@@ -555,7 +566,7 @@ mod tests {
             ))
         }
         let mut invoke_context =
-            ThisInvokeContext::new(&program_ids[0], Rent::default(), pre_accounts);
+            ThisInvokeContext::new(&program_ids[0], Rent::default(), pre_accounts, vec![]);
 
         // Check call depth increases and has a limit
         let mut depth_reached = 1;
@@ -1303,6 +1314,7 @@ mod tests {
             &caller_program_id,
             Rent::default(),
             vec![owned_preaccount, not_owned_preaccount],
+            vec![],
         );
         let metas = vec![
             AccountMeta::new(owned_key, false),