remove payer from vote instructions (#4475)
This commit is contained in:
parent
9670788bf5
commit
028e111fbc
|
@ -101,14 +101,10 @@ mod tests {
|
||||||
let votes = (0..MAX_RECENT_VOTES)
|
let votes = (0..MAX_RECENT_VOTES)
|
||||||
.map(|i| Vote::new(i as u64, Hash::default()))
|
.map(|i| Vote::new(i as u64, Hash::default()))
|
||||||
.collect::<Vec<_>>();
|
.collect::<Vec<_>>();
|
||||||
let vote_ix = vote_instruction::vote(
|
let vote_ix = vote_instruction::vote(&vote_keypair.pubkey(), &vote_keypair.pubkey(), votes);
|
||||||
&node_keypair.pubkey(),
|
|
||||||
&vote_keypair.pubkey(),
|
let mut vote_tx = Transaction::new_with_payer(vec![vote_ix], Some(&node_keypair.pubkey()));
|
||||||
&vote_keypair.pubkey(),
|
|
||||||
votes,
|
|
||||||
);
|
|
||||||
|
|
||||||
let mut vote_tx = Transaction::new_unsigned_instructions(vec![vote_ix]);
|
|
||||||
vote_tx.partial_sign(&[&node_keypair], Hash::default());
|
vote_tx.partial_sign(&[&node_keypair], Hash::default());
|
||||||
vote_tx.partial_sign(&[&vote_keypair], Hash::default());
|
vote_tx.partial_sign(&[&vote_keypair], Hash::default());
|
||||||
|
|
||||||
|
|
|
@ -333,13 +333,14 @@ impl ReplayStage {
|
||||||
|
|
||||||
// Send our last few votes along with the new one
|
// Send our last few votes along with the new one
|
||||||
let vote_ix = vote_instruction::vote(
|
let vote_ix = vote_instruction::vote(
|
||||||
&node_keypair.pubkey(),
|
|
||||||
&vote_account,
|
&vote_account,
|
||||||
&voting_keypair.pubkey(),
|
&voting_keypair.pubkey(),
|
||||||
locktower.recent_votes(),
|
locktower.recent_votes(),
|
||||||
);
|
);
|
||||||
|
|
||||||
let mut vote_tx = Transaction::new_unsigned_instructions(vec![vote_ix]);
|
let mut vote_tx =
|
||||||
|
Transaction::new_with_payer(vec![vote_ix], Some(&node_keypair.pubkey()));;
|
||||||
|
|
||||||
let blockhash = bank.last_blockhash();
|
let blockhash = bank.last_blockhash();
|
||||||
vote_tx.partial_sign(&[node_keypair.as_ref()], blockhash);
|
vote_tx.partial_sign(&[node_keypair.as_ref()], blockhash);
|
||||||
vote_tx.partial_sign(&[voting_keypair.as_ref()], blockhash);
|
vote_tx.partial_sign(&[voting_keypair.as_ref()], blockhash);
|
||||||
|
|
|
@ -26,16 +26,8 @@ pub enum VoteInstruction {
|
||||||
Vote(Vec<Vote>),
|
Vote(Vec<Vote>),
|
||||||
}
|
}
|
||||||
|
|
||||||
fn initialize_account(
|
fn initialize_account(vote_pubkey: &Pubkey, node_pubkey: &Pubkey, commission: u32) -> Instruction {
|
||||||
from_pubkey: &Pubkey,
|
let account_metas = vec![AccountMeta::new(*vote_pubkey, false)];
|
||||||
vote_pubkey: &Pubkey,
|
|
||||||
node_pubkey: &Pubkey,
|
|
||||||
commission: u32,
|
|
||||||
) -> Instruction {
|
|
||||||
let account_metas = vec![
|
|
||||||
AccountMeta::new(*from_pubkey, true),
|
|
||||||
AccountMeta::new(*vote_pubkey, false),
|
|
||||||
];
|
|
||||||
Instruction::new(
|
Instruction::new(
|
||||||
id(),
|
id(),
|
||||||
&VoteInstruction::InitializeAccount(*node_pubkey, commission),
|
&VoteInstruction::InitializeAccount(*node_pubkey, commission),
|
||||||
|
@ -53,20 +45,18 @@ pub fn create_account(
|
||||||
let space = VoteState::size_of() as u64;
|
let space = VoteState::size_of() as u64;
|
||||||
let create_ix =
|
let create_ix =
|
||||||
system_instruction::create_account(from_pubkey, vote_pubkey, lamports, space, &id());
|
system_instruction::create_account(from_pubkey, vote_pubkey, lamports, space, &id());
|
||||||
let init_ix = initialize_account(from_pubkey, vote_pubkey, node_pubkey, commission);
|
let init_ix = initialize_account(vote_pubkey, node_pubkey, commission);
|
||||||
vec![create_ix, init_ix]
|
vec![create_ix, init_ix]
|
||||||
}
|
}
|
||||||
|
|
||||||
fn metas_for_authorized_signer(
|
fn metas_for_authorized_signer(
|
||||||
from_pubkey: &Pubkey,
|
|
||||||
vote_pubkey: &Pubkey,
|
vote_pubkey: &Pubkey,
|
||||||
authorized_voter_pubkey: &Pubkey, // currently authorized
|
authorized_voter_pubkey: &Pubkey, // currently authorized
|
||||||
) -> Vec<AccountMeta> {
|
) -> Vec<AccountMeta> {
|
||||||
let mut account_metas = vec![AccountMeta::new(*from_pubkey, true)]; // sender
|
|
||||||
|
|
||||||
let is_own_signer = authorized_voter_pubkey == vote_pubkey;
|
let is_own_signer = authorized_voter_pubkey == vote_pubkey;
|
||||||
|
|
||||||
account_metas.push(AccountMeta::new(*vote_pubkey, is_own_signer)); // vote account
|
// vote account
|
||||||
|
let mut account_metas = vec![AccountMeta::new(*vote_pubkey, is_own_signer)];
|
||||||
|
|
||||||
if !is_own_signer {
|
if !is_own_signer {
|
||||||
account_metas.push(AccountMeta::new(*authorized_voter_pubkey, true)) // signer
|
account_metas.push(AccountMeta::new(*authorized_voter_pubkey, true)) // signer
|
||||||
|
@ -75,13 +65,11 @@ fn metas_for_authorized_signer(
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn authorize_voter(
|
pub fn authorize_voter(
|
||||||
from_pubkey: &Pubkey,
|
|
||||||
vote_pubkey: &Pubkey,
|
vote_pubkey: &Pubkey,
|
||||||
authorized_voter_pubkey: &Pubkey, // currently authorized
|
authorized_voter_pubkey: &Pubkey, // currently authorized
|
||||||
new_authorized_voter_pubkey: &Pubkey,
|
new_authorized_voter_pubkey: &Pubkey,
|
||||||
) -> Instruction {
|
) -> Instruction {
|
||||||
let account_metas =
|
let account_metas = metas_for_authorized_signer(vote_pubkey, authorized_voter_pubkey);
|
||||||
metas_for_authorized_signer(from_pubkey, vote_pubkey, authorized_voter_pubkey);
|
|
||||||
|
|
||||||
Instruction::new(
|
Instruction::new(
|
||||||
id(),
|
id(),
|
||||||
|
@ -91,16 +79,14 @@ pub fn authorize_voter(
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn vote(
|
pub fn vote(
|
||||||
from_pubkey: &Pubkey,
|
|
||||||
vote_pubkey: &Pubkey,
|
vote_pubkey: &Pubkey,
|
||||||
authorized_voter_pubkey: &Pubkey,
|
authorized_voter_pubkey: &Pubkey,
|
||||||
recent_votes: Vec<Vote>,
|
recent_votes: Vec<Vote>,
|
||||||
) -> Instruction {
|
) -> Instruction {
|
||||||
let mut account_metas =
|
let mut account_metas = metas_for_authorized_signer(vote_pubkey, authorized_voter_pubkey);
|
||||||
metas_for_authorized_signer(from_pubkey, vote_pubkey, authorized_voter_pubkey);
|
|
||||||
|
|
||||||
// request slot_hashes syscall account after vote_pubkey
|
// request slot_hashes syscall account after vote_pubkey
|
||||||
account_metas.insert(2, AccountMeta::new(slot_hashes::id(), false));
|
account_metas.insert(1, AccountMeta::new(slot_hashes::id(), false));
|
||||||
|
|
||||||
Instruction::new(id(), &VoteInstruction::Vote(recent_votes), account_metas)
|
Instruction::new(id(), &VoteInstruction::Vote(recent_votes), account_metas)
|
||||||
}
|
}
|
||||||
|
@ -116,13 +102,13 @@ pub fn process_instruction(
|
||||||
trace!("process_instruction: {:?}", data);
|
trace!("process_instruction: {:?}", data);
|
||||||
trace!("keyed_accounts: {:?}", keyed_accounts);
|
trace!("keyed_accounts: {:?}", keyed_accounts);
|
||||||
|
|
||||||
if keyed_accounts.len() < 2 {
|
if keyed_accounts.is_empty() {
|
||||||
Err(InstructionError::InvalidInstructionData)?;
|
Err(InstructionError::InvalidInstructionData)?;
|
||||||
}
|
}
|
||||||
|
|
||||||
// 0th index is the guy who paid for the transaction
|
// 0th index is vote account
|
||||||
let (me, rest) = &mut keyed_accounts.split_at_mut(2);
|
let (me, rest) = &mut keyed_accounts.split_at_mut(1);
|
||||||
let me = &mut me[1];
|
let me = &mut me[0];
|
||||||
|
|
||||||
// TODO: data-driven unpack and dispatch of KeyedAccounts
|
// TODO: data-driven unpack and dispatch of KeyedAccounts
|
||||||
match deserialize(data).map_err(|_| InstructionError::InvalidInstructionData)? {
|
match deserialize(data).map_err(|_| InstructionError::InvalidInstructionData)? {
|
||||||
|
@ -191,7 +177,6 @@ mod tests {
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
process_instruction(&vote(
|
process_instruction(&vote(
|
||||||
&Pubkey::default(),
|
|
||||||
&Pubkey::default(),
|
&Pubkey::default(),
|
||||||
&Pubkey::default(),
|
&Pubkey::default(),
|
||||||
vec![Vote::default()]
|
vec![Vote::default()]
|
||||||
|
@ -203,7 +188,6 @@ mod tests {
|
||||||
&Pubkey::default(),
|
&Pubkey::default(),
|
||||||
&Pubkey::default(),
|
&Pubkey::default(),
|
||||||
&Pubkey::default(),
|
&Pubkey::default(),
|
||||||
&Pubkey::default(),
|
|
||||||
)),
|
)),
|
||||||
Err(InstructionError::InvalidAccountData),
|
Err(InstructionError::InvalidAccountData),
|
||||||
);
|
);
|
||||||
|
|
|
@ -75,6 +75,21 @@ impl Transaction {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn new_with_payer(instructions: Vec<Instruction>, payer: Option<&Pubkey>) -> Self {
|
||||||
|
let message = Message::new_with_payer(instructions, payer);
|
||||||
|
Self::new_unsigned(message)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn new_signed_with_payer<T: KeypairUtil>(
|
||||||
|
instructions: Vec<Instruction>,
|
||||||
|
payer: Option<&Pubkey>,
|
||||||
|
signing_keypairs: &[&T],
|
||||||
|
recent_blockhash: Hash,
|
||||||
|
) -> Self {
|
||||||
|
let message = Message::new_with_payer(instructions, payer);
|
||||||
|
Self::new(signing_keypairs, message, recent_blockhash)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn new_unsigned_instructions(instructions: Vec<Instruction>) -> Self {
|
pub fn new_unsigned_instructions(instructions: Vec<Instruction>) -> Self {
|
||||||
let message = Message::new(instructions);
|
let message = Message::new(instructions);
|
||||||
Self::new_unsigned(message)
|
Self::new_unsigned(message)
|
||||||
|
@ -167,25 +182,11 @@ impl Transaction {
|
||||||
serialize(&self.message()).unwrap()
|
serialize(&self.message()).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Sign this transaction.
|
|
||||||
pub fn sign_unchecked<T: KeypairUtil>(&mut self, keypairs: &[&T], recent_blockhash: Hash) {
|
|
||||||
self.message.recent_blockhash = recent_blockhash;
|
|
||||||
let message_data = self.message_data();
|
|
||||||
self.signatures = keypairs
|
|
||||||
.iter()
|
|
||||||
.map(|keypair| keypair.sign_message(&message_data))
|
|
||||||
.collect();
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Check keys and keypair lengths, then sign this transaction.
|
/// Check keys and keypair lengths, then sign this transaction.
|
||||||
pub fn sign<T: KeypairUtil>(&mut self, keypairs: &[&T], recent_blockhash: Hash) {
|
pub fn sign<T: KeypairUtil>(&mut self, keypairs: &[&T], recent_blockhash: Hash) {
|
||||||
let signed_keys =
|
self.partial_sign(keypairs, recent_blockhash);
|
||||||
&self.message.account_keys[0..self.message.header.num_required_signatures as usize];
|
|
||||||
for (i, keypair) in keypairs.iter().enumerate() {
|
assert_eq!(self.is_signed(), true, "not enough keypairs");
|
||||||
assert_eq!(keypair.pubkey(), signed_keys[i], "keypair-pubkey mismatch");
|
|
||||||
}
|
|
||||||
assert_eq!(keypairs.len(), signed_keys.len(), "not enough keypairs");
|
|
||||||
self.sign_unchecked(keypairs, recent_blockhash);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Sign using some subset of required keys
|
/// Sign using some subset of required keys
|
||||||
|
|
|
@ -472,18 +472,19 @@ fn process_authorize_voter(
|
||||||
) -> ProcessResult {
|
) -> ProcessResult {
|
||||||
let (recent_blockhash, _fee_calculator) = rpc_client.get_recent_blockhash()?;
|
let (recent_blockhash, _fee_calculator) = rpc_client.get_recent_blockhash()?;
|
||||||
let ixs = vec![vote_instruction::authorize_voter(
|
let ixs = vec![vote_instruction::authorize_voter(
|
||||||
&config.keypair.pubkey(), // from
|
|
||||||
voting_account_pubkey, // vote account to update
|
voting_account_pubkey, // vote account to update
|
||||||
&authorized_voter_keypair.pubkey(), // current authorized voter (often the vote account itself)
|
&authorized_voter_keypair.pubkey(), // current authorized voter (often the vote account itself)
|
||||||
new_authorized_voter_pubkey, // new vote signer
|
new_authorized_voter_pubkey, // new vote signer
|
||||||
)];
|
)];
|
||||||
|
|
||||||
let mut tx = Transaction::new_signed_instructions(
|
let mut tx = Transaction::new_signed_with_payer(
|
||||||
&[&config.keypair, &authorized_voter_keypair],
|
|
||||||
ixs,
|
ixs,
|
||||||
|
Some(&config.keypair.pubkey()),
|
||||||
|
&[&config.keypair, &authorized_voter_keypair],
|
||||||
recent_blockhash,
|
recent_blockhash,
|
||||||
);
|
);
|
||||||
let signature_str = rpc_client.send_and_confirm_transaction(&mut tx, &[&config.keypair])?;
|
let signature_str = rpc_client
|
||||||
|
.send_and_confirm_transaction(&mut tx, &[&config.keypair, &authorized_voter_keypair])?;
|
||||||
Ok(signature_str.to_string())
|
Ok(signature_str.to_string())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue