1052 lines
38 KiB
Rust
1052 lines
38 KiB
Rust
//! A command-line executable for generating the chain's genesis config.
|
|
#![allow(clippy::integer_arithmetic)]
|
|
|
|
use {
|
|
base64::{prelude::BASE64_STANDARD, Engine},
|
|
clap::{crate_description, crate_name, value_t, value_t_or_exit, App, Arg, ArgMatches},
|
|
itertools::Itertools,
|
|
solana_clap_utils::{
|
|
input_parsers::{
|
|
cluster_type_of, pubkey_of, pubkeys_of, unix_timestamp_from_rfc3339_datetime,
|
|
},
|
|
input_validators::{
|
|
is_pubkey_or_keypair, is_rfc3339_datetime, is_slot, is_valid_percentage,
|
|
},
|
|
},
|
|
solana_entry::poh::compute_hashes_per_tick,
|
|
solana_genesis::{genesis_accounts::add_genesis_accounts, Base64Account},
|
|
solana_ledger::{blockstore::create_new_ledger, blockstore_options::LedgerColumnOptions},
|
|
solana_runtime::hardened_unpack::MAX_GENESIS_ARCHIVE_UNPACKED_SIZE,
|
|
solana_sdk::{
|
|
account::{Account, AccountSharedData, ReadableAccount, WritableAccount},
|
|
bpf_loader_upgradeable::UpgradeableLoaderState,
|
|
clock,
|
|
epoch_schedule::EpochSchedule,
|
|
fee_calculator::FeeRateGovernor,
|
|
genesis_config::{ClusterType, GenesisConfig},
|
|
inflation::Inflation,
|
|
native_token::sol_to_lamports,
|
|
poh_config::PohConfig,
|
|
pubkey::Pubkey,
|
|
rent::Rent,
|
|
signature::{Keypair, Signer},
|
|
signer::keypair::read_keypair_file,
|
|
stake::state::StakeState,
|
|
system_program, timing,
|
|
},
|
|
solana_stake_program::stake_state,
|
|
solana_vote_program::vote_state::{self, VoteState},
|
|
std::{
|
|
collections::HashMap,
|
|
error,
|
|
fs::File,
|
|
io::{self, Read},
|
|
path::PathBuf,
|
|
process,
|
|
str::FromStr,
|
|
time::Duration,
|
|
},
|
|
};
|
|
|
|
pub enum AccountFileFormat {
|
|
Pubkey,
|
|
Keypair,
|
|
}
|
|
|
|
fn pubkey_from_str(key_str: &str) -> Result<Pubkey, Box<dyn error::Error>> {
|
|
Pubkey::from_str(key_str).or_else(|_| {
|
|
let bytes: Vec<u8> = serde_json::from_str(key_str)?;
|
|
let keypair = Keypair::from_bytes(&bytes)
|
|
.map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e.to_string()))?;
|
|
Ok(keypair.pubkey())
|
|
})
|
|
}
|
|
|
|
pub fn load_genesis_accounts(file: &str, genesis_config: &mut GenesisConfig) -> io::Result<u64> {
|
|
let mut lamports = 0;
|
|
let accounts_file = File::open(file)?;
|
|
|
|
let genesis_accounts: HashMap<String, Base64Account> =
|
|
serde_yaml::from_reader(accounts_file)
|
|
.map_err(|err| io::Error::new(io::ErrorKind::Other, format!("{err:?}")))?;
|
|
|
|
for (key, account_details) in genesis_accounts {
|
|
let pubkey = pubkey_from_str(key.as_str()).map_err(|err| {
|
|
io::Error::new(
|
|
io::ErrorKind::Other,
|
|
format!("Invalid pubkey/keypair {key}: {err:?}"),
|
|
)
|
|
})?;
|
|
|
|
let owner_program_id = Pubkey::from_str(account_details.owner.as_str()).map_err(|err| {
|
|
io::Error::new(
|
|
io::ErrorKind::Other,
|
|
format!("Invalid owner: {}: {:?}", account_details.owner, err),
|
|
)
|
|
})?;
|
|
|
|
let mut account = AccountSharedData::new(account_details.balance, 0, &owner_program_id);
|
|
if account_details.data != "~" {
|
|
account.set_data(
|
|
BASE64_STANDARD
|
|
.decode(account_details.data.as_str())
|
|
.map_err(|err| {
|
|
io::Error::new(
|
|
io::ErrorKind::Other,
|
|
format!("Invalid account data: {}: {:?}", account_details.data, err),
|
|
)
|
|
})?,
|
|
);
|
|
}
|
|
account.set_executable(account_details.executable);
|
|
lamports += account.lamports();
|
|
genesis_config.add_account(pubkey, account);
|
|
}
|
|
|
|
Ok(lamports)
|
|
}
|
|
|
|
#[allow(clippy::cognitive_complexity)]
|
|
fn main() -> Result<(), Box<dyn error::Error>> {
|
|
let default_faucet_pubkey = solana_cli_config::Config::default().keypair_path;
|
|
let fee_rate_governor = FeeRateGovernor::default();
|
|
let (
|
|
default_target_lamports_per_signature,
|
|
default_target_signatures_per_slot,
|
|
default_fee_burn_percentage,
|
|
) = {
|
|
(
|
|
&fee_rate_governor.target_lamports_per_signature.to_string(),
|
|
&fee_rate_governor.target_signatures_per_slot.to_string(),
|
|
&fee_rate_governor.burn_percent.to_string(),
|
|
)
|
|
};
|
|
|
|
let rent = Rent::default();
|
|
let (
|
|
default_lamports_per_byte_year,
|
|
default_rent_exemption_threshold,
|
|
default_rent_burn_percentage,
|
|
) = {
|
|
(
|
|
&rent.lamports_per_byte_year.to_string(),
|
|
&rent.exemption_threshold.to_string(),
|
|
&rent.burn_percent.to_string(),
|
|
)
|
|
};
|
|
|
|
// vote account
|
|
let default_bootstrap_validator_lamports = &sol_to_lamports(500.0)
|
|
.max(VoteState::get_rent_exempt_reserve(&rent))
|
|
.to_string();
|
|
// stake account
|
|
let default_bootstrap_validator_stake_lamports = &sol_to_lamports(0.5)
|
|
.max(rent.minimum_balance(StakeState::size_of()))
|
|
.to_string();
|
|
|
|
let default_target_tick_duration =
|
|
timing::duration_as_us(&PohConfig::default().target_tick_duration);
|
|
let default_ticks_per_slot = &clock::DEFAULT_TICKS_PER_SLOT.to_string();
|
|
let default_cluster_type = "mainnet-beta";
|
|
let default_genesis_archive_unpacked_size = MAX_GENESIS_ARCHIVE_UNPACKED_SIZE.to_string();
|
|
|
|
let matches = App::new(crate_name!())
|
|
.about(crate_description!())
|
|
.version(solana_version::version!())
|
|
.arg(
|
|
Arg::with_name("creation_time")
|
|
.long("creation-time")
|
|
.value_name("RFC3339 DATE TIME")
|
|
.validator(is_rfc3339_datetime)
|
|
.takes_value(true)
|
|
.help("Time when the bootstrap validator will start the cluster [default: current system time]"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("bootstrap_validator")
|
|
.short("b")
|
|
.long("bootstrap-validator")
|
|
.value_name("IDENTITY_PUBKEY VOTE_PUBKEY STAKE_PUBKEY")
|
|
.takes_value(true)
|
|
.validator(is_pubkey_or_keypair)
|
|
.number_of_values(3)
|
|
.multiple(true)
|
|
.required(true)
|
|
.help("The bootstrap validator's identity, vote and stake pubkeys"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("ledger_path")
|
|
.short("l")
|
|
.long("ledger")
|
|
.value_name("DIR")
|
|
.takes_value(true)
|
|
.required(true)
|
|
.help("Use directory as persistent ledger location"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("faucet_lamports")
|
|
.short("t")
|
|
.long("faucet-lamports")
|
|
.value_name("LAMPORTS")
|
|
.takes_value(true)
|
|
.help("Number of lamports to assign to the faucet"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("faucet_pubkey")
|
|
.short("m")
|
|
.long("faucet-pubkey")
|
|
.value_name("PUBKEY")
|
|
.takes_value(true)
|
|
.validator(is_pubkey_or_keypair)
|
|
.requires("faucet_lamports")
|
|
.default_value(&default_faucet_pubkey)
|
|
.help("Path to file containing the faucet's pubkey"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("bootstrap_stake_authorized_pubkey")
|
|
.long("bootstrap-stake-authorized-pubkey")
|
|
.value_name("BOOTSTRAP STAKE AUTHORIZED PUBKEY")
|
|
.takes_value(true)
|
|
.validator(is_pubkey_or_keypair)
|
|
.help(
|
|
"Path to file containing the pubkey authorized to manage the bootstrap \
|
|
validator's stake [default: --bootstrap-validator IDENTITY_PUBKEY]",
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("bootstrap_validator_lamports")
|
|
.long("bootstrap-validator-lamports")
|
|
.value_name("LAMPORTS")
|
|
.takes_value(true)
|
|
.default_value(default_bootstrap_validator_lamports)
|
|
.help("Number of lamports to assign to the bootstrap validator"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("bootstrap_validator_stake_lamports")
|
|
.long("bootstrap-validator-stake-lamports")
|
|
.value_name("LAMPORTS")
|
|
.takes_value(true)
|
|
.default_value(default_bootstrap_validator_stake_lamports)
|
|
.help("Number of lamports to assign to the bootstrap validator's stake account"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("target_lamports_per_signature")
|
|
.long("target-lamports-per-signature")
|
|
.value_name("LAMPORTS")
|
|
.takes_value(true)
|
|
.default_value(default_target_lamports_per_signature)
|
|
.help(
|
|
"The cost in lamports that the cluster will charge for signature \
|
|
verification when the cluster is operating at target-signatures-per-slot",
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("lamports_per_byte_year")
|
|
.long("lamports-per-byte-year")
|
|
.value_name("LAMPORTS")
|
|
.takes_value(true)
|
|
.default_value(default_lamports_per_byte_year)
|
|
.help(
|
|
"The cost in lamports that the cluster will charge per byte per year \
|
|
for accounts with data",
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("rent_exemption_threshold")
|
|
.long("rent-exemption-threshold")
|
|
.value_name("NUMBER")
|
|
.takes_value(true)
|
|
.default_value(default_rent_exemption_threshold)
|
|
.help(
|
|
"amount of time (in years) the balance has to include rent for \
|
|
to qualify as rent exempted account",
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("rent_burn_percentage")
|
|
.long("rent-burn-percentage")
|
|
.value_name("NUMBER")
|
|
.takes_value(true)
|
|
.default_value(default_rent_burn_percentage)
|
|
.help("percentage of collected rent to burn")
|
|
.validator(is_valid_percentage),
|
|
)
|
|
.arg(
|
|
Arg::with_name("fee_burn_percentage")
|
|
.long("fee-burn-percentage")
|
|
.value_name("NUMBER")
|
|
.takes_value(true)
|
|
.default_value(default_fee_burn_percentage)
|
|
.help("percentage of collected fee to burn")
|
|
.validator(is_valid_percentage),
|
|
)
|
|
.arg(
|
|
Arg::with_name("vote_commission_percentage")
|
|
.long("vote-commission-percentage")
|
|
.value_name("NUMBER")
|
|
.takes_value(true)
|
|
.default_value("100")
|
|
.help("percentage of vote commission")
|
|
.validator(is_valid_percentage),
|
|
)
|
|
.arg(
|
|
Arg::with_name("target_signatures_per_slot")
|
|
.long("target-signatures-per-slot")
|
|
.value_name("NUMBER")
|
|
.takes_value(true)
|
|
.default_value(default_target_signatures_per_slot)
|
|
.help(
|
|
"Used to estimate the desired processing capacity of the cluster. \
|
|
When the latest slot processes fewer/greater signatures than this \
|
|
value, the lamports-per-signature fee will decrease/increase for \
|
|
the next slot. A value of 0 disables signature-based fee adjustments",
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("target_tick_duration")
|
|
.long("target-tick-duration")
|
|
.value_name("MILLIS")
|
|
.takes_value(true)
|
|
.help("The target tick rate of the cluster in milliseconds"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("hashes_per_tick")
|
|
.long("hashes-per-tick")
|
|
.value_name("NUM_HASHES|\"auto\"|\"sleep\"")
|
|
.takes_value(true)
|
|
.default_value("auto")
|
|
.help(
|
|
"How many PoH hashes to roll before emitting the next tick. \
|
|
If \"auto\", determine based on --target-tick-duration \
|
|
and the hash rate of this computer. If \"sleep\", for development \
|
|
sleep for --target-tick-duration instead of hashing",
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("ticks_per_slot")
|
|
.long("ticks-per-slot")
|
|
.value_name("TICKS")
|
|
.takes_value(true)
|
|
.default_value(default_ticks_per_slot)
|
|
.help("The number of ticks in a slot"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("slots_per_epoch")
|
|
.long("slots-per-epoch")
|
|
.value_name("SLOTS")
|
|
.validator(is_slot)
|
|
.takes_value(true)
|
|
.help("The number of slots in an epoch"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("enable_warmup_epochs")
|
|
.long("enable-warmup-epochs")
|
|
.help(
|
|
"When enabled epochs start short and will grow. \
|
|
Useful for warming up stake quickly during development"
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("primordial_accounts_file")
|
|
.long("primordial-accounts-file")
|
|
.value_name("FILENAME")
|
|
.takes_value(true)
|
|
.multiple(true)
|
|
.help("The location of pubkey for primordial accounts and balance"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("cluster_type")
|
|
.long("cluster-type")
|
|
.possible_values(&ClusterType::STRINGS)
|
|
.takes_value(true)
|
|
.default_value(default_cluster_type)
|
|
.help(
|
|
"Selects the features that will be enabled for the cluster"
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("max_genesis_archive_unpacked_size")
|
|
.long("max-genesis-archive-unpacked-size")
|
|
.value_name("NUMBER")
|
|
.takes_value(true)
|
|
.default_value(&default_genesis_archive_unpacked_size)
|
|
.help(
|
|
"maximum total uncompressed file size of created genesis archive",
|
|
),
|
|
)
|
|
.arg(
|
|
Arg::with_name("bpf_program")
|
|
.long("bpf-program")
|
|
.value_name("ADDRESS LOADER SBF_PROGRAM.SO")
|
|
.takes_value(true)
|
|
.number_of_values(3)
|
|
.multiple(true)
|
|
.help("Install a SBF program at the given address"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("upgradeable_program")
|
|
.long("upgradeable-program")
|
|
.value_name("ADDRESS UPGRADEABLE_LOADER SBF_PROGRAM.SO UPGRADE_AUTHORITY")
|
|
.takes_value(true)
|
|
.number_of_values(4)
|
|
.multiple(true)
|
|
.help("Install an upgradeable SBF program at the given address with the given upgrade authority (or \"none\")"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("inflation")
|
|
.required(false)
|
|
.long("inflation")
|
|
.takes_value(true)
|
|
.possible_values(&["pico", "full", "none"])
|
|
.help("Selects inflation"),
|
|
)
|
|
.get_matches();
|
|
|
|
let ledger_path = PathBuf::from(matches.value_of("ledger_path").unwrap());
|
|
|
|
let rent = Rent {
|
|
lamports_per_byte_year: value_t_or_exit!(matches, "lamports_per_byte_year", u64),
|
|
exemption_threshold: value_t_or_exit!(matches, "rent_exemption_threshold", f64),
|
|
burn_percent: value_t_or_exit!(matches, "rent_burn_percentage", u8),
|
|
};
|
|
|
|
fn rent_exempt_check(matches: &ArgMatches<'_>, name: &str, exempt: u64) -> io::Result<u64> {
|
|
let lamports = value_t_or_exit!(matches, name, u64);
|
|
|
|
if lamports < exempt {
|
|
Err(io::Error::new(
|
|
io::ErrorKind::Other,
|
|
format!(
|
|
"error: insufficient {name}: {lamports} for rent exemption, requires {exempt}"
|
|
),
|
|
))
|
|
} else {
|
|
Ok(lamports)
|
|
}
|
|
}
|
|
|
|
let bootstrap_validator_pubkeys = pubkeys_of(&matches, "bootstrap_validator").unwrap();
|
|
assert_eq!(bootstrap_validator_pubkeys.len() % 3, 0);
|
|
|
|
// Ensure there are no duplicated pubkeys in the --bootstrap-validator list
|
|
{
|
|
let mut v = bootstrap_validator_pubkeys.clone();
|
|
v.sort();
|
|
v.dedup();
|
|
if v.len() != bootstrap_validator_pubkeys.len() {
|
|
eprintln!("Error: --bootstrap-validator pubkeys cannot be duplicated");
|
|
process::exit(1);
|
|
}
|
|
}
|
|
|
|
let bootstrap_validator_lamports =
|
|
value_t_or_exit!(matches, "bootstrap_validator_lamports", u64);
|
|
|
|
let bootstrap_validator_stake_lamports = rent_exempt_check(
|
|
&matches,
|
|
"bootstrap_validator_stake_lamports",
|
|
rent.minimum_balance(StakeState::size_of()),
|
|
)?;
|
|
|
|
let bootstrap_stake_authorized_pubkey =
|
|
pubkey_of(&matches, "bootstrap_stake_authorized_pubkey");
|
|
let faucet_lamports = value_t!(matches, "faucet_lamports", u64).unwrap_or(0);
|
|
let faucet_pubkey = pubkey_of(&matches, "faucet_pubkey");
|
|
|
|
let ticks_per_slot = value_t_or_exit!(matches, "ticks_per_slot", u64);
|
|
|
|
let mut fee_rate_governor = FeeRateGovernor::new(
|
|
value_t_or_exit!(matches, "target_lamports_per_signature", u64),
|
|
value_t_or_exit!(matches, "target_signatures_per_slot", u64),
|
|
);
|
|
fee_rate_governor.burn_percent = value_t_or_exit!(matches, "fee_burn_percentage", u8);
|
|
|
|
let mut poh_config = PohConfig {
|
|
target_tick_duration: if matches.is_present("target_tick_duration") {
|
|
Duration::from_micros(value_t_or_exit!(matches, "target_tick_duration", u64))
|
|
} else {
|
|
Duration::from_micros(default_target_tick_duration)
|
|
},
|
|
..PohConfig::default()
|
|
};
|
|
|
|
let cluster_type = cluster_type_of(&matches, "cluster_type").unwrap();
|
|
|
|
match matches.value_of("hashes_per_tick").unwrap() {
|
|
"auto" => match cluster_type {
|
|
ClusterType::Development => {
|
|
let hashes_per_tick =
|
|
compute_hashes_per_tick(poh_config.target_tick_duration, 1_000_000);
|
|
poh_config.hashes_per_tick = Some(hashes_per_tick / 2); // use 50% of peak ability
|
|
}
|
|
ClusterType::Devnet | ClusterType::Testnet | ClusterType::MainnetBeta => {
|
|
poh_config.hashes_per_tick = Some(clock::DEFAULT_HASHES_PER_TICK);
|
|
}
|
|
},
|
|
"sleep" => {
|
|
poh_config.hashes_per_tick = None;
|
|
}
|
|
_ => {
|
|
poh_config.hashes_per_tick = Some(value_t_or_exit!(matches, "hashes_per_tick", u64));
|
|
}
|
|
}
|
|
|
|
let slots_per_epoch = if matches.value_of("slots_per_epoch").is_some() {
|
|
value_t_or_exit!(matches, "slots_per_epoch", u64)
|
|
} else {
|
|
match cluster_type {
|
|
ClusterType::Development => clock::DEFAULT_DEV_SLOTS_PER_EPOCH,
|
|
ClusterType::Devnet | ClusterType::Testnet | ClusterType::MainnetBeta => {
|
|
clock::DEFAULT_SLOTS_PER_EPOCH
|
|
}
|
|
}
|
|
};
|
|
let epoch_schedule = EpochSchedule::custom(
|
|
slots_per_epoch,
|
|
slots_per_epoch,
|
|
matches.is_present("enable_warmup_epochs"),
|
|
);
|
|
|
|
let mut genesis_config = GenesisConfig {
|
|
native_instruction_processors: vec![],
|
|
ticks_per_slot,
|
|
poh_config,
|
|
fee_rate_governor,
|
|
rent,
|
|
epoch_schedule,
|
|
cluster_type,
|
|
..GenesisConfig::default()
|
|
};
|
|
|
|
if let Ok(raw_inflation) = value_t!(matches, "inflation", String) {
|
|
let inflation = match raw_inflation.as_str() {
|
|
"pico" => Inflation::pico(),
|
|
"full" => Inflation::full(),
|
|
"none" => Inflation::new_disabled(),
|
|
_ => unreachable!(),
|
|
};
|
|
genesis_config.inflation = inflation;
|
|
}
|
|
|
|
let commission = value_t_or_exit!(matches, "vote_commission_percentage", u8);
|
|
|
|
let mut bootstrap_validator_pubkeys_iter = bootstrap_validator_pubkeys.iter();
|
|
loop {
|
|
let identity_pubkey = match bootstrap_validator_pubkeys_iter.next() {
|
|
None => break,
|
|
Some(identity_pubkey) => identity_pubkey,
|
|
};
|
|
let vote_pubkey = bootstrap_validator_pubkeys_iter.next().unwrap();
|
|
let stake_pubkey = bootstrap_validator_pubkeys_iter.next().unwrap();
|
|
|
|
genesis_config.add_account(
|
|
*identity_pubkey,
|
|
AccountSharedData::new(bootstrap_validator_lamports, 0, &system_program::id()),
|
|
);
|
|
|
|
let vote_account = vote_state::create_account_with_authorized(
|
|
identity_pubkey,
|
|
identity_pubkey,
|
|
identity_pubkey,
|
|
commission,
|
|
VoteState::get_rent_exempt_reserve(&rent).max(1),
|
|
);
|
|
|
|
genesis_config.add_account(
|
|
*stake_pubkey,
|
|
stake_state::create_account(
|
|
bootstrap_stake_authorized_pubkey
|
|
.as_ref()
|
|
.unwrap_or(identity_pubkey),
|
|
vote_pubkey,
|
|
&vote_account,
|
|
&rent,
|
|
bootstrap_validator_stake_lamports,
|
|
),
|
|
);
|
|
|
|
genesis_config.add_account(*vote_pubkey, vote_account);
|
|
}
|
|
|
|
if let Some(creation_time) = unix_timestamp_from_rfc3339_datetime(&matches, "creation_time") {
|
|
genesis_config.creation_time = creation_time;
|
|
}
|
|
|
|
if let Some(faucet_pubkey) = faucet_pubkey {
|
|
genesis_config.add_account(
|
|
faucet_pubkey,
|
|
AccountSharedData::new(faucet_lamports, 0, &system_program::id()),
|
|
);
|
|
}
|
|
|
|
solana_stake_program::add_genesis_accounts(&mut genesis_config);
|
|
if genesis_config.cluster_type == ClusterType::Development {
|
|
solana_runtime::genesis_utils::activate_all_features(&mut genesis_config);
|
|
}
|
|
|
|
if let Some(files) = matches.values_of("primordial_accounts_file") {
|
|
for file in files {
|
|
load_genesis_accounts(file, &mut genesis_config)?;
|
|
}
|
|
}
|
|
|
|
let max_genesis_archive_unpacked_size =
|
|
value_t_or_exit!(matches, "max_genesis_archive_unpacked_size", u64);
|
|
|
|
let issued_lamports = genesis_config
|
|
.accounts
|
|
.values()
|
|
.map(|account| account.lamports)
|
|
.sum::<u64>();
|
|
|
|
add_genesis_accounts(&mut genesis_config, issued_lamports - faucet_lamports);
|
|
|
|
let parse_address = |address: &str, input_type: &str| {
|
|
address.parse::<Pubkey>().unwrap_or_else(|err| {
|
|
eprintln!("Error: invalid {input_type} {address}: {err}");
|
|
process::exit(1);
|
|
})
|
|
};
|
|
|
|
let parse_program_data = |program: &str| {
|
|
let mut program_data = vec![];
|
|
File::open(program)
|
|
.and_then(|mut file| file.read_to_end(&mut program_data))
|
|
.unwrap_or_else(|err| {
|
|
eprintln!("Error: failed to read {program}: {err}");
|
|
process::exit(1);
|
|
});
|
|
program_data
|
|
};
|
|
|
|
if let Some(values) = matches.values_of("bpf_program") {
|
|
for (address, loader, program) in values.tuples() {
|
|
let address = parse_address(address, "address");
|
|
let loader = parse_address(loader, "loader");
|
|
let program_data = parse_program_data(program);
|
|
genesis_config.add_account(
|
|
address,
|
|
AccountSharedData::from(Account {
|
|
lamports: genesis_config.rent.minimum_balance(program_data.len()),
|
|
data: program_data,
|
|
executable: true,
|
|
owner: loader,
|
|
rent_epoch: 0,
|
|
}),
|
|
);
|
|
}
|
|
}
|
|
|
|
if let Some(values) = matches.values_of("upgradeable_program") {
|
|
for (address, loader, program, upgrade_authority) in values.tuples() {
|
|
let address = parse_address(address, "address");
|
|
let loader = parse_address(loader, "loader");
|
|
let program_data_elf = parse_program_data(program);
|
|
let upgrade_authority_address = if upgrade_authority == "none" {
|
|
Pubkey::default()
|
|
} else {
|
|
upgrade_authority.parse::<Pubkey>().unwrap_or_else(|_| {
|
|
read_keypair_file(upgrade_authority)
|
|
.map(|keypair| keypair.pubkey())
|
|
.unwrap_or_else(|err| {
|
|
eprintln!(
|
|
"Error: invalid upgrade_authority {upgrade_authority}: {err}"
|
|
);
|
|
process::exit(1);
|
|
})
|
|
})
|
|
};
|
|
|
|
let (programdata_address, _) =
|
|
Pubkey::find_program_address(&[address.as_ref()], &loader);
|
|
let mut program_data = bincode::serialize(&UpgradeableLoaderState::ProgramData {
|
|
slot: 0,
|
|
upgrade_authority_address: Some(upgrade_authority_address),
|
|
})
|
|
.unwrap();
|
|
program_data.extend_from_slice(&program_data_elf);
|
|
genesis_config.add_account(
|
|
programdata_address,
|
|
AccountSharedData::from(Account {
|
|
lamports: genesis_config.rent.minimum_balance(program_data.len()),
|
|
data: program_data,
|
|
owner: loader,
|
|
executable: false,
|
|
rent_epoch: 0,
|
|
}),
|
|
);
|
|
|
|
let program_data = bincode::serialize(&UpgradeableLoaderState::Program {
|
|
programdata_address,
|
|
})
|
|
.unwrap();
|
|
genesis_config.add_account(
|
|
address,
|
|
AccountSharedData::from(Account {
|
|
lamports: genesis_config.rent.minimum_balance(program_data.len()),
|
|
data: program_data,
|
|
owner: loader,
|
|
executable: true,
|
|
rent_epoch: 0,
|
|
}),
|
|
);
|
|
}
|
|
}
|
|
|
|
solana_logger::setup();
|
|
create_new_ledger(
|
|
&ledger_path,
|
|
&genesis_config,
|
|
max_genesis_archive_unpacked_size,
|
|
LedgerColumnOptions::default(),
|
|
)?;
|
|
|
|
println!("{genesis_config}");
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use {
|
|
super::*,
|
|
solana_sdk::genesis_config::GenesisConfig,
|
|
std::{collections::HashMap, fs::remove_file, io::Write, path::Path},
|
|
};
|
|
|
|
#[test]
|
|
fn test_append_primordial_accounts_to_genesis() {
|
|
// Test invalid file returns error
|
|
assert!(load_genesis_accounts("unknownfile", &mut GenesisConfig::default()).is_err());
|
|
|
|
let mut genesis_config = GenesisConfig::default();
|
|
|
|
let mut genesis_accounts = HashMap::new();
|
|
genesis_accounts.insert(
|
|
solana_sdk::pubkey::new_rand().to_string(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 2,
|
|
executable: false,
|
|
data: String::from("aGVsbG8="),
|
|
},
|
|
);
|
|
genesis_accounts.insert(
|
|
solana_sdk::pubkey::new_rand().to_string(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 1,
|
|
executable: true,
|
|
data: String::from("aGVsbG8gd29ybGQ="),
|
|
},
|
|
);
|
|
genesis_accounts.insert(
|
|
solana_sdk::pubkey::new_rand().to_string(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 3,
|
|
executable: true,
|
|
data: String::from("bWUgaGVsbG8gdG8gd29ybGQ="),
|
|
},
|
|
);
|
|
|
|
let serialized = serde_yaml::to_string(&genesis_accounts).unwrap();
|
|
let path = Path::new("test_append_primordial_accounts_to_genesis.yml");
|
|
let mut file = File::create(path).unwrap();
|
|
file.write_all(b"---\n").unwrap();
|
|
file.write_all(&serialized.into_bytes()).unwrap();
|
|
|
|
load_genesis_accounts(
|
|
"test_append_primordial_accounts_to_genesis.yml",
|
|
&mut genesis_config,
|
|
)
|
|
.expect("test_append_primordial_accounts_to_genesis.yml");
|
|
// Test valid file returns ok
|
|
|
|
remove_file(path).unwrap();
|
|
|
|
{
|
|
// Test all accounts were added
|
|
assert_eq!(genesis_config.accounts.len(), genesis_accounts.len());
|
|
|
|
// Test account data matches
|
|
for (pubkey_str, b64_account) in genesis_accounts.iter() {
|
|
let pubkey = pubkey_str.parse().unwrap();
|
|
assert_eq!(
|
|
b64_account.owner,
|
|
genesis_config.accounts[&pubkey].owner.to_string()
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.balance,
|
|
genesis_config.accounts[&pubkey].lamports
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.executable,
|
|
genesis_config.accounts[&pubkey].executable
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.data,
|
|
BASE64_STANDARD.encode(&genesis_config.accounts[&pubkey].data)
|
|
);
|
|
}
|
|
}
|
|
|
|
// Test more accounts can be appended
|
|
let mut genesis_accounts1 = HashMap::new();
|
|
genesis_accounts1.insert(
|
|
solana_sdk::pubkey::new_rand().to_string(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 6,
|
|
executable: true,
|
|
data: String::from("eW91IGFyZQ=="),
|
|
},
|
|
);
|
|
genesis_accounts1.insert(
|
|
solana_sdk::pubkey::new_rand().to_string(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 5,
|
|
executable: false,
|
|
data: String::from("bWV0YSBzdHJpbmc="),
|
|
},
|
|
);
|
|
genesis_accounts1.insert(
|
|
solana_sdk::pubkey::new_rand().to_string(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 10,
|
|
executable: false,
|
|
data: String::from("YmFzZTY0IHN0cmluZw=="),
|
|
},
|
|
);
|
|
|
|
let serialized = serde_yaml::to_string(&genesis_accounts1).unwrap();
|
|
let path = Path::new("test_append_primordial_accounts_to_genesis.yml");
|
|
let mut file = File::create(path).unwrap();
|
|
file.write_all(b"---\n").unwrap();
|
|
file.write_all(&serialized.into_bytes()).unwrap();
|
|
|
|
load_genesis_accounts(
|
|
"test_append_primordial_accounts_to_genesis.yml",
|
|
&mut genesis_config,
|
|
)
|
|
.expect("test_append_primordial_accounts_to_genesis.yml");
|
|
|
|
remove_file(path).unwrap();
|
|
|
|
// Test total number of accounts is correct
|
|
assert_eq!(
|
|
genesis_config.accounts.len(),
|
|
genesis_accounts.len() + genesis_accounts1.len()
|
|
);
|
|
|
|
// Test old accounts are still there
|
|
for (pubkey_str, b64_account) in genesis_accounts.iter() {
|
|
let pubkey = &pubkey_str.parse().unwrap();
|
|
assert_eq!(
|
|
b64_account.balance,
|
|
genesis_config.accounts[pubkey].lamports,
|
|
);
|
|
}
|
|
|
|
// Test new account data matches
|
|
for (pubkey_str, b64_account) in genesis_accounts1.iter() {
|
|
let pubkey = pubkey_str.parse().unwrap();
|
|
assert_eq!(
|
|
b64_account.owner,
|
|
genesis_config.accounts[&pubkey].owner.to_string()
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.balance,
|
|
genesis_config.accounts[&pubkey].lamports,
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.executable,
|
|
genesis_config.accounts[&pubkey].executable,
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.data,
|
|
BASE64_STANDARD.encode(&genesis_config.accounts[&pubkey].data),
|
|
);
|
|
}
|
|
|
|
// Test accounts from keypairs can be appended
|
|
let account_keypairs: Vec<_> = (0..3).map(|_| Keypair::new()).collect();
|
|
let mut genesis_accounts2 = HashMap::new();
|
|
genesis_accounts2.insert(
|
|
serde_json::to_string(&account_keypairs[0].to_bytes().to_vec()).unwrap(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 20,
|
|
executable: true,
|
|
data: String::from("Y2F0IGRvZw=="),
|
|
},
|
|
);
|
|
genesis_accounts2.insert(
|
|
serde_json::to_string(&account_keypairs[1].to_bytes().to_vec()).unwrap(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 15,
|
|
executable: false,
|
|
data: String::from("bW9ua2V5IGVsZXBoYW50"),
|
|
},
|
|
);
|
|
genesis_accounts2.insert(
|
|
serde_json::to_string(&account_keypairs[2].to_bytes().to_vec()).unwrap(),
|
|
Base64Account {
|
|
owner: solana_sdk::pubkey::new_rand().to_string(),
|
|
balance: 30,
|
|
executable: true,
|
|
data: String::from("Y29tYSBtb2Nh"),
|
|
},
|
|
);
|
|
|
|
let serialized = serde_yaml::to_string(&genesis_accounts2).unwrap();
|
|
let path = Path::new("test_append_primordial_accounts_to_genesis.yml");
|
|
let mut file = File::create(path).unwrap();
|
|
file.write_all(b"---\n").unwrap();
|
|
file.write_all(&serialized.into_bytes()).unwrap();
|
|
|
|
load_genesis_accounts(
|
|
"test_append_primordial_accounts_to_genesis.yml",
|
|
&mut genesis_config,
|
|
)
|
|
.expect("genesis");
|
|
|
|
remove_file(path).unwrap();
|
|
|
|
// Test total number of accounts is correct
|
|
assert_eq!(
|
|
genesis_config.accounts.len(),
|
|
genesis_accounts.len() + genesis_accounts1.len() + genesis_accounts2.len()
|
|
);
|
|
|
|
// Test old accounts are still there
|
|
for (pubkey_str, b64_account) in genesis_accounts {
|
|
let pubkey = pubkey_str.parse().unwrap();
|
|
assert_eq!(
|
|
b64_account.balance,
|
|
genesis_config.accounts[&pubkey].lamports,
|
|
);
|
|
}
|
|
|
|
// Test new account data matches
|
|
for (pubkey_str, b64_account) in genesis_accounts1 {
|
|
let pubkey = pubkey_str.parse().unwrap();
|
|
assert_eq!(
|
|
b64_account.owner,
|
|
genesis_config.accounts[&pubkey].owner.to_string(),
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.balance,
|
|
genesis_config.accounts[&pubkey].lamports,
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.executable,
|
|
genesis_config.accounts[&pubkey].executable,
|
|
);
|
|
|
|
assert_eq!(
|
|
b64_account.data,
|
|
BASE64_STANDARD.encode(&genesis_config.accounts[&pubkey].data),
|
|
);
|
|
}
|
|
|
|
// Test account data for keypairs matches
|
|
account_keypairs.iter().for_each(|keypair| {
|
|
let keypair_str = serde_json::to_string(&keypair.to_bytes().to_vec()).unwrap();
|
|
let pubkey = keypair.pubkey();
|
|
assert_eq!(
|
|
genesis_accounts2[&keypair_str].owner,
|
|
genesis_config.accounts[&pubkey].owner.to_string(),
|
|
);
|
|
|
|
assert_eq!(
|
|
genesis_accounts2[&keypair_str].balance,
|
|
genesis_config.accounts[&pubkey].lamports,
|
|
);
|
|
|
|
assert_eq!(
|
|
genesis_accounts2[&keypair_str].executable,
|
|
genesis_config.accounts[&pubkey].executable,
|
|
);
|
|
|
|
assert_eq!(
|
|
genesis_accounts2[&keypair_str].data,
|
|
BASE64_STANDARD.encode(&genesis_config.accounts[&pubkey].data),
|
|
);
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn test_genesis_account_struct_compatibility() {
|
|
let yaml_string_pubkey = "---
|
|
98frSc8R8toHoS3tQ1xWSvHCvGEADRM9hAm5qmUKjSDX:
|
|
balance: 4
|
|
owner: Gw6S9CPzR8jHku1QQMdiqcmUKjC2dhJ3gzagWduA6PGw
|
|
data:
|
|
executable: true
|
|
88frSc8R8toHoS3tQ1xWSvHCvGEADRM9hAm5qmUKjSDX:
|
|
balance: 3
|
|
owner: Gw7S9CPzR8jHku1QQMdiqcmUKjC2dhJ3gzagWduA6PGw
|
|
data: ~
|
|
executable: true
|
|
6s36rsNPDfRSvzwek7Ly3mQu9jUMwgqBhjePZMV6Acp4:
|
|
balance: 2
|
|
owner: DBC5d45LUHTCrq42ZmCdzc8A8ufwTaiYsL9pZY7KU6TR
|
|
data: aGVsbG8=
|
|
executable: false
|
|
8Y98svZv5sPHhQiPqZvqA5Z5djQ8hieodscvb61RskMJ:
|
|
balance: 1
|
|
owner: DSknYr8cPucRbx2VyssZ7Yx3iiRqNGD38VqVahkUvgV1
|
|
data: aGVsbG8gd29ybGQ=
|
|
executable: true";
|
|
|
|
let tmpfile = tempfile::NamedTempFile::new().unwrap();
|
|
let path = tmpfile.path();
|
|
let mut file = File::create(path).unwrap();
|
|
file.write_all(yaml_string_pubkey.as_bytes()).unwrap();
|
|
|
|
let mut genesis_config = GenesisConfig::default();
|
|
load_genesis_accounts(path.to_str().unwrap(), &mut genesis_config).expect("genesis");
|
|
remove_file(path).unwrap();
|
|
|
|
assert_eq!(genesis_config.accounts.len(), 4);
|
|
|
|
let yaml_string_keypair = "---
|
|
\"[17,12,234,59,35,246,168,6,64,36,169,164,219,96,253,79,238,202,164,160,195,89,9,96,179,117,255,239,32,64,124,66,233,130,19,107,172,54,86,32,119,148,4,39,199,40,122,230,249,47,150,168,163,159,83,233,97,18,25,238,103,25,253,108]\":
|
|
balance: 20
|
|
owner: 9ZfsP6Um1KU8d5gNzTsEbSJxanKYp5EPF36qUu4FJqgp
|
|
data: Y2F0IGRvZw==
|
|
executable: true
|
|
\"[36,246,244,43,37,214,110,50,134,148,148,8,205,82,233,67,223,245,122,5,149,232,213,125,244,182,26,29,56,224,70,45,42,163,71,62,222,33,229,54,73,136,53,174,128,103,247,235,222,27,219,129,180,77,225,174,220,74,201,123,97,155,159,234]\":
|
|
balance: 15
|
|
owner: F9dmtjJPi8vfLu1EJN4KkyoGdXGmVfSAhxz35Qo9RDCJ
|
|
data: bW9ua2V5IGVsZXBoYW50
|
|
executable: false
|
|
\"[103,27,132,107,42,149,72,113,24,138,225,109,209,31,158,6,26,11,8,76,24,128,131,215,156,80,251,114,103,220,111,235,56,22,87,5,209,56,53,12,224,170,10,66,82,42,11,138,51,76,120,27,166,200,237,16,200,31,23,5,57,22,131,221]\":
|
|
balance: 30
|
|
owner: AwAR5mAbNPbvQ4CvMeBxwWE8caigQoMC2chkWAbh2b9V
|
|
data: Y29tYSBtb2Nh
|
|
executable: true";
|
|
|
|
let tmpfile = tempfile::NamedTempFile::new().unwrap();
|
|
let path = tmpfile.path();
|
|
let mut file = File::create(path).unwrap();
|
|
file.write_all(yaml_string_keypair.as_bytes()).unwrap();
|
|
|
|
let mut genesis_config = GenesisConfig::default();
|
|
load_genesis_accounts(path.to_str().unwrap(), &mut genesis_config).expect("genesis");
|
|
remove_file(path).unwrap();
|
|
|
|
assert_eq!(genesis_config.accounts.len(), 3);
|
|
}
|
|
}
|