tokio-rs/bytes
1
0
Fork 0
mirror of https://github.com/tokio-rs/bytes.git synced 2026-08-25 00:00:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Make it safer.
This commit is contained in:
Stepan Koltsov
2021-02-16 12:44:33 -08:00
committed by GitHub
parent 268f6f80b4
commit 2428c152a6
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/buf/chain.rs
+5 -2
View File
@@ -135,7 +135,7 @@ where
U: Buf, U: Buf,
{ {
fn remaining(&self) -> usize { fn remaining(&self) -> usize {
self.a.remaining() + self.b.remaining() self.a.remaining().checked_add(self.b.remaining()).unwrap()
} }
fn chunk(&self) -> &[u8] { fn chunk(&self) -> &[u8] {
@@ -178,7 +178,10 @@ where
U: BufMut, U: BufMut,
{ {
fn remaining_mut(&self) -> usize { 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 { fn chunk_mut(&mut self) -> &mut UninitSlice {