191 lines
5.8 KiB
Rust
191 lines
5.8 KiB
Rust
#[cfg(RUSTC_WITH_SPECIALIZATION)]
|
|
use solana_frozen_abi::abi_example::AbiExample;
|
|
use {
|
|
crate::system_instruction_processor,
|
|
solana_program_runtime::invoke_context::ProcessInstructionWithContext,
|
|
solana_sdk::{feature_set, pubkey::Pubkey, stake, system_program},
|
|
std::fmt,
|
|
};
|
|
|
|
#[derive(Clone)]
|
|
pub struct Builtin {
|
|
pub name: String,
|
|
pub id: Pubkey,
|
|
pub process_instruction_with_context: ProcessInstructionWithContext,
|
|
}
|
|
|
|
impl Builtin {
|
|
pub fn new(
|
|
name: &str,
|
|
id: Pubkey,
|
|
process_instruction_with_context: ProcessInstructionWithContext,
|
|
) -> Self {
|
|
Self {
|
|
name: name.to_string(),
|
|
id,
|
|
process_instruction_with_context,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl fmt::Debug for Builtin {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "Builtin [name={}, id={}]", self.name, self.id)
|
|
}
|
|
}
|
|
|
|
#[cfg(RUSTC_WITH_SPECIALIZATION)]
|
|
impl AbiExample for Builtin {
|
|
fn example() -> Self {
|
|
Self {
|
|
name: String::default(),
|
|
id: Pubkey::default(),
|
|
process_instruction_with_context: |_invoke_context| Ok(()),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct Builtins {
|
|
/// Builtin programs that are always available
|
|
pub genesis_builtins: Vec<Builtin>,
|
|
|
|
/// Dynamic feature transitions for builtin programs
|
|
pub feature_transitions: Vec<BuiltinFeatureTransition>,
|
|
}
|
|
|
|
/// Actions taken by a bank when managing the list of active builtin programs.
|
|
#[derive(Debug, Clone)]
|
|
pub enum BuiltinAction {
|
|
Add(Builtin),
|
|
Remove(Pubkey),
|
|
}
|
|
|
|
/// State transition enum used for adding and removing builtin programs through
|
|
/// feature activations.
|
|
#[derive(Debug, Clone, AbiExample)]
|
|
pub enum BuiltinFeatureTransition {
|
|
/// Add a builtin program if a feature is activated.
|
|
Add {
|
|
builtin: Builtin,
|
|
feature_id: Pubkey,
|
|
},
|
|
/// Remove a builtin program if a feature is activated or
|
|
/// retain a previously added builtin.
|
|
RemoveOrRetain {
|
|
previously_added_builtin: Builtin,
|
|
addition_feature_id: Pubkey,
|
|
removal_feature_id: Pubkey,
|
|
},
|
|
}
|
|
|
|
impl BuiltinFeatureTransition {
|
|
pub fn to_action(
|
|
&self,
|
|
should_apply_action_for_feature: &impl Fn(&Pubkey) -> bool,
|
|
) -> Option<BuiltinAction> {
|
|
match self {
|
|
Self::Add {
|
|
builtin,
|
|
feature_id,
|
|
} => {
|
|
if should_apply_action_for_feature(feature_id) {
|
|
Some(BuiltinAction::Add(builtin.clone()))
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
Self::RemoveOrRetain {
|
|
previously_added_builtin,
|
|
addition_feature_id,
|
|
removal_feature_id,
|
|
} => {
|
|
if should_apply_action_for_feature(removal_feature_id) {
|
|
Some(BuiltinAction::Remove(previously_added_builtin.id))
|
|
} else if should_apply_action_for_feature(addition_feature_id) {
|
|
// Retaining is no different from adding a new builtin.
|
|
Some(BuiltinAction::Add(previously_added_builtin.clone()))
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Builtin programs that are always available
|
|
fn genesis_builtins() -> Vec<Builtin> {
|
|
vec![
|
|
Builtin::new(
|
|
"system_program",
|
|
system_program::id(),
|
|
system_instruction_processor::process_instruction,
|
|
),
|
|
Builtin::new(
|
|
"vote_program",
|
|
solana_vote_program::id(),
|
|
solana_vote_program::vote_processor::process_instruction,
|
|
),
|
|
Builtin::new(
|
|
"stake_program",
|
|
stake::program::id(),
|
|
solana_stake_program::stake_instruction::process_instruction,
|
|
),
|
|
Builtin::new(
|
|
"config_program",
|
|
solana_config_program::id(),
|
|
solana_config_program::config_processor::process_instruction,
|
|
),
|
|
]
|
|
}
|
|
|
|
/// Dynamic feature transitions for builtin programs
|
|
fn builtin_feature_transitions() -> Vec<BuiltinFeatureTransition> {
|
|
vec![
|
|
BuiltinFeatureTransition::Add {
|
|
builtin: Builtin::new(
|
|
"compute_budget_program",
|
|
solana_sdk::compute_budget::id(),
|
|
solana_compute_budget_program::process_instruction,
|
|
),
|
|
feature_id: feature_set::add_compute_budget_program::id(),
|
|
},
|
|
BuiltinFeatureTransition::Add {
|
|
builtin: Builtin::new(
|
|
"address_lookup_table_program",
|
|
solana_address_lookup_table_program::id(),
|
|
solana_address_lookup_table_program::processor::process_instruction,
|
|
),
|
|
feature_id: feature_set::versioned_tx_message_enabled::id(),
|
|
},
|
|
BuiltinFeatureTransition::Add {
|
|
builtin: Builtin::new(
|
|
"zk_token_proof_program",
|
|
solana_zk_token_sdk::zk_token_proof_program::id(),
|
|
solana_zk_token_proof_program::process_instruction,
|
|
),
|
|
feature_id: feature_set::zk_token_sdk_enabled::id(),
|
|
},
|
|
]
|
|
}
|
|
|
|
pub(crate) fn get() -> Builtins {
|
|
Builtins {
|
|
genesis_builtins: genesis_builtins(),
|
|
feature_transitions: builtin_feature_transitions(),
|
|
}
|
|
}
|
|
|
|
/// Returns the addresses of all builtin programs.
|
|
pub fn get_pubkeys() -> Vec<Pubkey> {
|
|
let builtins = get();
|
|
|
|
let mut pubkeys = Vec::new();
|
|
pubkeys.extend(builtins.genesis_builtins.iter().map(|b| b.id));
|
|
pubkeys.extend(builtins.feature_transitions.iter().filter_map(|f| match f {
|
|
BuiltinFeatureTransition::Add { builtin, .. } => Some(builtin.id),
|
|
BuiltinFeatureTransition::RemoveOrRetain { .. } => None,
|
|
}));
|
|
pubkeys
|
|
}
|