97 lines
2.8 KiB
Rust
97 lines
2.8 KiB
Rust
use solana_sdk::{
|
|
account::Account,
|
|
fee_calculator::FeeCalculator,
|
|
genesis_config::GenesisConfig,
|
|
pubkey::Pubkey,
|
|
rent::Rent,
|
|
signature::{Keypair, KeypairUtil},
|
|
system_program::{self, solana_system_program},
|
|
};
|
|
use solana_stake_program::stake_state;
|
|
use solana_vote_program::vote_state;
|
|
|
|
// The default stake placed with the bootstrap leader
|
|
pub const BOOTSTRAP_LEADER_LAMPORTS: u64 = 42;
|
|
|
|
pub struct GenesisConfigInfo {
|
|
pub genesis_config: GenesisConfig,
|
|
pub mint_keypair: Keypair,
|
|
pub voting_keypair: Keypair,
|
|
}
|
|
|
|
pub fn create_genesis_config(mint_lamports: u64) -> GenesisConfigInfo {
|
|
create_genesis_config_with_leader(mint_lamports, &Pubkey::new_rand(), 0)
|
|
}
|
|
|
|
pub fn create_genesis_config_with_leader(
|
|
mint_lamports: u64,
|
|
bootstrap_leader_pubkey: &Pubkey,
|
|
bootstrap_leader_stake_lamports: u64,
|
|
) -> GenesisConfigInfo {
|
|
let mint_keypair = Keypair::new();
|
|
let bootstrap_leader_voting_keypair = Keypair::new();
|
|
let bootstrap_leader_staking_keypair = Keypair::new();
|
|
|
|
let bootstrap_leader_vote_account = vote_state::create_account(
|
|
&bootstrap_leader_voting_keypair.pubkey(),
|
|
&bootstrap_leader_pubkey,
|
|
0,
|
|
bootstrap_leader_stake_lamports,
|
|
);
|
|
|
|
let rent = Rent::free();
|
|
|
|
let bootstrap_leader_stake_account = stake_state::create_account(
|
|
&bootstrap_leader_staking_keypair.pubkey(),
|
|
&bootstrap_leader_voting_keypair.pubkey(),
|
|
&bootstrap_leader_vote_account,
|
|
&rent,
|
|
bootstrap_leader_stake_lamports,
|
|
);
|
|
|
|
let accounts = vec![
|
|
(
|
|
mint_keypair.pubkey(),
|
|
Account::new(mint_lamports, 0, &system_program::id()),
|
|
),
|
|
(
|
|
*bootstrap_leader_pubkey,
|
|
Account::new(BOOTSTRAP_LEADER_LAMPORTS, 0, &system_program::id()),
|
|
),
|
|
(
|
|
bootstrap_leader_voting_keypair.pubkey(),
|
|
bootstrap_leader_vote_account,
|
|
),
|
|
(
|
|
bootstrap_leader_staking_keypair.pubkey(),
|
|
bootstrap_leader_stake_account,
|
|
),
|
|
];
|
|
|
|
// Bare minimum program set
|
|
let native_instruction_processors = vec![
|
|
solana_system_program(),
|
|
solana_bpf_loader_program!(),
|
|
solana_vote_program!(),
|
|
solana_stake_program!(),
|
|
];
|
|
|
|
let fee_calculator = FeeCalculator::new(0, 0); // most tests can't handle transaction fees
|
|
let mut genesis_config = GenesisConfig {
|
|
accounts,
|
|
native_instruction_processors,
|
|
fee_calculator,
|
|
rent,
|
|
..GenesisConfig::default()
|
|
};
|
|
|
|
solana_stake_program::add_genesis_accounts(&mut genesis_config);
|
|
solana_storage_api::rewards_pools::add_genesis_accounts(&mut genesis_config);
|
|
|
|
GenesisConfigInfo {
|
|
genesis_config,
|
|
mint_keypair,
|
|
voting_keypair: bootstrap_leader_voting_keypair,
|
|
}
|
|
}
|