tuxmain/qoi
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix potential decoder panic when checking padding

Browse source
This commit is contained in:
Ivan Smirnov 2022-01-03 13:04:22 +03:00
parent e6e156f974
commit fb8a43fd13
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/decode.rs
View file

@ -1,7 +1,7 @@
use std::io::Read; use std::io::Read;
// TODO: can be removed once https://github.com/rust-lang/rust/issues/74985 is stable // TODO: can be removed once https://github.com/rust-lang/rust/issues/74985 is stable
use bytemuck::{cast_slice, cast_slice_mut, Pod}; use bytemuck::{cast_slice_mut, Pod};
use crate::consts::{ use crate::consts::{
QOI_HEADER_SIZE, QOI_OP_DIFF, QOI_OP_INDEX, QOI_OP_LUMA, QOI_OP_RGB, QOI_OP_RGBA, QOI_OP_RUN, QOI_HEADER_SIZE, QOI_OP_DIFF, QOI_OP_INDEX, QOI_OP_LUMA, QOI_OP_RGB, QOI_OP_RGBA, QOI_OP_RUN,
@ -81,7 +81,7 @@ where
if unlikely(data.len() < QOI_PADDING_SIZE) { if unlikely(data.len() < QOI_PADDING_SIZE) {
return Err(Error::UnexpectedBufferEnd); return Err(Error::UnexpectedBufferEnd);
} else if unlikely(cast_slice::<_, [u8; QOI_PADDING_SIZE]>(data)[0] != QOI_PADDING) { } else if unlikely(&data[..QOI_PADDING_SIZE] != &QOI_PADDING) {
return Err(Error::InvalidPadding); return Err(Error::InvalidPadding);
} }