Merge pull request #7451 from reunanen:issue-7409

Fix findContours crash for very large images (#7451)

* Cast step to size_t in order to avoid integer overflow when processing very large images

* Change assert to CV_Assert
pull/7468/head
Juha Reunanen 8 years ago committed by Alexander Alekhin
parent 016359f406
commit 0f387cda8f
  1. 17
      modules/imgproc/src/contours.cpp

@ -306,10 +306,13 @@ cvStartFindContours( void* _img, CvMemStorage* storage,
scanner->cinfo_storage );
}
CV_Assert(step >= 0);
CV_Assert(size.height >= 1);
/* make zero borders */
int esz = CV_ELEM_SIZE(mat->type);
memset( img, 0, size.width*esz );
memset( img + step * (size.height - 1), 0, size.width*esz );
memset( img + static_cast<size_t>(step) * (size.height - 1), 0, size.width*esz );
img += step;
for( int y = 1; y < size.height - 1; y++, img += step )
@ -1005,6 +1008,8 @@ cvFindNextContour( CvContourScanner scanner )
bool haveSIMD = cv::checkHardwareSupport(CPU_SSE2);
#endif
CV_Assert(scanner->img_step >= 0);
icvEndProcessContour( scanner );
/* initialize local state */
@ -1126,7 +1131,7 @@ cvFindNextContour( CvContourScanner scanner )
is_hole = 1;
}
if( mode == 0 && (is_hole || img0[lnbd.y * step + lnbd.x] > 0) )
if( mode == 0 && (is_hole || img0[lnbd.y * static_cast<size_t>(step) + lnbd.x] > 0) )
goto resume_scan;
origin.y = y;
@ -1140,8 +1145,8 @@ cvFindNextContour( CvContourScanner scanner )
else
{
int lval = (img0_i ?
img0_i[lnbd.y * step_i + lnbd.x] :
(int)img0[lnbd.y * step + lnbd.x]) & 0x7f;
img0_i[lnbd.y * static_cast<size_t>(step_i) + lnbd.x] :
(int)img0[lnbd.y * static_cast<size_t>(step) + lnbd.x]) & 0x7f;
_CvContourInfo *cur = scanner->cinfo_table[lval];
/* find the first bounding contour */
@ -1153,11 +1158,11 @@ cvFindNextContour( CvContourScanner scanner )
if( par_info )
{
if( (img0_i &&
icvTraceContour_32s( img0_i + par_info->origin.y * step_i +
icvTraceContour_32s( img0_i + par_info->origin.y * static_cast<size_t>(step_i) +
par_info->origin.x, step_i, img_i + lnbd.x,
par_info->is_hole ) > 0) ||
(!img0_i &&
icvTraceContour( img0 + par_info->origin.y * step +
icvTraceContour( img0 + par_info->origin.y * static_cast<size_t>(step) +
par_info->origin.x, step, img + lnbd.x,
par_info->is_hole ) > 0) )
break;

Loading…
Cancel
Save