Updated types in YamuxFrameInner from i32 to u32

Author: 0xMimir

p2p/src/network/yamux/p2p_network_yamux_reducer.rs

@@ -89,7 +89,7 @@ impl P2pNetworkYamuxState {
                                 }
                             }
                             1 => {
-                                let difference = i32::from_be_bytes(b);
+                                let difference = u32::from_be_bytes(b);
                                 let frame = YamuxFrame {
                                     flags,
                                     stream_id,
@@ -100,7 +100,7 @@ impl P2pNetworkYamuxState {
                                 continue;
                             }
                             2 => {
-                                let opaque = i32::from_be_bytes(b);
+                                let opaque = u32::from_be_bytes(b);
                                 let frame = YamuxFrame {
                                     flags,
                                     stream_id,
@@ -348,16 +348,9 @@ impl P2pNetworkYamuxState {
                             });
                         }
                     }
-                    YamuxFrameInner::WindowUpdate { difference } => {
-                        if *difference < 0 {
-                            let error =
-                                P2pNetworkConnectionError::YamuxBadWindowUpdate(frame.stream_id);
-                            dispatcher.push(P2pNetworkSchedulerAction::Error { addr, error });
-                        } else {
-                            while let Some(frame) = pending_outgoing.pop_front() {
-                                dispatcher
-                                    .push(P2pNetworkYamuxAction::OutgoingFrame { addr, frame });
-                            }
+                    YamuxFrameInner::WindowUpdate { .. } => {
+                        while let Some(frame) = pending_outgoing.pop_front() {
+                            dispatcher.push(P2pNetworkYamuxAction::OutgoingFrame { addr, frame });
                         }
                     }
                     _ => {}
@@ -481,22 +474,13 @@ impl P2pNetworkYamuxState {
 }
 
 impl YamuxStreamState {
-    pub fn update_window(&mut self, ours: bool, difference: i32) {
+    pub fn update_window(&mut self, ours: bool, difference: u32) {
         let window = if ours {
             &mut self.window_ours
         } else {
             &mut self.window_theirs
         };
-        if difference < 0 {
-            let decreasing = (-difference) as u32;
-            if *window < decreasing {
-                *window = 0;
-            } else {
-                *window = (*window).wrapping_sub(decreasing);
-            }
-        } else {
-            let increasing = difference as u32;
-            *window = (*window).wrapping_add(increasing);
-        }
+
+        *window = (*window).wrapping_add(difference);
     }
 }

p2p/src/network/yamux/p2p_network_yamux_state.rs

@@ -95,7 +95,7 @@ bitflags::bitflags! {
 #[derive(Serialize, Deserialize, Debug, Clone, Copy)]
 pub struct YamuxPing {
     pub stream_id: StreamId,
-    pub opaque: i32,
+    pub opaque: u32,
     pub response: bool,
 }
 
@@ -227,8 +227,8 @@ impl YamuxFrame {
 #[derive(Serialize, Deserialize, Debug, Clone)]
 pub enum YamuxFrameInner {
     Data(Data),
-    WindowUpdate { difference: i32 },
-    Ping { opaque: i32 },
+    WindowUpdate { difference: u32 },
+    Ping { opaque: u32 },
     GoAway(Result<(), YamuxSessionError>),
 }