samples: handle corner cases (#25604)

Ticket 145028
openvinotoolkit · Jul 17, 2024 · 054c335 · 054c335
1 parent da58f14
commit 054c335
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/samples/c/common/opencv_c_wrapper/src/bmp_reader.c b/samples/c/common/opencv_c_wrapper/src/bmp_reader.c
@@ -83,7 +83,7 @@ int readBmpImage(const char* fileName, BitMap* image) {
     int padSize = image->width & 3;
     size_t row_size = (size_t)image->width * 3;
     char pad[3];
-    size_t size = row_size * image->height;
+    size_t size = row_size * (size_t)image->height;
 
     image->data = malloc(sizeof(char) * size);
     if (NULL == image->data) {

diff --git a/samples/cpp/common/utils/include/samples/common.hpp b/samples/cpp/common/utils/include/samples/common.hpp
@@ -255,6 +255,12 @@ static UNUSED void addRectangles(unsigned char* data,
                                  std::vector<int> rectangles,
                                  std::vector<int> classes,
                                  int thickness) {
+    if (height <= 0) {
+        throw std::runtime_error("height must be greater than 0");
+    }
+    if (width <= 0) {
+        throw std::runtime_error("width must be greater than 0");
+    }
     struct Color {
         unsigned char red;
         unsigned char green;