Move nonce into system program (#7645)

automerge
This commit is contained in:
Trent Nelson 2020-01-03 19:34:58 -05:00 committed by Grimes
parent 7002ccb866
commit 7e94cc2cc3
16 changed files with 1138 additions and 858 deletions

View File

@ -685,7 +685,6 @@ pub fn parse_create_address_with_seed(
"STAKE" => solana_stake_program::id(), "STAKE" => solana_stake_program::id(),
"VOTE" => solana_vote_program::id(), "VOTE" => solana_vote_program::id(),
"STORAGE" => solana_storage_program::id(), "STORAGE" => solana_storage_program::id(),
"NONCE" => solana_sdk::nonce_program::id(),
_ => pubkey_of(matches, "program_id").unwrap(), _ => pubkey_of(matches, "program_id").unwrap(),
}; };
@ -1769,7 +1768,7 @@ pub fn app<'ab, 'v>(name: &str, about: &'ab str, version: &'v str) -> App<'ab, '
.required(true) .required(true)
.help( .help(
"The program_id that the address will ultimately be used for, \n\ "The program_id that the address will ultimately be used for, \n\
or one of STAKE, VOTE, NONCE, and STORAGE keywords", or one of STAKE, VOTE, and STORAGE keywords",
), ),
) )
.arg( .arg(
@ -1985,9 +1984,9 @@ mod tests {
}; };
use solana_sdk::{ use solana_sdk::{
account::Account, account::Account,
nonce_program,
nonce_state::{Meta as NonceMeta, NonceState}, nonce_state::{Meta as NonceMeta, NonceState},
signature::{read_keypair_file, write_keypair_file}, signature::{read_keypair_file, write_keypair_file},
system_program,
transaction::TransactionError, transaction::TransactionError,
}; };
use std::{collections::HashMap, path::PathBuf}; use std::{collections::HashMap, path::PathBuf};
@ -2123,7 +2122,6 @@ mod tests {
for (name, program_id) in &[ for (name, program_id) in &[
("STAKE", solana_stake_program::id()), ("STAKE", solana_stake_program::id()),
("VOTE", solana_vote_program::id()), ("VOTE", solana_vote_program::id()),
("NONCE", solana_sdk::nonce_program::id()),
("STORAGE", solana_storage_program::id()), ("STORAGE", solana_storage_program::id()),
] { ] {
let test_create_address_with_seed = test_commands.clone().get_matches_from(vec![ let test_create_address_with_seed = test_commands.clone().get_matches_from(vec![
@ -2665,7 +2663,7 @@ mod tests {
value: json!(Account::new_data( value: json!(Account::new_data(
1, 1,
&NonceState::Initialized(NonceMeta::new(&config.keypair.pubkey()), blockhash), &NonceState::Initialized(NonceMeta::new(&config.keypair.pubkey()), blockhash),
&nonce_program::ID, &system_program::ID,
) )
.unwrap()), .unwrap()),
}); });
@ -2691,7 +2689,7 @@ mod tests {
value: json!(Account::new_data( value: json!(Account::new_data(
1, 1,
&NonceState::Initialized(NonceMeta::new(&bob_pubkey), blockhash), &NonceState::Initialized(NonceMeta::new(&bob_pubkey), blockhash),
&nonce_program::ID, &system_program::ID,
) )
.unwrap()), .unwrap()),
}); });

View File

@ -10,12 +10,14 @@ use solana_sdk::{
account::Account, account::Account,
account_utils::State, account_utils::State,
hash::Hash, hash::Hash,
nonce_instruction::{authorize, create_nonce_account, nonce, withdraw, NonceError},
nonce_program,
nonce_state::NonceState, nonce_state::NonceState,
pubkey::Pubkey, pubkey::Pubkey,
signature::{Keypair, KeypairUtil}, signature::{Keypair, KeypairUtil},
system_instruction::SystemError, system_instruction::{
create_nonce_account, nonce_advance, nonce_authorize, nonce_withdraw, NonceError,
SystemError,
},
system_program,
transaction::Transaction, transaction::Transaction,
}; };
@ -298,7 +300,7 @@ pub fn check_nonce_account(
nonce_authority: &Pubkey, nonce_authority: &Pubkey,
nonce_hash: &Hash, nonce_hash: &Hash,
) -> Result<(), Box<CliError>> { ) -> Result<(), Box<CliError>> {
if nonce_account.owner != nonce_program::ID { if nonce_account.owner != system_program::ID {
return Err(CliError::InvalidNonce(CliNonceError::InvalidAccountOwner).into()); return Err(CliError::InvalidNonce(CliNonceError::InvalidAccountOwner).into());
} }
let nonce_state: NonceState = nonce_account let nonce_state: NonceState = nonce_account
@ -330,7 +332,7 @@ pub fn process_authorize_nonce_account(
let (recent_blockhash, fee_calculator) = rpc_client.get_recent_blockhash()?; let (recent_blockhash, fee_calculator) = rpc_client.get_recent_blockhash()?;
let nonce_authority = nonce_authority.unwrap_or(&config.keypair); let nonce_authority = nonce_authority.unwrap_or(&config.keypair);
let ix = authorize(nonce_account, &nonce_authority.pubkey(), new_authority); let ix = nonce_authorize(nonce_account, &nonce_authority.pubkey(), new_authority);
let mut tx = Transaction::new_signed_with_payer( let mut tx = Transaction::new_signed_with_payer(
vec![ix], vec![ix],
Some(&config.keypair.pubkey()), Some(&config.keypair.pubkey()),
@ -405,7 +407,7 @@ pub fn process_create_nonce_account(
pub fn process_get_nonce(rpc_client: &RpcClient, nonce_account_pubkey: &Pubkey) -> ProcessResult { pub fn process_get_nonce(rpc_client: &RpcClient, nonce_account_pubkey: &Pubkey) -> ProcessResult {
let nonce_account = rpc_client.get_account(nonce_account_pubkey)?; let nonce_account = rpc_client.get_account(nonce_account_pubkey)?;
if nonce_account.owner != nonce_program::id() { if nonce_account.owner != system_program::id() {
return Err(CliError::RpcRequestError(format!( return Err(CliError::RpcRequestError(format!(
"{:?} is not a nonce account", "{:?} is not a nonce account",
nonce_account_pubkey nonce_account_pubkey
@ -442,7 +444,7 @@ pub fn process_new_nonce(
} }
let nonce_authority = nonce_authority.unwrap_or(&config.keypair); let nonce_authority = nonce_authority.unwrap_or(&config.keypair);
let ix = nonce(&nonce_account, &nonce_authority.pubkey()); let ix = nonce_advance(&nonce_account, &nonce_authority.pubkey());
let (recent_blockhash, fee_calculator) = rpc_client.get_recent_blockhash()?; let (recent_blockhash, fee_calculator) = rpc_client.get_recent_blockhash()?;
let mut tx = Transaction::new_signed_with_payer( let mut tx = Transaction::new_signed_with_payer(
vec![ix], vec![ix],
@ -467,7 +469,7 @@ pub fn process_show_nonce_account(
use_lamports_unit: bool, use_lamports_unit: bool,
) -> ProcessResult { ) -> ProcessResult {
let nonce_account = rpc_client.get_account(nonce_account_pubkey)?; let nonce_account = rpc_client.get_account(nonce_account_pubkey)?;
if nonce_account.owner != nonce_program::id() { if nonce_account.owner != system_program::id() {
return Err(CliError::RpcRequestError(format!( return Err(CliError::RpcRequestError(format!(
"{:?} is not a nonce account", "{:?} is not a nonce account",
nonce_account_pubkey nonce_account_pubkey
@ -515,7 +517,7 @@ pub fn process_withdraw_from_nonce_account(
let (recent_blockhash, fee_calculator) = rpc_client.get_recent_blockhash()?; let (recent_blockhash, fee_calculator) = rpc_client.get_recent_blockhash()?;
let nonce_authority = nonce_authority.unwrap_or(&config.keypair); let nonce_authority = nonce_authority.unwrap_or(&config.keypair);
let ix = withdraw( let ix = nonce_withdraw(
nonce_account, nonce_account,
&nonce_authority.pubkey(), &nonce_authority.pubkey(),
destination_account_pubkey, destination_account_pubkey,
@ -802,14 +804,14 @@ mod tests {
let valid = Account::new_data( let valid = Account::new_data(
1, 1,
&NonceState::Initialized(NonceMeta::new(&nonce_pubkey), blockhash), &NonceState::Initialized(NonceMeta::new(&nonce_pubkey), blockhash),
&nonce_program::ID, &system_program::ID,
); );
assert!(check_nonce_account(&valid.unwrap(), &nonce_pubkey, &blockhash).is_ok()); assert!(check_nonce_account(&valid.unwrap(), &nonce_pubkey, &blockhash).is_ok());
let invalid_owner = Account::new_data( let invalid_owner = Account::new_data(
1, 1,
&NonceState::Initialized(NonceMeta::new(&nonce_pubkey), blockhash), &NonceState::Initialized(NonceMeta::new(&nonce_pubkey), blockhash),
&system_program::ID, &Pubkey::new(&[1u8; 32]),
); );
assert_eq!( assert_eq!(
check_nonce_account(&invalid_owner.unwrap(), &nonce_pubkey, &blockhash), check_nonce_account(&invalid_owner.unwrap(), &nonce_pubkey, &blockhash),
@ -818,7 +820,7 @@ mod tests {
))), ))),
); );
let invalid_data = Account::new_data(1, &"invalid", &nonce_program::ID); let invalid_data = Account::new_data(1, &"invalid", &system_program::ID);
assert_eq!( assert_eq!(
check_nonce_account(&invalid_data.unwrap(), &nonce_pubkey, &blockhash), check_nonce_account(&invalid_data.unwrap(), &nonce_pubkey, &blockhash),
Err(Box::new(CliError::InvalidNonce( Err(Box::new(CliError::InvalidNonce(
@ -829,7 +831,7 @@ mod tests {
let invalid_hash = Account::new_data( let invalid_hash = Account::new_data(
1, 1,
&NonceState::Initialized(NonceMeta::new(&nonce_pubkey), hash(b"invalid")), &NonceState::Initialized(NonceMeta::new(&nonce_pubkey), hash(b"invalid")),
&nonce_program::ID, &system_program::ID,
); );
assert_eq!( assert_eq!(
check_nonce_account(&invalid_hash.unwrap(), &nonce_pubkey, &blockhash), check_nonce_account(&invalid_hash.unwrap(), &nonce_pubkey, &blockhash),
@ -839,7 +841,7 @@ mod tests {
let invalid_authority = Account::new_data( let invalid_authority = Account::new_data(
1, 1,
&NonceState::Initialized(NonceMeta::new(&Pubkey::new_rand()), blockhash), &NonceState::Initialized(NonceMeta::new(&Pubkey::new_rand()), blockhash),
&nonce_program::ID, &system_program::ID,
); );
assert_eq!( assert_eq!(
check_nonce_account(&invalid_authority.unwrap(), &nonce_pubkey, &blockhash), check_nonce_account(&invalid_authority.unwrap(), &nonce_pubkey, &blockhash),
@ -848,7 +850,7 @@ mod tests {
))), ))),
); );
let invalid_state = Account::new_data(1, &NonceState::Uninitialized, &nonce_program::ID); let invalid_state = Account::new_data(1, &NonceState::Uninitialized, &system_program::ID);
assert_eq!( assert_eq!(
check_nonce_account(&invalid_state.unwrap(), &nonce_pubkey, &blockhash), check_nonce_account(&invalid_state.unwrap(), &nonce_pubkey, &blockhash),
Err(Box::new(CliError::InvalidNonce( Err(Box::new(CliError::InvalidNonce(

View File

@ -1,7 +1,6 @@
use solana_sdk::{ use solana_sdk::{
clock::Epoch, genesis_config::OperatingMode, inflation::Inflation, clock::Epoch, genesis_config::OperatingMode, inflation::Inflation,
move_loader::solana_move_loader_program, nonce_program::solana_nonce_program, pubkey::Pubkey, move_loader::solana_move_loader_program, pubkey::Pubkey, system_program::solana_system_program,
system_program::solana_system_program,
}; };
#[macro_use] #[macro_use]
@ -59,7 +58,6 @@ pub fn get_programs(operating_mode: OperatingMode, epoch: Epoch) -> Option<Vec<(
solana_system_program(), solana_system_program(),
solana_bpf_loader_program!(), solana_bpf_loader_program!(),
solana_config_program!(), solana_config_program!(),
solana_nonce_program(),
solana_stake_program!(), solana_stake_program!(),
solana_storage_program!(), solana_storage_program!(),
solana_vest_program!(), solana_vest_program!(),
@ -77,7 +75,6 @@ pub fn get_programs(operating_mode: OperatingMode, epoch: Epoch) -> Option<Vec<(
if epoch == 0 { if epoch == 0 {
// Nonce, Voting, Staking and System Program only at epoch 0 // Nonce, Voting, Staking and System Program only at epoch 0
Some(vec![ Some(vec![
solana_nonce_program(),
solana_stake_program!(), solana_stake_program!(),
solana_system_program(), solana_system_program(),
solana_vote_program!(), solana_vote_program!(),
@ -127,7 +124,6 @@ pub fn get_entered_epoch_callback(operating_mode: OperatingMode) -> EnteredEpoch
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
use solana_sdk::nonce_program::solana_nonce_program;
use std::collections::HashSet; use std::collections::HashSet;
#[test] #[test]
@ -150,7 +146,7 @@ mod tests {
fn test_development_programs() { fn test_development_programs() {
assert_eq!( assert_eq!(
get_programs(OperatingMode::Development, 0).unwrap().len(), get_programs(OperatingMode::Development, 0).unwrap().len(),
11 10
); );
assert_eq!(get_programs(OperatingMode::Development, 1), None); assert_eq!(get_programs(OperatingMode::Development, 1), None);
} }
@ -173,7 +169,6 @@ mod tests {
assert_eq!( assert_eq!(
get_programs(OperatingMode::SoftLaunch, 0), get_programs(OperatingMode::SoftLaunch, 0),
Some(vec![ Some(vec![
solana_nonce_program(),
solana_stake_program!(), solana_stake_program!(),
solana_system_program(), solana_system_program(),
solana_vote_program!(), solana_vote_program!(),

View File

@ -5,6 +5,7 @@ use crate::bank::{HashAgeKind, TransactionProcessResult};
use crate::blockhash_queue::BlockhashQueue; use crate::blockhash_queue::BlockhashQueue;
use crate::message_processor::has_duplicates; use crate::message_processor::has_duplicates;
use crate::rent_collector::RentCollector; use crate::rent_collector::RentCollector;
use crate::system_instruction_processor::{get_system_account_kind, SystemAccountKind};
use log::*; use log::*;
use rayon::slice::ParallelSliceMut; use rayon::slice::ParallelSliceMut;
use solana_metrics::inc_new_counter_error; use solana_metrics::inc_new_counter_error;
@ -12,8 +13,8 @@ use solana_sdk::account::Account;
use solana_sdk::bank_hash::BankHash; use solana_sdk::bank_hash::BankHash;
use solana_sdk::clock::Slot; use solana_sdk::clock::Slot;
use solana_sdk::native_loader; use solana_sdk::native_loader;
use solana_sdk::nonce_state::NonceState;
use solana_sdk::pubkey::Pubkey; use solana_sdk::pubkey::Pubkey;
use solana_sdk::system_program;
use solana_sdk::transaction::Result; use solana_sdk::transaction::Result;
use solana_sdk::transaction::{Transaction, TransactionError}; use solana_sdk::transaction::{Transaction, TransactionError};
use std::collections::{HashMap, HashSet}; use std::collections::{HashMap, HashSet};
@ -133,15 +134,24 @@ impl Accounts {
if accounts.is_empty() || accounts[0].lamports == 0 { if accounts.is_empty() || accounts[0].lamports == 0 {
error_counters.account_not_found += 1; error_counters.account_not_found += 1;
Err(TransactionError::AccountNotFound) Err(TransactionError::AccountNotFound)
} else if accounts[0].owner != system_program::id() {
error_counters.invalid_account_for_fee += 1;
Err(TransactionError::InvalidAccountForFee)
} else if accounts[0].lamports < fee {
error_counters.insufficient_funds += 1;
Err(TransactionError::InsufficientFundsForFee)
} else { } else {
accounts[0].lamports -= fee; let min_balance = match get_system_account_kind(&accounts[0]).ok_or_else(|| {
Ok((accounts, tx_rent)) error_counters.invalid_account_for_fee += 1;
TransactionError::InvalidAccountForFee
})? {
SystemAccountKind::System => 0,
SystemAccountKind::Nonce => {
rent_collector.rent.minimum_balance(NonceState::size())
}
};
if accounts[0].lamports < fee + min_balance {
error_counters.insufficient_funds += 1;
Err(TransactionError::InsufficientFundsForFee)
} else {
accounts[0].lamports -= fee;
Ok((accounts, tx_rent))
}
} }
} }
} }
@ -627,14 +637,19 @@ mod tests {
use crate::accounts_db::tests::copy_append_vecs; use crate::accounts_db::tests::copy_append_vecs;
use crate::accounts_db::{get_temp_accounts_paths, AccountsDBSerialize}; use crate::accounts_db::{get_temp_accounts_paths, AccountsDBSerialize};
use crate::bank::HashAgeKind; use crate::bank::HashAgeKind;
use crate::rent_collector::RentCollector;
use bincode::serialize_into; use bincode::serialize_into;
use rand::{thread_rng, Rng}; use rand::{thread_rng, Rng};
use solana_sdk::account::Account; use solana_sdk::account::Account;
use solana_sdk::epoch_schedule::EpochSchedule;
use solana_sdk::fee_calculator::FeeCalculator; use solana_sdk::fee_calculator::FeeCalculator;
use solana_sdk::hash::Hash; use solana_sdk::hash::Hash;
use solana_sdk::instruction::CompiledInstruction; use solana_sdk::instruction::CompiledInstruction;
use solana_sdk::message::Message; use solana_sdk::message::Message;
use solana_sdk::nonce_state;
use solana_sdk::rent::Rent;
use solana_sdk::signature::{Keypair, KeypairUtil}; use solana_sdk::signature::{Keypair, KeypairUtil};
use solana_sdk::system_program;
use solana_sdk::sysvar; use solana_sdk::sysvar;
use solana_sdk::transaction::Transaction; use solana_sdk::transaction::Transaction;
use std::io::Cursor; use std::io::Cursor;
@ -642,10 +657,11 @@ mod tests {
use std::{thread, time}; use std::{thread, time};
use tempfile::TempDir; use tempfile::TempDir;
fn load_accounts_with_fee( fn load_accounts_with_fee_and_rent(
tx: Transaction, tx: Transaction,
ka: &Vec<(Pubkey, Account)>, ka: &Vec<(Pubkey, Account)>,
fee_calculator: &FeeCalculator, fee_calculator: &FeeCalculator,
rent_collector: &RentCollector,
error_counters: &mut ErrorCounters, error_counters: &mut ErrorCounters,
) -> Vec<(Result<TransactionLoadResult>, Option<HashAgeKind>)> { ) -> Vec<(Result<TransactionLoadResult>, Option<HashAgeKind>)> {
let mut hash_queue = BlockhashQueue::new(100); let mut hash_queue = BlockhashQueue::new(100);
@ -656,7 +672,6 @@ mod tests {
} }
let ancestors = vec![(0, 0)].into_iter().collect(); let ancestors = vec![(0, 0)].into_iter().collect();
let rent_collector = RentCollector::default();
let res = accounts.load_accounts( let res = accounts.load_accounts(
&ancestors, &ancestors,
&[tx], &[tx],
@ -664,11 +679,21 @@ mod tests {
vec![(Ok(()), Some(HashAgeKind::Extant))], vec![(Ok(()), Some(HashAgeKind::Extant))],
&hash_queue, &hash_queue,
error_counters, error_counters,
&rent_collector, rent_collector,
); );
res res
} }
fn load_accounts_with_fee(
tx: Transaction,
ka: &Vec<(Pubkey, Account)>,
fee_calculator: &FeeCalculator,
error_counters: &mut ErrorCounters,
) -> Vec<(Result<TransactionLoadResult>, Option<HashAgeKind>)> {
let rent_collector = RentCollector::default();
load_accounts_with_fee_and_rent(tx, ka, fee_calculator, &rent_collector, error_counters)
}
fn load_accounts( fn load_accounts(
tx: Transaction, tx: Transaction,
ka: &Vec<(Pubkey, Account)>, ka: &Vec<(Pubkey, Account)>,
@ -743,7 +768,7 @@ mod tests {
let key0 = keypair.pubkey(); let key0 = keypair.pubkey();
let key1 = Pubkey::new(&[5u8; 32]); let key1 = Pubkey::new(&[5u8; 32]);
let account = Account::new(1, 1, &Pubkey::default()); let account = Account::new(1, 0, &Pubkey::default());
accounts.push((key0, account)); accounts.push((key0, account));
let account = Account::new(2, 1, &Pubkey::default()); let account = Account::new(2, 1, &Pubkey::default());
@ -779,7 +804,7 @@ mod tests {
let keypair = Keypair::new(); let keypair = Keypair::new();
let key0 = keypair.pubkey(); let key0 = keypair.pubkey();
let account = Account::new(1, 1, &Pubkey::default()); let account = Account::new(1, 0, &Pubkey::default());
accounts.push((key0, account)); accounts.push((key0, account));
let instructions = vec![CompiledInstruction::new(1, &(), vec![0])]; let instructions = vec![CompiledInstruction::new(1, &(), vec![0])];
@ -841,6 +866,84 @@ mod tests {
); );
} }
#[test]
fn test_load_accounts_fee_payer_is_nonce() {
let mut error_counters = ErrorCounters::default();
let rent_collector = RentCollector::new(
0,
&EpochSchedule::default(),
500_000.0,
&Rent {
lamports_per_byte_year: 42,
..Rent::default()
},
);
let min_balance = rent_collector
.rent
.minimum_balance(nonce_state::NonceState::size());
let fee_calculator = FeeCalculator::new(min_balance, 0);
let nonce = Keypair::new();
let mut accounts = vec![(
nonce.pubkey(),
Account::new_data(
min_balance * 2,
&nonce_state::NonceState::Initialized(
nonce_state::Meta::new(&Pubkey::default()),
Hash::default(),
),
&system_program::id(),
)
.unwrap(),
)];
let instructions = vec![CompiledInstruction::new(1, &(), vec![0])];
let tx = Transaction::new_with_compiled_instructions(
&[&nonce],
&[],
Hash::default(),
vec![native_loader::id()],
instructions,
);
// Fee leaves min_balance balance succeeds
let loaded_accounts = load_accounts_with_fee_and_rent(
tx.clone(),
&accounts,
&fee_calculator,
&rent_collector,
&mut error_counters,
);
assert_eq!(loaded_accounts.len(), 1);
let (load_res, _hash_age_kind) = &loaded_accounts[0];
let (tx_accounts, _loaders, _rents) = load_res.as_ref().unwrap();
assert_eq!(tx_accounts[0].lamports, min_balance);
// Fee leaves zero balance fails
accounts[0].1.lamports = min_balance;
let loaded_accounts = load_accounts_with_fee_and_rent(
tx.clone(),
&accounts,
&fee_calculator,
&rent_collector,
&mut error_counters,
);
assert_eq!(loaded_accounts.len(), 1);
let (load_res, _hash_age_kind) = &loaded_accounts[0];
assert_eq!(*load_res, Err(TransactionError::InsufficientFundsForFee));
// Fee leaves non-zero, but sub-min_balance balance fails
accounts[0].1.lamports = 3 * min_balance / 2;
let loaded_accounts = load_accounts_with_fee_and_rent(
tx.clone(),
&accounts,
&fee_calculator,
&rent_collector,
&mut error_counters,
);
assert_eq!(loaded_accounts.len(), 1);
let (load_res, _hash_age_kind) = &loaded_accounts[0];
assert_eq!(*load_res, Err(TransactionError::InsufficientFundsForFee));
}
#[test] #[test]
fn test_load_accounts_no_loaders() { fn test_load_accounts_no_loaders() {
let mut accounts: Vec<(Pubkey, Account)> = Vec::new(); let mut accounts: Vec<(Pubkey, Account)> = Vec::new();
@ -850,7 +953,7 @@ mod tests {
let key0 = keypair.pubkey(); let key0 = keypair.pubkey();
let key1 = Pubkey::new(&[5u8; 32]); let key1 = Pubkey::new(&[5u8; 32]);
let mut account = Account::new(1, 1, &Pubkey::default()); let mut account = Account::new(1, 0, &Pubkey::default());
account.rent_epoch = 1; account.rent_epoch = 1;
accounts.push((key0, account)); accounts.push((key0, account));
@ -899,7 +1002,7 @@ mod tests {
let key5 = Pubkey::new(&[9u8; 32]); let key5 = Pubkey::new(&[9u8; 32]);
let key6 = Pubkey::new(&[10u8; 32]); let key6 = Pubkey::new(&[10u8; 32]);
let account = Account::new(1, 1, &Pubkey::default()); let account = Account::new(1, 0, &Pubkey::default());
accounts.push((key0, account)); accounts.push((key0, account));
let mut account = Account::new(40, 1, &Pubkey::default()); let mut account = Account::new(40, 1, &Pubkey::default());
@ -966,7 +1069,7 @@ mod tests {
let account = Account::new(1, 1, &Pubkey::default()); let account = Account::new(1, 1, &Pubkey::default());
accounts.push((key0, account)); accounts.push((key0, account));
let mut account = Account::new(40, 1, &Pubkey::default()); let mut account = Account::new(40, 0, &Pubkey::default());
account.executable = true; account.executable = true;
account.owner = Pubkey::default(); account.owner = Pubkey::default();
accounts.push((key1, account)); accounts.push((key1, account));
@ -1002,7 +1105,7 @@ mod tests {
let key0 = keypair.pubkey(); let key0 = keypair.pubkey();
let key1 = Pubkey::new(&[5u8; 32]); let key1 = Pubkey::new(&[5u8; 32]);
let account = Account::new(1, 1, &Pubkey::default()); let account = Account::new(1, 0, &Pubkey::default());
accounts.push((key0, account)); accounts.push((key0, account));
let mut account = Account::new(40, 1, &Pubkey::default()); let mut account = Account::new(40, 1, &Pubkey::default());
@ -1041,7 +1144,7 @@ mod tests {
let key1 = Pubkey::new(&[5u8; 32]); let key1 = Pubkey::new(&[5u8; 32]);
let key2 = Pubkey::new(&[6u8; 32]); let key2 = Pubkey::new(&[6u8; 32]);
let mut account = Account::new(1, 1, &Pubkey::default()); let mut account = Account::new(1, 0, &Pubkey::default());
account.rent_epoch = 1; account.rent_epoch = 1;
accounts.push((key0, account)); accounts.push((key0, account));

View File

@ -16,6 +16,7 @@ use crate::{
status_cache::{SlotDelta, StatusCache}, status_cache::{SlotDelta, StatusCache},
storage_utils, storage_utils,
storage_utils::StorageAccounts, storage_utils::StorageAccounts,
system_instruction_processor::{get_system_account_kind, SystemAccountKind},
transaction_batch::TransactionBatch, transaction_batch::TransactionBatch,
transaction_utils::OrderedIterator, transaction_utils::OrderedIterator,
}; };
@ -37,6 +38,7 @@ use solana_sdk::{
hash::{hashv, Hash}, hash::{hashv, Hash},
inflation::Inflation, inflation::Inflation,
native_loader, native_loader,
nonce_state::NonceState,
pubkey::Pubkey, pubkey::Pubkey,
signature::{Keypair, Signature}, signature::{Keypair, Signature},
slot_hashes::SlotHashes, slot_hashes::SlotHashes,
@ -1495,7 +1497,13 @@ impl Bank {
pub fn withdraw(&self, pubkey: &Pubkey, lamports: u64) -> Result<()> { pub fn withdraw(&self, pubkey: &Pubkey, lamports: u64) -> Result<()> {
match self.get_account(pubkey) { match self.get_account(pubkey) {
Some(mut account) => { Some(mut account) => {
if lamports > account.lamports { let min_balance = match get_system_account_kind(&account) {
Some(SystemAccountKind::Nonce) => {
self.rent_collector.rent.minimum_balance(NonceState::size())
}
_ => 0,
};
if lamports + min_balance > account.lamports {
return Err(TransactionError::InsufficientFundsForFee); return Err(TransactionError::InsufficientFundsForFee);
} }
@ -1873,7 +1881,7 @@ mod tests {
genesis_config::create_genesis_config, genesis_config::create_genesis_config,
instruction::{CompiledInstruction, Instruction, InstructionError}, instruction::{CompiledInstruction, Instruction, InstructionError},
message::{Message, MessageHeader}, message::{Message, MessageHeader},
nonce_instruction, nonce_state, nonce_state,
poh_config::PohConfig, poh_config::PohConfig,
rent::Rent, rent::Rent,
signature::{Keypair, KeypairUtil}, signature::{Keypair, KeypairUtil},
@ -2041,11 +2049,11 @@ mod tests {
assert_eq!(bank.last_blockhash(), genesis_config.hash()); assert_eq!(bank.last_blockhash(), genesis_config.hash());
// Initialize credit-debit and credit only accounts // Initialize credit-debit and credit only accounts
let account1 = Account::new(264, 1, &Pubkey::default()); let account1 = Account::new(264, 0, &Pubkey::default());
let account2 = Account::new(264, 1, &Pubkey::default()); let account2 = Account::new(264, 1, &Pubkey::default());
let account3 = Account::new(264, 1, &Pubkey::default()); let account3 = Account::new(264, 0, &Pubkey::default());
let account4 = Account::new(264, 1, &Pubkey::default()); let account4 = Account::new(264, 1, &Pubkey::default());
let account5 = Account::new(10, 1, &Pubkey::default()); let account5 = Account::new(10, 0, &Pubkey::default());
let account6 = Account::new(10, 1, &Pubkey::default()); let account6 = Account::new(10, 1, &Pubkey::default());
bank.store_account(&keypair1.pubkey(), &account1); bank.store_account(&keypair1.pubkey(), &account1);
@ -2161,7 +2169,7 @@ mod tests {
keypairs[0].pubkey(), keypairs[0].pubkey(),
Account::new( Account::new(
generic_rent_due_for_system_account + 2, generic_rent_due_for_system_account + 2,
1, 0,
&Pubkey::default(), &Pubkey::default(),
), ),
)); ));
@ -2169,7 +2177,7 @@ mod tests {
keypairs[1].pubkey(), keypairs[1].pubkey(),
Account::new( Account::new(
generic_rent_due_for_system_account + 2, generic_rent_due_for_system_account + 2,
1, 0,
&Pubkey::default(), &Pubkey::default(),
), ),
)); ));
@ -2177,7 +2185,7 @@ mod tests {
keypairs[2].pubkey(), keypairs[2].pubkey(),
Account::new( Account::new(
generic_rent_due_for_system_account + 2, generic_rent_due_for_system_account + 2,
1, 0,
&Pubkey::default(), &Pubkey::default(),
), ),
)); ));
@ -2185,23 +2193,23 @@ mod tests {
keypairs[3].pubkey(), keypairs[3].pubkey(),
Account::new( Account::new(
generic_rent_due_for_system_account + 2, generic_rent_due_for_system_account + 2,
1, 0,
&Pubkey::default(), &Pubkey::default(),
), ),
)); ));
account_pairs.push(( account_pairs.push((
keypairs[4].pubkey(), keypairs[4].pubkey(),
Account::new(10, 1, &Pubkey::default()), Account::new(10, 0, &Pubkey::default()),
)); ));
account_pairs.push(( account_pairs.push((
keypairs[5].pubkey(), keypairs[5].pubkey(),
Account::new(10, 1, &Pubkey::default()), Account::new(10, 0, &Pubkey::default()),
)); ));
account_pairs.push(( account_pairs.push((
keypairs[6].pubkey(), keypairs[6].pubkey(),
Account::new( Account::new(
(2 * generic_rent_due_for_system_account) + 24, (2 * generic_rent_due_for_system_account) + 24,
1, 0,
&Pubkey::default(), &Pubkey::default(),
), ),
)); ));
@ -2210,13 +2218,13 @@ mod tests {
keypairs[8].pubkey(), keypairs[8].pubkey(),
Account::new( Account::new(
generic_rent_due_for_system_account + 2 + 929, generic_rent_due_for_system_account + 2 + 929,
1, 0,
&Pubkey::default(), &Pubkey::default(),
), ),
)); ));
account_pairs.push(( account_pairs.push((
keypairs[9].pubkey(), keypairs[9].pubkey(),
Account::new(10, 1, &Pubkey::default()), Account::new(10, 0, &Pubkey::default()),
)); ));
// Feeding to MockProgram to test read only rent behaviour // Feeding to MockProgram to test read only rent behaviour
@ -2224,21 +2232,21 @@ mod tests {
keypairs[10].pubkey(), keypairs[10].pubkey(),
Account::new( Account::new(
generic_rent_due_for_system_account + 3, generic_rent_due_for_system_account + 3,
1, 0,
&Pubkey::default(), &Pubkey::default(),
), ),
)); ));
account_pairs.push(( account_pairs.push((
keypairs[11].pubkey(), keypairs[11].pubkey(),
Account::new(generic_rent_due_for_system_account + 3, 1, &mock_program_id), Account::new(generic_rent_due_for_system_account + 3, 0, &mock_program_id),
)); ));
account_pairs.push(( account_pairs.push((
keypairs[12].pubkey(), keypairs[12].pubkey(),
Account::new(generic_rent_due_for_system_account + 3, 1, &mock_program_id), Account::new(generic_rent_due_for_system_account + 3, 0, &mock_program_id),
)); ));
account_pairs.push(( account_pairs.push((
keypairs[13].pubkey(), keypairs[13].pubkey(),
Account::new(14, 23, &mock_program_id), Account::new(14, 22, &mock_program_id),
)); ));
for account_pair in account_pairs.iter() { for account_pair in account_pairs.iter() {
@ -2408,7 +2416,7 @@ mod tests {
bank.rent_collector.slots_per_year = 192.0; bank.rent_collector.slots_per_year = 192.0;
let payer = Keypair::new(); let payer = Keypair::new();
let payer_account = Account::new(400, 2, &system_program::id()); let payer_account = Account::new(400, 0, &system_program::id());
bank.store_account(&payer.pubkey(), &payer_account); bank.store_account(&payer.pubkey(), &payer_account);
let payee = Keypair::new(); let payee = Keypair::new();
@ -2424,9 +2432,9 @@ mod tests {
let mut total_rent_deducted = 0; let mut total_rent_deducted = 0;
// 400 - 130(Rent) - 180(Transfer) // 400 - 128(Rent) - 180(Transfer)
assert_eq!(bank.get_balance(&payer.pubkey()), 90); assert_eq!(bank.get_balance(&payer.pubkey()), 92);
total_rent_deducted += 130; total_rent_deducted += 128;
// 70 - 70(Rent) + 180(Transfer) - 21(Rent) // 70 - 70(Rent) + 180(Transfer) - 21(Rent)
assert_eq!(bank.get_balance(&payee.pubkey()), 159); assert_eq!(bank.get_balance(&payee.pubkey()), 159);
@ -2454,26 +2462,30 @@ mod tests {
// Since, validator 1 and validator 2 has equal smallest stake, it comes down to comparison // Since, validator 1 and validator 2 has equal smallest stake, it comes down to comparison
// between their pubkey. // between their pubkey.
let mut validator_1_portion = let tweak_1 = if validator_1_pubkey > validator_2_pubkey {
((validator_1_stake_lamports * rent_to_be_distributed) as f64 / 100.0) as u64; 1
if validator_1_pubkey > validator_2_pubkey { } else {
validator_1_portion += 1; 0
} };
let validator_1_portion =
((validator_1_stake_lamports * rent_to_be_distributed) as f64 / 100.0) as u64 + tweak_1;
assert_eq!( assert_eq!(
bank.get_balance(&validator_1_pubkey), bank.get_balance(&validator_1_pubkey),
validator_1_portion + 42 validator_1_portion + 42 - tweak_1,
); );
// Since, validator 1 and validator 2 has equal smallest stake, it comes down to comparison // Since, validator 1 and validator 2 has equal smallest stake, it comes down to comparison
// between their pubkey. // between their pubkey.
let mut validator_2_portion = let tweak_2 = if validator_2_pubkey > validator_1_pubkey {
((validator_2_stake_lamports * rent_to_be_distributed) as f64 / 100.0) as u64; 1
if validator_2_pubkey > validator_1_pubkey { } else {
validator_2_portion += 1; 0
} };
let validator_2_portion =
((validator_2_stake_lamports * rent_to_be_distributed) as f64 / 100.0) as u64 + tweak_2;
assert_eq!( assert_eq!(
bank.get_balance(&validator_2_pubkey), bank.get_balance(&validator_2_pubkey),
validator_2_portion + 42 validator_2_portion + 42 - tweak_2,
); );
let validator_3_portion = let validator_3_portion =
@ -2521,8 +2533,8 @@ mod tests {
}) })
.sum(); .sum();
let (generic_rent_due_for_system_account, _) = bank.rent_collector.rent.due( let (generic_rent_due_for_system_account, _) = bank.rent_collector.rent.due(
bank.get_minimum_balance_for_rent_exemption(1) - 1, bank.get_minimum_balance_for_rent_exemption(0) - 1,
1, 0,
slots_elapsed as f64 / bank.rent_collector.slots_per_year, slots_elapsed as f64 / bank.rent_collector.slots_per_year,
); );
@ -2554,7 +2566,7 @@ mod tests {
let t4 = system_transaction::transfer( let t4 = system_transaction::transfer(
&keypairs[6], &keypairs[6],
&keypairs[7].pubkey(), &keypairs[7].pubkey(),
49373, 48991,
genesis_config.hash(), genesis_config.hash(),
); );
let t5 = system_transaction::transfer( let t5 = system_transaction::transfer(
@ -2594,19 +2606,19 @@ mod tests {
let mut rent_collected = 0; let mut rent_collected = 0;
// 49374 - 49372(Rent) - 1(transfer) // 48992 - 48990(Rent) - 1(transfer)
assert_eq!(bank.get_balance(&keypairs[0].pubkey()), 1); assert_eq!(bank.get_balance(&keypairs[0].pubkey()), 1);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
// 49374 - 49372(Rent) - 1(transfer) // 48992 - 48990(Rent) + 1(transfer)
assert_eq!(bank.get_balance(&keypairs[1].pubkey()), 3); assert_eq!(bank.get_balance(&keypairs[1].pubkey()), 3);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
// 49374 - 49372(Rent) - 1(transfer) // 48992 - 48990(Rent) - 1(transfer)
assert_eq!(bank.get_balance(&keypairs[2].pubkey()), 1); assert_eq!(bank.get_balance(&keypairs[2].pubkey()), 1);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
// 49374 - 49372(Rent) - 1(transfer) // 48992 - 48990(Rent) + 1(transfer)
assert_eq!(bank.get_balance(&keypairs[3].pubkey()), 3); assert_eq!(bank.get_balance(&keypairs[3].pubkey()), 3);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
@ -2614,11 +2626,11 @@ mod tests {
assert_eq!(bank.get_balance(&keypairs[4].pubkey()), 10); assert_eq!(bank.get_balance(&keypairs[4].pubkey()), 10);
assert_eq!(bank.get_balance(&keypairs[5].pubkey()), 10); assert_eq!(bank.get_balance(&keypairs[5].pubkey()), 10);
// 98768 - 49372(Rent) - 49373(transfer) // 98004 - 48990(Rent) - 48991(transfer)
assert_eq!(bank.get_balance(&keypairs[6].pubkey()), 23); assert_eq!(bank.get_balance(&keypairs[6].pubkey()), 23);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
// 0 + 49373(transfer) - 917(Rent) // 0 + 48990(transfer) - 917(Rent)
assert_eq!( assert_eq!(
bank.get_balance(&keypairs[7].pubkey()), bank.get_balance(&keypairs[7].pubkey()),
generic_rent_due_for_system_account + 1 - 917 generic_rent_due_for_system_account + 1 - 917
@ -2630,7 +2642,7 @@ mod tests {
assert_eq!(account8.rent_epoch, bank.epoch + 1); assert_eq!(account8.rent_epoch, bank.epoch + 1);
rent_collected += 917; rent_collected += 917;
// 50303 - 49372(Rent) - 929(Transfer) // 49921 - 48900(Rent) - 929(Transfer)
assert_eq!(bank.get_balance(&keypairs[8].pubkey()), 2); assert_eq!(bank.get_balance(&keypairs[8].pubkey()), 2);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
@ -2644,15 +2656,15 @@ mod tests {
assert_eq!(account10.lamports, 12); assert_eq!(account10.lamports, 12);
rent_collected += 927; rent_collected += 927;
// 49375 - 49372(Rent) // 48993 - 48990(Rent)
assert_eq!(bank.get_balance(&keypairs[10].pubkey()), 3); assert_eq!(bank.get_balance(&keypairs[10].pubkey()), 3);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
// 49375 - 49372(Rent) + 1(Addition by program) // 48993 - 48990(Rent) + 1(Addition by program)
assert_eq!(bank.get_balance(&keypairs[11].pubkey()), 4); assert_eq!(bank.get_balance(&keypairs[11].pubkey()), 4);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
// 49375 - 49372(Rent) - 1(Deduction by program) // 48993 - 48990(Rent) - 1(Deduction by program)
assert_eq!(bank.get_balance(&keypairs[12].pubkey()), 2); assert_eq!(bank.get_balance(&keypairs[12].pubkey()), 2);
rent_collected += generic_rent_due_for_system_account; rent_collected += generic_rent_due_for_system_account;
@ -2996,6 +3008,45 @@ mod tests {
assert_eq!(bank.get_balance(&key.pubkey()), 1); assert_eq!(bank.get_balance(&key.pubkey()), 1);
} }
#[test]
fn test_bank_withdraw_from_nonce_account() {
let (mut genesis_config, _mint_keypair) = create_genesis_config(100_000);
genesis_config.rent.lamports_per_byte_year = 42;
let bank = Bank::new(&genesis_config);
let min_balance =
bank.get_minimum_balance_for_rent_exemption(nonce_state::NonceState::size());
let nonce = Keypair::new();
let nonce_account = Account::new_data(
min_balance + 42,
&nonce_state::NonceState::Initialized(
nonce_state::Meta::new(&Pubkey::default()),
Hash::default(),
),
&system_program::id(),
)
.unwrap();
bank.store_account(&nonce.pubkey(), &nonce_account);
assert_eq!(bank.get_balance(&nonce.pubkey()), min_balance + 42);
// Resulting in non-zero, but sub-min_balance balance fails
assert_eq!(
bank.withdraw(&nonce.pubkey(), min_balance / 2),
Err(TransactionError::InsufficientFundsForFee)
);
assert_eq!(bank.get_balance(&nonce.pubkey()), min_balance + 42);
// Resulting in exactly rent-exempt balance succeeds
bank.withdraw(&nonce.pubkey(), 42).unwrap();
assert_eq!(bank.get_balance(&nonce.pubkey()), min_balance);
// Account closure fails
assert_eq!(
bank.withdraw(&nonce.pubkey(), min_balance),
Err(TransactionError::InsufficientFundsForFee),
);
}
#[test] #[test]
fn test_bank_tx_fee() { fn test_bank_tx_fee() {
let arbitrary_transfer_amount = 42; let arbitrary_transfer_amount = 42;
@ -4538,7 +4589,7 @@ mod tests {
custodian_lamports, custodian_lamports,
)]; )];
let nonce_authority = nonce_authority.unwrap_or(nonce_keypair.pubkey()); let nonce_authority = nonce_authority.unwrap_or(nonce_keypair.pubkey());
setup_ixs.extend_from_slice(&nonce_instruction::create_nonce_account( setup_ixs.extend_from_slice(&system_instruction::create_nonce_account(
&custodian_keypair.pubkey(), &custodian_keypair.pubkey(),
&nonce_keypair.pubkey(), &nonce_keypair.pubkey(),
&nonce_authority, &nonce_authority,
@ -4588,7 +4639,7 @@ mod tests {
let nonce_hash = get_nonce(&bank, &nonce_pubkey).unwrap(); let nonce_hash = get_nonce(&bank, &nonce_pubkey).unwrap();
let tx = Transaction::new_signed_with_payer( let tx = Transaction::new_signed_with_payer(
vec![ vec![
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000), system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000),
], ],
Some(&custodian_pubkey), Some(&custodian_pubkey),
@ -4609,7 +4660,7 @@ mod tests {
let tx = Transaction::new_signed_with_payer( let tx = Transaction::new_signed_with_payer(
vec![ vec![
system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000), system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000),
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
], ],
Some(&custodian_pubkey), Some(&custodian_pubkey),
&[&custodian_keypair, &nonce_keypair], &[&custodian_keypair, &nonce_keypair],
@ -4628,7 +4679,7 @@ mod tests {
let nonce_hash = get_nonce(&bank, &nonce_pubkey).unwrap(); let nonce_hash = get_nonce(&bank, &nonce_pubkey).unwrap();
let mut tx = Transaction::new_signed_with_payer( let mut tx = Transaction::new_signed_with_payer(
vec![ vec![
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000), system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000),
], ],
Some(&custodian_pubkey), Some(&custodian_pubkey),
@ -4651,7 +4702,7 @@ mod tests {
let nonce_hash = get_nonce(&bank, &nonce_pubkey).unwrap(); let nonce_hash = get_nonce(&bank, &nonce_pubkey).unwrap();
let tx = Transaction::new_signed_with_payer( let tx = Transaction::new_signed_with_payer(
vec![ vec![
nonce_instruction::nonce(&missing_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&missing_pubkey, &nonce_pubkey),
system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000), system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000),
], ],
Some(&custodian_pubkey), Some(&custodian_pubkey),
@ -4670,7 +4721,7 @@ mod tests {
let tx = Transaction::new_signed_with_payer( let tx = Transaction::new_signed_with_payer(
vec![ vec![
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000), system_instruction::transfer(&custodian_pubkey, &nonce_pubkey, 100_000),
], ],
Some(&custodian_pubkey), Some(&custodian_pubkey),
@ -4680,6 +4731,39 @@ mod tests {
assert!(!bank.check_tx_durable_nonce(&tx)); assert!(!bank.check_tx_durable_nonce(&tx));
} }
#[test]
fn test_assign_from_nonce_account_fail() {
let (genesis_config, _mint_keypair) = create_genesis_config(100_000_000);
let bank = Arc::new(Bank::new(&genesis_config));
let nonce = Keypair::new();
let nonce_account = Account::new_data(
42424242,
&nonce_state::NonceState::Initialized(
nonce_state::Meta::new(&Pubkey::default()),
Hash::default(),
),
&system_program::id(),
)
.unwrap();
let blockhash = bank.last_blockhash();
bank.store_account(&nonce.pubkey(), &nonce_account);
let tx = Transaction::new_signed_instructions(
&[&nonce],
vec![system_instruction::assign(
&nonce.pubkey(),
&Pubkey::new(&[9u8; 32]),
)],
blockhash,
);
let expect = Err(TransactionError::InstructionError(
0,
InstructionError::ModifiedProgramId,
));
assert_eq!(bank.process_transaction(&tx), expect);
}
#[test] #[test]
fn test_durable_nonce_transaction() { fn test_durable_nonce_transaction() {
let (mut bank, _mint_keypair, custodian_keypair, nonce_keypair) = setup_nonce_with_bank( let (mut bank, _mint_keypair, custodian_keypair, nonce_keypair) = setup_nonce_with_bank(
@ -4725,7 +4809,7 @@ mod tests {
/* Durable Nonce transfer */ /* Durable Nonce transfer */
let durable_tx = Transaction::new_signed_with_payer( let durable_tx = Transaction::new_signed_with_payer(
vec![ vec![
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
system_instruction::transfer(&custodian_pubkey, &alice_pubkey, 100_000), system_instruction::transfer(&custodian_pubkey, &alice_pubkey, 100_000),
], ],
Some(&custodian_pubkey), Some(&custodian_pubkey),
@ -4746,7 +4830,7 @@ mod tests {
/* Durable Nonce re-use fails */ /* Durable Nonce re-use fails */
let durable_tx = Transaction::new_signed_with_payer( let durable_tx = Transaction::new_signed_with_payer(
vec![ vec![
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
system_instruction::transfer(&custodian_pubkey, &alice_pubkey, 100_000), system_instruction::transfer(&custodian_pubkey, &alice_pubkey, 100_000),
], ],
Some(&custodian_pubkey), Some(&custodian_pubkey),
@ -4770,7 +4854,7 @@ mod tests {
let durable_tx = Transaction::new_signed_with_payer( let durable_tx = Transaction::new_signed_with_payer(
vec![ vec![
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
system_instruction::transfer(&custodian_pubkey, &alice_pubkey, 100_000_000), system_instruction::transfer(&custodian_pubkey, &alice_pubkey, 100_000_000),
], ],
Some(&custodian_pubkey), Some(&custodian_pubkey),

View File

@ -2,7 +2,6 @@ use solana_sdk::{
account::Account, account::Account,
fee_calculator::FeeCalculator, fee_calculator::FeeCalculator,
genesis_config::GenesisConfig, genesis_config::GenesisConfig,
nonce_program::solana_nonce_program,
pubkey::Pubkey, pubkey::Pubkey,
rent::Rent, rent::Rent,
signature::{Keypair, KeypairUtil}, signature::{Keypair, KeypairUtil},
@ -75,7 +74,6 @@ pub fn create_genesis_config_with_leader(
// Bare minimum program set // Bare minimum program set
let native_instruction_processors = vec![ let native_instruction_processors = vec![
solana_system_program(), solana_system_program(),
solana_nonce_program(),
solana_bpf_loader_program!(), solana_bpf_loader_program!(),
solana_vote_program!(), solana_vote_program!(),
solana_stake_program!(), solana_stake_program!(),

View File

@ -7,8 +7,6 @@ use solana_sdk::instruction::{CompiledInstruction, InstructionError};
use solana_sdk::instruction_processor_utils; use solana_sdk::instruction_processor_utils;
use solana_sdk::loader_instruction::LoaderInstruction; use solana_sdk::loader_instruction::LoaderInstruction;
use solana_sdk::message::Message; use solana_sdk::message::Message;
use solana_sdk::nonce_instruction;
use solana_sdk::nonce_program;
use solana_sdk::pubkey::Pubkey; use solana_sdk::pubkey::Pubkey;
use solana_sdk::system_program; use solana_sdk::system_program;
use solana_sdk::transaction::TransactionError; use solana_sdk::transaction::TransactionError;
@ -200,13 +198,10 @@ pub struct MessageProcessor {
impl Default for MessageProcessor { impl Default for MessageProcessor {
fn default() -> Self { fn default() -> Self {
let instruction_processors: Vec<(Pubkey, ProcessInstruction)> = vec![ let instruction_processors: Vec<(Pubkey, ProcessInstruction)> = vec![(
( system_program::id(),
system_program::id(), system_instruction_processor::process_instruction,
system_instruction_processor::process_instruction, )];
),
(nonce_program::id(), nonce_instruction::process_instruction),
];
Self { Self {
instruction_processors, instruction_processors,

View File

@ -1,7 +1,7 @@
use solana_sdk::{ use solana_sdk::{
account::Account, account_utils::State, hash::Hash, instruction::CompiledInstruction, account::Account, account_utils::State, hash::Hash, instruction::CompiledInstruction,
instruction_processor_utils::limited_deserialize, nonce_instruction::NonceInstruction, instruction_processor_utils::limited_deserialize, nonce_state::NonceState, pubkey::Pubkey,
nonce_program, nonce_state::NonceState, pubkey::Pubkey, transaction::Transaction, system_instruction::SystemInstruction, system_program, transaction::Transaction,
}; };
pub fn transaction_uses_durable_nonce(tx: &Transaction) -> Option<&CompiledInstruction> { pub fn transaction_uses_durable_nonce(tx: &Transaction) -> Option<&CompiledInstruction> {
@ -12,12 +12,12 @@ pub fn transaction_uses_durable_nonce(tx: &Transaction) -> Option<&CompiledInstr
.filter(|maybe_ix| { .filter(|maybe_ix| {
let prog_id_idx = maybe_ix.program_id_index as usize; let prog_id_idx = maybe_ix.program_id_index as usize;
match message.account_keys.get(prog_id_idx) { match message.account_keys.get(prog_id_idx) {
Some(program_id) => nonce_program::check_id(&program_id), Some(program_id) => system_program::check_id(&program_id),
_ => false, _ => false,
} }
}) })
.filter(|maybe_ix| match limited_deserialize(&maybe_ix.data) { .filter(|maybe_ix| match limited_deserialize(&maybe_ix.data) {
Ok(NonceInstruction::Nonce) => true, Ok(SystemInstruction::NonceAdvance) => true,
_ => false, _ => false,
}) })
} }
@ -44,7 +44,6 @@ mod tests {
use super::*; use super::*;
use solana_sdk::{ use solana_sdk::{
hash::Hash, hash::Hash,
nonce_instruction,
nonce_state::{with_test_keyed_account, NonceAccount}, nonce_state::{with_test_keyed_account, NonceAccount},
pubkey::Pubkey, pubkey::Pubkey,
signature::{Keypair, KeypairUtil}, signature::{Keypair, KeypairUtil},
@ -61,7 +60,7 @@ mod tests {
let tx = Transaction::new_signed_instructions( let tx = Transaction::new_signed_instructions(
&[&from_keypair, &nonce_keypair], &[&from_keypair, &nonce_keypair],
vec![ vec![
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
system_instruction::transfer(&from_pubkey, &nonce_pubkey, 42), system_instruction::transfer(&from_pubkey, &nonce_pubkey, 42),
], ],
Hash::default(), Hash::default(),
@ -99,7 +98,7 @@ mod tests {
&[&from_keypair, &nonce_keypair], &[&from_keypair, &nonce_keypair],
vec![ vec![
system_instruction::transfer(&from_pubkey, &nonce_pubkey, 42), system_instruction::transfer(&from_pubkey, &nonce_pubkey, 42),
nonce_instruction::nonce(&nonce_pubkey, &nonce_pubkey), system_instruction::nonce_advance(&nonce_pubkey, &nonce_pubkey),
], ],
Hash::default(), Hash::default(),
); );
@ -115,7 +114,7 @@ mod tests {
let tx = Transaction::new_signed_instructions( let tx = Transaction::new_signed_instructions(
&[&from_keypair, &nonce_keypair], &[&from_keypair, &nonce_keypair],
vec![ vec![
nonce_instruction::withdraw(&nonce_pubkey, &nonce_pubkey, &from_pubkey, 42), system_instruction::nonce_withdraw(&nonce_pubkey, &nonce_pubkey, &from_pubkey, 42),
system_instruction::transfer(&from_pubkey, &nonce_pubkey, 42), system_instruction::transfer(&from_pubkey, &nonce_pubkey, 42),
], ],
Hash::default(), Hash::default(),
@ -162,7 +161,7 @@ mod tests {
let recent_blockhashes = create_test_recent_blockhashes(0); let recent_blockhashes = create_test_recent_blockhashes(0);
let authorized = nonce_account.unsigned_key().clone(); let authorized = nonce_account.unsigned_key().clone();
nonce_account nonce_account
.initialize(&authorized, &recent_blockhashes, &Rent::free()) .nonce_initialize(&authorized, &recent_blockhashes, &Rent::free())
.unwrap(); .unwrap();
assert!(verify_nonce(&nonce_account.account, &recent_blockhashes[0])); assert!(verify_nonce(&nonce_account.account, &recent_blockhashes[0]));
}); });
@ -186,7 +185,7 @@ mod tests {
let recent_blockhashes = create_test_recent_blockhashes(0); let recent_blockhashes = create_test_recent_blockhashes(0);
let authorized = nonce_account.unsigned_key().clone(); let authorized = nonce_account.unsigned_key().clone();
nonce_account nonce_account
.initialize(&authorized, &recent_blockhashes, &Rent::free()) .nonce_initialize(&authorized, &recent_blockhashes, &Rent::free())
.unwrap(); .unwrap();
assert!(!verify_nonce( assert!(!verify_nonce(
&nonce_account.account, &nonce_account.account,

View File

@ -1,14 +1,17 @@
use log::*; use log::*;
use solana_sdk::{ use solana_sdk::{
account::{get_signers, KeyedAccount}, account::{get_signers, Account, KeyedAccount},
account_utils::State,
instruction::InstructionError, instruction::InstructionError,
instruction_processor_utils::{limited_deserialize, next_keyed_account}, instruction_processor_utils::{limited_deserialize, next_keyed_account},
nonce_state::{NonceAccount, NonceState},
pubkey::Pubkey, pubkey::Pubkey,
system_instruction::{ system_instruction::{
create_address_with_seed, SystemError, SystemInstruction, MAX_PERMITTED_DATA_LENGTH, create_address_with_seed, SystemError, SystemInstruction, MAX_PERMITTED_DATA_LENGTH,
}, },
system_program, sysvar, system_program,
sysvar::{self, recent_blockhashes::RecentBlockhashes, rent::Rent, Sysvar},
}; };
use std::collections::HashSet; use std::collections::HashSet;
@ -69,9 +72,15 @@ fn finish_create_account(
program_id: &Pubkey, program_id: &Pubkey,
) -> Result<(), InstructionError> { ) -> Result<(), InstructionError> {
// if lamports == 0, the `from` account isn't touched // if lamports == 0, the `from` account isn't touched
if lamports != 0 && from.signer_key().is_none() { if lamports != 0 {
debug!("CreateAccount: `from` must sign transfer"); if from.signer_key().is_none() {
return Err(InstructionError::MissingRequiredSignature); debug!("CreateAccount: `from` must sign transfer");
return Err(InstructionError::MissingRequiredSignature);
}
if !from.account.data.is_empty() {
debug!("CreateAccount: `from` must not carry data");
return Err(InstructionError::InvalidArgument);
}
} }
// if it looks like the `to` account is already in use, bail // if it looks like the `to` account is already in use, bail
@ -112,6 +121,7 @@ fn finish_create_account(
debug!("Assign: program id {} invalid", program_id); debug!("Assign: program id {} invalid", program_id);
return Err(SystemError::InvalidProgramId.into()); return Err(SystemError::InvalidProgramId.into());
} }
to.account.owner = *program_id; to.account.owner = *program_id;
from.account.lamports -= lamports; from.account.lamports -= lamports;
@ -155,6 +165,11 @@ fn transfer_lamports(
return Err(InstructionError::MissingRequiredSignature); return Err(InstructionError::MissingRequiredSignature);
} }
if !from.account.data.is_empty() {
debug!("Transfer: `from` must not carry data");
return Err(InstructionError::InvalidArgument);
}
if lamports > from.account.lamports { if lamports > from.account.lamports {
debug!( debug!(
"Transfer: insufficient lamports ({}, need {})", "Transfer: insufficient lamports ({}, need {})",
@ -162,6 +177,7 @@ fn transfer_lamports(
); );
return Err(SystemError::ResultWithNegativeLamports.into()); return Err(SystemError::ResultWithNegativeLamports.into());
} }
from.account.lamports -= lamports; from.account.lamports -= lamports;
to.account.lamports += lamports; to.account.lamports += lamports;
Ok(()) Ok(())
@ -220,6 +236,56 @@ pub fn process_instruction(
let to = next_keyed_account(keyed_accounts_iter)?; let to = next_keyed_account(keyed_accounts_iter)?;
transfer_lamports(from, to, lamports) transfer_lamports(from, to, lamports)
} }
SystemInstruction::NonceAdvance => {
let me = &mut next_keyed_account(keyed_accounts_iter)?;
me.nonce_advance(
&RecentBlockhashes::from_keyed_account(next_keyed_account(keyed_accounts_iter)?)?,
&signers,
)
}
SystemInstruction::NonceWithdraw(lamports) => {
let me = &mut next_keyed_account(keyed_accounts_iter)?;
let to = &mut next_keyed_account(keyed_accounts_iter)?;
me.nonce_withdraw(
lamports,
to,
&RecentBlockhashes::from_keyed_account(next_keyed_account(keyed_accounts_iter)?)?,
&Rent::from_keyed_account(next_keyed_account(keyed_accounts_iter)?)?,
&signers,
)
}
SystemInstruction::NonceInitialize(authorized) => {
let me = &mut next_keyed_account(keyed_accounts_iter)?;
me.nonce_initialize(
&authorized,
&RecentBlockhashes::from_keyed_account(next_keyed_account(keyed_accounts_iter)?)?,
&Rent::from_keyed_account(next_keyed_account(keyed_accounts_iter)?)?,
)
}
SystemInstruction::NonceAuthorize(nonce_authority) => {
let me = &mut next_keyed_account(keyed_accounts_iter)?;
me.nonce_authorize(&nonce_authority, &signers)
}
}
}
#[derive(Copy, Clone, Debug, Eq, PartialEq)]
pub enum SystemAccountKind {
System,
Nonce,
}
pub fn get_system_account_kind(account: &Account) -> Option<SystemAccountKind> {
if system_program::check_id(&account.owner) {
if account.data.is_empty() {
Some(SystemAccountKind::System)
} else if let Ok(NonceState::Initialized(_, _)) = account.state() {
Some(SystemAccountKind::Nonce)
} else {
None
}
} else {
None
} }
} }
@ -232,9 +298,13 @@ mod tests {
use solana_sdk::account::Account; use solana_sdk::account::Account;
use solana_sdk::client::SyncClient; use solana_sdk::client::SyncClient;
use solana_sdk::genesis_config::create_genesis_config; use solana_sdk::genesis_config::create_genesis_config;
use solana_sdk::hash::{hash, Hash};
use solana_sdk::instruction::{AccountMeta, Instruction, InstructionError}; use solana_sdk::instruction::{AccountMeta, Instruction, InstructionError};
use solana_sdk::nonce_state;
use solana_sdk::signature::{Keypair, KeypairUtil}; use solana_sdk::signature::{Keypair, KeypairUtil};
use solana_sdk::system_instruction;
use solana_sdk::system_program; use solana_sdk::system_program;
use solana_sdk::sysvar;
use solana_sdk::transaction::TransactionError; use solana_sdk::transaction::TransactionError;
#[test] #[test]
@ -359,7 +429,7 @@ mod tests {
// create account with zero lamports tranferred // create account with zero lamports tranferred
let new_program_owner = Pubkey::new(&[9; 32]); let new_program_owner = Pubkey::new(&[9; 32]);
let from = Pubkey::new_rand(); let from = Pubkey::new_rand();
let mut from_account = Account::new(100, 0, &Pubkey::new_rand()); // not from system account let mut from_account = Account::new(100, 1, &Pubkey::new_rand()); // not from system account
let to = Pubkey::new_rand(); let to = Pubkey::new_rand();
let mut to_account = Account::new(0, 0, &Pubkey::default()); let mut to_account = Account::new(0, 0, &Pubkey::default());
@ -593,6 +663,29 @@ mod tests {
assert_eq!(populated_account, unchanged_account); assert_eq!(populated_account, unchanged_account);
} }
#[test]
fn test_create_from_account_is_nonce_fail() {
let nonce = Pubkey::new_rand();
let mut nonce_account = Account::new_data(
42,
&nonce_state::NonceState::Initialized(
nonce_state::Meta::new(&Pubkey::default()),
Hash::default(),
),
&system_program::id(),
)
.unwrap();
let mut from = KeyedAccount::new(&nonce, true, &mut nonce_account);
let new = Pubkey::new_rand();
let mut new_account = Account::default();
let mut to = KeyedAccount::new(&new, true, &mut new_account);
assert_eq!(
create_account(&mut from, &mut to, 42, 0, &Pubkey::new_rand()),
Err(InstructionError::InvalidArgument),
);
}
#[test] #[test]
fn test_assign_account_to_program() { fn test_assign_account_to_program() {
let new_program_owner = Pubkey::new(&[9; 32]); let new_program_owner = Pubkey::new(&[9; 32]);
@ -698,6 +791,31 @@ mod tests {
assert_eq!(to_account.lamports, 51); assert_eq!(to_account.lamports, 51);
} }
#[test]
fn test_transfer_lamports_from_nonce_account_fail() {
let from = Pubkey::new_rand();
let mut from_account = Account::new_data(
100,
&nonce_state::NonceState::Initialized(nonce_state::Meta::new(&from), Hash::default()),
&system_program::id(),
)
.unwrap();
assert_eq!(
get_system_account_kind(&from_account),
Some(SystemAccountKind::Nonce)
);
let to = Pubkey::new_rand();
let mut to_account = Account::new(1, 0, &Pubkey::new(&[3; 32])); // account owner should not matter
assert_eq!(
transfer_lamports(
&mut KeyedAccount::new(&from, true, &mut from_account),
&mut KeyedAccount::new(&to, false, &mut to_account),
50,
),
Err(InstructionError::InvalidArgument),
)
}
#[test] #[test]
fn test_system_unsigned_transaction() { fn test_system_unsigned_transaction() {
let (genesis_config, alice_keypair) = create_genesis_config(100); let (genesis_config, alice_keypair) = create_genesis_config(100);
@ -733,4 +851,469 @@ mod tests {
assert_eq!(bank_client.get_balance(&alice_pubkey).unwrap(), 50); assert_eq!(bank_client.get_balance(&alice_pubkey).unwrap(), 50);
assert_eq!(bank_client.get_balance(&mallory_pubkey).unwrap(), 50); assert_eq!(bank_client.get_balance(&mallory_pubkey).unwrap(), 50);
} }
fn process_nonce_instruction(instruction: &Instruction) -> Result<(), InstructionError> {
let mut accounts: Vec<_> = instruction
.accounts
.iter()
.map(|meta| {
if sysvar::recent_blockhashes::check_id(&meta.pubkey) {
sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
)
} else if sysvar::rent::check_id(&meta.pubkey) {
sysvar::rent::create_account(1, &Rent::free())
} else {
Account::default()
}
})
.collect();
{
let mut keyed_accounts_iter: Vec<_> = instruction
.accounts
.iter()
.zip(accounts.iter_mut())
.map(|(meta, account)| KeyedAccount::new(&meta.pubkey, meta.is_signer, account))
.collect();
super::process_instruction(
&Pubkey::default(),
&mut keyed_accounts_iter,
&instruction.data,
)
}
}
#[test]
fn test_process_nonce_ix_no_acc_data_fail() {
assert_eq!(
process_nonce_instruction(&system_instruction::nonce_advance(
&Pubkey::default(),
&Pubkey::default()
)),
Err(InstructionError::InvalidAccountData),
);
}
#[test]
fn test_process_nonce_ix_no_keyed_accs_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [],
&serialize(&SystemInstruction::NonceAdvance).unwrap()
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_nonce_ix_only_nonce_acc_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [KeyedAccount::new(
&Pubkey::default(),
true,
&mut Account::default(),
),],
&serialize(&SystemInstruction::NonceAdvance).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_nonce_ix_bad_recent_blockhash_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut Account::default(),),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut Account::default(),
),
],
&serialize(&SystemInstruction::NonceAdvance).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_nonce_ix_ok() {
let mut nonce_acc = nonce_state::create_account(1_000_000);
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut nonce_acc),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(
&sysvar::rent::id(),
false,
&mut sysvar::rent::create_account(1, &Rent::free()),
),
],
&serialize(&SystemInstruction::NonceInitialize(Pubkey::default())).unwrap(),
)
.unwrap();
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut nonce_acc,),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &hash(&serialize(&0).unwrap())); 32].into_iter(),
),
),
],
&serialize(&SystemInstruction::NonceAdvance).unwrap(),
),
Ok(()),
);
}
#[test]
fn test_process_withdraw_ix_no_acc_data_fail() {
assert_eq!(
process_nonce_instruction(&system_instruction::nonce_withdraw(
&Pubkey::default(),
&Pubkey::default(),
&Pubkey::default(),
1,
)),
Err(InstructionError::InvalidAccountData),
);
}
#[test]
fn test_process_withdraw_ix_no_keyed_accs_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [],
&serialize(&SystemInstruction::NonceWithdraw(42)).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_withdraw_ix_only_nonce_acc_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [KeyedAccount::new(
&Pubkey::default(),
true,
&mut Account::default(),
),],
&serialize(&SystemInstruction::NonceWithdraw(42)).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_withdraw_ix_bad_recent_blockhash_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut Account::default(),),
KeyedAccount::new(&Pubkey::default(), false, &mut Account::default(),),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut Account::default(),
),
],
&serialize(&SystemInstruction::NonceWithdraw(42)).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_withdraw_ix_bad_rent_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(&Pubkey::default(), true, &mut Account::default(),),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(&sysvar::rent::id(), false, &mut Account::default(),),
],
&serialize(&SystemInstruction::NonceWithdraw(42)).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_withdraw_ix_ok() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(&Pubkey::default(), true, &mut Account::default(),),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(
&sysvar::rent::id(),
false,
&mut sysvar::rent::create_account(1, &Rent::free())
),
],
&serialize(&SystemInstruction::NonceWithdraw(42)).unwrap(),
),
Ok(()),
);
}
#[test]
fn test_process_initialize_ix_no_keyed_accs_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [],
&serialize(&SystemInstruction::NonceInitialize(Pubkey::default())).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_initialize_ix_only_nonce_acc_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),],
&serialize(&SystemInstruction::NonceInitialize(Pubkey::default())).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_initialize_bad_recent_blockhash_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut Account::default(),
),
],
&serialize(&SystemInstruction::NonceInitialize(Pubkey::default())).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_initialize_ix_bad_rent_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(&sysvar::rent::id(), false, &mut Account::default(),),
],
&serialize(&SystemInstruction::NonceInitialize(Pubkey::default())).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_initialize_ix_ok() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(
&sysvar::rent::id(),
false,
&mut sysvar::rent::create_account(1, &Rent::free())
),
],
&serialize(&SystemInstruction::NonceInitialize(Pubkey::default())).unwrap(),
),
Ok(()),
);
}
#[test]
fn test_process_authorize_ix_ok() {
let mut nonce_acc = nonce_state::create_account(1_000_000);
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut nonce_acc),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(
&sysvar::rent::id(),
false,
&mut sysvar::rent::create_account(1, &Rent::free()),
),
],
&serialize(&SystemInstruction::NonceInitialize(Pubkey::default())).unwrap(),
)
.unwrap();
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [KeyedAccount::new(&Pubkey::default(), true, &mut nonce_acc,),],
&serialize(&SystemInstruction::NonceAuthorize(Pubkey::default(),)).unwrap(),
),
Ok(()),
);
}
#[test]
fn test_process_authorize_bad_account_data_fail() {
assert_eq!(
process_nonce_instruction(&system_instruction::nonce_authorize(
&Pubkey::default(),
&Pubkey::default(),
&Pubkey::default(),
)),
Err(InstructionError::InvalidAccountData),
);
}
#[test]
fn test_get_system_account_kind_system_ok() {
let system_account = Account::default();
assert_eq!(
get_system_account_kind(&system_account),
Some(SystemAccountKind::System)
);
}
#[test]
fn test_get_system_account_kind_nonce_ok() {
let nonce_account = Account::new_data(
42,
&nonce_state::NonceState::Initialized(
nonce_state::Meta::new(&Pubkey::default()),
Hash::default(),
),
&system_program::id(),
)
.unwrap();
assert_eq!(
get_system_account_kind(&nonce_account),
Some(SystemAccountKind::Nonce)
);
}
#[test]
fn test_get_system_account_kind_uninitialized_nonce_account_fail() {
assert_eq!(
get_system_account_kind(&nonce_state::create_account(42)),
None
);
}
#[test]
fn test_get_system_account_kind_system_owner_nonzero_nonnonce_data_fail() {
let other_data_account = Account::new_data(42, b"other", &Pubkey::default()).unwrap();
assert_eq!(get_system_account_kind(&other_data_account), None);
}
#[test]
fn test_get_system_account_kind_nonsystem_owner_with_nonce_state_data_fail() {
let nonce_account = Account::new_data(
42,
&nonce_state::NonceState::Initialized(
nonce_state::Meta::new(&Pubkey::default()),
Hash::default(),
),
&Pubkey::new_rand(),
)
.unwrap();
assert_eq!(get_system_account_kind(&nonce_account), None);
}
} }

View File

@ -7,7 +7,6 @@ use crate::{
fee_calculator::FeeCalculator, fee_calculator::FeeCalculator,
hash::{hash, Hash}, hash::{hash, Hash},
inflation::Inflation, inflation::Inflation,
nonce_program::solana_nonce_program,
poh_config::PohConfig, poh_config::PohConfig,
pubkey::Pubkey, pubkey::Pubkey,
rent::Rent, rent::Rent,
@ -65,7 +64,7 @@ pub fn create_genesis_config(lamports: u64) -> (GenesisConfig, Keypair) {
faucet_keypair.pubkey(), faucet_keypair.pubkey(),
Account::new(lamports, 0, &system_program::id()), Account::new(lamports, 0, &system_program::id()),
)], )],
&[solana_nonce_program(), solana_system_program()], &[solana_system_program()],
), ),
faucet_keypair, faucet_keypair,
) )

View File

@ -16,8 +16,6 @@ pub mod message;
pub mod move_loader; pub mod move_loader;
pub mod native_loader; pub mod native_loader;
pub mod native_token; pub mod native_token;
pub mod nonce_instruction;
pub mod nonce_program;
pub mod nonce_state; pub mod nonce_state;
pub mod packet; pub mod packet;
pub mod poh_config; pub mod poh_config;

View File

@ -1,665 +0,0 @@
use crate::{
account::{get_signers, KeyedAccount},
instruction::{AccountMeta, Instruction, InstructionError, WithSigner},
instruction_processor_utils::{limited_deserialize, next_keyed_account, DecodeError},
nonce_program::id,
nonce_state::{NonceAccount, NonceState},
pubkey::Pubkey,
system_instruction,
sysvar::{
recent_blockhashes::{self, RecentBlockhashes},
rent::{self, Rent},
Sysvar,
},
};
use num_derive::{FromPrimitive, ToPrimitive};
use serde_derive::{Deserialize, Serialize};
use thiserror::Error;
#[derive(Error, Debug, Clone, PartialEq, FromPrimitive, ToPrimitive)]
pub enum NonceError {
#[error("recent blockhash list is empty")]
NoRecentBlockhashes,
#[error("stored nonce is still in recent_blockhashes")]
NotExpired,
#[error("specified nonce does not match stored nonce")]
UnexpectedValue,
#[error("cannot handle request in current account state")]
BadAccountState,
}
impl<E> DecodeError<E> for NonceError {
fn type_of() -> &'static str {
"NonceError"
}
}
#[derive(Serialize, Deserialize, Debug, PartialEq, Clone)]
pub enum NonceInstruction {
/// `Nonce` consumes a stored nonce, replacing it with a successor
///
/// Expects 2 Accounts:
/// 0 - A NonceAccount
/// 1 - RecentBlockhashes sysvar
///
/// The current authority must sign a transaction executing this instrucion
Nonce,
/// `Withdraw` transfers funds out of the nonce account
///
/// Expects 4 Accounts:
/// 0 - A NonceAccount
/// 1 - A system account to which the lamports will be transferred
/// 2 - RecentBlockhashes sysvar
/// 3 - Rent sysvar
///
/// The `u64` parameter is the lamports to withdraw, which must leave the
/// account balance above the rent exempt reserve or at zero.
///
/// The current authority must sign a transaction executing this instruction
Withdraw(u64),
/// `Initialize` drives state of Uninitalized NonceAccount to Initialized,
/// setting the nonce value.
///
/// Expects 3 Accounts:
/// 0 - A NonceAccount in the Uninitialized state
/// 1 - RecentBlockHashes sysvar
/// 2 - Rent sysvar
///
/// The `Pubkey` parameter specifies the entity authorized to execute nonce
/// instruction on the account
///
/// No signatures are required to execute this instruction, enabling derived
/// nonce account addresses
Initialize(Pubkey),
/// `Authorize` changes the entity authorized to execute nonce instructions
/// on the account
///
/// Expects 1 Account:
/// 0 - A NonceAccount
///
/// The `Pubkey` parameter identifies the entity to authorize
///
/// The current authority must sign a transaction executing this instruction
Authorize(Pubkey),
}
pub fn create_nonce_account(
from_pubkey: &Pubkey,
nonce_pubkey: &Pubkey,
authority: &Pubkey,
lamports: u64,
) -> Vec<Instruction> {
vec![
system_instruction::create_account(
from_pubkey,
nonce_pubkey,
lamports,
NonceState::size() as u64,
&id(),
),
initialize(nonce_pubkey, authority),
]
}
fn initialize(nonce_pubkey: &Pubkey, authority: &Pubkey) -> Instruction {
Instruction::new(
id(),
&NonceInstruction::Initialize(*authority),
vec![
AccountMeta::new(*nonce_pubkey, false),
AccountMeta::new_readonly(recent_blockhashes::id(), false),
AccountMeta::new_readonly(rent::id(), false),
],
)
}
pub fn nonce(nonce_pubkey: &Pubkey, authorized_pubkey: &Pubkey) -> Instruction {
let account_metas = vec![
AccountMeta::new(*nonce_pubkey, false),
AccountMeta::new_readonly(recent_blockhashes::id(), false),
]
.with_signer(authorized_pubkey);
Instruction::new(id(), &NonceInstruction::Nonce, account_metas)
}
pub fn withdraw(
nonce_pubkey: &Pubkey,
authorized_pubkey: &Pubkey,
to_pubkey: &Pubkey,
lamports: u64,
) -> Instruction {
let account_metas = vec![
AccountMeta::new(*nonce_pubkey, false),
AccountMeta::new(*to_pubkey, false),
AccountMeta::new_readonly(recent_blockhashes::id(), false),
AccountMeta::new_readonly(rent::id(), false),
]
.with_signer(authorized_pubkey);
Instruction::new(id(), &NonceInstruction::Withdraw(lamports), account_metas)
}
pub fn authorize(
nonce_pubkey: &Pubkey,
authorized_pubkey: &Pubkey,
new_authority: &Pubkey,
) -> Instruction {
let account_metas = vec![AccountMeta::new(*nonce_pubkey, false)].with_signer(authorized_pubkey);
Instruction::new(
id(),
&NonceInstruction::Authorize(*new_authority),
account_metas,
)
}
pub fn process_instruction(
_program_id: &Pubkey,
keyed_accounts: &mut [KeyedAccount],
data: &[u8],
) -> Result<(), InstructionError> {
let signers = get_signers(keyed_accounts);
let keyed_accounts = &mut keyed_accounts.iter_mut();
let me = &mut next_keyed_account(keyed_accounts)?;
match limited_deserialize(data)? {
NonceInstruction::Nonce => me.nonce(
&RecentBlockhashes::from_keyed_account(next_keyed_account(keyed_accounts)?)?,
&signers,
),
NonceInstruction::Withdraw(lamports) => {
let to = &mut next_keyed_account(keyed_accounts)?;
me.withdraw(
lamports,
to,
&RecentBlockhashes::from_keyed_account(next_keyed_account(keyed_accounts)?)?,
&Rent::from_keyed_account(next_keyed_account(keyed_accounts)?)?,
&signers,
)
}
NonceInstruction::Initialize(authorized) => me.initialize(
&authorized,
&RecentBlockhashes::from_keyed_account(next_keyed_account(keyed_accounts)?)?,
&Rent::from_keyed_account(next_keyed_account(keyed_accounts)?)?,
),
NonceInstruction::Authorize(nonce_authority) => me.authorize(&nonce_authority, &signers),
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::{
account::Account,
hash::{hash, Hash},
nonce_state, system_program, sysvar,
};
use bincode::serialize;
fn process_instruction(instruction: &Instruction) -> Result<(), InstructionError> {
let mut accounts: Vec<_> = instruction
.accounts
.iter()
.map(|meta| {
if sysvar::recent_blockhashes::check_id(&meta.pubkey) {
sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
)
} else if sysvar::rent::check_id(&meta.pubkey) {
sysvar::rent::create_account(1, &Rent::free())
} else {
Account::default()
}
})
.collect();
{
let mut keyed_accounts: Vec<_> = instruction
.accounts
.iter()
.zip(accounts.iter_mut())
.map(|(meta, account)| KeyedAccount::new(&meta.pubkey, meta.is_signer, account))
.collect();
super::process_instruction(&Pubkey::default(), &mut keyed_accounts, &instruction.data)
}
}
#[test]
fn test_create_account() {
let from_pubkey = Pubkey::new_rand();
let nonce_pubkey = Pubkey::new_rand();
let authorized = nonce_pubkey;
let ixs = create_nonce_account(&from_pubkey, &nonce_pubkey, &authorized, 42);
assert_eq!(ixs.len(), 2);
let ix = &ixs[0];
assert_eq!(ix.program_id, system_program::id());
let pubkeys: Vec<_> = ix.accounts.iter().map(|am| am.pubkey).collect();
assert!(pubkeys.contains(&from_pubkey));
assert!(pubkeys.contains(&nonce_pubkey));
}
#[test]
fn test_process_nonce_ix_no_acc_data_fail() {
assert_eq!(
process_instruction(&nonce(&Pubkey::default(), &Pubkey::default())),
Err(InstructionError::InvalidAccountData),
);
}
#[test]
fn test_process_nonce_ix_no_keyed_accs_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [],
&serialize(&NonceInstruction::Nonce).unwrap()
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_nonce_ix_only_nonce_acc_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [KeyedAccount::new(
&Pubkey::default(),
true,
&mut Account::default(),
),],
&serialize(&NonceInstruction::Nonce).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_nonce_ix_bad_recent_blockhash_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut Account::default(),),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut Account::default(),
),
],
&serialize(&NonceInstruction::Nonce).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_nonce_ix_ok() {
let mut nonce_acc = nonce_state::create_account(1_000_000);
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut nonce_acc),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(
&sysvar::rent::id(),
false,
&mut sysvar::rent::create_account(1, &Rent::free()),
),
],
&serialize(&NonceInstruction::Initialize(Pubkey::default())).unwrap(),
)
.unwrap();
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut nonce_acc,),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &hash(&serialize(&0).unwrap())); 32].into_iter(),
),
),
],
&serialize(&NonceInstruction::Nonce).unwrap(),
),
Ok(()),
);
}
#[test]
fn test_process_withdraw_ix_no_acc_data_fail() {
assert_eq!(
process_instruction(&withdraw(
&Pubkey::default(),
&Pubkey::default(),
&Pubkey::default(),
1,
)),
Err(InstructionError::InvalidAccountData),
);
}
#[test]
fn test_process_withdraw_ix_no_keyed_accs_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [],
&serialize(&NonceInstruction::Withdraw(42)).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_withdraw_ix_only_nonce_acc_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [KeyedAccount::new(
&Pubkey::default(),
true,
&mut Account::default(),
),],
&serialize(&NonceInstruction::Withdraw(42)).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_withdraw_ix_bad_recent_blockhash_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut Account::default(),),
KeyedAccount::new(&Pubkey::default(), false, &mut Account::default(),),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut Account::default(),
),
],
&serialize(&NonceInstruction::Withdraw(42)).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_withdraw_ix_bad_rent_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(&Pubkey::default(), true, &mut Account::default(),),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(&sysvar::rent::id(), false, &mut Account::default(),),
],
&serialize(&NonceInstruction::Withdraw(42)).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_withdraw_ix_ok() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(&Pubkey::default(), true, &mut Account::default(),),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(
&sysvar::rent::id(),
false,
&mut sysvar::rent::create_account(1, &Rent::free())
),
],
&serialize(&NonceInstruction::Withdraw(42)).unwrap(),
),
Ok(()),
);
}
#[test]
fn test_process_initialize_ix_invalid_acc_data_fail() {
let authorized = Pubkey::default();
assert_eq!(
process_instruction(&initialize(&Pubkey::default(), &authorized)),
Err(InstructionError::InvalidAccountData),
);
}
#[test]
fn test_process_initialize_ix_no_keyed_accs_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [],
&serialize(&NonceInstruction::Initialize(Pubkey::default())).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_initialize_ix_only_nonce_acc_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),],
&serialize(&NonceInstruction::Initialize(Pubkey::default())).unwrap(),
),
Err(InstructionError::NotEnoughAccountKeys),
);
}
#[test]
fn test_process_initialize_bad_recent_blockhash_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut Account::default(),
),
],
&serialize(&NonceInstruction::Initialize(Pubkey::default())).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_initialize_ix_bad_rent_state_fail() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(&sysvar::rent::id(), false, &mut Account::default(),),
],
&serialize(&NonceInstruction::Initialize(Pubkey::default())).unwrap(),
),
Err(InstructionError::InvalidArgument),
);
}
#[test]
fn test_process_initialize_ix_ok() {
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(
&Pubkey::default(),
true,
&mut nonce_state::create_account(1_000_000),
),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(
&sysvar::rent::id(),
false,
&mut sysvar::rent::create_account(1, &Rent::free())
),
],
&serialize(&NonceInstruction::Initialize(Pubkey::default())).unwrap(),
),
Ok(()),
);
}
#[test]
fn test_process_authorize_ix_ok() {
let mut nonce_acc = nonce_state::create_account(1_000_000);
super::process_instruction(
&Pubkey::default(),
&mut [
KeyedAccount::new(&Pubkey::default(), true, &mut nonce_acc),
KeyedAccount::new(
&sysvar::recent_blockhashes::id(),
false,
&mut sysvar::recent_blockhashes::create_account_with_data(
1,
vec![(0u64, &Hash::default()); 32].into_iter(),
),
),
KeyedAccount::new(
&sysvar::rent::id(),
false,
&mut sysvar::rent::create_account(1, &Rent::free()),
),
],
&serialize(&NonceInstruction::Initialize(Pubkey::default())).unwrap(),
)
.unwrap();
assert_eq!(
super::process_instruction(
&Pubkey::default(),
&mut [KeyedAccount::new(&Pubkey::default(), true, &mut nonce_acc,),],
&serialize(&NonceInstruction::Authorize(Pubkey::default(),)).unwrap(),
),
Ok(()),
);
}
#[test]
fn test_process_authorize_bad_account_data_fail() {
assert_eq!(
process_instruction(&authorize(
&Pubkey::default(),
&Pubkey::default(),
&Pubkey::default(),
)),
Err(InstructionError::InvalidAccountData),
);
}
#[test]
fn test_custom_error_decode() {
use num_traits::FromPrimitive;
fn pretty_err<T>(err: InstructionError) -> String
where
T: 'static + std::error::Error + DecodeError<T> + FromPrimitive,
{
if let InstructionError::CustomError(code) = err {
let specific_error: T = T::decode_custom_error_to_enum(code).unwrap();
format!(
"{:?}: {}::{:?} - {}",
err,
T::type_of(),
specific_error,
specific_error,
)
} else {
"".to_string()
}
}
assert_eq!(
"CustomError(0): NonceError::NoRecentBlockhashes - recent blockhash list is empty",
pretty_err::<NonceError>(NonceError::NoRecentBlockhashes.into())
);
assert_eq!(
"CustomError(1): NonceError::NotExpired - stored nonce is still in recent_blockhashes",
pretty_err::<NonceError>(NonceError::NotExpired.into())
);
assert_eq!(
"CustomError(2): NonceError::UnexpectedValue - specified nonce does not match stored nonce",
pretty_err::<NonceError>(NonceError::UnexpectedValue.into())
);
assert_eq!(
"CustomError(3): NonceError::BadAccountState - cannot handle request in current account state",
pretty_err::<NonceError>(NonceError::BadAccountState.into())
);
}
}

View File

@ -1,5 +0,0 @@
crate::declare_id!("Nonce11111111111111111111111111111111111111");
pub fn solana_nonce_program() -> (String, crate::pubkey::Pubkey) {
("solana_nonce_program".to_string(), id())
}

View File

@ -3,9 +3,9 @@ use crate::{
account_utils::State, account_utils::State,
hash::Hash, hash::Hash,
instruction::InstructionError, instruction::InstructionError,
nonce_instruction::NonceError,
nonce_program,
pubkey::Pubkey, pubkey::Pubkey,
system_instruction::NonceError,
system_program,
sysvar::recent_blockhashes::RecentBlockhashes, sysvar::recent_blockhashes::RecentBlockhashes,
sysvar::rent::Rent, sysvar::rent::Rent,
}; };
@ -45,12 +45,12 @@ impl NonceState {
} }
pub trait NonceAccount { pub trait NonceAccount {
fn nonce( fn nonce_advance(
&mut self, &mut self,
recent_blockhashes: &RecentBlockhashes, recent_blockhashes: &RecentBlockhashes,
signers: &HashSet<Pubkey>, signers: &HashSet<Pubkey>,
) -> Result<(), InstructionError>; ) -> Result<(), InstructionError>;
fn withdraw( fn nonce_withdraw(
&mut self, &mut self,
lamports: u64, lamports: u64,
to: &mut KeyedAccount, to: &mut KeyedAccount,
@ -58,13 +58,13 @@ pub trait NonceAccount {
rent: &Rent, rent: &Rent,
signers: &HashSet<Pubkey>, signers: &HashSet<Pubkey>,
) -> Result<(), InstructionError>; ) -> Result<(), InstructionError>;
fn initialize( fn nonce_initialize(
&mut self, &mut self,
nonce_authority: &Pubkey, nonce_authority: &Pubkey,
recent_blockhashes: &RecentBlockhashes, recent_blockhashes: &RecentBlockhashes,
rent: &Rent, rent: &Rent,
) -> Result<(), InstructionError>; ) -> Result<(), InstructionError>;
fn authorize( fn nonce_authorize(
&mut self, &mut self,
nonce_authority: &Pubkey, nonce_authority: &Pubkey,
signers: &HashSet<Pubkey>, signers: &HashSet<Pubkey>,
@ -72,7 +72,7 @@ pub trait NonceAccount {
} }
impl<'a> NonceAccount for KeyedAccount<'a> { impl<'a> NonceAccount for KeyedAccount<'a> {
fn nonce( fn nonce_advance(
&mut self, &mut self,
recent_blockhashes: &RecentBlockhashes, recent_blockhashes: &RecentBlockhashes,
signers: &HashSet<Pubkey>, signers: &HashSet<Pubkey>,
@ -97,7 +97,7 @@ impl<'a> NonceAccount for KeyedAccount<'a> {
self.set_state(&NonceState::Initialized(meta, recent_blockhashes[0])) self.set_state(&NonceState::Initialized(meta, recent_blockhashes[0]))
} }
fn withdraw( fn nonce_withdraw(
&mut self, &mut self,
lamports: u64, lamports: u64,
to: &mut KeyedAccount, to: &mut KeyedAccount,
@ -137,7 +137,7 @@ impl<'a> NonceAccount for KeyedAccount<'a> {
Ok(()) Ok(())
} }
fn initialize( fn nonce_initialize(
&mut self, &mut self,
nonce_authority: &Pubkey, nonce_authority: &Pubkey,
recent_blockhashes: &RecentBlockhashes, recent_blockhashes: &RecentBlockhashes,
@ -161,7 +161,7 @@ impl<'a> NonceAccount for KeyedAccount<'a> {
self.set_state(&NonceState::Initialized(meta, recent_blockhashes[0])) self.set_state(&NonceState::Initialized(meta, recent_blockhashes[0]))
} }
fn authorize( fn nonce_authorize(
&mut self, &mut self,
nonce_authority: &Pubkey, nonce_authority: &Pubkey,
signers: &HashSet<Pubkey>, signers: &HashSet<Pubkey>,
@ -183,7 +183,7 @@ pub fn create_account(lamports: u64) -> Account {
lamports, lamports,
&NonceState::Uninitialized, &NonceState::Uninitialized,
NonceState::size(), NonceState::size(),
&nonce_program::id(), &system_program::id(),
) )
.expect("nonce_account") .expect("nonce_account")
} }
@ -205,7 +205,7 @@ mod test {
use super::*; use super::*;
use crate::{ use crate::{
account::KeyedAccount, account::KeyedAccount,
nonce_instruction::NonceError, system_instruction::NonceError,
sysvar::recent_blockhashes::{create_test_recent_blockhashes, RecentBlockhashes}, sysvar::recent_blockhashes::{create_test_recent_blockhashes, RecentBlockhashes},
}; };
use std::iter::FromIterator; use std::iter::FromIterator;
@ -238,20 +238,24 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(95); let recent_blockhashes = create_test_recent_blockhashes(95);
let authorized = keyed_account.unsigned_key().clone(); let authorized = keyed_account.unsigned_key().clone();
keyed_account keyed_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let state: NonceState = keyed_account.state().unwrap(); let state: NonceState = keyed_account.state().unwrap();
let stored = recent_blockhashes[0]; let stored = recent_blockhashes[0];
// First nonce instruction drives state from Uninitialized to Initialized // First nonce instruction drives state from Uninitialized to Initialized
assert_eq!(state, NonceState::Initialized(meta, stored)); assert_eq!(state, NonceState::Initialized(meta, stored));
let recent_blockhashes = create_test_recent_blockhashes(63); let recent_blockhashes = create_test_recent_blockhashes(63);
keyed_account.nonce(&recent_blockhashes, &signers).unwrap(); keyed_account
.nonce_advance(&recent_blockhashes, &signers)
.unwrap();
let state: NonceState = keyed_account.state().unwrap(); let state: NonceState = keyed_account.state().unwrap();
let stored = recent_blockhashes[0]; let stored = recent_blockhashes[0];
// Second nonce instruction consumes and replaces stored nonce // Second nonce instruction consumes and replaces stored nonce
assert_eq!(state, NonceState::Initialized(meta, stored)); assert_eq!(state, NonceState::Initialized(meta, stored));
let recent_blockhashes = create_test_recent_blockhashes(31); let recent_blockhashes = create_test_recent_blockhashes(31);
keyed_account.nonce(&recent_blockhashes, &signers).unwrap(); keyed_account
.nonce_advance(&recent_blockhashes, &signers)
.unwrap();
let state: NonceState = keyed_account.state().unwrap(); let state: NonceState = keyed_account.state().unwrap();
let stored = recent_blockhashes[0]; let stored = recent_blockhashes[0];
// Third nonce instruction for fun and profit // Third nonce instruction for fun and profit
@ -262,7 +266,7 @@ mod test {
let expect_nonce_lamports = keyed_account.account.lamports - withdraw_lamports; let expect_nonce_lamports = keyed_account.account.lamports - withdraw_lamports;
let expect_to_lamports = to_keyed.account.lamports + withdraw_lamports; let expect_to_lamports = to_keyed.account.lamports + withdraw_lamports;
keyed_account keyed_account
.withdraw( .nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -291,7 +295,7 @@ mod test {
let authorized = nonce_account.unsigned_key().clone(); let authorized = nonce_account.unsigned_key().clone();
let meta = Meta::new(&authorized); let meta = Meta::new(&authorized);
nonce_account nonce_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let pubkey = nonce_account.account.owner.clone(); let pubkey = nonce_account.account.owner.clone();
let mut nonce_account = KeyedAccount::new(&pubkey, false, nonce_account.account); let mut nonce_account = KeyedAccount::new(&pubkey, false, nonce_account.account);
@ -299,7 +303,7 @@ mod test {
assert_eq!(state, NonceState::Initialized(meta, stored)); assert_eq!(state, NonceState::Initialized(meta, stored));
let signers = HashSet::new(); let signers = HashSet::new();
let recent_blockhashes = create_test_recent_blockhashes(0); let recent_blockhashes = create_test_recent_blockhashes(0);
let result = nonce_account.nonce(&recent_blockhashes, &signers); let result = nonce_account.nonce_advance(&recent_blockhashes, &signers);
assert_eq!(result, Err(InstructionError::MissingRequiredSignature),); assert_eq!(result, Err(InstructionError::MissingRequiredSignature),);
}) })
} }
@ -317,10 +321,10 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(0); let recent_blockhashes = create_test_recent_blockhashes(0);
let authorized = keyed_account.unsigned_key().clone(); let authorized = keyed_account.unsigned_key().clone();
keyed_account keyed_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let recent_blockhashes = RecentBlockhashes::from_iter(vec![].into_iter()); let recent_blockhashes = RecentBlockhashes::from_iter(vec![].into_iter());
let result = keyed_account.nonce(&recent_blockhashes, &signers); let result = keyed_account.nonce_advance(&recent_blockhashes, &signers);
assert_eq!(result, Err(NonceError::NoRecentBlockhashes.into())); assert_eq!(result, Err(NonceError::NoRecentBlockhashes.into()));
}) })
} }
@ -338,9 +342,9 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(63); let recent_blockhashes = create_test_recent_blockhashes(63);
let authorized = keyed_account.unsigned_key().clone(); let authorized = keyed_account.unsigned_key().clone();
keyed_account keyed_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let result = keyed_account.nonce(&recent_blockhashes, &signers); let result = keyed_account.nonce_advance(&recent_blockhashes, &signers);
assert_eq!(result, Err(NonceError::NotExpired.into())); assert_eq!(result, Err(NonceError::NotExpired.into()));
}) })
} }
@ -356,7 +360,7 @@ mod test {
let mut signers = HashSet::new(); let mut signers = HashSet::new();
signers.insert(keyed_account.signer_key().unwrap().clone()); signers.insert(keyed_account.signer_key().unwrap().clone());
let recent_blockhashes = create_test_recent_blockhashes(63); let recent_blockhashes = create_test_recent_blockhashes(63);
let result = keyed_account.nonce(&recent_blockhashes, &signers); let result = keyed_account.nonce_advance(&recent_blockhashes, &signers);
assert_eq!(result, Err(NonceError::BadAccountState.into())); assert_eq!(result, Err(NonceError::BadAccountState.into()));
}) })
} }
@ -375,12 +379,12 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(63); let recent_blockhashes = create_test_recent_blockhashes(63);
let authorized = nonce_authority.unsigned_key().clone(); let authorized = nonce_authority.unsigned_key().clone();
nonce_account nonce_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let mut signers = HashSet::new(); let mut signers = HashSet::new();
signers.insert(nonce_authority.signer_key().unwrap().clone()); signers.insert(nonce_authority.signer_key().unwrap().clone());
let recent_blockhashes = create_test_recent_blockhashes(31); let recent_blockhashes = create_test_recent_blockhashes(31);
let result = nonce_account.nonce(&recent_blockhashes, &signers); let result = nonce_account.nonce_advance(&recent_blockhashes, &signers);
assert_eq!(result, Ok(())); assert_eq!(result, Ok(()));
}); });
}); });
@ -400,9 +404,9 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(63); let recent_blockhashes = create_test_recent_blockhashes(63);
let authorized = nonce_authority.unsigned_key().clone(); let authorized = nonce_authority.unsigned_key().clone();
nonce_account nonce_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let result = nonce_account.nonce(&recent_blockhashes, &signers); let result = nonce_account.nonce_advance(&recent_blockhashes, &signers);
assert_eq!(result, Err(InstructionError::MissingRequiredSignature),); assert_eq!(result, Err(InstructionError::MissingRequiredSignature),);
}); });
}); });
@ -426,7 +430,7 @@ mod test {
let expect_nonce_lamports = nonce_keyed.account.lamports - withdraw_lamports; let expect_nonce_lamports = nonce_keyed.account.lamports - withdraw_lamports;
let expect_to_lamports = to_keyed.account.lamports + withdraw_lamports; let expect_to_lamports = to_keyed.account.lamports + withdraw_lamports;
nonce_keyed nonce_keyed
.withdraw( .nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -459,7 +463,7 @@ mod test {
with_test_keyed_account(42, false, |mut to_keyed| { with_test_keyed_account(42, false, |mut to_keyed| {
let signers = HashSet::new(); let signers = HashSet::new();
let recent_blockhashes = create_test_recent_blockhashes(0); let recent_blockhashes = create_test_recent_blockhashes(0);
let result = nonce_keyed.withdraw( let result = nonce_keyed.nonce_withdraw(
nonce_keyed.account.lamports, nonce_keyed.account.lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -485,7 +489,7 @@ mod test {
let mut signers = HashSet::new(); let mut signers = HashSet::new();
signers.insert(nonce_keyed.signer_key().unwrap().clone()); signers.insert(nonce_keyed.signer_key().unwrap().clone());
let recent_blockhashes = create_test_recent_blockhashes(0); let recent_blockhashes = create_test_recent_blockhashes(0);
let result = nonce_keyed.withdraw( let result = nonce_keyed.nonce_withdraw(
nonce_keyed.account.lamports + 1, nonce_keyed.account.lamports + 1,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -513,7 +517,7 @@ mod test {
let nonce_expect_lamports = nonce_keyed.account.lamports - withdraw_lamports; let nonce_expect_lamports = nonce_keyed.account.lamports - withdraw_lamports;
let to_expect_lamports = to_keyed.account.lamports + withdraw_lamports; let to_expect_lamports = to_keyed.account.lamports + withdraw_lamports;
nonce_keyed nonce_keyed
.withdraw( .nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -529,7 +533,7 @@ mod test {
let nonce_expect_lamports = nonce_keyed.account.lamports - withdraw_lamports; let nonce_expect_lamports = nonce_keyed.account.lamports - withdraw_lamports;
let to_expect_lamports = to_keyed.account.lamports + withdraw_lamports; let to_expect_lamports = to_keyed.account.lamports + withdraw_lamports;
nonce_keyed nonce_keyed
.withdraw( .nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -559,7 +563,7 @@ mod test {
let authorized = nonce_keyed.unsigned_key().clone(); let authorized = nonce_keyed.unsigned_key().clone();
let meta = Meta::new(&authorized); let meta = Meta::new(&authorized);
nonce_keyed nonce_keyed
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let state: NonceState = nonce_keyed.state().unwrap(); let state: NonceState = nonce_keyed.state().unwrap();
let stored = recent_blockhashes[0]; let stored = recent_blockhashes[0];
@ -569,7 +573,7 @@ mod test {
let nonce_expect_lamports = nonce_keyed.account.lamports - withdraw_lamports; let nonce_expect_lamports = nonce_keyed.account.lamports - withdraw_lamports;
let to_expect_lamports = to_keyed.account.lamports + withdraw_lamports; let to_expect_lamports = to_keyed.account.lamports + withdraw_lamports;
nonce_keyed nonce_keyed
.withdraw( .nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -587,7 +591,7 @@ mod test {
let nonce_expect_lamports = nonce_keyed.account.lamports - withdraw_lamports; let nonce_expect_lamports = nonce_keyed.account.lamports - withdraw_lamports;
let to_expect_lamports = to_keyed.account.lamports + withdraw_lamports; let to_expect_lamports = to_keyed.account.lamports + withdraw_lamports;
nonce_keyed nonce_keyed
.withdraw( .nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -612,13 +616,13 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(0); let recent_blockhashes = create_test_recent_blockhashes(0);
let authorized = nonce_keyed.unsigned_key().clone(); let authorized = nonce_keyed.unsigned_key().clone();
nonce_keyed nonce_keyed
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
with_test_keyed_account(42, false, |mut to_keyed| { with_test_keyed_account(42, false, |mut to_keyed| {
let mut signers = HashSet::new(); let mut signers = HashSet::new();
signers.insert(nonce_keyed.signer_key().unwrap().clone()); signers.insert(nonce_keyed.signer_key().unwrap().clone());
let withdraw_lamports = nonce_keyed.account.lamports; let withdraw_lamports = nonce_keyed.account.lamports;
let result = nonce_keyed.withdraw( let result = nonce_keyed.nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -641,14 +645,14 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(95); let recent_blockhashes = create_test_recent_blockhashes(95);
let authorized = nonce_keyed.unsigned_key().clone(); let authorized = nonce_keyed.unsigned_key().clone();
nonce_keyed nonce_keyed
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
with_test_keyed_account(42, false, |mut to_keyed| { with_test_keyed_account(42, false, |mut to_keyed| {
let recent_blockhashes = create_test_recent_blockhashes(63); let recent_blockhashes = create_test_recent_blockhashes(63);
let mut signers = HashSet::new(); let mut signers = HashSet::new();
signers.insert(nonce_keyed.signer_key().unwrap().clone()); signers.insert(nonce_keyed.signer_key().unwrap().clone());
let withdraw_lamports = nonce_keyed.account.lamports + 1; let withdraw_lamports = nonce_keyed.account.lamports + 1;
let result = nonce_keyed.withdraw( let result = nonce_keyed.nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -671,14 +675,14 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(95); let recent_blockhashes = create_test_recent_blockhashes(95);
let authorized = nonce_keyed.unsigned_key().clone(); let authorized = nonce_keyed.unsigned_key().clone();
nonce_keyed nonce_keyed
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
with_test_keyed_account(42, false, |mut to_keyed| { with_test_keyed_account(42, false, |mut to_keyed| {
let recent_blockhashes = create_test_recent_blockhashes(63); let recent_blockhashes = create_test_recent_blockhashes(63);
let mut signers = HashSet::new(); let mut signers = HashSet::new();
signers.insert(nonce_keyed.signer_key().unwrap().clone()); signers.insert(nonce_keyed.signer_key().unwrap().clone());
let withdraw_lamports = nonce_keyed.account.lamports - min_lamports + 1; let withdraw_lamports = nonce_keyed.account.lamports - min_lamports + 1;
let result = nonce_keyed.withdraw( let result = nonce_keyed.nonce_withdraw(
withdraw_lamports, withdraw_lamports,
&mut to_keyed, &mut to_keyed,
&recent_blockhashes, &recent_blockhashes,
@ -706,7 +710,7 @@ mod test {
let stored = recent_blockhashes[0]; let stored = recent_blockhashes[0];
let authorized = keyed_account.unsigned_key().clone(); let authorized = keyed_account.unsigned_key().clone();
let meta = Meta::new(&authorized); let meta = Meta::new(&authorized);
let result = keyed_account.initialize(&authorized, &recent_blockhashes, &rent); let result = keyed_account.nonce_initialize(&authorized, &recent_blockhashes, &rent);
assert_eq!(result, Ok(())); assert_eq!(result, Ok(()));
let state: NonceState = keyed_account.state().unwrap(); let state: NonceState = keyed_account.state().unwrap();
assert_eq!(state, NonceState::Initialized(meta, stored)); assert_eq!(state, NonceState::Initialized(meta, stored));
@ -725,7 +729,7 @@ mod test {
signers.insert(keyed_account.signer_key().unwrap().clone()); signers.insert(keyed_account.signer_key().unwrap().clone());
let recent_blockhashes = RecentBlockhashes::from_iter(vec![].into_iter()); let recent_blockhashes = RecentBlockhashes::from_iter(vec![].into_iter());
let authorized = keyed_account.unsigned_key().clone(); let authorized = keyed_account.unsigned_key().clone();
let result = keyed_account.initialize(&authorized, &recent_blockhashes, &rent); let result = keyed_account.nonce_initialize(&authorized, &recent_blockhashes, &rent);
assert_eq!(result, Err(NonceError::NoRecentBlockhashes.into())); assert_eq!(result, Err(NonceError::NoRecentBlockhashes.into()));
}) })
} }
@ -741,10 +745,10 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(31); let recent_blockhashes = create_test_recent_blockhashes(31);
let authorized = keyed_account.unsigned_key().clone(); let authorized = keyed_account.unsigned_key().clone();
keyed_account keyed_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let recent_blockhashes = create_test_recent_blockhashes(0); let recent_blockhashes = create_test_recent_blockhashes(0);
let result = keyed_account.initialize(&authorized, &recent_blockhashes, &rent); let result = keyed_account.nonce_initialize(&authorized, &recent_blockhashes, &rent);
assert_eq!(result, Err(NonceError::BadAccountState.into())); assert_eq!(result, Err(NonceError::BadAccountState.into()));
}) })
} }
@ -759,7 +763,7 @@ mod test {
with_test_keyed_account(min_lamports - 42, true, |keyed_account| { with_test_keyed_account(min_lamports - 42, true, |keyed_account| {
let recent_blockhashes = create_test_recent_blockhashes(63); let recent_blockhashes = create_test_recent_blockhashes(63);
let authorized = keyed_account.unsigned_key().clone(); let authorized = keyed_account.unsigned_key().clone();
let result = keyed_account.initialize(&authorized, &recent_blockhashes, &rent); let result = keyed_account.nonce_initialize(&authorized, &recent_blockhashes, &rent);
assert_eq!(result, Err(InstructionError::InsufficientFunds)); assert_eq!(result, Err(InstructionError::InsufficientFunds));
}) })
} }
@ -778,11 +782,11 @@ mod test {
let stored = recent_blockhashes[0]; let stored = recent_blockhashes[0];
let authorized = nonce_account.unsigned_key().clone(); let authorized = nonce_account.unsigned_key().clone();
nonce_account nonce_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let authorized = &Pubkey::default().clone(); let authorized = &Pubkey::default().clone();
let meta = Meta::new(&authorized); let meta = Meta::new(&authorized);
let result = nonce_account.authorize(&Pubkey::default(), &signers); let result = nonce_account.nonce_authorize(&Pubkey::default(), &signers);
assert_eq!(result, Ok(())); assert_eq!(result, Ok(()));
let state: NonceState = nonce_account.state().unwrap(); let state: NonceState = nonce_account.state().unwrap();
assert_eq!(state, NonceState::Initialized(meta, stored)); assert_eq!(state, NonceState::Initialized(meta, stored));
@ -799,7 +803,7 @@ mod test {
with_test_keyed_account(min_lamports + 42, true, |nonce_account| { with_test_keyed_account(min_lamports + 42, true, |nonce_account| {
let mut signers = HashSet::new(); let mut signers = HashSet::new();
signers.insert(nonce_account.signer_key().unwrap().clone()); signers.insert(nonce_account.signer_key().unwrap().clone());
let result = nonce_account.authorize(&Pubkey::default(), &signers); let result = nonce_account.nonce_authorize(&Pubkey::default(), &signers);
assert_eq!(result, Err(NonceError::BadAccountState.into())); assert_eq!(result, Err(NonceError::BadAccountState.into()));
}) })
} }
@ -817,9 +821,9 @@ mod test {
let recent_blockhashes = create_test_recent_blockhashes(31); let recent_blockhashes = create_test_recent_blockhashes(31);
let authorized = &Pubkey::default().clone(); let authorized = &Pubkey::default().clone();
nonce_account nonce_account
.initialize(&authorized, &recent_blockhashes, &rent) .nonce_initialize(&authorized, &recent_blockhashes, &rent)
.unwrap(); .unwrap();
let result = nonce_account.authorize(&Pubkey::default(), &signers); let result = nonce_account.nonce_authorize(&Pubkey::default(), &signers);
assert_eq!(result, Err(InstructionError::MissingRequiredSignature)); assert_eq!(result, Err(InstructionError::MissingRequiredSignature));
}) })
} }

View File

@ -1,9 +1,12 @@
use crate::hash::hashv; use crate::hash::hashv;
use crate::instruction::{AccountMeta, Instruction}; use crate::instruction::{AccountMeta, Instruction, WithSigner};
use crate::instruction_processor_utils::DecodeError; use crate::instruction_processor_utils::DecodeError;
use crate::nonce_state::NonceState;
use crate::pubkey::Pubkey; use crate::pubkey::Pubkey;
use crate::system_program; use crate::system_program;
use crate::sysvar::{recent_blockhashes, rent};
use num_derive::{FromPrimitive, ToPrimitive}; use num_derive::{FromPrimitive, ToPrimitive};
use thiserror::Error;
#[derive(Serialize, Debug, Clone, PartialEq, FromPrimitive, ToPrimitive)] #[derive(Serialize, Debug, Clone, PartialEq, FromPrimitive, ToPrimitive)]
pub enum SystemError { pub enum SystemError {
@ -30,6 +33,24 @@ impl std::fmt::Display for SystemError {
} }
impl std::error::Error for SystemError {} impl std::error::Error for SystemError {}
#[derive(Error, Debug, Clone, PartialEq, FromPrimitive, ToPrimitive)]
pub enum NonceError {
#[error("recent blockhash list is empty")]
NoRecentBlockhashes,
#[error("stored nonce is still in recent_blockhashes")]
NotExpired,
#[error("specified nonce does not match stored nonce")]
UnexpectedValue,
#[error("cannot handle request in current account state")]
BadAccountState,
}
impl<E> DecodeError<E> for NonceError {
fn type_of() -> &'static str {
"NonceError"
}
}
/// maximum length of derived address seed /// maximum length of derived address seed
pub const MAX_ADDRESS_SEED_LEN: usize = 32; pub const MAX_ADDRESS_SEED_LEN: usize = 32;
@ -71,6 +92,51 @@ pub enum SystemInstruction {
space: u64, space: u64,
program_id: Pubkey, program_id: Pubkey,
}, },
/// `NonceAdvance` consumes a stored nonce, replacing it with a successor
///
/// Expects 2 Accounts:
/// 0 - A NonceAccount
/// 1 - RecentBlockhashes sysvar
///
/// The current authority must sign a transaction executing this instrucion
NonceAdvance,
/// `NonceWithdraw` transfers funds out of the nonce account
///
/// Expects 4 Accounts:
/// 0 - A NonceAccount
/// 1 - A system account to which the lamports will be transferred
/// 2 - RecentBlockhashes sysvar
/// 3 - Rent sysvar
///
/// The `u64` parameter is the lamports to withdraw, which must leave the
/// account balance above the rent exempt reserve or at zero.
///
/// The current authority must sign a transaction executing this instruction
NonceWithdraw(u64),
/// `NonceInitialize` drives state of Uninitalized NonceAccount to Initialized,
/// setting the nonce value.
///
/// Expects 3 Accounts:
/// 0 - A NonceAccount in the Uninitialized state
/// 1 - RecentBlockHashes sysvar
/// 2 - Rent sysvar
///
/// The `Pubkey` parameter specifies the entity authorized to execute nonce
/// instruction on the account
///
/// No signatures are required to execute this instruction, enabling derived
/// nonce account addresses
NonceInitialize(Pubkey),
/// `NonceAuthorize` changes the entity authorized to execute nonce instructions
/// on the account
///
/// Expects 1 Account:
/// 0 - A NonceAccount
///
/// The `Pubkey` parameter identifies the entity to authorize
///
/// The current authority must sign a transaction executing this instruction
NonceAuthorize(Pubkey),
} }
pub fn create_account( pub fn create_account(
@ -167,9 +233,82 @@ pub fn create_address_with_seed(
)) ))
} }
pub fn create_nonce_account(
from_pubkey: &Pubkey,
nonce_pubkey: &Pubkey,
authority: &Pubkey,
lamports: u64,
) -> Vec<Instruction> {
vec![
create_account(
from_pubkey,
nonce_pubkey,
lamports,
NonceState::size() as u64,
&system_program::id(),
),
Instruction::new(
system_program::id(),
&SystemInstruction::NonceInitialize(*authority),
vec![
AccountMeta::new(*nonce_pubkey, false),
AccountMeta::new_readonly(recent_blockhashes::id(), false),
AccountMeta::new_readonly(rent::id(), false),
],
),
]
}
pub fn nonce_advance(nonce_pubkey: &Pubkey, authorized_pubkey: &Pubkey) -> Instruction {
let account_metas = vec![
AccountMeta::new(*nonce_pubkey, false),
AccountMeta::new_readonly(recent_blockhashes::id(), false),
]
.with_signer(authorized_pubkey);
Instruction::new(
system_program::id(),
&SystemInstruction::NonceAdvance,
account_metas,
)
}
pub fn nonce_withdraw(
nonce_pubkey: &Pubkey,
authorized_pubkey: &Pubkey,
to_pubkey: &Pubkey,
lamports: u64,
) -> Instruction {
let account_metas = vec![
AccountMeta::new(*nonce_pubkey, false),
AccountMeta::new(*to_pubkey, false),
AccountMeta::new_readonly(recent_blockhashes::id(), false),
AccountMeta::new_readonly(rent::id(), false),
]
.with_signer(authorized_pubkey);
Instruction::new(
system_program::id(),
&SystemInstruction::NonceWithdraw(lamports),
account_metas,
)
}
pub fn nonce_authorize(
nonce_pubkey: &Pubkey,
authorized_pubkey: &Pubkey,
new_authority: &Pubkey,
) -> Instruction {
let account_metas = vec![AccountMeta::new(*nonce_pubkey, false)].with_signer(authorized_pubkey);
Instruction::new(
system_program::id(),
&SystemInstruction::NonceAuthorize(*new_authority),
account_metas,
)
}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
use crate::instruction::{Instruction, InstructionError};
fn get_keys(instruction: &Instruction) -> Vec<Pubkey> { fn get_keys(instruction: &Instruction) -> Vec<Pubkey> {
instruction.accounts.iter().map(|x| x.pubkey).collect() instruction.accounts.iter().map(|x| x.pubkey).collect()
@ -239,4 +378,56 @@ mod tests {
assert_eq!(get_keys(&instructions[0]), vec![alice_pubkey, bob_pubkey]); assert_eq!(get_keys(&instructions[0]), vec![alice_pubkey, bob_pubkey]);
assert_eq!(get_keys(&instructions[1]), vec![alice_pubkey, carol_pubkey]); assert_eq!(get_keys(&instructions[1]), vec![alice_pubkey, carol_pubkey]);
} }
#[test]
fn test_create_nonce_account() {
let from_pubkey = Pubkey::new_rand();
let nonce_pubkey = Pubkey::new_rand();
let authorized = nonce_pubkey;
let ixs = create_nonce_account(&from_pubkey, &nonce_pubkey, &authorized, 42);
assert_eq!(ixs.len(), 2);
let ix = &ixs[0];
assert_eq!(ix.program_id, system_program::id());
let pubkeys: Vec<_> = ix.accounts.iter().map(|am| am.pubkey).collect();
assert!(pubkeys.contains(&from_pubkey));
assert!(pubkeys.contains(&nonce_pubkey));
}
#[test]
fn test_nonce_error_decode() {
use num_traits::FromPrimitive;
fn pretty_err<T>(err: InstructionError) -> String
where
T: 'static + std::error::Error + DecodeError<T> + FromPrimitive,
{
if let InstructionError::CustomError(code) = err {
let specific_error: T = T::decode_custom_error_to_enum(code).unwrap();
format!(
"{:?}: {}::{:?} - {}",
err,
T::type_of(),
specific_error,
specific_error,
)
} else {
"".to_string()
}
}
assert_eq!(
"CustomError(0): NonceError::NoRecentBlockhashes - recent blockhash list is empty",
pretty_err::<NonceError>(NonceError::NoRecentBlockhashes.into())
);
assert_eq!(
"CustomError(1): NonceError::NotExpired - stored nonce is still in recent_blockhashes",
pretty_err::<NonceError>(NonceError::NotExpired.into())
);
assert_eq!(
"CustomError(2): NonceError::UnexpectedValue - specified nonce does not match stored nonce",
pretty_err::<NonceError>(NonceError::UnexpectedValue.into())
);
assert_eq!(
"CustomError(3): NonceError::BadAccountState - cannot handle request in current account state",
pretty_err::<NonceError>(NonceError::BadAccountState.into())
);
}
} }

View File

@ -3,10 +3,10 @@
use crate::hash::Hash; use crate::hash::Hash;
use crate::instruction::{CompiledInstruction, Instruction, InstructionError}; use crate::instruction::{CompiledInstruction, Instruction, InstructionError};
use crate::message::Message; use crate::message::Message;
use crate::nonce_instruction;
use crate::pubkey::Pubkey; use crate::pubkey::Pubkey;
use crate::short_vec; use crate::short_vec;
use crate::signature::{KeypairUtil, Signature}; use crate::signature::{KeypairUtil, Signature};
use crate::system_instruction;
use bincode::serialize; use bincode::serialize;
use std::result; use std::result;
@ -103,7 +103,8 @@ impl Transaction {
nonce_authority_pubkey: &Pubkey, nonce_authority_pubkey: &Pubkey,
nonce_hash: Hash, nonce_hash: Hash,
) -> Self { ) -> Self {
let nonce_ix = nonce_instruction::nonce(&nonce_account_pubkey, &nonce_authority_pubkey); let nonce_ix =
system_instruction::nonce_advance(&nonce_account_pubkey, &nonce_authority_pubkey);
instructions.insert(0, nonce_ix); instructions.insert(0, nonce_ix);
Self::new_signed_with_payer(instructions, payer, signing_keypairs, nonce_hash) Self::new_signed_with_payer(instructions, payer, signing_keypairs, nonce_hash)
} }