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
|
use crate::{new_rpc_client, Command, Result};
use clap::value_t_or_exit;
pub struct SplitTunnel;
impl Command for SplitTunnel {
fn name(&self) -> &'static str {
"split-tunnel"
}
fn clap_subcommand(&self) -> clap::App<'static, 'static> {
clap::SubCommand::with_name(self.name())
.about("Manage split tunneling")
.setting(clap::AppSettings::SubcommandRequiredElseHelp)
.subcommand(create_pid_subcommand())
}
fn run(&self, matches: &clap::ArgMatches<'_>) -> Result<()> {
match matches.subcommand() {
("pid", Some(pid_matches)) => Self::handle_pid_cmd(pid_matches),
_ => unreachable!("unhandled comand"),
}
}
}
fn create_pid_subcommand() -> clap::App<'static, 'static> {
clap::SubCommand::with_name("pid")
.about("Manage processes to exclude from the tunnel")
.setting(clap::AppSettings::SubcommandRequiredElseHelp)
.subcommand(
clap::SubCommand::with_name("add").arg(clap::Arg::with_name("pid").required(true)),
)
.subcommand(
clap::SubCommand::with_name("delete").arg(clap::Arg::with_name("pid").required(true)),
)
.subcommand(clap::SubCommand::with_name("clear"))
.subcommand(clap::SubCommand::with_name("list"))
}
impl SplitTunnel {
fn handle_pid_cmd(matches: &clap::ArgMatches<'_>) -> Result<()> {
match matches.subcommand() {
("add", Some(matches)) => {
let pid = value_t_or_exit!(matches.value_of("pid"), i32);
new_rpc_client()?.add_split_tunnel_process(pid)?;
Ok(())
}
("delete", Some(matches)) => {
let pid = value_t_or_exit!(matches.value_of("pid"), i32);
new_rpc_client()?.remove_split_tunnel_process(pid)?;
Ok(())
}
("clear", Some(_)) => {
new_rpc_client()?.clear_split_tunnel_processes()?;
Ok(())
}
("list", Some(_)) => {
let pids = new_rpc_client()?.get_split_tunnel_processes()?;
println!("Excluded PIDs:");
for pid in pids.iter() {
println!(" {}", pid);
}
Ok(())
}
_ => unreachable!("unhandled command"),
}
}
}
|
