archive/tuic
1
0
Fork 0
Code

fix header marshaling

Browse Source
This commit is contained in:
EAimTY 2023-03-04 19:42:53 +09:00
parent 540618769c
commit 2c2f5623a7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tuic/src/marshal.rs
View File

@ -1,5 +1,5 @@
use crate::{Address, Authenticate, Connect, Dissociate, Header, Heartbeat, Packet, VERSION}; use crate::{Address, Authenticate, Connect, Dissociate, Header, Heartbeat, Packet, VERSION};
use bytes::BufMut; use bytes::{BufMut, BytesMut};
use futures_util::{AsyncWrite, AsyncWriteExt}; use futures_util::{AsyncWrite, AsyncWriteExt};
use std::{ use std::{
io::{Error as IoError, Write}, io::{Error as IoError, Write},
@ -10,7 +10,7 @@ impl Header {
/// Marshals the header into an `AsyncWrite` stream /// Marshals the header into an `AsyncWrite` stream
#[cfg(feature = "async_marshal")] #[cfg(feature = "async_marshal")]
pub async fn async_marshal(&self, s: &mut (impl AsyncWrite + Unpin)) -> Result<(), IoError> { pub async fn async_marshal(&self, s: &mut (impl AsyncWrite + Unpin)) -> Result<(), IoError> {
let mut buf = vec![0; self.len()]; let mut buf = BytesMut::with_capacity(self.len());
self.write(&mut buf); self.write(&mut buf);
s.write_all(&buf).await s.write_all(&buf).await
} }
@ -18,7 +18,7 @@ impl Header {
/// Marshals the header into a `Write` stream /// Marshals the header into a `Write` stream
#[cfg(feature = "marshal")] #[cfg(feature = "marshal")]
pub fn marshal(&self, s: &mut impl Write) -> Result<(), IoError> { pub fn marshal(&self, s: &mut impl Write) -> Result<(), IoError> {
let mut buf = vec![0; self.len()]; let mut buf = BytesMut::with_capacity(self.len());
self.write(&mut buf); self.write(&mut buf);
s.write_all(&buf) s.write_all(&buf)
} }