// Take a look at the license at the top of the repository in the LICENSE file. #![crate_type = "bin"] #![allow(unused_must_use, non_upper_case_globals)] extern crate sysinfo; use std::io::{self, BufRead, Write}; use std::str::FromStr; use sysinfo::Signal::*; use sysinfo::{ CpuExt, NetworkExt, NetworksExt, Pid, ProcessExt, Signal, System, SystemExt, UserExt, }; const signals: &[Signal] = &[ Hangup, Interrupt, Quit, Illegal, Trap, Abort, Bus, FloatingPointException, Kill, User1, Segv, User2, Pipe, Alarm, Term, Child, Continue, Stop, TSTP, TTIN, TTOU, Urgent, XCPU, XFSZ, VirtualAlarm, Profiling, Winch, IO, Power, Sys, ]; fn print_help() { writeln!(&mut io::stdout(), "== Help menu =="); writeln!(&mut io::stdout(), "help : show this menu"); writeln!( &mut io::stdout(), "signals : show the available signals" ); writeln!( &mut io::stdout(), "refresh : reloads all processes' information" ); writeln!( &mut io::stdout(), "refresh [pid] : reloads corresponding process' information" ); writeln!( &mut io::stdout(), "refresh_disks : reloads only disks' information" ); writeln!( &mut io::stdout(), "refresh_users : reloads only users' information" ); writeln!( &mut io::stdout(), "show [pid | name] : show information of the given process \ corresponding to [pid | name]" ); writeln!( &mut io::stdout(), "kill [pid] [signal]: send [signal] to the process with this \ [pid]. 0 < [signal] < 32" ); writeln!( &mut io::stdout(), "cpus : Displays CPUs state" ); writeln!( &mut io::stdout(), "memory : Displays memory state" ); writeln!( &mut io::stdout(), "temperature : Displays components' temperature" ); writeln!( &mut io::stdout(), "disks : Displays disks' information" ); writeln!( &mut io::stdout(), "network : Displays network' information" ); writeln!( &mut io::stdout(), "all : Displays all process name and pid" ); writeln!( &mut io::stdout(), "uptime : Displays system uptime" ); writeln!( &mut io::stdout(), "boot_time : Displays system boot time" ); writeln!( &mut io::stdout(), "vendor_id : Displays CPU vendor id" ); writeln!(&mut io::stdout(), "brand : Displays CPU brand"); writeln!( &mut io::stdout(), "load_avg : Displays system load average" ); writeln!( &mut io::stdout(), "frequency : Displays CPU frequency" ); writeln!(&mut io::stdout(), "users : Displays all users"); writeln!( &mut io::stdout(), "system : Displays system information (such as name, version and hostname)" ); writeln!( &mut io::stdout(), "pid : Display this example's PID" ); writeln!(&mut io::stdout(), "quit : Exit the program"); } fn interpret_input(input: &str, sys: &mut System) -> bool { match input.trim() { "help" => print_help(), "refresh_disks" => { writeln!(&mut io::stdout(), "Refreshing disk list..."); sys.refresh_disks_list(); writeln!(&mut io::stdout(), "Done."); } "refresh_users" => { writeln!(&mut io::stdout(), "Refreshing user list..."); sys.refresh_users_list(); writeln!(&mut io::stdout(), "Done."); } "signals" => { let mut nb = 1i32; for sig in signals { writeln!(&mut io::stdout(), "{:2}:{:?}", nb, sig); nb += 1; } } "cpus" => { // Note: you should refresh a few times before using this, so that usage statistics // can be ascertained writeln!( &mut io::stdout(), "number of physical cores: {}", sys.physical_core_count() .map(|c| c.to_string()) .unwrap_or_else(|| "Unknown".to_owned()), ); writeln!( &mut io::stdout(), "total process usage: {}%", sys.global_cpu_info().cpu_usage() ); for proc_ in sys.cpus() { writeln!(&mut io::stdout(), "{:?}", proc_); } } "memory" => { writeln!(&mut io::stdout(), "total memory: {} KB", sys.total_memory()); writeln!(&mut io::stdout(), "used memory : {} KB", sys.used_memory()); writeln!(&mut io::stdout(), "total swap : {} KB", sys.total_swap()); writeln!(&mut io::stdout(), "used swap : {} KB", sys.used_swap()); } "quit" | "exit" => return true, "all" => { for (pid, proc_) in sys.processes() { writeln!( &mut io::stdout(), "{}:{} status={:?}", pid, proc_.name(), proc_.status() ); } } "frequency" => { writeln!( &mut io::stdout(), "{} MHz", sys.global_cpu_info().frequency() ); } "vendor_id" => { writeln!( &mut io::stdout(), "vendor ID: {}", sys.cpus()[0].vendor_id() ); } "brand" => { writeln!(&mut io::stdout(), "brand: {}", sys.cpus()[0].brand()); } "load_avg" => { let load_avg = sys.load_average(); writeln!(&mut io::stdout(), "one minute : {}%", load_avg.one); writeln!(&mut io::stdout(), "five minutes : {}%", load_avg.five); writeln!(&mut io::stdout(), "fifteen minutes: {}%", load_avg.fifteen); } e if e.starts_with("show ") => { let tmp: Vec<&str> = e.split(' ').collect(); if tmp.len() != 2 { writeln!( &mut io::stdout(), "show command takes a pid or a name in parameter!" ); writeln!(&mut io::stdout(), "example: show 1254"); } else if let Ok(pid) = Pid::from_str(tmp[1]) { match sys.process(pid) { Some(p) => writeln!(&mut io::stdout(), "{:?}", *p), None => writeln!(&mut io::stdout(), "pid \"{:?}\" not found", pid), }; } else { let proc_name = tmp[1]; for proc_ in sys.processes_by_name(proc_name) { writeln!(&mut io::stdout(), "==== {} ====", proc_.name()); writeln!(&mut io::stdout(), "{:?}", proc_); } } } "temperature" => { for component in sys.components() { writeln!(&mut io::stdout(), "{:?}", component); } } "network" => { for (interface_name, data) in sys.networks().iter() { writeln!( &mut io::stdout(), "{}:\n input data (new / total): {} / {} B\n output data (new / total): {} / {} B", interface_name, data.received(), data.total_received(), data.transmitted(), data.total_transmitted(), ); } } "show" => { writeln!( &mut io::stdout(), "'show' command expects a pid number or a process name" ); } e if e.starts_with("kill ") => { let tmp: Vec<&str> = e.split(' ').collect(); if tmp.len() != 3 { writeln!( &mut io::stdout(), "kill command takes the pid and a signal number in parameter!" ); writeln!(&mut io::stdout(), "example: kill 1254 9"); } else { let pid = Pid::from_str(tmp[1]).unwrap(); let signal = i32::from_str(tmp[2]).unwrap(); if signal < 1 || signal > 31 { writeln!( &mut io::stdout(), "Signal must be between 0 and 32 ! See the signals list with the \ signals command" ); } else { match sys.process(pid) { Some(p) => { if let Some(res) = p.kill_with(*signals.get(signal as usize - 1).unwrap()) { writeln!(&mut io::stdout(), "kill: {}", res,); } else { writeln!( &mut io::stdout(), "kill: signal not supported on this platform" ); } } None => { writeln!(&mut io::stdout(), "pid not found"); } }; } } } "disks" => { for disk in sys.disks() { writeln!(&mut io::stdout(), "{:?}", disk); } } "users" => { for user in sys.users() { writeln!(&mut io::stdout(), "{:?}", user.name()); } } "boot_time" => { writeln!(&mut io::stdout(), "{} seconds", sys.boot_time()); } "uptime" => { let up = sys.uptime(); let mut uptime = sys.uptime(); let days = uptime / 86400; uptime -= days * 86400; let hours = uptime / 3600; uptime -= hours * 3600; let minutes = uptime / 60; writeln!( &mut io::stdout(), "{} days {} hours {} minutes ({} seconds in total)", days, hours, minutes, up, ); } x if x.starts_with("refresh") => { if x == "refresh" { writeln!(&mut io::stdout(), "Getting processes' information..."); sys.refresh_all(); writeln!(&mut io::stdout(), "Done."); } else if x.starts_with("refresh ") { writeln!(&mut io::stdout(), "Getting process' information..."); if let Some(pid) = x .split(' ') .filter_map(|pid| pid.parse().ok()) .take(1) .next() { if sys.refresh_process(pid) { writeln!(&mut io::stdout(), "Process `{}` updated successfully", pid); } else { writeln!( &mut io::stdout(), "Process `{}` couldn't be updated...", pid ); } } else { writeln!(&mut io::stdout(), "Invalid [pid] received..."); } } else { writeln!( &mut io::stdout(), "\"{}\": Unknown command. Enter 'help' if you want to get the commands' \ list.", x ); } } "pid" => { writeln!( &mut io::stdout(), "PID: {}", sysinfo::get_current_pid().expect("failed to get PID") ); } "system" => { writeln!( &mut io::stdout(), "System name: {}\n\ System kernel version: {}\n\ System OS version: {}\n\ System OS (long) version: {}\n\ System host name: {}", sys.name().unwrap_or_else(|| "".to_owned()), sys.kernel_version() .unwrap_or_else(|| "".to_owned()), sys.os_version().unwrap_or_else(|| "".to_owned()), sys.long_os_version() .unwrap_or_else(|| "".to_owned()), sys.host_name().unwrap_or_else(|| "".to_owned()), ); } e => { writeln!( &mut io::stdout(), "\"{}\": Unknown command. Enter 'help' if you want to get the commands' \ list.", e ); } } false } fn main() { println!("Getting processes' information..."); let mut t = System::new_all(); println!("Done."); let t_stin = io::stdin(); let mut stin = t_stin.lock(); let mut done = false; println!("To get the commands' list, enter 'help'."); while !done { let mut input = String::new(); write!(&mut io::stdout(), "> "); io::stdout().flush(); stin.read_line(&mut input); if input.is_empty() { // The string is empty, meaning there is no '\n', meaning // that the user used CTRL+D so we can just quit! println!("\nLeaving, bye!"); break; } if (&input as &str).ends_with('\n') { input.pop(); } done = interpret_input(input.as_ref(), &mut t); } }