fixed potential buffer size overflow in Boost::update_weight, bug #1524 (thanks to benlemna for the patch)

13 years ago · 8bb422b1fc
parent f341ce2665
commit 8bb422b1fc
1 changed files with 1 additions and 1 deletions
--- a/modules/ml/src/boost.cpp
+++ b/modules/ml/src/boost.cpp
@ -1126,7 +1126,7 @@ CvBoost::update_weights( CvBoostTree* tree )
    int *sample_idx_buf;
    const int* sample_idx = 0;
    cv::AutoBuffer<uchar> inn_buf;
-    int _buf_size = (params.boost_type == LOGIT) || (params.boost_type == GENTLE) ? data->sample_count*sizeof(int) : 0;
+    size_t _buf_size = (params.boost_type == LOGIT) || (params.boost_type == GENTLE) ? data->sample_count*sizeof(int) : 0;
    if( !tree )
        _buf_size += n*sizeof(int);
    else