waalge
/
aiken
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed bug if byte array exceed 255 bytes

This commit is contained in:
zypeh 2022-08-25 21:13:29 +08:00 committed by Lucas
parent 10461779af
commit ef81427b62
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
crates/flat/src/encode/encoder.rs
View File

@ -80,7 +80,7 @@ impl Encoder {
return Err(Error::BufferNotByteAligned); return Err(Error::BufferNotByteAligned);
} }
self.write_blk(arr, &mut 0); self.write_blk(arr);
Ok(self) Ok(self)
} }
@ -277,20 +277,13 @@ impl Encoder {
/// Following that it writes the next 255 bytes from the array. /// Following that it writes the next 255 bytes from the array.
/// After reaching the end of the buffer we write a 0 byte. Only write 0 if the byte array is empty. /// After reaching the end of the buffer we write a 0 byte. Only write 0 if the byte array is empty.
/// This is byte alignment agnostic. /// This is byte alignment agnostic.
fn write_blk(&mut self, arr: &[u8], src_ptr: &mut usize) { fn write_blk(&mut self, arr: &[u8]) {
loop { let chunks = arr.chunks(255);
let src_len = arr.len() - *src_ptr;
let blk_len = src_len.min(255);
self.buffer.push(blk_len as u8); for chunk in chunks {
self.buffer.push(chunk.len() as u8);
if blk_len == 0 { self.buffer.extend(chunk);
return;
}
self.buffer.extend(&arr[*src_ptr..blk_len]);
*src_ptr += blk_len;
} }
self.buffer.push(0);
} }
} }