archive/tuic
1
0
Fork 0
Code

stop client from receiving heartbeat

Browse Source
This commit is contained in:
EAimTY 2023-01-27 01:33:46 +09:00
parent 823ae55024
commit 437776abb0
1 changed files with 45 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
tuic-quinn/src/lib.rs
View File

@ -78,27 +78,6 @@ impl<'conn, Side> Connection<'conn, Side> {
Ok(()) Ok(())
} }
pub async fn accept_datagram(&self, dg: Bytes) -> Result<Task, Error> {
let mut dg = Cursor::new(dg);
match Header::unmarshal(&mut dg).await? {
Header::Authenticate(_) => Err(Error::BadCommand("authenticate")),
Header::Connect(_) => Err(Error::BadCommand("connect")),
Header::Packet(pkt) => {
let model = self.model.recv_packet(pkt);
let pos = dg.position() as usize;
let buf = dg.into_inner().slice(pos..pos + *model.size() as usize);
Ok(Task::Packet(self.accept_packet_native(model, buf).await?))
}
Header::Dissociate(_) => Err(Error::BadCommand("dissociate")),
Header::Heartbeat(hb) => {
let _ = self.model.recv_heartbeat(hb);
Ok(Task::Heartbeat)
}
_ => unreachable!(),
}
}
pub fn collect_garbage(&self, timeout: Duration) { pub fn collect_garbage(&self, timeout: Duration) {
self.model.collect_garbage(timeout); self.model.collect_garbage(timeout);
} }
@ -156,6 +135,14 @@ impl<'conn> Connection<'conn, side::Client> {
Ok(Connect::new(Side::Client(model), send, recv)) Ok(Connect::new(Side::Client(model), send, recv))
} }
pub async fn heartbeat(&self) -> Result<(), Error> {
let model = self.model.send_heartbeat();
let mut buf = Vec::with_capacity(model.header().len());
model.header().marshal(&mut buf).await.unwrap();
self.conn.send_datagram(Bytes::from(buf))?;
Ok(())
}
pub async fn accept_uni_stream(&self, mut recv: RecvStream) -> Result<Task, Error> { pub async fn accept_uni_stream(&self, mut recv: RecvStream) -> Result<Task, Error> {
match Header::unmarshal(&mut recv).await? { match Header::unmarshal(&mut recv).await? {
Header::Authenticate(_) => Err(Error::BadCommand("authenticate")), Header::Authenticate(_) => Err(Error::BadCommand("authenticate")),
@ -187,12 +174,22 @@ impl<'conn> Connection<'conn, side::Client> {
} }
} }
pub async fn heartbeat(&self) -> Result<(), Error> { pub async fn accept_datagram(&self, dg: Bytes) -> Result<Task, Error> {
let model = self.model.send_heartbeat(); let mut dg = Cursor::new(dg);
let mut buf = Vec::with_capacity(model.header().len());
model.header().marshal(&mut buf).await.unwrap(); match Header::unmarshal(&mut dg).await? {
self.conn.send_datagram(Bytes::from(buf))?; Header::Authenticate(_) => Err(Error::BadCommand("authenticate")),
Ok(()) Header::Connect(_) => Err(Error::BadCommand("connect")),
Header::Packet(pkt) => {
let model = self.model.recv_packet(pkt);
let pos = dg.position() as usize;
let buf = dg.into_inner().slice(pos..pos + *model.size() as usize);
Ok(Task::Packet(self.accept_packet_native(model, buf).await?))
}
Header::Dissociate(_) => Err(Error::BadCommand("dissociate")),
Header::Heartbeat(_) => Err(Error::BadCommand("heartbeat")),
_ => unreachable!(),
}
} }
} }
@ -244,6 +241,27 @@ impl<'conn> Connection<'conn, side::Server> {
_ => unreachable!(), _ => unreachable!(),
} }
} }
pub async fn accept_datagram(&self, dg: Bytes) -> Result<Task, Error> {
let mut dg = Cursor::new(dg);
match Header::unmarshal(&mut dg).await? {
Header::Authenticate(_) => Err(Error::BadCommand("authenticate")),
Header::Connect(_) => Err(Error::BadCommand("connect")),
Header::Packet(pkt) => {
let model = self.model.recv_packet(pkt);
let pos = dg.position() as usize;
let buf = dg.into_inner().slice(pos..pos + *model.size() as usize);
Ok(Task::Packet(self.accept_packet_native(model, buf).await?))
}
Header::Dissociate(_) => Err(Error::BadCommand("dissociate")),
Header::Heartbeat(hb) => {
let _ = self.model.recv_heartbeat(hb);
Ok(Task::Heartbeat)
}
_ => unreachable!(),
}
}
} }
pub struct Connect { pub struct Connect {