Compare commits
6 Commits
90f679c366
...
33e540add4
Author | SHA1 | Date | |
---|---|---|---|
33e540add4 | |||
a315e2662c | |||
9000924590 | |||
a9b1ffa614 | |||
b1255c451e | |||
04ebc2290f |
5
.gitignore
vendored
5
.gitignore
vendored
@ -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
18
Cargo.toml
Normal 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
3
bin/docker-build.sh
Executable 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
25
src/hdtest.rs
Normal 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
78
src/nettest.rs
Normal 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),
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user