Skip to content

Commit

Permalink
Fix small problems in transaction-dos tool
Browse files Browse the repository at this point in the history
  • Loading branch information
KirillLykov committed Feb 18, 2022
1 parent 1351c1b commit bb1c3de
Showing 1 changed file with 13 additions and 5 deletions.
18 changes: 13 additions & 5 deletions transaction-dos/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ use {
commitment_config::CommitmentConfig,
instruction::{AccountMeta, Instruction},
message::Message,
packet::PACKET_DATA_SIZE,
pubkey::Pubkey,
rpc_port::DEFAULT_RPC_PORT,
signature::{read_keypair_file, Keypair, Signer},
Expand Down Expand Up @@ -174,7 +175,8 @@ fn run_transactions_dos(

let program_account = client.get_account(&program_id);

let message = Message::new(
let mut blockhash = client.get_latest_blockhash().expect("blockhash");
let mut message = Message::new_with_blockhash(
&[
Instruction::new_with_bytes(
Pubkey::new_unique(),
Expand All @@ -188,12 +190,12 @@ fn run_transactions_dos(
),
],
None,
&blockhash,
);

let mut latest_blockhash = Instant::now();
let mut last_log = Instant::now();
let mut count = 0;
let mut blockhash = client.get_latest_blockhash().expect("blockhash");

if just_calculate_fees {
let fee = client
Expand Down Expand Up @@ -267,6 +269,7 @@ fn run_transactions_dos(
loop {
if latest_blockhash.elapsed().as_secs() > 10 {
blockhash = client.get_latest_blockhash().expect("blockhash");
message.recent_blockhash = blockhash;
latest_blockhash = Instant::now();
}

Expand Down Expand Up @@ -378,7 +381,7 @@ fn run_transactions_dos(
let tx = Transaction::new(&signers, message, blockhash);
if !tested_size.load(Ordering::Relaxed) {
let ser_size = bincode::serialized_size(&tx).unwrap();
assert!(ser_size < 1200, "{}", ser_size);
assert!(ser_size < PACKET_DATA_SIZE as u64, "{}", ser_size);
tested_size.store(true, Ordering::Relaxed);
}
tx
Expand Down Expand Up @@ -657,7 +660,7 @@ pub mod test {
let tx = Transaction::new(&signers, message, blockhash);
let size = bincode::serialized_size(&tx).unwrap();
info!("size:{}", size);
assert!(size < 1200);
assert!(size < PACKET_DATA_SIZE as u64);
}

#[test]
Expand Down Expand Up @@ -698,6 +701,7 @@ pub mod test {
.collect();
let account_keypair_refs: Vec<_> = account_keypairs.iter().collect();
let mut start = Measure::start("total accounts run");

run_transactions_dos(
cluster.entry_point_info.rpc,
faucet_addr,
Expand All @@ -711,7 +715,11 @@ pub mod test {
program_keypair.pubkey(),
Some((
program_keypair,
String::from("../programs/bpf/c/out/tuner.so"),
format!(
"{}{}",
env!("CARGO_MANIFEST_DIR"),
"/../programs/bpf/c/out/tuner.so"
),
)),
&account_keypair_refs,
maybe_account_groups,
Expand Down

0 comments on commit bb1c3de

Please sign in to comment.