/
process_list.rs
92 lines (78 loc) · 2.38 KB
/
process_list.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/// A simple process list example using memflow
use clap::*;
use log::Level;
use memflow::prelude::v1::*;
fn main() -> Result<()> {
//let matches = parse_args();
//let chain = extract_args(&matches)?;
let mut registry = Registry::new();
// now use the same builder stuff?
let mut os = registry.instantiate_os("native", None, None)?;
// create inventory + os
//let inventory = Inventory::scan();
//let mut os = inventory.builder().os_chain(chain).build()?;
let process_list = os.process_info_list()?;
// Print process list, formatted
println!(
"{:>5} {:>10} {:>10} {:<}",
"PID", "SYS ARCH", "PROC ARCH", "NAME"
);
for p in process_list {
println!(
"{:>5} {:^10} {:^10} {} ({}) ({:?})",
p.pid, p.sys_arch, p.proc_arch, p.name, p.command_line, p.state
);
}
Ok(())
}
fn parse_args() -> ArgMatches {
Command::new("mfps example")
.version(crate_version!())
.author(crate_authors!())
.arg(Arg::new("verbose").short('v').action(ArgAction::Count))
.arg(
Arg::new("connector")
.long("connector")
.short('c')
.action(ArgAction::Append)
.required(false),
)
.arg(
Arg::new("os")
.long("os")
.short('o')
.action(ArgAction::Append)
.required(true),
)
.get_matches()
}
fn extract_args(matches: &ArgMatches) -> Result<OsChain<'_>> {
let log_level = match matches.get_count("verbose") {
0 => Level::Error,
1 => Level::Warn,
2 => Level::Info,
3 => Level::Debug,
4 => Level::Trace,
_ => Level::Trace,
};
simplelog::TermLogger::init(
log_level.to_level_filter(),
simplelog::Config::default(),
simplelog::TerminalMode::Stdout,
simplelog::ColorChoice::Auto,
)
.unwrap();
let conn_iter = matches
.indices_of("connector")
.zip(matches.get_many::<String>("connector"))
.map(|(a, b)| a.zip(b.map(String::as_str)))
.into_iter()
.flatten();
let os_iter = matches
.indices_of("os")
.zip(matches.get_many::<String>("os"))
.map(|(a, b)| a.zip(b.map(String::as_str)))
.into_iter()
.flatten();
OsChain::new(conn_iter, os_iter)
}