Panic on integer overflow in Chain::remaining (#482)

Make it safer.
2026-08-07 00:00:13 +02:00 · 2021-02-16 12:44:33 -08:00
parent 268f6f80b4
commit 2428c152a6
1 changed files with 5 additions and 2 deletions
@@ -135,7 +135,7 @@ where
    U: Buf,
 {
    fn remaining(&self) -> usize {
-        self.a.remaining() + self.b.remaining()
+        self.a.remaining().checked_add(self.b.remaining()).unwrap()
    }

    fn chunk(&self) -> &[u8] {
@@ -178,7 +178,10 @@ where
    U: BufMut,
 {
    fn remaining_mut(&self) -> usize {
-        self.a.remaining_mut() + self.b.remaining_mut()
+        self.a
+            .remaining_mut()
+            .checked_add(self.b.remaining_mut())
+            .unwrap()
    }

    fn chunk_mut(&mut self) -> &mut UninitSlice {