tribufu/rust-gamedig
1
0
Fork 0
mirror of https://github.com/tribufu/rust-gamedig synced 2026-05-18 09:35:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cli: Add option to create a packet capture

Browse source
This commit is contained in:
Douile 2023-11-22 02:23:32 +00:00
parent abbcae618f
commit 8d17ca4e48
No known key found for this signature in database
GPG key ID: 1EC3120F9A1028F2
2 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
crates/cli/Cargo.toml
View file

@ -11,8 +11,9 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[features]
default = ["json"]
default = ["json", "packet_capture"]
json = ["dep:serde", "dep:serde_json", "gamedig/serde"]
packet_capture = ["gamedig/packet_capture"]
[dependencies]
clap = { version = "4.1.11", features = ["derive"] }

9
crates/cli/src/main.rs
View file

@ -38,6 +38,10 @@ struct Cli {
#[arg(short, long, default_value = "generic")]
output_mode: OutputMode,
#[cfg(feature = "packet_capture")]
#[arg(short, long)]
capture: Option<String>,
/// Optional timeout settings for the server query.
#[command(flatten, next_help_heading = "Timeouts")]
timeout_settings: Option<TimeoutSettings>,
@ -177,6 +181,11 @@ fn main() -> Result<()> {
// Resolve the IP address
let ip = resolve_ip_or_domain(&args.ip, &mut extra_options)?;
#[cfg(feature = "packet_capture")]
unsafe {
gamedig::capture::simple_setup_capture(args.capture.clone());
}
// Query the server using game definition, parsed IP, and user command line
// flags.
let result = query_with_timeout_and_extra_settings(game, &ip, args.port, args.timeout_settings, extra_options)?;