wisp net

wisp/src/extensions/cert.rs

@@ -191,6 +191,7 @@ impl ProtocolExtension for CertAuthProtocolExtension {
 
 	async fn handle_packet(
 		&mut self,
+		_: u8,
 		_: Bytes,
 		_: &mut DynWebSocketRead,
 		_: &dyn LockingWebSocketWrite,

wisp/src/extensions/mod.rs

@@ -112,6 +112,7 @@ pub trait ProtocolExtension: std::fmt::Debug + Sync + Send + 'static {
 	/// Handle receiving a packet.
 	async fn handle_packet(
 		&mut self,
+		packet_type: u8,
 		packet: Bytes,
 		read: &mut DynWebSocketRead,
 		write: &dyn LockingWebSocketWrite,

wisp/src/extensions/motd.rs

@@ -56,6 +56,7 @@ impl ProtocolExtension for MotdProtocolExtension {
 
 	async fn handle_packet(
 		&mut self,
+		_: u8,
 		_: Bytes,
 		_: &mut DynWebSocketRead,
 		_: &dyn LockingWebSocketWrite,

wisp/src/extensions/password.rs

@@ -102,6 +102,7 @@ impl ProtocolExtension for PasswordProtocolExtension {
 
 	async fn handle_packet(
 		&mut self,
+		_: u8,
 		_: Bytes,
 		_: &mut DynWebSocketRead,
 		_: &dyn LockingWebSocketWrite,

wisp/src/extensions/udp.rs

@@ -48,6 +48,7 @@ impl ProtocolExtension for UdpProtocolExtension {
 
 	async fn handle_packet(
 		&mut self,
+		_: u8,
 		_: Bytes,
 		_: &mut DynWebSocketRead,
 		_: &dyn LockingWebSocketWrite,

wisp/src/mux/inner.rs

@@ -189,7 +189,7 @@ impl<R: WebSocketRead + 'static, W: WebSocketWrite + 'static> MuxInner<R, W> {
 		let (ch_tx, ch_rx) = mpsc::bounded(if self.role == Role::Server {
 			self.buffer_size as usize
 		} else {
-			usize::MAX
+			usize::MAX - 8
 		});
 
 		let should_flow_control = self.tcp_extensions.contains(&stream_type.into());

wisp/src/packet.rs

@@ -572,6 +572,7 @@ impl<'a> Packet<'a> {
 				{
 					extension
 						.handle_packet(
+							packet_type,
 							BytesMut::from(bytes).freeze(),
 							DynWebSocketRead::from_mut(read),
 							write,