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
99
100
|
mod audio;
mod network;
mod command;
mod state;
use crate::audio::Audio;
use crate::state::Server;
use argparse::ArgumentParser;
use argparse::Store;
use argparse::StoreTrue;
use colored::*;
use cpal::traits::StreamTrait;
use futures::channel::oneshot;
use futures::join;
use log::*;
use mumble_protocol::crypt::ClientCryptState;
use std::net::ToSocketAddrs;
use std::sync::Arc;
use std::sync::Mutex;
#[tokio::main]
async fn main() {
// setup logger
fern::Dispatch::new()
.format(|out, message, record| {
out.finish(format_args!(
"{} {}:{} {}",
//TODO runtime flag that disables color
match record.level() {
Level::Error => "ERROR".red(),
Level::Warn => "WARN ".yellow(),
Level::Info => "INFO ".normal(),
Level::Debug => "DEBUG".green(),
Level::Trace => "TRACE".normal(),
},
record.file().unwrap(),
record.line().unwrap(),
message
))
})
.level(log::LevelFilter::Debug)
.chain(std::io::stderr())
.apply().unwrap();
// Handle command line arguments
let mut server_host = "".to_string();
let mut server_port = 64738u16;
let mut username = "EchoBot".to_string();
let mut accept_invalid_cert = false;
{
let mut ap = ArgumentParser::new();
ap.set_description("Run the echo client example");
ap.refer(&mut server_host)
.add_option(&["--host"], Store, "Hostname of mumble server")
.required();
ap.refer(&mut server_port)
.add_option(&["--port"], Store, "Port of mumble server");
ap.refer(&mut username)
.add_option(&["--username"], Store, "User name used to connect");
ap.refer(&mut accept_invalid_cert).add_option(
&["--accept-invalid-cert"],
StoreTrue,
"Accept invalid TLS certificates",
);
ap.parse_args_or_exit();
}
let server_addr = (server_host.as_ref(), server_port)
.to_socket_addrs()
.expect("Failed to parse server address")
.next()
.expect("Failed to resolve server address");
// Oneshot channel for setting UDP CryptState from control task
// For simplicity we don't deal with re-syncing, real applications would have to.
let (crypt_state_sender, crypt_state_receiver) = oneshot::channel::<ClientCryptState>();
let audio = Audio::new();
audio.output_stream.play().unwrap();
let audio = Arc::new(Mutex::new(audio));
let server_state = Arc::new(Mutex::new(Server::new()));
// Run it
join!(
network::handle_tcp(
server_state,
server_addr,
server_host,
username,
accept_invalid_cert,
crypt_state_sender,
Arc::clone(&audio),
),
network::handle_udp(
server_addr,
crypt_state_receiver,
audio,
),
);
}
|
