reactor
/
ptth
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

🚧 working on a few things

main
_ 2021-07-19 02:11:13 +00:00
parent 27106de4e4
commit 97ef6d7fce
1 changed files with 78 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
prototypes/quic_demo/src/bin/client_gui.rs
View File

@ -45,7 +45,9 @@ fn main () -> anyhow::Result <()> {
let mut x = margin;
let mut y = margin;
let frame_status;
let mut frame_status;
let mut input_client_port;
let mut input_server_id;
let mut input_server_port;
@ -53,7 +55,7 @@ fn main () -> anyhow::Result <()> {
let mut but_close;
{
frame_status = Frame::new (10, 10, 800 - 20, 30, "0 ports forwarded");
frame_status = Frame::new (10, 10, 800 - 20, 30, "Forwarding 0 ports");
}
y += h + margin;
@ -110,6 +112,14 @@ fn main () -> anyhow::Result <()> {
set_active (&mut but_close, false);
}
let mut gui_port = GuiPort {
input_client_port,
input_server_id,
input_server_port,
but_open,
but_close,
};
// y += h + margin;
wind.end ();
@ -137,31 +147,25 @@ fn main () -> anyhow::Result <()> {
while app.wait () {
match fltk_rx.recv () {
Some (Message::OpenPort) => {
let connection_p2_p3 = connection_p2_p3.clone ();
let client_tcp_port = u16::from_str (&input_client_port.value ())?;
let server_id = input_server_id.value ();
let server_tcp_port = u16::from_str (&input_server_port.value ())?;
let (shutdown_flag, shutdown_flag_rx) = tokio::sync::watch::channel (true);
if let Ok (params) = gui_port.get_params () {
let connection_p2_p3 = connection_p2_p3.clone ();
let (shutdown_flag, shutdown_flag_rx) = tokio::sync::watch::channel (true);
let task = rt.spawn (async move {
forward_port (
let task = rt.spawn (forward_port (
connection_p2_p3,
client_tcp_port,
server_id,
server_tcp_port,
params,
shutdown_flag_rx
).await
});
));
forwarding_instance.replace (ForwardingInstance {
task,
shutdown_flag,
});
forwarding_instance.replace (ForwardingInstance {
task,
shutdown_flag,
});
set_active (&mut but_open, false);
set_active (&mut but_close, true);
but_open.set (true);
but_close.set (false);
gui_port.set_forwarding (true);
frame_status.set_label ("Forwarding 1 port");
}
},
Some (Message::ClosePort) => {
if let Some (mut old_instance) = forwarding_instance.take () {
@ -172,10 +176,8 @@ fn main () -> anyhow::Result <()> {
})?;
}
set_active (&mut but_open, true);
set_active (&mut but_close, false);
but_open.set (false);
but_close.set (true);
gui_port.set_forwarding (false);
frame_status.set_label ("Forwarding 0 ports");
},
None => (),
}
@ -200,12 +202,16 @@ struct ForwardingInstance {
async fn forward_port (
connection_p2_p3: quinn::Connection,
client_tcp_port: u16,
server_id: String,
server_tcp_port: u16,
params: ForwardingParams,
shutdown_flag_rx: tokio::sync::watch::Receiver <bool>,
) -> anyhow::Result <()>
{
let ForwardingParams {
client_tcp_port,
server_id,
server_tcp_port,
} = params;
let listener = TcpListener::bind (("127.0.0.1", client_tcp_port)).await?;
trace! ("Accepting local TCP connections from P1 on {}", client_tcp_port);
@ -307,3 +313,43 @@ async fn handle_p1 (
Ok (())
}
struct GuiPort {
input_client_port: Input,
input_server_id: Input,
input_server_port: Input,
but_open: Button,
but_close: Button,
}
struct ForwardingParams {
client_tcp_port: u16,
server_id: String,
server_tcp_port: u16,
}
impl GuiPort {
fn get_params (&self) -> anyhow::Result <ForwardingParams>
{
let client_tcp_port = u16::from_str (&self.input_client_port.value ())?;
let server_id = self.input_server_id.value ();
let server_tcp_port = u16::from_str (&self.input_server_port.value ())?;
Ok (ForwardingParams {
client_tcp_port,
server_id,
server_tcp_port,
})
}
fn set_forwarding (&mut self, x: bool) {
set_active (&mut self.input_client_port, !x);
set_active (&mut self.input_server_id, !x);
set_active (&mut self.input_server_port, !x);
set_active (&mut self.but_open, !x);
set_active (&mut self.but_close, x);
self.but_open.set (x);
self.but_close.set (!x);
}
}