diff --git a/projects/rocjpeg/src/rocjpeg_parser.cpp b/projects/rocjpeg/src/rocjpeg_parser.cpp index 3280edcd0a..e79a589a24 100644 --- a/projects/rocjpeg/src/rocjpeg_parser.cpp +++ b/projects/rocjpeg/src/rocjpeg_parser.cpp @@ -330,7 +330,7 @@ bool RocJpegStreamParser::ParseSOS() { uint32_t num_components = stream_[2]; - if (num_components > NUM_COMPONENTS - 1) { + if (num_components > NUM_COMPONENTS - 1 || num_components != jpeg_stream_parameters_.picture_parameter_buffer.num_components) { ERR("invalid number of component!") return false; }