Skip generator if non-unique transactions (#25039)

This commit is contained in:
sakridge 2022-05-10 09:22:24 -05:00 committed by GitHub
parent e27e9fb538
commit 6cfa19e2e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions

View File

@ -240,16 +240,16 @@ fn run_dos(
data.resize(params.data_size, 0); data.resize(params.data_size, 0);
} }
DataType::Transaction => { DataType::Transaction => {
let tp = params.transaction_params; let tp = params.transaction_params.clone();
info!("{:?}", tp); info!("{:?}", tp);
transaction_generator = Some(TransactionGenerator::new(tp)); let mut tg = TransactionGenerator::new(tp);
let tx = transaction_generator let tx = tg.generate(payer, &rpc_client);
.as_mut()
.unwrap()
.generate(payer, &rpc_client);
info!("{:?}", tx); info!("{:?}", tx);
data = bincode::serialize(&tx).unwrap(); data = bincode::serialize(&tx).unwrap();
if params.transaction_params.unique_transactions {
transaction_generator = Some(tg);
}
} }
DataType::GetAccountInfo => {} DataType::GetAccountInfo => {}
DataType::GetProgramAccounts => {} DataType::GetProgramAccounts => {}
@ -287,7 +287,7 @@ fn run_dos(
} }
if let Some(tg) = transaction_generator.as_mut() { if let Some(tg) = transaction_generator.as_mut() {
let tx = tg.generate(payer, &rpc_client); let tx = tg.generate(payer, &rpc_client);
info!("{:?}", tx); debug!("{:?}", tx);
data = bincode::serialize(&tx).unwrap(); data = bincode::serialize(&tx).unwrap();
} }
let res = socket.send_to(&data, target); let res = socket.send_to(&data, target);
@ -350,7 +350,7 @@ struct DosClientParameters {
transaction_params: TransactionParams, transaction_params: TransactionParams,
} }
#[derive(Args, Serialize, Deserialize, Debug, Default)] #[derive(Clone, Args, Serialize, Deserialize, Debug, Default)]
#[clap(rename_all = "kebab-case")] #[clap(rename_all = "kebab-case")]
struct TransactionParams { struct TransactionParams {
#[clap( #[clap(