archive/tuic
1
0
Fork 0
Code

fixes #167

Browse Source
This commit is contained in:
EAimTY 2023-05-27 18:43:52 +09:00
parent 62193b2596
commit 27c378e5b6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,7 +1,7 @@
#![doc = include_str!("../README.md")] #![doc = include_str!("../README.md")]
use self::side::Side; use self::side::Side;
use bytes::{BufMut, Bytes}; use bytes::{BufMut, Bytes, BytesMut};
use futures_util::{AsyncRead, AsyncReadExt, AsyncWrite, AsyncWriteExt}; use futures_util::{AsyncRead, AsyncReadExt, AsyncWrite, AsyncWriteExt};
use quinn::{ use quinn::{
Connection as QuinnConnection, ConnectionError, RecvStream, SendDatagramError, SendStream, Connection as QuinnConnection, ConnectionError, RecvStream, SendDatagramError, SendStream,
@ -67,7 +67,7 @@ impl<Side> Connection<Side> {
let model = self.model.send_packet(assoc_id, addr, max_pkt_size); let model = self.model.send_packet(assoc_id, addr, max_pkt_size);
for (header, frag) in model.into_fragments(pkt) { for (header, frag) in model.into_fragments(pkt) {
let mut buf = vec![0; header.len() + frag.len()]; let mut buf = BytesMut::with_capacity(header.len() + frag.len());
header.write(&mut buf); header.write(&mut buf);
buf.put_slice(frag); buf.put_slice(frag);
self.conn.send_datagram(Bytes::from(buf))?; self.conn.send_datagram(Bytes::from(buf))?;