506 lines
20 KiB
Rust
506 lines
20 KiB
Rust
use {
|
|
crate::parse_instruction::{
|
|
check_num_accounts, ParsableProgram, ParseInstructionError, ParsedInstructionEnum,
|
|
},
|
|
bincode::deserialize,
|
|
serde_json::json,
|
|
solana_sdk::{instruction::CompiledInstruction, message::AccountKeys},
|
|
solana_vote_program::vote_instruction::VoteInstruction,
|
|
};
|
|
|
|
pub fn parse_vote(
|
|
instruction: &CompiledInstruction,
|
|
account_keys: &AccountKeys,
|
|
) -> Result<ParsedInstructionEnum, ParseInstructionError> {
|
|
let vote_instruction: VoteInstruction = deserialize(&instruction.data)
|
|
.map_err(|_| ParseInstructionError::InstructionNotParsable(ParsableProgram::Vote))?;
|
|
match instruction.accounts.iter().max() {
|
|
Some(index) if (*index as usize) < account_keys.len() => {}
|
|
_ => {
|
|
// Runtime should prevent this from ever happening
|
|
return Err(ParseInstructionError::InstructionKeyMismatch(
|
|
ParsableProgram::Vote,
|
|
));
|
|
}
|
|
}
|
|
match vote_instruction {
|
|
VoteInstruction::InitializeAccount(vote_init) => {
|
|
check_num_vote_accounts(&instruction.accounts, 4)?;
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "initialize".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"rentSysvar": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"clockSysvar": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
"node": account_keys[instruction.accounts[3] as usize].to_string(),
|
|
"authorizedVoter": vote_init.authorized_voter.to_string(),
|
|
"authorizedWithdrawer": vote_init.authorized_withdrawer.to_string(),
|
|
"commission": vote_init.commission,
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::Authorize(new_authorized, authority_type) => {
|
|
check_num_vote_accounts(&instruction.accounts, 3)?;
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "authorize".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"clockSysvar": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"authority": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
"newAuthority": new_authorized.to_string(),
|
|
"authorityType": authority_type,
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::Vote(vote) => {
|
|
check_num_vote_accounts(&instruction.accounts, 4)?;
|
|
let vote = json!({
|
|
"slots": vote.slots,
|
|
"hash": vote.hash.to_string(),
|
|
"timestamp": vote.timestamp,
|
|
});
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "vote".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"slotHashesSysvar": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"clockSysvar": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
"voteAuthority": account_keys[instruction.accounts[3] as usize].to_string(),
|
|
"vote": vote,
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::UpdateVoteState(vote_state_update) => {
|
|
check_num_vote_accounts(&instruction.accounts, 4)?;
|
|
let vote_state_update = json!({
|
|
"lockouts": vote_state_update.lockouts,
|
|
"root": vote_state_update.root,
|
|
"hash": vote_state_update.hash.to_string(),
|
|
"timestamp": vote_state_update.timestamp,
|
|
});
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "updatevotestate".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"slotHashesSysvar": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"clockSysvar": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
"voteAuthority": account_keys[instruction.accounts[3] as usize].to_string(),
|
|
"voteStateUpdate": vote_state_update,
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::UpdateVoteStateSwitch(vote_state_update, hash) => {
|
|
check_num_vote_accounts(&instruction.accounts, 4)?;
|
|
let vote_state_update = json!({
|
|
"lockouts": vote_state_update.lockouts,
|
|
"root": vote_state_update.root,
|
|
"hash": vote_state_update.hash.to_string(),
|
|
"timestamp": vote_state_update.timestamp,
|
|
});
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "updatevotestateswitch".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"slotHashesSysvar": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"clockSysvar": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
"voteAuthority": account_keys[instruction.accounts[3] as usize].to_string(),
|
|
"voteStateUpdate": vote_state_update,
|
|
"hash": hash.to_string(),
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::Withdraw(lamports) => {
|
|
check_num_vote_accounts(&instruction.accounts, 3)?;
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "withdraw".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"destination": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"withdrawAuthority": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
"lamports": lamports,
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::UpdateValidatorIdentity => {
|
|
check_num_vote_accounts(&instruction.accounts, 3)?;
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "updateValidatorIdentity".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"newValidatorIdentity": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"withdrawAuthority": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::UpdateCommission(commission) => {
|
|
check_num_vote_accounts(&instruction.accounts, 2)?;
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "updateCommission".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"withdrawAuthority": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"commission": commission,
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::VoteSwitch(vote, hash) => {
|
|
check_num_vote_accounts(&instruction.accounts, 4)?;
|
|
let vote = json!({
|
|
"slots": vote.slots,
|
|
"hash": vote.hash.to_string(),
|
|
"timestamp": vote.timestamp,
|
|
});
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "voteSwitch".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"slotHashesSysvar": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"clockSysvar": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
"voteAuthority": account_keys[instruction.accounts[3] as usize].to_string(),
|
|
"vote": vote,
|
|
"hash": hash.to_string(),
|
|
}),
|
|
})
|
|
}
|
|
VoteInstruction::AuthorizeChecked(authority_type) => {
|
|
check_num_vote_accounts(&instruction.accounts, 4)?;
|
|
Ok(ParsedInstructionEnum {
|
|
instruction_type: "authorizeChecked".to_string(),
|
|
info: json!({
|
|
"voteAccount": account_keys[instruction.accounts[0] as usize].to_string(),
|
|
"clockSysvar": account_keys[instruction.accounts[1] as usize].to_string(),
|
|
"authority": account_keys[instruction.accounts[2] as usize].to_string(),
|
|
"newAuthority": account_keys[instruction.accounts[3] as usize].to_string(),
|
|
"authorityType": authority_type,
|
|
}),
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
fn check_num_vote_accounts(accounts: &[u8], num: usize) -> Result<(), ParseInstructionError> {
|
|
check_num_accounts(accounts, num, ParsableProgram::Vote)
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use {
|
|
super::*,
|
|
solana_sdk::{hash::Hash, message::Message, pubkey::Pubkey, sysvar},
|
|
solana_vote_program::{
|
|
vote_instruction,
|
|
vote_state::{Vote, VoteAuthorize, VoteInit},
|
|
},
|
|
};
|
|
|
|
#[test]
|
|
fn test_parse_vote_initialize_ix() {
|
|
let lamports = 55;
|
|
|
|
let commission = 10;
|
|
let node_pubkey = Pubkey::new_unique();
|
|
let vote_pubkey = Pubkey::new_unique();
|
|
let authorized_voter = Pubkey::new_unique();
|
|
let authorized_withdrawer = Pubkey::new_unique();
|
|
let vote_init = VoteInit {
|
|
node_pubkey,
|
|
authorized_voter,
|
|
authorized_withdrawer,
|
|
commission,
|
|
};
|
|
|
|
let instructions = vote_instruction::create_account(
|
|
&Pubkey::new_unique(),
|
|
&vote_pubkey,
|
|
&vote_init,
|
|
lamports,
|
|
);
|
|
let message = Message::new(&instructions, None);
|
|
assert_eq!(
|
|
parse_vote(
|
|
&message.instructions[1],
|
|
&AccountKeys::new(&message.account_keys, None)
|
|
)
|
|
.unwrap(),
|
|
ParsedInstructionEnum {
|
|
instruction_type: "initialize".to_string(),
|
|
info: json!({
|
|
"voteAccount": vote_pubkey.to_string(),
|
|
"rentSysvar": sysvar::rent::ID.to_string(),
|
|
"clockSysvar": sysvar::clock::ID.to_string(),
|
|
"node": node_pubkey.to_string(),
|
|
"authorizedVoter": authorized_voter.to_string(),
|
|
"authorizedWithdrawer": authorized_withdrawer.to_string(),
|
|
"commission": commission,
|
|
}),
|
|
}
|
|
);
|
|
assert!(parse_vote(
|
|
&message.instructions[1],
|
|
&AccountKeys::new(&message.account_keys[0..3], None)
|
|
)
|
|
.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_vote_authorize_ix() {
|
|
let vote_pubkey = Pubkey::new_unique();
|
|
let authorized_pubkey = Pubkey::new_unique();
|
|
let new_authorized_pubkey = Pubkey::new_unique();
|
|
let authority_type = VoteAuthorize::Voter;
|
|
let instruction = vote_instruction::authorize(
|
|
&vote_pubkey,
|
|
&authorized_pubkey,
|
|
&new_authorized_pubkey,
|
|
authority_type,
|
|
);
|
|
let message = Message::new(&[instruction], None);
|
|
assert_eq!(
|
|
parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys, None)
|
|
)
|
|
.unwrap(),
|
|
ParsedInstructionEnum {
|
|
instruction_type: "authorize".to_string(),
|
|
info: json!({
|
|
"voteAccount": vote_pubkey.to_string(),
|
|
"clockSysvar": sysvar::clock::ID.to_string(),
|
|
"authority": authorized_pubkey.to_string(),
|
|
"newAuthority": new_authorized_pubkey.to_string(),
|
|
"authorityType": authority_type,
|
|
}),
|
|
}
|
|
);
|
|
assert!(parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys[0..2], None)
|
|
)
|
|
.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_vote_ix() {
|
|
let hash = Hash::new_from_array([1; 32]);
|
|
let vote = Vote {
|
|
slots: vec![1, 2, 4],
|
|
hash,
|
|
timestamp: Some(1_234_567_890),
|
|
};
|
|
|
|
let vote_pubkey = Pubkey::new_unique();
|
|
let authorized_voter_pubkey = Pubkey::new_unique();
|
|
let instruction = vote_instruction::vote(&vote_pubkey, &authorized_voter_pubkey, vote);
|
|
let message = Message::new(&[instruction], None);
|
|
assert_eq!(
|
|
parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys, None)
|
|
)
|
|
.unwrap(),
|
|
ParsedInstructionEnum {
|
|
instruction_type: "vote".to_string(),
|
|
info: json!({
|
|
"voteAccount": vote_pubkey.to_string(),
|
|
"slotHashesSysvar": sysvar::slot_hashes::ID.to_string(),
|
|
"clockSysvar": sysvar::clock::ID.to_string(),
|
|
"voteAuthority": authorized_voter_pubkey.to_string(),
|
|
"vote": {
|
|
"slots": [1, 2, 4],
|
|
"hash": hash.to_string(),
|
|
"timestamp": 1_234_567_890,
|
|
},
|
|
}),
|
|
}
|
|
);
|
|
assert!(parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys[0..3], None)
|
|
)
|
|
.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_vote_withdraw_ix() {
|
|
let lamports = 55;
|
|
let vote_pubkey = Pubkey::new_unique();
|
|
let authorized_withdrawer_pubkey = Pubkey::new_unique();
|
|
let to_pubkey = Pubkey::new_unique();
|
|
let instruction = vote_instruction::withdraw(
|
|
&vote_pubkey,
|
|
&authorized_withdrawer_pubkey,
|
|
lamports,
|
|
&to_pubkey,
|
|
);
|
|
let message = Message::new(&[instruction], None);
|
|
assert_eq!(
|
|
parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys, None)
|
|
)
|
|
.unwrap(),
|
|
ParsedInstructionEnum {
|
|
instruction_type: "withdraw".to_string(),
|
|
info: json!({
|
|
"voteAccount": vote_pubkey.to_string(),
|
|
"destination": to_pubkey.to_string(),
|
|
"withdrawAuthority": authorized_withdrawer_pubkey.to_string(),
|
|
"lamports": lamports,
|
|
}),
|
|
}
|
|
);
|
|
assert!(parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys[0..2], None)
|
|
)
|
|
.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_vote_update_validator_identity_ix() {
|
|
let vote_pubkey = Pubkey::new_unique();
|
|
let authorized_withdrawer_pubkey = Pubkey::new_unique();
|
|
let node_pubkey = Pubkey::new_unique();
|
|
let instruction = vote_instruction::update_validator_identity(
|
|
&vote_pubkey,
|
|
&authorized_withdrawer_pubkey,
|
|
&node_pubkey,
|
|
);
|
|
let message = Message::new(&[instruction], None);
|
|
assert_eq!(
|
|
parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys, None)
|
|
)
|
|
.unwrap(),
|
|
ParsedInstructionEnum {
|
|
instruction_type: "updateValidatorIdentity".to_string(),
|
|
info: json!({
|
|
"voteAccount": vote_pubkey.to_string(),
|
|
"newValidatorIdentity": node_pubkey.to_string(),
|
|
"withdrawAuthority": authorized_withdrawer_pubkey.to_string(),
|
|
}),
|
|
}
|
|
);
|
|
assert!(parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys[0..2], None)
|
|
)
|
|
.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_vote_update_commission_ix() {
|
|
let commission = 10;
|
|
let vote_pubkey = Pubkey::new_unique();
|
|
let authorized_withdrawer_pubkey = Pubkey::new_unique();
|
|
let instruction = vote_instruction::update_commission(
|
|
&vote_pubkey,
|
|
&authorized_withdrawer_pubkey,
|
|
commission,
|
|
);
|
|
let message = Message::new(&[instruction], None);
|
|
assert_eq!(
|
|
parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys, None)
|
|
)
|
|
.unwrap(),
|
|
ParsedInstructionEnum {
|
|
instruction_type: "updateCommission".to_string(),
|
|
info: json!({
|
|
"voteAccount": vote_pubkey.to_string(),
|
|
"withdrawAuthority": authorized_withdrawer_pubkey.to_string(),
|
|
"commission": commission,
|
|
}),
|
|
}
|
|
);
|
|
assert!(parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys[0..1], None)
|
|
)
|
|
.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_vote_switch_ix() {
|
|
let hash = Hash::new_from_array([1; 32]);
|
|
let vote = Vote {
|
|
slots: vec![1, 2, 4],
|
|
hash,
|
|
timestamp: Some(1_234_567_890),
|
|
};
|
|
|
|
let vote_pubkey = Pubkey::new_unique();
|
|
let authorized_voter_pubkey = Pubkey::new_unique();
|
|
let proof_hash = Hash::new_from_array([2; 32]);
|
|
let instruction =
|
|
vote_instruction::vote_switch(&vote_pubkey, &authorized_voter_pubkey, vote, proof_hash);
|
|
let message = Message::new(&[instruction], None);
|
|
assert_eq!(
|
|
parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys, None)
|
|
)
|
|
.unwrap(),
|
|
ParsedInstructionEnum {
|
|
instruction_type: "voteSwitch".to_string(),
|
|
info: json!({
|
|
"voteAccount": vote_pubkey.to_string(),
|
|
"slotHashesSysvar": sysvar::slot_hashes::ID.to_string(),
|
|
"clockSysvar": sysvar::clock::ID.to_string(),
|
|
"voteAuthority": authorized_voter_pubkey.to_string(),
|
|
"vote": {
|
|
"slots": [1, 2, 4],
|
|
"hash": hash.to_string(),
|
|
"timestamp": 1_234_567_890,
|
|
},
|
|
"hash": proof_hash.to_string(),
|
|
}),
|
|
}
|
|
);
|
|
assert!(parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys[0..3], None)
|
|
)
|
|
.is_err());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_vote_authorized_checked_ix() {
|
|
let vote_pubkey = Pubkey::new_unique();
|
|
let authorized_pubkey = Pubkey::new_unique();
|
|
let new_authorized_pubkey = Pubkey::new_unique();
|
|
let authority_type = VoteAuthorize::Voter;
|
|
let instruction = vote_instruction::authorize_checked(
|
|
&vote_pubkey,
|
|
&authorized_pubkey,
|
|
&new_authorized_pubkey,
|
|
authority_type,
|
|
);
|
|
let message = Message::new(&[instruction], None);
|
|
assert_eq!(
|
|
parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys, None)
|
|
)
|
|
.unwrap(),
|
|
ParsedInstructionEnum {
|
|
instruction_type: "authorizeChecked".to_string(),
|
|
info: json!({
|
|
"voteAccount": vote_pubkey.to_string(),
|
|
"clockSysvar": sysvar::clock::ID.to_string(),
|
|
"authority": authorized_pubkey.to_string(),
|
|
"newAuthority": new_authorized_pubkey.to_string(),
|
|
"authorityType": authority_type,
|
|
}),
|
|
}
|
|
);
|
|
assert!(parse_vote(
|
|
&message.instructions[0],
|
|
&AccountKeys::new(&message.account_keys[0..3], None)
|
|
)
|
|
.is_err());
|
|
}
|
|
}
|