Fixed a small logic error; Updated Rust version to build against in the Docker build script

Reviewed-on: #1

.gitignore

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

Cargo.toml

@@ -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"] }

bin/docker-build.sh

@@ -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

src/hdtest.rs

@@ -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),
+    }
+}

src/nettest.rs

@@ -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),
+    }
+}