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

♻️ refactor

main
_ 2021-07-18 18:33:19 +00:00
parent b64042043b
commit 4820722ec2
2 changed files with 36 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
prototypes/quic_demo/src/bin/quic_demo_end_server.rs
View File

@ -27,31 +27,12 @@ async fn main () -> anyhow::Result <()> {
trace! ("Connecting to relay server"); trace! ("Connecting to relay server");
let server_id = opt.server_id.unwrap_or (43);
let quinn::NewConnection { let quinn::NewConnection {
connection,
mut bi_streams, mut bi_streams,
.. ..
} = endpoint.connect (&relay_addr, "localhost")?.await?; } = protocol::p4_connect_to_p3 (&endpoint, &relay_addr, server_id).await?;
let (mut send, mut recv) = connection.open_bi ().await?;
let our_id = opt.server_id.unwrap_or (43);
let req_buf = [
Command::CONNECT_P4_TO_P3.0,
our_id,
0,
0,
];
send.write_all (&req_buf).await?;
let mut resp_buf = [0, 0, 0, 0];
recv.read_exact (&mut resp_buf).await?;
assert_eq! (resp_buf, [
Command::OKAY.0,
Command::CONNECT_P4_TO_P3.0,
0,
0,
]);
trace! ("Accepting bi streams from P3"); trace! ("Accepting bi streams from P3");

33
prototypes/quic_demo/src/protocol.rs
View File

@ -103,3 +103,36 @@ pub async fn p2_connect_to_p5 (
Ok ((relay_send, relay_recv)) Ok ((relay_send, relay_recv))
} }
pub async fn p4_connect_to_p3 (
endpoint: &quinn::Endpoint,
relay_addr: &std::net::SocketAddr,
server_id: u8,
) -> anyhow::Result <quinn::NewConnection>
{
let new_conn = endpoint.connect (relay_addr, "localhost")?.await?;
let (mut send, mut recv) = new_conn.connection.open_bi ().await?;
let req_buf = [
Command::CONNECT_P4_TO_P3.0,
server_id,
0,
0,
];
send.write_all (&req_buf).await?;
let mut resp_buf = [0, 0, 0, 0];
recv.read_exact (&mut resp_buf).await?;
let expected = [
Command::OKAY.0,
Command::CONNECT_P4_TO_P3.0,
0,
0,
];
if resp_buf != expected {
bail! ("P4 didn't get OK response when connecting to P3");
}
Ok (new_conn)
}