Compare commits

...

6 Commits

5 changed files with 129 additions and 0 deletions

5
.gitignore vendored
View File

@ -14,3 +14,8 @@ Cargo.lock
# MSVC Windows builds of rustc generate these, which store debugging information
*.pdb
# Added by cargo
/target

18
Cargo.toml Normal file
View File

@ -0,0 +1,18 @@
[package]
name = "hardware-tests"
version = "0.1.0"
edition = "2021"
[[bin]]
name = "hdtest"
path = "src/hdtest.rs"
[[bin]]
name = "nettest"
path = "src/nettest.rs"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
chrono = "0.4.19"
clap = { version = "3.1.2", features = ["derive"] }

3
bin/docker-build.sh Executable file
View File

@ -0,0 +1,3 @@
#!/bin/sh
docker run --rm --user "$(id -u)":"$(id -g)" -v "$PWD":/usr/src/hardware-tests -w /usr/src/hardware-tests rust:1.62 cargo build

25
src/hdtest.rs Normal file
View File

@ -0,0 +1,25 @@
use std::fs;
use std::process::Command;
fn main() {
// run the dd command with a block size of 1 MB, 10K times (10GB file)
let output = Command::new("dd")
.arg("bs=1M")
.arg("count=10240")
.arg("if=/dev/zero")
.arg("of=./speed-test")
.output()
.expect("Failed to execute command");
// check that the command succeeded
assert!(output.status.success());
// for whatever reason, `dd` output ends up in stderr
println!("{}", String::from_utf8_lossy(&output.stderr));
// remove the test file
match fs::remove_file("./speed-test") {
Ok(()) => println!("Cleaning up..."),
Err(e) => println!("There was a problem during cleanup - {}", e),
}
}

78
src/nettest.rs Normal file
View File

@ -0,0 +1,78 @@
use chrono::prelude::*;
use clap::{Parser, Subcommand};
use std::{fs,process};
#[derive(Parser)]
#[clap(name = "Bit Goblin Network Tester", author, version, about = "Network testing app.", long_about = None)]
#[clap(propagate_version = true)]
struct Cli {
#[clap(subcommand)]
command: Commands,
}
#[derive(Subcommand)]
enum Commands {
// ping subcommand
Ping {
#[clap(default_value_t = String::from("8.8.8.8"))]
host: String
},
// bandwidth test subcommand
Bandwidth {
#[clap(default_value_t = String::from("https://www.bitgoblin.tech/hardware-tests/export-01.mp4"))]
download: String,
#[clap(default_value_t = String::from("./tempfile"))]
output: String,
},
}
fn main() {
let cli = Cli::parse();
// map subcommands back to the main command
match &cli.command {
Commands::Ping { host } => ping_host(host),
Commands::Bandwidth { download, output } => bandwidth_test(download, output)
}
}
// ping a host
fn ping_host(host: &str) {
println!("Pinging host {}", host);
// run the ping command with 100 pings
let output = process::Command::new("ping")
.arg(host)
.arg("-c 100")
.output()
.expect("Failed to execute command");
// check that the command succeeded
assert!(output.status.success());
// print out the ping results from stdout
println!("{}", String::from_utf8_lossy(&output.stdout));
}
// timed file copy test to guage bandwidth speeds
fn bandwidth_test(download: &str, output: &str) {
println!("Testing network bandwidth by downloading {}.", download);
// get start time so we can track how long it takes to complete
let start_time = Utc::now();
// do the download
// get finish time
let finish_time = Utc::now();
// compute time to complete
let comp_time = finish_time - start_time;
println!("{}", comp_time.num_milliseconds());
// clean up the test file
match fs::remove_file(output) {
Ok(()) => println!("Cleaning up..."),
Err(e) => println!("There was a problem during cleanup - {}", e),
}
}