Fix int max (#17)

* made length checker as mandatory

* Add len checker

* Rework len checker to count checker

Author: VinnitskiV

src_c/IMB_mem_manager.c

@@ -645,25 +645,10 @@ In/out variables:
     /* the displ is declared as int by MPI1 standard
        If c_info->num_procs*init_size  exceed INT_MAX value there is no way to run this sample
        */
-    if (
-#ifdef MPI1
-        !strcmp(Bmark->name, "Alltoallv") ||
-        !strcmp(Bmark->name, "Allgatherv") ||
-        !strcmp(Bmark->name, "Scatterv") ||
-        !strcmp(Bmark->name, "Gatherv")
-#elif defined NBC // MPI1
-        !strcmp(Bmark->name, "Ialltoallv") || !strcmp(Bmark->name, "Ialltoallv_pure") ||
-        !strcmp(Bmark->name, "Iallgatherv") || !strcmp(Bmark->name, "Iallgatherv_pure") ||
-        !strcmp(Bmark->name, "Iscatterv") || !strcmp(Bmark->name, "Iscatterv_pure") ||
-        !strcmp(Bmark->name, "Igatherv") || !strcmp(Bmark->name, "Igatherv_pure")
-#else // NBC // MPI1
-        0
-#endif // NBC // MPI1
-        ) {
-        if (s_len > INT_MAX || r_len > INT_MAX) {
-            Bmark->sample_failure = SAMPLE_FAILED_INT_OVERFLOW;
-            return;
-        }
+
+    if (s_len > INT_MAX || r_len > INT_MAX) {
+        Bmark->sample_failure = SAMPLE_FAILED_INT_OVERFLOW;
+        return;
     }
     /*===============================================*/
 

src_cpp/helpers/helper_IMB_functions.h

@@ -51,6 +51,7 @@ goods and services.
 #pragma once
 
 #include <algorithm>
+#include <climits>
 
 #ifdef MPIIO
 static int do_nonblocking_;
@@ -484,7 +485,26 @@ struct Bmark_descr {
         if (!result) {
             throw std::logic_error("wrong recv or send buffer requirement description on a benchmark");
         }
-//        printf(">> s_len=%ld, r_len=%ld\n", s_len, r_len);
+        if (flags.count(REDUCTION)) {
+            int red_size_dt;
+            MPI_Type_size(c_info->red_data_type, &red_size_dt);
+
+            if (s_len / red_size_dt > INT_MAX || r_len / red_size_dt > INT_MAX) {
+                Bmark->sample_failure = SAMPLE_FAILED_INT_OVERFLOW;
+                return;
+            }
+        }
+        else {
+            int s_size_dt,
+                r_size_dt;
+            MPI_Type_size(c_info->s_data_type, &s_size_dt);
+            MPI_Type_size(c_info->r_data_type, &r_size_dt);
+
+            if (s_len / s_size_dt > INT_MAX || r_len / r_size_dt > INT_MAX) {
+                Bmark->sample_failure = SAMPLE_FAILED_INT_OVERFLOW;
+                return;
+            }
+        }
 //---------------------------------------------------------------------------------------------------
 // --- STEP 3: set s_alloc and r_alloc AND all these ITERATIONS->s_offs,r_offs,...
 //---------------------------------------------------------------------------------------------------