DoS tool: generate transactions using several threads (#26286)

* add cli arg num_gen_threads * introduce many generating threads * add sender thread * add time measurments * cleanup * sort dependencies * revisit threads termination * make send_batch_size to be configurable * update Cargo.lock
solana-labs · Aug 1, 2022 · e74ad90 · e74ad90
1 parent ac91577
commit e74ad90
Show file tree

Hide file tree

Showing 4 changed files with 272 additions and 84 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/dos/Cargo.toml b/dos/Cargo.toml
@@ -12,6 +12,7 @@ description = "Tool to send various requests to cluster in order to evaluate the
 [dependencies]
 bincode = "1.3.3"
 clap = { version = "3.1.5", features = ["derive", "cargo"] }
+crossbeam-channel = "0.5.4"
 itertools = "0.10.3"
 log = "0.4.17"
 rand = "0.7.0"
@@ -22,6 +23,7 @@ solana-core = { path = "../core", version = "=1.11.5" }
 solana-faucet = { path = "../faucet", version = "=1.11.5" }
 solana-gossip = { path = "../gossip", version = "=1.11.5" }
 solana-logger = { path = "../logger", version = "=1.11.5" }
+solana-measure = { path = "../measure", version = "=1.11.5" }
 solana-net-utils = { path = "../net-utils", version = "=1.11.5" }
 solana-perf = { path = "../perf", version = "=1.11.5" }
 solana-rpc = { path = "../rpc", version = "=1.11.5" }

diff --git a/dos/src/cli.rs b/dos/src/cli.rs
@@ -48,6 +48,13 @@ pub struct DosClientParameters {
     #[clap(long, help = "Allow contacting private ip addresses")]
     pub allow_private_addr: bool,
 
+    #[clap(
+        long,
+        default_value = "1",
+        help = "Number of threads generating transactions"
+    )]
+    pub num_gen_threads: usize,
+
     #[clap(flatten)]
     pub transaction_params: TransactionParams,
 
@@ -57,9 +64,12 @@ pub struct DosClientParameters {
         help = "Submit transactions via QUIC"
     )]
     pub tpu_use_quic: bool,
+
+    #[clap(long, default_value = "16384", help = "Size of the transactions batch")]
+    pub send_batch_size: usize,
 }
 
-#[derive(Args, Serialize, Deserialize, Debug, Default, PartialEq, Eq)]
+#[derive(Args, Clone, Serialize, Deserialize, Debug, Default, PartialEq, Eq)]
 #[clap(rename_all = "kebab-case")]
 pub struct TransactionParams {
     #[clap(
@@ -219,6 +229,8 @@ mod tests {
                 allow_private_addr: false,
                 transaction_params: TransactionParams::default(),
                 tpu_use_quic: false,
+                num_gen_threads: 1,
+                send_batch_size: 16384,
             },
         );
     }
@@ -237,6 +249,8 @@ mod tests {
             "--num-signatures",
             "8",
             "--tpu-use-quic",
+            "--send-batch-size",
+            "1",
         ])
         .unwrap();
         assert_eq!(
@@ -249,6 +263,7 @@ mod tests {
                 data_input: None,
                 skip_gossip: false,
                 allow_private_addr: false,
+                num_gen_threads: 1,
                 transaction_params: TransactionParams {
                     num_signatures: Some(8),
                     valid_blockhash: false,
@@ -258,6 +273,7 @@ mod tests {
                     num_instructions: None,
                 },
                 tpu_use_quic: true,
+                send_batch_size: 1,
             },
         );
     }
@@ -277,6 +293,8 @@ mod tests {
             "transfer",
             "--num-instructions",
             "1",
+            "--send-batch-size",
+            "1",
         ])
         .unwrap();
         assert_eq!(
@@ -289,6 +307,7 @@ mod tests {
                 data_input: None,
                 skip_gossip: false,
                 allow_private_addr: false,
+                num_gen_threads: 1,
                 transaction_params: TransactionParams {
                     num_signatures: None,
                     valid_blockhash: true,
@@ -298,6 +317,7 @@ mod tests {
                     num_instructions: Some(1),
                 },
                 tpu_use_quic: false,
+                send_batch_size: 1,
             },
         );
 
@@ -332,6 +352,8 @@ mod tests {
             "transfer",
             "--num-instructions",
             "8",
+            "--send-batch-size",
+            "1",
         ])
         .unwrap();
         assert_eq!(
@@ -344,6 +366,7 @@ mod tests {
                 data_input: None,
                 skip_gossip: false,
                 allow_private_addr: false,
+                num_gen_threads: 1,
                 transaction_params: TransactionParams {
                     num_signatures: None,
                     valid_blockhash: true,
@@ -353,6 +376,7 @@ mod tests {
                     num_instructions: Some(8),
                 },
                 tpu_use_quic: false,
+                send_batch_size: 1,
             },
         );
     }
@@ -370,6 +394,8 @@ mod tests {
             "--valid-blockhash",
             "--transaction-type",
             "account-creation",
+            "--send-batch-size",
+            "1",
         ])
         .unwrap();
         assert_eq!(
@@ -382,6 +408,7 @@ mod tests {
                 data_input: None,
                 skip_gossip: false,
                 allow_private_addr: false,
+                num_gen_threads: 1,
                 transaction_params: TransactionParams {
                     num_signatures: None,
                     valid_blockhash: true,
@@ -391,6 +418,7 @@ mod tests {
                     num_instructions: None,
                 },
                 tpu_use_quic: false,
+                send_batch_size: 1,
             },
         );
     }