Commit cd64b504 authored by Alexander Alekhin's avatar Alexander Alekhin

imgcodecs: add overflow checks

imgcodecs: remove assert() usage

Origin commits:
- be524792
- 8a76fada
parent 443059e3
...@@ -42,6 +42,7 @@ ...@@ -42,6 +42,7 @@
#include "precomp.hpp" #include "precomp.hpp"
#include "bitstrm.hpp" #include "bitstrm.hpp"
#include "utils.hpp"
namespace cv namespace cv
{ {
...@@ -164,7 +165,7 @@ void RBaseStream::release() ...@@ -164,7 +165,7 @@ void RBaseStream::release()
void RBaseStream::setPos( int pos ) void RBaseStream::setPos( int pos )
{ {
assert( isOpened() && pos >= 0 ); CV_Assert(isOpened() && pos >= 0);
if( !m_file ) if( !m_file )
{ {
...@@ -181,14 +182,19 @@ void RBaseStream::setPos( int pos ) ...@@ -181,14 +182,19 @@ void RBaseStream::setPos( int pos )
int RBaseStream::getPos() int RBaseStream::getPos()
{ {
assert( isOpened() ); CV_Assert(isOpened());
return m_block_pos + (int)(m_current - m_start); int pos = validateToInt((m_current - m_start) + m_block_pos);
CV_Assert(pos >= m_block_pos); // overflow check
CV_Assert(pos >= 0); // overflow check
return pos;
} }
void RBaseStream::skip( int bytes ) void RBaseStream::skip( int bytes )
{ {
assert( bytes >= 0 ); CV_Assert(bytes >= 0);
uchar* old = m_current;
m_current += bytes; m_current += bytes;
CV_Assert(m_current >= old); // overflow check
} }
///////////////////////// RLByteStream //////////////////////////// ///////////////////////// RLByteStream ////////////////////////////
...@@ -220,7 +226,7 @@ int RLByteStream::getBytes( void* buffer, int count ) ...@@ -220,7 +226,7 @@ int RLByteStream::getBytes( void* buffer, int count )
{ {
uchar* data = (uchar*)buffer; uchar* data = (uchar*)buffer;
int readed = 0; int readed = 0;
assert( count >= 0 ); CV_Assert(count >= 0);
while( count > 0 ) while( count > 0 )
{ {
...@@ -371,7 +377,7 @@ void WBaseStream::writeBlock() ...@@ -371,7 +377,7 @@ void WBaseStream::writeBlock()
{ {
int size = (int)(m_current - m_start); int size = (int)(m_current - m_start);
assert( isOpened() ); CV_Assert(isOpened());
if( size == 0 ) if( size == 0 )
return; return;
...@@ -442,7 +448,7 @@ void WBaseStream::release() ...@@ -442,7 +448,7 @@ void WBaseStream::release()
int WBaseStream::getPos() int WBaseStream::getPos()
{ {
assert( isOpened() ); CV_Assert(isOpened());
return m_block_pos + (int)(m_current - m_start); return m_block_pos + (int)(m_current - m_start);
} }
...@@ -465,7 +471,7 @@ void WLByteStream::putBytes( const void* buffer, int count ) ...@@ -465,7 +471,7 @@ void WLByteStream::putBytes( const void* buffer, int count )
{ {
uchar* data = (uchar*)buffer; uchar* data = (uchar*)buffer;
assert( data && m_current && count >= 0 ); CV_Assert(data && m_current && count >= 0);
while( count ) while( count )
{ {
......
...@@ -92,6 +92,7 @@ bool BmpDecoder::readHeader() ...@@ -92,6 +92,7 @@ bool BmpDecoder::readHeader()
m_offset = m_strm.getDWord(); m_offset = m_strm.getDWord();
int size = m_strm.getDWord(); int size = m_strm.getDWord();
CV_Assert(size > 0); // overflow, 2Gb limit
if( size >= 36 ) if( size >= 36 )
{ {
......
...@@ -120,7 +120,7 @@ bool SunRasterDecoder::readHeader() ...@@ -120,7 +120,7 @@ bool SunRasterDecoder::readHeader()
m_type = IsColorPalette( m_palette, m_bpp ) ? CV_8UC3 : CV_8UC1; m_type = IsColorPalette( m_palette, m_bpp ) ? CV_8UC3 : CV_8UC1;
m_offset = m_strm.getPos(); m_offset = m_strm.getPos();
assert( m_offset == 32 + m_maplength ); CV_Assert(m_offset == 32 + m_maplength);
result = true; result = true;
} }
} }
...@@ -133,7 +133,7 @@ bool SunRasterDecoder::readHeader() ...@@ -133,7 +133,7 @@ bool SunRasterDecoder::readHeader()
m_offset = m_strm.getPos(); m_offset = m_strm.getPos();
assert( m_offset == 32 + m_maplength ); CV_Assert(m_offset == 32 + m_maplength);
result = true; result = true;
} }
} }
...@@ -226,7 +226,7 @@ bool SunRasterDecoder::readData( Mat& img ) ...@@ -226,7 +226,7 @@ bool SunRasterDecoder::readData( Mat& img )
code = m_strm.getByte(); code = m_strm.getByte();
if( len > line_end - tsrc ) if( len > line_end - tsrc )
{ {
assert(0); CV_Error(CV_StsInternal, "");
goto bad_decoding_1bpp; goto bad_decoding_1bpp;
} }
...@@ -367,7 +367,7 @@ bad_decoding_end: ...@@ -367,7 +367,7 @@ bad_decoding_end:
result = true; result = true;
break; break;
default: default:
assert(0); CV_Error(CV_StsInternal, "");
} }
} }
catch( ... ) catch( ... )
......
...@@ -54,7 +54,7 @@ ...@@ -54,7 +54,7 @@
#include <string.h> #include <string.h>
#include <limits.h> #include <limits.h>
#include <ctype.h> #include <ctype.h>
#include <assert.h> #include <assert.h> // FIX IT: remove this
#if defined WIN32 || defined WINCE #if defined WIN32 || defined WINCE
#if !defined _WIN32_WINNT #if !defined _WIN32_WINNT
......
...@@ -670,7 +670,7 @@ cvConvertImage( const CvArr* srcarr, CvArr* dstarr, int flags ) ...@@ -670,7 +670,7 @@ cvConvertImage( const CvArr* srcarr, CvArr* dstarr, int flags )
icvCvt_BGR2Gray_8u_C3C1R( s, s_step, d, d_step, size, swap_rb ); icvCvt_BGR2Gray_8u_C3C1R( s, s_step, d, d_step, size, swap_rb );
break; break;
case 33: case 33:
assert( swap_rb ); CV_Assert(swap_rb);
icvCvt_RGB2BGR_8u_C3R( s, s_step, d, d_step, size ); icvCvt_RGB2BGR_8u_C3R( s, s_step, d, d_step, size );
break; break;
case 41: case 41:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment