blob: b06b7df3eefbaa19c56ccf7daeab1b92e6299d72 (
plain)
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
93
94
95
96
97
98
|
use clap;
use {Command, Result};
use mullvad_ipc_client::DaemonRpcClient;
use talpid_types::net::{OpenVpnTunnelOptions, TunnelOptions};
pub struct Tunnel;
impl Command for Tunnel {
fn name(&self) -> &'static str {
"tunnel"
}
fn clap_subcommand(&self) -> clap::App<'static, 'static> {
clap::SubCommand::with_name(self.name())
.about("Manage tunnel specific options")
.setting(clap::AppSettings::SubcommandRequired)
.subcommand(
clap::SubCommand::with_name("openvpn")
.about("Manage options for OpenVPN tunnels")
.setting(clap::AppSettings::SubcommandRequired)
.subcommand(
clap::SubCommand::with_name("set")
.subcommand(
clap::SubCommand::with_name("mssfix").arg(
clap::Arg::with_name("mssfix")
.help(
"Sets the optional mssfix parameter. \
Set an empty string to clear it.",
)
.required(true),
),
)
.setting(clap::AppSettings::SubcommandRequired),
)
.subcommand(
clap::SubCommand::with_name("get")
.help("Retrieves the current setting for mssfix"),
),
)
}
fn run(&self, matches: &clap::ArgMatches) -> Result<()> {
if let Some(openvpn_matches) = matches.subcommand_matches("openvpn") {
Self::handle_openvpn_cmd(openvpn_matches)
} else {
unreachable!("No tunnel command given")
}
}
}
impl Tunnel {
fn handle_openvpn_cmd(matches: &clap::ArgMatches) -> Result<()> {
if let Some(set_matches) = matches.subcommand_matches("set") {
Self::set_openvpn_option(set_matches)
} else if let Some(_) = matches.subcommand_matches("get") {
let openvpn_options = Self::get_tunnel_options()?.openvpn;
Self::print_openvpn_tunnel_options(&openvpn_options);
Ok(())
} else {
unreachable!("Unrecognized subcommand");
}
}
fn set_openvpn_option(matches: &clap::ArgMatches) -> Result<()> {
if let Some(mssfix_args) = matches.subcommand_matches("mssfix") {
let mssfix_str = mssfix_args.value_of("mssfix").unwrap();
let mssfix: Option<u16> = if mssfix_str == "" {
None
} else {
Some(mssfix_str.parse()?)
};
let mut rpc = DaemonRpcClient::new()?;
rpc.set_openvpn_mssfix(mssfix)?;
println!("mssfix parameter updated");
Ok(())
} else {
unreachable!("Invalid option passed to 'openvpn set'");
}
}
fn get_tunnel_options() -> Result<TunnelOptions> {
let mut rpc = DaemonRpcClient::new()?;
Ok(rpc.get_tunnel_options()?)
}
fn print_openvpn_tunnel_options(options: &OpenVpnTunnelOptions) {
println!("OpenVPN tunnel options");
println!(
"\tmssfix: {}",
options
.mssfix
.map(|v| v.to_string())
.unwrap_or("UNSET".to_string())
);
}
}
|
