chris/mumble-web-proxy
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Make port in --server argument optional

Browse source
This commit is contained in:
Jonas Herzig 2019-02-05 13:48:14 +01:00
parent 8950e30690
commit f9e85618c8
1 changed files with 14 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
src/main.rs
View file

@ -66,7 +66,7 @@ fn main() {
.add_option( .add_option(
&["--server"], &["--server"],
Store, Store,
"Hostname and port of upstream mumble server", "Hostname and (optionally) port of the upstream Mumble server",
) )
.required(); .required();
ap.refer(&mut accept_invalid_certs).add_option( ap.refer(&mut accept_invalid_certs).add_option(
@ -78,13 +78,19 @@ fn main() {
ap.parse_args_or_exit(); ap.parse_args_or_exit();
} }
let mut upstream_parts = upstream.rsplitn(2, ':'); // Try parsing as raw IPv6 address first
let upstream_port: u16 = upstream_parts let (upstream_host, upstream_port) = match upstream.parse::<Ipv6Addr>() {
.next() Ok(_) => (upstream.as_ref(), 64738),
.expect("Missing upstream port") Err(_) => {
.parse() // Otherwise split off port from end
.expect("Failed to parse upstream port"); let mut upstream_parts = upstream.rsplitn(2, ':');
let upstream_host = upstream_parts.next().expect("Missing upstream host name"); let right = upstream_parts.next().expect("Empty upstream address");
match upstream_parts.next() {
Some(host) => (host, right.parse().expect("Failed to parse upstream port")),
None => (right, 64738),
}
}
};
let upstream_host = Box::leak(Box::new(upstream_host.to_owned())).as_str(); let upstream_host = Box::leak(Box::new(upstream_host.to_owned())).as_str();
let upstream_addr = (upstream_host, upstream_port) let upstream_addr = (upstream_host, upstream_port)
.to_socket_addrs() .to_socket_addrs()