Added a new CPU test suite with a CPU stress test tool.
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful

This commit is contained in:
Gregory Ballantine 2022-08-07 10:18:30 -04:00
parent 983b1cbeef
commit 0f887d0c76
4 changed files with 73 additions and 13 deletions

View File

@ -14,6 +14,7 @@ path = "src/main.rs"
[dependencies] [dependencies]
chrono = "0.4.20" chrono = "0.4.20"
clap = { version = "3.2.16", features = ["derive"] } clap = { version = "3.2.16", features = ["derive"] }
sysinfo = "0.25.1"
[package.metadata.deb] [package.metadata.deb]
depends = "fio" depends = "fio"

View File

@ -1,4 +1,5 @@
mod benchmarks; mod benchmarks;
mod stress;
mod text; mod text;
use clap::{Args, Parser, Subcommand}; use clap::{Args, Parser, Subcommand};
@ -10,12 +11,15 @@ struct Cli {
#[clap(subcommand)] #[clap(subcommand)]
command: Commands, command: Commands,
#[clap(short = 'l', default_value_t = 1, help = "Number of times to run test. Default = 1", global = true)] #[clap(short = 'l', long, default_value_t = 1, help = "Number of times to run test. Default = 1", global = true)]
loopcount: u8, loopcount: u8,
} }
#[derive(Subcommand)] #[derive(Subcommand)]
enum Commands { enum Commands {
// CPU benchmarks subcommand
#[clap(name = "cpu", about = "CPU benchmarks and stress tests.")]
Cpu(Cpu),
// disk benchmarks subcommand // disk benchmarks subcommand
#[clap(name = "disk", about = "Hard drive and SSD benchmarks.")] #[clap(name = "disk", about = "Hard drive and SSD benchmarks.")]
Disk(Disk), Disk(Disk),
@ -24,6 +28,22 @@ enum Commands {
Net(Net), Net(Net),
} }
#[derive(Args)]
struct Cpu {
#[clap(subcommand)]
cpu_commands: CpuCommands,
}
#[derive(Subcommand)]
enum CpuCommands {
// CPU stress test subcommand
#[clap(name = "stress", about = "Stress test the CPU with math!")]
StressTest {
#[clap(short = 't', long, default_value_t = 0, help = "Number of threads to use; defaults to CPU's max thread count.")]
threads: usize,
},
}
#[derive(Args)] #[derive(Args)]
struct Disk { struct Disk {
#[clap(subcommand)] #[clap(subcommand)]
@ -35,36 +55,36 @@ enum DiskCommands {
// sequential disk read subcommand // sequential disk read subcommand
#[clap(name = "read_seq", about = "Sequential disk read speed test.")] #[clap(name = "read_seq", about = "Sequential disk read speed test.")]
ReadSeqTest { ReadSeqTest {
#[clap(short = 'f', default_value_t = String::from("/tmp/disk-test.tmp"))] #[clap(short = 'f', long, default_value_t = String::from("/tmp/disk-test.tmp"))]
tempfile: String, tempfile: String,
#[clap(short = 's', default_value_t = 15)] #[clap(short = 's', long, default_value_t = 15)]
size: u8, size: u8,
}, },
// random disk read subcommand // random disk read subcommand
#[clap(name = "read_rand", about = "Random 4K disk read speed test.")] #[clap(name = "read_rand", about = "Random 4K disk read speed test.")]
ReadRandTest { ReadRandTest {
#[clap(short = 'f', default_value_t = String::from("/tmp/disk-test.tmp"))] #[clap(short = 'f', long, default_value_t = String::from("/tmp/disk-test.tmp"))]
tempfile: String, tempfile: String,
#[clap(short = 's', default_value_t = 15)] #[clap(short = 's', long, default_value_t = 15)]
size: u8, size: u8,
}, },
// sequential disk write subcommand // sequential disk write subcommand
#[clap(name = "write_seq", about = "Write a large file to determine sequential disk write speeds.")] #[clap(name = "write_seq", about = "Write a large file to determine sequential disk write speeds.")]
WriteSeqTest { WriteSeqTest {
#[clap(short = 't', default_value_t = String::from("/tmp/disk-test.tmp"))] #[clap(short = 't', long, default_value_t = String::from("/tmp/disk-test.tmp"))]
tempfile: String, tempfile: String,
#[clap(short = 's', default_value_t = 15)] #[clap(short = 's', long, default_value_t = 15)]
size: u8, size: u8,
}, },
// random 4K disk write subcommand // random 4K disk write subcommand
#[clap(name = "write_rand", about = "Write a bunch of smallfiles to determine random disk write speeds.")] #[clap(name = "write_rand", about = "Write a bunch of smallfiles to determine random disk write speeds.")]
WriteRandTest { WriteRandTest {
#[clap(short = 't', default_value_t = String::from("/tmp/disk-test.tmp"))] #[clap(short = 't', long, default_value_t = String::from("/tmp/disk-test.tmp"))]
tempfile: String, tempfile: String,
#[clap(short = 's', default_value_t = 15)] #[clap(short = 's', long, default_value_t = 15)]
size: u8, size: u8,
}, },
} }
@ -80,18 +100,18 @@ enum NetCommands {
// ping subcommand // ping subcommand
#[clap(name = "ping", about = "Ping a host to determine network latency.")] #[clap(name = "ping", about = "Ping a host to determine network latency.")]
Ping { Ping {
#[clap(short = 't', default_value_t = String::from("8.8.8.8"))] #[clap(short = 't', long, default_value_t = String::from("8.8.8.8"))]
host: String, host: String,
#[clap(short = 'c', default_value_t = 30)] #[clap(short = 'c', long, default_value_t = 30)]
count: u16, count: u16,
}, },
// bandwidth test subcommand // bandwidth test subcommand
#[clap(name = "bandwidth", about = "Downloads a remote file to determine network bandwidth.")] #[clap(name = "bandwidth", about = "Downloads a remote file to determine network bandwidth.")]
Bandwidth { Bandwidth {
#[clap(short = 'd', default_value_t = String::from("https://www.bitgoblin.tech/hardware-tests/export-01.mp4"))] #[clap(short = 'd', long, default_value_t = String::from("https://www.bitgoblin.tech/hardware-tests/export-01.mp4"))]
download: String, download: String,
#[clap(short = 'o', default_value_t = String::from("./tempfile"))] #[clap(short = 'o', long, default_value_t = String::from("./tempfile"))]
output: String, output: String,
}, },
} }
@ -101,6 +121,10 @@ fn main() {
// map subcommands back to the main command // map subcommands back to the main command
match &cli.command { match &cli.command {
Commands::Cpu(args) => match &args.cpu_commands {
CpuCommands::StressTest { threads } => stress::cpu::cpu_stress_math(*threads),
},
Commands::Disk(args) => match &args.disk_commands { Commands::Disk(args) => match &args.disk_commands {
DiskCommands::ReadSeqTest { tempfile, size } => { DiskCommands::ReadSeqTest { tempfile, size } => {
for i in 0..cli.loopcount { for i in 0..cli.loopcount {

34
src/stress/cpu.rs Normal file
View File

@ -0,0 +1,34 @@
use std::thread;
use sysinfo::{System,SystemExt};
pub fn cpu_stress_math(threads: usize) {
// fetch system information
let mut sys = System::new_all();
sys.refresh_all();
let num_cpus = sys.cpus().len();
let mut num_threads = threads;
if num_threads == 0 {
println!("Number of threads not specified, defaulting to CPU's thread count of {}.", num_cpus);
num_threads = num_cpus;
} else {
println!("Using specified thread count of {}", num_threads);
}
for i in 1..num_threads {
println!("Spawning thread number {}", i);
thread::spawn (|| {
worker();
});
}
println!("Using main as last thread");
worker();
}
fn worker() {
let mut _x = 0;
loop {
_x += 1;
_x -= 1;
}
}

1
src/stress/mod.rs Normal file
View File

@ -0,0 +1 @@
pub mod cpu;