Prevent imread from illegal memory access (Bug #2602)

The change is based on pull request #211.
This commit is contained in:
Andrey Kamaev 2012-12-06 19:42:17 +04:00
parent 7852b68c20
commit 1821d21f5b

View File

@ -229,6 +229,9 @@ bool JpegDecoder::readHeader()
if( m_f ) if( m_f )
jpeg_stdio_src( &state->cinfo, m_f ); jpeg_stdio_src( &state->cinfo, m_f );
} }
if (state->cinfo.src != 0)
{
jpeg_read_header( &state->cinfo, TRUE ); jpeg_read_header( &state->cinfo, TRUE );
m_width = state->cinfo.image_width; m_width = state->cinfo.image_width;
@ -236,6 +239,7 @@ bool JpegDecoder::readHeader()
m_type = state->cinfo.num_components > 1 ? CV_8UC3 : CV_8UC1; m_type = state->cinfo.num_components > 1 ? CV_8UC3 : CV_8UC1;
result = true; result = true;
} }
}
if( !result ) if( !result )
close(); close();