Merge branch 'main' of github.com:shinnku-nikaidou/layer7benchmark

Author: shinnku-nikaidou

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "layer7benchmark"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 description = "A simple benchmark tool for Layer 7 (HTTP) load testing"
 license = "MIT"
@@ -12,7 +12,3 @@ clap = { version = "4.5.38", features = ["derive"] }
 rand = "0.9.1"
 reqwest = { version = "0.12.15", features = ["json"] }
 tokio = { version = "1.45.0", features = ["full"] }
-openssl = { version = "0.10", features = ["vendored"] }
-
-[target.x86_64-unknown-linux-musl]
-linker = "x86_64-linux-musl-gcc"

Cargo.toml

@@ -11,5 +11,12 @@
     url to download (default is https://www.google.com)
 -t or --time u32
     time to download (default is 60)
+-h or --help
+    help for layer7benchmark
+-v or --version
+    version for layer7benchmark
+--header string
+    http header to send
+-i or --ip string
+    ip to send the request to (default is automatically resolved from the url)
 ```
-

README.md

@@ -10,4 +10,7 @@ pub struct Args {
 
     #[arg(short = 't', long = "time", default_value_t = 60)]
     pub time: u64,
+
+    #[arg(short='i',long="ip", default_value = "")]
+    pub ip: String,
 }

src/args.rs

@@ -1,20 +1,59 @@
 mod args;
+use core::panic;
 use std::time::Duration;
 
 use args::Args;
 use clap::Parser;
 use reqwest::Client;
+use std::net::SocketAddr;
+use tokio::net::lookup_host;
 use tokio::sync::broadcast;
+use url::Url;
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
     let args = Args::parse();
     let mut handles = Vec::new();
-    let client = Client::new();
+    let url = args.url.clone();
+    let parsed_url = Url::parse(&args.url)?;
+    let domain = parsed_url.host_str().unwrap_or_default();
+
+    println!("the download url is: {}", url);
+    println!("the domain is: {}", domain);
+    let mut addrs = lookup_host((domain, 0)).await?;
+    let ip;
+
+    match addrs.next() {
+        Some(socket_addr) => {
+            ip = socket_addr.ip();
+            println!(
+                "For domain '{}', no specific IP provided. System DNS lookup suggests: {}.",
+                domain, ip
+            );
+        }
+        None => {
+            eprintln!(
+                "Could not resolve IP for domain '{}' via system DNS lookup.",
+                domain
+            );
+            panic!("Please check your command or computer.");
+        }
+    }
+
+    let client = if args.ip.is_empty() {
+        // Client::new()
+        Client::builder()
+            .resolve(domain, SocketAddr::new(ip, 0))
+            .build()?
+    } else {
+        Client::builder()
+            .resolve(domain, format!("{}:0", args.ip).parse::<SocketAddr>()?)
+            .build()?
+    };
     let (shutdown_tx, _) = broadcast::channel(1);
 
     for _ in 0..args.concurrent_count {
-        let url = args.url.clone();
+        let url = url.clone();
         let client = client.clone();
         let mut shutdown_rx = shutdown_tx.subscribe();