fix for issue 237

IntelLabs · Oct 28, 2024 · 2905930 · 2905930
1 parent de38f03
commit 2905930
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/remote_function/udf_server.py b/remote_function/udf_server.py
@@ -56,6 +56,8 @@ def image_api():
         return_string += r_meta.encode("utf-8")
 
     os.remove(tmpfile)
+    if return_string == "" or return_string == None:
+        return "error"
     return return_string
 
 

diff --git a/src/ImageLoop.cc b/src/ImageLoop.cc
@@ -330,11 +330,15 @@ void ImageLoop::execute_remote_operations(
         continue;
       }
       cv::Mat dmat = write_image(responseBuffer[rindex]);
+      if (dmat.rows == 0 || dmat.cols == 0){
+        throw VCLException(ObjectEmpty,
+                                 "Invalid response from the remote server.");
+      }
       if (dmat.empty()) {
         pendingImages.push_back(img);
       }
 
-      img->shallow_copy_cv(dmat);
+      img->shallow_copy_cv(dmat);      
       img->update_op_completed();
 
       auto const result = imageMap.insert(